206 lines
4 KiB
C++
206 lines
4 KiB
C++
#ifndef _MTRACE_H
|
|
#define _MTRACE_H
|
|
//
|
|
// Copyright (c) Microsoft. All Rights Reserved
|
|
//
|
|
// THIS IS UNPUBLISHED PROPRIETARY SOURCE CODE OF Microsoft.
|
|
// The copyright notice above does not evidence any
|
|
// actual or intended publication of such source code.
|
|
//
|
|
// OneLiner : MWMIObject interface.
|
|
// DevUnit : wlbstest
|
|
// Author : Murtaza Hakim
|
|
//
|
|
// Description:
|
|
// -----------
|
|
|
|
// include files
|
|
//
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <windows.h>
|
|
#include <winbase.h>
|
|
#include <wbemidl.h>
|
|
#include <comdef.h>
|
|
|
|
using namespace std;
|
|
|
|
//
|
|
// use this call to do tracing.
|
|
// eg usage.
|
|
// TRACE( MTrace::WARNING, "running out of non paged pool\n");
|
|
|
|
#define TRACE(traceLevel, traceMessage) MYTRACE( (__LINE__ ), (__FILE__), (traceLevel), (traceMessage) );
|
|
|
|
class MTrace
|
|
{
|
|
public:
|
|
|
|
// the trace levels which can be set.
|
|
enum TraceLevel
|
|
{
|
|
FULL_TRACE = 0,
|
|
|
|
INFO = 1,
|
|
WARNING = 2,
|
|
SEVERE_ERROR = 3,
|
|
|
|
NO_TRACE = 100,
|
|
};
|
|
|
|
|
|
// the maximum single message size which can be logged.
|
|
enum
|
|
{
|
|
MAX_MESSAGE_SIZE = 2000,
|
|
};
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// gets instance of MTrace singleton class.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// none
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// the MTrace singleton object.
|
|
|
|
static MTrace*
|
|
Instance();
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// Initialize. The trace object will write the trace data to the screen and the traceFile
|
|
// if file is specified.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// level IN : level determines what is the high threshold for data to be output.
|
|
// traceFile IN : file to store trace output to. Any previous file is deleted when call is
|
|
// made.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// none
|
|
|
|
void
|
|
Initialize( TraceLevel level,
|
|
string traceFile );
|
|
|
|
|
|
void
|
|
Initialize( TraceLevel level );
|
|
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// Initializes the objects to which the singleton object will write data to.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// level IN : level determines what is the high threshold for data to be output.
|
|
// traceMessage IN : trace message to be logged.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// none
|
|
|
|
void
|
|
SendTraceOutput( TraceLevel level,
|
|
wstring traceMessage );
|
|
|
|
void
|
|
SendTraceOutput( TraceLevel level,
|
|
string traceMessage );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// Gets the present trace level
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// none
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// trace level set.
|
|
|
|
TraceLevel
|
|
GetLevel();
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// Sets the trace level.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// level IN : level to set.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// none
|
|
|
|
void
|
|
SetLevel(TraceLevel level );
|
|
|
|
|
|
protected:
|
|
MTrace();
|
|
|
|
private:
|
|
static MTrace* _instance;
|
|
|
|
string _traceFile;
|
|
|
|
TraceLevel _level;
|
|
|
|
void
|
|
FormatOutput( TraceLevel level,
|
|
string traceMessage );
|
|
|
|
};
|
|
|
|
//
|
|
// Ensure type safety
|
|
|
|
typedef class MTrace MTrace;
|
|
|
|
// helper functions.
|
|
//
|
|
|
|
void
|
|
MYTRACE( int lineNum, string fileName, MTrace::TraceLevel level, string traceMessage );
|
|
|
|
void
|
|
MYTRACE( int lineNum, string fileName, MTrace::TraceLevel level, wstring traceMessage );
|
|
|
|
|
|
// utility functions
|
|
void
|
|
CharToWChar( PCHAR pchCharString,
|
|
int iSizeOfCharString,
|
|
PWCHAR pwchCharString,
|
|
int iSizeOfWideCharString );
|
|
|
|
void
|
|
WCharToChar( PWCHAR pwchWideCharString,
|
|
int iSizeOfWCharString,
|
|
PCHAR pchCharString,
|
|
int iSizeOfCharString );
|
|
|
|
|
|
#endif
|