From 05498e0e33f575948026316b560666213535d5ed Mon Sep 17 00:00:00 2001
From: Andreas Schneider <asn@cryptomilk.org>
Date: Mon, 4 May 2015 16:16:27 +0200
Subject: [PATCH] pki_container: Add check for return value

CID: #1267982

Signed-off-by: Andreas Schneider <asn@cryptomilk.org>
---
 src/pki_container_openssh.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/pki_container_openssh.c b/src/pki_container_openssh.c
index c2aface2..c276f486 100644
--- a/src/pki_container_openssh.c
+++ b/src/pki_container_openssh.c
@@ -484,7 +484,10 @@ static int pki_private_key_encrypt(ssh_buffer privkey_buffer,
         return SSH_ERROR;
     }
     while (ssh_buffer_get_len(privkey_buffer) % cipher.blocksize != 0) {
-        buffer_add_u8(privkey_buffer, padding);
+        rc = buffer_add_u8(privkey_buffer, padding);
+        if (rc < 0) {
+            return SSH_ERROR;
+        }
         padding++;
     }