summaryrefslogtreecommitdiffhomepage
path: root/src/meta/pe_get_symbol_module_info.c
blob: 785a314856ac408ec63cd8bb2b8a226089b2c932 (plain)
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
/*****************************************************************************/
/*  pemagination: a (virtual) tour into portable bits and executable bytes   */
/*  Copyright (C) 2013--2016  Z. Gilboa                                      */
/*  Released under GPLv2 and GPLv3; see COPYING.PEMAGINE.                    */
/*****************************************************************************/

#include <psxtypes/psxtypes.h>
#include <pemagine/pemagine.h>


struct pe_module_info_ctx {
	uintptr_t			sym_addr;
	struct pe_ldr_tbl_entry *	image_ldr_tbl_entry;
};



static int pe_get_symbol_module_info_callback(
	struct pe_ldr_tbl_entry *	ldr_tbl_entry,
	enum pe_callback_reason		reason,
	void *				context)
{
	uintptr_t			image_base;
	uintptr_t			image_size;
	struct pe_module_info_ctx *	ctx;

	if (reason != PE_CALLBACK_REASON_ITEM)
		return 1;

	ctx = (struct pe_module_info_ctx *)context;
	image_base = (uintptr_t)(ldr_tbl_entry->dll_base);
	image_size = (uintptr_t)(ldr_tbl_entry->size_of_image);

	if ((ctx->sym_addr > image_base) && (ctx->sym_addr < image_base + image_size)) {
		/* within bounds */
		ctx->image_ldr_tbl_entry = ldr_tbl_entry;
		return 0;
	} else
		return 1;
}


struct pe_ldr_tbl_entry * pe_get_symbol_module_info(const void * sym_addr)
{
	struct pe_module_info_ctx ctx;

	ctx.sym_addr = (uintptr_t)sym_addr;
	ctx.image_ldr_tbl_entry = 0;

	pe_enum_modules_in_load_order(
		pe_get_symbol_module_info_callback,
		&ctx);

	return ctx.image_ldr_tbl_entry;
}