Как программно проверить использование памяти довольно переносимым способом? (C / C ++) - PullRequest
Новая
       13

Как программно проверить использование памяти довольно переносимым способом? (C / C ++)

15 голосов
/ 16 декабря 2008

Я пишу кроссплатформенный код C ++ (Windows, Mac). Есть ли способ проверить, сколько памяти используется текущим процессом? Очень надуманный фрагмент для иллюстрации: 

unsigned long m0 = GetMemoryInUse();
char *p = new char[ random_number ];
unsigned long m1 = GetMemoryInUse();
printf( "%d bytes used\n", (m1-m0) );

Конечно (m1-m0) должно равняться random_number, но я пытаюсь сделать это на более сложном уровне, включая возможные вызовы библиотеки, которые могут выделить память.

Следующие не являются предпочтительными:

  1. Используйте Valgrind (или его аналог)
  2. Используйте специальный распределитель памяти для отслеживания выделенная память.

Ответы [ 4 ]

18 голосов
/ 16 декабря 2008

Вот некоторый код, который я написал, чтобы попытаться сделать это переносимым способом. Он не идеален, но я думаю, что он должен по крайней мере дать указание, как это сделать на каждой из нескольких платформ.

(П.С. Я регулярно использую OSX и Linux, и знаю, что это хорошо работает. Я использую Windows реже, поэтому предостережения применяются к предложению Windows, но я думаю, что это правильно.) 

#ifdef __linux__
# include <sys/sysinfo.h>
#endif

#ifdef __APPLE__
# include <mach/task.h>
# include <mach/mach_init.h>
#endif

#ifdef _WINDOWS
# include <windows.h>
#else
# include <sys/resource.h>
#endif

/// The amount of memory currently being used by this process, in bytes.
/// By default, returns the full virtual arena, but if resident=true,
/// it will report just the resident set in RAM (if supported on that OS).
size_t memory_used (bool resident=false)
{
#if defined(__linux__)
    // Ugh, getrusage doesn't work well on Linux.  Try grabbing info
    // directly from the /proc pseudo-filesystem.  Reading from
    // /proc/self/statm gives info on your own process, as one line of
    // numbers that are: virtual mem program size, resident set size,
    // shared pages, text/code, data/stack, library, dirty pages.  The
    // mem sizes should all be multiplied by the page size.
    size_t size = 0;
    FILE *file = fopen("/proc/self/statm", "r");
    if (file) {
        unsigned long vm = 0;
        fscanf (file, "%ul", &vm);  // Just need the first num: vm size
        fclose (file);
       size = (size_t)vm * getpagesize();
    }
    return size;

#elif defined(__APPLE__)
    // Inspired by:
    // http://miknight.blogspot.com/2005/11/resident-set-size-in-mac-os-x.html
    struct task_basic_info t_info;
    mach_msg_type_number_t t_info_count = TASK_BASIC_INFO_COUNT;
    task_info(current_task(), TASK_BASIC_INFO, (task_info_t)&t_info, &t_info_count);
    size_t size = (resident ? t_info.resident_size : t_info.virtual_size);
    return size;

#elif defined(_WINDOWS)
    // According to MSDN...
    PROCESS_MEMORY_COUNTERS counters;
    if (GetProcessMemoryInfo (GetCurrentProcess(), &counters, sizeof (counters)))
        return counters.PagefileUsage;
    else return 0;

#else
    // No idea what platform this is
    return 0;   // Punt
#endif
}
2 голосов
/ 16 декабря 2008

Я использовал SIGAR API для получения всех видов информации, относящейся к системе, достаточно переносимой на основные платформы. Это также открытый исходный код (Apache). На самом деле нет необходимости заново изобретать колесо в этой относительно тривиальной, но утомительной работе.

2 голосов
/ 16 декабря 2008

Вы можете использовать шаблон «пул памяти». Все объекты в вашей программе выделяют / освобождают память из этого пула, чтобы вы могли узнать, сколько памяти вы используете.

2 голосов
/ 16 декабря 2008
  • Нет портативного способа сделать это.
  • Для большинства операционных систем даже нет надежного способа сделать это специально для этой ОС.
...