pki: Fixed a possible crash.

1 files changed, 6 insertions, 3 deletions

diff --git a/src/pki.c b/src/pki.c
index ef925dd..fc15a2c 100644
--- a/src/pki.c
+++ b/src/pki.c
@@ -34,10 +34,13 @@
 
 /**
  * @brief creates a new empty SSH key
- * @returns an empty ssh_key handle
+ * @returns an empty ssh_key handle, or NULL on error.
  */
-ssh_key ssh_key_new (void){
-  ssh_key ptr=malloc (sizeof (struct ssh_key_struct));
+ssh_key ssh_key_new (void) {
+  ssh_key ptr = malloc (sizeof (struct ssh_key_struct));
+  if (ptr == NULL) {
+      return NULL;
+  }
   ZERO_STRUCTP(ptr);
   return ptr;
 }