dxx-rebirth/common/include/u_mem.h

145 lines
4.8 KiB
C++

/*
* Portions of this file are copyright Rebirth contributors and licensed as
* described in COPYING.txt.
* Portions of this file are copyright Parallax Software and licensed
* according to the Parallax license below.
* See COPYING.txt for license details.
THE COMPUTER CODE CONTAINED HEREIN IS THE SOLE PROPERTY OF PARALLAX
SOFTWARE CORPORATION ("PARALLAX"). PARALLAX, IN DISTRIBUTING THE CODE TO
END-USERS, AND SUBJECT TO ALL OF THE TERMS AND CONDITIONS HEREIN, GRANTS A
ROYALTY-FREE, PERPETUAL LICENSE TO SUCH END-USERS FOR USE BY SUCH END-USERS
IN USING, DISPLAYING, AND CREATING DERIVATIVE WORKS THEREOF, SO LONG AS
SUCH USE, DISPLAY OR CREATION IS FOR NON-COMMERCIAL, ROYALTY OR REVENUE
FREE PURPOSES. IN NO EVENT SHALL THE END-USER USE THE COMPUTER CODE
CONTAINED HEREIN FOR REVENUE-BEARING PURPOSES. THE END-USER UNDERSTANDS
AND AGREES TO THE TERMS HEREIN AND ACCEPTS THE SAME BY USE OF THIS FILE.
COPYRIGHT 1993-1999 PARALLAX SOFTWARE CORPORATION. ALL RIGHTS RESERVED.
*/
#ifndef _U_MEM_H
#define _U_MEM_H
#include <stdlib.h>
#ifdef __cplusplus
#include "dxxsconf.h"
#include "compiler-exchange.h"
#include "compiler-type_traits.h"
#define MEM_K 1.5 // Dynamic array growth factor
#ifdef DEBUG_MEMORY_ALLOCATIONS
void mem_init(void);
void mem_display_blocks();
void * mem_malloc( unsigned int size, const char * var, const char * file, unsigned line) __attribute_alloc_size(1) __attribute_malloc();
void * mem_calloc( size_t nmemb, size_t size, const char * var, const char * filename, unsigned line) __attribute_alloc_size(1,2) __attribute_malloc();
void * mem_realloc( void * buffer, unsigned int size, const char * var, const char * file, int line ) __attribute_alloc_size(2);
extern void mem_free( void * buffer );
/* DPH: Changed malloc, etc. to d_malloc. Overloading system calls is very evil and error prone */
// Checks to see if any blocks are overwritten
void mem_validate_heap();
#define mem_calloc(nmemb,size,var,file,line) ((mem_calloc)((size) ? (nmemb) : 0, (size), (var), (file), (line)))
#else
#define mem_malloc(size,var,file,line) malloc((size))
#define mem_calloc(nmemb,size,var,file,line) calloc((nmemb), (size))
#define mem_realloc(ptr,size,var,file,line) realloc((ptr),(size))
#define mem_free free
static inline void mem_init(void)
{
}
#endif
template <typename T>
T *MALLOC(T *&r, std::size_t count, const char *var, const char *file, unsigned line)
{
static_assert(tt::is_pod<T>::value, "MALLOC cannot allocate non-POD");
return r = reinterpret_cast<T *>(mem_malloc(count, var, file, line));
}
template <typename T>
T *CALLOC(T *&r, std::size_t count, const char *var, const char *file, unsigned line)
{
static_assert(tt::is_pod<T>::value, "CALLOC cannot allocate non-POD");
return r = reinterpret_cast<T *>(mem_calloc(count, sizeof(T), var, file, line));
}
#define d_malloc(size) mem_malloc((size),"Unknown", __FILE__,__LINE__ )
#define d_calloc(nmemb,size) mem_calloc((nmemb),(size),"Unknown", __FILE__,__LINE__ )
#define d_realloc(ptr,size) mem_realloc((ptr),(size),"Unknown", __FILE__,__LINE__ )
template <typename T>
static inline void d_free(T *&ptr)
{
static_assert(tt::is_same<T, void>::value || tt::is_pod<T>::value, "d_free cannot free non-POD");
T *t = ptr;
ptr = NULL;
mem_free(t);
}
class BaseRAIIdmem
{
protected:
BaseRAIIdmem(const BaseRAIIdmem&) = delete;
BaseRAIIdmem& operator=(const BaseRAIIdmem&) = delete;
BaseRAIIdmem(BaseRAIIdmem &&that) : p(exchange(that.p, nullptr)) {}
BaseRAIIdmem& operator=(BaseRAIIdmem &&rhs)
{
std::swap(p, rhs.p);
return *this;
}
void *p;
BaseRAIIdmem(void *v = nullptr) : p(v) {}
~BaseRAIIdmem() {
#ifdef DEBUG_MEMORY_ALLOCATIONS
if (p) // Avoid bogus warning about freeing NULL
#endif
d_free(p);
}
};
template <typename T>
struct RAIIdmem : BaseRAIIdmem
{
static_assert(tt::is_pod<T>::value, "RAIIdmem cannot manage non-POD");
RAIIdmem() = default;
RAIIdmem(std::nullptr_t) : BaseRAIIdmem(nullptr) {}
explicit RAIIdmem(T *v) : BaseRAIIdmem(v) {}
#ifdef DXX_HAVE_CXX11_REF_QUALIFIER
operator T*() const & { return static_cast<T*>(p); }
operator T*() const && = delete;
#else
operator T*() const { return static_cast<T*>(p); }
#endif
T *operator->() const { return static_cast<T*>(p); }
};
template <typename T>
T *MALLOC(RAIIdmem<T> &r, std::size_t count, const char *var, const char *file, unsigned line)
{
T *p;
return r = RAIIdmem<T>(MALLOC<T>(p, count, var, file, line));
}
template <typename T>
T *CALLOC(RAIIdmem<T> &r, std::size_t count, const char *var, const char *file, unsigned line)
{
T *p;
return r = RAIIdmem<T>(CALLOC<T>(p, count, var, file, line));
}
#define MALLOC( var, type, count ) (MALLOC<type>(var, (count)*sizeof(type),#var, __FILE__,__LINE__ ))
#define CALLOC( var, type, count ) (CALLOC<type>(var, (count),#var, __FILE__,__LINE__ ))
typedef RAIIdmem<unsigned char> RAIIdubyte;
typedef RAIIdmem<char> RAIIdchar;
#endif
#endif // _U_MEM_H