mem.c

Go to the documentation of this file.

 
#include <arch/mem.h>
 
#include <arch/platform.h>
 
#ifdef __POSIX
 
#include <sys/resource.h>
#include <sys/time.h>
 
#if defined(__MACOS)
 
#include <mach/mach_init.h>
#include <mach/mach_port.h>
#include <mach/task.h>
#include <mach/task_info.h>
 
int mem_stat_setup(void)
{
        return 0;
}
 
size_t mem_stat_rss_current_get(void)
{
        struct mach_task_basic_info info;
        mach_msg_type_number_t count = MACH_TASK_BASIC_INFO_COUNT;
        if (__builtin_expect(task_info(mach_task_self(), MACH_TASK_BASIC_INFO,
                      (task_info_t)&info, &count) != KERN_SUCCESS, 0))
                return (size_t)0;
        return (size_t)info.resident_size;
}
 
#elif defined(__LINUX)
 
#include <ctype.h>
#include <fcntl.h>
#include <unistd.h>
 
static int proc_stat_fd;
static long linux_page_size;
 
int mem_stat_setup(void)
{
        proc_stat_fd = open("/proc/self/statm", O_RDONLY);
        if (proc_stat_fd == -1)
                return -1;
 
        linux_page_size = sysconf(_SC_PAGESIZE);
        return 0;
}
 
size_t mem_stat_rss_current_get(void)
{
        char res[40]; // sufficient for two 64 bit base 10 numbers and a space
        if (__builtin_expect(lseek(proc_stat_fd, 0, SEEK_SET) == -1 ||
                             read(proc_stat_fd, res, sizeof(res) - 1) == -1, 0))
                return (size_t)0;
 
        res[sizeof(res) - 1] = '\0';
 
        size_t i = 0;
        while (res[i] && !isspace(res[i])) {
                ++i;
        }
 
        return (size_t)(strtoull(&res[i], NULL, 10) * linux_page_size);
}
 
#else
 
int mem_stat_setup(void)
{
        return 0;
}
 
size_t mem_stat_rss_current_get(void)
{
        return 0;
}
 
#endif
 
size_t mem_stat_rss_max_get(void)
{
        struct rusage res;
        getrusage(RUSAGE_SELF, &res);
 
#ifdef __MACOS
        return (size_t)res.ru_maxrss;
#else
        return (size_t)res.ru_maxrss * 1024;
#endif
}
 
#endif
 
#ifdef __WINDOWS
 
#include <psapi.h>
#include <windows.h>
 
int mem_stat_setup(void)
{
        return 0;
}
 
size_t mem_stat_rss_current_get(void)
{
        PROCESS_MEMORY_COUNTERS info;
        GetProcessMemoryInfo(GetCurrentProcess(), &info, sizeof(info));
        return (size_t)info.WorkingSetSize;
}
 
size_t mem_stat_rss_max_get(void)
{
        PROCESS_MEMORY_COUNTERS info;
        GetProcessMemoryInfo(GetCurrentProcess(), &info, sizeof(info));
        return (size_t)info.PeakWorkingSetSize;
}
 
#endif