blob: d141593a68fcc2dce32816dcccba6778507e0c5a (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
//===-- MemoryUsage.cpp ---------------------------------------------------===//
//
// The KLEE Symbolic Virtual Machine
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "klee/Internal/System/MemoryUsage.h"
#include "klee/Config/config.h"
#ifdef HAVE_GPERFTOOLS_MALLOC_EXTENSION_H
#include "gperftools/malloc_extension.h"
#endif
#ifdef HAVE_MALLINFO
#include <malloc.h>
#endif
#ifdef HAVE_MALLOC_MALLOC_H
#include <malloc/malloc.h>
#endif
using namespace klee;
size_t util::GetTotalMallocUsage() {
#ifdef HAVE_GPERFTOOLS_MALLOC_EXTENSION_H
size_t value = 0;
MallocExtension::instance()->GetNumericProperty(
"generic.current_allocated_bytes", &value);
return value;
#elif HAVE_MALLINFO
struct mallinfo mi = ::mallinfo();
// The malloc implementation in glibc (pmalloc2)
// does not include mmap()'ed memory in mi.uordblks
// but other implementations (e.g. tcmalloc) do.
#if defined(__GLIBC__)
return (size_t)(unsigned)mi.uordblks + (unsigned)mi.hblkhd;
#else
return (unsigned)mi.uordblks;
#endif
#elif defined(HAVE_MALLOC_ZONE_STATISTICS) && defined(HAVE_MALLOC_MALLOC_H)
// Support memory usage on Darwin.
malloc_statistics_t Stats;
malloc_zone_statistics(malloc_default_zone(), &Stats);
return Stats.size_in_use;
#else // HAVE_MALLINFO
#warning Cannot get malloc info on this platform
return 0;
#endif
}
|
