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
92
93
94
95
96
97
|
/*
* Copyright (c) 2009, Giampaolo Rodola'. All rights reserved.
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include <stdio.h>
#include <windows.h>
#include <Python.h>
#include "ntextapi.h"
#include "global.h"
// A wrapper around GetModuleHandle and GetProcAddress.
static PVOID
ps_GetProcAddress(LPCSTR libname, LPCSTR procname) {
HMODULE mod;
FARPROC addr;
if ((mod = GetModuleHandleA(libname)) == NULL) {
PyErr_SetFromWindowsErrWithFilename(0, libname);
return NULL;
}
if ((addr = GetProcAddress(mod, procname)) == NULL) {
PyErr_SetFromWindowsErrWithFilename(0, procname);
return NULL;
}
return addr;
}
// A wrapper around LoadLibrary and GetProcAddress.
static PVOID
ps_GetProcAddressFromLib(LPCSTR libname, LPCSTR procname) {
HMODULE mod;
FARPROC addr;
if ((mod = LoadLibraryA(libname)) == NULL) {
PyErr_SetFromWindowsErrWithFilename(0, libname);
return NULL;
}
if ((addr = GetProcAddress(mod, procname)) == NULL) {
PyErr_SetFromWindowsErrWithFilename(0, procname);
FreeLibrary(mod);
return NULL;
}
FreeLibrary(mod);
return addr;
}
int
psutil_load_globals() {
// Mandatory.
psutil_NtQuerySystemInformation = ps_GetProcAddressFromLib(
"ntdll.dll", "NtQuerySystemInformation");
if (psutil_NtQuerySystemInformation == NULL)
return 1;
psutil_NtQueryInformationProcess = ps_GetProcAddress(
"ntdll.dll", "NtQueryInformationProcess");
if (! psutil_NtQueryInformationProcess)
return 1;
psutil_NtSetInformationProcess = ps_GetProcAddress(
"ntdll.dll", "NtSetInformationProcess");
if (! psutil_NtSetInformationProcess)
return 1;
psutil_WinStationQueryInformationW = ps_GetProcAddressFromLib(
"winsta.dll", "WinStationQueryInformationW");
if (! psutil_WinStationQueryInformationW)
return 1;
psutil_rtlIpv4AddressToStringA = ps_GetProcAddressFromLib(
"ntdll.dll", "RtlIpv4AddressToStringA");
if (! psutil_rtlIpv4AddressToStringA)
return 1;
psutil_rtlIpv6AddressToStringA = ps_GetProcAddressFromLib(
"ntdll.dll", "RtlIpv6AddressToStringA");
if (! psutil_rtlIpv6AddressToStringA)
return 1;
psutil_GetExtendedTcpTable = ps_GetProcAddressFromLib(
"iphlpapi.dll", "GetExtendedTcpTable");
if (! psutil_GetExtendedTcpTable)
return 1;
// Optionals.
psutil_GetActiveProcessorCount = ps_GetProcAddress(
"kernel32", "GetActiveProcessorCount");
return 0;
}
|