debug_new.h

Go to the documentation of this file.

// -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
// vim:tabstop=4:shiftwidth=4:expandtab:
 
/*
 * Copyright (C) 2004-2015 Wu Yongwei <adah at users dot sourceforge dot net>
 *
 * This software is provided 'as-is', without any express or implied
 * warranty. In no event will the authors be held liable for any
 * damages arising from the use of this software.
 *
 * Permission is granted to anyone to use this software for any purpose,
 * including commercial applications, and to alter it and redistribute
 * it freely, subject to the following restrictions:
 *
 * 1. The origin of this software must not be misrepresented; you must
 * not claim that you wrote the original software. If you use this
 * software in a product, an acknowledgement in the product
 * documentation would be appreciated but is not required.
 * 2. Altered source versions must be plainly marked as such, and must
 * not be misrepresented as being the original software.
 * 3. This notice may not be removed or altered from any source
 * distribution.
 *
 * This file is part of Stones of Nvwa:
 * http://sourceforge.net/projects/nvwa
 *
 */
 
#ifndef NVWA_DEBUG_NEW_H
#define NVWA_DEBUG_NEW_H
 
#include <new> // size_t/std::bad_alloc
#include <stdio.h> // FILE
#include "debug/nvwa/_nvwa.h" // NVWA_NAMESPACE_*
#include "debug/nvwa/c++11.h" // _NOEXCEPT
 
/* Special allocation/deallocation functions in the global scope */
void* operator new(size_t size, const char* file, int line);
void* operator new[](size_t size, const char* file, int line);
void operator delete(void* ptr, const char* file, int line) _NOEXCEPT;
void operator delete[](void* ptr, const char* file, int line) _NOEXCEPT;
 
NVWA_NAMESPACE_BEGIN
 
#ifndef _DEBUG_NEW_REDEFINE_NEW
#define _DEBUG_NEW_REDEFINE_NEW 1
#endif
 
#ifndef _DEBUG_NEW_TYPE
#define _DEBUG_NEW_TYPE 1
#endif
 
typedef void (*stacktrace_print_callback_t)(FILE* fp, void** stacktrace);
 
typedef bool (*leak_whitelist_callback_t)(char const* file, int line,
 void* addr, void** stacktrace);
 
/* Prototypes */
int check_leaks();
int check_mem_corruption();
 
/* Control variables */
extern bool new_autocheck_flag; // default to true: call check_leaks() on exit
extern bool new_verbose_flag; // default to false: no verbose information
extern FILE* new_output_fp; // default to stderr: output to console
extern const char* new_progname;// default to null; should be assigned argv[0]
extern stacktrace_print_callback_t stacktrace_print_callback;// default to null
extern leak_whitelist_callback_t leak_whitelist_callback; // default to null
 
# if _DEBUG_NEW_TYPE == 1
# define DEBUG_NEW NVWA::debug_new_recorder(__FILE__, __LINE__) ->* new
# else
# define DEBUG_NEW new(__FILE__, __LINE__)
# endif
 
# if _DEBUG_NEW_REDEFINE_NEW
# define new DEBUG_NEW
# endif
# ifdef _DEBUG_NEW_EMULATE_MALLOC
# include <stdlib.h>
# ifdef new
# define malloc(s) ((void*)(new char[s]))
# else
# define malloc(s) ((void*)(DEBUG_NEW char[s]))
# endif
# define free(p) delete[] (char*)(p)
# endif
 
class debug_new_recorder
{
 const char* _M_file;
 const int _M_line;
 void _M_process(void* ptr);
public:
 debug_new_recorder(const char* file, int line)
 : _M_file(file), _M_line(line) {}
 template <class _Tp> _Tp* operator->*(_Tp* ptr)
 { _M_process(ptr); return ptr; }
private:
 debug_new_recorder(const debug_new_recorder&);
 debug_new_recorder& operator=(const debug_new_recorder&);
};
 
class debug_new_counter
{
 static int _S_count;
public:
 debug_new_counter();
 ~debug_new_counter();
};
static debug_new_counter __debug_new_count;
 
NVWA_NAMESPACE_END
 
#endif // NVWA_DEBUG_NEW_H