diff --git a/client/cmdflashmem.c b/client/cmdflashmem.c index af8584ce7..2a9709ef0 100644 --- a/client/cmdflashmem.c +++ b/client/cmdflashmem.c @@ -260,9 +260,12 @@ int CmdFlashMemLoad(const char *Cmd) { break; } - data = realloc(data, datalen); - if (!data) { + uint8_t *newdata = realloc(data, datalen); + if (newdata == NULL) { + free(data); return 1; + } else { + data = newdata; } //Send to device