From 03550f8d2a7d80486fd6dd1d090fbe43980508de Mon Sep 17 00:00:00 2001
From: Zhiyi Zhang <zzhang@codeweavers.com>
Date: Fri, 17 Mar 2023 11:23:50 +0800
Subject: [PATCH] steam_helper: Support forwarding command line options to
 native steam.

CW-Bug-Id: #22030
---
 Makefile.in            |  2 +-
 steam_helper/steam.cpp | 94 +++++++++++++++++++++++++++---------------
 2 files changed, 62 insertions(+), 34 deletions(-)

diff --git a/Makefile.in b/Makefile.in
index 914be53c..15fb8276 100644
--- a/Makefile.in
+++ b/Makefile.in
@@ -354,7 +354,7 @@ all-dist: $(DIST_WINEOPENXR_JSON64)
 
 STEAMEXE_CFLAGS = -Wno-attributes
 STEAMEXE_CXXFLAGS = -Wno-attributes
-STEAMEXE_LDFLAGS = -L$(STEAMEXE_SRC)/32/ -L$(STEAMEXE_SRC)/64/ -lsteam_api -lmsi -lole32 -ldl -static-libgcc -static-libstdc++
+STEAMEXE_LDFLAGS = -L$(STEAMEXE_SRC)/32/ -L$(STEAMEXE_SRC)/64/ -lsteam_api -lshlwapi -lmsi -lole32 -ldl -static-libgcc -static-libstdc++
 
 STEAMEXE_WINEMAKER_ARGS = \
 	"-I$(SRC)/lsteamclient/steamworks_sdk_142/" \
diff --git a/steam_helper/steam.cpp b/steam_helper/steam.cpp
index 190a8c05..e4558fe4 100644
--- a/steam_helper/steam.cpp
+++ b/steam_helper/steam.cpp
@@ -36,6 +36,7 @@
 #define WIN32_NO_STATUS
 #include <windows.h>
 #include <winternl.h>
+#include <shlwapi.h>
 #include <shlobj.h>
 #include <string.h>
 #include <stdio.h>
@@ -1235,46 +1236,73 @@ run:
     }
 }
 
-static BOOL steam_protocol_handler(int argc, char *argv[])
+/* Forward stub steam.exe commands to the native steam client */
+static BOOL steam_command_handler(int argc, char *argv[])
 {
-    const char *steam_prefix = "steam://";
-    STARTUPINFOA si = { sizeof(si) };
-    PROCESS_INFORMATION pi;
-    char cmd[1024];
-    int size;
-    int i;
+    typedef NTSTATUS (WINAPI *__WINE_UNIX_SPAWNVP)(char *const argv[], int wait);
+    static __WINE_UNIX_SPAWNVP p__wine_unix_spawnvp;
+    NTSTATUS status = STATUS_UNSUCCESSFUL;
+    char **unix_argv;
+    HMODULE module;
+    int i, j;
+    static char *unix_steam[] =
+    {
+        (char *)"steam-runtime-steam-remote",
+        (char *)"steam",
+        NULL,
+    };
+
+    /* If there are command line options, only forward steam:// and options start with - */
+    if (argc > 1 && StrStrIA(argv[1], "steam://") != argv[1] && argv[1][0] != '-')
+        return FALSE;
+
+    if (!p__wine_unix_spawnvp)
+    {
+        module = GetModuleHandleA("ntdll.dll");
+        p__wine_unix_spawnvp = (__WINE_UNIX_SPAWNVP)GetProcAddress(module, "__wine_unix_spawnvp");
+        if (!p__wine_unix_spawnvp)
+        {
+            WINE_ERR("Failed to load __wine_unix_spawnvp().\n");
+            return FALSE;
+        }
+    }
+
+    if (!(unix_argv = static_cast<char **>(malloc((argc + 1) * sizeof(*unix_argv)))))
+    {
+        WINE_ERR("Out of memory.\n");
+        return FALSE;
+    }
 
     for (i = 1; i < argc; ++i)
-        if (!strcmp(argv[i], "--"))
+        unix_argv[i] = argv[i];
+    unix_argv[argc] = NULL;
+
+    for (i = 0; i < ARRAY_SIZE(unix_steam); ++i)
+    {
+        unix_argv[0] = unix_steam[i];
+
+        WINE_TRACE("Trying");
+        for (j = 0; j < argc; ++j)
+            WINE_TRACE(" %s", wine_dbgstr_a(unix_argv[j]));
+        WINE_TRACE("\n");
+
+        status = p__wine_unix_spawnvp(unix_argv, TRUE);
+        if (status == STATUS_SUCCESS)
             break;
-
-    if (i >= argc - 1)
-        return FALSE;
-    ++i;
-
-    if (strlen(argv[i]) < ARRAY_SIZE(steam_prefix))
-        return FALSE;
-
-    if (strncasecmp(argv[i], steam_prefix, ARRAY_SIZE(steam_prefix) - 1))
-        return FALSE;
-
-    size = snprintf(cmd, sizeof(cmd), "winebrowser \"%s\"", argv[i]);
-    if (size >= sizeof(cmd))
-    {
-        WINE_ERR("Argument is too large, argv[%d] %s.\n", i, wine_dbgstr_a(argv[i]));
-        return TRUE;
     }
+    free(unix_argv);
 
-    WINE_TRACE("Executing %s.\n", wine_dbgstr_a(cmd));
-    if (!CreateProcessA(NULL, cmd, NULL, NULL, FALSE, CREATE_UNICODE_ENVIRONMENT, NULL, NULL, &si, &pi))
+    if (status == STATUS_SUCCESS)
     {
-        WINE_ERR("Failed to create process %s, error %u.\n", wine_dbgstr_a(cmd), GetLastError());
-        return TRUE;
+        WINE_TRACE("Forwarded command to native steam.\n");
+    }
+    else
+    {
+        WINE_ERR("Forwarding");
+        for (i = 0; i < argc; ++i)
+            WINE_ERR(" %s", wine_dbgstr_a(argv[i]));
+        WINE_ERR(" to native steam failed, status %#lx.\n", status);
     }
-    FreeConsole();
-    WaitForSingleObject(pi.hProcess, INFINITE);
-    CloseHandle(pi.hThread);
-    CloseHandle(pi.hProcess);
     return TRUE;
 }
 
@@ -1482,7 +1510,7 @@ int main(int argc, char *argv[])
 
     WINE_TRACE("\n");
 
-    if (steam_protocol_handler(argc, argv))
+    if (steam_command_handler(argc, argv))
         return 0;
 
     if (getenv("SteamGameId"))