1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
/*
entry_win.cpp
This file is part of GammaRay, the Qt application inspection and manipulation tool.
SPDX-FileCopyrightText: 2014 Klarälvdalens Datakonsult AB, a KDAB Group company <info@kdab.com>
Author: Volker Krause <volker.krause@kdab.com>
SPDX-License-Identifier: GPL-2.0-or-later
Contact KDAB at <info@kdab.com> for commercial licensing options.
*/
#include <config-gammaray.h>
#include <windows.h>
#include <string>
typedef void (*gammaray_probe_inject)(void);
namespace {
static const std::wstring LOADER_NAME = L"gammaray_winloader";
static const std::string PROBE_NAME = GAMMARAY_PROBE_BASENAME;
}
extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID /*lpvReserved*/)
{
switch (dwReason) {
case DLL_PROCESS_ATTACH: {
wchar_t buffer[MAX_PATH * 2];
const int size = GetModuleFileNameW(hInstance, buffer, MAX_PATH * 2);
if (!size) {
OutputDebugStringW(L"GammaRay: GetModuleFileNameW failed");
break;
}
const std::wstring probeName(PROBE_NAME.cbegin(), PROBE_NAME.cend());
std::wstring path(buffer, size);
path.replace(path.find(LOADER_NAME), LOADER_NAME.length(), probeName);
HMODULE probe = GetModuleHandleW(path.c_str());
if (!probe) {
probe = LoadLibraryW(path.c_str());
if (!probe) {
OutputDebugStringW(L"GammaRay: Failed to load: ");
OutputDebugStringW(path.c_str());
break;
}
}
gammaray_probe_inject inject = ( gammaray_probe_inject )GetProcAddress(probe, "gammaray_probe_inject");
if (!inject) {
OutputDebugStringW(L"GammaRay: Failed to resolve gammaray_probe_inject");
break;
}
inject();
} break;
}
// return false to get unloaded
return FALSE; // krazy:exclude=captruefalse
}
|
