export RUNUTILS="$RUNDIR/utils"

export RUNSTATIC="$RUNDIR/static"

[ "$RIM_SYS_TOOLS" == 1 ] && \

export PATH="$SYS_PATH:/bin:/sbin:/usr/bin:/usr/sbin:$RUNSTATIC:$RUNUTILS"||\

export PATH="$RUNSTATIC:$RUNUTILS:$SYS_PATH:/bin:/sbin:/usr/bin:/usr/sbin"

if [ -f "$URUNTIME" ]

then export RUNIMAGE="$URUNTIME" && return

fi

local rpth="$(realpath "$1" 2>/dev/null)"

local wrpth="$(realpath "$(which_exe "$1")" 2>/dev/null)"

if is_exe "$rpth"

then export RUNIMAGE="$rpth"

elif is_exe "$wrpth"

then export RUNIMAGE="$wrpth"

else [ -n "$2" ] && \

export RUNIMAGE="$2"||\

export RUNIMAGE="$1"

fi

local rspth="$(realpath -s "$1" 2>/dev/null)"

local wrspth="$(realpath -s "$(which_exe "$1")" 2>/dev/null)"

if is_exe "$rspth"

then export RUNSRC="$rspth"

elif is_exe "$wrspth"

then export RUNSRC="$wrspth"

else [ -n "$2" ] && \

export RUNSRC="$2"||\

export RUNSRC="$1"

fi

if [ -d "$URUNTIME_DIR" ]

then export RUNIMAGEDIR="$URUNTIME_DIR"

else export RUNIMAGEDIR="$(dirname "$RUNIMAGE" 2>/dev/null)"

fi

export RUNROOTFS_VERSION="$(cat "$RUNROOTFS/.version" \

export RUNROOTFSTYPE="$(cat "$RUNROOTFS/.type" 2>/dev/null)"

echo -e "${RED}[ ERROR ][$(date +"%Y.%m.%d %T")]: $@ $RESETCOLOR" 1>&2

if [ "$NOT_TERM" == 1 ]

then notify-send -a 'RunImage Error' "$(echo -e "$@"|nocolor)" 2>/dev/null &

fi

if [ "$RIM_QUIET_MODE" != 1 ]

then echo -e "${GREEN}[ INFO ][$(date +"%Y.%m.%d %T")]: $@ $RESETCOLOR" 1>&2

if [[ "$NOT_TERM" == 1 && "$RIM_NOTIFY" == 1 ]]

then notify-send -a 'RunImage Info' "$(echo -e "$@"|nocolor)" 2>/dev/null &

fi

fi

if [[ "$RIM_QUIET_MODE" != 1 && "$RIM_NO_WARN" != 1 ]]

then echo -e "${YELLOW}[ WARNING ][$(date +"%Y.%m.%d %T")]: $@ $RESETCOLOR" 1>&2

if [[ "$NOT_TERM" == 1 && "$RIM_NOTIFY" == 1 ]]

then notify-send -a 'RunImage Warning' "$(echo -e "$@"|nocolor)" 2>/dev/null &

fi

fi

time_out=1000 # =~ 3 sec

wait_time=0

while true

do

if [ "$wait_time" -le "$time_out" ]

then

if is_pid "$1" && [ -n "$(ls -A "$2" 2>/dev/null)" ]

then

return 0

else

wait_time="$(( $wait_time + 1 ))"

sleep 0.0001 2>/dev/null

fi

else

return 1

fi

done

while true

do

read -p "$(echo -e "${RED}$1 ${GREEN}(y/n) ${BLUE}> $RESETCOLOR")" yn

case $yn in

[Yy] ) return 0 ;;

[Nn] ) return 1 ;;

esac

done

[ ! -n "$1" ] && \

return 1

if [ -n "$2" ]

then [ "$(check_url_stat_code "$1")" == "$2" ]

else [ "$(check_url_stat_code "$1")" == "200" ]

fi

err_no_downloader() {

error_msg "Downloader not found!"

cleanup force

exit 1

}

rm_fail_dl() {

[ -f "$FILEDIR/$FILENAME" ] && \

rm -rf "$FILEDIR/$FILENAME" \

"$FILEDIR/$FILENAME"*.aria2

}

dl_ret() {

if [ "$1" != 0 ]

then

rm_fail_dl

dl_repeat && \

try_dl "$URL" "$FILEDIR/$FILENAME"||\

return 1

else return 0

fi

}

dl_repeat() {

[ "$NO_DL_REPEAT" == 1 ] && \

return 1

DL_REP_TITLE="Download interrupted!"

DL_REP_TEXT="Failed to download: $FILENAME from $(echo "$URL"|gawk -F/ '{print$3"/"$4}') \nWould you like to repeat it?"

if [[ "$NOT_TERM" != 1 || "$NO_DL_GUI" == 1 ]]

then

yn_case "$DL_REP_TEXT"||return 1

elif is_exe_exist yad

then

yad --image="dialog-error" --button="CANCEL:1" --center \

--button="REPEAT:0" --title="$DL_REP_TITLE" \

--text="$DL_REP_TEXT" --on-top --fixed

elif is_exe_exist zenity

then

zenity --question --title="$DL_REP_TITLE" --no-wrap \

--text="$DL_REP_TEXT"

else return 1

fi

}

if [ -n "$1" ]

then

URL="$1"

if [ -n "$2" ]

then

if [ -d "$2" ]

then

FILEDIR="$2"

FILENAME="$(basename "$1")"

else

FILEDIR="$(dirname "$2")"

FILENAME="$(basename "$2")"

fi

else

FILEDIR="."

FILENAME="$(basename "$1")"

fi

if is_url "$URL"

then

WGET_ARGS=(--no-check-certificate -t 3 -T 5 -w 0.5 "$URL" -O "$FILEDIR/$FILENAME")

[ ! -d "$FILEDIR" ] && \

try_mkdir "$FILEDIR"

if [[ "$NOT_TERM" == 1 && "$NO_DL_GUI" != 1 ]] && \

(is_exe_exist yad||is_exe_exist zenity)

then

set -o pipefail

dl_progress() {

[[ "$URL" =~ '&key=' ]] && \

local URL="$(echo "$URL"|sed "s|\&key=.*||g")"

[[ "$URL" =~ '&' && ! "$URL" =~ '&' ]] && \

local URL="$(echo "$URL"|sed "s|\&|\&|g")"

if is_exe_exist yad

then

yad --progress --percentage=0 --text="Download:\t$FILENAME\n$URL" \

--auto-close --no-escape --selectable-labels --auto-kill \

--center --on-top --fixed --no-buttons --undecorated --skip-taskbar

elif is_exe_exist zenity

then

zenity --progress --text="Connecting to $URL" --width=650 --height=40 \

--auto-close --no-cancel --title="Download: $FILENAME"

else return 1

fi

}

dl_progress_pulsate() {

local ret=1

[[ "$URL" =~ '&key=' ]] && \

local URL="$(echo "$URL"|sed "s|\&key=.*||g")"

[[ "$URL" =~ '&' && ! "$URL" =~ '&' ]] && \

local URL="$(echo "$URL"|sed "s|\&|\&|g")"

if is_exe_exist yad

then

local yad_args=(

--progress --pulsate --text="Download:\t$FILENAME\n$URL"

--width=650 --height=40 --undecorated --skip-taskbar

--no-buttons --text-align center --auto-close --auto-kill

--center --fixed --on-top --no-escape --selectable-labels

)

"$@" &

local exec_pid="$!"

if is_pid "$exec_pid"

then

(while is_pid "$exec_pid"

do echo -e "#\n" ; sleep 0.1 2>/dev/null

done)|yad "${yad_args[@]}" &>/dev/null &

local yad_pid="$!"

wait "$exec_pid" &>/dev/null

ret="$?"

kill "$yad_pid" &>/dev/null

fi

elif is_exe_exist zenity

then

"$@"|zenity --progress --pulsate --text="$URL" --width=650 --height=40 \

--auto-close --no-cancel --title="Download: $FILENAME"

ret="$?"

fi

return $ret

}

if [ "$NO_ARIA2C" != 1 ] && \

is_exe_exist aria2c

then

aria2c --no-conf -R -x 13 -s 13 --allow-overwrite --summary-interval=1 -o \

"$FILENAME" -d "$FILEDIR" "$URL"|grep --line-buffered 'ETA'|\

sed -u 's|(.*)| &|g;s|(||g;s|)||g;s|\[||g;s|\]||g'|\

gawk '{print$3"\n#Downloading at "$3,$2,$5,$6;system("")}'|\

dl_progress

elif is_exe_exist curl

then

curl -R --progress-bar --insecure --fail -L "$URL" -o \

"$FILEDIR/$FILENAME" |& tr '\r' '\n'|sed '0,/100/{/100/d;}'|\

sed -ur 's|[# ]+||g;s|.*=.*||g;s|.*|#Downloading at &\n&|g'|\

dl_progress

elif is_exe_exist wget2

then

dl_progress_pulsate wget2 "${WGET_ARGS[@]}"

elif is_exe_exist wget

then

wget "${WGET_ARGS[@]}"|& tr '\r' '\n'|\

sed -u 's/.* \([0-9]\+%\)\ \+\([0-9,.]\+.\) \(.*\)/\1\n#Downloading at \1\ ETA: \3/; s/^20[0-9][0-9].*/#Done./'|\

dl_progress

else

err_no_downloader

fi

dl_ret "${PIPESTATUS[0]}"||return 1

else

if [ "$NO_ARIA2C" != 1 ] && is_exe_exist aria2c

then

aria2c --no-conf -R -x 13 -s 13 --allow-overwrite -d "$FILEDIR" -o "$FILENAME" "$URL"

elif is_exe_exist curl

then

curl -R --progress-bar --insecure --fail -L "$URL" -o "$FILEDIR/$FILENAME"

elif is_exe_exist wget2

then

wget2 -q --force-progress "${WGET_ARGS[@]}"

elif is_exe_exist wget

then

wget -q --show-progress "${WGET_ARGS[@]}"

else

err_no_downloader

fi

dl_ret "$?"||return 1

fi

else

error_msg "$FILENAME not found in $(echo "$URL"|gawk -F/ '{print$3"/"$4}')"

return 1

fi

else

error_msg "Specify download URL!"

return 1

fi

(local ret=1

unset rmnvsrc

if [[ -n "$1" || -n "$nvidia_version" ]]

then

[ ! -n "$nvidia_version" ] && \

nvidia_version="$1"

[ ! -n "$NVDRVARCH" ] && \

NVDRVARCH="$(uname -m)"

[[ -d "$2" && ! -n "$NVIDIA_DRIVERS_DIR" ]] && \

export NVIDIA_DRIVERS_DIR="$2"

[[ ! -d "$2" && ! -n "$NVIDIA_DRIVERS_DIR" ]] && \

export NVIDIA_DRIVERS_DIR="."

[ ! -n "$nvidia_driver_image" ] && \

nvidia_driver_image="$nvidia_version.nv.drv"

try_mkdir "$NVIDIA_DRIVERS_DIR"

NVBINS=(

'mkprecompiled' 'nvidia-cuda-mps-control' 'nvidia-cuda-mps-server'

'nvidia-debugdump' 'nvidia-installer' 'nvidia-modprobe'

'nvidia-ngx-updater' 'tls_test' 'nvidia-persistenced' 'nvidia-powerd'

'nvidia-settings' 'nvidia-smi' 'nvidia-xconfig' 'nvidia-pcc'

'nvidia-cuda-mps-srv' 'nvidia-bug-report.sh' 'nvidia-sleep.sh'

)

if [[ "$RIM_SYS_NVLIBS" == 1 || "$NVDRVARCH" != 'x86_64' ]]

then

info_msg "Find Nvidia ${nvidia_version} local libs, please wait..."

cp_nvfiles() (

local sys_pth sys_pths

local dir="$1"

shift

mkdir -p "$dir" && \

cd "$dir"||return 1

for item in "$@"

do

OLD_IFS="$IFS"

IFS=$'\n'

case "$dir" in

64|32) sys_pths="$(realpath "$item" 2>/dev/null)" ;;

bin) sys_pths="$(command -v "$item" 2>/dev/null)" ;;

wine) sys_pths=($(find /usr -type f -name "$item" 2>/dev/null)) ;;

.) sys_pths=($(ls "$item" 2>/dev/null)) ;;

*) sys_pths=($(find /etc/ /usr/share -name "*${item}" -type f 2>/dev/null)) ;;

esac

IFS="$OLD_IFS"

if [ -n "$sys_pths" ]

then

for sys_pth in "${sys_pths[@]}"

do

local file="$(basename "$sys_pth")"

if [[ ! -e "$file" && -e "$sys_pth" ]]

then cp -f "$sys_pth" "$file"

fi

done

fi

done

)

[ "$NVDRVARCH" != 'x86_64' ] && \

RIM_NO_32BIT_NVLIBS_CHECK=1

NVTRASH_LIBS=('libnvidia-container*')

NVCONFS=('nvidia*.conf' 'nvidia.icd')

NVJSONS=('nvidia*.json')

NVPROFS=(

"nvidia-application-profiles-${nvidia_version}-*"

'nvoptix.bin'

)

NVWINELS=('*nvngx*.dll')

NVLICENSES=(

'/usr/share/licenses/nvidia-utils/LICENSE'

/usr/share/doc/nvidia-driver-*/LICENSE

)

NVLIBS="$(if is_exe_exist ldconfig

for lib in "${NVTRASH_LIBS[@]}"

do NVLIBS="$(grep -v "$lib"<<<"$NVLIBS")"

done

NVLIBS64=($(grep -E "/lib/|/${NVDRVARCH}-linux-gnu/"<<<"$NVLIBS"))

for pth in lib "lib/${NVDRVARCH}-linux-gnu"

do

libs=(

"/usr/$pth/vdpau/libvdpau_nvidia.so.1"

"/usr/$pth/xorg/modules/drivers/nvidia_drv.so"

"/usr/$pth/nvidia/xorg/libglxserver_nvidia.so"

"/usr/$pth/nvidia/libglxserver_nvidia.so"

)

for lib in "${libs[@]}"

do [ -e "$lib" ] && NVLIBS64+=("$lib")

done

done

if [ ! -n "$NVLIBS64" ]

then

error_msg "Nvidia libraries are not found in your system!"

if [[ "$NVLIBS_DLFAILED" == 1 || "$NVDRVARCH" != 'x86_64' ]]

then return 1

else

RIM_SYS_NVLIBS=0 get_nvidia_driver_image

return $?

fi

fi

NVLIBS32=($(grep -E '/lib32/|/i386-linux-gnu/'<<<"$NVLIBS"))

for pth in lib32 i386-linux-gnu

do

nvvdpau="/usr/$pth/vdpau/libvdpau_nvidia.so"

[ -e "$nvvdpau" ] && NVLIBS32+=("$nvvdpau")

done

if [[ ! -n "$NVLIBS32" && "$RIM_NO_32BIT_NVLIBS_CHECK" != 1 ]]

then

error_msg "Nvidia 32-bit libraries are not found in your system!"

info_msg "Use ${YELLOW}RIM_NO_32BIT_NVLIBS_CHECK=1 ${GREEN}if they are not required."

if [ "$NVLIBS_DLFAILED" == 1 ]

then return 1

else

RIM_SYS_NVLIBS=0 get_nvidia_driver_image

return $?

fi

fi

info_msg "Creating a driver directory structure..."

(cd "$NVIDIA_DRIVERS_DIR" && \

mkdir -p "$nvidia_version" && \

cd "$nvidia_version"

cp_nvfiles 32 "${NVLIBS32[@]}"

cp_nvfiles 64 "${NVLIBS64[@]}"

cp_nvfiles bin "${NVBINS[@]}"

cp_nvfiles conf "${NVCONFS[@]}"

cp_nvfiles json "${NVJSONS[@]}"

if [[ ! -e 'profiles' ]]

then

if [ -d '/usr/share/nvidia' ]

then cp -TLrf '/usr/share/nvidia' 'profiles'

else cp_nvfiles profiles "${NVPROFS[@]}"

fi

fi

if [ ! -e 'wine' ]

then

if [ -d '/usr/lib/nvidia/wine' ]

then cp -TLrf '/usr/lib/nvidia/wine' 'wine'

else cp_nvfiles wine "${NVWINELS[@]}"

fi

fi

cp_nvfiles . "${NVLICENSES[@]}")

else

info_msg "Downloading Nvidia ${nvidia_version} driver, please wait..."

nvidia_driver_run="NVIDIA-Linux-x86_64-${nvidia_version}.run"

driver_url_list=(

"https://storage.yandexcloud.net/runimage/nvidia-drivers/$nvidia_driver_image"

"https://huggingface.co/runimage/nvidia-drivers/resolve/main/releases/$nvidia_driver_image"

"https://github.com/VHSgunzo/runimage-nvidia-drivers/releases/download/v${nvidia_version}/$nvidia_driver_image"

"https://download.nvidia.com/XFree86/Linux-x86_64/${nvidia_version}/$nvidia_driver_run"

"https://us.download.nvidia.com/tesla/${nvidia_version}/$nvidia_driver_run"

"https://developer.nvidia.com/downloads/vulkan-beta-${nvidia_version//.}-linux"

"https://developer.nvidia.com/vulkan-beta-${nvidia_version//.}-linux"

"https://developer.nvidia.com/linux-${nvidia_version//.}"

)

if try_dl "${driver_url_list[0]}" "$NVIDIA_DRIVERS_DIR"||\

try_dl "${driver_url_list[1]}" "$NVIDIA_DRIVERS_DIR"||\

try_dl "${driver_url_list[2]}" "$NVIDIA_DRIVERS_DIR"

then return 0

elif try_dl "${driver_url_list[3]}" "$NVIDIA_DRIVERS_DIR"||\

try_dl "${driver_url_list[4]}" "$NVIDIA_DRIVERS_DIR"||\

try_dl "${driver_url_list[5]}" "$NVIDIA_DRIVERS_DIR" "$nvidia_driver_run"||\

try_dl "${driver_url_list[6]}" "$NVIDIA_DRIVERS_DIR" "$nvidia_driver_run"||\

try_dl "${driver_url_list[7]}" "$NVIDIA_DRIVERS_DIR" "$nvidia_driver_run"

then

trash_libs="libEGL.so* libGLdispatch.so* *.swidtag *.la \

chmod u+x "$NVIDIA_DRIVERS_DIR/$nvidia_driver_run"

info_msg "Unpacking $nvidia_driver_run..."

(cd "$NVIDIA_DRIVERS_DIR" && \

"./$nvidia_driver_run" --target "$nvidia_version" -x &>/dev/null

rm -f "$nvidia_driver_run")

info_msg "Creating a driver directory structure..."

(cd "$NVIDIA_DRIVERS_DIR/$nvidia_version" && \

rm -rf html kernel* libglvnd_install_checker 32/libglvnd_install_checker \

supported-gpus systemd *.gz *.bz2 *.txt .manifest *.desktop *.png firmware *.h

for temp in $(ls *.template 2>/dev/null) ; do mv "$temp" "${temp%.template}" ; done

try_mkdir profiles && mv *application-profiles* profiles

[ -f "nvoptix.bin" ] && mv nvoptix.bin profiles

[ -n "$(ls *.dll 2>/dev/null)" ] && try_mkdir wine && mv *.dll wine

try_mkdir json && mv *.json json

try_mkdir conf && mv *.conf *.icd conf

for lib in $trash_libs ; do rm -f $lib 32/$lib ; done

try_mkdir bin && mv *.sh bin

for binary in "${NVBINS[@]}" ; do [ -f "$binary" ] && mv $binary bin ; done

try_mkdir 64 && mv *.so* 64

[ -d "tls" ] && mv tls/* 64 && rm -rf tls

[ -d "32/tls" ] && mv 32/tls/* 32 && rm -rf 32/tls)

else

error_msg "Failed to download nvidia driver!"

if [ "$NVLIBS_SYSFAILED" != 1 ]

then

RIM_SYS_NVLIBS=1 NVLIBS_DLFAILED=1 get_nvidia_driver_image

return $?

fi

return 1

fi

fi

if [ -e "$NVIDIA_DRIVERS_DIR/$nvidia_version/64/libGLX_nvidia.so.$nvidia_version" ]

then

info_msg "Creating a SquashFS driver image..."

info_msg "$NVIDIA_DRIVERS_DIR/$nvidia_driver_image"

echo -en "$BLUE"

if "$MKSQFS" "$NVIDIA_DRIVERS_DIR/$nvidia_version" "$NVIDIA_DRIVERS_DIR/$nvidia_driver_image" \

-root-owned -no-xattrs -noappend -b 1M -comp zstd -Xcompression-level 1 -quiet

then ret=0

else error_msg "Failed to create Nvidia driver image!"

fi

echo -en "$RESETCOLOR"

else

error_msg "libGLX_nvidia.so.$nvidia_version not found in the source directory of the driver!"

rmnvsrc=1

fi

if [ -d "$NVIDIA_DRIVERS_DIR/$nvidia_version" ] && [[ "$ret" == 0 || "$rmnvsrc" == 1 ]]

then

info_msg "Deleting the source directory of the driver..."

rm -rf "$NVIDIA_DRIVERS_DIR/$nvidia_version"

fi

if [[ "$RIM_SYS_NVLIBS" == 1 && "$rmnvsrc" == 1 ]]

then

RIM_SYS_NVLIBS=0 NVLIBS_SYSFAILED=1 get_nvidia_driver_image

return $?

fi

else

error_msg "You must specify the nvidia driver version!"

fi

return $ret)

if [[ -n "$1" && -n "$(echo "$1"|grep -o "\.nv\.drv$")" ]]

then

[ ! -n "$nvidia_version" ] && \

nvidia_version="$(echo "$1"|sed 's|.nv.drv||g')"

[ ! -n "$NVDRVMNT" ] && \

NVDRVMNT="$RUNPIDDIR/mnt/nv${nvidia_version}drv"

info_msg "Mounting the nvidia driver image: $(basename "$1")"

try_mkdir "$NVDRVMNT"

"$SQFUSE" -f -o ro,nodev,uid=$EUID,gid=$EGID \

"$1" "$NVDRVMNT" &>/dev/null &

FUSE_PID="$!"

FUSE_PIDS="$FUSE_PID $FUSE_PIDS"

if mount_exist "$FUSE_PID" "$NVDRVMNT"

then

nvidia_driver_dir="$NVDRVMNT"

else

error_msg "Failed to mount the nvidia driver image!"

rm -f "$1" 2>/dev/null

return 1

fi

else

error_msg "You must specify the nvidia driver image!"

return 1

fi

unset NVIDIA_DRIVER_BIND NVLIBS_DLFAILED NVLIBS_SYSFAILED

is_inside_ver_eq() { [ "$(cat "$RUNROOTFS/etc/ld.so.version" 2>/dev/null)" == "$RUNROOTFS_VERSION-$nvidia_version" ] ; }

print_nv_drv_dir() { info_msg "Found nvidia driver directory: $(basename "$nvidia_driver_dir")" ; }

update_ld_cache() {

if [[ "$(cat "$RUNCACHEDIR/ld.so.version" 2>/dev/null)" != "$RUNROOTFS_VERSION-$nvidia_version" ]]||\

([ -f "$RUNROOTFS/etc/ld.so.version" ] && ! is_inside_ver_eq)||\

([ -w "$RUNROOTFS" ] && ! is_inside_ver_eq)

then

info_msg "Updating the nvidia library cache..."

if (RIM_SANDBOX_NET=0 RIM_NO_NET=0 RIM_WAIT_RPIDS_EXIT=0 \

RIM_NO_BWRAP_OVERLAY=1 SSRV_SOCK="unix:$RUNPIDDIR/ldupd" \

bwrun /usr/bin/ldconfig -C "$RUNPIDDIR/ld.so.cache" 2>/dev/null </dev/null)

then

try_mkdir "$RUNCACHEDIR"

if mv -f "$RUNPIDDIR/ld.so.cache" \

"$RUNCACHEDIR/ld.so.cache" 2>/dev/null

then

echo "$RUNROOTFS_VERSION-$nvidia_version" > \

"$RUNCACHEDIR/ld.so.version"

if [ -w "$RUNROOTFS" ]

then

cp -f "$RUNCACHEDIR/ld.so.cache" \

"$RUNROOTFS/etc/ld.so.cache" 2>/dev/null

echo "$RUNROOTFS_VERSION-$nvidia_version" > \

"$RUNROOTFS/etc/ld.so.version"

fi

else

error_msg "Failed to merge nvidia library cache!"

return 1

fi

else

error_msg "Failed to update nvidia library cache!"

return 1

fi

fi

}

NVDRVARCH="$(uname -m)"

if [ -e '/sys/module/nvidia/version' ]||\

grep -owm1 nvidia /proc/modules &>/dev/null

then

unset NVDRVMNT nvidia_driver_dir

[[ -n "$RIM_NVIDIA_DRIVERS_DIR" && ! -d "$RIM_NVIDIA_DRIVERS_DIR" ]] && \

try_mkdir "$RIM_NVIDIA_DRIVERS_DIR"

[ -d "$RIM_NVIDIA_DRIVERS_DIR" ] && \

NVIDIA_DRIVERS_DIR="$RIM_NVIDIA_DRIVERS_DIR"||\

NVIDIA_DRIVERS_DIR="$RUNIMAGEDIR/nvidia-drivers"

export NVIDIA_DRIVERS_DIR

if [ -e '/sys/module/nvidia/version' ]

then

nvidia_version="$(cat /sys/module/nvidia/version 2>/dev/null)"

elif modinfo nvidia &>/dev/null

then

nvidia_version="$(modinfo -F version nvidia 2>/dev/null)"

elif nvidia-smi &>/dev/null

then

nvidia_version="$(nvidia-smi --query-gpu=driver_version --format=csv,noheader|head -1)"

else

if [ -d "/usr/lib/${NVDRVARCH}-linux-gnu" ]

then

nvidia_version="$(basename "$(realpath /usr/lib/${NVDRVARCH}-linux-gnu/libGLX_nvidia.so.*|head -1)"|tail -c +18)"

else

nvidia_version="$(basename "$(realpath /usr/lib/libGLX_nvidia.so.*)"|tail -c +18)"

fi

fi

if [[ -n "$nvidia_version" && "$nvidia_version" != '*' ]]

then

nvidia_version_inside="$(basename "$(realpath "$RUNROOTFS/usr/lib/libGLX_nvidia.so".*|head -1)"|tail -c +18)"

if [ "$nvidia_version" != "$nvidia_version_inside" ]

then

if [[ -n "$nvidia_version_inside" && "$nvidia_version_inside" != '*' ]]

then

nvidia_driver_image="$nvidia_version.nv.drv"

NVDRVMNT="$RUNPIDDIR/mnt/nv${nvidia_version}drv"

[ "$nvidia_version_inside" != "000.00.00" ] && \

warn_msg "Nvidia driver version mismatch detected, trying to fix it"

if [ ! -f "$NVIDIA_DRIVERS_DIR/$nvidia_version/64/libGLX_nvidia.so.$nvidia_version" ] && \

[ ! -f "$RUNIMAGEDIR/$nvidia_driver_image" ] && \

[ ! -f "$NVIDIA_DRIVERS_DIR/$nvidia_driver_image" ] && \

[ ! -f "$NVDRVMNT/64/libGLX_nvidia.so.$nvidia_version" ] && \

[ ! -f "$RUNDIR/nvidia-drivers/$nvidia_version/64/libGLX_nvidia.so.$nvidia_version" ] && \

[ ! -f "$RUNDIR/nvidia-drivers/$nvidia_driver_image" ]

then

if RIM_NOTIFY=1 RIM_QUIET_MODE=0 get_nvidia_driver_image

then

mount_nvidia_driver_image "$NVIDIA_DRIVERS_DIR/$nvidia_driver_image"

else

nvidia_driver_dir="$NVIDIA_DRIVERS_DIR/$nvidia_version"

fi

else

if [ -f "$NVDRVMNT/64/libGLX_nvidia.so.$nvidia_version" ]

then

nvidia_driver_dir="$NVDRVMNT"

print_nv_drv_dir

elif [ -f "$NVIDIA_DRIVERS_DIR/$nvidia_version/64/libGLX_nvidia.so.$nvidia_version" ]

then

nvidia_driver_dir="$NVIDIA_DRIVERS_DIR/$nvidia_version"

print_nv_drv_dir

elif [ -f "$RUNIMAGEDIR/$nvidia_driver_image" ]

then

mount_nvidia_driver_image "$RUNIMAGEDIR/$nvidia_driver_image"

elif [ -f "$NVIDIA_DRIVERS_DIR/$nvidia_driver_image" ]

then

mount_nvidia_driver_image "$NVIDIA_DRIVERS_DIR/$nvidia_driver_image"

elif [ -f "$RUNDIR/nvidia-drivers/$nvidia_version/64/libGLX_nvidia.so.$nvidia_version" ]

then

nvidia_driver_dir="$RUNDIR/nvidia-drivers/$nvidia_version"

print_nv_drv_dir

elif [ -f "$RUNDIR/nvidia-drivers/$nvidia_driver_image" ]

then

mount_nvidia_driver_image "$RUNDIR/nvidia-drivers/$nvidia_driver_image"

fi

fi

else

error_msg "Nvidia driver not found in RunImage!"

return 1

fi

if [ -f "$nvidia_driver_dir/64/libGLX_nvidia.so.$nvidia_version" ]

then

nvidia_libs_list="libcuda.so libEGL_nvidia.so libGLESv1_CM_nvidia.so libnvidia-opencl.so \

for lib in ${nvidia_libs_list}

do

if [ -f "$RUNROOTFS/usr/lib/${lib}.${nvidia_version_inside}" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/64/${lib}.${nvidia_version}"

"/usr/lib/${lib}.${nvidia_version_inside}")

fi

if [ -f "$RUNROOTFS/usr/lib32/${lib}.${nvidia_version_inside}" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/32/${lib}.${nvidia_version}"

"/usr/lib32/${lib}.${nvidia_version_inside}")

fi

done

if [ -f "$RUNROOTFS/usr/lib/libnvidia-api.so.1" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/64/libnvidia-api.so.1"

"/usr/lib/libnvidia-api.so.1")

fi

if [ -f "$RUNROOTFS/usr/lib/xorg/modules/drivers/nvidia_drv.so" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/64/nvidia_drv.so"

"/usr/lib/xorg/modules/drivers/nvidia_drv.so")

fi

if [ -f "$RUNROOTFS/usr/lib/nvidia/xorg/libglxserver_nvidia.so.${nvidia_version_inside}" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/64/libglxserver_nvidia.so.${nvidia_version}"

"/usr/lib/nvidia/xorg/libglxserver_nvidia.so.${nvidia_version_inside}")

fi

if [ -f "$RUNROOTFS/usr/lib/vdpau/libvdpau_nvidia.so.${nvidia_version_inside}" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/64/libvdpau_nvidia.so.${nvidia_version}"

"/usr/lib/vdpau/libvdpau_nvidia.so.${nvidia_version_inside}")

fi

if [ -f "$RUNROOTFS/usr/lib32/vdpau/libvdpau_nvidia.so.${nvidia_version_inside}" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/32/libvdpau_nvidia.so.${nvidia_version}"

"/usr/lib32/vdpau/libvdpau_nvidia.so.${nvidia_version_inside}")

fi

if [ -f "$RUNROOTFS/usr/share/egl/egl_external_platform.d/15_nvidia_gbm.json" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/json/15_nvidia_gbm.json"

"/usr/share/egl/egl_external_platform.d/15_nvidia_gbm.json")

fi

if [ -f "$RUNROOTFS/usr/share/glvnd/egl_vendor.d/10_nvidia.json" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/json/10_nvidia.json"

"/usr/share/glvnd/egl_vendor.d/10_nvidia.json")

fi

if [ -f "$RUNROOTFS/usr/share/vulkan/icd.d/nvidia_icd.json" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/json/nvidia_icd.json"

"/usr/share/vulkan/icd.d/nvidia_icd.json")

fi

if [ -f "$RUNROOTFS/usr/share/vulkan/implicit_layer.d/nvidia_layers.json" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/json/nvidia_layers.json"

"/usr/share/vulkan/implicit_layer.d/nvidia_layers.json")

fi

if [ -f "$RUNROOTFS/usr/share/vulkansc/icd.d/nvidia_icd_vksc.json" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/json/nvidia_icd_vksc.json"

"/usr/share/vulkansc/icd.d/nvidia_icd_vksc.json")

fi

if [ -f "$RUNROOTFS/etc/OpenCL/vendors/nvidia.icd" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/conf/nvidia.icd"

"/etc/OpenCL/vendors/nvidia.icd")

fi

if [ -d "$RUNROOTFS/usr/share/nvidia" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/profiles"

"/usr/share/nvidia")

fi

if [ -d "$RUNROOTFS/usr/lib/nvidia/wine" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try"

"$nvidia_driver_dir/wine"

"/usr/lib/nvidia/wine")

fi

if [ -w "$RUNROOTFS" ]

then

if [[ ! -d "$RUNROOTFS/usr/bin/nvidia" || \

! -d "$RUNROOTFS/usr/lib/nvidia/64" || \

! -d "$RUNROOTFS/usr/lib/nvidia/32" ]]

then

mkdir -p "$RUNROOTFS/usr/bin/nvidia"

mkdir -p "$RUNROOTFS/usr/lib/nvidia/64"

mkdir -p "$RUNROOTFS/usr/lib/nvidia/32"

fi

if [ ! -f "$RUNROOTFS/etc/ld.so.conf.d/nvidia.conf" ]

then

mkdir -p "$RUNROOTFS/etc/ld.so.conf.d"

echo -e "/usr/lib/nvidia/64\n/usr/lib/nvidia/32" > \

"$RUNROOTFS/etc/ld.so.conf.d/nvidia.conf"

rm -f "$RUNCACHEDIR/ld.so."*

fi

fi

if [ -d "$RUNROOTFS/usr/bin/nvidia" ] && \

[ -d "$RUNROOTFS/usr/lib/nvidia/64" ] && \

[ -d "$RUNROOTFS/usr/lib/nvidia/32" ]

then

NVIDIA_DRIVER_BIND+=("--ro-bind-try" "$nvidia_driver_dir/bin" "/usr/bin/nvidia"

"--ro-bind-try" "$nvidia_driver_dir/64" "/usr/lib/nvidia/64"

"--ro-bind-try" "$nvidia_driver_dir/32" "/usr/lib/nvidia/32")

add_bin_pth '/usr/bin/nvidia'

add_lib_pth '/usr/lib/nvidia/64:/usr/lib/nvidia/32'

elif [ -d "$nvidia_driver_dir/bin" ] && \

[ -d "$nvidia_driver_dir/64" ] && \

[ -d "$nvidia_driver_dir/32" ]

then

add_bin_pth "$nvidia_driver_dir/bin"

add_lib_pth "$nvidia_driver_dir/64:$nvidia_driver_dir/32"

fi

if update_ld_cache && [ ! -w "$RUNROOTFS" ]

then BIND_LDSO_CACHE=1

fi

if [ "$RIM_UNSHARE_RUN" != 1 ]

then

NVXSOCKET="$(ls /run/nvidia-xdriver-* 2>/dev/null|head -1)"

if [ -S "$NVXSOCKET" ]

then XDG_RUN_BIND+=("--bind-try" "$NVXSOCKET" "$NVXSOCKET")

fi

fi

else

error_msg "Nvidia driver not found!"

return 1