add command to ssh key when using PUT

src/api/ssh_keys_post.c

@@ -41,13 +41,12 @@ int callback_ssh_keys_post(const struct _u_request * request,
         return send_simple_response(response, 400, "error", "missing ssh-key");
     }
 
+    // add cryptops-client command to ssh-key
     char * ssh_key_with_command;
-
     add_ssh_command(&ssh_key_with_command, ssh_key);
 
-    asprintf(&ssh_key_with_command, "%s\n", ssh_key_with_command);
-
     // Write SSH key to file
+    asprintf(&ssh_key_with_command, "%s\n", ssh_key_with_command);
     fprintf(authorized_keys, ssh_key_with_command);
     fclose(authorized_keys);
 

src/api/ssh_keys_put.c

@@ -50,7 +50,12 @@ int callback_ssh_keys_put(const struct _u_request * request,
         return send_simple_response(response, 400, "error", "missing ssh-key");
     }
 
-    r = replace_ssh_key(id, ssh_key);
+    // add cryptops-client command to ssh-key
+    char * ssh_key_with_command;
+    add_ssh_command(&ssh_key_with_command, ssh_key);
+
+    r = replace_ssh_key(id, ssh_key_with_command);
+
     if (r < 0)
     {
         if (r == -1)