代码之家 › 专栏 › 技术社区 › BenAlabaster

如何以相当可移植的方式以编程方式检查内存使用?(C/C++)

portability memory c++

BenAlabaster · 技术社区 · 16 年前

我正在编写跨平台的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)应该等于随机数,但我正试图在更复杂的层次上实现这一点,包括可能的可以分配内存的库调用。

以下是不可取的:

使用Valgrind(或其同类产品)
使用自定义内存分配器跟踪分配的内存。

4 回复 | 直到 10 年前

James Curran 16 年前

没有便携式的方法可以做到这一点。
对于大多数操作系统,甚至没有一种可靠的方法来实现特定于该操作系统的操作。

timrau 10 年前

我写了一些代码来尝试用一种可移植的方式来实现这一点。它并不完美,但我认为它至少应该为如何在几个平台上实现这一点提供一个指针。

(P.S.我经常使用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
}

Qubeuc 16 年前

您可以使用“内存池”模式。程序中的所有对象都从该池中分配/取消分配内存,以便了解您使用了多少内存。

Gilles 'SO- stop being evil' 12 年前

我已经用过 SIGAR API 在主要平台上非常方便地获取各种与系统相关的信息。它也是开源的(Apache)。真的没有必要在这些相对琐碎但乏味的工作上重新发明轮子。