pki: Add ssh_pki_import_pubkey_file().

2 files changed, 76 insertions, 0 deletions

diff --git a/include/libssh/libssh.h b/include/libssh/libssh.h
index b74a2bc..cbe9ff1 100644
--- a/include/libssh/libssh.h
+++ b/include/libssh/libssh.h
@@ -440,6 +440,9 @@ LIBSSH_API int ssh_pki_import_pubkey_base64(ssh_session session,
 LIBSSH_API int ssh_pki_import_pubkey_blob(ssh_session session,
                                           const ssh_string key_blob,
                                           ssh_key *pkey);
+LIBSSH_API int ssh_pki_import_pubkey_file(ssh_session session,
+                                          const char *filename,
+                                          ssh_key *pkey);
 LIBSSH_API ssh_string ssh_pki_publickey_to_blob(const ssh_key key);
 
 LIBSSH_API int ssh_userauth_pki_pubkey(ssh_session session, const char *username,
diff --git a/src/pki.c b/src/pki.c
index b87e5e2..bafe7a4 100644
--- a/src/pki.c
+++ b/src/pki.c
@@ -33,6 +33,7 @@
 #include "config.h"
 
 #include <errno.h>
+#include <ctype.h>
 #include <stdio.h>
 #include <fcntl.h>
 #include <sys/stat.h>
@@ -605,6 +606,78 @@ fail:
     return SSH_ERROR;
 }
 
+int ssh_pki_import_pubkey_file(ssh_session session, const char *filename,
+        ssh_key *pkey)
+{
+    enum ssh_keytypes_e type;
+    struct stat sb;
+    char *key_buf, *p;
+    const char *q;
+    FILE *file;
+    off_t size;
+    int rc;
+
+    if (session == NULL || pkey == NULL) {
+        return SSH_ERROR;
+    }
+
+    if (filename == NULL || *filename == '\0') {
+        return SSH_ERROR;
+    }
+
+    rc = stat(filename, &sb);
+    if (rc < 0) {
+        ssh_set_error(session, SSH_REQUEST_DENIED,
+                      "Error gettint stat of %s: %s",
+                      filename, strerror(errno));
+        return SSH_ERROR;
+    }
+
+    file = fopen(filename, "r");
+    if (file == NULL) {
+        ssh_set_error(session, SSH_REQUEST_DENIED,
+                      "Error opening %s: %s",
+                      filename, strerror(errno));
+        return SSH_ERROR;
+    }
+
+    key_buf = malloc(sb.st_size + 1);
+    if (key_buf == NULL) {
+        fclose(file);
+        ssh_set_error_oom(session);
+        return SSH_ERROR;
+    }
+
+    size = fread(key_buf, 1, sb.st_size, file);
+    fclose(file);
+
+    if (size != sb.st_size) {
+        SAFE_FREE(key_buf);
+        ssh_set_error(session, SSH_FATAL,
+                      "Error reading %s: %s",
+                      filename, strerror(errno));
+        return SSH_ERROR;
+    }
+
+    q = p = key_buf;
+    while (!isspace((int)*p)) p++;
+    *p = '\0';
+
+    type = ssh_key_type_from_name(q);
+    if (type == SSH_KEYTYPE_UNKNOWN) {
+        SAFE_FREE(key_buf);
+        return SSH_ERROR;
+    }
+    q = ++p;
+    while (!isspace((int)*p)) p++;
+    *p = '\0';
+
+    rc = ssh_pki_import_pubkey_base64(session, q, type, pkey);
+    SAFE_FREE(key_buf);
+
+    return rc;
+}
+
 /**
  * @brief Generate and duplicate a public key from a private key.
  *