misc/xfire/xfiregameclient.cpp
changeset 2830 571a9deffeac
child 2920 5cf48744a700
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/misc/xfire/xfiregameclient.cpp	Sat Feb 20 09:33:20 2010 +0000
@@ -0,0 +1,121 @@
+/* This file is NOT open source. See "license.txt" to read the full license provided with the Xfire SDK.*/
+
+#define WIN32_LEAN_AND_MEAN
+#include <windows.h>
+#include <tlhelp32.h>
+
+#include "xfiregameclient.h"
+
+static HMODULE g_toucan_dll = NULL;
+static void HelperInit();
+static HMODULE HelperGetToucanDLL();
+
+typedef int (*XfireSetCustomGameDataAFunction)(int , const char **, const char **);
+typedef int (*XfireSetCustomGameDataWFunction)(int , const wchar_t **, const wchar_t **);
+typedef int (*XfireSetCustomGameDataUTF8Function)(int , const char **, const char **);
+
+static XfireSetCustomGameDataAFunction ptr_XfireSetCustomGameDataA = NULL;
+static XfireSetCustomGameDataWFunction ptr_XfireSetCustomGameDataW = NULL;
+static XfireSetCustomGameDataUTF8Function ptr_XfireSetCustomGameDataUTF8 = NULL;
+
+/* make sure we are going to call the ANSI version */
+#ifdef MODULEENTRY32
+#undef MODULEENTRY32
+#endif
+
+#ifdef Module32First
+#undef Module32First
+#endif
+
+#ifdef Module32Next
+#undef Module32Next
+#endif 
+
+
+int XfireIsLoaded()
+{
+	HelperInit();
+	if (ptr_XfireSetCustomGameDataA &&
+		ptr_XfireSetCustomGameDataW &&
+		ptr_XfireSetCustomGameDataUTF8)
+		return 1;
+	return 0;
+}
+
+int XfireSetCustomGameDataA(int num_keys, const char **keys, const char **values)
+{
+	HelperInit();
+	if (ptr_XfireSetCustomGameDataA)
+		return ptr_XfireSetCustomGameDataA(num_keys, keys, values);
+	return 1;
+}
+
+int XfireSetCustomGameDataW(int num_keys, const wchar_t **keys, const wchar_t **values)
+{
+	HelperInit();
+	if (ptr_XfireSetCustomGameDataW)
+		return ptr_XfireSetCustomGameDataW(num_keys, keys, values);
+	return 1;
+}
+
+int XfireSetCustomGameDataUTF8(int num_keys, const char **keys, const char **values)
+{
+	HelperInit();
+	if (ptr_XfireSetCustomGameDataUTF8)
+		return ptr_XfireSetCustomGameDataUTF8(num_keys, keys, values);
+	return 1;
+}
+
+/* ------------------------------------------------------------------------- */
+static void HelperInit()
+{
+	if (!ptr_XfireSetCustomGameDataA ||
+		!ptr_XfireSetCustomGameDataW ||
+		!ptr_XfireSetCustomGameDataUTF8)
+	{
+		HMODULE toucan_dll = HelperGetToucanDLL();
+		if (toucan_dll)
+		{
+			ptr_XfireSetCustomGameDataA = (XfireSetCustomGameDataAFunction)::GetProcAddress(toucan_dll, "ToucanSendGameClientDataA_V1");
+			ptr_XfireSetCustomGameDataW = (XfireSetCustomGameDataWFunction)::GetProcAddress(toucan_dll, "ToucanSendGameClientDataW_V1");
+			ptr_XfireSetCustomGameDataUTF8 = (XfireSetCustomGameDataUTF8Function)::GetProcAddress(toucan_dll, "ToucanSendGameClientDataUTF8_V1");
+		}
+	}
+}
+
+
+static HMODULE HelperGetToucanDLL()
+{
+	if (g_toucan_dll)
+		return g_toucan_dll;
+
+	/*
+	** We need to enumerate the DLLs loaded to find toucan dll.
+	** This is done because the toucan dll changes with each update.
+	** The toucan dll has the following format. "xfire_toucan_{BUILD_NUMBER}.dll"
+	** We simply try to find a dll w/ the prefix "xfire_toucan"
+	*/
+	HANDLE snapshot_handle = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, GetCurrentProcessId());
+	if (snapshot_handle != INVALID_HANDLE_VALUE)
+	{
+		MODULEENTRY32 module_entry;
+		module_entry.dwSize = sizeof(MODULEENTRY32); 
+
+		BOOL result = Module32First(snapshot_handle, &module_entry);
+		char module_name[] = "xfire_toucan";
+		DWORD module_name_len = sizeof(module_name)-1;
+		while (result)
+		{
+			if (CompareStringA(LOCALE_USER_DEFAULT, NORM_IGNORECASE, module_entry.szModule, module_name_len, module_name, module_name_len) == CSTR_EQUAL)
+			{
+				g_toucan_dll = module_entry.hModule;
+				break;
+			}
+			result = Module32Next(snapshot_handle, &module_entry);
+		}
+
+		CloseHandle(snapshot_handle);
+	}
+
+	return g_toucan_dll;
+}
\ No newline at end of file