Update aws-ssm-ec2-proxy-command.sh

Author: qoomon

aws-ssm-ec2-proxy-command.sh

@@ -21,7 +21,7 @@
 #
 # Open SSH Connection
 #   ssh <INSTANCE_USER>@<INSTANCE_ID>
-#   
+#
 #   Ensure AWS CLI environment variables are set properly
 #   e.g. AWS_PROFILE='default' ssh ec2-user@i-xxxxxxxxxxxxxxxx
 #
@@ -52,13 +52,19 @@ aws ssm send-command \
   --document-name 'AWS-RunShellScript' \
   --comment "Add an SSH public key to authorized_keys for ${ssh_public_key_timeout} seconds" \
   --parameters commands="\"
-    mkdir -p ~${ssh_user}/.ssh && cd ~${ssh_user}/.ssh || exit 1
-    
+    ssh_user_home=\$(getent passwd '${ssh_user}' | cut -d: -f6)
+    if [ -z \\\"\${ssh_user_home}\\\" ]
+    then
+      exit 1
+    fi
+
+    mkdir -p \\\"\${ssh_user_home}/.ssh\\\" && cd \\\"\${ssh_user_home}/.ssh\\\" || exit 1
+
     authorized_key='${ssh_public_key} ssm-session'
     echo \\\"\${authorized_key}\\\" >> authorized_keys
-    
+
     sleep ${ssh_public_key_timeout}
-    
+
     grep -v -F \\\"\${authorized_key}\\\" authorized_keys > .authorized_keys
     mv .authorized_keys authorized_keys
   \""