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/getinmemory.c | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'docs/examples/getinmemory.c') diff --git a/docs/examples/getinmemory.c b/docs/examples/getinmemory.c index f5e8942f8..a21a2aafc 100644 --- a/docs/examples/getinmemory.c +++ b/docs/examples/getinmemory.c @@ -5,7 +5,7 @@ * | (__| |_| | _ <| |___ * \___|\___/|_| \_\_____| * - * Copyright (C) 1998 - 2015, 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 @@ -42,13 +42,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 == NULL) { /* 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