#!/bin/bash
# halt on any error for safety and proper pipe handling
set -euo pipefail ; # <- this semicolon and comment make options apply
# even when script is corrupt by CRLF line terminators (issue #75)
# empty line must follow this comment for immediate fail with CRLF newlines
backup_path = "/opt/nvidia/libnvidia-encode-backup"
silent_flag = ''
print_usage( ) { printf '
SYNOPSIS
patch.sh [ -s] [ -r| -h| -c VERSION| -l]
DESCRIPTION
The patch for Nvidia drivers to remove NVENC session limit
-s Silent mode ( No output)
-r Rollback to original ( Restore lib from backup)
-h Print this help message
-c VERSION Check if version VERSION supported by this patch.
Returns true exit code ( 0) if version is supported.
-l List supported driver versions
'
}
# shellcheck disable=SC2209
opmode = "patch"
while getopts 'rshc:l' flag; do
case " ${ flag } " in
r) opmode = " ${ opmode } rollback " ; ;
s) silent_flag = 'true' ; ;
h) opmode = " ${ opmode } help " ; ;
c) opmode = " ${ opmode } checkversion " ; checked_version = " $OPTARG " ; ;
l) opmode = " ${ opmode } listversions " ; ;
*) echo "Incorrect option specified in command line" ; exit 2 ; ;
esac
done
if [ [ $silent_flag ] ] ; then
exec 1> /dev/null
fi
declare -A patch_list = (
[ "375.39" ] = 's/\x85\xC0\x89\xC5\x75\x18/\x29\xC0\x89\xC5\x90\x90/g'
[ "390.77" ] = 's/\x85\xC0\x89\xC5\x75\x18/\x29\xC0\x89\xC5\x90\x90/g'
[ "390.87" ] = 's/\x85\xC0\x89\xC5\x75\x18/\x29\xC0\x89\xC5\x90\x90/g'
[ "396.24" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "396.26" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "396.37" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g' #added info from https://github.com/keylase/nvidia-patch/issues/6#issuecomment-406895356
# break nvenc.c:236,layout asm,step-mode,step,break *0x00007fff89f9ba45
# libnvidia-encode.so @ 0x15a45; test->sub, jne->nop-nop-nop-nop-nop-nop
[ "396.54" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "410.48" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "410.57" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "410.73" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "410.78" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "410.79" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "410.93" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "410.104" ] = 's/\x85\xC0\x89\xC5\x0F\x85\x96\x00\x00\x00/\x29\xC0\x89\xC5\x90\x90\x90\x90\x90\x90/g'
[ "415.18" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x40\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "415.25" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x40\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "415.27" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x40\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "418.30" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x40\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "418.43" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x40\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "418.56" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x40\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "418.67" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x40\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "418.74" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "418.87.00" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "418.87.01" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "418.88" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "430.09" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "430.14" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "430.26" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "430.34" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "430.40" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "430.50" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "435.17" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "435.21" ] = 's/\x00\x00\x00\x84\xc0\x0f\x84\x0f\xfd\xff\xff/\x00\x00\x00\x84\xc0\x90\x90\x90\x90\x90\x90/g'
[ "440.26" ] = 's/\xff\x90\x98\x00\x00\x00\x84\xc0\x0f\x84/\xff\x90\x98\x00\x00\x00\x0c\x01\x0f\x84/g'
[ "440.31" ] = 's/\xff\x90\x98\x00\x00\x00\x84\xc0\x0f\x84/\xff\x90\x98\x00\x00\x00\x0c\x01\x0f\x84/g'
[ "440.33.01" ] = 's/\xff\x90\x98\x00\x00\x00\x84\xc0\x0f\x84/\xff\x90\x98\x00\x00\x00\x0c\x01\x0f\x84/g'
[ "440.36" ] = 's/\x85\xc0\x41\x89\xc4\x75\x1f/\x31\xc0\x41\x89\xc4\x75\x1f/g'
)
declare -A object_list = (
[ "375.39" ] = 'libnvidia-encode.so'
[ "390.77" ] = 'libnvidia-encode.so'
[ "390.87" ] = 'libnvidia-encode.so'
[ "396.24" ] = 'libnvidia-encode.so'
[ "396.26" ] = 'libnvidia-encode.so'
[ "396.37" ] = 'libnvidia-encode.so'
[ "396.54" ] = 'libnvidia-encode.so'
[ "410.48" ] = 'libnvidia-encode.so'
[ "410.57" ] = 'libnvidia-encode.so'
[ "410.73" ] = 'libnvidia-encode.so'
[ "410.78" ] = 'libnvidia-encode.so'
[ "410.79" ] = 'libnvidia-encode.so'
[ "410.93" ] = 'libnvidia-encode.so'
[ "410.104" ] = 'libnvidia-encode.so'
[ "415.18" ] = 'libnvcuvid.so'
[ "415.25" ] = 'libnvcuvid.so'
[ "415.27" ] = 'libnvcuvid.so'
[ "418.30" ] = 'libnvcuvid.so'
[ "418.43" ] = 'libnvcuvid.so'
[ "418.56" ] = 'libnvcuvid.so'
[ "418.67" ] = 'libnvcuvid.so'
[ "418.74" ] = 'libnvcuvid.so'
[ "418.87.00" ] = 'libnvcuvid.so'
[ "418.87.01" ] = 'libnvcuvid.so'
[ "418.88" ] = 'libnvcuvid.so'
[ "430.09" ] = 'libnvcuvid.so'
[ "430.14" ] = 'libnvcuvid.so'
[ "430.26" ] = 'libnvcuvid.so'
[ "430.34" ] = 'libnvcuvid.so'
[ "430.40" ] = 'libnvcuvid.so'
[ "430.50" ] = 'libnvcuvid.so'
[ "435.17" ] = 'libnvcuvid.so'
[ "435.21" ] = 'libnvcuvid.so'
[ "440.26" ] = 'libnvcuvid.so'
[ "440.31" ] = 'libnvcuvid.so'
[ "440.33.01" ] = 'libnvcuvid.so'
[ "440.36" ] = 'libnvidia-encode.so'
)
check_version_supported ( ) {
local ver = " $1 "
[ [ " ${ patch_list [ $ver ]+isset } " && " ${ object_list [ $ver ]+isset } " ] ]
}
get_supported_versions ( ) {
for drv in " ${ !patch_list[@] } " ; do
[ [ " ${ object_list [ $drv ]+isset } " ] ] && echo " $drv "
done | sort -t. -n
return 0
}
patch_common ( ) {
NVIDIA_SMI = " $( command -v nvidia-smi || true ) "
if [ [ ! " $NVIDIA_SMI " ] ] ; then
echo 'nvidia-smi utility not found. Probably driver is not installed.'
exit 1
fi
if ! driver_version = $( " $NVIDIA_SMI " --query-gpu= driver_version --format= csv,noheader,nounits | head -n 1) ; then
echo 'Something went wrong. Check nvidia driver'
exit 1
fi
echo " Detected nvidia driver version: $driver_version "
if ! check_version_supported " $driver_version " ; then
echo " Patch for this ( $driver_version ) nvidia driver not found. "
echo "Patch is available for versions: "
get_supported_versions
exit 1
fi
patch = " ${ patch_list [ $driver_version ] } "
object = " ${ object_list [ $driver_version ] } "
declare -a driver_locations = (
'/usr/lib/x86_64-linux-gnu'
'/usr/lib/x86_64-linux-gnu/nvidia/current/'
'/usr/lib64'
" /usr/lib/nvidia- ${ driver_version %%.* } "
)
dir_found = ''
for driver_dir in " ${ driver_locations [@] } " ; do
if [ [ -e " $driver_dir / $object . $driver_version " ] ] ; then
dir_found = 'true'
break
fi
done
[ [ " $dir_found " ] ] || { echo "ERROR: cannot detect driver directory" ; exit 1; }
}
rollback ( ) {
patch_common
if [ [ -f " $backup_path / $object . $driver_version " ] ] ; then
cp -p " $backup_path / $object . $driver_version " \
" $driver_dir / $object . $driver_version "
echo " Restore from backup $object . $driver_version "
else
echo "Backup not found. Try to patch first."
exit 1
fi
}
patch ( ) {
patch_common
if [ [ -f " $backup_path / $object . $driver_version " ] ] ; then
bkp_hash = " $( sha1sum " $backup_path / $object . $driver_version " | cut -f1 -d\ ) "
drv_hash = " $( sha1sum " $driver_dir / $object . $driver_version " | cut -f1 -d\ ) "
if [ [ " $bkp_hash " != " $drv_hash " ] ] ; then
echo "Backup exists and driver file differ from backup. Skipping patch."
return 0
fi
else
echo " Attention! Backup not found. Copying current $object to backup. "
mkdir -p " $backup_path "
cp -p " $driver_dir / $object . $driver_version " \
" $backup_path / $object . $driver_version "
fi
sha1sum " $backup_path / $object . $driver_version "
sed " $patch " " $backup_path / $object . $driver_version " > \
" ${ PATCH_OUTPUT_DIR - $driver_dir } / $object . $driver_version "
sha1sum " ${ PATCH_OUTPUT_DIR - $driver_dir } / $object . $driver_version "
ldconfig
echo "Patched!"
}
query_version_support ( ) {
if check_version_supported " $checked_version " ; then
echo "SUPPORTED"
exit 0
else
echo "NOT SUPPORTED"
exit 1
fi
}
list_supported_versions ( ) {
get_supported_versions
}
case " ${ opmode } " in
patch) patch ; ;
patchrollback) rollback ; ;
patchhelp) print_usage ; exit 2 ; ;
patchcheckversion) query_version_support ; ;
patchlistversions) list_supported_versions ; ;
*) echo "Incorrect combination of flags. Use option -h to get help."
exit 2 ; ;
esac