public/gpu-screen-recorder-ui
1
0
Fork 0
mirror of https://repo.dec05eba.com/gpu-screen-recorder-ui synced 2026-05-07 23:20:26 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

Detect non-steam games in steam launcher

Browse Source
This commit is contained in:
dec05eba
2026-04-28 08:22:21 +02:00
parent 42baaff9b0
commit e3ccfed987
1 changed files with 40 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
tools/gsr-game-tracker/main.c
View File

@@ -5,6 +5,7 @@
#include <errno.h> #include <errno.h>
#include <fcntl.h> #include <fcntl.h>
#include <dirent.h> #include <dirent.h>
#include <locale.h>
#include <sys/socket.h> #include <sys/socket.h>
#include <linux/netlink.h> #include <linux/netlink.h>
#include <linux/connector.h> #include <linux/connector.h>
@@ -96,26 +97,7 @@ static size_t get_argv_len(const char *cmdline, size_t size) {
static const char* memchr_reverse(const char *p, int c, size_t size) { static const char* memchr_reverse(const char *p, int c, size_t size) {
for(size_t i = 0; i < size; ++i) { for(size_t i = 0; i < size; ++i) {
if(p[size - 1 - i] == c) if(p[size - 1 - i] == c)
return p + i; return &p[size - 1 - i];
}
return NULL;
}
static const char* get_arg_by_index(const char *cmdline, size_t size, size_t index) {
size_t current_index = 0;
size_t cmdline_index = 0;
while (index < size) {
const char *arg_start = cmdline + cmdline_index;
const char *arg_end = memchr(arg_start, '\0', size - cmdline_index);
if(!arg_end)
break;
const size_t arg_len = arg_end - arg_start;
if(current_index == index)
return arg_start;
cmdline_index += arg_len + 1;
current_index += 1;
} }
return NULL; return NULL;
} }
@@ -125,12 +107,7 @@ static bool memeql(const char *haystack, size_t haystack_size, const char *needl
return haystack_size == needle_size && memcmp(haystack, needle, needle_size) == 0; return haystack_size == needle_size && memcmp(haystack, needle, needle_size) == 0;
} }
static bool starts_with(const char *haystack, size_t haystack_size, const char *needle) { static bool is_wine_binary(const char *argv0, size_t size) {
const size_t needle_size = strlen(needle);
return haystack_size >= needle_size && memcmp(haystack, needle, needle_size) == 0;
}
static int is_wine_binary(const char *argv0, size_t size) {
const char *base = memchr_reverse(argv0, '/', size); const char *base = memchr_reverse(argv0, '/', size);
base = base ? base + 1 : argv0; base = base ? base + 1 : argv0;
const size_t base_len = argv0 + size - base; const size_t base_len = argv0 + size - base;
@@ -140,14 +117,21 @@ static int is_wine_binary(const char *argv0, size_t size) {
memeql(base, base_len, "wine64-preloader"); memeql(base, base_len, "wine64-preloader");
} }
static int has_game_arch_suffix(const char *argv0, size_t size) { static bool is_wine_server(const char *argv0, size_t size) {
const char *base = memchr_reverse(argv0, '/', size);
base = base ? base + 1 : argv0;
const size_t base_len = argv0 + size - base;
return memeql(base, base_len, "wineserver");
}
static bool has_game_arch_suffix(const char *argv0, size_t size) {
static const char *suffixes[] = { ".x86_64", ".x64", ".x86" }; static const char *suffixes[] = { ".x86_64", ".x64", ".x86" };
for (int i = 0; i < 3; i++) { for (int i = 0; i < 3; i++) {
const size_t slen = strlen(suffixes[i]); const size_t slen = strlen(suffixes[i]);
if (size >= slen && memcmp(argv0 + size - slen, suffixes[i], slen) == 0) if (size >= slen && memcmp(argv0 + size - slen, suffixes[i], slen) == 0)
return 1; return true;
} }
return 0; return false;
} }
static void check_process(pid_t pid) { static void check_process(pid_t pid) {
@@ -155,24 +139,27 @@ static void check_process(pid_t pid) {
char path[64]; char path[64];
ssize_t env_n, cmd_n; ssize_t env_n, cmd_n;
bool is_steam_fossilize = false; bool is_steam_app = false;
bool is_stream_script = false; const char *steam_overlay_game_id = NULL;
/* Proton launched outside steam has this while it doesn't have SteamAppId nor is the process called wine */ /* Proton launched for a non-steam game has this while it doesn't have SteamAppId nor is the process called wine */
bool has_wine_prefix_env = false; bool has_wine_env = false;
snprintf(path, sizeof(path), "/proc/%d/environ", pid); snprintf(path, sizeof(path), "/proc/%d/environ", pid);
env_n = read_file(path, environ_buf, sizeof(environ_buf)); env_n = read_file(path, environ_buf, sizeof(environ_buf));
if (env_n > 0) { if (env_n > 0) {
is_steam_fossilize = env_get(environ_buf, env_n, "STEAM_FOSSILIZE_DUMP_PATH_READ_ONLY"); steam_overlay_game_id = env_get(environ_buf, env_n, "SteamOverlayGameId"); /* This is set for non-steam games added to steam library */
is_stream_script = env_get(environ_buf, env_n, "STEAMSCRIPT_VERSION"); const char *appid = env_get(environ_buf, env_n, "SteamAppId"); /* This is set for regular steam apps */
has_wine_prefix_env = env_get(environ_buf, env_n, "WINEPREFIX");
const char *appid = env_get(environ_buf, env_n, "SteamAppId");
if (!appid) appid = env_get(environ_buf, env_n, "SteamGameId"); if (!appid) appid = env_get(environ_buf, env_n, "SteamGameId");
if (!appid) appid = steam_overlay_game_id;
if (appid && appid[0] >= '1' && appid[0] <= '9') { if (appid && appid[0] >= '1' && appid[0] <= '9') {
add_game(pid); add_game(pid);
return; return;
} }
has_wine_env = env_get(environ_buf, env_n, "WINELOADER") != NULL;
has_wine_env = has_wine_env || (env_get(environ_buf, env_n, "WINELOADERNOEXEC") != NULL);
is_steam_app = env_get(environ_buf, env_n, "STEAM_BASE_FOLDER") != NULL;
} }
snprintf(path, sizeof(path), "/proc/%d/cmdline", pid); snprintf(path, sizeof(path), "/proc/%d/cmdline", pid);
@@ -183,20 +170,23 @@ static void check_process(pid_t pid) {
const char *argv0 = cmdline_buf; const char *argv0 = cmdline_buf;
const size_t argv0_len = get_argv_len(cmdline_buf, cmd_n); const size_t argv0_len = get_argv_len(cmdline_buf, cmd_n);
const char *argv1 = get_arg_by_index(cmdline_buf, cmd_n, 1); const char *argv1 = cmdline_buf + argv0_len + 1;
const size_t argv1_len = argv1 ? get_argv_len(argv1, cmdline_buf + cmd_n - argv1) : 0; const size_t argv1_len = (size_t)cmd_n > argv0_len + 1 ? get_argv_len(argv1, cmd_n - (argv0_len + 1)) : 0;
if (cmd_n > 0 && (is_wine_binary(argv0, argv0_len) || has_game_arch_suffix(argv0, argv0_len) || has_wine_prefix_env) if (cmd_n > 0 && (is_wine_binary(argv0, argv0_len) || has_game_arch_suffix(argv0, argv0_len) || has_wine_env)
&& !is_steam_fossilize && !is_wine_server(argv0, argv0_len)
&& !is_stream_script
&& !memeql(argv1, argv1_len, "--version") && !memeql(argv1, argv1_len, "--version")
&& !starts_with(argv0, argv0_len, "C:\\windows\\system32")) && (!is_steam_app || steam_overlay_game_id != NULL))
{ {
//fprintf(stderr, "argv0: |%.*s|\n", (int)argv0_len, argv0);
//fprintf(stderr, "*******\n");
//fprintf(stderr, "env: "); //fprintf(stderr, "env: ");
//write(STDOUT_FILENO, environ_buf, env_n); //write(STDOUT_FILENO, environ_buf, env_n);
//fprintf(stderr, "\ncmdline: "); //fprintf(stderr, "\ncmdline: ");
//write(STDOUT_FILENO, cmdline_buf, cmd_n); //write(STDOUT_FILENO, cmdline_buf, cmd_n);
//fprintf(stderr, "\n"); //fprintf(stderr, "\n");
//fprintf(stderr, "*******\n");
//fprintf(stderr, "\n\n\n");
add_game(pid); add_game(pid);
} }
} }
@@ -301,6 +291,8 @@ static void scan_existing_processes(void) {
} }
int main(void) { int main(void) {
setlocale(LC_ALL, "C"); // Sigh... stupid C
memset(games, 0, sizeof(games)); memset(games, 0, sizeof(games));
int sock = setup_netlink(); int sock = setup_netlink();