Changeset - r18056:a3aa5f0ca569
[Not reviewed]
master
0 3 0
frosch - 13 years ago 2011-09-03 12:45:45
frosch@openttd.org
(svn r22881) -Fix (r22875): GCC warnings on 64bit systems.
3 files changed with 28 insertions and 4 deletions:
0 comments (0 inline, 0 general)
src/core/alloc_func.hpp
Show inline comments
 
@@ -23,6 +23,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 <typename T>
 
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
 
 * type.
 
@@ -43,7 +66,7 @@ static FORCEINLINE T *MallocT(size_t num
 
	if (num_elements == 0) return NULL;
 

	
 
	/* Ensure the size does not overflow. */
 
	if (num_elements > SIZE_MAX / sizeof(T)) MallocError(SIZE_MAX);
 
	CheckAllocationConstraints<T>(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,
 
	}
 

	
 
	/* Ensure the size does not overflow. */
 
	if (num_elements > SIZE_MAX / sizeof(T)) MallocError(SIZE_MAX);
 
	CheckAllocationConstraints<T>(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,
 

	
 
/** 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<T>(num_elements), \
 
	(T*)alloca((num_elements) * sizeof(T)))
 

	
 
#endif /* ALLOC_FUNC_HPP */
src/pathfinder/npf/queue.cpp
Show inline comments
 
@@ -235,7 +235,7 @@ void Hash::Init(Hash_HashProc *hash, uin
 
	uint i;
 

	
 
	/* Ensure the size won't overflow. */
 
	assert(num_buckets < SIZE_MAX / (sizeof(*this->buckets) + sizeof(*this->buckets_in_use)));
 
	CheckAllocationConstraints(sizeof(*this->buckets) + sizeof(*this->buckets_in_use), num_buckets);
 

	
 
	this->hash = hash;
 
	this->size = 0;
src/stdafx.h
Show inline comments
 
@@ -339,6 +339,7 @@ assert_compile(sizeof(uint64) == 8);
 
assert_compile(sizeof(uint32) == 4);
 
assert_compile(sizeof(uint16) == 2);
 
assert_compile(sizeof(uint8)  == 1);
 
assert_compile(SIZE_MAX >= UINT32_MAX);
 

	
 
#ifndef M_PI_2
 
#define M_PI_2 1.57079632679489661923
0 comments (0 inline, 0 general)