[<prev] [next>] [thread-next>] [day] [month] [year] [list]
Message-Id: <20230412093048.3005276-1-zenghao@kylinos.cn>
Date: Wed, 12 Apr 2023 17:30:48 +0800
From: Hao Zeng <zenghao@...inos.cn>
To: chenhuacai@...nel.org, rostedt@...dmis.org, zhangqing@...ngson.cn,
linux-kernel@...r.kernel.org
Cc: Hao Zeng <zenghao@...inos.cn>
Subject: [PATCH] recordmcount: Fix memory leaks in the uwrite function
Common realloc mistake: 'file_append' nulled but not freed upon failure
Signed-off-by: Hao Zeng <zenghao@...inos.cn>
---
scripts/recordmcount.c | 17 +++++++++--------
1 file changed, 9 insertions(+), 8 deletions(-)
diff --git a/scripts/recordmcount.c b/scripts/recordmcount.c
index e30216525325..2b7173a86d4c 100644
--- a/scripts/recordmcount.c
+++ b/scripts/recordmcount.c
@@ -110,22 +110,23 @@ 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;
if (file_ptr + count >= file_end) {
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) {
+ perror("write");
+ file_append_cleanup();
+ mmap_cleanup();
+ return -1;
+ }
+ file_append = p;
file_append_size = aoffset;
}
- if (!file_append) {
- perror("write");
- file_append_cleanup();
- mmap_cleanup();
- return -1;
- }
if (file_ptr < file_end) {
cnt = file_end - file_ptr;
} else {
--
2.37.2
No virus found
Checked by Hillstone Network AntiVirus
Powered by blists - more mailing lists