#!/usr/bin/env bash ## Dependencies: bash gzip fuse2 (or fuse3) tar coreutils # Prevent launching as root if [ $EUID = 0 ] && [ -z "$ALLOW_ROOT" ]; then echo "Do not run this script as root!" echo echo "If you really need to run it as root and you know what you are doing," echo "set ALLOW_ROOT environment variable." exit 1 fi script_version="1.15" # Full path to the script script_literal="${BASH_SOURCE[0]}" script_name="$(basename "${script_literal}")" script="$(readlink -f "${script_literal}")" # MD5 of the last 1 MB of the script script_md5="$(tail -c 1000000 "${script}" | md5sum | head -c 7)" script_id="${RANDOM}" # Working directory where the utils will be extracted # And where the image will be mounted # The default path is /tmp/scriptname_username_scriptmd5 # And if /tmp is mounted with noexec, the default path # is ~/.local/share/Conty/scriptname_username_scriptmd5 conty_dir_name="$(basename "${script}")"_"${USER}"_"${script_md5}" if [ -z "${BASE_DIR}" ]; then export working_dir=/tmp/"${conty_dir_name}" else export working_dir="${BASE_DIR}"/"${conty_dir_name}" fi mount_point="${working_dir}"/mnt # It's important to set correct sizes below, otherwise there will be # a problem with mounting the image due to an incorrectly calculated offset. # The size of this script scriptsize=21361 # The size of the utils.tar.gz archive # utils.tar.gz contains bwrap, squashfuse and dwarfs binaries utilssize=7418944 # Offset where the image is stored offset=$((scriptsize+utilssize)) # Set to 1 if you are using an image compressed with dwarfs instead of squashfs dwarfs_image=0 if [ "$1" = "--help" ] || [ "$1" = "-h" ] || ([ -z "$1" ] && [ ! -L "${script_literal}" ]); then echo "Usage: ./conty.sh command command_arguments" echo echo "Arguments:" echo echo -e "-v \tShow version of this script" echo -e "-e \tExtract the squashfs/dwarfs image" echo -e "-o \tShow the image offset" echo -e "-u \tUpdate all packages inside the container" echo -e "\tThis will update all packages inside the container and will rebuild" echo -e "\tthe image. This may take quite a lot of time, depending" echo -e "\ton your hardware and internet speed. Additional disk space" echo -e "\t(about 6x the size of the current file) is needed during" echo -e "\tthe update process." echo -e "-U \tThe same as -u but will also update the init script (conty-start.sh)" echo -e "\tand the integrated utils." echo echo "Environment variables:" echo echo -e "DISABLE_NET \tDisables network access" echo -e "SANDBOX \tEnables filesystem sandbox" echo -e "BIND \t\tBinds directories and files (separated by space) from the" echo -e "\t\thost system to the container. All specified items must exist." echo -e "\t\tFor example, BIND=\"/home/username/.config /etc/pacman.conf\"" echo -e "HOME_DIR \tSets the HOME directory to a custom location." echo -e "\t\tFor example, HOME_DIR=\"/home/username/custom_home\"" echo -e "\t\tIf you set this, HOME inside the container will still appear" echo -e "\t\tas /home/username, but actually a custom directory will be" echo -e "\t\tused for it." echo -e "USE_SYS_UTILS \tMakes the script to use squashfuse/dwarfs and bwrap" echo -e "\t\tinstalled on the system instead of the builtin ones." echo -e "NVIDIA_FIX \tAutomatically download and bind the required Nvidia" echo -e "\t\tlibraries if the kernel module version in the system differs" echo -e "\t\tfrom the Nvidia libraries version inside the container." echo -e "\t\tThis should fix the graphics acceleration problems on Nvidia." echo -e "SUDO_MOUNT \tMakes the script to mount the squashfs image by using" echo -e "\t\tthe regular mount command instead of squashfuse. In this" echo -e "\t\tcase root rights will be requested (via sudo) when mounting" echo -e "\t\tand unmounting. This doesn't work for dwarfs-compressed images." echo -e "BASE_DIR \tSets a custom directory where Conty will extract" echo -e "\t\tits builtin utilities and mount the image." echo -e "\t\tThe default location is /tmp." echo -e "QUIET_MODE \tDisables all non-error Conty messages." echo -e "\t\tDoesn't affect the output of applications." echo echo "Additional notes:" echo echo "If you enable SANDBOX but don't set BIND or HOME_DIR, then no system" echo "directories/files will be available at all inside the container and a fake" echo "temporary HOME directory will be used." echo echo "If the script is a symlink to itself but with a different name," echo "then the symlinked script will automatically run a program according" echo "to its name. For instance, if the script is a symlink with the name \"wine\"," echo "then it will automatically run wine during launch." echo echo "Besides updating all packages, you can also remove and install packages using" echo "the same -u (or -U) argument. To install packages add them as additional" echo "arguments, and to remove packages add a minus sign (-) before their names." echo "To install: ./conty.sh -u pkgname1 pkgname2 pkgname3" echo "To remove: ./conty.sh -u -pkgname1 -pkgname2 -pkgname3" echo "In this case Conty will update all packages and will additionally" echo "install and/or remove specified packages." exit elif [ "$1" = "-v" ]; then echo "${script_version}" exit elif [ "$1" = "-e" ]; then if [ "${dwarfs_image}" = 1 ]; then if command -v dwarfsextract 1>/dev/null; then dwarfsextract -i "${script}" -o "$(basename "${script}")"_files else echo "To extract the image install dwarfs." fi else if command -v unsquashfs 1>/dev/null; then unsquashfs -o $offset -user-xattrs -d "$(basename "${script}")"_files "${script}" else echo "To extract the image install squashfs-tools." fi fi exit elif [ "$1" = "-o" ]; then echo $offset exit fi show_msg () { if [ "${QUIET_MODE}" != 1 ]; then echo "$@" fi } exec_test () { mkdir -p "${working_dir}" exec_test_file="${working_dir}"/exec_test rm -f "${exec_test_file}" touch "${exec_test_file}" chmod +x "${exec_test_file}" [ -x "${exec_test_file}" ] } launch_wrapper () { if [ "$1" = "mount" ]; then ${use_sudo} "$@" elif [ "${USE_SYS_UTILS}" = 1 ]; then "$@" else "${working_dir}"/utils/ld-linux-x86-64.so.2 --library-path "${working_dir}"/utils "$@" fi } # Check if FUSE is installed when SUDO_MOUNT is not enabled if [ "${SUDO_MOUNT}" != 1 ] || [ "${dwarfs_image}" = 1 ]; then if ! command -v fusermount3 1>/dev/null && ! command -v fusermount 1>/dev/null; then echo "Please install fuse2 or fuse3 and run the script again!" exit 1 fi if command -v fusermount3 1>/dev/null; then fuse_version=3 fi fi # Extract utils.tar.gz mkdir -p "${working_dir}" if [ "${USE_SYS_UTILS}" != 1 ]; then # Check if filesystem of the working_dir is mounted without noexec if ! exec_test; then if [ -z "${BASE_DIR}" ]; then export working_dir="${HOME}"/.local/share/Conty/"${conty_dir_name}" mount_point="${working_dir}"/mnt fi if ! exec_test; then echo "Seems like /tmp is mounted with noexec or you don't have write access!" echo "Please remount it without noexec or set BASE_DIR to a different location." exit 1 fi fi if [ "${dwarfs_image}" = 1 ]; then mount_tool="${working_dir}"/utils/dwarfs${fuse_version} else mount_tool="${working_dir}"/utils/squashfuse${fuse_version} fi bwrap="${working_dir}"/utils/bwrap if [ ! -f "${mount_tool}" ] || [ ! -f "${bwrap}" ]; then tail -c +$((scriptsize+1)) "${script}" | head -c $utilssize | tar -C "${working_dir}" -zxf - if [ ! -f "${mount_tool}" ] || [ ! -f "${bwrap}" ]; then clear echo "The utilities were not extracted!" echo "Perhaps something is wrong with the integrated utils.tar.gz." exit 1 fi chmod +x "${mount_tool}" chmod +x "${bwrap}" fi else if ! command -v bwrap 1>/dev/null; then echo "USE_SYS_UTILS is enabled, but bwrap is not installed!" echo "Please install it and run the script again." exit 1 fi bwrap=bwrap if [ "${dwarfs_image}" = 1 ]; then if ! command -v dwarfs 1>/dev/null; then echo "USE_SYS_UTILS is enabled, but dwarfs is not installed!" echo "Please install it and run the script again." exit 1 fi mount_tool=dwarfs else if ! command -v squashfuse 1>/dev/null && [ "${SUDO_MOUNT}" != 1 ]; then echo "USE_SYS_UTILS is enabled, but squshfuse is not installed!" echo "Please install it and run the script again." echo "Or enable SUDO_MOUNT to mount the image using the regular" echo "mount command instead of squashfuse." exit 1 fi mount_tool=squashfuse fi show_msg "Using system-wide ${mount_tool} and bwrap" fi if [ "${SUDO_MOUNT}" = 1 ] && [ "${dwarfs_image}" != 1 ]; then show_msg "Using regular mount command (sudo mount) instead of squashfuse" mount_tool=mount use_sudo=sudo fi if [ "$1" = "-u" ] || [ "$1" = "-U" ]; then if [ "${dwarfs_image}" = 1 ]; then echo "This feature is currently unimplemented for dwarfs-compressed images." exit 1 fi OLD_PWD="${PWD}" # Check if current directory is writable # And if it's not, use ~/.local/share/Conty as a working directory if ! touch test_rw 2>/dev/null; then update_temp_dir="${HOME}"/.local/share/Conty/conty_update_temp else update_temp_dir="${OLD_PWD}"/conty_update_temp fi rm -f test_rw # Remove conty_update_temp directory if it already exists chmod -R 700 "${update_temp_dir}" 2>/dev/null rm -rf "${update_temp_dir}" mkdir -p "${update_temp_dir}" cd "${update_temp_dir}" || exit 1 # Since Conty is used here to update itself, it's necessary to disable # some environment variables for this to work properly unset NVIDIA_FIX unset DISABLE_NET unset HOME_DIR unset BIND # Enable SANDBOX export SANDBOX=1 # Extract the squashfs image clear echo "Extracting the squashfs image" bash "${script}" --bind "${update_temp_dir}" "${update_temp_dir}" \ --bind "${script}" /tmp/conty.sh \ unsquashfs -o $offset -user-xattrs -d sqfs /tmp/conty.sh # Download or extract the utils.tar.gz and the init script depending # on what command line argument is used (-u or -U) clear if [ "$1" = "-U" ] && command -v wget 1>/dev/null; then echo "Downloading the init script and the utils" wget -q --show-progress "https://github.com/Kron4ek/Conty/raw/master/conty-start.sh" wget -q --show-progress "https://github.com/Kron4ek/Conty/raw/master/utils.tar.gz" fi if [ ! -s conty-start.sh ] || [ ! -s utils.tar.gz ]; then echo "Extracting the init script and the integrated utils" tail -c +$((scriptsize+1)) "${script}" | head -c $utilssize > utils.tar.gz head -c $scriptsize "${script}" > conty-start.sh fi # Check if there are additional arguments passed shift if [ -n "$1" ]; then packagelist="$@" # Check which packages to install and which ones to remove for i in ${packagelist}; do if [ "$(echo "${i}" | head -c 1)" = "-" ]; then export pkgsremove="${pkgsremove} $(echo "${i}" | tail -c +2)" else export pkgsinstall="${pkgsinstall} ${i}" fi done fi # Generate a script to perform inside Conty # It updates Arch mirrorlist # Updates keyrings # Updates all installed packages # Installs additional packages (if requested) # Removes packages (if requested) # Clears package cache # Updates SSL CA certificates # Generates locales cat < container-update.sh reflector --protocol https --score 3 --sort rate --save /etc/pacman.d/mirrorlist fakeroot -- pacman -Syy 2>/dev/null fakeroot -- pacman --noconfirm -S archlinux-keyring 2>/dev/null fakeroot -- pacman --noconfirm -S chaotic-keyring 2>/dev/null rm -rf /etc/pacman.d/gnupg fakeroot -- pacman-key --init echo "keyserver hkps://keyserver.ubuntu.com" >> /etc/pacman.d/gnupg/gpg.conf fakeroot -- pacman-key --populate archlinux fakeroot -- pacman-key --populate chaotic fakeroot -- pacman --noconfirm --overwrite "*" -Su 2>/dev/null fakeroot -- pacman --noconfirm -Runs ${pkgsremove} 2>/dev/null fakeroot -- pacman --noconfirm -S ${pkgsinstall} 2>/dev/null rm -f /var/cache/pacman/pkg/* update-ca-trust locale-gen EOF rm -f sqfs/etc/resolv.conf cp /etc/resolv.conf sqfs/etc/resolv.conf mkdir -p sqfs/run/shm # Execute the previously generated script clear echo "Updating and installing packages" bash "${script}" --bind sqfs / --ro-bind /sys /sys --dev-bind /dev /dev \ --proc /proc --bind "${update_temp_dir}" "${update_temp_dir}" \ bash container-update.sh # Create a squashfs image clear echo "Creating a squashfs image" bash "${script}" --bind "${update_temp_dir}" "${update_temp_dir}" \ mksquashfs sqfs image -b 256K -comp zstd -Xcompression-level 14 # Combine into a single executable clear echo "Combining everything into a single executable" cat conty-start.sh utils.tar.gz image > conty_updated.sh chmod +x conty_updated.sh mv -f "${script}" "${script}".old."${script_md5}" 2>/dev/null mv -f conty_updated.sh "${script}" 2>/dev/null || move_failed=1 if [ "${move_failed}" = 1 ]; then mv -f conty_updated.sh "${OLD_PWD}" 2>/dev/null || \ mv -f conty_updated.sh "${HOME}" 2>/dev/null fi chmod -R 700 sqfs 2>/dev/null rm -rf "${update_temp_dir}" clear echo "Conty has been updated!" if [ "${move_failed}" = 1 ]; then echo echo "Replacing ${script} with the new one failed!" echo echo "You can find conty_updated.sh in the current working" echo "directory or in your HOME." fi exit fi run_bwrap () { if [ "$DISABLE_NET" = 1 ]; then show_msg "Network is disabled" net="--unshare-net" fi if [ "$SANDBOX" = 1 ]; then show_msg "Filesystem sandbox is enabled" dirs="--tmpfs /home --dir ${HOME} --tmpfs /opt --tmpfs /mnt \ --tmpfs /media --tmpfs /var --tmpfs /run --symlink /run /var/run \ --bind-try /run/user /run/user --bind-try /run/dbus /run/dbus \ --tmpfs /tmp --ro-bind-try /tmp/.X11-unix /tmp/.X11-unix" # unshare="--unshare-user-try --unshare-pid --unshare-uts --unshare-cgroup-try \ # --hostname Conty" else dirs="--bind-try /home /home --bind-try /mnt /mnt --bind-try /opt /opt \ --bind-try /media /media --bind-try /run /run --bind-try /var /var \ --bind-try ${HOME} ${HOME}" fi if [ -n "${HOME_DIR}" ]; then show_msg "Set home directory to ${HOME_DIR}" dirs="${dirs} --bind ${HOME_DIR} ${HOME}" fi if [ -n "$BIND" ]; then show_msg "Bound items: ${BIND}" for i in ${BIND}; do bind="${bind} --bind ${i} ${i}" done dirs="${dirs} ${bind}" fi # Set XAUTHORITY variable if it's missing (which is unlikely) if [ -z "${XAUTHORITY}" ]; then XAUTHORITY="${HOME}"/.Xauthority fi show_msg launch_wrapper "${bwrap}" --ro-bind "${mount_point}" / \ --dev-bind /dev /dev \ --ro-bind /sys /sys \ --bind-try /tmp /tmp \ --proc /proc \ --ro-bind-try /etc/resolv.conf /etc/resolv.conf \ --ro-bind-try /etc/hosts /etc/hosts \ --ro-bind-try /etc/nsswitch.conf /etc/nsswitch.conf \ --ro-bind-try /etc/passwd /etc/passwd \ --ro-bind-try /etc/group /etc/group \ --ro-bind-try /etc/machine-id /etc/machine-id \ --ro-bind-try /etc/asound.conf /etc/asound.conf \ --ro-bind-try /etc/localtime /etc/localtime \ ${dirs} \ ${net} \ ${nvidia_driver_bind} \ --ro-bind-try "${XAUTHORITY}" "${XAUTHORITY}" \ --setenv PATH "${CUSTOM_PATH}" \ "$@" } # Function that checks if the Nvidia kernel module loaded in the # system matches the version of the Nvidia libraries inside the container # and downloads corresponding Nvidia libs from the official site if they # are not the same. Also binds the downloaded libraries to the container. # # This is absolutely necessary for Nvidia GPUs, otherwise graphics # acceleration will not work. bind_nvidia_driver () { # Path to store downloaded Nvidia drivers nvidia_drivers_dir="${HOME}"/.local/share/Conty/nvidia-drivers # Check if the Nvidia module is loaded # If it's loaded, then likely Nvidia GPU is being used if lsmod | grep nvidia 1>/dev/null || nvidia-smi 1>/dev/null; then if nvidia-smi 1>/dev/null; then nvidia_version="$(nvidia-smi --query-gpu=driver_version --format=csv,noheader)" elif modinfo nvidia &>/dev/null; then nvidia_version="$(modinfo -F version nvidia 2>/dev/null)" else if [ -d /usr/lib/x86_64-linux-gnu ]; then nvidia_version="$(basename /usr/lib/x86_64-linux-gnu/libGLX_nvidia.so.*.* | tail -c +18)" else nvidia_version="$(basename /usr/lib/libGLX_nvidia.so.*.* | tail -c +18)" fi fi # Check if the kernel module version is different from the # libraries version inside the container if [ -n "${nvidia_version}" ]; then nvidia_version_inside="$(basename "${mount_point}"/usr/lib/libGLX_nvidia.so.*.* | tail -c +18)" if [ "$(cat "${nvidia_drivers_dir}"/current_version.txt 2>/dev/null)" != "${nvidia_version}" ] \ && [ "${nvidia_version}" != "${nvidia_version_inside}" ]; then echo "Nvidia driver version mismatch detected, trying to fix" OLD_PWD="${PWD}" mkdir -p "${nvidia_drivers_dir}" cd "${nvidia_drivers_dir}" rm -rf nvidia-driver rm -f nvidia.run echo "Downloading Nvidia ${nvidia_version}, please wait" # Try to download from the default Nvidia url driver_url="https://us.download.nvidia.com/XFree86/Linux-x86_64/${nvidia_version}/NVIDIA-Linux-x86_64-${nvidia_version}.run" wget -q --show-progress "${driver_url}" -O nvidia.run # If the previous download failed, get url from flathub if [ ! -s nvidia.run ]; then rm -f nvidia.run driver_url="https:$(wget -q "https://raw.githubusercontent.com/flathub/org.freedesktop.Platform.GL.nvidia/master/data/nvidia-${nvidia_version}-i386.data" \ -O - | cut -d ':' -f 6)" wget -q --show-progress "${driver_url}" -O nvidia.run fi if [ -s nvidia.run ]; then chmod +x nvidia.run echo "Unpacking nvidia.run..." ./nvidia.run -x &>/dev/null rm nvidia.run mv NVIDIA-Linux-x86_64-${nvidia_version} nvidia-driver echo ${nvidia_version} > current_version.txt fi cd "${OLD_PWD}" fi # Bind the downloaded Nvidia libs to the container if [ -d "${nvidia_drivers_dir}"/nvidia-driver ]; then nvidia_libs_list="libcuda.so libEGL_nvidia.so libGLESv1_CM_nvidia.so \ libGLESv2_nvidia.so libGLX_nvidia.so libnvcuvid.so libnvidia-cbl.so \ libnvidia-cfg.so libnvidia-eglcore.so libnvidia-encode.so libnvidia-fbc.so \ libnvidia-glcore.so libnvidia-glsi.so libnvidia-glvkspirv.so libnvidia-ifr.so \ libnvidia-ml.so libnvidia-ngx.so libnvidia-opticalflow.so libnvidia-ptxjitcompiler.so \ libnvidia-rtcore.so libnvidia-tls.so libnvoptix.so" for lib in ${nvidia_libs_list}; do if [ -f "${mount_point}"/usr/lib/${lib}.${nvidia_version_inside} ]; then nvidia_driver_bind="${nvidia_driver_bind} \ --ro-bind-try ${nvidia_drivers_dir}/nvidia-driver/${lib}.${nvidia_version} \ /usr/lib/${lib}.${nvidia_version_inside}" fi if [ -f "${mount_point}"/usr/lib32/${lib}.${nvidia_version_inside} ]; then nvidia_driver_bind="${nvidia_driver_bind} \ --ro-bind-try ${nvidia_drivers_dir}/nvidia-driver/32/${lib}.${nvidia_version} \ /usr/lib32/${lib}.${nvidia_version_inside}" fi if [ -f "${mount_point}"/usr/lib/nvidia/xorg/libglxserver_nvidia.so.${nvidia_version_inside} ]; then nvidia_driver_bind="${nvidia_driver_bind} \ --ro-bind-try ${nvidia_drivers_dir}/nvidia-driver/libglxserver_nvidia.so.${nvidia_version} \ /usr/lib/nvidia/xorg/libglxserver_nvidia.so.${nvidia_version_inside}" fi if [ -f "${mount_point}"/usr/lib/vdpau/libvdpau_nvidia.so.${nvidia_version_inside} ]; then nvidia_driver_bind="${nvidia_driver_bind} \ --ro-bind-try ${nvidia_drivers_dir}/nvidia-driver/libvdpau_nvidia.so.${nvidia_version} \ /usr/lib/vdpau/libvdpau_nvidia.so.${nvidia_version_inside}" fi if [ -f "${mount_point}"/usr/lib32/vdpau/libvdpau_nvidia.so.${nvidia_version_inside} ]; then nvidia_driver_bind="${nvidia_driver_bind} \ --ro-bind-try ${nvidia_drivers_dir}/nvidia-driver/32/libvdpau_nvidia.so.${nvidia_version} \ /usr/lib32/vdpau/libvdpau_nvidia.so.${nvidia_version_inside}" fi done fi fi fi } trap_exit () { rm -f "${working_dir}"/running_"${script_id}" if [ ! "$(ls "${working_dir}"/running_* 2>/dev/null)" ]; then fusermount${fuse_version} -uz "${mount_point}" 2>/dev/null || \ ${use_sudo} umount --lazy "${mount_point}" 2>/dev/null rm -rf "${working_dir}" fi exit } trap 'trap_exit' EXIT # Mount the squashfs image mkdir -p "${mount_point}" if [ "$(ls "${mount_point}" 2>/dev/null)" ] || \ ( [ "${dwarfs_image}" != 1 ] && launch_wrapper "${mount_tool}" -o offset="${offset}",ro "${script}" "${mount_point}" ) || \ launch_wrapper "${mount_tool}" "${script}" "${mount_point}" -o offset="${offset}" -o debuglevel=error; then echo 1 > "${working_dir}"/running_"${script_id}" show_msg "Running Conty" if [ "${NVIDIA_FIX}" = 1 ]; then bind_nvidia_driver fi if [ -L "${script_literal}" ] && [ -f "${mount_point}"/usr/bin/"${script_name}" ]; then export CUSTOM_PATH="/bin:/sbin:/usr/bin:/usr/sbin:/usr/lib/jvm/default/bin" show_msg "Autostarting ${script_name}" run_bwrap "${script_name}" "$@" else export CUSTOM_PATH="/bin:/sbin:/usr/bin:/usr/sbin:/usr/lib/jvm/default/bin:/usr/local/bin:/usr/local/sbin:${PATH}" run_bwrap "$@" fi else echo "Mounting the image failed!" exit 1 fi exit