Update start.sh

start.sh

@@ -7,14 +7,28 @@ print_host_details() {
     echo "* The host File of this container is: $(cat /etc/hosts)"
     echo "* The Sudoers of this container is: $(cat /etc/sudoers)"
     echo "* ID of the user running the script:$(whoami) *ID : $(id -u) * Group: $(id -g)"
-    echo "* Changing User to Admin :$(echo "toor" || su - admin)"
+    #echo "* Changing User to Admin :$(echo "toor" || su - admin)"
     echo "* Current User WHO AM I $(whoami)"
     echo "* Status of Admin: $(id admin 2>/dev/null || echo 'Admin user not found')"
 }
+# Function to generate SSH host keys if missing
+generate_ssh_keys() {
+    if [ ! -f /etc/ssh/ssh_host_rsa_key ]; then
+        echo "* Generating SSH host keys *"        
+        yes y | ssh-keygen -t rsa -b 2048 -f /etc/ssh/ssh_host_rsa_key -N "" || { echo "Failed to generate RSA key"; exit 1; }
+        yes y | ssh-keygen -t ecdsa -b 256 -f /etc/ssh/ssh_host_ecdsa_key -N "" || { echo "Failed to generate ECDSA key"; exit 1; }
+        yes y | ssh-keygen -t ed25519 -f /etc/ssh/ssh_host_ed25519_key -N "" || { echo "Failed to generate ED25519 key"; exit 1; }
+        ssh-keyscan -p 2222 127.0.0.1 >> /home/admin/.ssh/known_hosts
+        cp /home/admin/.ssh/known_hosts /home/admin/.ssh/ssh_known_host
+        #ssh-keygen -t rsa -b 2048 -f /etc/ssh/ssh_host_rsa_key -N "" 
+        #ssh-keygen -t ecdsa -b 256 -f /etc/ssh/ssh_host_ecdsa_key -N "" 
+        #ssh-keygen -t ed25519 -f /etc/ssh/ssh_host_ed25519_key -N "" 
+    fi
+}
 # Function to start SSH service if not running
 start_ssh_service() {
     if ! pgrep -x "sshd" >/dev/null; then
-        echo "* Starting SSH server on port 2222 *"
+        echo "* Starting SSH server at port 2222 *"
         /usr/sbin/sshd -p 2222 || { echo "Failed to start SSH server"; exit 1; }
         #/usr/sbin/sshd -D -e -ddd|| { echo "Failed to start SSH server"; exit 1; }
     else