diff mbox

[U-Boot,V2] common: cli_hush: avoid memory leak

Message ID 1448590322-14135-1-git-send-email-Peng.Fan@freescale.com
State Accepted
Delegated to: Tom Rini
Headers show

Commit Message

Peng Fan Nov. 27, 2015, 2:12 a.m. UTC
Need to free memory avoid memory leak, when error.

Signed-off-by: Peng Fan <Peng.Fan@freescale.com>
Reviewed-by: Simon Glass <sjg@chromium.org>
Cc: Simon Glass <sjg@chromium.org>
Cc: Tom Rini <trini@konsulko.com>
---

Changes V2:
 Take Simon's comments, fix another memory leak when child->argv is NULL.
 Add Simon's review tag.

 common/cli_hush.c | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

Comments

Simon Glass Nov. 27, 2015, 2:52 a.m. UTC | #1
On 26 November 2015 at 18:12, Peng Fan <Peng.Fan@freescale.com> wrote:
> Need to free memory avoid memory leak, when error.
>
> Signed-off-by: Peng Fan <Peng.Fan@freescale.com>
> Reviewed-by: Simon Glass <sjg@chromium.org>
> Cc: Simon Glass <sjg@chromium.org>
> Cc: Tom Rini <trini@konsulko.com>
> ---
>
> Changes V2:
>  Take Simon's comments, fix another memory leak when child->argv is NULL.
>  Add Simon's review tag.
>
>  common/cli_hush.c | 9 +++++++--
>  1 file changed, 7 insertions(+), 2 deletions(-)

Looks good!

- Simon
Tom Rini Dec. 6, 2015, 10:07 p.m. UTC | #2
On Fri, Nov 27, 2015 at 10:12:02AM +0800, Peng Fan wrote:

> Need to free memory avoid memory leak, when error.
> 
> Signed-off-by: Peng Fan <Peng.Fan@freescale.com>
> Reviewed-by: Simon Glass <sjg@chromium.org>
> Cc: Simon Glass <sjg@chromium.org>
> Cc: Tom Rini <trini@konsulko.com>

Applied to u-boot/master, thanks!
diff mbox

Patch

diff --git a/common/cli_hush.c b/common/cli_hush.c
index f075459..7f69c06 100644
--- a/common/cli_hush.c
+++ b/common/cli_hush.c
@@ -2471,11 +2471,16 @@  static int done_word(o_string *dest, struct p_context *ctx)
 		}
 		argc = ++child->argc;
 		child->argv = realloc(child->argv, (argc+1)*sizeof(*child->argv));
-		if (child->argv == NULL) return 1;
+		if (child->argv == NULL) {
+			free(str);
+			return 1;
+		}
 		child->argv_nonnull = realloc(child->argv_nonnull,
 					(argc+1)*sizeof(*child->argv_nonnull));
-		if (child->argv_nonnull == NULL)
+		if (child->argv_nonnull == NULL) {
+			free(str);
 			return 1;
+		}
 		child->argv[argc-1]=str;
 		child->argv_nonnull[argc-1] = dest->nonnull;
 		child->argv[argc]=NULL;