summaryrefslogtreecommitdiff
path: root/scripts
diff options
context:
space:
mode:
authorHao Zeng <zenghao@kylinos.cn>2023-04-26 09:05:27 +0800
committerSteven Rostedt (Google) <rostedt@goodmis.org>2023-04-25 21:10:20 -0400
commitfa359d068574d29e7d2f0fdd0ebe4c6a12b5cfb9 (patch)
treeea4ce7f88d231cba56a22c894f820e18e9408d2f /scripts
parent41d8fba193b36ac1208d8f8489390b93675fab7b (diff)
downloadlinux-fa359d068574d29e7d2f0fdd0ebe4c6a12b5cfb9.tar.gz
recordmcount: Fix memory leaks in the uwrite function
Common realloc mistake: 'file_append' nulled but not freed upon failure Link: https://lkml.kernel.org/r/20230426010527.703093-1-zenghao@kylinos.cn Signed-off-by: Hao Zeng <zenghao@kylinos.cn> Suggested-by: Steven Rostedt <rostedt@goodmis.org> Signed-off-by: Steven Rostedt (Google) <rostedt@goodmis.org>
Diffstat (limited to 'scripts')
-rw-r--r--scripts/recordmcount.c6
1 files changed, 5 insertions, 1 deletions
diff --git a/scripts/recordmcount.c b/scripts/recordmcount.c
index e30216525325..40ae6b2c7a6d 100644
--- a/scripts/recordmcount.c
+++ b/scripts/recordmcount.c
@@ -110,6 +110,7 @@ static ssize_t uwrite(void const *const buf, size_t const count)
{
size_t cnt = count;
off_t idx = 0;
+ void *p = NULL;
file_updated = 1;
@@ -117,7 +118,10 @@ static ssize_t uwrite(void const *const buf, size_t const count)
off_t aoffset = (file_ptr + count) - file_end;
if (aoffset > file_append_size) {
- file_append = realloc(file_append, aoffset);
+ p = realloc(file_append, aoffset);
+ if (!p)
+ free(file_append);
+ file_append = p;
file_append_size = aoffset;
}
if (!file_append) {