Dalamud/lib/CoreCLR/boot.cpp

#define WIN32_LEAN_AND_MEAN

#include <cstdio>
#include <filesystem>
#include <Windows.h>
#include <Shlobj.h>
#include "CoreCLR.h"
#include "..\..\Dalamud.Boot\logging.h"

FILE* g_CmdStream;
void ConsoleSetup(const std::wstring console_name)
{
    if (!AllocConsole())
        return;

    SetConsoleTitleW(console_name.c_str());
    freopen_s(&g_CmdStream, "CONOUT$", "w", stdout);
    freopen_s(&g_CmdStream, "CONOUT$", "w", stderr);
    freopen_s(&g_CmdStream, "CONIN$", "r", stdin);
    SetConsoleOutputCP(CP_UTF8);
}

void ConsoleTeardown()
{
    FreeConsole();
}

std::optional<CoreCLR> g_clr;

int InitializeClrAndGetEntryPoint(
    void* calling_module,
    std::wstring runtimeconfig_path,
    std::wstring module_path,
    std::wstring entrypoint_assembly_name,
    std::wstring entrypoint_method_name,
    std::wstring entrypoint_delegate_type_name,
    void** entrypoint_fn)
{
    g_clr.emplace(calling_module);

    int result;
    SetEnvironmentVariable(L"DOTNET_MULTILEVEL_LOOKUP", L"0");
    //SetEnvironmentVariable(L"COMPlus_legacyCorruptedStateExceptionsPolicy", L"1");
    SetEnvironmentVariable(L"DOTNET_legacyCorruptedStateExceptionsPolicy", L"1");
    SetEnvironmentVariable(L"COMPLUS_ForceENC", L"1");

    wchar_t* dotnet_path;
    wchar_t* _appdata;

    std::wstring buffer;
    buffer.resize(0);
    result = GetEnvironmentVariableW(L"DALAMUD_RUNTIME", &buffer[0], 0);

    if (result)
    {
        buffer.resize(result); // The first pass returns the required length
        result = GetEnvironmentVariableW(L"DALAMUD_RUNTIME", &buffer[0], result);
        dotnet_path = _wcsdup(buffer.c_str());
    }
    else
    {
        result = SHGetKnownFolderPath(FOLDERID_RoamingAppData, KF_FLAG_DEFAULT, nullptr, &_appdata);

        if (result != 0)
        {
            logging::print<logging::E>("Unable to get RoamingAppData path (err={})", result);
            return result;
        }

        std::filesystem::path fs_app_data(_appdata);
        dotnet_path = _wcsdup(fs_app_data.append("XIVLauncher").append("runtime").c_str());
    }

    // =========================================================================== //

    logging::print<logging::I>(L"with dotnet_path: {}", dotnet_path);
    logging::print<logging::I>(L"with config_path: {}", runtimeconfig_path.c_str());
    logging::print<logging::I>(L"with module_path: {}", module_path.c_str());

    if (!std::filesystem::exists(dotnet_path))
    {
        logging::print<logging::E>("Error: Unable to find .NET runtime path");
        return 1;
    }

    get_hostfxr_parameters init_parameters
    {
        sizeof(get_hostfxr_parameters),
        nullptr,
        dotnet_path,
    };

    logging::print<logging::I>("Loading hostfxr...");
    if ((result = g_clr->load_hostfxr(&init_parameters)) != 0)
    {
        logging::print<logging::E>("Failed to load the `hostfxr` library (err=0x{:08x})", result);
        return result;
    }
    logging::print<logging::I>("Done!");

    // =========================================================================== //

    hostfxr_initialize_parameters runtime_parameters
    {
        sizeof(hostfxr_initialize_parameters),
        module_path.c_str(),
        dotnet_path,
    };

    logging::print<logging::I>("Loading coreclr... ");
    if ((result = g_clr->load_runtime(runtimeconfig_path, &runtime_parameters)) != 0)
    {
        logging::print<logging::E>("Failed to load coreclr (err={})", result);
        return result;
    }
    logging::print<logging::I>("Done!");

    // =========================================================================== //

    logging::print<logging::I>("Loading module...");
    if ((result = g_clr->load_assembly_and_get_function_pointer(
        module_path.c_str(),
        entrypoint_assembly_name.c_str(),
        entrypoint_method_name.c_str(),
        entrypoint_delegate_type_name.c_str(),
        nullptr, entrypoint_fn)) != 0)
    {
        logging::print<logging::E>("Failed to load module (err={})", result);
        return result;
    }
    logging::print<logging::I>("Done!");

    // =========================================================================== //

    return 0;
}