服務核心 - 工具類

雖然類名稱為CWHService，我理解更多的是工具函數。

主要接口功能有：

1）SetClipboardString設置字符串到windows剪貼板

2）GetMachineID獲取機器標識，網卡地址MD5加密；

3）GetMachineIDEx獲取機器標識，網卡地址和硬盤信息MD5加密；

4）RegisterHotKey與UnRegisterHotKey為注冊熱鍵和注銷熱鍵；

5）GetWorkDirectory獲取進程所在路徑；

6）GetModuleVersion獲取軟件版本信息；

7）CompressData與UnCompressData為壓縮和解壓數據，需要zlib庫。

//

//壓縮文件
#include "Compress\ZLib.h"

//鏈接文件
#ifndef _DEBUG
    #pragma comment(lib,"Version")
    #pragma comment(lib,"Compress\\ZLib.lib")
#else
    #pragma comment(lib,"Version")
    #pragma comment(lib,"Compress\\ZLibD.lib")
#endif

//

//狀態信息
struct tagAstatInfo
{
    ADAPTER_STATUS                    AdapterStatus;                        //網卡狀態
    NAME_BUFFER                        NameBuff[16];                        //名字緩沖
};

//

//構造函數
CWHService::CWHService()
{
}

//拷貝字符
bool CWHService::SetClipboardString(LPCTSTR pszString)
{
    //變量定義
    HANDLE hData=NULL;
    BOOL bOpenClopboard=FALSE;

    //執行邏輯
    __try
    {
        //打開拷貝
        bOpenClopboard=OpenClipboard(AfxGetMainWnd()->m_hWnd);
        if (bOpenClopboard==FALSE) __leave;

        //清空拷貝
        if (EmptyClipboard()==FALSE) __leave;

        //申請內存
        HANDLE hData=GlobalAlloc(GMEM_MOVEABLE|GMEM_ZEROINIT,CountStringBuffer(pszString));
        if (hData==NULL) __leave;

        //復制數據
        lstrcpy((LPTSTR)GlobalLock(hData),pszString);
        GlobalUnlock(hData);

        //設置數據
        #ifndef _UNICODE
            ::SetClipboardData(CF_TEXT,hData);
        #else
            ::SetClipboardData(CF_UNICODETEXT,hData);
        #endif
    }

    //終止程序
    __finally
    {
        //釋放內存
        if (hData!=NULL) GlobalUnlock(hData);

        //關閉拷貝
        if (bOpenClopboard==TRUE) CloseClipboard();

        //錯誤判斷
        if (AbnormalTermination()==TRUE)
        {
            ASSERT(FALSE);
        }
    }

    return true;
}

//機器標識
bool CWHService::GetMachineID(TCHAR szMachineID[LEN_MACHINE_ID])
{
    //變量定義
    TCHAR szMACAddress[LEN_NETWORK_ID]=TEXT("");

    //網卡標識
    GetMACAddress(szMACAddress);

    //轉換信息
    ASSERT(LEN_MACHINE_ID>=LEN_MD5);
    CWHEncrypt::MD5Encrypt(szMACAddress,szMachineID);

    return true;
}

//機器標識
bool CWHService::GetMachineIDEx(TCHAR szMachineID[LEN_MACHINE_ID])
{
    //變量定義
    TCHAR szMACAddress[LEN_NETWORK_ID]=TEXT("");

    //網卡標識
    WORD wMacSize=GetMACAddress(szMACAddress);

    //硬盤標識
    DWORD *pbuf=(DWORD*)szMACAddress;
    WORD wIndex=(wMacSize+sizeof(DWORD)-1)/sizeof(DWORD);
    LPCTSTR pszHardDisk[]={TEXT("C:\\"),TEXT("D:\\"),TEXT("E:\\")};
    for (WORD i=wIndex;i<CountArray(pszHardDisk);i++)
    {
        ASSERT(CountArray(pszHardDisk)>(i-wIndex));
        GetVolumeInformation(pszHardDisk[i-wIndex],NULL,0,pbuf+i,NULL,NULL,0,NULL);
    }

    //轉換信息
    ASSERT(LEN_MACHINE_ID>=LEN_MD5);
    CWHEncrypt::MD5Encrypt(szMACAddress,szMachineID);

    return true;
}

//網卡地址
bool CWHService::GetMACAddress(TCHAR szMACAddress[LEN_NETWORK_ID])
{
    //變量定義
    HINSTANCE hInstance=NULL;

    //執行邏輯
    __try
    {
        //加載 DLL
        hInstance=LoadLibrary(TEXT("NetApi32.dll"));
        if (hInstance==NULL) __leave;

        //獲取函數
        typedef BYTE __stdcall NetBiosProc(NCB * Ncb);
        NetBiosProc * pNetBiosProc=(NetBiosProc *)GetProcAddress(hInstance,"Netbios");
        if (pNetBiosProc==NULL) __leave;

        //變量定義
        NCB Ncb;
        LANA_ENUM LanaEnum;
        ZeroMemory(&Ncb,sizeof(Ncb));
        ZeroMemory(&LanaEnum,sizeof(LanaEnum));

        //枚舉網卡
        Ncb.ncb_command=NCBENUM;
        Ncb.ncb_length=sizeof(LanaEnum);
        Ncb.ncb_buffer=(BYTE *)&LanaEnum;
        if ((pNetBiosProc(&Ncb)!=NRC_GOODRET)||(LanaEnum.length==0)) __leave;

        //獲取地址
        if (LanaEnum.length>0)
        {
            //變量定義
            tagAstatInfo Adapter;
            ZeroMemory(&Adapter,sizeof(Adapter));

            //重置網卡
            Ncb.ncb_command=NCBRESET;
            Ncb.ncb_lana_num=LanaEnum.lana[0];
            if (pNetBiosProc(&Ncb)!=NRC_GOODRET) __leave;

            //獲取狀態
            Ncb.ncb_command=NCBASTAT;
            Ncb.ncb_length=sizeof(Adapter);
            Ncb.ncb_buffer=(BYTE *)&Adapter;
            Ncb.ncb_lana_num=LanaEnum.lana[0];
            strcpy((char *)Ncb.ncb_callname,"*");
            if (pNetBiosProc(&Ncb)!=NRC_GOODRET) __leave;

            //獲取地址
            for (INT i=0;i<6;i++)
            {
                ASSERT((i*2)<LEN_NETWORK_ID);
                _stprintf(&szMACAddress[i*2],TEXT("%02X"),Adapter.AdapterStatus.adapter_address[i]);
            }
        }
    }

    //結束清理
    __finally
    {
        //釋放資源
        if (hInstance!=NULL)
        {
            FreeLibrary(hInstance);
            hInstance=NULL;
        }

        //錯誤斷言
        if (AbnormalTermination()==TRUE)
        {
            ASSERT(FALSE);
        }
    }

    return true;
}

//注銷熱鍵
bool CWHService::UnRegisterHotKey(HWND hWnd, UINT uKeyID)
{
    //注銷熱鍵
    BOOL bSuccess=::UnregisterHotKey(hWnd,uKeyID);

    return (bSuccess==TRUE)?true:false;
}

//注冊熱鍵
bool CWHService::RegisterHotKey(HWND hWnd, UINT uKeyID, WORD wHotKey)
{
    //變量定義
    BYTE cbModifiers=0;
    if (HIBYTE(wHotKey)&HOTKEYF_ALT) cbModifiers|=MOD_ALT;
    if (HIBYTE(wHotKey)&HOTKEYF_SHIFT) cbModifiers|=MOD_SHIFT;
    if (HIBYTE(wHotKey)&HOTKEYF_CONTROL) cbModifiers|=MOD_CONTROL;

    //注冊熱鍵
    BOOL bSuccess=::RegisterHotKey(hWnd,uKeyID,cbModifiers,LOBYTE(wHotKey));

    return (bSuccess==TRUE)?true:false;
}

//進程目錄
bool CWHService::GetWorkDirectory(TCHAR szWorkDirectory[], WORD wBufferCount)
{
    //模塊路徑
    TCHAR szModulePath[MAX_PATH]=TEXT("");
    GetModuleFileName(AfxGetInstanceHandle(),szModulePath,CountArray(szModulePath));

    //分析文件
    for (INT i=lstrlen(szModulePath);i>=0;i--)
    {
        if (szModulePath[i]==TEXT('\\'))
        {
            szModulePath[i]=0;
            break;
        }
    }

    //設置結果
    ASSERT(szModulePath[0]!=0);
    lstrcpyn(szWorkDirectory,szModulePath,wBufferCount);

    return true;
}

//文件版本
bool CWHService::GetModuleVersion(LPCTSTR pszModuleName, DWORD & dwVersionInfo)
{
    //設置結果
    dwVersionInfo=0L;

    //接收緩沖
    BYTE cbInfoBuffer[1024];
    ZeroMemory(cbInfoBuffer,sizeof(cbInfoBuffer));

    //模塊信息
    DWORD dwFileHandle=NULL;
    if (GetFileVersionInfo(pszModuleName,dwFileHandle,sizeof(cbInfoBuffer),cbInfoBuffer)==FALSE) return false;

    //獲取信息
    UINT uQuerySize=0;
    VS_FIXEDFILEINFO * pFixedFileInfo=NULL;
    if (VerQueryValue(cbInfoBuffer,TEXT("\\"),(VOID * *)&pFixedFileInfo,&uQuerySize)==FALSE) return false;

    //設置結果
    if ((pFixedFileInfo!=NULL)&&(uQuerySize==sizeof(VS_FIXEDFILEINFO)))
    {
        //設置版本
        WORD wVersion1=HIWORD(pFixedFileInfo->dwFileVersionMS);
        WORD wVersion2=LOWORD(pFixedFileInfo->dwFileVersionMS);
        WORD wVersion3=HIWORD(pFixedFileInfo->dwFileVersionLS);
        WORD wVersion4=LOWORD(pFixedFileInfo->dwFileVersionLS);
        dwVersionInfo=MAKELONG(MAKEWORD(wVersion4,wVersion3),MAKEWORD(wVersion2,wVersion1));

        return true;
    }

    return false;
}

//壓縮數據
ULONG CWHService::CompressData(LPBYTE pcbSourceData, ULONG lSourceSize, BYTE cbResultData[], ULONG lResultSize)
{
    //壓縮數據
    if (compress(cbResultData,&lResultSize,pcbSourceData,lSourceSize)==0L)
    {
        return lResultSize;
    }

    return 0L;
}

//解壓數據
ULONG CWHService::UnCompressData(LPBYTE pcbSourceData, ULONG lSourceSize, BYTE cbResultData[], ULONG lResultSize)
{
    //解壓數據
    if (uncompress(cbResultData,&lResultSize,pcbSourceData,lSourceSize)==0L)
    {
        return lResultSize;
    }

    return 0L;
}

//

?