/* 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;

}