HLMetaModOfficial/loader/gamedll.cpp

/**
 * vim: set ts=4 sw=4 tw=99 noet :
 * ======================================================
 * Metamod:Source
 * Copyright (C) 2004-2015 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 <stdio.h>
#include <ctype.h>
#include <string.h>
#include <assert.h>
#include <stdlib.h>
#include <stddef.h>
#include "loader.h"
#include <sh_memfuncinfo.h>
#include <sh_memory.h>
#include "utility.h"
#include "gamedll.h"

class IServerGameDLL;
class ISource2ServerConfig;

#define MAX_GAMEDLL_PATHS	10

IGameDllBridge* gamedll_bridge = NULL;
static int game_info_detected = 0;
static char game_name[128];
static char gamedll_paths[MAX_GAMEDLL_PATHS][PLATFORM_MAX_PATH];
static void *gamedll_libs[MAX_GAMEDLL_PATHS];
static unsigned int gamedll_path_count = 0;
static void *gamedll_lib = NULL;
static IServerGameDLL *gamedll_iface = NULL;
static ISource2ServerConfig *config_iface = NULL;
static QueryValveInterface gamedll_qvi = NULL;
static int gamedll_version = 0;
static int isgd_shutdown_index = -1;
static int is2sc_allowdedi_index = 21;
static char mm_path[PLATFORM_MAX_PATH];
static bool g_is_source2 = false;

#if defined _WIN32
#define SERVER_NAME			"server.dll"
#if defined _WIN64
#define PLATFORM_NAME		"win64"
#else
#define PLATFORM_NAME		"win32"
#endif
#elif defined __APPLE__
#define SERVER_NAME			"server.dylib"
#if defined __amd64__
#define PLATFORM_NAME		"osx64"
#else
#define PLATFORM_NAME		"osx32"
#endif
#elif defined __linux__
#define SERVER_NAME			"server" LIB_SUFFIX
#if defined __amd64__
#define PLATFORM_NAME		"linuxsteamrt64"
#else
#define PLATFORM_NAME		"linuxsteamrt32"
#endif
#endif

static bool
mm_DetectGameInformation()
{
	char game_path[PLATFORM_MAX_PATH];

	if (game_info_detected)
		return game_info_detected == 1 ? true : false;

	game_info_detected = -1;

	mm_GetGameName(game_name, sizeof(game_name));

	if (!mm_GetFileOfAddress((void*)mm_DetectGameInformation, mm_path, sizeof(mm_path)))
	{
		mm_LogFatal("Could not locate Metamod loader library path");
		return false;
	}

	if (!mm_ResolvePath(game_name, game_path, sizeof(game_path), g_is_source2))
	{
		mm_LogFatal("Could not resolve path: %s", game_name);
		return false;
	}

	FILE *fp;
	char gameinfo_path[PLATFORM_MAX_PATH];

	bool is_source2 = false;
	mm_PathFormat(gameinfo_path, sizeof(gameinfo_path), "%s/gameinfo.txt", game_path);
	if ((fp = fopen(gameinfo_path, "rt")) == NULL)
	{
		// Try Source2 gameinfo
		mm_PathFormat(gameinfo_path, sizeof(gameinfo_path), "%s/gameinfo.gi", game_path);
		if ((fp = fopen(gameinfo_path, "rt")) == NULL)
		{
			mm_LogFatal("Could not read file: %s", gameinfo_path);
			return false;
		}
		else
		{
			is_source2 = true;
		}
	}

	char temp_path[PLATFORM_MAX_PATH];
	char cur_path[PLATFORM_MAX_PATH];
	getcwd(cur_path, sizeof(cur_path));

	char *ptr;
	const char *lptr;
	bool search = false;
	char buffer[255], key[128], val[128];
	while (!feof(fp) && fgets(buffer, sizeof(buffer), fp) != NULL)
	{
		mm_TrimComments(buffer);
		mm_TrimLeft(buffer);
		mm_TrimRight(buffer);

		if (stricmp(buffer, "SearchPaths") == 0)
			search = true;

		if (!search)
			continue;

		mm_KeySplit(buffer, key, sizeof(key) - 1, val, sizeof(val) - 1);
		if (stricmp(key, "Game") != 0 && stricmp(key, "GameBin") != 0)
			continue;

		if (strncmp(val, "|gameinfo_path|", sizeof("|gameinfo_path|") - 1) == 0)
		{
			ptr = &val[sizeof("|gameinfo_path|") - 1];
			if (ptr[0] == '.')
				ptr++;
			lptr = game_path;
		}
		else
		{
			ptr = val;
			lptr = cur_path;
		}

		char *pRelPath = is_source2 ? "../../" : "";
		char *pOSDir = is_source2 ? PLATFORM_NAME "/" : "";
		if (stricmp(key, "GameBin") == 0)
			mm_PathFormat(temp_path, sizeof(temp_path), "%s/%s%s/%s" SERVER_NAME, lptr, pRelPath, ptr, pOSDir);
		else if (!ptr[0])
			mm_PathFormat(temp_path, sizeof(temp_path), "%s/%sbin/%s" SERVER_NAME, lptr, pRelPath, pOSDir);
		else
			mm_PathFormat(temp_path, sizeof(temp_path), "%s/%s%s/bin/%s" SERVER_NAME, lptr, pRelPath, ptr, pOSDir);

		if (mm_PathCmp(mm_path, temp_path))
			continue;

		FILE *exists = fopen(temp_path, "rb");
		if (!exists)
			continue;
		fclose(exists);

		/* exists is still non-NULL... use this as a flag */
		for (unsigned int i = 0; i < gamedll_path_count; i++)
		{
			if (mm_PathCmp(gamedll_paths[i], temp_path))
			{
				exists = NULL;
				break;
			}
		}

		if (!exists)
			continue;

		mm_Format(gamedll_paths[gamedll_path_count],
				  PLATFORM_MAX_PATH,
				  "%s",
				  temp_path);
		gamedll_path_count++;

		if (gamedll_path_count == MAX_GAMEDLL_PATHS)
			break;
	}
	fclose(fp);

	game_info_detected = 1;

	if (gamedll_path_count == 0)
	{
		mm_LogFatal("Could not detect any valid game paths in gameinfo file");
		return false;
	}

	return true;
}

static void
mm_FreeCachedLibraries()
{
	for (unsigned int i = 0; i < gamedll_path_count; i++)
	{
		if (gamedll_libs[i] == NULL)
			continue;
		mm_UnloadLibrary(gamedll_libs[i]);
	}
}

static void
mm_PatchDllInit(bool patch);

static void
mm_PatchDllShutdown();

static void
mm_PatchAllowDedicated(bool patch);

static void
mm_PatchConnect(bool patch);

static void *isgd_orig_init = NULL;
static void *isgd_orig_shutdown = NULL;
static void *is2sc_orig_allowdedi = NULL;
static void *is2sc_orig_connect = NULL;

class VEmptyClass
{
};

gamedll_bridge_info g_bridge_info;

// Source2 - Rough start order
// CreateInterfaceFn (IS2SC) - hook Connect and AllowDedicatedServer
// IS2SC::Connect - save factory pointer. return orig. remove hook.
// IS2SC::AllowDedicatedServer - return true. remove hook.
// CreateInterfaceFn (IS2S) - hook Init and Shutdown
// IS2S::Init - do same as old ISGD::DLLInit, including core load. return orig. remove hook.
// IS2S::Shutdown - <-- this

enum InitReturnVal_t
{
	INIT_FAILED = 0,
	INIT_OK,

	INIT_LAST_VAL,
};

class ISource2ServerConfig
{
public:
	virtual bool	Connect(QueryValveInterface factory)
	{
		g_bridge_info.engineFactory = factory;
		g_bridge_info.fsFactory = factory;
		g_bridge_info.physicsFactory = factory;


		/* Call the original */
		bool result;
		{
			union
			{
				bool(VEmptyClass::*mfpnew)(QueryValveInterface factory);
#if defined _WIN32
				void *addr;
			} u;
			u.addr = is2sc_orig_connect;
#else
				struct
				{
					void *addr;
					intptr_t adjustor;
				} s;
		} u;
			u.s.addr = is2sc_orig_connect;
			u.s.adjustor = 0;
#endif
			result = (((VEmptyClass *) config_iface)->*u.mfpnew)(factory);
		}

		mm_PatchConnect(false);

		return result;
	}
	virtual bool	AllowDedicatedServers(int universe) const
	{
		mm_PatchAllowDedicated(false);
		return true;
	}
};

class ISource2Server
{
public:
	virtual bool Connect(QueryValveInterface factory) { return true; }
	virtual void Disconnect() {}
	virtual void *QueryInterface(const char *pInterfaceName) { return nullptr; }

	virtual InitReturnVal_t Init()
	{
		mm_backend = MMBackend_DOTA;

		char error[255];
		if (!mm_LoadMetamodLibrary(mm_backend, error, sizeof(error)))
		{
			mm_LogFatal("Detected engine %d but could not load: %s", mm_backend, error);
		}
		else
		{
			typedef IGameDllBridge *(*GetGameDllBridge)();
			GetGameDllBridge get_bridge = (GetGameDllBridge)mm_GetProcAddress("GetGameDllBridge");
			if (get_bridge == NULL)
			{
				mm_UnloadMetamodLibrary();
				mm_LogFatal("Detected engine %d but could not find GetGameDllBridge callback", mm_backend);
			}
			else
			{
				gamedll_bridge = get_bridge();
			}
		}

		if (gamedll_bridge)
		{
			g_bridge_info.pGlobals = nullptr;// pGlobals;
			g_bridge_info.dllVersion = gamedll_version;
			g_bridge_info.isgd = gamedll_iface;
			g_bridge_info.gsFactory = gamedll_qvi;
			g_bridge_info.vsp_listener_path = mm_path;

			strcpy(error, "Unknown error");
			if (!gamedll_bridge->DLLInit_Pre(&g_bridge_info, error, sizeof(error)))
			{
				gamedll_bridge = NULL;
				mm_UnloadMetamodLibrary();
				mm_LogFatal("Unknown error loading Metamod for engine %d: %s", mm_backend, error);	
			}
		}

		/* Call the original */
		InitReturnVal_t result;
		{
			union
			{
				InitReturnVal_t(VEmptyClass::*mfpnew)();
#if defined _WIN32
				void *addr;
			} u;
			u.addr = isgd_orig_init;
#else
				struct
				{
					void *addr;
					intptr_t adjustor;
				} s;
		} u;
			u.s.addr = isgd_orig_init;
			u.s.adjustor = 0;
#endif
			result = (((VEmptyClass *)gamedll_iface)->*u.mfpnew)();
		}

		/**
		 * :TODO: possible logic hole here, what happens if the gamedll REALLY returns false? 
		 * I'm pretty sure we'll die horribly.
		 */

		if (!result)
		{
			gamedll_bridge->Unload();
			mm_UnloadMetamodLibrary();
			gamedll_bridge = NULL;
		}
		else if (gamedll_bridge != NULL)
		{
			gamedll_bridge->DLLInit_Post(&isgd_shutdown_index);
			assert(isgd_shutdown_index != -1);
			mm_PatchDllShutdown();
		}

		mm_PatchDllInit(false);

		return result;
	}

	virtual void Shutdown()
	{
		gamedll_bridge->Unload();
		gamedll_bridge = NULL;
		mm_UnloadMetamodLibrary();

		/* Call original function */
		{
			union
			{
				void (VEmptyClass::*mfpnew)();
#if defined _WIN32
				void *addr;
			} u;
			u.addr = isgd_orig_shutdown;
#else
				struct
				{
					void *addr;
					intptr_t adjustor;
				} s;
			} u;
			u.s.addr = isgd_orig_shutdown;
			u.s.adjustor = 0;
#endif
			(((VEmptyClass *)gamedll_iface)->*u.mfpnew)();
		}

		mm_UnloadLibrary(gamedll_lib);
		gamedll_lib = NULL;
	}
};

class IServerGameDLL
{
public:
	virtual bool DLLInit(QueryValveInterface engineFactory, 
						 QueryValveInterface physicsFactory, 
						 QueryValveInterface fileSystemFactory, 
						 void *pGlobals)
	{
		mm_backend = mm_DetermineBackend(engineFactory, gamedll_qvi, game_name);

		char error[255];
		if (mm_backend == MMBackend_UNKNOWN)
		{
			mm_LogFatal("Could not detect engine version");
		}
		else
		{
			if (!mm_LoadMetamodLibrary(mm_backend, error, sizeof(error)))
			{
				mm_LogFatal("Detected engine %d but could not load: %s", mm_backend, error);
			}
			else
			{
				typedef IGameDllBridge *(*GetGameDllBridge)();
				GetGameDllBridge get_bridge = (GetGameDllBridge)mm_GetProcAddress("GetGameDllBridge");
				if (get_bridge == NULL)
				{
					mm_UnloadMetamodLibrary();
					mm_LogFatal("Detected engine %d but could not find GetGameDllBridge callback", mm_backend);
				}
				else
				{
					gamedll_bridge = get_bridge();
				}
			}
		}

		if (gamedll_bridge)
		{
			g_bridge_info.engineFactory = (QueryValveInterface)engineFactory;
			g_bridge_info.physicsFactory = (QueryValveInterface)physicsFactory;
			g_bridge_info.fsFactory = (QueryValveInterface)fileSystemFactory;
			g_bridge_info.pGlobals = pGlobals;
			g_bridge_info.dllVersion = gamedll_version;
			g_bridge_info.isgd = gamedll_iface;
			g_bridge_info.gsFactory = gamedll_qvi;
			g_bridge_info.vsp_listener_path = mm_path;

			strcpy(error, "Unknown error");
			if (!gamedll_bridge->DLLInit_Pre(&g_bridge_info, error, sizeof(error)))
			{
				gamedll_bridge = NULL;
				mm_UnloadMetamodLibrary();
				mm_LogFatal("Unknown error loading Metamod for engine %d: %s", mm_backend, error);	
			}
		}

		/* Call the original */
		bool result;
		{
			union
			{
				bool (VEmptyClass::*mfpnew)(QueryValveInterface engineFactory, 
										 	QueryValveInterface physicsFactory, 
										 	QueryValveInterface fileSystemFactory, 
										 	void *pGlobals);
#if defined _WIN32
				void *addr;
			} u;
			u.addr = isgd_orig_init;
#else
				struct
				{
					void *addr;
					intptr_t adjustor;
				} s;
			} u;
			u.s.addr = isgd_orig_init;
			u.s.adjustor = 0;
#endif
			result = (((VEmptyClass *)gamedll_iface)->*u.mfpnew)(engineFactory,
																 physicsFactory,
																 fileSystemFactory,
																 pGlobals);
		}

		/**
		 * :TODO: possible logic hole here, what happens if the gamedll REALLY returns false? 
		 * I'm pretty sure we'll die horribly.
		 */

		if (!result)
		{
			gamedll_bridge->Unload();
			mm_UnloadMetamodLibrary();
			gamedll_bridge = NULL;
		}
		else if (gamedll_bridge != NULL)
		{
			gamedll_bridge->DLLInit_Post(&isgd_shutdown_index);
			assert(isgd_shutdown_index != -1);
			mm_PatchDllShutdown();
		}

		mm_PatchDllInit(false);

		return result;
	}

	virtual void DLLShutdown()
	{
		gamedll_bridge->Unload();
		gamedll_bridge = NULL;
		mm_UnloadMetamodLibrary();

		/* Call original function */
		{
			union
			{
				void (VEmptyClass::*mfpnew)();
#if defined _WIN32
				void *addr;
			} u;
			u.addr = isgd_orig_shutdown;
#else
				struct
				{
					void *addr;
					intptr_t adjustor;
				} s;
			} u;
			u.s.addr = isgd_orig_shutdown;
			u.s.adjustor = 0;
#endif
			(((VEmptyClass *)gamedll_iface)->*u.mfpnew)();
		}

		mm_UnloadLibrary(gamedll_lib);
		gamedll_lib = NULL;
	}
};

static IServerGameDLL isgd_thunk;
static ISource2Server is2s_thunk;
static ISource2ServerConfig is2sc_thunk;

static void
mm_PatchDllInit(bool patch)
{
	void **vtable_src;
	void **vtable_dest;
	SourceHook::MemFuncInfo mfp;

	if (g_is_source2)
	{
		SourceHook::GetFuncInfo(&ISource2Server::Init, mfp);
	}
	else
	{
		SourceHook::GetFuncInfo(&IServerGameDLL::DLLInit, mfp);
	}

	assert(mfp.isVirtual);
	assert(mfp.thisptroffs == 0);
	assert(mfp.vtbloffs == 0);

	if (g_is_source2)
	{
		vtable_src = (void **)*(void **)&is2s_thunk;
	}
	else
	{
		vtable_src = (void **)*(void **)&isgd_thunk;
	}
	vtable_dest = (void **)*(void **)gamedll_iface;

	SourceHook::SetMemAccess(&vtable_dest[mfp.vtblindex],
							 sizeof(void*),
							 SH_MEM_READ|SH_MEM_WRITE|SH_MEM_EXEC);

	if (patch)
	{
		assert(isgd_orig_init == NULL);
		isgd_orig_init = vtable_dest[mfp.vtblindex];
		vtable_dest[mfp.vtblindex] = vtable_src[mfp.vtblindex];
	}
	else
	{
		assert(isgd_orig_init != NULL);
		vtable_dest[mfp.vtblindex] = isgd_orig_init;
		isgd_orig_init = NULL;
	}
}

static void
mm_PatchDllShutdown()
{
	void **vtable_src;
	void **vtable_dest;
	SourceHook::MemFuncInfo mfp;

	mfp.isVirtual = false;
	if (g_is_source2)
	{
		SourceHook::GetFuncInfo(&ISource2Server::Shutdown, mfp);
	}
	else
	{
		SourceHook::GetFuncInfo(&IServerGameDLL::DLLShutdown, mfp);
	}
	assert(mfp.isVirtual);
	assert(mfp.thisptroffs == 0);
	assert(mfp.vtbloffs == 0);

	if (g_is_source2)
	{
		vtable_src = (void **)*(void **)&is2s_thunk;
	}
	else
	{
		vtable_src = (void **)*(void **)&isgd_thunk;
	}
	vtable_dest = (void **)*(void **)gamedll_iface;

	isgd_orig_shutdown = vtable_dest[isgd_shutdown_index];
	vtable_dest[isgd_shutdown_index] = vtable_src[mfp.vtblindex];
}

static void
mm_PatchAllowDedicated(bool patch)
{
	void **vtable_src;
	void **vtable_dest;
	SourceHook::MemFuncInfo mfp;

	SourceHook::GetFuncInfo(&ISource2ServerConfig::AllowDedicatedServers, mfp);

	assert(mfp.isVirtual);
	assert(mfp.thisptroffs == 0);
	assert(mfp.vtbloffs == 0);

	vtable_src = (void **) *(void **) &is2sc_thunk;
	vtable_dest = (void **) *(void **) config_iface;

	SourceHook::SetMemAccess(&vtable_dest[is2sc_allowdedi_index],
		sizeof(void*),
		SH_MEM_READ | SH_MEM_WRITE | SH_MEM_EXEC);

	if (patch)
	{
		assert(is2sc_orig_allowdedi == NULL);
		is2sc_orig_allowdedi = vtable_dest[is2sc_allowdedi_index];
		vtable_dest[is2sc_allowdedi_index] = vtable_src[mfp.vtblindex];
	}
	else
	{
		assert(is2sc_orig_allowdedi != NULL);
		vtable_dest[is2sc_allowdedi_index] = is2sc_orig_allowdedi;
		is2sc_orig_allowdedi = NULL;
	}
}

static void
mm_PatchConnect(bool patch)
{
	void **vtable_src;
	void **vtable_dest;
	SourceHook::MemFuncInfo mfp;

	SourceHook::GetFuncInfo(&ISource2ServerConfig::Connect, mfp);

	assert(mfp.isVirtual);
	assert(mfp.thisptroffs == 0);
	assert(mfp.vtbloffs == 0);

	vtable_src = (void **) *(void **) &is2sc_thunk;
	vtable_dest = (void **) *(void **) config_iface;

	SourceHook::SetMemAccess(&vtable_dest[mfp.vtblindex],
		sizeof(void*),
		SH_MEM_READ | SH_MEM_WRITE | SH_MEM_EXEC);

	if (patch)
	{
		assert(is2sc_orig_connect == NULL);
		is2sc_orig_connect = vtable_dest[mfp.vtblindex];
		vtable_dest[mfp.vtblindex] = vtable_src[mfp.vtblindex];
	}
	else
	{
		assert(is2sc_orig_connect != NULL);
		vtable_dest[mfp.vtblindex] = is2sc_orig_connect;
		is2sc_orig_connect = NULL;
	}
}


void *
mm_GameDllRequest(const char *name, int *ret)
{
	if (strncmp(name, "Source2ServerConfig", 19) == 0)
	{
		g_is_source2 = true;
		if (!mm_DetectGameInformation())
		{
			if (ret != NULL)
				*ret = 1;
			return NULL;
		}

		void *lib;
		char error[255];
		void *ptr = NULL;
		QueryValveInterface qvi;
		for (unsigned int i = 0; i < gamedll_path_count; i++)
		{
			if (gamedll_libs[i] == NULL)
			{
				lib = mm_LoadLibrary(gamedll_paths[i], error, sizeof(error));
				if (lib == NULL)
					continue;
				gamedll_libs[i] = lib;
			}
			lib = gamedll_libs[i];
			qvi = (QueryValveInterface)mm_GetLibAddress(lib, "CreateInterface");
			if (qvi == NULL)
				continue;
			ptr = qvi(name, ret);
			if (ptr != NULL)
			{
				gamedll_libs[i] = NULL;
				break;
			}
		}

		if (ptr != NULL)
		{
			mm_FreeCachedLibraries();
			gamedll_lib = lib;
			config_iface = (ISource2ServerConfig *) ptr;
			gamedll_qvi = qvi;

			mm_PatchConnect(true);
			mm_PatchAllowDedicated(true);

			if (ret != NULL)
				*ret = 0;
			return ptr;
		}
	}
	else if (strncmp(name, "Source2Server0", 14) == 0)
	{
		gamedll_iface = (IServerGameDLL *)gamedll_qvi(name, ret);
		gamedll_version = atoi(&name[13]);
		mm_PatchDllInit(true);

		if (ret != NULL)
			*ret = 0;
		return gamedll_iface;
	}
	else if (strncmp(name, "ServerGameDLL", 13) == 0)
	{
		if (!mm_DetectGameInformation())
		{
			if (ret != NULL)
				*ret = 1;
			return NULL;
		}

		void *lib;
		char error[255];
		void *ptr = NULL;
		QueryValveInterface qvi;
		for (unsigned int i = 0; i < gamedll_path_count; i++)
		{
			if (gamedll_libs[i] == NULL)
			{
				lib = mm_LoadLibrary(gamedll_paths[i], error, sizeof(error));
				if (lib == NULL)
					continue;
				gamedll_libs[i] = lib;
			}
			lib = gamedll_libs[i];
			qvi = (QueryValveInterface)mm_GetLibAddress(lib, "CreateInterface");
			if (qvi == NULL)
				continue;
			ptr = qvi(name, ret);
			if (ptr != NULL)
			{
				gamedll_libs[i] = NULL;
				break;
			}
		}

		if (ptr != NULL)
		{
			mm_FreeCachedLibraries();	
			gamedll_lib = lib;
			gamedll_iface = (IServerGameDLL *)ptr;
			gamedll_qvi = qvi;
			gamedll_version = atoi(&name[13]);
			mm_PatchDllInit(true);

			if (ret != NULL)
				*ret = 0;
			return ptr;
		}
	}
	else if (gamedll_lib != NULL && gamedll_bridge == NULL)
	{
		return gamedll_qvi(name, ret);
	}
	else if (game_info_detected == 0)
	{
		mm_LogFatal("Received interface request too early: %s", name);
	}

	if (ret != NULL)
		*ret = 1;
	return NULL;
}