refactor: merge memory_used.hpp and system_version.hpp into system_info.hpp

This commit is contained in:
2026-04-16 22:48:47 +08:00
parent 278fcef295
commit b07adea771
4 changed files with 34 additions and 41 deletions

View File

@@ -0,0 +1,90 @@
#pragma once
#include <string>
#include <Cubed/tools/log.hpp>
#ifdef _WIN32
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <psapi.h>
typedef LONG (WINAPI* RtlGetVersionPtr)(PRTL_OSVERSIONINFOW);
#elif defined(__linux__)
#include <sys/resource.h>
#include <unistd.h>
#include <fstream>
#endif
namespace Tools {
inline bool get_os_version(std::string& str) {
#ifdef _WIN32
HMODULE hntdll = GetModuleHandleW(L"ntdll.dll");
if (!hntdll) return false;
auto prtl_get_version = reinterpret_cast<RtlGetVersionPtr>(
GetProcAddress(hntdll, "RtlGetVersion"));
if (!prtl_get_version) return false;
RTL_OSVERSIONINFOW osvi = { 0 };
osvi.dwOSVersionInfoSize = sizeof(osvi);
if (prtl_get_version(&osvi) != 0) return false;
if (osvi.dwMajorVersion == 10) {
if (osvi.dwBuildNumber >= 22000) {
str = "Windows 11 Build " + std::to_string(osvi.dwBuildNumber);
} else {
str = "Windows 10 Build " + std::to_string(osvi.dwBuildNumber);
}
} else {
str = "Windows Build " + std::to_string(osvi.dwBuildNumber);
}
return true;
#elif defined(__linux__)
std::ifstream file("/etc/os-release");
if (!file.is_open()) {
Logger::error("Can't Open /etc/os-release");
return false;
}
std::string line;
while (std::getline(file, line)) {
size_t eq_pos = line.find("=");
if (eq_pos == std::string::npos) {
continue;
}
std::string key = line.substr(0, eq_pos);
if (key != "PRETTY_NAME") {
continue;
}
str = line.substr(eq_pos + 1);
if (str.size() >= 2 && str.front() == '"' && str.back() == '"') {
str = str.substr(1, str.size() - 2);
return true;
}
}
return false;
#else
str = "Unknown OS";
return false;
#endif
}
inline size_t get_current_rss() {
#ifdef _WIN32
PROCESS_MEMORY_COUNTERS_EX pmc;
if (GetProcessMemoryInfo(GetCurrentProcess(), (PROCESS_MEMORY_COUNTERS*)&pmc, sizeof(pmc))) {
return pmc.WorkingSetSize;
}
return 0;
#elif defined(__linux__)
std::ifstream statm("/proc/self/statm");
long vsz = 0, rss_pages = 0;
statm >> vsz >> rss_pages;
statm.close();
long page_size = sysconf(_SC_PAGESIZE);
return rss_pages * page_size;
#else
return 0; // Unsupported platform
#endif
}
}