#ifndef __AIRMEMORY_H__
#define __AIRMEMORY_H__

#include<Windows.h>
#include<psapi.h>
#pragma comment(lib, "psapi.lib")

#define MaxSize 256

#define CALL 0xE8
#define JMP 0xE9
#define JNE 0x850F
#define JE 0x840F

class AirMemory{
    private:
        DWORD BaseAddress;
        DWORD Memory_Start;
        DWORD Memory_End;
        DWORD MemoryDump;
        bool IsDLL;
        DWORD OldAddr;
        DWORD OldSize;
        DWORD OldProtect;

        bool FullAccess(DWORD dwAddr, DWORD dwSize);
        bool RestoreProtect();

    public:
        AirMemory();
        void Init(char *ModuleName = NULL);
        void CreateMemoryDump();
        DWORD AobScan(char *Aob, int Result = 0);
        bool MemoryWriter(DWORD dwAddr, char *Code);
        bool RestoreMemory(DWORD dwAddr, DWORD dwSize);
        bool WriteHook(DWORD dwPrev, WORD OpCode, void(*dwNext)(), DWORD *RetAddr = NULL, DWORD dwAdd = 0);
        void GetDumpInfo(DWORD *MS, DWORD *ME, DWORD *MD);
        DWORD GetBaseAddress();
        void PointerHook(DWORD dwPointer, void(*NewFunction)(), DWORD *OldFunction);
        DWORD AutoVMHook(DWORD dwFunction, void(*dwNext)(), DWORD *RetAddr, DWORD dwMinAddr = 0);
};

#endif