mai2: segatools added

mai2hook/config.c

@@ -0,0 +1,41 @@
+#include <assert.h>
+#include <stddef.h>
+
+#include "board/config.h"
+
+#include "hooklib/config.h"
+#include "hooklib/dvd.h"
+
+#include "mai2hook/config.h"
+
+#include "platform/config.h"
+
+void mai2_dll_config_load(
+        struct mai2_dll_config *cfg,
+        const wchar_t *filename)
+{
+    assert(cfg != NULL);
+    assert(filename != NULL);
+
+    GetPrivateProfileStringW(
+            L"mai2io",
+            L"path",
+            L"",
+            cfg->path,
+            _countof(cfg->path),
+            filename);
+}
+
+void mai2_hook_config_load(
+        struct mai2_hook_config *cfg,
+        const wchar_t *filename)
+{
+    assert(cfg != NULL);
+    assert(filename != NULL);
+
+    platform_config_load(&cfg->platform, filename);
+    aime_config_load(&cfg->aime, filename);
+    dvd_config_load(&cfg->dvd, filename);
+    io4_config_load(&cfg->io4, filename);
+    mai2_dll_config_load(&cfg->dll, filename);
+}

mai2hook/config.h

@@ -0,0 +1,27 @@
+#pragma once
+
+#include <stddef.h>
+
+#include "board/config.h"
+
+#include "hooklib/dvd.h"
+
+#include "mai2hook/mai2-dll.h"
+
+#include "platform/config.h"
+
+struct mai2_hook_config {
+    struct platform_config platform;
+    struct aime_config aime;
+    struct dvd_config dvd;
+    struct io4_config io4;
+    struct mai2_dll_config dll;
+};
+
+void mai2_dll_config_load(
+        struct mai2_dll_config *cfg,
+        const wchar_t *filename);
+
+void mai2_hook_config_load(
+        struct mai2_hook_config *cfg,
+        const wchar_t *filename);

mai2hook/dllmain.c

@@ -0,0 +1,116 @@
+#include <windows.h>
+
+#include "board/io4.h"
+#include "board/sg-reader.h"
+#include "board/vfd.h"
+
+#include "hook/process.h"
+
+#include "hooklib/serial.h"
+#include "hooklib/spike.h"
+
+#include "mai2hook/config.h"
+#include "mai2hook/io4.h"
+#include "mai2hook/mai2-dll.h"
+#include "mai2hook/unity.h"
+
+#include "platform/platform.h"
+
+#include "util/dprintf.h"
+
+static HMODULE mai2_hook_mod;
+static process_entry_t mai2_startup;
+static struct mai2_hook_config mai2_hook_cfg;
+
+/* This hook is based on mu3hook, with leaked mai2hook i/o codes. */
+
+static DWORD CALLBACK mai2_pre_startup(void)
+{
+    HRESULT hr;
+
+    dprintf("--- Begin mai2_pre_startup ---\n");
+
+    /* Load config */
+
+    mai2_hook_config_load(&mai2_hook_cfg, L".\\segatools.ini");
+
+    /* Hook Win32 APIs */
+
+    dvd_hook_init(&mai2_hook_cfg.dvd, mai2_hook_mod);
+    serial_hook_init();
+
+    /* Initialize emulation hooks */
+
+    hr = platform_hook_init(
+            &mai2_hook_cfg.platform,
+            "SDEZ",
+            "ACA1",
+            mai2_hook_mod);
+
+    if (FAILED(hr)) {
+        goto fail;
+    }
+
+    hr = sg_reader_hook_init(&mai2_hook_cfg.aime, 1, mai2_hook_mod);
+
+    if (FAILED(hr)) {
+        goto fail;
+    }
+
+    hr = vfd_hook_init(2);
+
+    if (FAILED(hr)) {
+        goto fail;
+    }
+
+    hr = mai2_dll_init(&mai2_hook_cfg.dll, mai2_hook_mod);
+
+    if (FAILED(hr)) {
+        goto fail;
+    }
+
+    hr = mai2_io4_hook_init(&mai2_hook_cfg.io4);
+
+    if (FAILED(hr)) {
+        goto fail;
+    }
+
+    /* Initialize Unity native plugin DLL hooks
+
+       There seems to be an issue with other DLL hooks if `LoadLibraryW` is
+       hooked earlier in the `mai2hook` initialization. */
+
+    unity_hook_init();
+
+    /* Initialize debug helpers */
+
+    spike_hook_init(L".\\segatools.ini");
+
+    dprintf("---  End  mai2_pre_startup ---\n");
+
+    /* Jump to EXE start address */
+
+    return mai2_startup();
+
+fail:
+    ExitProcess(EXIT_FAILURE);
+}
+
+BOOL WINAPI DllMain(HMODULE mod, DWORD cause, void *ctx)
+{
+    HRESULT hr;
+
+    if (cause != DLL_PROCESS_ATTACH) {
+        return TRUE;
+    }
+
+    mai2_hook_mod = mod;
+
+    hr = process_hijack_startup(mai2_pre_startup, &mai2_startup);
+
+    if (!SUCCEEDED(hr)) {
+        dprintf("Failed to hijack process startup: %x\n", (int) hr);
+    }
+
+    return SUCCEEDED(hr);
+}

mai2hook/io4.c

@@ -0,0 +1,153 @@
+#include <windows.h>
+
+#include <assert.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "board/io4.h"
+
+#include "mai2hook/mai2-dll.h"
+
+#include "util/dprintf.h"
+
+static HRESULT mai2_io4_poll(void *ctx, struct io4_state *state);
+static uint16_t coins;
+
+static const struct io4_ops mai2_io4_ops = {
+    .poll = mai2_io4_poll,
+};
+
+HRESULT mai2_io4_hook_init(const struct io4_config *cfg)
+{
+    HRESULT hr;
+
+    assert(mai2_dll.init != NULL);
+
+    hr = io4_hook_init(cfg, &mai2_io4_ops, NULL);
+
+    if (FAILED(hr)) {
+        return hr;
+    }
+
+    return mai2_dll.init();
+}
+
+static HRESULT mai2_io4_poll(void *ctx, struct io4_state *state)
+{
+    uint8_t opbtn;
+    uint16_t player1;
+    uint16_t player2;
+    HRESULT hr;
+
+    assert(mai2_dll.poll != NULL);
+    assert(mai2_dll.get_opbtns != NULL);
+    assert(mai2_dll.get_gamebtns != NULL);
+
+    memset(state, 0, sizeof(*state));
+
+    hr = mai2_dll.poll();
+
+    if (FAILED(hr)) {
+        return hr;
+    }
+
+    opbtn = 0;
+    player1 = 0;
+    player2 = 0;
+
+    mai2_dll.get_opbtns(&opbtn);
+    mai2_dll.get_gamebtns(&player1, &player2);
+
+    if (opbtn & MAI2_IO_OPBTN_TEST) {
+        state->buttons[0] |= IO4_BUTTON_TEST;
+    }
+
+    if (opbtn & MAI2_IO_OPBTN_SERVICE) {
+        state->buttons[0] |= IO4_BUTTON_SERVICE;
+    }
+
+    if (opbtn & MAI2_IO_OPBTN_COIN) {
+        coins++;
+    }
+    state->chutes[0] = coins << 8;
+
+    // Buttons around screen are active-low, select button is active-high
+
+    // Player 1
+
+    if (!(player1 & MAI2_IO_GAMEBTN_1)) {
+        state->buttons[0] |= 1 << 2;
+    }
+
+    if (!(player1 & MAI2_IO_GAMEBTN_2)) {
+        state->buttons[0] |= 1 << 3;
+    }
+
+    if (!(player1 & MAI2_IO_GAMEBTN_3)) {
+        state->buttons[0] |= 1 << 0;
+    }
+
+    if (!(player1 & MAI2_IO_GAMEBTN_4)) {
+        state->buttons[0] |= 1 << 15;
+    }
+
+    if (!(player1 & MAI2_IO_GAMEBTN_5)) {
+        state->buttons[0] |= 1 << 14;
+    }
+
+    if (!(player1 & MAI2_IO_GAMEBTN_6)) {
+        state->buttons[0] |= 1 << 13;
+    }
+
+    if (!(player1 & MAI2_IO_GAMEBTN_7)) {
+        state->buttons[0] |= 1 << 12;
+    }
+
+    if (!(player1 & MAI2_IO_GAMEBTN_8)) {
+        state->buttons[0] |= 1 << 11;
+    }
+
+    if (player1 & MAI2_IO_GAMEBTN_SELECT) {
+        state->buttons[0] |= 1 << 1;
+    }
+
+    // Player 2
+
+    if (!(player2 & MAI2_IO_GAMEBTN_1)) {
+        state->buttons[1] |= 1 << 2;
+    }
+
+    if (!(player2 & MAI2_IO_GAMEBTN_2)) {
+        state->buttons[1] |= 1 << 3;
+    }
+
+    if (!(player2 & MAI2_IO_GAMEBTN_3)) {
+        state->buttons[1] |= 1 << 0;
+    }
+
+    if (!(player2 & MAI2_IO_GAMEBTN_4)) {
+        state->buttons[1] |= 1 << 15;
+    }
+
+    if (!(player2 & MAI2_IO_GAMEBTN_5)) {
+        state->buttons[1] |= 1 << 14;
+    }
+
+    if (!(player2 & MAI2_IO_GAMEBTN_6)) {
+        state->buttons[1] |= 1 << 13;
+    }
+
+    if (!(player2 & MAI2_IO_GAMEBTN_7)) {
+        state->buttons[1] |= 1 << 12;
+    }
+
+    if (!(player2 & MAI2_IO_GAMEBTN_8)) {
+        state->buttons[1] |= 1 << 11;
+    }
+
+    if (player2 & MAI2_IO_GAMEBTN_SELECT) {
+        state->buttons[1] |= 1 << 4;
+    }
+
+    return S_OK;
+}

mai2hook/io4.h

@@ -0,0 +1,7 @@
+#pragma once
+
+#include <windows.h>
+
+#include "board/io4.h"
+
+HRESULT mai2_io4_hook_init(const struct io4_config *cfg);

mai2hook/mai2-dll.c

@@ -0,0 +1,109 @@
+#include <windows.h>
+
+#include <assert.h>
+#include <stdlib.h>
+
+#include "mai2hook/mai2-dll.h"
+
+#include "util/dll-bind.h"
+#include "util/dprintf.h"
+
+const struct dll_bind_sym mai2_dll_syms[] = {
+    {
+        .sym = "mai2_io_init",
+        .off = offsetof(struct mai2_dll, init),
+    }, {
+        .sym = "mai2_io_poll",
+        .off = offsetof(struct mai2_dll, poll),
+    }, {
+        .sym = "mai2_io_get_opbtns",
+        .off = offsetof(struct mai2_dll, get_opbtns),
+    }, {
+        .sym = "mai2_io_get_gamebtns",
+        .off = offsetof(struct mai2_dll, get_gamebtns),
+    }
+};
+
+struct mai2_dll mai2_dll;
+
+// Copypasta DLL binding and diagnostic message boilerplate.
+// Not much of this lends itself to being easily factored out. Also there
+// will be a lot of API-specific branching code here eventually as new API
+// versions get defined, so even though these functions all look the same
+// now this won't remain the case forever.
+
+HRESULT mai2_dll_init(const struct mai2_dll_config *cfg, HINSTANCE self)
+{
+    uint16_t (*get_api_version)(void);
+    const struct dll_bind_sym *sym;
+    HINSTANCE owned;
+    HINSTANCE src;
+    HRESULT hr;
+
+    assert(cfg != NULL);
+    assert(self != NULL);
+
+    if (cfg->path[0] != L'\0') {
+        owned = LoadLibraryW(cfg->path);
+
+        if (owned == NULL) {
+            hr = HRESULT_FROM_WIN32(GetLastError());
+            dprintf("Maimai DX IO: Failed to load IO DLL: %lx: %S\n",
+                    hr,
+                    cfg->path);
+
+            goto end;
+        }
+
+        dprintf("Maimai DX IO: Using custom IO DLL: %S\n", cfg->path);
+        src = owned;
+    } else {
+        owned = NULL;
+        src = self;
+    }
+
+    get_api_version = (void *) GetProcAddress(src, "mai2_io_get_api_version");
+
+    if (get_api_version != NULL) {
+        mai2_dll.api_version = get_api_version();
+    } else {
+        mai2_dll.api_version = 0x0100;
+        dprintf("Custom IO DLL does not expose mai2_io_get_api_version, "
+                "assuming API version 1.0.\n"
+                "Please ask the developer to update their DLL.\n");
+    }
+
+    if (mai2_dll.api_version >= 0x0200) {
+        hr = E_NOTIMPL;
+        dprintf("Maimai DX IO: Custom IO DLL implements an unsupported "
+                "API version (%#04x). Please update Segatools.\n",
+                mai2_dll.api_version);
+
+        goto end;
+    }
+
+    sym = mai2_dll_syms;
+    hr = dll_bind(&mai2_dll, src, &sym, _countof(mai2_dll_syms));
+
+    if (FAILED(hr)) {
+        if (src != self) {
+            dprintf("Maimai DX IO: Custom IO DLL does not provide function "
+                    "\"%s\". Please contact your IO DLL's developer for "
+                    "further assistance.\n",
+                    sym->sym);
+
+            goto end;
+        } else {
+            dprintf("Internal error: could not reflect \"%s\"\n", sym->sym);
+        }
+    }
+
+    owned = NULL;
+
+end:
+    if (owned != NULL) {
+        FreeLibrary(owned);
+    }
+
+    return hr;
+}

mai2hook/mai2-dll.h

@@ -0,0 +1,21 @@
+#pragma once
+
+#include <windows.h>
+
+#include "mai2io/mai2io.h"
+
+struct mai2_dll {
+    uint16_t api_version;
+    HRESULT (*init)(void);
+    HRESULT (*poll)(void);
+    void (*get_opbtns)(uint8_t *opbtn);
+    void (*get_gamebtns)(uint16_t *player1, uint16_t *player2);
+};
+
+struct mai2_dll_config {
+    wchar_t path[MAX_PATH];
+};
+
+extern struct mai2_dll mai2_dll;
+
+HRESULT mai2_dll_init(const struct mai2_dll_config *cfg, HINSTANCE self);

mai2hook/mai2hook.def

@@ -0,0 +1,18 @@
+LIBRARY mai2hook
+
+EXPORTS
+    aime_io_get_api_version
+    aime_io_init
+    aime_io_led_set_color
+    aime_io_nfc_get_aime_id
+    aime_io_nfc_get_felica_id
+    aime_io_nfc_poll
+    amDllVideoClose @2
+    amDllVideoGetVBiosVersion @4
+    amDllVideoOpen @1
+    amDllVideoSetResolution @3
+    mai2_io_get_api_version
+    mai2_io_get_gamebtns
+    mai2_io_get_opbtns
+    mai2_io_init
+    mai2_io_poll

mai2hook/meson.build

@@ -0,0 +1,31 @@
+shared_library(
+    'mai2hook',
+    name_prefix : '',
+    include_directories : inc,
+    implicit_include_directories : false,
+    vs_module_defs : 'mai2hook.def',
+    c_pch : '../precompiled.h',
+    dependencies : [
+        capnhook.get_variable('hook_dep'),
+        capnhook.get_variable('hooklib_dep'),
+    ],
+    link_with : [
+        aimeio_lib,
+        board_lib,
+        hooklib_lib,
+        mai2io_lib,
+        platform_lib,
+        util_lib,
+    ],
+    sources : [
+        'config.c',
+        'config.h',
+        'dllmain.c',
+        'io4.c',
+        'io4.h',
+        'mai2-dll.c',
+        'mai2-dll.h',
+        'unity.h',
+        'unity.c',
+    ],
+)

mai2hook/unity.c

@@ -0,0 +1,95 @@
+#include <stdbool.h>
+
+#include <windows.h>
+
+#include "hook/table.h"
+
+#include "hooklib/dll.h"
+#include "hooklib/path.h"
+
+#include "util/dprintf.h"
+
+static void dll_hook_insert_hooks(HMODULE target);
+
+static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name);
+static HMODULE (WINAPI *next_LoadLibraryW)(const wchar_t *name);
+
+static const struct hook_symbol unity_kernel32_syms[] = {
+    {
+        .name = "LoadLibraryW",
+        .patch = my_LoadLibraryW,
+        .link = (void **) &next_LoadLibraryW,
+    },
+};
+
+static const wchar_t *target_modules[] = {
+    L"mono-2.0-bdwgc.dll",
+    L"cri_ware_unity.dll",
+};
+static const size_t target_modules_len = _countof(target_modules);
+
+void unity_hook_init(void)
+{
+    dll_hook_insert_hooks(NULL);
+}
+
+static void dll_hook_insert_hooks(HMODULE target)
+{
+    hook_table_apply(
+            target,
+            "kernel32.dll",
+            unity_kernel32_syms,
+            _countof(unity_kernel32_syms));
+}
+
+static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name)
+{
+    const wchar_t *name_end;
+    const wchar_t *target_module;
+    bool already_loaded;
+    HMODULE result;
+    size_t name_len;
+    size_t target_module_len;
+
+    if (name == NULL) {
+        SetLastError(ERROR_INVALID_PARAMETER);
+
+        return NULL;
+    }
+
+    // Check if the module is already loaded
+    already_loaded = GetModuleHandleW(name) != NULL;
+
+    // Must call the next handler so the DLL reference count is incremented
+    result = next_LoadLibraryW(name);
+
+    if (!already_loaded && result != NULL) {
+        name_len = wcslen(name);
+
+        for (size_t i = 0; i < target_modules_len; i++) {
+            target_module = target_modules[i];
+            target_module_len = wcslen(target_module);
+
+            // Check if the newly loaded library is at least the length of
+            // the name of the target module
+            if (name_len < target_module_len) {
+                continue;
+            }
+
+            name_end = &name[name_len - target_module_len];
+
+            // Check if the name of the newly loaded library is one of the
+            // modules the path hooks should be injected into
+            if (_wcsicmp(name_end, target_module) != 0) {
+                continue;
+            }
+
+            dprintf("Unity: Loaded %S\n", target_module);
+
+            dll_hook_insert_hooks(result);
+            path_hook_insert_hooks(result);
+        }
+    }
+
+    return result;
+}

mai2hook/unity.h

@@ -0,0 +1,3 @@
+#pragma once
+
+void unity_hook_init(void);