Update bootstrap.sh

bootstrap.sh

@@ -18,6 +18,8 @@ TEMP_SCRIPT_CLEANER="/root/firstlogincleaner.sh"
 #################. 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
 
 
@@ -89,7 +91,7 @@ sed -i 's/^#\?PasswordAuthentication.*/PasswordAuthentication no/' /etc/ssh/sshd
 systemctl restart sshd
 
 # Nettoyage différé
-nohup /root/first-login-cleaner.sh >/dev/null 2>&1 &
+nohup /root/firstlogincleaner.sh >/dev/null 2>&1 &
 exit 0
 EOF
 chmod +x $TEMP_SCRIPT
@@ -99,8 +101,8 @@ chmod +x $TEMP_SCRIPT
 cat << 'EOF' > $TEMP_SCRIPT_WRAPPER
 #!/bin/bash
 
-if [ -f /root/first-login.sh ]; then
-    /root/first-login.sh
+if [ -f /root/firstlogin.sh ]; then
+    /root/firstlogin.sh
 fi
 
 exit 0
@@ -112,21 +114,22 @@ cat << 'EOF' > $TEMP_SCRIPT_CLEANER
 #!/bin/bash
 
 # Attendre la fin du login
-sleep 10
+sleep 2
 
 # Supprimer la ligne PAM
-sed -i '/first-login-wrapper.sh/d' /etc/pam.d/sshd
+sed -i '/firstloginwrapper.sh/d' /etc/pam.d/sshd
 
 # Supprimer les scripts
-rm -f /root/first-login.sh
-rm -f /root/first-login-wrapper.sh
-rm -f /root/first-login-cleaner.sh
+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/first-login-wrapper.sh" >> /etc/pam.d/sshd
+echo "auth optional pam_exec.so stdout /root/firstloginwrapper.sh" >> /etc/pam.d/sshd
 
 
 #################. Hardening binaries
@@ -135,6 +138,7 @@ mv $REBOOTBIN $REBOOTBINOLD
 cat << 'EOF' > $REBOOTBIN
 #!/bin/bash
 touch /var/log/restart-flag
+echo "[REBOOT] Rebooting server..."
 sleep 1
 /usr/sbin/reboot.old
 EOF
@@ -144,6 +148,7 @@ chmod +x $REBOOTBIN
 cat << 'EOF' > $SHUTBIN
 #!/bin/bash
 touch /var/log/restart-flag
+echo "[SHUTDOWN] Stopping server..."
 sleep 1
 shutdown -h now
 EOF