BF3 plugin update. Added context + identity support.

1 files changed, 179 insertions, 83 deletions

diff --git a/plugins/bf3/bf3.cpp b/plugins/bf3/bf3.cpp
index 508da5160..5b7f6c69e 100644
--- a/plugins/bf3/bf3.cpp
+++ b/plugins/bf3/bf3.cpp
@@ -1,6 +1,7 @@
 /* Copyright (C) 2010-2011, Snares <snares@users.sourceforge.net>
    Copyright (C) 2005-2011, Thorvald Natvig <thorvald@natvig.com>
    Copyright (C) 2011, Ryan Austin <ryan@gameforcecenters.com>
+   Copyright (C) 2012, Bojan Hartmann <bogie@bawki.de>
 
    All rights reserved.
 
@@ -31,116 +32,211 @@
 */
 
 #include "../mumble_plugin_win32.h"
+bool ptr_chain_valid = false;
+BYTE *pmodule_bf3;
+
+// Magic ptrs
+BYTE* const state_ptr = (BYTE *) 0x023C7A9C;
+
+// Vector ptrs
+BYTE* const avatar_pos_ptr = (BYTE *) 0x023C7A30;
+BYTE* const avatar_front_ptr = (BYTE *) 0x023C7A60;
+BYTE* const avatar_top_ptr = (BYTE *) 0x023C7A50;
+
+// Context ptrs
+BYTE* const ipport_ptr = (BYTE *) 0x02396F40;
+
+// Identity ptrs
+BYTE *team_state_ptr;
+BYTE *squad_state_ptr;
+BYTE *squad_lead_state_ptr;
+
+// Offsets
+static int base_offset = 0x01F24F58;
+static int identity_offset1 = 0x1C;
+static int identity_offset2 = 0xBC;
+static int identity_offset3 = 0x36C;
+static int identity_offset4 = 0x8;
+static int squad_state_offset = 0x104;
+static int squad_lead_state_offset = 0x108;
+static int team_state_offset = 0x31C;
+
+enum state_values {
+    STATE_UNKN = 0,
+    STATE_LOADING = 1,
+    STATE_IN_GAME = 2,
+    STATE_IN_MENU = 3
+};
+
+inline bool resolve_ptrs() {
+    team_state_ptr = squad_state_ptr = squad_lead_state_ptr = NULL;
+
+    /*
+    Magic:
+        state : 0x023C7A9C                                                      BYTE        2 when playing
+                                                                                            3 while in menu/dead
+
+    Context:
+        IP:Port of server: 0x02396F40                                           char[128]   ip:port of the server
+
+    Identity:
+        Squad state: BF3.exe+0x01F24F58 + 0x1C + 0xBC + 0x36C + 0x8 + 0x104     BYTE        0 is not in squad; 1 is in Alpha squad, 2 Bravo, ... , 9 India
+        SLead state: BF3.exe+0x01F24F58 + 0x1c + 0xBC + 0x36C + 0x8 + 0x108     BYTE        0 is not lead; 1 is lead
+        Team state:  BF3.exe+0x01F24F58 + 0x1C + 0xBC + 0x31C                   BYTE        1 is blufor (US team, for example), 2 is opfor (RU), 0 is probably upcoming spec mode
+    */
+
+    BYTE *base_bf3 = peekProc<BYTE *>(pmodule_bf3 + base_offset);
+    if (!base_bf3)
+        return false;
 
-static int fetch(float *avatar_pos, float *avatar_front, float *avatar_top, float *camera_pos, float *camera_front, float *camera_top, std::string &context, std::wstring &) {
-	for (int i=0;i<3;i++)
-		avatar_pos[i] = avatar_front[i] = avatar_top[i] = camera_pos[i] = camera_front[i] = camera_top[i] = 0.0f;
-
-	//char ccontext[128];
-	char state;
-	bool ok;
-
-	ok = peekProc((BYTE *) 0x023C7A9C, &state, 1); // State value
-	if (! ok)
-		return false;
-	
-	/*
-	    Get context string; in this plugin this will be an
-	    ip:port (char 256 bytes) string
-
-	ccontext[127] = 0;
-	context = std::string(ccontext);
-	if ( state == 3 ) 
-		context = std::string("menu");
-	if ( state == 2 ) 
-		context = std::string("ingame");
-	if ( state == 1 ) 
-		context = std::string("loading");
-	*/
-
-	// Current assesment of state values are:
-	// 3 = in main menu
-	// 2 = in game
-	// 1 = loading
-	if ( state != 2 )
-		return true;	// This results in all vectors being zero which tells Mumble to ignore them.
-	
-	ok = peekProc((BYTE *) 0x023C7A30, avatar_pos, 12) &&
-		peekProc((BYTE *)  0x023C7A60, avatar_front, 12) &&
-		peekProc((BYTE *)  0x023C7A50, avatar_top, 12);
-
-	if (! ok)
-		return false;
-
-	// Flip our front vector
-	for (int i=0;i<3;i++) {
-		avatar_front[i] = -avatar_front[i];
-	}
-
-	// Convert from right to left handed
-	avatar_pos[0] = -avatar_pos[0];
-	avatar_front[0] = -avatar_front[0];
-	avatar_top[0] = -avatar_top[0];
-
-	for (int i=0;i<3;i++) {
-		camera_pos[i] = avatar_pos[i];
-		camera_front[i] = avatar_front[i];
-		camera_top[i] = avatar_top[i];
-	}
-
-	return ok;
+    BYTE *offset_ptr1 = peekProc<BYTE *>(base_bf3 + identity_offset1);
+    BYTE *offset_ptr2 = peekProc<BYTE *>(offset_ptr1 + identity_offset2);
+    BYTE *offset_ptr3 = peekProc<BYTE *>(offset_ptr2 + identity_offset3);
+    BYTE *offset_ptr4 = peekProc<BYTE *>(offset_ptr3 + identity_offset4);
+
+    squad_state_ptr = offset_ptr4 + squad_state_offset;
+    squad_lead_state_ptr = offset_ptr4 + squad_lead_state_offset;
+    team_state_ptr = offset_ptr2 + team_state_offset;
+    if (!squad_state_ptr || !squad_lead_state_ptr || !team_state_ptr)
+        return false;
+
+    return true;
+}
+
+static int fetch(float *avatar_pos, float *avatar_front, float *avatar_top, float *camera_pos, float *camera_front, float *camera_top, std::string &context, std::wstring &identity) {
+    for (int i=0;i<3;i++)
+        avatar_pos[i] = avatar_front[i] = avatar_top[i] = camera_pos[i] = camera_front[i] = camera_top[i] = 0.0f;
+
+    char ccontext[128];
+    char state;
+    BYTE squad_state;
+    BYTE is_squadleader;
+    BYTE team_state;
+    bool ok;
+
+    ok = peekProc(state_ptr, &state, 1); // State value
+    if (! ok)
+        return false;
+
+    if (state != STATE_IN_GAME && state != STATE_IN_MENU) {
+        ptr_chain_valid = false;
+        context.clear();
+        identity.clear();
+        return true;
+    } else if (!ptr_chain_valid) {
+        if (!resolve_ptrs())
+            return false;
+        ptr_chain_valid = true;
+    }
+
+    ok = peekProc(avatar_pos_ptr, avatar_pos, 12) &&
+        peekProc(avatar_front_ptr, avatar_front, 12) &&
+        peekProc(avatar_top_ptr, avatar_top, 12) &&
+        peekProc(squad_state_ptr,&squad_state,1) &&
+        peekProc(squad_lead_state_ptr, &is_squadleader,1) &&
+        peekProc(team_state_ptr,&team_state,1) &&
+        peekProc(ipport_ptr,ccontext,128);
+
+    if (! ok)
+        return false;
+
+    ccontext[127] = 0;
+    if (ccontext[0] != '0') {
+        std::ostringstream ocontext;
+        ocontext << "{ \"ipport\": \"" << ccontext << "\" }";
+
+        context = ocontext.str();
+
+        /*
+            Get identity string.
+        */
+        std::wostringstream oidentity;
+        oidentity << "{"
+        << "\"squad\":" << static_cast<unsigned int>(squad_state) << ","
+        << "\"squad_leader\":" << (is_squadleader ? "true" : "false") << ",";
+        if (team_state == 0)
+            oidentity << "\"team\": \"SPEC\"";
+        else if (team_state == 1)
+            oidentity << "\"team\": \"US\"";
+        else if (team_state == 2)
+            oidentity << "\"team\": \"RU\"";
+        oidentity << "}";
+        identity = oidentity.str();
+    }
+
+    // Flip our front vector
+    for (int i=0;i<3;i++) {
+        avatar_front[i] = -avatar_front[i];
+    }
+
+    // Convert from right to left handed
+    avatar_pos[0] = -avatar_pos[0];
+    avatar_front[0] = -avatar_front[0];
+    avatar_top[0] = -avatar_top[0];
+
+    for (int i=0;i<3;i++) {
+        camera_pos[i] = avatar_pos[i];
+        camera_front[i] = avatar_front[i];
+        camera_top[i] = avatar_top[i];
+    }
+
+    return ok;
 }
 
 static int trylock(const std::multimap<std::wstring, unsigned long long int> &pids) {
 
-	if (! initialize(pids, L"bf3.exe"))
-		return false;
+    if (! initialize(pids, L"bf3.exe"))
+        return false;
+
+    pmodule_bf3 = getModuleAddr(L"bf3.exe");
+    if (!pmodule_bf3)
+        return false;
 
-	float apos[3], afront[3], atop[3], cpos[3], cfront[3], ctop[3];
-	std::string context;
-	std::wstring identity;
+    float apos[3], afront[3], atop[3], cpos[3], cfront[3], ctop[3];
+    std::string context;
+    std::wstring identity;
 
-	if (!fetch(apos, afront, atop, cpos, cfront, ctop, context, identity)) {
-		generic_unlock();
-		return false;
-	}
+    if (!fetch(apos, afront, atop, cpos, cfront, ctop, context, identity)) {
+        generic_unlock();
+        return false;
+    }
 
-	return true;
+    return true;
 }
 
 static const std::wstring longdesc() {
-	return std::wstring(L"Supports Battlefield 3. No identity or context support.");
+    return std::wstring(L"Supports Battlefield 3 with context and identity support.");
 }
 
 static std::wstring description(L"Battlefield 3 v944019");
 static std::wstring shortname(L"Battlefield 3");
 
 static int trylock1() {
-	return trylock(std::multimap<std::wstring, unsigned long long int>());
+    return trylock(std::multimap<std::wstring, unsigned long long int>());
 }
 
 static MumblePlugin bf3plug = {
-	MUMBLE_PLUGIN_MAGIC,
-	description,
-	shortname,
-	NULL,
-	NULL,
-	trylock1,
-	generic_unlock,
-	longdesc,
-	fetch
+    MUMBLE_PLUGIN_MAGIC,
+    description,
+    shortname,
+    NULL,
+    NULL,
+    trylock1,
+    generic_unlock,
+    longdesc,
+    fetch
 };
 
 static MumblePlugin2 bf3plug2 = {
-	MUMBLE_PLUGIN_MAGIC_2,
-	MUMBLE_PLUGIN_VERSION,
-	trylock
+    MUMBLE_PLUGIN_MAGIC_2,
+    MUMBLE_PLUGIN_VERSION,
+    trylock
 };
 
 extern "C" __declspec(dllexport) MumblePlugin *getMumblePlugin() {
-	return &bf3plug;
+    return &bf3plug;
 }
 
 extern "C" __declspec(dllexport) MumblePlugin2 *getMumblePlugin2() {
-	return &bf3plug2;
+    return &bf3plug2;
 }