libfunctionpatcher/include/function_patcher/fpatching_defines.h

#pragma once

#include <stdint.h>

typedef enum function_replacement_library_type_t {
    LIBRARY_AVM,
    LIBRARY_CAMERA,
    LIBRARY_COREINIT,
    LIBRARY_DC,
    LIBRARY_DMAE,
    LIBRARY_DRMAPP,
    LIBRARY_ERREULA,
    LIBRARY_GX2,
    LIBRARY_H264,
    LIBRARY_LZMA920,
    LIBRARY_MIC,
    LIBRARY_NFC,
    LIBRARY_NIO_PROF,
    LIBRARY_NLIBCURL,
    LIBRARY_NLIBNSS,
    LIBRARY_NLIBNSS2,
    LIBRARY_NN_AC,
    LIBRARY_NN_ACP,
    LIBRARY_NN_ACT,
    LIBRARY_NN_AOC,
    LIBRARY_NN_BOSS,
    LIBRARY_NN_CCR,
    LIBRARY_NN_CMPT,
    LIBRARY_NN_DLP,
    LIBRARY_NN_EC,
    LIBRARY_NN_FP,
    LIBRARY_NN_HAI,
    LIBRARY_NN_HPAD,
    LIBRARY_NN_IDBE,
    LIBRARY_NN_NDM,
    LIBRARY_NN_NETS2,
    LIBRARY_NN_NFP,
    LIBRARY_NN_NIM,
    LIBRARY_NN_OLV,
    LIBRARY_NN_PDM,
    LIBRARY_NN_SAVE,
    LIBRARY_NN_SL,
    LIBRARY_NN_SPM,
    LIBRARY_NN_TEMP,
    LIBRARY_NN_UDS,
    LIBRARY_NN_VCTL,
    LIBRARY_NSYSCCR,
    LIBRARY_NSYSHID,
    LIBRARY_NSYSKBD,
    LIBRARY_NSYSNET,
    LIBRARY_NSYSUHS,
    LIBRARY_NSYSUVD,
    LIBRARY_NTAG,
    LIBRARY_PADSCORE,
    LIBRARY_PROC_UI,
    LIBRARY_SND_CORE,
    LIBRARY_SND_USER,
    LIBRARY_SNDCORE2,
    LIBRARY_SNDUSER2,
    LIBRARY_SWKBD,
    LIBRARY_SYSAPP,
    LIBRARY_TCL,
    LIBRARY_TVE,
    LIBRARY_UAC,
    LIBRARY_UAC_RPL,
    LIBRARY_USB_MIC,
    LIBRARY_UVC,
    LIBRARY_UVD,
    LIBRARY_VPAD,
    LIBRARY_VPADBASE,
    LIBRARY_ZLIB125,
    LIBRARY_OTHER,
} function_replacement_library_type_t;

#define MAXIMUM_FUNCTION_NAME_LENGTH                        100
#define FUNCTION_PATCHER_METHOD_STORE_SIZE                  40

typedef enum FunctionPatcherFunctionType {
    FUNCTION_PATCHER_STATIC_FUNCTION       = 0,
    FUNCTION_PATCHER_DYNAMIC_FUNCTION      = 1
} FunctionPatcherFunctionType;

typedef enum FunctionPatcherTargetProcess {
    FP_TARGET_PROCESS_ALL                   = 0xFF,
    FP_TARGET_PROCESS_ROOT_RPX              = 1,
    FP_TARGET_PROCESS_WII_U_MENU            = 2,
    FP_TARGET_PROCESS_TVII                  = 3,
    FP_TARGET_PROCESS_E_MANUAL              = 4,
    FP_TARGET_PROCESS_HOME_MENU             = 5,
    FP_TARGET_PROCESS_ERROR_DISPLAY         = 6,
    FP_TARGET_PROCESS_MINI_MIIVERSE         = 7,
    FP_TARGET_PROCESS_BROWSER               = 8,
    FP_TARGET_PROCESS_MIIVERSE              = 9,
    FP_TARGET_PROCESS_ESHOP                 = 10,
    FP_TARGET_PROCESS_PFID_11               = 11,
    FP_TARGET_PROCESS_DOWNLOAD_MANAGER      = 12,
    FP_TARGET_PROCESS_PFID_13               = 13,
    FP_TARGET_PROCESS_PFID_14               = 14,
    FP_TARGET_PROCESS_GAME                  = 15,
    FP_TARGET_PROCESS_GAME_AND_MENU         = 16,
} FunctionPatcherTargetProcess;

#define FUNCTION_REPLACEMENT_DATA_STRUCT_VERSION            0x00000001

typedef struct function_replacement_data_t {
    uint32_t                            VERSION;
    uint32_t                            physicalAddr;                                       /* [needs to be filled]  */
    uint32_t                            virtualAddr;                                        /* [needs to be filled]  */
    uint32_t                            replaceAddr;                                        /* [needs to be filled] Address of our replacement function */
    uint32_t                            replaceCall;                                        /* [needs to be filled] Address to access the real_function */
    function_replacement_library_type_t library;                                            /* [needs to be filled] rpl where the function we want to replace is. */
    char                                function_name[MAXIMUM_FUNCTION_NAME_LENGTH];        /* [needs to be filled] name of the function we want to replace */
    uint32_t                            realAddr;                                           /* [will be filled] Address of the real function we want to replace. */
    volatile uint32_t                   replace_data [FUNCTION_PATCHER_METHOD_STORE_SIZE];  /* [will be filled] Space for us to store some jump instructions */
    uint32_t                            restoreInstruction;                                 /* [will be filled] Copy of the instruction we replaced to jump to our code. */
    FunctionPatcherFunctionType         functionType;                                       /* [will be filled] */
    uint8_t                             alreadyPatched;                                     /* [will be filled] */
    FunctionPatcherTargetProcess        targetProcess;                                      /* [will be filled] */
} function_replacement_data_t;


#define REPLACE_FUNCTION(x, lib, function_name) \
        REPLACE_FUNCTION_FOR_PROCESS(x, lib, function_name, FP_TARGET_PROCESS_GAME_AND_MENU)

#define REPLACE_FUNCTION_FOR_PROCESS(x, lib, function_name, process) \
        REPLACE_FUNCTION_EX(x, lib, # function_name, 0, 0, process)

#define REPLACE_FUNCTION_VIA_ADDRESS(x, physicalAddress, effectiveAddress) \
        REPLACE_FUNCTION_VIA_ADDRESS_FOR_PROCESS(x, physicalAddress, effectiveAddress, FP_TARGET_PROCESS_ALL)

#define REPLACE_FUNCTION_VIA_ADDRESS_FOR_PROCESS(x, physicalAddress, effectiveAddress, process) \
        REPLACE_FUNCTION_EX(x, LIBRARY_OTHER, # x, physicalAddress, effectiveAddress, process)

#define REPLACE_FUNCTION_EX(x, lib, function_name, physicalAddress, effectiveAddress, process) \
    { \
        FUNCTION_REPLACEMENT_DATA_STRUCT_VERSION, \
        physicalAddress, \
        effectiveAddress, \
        (uint32_t) my_ ## x, \
        (uint32_t) &real_ ## x, \
        lib, \
        function_name, \
        0, \
        {}, \
        0, \
        FUNCTION_PATCHER_STATIC_FUNCTION, \
        0, \
        process \
    }

#define DECL_FUNCTION(res, name, ...) \
        res (* real_ ## name)(__VA_ARGS__) __attribute__((section(".data"))); \
        res my_ ## name(__VA_ARGS__)