memtest.h

Go to the documentation of this file.

#pragma once
 
#include <Util/conscience_log.h>
 
using namespace conscience_utils::logging;
 
#if defined(__linux__)
    #include <fstream>
#elif defined(__APPLE__)
    #include <mach/mach.h>
#endif
 
inline size_t getCurrentRSS()
{
#if defined(__linux__)
    std::ifstream file("/proc/self/status");
    std::string key;
 
    while (file >> key) {
        if (key == "VmRSS:") {
            size_t value = 0;
            std::string unit;
            file >> value >> unit;
            return value * 1024;
        }
 
        std::string rest;
        std::getline(file, rest);
    }
 
    return 0;
 
#elif defined(__APPLE__)
    mach_task_basic_info info;
    mach_msg_type_number_t count = MACH_TASK_BASIC_INFO_COUNT;
 
    const kern_return_t result = task_info(
        mach_task_self(),
        MACH_TASK_BASIC_INFO,
        reinterpret_cast<task_info_t>(&info),
        &count
    );
 
    if (result != KERN_SUCCESS) {
        return 0;
    }
 
    return static_cast<size_t>(info.resident_size);
 
#else
    return 0;
#endif
}
 
inline void logRSS(const std::string& label)
{
    const double mb = static_cast<double>(getCurrentRSS()) / 1024.0 / 1024.0;
    LOG_INFO("[RSS] " + label + ": " + std::to_string(mb) + " MB");
}