From 23524bf85b887adbc513bc015c9530355967bc04 Mon Sep 17 00:00:00 2001 From: Kruzya Date: Sat, 15 Sep 2018 08:55:11 +0300 Subject: examples: Fix memory leaks from realloc errors Make sure to not overwrite the reallocated pointer in realloc() calls to avoid a memleak on memory errors. --- docs/examples/postinmemory.c | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'docs/examples/postinmemory.c') diff --git a/docs/examples/postinmemory.c b/docs/examples/postinmemory.c index 488d227be..176f24af3 100644 --- a/docs/examples/postinmemory.c +++ b/docs/examples/postinmemory.c @@ -5,7 +5,7 @@ * | (__| |_| | _ <| |___ * \___|\___/|_| \_\_____| * - * Copyright (C) 1998 - 2017, Daniel Stenberg, , et al. + * Copyright (C) 1998 - 2018, Daniel Stenberg, , et al. * * This software is licensed as described in the file COPYING, which * you should have received as part of this distribution. The terms @@ -39,13 +39,14 @@ WriteMemoryCallback(void *contents, size_t size, size_t nmemb, void *userp) size_t realsize = size * nmemb; struct MemoryStruct *mem = (struct MemoryStruct *)userp; - mem->memory = realloc(mem->memory, mem->size + realsize + 1); - if(mem->memory == NULL) { + char *ptr = realloc(mem->memory, mem->size + realsize + 1); + if(!ptr) { /* out of memory! */ printf("not enough memory (realloc returned NULL)\n"); return 0; } + mem->memory = ptr; memcpy(&(mem->memory[mem->size]), contents, realsize); mem->size += realsize; mem->memory[mem->size] = 0; -- cgit v1.2.3