Tools/Xabsl2/yabsl/vsyabsl/common.cpp

Go to the documentation of this file.

/***************************************************************************
         Copyright (c) Microsoft Corporation, All rights reserved.             
    This code sample is provided "AS IS" without warranty of any kind, 
    it is not recommended for use in a production environment.
***************************************************************************/

#include "common.h"
#include <stdio.h>
#include <stdarg.h>

/*---------------------------------------------------------
  Reference counting
-----------------------------------------------------------*/
ULONG IncRefCount( inout ULONG* refCount )
{
  return (ULONG)InterlockedIncrement( (LONG*)refCount );
}

ULONG DecRefCount( inout ULONG* refCount )
{
  return (ULONG)InterlockedDecrement( (LONG*)refCount );
}



/*---------------------------------------------------------
  Debugging
-----------------------------------------------------------*/
static const char*      g_traceModuleName = NULL;
static CRITICAL_SECTION g_traceGuard;
static DWORD            g_traceThreadID;

void Trace( const char* format, ... )
{
  va_list args;
  va_start( args, format );

  EnterCriticalSection( &g_traceGuard );

  if (g_traceModuleName)
  {
    OutputDebugStringA(g_traceModuleName);
    OutputDebugStringA( ": " );
  }

  char buffer[1024];
  
  DWORD threadID = GetCurrentThreadId();
  if (threadID != g_traceThreadID)
  {
    _snprintf( buffer, 1023, "tid 0x%x: ", threadID );
    buffer[1023] = 0;
    OutputDebugStringA(buffer);
  }

  _vsnprintf( buffer, 1024, format, args );
  buffer[1023] = 0;
  OutputDebugStringA(buffer);

  OutputDebugStringA( "\n" );

  LeaveCriticalSection( &g_traceGuard );

  va_end( args );
  return;
}


/*---------------------------------------------------------
  Track object life times
-----------------------------------------------------------*/
struct ObjectInfo
{
  ObjectInfo* next;
  const char* name;
  IUnknown*   obj;
};

static ObjectInfo*      objectList = NULL;

void TraceDumpObjects()
{
  EnterCriticalSection( &g_traceGuard );
  
  TRACE("live objects:");
  for (ObjectInfo* objectInfo = objectList; objectInfo; objectInfo = objectInfo->next )
  {
    ULONG refCount = objectInfo->obj->AddRef();
    if (refCount > 0)
    {
      objectInfo->obj->Release();
      refCount--;
    }
    TRACE3( "  0x%x, refcount %u, %s", objectInfo->obj, refCount, objectInfo->name); 
  }

  LeaveCriticalSection( &g_traceGuard );
}

void TraceInit( in const char* moduleName )
{
  InitializeCriticalSection( &g_traceGuard );
  g_traceThreadID     = GetCurrentThreadId();
  
  g_traceModuleName = moduleName;  
  objectList        = NULL;
  
}

void TraceDone()
{
  EnterCriticalSection( &g_traceGuard );

  if (objectList)
  {
    TRACE( "found live objects:" );
    TraceDumpObjects();
  }

  ObjectInfo* objectInfo;
  ObjectInfo* objectNext;

  objectInfo = objectList;
  while (objectInfo)
  {
    objectNext = objectInfo->next;
    FREE(objectInfo);
    objectInfo = objectNext;
  }

  objectList = NULL;

  LeaveCriticalSection( &g_traceGuard );
  DeleteCriticalSection( &g_traceGuard );
}



void TraceCreate( const char* name, IUnknown* obj )
{
  EnterCriticalSection( &g_traceGuard );

  ObjectInfo* objectInfo = ALLOC(ObjectInfo);
  if (!objectInfo) return;
  if (!obj)        
  {
    TRACE1("TraceCreate: creation of %s failed", name );
    return;
  }

  TRACE2( "create %s at 0x%x", name, obj );
  objectInfo->name = name;
  objectInfo->obj  = obj;
  objectInfo->next = objectList;
  objectList       = objectInfo;

  LeaveCriticalSection( &g_traceGuard );
}

void TraceDestroy( IUnknown* obj )
{
  EnterCriticalSection( &g_traceGuard );

  ObjectInfo* objectInfo;
  ObjectInfo* objectPrev = NULL;

  for (objectInfo = objectList; objectInfo; objectInfo = objectInfo->next)
  {
    if (objectInfo->obj == obj) break;
    objectPrev = objectInfo;
  }

  if (!objectInfo)
  {
    TRACE1( "TraceDestroy: destroying unknown object at 0x%x", obj );
  }
  else
  {
    TRACE2( "destroy %s at 0x%8x", objectInfo->name, objectInfo->obj );
    if (objectPrev)
      objectPrev->next = objectInfo->next;
    else
      objectList = objectInfo->next;

    FREE(objectInfo);
  }

  LeaveCriticalSection( &g_traceGuard );
}

---