httpserver: properly implement persistance

Author: RaphaelIT7

source/lua.cpp

@@ -238,7 +238,6 @@ void Lua::RemoveLuaData(GarrysMod::Lua::ILuaInterface* LUA)
 	Msg("holylib - Removed thread data %p\n", data);
 }
 
-
 static void LuaCheck(const CCommand& args)
 {
 	if (!g_Lua)

source/lua.h

@@ -27,7 +27,7 @@ namespace Lua
 	// This will be required when we work with multiple ILuaInterface's
 	struct StateData
 	{
-		int iReference = -1; // Reference to this structure to stop GC
+		void* pOtherData[4]; // If any other plugin wants to use this, they can.
 	};
 
 	extern Lua::StateData* GetLuaData(GarrysMod::Lua::ILuaInterface* LUA);

source/modules/httpserver.cpp

@@ -11,6 +11,7 @@ class CHTTPServerModule : public IModule
 public:
 	virtual void LuaInit(bool bServerInit) OVERRIDE;
 	virtual void LuaShutdown() OVERRIDE;
+	virtual void Shutdown() OVERRIDE;
 	virtual void Think(bool bSimulating) OVERRIDE;
 	virtual const char* Name() { return "httpserver"; };
 	virtual int Compatibility() { return LINUX32 | LINUX64 | WINDOWS32 | WINDOWS64; };
@@ -49,9 +50,14 @@ enum
 	HTTPSERVER_OFFLINE
 };
 
+static std::unordered_set<HttpServer*> g_pHttpServers;
 class HttpServer
 {
 public:
+	HttpServer() {
+		g_pHttpServers.insert(this);
+	}
+
 	~HttpServer()
 	{
 		if (!ThreadInMainThread())
@@ -67,6 +73,7 @@ class HttpServer
 			Util::ReferenceFree(ref, "HttpServer::~HttpServer - Handler references");
 
 		m_pHandlerReferences.clear();
+		g_pHttpServers.erase(this);
 	}
 
 	void Start(const char* address, unsigned short port);
@@ -128,11 +135,11 @@ class HttpServer
 
 private:
 	unsigned char m_iStatus = HTTPSERVER_OFFLINE;
-	unsigned short m_iPort;
+	unsigned short m_iPort = 0;
 	unsigned int m_iThreadSleep = 5; // How long the threads sleep / wait for a request to be handled
 	bool m_bUpdate = false;
 	bool m_bInUpdate = false;
-	std::string m_strAddress;
+	std::string m_strAddress = "";
 	std::vector<HttpRequest*> m_pRequests;
 	std::vector<int> m_pHandlerReferences; // Contains the Lua references to the handler functions.
 	httplib::Server m_pServer;
@@ -743,11 +750,11 @@ LUA_FUNCTION_STATIC(httpserver_Destroy)
 
 LUA_FUNCTION_STATIC(httpserver_GetAll)
 {
-	LUA->PreCreateTable(g_pPushedHttpServer.size(), 0);
+	LUA->PreCreateTable(g_pHttpServers.size(), 0);
 		int idx = 0;
-		for (auto& [_, value] : g_pPushedHttpServer)
+		for (auto& server : g_pHttpServers)
 		{
-			value->Push();
+			Push_HttpServer(server);
 			Util::RawSetI(LUA, -2, ++idx);
 		}
 
@@ -758,11 +765,11 @@ LUA_FUNCTION_STATIC(httpserver_FindByName)
 {
 	std::string strName = LUA->CheckString(1);
 	bool bPushed = false;
-	for (auto& [_, value] : g_pPushedHttpServer)
+	for (auto& server : g_pHttpServers)
 	{
-		if (((HttpServer*)value->GetData())->GetName() == strName)
+		if (server->GetName() == strName)
 		{
-			value->Push();
+			Push_HttpServer(server);
 			bPushed = true;
 		}
 	}
@@ -862,12 +869,22 @@ void CHTTPServerModule::LuaShutdown()
 {
 	Util::NukeTable("httpserver");
 
-	// Shouls we allow HttpServers to persist across map changes? Maybe later, but if anyone wants this request it.
+	// HttpServers WILL persist across map changes.
 	DeleteAll_HttpResponse();
 	DeleteAll_HttpRequest();
 	DeleteAll_HttpServer();
 }
 
+void CHTTPServerModule::Shutdown()
+{
+	std::vector<HttpServer*> httpServers; // Copy of g_pHttpServers as when deleting we can't iterate over it.
+	for (auto& server : g_pHttpServers)
+		httpServers.push_back(server);
+
+	for (auto& server : httpServers)
+		delete server;
+}
+
 void CHTTPServerModule::Think(bool simulating)
 {
 	VPROF_BUDGET("HolyLib - CHTTPServerModule::Think", VPROF_BUDGETGROUP_HOLYLIB);