feat(scripting/player): GetIPAddress

Author: skuzzis

plugin_files/scripting/includes/swiftly/player.h

@@ -139,6 +139,15 @@ class Player
         else
             return 0;
     }
+
+    const char *GetIPAddress()
+    {
+        void *player_GetIPAddress = FetchFunctionPtr(nullptr, "scripting_Player_GetIPAddress");
+        if (player_GetIPAddress)
+            return reinterpret_cast<Player_GetIPAddress>(player_GetIPAddress)(this->m_playerSlot);
+        else
+            return "";
+    }
 };
 
 #endif

plugin_files/scripting/includes/swiftly/swiftly_memory.h

@@ -35,6 +35,8 @@ typedef uint32_t (*Player_GetConnectedTime)(uint32_t);
 typedef int (*Player_FetchStat)(uint32_t, uint32_t);
 typedef void (*Player_SetStat)(uint32_t, uint32_t, int);
 
+typedef const char *(*Player_GetIPAddress)(uint32_t);
+
 typedef void (*Player_SendMessage)(uint32_t, HudDestination, const char *);
 typedef void (*Players_SendMessage)(HudDestination, const char *);
 

src/common.h

@@ -14,6 +14,7 @@
 #define SITE_UNKNOWN 2
 
 #include <public/playerslot.h>
+#include <public/inetchannelinfo.h>
 #include <ISmmPlugin.h>
 #include <iplayerinfo.h>
 #include <sh_vector.h>

src/components/Plugins/src/scripting/Player.cpp

@@ -342,4 +342,13 @@ SMM_API void scripting_Player_SetMatchStat(uint32 playerId, PlayerStat stat, int
         playerController->m_pActionTrackingServices->m_matchStats().m_iAssists = value;
     else if (stat == PlayerStat::DEATHS)
         playerController->m_pActionTrackingServices->m_matchStats().m_iDeaths = value;
+}
+
+SMM_API const char *scripting_Player_GetIPAddress(uint32 playerId)
+{
+    Player *player = g_playerManager->GetPlayer(playerId);
+    if (!player)
+        return "";
+
+    return player->GetIPAddress().c_str();
 }

src/entrypoint.cpp

@@ -240,7 +240,7 @@ void SwiftlyPlugin::Hook_OnClientConnected(CPlayerSlot slot, const char *pszName
 {
     if (bFakePlayer)
     {
-        Player *player = new Player(true, slot.Get(), pszName, 0);
+        Player *player = new Player(true, slot.Get(), pszName, 0, "127.0.0.1");
         g_playerManager->RegisterPlayer(player);
     }
 }
@@ -249,7 +249,8 @@ bool scripting_OnClientConnect(const OnClientConnect *e);
 
 bool SwiftlyPlugin::Hook_ClientConnect(CPlayerSlot slot, const char *pszName, uint64 xuid, const char *pszNetworkID, bool unk1, CBufferString *pRejectReason)
 {
-    Player *player = new Player(false, slot.Get(), pszName, xuid);
+    std::string ip_address = explode(pszNetworkID, ":")[0];
+    Player *player = new Player(false, slot.Get(), pszName, xuid, ip_address);
     g_playerManager->RegisterPlayer(player);
 
     player->SetConnected(true);

src/player/Player.h

@@ -93,13 +93,14 @@ enum ENetworkDisconnectionReason
 class Player
 {
 public:
-    Player(bool m_isFakeClient, int m_slot, const char *m_name, uint64 m_xuid)
+    Player(bool m_isFakeClient, int m_slot, const char *m_name, uint64 m_xuid, std::string ip_address)
     {
         this->slot = m_slot;
         this->isFakeClient = m_isFakeClient;
         this->connectTime = std::time(0);
         this->name = m_name;
         this->xuid = m_xuid;
+        this->ip_address = ip_address;
     }
 
     ~Player()
@@ -115,6 +116,7 @@ class Player
     inline void SetEHandlerIdx(int eHandleId) { this->eHandleId = eHandleId; };
     inline int GetEHandlerIdx() { return this->eHandleId; };
     inline uint32 GetConnectedTime() { return (std::time(0) - this->connectTime); };
+    inline std::string GetIPAddress() { return this->ip_address; }
 
     inline void SetConnected(bool connected) { this->isConnected = connected; };
     inline bool IsConnected() { return this->isConnected; };
@@ -145,6 +147,7 @@ class Player
     std::time_t connectTime;
     const char *name;
     uint64 xuid;
+    std::string ip_address = "0.0.0.0";
 
     std::map<std::string, std::any> internalVars;
 };