memprb.cpp

著名的 helix realplayer 基于手机 symbian 系统的 播放器全套源代码

/* ***** BEGIN LICENSE BLOCK ***** 
 * Version: RCSL 1.0/RPSL 1.0 
 *  
 * Portions Copyright (c) 1995-2002 RealNetworks, Inc. All Rights Reserved. 
 *      
 * The contents of this file, and the files included with this file, are 
 * subject to the current version of the RealNetworks Public Source License 
 * Version 1.0 (the "RPSL") available at 
 * http://www.helixcommunity.org/content/rpsl unless you have licensed 
 * the file under the RealNetworks Community Source License Version 1.0 
 * (the "RCSL") available at http://www.helixcommunity.org/content/rcsl, 
 * in which case the RCSL will apply. You may also obtain the license terms 
 * directly from RealNetworks.  You may not use this file except in 
 * compliance with the RPSL or, if you have a valid RCSL with RealNetworks 
 * applicable to this file, the RCSL.  Please see the applicable RPSL or 
 * RCSL for the rights, obligations and limitations governing use of the 
 * contents of the file.  
 *  
 * This file is part of the Helix DNA Technology. RealNetworks is the 
 * developer of the Original Code and owns the copyrights in the portions 
 * it created. 
 *  
 * This file, and the files included with this file, is distributed and made 
 * available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 
 * EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS ALL SUCH WARRANTIES, 
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS 
 * FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 
 * 
 * Technology Compatibility Kit Test Suite(s) Location: 
 *    http://www.helixcommunity.org/content/tck 
 * 
 * Contributor(s): 
 *  
 * ***** END LICENSE BLOCK ***** */ 

/****************************************************************************
 *  Defines
 */
#define MAX_DUMPLINE_LENGTH	1024
#define DUMPBLOCKID_START_CHAR	'{'
#define DUMPMEMUSED_START_CHAR	','
#define DUMPDATA_START_STRING	" Data:"	

#define mem_max(x, y)	((x) > (y) ? (x) : (y))


/****************************************************************************
 *  Includes
 */
#ifdef _DEBUG
#include <io.h>

#include <stdio.h>
#include <string.h>
#include <malloc.h>
#endif	// _DEBUG

#include "hxassert.h"
#include "hxmemprb.h"


/****************************************************************************
 *  Locals
 */
#ifdef _DEBUG
static LONG32 CalcUsedMem(_CrtMemState *MemState, BOOL bWithCrt);


static const MemBlockInfo EmptyMemBlockInfo =
{
    0,
    0,
    NULL,
    NULL
};

CHXMemProbe* CHXMemProbe::mz_pActiveProbe = NULL;

BOOL CHXMemProbe::m_bMutexInitialized = FALSE;
CRITICAL_SECTION CHXMemProbe::m_Mutex;
#endif  /* _DEBUG */


CHXMemProbe::CHXMemProbe(BOOL bIsFrozen)
#ifdef _DEBUG
    :	m_bIsFrozen(TRUE),
	m_MemFreeze(EmptyMemBlockInfo),
	m_pMemFreezeIter(NULL),
	m_MemUsed(0),
	m_MemUsedWithCrt(0),
	m_MaxMemUsed(0),
	m_MaxMemUsedWithCrt(0),
	m_BaseMemUsed(0),
	m_BaseMemUsedWithCrt(0),
	m_pParent(NULL),
	m_pChild(NULL)
#endif /* _DEBUG */
{
#ifdef _DEBUG
    if (!m_bMutexInitialized)
    {
	InitializeCriticalSection(&m_Mutex);
	m_bMutexInitialized = TRUE;
    }

    _CrtSetDbgFlag( _CrtSetDbgFlag(_CRTDBG_REPORT_FLAG) |
		    _CRTDBG_ALLOC_MEM_DF | 
		    _CRTDBG_CHECK_CRT_DF);

    _CrtMemCheckpoint(&m_MemRefPoint);
    m_MemFreezePoint = m_MemRefPoint;

    if (!bIsFrozen)
    {
	Reset();  // activate the probe
    }
#endif  /* _DEBUG */
}


CHXMemProbe::~CHXMemProbe()
{
#ifdef _DEBUG
    if (!m_bIsFrozen)
    {
	Freeze();
    }
    DestroyBlockInfo(m_MemFreeze);
#endif  /* _DEBUG */
}


BOOL CHXMemProbe::IsRestored(BOOL bWithCrt)
{
#ifdef _DEBUG
    BOOL bSignificantDiff;
    _CrtMemState MemRefPointDiff;

    if (!m_bIsFrozen)
    {
	_CrtMemCheckpoint(&m_MemFreezePoint);
    }

    bSignificantDiff = _CrtMemDifference(&MemRefPointDiff, &m_MemRefPoint, &m_MemFreezePoint);

    if (bWithCrt)
    {
	return !bSignificantDiff;
    }

    return ((CalcUsedMem(&MemRefPointDiff, HXDBG_CLIENT_MEM) == 0) ? TRUE : FALSE);
#else  /* _DEBUG */
    return TRUE;
#endif  /* _DEBUG */
}


LONG32 CHXMemProbe::GetMemUsed(BOOL bWithCrt, BOOL bWithBase, CHXMemProbe *pReferenceProbe)
{
#ifdef _DEBUG
    LONG32 BaseMem;
    _CrtMemState MemRefPointDiff;
    
    if (bWithBase)
    {
	BaseMem = bWithCrt ? m_BaseMemUsedWithCrt : m_BaseMemUsed;

	if (pReferenceProbe != NULL)
	{
	    BaseMem -= (bWithCrt ? pReferenceProbe->m_BaseMemUsedWithCrt : pReferenceProbe->m_BaseMemUsed);
	}
    }
    else
    {
	BaseMem = 0;
    }

    if (!m_bIsFrozen)
    {
	_CrtMemCheckpoint(&m_MemFreezePoint);
    }

    (void) _CrtMemDifference(&MemRefPointDiff, &m_MemRefPoint, &m_MemFreezePoint);

    return (BaseMem + CalcUsedMem(&MemRefPointDiff, bWithCrt));
#else  /* DEBUG */
    return 0;
#endif  /* _DEBUG */
}


LONG32 CHXMemProbe::GetBaseMemUsed(BOOL bWithCrt)
{
#ifdef _DEBUG
    return bWithCrt ? m_BaseMemUsedWithCrt : m_BaseMemUsed;
#else  /* _DEBUG */
    return 0;
#endif  /* _DEBUG */
}


HX_RESULT CHXMemProbe::DumpMemUsedSinceReset(FILE *pFile, BOOL bWithCrt)
{
#ifdef _DEBUG
    FILE *pTempFile;
    char Buffer[MAX_DUMPLINE_LENGTH]; /* Flawfinder: ignore */
    HX_RESULT RetVal = HXR_OK;

    pTempFile = tmpfile();

    if (pTempFile == NULL)
    {
	RetVal = HXR_FAIL;
    }

    if (SUCCEEDED(RetVal))
    {
	_DumpMemUsedSinceReset(pTempFile, bWithCrt);

	if (fseek(pTempFile, 0, SEEK_SET) != 0)
	{
	    RetVal = HXR_FAIL;
	}
    }

    while (SUCCEEDED(RetVal) && (fgets(Buffer, sizeof(Buffer), pTempFile) != NULL))
    {
	if (fputs(Buffer, pFile) == EOF)
	{
	    RetVal = HXR_FAIL;
	}
    }

    if (pTempFile != NULL)
    {
	fclose(pTempFile);
    }

    return RetVal;
#endif  /* _DEBUG */

    return HXR_OK;
}


#ifdef _DEBUG
void CHXMemProbe::_DumpMemUsedSinceReset(FILE *pFile, BOOL bWithCrt)
{
    int OldFlag;
    int NewFlag;
    _HFILE hOldFile;
    _HFILE hNewFile;
    int OldMode;

    // Set Debug Flag
    OldFlag = _CrtSetDbgFlag(_CRTDBG_REPORT_FLAG);
    if (bWithCrt)
    {
	NewFlag = OldFlag | _CRTDBG_CHECK_CRT_DF;
    }
    else
    {
	NewFlag = OldFlag & (~_CRTDBG_CHECK_CRT_DF);
    }

    (void) _CrtSetDbgFlag(NewFlag);

    // Set Output File
    if (pFile == NULL)
    {
	pFile = (FILE *) stdout;
    }

    hNewFile = (_HFILE) _get_osfhandle(_fileno(pFile));

    OldMode = _CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE);
    hOldFile = _CrtSetReportFile(_CRT_WARN, hNewFile);

    // Dump
    _CrtMemDumpAllObjectsSince(&m_MemRefPoint);
    
    // Restore Debug Mode
    (void) _CrtSetDbgFlag(OldFlag);
    (void) _CrtSetReportMode(_CRT_WARN, OldMode);
    (void) _CrtSetReportFile(_CRT_WARN, hOldFile);
}
#endif  /* _DEBUG */


ULONG32 CHXMemProbe::GetMaxMemUsed(BOOL bWithCrt, BOOL bWithBase, CHXMemProbe *pReferenceProbe)
{
#ifdef _DEBUG
    LONG32 BaseMem;
    LONG32 MaxMem;

    if (bWithBase)
    {
	BaseMem = bWithCrt ? m_BaseMemUsedWithCrt : m_BaseMemUsed;

	if (pReferenceProbe != NULL)
	{
	    BaseMem -= (bWithCrt ? pReferenceProbe->m_BaseMemUsedWithCrt : pReferenceProbe->m_BaseMemUsed);
	}
    }
    else
    {
	BaseMem = 0;
    }

    if (m_pChild == NULL)
    {
	MaxMem = bWithCrt ? m_MaxMemUsedWithCrt : m_MaxMemUsed;	
    }
    else
    {
	LONG32 uChildMax;

	uChildMax = (LONG32) m_pChild->GetMaxMemUsed(bWithCrt, PNDBG_NOBASE_MEM);

	if (bWithCrt)
	{
	    MaxMem = mem_max(uChildMax + m_MemUsedWithCrt, m_MaxMemUsedWithCrt);
	}
	else
	{
	    MaxMem = mem_max(uChildMax + m_MemUsed, m_MaxMemUsed);
	}
    }

    return mem_max(BaseMem + MaxMem, 0);
#else  /* DEBUG */
    return 0;
#endif  /* _DEBUG */
}


LONG32 CHXMemProbe::GetGlobalMemUsed(BOOL bWithCrt)
{
#ifdef _DEBUG
    _CrtMemState MemRefPoint;

    _CrtMemCheckpoint(&MemRefPoint);

    return CalcUsedMem(&MemRefPoint, bWithCrt);
#else  /* _DEBUG */
    return 0;
#endif  /* _DEBUG */
}


ULONG32 CHXMemProbe::GetGlobalMaxMemUsed(void)
{
#ifdef _DEBUG
    _CrtMemState MemRefPoint;

    _CrtMemCheckpoint(&MemRefPoint);

    return (ULONG32) MemRefPoint.lHighWaterCount;
#else  /* _DEBUG */
    return 0;
#endif  /* _DEBUG */
}


ULONG32 CHXMemProbe::SetGlobalMemBreak(ULONG32 uMemBlockId)
{
#ifdef _DEBUG
    return _CrtSetBreakAlloc(uMemBlockId);
#else  /* _DEBUG */
    return 0;
#endif  /* _DEBUG */
}


HX_RESULT CHXMemProbe::Reset(void)
{
#ifdef _DEBUG
    if (!m_bMutexInitialized)
    {
	InitializeCriticalSection(&m_Mutex);
	m_bMutexInitialized = TRUE;
    }

    if (!m_bIsFrozen)
    {
	Freeze();
    }

    EnterCriticalSection(&m_Mutex);

    DestroyBlockInfo(m_MemFreeze);

    _CrtMemCheckpoint(&m_MemRefPoint);

    m_MemUsed = 0;
    m_MemUsedWithCrt = 0;
    m_MaxMemUsed = 0;
    m_MaxMemUsedWithCrt = 0;
    m_BaseMemUsed = 0;
    m_BaseMemUsedWithCrt = 0;

    m_pChild = NULL;
    m_pParent = mz_pActiveProbe;
    if (m_pParent != NULL)
    {
	m_pParent->m_pChild = this;

	m_BaseMemUsedWithCrt = m_pParent->m_MemUsedWithCrt + m_pParent->m_BaseMemUsedWithCrt;
	m_BaseMemUsed = m_pParent->m_MemUsed + m_pParent->m_BaseMemUsed;
    }
    mz_pActiveProbe = this;

    _CrtSetAllocHook(MemProbeHook);

    m_bIsFrozen = FALSE;
    
    LeaveCriticalSection(&m_Mutex);
#endif  /* _DEBUG */

    return HXR_OK;
}


HX_RESULT CHXMemProbe::Freeze(BOOL bGenerateDetailInfo, BOOL bWithCrt)