ogboot/bin/setsmbpass

#!/bin/bash

PROG=$(basename $0)
OPENGNSYS=${OPENGNSYS:-"/opt/opengnsys"}
OGCFGFILE=$OPENGNSYS/etc/opengnsys.json
SAMBAUSER="opengnsys"  # Usuario por defecto.
TFTPDIR=$OPENGNSYS/tftpboot
INITRD=oginitrd.img
TMPDIR=/tmp/oglive$$
let CHANGES=0

# Control básico de errores.
if [ "$USER" != "root" ] && [ "$USER" != "ogboot" ] && ! id -nG "$USER" | grep -qw "ogboot"; then
    echo "$PROG: Error: solo ejecutable por root, ogboot o miembros del grupo ogboot" >&2
    exit 1
fi

case $# in
    0)  # Cambios en todos los clientes ogLive instalados.
        if which oglivecli &>/dev/null; then
            LIST=$(oglivecli list | awk '{print $2}')
        else
            LIST="ogclient"
        fi ;;
    1)  # Cambios en único ogLive (AVISO: puede crear inconsistencias con otros ogLive).
        LIST="$1" ;;
    *)  # Error de formato.
        echo "$PROG: Error de ejecución" >&2
        echo "Formato: $PROG ogLive"
        exit 1 ;;
esac

# Recuperar eco de consola si se corta el proceso.
trap "stty echo 2>/dev/null" KILL
# Buscar todos los clients ogLive instalados.
for OGLIVE in $LIST; do
    # Crear clave para usuario de acceso a los recursos.
    CLIENTINITRD="$TFTPDIR/$OGLIVE/$INITRD"
    if [ -r "$CLIENTINITRD" ]; then
        if [ -z "$SAMBAPASS" ]; then
            # Obtener clave del teclado sin eco en pantalla.
            stty -echo 2>/dev/null
            echo -n "Clave del usuario Samba: "
            read -r SAMBAPASS
            # Solo se deben aceptar números y letras para la clave de acceso.
            if [[ "$SAMBAPASS" =~ [^a-zA-Z0-9] ]]; then
                echo
                echo "$PROG: Error: la clave solo debe contener caracteres alfanuméricos" >&2
                stty echo 2>/dev/null
                exit 2
            fi
            echo
            # Obtener confirmación clave sin eco en pantalla.
            echo -n "Confirmar clave: "
            read -r SAMBAPASS2
            echo
            stty echo 2>/dev/null
            if [ "$SAMBAPASS" != "$SAMBAPASS2" ]; then
                echo "$PROG: Error: las claves no coinciden" >&2
                exit 2
            fi
        fi
        # Editar la parte de acceso del cliente:
        #    descomprimir Initrd, sustituir clave y recomprimir Initrd).
        echo "Configurando cliente \"$OGLIVE\" ..."
        mkdir -p $TMPDIR
        cd $TMPDIR || { echo "Error: no se pudo cambiar al directorio temporal."; exit 3; }

        # Verificar si el archivo es gzip o lz4 antes de descomprimir.
        if file "$CLIENTINITRD" | grep -q "gzip compressed data"; then
            if ! gzip -dc "$CLIENTINITRD" | sudo cpio -im; then
                echo "Error: No se pudo descomprimir y extraer $CLIENTINITRD con gzip."
                exit 4
            fi
            COMPRESS_CMD="gzip -9c"
        elif file "$CLIENTINITRD" | grep -q "LZ4 compressed data"; then
            if ! lz4 -d "$CLIENTINITRD" | sudo cpio -im; then
                echo "Error: No se pudo descomprimir y extraer $CLIENTINITRD con lz4."
                exit 4
            fi
            COMPRESS_CMD="lz4 -c"
        else
            echo "Error: $CLIENTINITRD no está en formato gzip o lz4."
            exit 4
        fi

        if [ -f scripts/ogfunctions ]; then
            sudo sed -i "s/OPTIONS=\(.*\)user=\w*\(.*\)pass=\w*\(.*\)/OPTIONS=\1user=$SAMBAUSER\2pass=$SAMBAPASS\3/" scripts/ogfunctions
            # TEMPORAL: solución ticket 554, actualizar cliente en caché (ogLive r3257).
            sudo sed -i "s/busybox reboot/reboot/" scripts/ogfunctions
            # FIN CÓDIGO TEMPORAL.
            # Ticket 565, preparar acceso Rsync cliente.
            echo "$SAMBAPASS" | sudo tee scripts/passrsync > /dev/null
            # Guardar tokens de seguridad.
            cat <<EOT | sudo tee scripts/client.cfg > /dev/null
CLIENTID=$(jq -r .client.id $OGCFGFILE)
CLIENTSECRET=$(jq -r .client.secret $OGCFGFILE)
EOT
            sudo chown root.root scripts/passrsync scripts/client.cfg
            sudo chmod 400 scripts/passrsync scripts/client.cfg
            pwd
            # Paso 1: Buscar archivos
            sudo find . -print > /tmp/filelist.txt
            if [ $? -ne 0 ]; then
                echo "Error: No se pudo encontrar los archivos."
                exit 5
            fi

            # Paso 2: Crear el archivo CPIO
            sudo cpio -H newc -oa < /tmp/filelist.txt > /tmp/initrd.cpio
            if [ $? -ne 0 ]; then
                echo "Error: No se pudo crear el archivo CPIO."
                exit 5
            fi

            # Paso 3: Comprimir el archivo CPIO
            sudo $COMPRESS_CMD < /tmp/initrd.cpio > /tmp/initrd.cpio.gz
            if [ $? -ne 0 ]; then
                echo "Error: No se pudo comprimir el archivo CPIO."
                exit 5
            fi
	    echo $CLIENTINITRD
            # Paso 4: Escribir el archivo comprimido en el destino
            # sudo tee "$CLIENTINITRD" < /tmp/initrd.cpio.gz > /dev/null
            sudo dd if=/tmp/initrd.cpio.gz of=/opt/opengnsys/tftpboot/ogLive-5.11.0-r20210413/oginitrd.img
            if [ $? -ne 0 ]; then
                echo "Error: No se pudo escribir el archivo comprimido en el destino."
                exit 5
            fi
            sudo rm /tmp/initrd.cpio /tmp/initrd.cpio.gz
        else
            echo "$PROG: Aviso: no se ha modificado la clave del cliente \"$OGLIVE\"."
        fi
        sudo rm -fr $TMPDIR
        # Calcular suma de comprobación.
        sudo md5sum "$CLIENTINITRD" | cut -f1 -d" " | sudo tee "$CLIENTINITRD.sum" > /dev/null
        let CHANGES++
    else
        echo "$PROG: Cliente \"$OGLIVE\" no accesible."
    fi
done

if [[ $CHANGES != 0 ]]; then
    # Ticket 565, preparar acceso Rsync servidor.
    echo "$SAMBAUSER:$SAMBAPASS" | sudo tee /etc/rsyncd.secrets > /dev/null
    echo "chown debugging"
    sudo chown root:root /etc/rsyncd.secrets
    sudo chmod 600 /etc/rsyncd.secrets
    # Cambiar clave Samba.
    echo -ne "$SAMBAPASS\n$SAMBAPASS\n" | sudo smbpasswd -a -s $SAMBAUSER
    echo "setsmbpass finish"
else
    echo "$PROG: Aviso: no se ha modificado la clave de ningún cliente."
fi