Update bootstrap.sh

bootstrap.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 
-### 1. VARIABLES
+#################. VARIABLES
 MOTD="/etc/update-motd.d/01-custom"
 REBOOTBIN="/usr/sbin/reboot"
 REBOOTBINOLD="/usr/sbin/reboot.old"
@@ -10,16 +10,20 @@ REBOOTHANDLER="/root/scripts/reboot_handler.sh"
 CRONTABTMP="/tmp/crontab.root.tmp"
 HOSTNAME=$(hostname)
 KEYFILE="/root/.ssh/$HOSTNAME"
-MOTD_TEMP="/etc/update-motd/02-temp"
+TEMP_SCRIPT="/root/firstlogin.sh"
+TEMP_SCRIPT_WRAPPER="/root/firstloginwrapper.sh"
+TEMP_SCRIPT_CLEANER="/root/firstlogincleaner.sh"
 
 
-## 2. ALIASES
+#################. ALIASES
 echo "alias ll='ls -l --color=auto'" >> ~/.bashrc
 echo "alias l='ls -lAh --color=auto'" >> ~/.bashrc
+echo "" >> ~/.bashrc
+echo 'PS1="\u\[\e[0m\]@\[\e[31m\]\h\[\e[0m\]:\[\e[34m\]\w\[\e[0m\]# "' >> ~/.bashrc
 source ~/.bashrc
 
 
-## 3. UPDATE & ESSENTIALS
+#################. UPDATE & ESSENTIALS
 apt update && apt -o Dpkg::Options::="--force-confold" upgrade -y
 apt install -y vim inxi fastfetch htop ncdu net-tools
 
@@ -56,29 +60,27 @@ fi
 EOF
 chmod +x $MOTD
 
-### 4. Création du script first-login
+#################. Création du script first-login
-
+cat << 'EOF' > $TEMP_SCRIPT
-cat << 'EOF' > $MOTD_TEMP
-
 #!/bin/bash
-RED='\033[0;31m'
-BLUE='\033[0;34m'
-NC='\033[0m' # No Color
 
 HOSTNAME=$(hostname)
 KEYFILE="/root/.ssh/$HOSTNAME"
 
 echo
-echo -e "${BLUE}===============================${NC}"
+echo "==============================="
-echo -e "${BLUE} Clé privée SSH à conserver !${NC}"
+echo " Clé privée SSH à conserver !"
-echo -e "${BLUE}===============================${NC}"
+echo "==============================="
 echo
-cat "${KEYFILE}"
+
+# Affichage dans TON shell
+cat "$KEYFILE"
+
 echo
-echo "${BLUE}===============================${NC}"
+echo "==============================="
 
 # Suppression de la clé privée
-rm -f "${KEYFILE}"
+rm -f "$KEYFILE"
 
 # Durcissement SSH
 sed -i 's/^#\?PubkeyAuthentication.*/PubkeyAuthentication yes/' /etc/ssh/sshd_config
@@ -87,16 +89,56 @@ sed -i 's/^#\?PermitRootLogin.*/PermitRootLogin prohibit-password/' /etc/ssh/ssh
 sed -i 's/^#\?PasswordAuthentication.*/PasswordAuthentication no/' /etc/ssh/sshd_config
 
 systemctl restart sshd
-rm -fr $MOTD_TEMP
-EOF
-chmod +x $MOTD_TEMP
 
-### 5. Hardening binaries
+# Nettoyage différé
+nohup /root/firstlogincleaner.sh >/dev/null 2>&1 &
+exit 0
+EOF
+chmod +x $TEMP_SCRIPT
+
+
+
+cat << 'EOF' > $TEMP_SCRIPT_WRAPPER
+#!/bin/bash
+
+if [ -f /root/firstlogin.sh ]; then
+    /root/firstlogin.sh
+fi
+
+exit 0
+EOF
+chmod +x $TEMP_SCRIPT_WRAPPER
+
+
+cat << 'EOF' > $TEMP_SCRIPT_CLEANER
+#!/bin/bash
+
+# Attendre la fin du login
+sleep 2
+
+# Supprimer la ligne PAM
+sed -i '/firstloginwrapper.sh/d' /etc/pam.d/sshd
+
+# Supprimer les scripts
+rm -f /root/firstlogin.sh
+rm -f /root/firstloginwrapper.sh
+rm -f /root/firstlogincleaner.sh
+rm -f /root/bootstrap.sh
+
+exit 0
+EOF
+chmod +x $TEMP_SCRIPT_CLEANER
+
+echo "auth optional pam_exec.so stdout /root/firstloginwrapper.sh" >> /etc/pam.d/sshd
+
+
+#################. Hardening binaries
 # REBOOT WRAPPER
 mv $REBOOTBIN $REBOOTBINOLD
 cat << 'EOF' > $REBOOTBIN
 #!/bin/bash
 touch /var/log/restart-flag
+echo "[REBOOT] Rebooting server..."
 sleep 1
 /usr/sbin/reboot.old
 EOF
@@ -106,13 +148,14 @@ chmod +x $REBOOTBIN
 cat << 'EOF' > $SHUTBIN
 #!/bin/bash
 touch /var/log/restart-flag
+echo "[SHUTDOWN] Stopping server..."
 sleep 1
 shutdown -h now
 EOF
 chmod +x $SHUTBIN
 
 
-### 6. REBOOT HANDLER
+#################. REBOOT HANDLER
 mkdir -p $SCRIPTSDIR
 cat << 'EOF' > $REBOOTHANDLER
 #!/bin/bash
@@ -168,3 +211,8 @@ ssh-keygen -t ed25519 -C "$HOSTNAME" -f "$KEYFILE" -N ""
 # Ajouter la clé publique dans authorized_keys
 cat "${KEYFILE}.pub" >> /root/.ssh/authorized_keys
 chmod 600 /root/.ssh/authorized_keys
+
+rm -fr /etc/systemd/system/bootstrap.service
+rm -fr /etc/systemd/system/multi-user.target.wants/bootstrap.service
+rm -fr /root/run-bootstrap.sh
+systemctl daemon-reload