djhackersdev_bemanitools/src/main/ddrhook2/dllmain.c

#include <windows.h>

#include <stdbool.h>

#include "avs-util/core-interop.h"

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

#include "core/log.h"
#include "core/thread.h"

#include "ddrhook-util/_com4.h"
#include "ddrhook-util/extio.h"
#include "ddrhook-util/gfx.h"
#include "ddrhook-util/p3io.h"
#include "ddrhook-util/spike.h"
#include "ddrhook-util/usbmem.h"

#include "ddrhook2/master.h"

#include "hook/iohook.h"

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

#include "imports/avs.h"

#include "p3ioemu/emu.h"

#include "util/cmdline.h"
#include "util/defs.h"

static bool my_dll_entry_init(char *sidcode, struct property_node *param);
static bool my_dll_entry_main(void);

bool standard_def;
bool _15khz;

static bool my_dll_entry_init(char *sidcode, struct property_node *param)
{
    int argc;
    char **argv;
    bool com4;
    bool ok;
    int i;
    char usbmem_data_path_p1[MAX_PATH] = "";
    char usbmem_data_path_p2[MAX_PATH] = "";
    bool usbmem_enabled = false;

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

    com4 = true;
    args_recover(&argc, &argv);

    for (i = 1; i < argc; i++) {
        if (argv[i][0] != '-') {
            continue;
        }

        switch (argv[i][1]) {
            case 'o':
                standard_def = true;

                break;

            case 'w':
                gfx_set_windowed();

                break;

            case 'u':
                /* Don't emulate P3IO COM4 and its downstream devices, use the
                   Windows COM4 port instead. */
                com4 = false;

                break;

            case 'a':
                /* Specify a USB memory path for P1 */
                if (i + 1 < argc) {
                    strcpy(usbmem_data_path_p1, argv[i + 1]);
                    usbmem_enabled = true;
                    i++; // Move forward one to skip the path parameter
                }

                break;

            case 'b':
                /* Specify a USB memory path for P2 */
                if (i + 1 < argc) {
                    strcpy(usbmem_data_path_p2, argv[i + 1]);
                    usbmem_enabled = true;
                    i++; // Move forward one to skip the path parameter
                }

                break;
        }
    }

    args_free(argc, argv);

    gfx_set_is_modern();

    iohook_push_handler(p3io_emu_dispatch_irp);
    iohook_push_handler(extio_dispatch_irp);
    iohook_push_handler(spike_dispatch_irp);
    iohook_push_handler(usbmem_dispatch_irp);

    if (com4) {
        /* See ddrhook2/p3io.c for details. */
        iohook_push_handler(com4_dispatch_irp);
    }

    rs232_hook_init();

    ddrhook2_master_insert_hooks(NULL);
    p3io_ddr_init();
    extio_init();
    usbmem_init(usbmem_data_path_p1, usbmem_data_path_p2, usbmem_enabled);
    spike_init();
    com4_init();

    log_info("Initializing DDR IO backend");

    core_log_impl_assign(ddr_io_set_loggers);

    ok = ddr_io_init(
        core_thread_create_impl_get(),
        core_thread_join_impl_get(),
        core_thread_destroy_impl_get());

    if (!ok) {
        return false;
    }

    if (com4) {
        log_info("Initializing card reader backend");

        core_log_impl_assign(eam_io_set_loggers);

        ok = eam_io_init(
            core_thread_create_impl_get(),
            core_thread_join_impl_get(),
            core_thread_destroy_impl_get());

        if (!ok) {
            return false;
        }
    }

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

    return app_hook_invoke_init(sidcode, param);
}

static bool my_dll_entry_main(void)
{
    bool result;

    result = app_hook_invoke_main();

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

    log_misc("Shutting down DDR IO backend");
    ddr_io_fini();

    com4_fini();
    spike_fini();
    usbmem_fini();
    extio_fini();
    p3io_emu_fini();

    return result;
}

BOOL WINAPI DllMain(HMODULE self, DWORD reason, void *ctx)
{
    if (reason != DLL_PROCESS_ATTACH) {
        goto end;
    }

    // Use AVS APIs
    avs_util_core_interop_thread_avs_impl_set();
    avs_util_core_interop_log_avs_impl_set();

    app_hook_init(my_dll_entry_init, my_dll_entry_main);

end:
    return TRUE;
}