Пришел к такому решению, оно наполовину рабочее... в общем вот код:
| Код |
typedef HRESULT(__stdcall *IWbemClassObject_Get)(void *__this, LPCWSTR, LONG, VARIANT*, CIMTYPE*, LONG*);
IWbemClassObject_Get True_Get;
__declspec(dllexport) HRESULT __stdcall Hooked_Get(void *__this, LPCWSTR wszName, LONG lFlags, VARIANT *pVal, CIMTYPE *pvtType, LONG *plFlavor)
{
printf("wszName: %S\n", wszName);
std::cout << "lFlags:" << lFlags << std::endl;
std::cout << "pVal:" << pVal << std::endl;
std::cout << "pvtType:" << pvtType << std::endl;
std::cout << "plFlavor:" << plFlavor << std::endl;
return True_Get(__this, wszName, lFlags, pVal, pvtType, plFlavor);
}
template<typename T>
void HookFunction(const char *module, char *signature, T &fn_real, PVOID fn_mine)
{
HookFunction<T>(DetourFindFunction(module, signature), fn_real, fn_mine);
}
template<typename T>
void HookFunction(DWORD address, T &fn_real, PVOID fn_mine)
{
HookFunction<T>(reinterpret_cast<PVOID>(address), fn_real, fn_mine);
}
template<typename T>
void HookFunction(PVOID target, T &fn_real, PVOID fn_mine)
{
fn_real = reinterpret_cast<T>(target);
HookFunction<T>(fn_real, fn_mine);
}
template<typename T>
void HookFunction(T &fn_real, PVOID fn_mine)
{
DetourAttach(&(PVOID&)fn_real, fn_mine);
}
void ApplyHooks(LPVOID address) //Function address 0x6FD9B723
{
DetourTransactionBegin();
DetourUpdateThread(GetCurrentThread());
HookFunction<IWbemClassObject_Get>((FARPROC)address, True_Get, Hooked_Get);
DetourTransactionCommit();
}
BOOL APIENTRY DllMain(HINSTANCE hinst, DWORD dwReason, LPVOID reserved)
{
AllocConsole();
freopen_s((FILE**)stdout, "CONOUT$", "w", stdout);
HMODULE hLib = GetModuleHandle(L"fastprox.dll");
if (hLib)
{
FARPROC hAdd = GetProcAddress(hLib, "?Get@CWbemObject@@UAGJPBGJPAUtagVARIANT@@PAJ2@Z");
if (hAdd)
{
if (dwReason == DLL_THREAD_ATTACH)
{
std::cout << "DLL_THREAD_ATTACH" << std::endl;
DisableThreadLibraryCalls(hinst); // Почему-то если его убрать, то через какой-то время dxdiag.exe выбрасывает ошибку
CreateThread(0, 0, (LPTHREAD_START_ROUTINE)ApplyHooks, hAdd, 0, 0);
}
}
}
return TRUE;
}
|
И он мне выдает вот такой списочек в консоли:
Но есть два непонятных для меня момент -
1). Без DisableThreadLibraryCalls(hinst) в DllMain(...) через какое-то время dxdiag.exe выбивает ошибку, хотя какие-то данные в консоли отобразились.
2). Потерпел фиаско при попытке изменить какие-либо значения, в dxdiag.exe они не поменялись. Предполагаю, что это как-то связано с DisableThreadLibraryCalls... но см.п.1
Пример кода того как пытался поменять значения у строковых переменных. Причем если после изменения вывести в консоль pVal, то оно будет изменено на "NO_DATA", а в dxdiag.exe изменениями и не пахнет.
| Код |
pVal->vt = VT_BSTR;
V_BSTR(pVal) = L"NO_DATA";
|
 |
