From b2db51d7ee05ffd381b00e6d41ebb57fe97d8322 Mon Sep 17 00:00:00 2001
From: MacRimi <123239993+MacRimi@users.noreply.github.com>
Date: Fri, 20 Dec 2024 19:37:16 +0100
Subject: [PATCH] Update add_hw_acceleration_lxc.sh

---
 scripts/add_hw_acceleration_lxc.sh | 61 ++++++++++++++++++++++++++++--
 1 file changed, 57 insertions(+), 4 deletions(-)

diff --git a/scripts/add_hw_acceleration_lxc.sh b/scripts/add_hw_acceleration_lxc.sh
index ec8767f..962f09a 100644
--- a/scripts/add_hw_acceleration_lxc.sh
+++ b/scripts/add_hw_acceleration_lxc.sh
@@ -1,12 +1,12 @@
 #!/bin/bash
 
-# Funciones y colores
+# Colores para salida
 YW="\033[33m"; GN="\033[1;92m"; RD="\033[01;31m"; CL="\033[m"
 msg_info() { echo -ne " ${YW}[INFO] $1...${CL}"; }
 msg_ok() { echo -e " ${GN}[OK] $1${CL}"; }
 msg_error() { echo -e " ${RD}[ERROR] $1${CL}"; }
 
-# Validaciones
+# Validar la versión de Proxmox
 validate_pve_version() {
     if ! pveversion | grep -Eq "pve-manager/(8\\.[0-9]+)"; then
         msg_error "Se requiere Proxmox VE 8.0 o superior."
@@ -14,6 +14,7 @@ validate_pve_version() {
     fi
 }
 
+# Selección del contenedor LXC
 select_container() {
     CONTAINERS=$(pct list | awk 'NR>1 {print $1, $3}' | xargs -n2)
     CONTAINER_ID=$(whiptail --title "Seleccionar Contenedor" --menu "Selecciona un contenedor LXC:" 15 60 5 $CONTAINERS 3>&1 1>&2 2>&3)
@@ -25,10 +26,52 @@ select_container() {
     msg_ok "Contenedor seleccionado: $CONTAINER_ID"
 }
 
+# Asegurarse de que el contenedor sea privilegiado
+ensure_privileged_container() {
+    CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
+
+    # Validar que el archivo de configuración exista
+    if [ ! -f "$CONFIG_FILE" ]; then
+        msg_error "Archivo de configuración del contenedor $CONTAINER_ID no encontrado."
+        exit 1
+    fi
+
+    # Verificar si el contenedor es no privilegiado
+    if grep -q "^unprivileged: 1" "$CONFIG_FILE"; then
+        msg_info "El contenedor es no privilegiado. Cambiando a privilegiado..."
+
+        # Cambiar la configuración a privilegiado
+        sed -i "s/^unprivileged: 1/unprivileged: 0/" "$CONFIG_FILE"
+
+        # Ajustar permisos del sistema de archivos si el almacenamiento es 'dir'
+        STORAGE_TYPE=$(pct config "$CONTAINER_ID" | grep "^rootfs:" | awk -F, '{print $2}' | cut -d'=' -f2)
+        if [[ "$STORAGE_TYPE" == "dir" ]]; then
+            STORAGE_PATH=$(pct config "$CONTAINER_ID" | grep "^rootfs:" | awk '{print $2}' | cut -d',' -f1)
+            chown -R root:root "$STORAGE_PATH"
+            msg_ok "Permisos ajustados para almacenamiento en directorios."
+        fi
+
+        msg_ok "Contenedor cambiado a privilegiado."
+    else
+        msg_ok "El contenedor ya es privilegiado."
+    fi
+}
+
+# Apagar el contenedor si está corriendo
+stop_container_if_running() {
+    if pct status "$CONTAINER_ID" | grep -q "status: running"; then
+        msg_info "Apagando el contenedor $CONTAINER_ID..."
+        pct stop "$CONTAINER_ID"
+        msg_ok "Contenedor $CONTAINER_ID apagado."
+    fi
+}
+
+# Configurar iGPU en el contenedor
 configure_lxc_for_igpu() {
     CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
     CONFIG_PRESENT=true
 
+    # Verificar cada línea específica para la configuración de iGPU
     if ! grep -q "lxc.cgroup2.devices.allow: c 226:0 rwm" "$CONFIG_FILE"; then
         CONFIG_PRESENT=false
     fi
@@ -66,9 +109,16 @@ configure_lxc_for_igpu() {
     msg_ok "Configuración de iGPU añadida al contenedor."
 }
 
-
-install_igpu_in_container() {
+# Iniciar el contenedor
+start_container() {
+    msg_info "Iniciando el contenedor $CONTAINER_ID..."
     pct start "$CONTAINER_ID"
+    msg_ok "Contenedor $CONTAINER_ID iniciado."
+}
+
+# Instalar controladores de iGPU en el contenedor
+install_igpu_in_container() {
+    msg_info "Instalando controladores de iGPU dentro del contenedor..."
     pct exec "$CONTAINER_ID" -- bash -c "
     apt-get update && \
     apt-get install -y va-driver-all intel-opencl-icd vainfo && \
@@ -80,6 +130,9 @@ install_igpu_in_container() {
 # Ejecución
 validate_pve_version
 select_container
+ensure_privileged_container
+stop_container_if_running
 configure_lxc_for_igpu
+start_container
 install_igpu_in_container
 msg_ok "Aceleración gráfica iGPU configurada correctamente."