mirror of
https://github.com/alliedmodders/metamod-source.git
synced 2025-02-20 13:54:14 +01:00
Improve game detection further for some engines when -game missing
This commit is contained in:
Nicholas Hastings
parent
4cff6368d2
commit
7a4845e4ac
@ -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 <time.h>
|
||||
#include <assert.h>
|
||||
#include <stdarg.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include "loader.h"
|
||||
#include "serverplugin.h"
|
||||
#include "gamedll.h"
|
||||
#include "utility.h"
|
||||
#if defined __APPLE__
|
||||
#include <crt_externs.h>
|
||||
#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 <time.h>
|
||||
#include <assert.h>
|
||||
#include <stdarg.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include "loader.h"
|
||||
#include "serverplugin.h"
|
||||
#include "gamedll.h"
|
||||
#include "utility.h"
|
||||
#if defined __APPLE__
|
||||
#include <crt_externs.h>
|
||||
#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;
|
||||
}
|
||||
|
||||
|
||||
@ -34,6 +34,13 @@
|
||||
#include "loader.h"
|
||||
#include "utility.h"
|
||||
|
||||
#if defined __linux__
|
||||
#include <link.h>
|
||||
|
||||
#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<uintptr_t>(info.AllocationBase);
|
||||
|
||||
/* All this is for our insane sanity checks :o */
|
||||
dos = reinterpret_cast<IMAGE_DOS_HEADER *>(baseAddr);
|
||||
pe = reinterpret_cast<IMAGE_NT_HEADERS *>(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<uintptr_t>(info.dli_fbase);
|
||||
file = reinterpret_cast<Elf32_Ehdr *>(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<Elf32_Phdr *>(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<void *>(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<char *>(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;
|
||||
}
|
||||
|
||||
@ -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_ */
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user