mirror of
https://github.com/EricPlayZ/EGameTools.git
synced 2025-07-18 17:37:53 +08:00
305 lines
8.8 KiB
C++
305 lines
8.8 KiB
C++
#define WIN32_LEAN_AND_MEAN
|
|
#ifndef _CRT_SECURE_NO_WARNINGS
|
|
#define _CRT_SECURE_NO_WARNINGS
|
|
#endif
|
|
|
|
#include <assert.h>
|
|
#include <windows.h>
|
|
#include <tchar.h>
|
|
#include <stdio.h>
|
|
#include <malloc.h>
|
|
|
|
#include "../MemoryModule.h"
|
|
|
|
typedef int (*addProc)(int);
|
|
typedef int (*addNumberProc)(int, int);
|
|
|
|
// Thanks to Tim Cooper (from http://stackoverflow.com/a/8584708)
|
|
const char *sstrstr(const char *haystack, const char *needle, size_t length) {
|
|
size_t needle_length = strlen(needle);
|
|
size_t i;
|
|
|
|
for (i = 0; i < length; i++) {
|
|
if (i + needle_length > length) {
|
|
return NULL;
|
|
}
|
|
|
|
if (strncmp(&haystack[i], needle, needle_length) == 0) {
|
|
return &haystack[i];
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
const wchar_t *swcsstr(const wchar_t *haystack, const wchar_t *needle, size_t length) {
|
|
size_t needle_length = wcslen(needle);
|
|
size_t i;
|
|
|
|
for (i = 0; i < length; i++) {
|
|
if (i + needle_length > length) {
|
|
return NULL;
|
|
}
|
|
|
|
if (wcsncmp(&haystack[i], needle, needle_length) == 0) {
|
|
return &haystack[i];
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
BOOL CheckResourceStrings(LPVOID data, DWORD size, const char *first, const wchar_t *second) {
|
|
const char *first_pos;
|
|
const wchar_t *second_pos;
|
|
const wchar_t *src;
|
|
|
|
if (data == NULL || size == 0) {
|
|
return FALSE;
|
|
}
|
|
|
|
first_pos = sstrstr((const char *) data, first, size);
|
|
if (first_pos == NULL) {
|
|
fprintf(stderr, "ERROR: data doesn't start with %s\n", first);
|
|
return FALSE;
|
|
}
|
|
|
|
src = (const wchar_t *) (((const char *) data) + strlen(first) + 1);
|
|
second_pos = swcsstr(src, second, (size - strlen(first) - 1) / sizeof(wchar_t));
|
|
if (second_pos == NULL) {
|
|
fwprintf(stderr, L"ERROR: data doesn't continue with %s\n", second);
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL LoadFromMemory(char *filename)
|
|
{
|
|
FILE *fp;
|
|
unsigned char *data=NULL;
|
|
long size;
|
|
size_t read;
|
|
HMEMORYMODULE handle = NULL;
|
|
addNumberProc addNumber;
|
|
addNumberProc addNumber2;
|
|
HMEMORYRSRC resourceInfo;
|
|
DWORD resourceSize;
|
|
LPVOID resourceData;
|
|
TCHAR buffer[100];
|
|
BOOL result = TRUE;
|
|
|
|
fp = fopen(filename, "rb");
|
|
if (fp == NULL)
|
|
{
|
|
printf("Can't open DLL file \"%s\".", filename);
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
|
|
fseek(fp, 0, SEEK_END);
|
|
size = ftell(fp);
|
|
assert(size > 0);
|
|
data = (unsigned char *)malloc(size);
|
|
assert(data != NULL);
|
|
fseek(fp, 0, SEEK_SET);
|
|
read = fread(data, 1, size, fp);
|
|
assert(read == static_cast<size_t>(size));
|
|
fclose(fp);
|
|
|
|
handle = MemoryLoadLibrary(data, size);
|
|
if (handle == NULL)
|
|
{
|
|
_tprintf(_T("Can't load library from memory.\n"));
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
|
|
addNumber = (addNumberProc)MemoryGetProcAddress(handle, NULL);
|
|
if (addNumber != NULL) {
|
|
_tprintf(_T("MemoryGetProcAddress(NULL) returned %p\n"), addNumber);
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
|
|
addNumber = (addNumberProc)MemoryGetProcAddress(handle, reinterpret_cast<LPCSTR>(0xff));
|
|
if (addNumber != NULL) {
|
|
_tprintf(_T("MemoryGetProcAddress(0xff) returned %p\n"), addNumber);
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
|
|
addNumber = (addNumberProc)MemoryGetProcAddress(handle, "addNumbers");
|
|
if (!addNumber) {
|
|
_tprintf(_T("MemoryGetProcAddress(\"addNumber\") returned NULL\n"));
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
_tprintf(_T("From memory: %d\n"), addNumber(1, 2));
|
|
|
|
// the DLL only exports one function, try to load by ordinal value
|
|
addNumber2 = (addNumberProc)MemoryGetProcAddress(handle, reinterpret_cast<LPCSTR>(0x01));
|
|
if (addNumber != addNumber2) {
|
|
_tprintf(_T("MemoryGetProcAddress(0x01) returned %p (expected %p)\n"), addNumber2, addNumber);
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
|
|
resourceInfo = MemoryFindResource(handle, MAKEINTRESOURCE(VS_VERSION_INFO), RT_VERSION);
|
|
_tprintf(_T("MemoryFindResource returned 0x%p\n"), resourceInfo);
|
|
|
|
if (resourceInfo != NULL) {
|
|
resourceSize = MemorySizeofResource(handle, resourceInfo);
|
|
resourceData = MemoryLoadResource(handle, resourceInfo);
|
|
_tprintf(_T("Memory resource data: %ld bytes at 0x%p\n"), resourceSize, resourceData);
|
|
|
|
MemoryLoadString(handle, 1, buffer, sizeof(buffer));
|
|
_tprintf(_T("String1: %s\n"), buffer);
|
|
|
|
MemoryLoadString(handle, 20, buffer, sizeof(buffer));
|
|
_tprintf(_T("String2: %s\n"), buffer);
|
|
} else {
|
|
result = FALSE;
|
|
}
|
|
|
|
resourceInfo = MemoryFindResource(handle, _T("stringres"), RT_RCDATA);
|
|
_tprintf(_T("MemoryFindResource returned 0x%p\n"), resourceInfo);
|
|
if (resourceInfo != NULL) {
|
|
resourceSize = MemorySizeofResource(handle, resourceInfo);
|
|
resourceData = MemoryLoadResource(handle, resourceInfo);
|
|
|
|
_tprintf(_T("Memory resource data: %ld bytes at 0x%p\n"), resourceSize, resourceData);
|
|
if (!CheckResourceStrings(resourceData, resourceSize, "This is a ANSI string", L"This is a UNICODE string")) {
|
|
result = FALSE;
|
|
}
|
|
} else {
|
|
result = FALSE;
|
|
}
|
|
|
|
resourceInfo = MemoryFindResource(handle, _T("stringres1"), RT_RCDATA);
|
|
_tprintf(_T("MemoryFindResource returned 0x%p\n"), resourceInfo);
|
|
if (resourceInfo != NULL) {
|
|
resourceSize = MemorySizeofResource(handle, resourceInfo);
|
|
resourceData = MemoryLoadResource(handle, resourceInfo);
|
|
|
|
_tprintf(_T("Memory resource data: %ld bytes at 0x%p\n"), resourceSize, resourceData);
|
|
if (!CheckResourceStrings(resourceData, resourceSize, "This is ANSI string 1", L"This is UNICODE string 1")) {
|
|
result = FALSE;
|
|
}
|
|
} else {
|
|
result = FALSE;
|
|
}
|
|
|
|
|
|
resourceInfo = MemoryFindResource(handle, _T("stringres2"), RT_RCDATA);
|
|
_tprintf(_T("MemoryFindResource returned 0x%p\n"), resourceInfo);
|
|
if (resourceInfo != NULL) {
|
|
resourceSize = MemorySizeofResource(handle, resourceInfo);
|
|
resourceData = MemoryLoadResource(handle, resourceInfo);
|
|
|
|
_tprintf(_T("Memory resource data: %ld bytes at 0x%p\n"), resourceSize, resourceData);
|
|
if (!CheckResourceStrings(resourceData, resourceSize, "This is ANSI string 2", L"This is UNICODE string 2")) {
|
|
result = FALSE;
|
|
}
|
|
} else {
|
|
result = FALSE;
|
|
}
|
|
|
|
|
|
resourceInfo = MemoryFindResource(handle, _T("stringres3"), RT_RCDATA);
|
|
_tprintf(_T("MemoryFindResource returned 0x%p\n"), resourceInfo);
|
|
if (resourceInfo != NULL) {
|
|
resourceSize = MemorySizeofResource(handle, resourceInfo);
|
|
resourceData = MemoryLoadResource(handle, resourceInfo);
|
|
|
|
_tprintf(_T("Memory resource data: %ld bytes at 0x%p\n"), resourceSize, resourceData);
|
|
if (!CheckResourceStrings(resourceData, resourceSize, "This is ANSI string 3", L"This is UNICODE string 3")) {
|
|
result = FALSE;
|
|
}
|
|
} else {
|
|
result = FALSE;
|
|
}
|
|
|
|
exit:
|
|
MemoryFreeLibrary(handle);
|
|
free(data);
|
|
return result;
|
|
}
|
|
|
|
BOOL LoadExportsFromMemory(char *filename)
|
|
{
|
|
FILE *fp;
|
|
unsigned char *data=NULL;
|
|
long size;
|
|
size_t read;
|
|
HMEMORYMODULE handle = NULL;
|
|
int i;
|
|
BOOL result = TRUE;
|
|
|
|
fp = fopen(filename, "rb");
|
|
if (fp == NULL)
|
|
{
|
|
printf("Can't open DLL file \"%s\".", filename);
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
|
|
fseek(fp, 0, SEEK_END);
|
|
size = ftell(fp);
|
|
assert(size > 0);
|
|
data = (unsigned char *)malloc(size);
|
|
assert(data != NULL);
|
|
fseek(fp, 0, SEEK_SET);
|
|
read = fread(data, 1, size, fp);
|
|
assert(read == static_cast<size_t>(size));
|
|
fclose(fp);
|
|
|
|
handle = MemoryLoadLibrary(data, size);
|
|
if (handle == NULL)
|
|
{
|
|
_tprintf(_T("Can't load library from memory.\n"));
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
|
|
for (i = 1; i <= 100; i++) {
|
|
char name[100];
|
|
sprintf(name, "add%d", i);
|
|
addProc addNumber = (addProc)MemoryGetProcAddress(handle, name);
|
|
if (!addNumber) {
|
|
_tprintf(_T("MemoryGetProcAddress(\"%s\") returned NULL\n"), name);
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
int result = addNumber(1);
|
|
if (result != 1 + i) {
|
|
_tprintf(_T("(\"%s\") returned %d, expected %d\n"), name, result, 1 + i);
|
|
result = FALSE;
|
|
goto exit;
|
|
}
|
|
_tprintf(_T("%s: %d\n"), name, result);
|
|
}
|
|
exit:
|
|
MemoryFreeLibrary(handle);
|
|
free(data);
|
|
return result;
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
if (argc < 2) {
|
|
fprintf(stderr, "USAGE: %s <filename.dll>\n", argv[0]);
|
|
return 1;
|
|
}
|
|
|
|
if (!strstr((const char *) argv[1], "exports")) {
|
|
if (!LoadFromMemory(argv[1])) {
|
|
return 2;
|
|
}
|
|
} else {
|
|
if (!LoadExportsFromMemory(argv[1])) {
|
|
return 2;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|