#include"Godot.h"

bool InsertGodotHook_X64() {
	const BYTE bytes[] = { 0x8B,0x40,0xFC,0x83,0xF8,0x01,0x83,0xD0,0xFF,0x41,0x39,0xC6 };

	ULONG64 range = min(processStopAddress - processStartAddress, MAX_REL_ADDR);
	for (auto addr : Util::SearchMemory(bytes, sizeof(bytes), PAGE_EXECUTE, processStartAddress, processStartAddress + range)) {
		HookParam myhp;
		myhp.address = addr;

		myhp.type = USING_STRING | CODEC_UTF16 | NO_CONTEXT; // /HQ 不使用上下文区分 把所有线程的文本都提取
		//myhp.padding = 0xc;//[esp+4]+padding
		// data_offset
		myhp.offset=get_reg(regs::rax);
		myhp.text_fun = [](hook_stack* stack, HookParam *hp, uintptr_t* data, uintptr_t* split, size_t* count)
		{
			*data=(stack->rax);
			int len = *(int*)(*data - 4);
			if(len!=wcslen((wchar_t*)*data))return;
			*count=len*2;
		};
		char nameForUser[HOOK_NAME_SIZE] = "RichTextLabel_add_text";
		
		ConsoleOutput("Insert: Godot_add_text_X64 Hook ");
		return NewHook(myhp, nameForUser);
	}

	ConsoleOutput("Godot_x64: pattern not found");
	return false;
}
bool InsertGodotHook2_X64() {
	
	/*
	* Sample games:
	* https://vndb.org/r109138
	*/
	const BYTE bytes[] = {
		0x48, 0x8B, 0x94, 0x24, XX4,       // mov rdx,[rsp+000001C0]	<- hook here
		0x4C, 0x89, 0xE1,                  // mov rcx,r12
		0xE8, XX4,                         // call NULL-Windows.exe+D150
		0x49, 0x8B, 0x06,                  // mov rax,[r14]
		0x48, 0x85, 0xC0,                  // test rax,rax
		0x0F, 0x85, XX4                    // jne NULL-Windows.exe+A359D4

	};

	ULONG64 range = min(processStopAddress - processStartAddress, X64_MAX_REL_ADDR);
	for (auto addr : Util::SearchMemory(bytes, sizeof(bytes), PAGE_EXECUTE, processStartAddress, processStartAddress + range)) {
		HookParam hp;
		hp.address = addr;
		hp.offset=get_reg(regs::rcx);
		hp.type = USING_STRING | CODEC_UTF16;
		ConsoleOutput("INSERT Godot2_x64 Hook ");
		return NewHook(hp, "Godot2_x64");
	}

	ConsoleOutput("Godot2_x64: pattern not found");
	return false;
}
bool Godot::attach_function(){
	auto _= InsertGodotHook_X64();
	_=InsertGodotHook2_X64()||_;
	return _;
}