diff --git a/loader/loader.cpp b/loader/loader.cpp index 5dac2e5..9ae815a 100644 --- a/loader/loader.cpp +++ b/loader/loader.cpp @@ -1,384 +1,405 @@ -/** - * vim: set ts=4 sw=4 tw=99 noet : - * ====================================================== - * Metamod:Source - * Copyright (C) 2004-2010 AlliedModders LLC and authors. - * All rights reserved. - * ====================================================== - * - * This software is provided 'as-is', without any express or implied warranty. - * In no event will the authors be held liable for any damages arising from - * the use of this software. - * - * Permission is granted to anyone to use this software for any purpose, - * including commercial applications, and to alter it and redistribute it - * freely, subject to the following restrictions: - * - * 1. The origin of this software must not be misrepresented; you must not - * claim that you wrote the original software. If you use this software in a - * product, an acknowledgment in the product documentation would be - * appreciated but is not required. - * 2. Altered source versions must be plainly marked as such, and must not be - * misrepresented as being the original software. - * 3. This notice may not be removed or altered from any source distribution. - */ - -#include -#include -#include -#include -#include -#include -#include "loader.h" -#include "serverplugin.h" -#include "gamedll.h" -#include "utility.h" -#if defined __APPLE__ -#include -#endif - -static HMODULE mm_library = NULL; -static char mm_fatal_logfile[PLATFORM_MAX_PATH] = "metamod-fatal.log"; -MetamodBackend mm_backend = MMBackend_UNKNOWN; - -extern void -mm_LogFatal(const char *message, ...) -{ - FILE *fp; - time_t t; - va_list ap; - char header[256]; - - fp = fopen(mm_fatal_logfile, "at"); - if (!fp && (fp = fopen("metamod-fatal.log", "at")) == NULL) - return; - - t = time(NULL); - strftime(header, sizeof(header), "%m/%d/%Y - %H:%M:%S", localtime(&t)); - fprintf(fp, "L %s: ", header); - - va_start(ap, message); - vfprintf(fp, message, ap); - va_end(ap); - - fprintf(fp, "\n"); - - fclose(fp); -} - -static const char *backend_names[] = -{ - "1.ep1", - "2.darkm", - "2.ep2", - "2.bgt", - "2.eye", - "2.css", - "2.ep2v", - "2.l4d", - "2.l4d2", - "2.swarm", - "2.portal2", - "2.csgo", - "2.dota", - "2.hl2dm", - "2.dods", - "2.tf2", - "2.nd", - "2.sdk2013", - "2.blade", - "2.insurgency", - "2.contagion", -}; - -#if defined _WIN32 -#define LIBRARY_EXT ".dll" -#define LIBRARY_MINEXT ".dll" -#elif defined __APPLE__ -#define LIBRARY_EXT ".dylib" -#define LIBRARY_MINEXT ".dylib" +/** + * vim: set ts=4 sw=4 tw=99 noet : + * ====================================================== + * Metamod:Source + * Copyright (C) 2004-2010 AlliedModders LLC and authors. + * All rights reserved. + * ====================================================== + * + * This software is provided 'as-is', without any express or implied warranty. + * In no event will the authors be held liable for any damages arising from + * the use of this software. + * + * Permission is granted to anyone to use this software for any purpose, + * including commercial applications, and to alter it and redistribute it + * freely, subject to the following restrictions: + * + * 1. The origin of this software must not be misrepresented; you must not + * claim that you wrote the original software. If you use this software in a + * product, an acknowledgment in the product documentation would be + * appreciated but is not required. + * 2. Altered source versions must be plainly marked as such, and must not be + * misrepresented as being the original software. + * 3. This notice may not be removed or altered from any source distribution. + */ + +#include +#include +#include +#include +#include +#include +#include "loader.h" +#include "serverplugin.h" +#include "gamedll.h" +#include "utility.h" +#if defined __APPLE__ +#include +#endif + +static HMODULE mm_library = NULL; +static char mm_fatal_logfile[PLATFORM_MAX_PATH] = "metamod-fatal.log"; +MetamodBackend mm_backend = MMBackend_UNKNOWN; + +extern void +mm_LogFatal(const char *message, ...) +{ + FILE *fp; + time_t t; + va_list ap; + char header[256]; + + fp = fopen(mm_fatal_logfile, "at"); + if (!fp && (fp = fopen("metamod-fatal.log", "at")) == NULL) + return; + + t = time(NULL); + strftime(header, sizeof(header), "%m/%d/%Y - %H:%M:%S", localtime(&t)); + fprintf(fp, "L %s: ", header); + + va_start(ap, message); + vfprintf(fp, message, ap); + va_end(ap); + + fprintf(fp, "\n"); + + fclose(fp); +} + +static const char *backend_names[] = +{ + "1.ep1", + "2.darkm", + "2.ep2", + "2.bgt", + "2.eye", + "2.css", + "2.ep2v", + "2.l4d", + "2.l4d2", + "2.swarm", + "2.portal2", + "2.csgo", + "2.dota", + "2.hl2dm", + "2.dods", + "2.tf2", + "2.nd", + "2.sdk2013", + "2.blade", + "2.insurgency", + "2.contagion", +}; + +#if defined _WIN32 +#define LIBRARY_EXT ".dll" +#define LIBRARY_MINEXT ".dll" +#elif defined __APPLE__ +#define LIBRARY_EXT ".dylib" +#define LIBRARY_MINEXT ".dylib" #elif defined __linux__ #define LIBRARY_EXT LIB_SUFFIX -#define LIBRARY_MINEXT ".so" -#endif - -bool -mm_LoadMetamodLibrary(MetamodBackend backend, char *buffer, size_t maxlength) -{ - size_t len, temp_len; - char mm_path[PLATFORM_MAX_PATH * 2]; - - /* Get our path */ - if (!mm_GetFileOfAddress((void*)mm_GetFileOfAddress, mm_path, sizeof(mm_path))) - return false; - - len = strlen(mm_path); - temp_len = strlen("server" LIBRARY_EXT); - if (len < temp_len) - return false; - - /* Build log file name */ - mm_path[len - temp_len] = '\0'; - mm_Format(mm_fatal_logfile, - sizeof(mm_fatal_logfile), - "%smetamod-fatal.log", - mm_path); - - /* Replace server.dll with the new binary we want */ - mm_Format(&mm_path[len - temp_len], - sizeof(mm_path) - (len - temp_len), - "metamod.%s" LIBRARY_MINEXT, - backend_names[backend]); - - mm_library = (HMODULE)mm_LoadLibrary(mm_path, buffer, maxlength); - - return (mm_library != NULL); -} - -void -mm_UnloadMetamodLibrary() -{ - mm_UnloadLibrary(mm_library); - mm_library = NULL; -} - -#if defined _WIN32 -#define EXPORT extern "C" __declspec(dllexport) -#elif defined __GNUC__ -#if __GNUC__ == 4 -#define EXPORT extern "C" __attribute__ ((visibility("default"))) -#else -#define EXPORT extern "C" -#endif -#endif - -EXPORT void * -CreateInterface(const char *name, int *ret) -{ - /* If we've got a VSP bridge, do nothing. */ - if (vsp_bridge != NULL) - { - if (ret != NULL) - *ret = 1; - return NULL; - } - - void *ptr; - if (strncmp(name, "ISERVERPLUGINCALLBACKS", 22) == 0) - { - /* Either load as VSP or start VSP listener */ - ptr = mm_GetVspCallbacks(atoi(&name[22])); - } - else if (gamedll_bridge == NULL) - { - /* Load as gamedll */ - ptr = mm_GameDllRequest(name, ret); - } - else - { - /* If we've got a gamedll bridge, forward the request. */ - return gamedll_bridge->QueryInterface(name, ret); - } - - if (ret != NULL) - *ret = (ptr != NULL) ? 0 : 1; - - return ptr; -} - -void * -mm_GetProcAddress(const char *name) -{ - return mm_GetLibAddress(mm_library, name); -} - -void -mm_GetGameName(char *buffer, size_t size) -{ - buffer[0] = '\0'; - -#if defined _WIN32 - static char game[128]; - - LPWSTR pCmdLine = GetCommandLineW(); - int argc; - LPWSTR *wargv = CommandLineToArgvW(pCmdLine, &argc); - for (int i = 0; i < argc; ++i) - { - if (wcscmp(wargv[i], L"-game") != 0) - continue; - - if (++i >= argc) - break; - - wcstombs(buffer, wargv[i], size); - buffer[size-1] = '\0'; - break; - } - - LocalFree(wargv); - -#elif defined __APPLE__ - int argc = *_NSGetArgc(); - char **argv = *_NSGetArgv(); - for (int i = 0; i < argc; ++i) - { - if (strcmp(argv[i], "-game") != 0) - continue; - - if (++i >= argc) - break; - - strncpy(buffer, argv[i], size); - buffer[size-1] = '\0'; - break; - } - -#elif defined __linux__ - FILE *pFile = fopen("/proc/self/cmdline", "rb"); - if (pFile) - { - char *arg = NULL; - size_t argsize = 0; - bool bNextIsGame = false; - - while (getdelim(&arg, &argsize, 0, pFile) != -1) - { - if (bNextIsGame) - { - strncpy(buffer, arg, size); - buffer[size-1] = '\0'; - break; - } - - if (strcmp(arg, "-game") == 0) - { - bNextIsGame = true; - } - } - - free(arg); - fclose(pFile); - } -#else -#error unsupported platform -#endif - - if (buffer[0] == 0) - { - strncpy(buffer, ".", size); - } -} - -MetamodBackend -mm_DetermineBackend(QueryValveInterface engineFactory, QueryValveInterface serverFactory, const char *game_name) -{ - if (engineFactory("VEngineServer024", NULL) != NULL) - { - return MMBackend_DOTA; - } - else if (engineFactory("VEngineServer023", NULL) != NULL) - { - if (engineFactory("IEngineSoundServer004", NULL) != NULL) - { - return MMBackend_Insurgency; - } - return MMBackend_CSGO; - } - else if (engineFactory("VEngineServer022", NULL) != NULL && - engineFactory("VEngineCvar007", NULL) != NULL) - { +#define LIBRARY_MINEXT ".so" +#endif + +bool +mm_LoadMetamodLibrary(MetamodBackend backend, char *buffer, size_t maxlength) +{ + size_t len, temp_len; + char mm_path[PLATFORM_MAX_PATH * 2]; + + /* Get our path */ + if (!mm_GetFileOfAddress((void*)mm_GetFileOfAddress, mm_path, sizeof(mm_path))) + return false; + + len = strlen(mm_path); + temp_len = strlen("server" LIBRARY_EXT); + if (len < temp_len) + return false; + + /* Build log file name */ + mm_path[len - temp_len] = '\0'; + mm_Format(mm_fatal_logfile, + sizeof(mm_fatal_logfile), + "%smetamod-fatal.log", + mm_path); + + /* Replace server.dll with the new binary we want */ + mm_Format(&mm_path[len - temp_len], + sizeof(mm_path) - (len - temp_len), + "metamod.%s" LIBRARY_MINEXT, + backend_names[backend]); + + mm_library = (HMODULE)mm_LoadLibrary(mm_path, buffer, maxlength); + + return (mm_library != NULL); +} + +void +mm_UnloadMetamodLibrary() +{ + mm_UnloadLibrary(mm_library); + mm_library = NULL; +} + +#if defined _WIN32 +#define EXPORT extern "C" __declspec(dllexport) +#elif defined __GNUC__ +#if __GNUC__ == 4 +#define EXPORT extern "C" __attribute__ ((visibility("default"))) +#else +#define EXPORT extern "C" +#endif +#endif + +EXPORT void * +CreateInterface(const char *name, int *ret) +{ + /* If we've got a VSP bridge, do nothing. */ + if (vsp_bridge != NULL) + { + if (ret != NULL) + *ret = 1; + return NULL; + } + + void *ptr; + if (strncmp(name, "ISERVERPLUGINCALLBACKS", 22) == 0) + { + /* Either load as VSP or start VSP listener */ + ptr = mm_GetVspCallbacks(atoi(&name[22])); + } + else if (gamedll_bridge == NULL) + { + /* Load as gamedll */ + ptr = mm_GameDllRequest(name, ret); + } + else + { + /* If we've got a gamedll bridge, forward the request. */ + return gamedll_bridge->QueryInterface(name, ret); + } + + if (ret != NULL) + *ret = (ptr != NULL) ? 0 : 1; + + return ptr; +} + +void * +mm_GetProcAddress(const char *name) +{ + return mm_GetLibAddress(mm_library, name); +} + +void +mm_GetGameName(char *buffer, size_t size) +{ + buffer[0] = '\0'; + +#if defined _WIN32 + static char game[128]; + + LPWSTR pCmdLine = GetCommandLineW(); + int argc; + LPWSTR *wargv = CommandLineToArgvW(pCmdLine, &argc); + for (int i = 0; i < argc; ++i) + { + if (wcscmp(wargv[i], L"-game") != 0) + continue; + + if (++i >= argc) + break; + + wcstombs(buffer, wargv[i], size); + buffer[size-1] = '\0'; + break; + } + + LocalFree(wargv); + +#elif defined __APPLE__ + int argc = *_NSGetArgc(); + char **argv = *_NSGetArgv(); + for (int i = 0; i < argc; ++i) + { + if (strcmp(argv[i], "-game") != 0) + continue; + + if (++i >= argc) + break; + + strncpy(buffer, argv[i], size); + buffer[size-1] = '\0'; + break; + } + +#elif defined __linux__ + FILE *pFile = fopen("/proc/self/cmdline", "rb"); + if (pFile) + { + char *arg = NULL; + size_t argsize = 0; + bool bNextIsGame = false; + + while (getdelim(&arg, &argsize, 0, pFile) != -1) + { + if (bNextIsGame) + { + strncpy(buffer, arg, size); + buffer[size-1] = '\0'; + break; + } + + if (strcmp(arg, "-game") == 0) + { + bNextIsGame = true; + } + } + + free(arg); + fclose(pFile); + } +#else +#error unsupported platform +#endif + + if (buffer[0] == 0) + { + strncpy(buffer, ".", size); + } +} + +MetamodBackend +mm_DetermineBackend(QueryValveInterface engineFactory, QueryValveInterface serverFactory, const char *game_name) +{ + if (engineFactory("VEngineServer024", NULL) != NULL) + { + return MMBackend_DOTA; + } + else if (engineFactory("VEngineServer023", NULL) != NULL) + { + if (engineFactory("IEngineSoundServer004", NULL) != NULL) + { + return MMBackend_Insurgency; + } + return MMBackend_CSGO; + } + else if (engineFactory("VEngineServer022", NULL) != NULL && + engineFactory("VEngineCvar007", NULL) != NULL) + { if (engineFactory("EngineTraceServer004", NULL) != NULL) { if (engineFactory("XboxSystemInterface001", NULL) != NULL) { return MMBackend_AlienSwarm; - } - - if (strcmp(game_name, "portal2") == 0) - { - return MMBackend_Portal2; - } - - return MMBackend_Blade; - } - else if (engineFactory("VPrecacheSystem001", NULL) != NULL) - { - if (engineFactory("ServerGameTags002", NULL) != NULL) - { - return MMBackend_NuclearDawn; - } - else if (strcmp(game_name, "contagion") == 0) - { - return MMBackend_Contagion; - } - else - { - return MMBackend_Left4Dead2; - } - } - - return MMBackend_Left4Dead; - } - else if (engineFactory("VEngineServer021", NULL) != NULL) - { - /* Check for OB */ - if (engineFactory("VEngineCvar004", NULL) != NULL) - { - if (engineFactory("VModelInfoServer002", NULL) != NULL) - { - /* BGT has same iface version numbers and libs as ep2 */ - if (strcmp(game_name, "pm") == 0) - { - return MMBackend_BloodyGoodTime; - } - else - { - return MMBackend_Episode2; - } - } - else if (engineFactory("VModelInfoServer003", NULL) != NULL) - { - if (engineFactory("VFileSystem017", NULL) != NULL) - { - return MMBackend_EYE; - } - else if (strcmp(game_name, "cstrike") == 0) - { - return MMBackend_CSS; - } - else if (strcmp(game_name, "tf") == 0) - { - return MMBackend_TF2; - } - else if (strcmp(game_name, "dod") == 0) - { - return MMBackend_DODS; - } - else if (strcmp(game_name, "hl2mp") == 0) - { - return MMBackend_HL2DM; - } - else - { - return MMBackend_SDK2013; - } - } - } - /* Check for Episode One/Old Engine */ - else if (engineFactory("VModelInfoServer001", NULL) != NULL && - (engineFactory("VEngineCvar003", NULL) != NULL || - engineFactory("VEngineCvar002", NULL) != NULL)) - { - /* Check for Dark Messiah which has a weird directory structure */ - if (strcmp(game_name, ".") == 0) - { - return MMBackend_DarkMessiah; - } - return MMBackend_Episode1; - } - } - - return MMBackend_UNKNOWN; -} - + } + + void *lib = (void *)serverFactory; + void *addr; + if (strcmp(game_name, "portal2") == 0 + || (addr = mm_FindPattern(lib, "baseportalcombatweapon", sizeof("baseportalcombatweapon") - 1))) + { + return MMBackend_Portal2; + } + + return MMBackend_Blade; + } + else if (engineFactory("VPrecacheSystem001", NULL) != NULL) + { + if (engineFactory("ServerGameTags002", NULL) != NULL) + { + return MMBackend_NuclearDawn; + } + else + { + void *lib = (void *)serverFactory; + void *addr; + if (strcmp(game_name, "contagion") == 0 + || (addr = mm_FindPattern(lib, "Contagion_Chat_All", sizeof("Contagion_Chat_All") - 1))) + { + return MMBackend_Contagion; + } + else + { + return MMBackend_Left4Dead2; + } + } + } + + return MMBackend_Left4Dead; + } + else if (engineFactory("VEngineServer021", NULL) != NULL) + { + /* Check for OB */ + if (engineFactory("VEngineCvar004", NULL) != NULL) + { + if (engineFactory("VModelInfoServer002", NULL) != NULL) + { + /* BGT has same iface version numbers and libs as ep2 */ + void *lib = (void *)serverFactory; + void *addr; + if (strcmp(game_name, "pm") == 0 + || (addr = mm_FindPattern(lib, "DT_PMPlayerResource", sizeof("DT_PMPlayerResource") - 1))) + { + return MMBackend_BloodyGoodTime; + } + else + { + return MMBackend_Episode2; + } + } + else if (engineFactory("VModelInfoServer003", NULL) != NULL) + { + if (engineFactory("VFileSystem017", NULL) != NULL) + { + return MMBackend_EYE; + } + else + { + void *lib = (void *)serverFactory; + void *addr; + if (strcmp(game_name, "cstrike") == 0 + || (addr = mm_FindPattern(lib, "DT_CSPlayerResource", sizeof("DT_CSPlayerResource") - 1))) + { + return MMBackend_CSS; + } + else if (strcmp(game_name, "tf") == 0 + || (addr = mm_FindPattern(lib, "DT_TFPlayerResource", sizeof("DT_TFPlayerResource") - 1))) + { + return MMBackend_TF2; + } + else if (strcmp(game_name, "dod") == 0 + || (addr = mm_FindPattern(lib, "DT_DODPlayerResource", sizeof("DT_DODPlayerResource") - 1))) + { + return MMBackend_DODS; + } + else if (strcmp(game_name, "hl2mp") == 0 + || (addr = mm_FindPattern(lib, "Half-Life 2 Deathmatch", sizeof("Half-Life 2 Deathmatch") - 1))) + { + return MMBackend_HL2DM; + } + else + { + return MMBackend_SDK2013; + } + } + } + } + /* Check for Episode One/Old Engine */ + else if (engineFactory("VModelInfoServer001", NULL) != NULL && + (engineFactory("VEngineCvar003", NULL) != NULL || + engineFactory("VEngineCvar002", NULL) != NULL)) + { + /* Check for Dark Messiah which has a weird directory structure */ + if (strcmp(game_name, ".") == 0) + { + return MMBackend_DarkMessiah; + } + return MMBackend_Episode1; + } + } + + return MMBackend_UNKNOWN; +} + diff --git a/loader/utility.cpp b/loader/utility.cpp index 56d158d..032a831 100644 --- a/loader/utility.cpp +++ b/loader/utility.cpp @@ -34,6 +34,13 @@ #include "loader.h" #include "utility.h" +#if defined __linux__ +#include + +#define PAGE_SIZE 4096 +#define PAGE_ALIGN_UP(x) ((x + PAGE_SIZE - 1) & ~(PAGE_SIZE - 1)) +#endif + #if defined _WIN32 static void mm_GetPlatformError(char *buffer, size_t maxlength) @@ -345,3 +352,228 @@ mm_GetFileOfAddress(void *pAddr, char *buffer, size_t maxlength) return true; } +struct DynLibInfo +{ + void *baseAddress; + size_t memorySize; +}; + +static bool +mm_GetLibraryInfo(const void *libPtr, DynLibInfo &lib) +{ + uintptr_t baseAddr; + + if (libPtr == NULL) + { + return false; + } + +#ifdef _WIN32 + + MEMORY_BASIC_INFORMATION info; + IMAGE_DOS_HEADER *dos; + IMAGE_NT_HEADERS *pe; + IMAGE_FILE_HEADER *file; + IMAGE_OPTIONAL_HEADER *opt; + + if (!VirtualQuery(libPtr, &info, sizeof(MEMORY_BASIC_INFORMATION))) + { + return false; + } + + baseAddr = reinterpret_cast(info.AllocationBase); + + /* All this is for our insane sanity checks :o */ + dos = reinterpret_cast(baseAddr); + pe = reinterpret_cast(baseAddr + dos->e_lfanew); + file = &pe->FileHeader; + opt = &pe->OptionalHeader; + + /* Check PE magic and signature */ + if (dos->e_magic != IMAGE_DOS_SIGNATURE || pe->Signature != IMAGE_NT_SIGNATURE || opt->Magic != IMAGE_NT_OPTIONAL_HDR32_MAGIC) + { + return false; + } + + /* Check architecture, which is 32-bit/x86 right now + * Should change this for 64-bit if Valve gets their act together + */ + if (file->Machine != IMAGE_FILE_MACHINE_I386) + { + return false; + } + + /* For our purposes, this must be a dynamic library */ + if ((file->Characteristics & IMAGE_FILE_DLL) == 0) + { + return false; + } + + /* Finally, we can do this */ + lib.memorySize = opt->SizeOfImage; + +#elif defined __linux__ + + Dl_info info; + Elf32_Ehdr *file; + Elf32_Phdr *phdr; + uint16_t phdrCount; + + if (!dladdr(libPtr, &info)) + { + return false; + } + + if (!info.dli_fbase || !info.dli_fname) + { + return false; + } + + /* This is for our insane sanity checks :o */ + baseAddr = reinterpret_cast(info.dli_fbase); + file = reinterpret_cast(baseAddr); + + /* Check ELF magic */ + if (memcmp(ELFMAG, file->e_ident, SELFMAG) != 0) + { + return false; + } + + /* Check ELF version */ + if (file->e_ident[EI_VERSION] != EV_CURRENT) + { + return false; + } + + /* Check ELF architecture, which is 32-bit/x86 right now + * Should change this for 64-bit if Valve gets their act together + */ + if (file->e_ident[EI_CLASS] != ELFCLASS32 || file->e_machine != EM_386 || file->e_ident[EI_DATA] != ELFDATA2LSB) + { + return false; + } + + /* For our purposes, this must be a dynamic library/shared object */ + if (file->e_type != ET_DYN) + { + return false; + } + + phdrCount = file->e_phnum; + phdr = reinterpret_cast(baseAddr + file->e_phoff); + + for (uint16_t i = 0; i < phdrCount; i++) + { + Elf32_Phdr &hdr = phdr[i]; + + /* We only really care about the segment with executable code */ + if (hdr.p_type == PT_LOAD && hdr.p_flags == (PF_X|PF_R)) + { + /* From glibc, elf/dl-load.c: + * c->mapend = ((ph->p_vaddr + ph->p_filesz + GLRO(dl_pagesize) - 1) + * & ~(GLRO(dl_pagesize) - 1)); + * + * In glibc, the segment file size is aligned up to the nearest page size and + * added to the virtual address of the segment. We just want the size here. + */ + lib.memorySize = PAGE_ALIGN_UP(hdr.p_filesz); + break; + } + } + +#elif defined __APPLE__ + + Dl_info info; + struct mach_header *file; + struct segment_command *seg; + uint32_t cmd_count; + + if (!dladdr(libPtr, &info)) + { + return false; + } + + if (!info.dli_fbase || !info.dli_fname) + { + return false; + } + + /* This is for our insane sanity checks :o */ + baseAddr = (uintptr_t)info.dli_fbase; + file = (struct mach_header *)baseAddr; + + /* Check Mach-O magic */ + if (file->magic != MH_MAGIC) + { + return false; + } + + /* Check architecture (32-bit/x86) */ + if (file->cputype != CPU_TYPE_I386 || file->cpusubtype != CPU_SUBTYPE_I386_ALL) + { + return false; + } + + /* For our purposes, this must be a dynamic library */ + if (file->filetype != MH_DYLIB) + { + return false; + } + + cmd_count = file->ncmds; + seg = (struct segment_command *)(baseAddr + sizeof(struct mach_header)); + + /* Add up memory sizes of mapped segments */ + for (uint32_t i = 0; i < cmd_count; i++) + { + if (seg->cmd == LC_SEGMENT) + { + lib.memorySize += seg->vmsize; + } + + seg = (struct segment_command *)((uintptr_t)seg + seg->cmdsize); + } + +#endif + + lib.baseAddress = reinterpret_cast(baseAddr); + + return true; +} + +void *mm_FindPattern(const void *libPtr, const char *pattern, size_t len) +{ + DynLibInfo lib; + bool found; + char *ptr, *end; + + memset(&lib, 0, sizeof(DynLibInfo)); + + if (!mm_GetLibraryInfo(libPtr, lib)) + { + return NULL; + } + + ptr = reinterpret_cast(lib.baseAddress); + end = ptr + lib.memorySize - len; + + while (ptr < end) + { + found = true; + for (register size_t i = 0; i < len; i++) + { + if (pattern[i] != '\x2A' && pattern[i] != ptr[i]) + { + found = false; + break; + } + } + + if (found) + return ptr; + + ptr++; + } + + return NULL; +} diff --git a/loader/utility.h b/loader/utility.h index ca9f2e0..c586d1b 100644 --- a/loader/utility.h +++ b/loader/utility.h @@ -66,5 +66,8 @@ mm_PathCmp(const char *path1, const char *path2); extern bool mm_GetFileOfAddress(void *pAddr, char *buffer, size_t maxlength); +extern void * +mm_FindPattern(const void *libPtr, const char *pattern, size_t len); + #endif /* _INCLUDE_METAMOD_SOURCE_LOADER_UTILITY_H_ */