X-Git-Url: https://git.tokkee.org/?p=sysdb.git;a=blobdiff_plain;f=src%2Ftools%2Fsysdb%2Fcommand.c;h=9dfb6194d40b9db217ab66a519e3d6995bf69c8b;hp=505124cae243b8e974c1790f8d04290e38f6c4c4;hb=0b865cb35c5f74e96c1cf984e5c9cff57efd8269;hpb=7f87e1631daf2ee4e217260c1ced2ab9b76cb457

diff --git a/src/tools/sysdb/command.c b/src/tools/sysdb/command.c
index 505124c..9dfb619 100644
--- a/src/tools/sysdb/command.c
+++ b/src/tools/sysdb/command.c
@@ -40,6 +40,7 @@
 
 #include <assert.h>
 #include <ctype.h>
+#include <stdlib.h>
 #include <string.h>
 
 /*
@@ -117,8 +118,11 @@ sdb_command_exec(sdb_input_t *input)
 		 * sends back. We'll wait for the first reply and then return to the
 		 * main loop which will handle any subsequent replies, including
 		 * eventually the reply to the query (if it's not the first reply). */
-		if (sdb_command_print_reply(input->client) < 0)
+		if (sdb_command_print_reply(input->client) < 0) {
+			if (data)
+				free(data);
 			return NULL;
+		}
 	}
 
 	sdb_strbuf_skip(input->input, 0, input->query_len);