Update start.sh

start.sh

@@ -13,9 +13,13 @@ print_host_details() {
 generate_ssh_keys() {
     if [ ! -f /etc/ssh/ssh_host_rsa_key ]; then
         echo "* Generating SSH host keys *"
-        ssh-keygen -t rsa -b 2048 -f /etc/ssh/ssh_host_rsa_key -N "" || { echo "Failed to generate RSA key"; exit 1; }
-        ssh-keygen -t ecdsa -b 256 -f /etc/ssh/ssh_host_ecdsa_key -N "" || { echo "Failed to generate ECDSA key"; exit 1; }
-        ssh-keygen -t ed25519 -f /etc/ssh/ssh_host_ed25519_key -N "" || { echo "Failed to generate ED25519 key"; exit 1; }
+        
+        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-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
 }
 
@@ -52,9 +56,9 @@ generate_admin_ssh_keys() {
         echo "* Generating SSH keys for 'admin' user *"
         mkdir -p /home/admin/.ssh || { echo "Failed to create .ssh directory"; exit 1; }
         ssh-keygen -t rsa -b 2048 -f /home/admin/.ssh/id_rsa -q -N "" || { echo "Failed to generate admin SSH key"; exit 1; }
-        chmod 755 /home/admin/.ssh || { echo "Failed to set .ssh directory permissions"; exit 1; }
-        chmod 755 /home/admin/.ssh/id_rsa || { echo "Failed to set private key permissions"; exit 1; }
-        chmod 755 /home/admin/.ssh/id_rsa.pub || { echo "Failed to set public key permissions"; exit 1; }
+        chmod 700 /home/admin/.ssh || { echo "Failed to set .ssh directory permissions"; exit 1; }
+        chmod 600 /home/admin/.ssh/id_rsa || { echo "Failed to set private key permissions"; exit 1; }
+        chmod 644 /home/admin/.ssh/id_rsa.pub || { echo "Failed to set public key permissions"; exit 1; }
     fi
 }
 
@@ -63,7 +67,7 @@ add_admin_authorized_keys() {
     if [ ! -f /home/admin/.ssh/authorized_keys ]; then
         echo "* Adding public key to authorized_keys for 'admin' user *"
         cat /home/admin/.ssh/id_rsa.pub >> /home/admin/.ssh/authorized_keys || { echo "Failed to add public key to authorized_keys"; exit 1; }
-        chmod 755 /home/admin/.ssh/authorized_keys || { echo "Failed to set authorized_keys permissions"; exit 1; }
+        chmod 600 /home/admin/.ssh/authorized_keys || { echo "Failed to set authorized_keys permissions"; exit 1; }
     fi
 }
 
@@ -105,7 +109,7 @@ run_webssh_application() {
 
 # Function to keep the container running
 keep_container_running() {
-    tail -f /dev/null
+    tail -f /var/log/auth.log
 }
 
 # Main script execution