#include "dispatch.h"


#define MAIN_OFFSET(offset) ((void *)((0x8048000) + (offset)))
#define SUBHOOK_INIT(func, addr)    \
    fn##func func = (fn##func)addr; \
    FuncHook h##func

#define SUBHOOK_SETUP(func) h##func.Hook((void **)&func, (void *)_##func)

#define INIT_HOOK(func, addr)										\
	using Type_##func = decltype(hook_##func);						\
	Type_##func * old_##func = (Type_##func *)addr;					\
	FuncHook mhook_##func

#define HOOK_SETUP(func) mhook_##func.Hook((void **)&old_##func, (void *)hook_##func)

class Controller
{
public:
	SINGLETON_DEFINE_S(Controller);

private:
	Controller();
	~Controller();
public:
	void init();


private:

	static int hook_IPacketDispatcher_ParamBase_dispatch_template(int a1, int a2, int a3);
	/**
	 * @brief hook修复镶嵌徽章
	 * @param pDispatcher_UseJewel
	 * @param pUser
	 * @param pBuf
	 * @return
	*/
	static int hook_UseJewel_dispatch_sig(void* pDispatcher_UseJewel, CUser* pUser, PacketBuf* pBuf);

	//使用蜜蜡
	static int hook_Dispatcher_ModItemAttr_dispatch_sig(Dispatcher_ModItemAttr* dis_mod, CUser* user, PacketBuf* pBuf);


	/**
	 * @brief hook收包处理
	 * @param a1
	 * @param a2
	 * @param a3
	 * @param a4
	 * @param src
	 * @param a6
	 * @param a7
	 * @param a8
	 * @return
	*/
	static int hook_PacketDispatcher_doDispatch(PacketDispatcher* a1, CUser* a2, int a3, int packet_id, char* packet_src, int pecakt_len, int a7, int a8);

	static int hook_DisPatcher_MoveMap_dispatch_sig(void* a1, CUser* pUser, PacketBuf* pBuf);

	static int hook_Inter_LoadEtc_dispatch_sig(void* a1, CUser* pUser, char* a3);

	static int hook_DisPatcher_ReturnToSelectCharacter_dispatch_sig(void* a1, CUser* pUser, char* a3);

private:

	INIT_HOOK(IPacketDispatcher_ParamBase_dispatch_template, base::IPacketDispatcher::ParamBase::dispatch_template);
	INIT_HOOK(PacketDispatcher_doDispatch, base::PacketDispatcher::doDispatch);
	INIT_HOOK(Dispatcher_ModItemAttr_dispatch_sig, base::Dispatcher_ModItemAttr::dispatch_sig);
	INIT_HOOK(UseJewel_dispatch_sig, base::Dispatcher_UseJewel::dispatch_sig);
	INIT_HOOK(DisPatcher_MoveMap_dispatch_sig, base::DisPatcher_MoveMap::dispatch_sig);
	INIT_HOOK(Inter_LoadEtc_dispatch_sig, base::Inter_LoadEtc::dispatch_sig);
	INIT_HOOK(DisPatcher_ReturnToSelectCharacter_dispatch_sig, base::DisPatcher_ReturnToSelectCharacter::dispatch_sig);
};