mirror of
https://github.com/zhenyan121/Cubed.git
synced 2026-06-18 00:27:02 +08:00
refactor: merge memory_used.hpp and system_version.hpp into system_info.hpp
This commit is contained in:
90
include/Cubed/tools/system_info.hpp
Normal file
90
include/Cubed/tools/system_info.hpp
Normal 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
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user