lkml.org 
[lkml]   [2019]   [Oct]   [28]   [last100]   RSS Feed
Views: [wrap][no wrap]   [headers]  [forward] 
 
Messages in this thread
Patch in this message
/
From
Subject[PATCH] perf kmem: Fix memory leak in __cmd_record()
Date
There are memory leaks in __cmd_record() found by visual inspection.
calloc() and strdup() are used to allocate memory for rec_argv pointer
array and argv, which should be freed before the function returns.

In addition, checking whether strdup() is success or not, if failure,
then go to the failure path to free memory and return the function.

Fixes: ba77c9e11111 ("perf: Add 'perf kmem' tool")
Signed-off-by: Yunfeng Ye <yeyunfeng@huawei.com>
---
tools/perf/builtin-kmem.c | 17 ++++++++++++++---
1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/tools/perf/builtin-kmem.c b/tools/perf/builtin-kmem.c
index 9661671cc26e..6a62acfc9470 100644
--- a/tools/perf/builtin-kmem.c
+++ b/tools/perf/builtin-kmem.c
@@ -1847,6 +1847,7 @@ static int __cmd_record(int argc, const char **argv)
};
unsigned int rec_argc, i, j;
const char **rec_argv;
+ int ret = -ENOMEM;

rec_argc = ARRAY_SIZE(record_args) + argc - 1;
if (kmem_slab)
@@ -1873,10 +1874,20 @@ static int __cmd_record(int argc, const char **argv)
rec_argv[i] = strdup(page_events[j]);
}

- for (j = 1; j < (unsigned int)argc; j++, i++)
- rec_argv[i] = argv[j];
+ for (j = 0; j < i; j++)
+ if (!rec_argv[j]) /* check strdup success or not */
+ goto out;
+
+ rec_argc = i;
+ for (j = 1; j < (unsigned int)argc; j++, rec_argc++)
+ rec_argv[rec_argc] = argv[j];

- return cmd_record(i, rec_argv);
+ ret = cmd_record(rec_argc, rec_argv);
+out:
+ for (i--; (int)i >= 0; i--)
+ free((void *)rec_argv[i]);
+ free(rec_argv);
+ return ret;
}

static int kmem_config(const char *var, const char *value, void *cb __maybe_unused)
--
2.7.4
\
 
 \ /
  Last update: 2019-10-28 07:48    [W:0.024 / U:0.192 seconds]
©2003-2020 Jasper Spaans|hosted at Digital Ocean and TransIP|Read the blog|Advertise on this site