get_string*: fix a memory leak
authorJonathan Neuschäfer <j.neuschaefer@gmx.net>
Tue, 19 Jul 2011 14:06:56 +0000 (16:06 +0200)
committerGreg Kroah-Hartman <gregkh@suse.de>
Sat, 22 Oct 2011 16:11:59 +0000 (18:11 +0200)
Signed-off-by: Greg Kroah-Hartman <gregkh@suse.de>
bti.c

diff --git a/bti.c b/bti.c
index db9cadc..6f77ddc 100644 (file)
--- a/bti.c
+++ b/bti.c
@@ -100,8 +100,10 @@ static char *get_string(const char *name)
                exit(1);
        if (name != NULL)
                fprintf(stdout, "%s", name);
-       if (!fgets(string, 999, stdin))
+       if (!fgets(string, 999, stdin)) {
+               free(string);
                return NULL;
+       }
        temp = strchr(string, '\n');
        if (temp)
                *temp = '\0';
@@ -835,8 +837,10 @@ static char *get_string_from_stdin(void)
        if (!string)
                return NULL;
 
-       if (!fgets(string, 999, stdin))
+       if (!fgets(string, 999, stdin)) {
+               free(string);
                return NULL;
+       }
        temp = strchr(string, '\n');
        if (temp)
                *temp = '\0';