HPM: Hooking!

http://hercules.ws/board/topic/2399-hpm-hooking-now-available/
Signed-off-by: shennetsind <ind@henn.et>

1 files changed, 173 insertions, 0 deletions

diff --git a/src/plugins/HPMHooking.c b/src/plugins/HPMHooking.c
new file mode 100644
index 000000000..8b4f06ae0
--- /dev/null
+++ b/src/plugins/HPMHooking.c
@@ -0,0 +1,173 @@
+// Copyright (c) Hercules Dev Team, licensed under GNU GPL.
+// See the LICENSE file
+// Sample Hercules Plugin
+
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+
+#include "../common/HPMi.h"
+#include "../common/mmo.h"
+#include "../common/socket.h"
+#include "../common/malloc.h"
+#include "../common/db.h"
+
+#include "../map/map.h"
+#include "../map/path.h"
+#include "../map/chrif.h"
+#include "../map/clif.h"
+#include "../map/duel.h"
+#include "../map/intif.h"
+#include "../map/npc.h"
+#include "../map/pc.h"
+#include "../map/status.h"
+#include "../map/mob.h"
+#include "../map/npc.h"
+#include "../map/chat.h"
+#include "../map/itemdb.h"
+#include "../map/storage.h"
+#include "../map/skill.h"
+#include "../map/trade.h"
+#include "../map/party.h"
+#include "../map/unit.h"
+#include "../map/battle.h"
+#include "../map/battleground.h"
+#include "../map/quest.h"
+#include "../map/script.h"
+#include "../map/mapreg.h"
+#include "../map/guild.h"
+#include "../map/pet.h"
+#include "../map/homunculus.h"
+#include "../map/instance.h"
+#include "../map/mercenary.h"
+#include "../map/elemental.h"
+#include "../map/atcommand.h"
+#include "../map/log.h"
+#include "../map/mail.h"
+#include "../map/irc-bot.h"
+
+HPExport struct hplugin_info pinfo = {
+	"HPMHooking",   // Plugin name
+	SERVER_TYPE_MAP,// Which server types this plugin works with?
+	"0.1",          // Plugin version
+	HPM_VERSION,    // HPM Version (don't change, macro is automatically updated)
+};
+
+#define HP_POP(x) #x , (void**)(&x)
+#define HP_POP2(x) (void*)x
+DBMap *hp_db;/* hooking points db -- for quick lookup */
+
+struct HookingPointData {
+	char* name;
+	void **sref;
+	void *tref;
+	int idx;
+};
+
+struct HPMHookPoint {
+	void *func;
+	unsigned int pID;
+};
+
+struct HPMHooksCore {
+	#include "../plugins/HPMHooking/HPMHooking.HPMHooksCore.inc"
+	struct {
+		int total;
+	} data;
+} HPMHooks;
+
+bool *HPMforce_return;
+
+void HPM_HP_final(void);
+void HPM_HP_load(void);
+
+HPExport void server_post_final (void) {
+	HPM_HP_final();
+}
+
+HPExport bool Hooked (bool *fr) {
+	HPMforce_return = fr;
+	DB = GET_SYMBOL("DB");
+	iMalloc = GET_SYMBOL("iMalloc");
+#include "../plugins/HPMHooking/HPMHooking.GetSymbol.inc"
+	HPM_HP_load();
+	return true;
+}
+
+
+HPExport bool HPM_Plugin_AddHook(enum HPluginHookType type, const char *target, void *hook, unsigned int pID) {
+	struct HookingPointData *hpd;
+	
+	if( hp_db && (hpd = strdb_get(hp_db,target)) ) {
+		struct HPMHookPoint **hp = NULL;
+		int *count = NULL;
+		
+		if( type == HOOK_TYPE_PRE ) {
+			hp = (struct HPMHookPoint **)((char *)&HPMHooks.list + (sizeof(struct HPMHookPoint *)*hpd->idx));
+			count = (int *)((char *)&HPMHooks.count + (sizeof(int)*hpd->idx));
+		} else {
+			hp = (struct HPMHookPoint **)((char *)&HPMHooks.list + (sizeof(struct HPMHookPoint *)*(hpd->idx+1)));
+			count = (int *)((char *)&HPMHooks.count + (sizeof(int)*(hpd->idx+1)));
+		}
+		
+		if( hp ) {
+			*count += 1;
+			
+			RECREATE(*hp, struct HPMHookPoint, *count);
+			
+			(*hp)[*count - 1].func = hook;
+			(*hp)[*count - 1].pID = pID;
+			
+			*(hpd->sref) = hpd->tref;
+			
+			return true;
+		}
+	}
+	
+	return false;
+}
+
+#include "../plugins/HPMHooking/HPMHooking.Hooks.inc"
+
+void HPM_HP_final(void) {
+	int i, len = HPMHooks.data.total * 2;
+	
+	if( hp_db )
+		db_destroy(hp_db);
+	
+	for(i = 0; i < len; i++) {
+		int *count = (int *)((char *)&HPMHooks.count + (sizeof(int)*(i)));
+		
+		if( count && *count ) {
+			struct HPMHookPoint **hp = (struct HPMHookPoint **)((char *)&HPMHooks.list + (sizeof(struct HPMHookPoint *)*(i)));
+			
+			if( hp && *hp )
+				aFree(*hp);
+		}
+	}
+	
+}
+
+void HPM_HP_load(void) {
+	#include "../plugins/HPMHooking/HPMHooking.HookingPoints.inc"
+	int i, len = ARRAYLENGTH(HookingPoints);
+	
+	memset(&HPMHooks,0,sizeof(struct HPMHooksCore));
+	
+	hp_db = strdb_alloc(DB_OPT_BASE|DB_OPT_DUP_KEY|DB_OPT_RELEASE_DATA, HookingPointsLenMax);
+	
+	for(i = 0; i < len; i++) {
+		struct HookingPointData *hpd = NULL;
+		
+		CREATE(hpd, struct HookingPointData, 1);
+		
+		memcpy(hpd, &HookingPoints[i], sizeof(struct HookingPointData));
+		
+		strdb_put(hp_db, HookingPoints[i].name, hpd);
+		
+		HPMHooks.data.total++;
+	}
+	
+	#include "../plugins/HPMHooking/HPMHooking.sources.inc"
+}
+