#include"AGS.h"


bool InsertAGSHook()
{
    
  const BYTE bytes1[] = {
    /*.text:0043E3A0 55                            push    ebp
.text : 0043E3A1 8B EC                         mov     ebp, esp
.text : 0043E3A3 83 EC 38                      sub     esp, 38h
.text : 0043E3A6 53                            push    ebx
.text : 0043E3A7 56                            push    esi
.text : 0043E3A8 8B F1                         mov     esi, ecx*/
    0x55,
    0x8b,0xec,
    0x83,0xec,0x38,0x53,0x56,0x8b,0xf1
  };
    
  ULONG addr = MemDbg::findBytes(bytes1, sizeof(bytes1), processStartAddress, processStopAddress); 
  if (!addr) { 
    return false;
  }  
  const BYTE bytes2[] = {
    /*	.text:0043E95E FF 75 08                      push[ebp + arg_0]
.text:0043E961 8B CE                         mov     ecx, esi
.text : 0043E963 E8 38 FA FF FF                call    sub_43E3A0*/
      0xff,0x75,0x08,
      0x8b,0xce
  };
  bool ok = false;
    
  auto addrs = findrelativecall(bytes2, sizeof(bytes2), addr, processStartAddress, processStopAddress);
  
  for(auto addr :addrs){ 
    addr = findfuncstart(addr);
    if (!addr)continue; 
    HookParam hp;
    hp.address = addr;
    hp.offset=get_reg(regs::eax);
    hp.type = USING_STRING;
    ConsoleOutput("INSERT HOOK_AGS %p",addr);
    
    ok |= NewHook(hp, "HOOK_AGS");
  } 
  
  
  return ok;

}
		 
namespace{
  bool hook2(){
    //誘惑女教師～熟れた蜜の味～
    for (auto addr : findiatcallormov_all((DWORD)TextOutA,processStartAddress,processStartAddress,processStopAddress,PAGE_EXECUTE)) { 
      
      auto funcaddr = findfuncstart(addr,0x1000);
      ConsoleOutput("funcaddr %p",funcaddr);
      if (!funcaddr) continue; 
      BYTE sig1[]={0x68,0x00,0x80,0x00,0x00,0x6a,0x00};
      BYTE sig2[]={0x2D,0xC0,0x00,0x00,0x00,0xC1,0xE0,0x08};
      BYTE sig3[]={0x83,0xC0,0x80,0xC1,0xE0,0x08};
      BYTE sig4[]={0x3C,0xA0,0x0F,0xB6,0xC0};
      int found=0;
      for(auto sigsz:std::vector<std::pair<BYTE*,int>>{{sig1,sizeof(sig1)},{sig2,sizeof(sig2)},{sig3,sizeof(sig3)},{sig4,sizeof(sig4)}}){
        auto fd= MemDbg::findBytes(sigsz.first, sigsz.second, funcaddr, addr);
        ConsoleOutput("%p",fd);
        if(fd)found+=1;
      }
      if(found==4){ 
          HookParam hp;
          hp.address = funcaddr;
          hp.type  = DATA_INDIRECT;
          hp.offset=get_stack(1);
          hp.index=0;
        return NewHook(hp, "AGS"); 
      }
    }
    return false;
  }
}
		
bool AGS::attach_function() {  
     
    return InsertAGSHook()||hook2();
}