djhackersdev_bemanitools/src/main/jbhook2/dllmain.c

#include <windows.h>

#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>

#include "bemanitools/eamio.h"
#include "bemanitools/jbio.h"

#include "hook/iohook.h"
#include "hook/table.h"

#include "hooklib/adapter.h"
#include "hooklib/app.h"
#include "hooklib/rs232.h"
#include "hooklib/setupapi.h"

#include "imports/avs.h"

#include "jbhook2/options.h"

#include "jbhook-util/acio.h"
#include "jbhook-util/eamuse.h"

#include "jbhook-util-p3io/gfx.h"
#include "jbhook-util-p3io/p3io.h"

#include "p3ioemu/devmgr.h"
#include "p3ioemu/emu.h"

#include "p4ioemu/device.h"
#include "p4ioemu/setupapi.h"

#include "security/id.h"
#include "security/mcode.h"

#include "util/defs.h"
#include "util/log.h"
#include "util/thread.h"

static struct options options;

static bool my_dll_entry_init(char *sidcode, struct property_node *param)
{
    bool eam_io_ok;
    bool jb_io_ok;

    eam_io_ok = false;
    jb_io_ok = false;

    log_info("--- Begin jbhook dll_entry_init ---");

    log_assert(sidcode != NULL);

    if(options.vertical) {
        jbhook_util_gfx_install_vertical_hooks();
    }

    if (!options.disable_p3ioemu) {
        iohook_push_handler(p3io_emu_dispatch_irp);
        p3io_setupapi_insert_hooks(NULL);

        // pcbid and eamid are only used here for sec check, the ones used for
        // network access are taken from ea3-config.xml.
        // When booting knit append (and on no other version), the code must
        // actually match, so we use the one from ea3-config as well.
        // Example sidcode format: `K44JBA2012072301`
        struct security_mcode security_mcode_j44 = {
            .header = SECURITY_MCODE_HEADER,
            .unkn = SECURITY_MCODE_UNKN_C,
            .game = {sidcode[0], sidcode[1], sidcode[2]},
            .region = sidcode[3],
            .cabinet = sidcode[4],
            .revision = sidcode[5],
        };

        jbhook_util_p3io_init(
            &security_mcode_j44,
            &security_id_default, &security_id_default);

        log_info("Starting up jubeat IO backend");

        jb_io_set_loggers(
            log_impl_misc, log_impl_info, log_impl_warning, log_impl_fatal);

        jb_io_ok =
            jb_io_init(avs_thread_create, avs_thread_join, avs_thread_destroy);

        if (!jb_io_ok) {
            goto fail;
        }
    }

    if (!options.disable_cardemu) {
        iohook_push_handler(jbhook_util_ac_io_port_dispatch_irp);
        rs232_hook_init();
        jbhook_util_ac_io_port_init(L"COM1");
        jbhook_util_ac_io_set_iccb();

        log_info("Starting up card reader backend");

        eam_io_set_loggers(
            log_impl_misc, log_impl_info, log_impl_warning, log_impl_fatal);

        eam_io_ok =
            eam_io_init(avs_thread_create, avs_thread_join, avs_thread_destroy);

        if (!eam_io_ok) {
            goto fail;
        }
    }

    log_info("---  End  jbhook dll_entry_init ---");

    char *sidcode_in_ea3_config = strdup(sidcode);

    bool ret = app_hook_invoke_init(sidcode, param);

    // If the game is append, the mcode `cabinet` is forced to C. This is bad if
    // p3io was configured to respond as A! Help the user help themsevles...
    if(strcmp(sidcode_in_ea3_config, sidcode) != 0) {
        log_warning("sidcode changed after running game DLL init (%s -> %s)", sidcode_in_ea3_config, sidcode);
        log_warning("This will trigger a security error. Modify ea3-config.xml <soft> section to match the second value!");

        // abort the boot
        return false;
    }

    free(sidcode_in_ea3_config);

    return ret;

fail:
    if (eam_io_ok) {
        eam_io_fini();
    }

    if (jb_io_ok) {
        jb_io_fini();
    }

    return false;
}

static bool my_dll_entry_main(void)
{
    bool result;

    result = app_hook_invoke_main();

    log_info("Shutting down card reader backend");
    eam_io_fini();

    log_info("Shutting down Jubeat IO backend");
    jb_io_fini();

    if (!options.disable_cardemu) {
        jbhook_util_ac_io_port_fini();
    }

    if (!options.disable_p3ioemu) {
        jbhook_util_p3io_fini();
    }

    options_fini(&options);

    return result;
}

static HWND CDECL
my_mwindow_create(HINSTANCE, void *, const char *, DWORD, DWORD, BOOL);
static HWND(CDECL *real_mwindow_create)(
    HINSTANCE, void *, const char *, DWORD, DWORD, BOOL);

static const struct hook_symbol init_hook_syms[] = {
    {
        .name = "mwindow_create",
        .patch = my_mwindow_create,
        .link = (void **) &real_mwindow_create,
    },
};

static HWND CDECL my_mwindow_create(
    HINSTANCE hinstance,
    void *callback,
    const char *window_title,
    DWORD window_width,
    DWORD window_height,
    BOOL fullscreen)
{
    log_info("-------------------------------------------------------------");
    log_info("---------------- Begin jbhook mwindow_create ----------------");
    log_info("-------------------------------------------------------------");

    if(options.vertical) {
        DWORD tmp = window_width;
        window_width = window_height;
        window_height = tmp;
    }

    fullscreen = !options.windowed;

    return real_mwindow_create(
        hinstance,
        callback,
        window_title,
        window_width,
        window_height,
        fullscreen);
}

BOOL WINAPI DllMain(HMODULE mod, DWORD reason, void *ctx)
{
    if (reason != DLL_PROCESS_ATTACH) {
        return TRUE;
    }

    log_to_external(
        log_body_misc, log_body_info, log_body_warning, log_body_fatal);

    options_init_from_cmdline(&options);

    hook_table_apply(
        NULL, "mwindow.dll", init_hook_syms, lengthof(init_hook_syms));

    app_hook_init(my_dll_entry_init, my_dll_entry_main);

    if (!options.disable_adapteremu) {
        adapter_hook_init();
    }

    jbhook_util_eamuse_hook_init();

    return TRUE;
}