mirror of
https://github.com/zhenyan121/Cubed.git
synced 2026-06-17 16:17:02 +08:00
121 lines
3.0 KiB
C++
121 lines
3.0 KiB
C++
#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
|
|
}
|
|
|
|
inline std::string get_cpu_info() {
|
|
#ifdef _WIN32
|
|
|
|
#elif defined (__linux__)
|
|
std::ifstream file("/proc/cpuinfo");
|
|
if (!file.is_open()) {
|
|
return std::string{"Unkown"};
|
|
}
|
|
std::string line;
|
|
while (std::getline(file, line)) {
|
|
size_t pos = line.find(":");
|
|
if (pos == std::string::npos) {
|
|
continue;
|
|
}
|
|
std::string key = line.substr(0, pos - 1);
|
|
key = key.erase(key.find_last_not_of(" \t\n") + 1);
|
|
if (key != "model name") {
|
|
continue;
|
|
}
|
|
return line.substr(pos + 2);
|
|
}
|
|
return std::string{"Unkown"};
|
|
#else
|
|
return std::string{"Unkown"};
|
|
#endif
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|