From f0a362fdbe525cd938efaf0ff9afab3903e82cdf Mon Sep 17 00:00:00 2001
From: Andreas Schneider <asn@cryptomilk.org>
Date: Wed, 19 Jun 2013 12:19:05 +0200
Subject: agent: Fix possible memory leak.

---
 src/agent.c | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'src/agent.c')

diff --git a/src/agent.c b/src/agent.c
index 95cf6a12..897641b1 100644
--- a/src/agent.c
+++ b/src/agent.c
@@ -287,9 +287,14 @@ int ssh_agent_get_ident_count(struct ssh_session_struct *session) {
 
   /* send message to the agent requesting the list of identities */
   request = ssh_buffer_new();
+  if (request == NULL) {
+      ssh_set_error_oom(request);
+      return -1;
+  }
   if (buffer_add_u8(request, c1) < 0) {
-    ssh_set_error(session, SSH_FATAL, "Not enough space");
-    return -1;
+      ssh_set_error_oom(request);
+      ssh_buffer_free(request);
+      return -1;
   }
 
   reply = ssh_buffer_new();
-- 
cgit v1.2.3