* and improve reusability
*/
-#include <ddk/ntddk.h>
-#include <debug.h>
#include <stddef.h>
+#define NTOS_MODE_USER
+#include <ntos.h>
+
+#define NDEBUG
+#include <debug.h>
#include <epsapi.h>
)
{
register NTSTATUS nErrCode = STATUS_SUCCESS;
- PSYSTEM_MODULES psmModules = NULL;
+ PSYSTEM_MODULE_INFORMATION psmModules = NULL;
#if 0
__try
NTAPI
PsaCaptureSystemModules
(
- OUT PSYSTEM_MODULES * SystemModules
+ OUT PSYSTEM_MODULE_INFORMATION * SystemModules
)
{
SIZE_T nSize = 0;
register NTSTATUS nErrCode;
- register PSYSTEM_MODULES psmModules = (PSYSTEM_MODULES)&nSize;
+ register PSYSTEM_MODULE_INFORMATION psmModules = (PSYSTEM_MODULE_INFORMATION)&nSize;
#if 0
__try
NTAPI
PsaWalkSystemModules
(
- IN PSYSTEM_MODULES SystemModules,
+ IN PSYSTEM_MODULE_INFORMATION SystemModules,
IN PSYSMOD_ENUM_ROUTINE Callback,
IN OUT PVOID CallbackContext
)
for(i = 0; i < SystemModules->Count; ++ i)
{
/* return current module to the callback */
- nErrCode = Callback(&(SystemModules->Modules[i]), CallbackContext);
+ nErrCode = Callback(&(SystemModules->Module[i]), CallbackContext);
if(!NT_SUCCESS(nErrCode))
/* failure */
return STATUS_SUCCESS;
}
-PSYSTEM_MODULE_INFORMATION
+PSYSTEM_MODULE_INFORMATION_ENTRY
FASTCALL
PsaWalkFirstSystemModule
(
- IN PSYSTEM_MODULES SystemModules
+ IN PSYSTEM_MODULE_INFORMATION SystemModules
)
{
- return &(SystemModules->Modules[0]);
+ return &(SystemModules->Module[0]);
}
-PSYSTEM_MODULE_INFORMATION
+PSYSTEM_MODULE_INFORMATION_ENTRY
FASTCALL
PsaWalkNextSystemModule
(
IN PSYSTEM_MODULE_INFORMATION CurrentSystemModule
)
{
- return (PSYSTEM_MODULE_INFORMATION)
+ return (PSYSTEM_MODULE_INFORMATION_ENTRY)
(
(ULONG_PTR)CurrentSystemModule +
(
- offsetof(SYSTEM_MODULES, Modules[1]) -
- offsetof(SYSTEM_MODULES, Modules[0])
+ offsetof(SYSTEM_MODULE_INFORMATION, Module[1]) -
+ offsetof(SYSTEM_MODULE_INFORMATION, Module[0])
)
);
}