Version 1.08 with public source code release.

This commit is contained in:
icex2
2020-10-03 20:56:55 +02:00
parent 656148121d
commit 762bf140c0
555 changed files with 55502 additions and 3 deletions
+294
View File
@@ -0,0 +1,294 @@
/**
* Hook library for Pump It Up: Prime
*/
#define LOG_MODULE "prihook"
#include <string.h>
#include "capnhook/hook/lib-main.h"
#include "capnhook/hooklib/redir.h"
#include "hook/core/piu-utils.h"
#include "hook/patch/amixer-block.h"
#include "hook/patch/blacklist-url.h"
#include "hook/patch/block-keyboard-grab.h"
#include "hook/patch/gfx.h"
#include "hook/patch/hasp.h"
#include "hook/patch/hdd-check.h"
#include "hook/patch/hook-mon.h"
#include "hook/patch/main-loop.h"
#include "hook/patch/mounts.h"
#include "hook/patch/network.h"
#include "hook/patch/piuio.h"
#include "hook/patch/piuio-exit.h"
#include "hook/patch/redir.h"
#include "hook/patch/ram-wipe.h"
#include "hook/patch/sigsegv.h"
#include "hook/patch/sound.h"
#include "hook/patch/usb-emu.h"
#include "hook/patch/usb-init-fix.h"
#include "hook/patch/usb-mnt.h"
#include "hook/patch/usbfs.h"
#include "util/fs.h"
#include "util/glibc.h"
#include "util/net.h"
#include "util/sys-info.h"
#include "options.h"
#include "usb-updates.h"
/* Compiled binary data from data folder. Symbols are defined by compiler */
extern const uint8_t _binary_prihook_hdd_raw_start[];
extern const uint8_t _binary_prihook_hdd_raw_end[];
extern const uint8_t _binary_prihook_hasp_key_start[];
extern const uint8_t _binary_prihook_hasp_key_end[];
static void prihook_bootstrapping(int argc, char** argv, struct prihook_options* options, char** game_data_path)
{
log_assert(argc >= 0);
log_assert(argv);
log_assert(options);
log_assert(game_data_path);
if (hook_core_piu_block_non_piu_process_recursion(argc, argv)) {
return;
}
hook_core_piu_log_init(argc, argv);
if (!prihook_options_init(argc, argv, options)) {
exit(0);
}
hook_core_piu_log_info(argc, argv);
*game_data_path = hook_core_piu_utils_get_and_verify_game_data_path(argc, argv);
hook_core_piu_utils_verify_root_user();
hook_core_piu_utils_fix_locale();
log_info("Initializing, piu game data path: %s", *game_data_path);
}
static void prihook_patch_fs_init(struct prihook_options* options)
{
log_assert(options);
patch_hook_mon_init(
options->patch.hook_mon.io,
options->patch.hook_mon.file,
options->patch.hook_mon.fs,
options->patch.hook_mon.usb,
options->patch.hook_mon.open);
patch_redir_init();
}
static void prihook_fs_redir_settings(struct prihook_options* options)
{
log_assert(options);
if (options->game.settings) {
if (!util_fs_path_exists(options->game.settings)) {
log_info("Settings path %s does not exist, creating", options->game.settings);
if (!util_fs_mkdir(options->game.settings)) {
log_error("Creating directory %s failed", options->game.settings);
}
}
char* abs_path = util_fs_get_abs_path(options->game.settings);
cnh_redir_add("/SETTINGS", abs_path);
free(abs_path);
} else {
log_warn("No settings path specified");
}
}
static void prihook_fs_redir_usb()
{
/* Needed on newer kernels */
cnh_redir_add("/proc/bus/usb/devices", "/sys/kernel/debug/usb/devices");
patch_usbfs_init();
}
static void prihook_fs_redirs_init(struct prihook_options* options, const char* game_data_path)
{
log_assert(options);
log_assert(game_data_path);
prihook_fs_redir_settings(options);
prihook_fs_redir_usb();
}
static void prihook_patch_fs_mounting_init()
{
patch_mounts_init();
patch_usb_mnt_init();
}
static void prihook_patch_gfx_init(struct prihook_options* options)
{
log_assert(options);
patch_gfx_init();
if (options->patch.gfx.windowed) {
patch_gfx_force_window_mode();
}
}
static void prihook_patch_main_loop_init(struct prihook_options* options)
{
log_assert(options);
patch_main_loop_init(false, options->patch.main_loop.disable_built_in_inputs, false);
if (options->patch.main_loop.x11_input_handler_api_lib) {
char* abs_path_x11_input_hook_lib = util_fs_get_abs_path(options->patch.main_loop.x11_input_handler_api_lib);
patch_main_loop_add_x11_input_handler(abs_path_x11_input_hook_lib);
free(abs_path_x11_input_hook_lib);
}
patch_block_keyboard_grab_init();
}
static void prihook_patch_sound_init(struct prihook_options* options)
{
log_assert(options);
patch_sound_init(options->patch.sound.device);
patch_axmier_block_init();
}
static void prihook_patch_sigsegv_init(struct prihook_options* options)
{
log_assert(options);
patch_sigsegv_init(options->patch.sigsegv.halt_on_segv);
}
static void prihook_patch_dongle_init()
{
patch_hasp_init((const uint8_t*) _binary_prihook_hasp_key_start,
((uintptr_t) &_binary_prihook_hasp_key_end - (uintptr_t) &_binary_prihook_hasp_key_start));
}
static void prihook_patch_hdd_check_init()
{
patch_hdd_check_init((const uint8_t*) _binary_prihook_hdd_raw_start,
((uintptr_t) &_binary_prihook_hdd_raw_end - (uintptr_t) &_binary_prihook_hdd_raw_start));
}
static void prihook_patch_ram_wipe_init()
{
patch_ram_wipe_init();
}
static void prihook_patch_piuio_init(struct prihook_options* options)
{
log_assert(options);
/* Hook before IO emulation */
if (options->patch.piuio.exit_test_serv) {
patch_piuio_exit_init();
}
patch_usb_init_fix_init();
patch_usb_emu_init();
if (options->patch.piuio.api_lib) {
char* abs_path_iolib = util_fs_get_abs_path(options->patch.piuio.api_lib);
patch_piuio_init(abs_path_iolib);
free(abs_path_iolib);
}
}
static void prihook_patch_net_init(struct prihook_options* options)
{
patch_blacklist_url_init();
/* update server */
patch_blacklist_url_add("http://update_kr");
/* game server */
patch_blacklist_url_add("http://www2.piugame.com");
patch_network_init();
if (options->patch.net.server) {
char* address;
uint16_t port;
if (!util_net_split_ipv4_str(options->patch.net.server, &address, &port)) {
log_warn("Invalid format for option network address specified, ignored");
} else {
patch_network_redirect_server_address("115.68.108.183", address, (uint16_t) port);
free(address);
}
} else {
// Redirect to avoid phoning home if no server address is set
patch_network_redirect_server_address("115.68.108.183", "127.0.0.1", 12345);
}
// Idk what this address is...redirect to something save for now
patch_network_redirect_server_address("45.76.222.114", "127.0.0.1", 12345);
}
static void prihook_patch_usb_init()
{
prihook_usb_updates_init();
}
void prihook_constructor(void)
{
/* Nothing here */
}
void prihook_destructor(void)
{
/* Nothing here */
}
void prihook_trap_before_main(int argc, char** argv)
{
log_assert(argc >= 0);
log_assert(argv);
char* game_data_path;
struct prihook_options options;
prihook_bootstrapping(argc, argv, &options, &game_data_path);
prihook_patch_fs_init(&options);
prihook_fs_redirs_init(&options, game_data_path);
prihook_patch_fs_mounting_init();
prihook_patch_gfx_init(&options);
prihook_patch_main_loop_init(&options);
prihook_patch_sound_init(&options);
prihook_patch_sigsegv_init(&options);
prihook_patch_dongle_init();
prihook_patch_hdd_check_init();
prihook_patch_ram_wipe_init();
prihook_patch_piuio_init(&options);
prihook_patch_net_init(&options);
prihook_patch_usb_init();
free(game_data_path);
log_info("Hooking finished");
}
void prihook_trap_after_main(void)
{
patch_piuio_shutdown();
}
LIB_MAIN_CONSTRUCTOR(prihook_constructor);
LIB_MAIN_DESTRUCTOR(prihook_destructor);
LIB_MAIN_TRAP_MAIN(prihook_trap_before_main, prihook_trap_after_main);
+183
View File
@@ -0,0 +1,183 @@
#include "hook/pri/options.h"
#include "util/options.h"
#define PRIHOOK_OPTIONS_STR_GAME_SETTINGS "game.settings"
#define PRIHOOK_OPTIONS_STR_PATCH_GFX_WINDOWED "patch.gfx.windowed"
#define PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_FILE "patch.hook_mon.file"
#define PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_FS "patch.hook_mon.fs"
#define PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_IO "patch.hook_mon.io"
#define PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_OPEN "patch.hook_mon.open"
#define PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_USB "patch.hook_mon.usb"
#define PRIHOOK_OPTIONS_STR_PATCH_HOOK_MAIN_LOOP_DISABLE_BUILT_IN_INPUTS "patch.hook_main_loop.disable_built_in_inputs"
#define PRIHOOK_OPTIONS_STR_PATCH_HOOK_MAIN_LOOP_X11_INPUT_HANDLER "patch_hook_main_loop.x11_input_handler"
#define PRIHOOK_OPTIONS_STR_PATCH_NET_SERVER "patch.net.server"
#define PRIHOOK_OPTIONS_STR_PATCH_PIUIO_EMU_LIB "patch.piuio.emu_lib"
#define PRIHOOK_OPTIONS_STR_PATCH_PIUIO_EXIT_TEST_SERV "patch.piuio_exit.test_serv"
#define PRIHOOK_OPTIONS_STR_PATCH_SOUND_DEVICE "patch.sound.device"
#define PRIHOOK_OPTIONS_STR_PATCH_SIGSEGV_HALT_ON_SEGV "patch.sigsegv.halt_on_segv"
#define PRIHOOK_OPTIONS_STR_PATCH_UTIL_LOG_FILE "util.log.file"
#define PRIHOOK_OPTIONS_STR_PATCH_UTIL_LOG_LEVEL "util.log.level"
const struct util_options_def prihook_options_def[] = {
{
.name = PRIHOOK_OPTIONS_STR_GAME_SETTINGS,
.description = "Path to game settings (SETTINGS) folder",
.param = 's',
.type = UTIL_OPTIONS_TYPE_STR,
.default_value.str = "./save",
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_GFX_WINDOWED,
.description = "Force game into window mode",
.param = 'w',
.type = UTIL_OPTIONS_TYPE_BOOL,
.default_value.b = false,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_FILE,
.description = "Enable file call monitoring",
.param = 'f',
.type = UTIL_OPTIONS_TYPE_BOOL,
.default_value.b = false,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_FS,
.description = "Enable file system related call monitoring",
.param = 'g',
.type = UTIL_OPTIONS_TYPE_BOOL,
.default_value.b = false,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_IO,
.description = "Enable IO syscall monitoring",
.param = 'i',
.type = UTIL_OPTIONS_TYPE_BOOL,
.default_value.b = false,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_OPEN,
.description = "Enable any file/IO open call monitoring",
.param = 'j',
.type = UTIL_OPTIONS_TYPE_BOOL,
.default_value.b = false,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_USB,
.description = "Enable libusb call monitoring",
.param = 'u',
.type = UTIL_OPTIONS_TYPE_BOOL,
.default_value.b = false,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_HOOK_MAIN_LOOP_DISABLE_BUILT_IN_INPUTS,
.description = "Disable the built in keyboard inputs for test (F1), service (PRI) and clear (F3)",
.param = 'k',
.type = UTIL_OPTIONS_TYPE_BOOL,
.default_value.b = false,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_HOOK_MAIN_LOOP_X11_INPUT_HANDLER,
.description = "Path to a library implementing the x11-input-handler api to capture X11 keyboard inputs",
.param = 'q',
.type = UTIL_OPTIONS_TYPE_STR,
.default_value.str = NULL,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_NET_SERVER,
.description = "Address and port of pumpnet server (e.g. 127.0.0.1:1234)",
.param = 'n',
.type = UTIL_OPTIONS_TYPE_STR,
.default_value.str = NULL,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_PIUIO_EMU_LIB,
.description = "Path to library implementing the piuio api for piuio emulation",
.param = 'p',
.type = UTIL_OPTIONS_TYPE_STR,
.default_value.str = NULL,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_PIUIO_EXIT_TEST_SERV,
.description = "Enable game exit on Test + Service",
.param = 'e',
.type = UTIL_OPTIONS_TYPE_BOOL,
.default_value.b = false,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_SOUND_DEVICE,
.description = "Select the sound device to open on snd_pcm_open",
.param = 'a',
.type = UTIL_OPTIONS_TYPE_STR,
.default_value.str = "dmix",
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_SIGSEGV_HALT_ON_SEGV,
.description = "Halt on sigsegv to attach a debugger to the process",
.param = 'd',
.type = UTIL_OPTIONS_TYPE_BOOL,
.default_value.str = false,
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_UTIL_LOG_FILE,
.description = "Print the log output to the specified file",
.param = 'o',
.type = UTIL_OPTIONS_TYPE_STR,
.default_value.str = "pumptools.log",
},
{
.name = PRIHOOK_OPTIONS_STR_PATCH_UTIL_LOG_LEVEL,
.description = "Set the log level (0-4)",
.param = 'l',
.type = UTIL_OPTIONS_TYPE_INT,
.default_value.i = LOG_LEVEL_DEBUG,
},
};
const struct util_options_defs prihook_options_defs = {
.usage_header =
"Pumptools prihook for Pump It Up: Prime, build " __DATE__ " " __TIME__ ", gitrev " STRINGIFY(GITREV) "\n"
"Usage: LD_PRELOAD=./prihook.so [game exec] [game data path] <options>",
.usage_param = 'h',
.defs = prihook_options_def,
.ndefs = lengthof(prihook_options_def)
};
bool prihook_options_init(int argc, char** argv, struct prihook_options* options)
{
log_assert(argv);
log_assert(options);
struct util_options_opts* options_opt;
util_options_init(argc, argv);
options_opt = util_options_get(&prihook_options_defs);
if (!options_opt) {
return false;
}
options->game.settings = util_options_get_str(options_opt, PRIHOOK_OPTIONS_STR_GAME_SETTINGS);
options->patch.gfx.windowed = util_options_get_bool(options_opt, PRIHOOK_OPTIONS_STR_PATCH_GFX_WINDOWED);
options->patch.hook_mon.file = util_options_get_bool(options_opt, PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_FILE);
options->patch.hook_mon.fs = util_options_get_bool(options_opt, PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_FS);
options->patch.hook_mon.io = util_options_get_bool(options_opt, PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_IO);
options->patch.hook_mon.open = util_options_get_bool(options_opt, PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_OPEN);
options->patch.hook_mon.usb = util_options_get_bool(options_opt, PRIHOOK_OPTIONS_STR_PATCH_HOOK_MON_USB);
options->patch.main_loop.disable_built_in_inputs = util_options_get_bool(options_opt,
PRIHOOK_OPTIONS_STR_PATCH_HOOK_MAIN_LOOP_DISABLE_BUILT_IN_INPUTS);
options->patch.main_loop.x11_input_handler_api_lib =
util_options_get_str(options_opt, PRIHOOK_OPTIONS_STR_PATCH_HOOK_MAIN_LOOP_X11_INPUT_HANDLER);
options->patch.net.server = util_options_get_str(options_opt, PRIHOOK_OPTIONS_STR_PATCH_NET_SERVER);
options->patch.piuio.api_lib = util_options_get_str(options_opt, PRIHOOK_OPTIONS_STR_PATCH_PIUIO_EMU_LIB);
options->patch.piuio.exit_test_serv =
util_options_get_bool(options_opt, PRIHOOK_OPTIONS_STR_PATCH_PIUIO_EXIT_TEST_SERV);
options->patch.sound.device = util_options_get_str(options_opt, PRIHOOK_OPTIONS_STR_PATCH_SOUND_DEVICE);
options->patch.sigsegv.halt_on_segv =
util_options_get_bool(options_opt, PRIHOOK_OPTIONS_STR_PATCH_SIGSEGV_HALT_ON_SEGV);
options->log.file = util_options_get_str(options_opt, PRIHOOK_OPTIONS_STR_PATCH_UTIL_LOG_FILE);
options->log.level =
(enum util_log_level) util_options_get_int(options_opt, PRIHOOK_OPTIONS_STR_PATCH_UTIL_LOG_LEVEL);
return true;
}
+58
View File
@@ -0,0 +1,58 @@
#ifndef PRIHOOK_OPTIONS_H
#define PRIHOOK_OPTIONS_H
#include <stdbool.h>
#include <stdint.h>
#include "util/log.h"
struct prihook_options {
struct game {
const char* settings;
} game;
struct patch {
struct gfx {
bool windowed;
} gfx;
struct hook_mon {
bool file;
bool fs;
bool io;
bool open;
bool usb;
} hook_mon;
struct main_loop {
bool disable_built_in_inputs;
const char* x11_input_handler_api_lib;
} main_loop;
struct net {
const char* server;
} net;
struct piuio {
const char* api_lib;
bool exit_test_serv;
} piuio;
struct sound {
const char* device;
} sound;
struct sigsegv {
bool halt_on_segv;
} sigsegv;
} patch;
struct log {
const char* file;
enum util_log_level level;
} log;
};
bool prihook_options_init(int argc, char** argv, struct prihook_options* options);
#endif
+1
View File
@@ -0,0 +1 @@
DUMMY NOT RELEASED YET
+1
View File
@@ -0,0 +1 @@
DUMMY NOT RELEASED YET
+27
View File
@@ -0,0 +1,27 @@
#define LOG_MODULE "prihook-usb-updates"
#include <string.h>
#include "capnhook/hook/filehook.h"
#include "util/log.h"
static enum cnh_result prihook_usb_updates_filehook(struct cnh_filehook_irp *irp)
{
if (irp->op != CNH_FILEHOOK_IRP_OP_OPEN) {
return cnh_filehook_invoke_next(irp);
}
if (!strcmp(irp->open_filename, "/mnt/0/prime_") || !strcmp(irp->open_filename, "/mnt/1/prime_")) {
log_info("Blocking usb update %s", irp->open_filename);
return CNH_RESULT_NO_SUCH_FILE_OR_DIR;
}
return cnh_filehook_invoke_next(irp);
}
void prihook_usb_updates_init(void)
{
cnh_filehook_push_handler(prihook_usb_updates_filehook);
}
+12
View File
@@ -0,0 +1,12 @@
/**
* Block updates from usb flash drives on Prime
*/
#ifndef PRIHOOK_USB_UPDATES_H
#define PATCH_USB_UPDATES_H
/**
* Initialize patch module
*/
void prihook_usb_updates_init(void);
#endif