#pragma once

// texthook.h
// 8/24/2013 jichi
// Branch: IHF_DLL/IHF_CLIENT.h, rev 133
//
// 8/24/2013 TODO:
// - Clean up this file
// - Reduce global variables. Use namespaces or singleton classes instead.

#include "src/tree/avl.h"
#include "include/types.h"
#include <windows.h>

// jichi 12/25/2013: Header in each message sent to vnrsrv
// There are totally three elements
// - 0x0 dwAddr  hook address
// - 0x4 dwRetn  return address
// - 0x8 dwSplit split value
#define HEADER_SIZE 0xc

extern int current_hook;
extern WCHAR dll_mutex[];
//extern WCHAR dll_name[];
extern DWORD trigger;
//extern DWORD current_process_id;

// jichi 6/3/2014: Get memory range of the current module
extern DWORD processStartAddress,
             processStopAddress;

template <class T, class D, class fComp, class fCopy, class fLength>
class AVLTree;
struct FunctionInfo {
  DWORD addr;
  DWORD module;
  DWORD size;
  LPWSTR name;
};
struct SCMP;
struct SCPY;
struct SLEN;
extern AVLTree<char, FunctionInfo, SCMP, SCPY, SLEN> *tree;

void InitFilterTable();

// jichi 9/25/2013: This class will be used by NtMapViewOfSectionfor
// interprocedure communication, where constructor/destructor will NOT work.
class TextHook : public Hook
{
  int UnsafeInsertHookCode();
  DWORD UnsafeSend(DWORD dwDataBase, DWORD dwRetn);
public:
  int InsertHook();
  int InsertHookCode();
  int InitHook(const HookParam &hp, LPCSTR name = 0, WORD set_flag = 0);
  int InitHook(LPVOID addr, DWORD data, DWORD data_ind,
      DWORD split_off, DWORD split_ind, WORD type, DWORD len_off = 0);
  DWORD Send(DWORD dwDataBase, DWORD dwRetn);
  int RecoverHook();
  int RemoveHook();
  int ClearHook();
  int ModifyHook(const HookParam&);
  int SetHookName(LPCSTR name);
  int GetLength(DWORD base, DWORD in); // jichi 12/25/2013: Return 0 if failed
  void CoolDown(); // jichi 9/28/2013: flush instruction cache on wine
};

extern TextHook *hookman,
                *current_available;

//void InitDefaultHook();

struct FilterRange { DWORD lower, upper; };
extern FilterRange *filter;

extern bool running,
            live;

extern HANDLE hPipe,
              hmMutex;

DWORD WINAPI WaitForPipe(LPVOID lpThreadParameter);
DWORD WINAPI CommandPipe(LPVOID lpThreadParameter);

//void RequestRefreshProfile();

//typedef DWORD (*InsertHookFun)(DWORD);
//typedef DWORD (*IdentifyEngineFun)();
//typedef DWORD (*InsertDynamicHookFun)(LPVOID addr, DWORD frame, DWORD stack);
//extern IdentifyEngineFun IdentifyEngine;
//extern InsertDynamicHookFun InsertDynamicHook;

// jichi 9/28/2013: Protect pipeline in wine
void CliLockPipe();
void CliUnlockPipe();

// EOF