-
Notifications
You must be signed in to change notification settings - Fork 1
/
imports.cpp
91 lines (75 loc) · 3.07 KB
/
imports.cpp
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#include "nssm.h"
imports_t imports;
/*
Try to set up function pointers.
In this first implementation it is not an error if we can't load them
because we aren't currently trying to load any functions which we
absolutely need. If we later add some indispensible imports we can
return non-zero here to force an application exit.
*/
HMODULE get_dll(const TCHAR *dll, unsigned long *error) {
*error = 0;
HMODULE ret = LoadLibrary(dll);
if (! ret) {
*error = GetLastError();
if (*error != ERROR_PROC_NOT_FOUND) log_event(EVENTLOG_WARNING_TYPE, NSSM_EVENT_LOADLIBRARY_FAILED, dll, error_string(*error), 0);
}
return ret;
}
FARPROC get_import(HMODULE library, const char *function, unsigned long *error) {
*error = 0;
FARPROC ret = GetProcAddress(library, function);
if (! ret) {
*error = GetLastError();
if (*error != ERROR_PROC_NOT_FOUND) {
TCHAR *function_name;
if (! from_utf8(function, &function_name, 0)) {
log_event(EVENTLOG_WARNING_TYPE, NSSM_EVENT_GETPROCADDRESS_FAILED, function_name, error_string(*error), 0);
HeapFree(GetProcessHeap(), 0, function_name);
}
}
}
return ret;
}
int get_imports() {
unsigned long error;
ZeroMemory(&imports, sizeof(imports));
imports.kernel32 = get_dll(_T("kernel32.dll"), &error);
if (imports.kernel32) {
imports.AttachConsole = (AttachConsole_ptr) get_import(imports.kernel32, "AttachConsole", &error);
if (! imports.AttachConsole) {
if (error != ERROR_PROC_NOT_FOUND) return 2;
}
imports.QueryFullProcessImageName = (QueryFullProcessImageName_ptr) get_import(imports.kernel32, QUERYFULLPROCESSIMAGENAME_EXPORT, &error);
if (! imports.QueryFullProcessImageName) {
if (error != ERROR_PROC_NOT_FOUND) return 3;
}
imports.SleepConditionVariableCS = (SleepConditionVariableCS_ptr) get_import(imports.kernel32, "SleepConditionVariableCS", &error);
if (! imports.SleepConditionVariableCS) {
if (error != ERROR_PROC_NOT_FOUND) return 4;
}
imports.WakeConditionVariable = (WakeConditionVariable_ptr) get_import(imports.kernel32, "WakeConditionVariable", &error);
if (! imports.WakeConditionVariable) {
if (error != ERROR_PROC_NOT_FOUND) return 5;
}
}
else if (error != ERROR_MOD_NOT_FOUND) return 1;
imports.advapi32 = get_dll(_T("advapi32.dll"), &error);
if (imports.advapi32) {
imports.CreateWellKnownSid = (CreateWellKnownSid_ptr) get_import(imports.advapi32, "CreateWellKnownSid", &error);
if (! imports.CreateWellKnownSid) {
if (error != ERROR_PROC_NOT_FOUND) return 7;
}
imports.IsWellKnownSid = (IsWellKnownSid_ptr) get_import(imports.advapi32, "IsWellKnownSid", &error);
if (! imports.IsWellKnownSid) {
if (error != ERROR_PROC_NOT_FOUND) return 8;
}
}
else if (error != ERROR_MOD_NOT_FOUND) return 6;
return 0;
}
void free_imports() {
if (imports.kernel32) FreeLibrary(imports.kernel32);
if (imports.advapi32) FreeLibrary(imports.advapi32);
ZeroMemory(&imports, sizeof(imports));
}