#include #include #include #include #include "bemanitools/eamio.h" #include "bemanitools/iidxio.h" #include "cconfig/cconfig-hook.h" #include "ezusb-emu/desc.h" #include "ezusb-emu/device.h" #include "ezusb-iidx-emu/msg.h" #include "ezusb-iidx-emu/node-security-plug.h" #include "ezusb-iidx-emu/node-serial.h" #include "ezusb-iidx-emu/nodes.h" #include "hook/d3d9.h" #include "hook/iohook.h" #include "hook/table.h" #include "hooklib/acp.h" #include "hooklib/adapter.h" #include "hooklib/rs232.h" #include "hooklib/setupapi.h" #include "iidxhook-util/acio.h" #include "iidxhook-util/chart-patch.h" #include "iidxhook-util/clock.h" #include "iidxhook-util/config-eamuse.h" #include "iidxhook-util/config-gfx.h" #include "iidxhook-util/config-misc.h" #include "iidxhook-util/config-sec.h" #include "iidxhook-util/d3d9.h" #include "iidxhook-util/eamuse.h" #include "iidxhook-util/effector.h" #include "iidxhook-util/settings.h" #include "iidxhook2/config-iidxhook2.h" #include "util/log.h" #include "util/thread.h" #define IIDXHOOK2_INFO_HEADER \ "iidxhook for DistorteD" \ ", build " __DATE__ " " __TIME__ ", gitrev " STRINGIFY(GITREV) #define IIDXHOOK2_CMD_USAGE \ "Usage: inject.exe iidxhook2.dll [options...]" static const irp_handler_t iidxhook_handlers[] = { ezusb_emu_device_dispatch_irp, iidxhook_util_acio_dispatch_irp, iidxhook_util_chart_patch_dispatch_irp, settings_hook_dispatch_irp, }; static const hook_d3d9_irp_handler_t iidxhook_d3d9_handlers[] = { iidxhook_util_d3d9_irp_handler, }; static HANDLE STDCALL my_OpenProcess(DWORD, BOOL, DWORD); static HANDLE(STDCALL *real_OpenProcess)(DWORD, BOOL, DWORD); static bool iidxhook_init_check; static const struct hook_symbol init_hook_syms[] = { {.name = "OpenProcess", .patch = my_OpenProcess, .link = (void **) &real_OpenProcess}, }; static void iidxhook2_setup_d3d9_hooks( const struct iidxhook_config_gfx *config_gfx, const struct iidxhook_config_iidxhook2 *config_iidxhook2) { struct iidxhook_util_d3d9_config d3d9_config; log_warning( "d3d8 hook module deprecated, using d3d9 hook modules requiring " "d3d8to9 to work!"); iidxhook_util_d3d9_init_config(&d3d9_config); d3d9_config.windowed = config_gfx->windowed; d3d9_config.framed = config_gfx->framed; d3d9_config.override_window_width = config_gfx->window_width; d3d9_config.override_window_height = config_gfx->window_height; d3d9_config.framerate_limit = config_gfx->frame_rate_limit; d3d9_config.iidx13_fix_song_select_bg = config_iidxhook2->distorted_ms_bg_fix; d3d9_config.iidx11_to_17_fix_uvs_bg_videos = config_gfx->bgvideo_uv_fix; d3d9_config.scale_back_buffer_width = config_gfx->scale_back_buffer_width; d3d9_config.scale_back_buffer_height = config_gfx->scale_back_buffer_height; d3d9_config.scale_back_buffer_filter = config_gfx->scale_back_buffer_filter; d3d9_config.forced_refresh_rate = config_gfx->forced_refresh_rate; d3d9_config.device_adapter = config_gfx->device_adapter; if (config_gfx->monitor_check == 0) { log_info("Auto monitor check enabled"); d3d9_config.iidx09_to_19_monitor_check_cb = iidxhook_util_chart_patch_set_refresh_rate; iidxhook_util_chart_patch_init( IIDXHOOK_UTIL_CHART_PATCH_TIMEBASE_9_TO_13); } else if (config_gfx->monitor_check > 0) { log_info( "Manual monitor check, resulting refresh rate: %f", config_gfx->monitor_check); iidxhook_util_chart_patch_init( IIDXHOOK_UTIL_CHART_PATCH_TIMEBASE_9_TO_13); iidxhook_util_chart_patch_set_refresh_rate(config_gfx->monitor_check); } iidxhook_util_d3d9_configure(&d3d9_config); hook_d3d9_init(iidxhook_d3d9_handlers, lengthof(iidxhook_d3d9_handlers)); } /** * This seems to be a good entry point to intercept * before the game calls anything important */ HANDLE STDCALL my_OpenProcess(DWORD dwDesiredAccess, BOOL bInheritHandle, DWORD dwProcessId) { struct cconfig *config; struct iidxhook_util_config_eamuse config_eamuse; struct iidxhook_config_gfx config_gfx; struct iidxhook_config_iidxhook2 config_iidxhook2; struct iidxhook_config_misc config_misc; struct iidxhook_config_sec config_sec; if (iidxhook_init_check) { goto skip; } iidxhook_init_check = true; log_info("-------------------------------------------------------------"); log_info("--------------- Begin iidxhook my_OpenProcess ---------------"); log_info("-------------------------------------------------------------"); config = cconfig_init(); iidxhook_util_config_eamuse_init(config); iidxhook_config_gfx_init(config); iidxhook_config_iidxhook2_init(config); iidxhook_config_misc_init(config); iidxhook_config_sec_init(config); if (!cconfig_hook_config_init( config, IIDXHOOK2_INFO_HEADER "\n" IIDXHOOK2_CMD_USAGE, CCONFIG_CMD_USAGE_OUT_DBG)) { cconfig_finit(config); exit(EXIT_FAILURE); } iidxhook_util_config_eamuse_get(&config_eamuse, config); iidxhook_config_gfx_get(&config_gfx, config); iidxhook_config_iidxhook2_get(&config_iidxhook2, config); iidxhook_config_misc_get(&config_misc, config); iidxhook_config_sec_get(&config_sec, config); cconfig_finit(config); log_info(IIDXHOOK2_INFO_HEADER); log_info("Initializing iidxhook..."); /* Round plug security */ ezusb_iidx_emu_node_security_plug_set_boot_version( &config_sec.boot_version); ezusb_iidx_emu_node_security_plug_set_boot_seeds(config_sec.boot_seeds); ezusb_iidx_emu_node_security_plug_set_plug_black_mcode( &config_sec.black_plug_mcode); ezusb_iidx_emu_node_security_plug_set_pcbid(&config_eamuse.pcbid); /* eAmusement server IP */ eamuse_set_addr(&config_eamuse.server); eamuse_check_connection(); /* Patch rteffect.dll calls */ if (config_misc.rteffect_stub) { effector_hook_init(); } /* Direct3D and USER32 hooks */ iidxhook2_setup_d3d9_hooks(&config_gfx, &config_iidxhook2); /* Disable operator menu clock setting system clock time */ if (config_misc.disable_clock_set) { iidxhook_util_clock_hook_init(); } /* Start up IIDXIO.DLL */ log_info("Starting IIDX IO backend"); iidx_io_set_loggers( log_impl_misc, log_impl_info, log_impl_warning, log_impl_fatal); if (!iidx_io_init(thread_create, thread_join, thread_destroy)) { log_fatal("Initializing IIDX IO backend failed"); } /* Start up EAMIO.DLL */ log_misc("Initializing card reader backend"); eam_io_set_loggers( log_impl_misc, log_impl_info, log_impl_warning, log_impl_fatal); if (!eam_io_init(thread_create, thread_join, thread_destroy)) { log_fatal("Initializing card reader backend failed"); } /* Set up IO emulation hooks _after_ IO API setup to allow API implementations with real IO devices */ iohook_init(iidxhook_handlers, lengthof(iidxhook_handlers)); hook_setupapi_init(&ezusb_emu_desc_device.setupapi); ezusb_emu_device_hook_init(ezusb_iidx_emu_msg_init()); /* Card reader emulation, same issue with hooking as IO emulation */ rs232_hook_init(); iidxhook_util_acio_init(true); log_info("-------------------------------------------------------------"); log_info("---------------- End iidxhook my_OpenProcess ----------------"); log_info("-------------------------------------------------------------"); skip: return real_OpenProcess(dwDesiredAccess, bInheritHandle, dwProcessId); } /** * Hook library for DistorteD */ BOOL WINAPI DllMain(HMODULE mod, DWORD reason, void *ctx) { if (reason == DLL_PROCESS_ATTACH) { #ifdef DEBUG_HOOKING FILE *file = fopen("iidxhook.dllmain.log", "w+"); log_to_writer(log_writer_file, file); #else log_to_writer(log_writer_null, NULL); #endif /* Bootstrap hook for further init tasks (see above) */ hook_table_apply( NULL, "kernel32.dll", init_hook_syms, lengthof(init_hook_syms)); /* Actual hooks for game specific stuff */ acp_hook_init(); adapter_hook_init(); eamuse_hook_init(); settings_hook_init(); #ifdef DEBUG_HOOKING fflush(file); fclose(file); #endif /* Logging to file and other destinations is handled by inject */ log_to_writer(log_writer_debug, NULL); } return TRUE; }