196 lines
6 KiB
C++
196 lines
6 KiB
C++
|
//+---------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Microsoft Windows
|
||
|
|
// Copyright (C) Microsoft Corporation, 1996.
|
||
|
|
//
|
||
|
|
// File: mydebug.hxx
|
||
|
|
//
|
||
|
|
// Contents: Functions to manage the debug heap
|
||
|
|
//
|
||
|
|
// Classes: ( none )
|
||
|
|
//
|
||
|
|
// Functions: void* operator new( size_t )
|
||
|
|
// void* operator new( size_t, const char*, int )
|
||
|
|
// void* my_new( size_t, const char*, int )
|
||
|
|
// void operator delete( void* )
|
||
|
|
//
|
||
|
|
// Coupling:
|
||
|
|
//
|
||
|
|
// Notes: To enable the debug version, include this header file, use the
|
||
|
|
// NEW macro instead of the new keyword, and include the following
|
||
|
|
// line in your sources file:
|
||
|
|
// DEBUG_CRTS=1
|
||
|
|
//
|
||
|
|
// History: 10-20-1996 ericne Created
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
#ifndef _MYDEBUG_
|
||
|
|
|
||
|
|
#define _MYDEBUG_
|
||
|
|
|
||
|
|
#ifdef _DEBUG
|
||
|
|
|
||
|
|
#include <windows.h>
|
||
|
|
#include <crtdbg.h>
|
||
|
|
#include <malloc.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <new.h>
|
||
|
|
|
||
|
|
// This macro strips the path information from a file name
|
||
|
|
#define __MYFILE__ ( strrchr( __FILE__, '\\' ) + 1 )
|
||
|
|
|
||
|
|
//+--------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: my_new
|
||
|
|
//
|
||
|
|
// Synopsis:
|
||
|
|
//
|
||
|
|
// Arguments: [size] --
|
||
|
|
// [file] --
|
||
|
|
// [line] --
|
||
|
|
//
|
||
|
|
// Returns:
|
||
|
|
//
|
||
|
|
// History: 10-20-1996 ericne Created
|
||
|
|
//
|
||
|
|
// Notes:
|
||
|
|
//
|
||
|
|
//---------------------------------------------------------------------
|
||
|
|
|
||
|
|
inline void * _CRTAPI1 my_new( size_t size, const char* file, int line )
|
||
|
|
{
|
||
|
|
|
||
|
|
void *pvoid = NULL;
|
||
|
|
_PNH old_new_handler = NULL;
|
||
|
|
|
||
|
|
do
|
||
|
|
{
|
||
|
|
|
||
|
|
// Check the new handler routine again in case another
|
||
|
|
// thread has changed it
|
||
|
|
old_new_handler = _query_new_handler( );
|
||
|
|
|
||
|
|
// Call _malloc_dbg to allocate space on the debug heap
|
||
|
|
pvoid = _malloc_dbg( size, _NORMAL_BLOCK, file, line );
|
||
|
|
|
||
|
|
} while( ( NULL == pvoid ) &&
|
||
|
|
( 1 != _query_new_mode( ) ) &&
|
||
|
|
( NULL != old_new_handler ) &&
|
||
|
|
( 0 != old_new_handler( size ) ) );
|
||
|
|
|
||
|
|
// Short-circuit evaluation of the above expression prevents the
|
||
|
|
// new handler from being called if old_new_handler is NULL
|
||
|
|
|
||
|
|
// Return the pointer
|
||
|
|
return( pvoid );
|
||
|
|
|
||
|
|
} // my_new
|
||
|
|
|
||
|
|
//+--------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: new
|
||
|
|
//
|
||
|
|
// Synopsis: Calls my_new. This function is needed to shadow the
|
||
|
|
// default global new operator
|
||
|
|
//
|
||
|
|
// Arguments: [size] -- The size of the memory to be allocated
|
||
|
|
//
|
||
|
|
// Returns: void *
|
||
|
|
//
|
||
|
|
// History: 10-20-1996 ericne Created
|
||
|
|
//
|
||
|
|
// Notes:
|
||
|
|
//
|
||
|
|
//---------------------------------------------------------------------
|
||
|
|
|
||
|
|
inline void * _CRTAPI1 operator new( size_t size )
|
||
|
|
{
|
||
|
|
|
||
|
|
return( my_new( size, NULL, 0 ) );
|
||
|
|
|
||
|
|
} // new
|
||
|
|
|
||
|
|
//+--------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: new
|
||
|
|
//
|
||
|
|
// Synopsis: Calls my_new.
|
||
|
|
//
|
||
|
|
// Arguments: [size] -- Size of the user's memory block
|
||
|
|
// [file] -- name of source file requesting the allocation
|
||
|
|
// [line] -- line in source file of allocation request
|
||
|
|
//
|
||
|
|
// Returns:
|
||
|
|
//
|
||
|
|
// History: 10-20-1996 ericne Created
|
||
|
|
//
|
||
|
|
// Notes: If you use the NEW macro, this function will be called
|
||
|
|
// with __MYFILE__ and __LINE__ as the 2nd and 3rd params
|
||
|
|
//
|
||
|
|
//---------------------------------------------------------------------
|
||
|
|
|
||
|
|
inline void * _CRTAPI1 operator new( size_t size,
|
||
|
|
const char* file,
|
||
|
|
int line )
|
||
|
|
{
|
||
|
|
|
||
|
|
return( my_new( size, file, line ) );
|
||
|
|
|
||
|
|
} // new
|
||
|
|
|
||
|
|
//+--------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: delete
|
||
|
|
//
|
||
|
|
// Synopsis: calls _free_dbg to clean up the debug heap
|
||
|
|
//
|
||
|
|
// Arguments: [pMemory] -- pointer to the users memory
|
||
|
|
//
|
||
|
|
// Returns: void
|
||
|
|
//
|
||
|
|
// History: 10-20-1996 ericne Created
|
||
|
|
//
|
||
|
|
// Notes:
|
||
|
|
//
|
||
|
|
//---------------------------------------------------------------------
|
||
|
|
|
||
|
|
inline void _CRTAPI1 operator delete( void *pMemory )
|
||
|
|
{
|
||
|
|
|
||
|
|
// free the memory
|
||
|
|
_free_dbg( pMemory, _NORMAL_BLOCK );
|
||
|
|
|
||
|
|
} // delete
|
||
|
|
|
||
|
|
#define NEW new( __MYFILE__, __LINE__ )
|
||
|
|
|
||
|
|
#define DEBUG_INIT( ) \
|
||
|
|
_CrtSetReportMode( _CRT_WARN, _CRTDBG_MODE_FILE ); \
|
||
|
|
_CrtSetReportFile( _CRT_WARN, _CRTDBG_FILE_STDOUT ); \
|
||
|
|
_CrtSetReportMode( _CRT_ERROR, _CRTDBG_MODE_FILE ); \
|
||
|
|
_CrtSetReportFile( _CRT_ERROR, _CRTDBG_FILE_STDOUT )
|
||
|
|
|
||
|
|
#define SET_CRT_DEBUG_FIELD( field ) \
|
||
|
|
_CrtSetDbgFlag( (field) | _CrtSetDbgFlag( _CRTDBG_REPORT_FLAG ) )
|
||
|
|
|
||
|
|
#define CLEAR_CRT_DEBUG_FIELD( field ) \
|
||
|
|
_CrtSetDbgFlag( ~(field) & _CrtSetDbgFlag( _CRTDBG_REPORT_FLAG ) )
|
||
|
|
|
||
|
|
#else
|
||
|
|
|
||
|
|
#define NEW new
|
||
|
|
|
||
|
|
#define DEBUG_INIT( ) ( (void) 0 )
|
||
|
|
|
||
|
|
#define SET_CRT_DEBUG_FIELD( field ) ( (void) 0 )
|
||
|
|
|
||
|
|
#define CLEAR_CRT_DEBUG_FIELD( field ) ( (void) 0 )
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
|