mem.cpp

来自「开放源码的编译器open watcom 1.6.0版的源代码」· C++ 代码 · 共 268 行

/****************************************************************************
*
*                            Open Watcom Project
*
*    Portions Copyright (c) 1983-2002 Sybase, Inc. All Rights Reserved.
*
*  ========================================================================
*
*    This file contains Original Code and/or Modifications of Original
*    Code as defined in and that are subject to the Sybase Open Watcom
*    Public License version 1.0 (the 'License'). You may not use this file
*    except in compliance with the License. BY USING THIS FILE YOU AGREE TO
*    ALL TERMS AND CONDITIONS OF THE LICENSE. A copy of the License is
*    provided with the Original Code and Modifications, and is also
*    available at www.sybase.com/developer/opensource.
*
*    The Original Code and all software distributed under the License are
*    distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
*    EXPRESS OR IMPLIED, AND SYBASE AND ALL CONTRIBUTORS HEREBY DISCLAIM
*    ALL SUCH WARRANTIES, INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF
*    MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR
*    NON-INFRINGEMENT. Please see the License for the specific language
*    governing rights and limitations under the License.
*
*  ========================================================================
*
* Description:  WHEN YOU FIGURE OUT WHAT THIS FILE DOES, PLEASE
*               DESCRIBE IT HERE!
*
****************************************************************************/


#include <malloc.h>

#ifndef STANDALONE_MERGER
#  include <dr.h>
#endif

#include "death.h"
#include "mem.h"
#include "debuglog.h"

#ifdef TRACKER
extern "C" {
#include "trmem.h"

#ifndef MINALLOC
#define MINALLOC 0
#endif
}

#ifndef STANDALONE_MERGER
#include <wmsgdlg.hpp>
#endif

    _trmem_hdl TrHdl;

    #pragma initialize 40;

    class Memory : public DebuggingLog
    {
    public:
            Memory();
            ~Memory();

            unsigned    _numMessages;
    };

    static Memory bogus;        // just need to get the ctor's called

    void PrintLine( void *, const char *buf, size_t len )
    {
        bogus.write( buf, len );
        bogus._numMessages++;
    }
#endif

void *operator new( size_t size )
//-------------------------------
{
    void *p;
#ifdef TRACKER
    _trmem_who  caller;

    caller = _trmem_guess_who();
#endif

    for(;;) {
#ifdef TRACKER
        p = _trmem_alloc( size, _trmem_guess_who(), TrHdl );
#else
        p = malloc( size );
#endif
        if( p != NULL || size == 0 ) break;     // successful allocation

#ifndef STANDALONE_MERGER
            if( !DRSwap() ) break;  // dwarf managed to swap out, so try again
#else
            break;                  // no recovery we can do, throw except and die.
#endif
    }
    if( p == NULL && size != 0 ) throw DEATH_BY_OUT_OF_MEMORY;
    return p;
}

extern "C" {

void * WBRAlloc( size_t size )
//----------------------------
// note: code directly cloned from above since we need to be able to trace
// calling functions when the memory tracker is in.
{
    void *p;
#ifdef TRACKER
    _trmem_who  caller;

    caller = _trmem_guess_who();
#endif

    for(;;) {
#ifdef TRACKER
        p = _trmem_alloc( size, _trmem_guess_who(), TrHdl );
#else
        p = malloc( size );
#endif
        if( p != NULL || size == 0 ) break;   // successful allocation

#ifndef STANDALONE_MERGER
            if( !DRSwap() ) break;  // dwarf managed to swap out, so try again
#else
            break;                  // no recovery we can do, throw except and die.
#endif
    }
    if( p == NULL && size != 0 ) throw DEATH_BY_OUT_OF_MEMORY;
    return p;
}

void * WBRRealloc( void * p, size_t size )
//----------------------------------------
// note: code cloned from above since we need to be able to trace
// calling functions when the memory tracker is in.
{
#ifdef TRACKER
    _trmem_who  caller;

    caller = _trmem_guess_who();
#endif

    for(;;) {
#ifdef TRACKER
        p = _trmem_realloc( p, size, caller, TrHdl );
#else
        p = realloc( p, size );
#endif
        if( p != NULL || size == 0 ) break;         // successful allocation

#ifndef STANDALONE_MERGER
            if( !DRSwap() ) break;  // dwarf managed to swap out, so try again
#else
            break;                  // no recovery we can do, throw except and die.
#endif
    }
    if( p == NULL && size != 0 ) throw DEATH_BY_OUT_OF_MEMORY;
    return p;
}

void WBRFree( void *p )
//---------------------
{
    if( p == NULL ) return;
#ifdef TRACKER
    _trmem_free( p, _trmem_guess_who(), TrHdl );
#else
    free( p );
#endif
}

#ifndef STANDALONE_MERGER
#if 0
extern void *GUIAlloc( unsigned a )
//---------------------------------
{
    return WBRAlloc( a );
}

extern void *GUIRealloc( void *ptr, unsigned size )
//-------------------------------------------------
{
    return WBRRealloc( ptr, size );
}

extern void GUIFree( void *p )
//----------------------------
{
    WBRFree( p );
}

extern void GUIMemOpen()
{
}

extern void GUIMemClose( void )
{
}

extern void GUIMemPrtUsage( void )
{
}
#endif
#endif // STANDALONE_MERGER

};  // extern "C"


void operator delete( void *p )
/*****************************/
{
    if( p == NULL ) return;
#ifdef TRACKER
    _trmem_free( p, _trmem_guess_who(), TrHdl );
#else
    free( p );
#endif
}

#ifdef TRACKER

static void GetOffset()
//-------------------------
{
    GetOffset();
    // get an offset of a function to match to map file, and make sure it's referenced
}

Memory::Memory()
        : DebuggingLog( "memory" )
        , _numMessages( 0 )
//--------------------------------
{
    DebuggingLog::printf( "Address of GetOffset is %#p\n\n", &GetOffset );
    TrHdl = _trmem_open( malloc, free, realloc, NULL, NULL, PrintLine,
            _TRMEM_ALLOC_SIZE_0 | _TRMEM_REALLOC_SIZE_0 |
            _TRMEM_OUT_OF_MEMORY | _TRMEM_CLOSE_CHECK_FREE );

    _trmem_set_min_alloc( MINALLOC, TrHdl );

#ifdef __WINDOWS__
    _trmem_prt_use_seg_num( TrHdl, TRUE );
#endif
}

Memory::~Memory()
//---------------
{
    _trmem_prt_list( TrHdl );
    _trmem_close( TrHdl );
    if( _numMessages > 1 ) {
#ifdef __WINDOWS__
        WMessageDialog::message( NULL, MsgWarning, MsgOk,
                         "Memory problems detected", "Browser Memory Tracker" );
#else
        fprintf( stderr, "Memory problems detected" );
#endif
    }
}

#endif // Tracker