# HG changeset patch # User rubidium # Date 2007-12-08 14:50:41 # Node ID 0f84abbbbbf68096feec88279bf3041e8d086bbb # Parent ed052f4e2a3a23d7832d35d9c4b02661fe129a89 (svn r11597) -Change: replace all remaining instances of (re|m|c)alloc with (Re|M|C)allocT and add a check for out-of-memory situations to the *allocT functions. diff --git a/src/bmp.cpp b/src/bmp.cpp --- a/src/bmp.cpp +++ b/src/bmp.cpp @@ -354,7 +354,7 @@ bool BmpReadBitmap(BmpBuffer *buffer, Bm { assert(info != NULL && data != NULL); - data->bitmap = (byte*)calloc(info->width * info->height, ((info->bpp == 24) ? 3 : 1) * sizeof(byte)); + data->bitmap = CallocT(info->width * info->height * ((info->bpp == 24) ? 3 : 1)); if (data->bitmap == NULL) return false; /* Load image */ diff --git a/src/fontcache.cpp b/src/fontcache.cpp --- a/src/fontcache.cpp +++ b/src/fontcache.cpp @@ -365,7 +365,7 @@ static void SetGlyphPtr(FontSize size, W void *AllocateFont(size_t size) { - return malloc(size); + return MallocT(size); } diff --git a/src/helpers.hpp b/src/helpers.hpp --- a/src/helpers.hpp +++ b/src/helpers.hpp @@ -12,6 +12,7 @@ template FORCEINLINE T* MallocT(size_t num_elements) { T *t_ptr = (T*)malloc(num_elements * sizeof(T)); + if (t_ptr == NULL && num_elements != 0) error("Out of memory. Cannot allocate %i bytes", num_elements * sizeof(T)); return t_ptr; } /** When allocating using malloc/calloc in C++ it is usually needed to cast the return value @@ -19,6 +20,7 @@ template FORCEINLINE T* Mal template FORCEINLINE T* CallocT(size_t num_elements) { T *t_ptr = (T*)calloc(num_elements, sizeof(T)); + if (t_ptr == NULL && num_elements != 0) error("Out of memory. Cannot allocate %i bytes", num_elements * sizeof(T)); return t_ptr; } /** When allocating using malloc/calloc in C++ it is usually needed to cast the return value @@ -26,6 +28,7 @@ template FORCEINLINE T* Cal template FORCEINLINE T* ReallocT(T* t_ptr, size_t num_elements) { t_ptr = (T*)realloc(t_ptr, num_elements * sizeof(T)); + if (t_ptr == NULL && num_elements != 0) error("Out of memory. Cannot reallocate %i bytes", num_elements * sizeof(T)); return t_ptr; } diff --git a/src/misc/blob.hpp b/src/misc/blob.hpp --- a/src/misc/blob.hpp +++ b/src/misc/blob.hpp @@ -298,7 +298,7 @@ public: /** all allocation should happen here */ static FORCEINLINE CHdr* RawAlloc(bsize_t num_bytes) { - return (CHdr*)malloc(num_bytes); + return (CHdr*)MallocT(num_bytes); } /** all deallocations should happen here */ diff --git a/src/misc/fixedsizearray.hpp b/src/misc/fixedsizearray.hpp --- a/src/misc/fixedsizearray.hpp +++ b/src/misc/fixedsizearray.hpp @@ -34,7 +34,7 @@ struct CFixedSizeArrayT { CFixedSizeArrayT() { // allocate block for header + items (don't construct items) - m_items = (Titem*)(((int8*)malloc(ThdrSize + Tcapacity * sizeof(Titem))) + ThdrSize); + m_items = (Titem*)((MallocT(ThdrSize + Tcapacity * sizeof(Titem))) + ThdrSize); SizeRef() = 0; // initial number of items RefCnt() = 1; // initial reference counter } diff --git a/src/queue.cpp b/src/queue.cpp --- a/src/queue.cpp +++ b/src/queue.cpp @@ -310,7 +310,7 @@ void init_Hash(Hash* h, Hash_HashProc* h h->hash = hash; h->size = 0; h->num_buckets = num_buckets; - h->buckets = (HashNode*)malloc(num_buckets * (sizeof(*h->buckets) + sizeof(*h->buckets_in_use))); + h->buckets = (HashNode*)MallocT(num_buckets * (sizeof(*h->buckets) + sizeof(*h->buckets_in_use))); #ifdef HASH_DEBUG debug("Buckets = %p", h->buckets); #endif diff --git a/src/saveload.cpp b/src/saveload.cpp --- a/src/saveload.cpp +++ b/src/saveload.cpp @@ -575,7 +575,7 @@ static void SlString(void *ptr, size_t l if (len == 0) { *(char**)ptr = NULL; } else { - *(char**)ptr = (char*)malloc(len + 1); // terminating '\0' + *(char**)ptr = MallocT(len + 1); // terminating '\0' ptr = *(char**)ptr; SlCopyBytes(ptr, len); } @@ -1060,7 +1060,7 @@ static void WriteLZO(uint size) static bool InitLZO() { _sl.bufsize = LZO_SIZE; - _sl.buf = _sl.buf_ori = (byte*)malloc(LZO_SIZE); + _sl.buf = _sl.buf_ori = MallocT(LZO_SIZE); return true; } @@ -1085,7 +1085,7 @@ static void WriteNoComp(uint size) static bool InitNoComp() { _sl.bufsize = LZO_SIZE; - _sl.buf = _sl.buf_ori = (byte*)malloc(LZO_SIZE); + _sl.buf = _sl.buf_ori = MallocT(LZO_SIZE); return true; } @@ -1154,7 +1154,7 @@ static bool InitReadZlib() if (inflateInit(&_z) != Z_OK) return false; _sl.bufsize = 4096; - _sl.buf = _sl.buf_ori = (byte*)malloc(4096 + 4096); // also contains fread buffer + _sl.buf = _sl.buf_ori = MallocT(4096 + 4096); // also contains fread buffer return true; } @@ -1194,7 +1194,7 @@ static bool InitWriteZlib() if (deflateInit(&_z, 6) != Z_OK) return false; _sl.bufsize = 4096; - _sl.buf = _sl.buf_ori = (byte*)malloc(4096); // also contains fread buffer + _sl.buf = _sl.buf_ori = MallocT(4096); // also contains fread buffer return true; } diff --git a/src/settings.cpp b/src/settings.cpp --- a/src/settings.cpp +++ b/src/settings.cpp @@ -80,7 +80,7 @@ static SettingsMemoryPool *pool_new(uint SettingsMemoryPool *p; if (minsize < 4096 - 12) minsize = 4096 - 12; - p = (SettingsMemoryPool*)malloc(sizeof(SettingsMemoryPool) - 1 + minsize); + p = (SettingsMemoryPool*)MallocT(sizeof(SettingsMemoryPool) - 1 + minsize); p->pos = 0; p->size = minsize; p->next = NULL; diff --git a/src/spritecache.cpp b/src/spritecache.cpp --- a/src/spritecache.cpp +++ b/src/spritecache.cpp @@ -491,7 +491,7 @@ const void *GetRawSprite(SpriteID sprite void GfxInitSpriteMem() { /* initialize sprite cache heap */ - if (_spritecache_ptr == NULL) _spritecache_ptr = (MemBlock*)malloc(_sprite_cache_size * 1024 * 1024); + if (_spritecache_ptr == NULL) _spritecache_ptr = (MemBlock*)MallocT(_sprite_cache_size * 1024 * 1024); /* A big free block */ _spritecache_ptr->size = ((_sprite_cache_size * 1024 * 1024) - sizeof(MemBlock)) | S_FREE_MASK; diff --git a/src/spriteloader/png.cpp b/src/spriteloader/png.cpp --- a/src/spriteloader/png.cpp +++ b/src/spriteloader/png.cpp @@ -144,7 +144,7 @@ static bool LoadPNG(SpriteLoader::Sprite pixelsize = sizeof(uint8); } - row_pointer = (png_byte *)malloc(info_ptr->width * pixelsize); + row_pointer = (png_byte *)MallocT(info_ptr->width * pixelsize); if (row_pointer == NULL) { png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); return false; diff --git a/src/texteff.cpp b/src/texteff.cpp --- a/src/texteff.cpp +++ b/src/texteff.cpp @@ -292,7 +292,7 @@ TextEffectID AddTextEffect(StringID msg, /* If there is none found, we grow the array */ if (i == _num_text_effects) { _num_text_effects += 25; - _text_effect_list = (TextEffect*) realloc(_text_effect_list, _num_text_effects * sizeof(TextEffect)); + _text_effect_list = ReallocT(_text_effect_list, _num_text_effects); for (; i < _num_text_effects; i++) _text_effect_list[i].string_id = INVALID_STRING_ID; i = _num_text_effects - 1; } diff --git a/src/video/dedicated_v.cpp b/src/video/dedicated_v.cpp --- a/src/video/dedicated_v.cpp +++ b/src/video/dedicated_v.cpp @@ -133,7 +133,7 @@ const char *VideoDriver_Dedicated::Start { int bpp = BlitterFactoryBase::GetCurrentBlitter()->GetScreenDepth(); if (bpp == 0) _dedicated_video_mem = NULL; - else _dedicated_video_mem = malloc(_cur_resolution[0] * _cur_resolution[1] * (bpp / 8)); + else _dedicated_video_mem = MallocT(_cur_resolution[0] * _cur_resolution[1] * (bpp / 8)); _screen.width = _screen.pitch = _cur_resolution[0]; _screen.height = _cur_resolution[1]; diff --git a/src/win32.cpp b/src/win32.cpp --- a/src/win32.cpp +++ b/src/win32.cpp @@ -322,7 +322,7 @@ static void SubmitFile(HWND wnd, const T size = GetFileSize(h, NULL); if (size > 500000) goto error1; - mem = malloc(size); + mem = MallocT(size); if (mem == NULL) goto error1; if (!ReadFile(h, mem, size, &read, NULL) || read != size) goto error2;