cWinOSWrap.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_cWinOSWrap_H
#define _INC_cWinOSWrap_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "../WinAPI/WinTypes.h"
#include "GrayCore/include/cOSModule.h"
 
#ifdef _WIN32
namespace GrayLib
{
        enum WINOSD_TYPE
        {
                WINOSD_Kernel32,
                WINOSD_User32,
                WINOSD_PSAPI,
                WINOSD_ADVAPI32,
                WINOSD_IPHLPAPI,        // GetAdaptersAddresses
                WINOSD_QTY,
        };
 
        class GRAYLIB_LINK cWinOSWrap : public cSingleton < cWinOSWrap >
        {
 
                friend class cSingleton < cWinOSWrap >;
                CHEAPOBJECT_IMPL;
 
        public:
                //typedef DWORD(WINAPI* FUNC_GetProcessId_t)(HANDLE hProcess);
                //typedef BOOL(WINAPI* FUNC_CancelIo_t)(HANDLE h);      // todo CancelIoEx?
                typedef BOOL(WINAPI* FUNC_ChangeWindowMessageFilter_t)(UINT message, DWORD dwFlag);
                typedef BOOL(WINAPI* FUNC_EnumProcessModulesEx)(HANDLE hProcess, OUT HMODULE *lphModule, DWORD cb, OUT LPDWORD lpcbNeeded, DWORD dwFilterFlag);
                typedef LONG(WINAPI* FUNC_RegDeleteKeyExW)(HKEY hKey, LPCWSTR lpSubKey, REGSAM samDesired, DWORD Reserved);
                typedef LONG(WINAPI* FUNC_RegDeleteKeyExA)(HKEY hKey, LPCSTR lpSubKey, REGSAM samDesired, DWORD Reserved);
 
        public:
                // cOSModuleFunc<FUNC_CancelIo_t> m_pfnCancelIo;                        // m_modKernel32 Vista
                cOSModuleFunc<FUNC_ChangeWindowMessageFilter_t> m_funcChangeWindowMessageFilter; // m_modUser32 Vista
                cOSModuleFunc<FUNC_EnumProcessModulesEx> m_funcEnumProcessModulesEx;    // m_modPSAPI Vista, Win7
                cOSModuleFunc<FUNC_RegDeleteKeyExA> m_funcRegDeleteKeyExA;      // m_modAdvapi, XP Pro x64, Vista
 
        public:
                // static const char* k_ModuleName[WINOSD_QTY + 1];
                cOSModule m_mod[WINOSD_QTY + 1];        
 
        protected:
                cWinOSWrap()
                        : cSingleton<cWinOSWrap>(this, typeid(cWinOSWrap))
                {
                }
                ~cWinOSWrap()
                {
                }
 
        public:
                HRESULT BindModule(cOSModule& mod, const FILECHAR_t* pszName)
                {
                        if (mod.isValidModule())        // already done.
                                return S_OK;
                        FILECHAR_t szPath[_MAX_PATH];
                        StrLen_t nLen = cSystemInfo::GetSystemDir(szPath, STRMAX(szPath));
                        nLen = cFilePath::CombineFilePathA(szPath, STRMAX(szPath), nLen, pszName);
                        HRESULT hRes = mod.LoadModule(szPath);
                        if (FAILED(hRes))
                                return hRes;
                        ASSERT(mod.isValidModule());
                        return nLen;
                }
 
                HRESULT BindKernel32()
                {
                        HRESULT hRes = BindModule(m_mod[WINOSD_Kernel32], _FN("kernel32.dll"));
                        if (FAILED(hRes) || hRes == S_OK)
                        {
                                return hRes;
                        }
                        return S_OK;
                }
                HRESULT BindUser32()
                {
                        HRESULT hRes = BindModule(m_mod[WINOSD_User32], _FN("user32.dll"));
                        if (FAILED(hRes) || hRes == S_OK)
                        {
                                return hRes;
                        }
                        // Vista
                        m_funcChangeWindowMessageFilter.put_FuncGeneric(m_mod[WINOSD_User32].GetSymbolAddress("ChangeWindowMessageFilter"));
                        return S_OK;
                }
                HRESULT BindPSAPI()
                {
                        HRESULT hRes = BindModule(m_mod[WINOSD_PSAPI], _FN("psapi.dll"));
                        if (FAILED(hRes) || hRes == S_OK)
                        {
                                return hRes;
                        }
                        // Vista and Win7
                        m_funcEnumProcessModulesEx.put_FuncGeneric(m_mod[WINOSD_PSAPI].GetSymbolAddress("EnumProcessModulesEx"));
                        return S_OK;
                }
                HRESULT BindAdvapi()
                {
                        HRESULT hRes = BindModule(m_mod[WINOSD_ADVAPI32], _FN("Advapi32.dll"));
                        if (FAILED(hRes) || hRes == S_OK)
                        {
                                return hRes;
                        }
                        m_funcRegDeleteKeyExA.put_FuncGeneric(m_mod[WINOSD_ADVAPI32].GetSymbolAddress("RegDeleteKeyExA"));
                        return S_OK;
                }
        };
};
#endif  // _WIN32
#endif