holylib: added HolyLib:PostEntityConstructor

Author: RaphaelIT7

README.md

@@ -24,6 +24,7 @@ If you already had a `ghostinj.dll`, you can rename it to `ghostinj2.dll` and it
 3. Put the `gmsv_holylib_linux.so` into the `garrysmod/lua/bin/` directory.  
 
 ## Next Update
+\- [+] Added `HolyLib.InvalidateBoneCache` function and the `HolyLib:PostEntityConstructor` hook.  
 \- [#] `util.AsyncCompress/Decompress` now use two seperate threadpools(You can change their size with the new convars).  
 \- [#] Fixed a small reference leak in `util.AsyncCompress/Decompress`. The function was leaked so it's not too big.  
 
@@ -167,6 +168,10 @@ hook.Add("HolyLib:GetGModTags", "Example", function()
 end)
 ```
 
+#### HolyLib:PostEntityConstructor(Entity ent, String className)
+Called before `CBaseEntity::PostConstructor` is called.  
+This should allow you to set the `EFL_SERVER_ONLY` flag properly.  
+
 ## gameevent
 This module contains additional functions for the gameevent library.  
 With the Add/Get/RemoveClient* functions you can control the gameevents that are networked to a client which can be useful.  
@@ -1624,7 +1629,8 @@ It now throws a warning instead of crashing -> https://github.com/Facepunch/garr
 `filesystem.TimeCreated` & `filesystem.TimeAccessed` -> https://github.com/Facepunch/garrysmod-requests/issues/1633  
 `systimer` module -> https://github.com/Facepunch/garrysmod-requests/issues/1671  
 `pas` module -> https://github.com/Facepunch/garrysmod-requests/issues/140  
-`HolyLib.InvalidateBoneCache` -> `https://github.com/Facepunch/garrysmod-requests/issues/1920`  
+`HolyLib.InvalidateBoneCache` -> https://github.com/Facepunch/garrysmod-requests/issues/1920  
+`HolyLib:PostEntityConstructor` -> https://github.com/Facepunch/garrysmod-requests/issues/2440  
 
 # Things planned to add:
 https://github.com/Facepunch/garrysmod-requests/issues/1884  
@@ -1636,7 +1642,6 @@ https://github.com/Facepunch/garrysmod-requests/issues/1323
 (Should be possible?)https://github.com/Facepunch/garrysmod-requests/issues/756  
 (Gonna make a seperate ConVar for it)https://github.com/Facepunch/garrysmod-requests/issues/2120  
 https://github.com/Facepunch/garrysmod-requests/issues/1472  
-https://github.com/Facepunch/garrysmod-requests/issues/2440  
 (Maybe)https://github.com/Facepunch/garrysmod-requests/issues/2129  
 (Maybe)https://github.com/Facepunch/garrysmod-requests/issues/1962  
 (Maybe)https://github.com/Facepunch/garrysmod-requests/issues/1699    

source/detours.h

@@ -67,7 +67,7 @@ namespace Symbols
 	typedef CBasePlayer* (*Get_Player)(int iStackPos, bool shouldError);
 	extern const std::vector<Symbol> Get_PlayerSym;
 
-	typedef void (*Push_Entity)(CBaseEntity*);
+	typedef void (*Push_Entity)(CBaseEntity* ent);
 	extern const std::vector<Symbol> Push_EntitySym;
 
 	typedef CBaseEntity* (*Get_Entity)(int iStackPos, bool shouldError);
@@ -79,6 +79,9 @@ namespace Symbols
 	typedef void (*CBaseAnimating_InvalidateBoneCache)(void* ent);
 	extern const std::vector<Symbol> CBaseAnimating_InvalidateBoneCacheSym;
 
+	typedef void (*CBaseEntity_PostConstructor)(void* ent, const char* szClassname);
+	extern const std::vector<Symbol> CBaseEntity_PostConstructorSym;
+
 	//---------------------------------------------------------------------------------
 	// Purpose: holylib Symbols
 	//---------------------------------------------------------------------------------

source/modules/holylib.cpp

@@ -235,6 +235,19 @@ LUA_FUNCTION_STATIC(InvalidateBoneCache)
 	return 0;
 }
 
+static Detouring::Hook detour_CBaseEntity_PostConstructor;
+static void hook_CBaseEntity_PostConstructor(CBaseEntity* pEnt, const char* szClassname)
+{
+	if (Lua::PushHook("HolyLib:PostEntityConstructor"))
+	{
+		Util::Push_Entity(pEnt);
+		g_Lua->PushString(szClassname);
+		g_Lua->CallFunctionProtected(3, 0, true);
+	}
+
+	detour_CBaseEntity_PostConstructor.GetTrampoline<Symbols::CBaseEntity_PostConstructor>()(pEnt, szClassname);
+}
+
 void CHolyLibModule::LuaInit(bool bServerInit)
 {
 	if (!bServerInit)
@@ -283,6 +296,12 @@ void CHolyLibModule::InitDetour(bool bPreServer)
 		(void*)hook_CServerGameDLL_ShouldHideServer, m_pID
 	);
 
+	Detour::Create(
+		&detour_CBaseEntity_PostConstructor, "CBaseEntity::PostConstructor",
+		server_loader.GetModule(), Symbols::CBaseEntity_PostConstructorSym,
+		(void*)hook_CBaseEntity_PostConstructor, m_pID
+	);
+
 	SourceSDK::ModuleLoader engine_loader("engine");
 	Detour::Create(
 		&detour_GetGModServerTags, "GetGModServerTags",

source/symbols.cpp

@@ -48,6 +48,10 @@ namespace Symbols
 		Symbol::FromName("_ZN14CBaseAnimating19InvalidateBoneCacheEv"),
 	};
 
+	const std::vector<Symbol> CBaseEntity_PostConstructorSym = {
+		Symbol::FromName("_ZN11CBaseEntity15PostConstructorEPKc"),
+	};
+
 	//---------------------------------------------------------------------------------
 	// Purpose: holylib Symbols
 	//---------------------------------------------------------------------------------