Browse Source

Ajout de 'entrypoint.sh'

penta 3 days ago
parent
commit
66dc313818
1 changed files with 248 additions and 0 deletions
  1. 248 0
      entrypoint.sh

+ 248 - 0
entrypoint.sh

@@ -0,0 +1,248 @@
+#!/bin/bash
+
+# Adapted from docker-selkies-glx-desktop for an XFCE-only NVIDIA image.
+# This Source Code Form is subject to the Mozilla Public License, v. 2.0.
+
+set -Eeuo pipefail
+
+trap 'echo "Desktop entrypoint received a termination signal"' HUP INT QUIT TERM
+
+export DISPLAY="${DISPLAY:-:20}"
+export DISPLAY_SIZEW="${DISPLAY_SIZEW:-1920}"
+export DISPLAY_SIZEH="${DISPLAY_SIZEH:-1080}"
+export DISPLAY_REFRESH="${DISPLAY_REFRESH:-60}"
+export DISPLAY_DPI="${DISPLAY_DPI:-96}"
+export DISPLAY_CDEPTH="${DISPLAY_CDEPTH:-24}"
+export VIDEO_PORT="${VIDEO_PORT:-DFP}"
+export XDG_RUNTIME_DIR="${XDG_RUNTIME_DIR:-/tmp/runtime-${USER:-ubuntu}}"
+export PIPEWIRE_RUNTIME_DIR="${PIPEWIRE_RUNTIME_DIR:-${XDG_RUNTIME_DIR}}"
+export PULSE_RUNTIME_PATH="${PULSE_RUNTIME_PATH:-${XDG_RUNTIME_DIR}/pulse}"
+export PULSE_SERVER="${PULSE_SERVER:-unix:${PULSE_RUNTIME_PATH}/native}"
+
+until [[ -d "${XDG_RUNTIME_DIR}" ]]; do
+    sleep 0.5
+done
+
+sudo-root chown "$(id -u):$(id -g)" "${HOME}" || true
+
+if [[ -n "${PASSWD:-}" ]]; then
+    printf '%s:%s\n' "${USER}" "${PASSWD}" | sudo-root chpasswd || true
+fi
+
+sudo-root rm -f "/tmp/.X${DISPLAY#:}-lock" || true
+sudo-root install -d -m 1777 /tmp/.X11-unix /tmp/.ICE-unix
+rm -rf "${HOME}/.cache" || true
+
+sudo-root ln -snf "/usr/share/zoneinfo/${TZ:-UTC}" /etc/localtime || true
+printf '%s\n' "${TZ:-UTC}" | sudo-root tee /etc/timezone >/dev/null || true
+
+# Joystick interposer support.
+SELKIES_INTERPOSER_PATH="$(
+    find /usr/lib /usr/lib64 /usr/local/lib \
+        -name 'selkies_joystick_interposer.so' \
+        -print -quit 2>/dev/null || true
+)"
+
+if [[ -n "${SELKIES_INTERPOSER_PATH}" ]]; then
+    export LD_PRELOAD="${SELKIES_INTERPOSER_PATH}${LD_PRELOAD:+:${LD_PRELOAD}}"
+    export SDL_JOYSTICK_DEVICE="${SDL_JOYSTICK_DEVICE:-/dev/input/js0}"
+
+    sudo-root install -d -m 1777 /dev/input || true
+    for joystick in /dev/input/js{0..3}; do
+        sudo-root touch "${joystick}" || true
+        sudo-root chmod 0777 "${joystick}" || true
+    done
+fi
+
+install_nvidia_userspace() {
+    command -v nvidia-smi >/dev/null 2>&1 || {
+        echo "NVIDIA Container Toolkit has not exposed nvidia-smi." >&2
+        return 1
+    }
+
+    if command -v nvidia-xconfig >/dev/null 2>&1; then
+        return 0
+    fi
+
+    local driver_arch driver_version installer archive_dir
+
+    driver_arch="$(
+        dpkg --print-architecture |
+        sed -e 's/arm64/aarch64/' \
+            -e 's/armhf/32bit-ARM/' \
+            -e 's/i.*86/x86/' \
+            -e 's/amd64/x86_64/' \
+            -e 's/unknown/x86_64/'
+    )"
+
+    driver_version="${NVIDIA_DRIVER_VERSION:-}"
+
+    if [[ -z "${driver_version}" && -f /proc/driver/nvidia/version ]]; then
+        driver_version="$(
+            head -n 1 /proc/driver/nvidia/version |
+            awk '{for (i=1; i<=NF; i++) if ($i ~ /^[0-9]+\.[0-9.]+$/) {print $i; exit}}'
+        )"
+    fi
+
+    if [[ -z "${driver_version}" ]]; then
+        driver_version="$(
+            nvidia-smi --query-gpu=driver_version --format=csv,noheader |
+            head -n 1
+        )"
+    fi
+
+    if [[ -z "${driver_version}" ]]; then
+        echo "Unable to determine the host NVIDIA driver version." >&2
+        return 1
+    fi
+
+    installer="/tmp/NVIDIA-Linux-${driver_arch}-${driver_version}.run"
+    archive_dir="/tmp/NVIDIA-Linux-${driver_arch}-${driver_version}"
+
+    echo "Installing NVIDIA ${driver_version} userspace GLX components"
+
+    curl -fsSL \
+        "https://international.download.nvidia.com/XFree86/Linux-${driver_arch}/${driver_version}/NVIDIA-Linux-${driver_arch}-${driver_version}.run" \
+        -o "${installer}" \
+    || curl -fsSL \
+        "https://international.download.nvidia.com/tesla/${driver_version}/NVIDIA-Linux-${driver_arch}-${driver_version}.run" \
+        -o "${installer}"
+
+    rm -rf "${archive_dir}"
+    (
+        cd /tmp
+        sh "${installer}" -x
+    )
+
+    sudo-root "${archive_dir}/nvidia-installer" \
+        --silent \
+        --no-kernel-module \
+        --no-install-compat32-libs \
+        --no-nouveau-check \
+        --no-nvidia-modprobe \
+        --no-systemd \
+        --no-rpms \
+        --no-backup \
+        --no-check-for-alternate-installs
+
+    rm -rf "${installer}" "${archive_dir}"
+}
+
+install_nvidia_userspace
+
+# Always select the first GPU visible inside the container. Device indexes are
+# renumbered by NVIDIA Container Toolkit, so the host-side index is unreliable.
+GPU_SELECT="$(
+    nvidia-smi --query-gpu=uuid --format=csv,noheader |
+    head -n 1
+)"
+
+if [[ -z "${GPU_SELECT}" ]]; then
+    echo "No visible NVIDIA GPU was detected." >&2
+    exit 1
+fi
+
+HEX_ID="$(
+    nvidia-smi \
+        --id="${GPU_SELECT}" \
+        --query-gpu=pci.bus_id \
+        --format=csv,noheader |
+    head -n 1
+)"
+
+IFS=':.' read -r domain bus device function <<<"${HEX_ID}"
+
+BUS_ID="PCI:$((16#${bus}))@$((16#${domain})):$((16#${device})):$((16#${function}))"
+MODELINE="$(cvt -r "${DISPLAY_SIZEW}" "${DISPLAY_SIZEH}" "${DISPLAY_REFRESH}" | sed -n '2p')"
+MODE_NAME="$(awk '{print $2}' <<<"${MODELINE}" | tr -d '"')"
+
+if [[ "${VIDEO_PORT,,}" == "none" ]]; then
+    CONNECTED_MONITOR=(--use-display-device=None)
+else
+    CONNECTED_MONITOR=(--connected-monitor="${VIDEO_PORT}")
+fi
+
+sudo-root rm -f /etc/X11/xorg.conf
+
+sudo-root nvidia-xconfig \
+    --virtual="${DISPLAY_SIZEW}x${DISPLAY_SIZEH}" \
+    --depth="${DISPLAY_CDEPTH}" \
+    --mode="${MODE_NAME}" \
+    --allow-empty-initial-configuration \
+    --no-probe-all-gpus \
+    --busid="${BUS_ID}" \
+    --include-implicit-metamodes \
+    --mode-debug \
+    --no-sli \
+    --no-base-mosaic \
+    --only-one-x-screen \
+    "${CONNECTED_MONITOR[@]}"
+
+sudo-root sed -i \
+    '/Driver[[:space:]]*"nvidia"/a\    Option "ModeValidation" "NoMaxPClkCheck,NoEdidMaxPClkCheck,NoMaxSizeCheck,NoHorizSyncCheck,NoVertRefreshCheck,NoVirtualSizeCheck,NoExtendedGpuCapabilitiesCheck,NoTotalSizeCheck,NoDualLinkDVICheck,NoDisplayPortBandwidthCheck,AllowNon3DVisionModes,AllowNonHDMI3DModes,AllowNonEdidModes,NoEdidHDMI2Check,AllowDpInterlaced"' \
+    /etc/X11/xorg.conf
+
+sudo-root sed -i \
+    '/Driver[[:space:]]*"nvidia"/a\    Option "AllowExternalGpus" "True"' \
+    /etc/X11/xorg.conf
+
+sudo-root sed -i \
+    '/Section[[:space:]]*"Monitor"/a\    '"${MODELINE}" \
+    /etc/X11/xorg.conf
+
+sudo-root sed -i '/"DPMS"/d' /etc/X11/xorg.conf
+sudo-root sed -i \
+    '/Section[[:space:]]*"Monitor"/a\    Option "DPMS" "False"' \
+    /etc/X11/xorg.conf
+
+cat <<'EOF' | sudo-root tee -a /etc/X11/xorg.conf >/dev/null
+Section "ServerFlags"
+    Option "DontVTSwitch" "True"
+    Option "DontZap" "True"
+    Option "AllowMouseOpenFail" "True"
+    Option "AutoAddGPU" "False"
+EndSection
+EOF
+
+sudo-root ln -snf /dev/ptmx /dev/tty7
+
+/usr/lib/xorg/Xorg \
+    "${DISPLAY}" \
+    vt7 \
+    -noreset \
+    -novtswitch \
+    -sharevts \
+    -nolisten tcp \
+    -ac \
+    -dpi "${DISPLAY_DPI}" \
+    +extension COMPOSITE \
+    +extension DAMAGE \
+    +extension GLX \
+    +extension RANDR \
+    +extension RENDER \
+    +extension MIT-SHM \
+    +extension XFIXES \
+    +extension XTEST \
+    &
+
+XORG_PID=$!
+
+echo "Waiting for X socket ${DISPLAY}"
+until [[ -S "/tmp/.X11-unix/X${DISPLAY#*:}" ]]; do
+    if ! kill -0 "${XORG_PID}" 2>/dev/null; then
+        echo "Xorg exited before creating its socket." >&2
+        exit 1
+    fi
+    sleep 0.5
+done
+
+xset -dpms || true
+xset s off || true
+xset s noblank || true
+
+export XDG_SESSION_ID="${DISPLAY#*:}"
+
+xfce4-session &
+XFCE_PID=$!
+
+wait "${XFCE_PID}"