|
 |
feffc7 |
/*****************************************************************************/
|
|
|
feffc7 |
/* pemagination: a (virtual) tour into portable bits and executable bytes */
|
|
|
feffc7 |
/* Copyright (C) 2013,2014,2015 Z. Gilboa */
|
|
|
feffc7 |
/* Released under GPLv2 and GPLv3; see COPYING.PEMAGINE. */
|
|
|
feffc7 |
/*****************************************************************************/
|
|
|
feffc7 |
|
|
|
feffc7 |
#include <psxtypes/psxtypes.h>
|
|
|
feffc7 |
#include <pemagine/pe_consts.h>
|
|
|
feffc7 |
#include <pemagine/pe_structs.h>
|
|
|
feffc7 |
#include <pemagine/pemagine.h>
|
|
|
feffc7 |
#include "pe_impl.h"
|
|
|
feffc7 |
|
|
|
feffc7 |
pe_api
|
|
|
feffc7 |
void * pe_get_first_module_handle(void)
|
|
|
feffc7 |
{
|
|
|
feffc7 |
struct pe_peb_ldr_data * peb_ldr_data;
|
|
|
feffc7 |
struct pe_list_entry * peb_list_entry;
|
|
|
feffc7 |
intptr_t peb_tbl_addr;
|
|
|
feffc7 |
struct pe_ldr_tbl_entry * peb_ldr_tbl_entry;
|
|
|
feffc7 |
|
|
|
feffc7 |
peb_ldr_data = (struct pe_peb_ldr_data *)pe_get_peb_ldr_data_address();
|
|
|
feffc7 |
peb_list_entry = (struct pe_list_entry *)peb_ldr_data->in_load_order_module_list.flink;
|
|
|
feffc7 |
peb_tbl_addr = (intptr_t)peb_list_entry - IN_LOAD_ORDER_MODULE_LIST_OFFSET;
|
|
|
feffc7 |
peb_ldr_tbl_entry = (struct pe_ldr_tbl_entry *)peb_tbl_addr;
|
|
|
feffc7 |
|
|
|
feffc7 |
return peb_ldr_tbl_entry->dll_base;
|
|
|
feffc7 |
}
|
|
|
feffc7 |
|
|
|
feffc7 |
pe_api
|
|
|
feffc7 |
void * pe_get_module_handle(const wchar16_t * name)
|
|
|
feffc7 |
{
|
|
|
feffc7 |
wchar16_t * src;
|
|
|
feffc7 |
wchar16_t * dst;
|
|
|
feffc7 |
|
|
|
feffc7 |
size_t match;
|
|
|
feffc7 |
size_t len;
|
|
|
feffc7 |
|
|
|
feffc7 |
struct pe_peb_ldr_data * peb_ldr_data;
|
|
|
feffc7 |
struct pe_list_entry * plist_head;
|
|
|
feffc7 |
|
|
|
feffc7 |
struct pe_list_entry * plist_current;
|
|
|
feffc7 |
struct pe_ldr_tbl_entry * ldr_tbl_entry = 0;
|
|
|
feffc7 |
|
|
|
feffc7 |
peb_ldr_data = (struct pe_peb_ldr_data *)pe_get_peb_ldr_data_address();
|
|
|
feffc7 |
plist_head = (struct pe_list_entry *)peb_ldr_data->in_load_order_module_list.flink;
|
|
|
feffc7 |
plist_current = plist_head;
|
|
|
feffc7 |
|
|
|
feffc7 |
len = pe_impl_strlen_utf16(name);
|
|
|
feffc7 |
|
|
|
feffc7 |
do {
|
|
|
feffc7 |
ldr_tbl_entry = (struct pe_ldr_tbl_entry *)plist_current - IN_LOAD_ORDER_MODULE_LIST_OFFSET;
|
|
|
feffc7 |
|
|
|
feffc7 |
if (ldr_tbl_entry->base_dll_name.strlen == len)
|
|
|
feffc7 |
dst = (wchar16_t *)ldr_tbl_entry->base_dll_name.buffer;
|
|
|
feffc7 |
else if (ldr_tbl_entry->full_dll_name.strlen == len)
|
|
|
feffc7 |
dst = (wchar16_t *)ldr_tbl_entry->full_dll_name.buffer;
|
|
|
feffc7 |
else
|
|
|
feffc7 |
dst = (wchar16_t *)0;
|
|
|
feffc7 |
|
|
|
feffc7 |
if ((intptr_t)(dst)) {
|
|
|
feffc7 |
src = (wchar16_t *)name;
|
|
|
feffc7 |
match = 0;
|
|
|
feffc7 |
|
|
|
feffc7 |
while ((match < len)
|
|
|
feffc7 |
&& ((pe_impl_utf16_char_to_lower(*src)) == (pe_impl_utf16_char_to_lower(*dst)))) {
|
|
|
feffc7 |
src = (wchar16_t *)pe_va_from_rva(src,sizeof(wchar16_t));
|
|
|
feffc7 |
dst = (wchar16_t *)pe_va_from_rva(dst,sizeof(wchar16_t));
|
|
|
feffc7 |
match+=sizeof(wchar16_t);
|
|
|
feffc7 |
}
|
|
|
feffc7 |
|
|
|
feffc7 |
if (match == len)
|
|
|
feffc7 |
return ldr_tbl_entry->dll_base;
|
|
|
feffc7 |
}
|
|
|
feffc7 |
|
|
|
feffc7 |
plist_current = plist_current->flink;
|
|
|
feffc7 |
} while (plist_current != plist_head);
|
|
|
feffc7 |
|
|
|
feffc7 |
/* address not found */
|
|
|
feffc7 |
return 0;
|
|
|
feffc7 |
}
|