summaryrefslogtreecommitdiff
path: root/cgpt/cgpt_common.c
diff options
context:
space:
mode:
authorFletcher Woodruff <fletcherw@chromium.org>2019-03-11 13:33:40 -0600
committerchrome-bot <chrome-bot@chromium.org>2019-03-13 21:05:07 -0700
commit2f6381ec0360b96ba836168b4635ceaaf4aff262 (patch)
treeba73fe92a449b3a965d143aed468e491e8914154 /cgpt/cgpt_common.c
parent3bfaab121cbafbd5c6d57004df6784866b8de5de (diff)
downloadvboot-2f6381ec0360b96ba836168b4635ceaaf4aff262.tar.gz
Fix memory leaks in cgpt
DriveOpen mallocs a couple of buffers, but DriveClose only freed them if a particular flag was passed causing GptSave to be called. Move the free calls out of GptSave so that the buffers are always freed, and add DriveClose calls to a couple of cgpt functions that are missing them. BUG=chromium:940543 TEST=precq passes, manual testing with cgpt/valgrind shows that memory leaks for cgpt find, show are fixed. BRANCH=none Change-Id: I58aeddfa6b8b4715ba4f8e064e95a660371a01c9 Reviewed-on: https://chromium-review.googlesource.com/1516413 Commit-Ready: Fletcher Woodruff <fletcherw@chromium.org> Tested-by: Fletcher Woodruff <fletcherw@chromium.org> Reviewed-by: Julius Werner <jwerner@chromium.org> Reviewed-by: Mike Frysinger <vapier@chromium.org>
Diffstat (limited to 'cgpt/cgpt_common.c')
-rw-r--r--cgpt/cgpt_common.c21
1 files changed, 9 insertions, 12 deletions
diff --git a/cgpt/cgpt_common.c b/cgpt/cgpt_common.c
index a5550b4c..c129d92d 100644
--- a/cgpt/cgpt_common.c
+++ b/cgpt/cgpt_common.c
@@ -286,18 +286,6 @@ static int GptSave(struct drive *drive) {
}
}
- if (drive->gpt.primary_header)
- free(drive->gpt.primary_header);
- drive->gpt.primary_header = 0;
- if (drive->gpt.primary_entries)
- free(drive->gpt.primary_entries);
- drive->gpt.primary_entries = 0;
- if (drive->gpt.secondary_header)
- free(drive->gpt.secondary_header);
- drive->gpt.secondary_header = 0;
- if (drive->gpt.secondary_entries)
- free(drive->gpt.secondary_entries);
- drive->gpt.secondary_entries = 0;
return errors ? -1 : 0;
}
@@ -388,6 +376,15 @@ int DriveClose(struct drive *drive, int update_as_needed) {
}
}
+ free(drive->gpt.primary_header);
+ drive->gpt.primary_header = NULL;
+ free(drive->gpt.primary_entries);
+ drive->gpt.primary_entries = NULL;
+ free(drive->gpt.secondary_header);
+ drive->gpt.secondary_header = NULL;
+ free(drive->gpt.secondary_entries);
+ drive->gpt.secondary_entries = NULL;
+
// Sync early! Only sync file descriptor here, and leave the whole system sync
// outside cgpt because whole system sync would trigger tons of disk accesses
// and timeout tests.