mirror of
https://github.com/esuo1198/TaikoArcadeLoader.git
synced 2026-10-02 03:37:57 +03:00
120 lines
8.8 KiB
C
120 lines
8.8 KiB
C
#pragma once
|
|
#include <MinHook.h>
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <toml.h>
|
|
#include <windows.h>
|
|
|
|
typedef int8_t i8;
|
|
typedef int16_t i16;
|
|
typedef int32_t i32;
|
|
typedef int64_t i64;
|
|
typedef uint8_t u8;
|
|
typedef uint16_t u16;
|
|
typedef uint32_t u32;
|
|
typedef uint64_t u64;
|
|
typedef void event ();
|
|
|
|
#ifdef DEFINE_GUID
|
|
#undef DEFINE_GUID
|
|
#endif
|
|
|
|
#define DEFINE_GUID(name, l, w1, w2, b1, b2, b3, b4, b5, b6, b7, b8) const GUID name = { l, w1, w2, { b1, b2, b3, b4, b5, b6, b7, b8 } }
|
|
|
|
#ifdef BASE_ADDRESS
|
|
#define ASLR(address, handle) ((u64)handle + (u64)address - (u64)BASE_ADDRESS)
|
|
#endif
|
|
|
|
#define FUNCTION_PTR(returnType, callingConvention, function, location, ...) \
|
|
returnType (callingConvention *function) (__VA_ARGS__) = (returnType (callingConvention *) (__VA_ARGS__)) (location)
|
|
|
|
#define PROC_ADDRESS(libraryName, procName) GetProcAddress (LoadLibrary (TEXT (libraryName)), procName)
|
|
|
|
#define HOOK(returnType, callingConvention, functionName, location, ...) \
|
|
typedef returnType callingConvention (*functionName) (__VA_ARGS__); \
|
|
functionName original##functionName = NULL; \
|
|
void *where##functionName = (void *)location; \
|
|
returnType callingConvention implOf##functionName (__VA_ARGS__)
|
|
|
|
#define HOOK_DYNAMIC(returnType, callingConvention, functionName, ...) \
|
|
typedef returnType callingConvention (*functionName) (__VA_ARGS__); \
|
|
functionName original##functionName = NULL; \
|
|
void *where##functionName = NULL; \
|
|
returnType callingConvention implOf##functionName (__VA_ARGS__)
|
|
|
|
#define INSTALL_HOOK(functionName) \
|
|
{ \
|
|
MH_Initialize (); \
|
|
MH_CreateHook ((void *)where##functionName, (void *)implOf##functionName, (void **)(&original##functionName)); \
|
|
MH_EnableHook ((void *)where##functionName); \
|
|
}
|
|
|
|
#define INSTALL_HOOK_DYNAMIC(functionName, location) \
|
|
{ \
|
|
where##functionName = (void *)location; \
|
|
INSTALL_HOOK (functionName); \
|
|
}
|
|
|
|
#define WRITE_MEMORY(location, type, ...) \
|
|
{ \
|
|
const type data[] = { __VA_ARGS__ }; \
|
|
DWORD oldProtect; \
|
|
VirtualProtect ((void *)(location), sizeof (data), PAGE_EXECUTE_READWRITE, &oldProtect); \
|
|
memcpy ((void *)(location), data, sizeof (data)); \
|
|
VirtualProtect ((void *)(location), sizeof (data), oldProtect, &oldProtect); \
|
|
}
|
|
|
|
#define WRITE_MEMORY_STRING(location, data, length) \
|
|
{ \
|
|
DWORD oldProtect; \
|
|
VirtualProtect ((void *)(location), length, PAGE_EXECUTE_READWRITE, &oldProtect); \
|
|
memcpy ((void *)(location), data, length); \
|
|
VirtualProtect ((void *)(location), length, oldProtect, &oldProtect); \
|
|
}
|
|
|
|
#define WRITE_NOP(location, count) \
|
|
{ \
|
|
DWORD oldProtect; \
|
|
VirtualProtect ((void *)(location), (size_t)(count), PAGE_EXECUTE_READWRITE, &oldProtect); \
|
|
for (size_t i = 0; i < (size_t)(count); i++) \
|
|
*((uint8_t *)(location) + i) = 0x90; \
|
|
VirtualProtect ((void *)(location), (size_t)(count), oldProtect, &oldProtect); \
|
|
}
|
|
|
|
#define WRITE_NULL(location, count) \
|
|
{ \
|
|
DWORD oldProtect; \
|
|
VirtualProtect ((void *)(location), (size_t)(count), PAGE_EXECUTE_READWRITE, &oldProtect); \
|
|
for (size_t i = 0; i < (size_t)(count); i++) \
|
|
*((uint8_t *)(location) + i) = 0x00; \
|
|
VirtualProtect ((void *)(location), (size_t)(count), oldProtect, &oldProtect); \
|
|
}
|
|
|
|
#define COUNTOFARR(arr) sizeof (arr) / sizeof (arr[0])
|
|
#define RETURN_FALSE(returnType, callingConvention, functionName, ...) \
|
|
returnType callingConvention functionName (__VA_ARGS__) { return 0; }
|
|
#define RETURN_TRUE(returnType, callingConvention, functionName, ...) \
|
|
returnType callingConvention functionName (__VA_ARGS__) { return 1; }
|
|
|
|
#define INFO_COLOUR FOREGROUND_GREEN
|
|
#define WARNING_COLOUR (FOREGROUND_RED | FOREGROUND_GREEN)
|
|
#define ERROR_COLOUR FOREGROUND_RED
|
|
#define printInfo(format, ...) printColour (INFO_COLOUR, format, __VA_ARGS__)
|
|
#define printWarning(format, ...) printColour (WARNING_COLOUR, format, __VA_ARGS__)
|
|
#define printError(format, ...) printColour (ERROR_COLOUR, format, __VA_ARGS__)
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
char *configPath (char *name);
|
|
toml_table_t *openConfig (char *configFilePath);
|
|
toml_table_t *openConfigSection (toml_table_t *config, char *sectionName);
|
|
bool readConfigBool (toml_table_t *table, char *key, bool notFoundValue);
|
|
int64_t readConfigInt (toml_table_t *table, char *key, int64_t notFoundValue);
|
|
char *readConfigString (toml_table_t *table, char *key, char *notFoundValue);
|
|
void printColour (int colour, const char *format, ...);
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|