Add core->requestHTTP4. Force IPV4 requests for annonce. Use bind add…

…ress if provided.
openmultiplayer · Nov 24, 2023 · c4e1ff8 · c4e1ff8
1 parent 7825487
commit c4e1ff8
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 3 deletions.
diff --git a/SDK/include/core.hpp b/SDK/include/core.hpp
@@ -344,6 +344,13 @@ struct ICore : public IExtensible, public ILogger
 			network->update();
 		}
 	}
+
+	/// Launch an HTTP request (IPV4) and read the response
+	/// @param handler The handler that will handle the response
+	/// @param type The request type
+	/// @param url The URL
+	/// @param[opt] data The POST data
+	virtual void requestHTTP4(HTTPResponseHandler* handler, HTTPRequestType type, StringView url, StringView data = StringView()) = 0;
 };
 
 /// Helper class to get streamer config properties

diff --git a/Server/Components/LegacyNetwork/legacy_network_impl.cpp b/Server/Components/LegacyNetwork/legacy_network_impl.cpp
@@ -859,7 +859,7 @@ void RakNetLegacyNetwork::start()
 		if (*config.getBool("announce"))
 		{
 			const String get = "https://api.open.mp/0.3.7/announce/" + std::to_string(port);
-			core->requestHTTP(new AnnounceHTTPResponseHandler(core), HTTPRequestType::HTTPRequestType_Get, get.data());
+			core->requestHTTP4(new AnnounceHTTPResponseHandler(core), HTTPRequestType::HTTPRequestType_Get, get.data());
 		}
 	}
 

diff --git a/Server/Source/core_impl.hpp b/Server/Source/core_impl.hpp
@@ -824,11 +824,13 @@ class Config final : public IEarlyConfig
 class HTTPAsyncIO
 {
 public:
-	HTTPAsyncIO(HTTPResponseHandler* handler, HTTPRequestType type, StringView url, StringView data)
+	HTTPAsyncIO(HTTPResponseHandler* handler, HTTPRequestType type, StringView url, StringView data, bool force_v4 = false, StringView bindAddr = "")
 		: handler(handler)
 		, type(type)
 		, url(url)
 		, data(data)
+		, force_v4(force_v4)
+		, bindAddr(bindAddr)
 		, finished(false)
 		, response(0)
 	{
@@ -903,6 +905,12 @@ class HTTPAsyncIO
 		request.set_write_timeout(Seconds(5));
 		request.set_keep_alive(true);
 
+		if (params->force_v4)
+			request.set_address_family(AF_INET);
+
+		if (!params->bindAddr.empty())
+			request.set_interface(params->bindAddr);
+
 		// Run request
 		httplib::Result res(nullptr, httplib::Error::Canceled);
 		switch (type)
@@ -937,6 +945,9 @@ class HTTPAsyncIO
 	String url;
 	String data;
 
+	bool force_v4;
+	String bindAddr;
+
 	std::atomic_bool finished;
 	int response;
 	String body;
@@ -2091,4 +2102,10 @@ class Core final : public ICore, public PlayerConnectEventHandler, public Consol
 		}
 		return false;
 	}
-};
+
+	void requestHTTP4(HTTPResponseHandler* handler, HTTPRequestType type, StringView url, StringView data) override
+	{
+		HTTPAsyncIO* httpIO = new HTTPAsyncIO(handler, type, url, data, true, config.getString("network.bind"));
+		httpFutures.emplace(httpIO);
+	}
+};