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
98
99
100
101
102
|
/*****************************************************************************/
/* pemagination: a (virtual) tour into portable bits and executable bytes */
/* Copyright (C) 2013,2014,2015 Z. Gilboa */
/* Released under GPLv2 and GPLv3; see COPYING.PEMAGINE. */
/*****************************************************************************/
#include <psxtypes/psxtypes.h>
#include <pemagine/pe_consts.h>
#include <pemagine/pe_structs.h>
#include <pemagine/pemagine.h>
#include "pe_impl.h"
static int pe_enum_modules(
pe_enum_modules_callback * callback,
const uintptr_t in_order_member_offset,
void * context)
{
int ret;
uintptr_t peb_tbl_addr;
struct pe_peb_ldr_data * peb_ldr_data;
struct pe_ldr_tbl_entry * ldr_tbl_entry;
struct pe_list_entry * plist_head;
struct pe_list_entry * plist_current;
peb_ldr_data = (struct pe_peb_ldr_data *)pe_get_peb_ldr_data_address();
plist_head = (struct pe_list_entry *)0;
if ((intptr_t)peb_ldr_data == 0) {
callback(
0,
PE_CALLBACK_REASON_ERROR,
context);
return -1;
}
ret = callback(
0,
PE_CALLBACK_REASON_INIT,
context);
if (ret <= 0)
return ret;
switch (in_order_member_offset) {
case IN_LOAD_ORDER_MODULE_LIST_OFFSET:
plist_head = peb_ldr_data->in_load_order_module_list.flink;
break;
case IN_MEMORY_ORDER_MODULE_LIST_OFFSET:
plist_head = peb_ldr_data->in_memory_order_module_list.flink;
break;
case IN_INITIALIZATION_ORDER_MODULE_LIST_OFFSET:
plist_head = peb_ldr_data->in_init_order_module_list.flink;
break;
}
plist_current = plist_head;
do
{
peb_tbl_addr = (uintptr_t)plist_current - in_order_member_offset;
ldr_tbl_entry = (struct pe_ldr_tbl_entry *)peb_tbl_addr;
ret = callback(
ldr_tbl_entry,
PE_CALLBACK_REASON_ITEM,
context);
if (ret <= 0)
return ret;
else
plist_current = plist_current->flink;
} while (plist_current != plist_head);
ret = callback(
ldr_tbl_entry,
PE_CALLBACK_REASON_DONE,
context);
return ret;
}
pe_api
int pe_enum_modules_in_load_order(pe_enum_modules_callback * fn, void * ctx)
{
return pe_enum_modules(fn,IN_LOAD_ORDER_MODULE_LIST_OFFSET,ctx);
}
pe_api
int pe_enum_modules_in_memory_order(pe_enum_modules_callback * fn, void * ctx)
{
return pe_enum_modules(fn,IN_MEMORY_ORDER_MODULE_LIST_OFFSET,ctx);
}
pe_api
int pe_enum_modules_in_init_order(pe_enum_modules_callback * fn, void * ctx)
{
return pe_enum_modules(fn,IN_INITIALIZATION_ORDER_MODULE_LIST_OFFSET,ctx);
}
|
