hookwmi
| /* x360ce - XBOX360 Controler Emulator |
| * Copyright (C) 2002-2010 ToCA Edit |
| * |
| * x360ce is free software: you can redistribute it and/or modify it under the terms |
| * of the GNU Lesser General Public License as published by the Free Software Found- |
| * ation, either version 3 of the License, or (at your option) any later version. |
| * |
| * x360ce is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; |
| * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR |
| * PURPOSE. See the GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License along with x360ce. |
| * If not, see <http://www.gnu.org/licenses/>. |
| */ |
| |
| #include "stdafx.h" |
| #include "globals.h" |
| #include "InputHook.h" |
| #include "Utilities\Log.h" |
| |
| #define CINTERFACE |
| #define _WIN32_DCOM |
| #include <wbemidl.h> |
| #include <ole2.h> |
| #include <oleauto.h> |
| #include <dinput.h> |
| |
| // COM CLSIDs |
| #pragma comment(lib,"wbemuuid.lib") |
| |
| // EasyHook handles |
| TRACED_HOOK_HANDLE hHookGet = NULL; |
| TRACED_HOOK_HANDLE hHookNext = NULL; |
| TRACED_HOOK_HANDLE hHookCreateInstanceEnum = NULL; |
| TRACED_HOOK_HANDLE hHookConnectServer = NULL; |
| TRACED_HOOK_HANDLE hHookCoCreateInstance = NULL; |
| TRACED_HOOK_HANDLE hHookCoUninitialize = NULL; |
| |
| |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| void (WINAPI *OriginalCoUninitialize)() = NULL; |
| HRESULT (WINAPI *OriginalCoCreateInstance)(__in REFCLSID rclsid, |
| __in_opt LPUNKNOWN pUnkOuter, |
| __in DWORD dwClsContext, |
| __in REFIID riid, |
| __deref_out LPVOID FAR* ppv) = NULL; |
| |
| |
| HRESULT ( STDMETHODCALLTYPE *OriginalConnectServer )( |
| IWbemLocator * This, |
| /* [in] */ const BSTR strNetworkResource, |
| /* [in] */ const BSTR strUser, |
| /* [in] */ const BSTR strPassword, |
| /* [in] */ const BSTR strLocale, |
| /* [in] */ long lSecurityFlags, |
| /* [in] */ const BSTR strAuthority, |
| /* [in] */ IWbemContext *pCtx, |
| /* [out] */ IWbemServices **ppNamespace) = NULL; |
| |
| HRESULT ( STDMETHODCALLTYPE *OriginalCreateInstanceEnum )( |
| IWbemServices * This, |
| /* [in] */ __RPC__in const BSTR strFilter, |
| /* [in] */ long lFlags, |
| /* [in] */ __RPC__in_opt IWbemContext *pCtx, |
| /* [out] */ __RPC__deref_out_opt IEnumWbemClassObject **ppEnum) = NULL; |
| |
| HRESULT ( STDMETHODCALLTYPE *OriginalNext )( |
| IEnumWbemClassObject * This, |
| /* [in] */ long lTimeout, |
| /* [in] */ ULONG uCount, |
| /* [length_is][size_is][out] */ __RPC__out_ecount_part(uCount, *puReturned) IWbemClassObject **apObjects, |
| /* [out] */ __RPC__out ULONG *puReturned) = NULL; |
| |
| HRESULT ( STDMETHODCALLTYPE *OriginalGet )( |
| IWbemClassObject * This, |
| /* [string][in] */ LPCWSTR wszName, |
| /* [in] */ long lFlags, |
| /* [unique][in][out] */ VARIANT *pVal, |
| /* [unique][in][out] */ CIMTYPE *pType, |
| /* [unique][in][out] */ long *plFlavor) = NULL; |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| HRESULT STDMETHODCALLTYPE HookGet( |
| IWbemClassObject * This, |
| /* [string][in] */ LPCWSTR wszName, |
| /* [in] */ long lFlags, |
| /* [unique][in][out] */ VARIANT *pVal, |
| /* [unique][in][out] */ CIMTYPE *pType, |
| /* [unique][in][out] */ long *plFlavor) |
| { |
| if(!InputHook_Config()->bEnabled) return OriginalGet(This,wszName,lFlags,pVal,pType,plFlavor); |
| |
| WriteLog(LOG_HOOKWMI,L"HookGet"); |
| HRESULT hr = OriginalGet(This,wszName,lFlags,pVal,pType,plFlavor); |
| |
| //WriteLog(L"wszName %s pVal->vt %d pType %d",wszName,pVal->vt,&pType); |
| //if( pVal->vt == VT_BSTR) WriteLog(L"%s",pVal->bstrVal); |
| |
| if( pVal->vt == VT_BSTR && pVal->bstrVal != NULL ) { |
| //WriteLog(L"%s"),pVal->bstrVal); |
| DWORD dwPid = 0, dwVid = 0; |
| WCHAR* strVid = wcsstr( pVal->bstrVal, L"VID_" ); |
| if(strVid && swscanf_s( strVid, L"VID_%4X", &dwVid ) != 1 ) |
| return hr; |
| WCHAR* strPid = wcsstr( pVal->bstrVal, L"PID_" ); |
| if(strPid && swscanf_s( strPid, L"PID_%4X", &dwPid ) != 1 ) |
| return hr; |
| |
| for(WORD i = 0; i < 4; i++) { |
| if(InputHook_GamepadConfig(i)->bEnabled && InputHook_GamepadConfig(i)->dwVID == dwVid && InputHook_GamepadConfig(i)->dwPID == dwPid) { |
| WCHAR* strUSB = wcsstr( pVal->bstrVal, L"USB" ); |
| WCHAR tempstr[MAX_PATH]; |
| if( strUSB ) { |
| BSTR Hookbstr=NULL; |
| WriteLog(LOG_HOOKWMI,L"Original DeviceID = %s",pVal->bstrVal); |
| if(InputHook_Config()->dwHookMode >= 2) swprintf_s(tempstr,L"USB\\VID_%04X&PID_%04X&IG_%02d", InputHook_Config()->dwHookVID, InputHook_Config()->dwHookPID,i ); |
| else swprintf_s(tempstr,L"USB\\VID_%04X&PID_%04X&IG_%02d", InputHook_GamepadConfig(i)->dwVID , InputHook_GamepadConfig(i)->dwPID,i ); |
| Hookbstr=SysAllocString(tempstr); |
| pVal->bstrVal = Hookbstr; |
| WriteLog(LOG_HOOKWMI,L"Hook DeviceID = %s",pVal->bstrVal); |
| return hr; |
| } |
| WCHAR* strHID = wcsstr( pVal->bstrVal, L"HID" ); |
| if( strHID ) { |
| BSTR Hookbstr=NULL; |
| WriteLog(LOG_HOOKWMI,L"Original DeviceID = %s",pVal->bstrVal); |
| if(InputHook_Config()->dwHookMode >= 2) swprintf_s(tempstr,L"HID\\VID_%04X&PID_%04X&IG_%02d", InputHook_Config()->dwHookVID, InputHook_Config()->dwHookPID,i ); |
| else swprintf_s(tempstr,L"HID\\VID_%04X&PID_%04X&IG_%02d", InputHook_GamepadConfig(i)->dwVID , InputHook_GamepadConfig(i)->dwPID,i ); |
| Hookbstr=SysAllocString(tempstr); |
| pVal->bstrVal = Hookbstr; |
| WriteLog(LOG_HOOKWMI,L"Hook DeviceID = %s",pVal->bstrVal); |
| return hr; |
| } |
| } |
| } |
| } |
| return hr; |
| } |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| HRESULT STDMETHODCALLTYPE HookNext( |
| IEnumWbemClassObject * This, |
| /* [in] */ long lTimeout, |
| /* [in] */ ULONG uCount, |
| /* [length_is][size_is][out] */ __RPC__out_ecount_part(uCount, *puReturned) IWbemClassObject **apObjects, |
| /* [out] */ __RPC__out ULONG *puReturned) |
| { |
| if(!InputHook_Config()->bEnabled) return OriginalNext(This,lTimeout,uCount,apObjects,puReturned); |
| WriteLog(LOG_HOOKWMI,L"HookNext"); |
| HRESULT hr; |
| IWbemClassObject* pDevices; |
| |
| hr = OriginalNext(This,lTimeout,uCount,apObjects,puReturned); |
| |
| if(apObjects) { |
| if(*apObjects) { |
| pDevices = *apObjects; |
| if(!OriginalGet) { |
| OriginalGet = pDevices->lpVtbl->Get; |
| hHookGet = new HOOK_TRACE_INFO(); |
| |
| LhInstallHook(OriginalGet,HookGet,static_cast<PVOID>(NULL),hHookGet); |
| LhSetExclusiveACL(ACLEntries, 0, hHookGet); |
| } |
| } |
| } |
| |
| return hr; |
| } |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| HRESULT STDMETHODCALLTYPE HookCreateInstanceEnum( |
| IWbemServices * This, |
| /* [in] */ __RPC__in const BSTR strFilter, |
| /* [in] */ long lFlags, |
| /* [in] */ __RPC__in_opt IWbemContext *pCtx, |
| /* [out] */ __RPC__deref_out_opt IEnumWbemClassObject **ppEnum) |
| { |
| if(!InputHook_Config()->bEnabled) return OriginalCreateInstanceEnum(This,strFilter,lFlags,pCtx,ppEnum); |
| WriteLog(LOG_HOOKWMI,L"HookCreateInstanceEnum"); |
| HRESULT hr; |
| IEnumWbemClassObject* pEnumDevices = NULL; |
| |
| hr = OriginalCreateInstanceEnum(This,strFilter,lFlags,pCtx,ppEnum); |
| |
| |
| if(ppEnum) { |
| if(*ppEnum) { |
| pEnumDevices = *ppEnum; |
| |
| if(!OriginalNext) { |
| OriginalNext = pEnumDevices->lpVtbl->Next; |
| hHookNext = new HOOK_TRACE_INFO(); |
| |
| LhInstallHook(OriginalNext,HookNext,static_cast<PVOID>(NULL),hHookNext); |
| LhSetExclusiveACL(ACLEntries, 0, hHookNext); |
| |
| } |
| } |
| } |
| |
| return hr; |
| } |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| HRESULT STDMETHODCALLTYPE HookConnectServer( |
| IWbemLocator * This, |
| /* [in] */ const BSTR strNetworkResource, |
| /* [in] */ const BSTR strUser, |
| /* [in] */ const BSTR strPassword, |
| /* [in] */ const BSTR strLocale, |
| /* [in] */ long lSecurityFlags, |
| /* [in] */ const BSTR strAuthority, |
| /* [in] */ IWbemContext *pCtx, |
| /* [out] */ IWbemServices **ppNamespace) |
| |
| { |
| if(!InputHook_Config()->bEnabled) return OriginalConnectServer(This,strNetworkResource,strUser,strPassword,strLocale,lSecurityFlags,strAuthority,pCtx,ppNamespace); |
| WriteLog(LOG_HOOKWMI,L"HookConnectServer"); |
| HRESULT hr; |
| IWbemServices* pIWbemServices = NULL; |
| |
| hr = OriginalConnectServer(This,strNetworkResource,strUser,strPassword,strLocale,lSecurityFlags,strAuthority,pCtx,ppNamespace); |
| |
| if(ppNamespace) { |
| if(*ppNamespace) { |
| pIWbemServices = *ppNamespace; |
| |
| if(!OriginalCreateInstanceEnum) { |
| OriginalCreateInstanceEnum = pIWbemServices->lpVtbl->CreateInstanceEnum; |
| hHookCreateInstanceEnum = new HOOK_TRACE_INFO(); |
| |
| LhInstallHook(OriginalCreateInstanceEnum,HookCreateInstanceEnum,static_cast<PVOID>(NULL),hHookCreateInstanceEnum); |
| LhSetExclusiveACL(ACLEntries, 0, hHookCreateInstanceEnum); |
| } |
| } |
| } |
| |
| return hr; |
| } |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| HRESULT WINAPI HookCoCreateInstance(__in REFCLSID rclsid, |
| __in_opt LPUNKNOWN pUnkOuter, |
| __in DWORD dwClsContext, |
| __in REFIID riid, |
| __deref_out LPVOID FAR* ppv) |
| { |
| if(!InputHook_Config()->bEnabled) return OriginalCoCreateInstance(rclsid,pUnkOuter,dwClsContext,riid,ppv); |
| HRESULT hr; |
| IWbemLocator* pIWbemLocator = NULL; |
| |
| hr = OriginalCoCreateInstance(rclsid,pUnkOuter,dwClsContext,riid,ppv); |
| |
| if(ppv && (riid == IID_IWbemLocator)) { |
| pIWbemLocator = static_cast<IWbemLocator*>(*ppv); |
| if(pIWbemLocator) { |
| WriteLog(LOG_HOOKWMI,L"HookCoCreateInstance"); |
| if(!OriginalConnectServer) { |
| OriginalConnectServer = pIWbemLocator->lpVtbl->ConnectServer; |
| hHookConnectServer = new HOOK_TRACE_INFO(); |
| |
| LhInstallHook(OriginalConnectServer,HookConnectServer,static_cast<PVOID>(NULL),hHookConnectServer); |
| LhSetExclusiveACL(ACLEntries, 0, hHookConnectServer); |
| } |
| } |
| } |
| |
| return hr; |
| } |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| void HookCoUninitialize() |
| { |
| WriteLog(LOG_HOOKWMI,L"HookCoUninitialize"); |
| |
| if(OriginalGet) { |
| WriteLog(LOG_HOOKWMI,L"HookGet:: Removing Hook"); |
| |
| LhUninstallHook(hHookGet); |
| LhWaitForPendingRemovals(); |
| SAFE_DELETE(hHookGet); |
| OriginalGet = NULL; |
| } |
| |
| if(OriginalNext) { |
| WriteLog(LOG_HOOKWMI,L"HookNext:: Removing Hook"); |
| |
| LhUninstallHook(hHookNext); |
| LhWaitForPendingRemovals(); |
| SAFE_DELETE(hHookNext); |
| OriginalNext=NULL; |
| } |
| |
| if(OriginalCreateInstanceEnum) { |
| WriteLog(LOG_HOOKWMI,L"HookCreateInstanceEnum:: Removing Hook"); |
| |
| LhUninstallHook(hHookCreateInstanceEnum); |
| LhWaitForPendingRemovals(); |
| SAFE_DELETE(hHookCreateInstanceEnum); |
| OriginalCreateInstanceEnum=NULL; |
| } |
| |
| if(OriginalConnectServer) { |
| WriteLog(LOG_HOOKWMI,L"HookConnectServer:: Removing Hook"); |
| |
| LhUninstallHook(hHookConnectServer); |
| LhWaitForPendingRemovals(); |
| SAFE_DELETE(hHookConnectServer); |
| OriginalConnectServer=NULL; |
| } |
| OriginalCoUninitialize(); |
| } |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| void HookWMI() |
| { |
| if(!OriginalCoCreateInstance) { |
| OriginalCoCreateInstance = CoCreateInstance; |
| hHookCoCreateInstance = new HOOK_TRACE_INFO(); |
| WriteLog(LOG_IHOOK,L"HookCoCreateInstance:: Hooking"); |
| |
| LhInstallHook(OriginalCoCreateInstance,HookCoCreateInstance,static_cast<PVOID>(NULL),hHookCoCreateInstance); |
| LhSetExclusiveACL(ACLEntries, 0, hHookCoCreateInstance); |
| |
| } |
| if(!OriginalCoUninitialize) { |
| OriginalCoUninitialize = CoUninitialize; |
| hHookCoUninitialize = new HOOK_TRACE_INFO(); |
| WriteLog(LOG_IHOOK,L"HookCoUninitialize:: Hooking"); |
| |
| LhInstallHook(OriginalCoUninitialize,HookCoUninitialize,static_cast<PVOID>(NULL),hHookCoUninitialize); |
| LhSetExclusiveACL(ACLEntries, 0, hHookCoUninitialize); |
| |
| } |
| } |
| ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| void HookWMIClean() |
| { |
| SAFE_DELETE(hHookGet); |
| SAFE_DELETE(hHookNext); |
| SAFE_DELETE(hHookCreateInstanceEnum); |
| SAFE_DELETE(hHookConnectServer); |
| SAFE_DELETE(hHookCoCreateInstance); |
| SAFE_DELETE(hHookCoUninitialize); |
| } |
|
http://code.google.com/p/x360ce/source/browse/trunk/x360ce/x360ce/InputHook/HookWMI.cpp?r=410