thirdpartymirrors/lsfg-vk
1
0
Fork 0
mirror of https://github.com/PancakeTAS/lsfg-vk.git synced 2025-10-30 07:01:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(exe): update c++ implementation to get exe

Browse source
This commit is contained in:
PancakeTAS 2025-08-07 16:52:54 +02:00 • committed by Pancake
parent f5b55c7f83
commit 7ddf1bdbde
1 changed files with 25 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
src/utils/utils.cpp
View file

@ -8,7 +8,6 @@
#include <unistd.h> #include <unistd.h>
#include <unordered_map> #include <unordered_map>
#include <filesystem>
#include <algorithm> #include <algorithm>
#include <optional> #include <optional>
#include <iostream> #include <iostream>
@ -210,20 +209,25 @@ void Utils::resetLimitN(const std::string& id) noexcept {
} }
std::pair<std::string, std::string> Utils::getProcessName() { std::pair<std::string, std::string> Utils::getProcessName() {
// check override first
const char* process_name = std::getenv("LSFG_PROCESS"); const char* process_name = std::getenv("LSFG_PROCESS");
if (process_name && *process_name != '\0') if (process_name && *process_name != '\0')
return { process_name, process_name }; return { process_name, process_name };
// then check benchmark flag
const char* benchmark_flag = std::getenv("LSFG_BENCHMARK"); const char* benchmark_flag = std::getenv("LSFG_BENCHMARK");
if (benchmark_flag) if (benchmark_flag)
return { "benchmark", "benchmark" }; return { "benchmark", "benchmark" };
std::array<char, 4096> exe{}; std::array<char, 4096> exe{};
// find executed binary
const ssize_t exe_len = readlink("/proc/self/exe", exe.data(), exe.size() - 1); const ssize_t exe_len = readlink("/proc/self/exe", exe.data(), exe.size() - 1);
if (exe_len <= 0) if (exe_len <= 0)
return { "Unknown Process", "unknown" }; return { "Unknown Process", "unknown" };
exe.at(static_cast<size_t>(exe_len)) = '\0'; exe.at(static_cast<size_t>(exe_len)) = '\0';
std::string exe_str(exe.data());
// find command name as well
std::ifstream comm_file("/proc/self/comm"); std::ifstream comm_file("/proc/self/comm");
if (!comm_file.is_open()) if (!comm_file.is_open())
return { std::string(exe.data()), "unknown" }; return { std::string(exe.data()), "unknown" };
@ -234,28 +238,33 @@ std::pair<std::string, std::string> Utils::getProcessName() {
if (comm_str.back() == '\n') if (comm_str.back() == '\n')
comm_str.pop_back(); comm_str.pop_back();
// replace binary with exe for wine apps
if (exe_str.find("wine") != std::string::npos
|| exe_str.find("proton") != std::string::npos) {
// For .exe apps running through Proton/Wine std::ifstream proc_maps("/proc/self/maps");
if (!proc_maps.is_open())
return{ exe_str, comm_str };
std::ifstream maps_file("/proc/self/maps");
std::string line; std::string line;
while (std::getline(proc_maps, line)) {
if (!line.ends_with(".exe"))
continue;
while (getline(maps_file, line)) { const size_t pos = line.find_last_of(' ');
if (line.find(".exe") != std::string::npos) { if (pos == std::string::npos)
continue;
// Backslash needs to be a raw string literal const std::string exe_name = line.substr(pos + 1);
size_t path_start = line.find("/") || line.find(R"(\)"); if (exe_name.empty())
if (path_start != std::string::npos) { continue;
std::string path = line.substr(path_start);
comm_str = std::filesystem::path(path).filename().string(); exe_str = exe_name;
} else { break;
// If nothing found, default to 0
path_start = 0;
}
} }
} }
return{ std::string(exe.data()), comm_str }; return{ exe_str, comm_str };
} }
std::string Utils::getConfigFile() { std::string Utils::getConfigFile() {