Create .ssh dir if not exsit

Under image mode, the ~/.ssh dir is created automatically
when execute ssh-keygen, so create it if not exsit

Signed-off-by: Lili Zhu <lizhu@redhat.com>

Author: iccaszhulili

virttest/ssh_key.py

@@ -65,11 +65,25 @@ def get_public_key(client_user=None):
                 'ssh-keygen -t rsa -q -N "" -f %s' % rsa_private_key_path, shell=True
             )
         else:
-            process.system(
-                "su - %s -c 'ssh-keygen -t rsa -q -N \"\" -f %s'"
-                % (client_user, rsa_private_key_path),
-                shell=True,
-            )
+            try:
+                process.system(
+                    "su - %s -c 'ssh-keygen -t rsa -q -N \"\" -f $HOME/.ssh/id_rsa'"
+                    % client_user,
+                    shell=True,
+                )
+            except Exception as e:
+                if "No such file or directory" in str(e):
+                    LOG.info("Creating directory %s", ssh_conf_path)
+                    os.makedirs(ssh_conf_path, mode=0o700)
+                    process.run("chown -R %s:%s %s" %
+                                (client_user, client_user, ssh_conf_path), shell=True)
+                    process.system(
+                        "su - %s -c 'ssh-keygen -t rsa -q -N \"\" -f $HOME/.ssh/id_rsa'"
+                        % client_user,
+                        shell=True,
+                    )
+                else:
+                    raise
         public_key_path = rsa_public_key_path
 
     public_key = open(public_key_path, "r")