123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936 |
- /* Profile heap and stack memory usage of running program.
- Copyright (C) 1998-2019 Free Software Foundation, Inc.
- This file is part of the GNU C Library.
- Contributed by Ulrich Drepper <drepper@cygnus.com>, 1998.
- The GNU C Library is free software; you can redistribute it and/or
- modify it under the terms of the GNU Lesser General Public
- License as published by the Free Software Foundation; either
- version 2.1 of the License, or (at your option) any later version.
- The GNU C Library is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- Lesser General Public License for more details.
- You should have received a copy of the GNU Lesser General Public
- License along with the GNU C Library; if not, see
- <http://www.gnu.org/licenses/>. */
- #include <assert.h>
- #include <atomic.h>
- #include <dlfcn.h>
- #include <errno.h>
- #include <fcntl.h>
- #include <inttypes.h>
- #include <signal.h>
- #include <stdarg.h>
- #include <stdbool.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <unistd.h>
- #include <stdint.h>
- #include <sys/mman.h>
- #include <sys/time.h>
- #include <memusage.h>
- /* Pointer to the real functions. These are determined used `dlsym'
- when really needed. */
- static void *(*mallocp)(size_t);
- static void *(*reallocp) (void *, size_t);
- static void *(*callocp) (size_t, size_t);
- static void (*freep) (void *);
- static void *(*mmapp) (void *, size_t, int, int, int, off_t);
- static void *(*mmap64p) (void *, size_t, int, int, int, off64_t);
- static int (*munmapp) (void *, size_t);
- static void *(*mremapp) (void *, size_t, size_t, int, void *);
- enum
- {
- idx_malloc = 0,
- idx_realloc,
- idx_calloc,
- idx_free,
- idx_mmap_r,
- idx_mmap_w,
- idx_mmap_a,
- idx_mremap,
- idx_munmap,
- idx_last
- };
- struct header
- {
- size_t length;
- size_t magic;
- };
- #define MAGIC 0xfeedbeaf
- static memusage_cntr_t calls[idx_last];
- static memusage_cntr_t failed[idx_last];
- static memusage_size_t total[idx_last];
- static memusage_size_t grand_total;
- static memusage_cntr_t histogram[65536 / 16];
- static memusage_cntr_t large;
- static memusage_cntr_t calls_total;
- static memusage_cntr_t inplace;
- static memusage_cntr_t decreasing;
- static memusage_cntr_t realloc_free;
- static memusage_cntr_t inplace_mremap;
- static memusage_cntr_t decreasing_mremap;
- static memusage_size_t current_heap;
- static memusage_size_t peak_use[3];
- static __thread uintptr_t start_sp;
- /* A few macros to make the source more readable. */
- #define peak_heap peak_use[0]
- #define peak_stack peak_use[1]
- #define peak_total peak_use[2]
- #define DEFAULT_BUFFER_SIZE 32768
- static size_t buffer_size;
- static int fd = -1;
- static bool not_me;
- static int initialized;
- static bool trace_mmap;
- extern const char *__progname;
- struct entry
- {
- uint64_t heap;
- uint64_t stack;
- uint32_t time_low;
- uint32_t time_high;
- };
- static struct entry buffer[2 * DEFAULT_BUFFER_SIZE];
- static uatomic32_t buffer_cnt;
- static struct entry first;
- /* Update the global data after a successful function call. */
- static void
- update_data (struct header *result, size_t len, size_t old_len)
- {
- if (result != NULL)
- {
- /* Record the information we need and mark the block using a
- magic number. */
- result->length = len;
- result->magic = MAGIC;
- }
- /* Compute current heap usage and compare it with the maximum value. */
- memusage_size_t heap
- = catomic_exchange_and_add (¤t_heap, len - old_len) + len - old_len;
- catomic_max (&peak_heap, heap);
- /* Compute current stack usage and compare it with the maximum
- value. The base stack pointer might not be set if this is not
- the main thread and it is the first call to any of these
- functions. */
- if (__glibc_unlikely (!start_sp))
- start_sp = GETSP ();
- uintptr_t sp = GETSP ();
- #ifdef STACK_GROWS_UPWARD
- /* This can happen in threads where we didn't catch the thread's
- stack early enough. */
- if (__glibc_unlikely (sp < start_sp))
- start_sp = sp;
- size_t current_stack = sp - start_sp;
- #else
- /* This can happen in threads where we didn't catch the thread's
- stack early enough. */
- if (__glibc_unlikely (sp > start_sp))
- start_sp = sp;
- size_t current_stack = start_sp - sp;
- #endif
- catomic_max (&peak_stack, current_stack);
- /* Add up heap and stack usage and compare it with the maximum value. */
- catomic_max (&peak_total, heap + current_stack);
- /* Store the value only if we are writing to a file. */
- if (fd != -1)
- {
- uatomic32_t idx = catomic_exchange_and_add (&buffer_cnt, 1);
- if (idx + 1 >= 2 * buffer_size)
- {
- /* We try to reset the counter to the correct range. If
- this fails because of another thread increasing the
- counter it does not matter since that thread will take
- care of the correction. */
- uatomic32_t reset = (idx + 1) % (2 * buffer_size);
- catomic_compare_and_exchange_val_acq (&buffer_cnt, reset, idx + 1);
- if (idx >= 2 * buffer_size)
- idx = reset - 1;
- }
- assert (idx < 2 * DEFAULT_BUFFER_SIZE);
- buffer[idx].heap = current_heap;
- buffer[idx].stack = current_stack;
- GETTIME (buffer[idx].time_low, buffer[idx].time_high);
- /* Write out buffer if it is full. */
- if (idx + 1 == buffer_size)
- write (fd, buffer, buffer_size * sizeof (struct entry));
- else if (idx + 1 == 2 * buffer_size)
- write (fd, &buffer[buffer_size], buffer_size * sizeof (struct entry));
- }
- }
- /* Interrupt handler. */
- static void
- int_handler (int signo)
- {
- /* Nothing gets allocated. Just record the stack pointer position. */
- update_data (NULL, 0, 0);
- }
- /* Find out whether this is the program we are supposed to profile.
- For this the name in the variable `__progname' must match the one
- given in the environment variable MEMUSAGE_PROG_NAME. If the variable
- is not present every program assumes it should be profiling.
- If this is the program open a file descriptor to the output file.
- We will write to it whenever the buffer overflows. The name of the
- output file is determined by the environment variable MEMUSAGE_OUTPUT.
- If the environment variable MEMUSAGE_BUFFER_SIZE is set its numerical
- value determines the size of the internal buffer. The number gives
- the number of elements in the buffer. By setting the number to one
- one effectively selects unbuffered operation.
- If MEMUSAGE_NO_TIMER is not present an alarm handler is installed
- which at the highest possible frequency records the stack pointer. */
- static void
- me (void)
- {
- const char *env = getenv ("MEMUSAGE_PROG_NAME");
- size_t prog_len = strlen (__progname);
- initialized = -1;
- mallocp = (void *(*)(size_t))dlsym (RTLD_NEXT, "malloc");
- reallocp = (void *(*)(void *, size_t))dlsym (RTLD_NEXT, "realloc");
- callocp = (void *(*)(size_t, size_t))dlsym (RTLD_NEXT, "calloc");
- freep = (void (*)(void *))dlsym (RTLD_NEXT, "free");
- mmapp = (void *(*)(void *, size_t, int, int, int, off_t))dlsym (RTLD_NEXT,
- "mmap");
- mmap64p =
- (void *(*)(void *, size_t, int, int, int, off64_t))dlsym (RTLD_NEXT,
- "mmap64");
- mremapp = (void *(*)(void *, size_t, size_t, int, void *))dlsym (RTLD_NEXT,
- "mremap");
- munmapp = (int (*)(void *, size_t))dlsym (RTLD_NEXT, "munmap");
- initialized = 1;
- if (env != NULL)
- {
- /* Check for program name. */
- size_t len = strlen (env);
- if (len > prog_len || strcmp (env, &__progname[prog_len - len]) != 0
- || (prog_len != len && __progname[prog_len - len - 1] != '/'))
- not_me = true;
- }
- /* Only open the file if it's really us. */
- if (!not_me && fd == -1)
- {
- const char *outname;
- if (!start_sp)
- start_sp = GETSP ();
- outname = getenv ("MEMUSAGE_OUTPUT");
- if (outname != NULL && outname[0] != '\0'
- && (access (outname, R_OK | W_OK) == 0 || errno == ENOENT))
- {
- fd = creat64 (outname, 0666);
- if (fd == -1)
- /* Don't do anything in future calls if we cannot write to
- the output file. */
- not_me = true;
- else
- {
- /* Write the first entry. */
- first.heap = 0;
- first.stack = 0;
- GETTIME (first.time_low, first.time_high);
- /* Write it two times since we need the starting and end time. */
- write (fd, &first, sizeof (first));
- write (fd, &first, sizeof (first));
- /* Determine the buffer size. We use the default if the
- environment variable is not present. */
- buffer_size = DEFAULT_BUFFER_SIZE;
- const char *str_buffer_size = getenv ("MEMUSAGE_BUFFER_SIZE");
- if (str_buffer_size != NULL)
- {
- buffer_size = atoi (str_buffer_size);
- if (buffer_size == 0 || buffer_size > DEFAULT_BUFFER_SIZE)
- buffer_size = DEFAULT_BUFFER_SIZE;
- }
- /* Possibly enable timer-based stack pointer retrieval. */
- if (getenv ("MEMUSAGE_NO_TIMER") == NULL)
- {
- struct sigaction act;
- act.sa_handler = (sighandler_t) &int_handler;
- act.sa_flags = SA_RESTART;
- sigfillset (&act.sa_mask);
- if (sigaction (SIGPROF, &act, NULL) >= 0)
- {
- struct itimerval timer;
- timer.it_value.tv_sec = 0;
- timer.it_value.tv_usec = 1;
- timer.it_interval = timer.it_value;
- setitimer (ITIMER_PROF, &timer, NULL);
- }
- }
- }
- }
- if (!not_me && getenv ("MEMUSAGE_TRACE_MMAP") != NULL)
- trace_mmap = true;
- }
- }
- /* Record the initial stack position. */
- static void
- __attribute__ ((constructor))
- init (void)
- {
- start_sp = GETSP ();
- if (!initialized)
- me ();
- }
- /* `malloc' replacement. We keep track of the memory usage if this is the
- correct program. */
- void *
- malloc (size_t len)
- {
- struct header *result = NULL;
- /* Determine real implementation if not already happened. */
- if (__glibc_unlikely (initialized <= 0))
- {
- if (initialized == -1)
- return NULL;
- me ();
- }
- /* If this is not the correct program just use the normal function. */
- if (not_me)
- return (*mallocp)(len);
- /* Keep track of number of calls. */
- catomic_increment (&calls[idx_malloc]);
- /* Keep track of total memory consumption for `malloc'. */
- catomic_add (&total[idx_malloc], len);
- /* Keep track of total memory requirement. */
- catomic_add (&grand_total, len);
- /* Remember the size of the request. */
- if (len < 65536)
- catomic_increment (&histogram[len / 16]);
- else
- catomic_increment (&large);
- /* Total number of calls of any of the functions. */
- catomic_increment (&calls_total);
- /* Do the real work. */
- result = (struct header *) (*mallocp)(len + sizeof (struct header));
- if (result == NULL)
- {
- catomic_increment (&failed[idx_malloc]);
- return NULL;
- }
- /* Update the allocation data and write out the records if necessary. */
- update_data (result, len, 0);
- /* Return the pointer to the user buffer. */
- return (void *) (result + 1);
- }
- /* `realloc' replacement. We keep track of the memory usage if this is the
- correct program. */
- void *
- realloc (void *old, size_t len)
- {
- struct header *result = NULL;
- struct header *real;
- size_t old_len;
- /* Determine real implementation if not already happened. */
- if (__glibc_unlikely (initialized <= 0))
- {
- if (initialized == -1)
- return NULL;
- me ();
- }
- /* If this is not the correct program just use the normal function. */
- if (not_me)
- return (*reallocp)(old, len);
- if (old == NULL)
- {
- /* This is really a `malloc' call. */
- real = NULL;
- old_len = 0;
- }
- else
- {
- real = ((struct header *) old) - 1;
- if (real->magic != MAGIC)
- /* This is no memory allocated here. */
- return (*reallocp)(old, len);
- old_len = real->length;
- }
- /* Keep track of number of calls. */
- catomic_increment (&calls[idx_realloc]);
- if (len > old_len)
- {
- /* Keep track of total memory consumption for `realloc'. */
- catomic_add (&total[idx_realloc], len - old_len);
- /* Keep track of total memory requirement. */
- catomic_add (&grand_total, len - old_len);
- }
- if (len == 0 && old != NULL)
- {
- /* Special case. */
- catomic_increment (&realloc_free);
- /* Keep track of total memory freed using `free'. */
- catomic_add (&total[idx_free], real->length);
- /* Update the allocation data and write out the records if necessary. */
- update_data (NULL, 0, old_len);
- /* Do the real work. */
- (*freep) (real);
- return NULL;
- }
- /* Remember the size of the request. */
- if (len < 65536)
- catomic_increment (&histogram[len / 16]);
- else
- catomic_increment (&large);
- /* Total number of calls of any of the functions. */
- catomic_increment (&calls_total);
- /* Do the real work. */
- result = (struct header *) (*reallocp)(real, len + sizeof (struct header));
- if (result == NULL)
- {
- catomic_increment (&failed[idx_realloc]);
- return NULL;
- }
- /* Record whether the reduction/increase happened in place. */
- if (real == result)
- catomic_increment (&inplace);
- /* Was the buffer increased? */
- if (old_len > len)
- catomic_increment (&decreasing);
- /* Update the allocation data and write out the records if necessary. */
- update_data (result, len, old_len);
- /* Return the pointer to the user buffer. */
- return (void *) (result + 1);
- }
- /* `calloc' replacement. We keep track of the memory usage if this is the
- correct program. */
- void *
- calloc (size_t n, size_t len)
- {
- struct header *result;
- size_t size = n * len;
- /* Determine real implementation if not already happened. */
- if (__glibc_unlikely (initialized <= 0))
- {
- if (initialized == -1)
- return NULL;
- me ();
- }
- /* If this is not the correct program just use the normal function. */
- if (not_me)
- return (*callocp)(n, len);
- /* Keep track of number of calls. */
- catomic_increment (&calls[idx_calloc]);
- /* Keep track of total memory consumption for `calloc'. */
- catomic_add (&total[idx_calloc], size);
- /* Keep track of total memory requirement. */
- catomic_add (&grand_total, size);
- /* Remember the size of the request. */
- if (size < 65536)
- catomic_increment (&histogram[size / 16]);
- else
- catomic_increment (&large);
- /* Total number of calls of any of the functions. */
- ++calls_total;
- /* Do the real work. */
- result = (struct header *) (*mallocp)(size + sizeof (struct header));
- if (result == NULL)
- {
- catomic_increment (&failed[idx_calloc]);
- return NULL;
- }
- /* Update the allocation data and write out the records if necessary. */
- update_data (result, size, 0);
- /* Do what `calloc' would have done and return the buffer to the caller. */
- return memset (result + 1, '\0', size);
- }
- /* `free' replacement. We keep track of the memory usage if this is the
- correct program. */
- void
- free (void *ptr)
- {
- struct header *real;
- /* Determine real implementation if not already happened. */
- if (__glibc_unlikely (initialized <= 0))
- {
- if (initialized == -1)
- return;
- me ();
- }
- /* If this is not the correct program just use the normal function. */
- if (not_me)
- {
- (*freep) (ptr);
- return;
- }
- /* `free (NULL)' has no effect. */
- if (ptr == NULL)
- {
- catomic_increment (&calls[idx_free]);
- return;
- }
- /* Determine the pointer to the header. */
- real = ((struct header *) ptr) - 1;
- if (real->magic != MAGIC)
- {
- /* This block wasn't allocated here. */
- (*freep) (ptr);
- return;
- }
- /* Keep track of number of calls. */
- catomic_increment (&calls[idx_free]);
- /* Keep track of total memory freed using `free'. */
- catomic_add (&total[idx_free], real->length);
- /* Update the allocation data and write out the records if necessary. */
- update_data (NULL, 0, real->length);
- /* Do the real work. */
- (*freep) (real);
- }
- /* `mmap' replacement. We do not have to keep track of the size since
- `munmap' will get it as a parameter. */
- void *
- mmap (void *start, size_t len, int prot, int flags, int fd, off_t offset)
- {
- void *result = NULL;
- /* Determine real implementation if not already happened. */
- if (__glibc_unlikely (initialized <= 0))
- {
- if (initialized == -1)
- return NULL;
- me ();
- }
- /* Always get a block. We don't need extra memory. */
- result = (*mmapp)(start, len, prot, flags, fd, offset);
- if (!not_me && trace_mmap)
- {
- int idx = (flags & MAP_ANON
- ? idx_mmap_a : prot & PROT_WRITE ? idx_mmap_w : idx_mmap_r);
- /* Keep track of number of calls. */
- catomic_increment (&calls[idx]);
- /* Keep track of total memory consumption for `malloc'. */
- catomic_add (&total[idx], len);
- /* Keep track of total memory requirement. */
- catomic_add (&grand_total, len);
- /* Remember the size of the request. */
- if (len < 65536)
- catomic_increment (&histogram[len / 16]);
- else
- catomic_increment (&large);
- /* Total number of calls of any of the functions. */
- catomic_increment (&calls_total);
- /* Check for failures. */
- if (result == NULL)
- catomic_increment (&failed[idx]);
- else if (idx == idx_mmap_w)
- /* Update the allocation data and write out the records if
- necessary. Note the first parameter is NULL which means
- the size is not tracked. */
- update_data (NULL, len, 0);
- }
- /* Return the pointer to the user buffer. */
- return result;
- }
- /* `mmap64' replacement. We do not have to keep track of the size since
- `munmap' will get it as a parameter. */
- void *
- mmap64 (void *start, size_t len, int prot, int flags, int fd, off64_t offset)
- {
- void *result = NULL;
- /* Determine real implementation if not already happened. */
- if (__glibc_unlikely (initialized <= 0))
- {
- if (initialized == -1)
- return NULL;
- me ();
- }
- /* Always get a block. We don't need extra memory. */
- result = (*mmap64p)(start, len, prot, flags, fd, offset);
- if (!not_me && trace_mmap)
- {
- int idx = (flags & MAP_ANON
- ? idx_mmap_a : prot & PROT_WRITE ? idx_mmap_w : idx_mmap_r);
- /* Keep track of number of calls. */
- catomic_increment (&calls[idx]);
- /* Keep track of total memory consumption for `malloc'. */
- catomic_add (&total[idx], len);
- /* Keep track of total memory requirement. */
- catomic_add (&grand_total, len);
- /* Remember the size of the request. */
- if (len < 65536)
- catomic_increment (&histogram[len / 16]);
- else
- catomic_increment (&large);
- /* Total number of calls of any of the functions. */
- catomic_increment (&calls_total);
- /* Check for failures. */
- if (result == NULL)
- catomic_increment (&failed[idx]);
- else if (idx == idx_mmap_w)
- /* Update the allocation data and write out the records if
- necessary. Note the first parameter is NULL which means
- the size is not tracked. */
- update_data (NULL, len, 0);
- }
- /* Return the pointer to the user buffer. */
- return result;
- }
- /* `mremap' replacement. We do not have to keep track of the size since
- `munmap' will get it as a parameter. */
- void *
- mremap (void *start, size_t old_len, size_t len, int flags, ...)
- {
- void *result = NULL;
- va_list ap;
- va_start (ap, flags);
- void *newaddr = (flags & MREMAP_FIXED) ? va_arg (ap, void *) : NULL;
- va_end (ap);
- /* Determine real implementation if not already happened. */
- if (__glibc_unlikely (initialized <= 0))
- {
- if (initialized == -1)
- return NULL;
- me ();
- }
- /* Always get a block. We don't need extra memory. */
- result = (*mremapp)(start, old_len, len, flags, newaddr);
- if (!not_me && trace_mmap)
- {
- /* Keep track of number of calls. */
- catomic_increment (&calls[idx_mremap]);
- if (len > old_len)
- {
- /* Keep track of total memory consumption for `malloc'. */
- catomic_add (&total[idx_mremap], len - old_len);
- /* Keep track of total memory requirement. */
- catomic_add (&grand_total, len - old_len);
- }
- /* Remember the size of the request. */
- if (len < 65536)
- catomic_increment (&histogram[len / 16]);
- else
- catomic_increment (&large);
- /* Total number of calls of any of the functions. */
- catomic_increment (&calls_total);
- /* Check for failures. */
- if (result == NULL)
- catomic_increment (&failed[idx_mremap]);
- else
- {
- /* Record whether the reduction/increase happened in place. */
- if (start == result)
- catomic_increment (&inplace_mremap);
- /* Was the buffer increased? */
- if (old_len > len)
- catomic_increment (&decreasing_mremap);
- /* Update the allocation data and write out the records if
- necessary. Note the first parameter is NULL which means
- the size is not tracked. */
- update_data (NULL, len, old_len);
- }
- }
- /* Return the pointer to the user buffer. */
- return result;
- }
- /* `munmap' replacement. */
- int
- munmap (void *start, size_t len)
- {
- int result;
- /* Determine real implementation if not already happened. */
- if (__glibc_unlikely (initialized <= 0))
- {
- if (initialized == -1)
- return -1;
- me ();
- }
- /* Do the real work. */
- result = (*munmapp)(start, len);
- if (!not_me && trace_mmap)
- {
- /* Keep track of number of calls. */
- catomic_increment (&calls[idx_munmap]);
- if (__glibc_likely (result == 0))
- {
- /* Keep track of total memory freed using `free'. */
- catomic_add (&total[idx_munmap], len);
- /* Update the allocation data and write out the records if
- necessary. */
- update_data (NULL, 0, len);
- }
- else
- catomic_increment (&failed[idx_munmap]);
- }
- return result;
- }
- /* Write some statistics to standard error. */
- static void
- __attribute__ ((destructor))
- dest (void)
- {
- int percent, cnt;
- unsigned long int maxcalls;
- /* If we haven't done anything here just return. */
- if (not_me)
- return;
- /* If we should call any of the memory functions don't do any profiling. */
- not_me = true;
- /* Finish the output file. */
- if (fd != -1)
- {
- /* Write the partially filled buffer. */
- if (buffer_cnt > buffer_size)
- write (fd, buffer + buffer_size,
- (buffer_cnt - buffer_size) * sizeof (struct entry));
- else
- write (fd, buffer, buffer_cnt * sizeof (struct entry));
- /* Go back to the beginning of the file. We allocated two records
- here when we opened the file. */
- lseek (fd, 0, SEEK_SET);
- /* Write out a record containing the total size. */
- first.stack = peak_total;
- write (fd, &first, sizeof (struct entry));
- /* Write out another record containing the maximum for heap and
- stack. */
- first.heap = peak_heap;
- first.stack = peak_stack;
- GETTIME (first.time_low, first.time_high);
- write (fd, &first, sizeof (struct entry));
- /* Close the file. */
- close (fd);
- fd = -1;
- }
- /* Write a colorful statistic. */
- fprintf (stderr, "\n\
- \e[01;32mMemory usage summary:\e[0;0m heap total: %llu, heap peak: %lu, stack peak: %lu\n\
- \e[04;34m total calls total memory failed calls\e[0m\n\
- \e[00;34m malloc|\e[0m %10lu %12llu %s%12lu\e[00;00m\n\
- \e[00;34mrealloc|\e[0m %10lu %12llu %s%12lu\e[00;00m (nomove:%ld, dec:%ld, free:%ld)\n\
- \e[00;34m calloc|\e[0m %10lu %12llu %s%12lu\e[00;00m\n\
- \e[00;34m free|\e[0m %10lu %12llu\n",
- (unsigned long long int) grand_total, (unsigned long int) peak_heap,
- (unsigned long int) peak_stack,
- (unsigned long int) calls[idx_malloc],
- (unsigned long long int) total[idx_malloc],
- failed[idx_malloc] ? "\e[01;41m" : "",
- (unsigned long int) failed[idx_malloc],
- (unsigned long int) calls[idx_realloc],
- (unsigned long long int) total[idx_realloc],
- failed[idx_realloc] ? "\e[01;41m" : "",
- (unsigned long int) failed[idx_realloc],
- (unsigned long int) inplace,
- (unsigned long int) decreasing,
- (unsigned long int) realloc_free,
- (unsigned long int) calls[idx_calloc],
- (unsigned long long int) total[idx_calloc],
- failed[idx_calloc] ? "\e[01;41m" : "",
- (unsigned long int) failed[idx_calloc],
- (unsigned long int) calls[idx_free],
- (unsigned long long int) total[idx_free]);
- if (trace_mmap)
- fprintf (stderr, "\
- \e[00;34mmmap(r)|\e[0m %10lu %12llu %s%12lu\e[00;00m\n\
- \e[00;34mmmap(w)|\e[0m %10lu %12llu %s%12lu\e[00;00m\n\
- \e[00;34mmmap(a)|\e[0m %10lu %12llu %s%12lu\e[00;00m\n\
- \e[00;34m mremap|\e[0m %10lu %12llu %s%12lu\e[00;00m (nomove: %ld, dec:%ld)\n\
- \e[00;34m munmap|\e[0m %10lu %12llu %s%12lu\e[00;00m\n",
- (unsigned long int) calls[idx_mmap_r],
- (unsigned long long int) total[idx_mmap_r],
- failed[idx_mmap_r] ? "\e[01;41m" : "",
- (unsigned long int) failed[idx_mmap_r],
- (unsigned long int) calls[idx_mmap_w],
- (unsigned long long int) total[idx_mmap_w],
- failed[idx_mmap_w] ? "\e[01;41m" : "",
- (unsigned long int) failed[idx_mmap_w],
- (unsigned long int) calls[idx_mmap_a],
- (unsigned long long int) total[idx_mmap_a],
- failed[idx_mmap_a] ? "\e[01;41m" : "",
- (unsigned long int) failed[idx_mmap_a],
- (unsigned long int) calls[idx_mremap],
- (unsigned long long int) total[idx_mremap],
- failed[idx_mremap] ? "\e[01;41m" : "",
- (unsigned long int) failed[idx_mremap],
- (unsigned long int) inplace_mremap,
- (unsigned long int) decreasing_mremap,
- (unsigned long int) calls[idx_munmap],
- (unsigned long long int) total[idx_munmap],
- failed[idx_munmap] ? "\e[01;41m" : "",
- (unsigned long int) failed[idx_munmap]);
- /* Write out a histoogram of the sizes of the allocations. */
- fprintf (stderr, "\e[01;32mHistogram for block sizes:\e[0;0m\n");
- /* Determine the maximum of all calls for each size range. */
- maxcalls = large;
- for (cnt = 0; cnt < 65536; cnt += 16)
- if (histogram[cnt / 16] > maxcalls)
- maxcalls = histogram[cnt / 16];
- for (cnt = 0; cnt < 65536; cnt += 16)
- /* Only write out the nonzero entries. */
- if (histogram[cnt / 16] != 0)
- {
- percent = (histogram[cnt / 16] * 100) / calls_total;
- fprintf (stderr, "%5d-%-5d%12lu ", cnt, cnt + 15,
- (unsigned long int) histogram[cnt / 16]);
- if (percent == 0)
- fputs (" <1% \e[41;37m", stderr);
- else
- fprintf (stderr, "%3d%% \e[41;37m", percent);
- /* Draw a bar with a length corresponding to the current
- percentage. */
- percent = (histogram[cnt / 16] * 50) / maxcalls;
- while (percent-- > 0)
- fputc ('=', stderr);
- fputs ("\e[0;0m\n", stderr);
- }
- if (large != 0)
- {
- percent = (large * 100) / calls_total;
- fprintf (stderr, " large %12lu ", (unsigned long int) large);
- if (percent == 0)
- fputs (" <1% \e[41;37m", stderr);
- else
- fprintf (stderr, "%3d%% \e[41;37m", percent);
- percent = (large * 50) / maxcalls;
- while (percent-- > 0)
- fputc ('=', stderr);
- fputs ("\e[0;0m\n", stderr);
- }
- /* Any following malloc/free etc. calls should generate statistics again,
- because otherwise freeing something that has been malloced before
- this destructor (including struct header in front of it) wouldn't
- be properly freed. */
- not_me = false;
- }
|