1 /* This file is NOT open source. See "license.txt" to read the full license provided with the Xfire SDK. */

     2 

     3 #define WIN32_LEAN_AND_MEAN

     4 #include <windows.h>

     5 #include <tlhelp32.h>

     6 

     7 #include "xfiregameclient.h"

     8 

     9 static HMODULE g_toucan_dll = NULL;

    10 static void HelperInit();

    11 static HMODULE HelperGetToucanDLL();

    12 

    13 typedef int (*XfireSetCustomGameDataAFunction)(int , const char **, const char **);

    14 typedef int (*XfireSetCustomGameDataWFunction)(int , const wchar_t **, const wchar_t **);

    15 typedef int (*XfireSetCustomGameDataUTF8Function)(int , const char **, const char **);

    16 

    17 static XfireSetCustomGameDataAFunction ptr_XfireSetCustomGameDataA = NULL;

    18 static XfireSetCustomGameDataWFunction ptr_XfireSetCustomGameDataW = NULL;

    19 static XfireSetCustomGameDataUTF8Function ptr_XfireSetCustomGameDataUTF8 = NULL;

    20 

    21 /* make sure we are going to call the ANSI version */

    22 #ifdef MODULEENTRY32

    23 #undef MODULEENTRY32

    24 #endif

    25 

    26 #ifdef Module32First

    27 #undef Module32First

    28 #endif

    29 

    30 #ifdef Module32Next

    31 #undef Module32Next

    32 #endif

    33 

    34 

    35 int XfireIsLoaded()

    36 {

    37     HelperInit();

    38     if (ptr_XfireSetCustomGameDataA &&

    39         ptr_XfireSetCustomGameDataW &&

    40         ptr_XfireSetCustomGameDataUTF8)

    41         return 1;

    42     return 0;

    43 }

    44 

    45 int XfireSetCustomGameDataA(int num_keys, const char **keys, const char **values)

    46 {

    47     HelperInit();

    48     if (ptr_XfireSetCustomGameDataA)

    49         return ptr_XfireSetCustomGameDataA(num_keys, keys, values);

    50     return 1;

    51 }

    52 

    53 int XfireSetCustomGameDataW(int num_keys, const wchar_t **keys, const wchar_t **values)

    54 {

    55     HelperInit();

    56     if (ptr_XfireSetCustomGameDataW)

    57         return ptr_XfireSetCustomGameDataW(num_keys, keys, values);

    58     return 1;

    59 }

    60 

    61 int XfireSetCustomGameDataUTF8(int num_keys, const char **keys, const char **values)

    62 {

    63     HelperInit();

    64     if (ptr_XfireSetCustomGameDataUTF8)

    65         return ptr_XfireSetCustomGameDataUTF8(num_keys, keys, values);

    66     return 1;

    67 }

    68 

    69 /* ------------------------------------------------------------------------- */

    70 static void HelperInit()

    71 {

    72     if (!ptr_XfireSetCustomGameDataA ||

    73         !ptr_XfireSetCustomGameDataW ||

    74         !ptr_XfireSetCustomGameDataUTF8)

    75     {

    76         HMODULE toucan_dll = HelperGetToucanDLL();

    77         if (toucan_dll)

    78         {

    79             ptr_XfireSetCustomGameDataA = (XfireSetCustomGameDataAFunction)::GetProcAddress(toucan_dll, "ToucanSendGameClientDataA_V1");

    80             ptr_XfireSetCustomGameDataW = (XfireSetCustomGameDataWFunction)::GetProcAddress(toucan_dll, "ToucanSendGameClientDataW_V1");

    81             ptr_XfireSetCustomGameDataUTF8 = (XfireSetCustomGameDataUTF8Function)::GetProcAddress(toucan_dll, "ToucanSendGameClientDataUTF8_V1");

    82         }

    83     }

    84 }

    85 

    86 

    87 static HMODULE HelperGetToucanDLL()

    88 {

    89     if (g_toucan_dll)

    90         return g_toucan_dll;

    91 

    92     /*

    93     ** We need to enumerate the DLLs loaded to find toucan dll.

    94     ** This is done because the toucan dll changes with each update.

    95     ** The toucan dll has the following format. "xfire_toucan_{BUILD_NUMBER}.dll"

    96     ** We simply try to find a dll w/ the prefix "xfire_toucan"

    97     */

    98     HANDLE snapshot_handle = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, GetCurrentProcessId());

    99     if (snapshot_handle != INVALID_HANDLE_VALUE)

   100     {

   101         MODULEENTRY32 module_entry;

   102         module_entry.dwSize = sizeof(MODULEENTRY32);

   103 

   104         BOOL result = Module32First(snapshot_handle, &module_entry);

   105         char module_name[] = "xfire_toucan";

   106         DWORD module_name_len = sizeof(module_name)-1;

   107         while (result)

   108         {

   109             if (CompareStringA(LOCALE_USER_DEFAULT, NORM_IGNORECASE, module_entry.szModule, module_name_len, module_name, module_name_len) == CSTR_EQUAL)

   110             {

   111                 g_toucan_dll = module_entry.hModule;

   112                 break;

   113             }

   114             result = Module32Next(snapshot_handle, &module_entry);

   115         }

   116 

   117         CloseHandle(snapshot_handle);

   118     }

   119 

   120     return g_toucan_dll;

   121 }