-
Notifications
You must be signed in to change notification settings - Fork 30
/
imports.cpp
78 lines (64 loc) · 2.33 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
#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();
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();
TCHAR *function_name;
#ifdef UNICODE
size_t buflen;
mbstowcs_s(&buflen, NULL, 0, function, _TRUNCATE);
function_name = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, buflen * sizeof(TCHAR));
if (function_name) mbstowcs_s(&buflen, function_name, buflen, function, _TRUNCATE);
#else
function_name = function;
#endif
log_event(EVENTLOG_WARNING_TYPE, NSSM_EVENT_GETPROCADDRESS_FAILED, function_name, error_string(*error), 0);
#ifdef UNICODE
if (function_name) HeapFree(GetProcessHeap(), 0, function_name);
#endif
}
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.SleepConditionVariableCS = (SleepConditionVariableCS_ptr) get_import(imports.kernel32, "SleepConditionVariableCS", &error);
if (! imports.SleepConditionVariableCS) {
if (error != ERROR_PROC_NOT_FOUND) return 3;
}
imports.WakeConditionVariable = (WakeConditionVariable_ptr) get_import(imports.kernel32, "WakeConditionVariable", &error);
if (! imports.WakeConditionVariable) {
if (error != ERROR_PROC_NOT_FOUND) return 4;
}
}
else if (error != ERROR_MOD_NOT_FOUND) return 1;
return 0;
}
void free_imports() {
if (imports.kernel32) FreeLibrary(imports.kernel32);
ZeroMemory(&imports, sizeof(imports));
}