working, windows
This commit is contained in:
maksw committed
commit
8c002299dd
37 files changed
+4627
No files matched your search
+252
@@ -0,0 +1,252 @@
|
||||
typedef void* HWND;
|
||||
typedef void* LPUNKNOWN;
|
||||
typedef void* REFIID;
|
||||
typedef void* REFCLSID;
|
||||
typedef long HRESULT;
|
||||
typedef unsigned long DWORD;
|
||||
typedef unsigned int UINT;
|
||||
typedef int BOOL;
|
||||
typedef void* HANDLE;
|
||||
typedef void* LPVOID;
|
||||
typedef long LONG;
|
||||
typedef long LONG_PTR;
|
||||
typedef const char* LPCSTR;
|
||||
typedef size_t SIZE_T;
|
||||
typedef void* HMODULE;
|
||||
typedef HMODULE HINSTANCE;
|
||||
typedef void* FARPROC;
|
||||
typedef DWORD (*LPTHREAD_START_ROUTINE)(LPVOID lpThreadParameter);
|
||||
typedef union _LARGE_INTEGER {
|
||||
struct {
|
||||
unsigned long LowPart;
|
||||
long HighPart;
|
||||
};
|
||||
long long QuadPart;
|
||||
} LARGE_INTEGER;
|
||||
#define S_OK ((HRESULT)0L)
|
||||
#define TRUE 1
|
||||
#define FALSE 0
|
||||
#include <cstdlib>
|
||||
#include <cstdint>
|
||||
|
||||
extern "C" {
|
||||
|
||||
typedef HANDLE (*HeapCreate_t)(DWORD, SIZE_T, SIZE_T);
|
||||
typedef BOOL (*HeapDestroy_t)(HANDLE);
|
||||
typedef LPVOID (*HeapAlloc_t)(HANDLE, DWORD, SIZE_T);
|
||||
typedef BOOL (*HeapFree_t)(HANDLE, DWORD, LPVOID);
|
||||
typedef LPVOID (*HeapReAlloc_t)(HANDLE a, DWORD b, LPVOID c, SIZE_T d);
|
||||
|
||||
HeapCreate_t pHeapCreate = nullptr;
|
||||
HeapDestroy_t pHeapDestroy = nullptr;
|
||||
HeapAlloc_t pHeapAlloc = nullptr;
|
||||
HeapFree_t pHeapFree = nullptr;
|
||||
HeapReAlloc_t pHeapReAlloc = nullptr;
|
||||
|
||||
// Memory
|
||||
__declspec(dllexport) HANDLE HeapCreate(DWORD a, SIZE_T b, SIZE_T c) { return pHeapCreate(a, b, c); }
|
||||
__declspec(dllexport) BOOL HeapDestroy(HANDLE a) { return pHeapDestroy(a); }
|
||||
__declspec(dllexport) LPVOID HeapAlloc(HANDLE a, DWORD b, SIZE_T c) { return pHeapAlloc(a, b, c); }
|
||||
__declspec(dllexport) BOOL HeapFree(HANDLE a, DWORD b, LPVOID c) { return pHeapFree(a, b, c); }
|
||||
__declspec(dllexport) LPVOID HeapReAlloc(HANDLE a, DWORD b, LPVOID c, SIZE_T d) { return pHeapReAlloc(a, b, c, d); }
|
||||
|
||||
// Threads (fake)
|
||||
__declspec(dllexport) DWORD GetCurrentThreadId() { return 1; }
|
||||
__declspec(dllexport) HANDLE GetCurrentThread() { return (HANDLE)1; }
|
||||
__declspec(dllexport) HANDLE CreateThread(void*, SIZE_T, LPTHREAD_START_ROUTINE, void*, DWORD, DWORD*) { return (HANDLE)1; }
|
||||
__declspec(dllexport) DWORD WaitForSingleObject(HANDLE, DWORD) { return 0; }
|
||||
__declspec(dllexport) BOOL CloseHandle(HANDLE) { return 1; }
|
||||
|
||||
typedef DWORD (*GetTickCount_t)();
|
||||
typedef BOOL (*QueryPerformanceCounter_t)(LARGE_INTEGER* li);
|
||||
typedef BOOL (*QueryPerformanceFrequency_t)(LARGE_INTEGER* li);
|
||||
|
||||
GetTickCount_t pGetTickCount = nullptr;
|
||||
QueryPerformanceCounter_t pQueryPerformanceCounter = nullptr;
|
||||
QueryPerformanceFrequency_t pQueryPerformanceFrequency = nullptr;
|
||||
|
||||
// Timing
|
||||
__declspec(dllexport) DWORD GetTickCount() {
|
||||
return pGetTickCount();
|
||||
}
|
||||
|
||||
__declspec(dllexport) BOOL QueryPerformanceCounter(LARGE_INTEGER* li) {
|
||||
return pQueryPerformanceCounter(li);
|
||||
}
|
||||
|
||||
__declspec(dllexport) BOOL QueryPerformanceFrequency(LARGE_INTEGER* li) {
|
||||
return pQueryPerformanceFrequency(li);
|
||||
}
|
||||
|
||||
// Module / file
|
||||
__declspec(dllexport) HMODULE GetModuleHandleA(LPCSTR) { return (HMODULE)1; }
|
||||
__declspec(dllexport) FARPROC GetProcAddress(HMODULE, LPCSTR) { return nullptr; }
|
||||
__declspec(dllexport) HMODULE LoadLibraryA(LPCSTR) { return (HMODULE)1; }
|
||||
|
||||
// Process
|
||||
__declspec(dllexport) DWORD GetCurrentProcessId() { return 1; }
|
||||
__declspec(dllexport) HANDLE GetCurrentProcess() { return (HANDLE)1; }
|
||||
__declspec(dllexport) void ExitProcess(UINT) { while(1); }
|
||||
|
||||
typedef void (*OutputDebugStringA_t)(LPCSTR lpOutputString);
|
||||
|
||||
OutputDebugStringA_t pOutputDebugStringA = nullptr;
|
||||
|
||||
__declspec(dllexport) void OutputDebugStringA(LPCSTR lpOutputString) {
|
||||
pOutputDebugStringA(lpOutputString);
|
||||
}
|
||||
|
||||
HMODULE LoadLibraryExA(LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags);
|
||||
|
||||
#define IMAGE_DOS_SIGNATURE 0x5A4D // MZ
|
||||
#define IMAGE_NT_SIGNATURE 0x00004550 // PE00
|
||||
#define IMAGE_DIRECTORY_ENTRY_EXPORT 0
|
||||
|
||||
struct IMAGE_DOS_HEADER {
|
||||
uint16_t e_magic;
|
||||
uint16_t e_cblp;
|
||||
uint16_t e_cp;
|
||||
uint16_t e_crlc;
|
||||
uint16_t e_cparhdr;
|
||||
uint16_t e_minalloc;
|
||||
uint16_t e_maxalloc;
|
||||
uint16_t e_ss;
|
||||
uint16_t e_sp;
|
||||
uint16_t e_csum;
|
||||
uint16_t e_ip;
|
||||
uint16_t e_cs;
|
||||
uint16_t e_lfarlc;
|
||||
uint16_t e_ovno;
|
||||
uint16_t e_res[4];
|
||||
uint16_t e_oemid;
|
||||
uint16_t e_oeminfo;
|
||||
uint16_t e_res2[10];
|
||||
uint32_t e_lfanew;
|
||||
};
|
||||
|
||||
struct IMAGE_FILE_HEADER {
|
||||
uint16_t Machine;
|
||||
uint16_t NumberOfSections;
|
||||
uint32_t TimeDateStamp;
|
||||
uint32_t PointerToSymbolTable;
|
||||
uint32_t NumberOfSymbols;
|
||||
uint16_t SizeOfOptionalHeader;
|
||||
uint16_t Characteristics;
|
||||
};
|
||||
|
||||
struct IMAGE_DATA_DIRECTORY {
|
||||
uint32_t VirtualAddress;
|
||||
uint32_t Size;
|
||||
};
|
||||
|
||||
struct IMAGE_OPTIONAL_HEADER64 {
|
||||
uint16_t Magic;
|
||||
uint8_t MajorLinkerVersion;
|
||||
uint8_t MinorLinkerVersion;
|
||||
uint32_t SizeOfCode;
|
||||
uint32_t SizeOfInitializedData;
|
||||
uint32_t SizeOfUninitializedData;
|
||||
uint32_t AddressOfEntryPoint;
|
||||
uint32_t BaseOfCode;
|
||||
uint64_t ImageBase;
|
||||
uint32_t SectionAlignment;
|
||||
uint32_t FileAlignment;
|
||||
uint16_t MajorOperatingSystemVersion;
|
||||
uint16_t MinorOperatingSystemVersion;
|
||||
uint16_t MajorImageVersion;
|
||||
uint16_t MinorImageVersion;
|
||||
uint16_t MajorSubsystemVersion;
|
||||
uint16_t MinorSubsystemVersion;
|
||||
uint32_t Win32VersionValue;
|
||||
uint32_t SizeOfImage;
|
||||
uint32_t SizeOfHeaders;
|
||||
uint32_t CheckSum;
|
||||
uint16_t Subsystem;
|
||||
uint16_t DllCharacteristics;
|
||||
uint64_t SizeOfStackReserve;
|
||||
uint64_t SizeOfStackCommit;
|
||||
uint64_t SizeOfHeapReserve;
|
||||
uint64_t SizeOfHeapCommit;
|
||||
uint32_t LoaderFlags;
|
||||
uint32_t NumberOfRvaAndSizes;
|
||||
IMAGE_DATA_DIRECTORY DataDirectory[16];
|
||||
};
|
||||
|
||||
struct IMAGE_NT_HEADERS64 {
|
||||
uint32_t Signature;
|
||||
IMAGE_FILE_HEADER FileHeader;
|
||||
IMAGE_OPTIONAL_HEADER64 OptionalHeader;
|
||||
};
|
||||
|
||||
struct IMAGE_EXPORT_DIRECTORY {
|
||||
uint32_t Characteristics;
|
||||
uint32_t TimeDateStamp;
|
||||
uint16_t MajorVersion;
|
||||
uint16_t MinorVersion;
|
||||
uint32_t Name;
|
||||
uint32_t Base;
|
||||
uint32_t NumberOfFunctions;
|
||||
uint32_t NumberOfNames;
|
||||
uint32_t AddressOfFunctions; // RVA from base of image
|
||||
uint32_t AddressOfNames; // RVA from base of image
|
||||
uint32_t AddressOfNameOrdinals; // RVA from base of image
|
||||
};
|
||||
|
||||
int strcmp(const char* a, const char* b) {
|
||||
while (*a && (*a == *b)) {
|
||||
++a;
|
||||
++b;
|
||||
}
|
||||
return static_cast<unsigned char>(*a) - static_cast<unsigned char>(*b);
|
||||
}
|
||||
|
||||
FARPROC GetRealGetProcAddress(HMODULE hModule) {
|
||||
auto base = reinterpret_cast<uint8_t*>(hModule);
|
||||
auto dos = reinterpret_cast<IMAGE_DOS_HEADER*>(base);
|
||||
if (dos->e_magic != IMAGE_DOS_SIGNATURE) return nullptr;
|
||||
|
||||
auto nt = reinterpret_cast<IMAGE_NT_HEADERS64*>(base + dos->e_lfanew);
|
||||
if (nt->Signature != IMAGE_NT_SIGNATURE) return nullptr;
|
||||
|
||||
auto exportDirRVA = nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].VirtualAddress;
|
||||
auto exportDir = reinterpret_cast<IMAGE_EXPORT_DIRECTORY*>(base + exportDirRVA);
|
||||
|
||||
auto names = reinterpret_cast<uint32_t*>(base + exportDir->AddressOfNames);
|
||||
auto ordinals = reinterpret_cast<uint16_t*>(base + exportDir->AddressOfNameOrdinals);
|
||||
auto functions = reinterpret_cast<uint32_t*>(base + exportDir->AddressOfFunctions);
|
||||
|
||||
for (uint32_t i = 0; i < exportDir->NumberOfNames; i++) {
|
||||
const char* name = reinterpret_cast<const char*>(base + names[i]);
|
||||
if (strcmp(name, "GetProcAddress") == 0) {
|
||||
uint16_t ord = ordinals[i];
|
||||
uint32_t funcRVA = functions[ord];
|
||||
return reinterpret_cast<FARPROC>(base + funcRVA);
|
||||
}
|
||||
}
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
typedef FARPROC (*GetProcAddress_t)(HMODULE, LPCSTR);
|
||||
GetProcAddress_t pGetProcAddress = nullptr;
|
||||
|
||||
BOOL DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved) {
|
||||
if (fdwReason == 1) { // DLL_PROCESS_ATTACH
|
||||
HMODULE kernel32 = LoadLibraryExA("kernel32.dll", NULL, 0x00000800);
|
||||
pGetProcAddress = (GetProcAddress_t)GetRealGetProcAddress(kernel32);
|
||||
|
||||
pHeapCreate = (HeapCreate_t)pGetProcAddress(kernel32, "HeapCreate");
|
||||
pHeapDestroy = (HeapDestroy_t)pGetProcAddress(kernel32, "HeapDestroy");
|
||||
pHeapAlloc = (HeapAlloc_t)pGetProcAddress(kernel32, "HeapAlloc");
|
||||
pHeapFree = (HeapFree_t)pGetProcAddress(kernel32, "HeapFree");
|
||||
pHeapReAlloc = (HeapReAlloc_t)pGetProcAddress(kernel32, "HeapReAlloc");
|
||||
|
||||
pGetTickCount = (GetTickCount_t)pGetProcAddress(kernel32, "GetTickCount");
|
||||
pQueryPerformanceCounter = (QueryPerformanceCounter_t)pGetProcAddress(kernel32, "QueryPerformanceCounter");
|
||||
pQueryPerformanceFrequency = (QueryPerformanceFrequency_t)pGetProcAddress(kernel32, "QueryPerformanceFrequency");
|
||||
|
||||
pOutputDebugStringA = (OutputDebugStringA_t)pGetProcAddress(kernel32, "OutputDebugStringA");
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in new issue
Block a user