1. #if defined(__APPLE__)
    2. #include <mach/task.h>
    3. #include <mach/mach_init.h>
    5. int64_t Stats::GetMemoryRSS() {
    6.   task_t task = MACH_PORT_NULL;
    7.   struct task_basic_info t_info;
    8.   mach_msg_type_number_t t_info_count = TASK_BASIC_INFO_COUNT;
    9.   if (task_for_pid(current_task(), getpid(), &task) != KERN_SUCCESS) return 0;
    10.   task_info(task, TASK_BASIC_INFO, (task_info_t)&t_info, &t_info_count);
    11.   return static_cast<int64_t>(t_info.resident_size);
    12. }
    13. #else
    14. #include <fcntl.h>
    16. #include <string>
    17. #include <cstdio>
    18. #include <cstring>
    20. int64_t Stats::GetMemoryRSS() {
    21.   int fd, count;
    22.   char buf[4096], filename[256];
    23.   snprintf(filename, sizeof(filename), "/proc/%d/stat", getpid());
    24.   if ((fd = open(filename, O_RDONLY)) == -1) return 0;
    25.   if (read(fd, buf, sizeof(buf)) <= 0) {
    26.     close(fd);
    27.     return 0;
    28.   }
    29.   close(fd);
    31.   char *start = buf;
    32.   count = 23;    // RSS is the 24th field in /proc/<pid>/stat
    33.   while (start && count--) {
    34.     start = strchr(start, ' ');
    35.     if (start) start++;
    36.   }
    37.   if (!start) return 0;
    38.   char *stop = strchr(start, ' ');
    39.   if (!stop) return 0;
    40.   *stop = '\0';
    41.   int rss = std::atoi(start);
    42.   return static_cast<int64_t>(rss * sysconf(_SC_PAGESIZE));
    43. }
    44. #endif