From 39a27d187278fc2d0cc05ef2ad59a872c4f19b2b Mon Sep 17 00:00:00 2001 From: Christian Heller Date: Fri, 21 Nov 2014 00:29:55 +0100 Subject: [PATCH] Client: Add size test to array_append(). --- src/client/array_append.c | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/client/array_append.c b/src/client/array_append.c index b2bbab9..52bcf97 100644 --- a/src/client/array_append.c +++ b/src/client/array_append.c @@ -7,9 +7,10 @@ #include "array_append.h" #include /* size_t */ -#include /* uint32_t */ +#include /* uint32_t, UINT32_MAX */ #include /* free() */ #include /* memcpy() */ +#include "../common/rexit.h" /* exit_trouble() */ #include "../common/try_malloc.h" /* try_malloc() */ @@ -17,12 +18,12 @@ extern void array_append(uint32_t old_n, size_t region_size, void * new_region, void ** ptr_old_array) { + exit_trouble(UINT32_MAX<(old_n+1)*region_size, __func__, "too large sizes"); uint32_t old_size = old_n * region_size; uint32_t new_size = old_size + region_size; char * new_array = try_malloc(new_size, __func__); memcpy(new_array, * ptr_old_array, old_size); memcpy(new_array + (old_n * region_size), new_region, region_size); free(* ptr_old_array); - * ptr_old_array = new_array; + *ptr_old_array = new_array; } - -- 2.30.2