From 5ccd091fd1bdf9a92b2424bba2dd63b372df0639 Mon Sep 17 00:00:00 2001 From: frosch Date: Sat, 3 Sep 2011 12:45:45 +0000 Subject: (svn r22881) -Fix (r22875): GCC warnings on 64bit systems. --- src/core/alloc_func.hpp | 29 ++++++++++++++++++++++++++--- 1 file changed, 26 insertions(+), 3 deletions(-) (limited to 'src/core/alloc_func.hpp') diff --git a/src/core/alloc_func.hpp b/src/core/alloc_func.hpp index 7bd09b004..4ba8c5f53 100644 --- a/src/core/alloc_func.hpp +++ b/src/core/alloc_func.hpp @@ -22,6 +22,29 @@ void NORETURN MallocError(size_t size); void NORETURN ReallocError(size_t size); +/** + * Checks whether allocating memory would overflow size_t. + * + * @param element_size Size of the structure to allocate. + * @param num_elements Number of elements to allocate. + */ +static inline void CheckAllocationConstraints(size_t element_size, size_t num_elements) +{ + if (num_elements > SIZE_MAX / element_size) MallocError(SIZE_MAX); +} + +/** + * Checks whether allocating memory would overflow size_t. + * + * @tparam T Structure to allocate. + * @param num_elements Number of elements to allocate. + */ +template +static inline void CheckAllocationConstraints(size_t num_elements) +{ + CheckAllocationConstraints(sizeof(T), num_elements); +} + /** * Simplified allocation function that allocates the specified number of * elements of the given type. It also explicitly casts it to the requested @@ -43,7 +66,7 @@ static FORCEINLINE T *MallocT(size_t num_elements) if (num_elements == 0) return NULL; /* Ensure the size does not overflow. */ - if (num_elements > SIZE_MAX / sizeof(T)) MallocError(SIZE_MAX); + CheckAllocationConstraints(num_elements); T *t_ptr = (T*)malloc(num_elements * sizeof(T)); if (t_ptr == NULL) MallocError(num_elements * sizeof(T)); @@ -100,7 +123,7 @@ static FORCEINLINE T *ReallocT(T *t_ptr, size_t num_elements) } /* Ensure the size does not overflow. */ - if (num_elements > SIZE_MAX / sizeof(T)) MallocError(SIZE_MAX); + CheckAllocationConstraints(num_elements); t_ptr = (T*)realloc(t_ptr, num_elements * sizeof(T)); if (t_ptr == NULL) ReallocError(num_elements * sizeof(T)); @@ -109,7 +132,7 @@ static FORCEINLINE T *ReallocT(T *t_ptr, size_t num_elements) /** alloca() has to be called in the parent function, so define AllocaM() as a macro */ #define AllocaM(T, num_elements) \ - ((num_elements) > SIZE_MAX / sizeof(T) && (MallocError(SIZE_MAX), NULL), \ + (CheckAllocationConstraints(num_elements), \ (T*)alloca((num_elements) * sizeof(T))) #endif /* ALLOC_FUNC_HPP */ -- cgit v1.2.3-54-g00ecf