#include"KiriKiri.h"
bool InsertKiriKiriZHook()
	{
		
		/*
		* Sample games:
		* RJ351843
		*/
		const BYTE bytes[] = {
			0xCC,                                                // int 3 
			0x4C, 0x89, 0x44, 0x24, 0x18,                        // mov [rsp+18],r8       <- hook here
			0x48, 0x89, 0x54, 0x24, 0x10,                        // mov [rsp+10],rdx
			0x53,                                                // push rbx
			0x56,                                                // push rsi
			0x57,                                                // push rdi
			0x41, 0x54,                                          // push r12
			0x41, 0x55,                                          // push r13
			0x41, 0x56,                                          // push r14
			0x41, 0x57,                                          // push r15
			0x48, 0x83, 0xEC, 0x40,                              // sub rsp,40
			0x48, 0xC7, 0x44, 0x24, 0x30, 0xFE, 0xFF, 0xFF, 0xFF // mov qword ptr [rsp+30],FFFFFFFFFFFFFFFE
		};

		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 + 1;
			hp.offset=get_reg(regs::rcx);
			hp.index = 0x18;
			hp.type = CODEC_UTF16 | DATA_INDIRECT; 
			return NewHook(hp, "KiriKiriZ");
		} 
		return false;
	}
bool Insertkrkrz64Hook()
{
  const BYTE BYTES[] = {
      0x41,0x0F,0xB7,0x44,0x24,0x04,
      0x89,0x43,0x20,
      0x41,0x0F,0xB7,0x44,0x24,0x06,
      0x89,0x43,0x24,
      0x41,0x0F,0xBF,0x44,0x24,0x0C,
      0x89,0x43,0x14
  };
  auto addrs = Util::SearchMemory(BYTES, sizeof(BYTES), PAGE_EXECUTE_READ, processStartAddress, processStopAddress);
  ConsoleOutput("%p %p", processStartAddress, processStopAddress);
  for (auto addr : addrs) {
    ConsoleOutput("krkrz64 %p", addr);
    const BYTE funcstart[] = { 0xcc,0xcc,0xcc,0xcc };
    addr = reverseFindBytes(funcstart, sizeof(funcstart), addr - 0x1000, addr);
    if (addr == 0)continue;
    addr += 4;
    HookParam hp;
    hp.address = addr;
    hp.type = CODEC_UTF16| DATA_INDIRECT;
    hp.offset=get_reg(regs::rcx);
    hp.index = 0x18;
    ConsoleOutput("krkrz64 %p %x", addr);
    return NewHook(hp, "krkrz64");
  }

  ConsoleOutput("krkrz64 failed");
  return false;
}
bool KiriKiri::attach_function() { 
  return Insertkrkrz64Hook()||InsertKiriKiriZHook();
}