1 /*
2 ** Zabbix
3 ** Copyright (C) 2001-2021 Zabbix SIA
4 **
5 ** This program is free software; you can redistribute it and/or modify
6 ** it under the terms of the GNU General Public License as published by
7 ** the Free Software Foundation; either version 2 of the License, or
8 ** (at your option) any later version.
9 **
10 ** This program is distributed in the hope that it will be useful,
11 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
12 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 ** GNU General Public License for more details.
14 **
15 ** You should have received a copy of the GNU General Public License
16 ** along with this program; if not, write to the Free Software
17 ** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 **/
19
20 #include "common.h"
21 #include "symbols.h"
22
23 #include "log.h"
24
25 DWORD (__stdcall *zbx_GetGuiResources)(HANDLE, DWORD) = NULL;
26 BOOL (__stdcall *zbx_GetProcessIoCounters)(HANDLE, PIO_COUNTERS) = NULL;
27 BOOL (__stdcall *zbx_GetPerformanceInfo)(PPERFORMANCE_INFORMATION, DWORD) = NULL;
28 BOOL (__stdcall *zbx_GlobalMemoryStatusEx)(LPMEMORYSTATUSEX) = NULL;
29 BOOL (__stdcall *zbx_GetFileInformationByHandleEx)(HANDLE, FILE_INFO_BY_HANDLE_CLASS, LPVOID, DWORD) = NULL;
30
GetProcAddressAndLog(HMODULE hModule,const char * procName)31 static FARPROC GetProcAddressAndLog(HMODULE hModule, const char *procName)
32 {
33 FARPROC ptr;
34
35 if (NULL == (ptr = GetProcAddress(hModule, procName)))
36 zabbix_log(LOG_LEVEL_DEBUG, "unable to resolve symbol '%s'", procName);
37
38 return ptr;
39 }
40
import_symbols()41 void import_symbols()
42 {
43 HMODULE hModule;
44
45 if (NULL != (hModule = GetModuleHandle(TEXT("USER32.DLL"))))
46 zbx_GetGuiResources = (DWORD (__stdcall *)(HANDLE, DWORD))GetProcAddressAndLog(hModule, "GetGuiResources");
47 else
48 zabbix_log(LOG_LEVEL_DEBUG, "unable to get handle to USER32.DLL");
49
50 if (NULL != (hModule = GetModuleHandle(TEXT("KERNEL32.DLL"))))
51 {
52 zbx_GetProcessIoCounters = (BOOL (__stdcall *)(HANDLE, PIO_COUNTERS))GetProcAddressAndLog(hModule, "GetProcessIoCounters");
53 zbx_GlobalMemoryStatusEx = (BOOL (__stdcall *)(LPMEMORYSTATUSEX))GetProcAddressAndLog(hModule, "GlobalMemoryStatusEx");
54 zbx_GetFileInformationByHandleEx = (BOOL (__stdcall *)(HANDLE, FILE_INFO_BY_HANDLE_CLASS, LPVOID,
55 DWORD))GetProcAddressAndLog(hModule, "GetFileInformationByHandleEx");
56 }
57 else
58 zabbix_log(LOG_LEVEL_DEBUG, "unable to get handle to KERNEL32.DLL");
59
60 if (NULL != (hModule = GetModuleHandle(TEXT("PSAPI.DLL"))))
61 zbx_GetPerformanceInfo = (BOOL (__stdcall *)(PPERFORMANCE_INFORMATION, DWORD))GetProcAddressAndLog(hModule, "GetPerformanceInfo");
62 else
63 zabbix_log(LOG_LEVEL_DEBUG, "unable to get handle to PSAPI.DLL");
64 }
65