iframe-proxy | Sunbelt Computer Software

232 lines (176 loc) · 5.61 KB
#include <process.h>
#include <windows.h>
#include <TlHelp32.h>
#include "steam_api.h"
#include <string>
const uint32 DFHACK_STEAM_APPID = 2346660;
const uint32 DF_STEAM_APPID = 975370;
static BOOL is_running_on_wine() {
    typedef const char* (CDECL wine_get_version)(void);
    static wine_get_version* pwine_get_version;
    HMODULE hntdll = GetModuleHandle("ntdll.dll");
    if(!hntdll)
        return FALSE;
    pwine_get_version = (wine_get_version*) GetProcAddress(hntdll, "wine_get_version");
    return !!pwine_get_version;
static LPCWSTR launch_via_steam_posix() {
    const char* argv[] = { "/bin/sh", "-c", "\"steam -applaunch 975370\"", NULL };
    // does not return on success
    _execv(argv[0], argv);
    return L"Could not launch Dwarf Fortress";
static LPCWSTR launch_via_steam_windows() {
    STARTUPINFOW si;
    PROCESS_INFORMATION pi;
    ZeroMemory(&si, sizeof(si));
    si.cb = sizeof(si);
    ZeroMemory(&pi, sizeof(pi));
    WCHAR steamPath[1024] = L"";
    DWORD datasize = 1024;
    LONG retCode = RegGetValueW(HKEY_CURRENT_USER, L"SOFTWARE\\Valve\\Steam",
            L"SteamExe", RRF_RT_REG_SZ, NULL, &steamPath, &datasize);
    if (retCode != ERROR_SUCCESS)
        return L"Could not find Steam client executable";
    WCHAR commandLine[1024] = L"steam.exe -applaunch 975370";
    BOOL res = CreateProcessW(steamPath, commandLine,
        NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi);
    if (res)
        WaitForSingleObject(pi.hProcess, INFINITE);
        CloseHandle(pi.hProcess);
        CloseHandle(pi.hThread);
        return NULL;
        return L"Could not launch Dwarf Fortress";
static LPCWSTR launch_direct() {
    STARTUPINFOW si;
    PROCESS_INFORMATION pi;
    ZeroMemory(&si, sizeof(si));
    si.cb = sizeof(si);
    ZeroMemory(&pi, sizeof(pi));
    BOOL res = CreateProcessW(L"Dwarf Fortress.exe",
            NULL, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi);
    if (res)
        WaitForSingleObject(pi.hProcess, INFINITE);
        CloseHandle(pi.hProcess);
        CloseHandle(pi.hThread);
        return NULL;
    return L"Could not launch via non-steam fallback method";
DWORD findDwarfFortressProcess()
    PROCESSENTRY32W entry;
    entry.dwSize = sizeof(PROCESSENTRY32W);
    const auto snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, NULL);
    if (!Process32FirstW(snapshot, &entry))
        CloseHandle(snapshot);
        return -1;
        std::wstring executableName(entry.szExeFile);
        if (executableName == L"Dwarf Fortress.exe")
            CloseHandle(snapshot);
            return entry.th32ProcessID;
    } while (Process32NextW(snapshot, &entry));
    CloseHandle(snapshot);
    return -1;
bool waitForDF() {
    DWORD df_pid = findDwarfFortressProcess();
    if (df_pid == -1)
        return false;
    HANDLE hDF = OpenProcess(PROCESS_QUERY_INFORMATION | SYNCHRONIZE, FALSE, df_pid);
    // in the future open an IPC connection so that we can proxy SteamAPI calls for the DFSteam module
    // this will eventuallyh need to become a loop with a WaitForMultipleObjects call
    WaitForSingleObject(hDF, INFINITE);
    CloseHandle(hDF);
    return true;
int WINAPI wWinMain(_In_ HINSTANCE hInstance, _In_opt_ HINSTANCE hPrevInstance, _In_ LPWSTR lpCmdLine, _In_ int nShowCmd) {
    // initialize steam context
    if (SteamAPI_RestartAppIfNecessary(DFHACK_STEAM_APPID))
        exit(0);
    if (waitForDF())
        exit(0);
    if (!SteamAPI_Init())
        // could not initialize steam context, attempt fallback launch
        LPCWSTR err = launch_direct();
        if (err != NULL)
            MessageBoxW(NULL, err, NULL, 0);
            exit(1);
        exit(0);
    bool wine = is_running_on_wine();
    if (wine)
        // attempt launch via steam client
        LPCWSTR err = launch_via_steam_posix();
        if (err != NULL)
            // steam client launch failed, attempt fallback launch
            err = launch_direct();
        if (err != NULL)
            MessageBoxW(NULL, err, NULL, 0);
            exit(1);
        exit(0);
    // steam detected and not running in wine
    bool df_installed = SteamApps()->BIsAppInstalled(DF_STEAM_APPID);
    if (!df_installed)
        // Steam DF is not installed. Assume DF is installed in same directory as DFHack and do a fallback launch
        LPCWSTR err = launch_direct();
        if (err != NULL)
            MessageBoxW(NULL, err, NULL, 0);
            exit(1);
        exit(0);
    // obtain DF app path
    char buf[2048] = "";
    int b1 = SteamApps()->GetAppInstallDir(DFHACK_STEAM_APPID, (char*)&buf, 2048);
    std::string dfhack_install_folder = (b1 != -1) ? std::string(buf) : "";
    int b2 = SteamApps()->GetAppInstallDir(DF_STEAM_APPID, (char*)&buf, 2048);
    std::string df_install_folder = (b2 != -1) ? std::string(buf) : "";
    if (df_install_folder != dfhack_install_folder)
        // DF and DFHack are not installed in the same library
        MessageBoxW(NULL, L"DFHack and Dwarf Fortress must be installed in the same Steam library.\nAborting.", NULL, 0);
        exit(1);
    if (waitForDF())
        exit(0);
    LPCWSTR err = launch_via_steam_windows();
    if (err != NULL)
        MessageBoxW(NULL, err, NULL, 0);
        exit(1);
    int counter = 0;
    while (!waitForDF()) {
        if (counter++ > 60)
            MessageBoxW(NULL, L"Dwarf Fortress took too long to launch, aborting", NULL, 0);
            exit(1);
        Sleep(1000);
    exit(0);
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

launchdf.cpp

Latest commit

History

launchdf.cpp

File metadata and controls
