#include <efi.h>
#include <efilib.h>
#include <intrin.h>
#include "print.h"

#pragma region types
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 const void* LPCVOID;
typedef long LONG;
typedef long LONG_PTR;
typedef char* LPSTR;
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;

typedef int BOOL;
typedef unsigned long DWORD;
typedef void* LPVOID;
typedef LPVOID LPCRITICAL_SECTION;

// Basic types
typedef void* PVOID;
typedef unsigned long DWORD;
typedef int BOOL;
typedef unsigned int UINT;
typedef unsigned short USHORT;
typedef unsigned long long ULONG64;
typedef unsigned long ULONG;
typedef void* HANDLE;
typedef long LONG;
typedef unsigned short WORD;
typedef unsigned char BYTE;
typedef long LONG_PTR;
typedef unsigned long long ULONG_PTR;
typedef unsigned char* LPBYTE;
typedef ULONG_PTR DWORD_PTR;

// Pointer to Const types
typedef const char* LPCSTR;
typedef const char* LPCTSTR;
typedef const void* LPCVOID;
typedef const wchar_t* LPCWSTR;

// Pointers to variables
typedef DWORD* LPDWORD;
typedef long* PLONG;
typedef char* LPSTR;
typedef char* LPCH;
typedef wchar_t* LPWSTR;
typedef wchar_t* LPWCH;
typedef BOOL* LPBOOL;
typedef void* LPVOID;
typedef void* HMODULE;

typedef long* LPLONG;
typedef ULONG64* PULONG64;

// Structures and callbacks
typedef struct _SECURITY_ATTRIBUTES {
    DWORD nLength;
    LPVOID lpSecurityDescriptor;
    BOOL bInheritHandle;
} SECURITY_ATTRIBUTES, *LPSECURITY_ATTRIBUTES;

typedef struct _OVERLAPPED {
    ULONG_PTR Internal;
    ULONG_PTR InternalHigh;
    DWORD Offset;
    DWORD OffsetHigh;
    HANDLE hEvent;
} OVERLAPPED, *LPOVERLAPPED;

typedef struct _STARTUPINFOA {
    DWORD cb;
    LPSTR lpReserved;
    LPSTR lpDesktop;
    LPSTR lpTitle;
    DWORD dwX;
    DWORD dwY;
    DWORD dwXSize;
    DWORD dwYSize;
    DWORD dwXCountChars;
    DWORD dwYCountChars;
    DWORD dwFillAttribute;
    DWORD dwFlags;
    USHORT wShowWindow;
    USHORT cbReserved2;
    LPBYTE lpReserved2;
    HANDLE hStdInput;
    HANDLE hStdOutput;
    HANDLE hStdError;
} STARTUPINFOA, *LPSTARTUPINFOA;

typedef struct _FILETIME {
    DWORD dwLowDateTime;
    DWORD dwHighDateTime;
} FILETIME, *LPFILETIME;

typedef struct _CPINFO {
    UINT MaxCharSize;
    BYTE DefaultChar[2];
    BYTE LeadByte[12];
} CPINFO, *LPCPINFO;

typedef unsigned short* LPWORD;
typedef char* PSTR;

// Function pointer for Fiber Local Storage callback
typedef void (*PFLS_CALLBACK_FUNCTION)(PVOID lpFlsData);

// Heap information enumeration
typedef enum _HEAP_INFORMATION_CLASS {
    HeapCompatibilityInformation,
    HeapEnableTerminationOnCorruption
} HEAP_INFORMATION_CLASS;

typedef struct _SYSTEM_INFO {
    union {
        DWORD dwOemId;
        struct {
            WORD wProcessorArchitecture;
            WORD wReserved;
        } DUMMYSTRUCTNAME;
    } DUMMYUNIONNAME;
    DWORD dwPageSize;
    LPVOID lpMinimumApplicationAddress;
    LPVOID lpMaximumApplicationAddress;
    DWORD_PTR dwActiveProcessorMask;
    DWORD dwNumberOfProcessors;
    DWORD dwProcessorType;
    DWORD dwAllocationGranularity;
    WORD wProcessorLevel;
    WORD wProcessorRevision;
} SYSTEM_INFO, *LPSYSTEM_INFO;

// Exception handling types
typedef PVOID PVECTORED_EXCEPTION_HANDLER;
typedef struct _CONTEXT {} CONTEXT, *PCONTEXT;
typedef struct _RUNTIME_FUNCTION {} RUNTIME_FUNCTION, *PRUNTIME_FUNCTION;
typedef LONG(*LPTOP_LEVEL_EXCEPTION_FILTER)(PVOID ExceptionInfo);

// Locale and Codepage types
typedef DWORD LCID;
typedef DWORD LCTYPE;
#pragma endregion

// --- Memory ---
HANDLE HeapCreate(DWORD, SIZE_T, SIZE_T) {
    PRINT(L"Called: HeapCreate\n");
    return nullptr;
}

BOOL HeapDestroy(HANDLE) {
    PRINT(L"Called: HeapDestroy\n");
    return TRUE; // nothing to do
}

LPVOID HeapAlloc(HANDLE, DWORD, SIZE_T size) {
    PRINT(L"Called: HeapAlloc with size: %ull\n", size);
    return AllocatePool(size);
}

BOOL HeapFree(HANDLE, DWORD, LPVOID ptr) {
    PRINT(L"Called: HeapFree with ptr: %p\n", ptr);
    if (ptr)
        FreePool(ptr);
    return TRUE;
}

LPVOID HeapReAlloc(HANDLE, DWORD, LPVOID ptr, SIZE_T size) {
    PRINT(L"Called: HeapReAlloc with size: %ull and ptr: %p\n", size, ptr);
    if (!ptr) return AllocatePool(size);
    LPVOID newPtr = AllocatePool(size);
    if (!newPtr) return nullptr;
    // No original size info; can't fully emulate, user must handle
    CopyMem(newPtr, ptr, size);
    FreePool(ptr);
    return newPtr;
}

SIZE_T HeapSize(HANDLE, DWORD, LPCVOID) {
    PRINT(L"Called: HeapSize\n");
    return 0;
}

// --- Threads (fake, UEFI is single-threaded at boot) ---
HANDLE GetCurrentThread() {
    PRINT(L"Called: GetCurrentThread\n");
    return reinterpret_cast<HANDLE>(1);
}

HANDLE CreateThread(void*, SIZE_T, LPTHREAD_START_ROUTINE, void*, DWORD, DWORD*) {
    PRINT(L"Called: CreateThread\n");
    return reinterpret_cast<HANDLE>(1);
}

DWORD WaitForSingleObject(HANDLE, DWORD) {
    PRINT(L"Called: WaitForSingleObject\n");
    return 0;
}

// --- Timing ---
static unsigned long long tsc_freq = 0;

static void init_tsc_freq() {
    int cpuInfo[4] = {0};
    __cpuid(cpuInfo, 0x15);

    if (cpuInfo[0] && cpuInfo[1]) {
        // leaf 0x15 gives: eax = denom, ebx = numer, ecx = crystal Hz
        unsigned eax = cpuInfo[0]; // denom
        unsigned ebx = cpuInfo[1]; // numer
        unsigned ecx = cpuInfo[2]; // crystal freq Hz

        if (ecx != 0) {
            tsc_freq = (unsigned long long)ecx * ebx / eax;
            return;
        }
    }

    // fallback: CPUID 0x16 (nominal core frequency MHz)
    __cpuid(cpuInfo, 0x16);
    if (cpuInfo[0]) {
        tsc_freq = (unsigned long long)cpuInfo[0] * 1000000ULL;
    }
}

BOOL QueryPerformanceCounter(LARGE_INTEGER *li) {
    if (!tsc_freq) init_tsc_freq();
    unsigned long long tsc = __rdtsc();
    // convert to microseconds
    li->QuadPart = (tsc * 1000000ULL) / tsc_freq;
    return TRUE;
}

BOOL QueryPerformanceFrequency(LARGE_INTEGER *li) {
    if (!tsc_freq) init_tsc_freq();
    // frequency in Hz (ticks per second)
    li->QuadPart = 1000000ULL; // since we convert to µs above
    return TRUE;
}

DWORD GetTickCount() {
    LARGE_INTEGER counter;
    QueryPerformanceCounter(&counter);
    return (DWORD)(counter.QuadPart / 1000ULL); // ms
}

// --- Module / File ---
HMODULE GetModuleHandleA(LPCSTR name) {
    PRINT(L"Called: GetModuleHandleA with name: %a\n", name);
    return reinterpret_cast<HMODULE>(1);
}

FARPROC GetProcAddress(HMODULE, LPCSTR func) {
    PRINT(L"Called: GetProcAddress with string: %a\n", func);
    return nullptr;
}

HMODULE LoadLibraryA(LPCSTR) {
    PRINT(L"Called: LoadLibraryA\n");
    return reinterpret_cast<HMODULE>(1);
}

// --- Process ---
DWORD GetCurrentProcessId() {
    PRINT(L"Called: GetCurrentProcessId\n");
    return 1;
}

void ExitProcess(UINT uExitCode) {
    PRINT(L"Called: ExitProcess\n");
    while(1); // halt
}

BOOL InitializeCriticalSectionAndSpinCount(LPCRITICAL_SECTION lpCriticalSection, DWORD dwSpinCount) {
    PRINT(L"Called: InitializeCriticalSectionAndSpinCount\n");
    return TRUE;
}

DWORD GetSystemDirectoryA(LPSTR lpBuffer, DWORD uSize) {
    PRINT(L"Called: GetSystemDirectoryA\n");
    if (lpBuffer && uSize > 0) {
        *lpBuffer = '\0';
        return 1;
    }
    return 0;
}

DWORD GetWindowsDirectoryA(LPSTR lpBuffer, DWORD uSize) {
    PRINT(L"Called: GetWindowsDirectoryA\n");
    if (lpBuffer && uSize > 0) {
        *lpBuffer = '\0';
        return 1;
    }
    return 0;
}

DWORD GetModuleFileNameA(HMODULE hModule, LPSTR lpFilename, DWORD nSize) {
    PRINT(L"Called: GetModuleFileNameA\n");
    if (lpFilename && nSize > 0) {
        *lpFilename = '\0';
        return 1;
    }
    return 0;
}

void Sleep(DWORD dwMilliseconds) {
    PRINT(L"Called: Sleep\n");
    UINTN microseconds = (UINTN)dwMilliseconds * 1000;
    BS->Stall(microseconds);
}

DWORD SetErrorMode(DWORD uMode) {
    PRINT(L"Called: SetErrorMode\n");
    return 0;
}

DWORD GetEnvironmentVariableA(LPCTSTR lpName, LPSTR lpBuffer, DWORD nSize) {
    PRINT(L"Called: GetEnvironmentVariableA\n");
    if (lpBuffer && nSize > 0) lpBuffer[0] = 0;
    return 1;
}

BOOL CloseHandle(HANDLE hObject) {
    PRINT(L"Called: CloseHandle\n");
    return TRUE;
}

DWORD GetLastError() {
    PRINT(L"Called: GetLastError\n");
    return 0;
}

DWORD GetFileSize(HANDLE hFile, LPDWORD lpFileSizeHigh) {
    PRINT(L"Called: GetFileSize\n");
    if (lpFileSizeHigh) *lpFileSizeHigh = 0;
    return 0;
}

HANDLE CreateFileA(LPCTSTR lpFileName, DWORD dwDesiredAccess, DWORD dwShareMode, LPSECURITY_ATTRIBUTES lpSecurityAttributes, DWORD dwCreationDisposition, DWORD dwFlagsAndAttributes, HANDLE hTemplateFile) {
    PRINT(L"Called: CreateFileA\n");
    return NULL;
}

BOOL ReadFile(HANDLE hFile, LPVOID lpBuffer, DWORD nNumberOfBytesToRead, LPDWORD lpNumberOfBytesRead, LPOVERLAPPED lpOverlapped) {
    PRINT(L"Called: ReadFile\n");
    *lpNumberOfBytesRead = 0;
    return TRUE;
}

DWORD SetFilePointer(HANDLE hFile, LONG lDistanceToMove, PLONG lpDistanceToMoveHigh, DWORD dwMoveMethod) {
    PRINT(L"Called: SetFilePointer\n");
    return 0;
}

BOOL FreeLibrary(HMODULE hLibModule) {
    PRINT(L"Called: FreeLibrary\n");
    return TRUE;
}

int GetThreadPriority(HANDLE hThread) {
    PRINT(L"Called: GetThreadPriority\n");
    return 0;
}

PVOID RemoveVectoredExceptionHandler(PVOID Handle) {
    PRINT(L"Called: RemoveVectoredExceptionHandler\n");
    return NULL;
}

void RaiseException(DWORD dwExceptionCode, DWORD dwExceptionFlags, DWORD nNumberOfArguments, const ULONG_PTR* lpArguments) {
    PRINT(L"Called: RaiseException\n");
    return;
}

PVOID AddVectoredExceptionHandler(ULONG FirstHandler, PVECTORED_EXCEPTION_HANDLER VectoredHandler) {
    PRINT(L"Called: AddVectoredExceptionHandler\n");
    return NULL;
}

DWORD_PTR SetThreadAffinityMask(HANDLE hThread, DWORD_PTR dwThreadAffinityMask) {
    PRINT(L"Called: SetThreadAffinityMask\n");
    return 1;
}

DWORD ResumeThread(HANDLE hThread) {
    PRINT(L"Called: ResumeThread\n");
    return 1;
}

BOOL SetThreadPriority(HANDLE hThread, int nPriority) {
    PRINT(L"Called: SetThreadPriority\n");
    return TRUE;
}

void InitializeCriticalSection(LPCRITICAL_SECTION lpCriticalSection) {
    PRINT(L"Called: InitializeCriticalSection\n");
}

void DeleteCriticalSection(LPCRITICAL_SECTION lpCriticalSection) {
    PRINT(L"Called: DeleteCriticalSection\n");
}

void EnterCriticalSection(LPCRITICAL_SECTION lpCriticalSection) {
    PRINT(L"Called: EnterCriticalSection\n");
}

void LeaveCriticalSection(LPCRITICAL_SECTION lpCriticalSection) {
    PRINT(L"Called: LeaveCriticalSection\n");
}

DWORD GetCurrentThreadId() {
    PRINT(L"Called: GetCurrentThreadId\n");
    return 1;
}

HANDLE CreateMutexA(LPSECURITY_ATTRIBUTES lpMutexAttributes, BOOL bInitialOwner, LPCTSTR lpName) {
    PRINT(L"Called: CreateMutexA with name: lpName: %a\n", lpName);
    return reinterpret_cast<HANDLE>(1);
}

BOOL ReleaseMutex(HANDLE hMutex) {
    PRINT(L"Called: ReleaseMutex\n");
    return TRUE;
}

HANDLE CreateSemaphoreA(LPSECURITY_ATTRIBUTES lpSemaphoreAttributes, LONG lInitialCount, LONG lMaximumCount, LPCTSTR lpName) {
    PRINT(L"Called: CreateSemaphoreA with name: lpName: %a\n", lpName);
    return reinterpret_cast<HANDLE>(1);
}

BOOL ReleaseSemaphore(HANDLE hSemaphore, LONG lReleaseCount, LPLONG lpPreviousCount) {
    PRINT(L"Called: ReleaseSemaphore\n");
    if (lpPreviousCount) *lpPreviousCount = 0;
    return TRUE;
}

BOOL TerminateProcess(HANDLE hProcess, UINT uExitCode) {
    PRINT(L"Called: TerminateProcess\n");
    return TRUE;
}

HANDLE GetCurrentProcess() {
    PRINT(L"Called: GetCurrentProcess\n");
    return reinterpret_cast<HANDLE>(1);
}

LPTOP_LEVEL_EXCEPTION_FILTER UnhandledExceptionFilter(LPTOP_LEVEL_EXCEPTION_FILTER lpTopLevelExceptionFilter) {
    PRINT(L"Called: UnhandledExceptionFilter\n");
    return NULL;
}

LPTOP_LEVEL_EXCEPTION_FILTER SetUnhandledExceptionFilter(LPTOP_LEVEL_EXCEPTION_FILTER lpTopLevelExceptionFilter) {
    PRINT(L"Called: SetUnhandledExceptionFilter\n");
    return NULL;
}

BOOL IsDebuggerPresent() {
    PRINT(L"Called: IsDebuggerPresent\n");
    return FALSE;
}

PVOID RtlVirtualUnwind(ULONG HandlerType, ULONG64 ImageBase, ULONG64 ControlPc, PRUNTIME_FUNCTION FunctionEntry, PCONTEXT ContextRecord, PVOID* HandlerData, PULONG64 EstablisherFrame, PCONTEXT ContextRecord2) {
    PRINT(L"Called: RtlVirtualUnwind\n");
    return NULL;
}

PRUNTIME_FUNCTION RtlLookupFunctionEntry(ULONG64 ControlPc, PULONG64 ImageBase, PVOID HistoryTable) {
    PRINT(L"Called: RtlLookupFunctionEntry\n");
    return NULL;
}

void RtlCaptureContext(PCONTEXT ContextRecord) {
    PRINT(L"Called: RtlCaptureContext\n");
}

void RtlUnwindEx(PVOID TargetFrame, PVOID TargetIp, PVOID ExceptionRecord, PVOID ReturnValue, PCONTEXT ContextRecord, PVOID HistoryTable) {
    PRINT(L"Called: RtlUnwindEx\n");
}

BOOL WriteFile(HANDLE hFile, LPCVOID lpBuffer, DWORD nNumberOfBytesToWrite, LPDWORD lpNumberOfBytesWritten, LPOVERLAPPED lpOverlapped) {
    PRINT(L"Called: WriteFile\n");
    if (lpNumberOfBytesWritten) *lpNumberOfBytesWritten = 0;
    return TRUE;
}

HMODULE GetModuleHandleW(LPCWSTR lpModuleName) {
    PRINT(L"Called: GetModuleHandleW with name: %s\n", lpModuleName);
    return NULL;
}

int SetHandleCount(int n) {
    PRINT(L"Called: SetHandleCount\n");
    return 0;
}

HANDLE GetStdHandle(DWORD nStdHandle) {
    PRINT(L"Called: GetStdHandle\n");
    return reinterpret_cast<HANDLE>(1);
}

DWORD GetFileType(HANDLE hFile) {
    PRINT(L"Called: GetFileType\n");
    return 0;
}

void GetStartupInfoA(LPSTARTUPINFOA lpStartupInfo) {
    PRINT(L"Called: GetStartupInfoA\n");
}

LPCH GetEnvironmentStrings() {
    PRINT(L"Called: GetEnvironmentStrings\n");
    return NULL;
}

BOOL FreeEnvironmentStringsA(LPCH lpszString) {
    PRINT(L"Called: FreeEnvironmentStringsA\n");
    return TRUE;
}

LPWCH GetEnvironmentStringsW() {
    PRINT(L"Called: GetEnvironmentStringsW\n");
    return NULL;
}

BOOL FreeEnvironmentStringsW(LPWCH lpszString) {
    PRINT(L"Called: FreeEnvironmentStringsW\n");
    return TRUE;
}

void GetSystemTimeAsFileTime(LPFILETIME lpSystemTimeAsFileTime) {
    PRINT(L"Called: GetSystemTimeAsFileTime\n");
}

int GetACP() {
    PRINT(L"Called: GetACP\n");
    return 0;
}

BOOL GetCPInfo(UINT CodePage, LPCPINFO lpCPInfo) {
    PRINT(L"Called: GetCPInfo\n");
    return FALSE;
}

UINT GetOEMCP() {
    PRINT(L"Called: GetOEMCP\n");
    return 0;
}

BOOL IsValidCodePage(UINT CodePage) {
    PRINT(L"Called: IsValidCodePage\n");
    return FALSE;
}

int WideCharToMultiByte(UINT CodePage, DWORD dwFlags, LPCWSTR lpWideCharStr, int cchWideChar, LPSTR lpMultiByteStr, int cbMultiByte, LPCSTR lpDefaultChar, LPBOOL lpUsedDefaultChar) {
    PRINT(L"Called: WideCharToMultiByte\n");
    if (lpWideCharStr == NULL || cchWideChar <= 0) {
        return 0; // Return 0 for invalid input
    }
    
    int convertedCount = 0;
    while (*lpWideCharStr != '\0' && convertedCount < cchWideChar) {
        *lpMultiByteStr = (char)*lpWideCharStr;
        lpMultiByteStr++;
        lpWideCharStr++;
        convertedCount++;
    }
    
    // Null-terminate the string if there is space
    if (convertedCount < cchWideChar) {
        *lpMultiByteStr = '\0';
    }
    
    return convertedCount;
}

int LCMapStringA(LCID Locale, DWORD dwMapFlags, LPCSTR lpSrcStr, int cchSrc, LPSTR lpDestStr, int cchDest) {
    PRINT(L"Called: LCMapStringA\n");
    return 0;
}

int LCMapStringW(LCID Locale, DWORD dwMapFlags, LPCWSTR lpSrcStr, int cchSrc, LPWSTR lpDestStr, int cchDest) {
    PRINT(L"Called: LCMapStringW\n");
    return 0;
}

BOOL GetStringTypeA(LCID Locale, DWORD dwInfoType, LPCSTR lpSrcStr, int cchSrc, LPWORD lpCharType) {
    PRINT(L"Called: GetStringTypeA\n");
    return FALSE;
}

BOOL GetStringTypeW(DWORD dwInfoType, LPCWSTR lpSrcStr, int cchSrc, LPWORD lpCharType) {
    PRINT(L"Called: GetStringTypeW\n");
    return FALSE;
}

int GetLocaleInfoA(LCID Locale, LCTYPE LCType, LPSTR lpLCData, int cchData) {
    PRINT(L"Called: GetLocaleInfoA\n");
    *lpLCData = '\0';
    return 1;
}

BOOL FlsSetValue(DWORD dwFlsIndex, PVOID lpFlsData) {
    PRINT(L"Called: FlsSetValue\n");
    return TRUE;
}

PSTR GetCommandLineA() {
    PRINT(L"Called: GetCommandLineA\n");
    return NULL;
}

LPVOID EncodePointer(LPVOID Ptr) {
    PRINT(L"Called: EncodePointer\n");
    return Ptr;
}

LPVOID DecodePointer(LPVOID Ptr) {
    PRINT(L"Called: DecodePointer\n");
    return Ptr;
}

BOOL FlsGetValue(DWORD dwFlsIndex) {
    PRINT(L"Called: FlsGetValue\n");
    return NULL;
}

BOOL FlsFree(DWORD dwFlsIndex) {
    PRINT(L"Called: FlsFree\n");
    return TRUE;
}

void SetLastError(DWORD dwErrCode) {
    PRINT(L"Called: SetLastError\n");
}

DWORD FlsAlloc(PFLS_CALLBACK_FUNCTION lpCallback) {
    PRINT(L"Called: FlsAlloc\n");
    return 0;
}

BOOL HeapSetInformation(HANDLE HeapHandle, HEAP_INFORMATION_CLASS HeapInformationClass, PVOID HeapInformation, SIZE_T HeapInformationLength) {
    PRINT(L"Called: HeapSetInformation\n");
    return TRUE;
}

void GetSystemInfo(LPSYSTEM_INFO lpSystemInfo) {
    PRINT(L"Called: GetSystemInfo\n");
    if (lpSystemInfo) {
        // Fill the struct with dummy values consistent with an x64 environment
        lpSystemInfo->DUMMYUNIONNAME.DUMMYSTRUCTNAME.wProcessorArchitecture = 9; // PROCESSOR_ARCHITECTURE_AMD64
        lpSystemInfo->dwPageSize = 0x1000; // 4 KB page size
        lpSystemInfo->lpMinimumApplicationAddress = (LPVOID)0x10000;
        lpSystemInfo->lpMaximumApplicationAddress = (LPVOID)0x7FFFFFFFFFFFFFFF;
        lpSystemInfo->dwActiveProcessorMask = 1;
        lpSystemInfo->dwNumberOfProcessors = 1;
        lpSystemInfo->dwProcessorType = 8664; // PROCESSOR_AMD_X8664
        lpSystemInfo->dwAllocationGranularity = 0x10000; // 64 KB
        lpSystemInfo->wProcessorLevel = 6;
        lpSystemInfo->wProcessorRevision = 0;
    }
}

int MultiByteToWideChar(UINT CodePage, DWORD dwFlags, LPCSTR lpMultiByteStr, int cbMultiByte, LPWSTR lpWideCharStr, int cchWideChar) {
    PRINT(L"Called: MultiByteToWideChar\n");
    if (lpMultiByteStr == NULL || cchWideChar <= 0) {
        return 0; // Return 0 for invalid input
    }
    
    int convertedCount = 0;
    while (*lpMultiByteStr != '\0' && convertedCount < cchWideChar) {
        *lpWideCharStr = (wchar_t)*lpMultiByteStr;
        lpMultiByteStr++;
        lpWideCharStr++;
        convertedCount++;
    }
    
    // Null-terminate the string if there is space
    if (convertedCount < cchWideChar) {
        *lpWideCharStr = L'\0';
    }
    
    return convertedCount;
}

extern int stricmp(const char *s1, const char *s2);

void* ResolveKernel32(const char* functionName) {
    if (stricmp(functionName, "HeapCreate") == 0) {
        return HeapCreate;
    } else if (stricmp(functionName, "HeapDestroy") == 0) {
        return HeapDestroy;
    } else if (stricmp(functionName, "HeapAlloc") == 0) {
        return HeapAlloc;
    } else if (stricmp(functionName, "HeapFree") == 0) {
        return HeapFree;
    } else if (stricmp(functionName, "HeapReAlloc") == 0) {
        return HeapReAlloc;
    } else if (stricmp(functionName, "HeapSize") == 0) {
        return HeapSize;
    } else if (stricmp(functionName, "GetCurrentThreadId") == 0) {
        return GetCurrentThreadId;
    } else if (stricmp(functionName, "GetCurrentThread") == 0) {
        return GetCurrentThread;
    } else if (stricmp(functionName, "CreateThread") == 0) {
        return CreateThread;
    } else if (stricmp(functionName, "WaitForSingleObject") == 0) {
        return WaitForSingleObject;
    } else if (stricmp(functionName, "CloseHandle") == 0) {
        return CloseHandle;
    } else if (stricmp(functionName, "QueryPerformanceCounter") == 0) {
        return QueryPerformanceCounter;
    } else if (stricmp(functionName, "QueryPerformanceFrequency") == 0) {
        return QueryPerformanceFrequency;
    } else if (stricmp(functionName, "GetTickCount") == 0) {
        return GetTickCount;
    } else if (stricmp(functionName, "GetModuleHandleA") == 0) {
        return GetModuleHandleA;
    } else if (stricmp(functionName, "GetProcAddress") == 0) {
        return GetProcAddress;
    } else if (stricmp(functionName, "LoadLibraryA") == 0) {
        return LoadLibraryA;
    } else if (stricmp(functionName, "GetCurrentProcessId") == 0) {
        return GetCurrentProcessId;
    } else if (stricmp(functionName, "GetCurrentProcess") == 0) {
        return GetCurrentProcess;
    } else if (stricmp(functionName, "ExitProcess") == 0) {
        return ExitProcess;
    } else if (stricmp(functionName, "InitializeCriticalSectionAndSpinCount") == 0) {
        return InitializeCriticalSectionAndSpinCount;
    } else if (stricmp(functionName, "GetSystemDirectoryA") == 0) {
        return GetSystemDirectoryA;
    } else if (stricmp(functionName, "GetWindowsDirectoryA") == 0) {
        return GetWindowsDirectoryA;
    } else if (stricmp(functionName, "GetModuleFileNameA") == 0) {
        return GetModuleFileNameA;
    } else if (stricmp(functionName, "Sleep") == 0) {
        return Sleep;
    } else if (stricmp(functionName, "SetErrorMode") == 0) {
        return SetErrorMode;
    } else if (stricmp(functionName, "GetEnvironmentVariableA") == 0) {
        return GetEnvironmentVariableA;
    } else if (stricmp(functionName, "CloseHandle") == 0) {
        return CloseHandle;
    } else if (stricmp(functionName, "GetLastError") == 0) {
        return GetLastError;
    } else if (stricmp(functionName, "GetFileSize") == 0) {
        return GetFileSize;
    } else if (stricmp(functionName, "CreateFileA") == 0) {
        return CreateFileA;
    } else if (stricmp(functionName, "ReadFile") == 0) {
        return ReadFile;
    } else if (stricmp(functionName, "SetFilePointer") == 0) {
        return SetFilePointer;
    } else if (stricmp(functionName, "FreeLibrary") == 0) {
        return FreeLibrary;
    } else if (stricmp(functionName, "GetThreadPriority") == 0) {
        return GetThreadPriority;
    } else if (stricmp(functionName, "RemoveVectoredExceptionHandler") == 0) {
        return RemoveVectoredExceptionHandler;
    } else if (stricmp(functionName, "RaiseException") == 0) {
        return RaiseException;
    } else if (stricmp(functionName, "AddVectoredExceptionHandler") == 0) {
        return AddVectoredExceptionHandler;
    } else if (stricmp(functionName, "SetThreadAffinityMask") == 0) {
        return SetThreadAffinityMask;
    } else if (stricmp(functionName, "ResumeThread") == 0) {
        return ResumeThread;
    } else if (stricmp(functionName, "SetThreadPriority") == 0) {
        return SetThreadPriority;
    } else if (stricmp(functionName, "InitializeCriticalSection") == 0) {
        return InitializeCriticalSection;
    } else if (stricmp(functionName, "DeleteCriticalSection") == 0) {
        return DeleteCriticalSection;
    } else if (stricmp(functionName, "EnterCriticalSection") == 0) {
        return EnterCriticalSection;
    } else if (stricmp(functionName, "LeaveCriticalSection") == 0) {
        return LeaveCriticalSection;
    } else if (stricmp(functionName, "GetCurrentThreadId") == 0) {
        return GetCurrentThreadId;
    } else if (stricmp(functionName, "CreateMutexA") == 0) {
        return CreateMutexA;
    } else if (stricmp(functionName, "ReleaseMutex") == 0) {
        return ReleaseMutex;
    } else if (stricmp(functionName, "CreateSemaphoreA") == 0) {
        return CreateSemaphoreA;
    } else if (stricmp(functionName, "ReleaseSemaphore") == 0) {
        return ReleaseSemaphore;
    } else if (stricmp(functionName, "TerminateProcess") == 0) {
        return TerminateProcess;
    } else if (stricmp(functionName, "GetCurrentProcess") == 0) {
        return GetCurrentProcess;
    } else if (stricmp(functionName, "UnhandledExceptionFilter") == 0) {
        return UnhandledExceptionFilter;
    } else if (stricmp(functionName, "SetUnhandledExceptionFilter") == 0) {
        return SetUnhandledExceptionFilter;
    } else if (stricmp(functionName, "IsDebuggerPresent") == 0) {
        return IsDebuggerPresent;
    } else if (stricmp(functionName, "RtlVirtualUnwind") == 0) {
        return RtlVirtualUnwind;
    } else if (stricmp(functionName, "RtlLookupFunctionEntry") == 0) {
        return RtlLookupFunctionEntry;
    } else if (stricmp(functionName, "RtlCaptureContext") == 0) {
        return RtlCaptureContext;
    } else if (stricmp(functionName, "RtlUnwindEx") == 0) {
        return RtlUnwindEx;
    } else if (stricmp(functionName, "WriteFile") == 0) {
        return WriteFile;
    } else if (stricmp(functionName, "GetModuleHandleW") == 0) {
        return GetModuleHandleW;
    } else if (stricmp(functionName, "SetHandleCount") == 0) {
        return SetHandleCount;
    } else if (stricmp(functionName, "GetStdHandle") == 0) {
        return GetStdHandle;
    } else if (stricmp(functionName, "GetFileType") == 0) {
        return GetFileType;
    } else if (stricmp(functionName, "GetStartupInfoA") == 0) {
        return GetStartupInfoA;
    } else if (stricmp(functionName, "GetEnvironmentStrings") == 0) {
        return GetEnvironmentStrings;
    } else if (stricmp(functionName, "FreeEnvironmentStringsA") == 0) {
        return FreeEnvironmentStringsA;
    } else if (stricmp(functionName, "GetEnvironmentStringsW") == 0) {
        return GetEnvironmentStringsW;
    } else if (stricmp(functionName, "FreeEnvironmentStringsW") == 0) {
        return FreeEnvironmentStringsW;
    } else if (stricmp(functionName, "GetSystemTimeAsFileTime") == 0) {
        return GetSystemTimeAsFileTime;
    } else if (stricmp(functionName, "GetACP") == 0) {
        return GetACP;
    } else if (stricmp(functionName, "GetCPInfo") == 0) {
        return GetCPInfo;
    } else if (stricmp(functionName, "GetOEMCP") == 0) {
        return GetOEMCP;
    } else if (stricmp(functionName, "IsValidCodePage") == 0) {
        return IsValidCodePage;
    } else if (stricmp(functionName, "WideCharToMultiByte") == 0) {
        return WideCharToMultiByte;
    } else if (stricmp(functionName, "LCMapStringA") == 0) {
        return LCMapStringA;
    } else if (stricmp(functionName, "LCMapStringW") == 0) {
        return LCMapStringW;
    } else if (stricmp(functionName, "GetStringTypeA") == 0) {
        return GetStringTypeA;
    } else if (stricmp(functionName, "GetStringTypeW") == 0) {
        return GetStringTypeW;
    } else if (stricmp(functionName, "GetLocaleInfoA") == 0) {
        return GetLocaleInfoA;
    } else if (stricmp(functionName, "FlsSetValue") == 0) {
        return FlsSetValue;
    } else if (stricmp(functionName, "GetCommandLineA") == 0) {
        return GetCommandLineA;
    } else if (stricmp(functionName, "EncodePointer") == 0) {
        return EncodePointer;
    } else if (stricmp(functionName, "DecodePointer") == 0) {
        return DecodePointer;
    } else if (stricmp(functionName, "FlsGetValue") == 0) {
        return FlsGetValue;
    } else if (stricmp(functionName, "FlsFree") == 0) {
        return FlsFree;
    } else if (stricmp(functionName, "SetLastError") == 0) {
        return SetLastError;
    } else if (stricmp(functionName, "FlsAlloc") == 0) {
        return FlsAlloc;
    } else if (stricmp(functionName, "GetSystemInfo") == 0) {
        return GetSystemInfo;
    } else if (stricmp(functionName, "HeapSetInformation") == 0) {
        return HeapSetInformation;
    } else if (stricmp(functionName, "MultiByteToWideChar") == 0) {
        return MultiByteToWideChar;
    }
    
    Print(L"Unknown function: %a\n", functionName);
    return NULL;
}