Add server-side getServerIpFromMasterServer

Author: botder

Server/mods/deathmatch/logic/luadefs/CLuaGenericDefs.cpp

@@ -12,6 +12,12 @@
 #include "CLuaGenericDefs.h"
 #include "CStaticFunctionDefinitions.h"
 #include "CScriptArgReader.h"
+#include "CMasterServerAnnouncer.h"
+
+static auto GetServerIpFromMasterServer() -> std::string
+{
+    return g_pGame->GetMasterServerAnnouncer()->GetRemoteAddress();
+}
 
 void CLuaGenericDefs::LoadFunctions()
 {
@@ -24,6 +30,7 @@ void CLuaGenericDefs::LoadFunctions()
         {"outputConsole", ArgumentParserWarn<false, OutputConsole>},
         {"outputDebugString", ArgumentParserWarn<false, OutputScriptDebugLog>},
         {"outputServerLog", ArgumentParserWarn<false, OutputServerLog>},
+        {"getServerIpFromMasterServer", ArgumentParser<GetServerIpFromMasterServer>},
         {"getServerName", ArgumentParserWarn<nullptr, GetServerName>},
         {"getServerHttpPort", ArgumentParserWarn<nullptr, GetServerHttpPort>},
         {"getServerPassword", ArgumentParserWarn<nullptr, GetServerPassword>},

Server/mods/deathmatch/utils/CMasterServerAnnouncer.h

@@ -10,6 +10,7 @@
 
 #pragma once
 
+#include "ASE.h"
 #include "version.h"
 
 struct SMasterServerDefinition
@@ -129,10 +130,15 @@ class CMasterServer : public CRefCountable
             if (m_Stage < ANNOUNCE_STAGE_REMINDER)
             {
                 m_Stage = ANNOUNCE_STAGE_REMINDER;
+
+                CArgMap argMap;
+                argMap.SetFromString(result.pData);
+
+                if (result.iErrorCode == 200)
+                    m_remoteAddress = argMap.Get("remote_addr");
+
                 if (!m_Definition.bHideSuccess)
                 {
-                    CArgMap argMap;
-                    argMap.SetFromString(result.pData);
                     SString strOkMessage = argMap.Get("ok_message");
 
                     // Log successful initial announcement
@@ -178,6 +184,9 @@ class CMasterServer : public CRefCountable
 
     const SMasterServerDefinition& GetDefinition() const { return m_Definition; }
 
+    bool               HasRemoteAddress() const noexcept { return !m_remoteAddress.empty(); }
+    const std::string& GetRemoteAddress() const noexcept { return m_remoteAddress; }
+
     //
     // Get http downloader used for master server comms etc.
     //
@@ -192,6 +201,7 @@ class CMasterServer : public CRefCountable
     long long                     m_llLastAnnounceTime;
     long long                     m_llLastPushTime;
     const SMasterServerDefinition m_Definition;
+    std::string                   m_remoteAddress;
 };
 
 ////////////////////////////////////////////////////////////////////
@@ -270,6 +280,21 @@ class CMasterServerAnnouncer
         }
     }
 
+    /*
+     * @brief Get remote address of the first master server that has it.
+     */
+    const std::string& GetRemoteAddress() const noexcept
+    {
+        for (CMasterServer* masterServer : m_MasterServerList)
+        {
+            if (masterServer->HasRemoteAddress())
+                return masterServer->GetRemoteAddress();
+        }
+
+        static std::string empty;
+        return empty;
+    }
+
 protected:
     std::vector<CMasterServer*> m_MasterServerList;
 };