Win32函数封装(二)---控制台输出(多用于调试),文件读写,dx注册

// DebugTool.h: interface for the CDebugTool class.
//兰海 lanhai(guoliangw@hotmail.com) http://blog.csdn.net/lanhai/ 调试类 2006-08-06
//出于 助人为乐目的发布,可以任意转载,完善及丰富,但需保留此声明,用于商用请充分测试
//////////////////////////////////////////////////////////////////////
#ifndef __DEBUGTOOL_H
#define __DEBUGTOOL_H

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include <unknwn.h>
class CDebugTool
{
public:
CDebugTool();
virtual ~CDebugTool();

//对外接口
public:
void ConsolOut(char* szinfo,...); //输出信息
void ConsolOutNoRN(char* szinfo,...);
HANDLE CreateBufferFile(LPSTR lpName);
void WriteBuffer2File(HANDLE aFile,LPSTR apBuff,long lSize);
long GetModuleDir(LPSTR lpBuff);
HRESULT AddToRot(IUnknown* pUnkGraph, DWORD* pdwRegister);
void RemoveFromRot(DWORD pdwRegister);
void ConsolDel();

HANDLE OpenDbReadFile(LPSTR lpName);
DWORD GetDbFileSize(HANDLE hFile);
BOOL ReadDbFile(HANDLE hFile,LPSTR lpBuff,long lBuffLen);

DWORD GetTheFileSize(char* filename) ;

//内部函数
protected:
void ConsolGet();
protected:
HANDLE mhOut;
BOOL mbSelConsol;
HANDLE mhFile;
BOOL mbRN;

};

#endif //__DEBUGTOOL_H

// DebugTool.cpp: implementation of the CDebugTool class.
//兰海 lanhai(guoliangw@hotmail.com) http://blog.csdn.net/lanhai/ 调试类 2006-08-06

//出于 助人为乐目的发布,可以任意转载,完善及丰富,但需保留此声明,用于商用请充分测试
//////////////////////////////////////////////////////////////////////

#include <windows.h>
#include "DebugTool.h"
#include <stdio.h>
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CDebugTool::CDebugTool()
{
mhOut = NULL;
mbSelConsol = FALSE;
mbRN = FALSE;
}

CDebugTool::~CDebugTool()
{
ConsolDel();
}
void CDebugTool::ConsolGet()
{
if(!mhOut)
{
AllocConsole();
mhOut = GetStdHandle(STD_OUTPUT_HANDLE);
mbSelConsol = TRUE;

// mhOut = GetStdHandle(STD_OUTPUT_HANDLE);
// if(INVALID_HANDLE_VALUE == mhOut)
// {
// AllocConsole();
// mhOut = GetStdHandle(STD_OUTPUT_HANDLE);
// mbSelConsol = TRUE;
// }
}
}
void CDebugTool::ConsolDel()
{
if(mhOut && mbSelConsol)
{
FreeConsole();
mhOut =NULL;
mbRN = FALSE;
}

}
void CDebugTool::ConsolOut(char* szinfo,...)
{

va_list argptr;
char strPrintBuff[512];

//把可变参数序列化到strPrintBuff中
va_start(argptr,szinfo);
vsprintf(strPrintBuff, szinfo, argptr);
va_end( argptr );

ConsolGet();

if(mhOut)
{
if(mbRN)
{
WriteConsole(mhOut,"/r/n",lstrlen("/r/n"),NULL,NULL);
}
WriteConsole(mhOut,strPrintBuff,lstrlen(strPrintBuff),NULL,NULL);
WriteConsole(mhOut,"/r/n",lstrlen("/r/n"),NULL,NULL);
}
}
void CDebugTool::ConsolOutNoRN(char* szinfo,...)
{
va_list argptr;
char strPrintBuff[512];

//把可变参数序列化到strPrintBuff中
va_start(argptr,szinfo);
vsprintf(strPrintBuff, szinfo, argptr);
va_end( argptr );

ConsolGet();

if(mhOut)
{
WriteConsole(mhOut,strPrintBuff,lstrlen(strPrintBuff),NULL,NULL);
WriteConsole(mhOut,"/r",lstrlen("/r"),NULL,NULL);

mbRN = TRUE;
}
}
HANDLE CDebugTool::CreateBufferFile(LPSTR lpName)
{
//mhFile = CreateFile(lpName,NULL,FILE_SHARE_READ,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
mhFile = CreateFile(lpName,GENERIC_WRITE,0,0,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,0);

return mhFile;
}
void CDebugTool::WriteBuffer2File(HANDLE aFile,LPSTR apBuff,long lSize)
{
DWORD dwWrite;
if(aFile)
{
WriteFile(aFile,apBuff,lSize,&dwWrite,NULL);
}
}
long CDebugTool::GetModuleDir(LPSTR lpBuff)
{
char szPath[MAX_PATH] = {0};

if(lpBuff == NULL)
return 0;

GetModuleFileName(NULL,szPath,sizeof(szPath));
char* pPos = NULL;
pPos = strrchr(szPath,'//');
pPos++;
*pPos = '/0';
lstrcpy(lpBuff,szPath);
return lstrlen(lpBuff);
}
HRESULT CDebugTool::AddToRot(IUnknown* pUnkGraph, DWORD* pdwRegister)
{
IMoniker* pMoniker;
IRunningObjectTable* pROT;
if(FAILED(GetRunningObjectTable(0, &pROT))){
return E_FAIL;
}
WCHAR wsz[256];
wsprintfW(wsz, L"FilterGraph %08x pid %08x", (DWORD_PTR)pUnkGraph,
GetCurrentProcessId());
HRESULT hr = CreateItemMoniker(L"!", wsz, &pMoniker);
if(SUCCEEDED(hr)){
hr = pROT->Register(ROTFLAGS_REGISTRATIONKEEPSALIVE, pUnkGraph,
pMoniker, pdwRegister);
pMoniker->Release();
}
pROT->Release();
return hr;
}
void CDebugTool::RemoveFromRot(DWORD pdwRegister)
{
IRunningObjectTable* pROT;
if(SUCCEEDED(GetRunningObjectTable(0, &pROT))){
pROT->Revoke(pdwRegister);
pROT->Release();
}
}
HANDLE CDebugTool::OpenDbReadFile(LPSTR lpName)
{
HANDLE hFile = NULL;
hFile = CreateFile(lpName,GENERIC_READ,FILE_SHARE_READ,0,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,0);
return hFile;
}
DWORD CDebugTool::GetDbFileSize(HANDLE hFile)
{
DWORD dwSize = 0;
if(INVALID_FILE_SIZE==GetFileSize(hFile,&dwSize) )
return 0;
return dwSize;
}
BOOL CDebugTool::ReadDbFile(HANDLE hFile,LPSTR lpBuff,long lBuffLen)
{
DWORD dwRead = 0;
LONG lret = 0;
lret = ReadFile(hFile,lpBuff,lBuffLen,&dwRead ,NULL);
if(lret == 0)
return 0;
return dwRead;
}
DWORD CDebugTool::GetTheFileSize(char* filename)
{
HANDLE hFile = INVALID_HANDLE_VALUE;
HRESULT hr;
hFile = CreateFile(filename,GENERIC_READ,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
if( INVALID_HANDLE_VALUE == hFile )
{
hr = HRESULT_FROM_WIN32( GetLastError() );
return 0;
}

int dwLength = GetFileSize( hFile, NULL );
CloseHandle(hFile);
return dwLength;
}

lanhai 2007.06.12