tools/trace-malloc/allocation-stacks.c

   1 /* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
   2  *
   3  * ***** BEGIN LICENSE BLOCK *****
   4  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   5  *
   6  * The contents of this file are subject to the Mozilla Public License Version
   7  * 1.1 (the "License"); you may not use this file except in compliance with
   8  * the License. You may obtain a copy of the License at
   9  * http://www.mozilla.org/MPL/
  10  *
  11  * Software distributed under the License is distributed on an "AS IS" basis,
  12  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13  * for the specific language governing rights and limitations under the
  14  * License.
  15  *
  16  * The Original Code is nsTraceMalloc.c/bloatblame.c code, released
  17  * April 19, 2000.
  18  *
  19  * The Initial Developer of the Original Code is
  20  * Netscape Communications Corporation.
  21  * Portions created by the Initial Developer are Copyright (C) 2000
  22  * the Initial Developer. All Rights Reserved.
  23  *
  24  * Contributor(s):
  25  *   Brendan Eich, 14-April-2000
  26  *   L. David Baron, 2001-06-07, created leakstats.c based on bloatblame.c
  27  *   L. David Baron, 2004-02-19, created allocation-stats.c based
  28  *   on leakstats.c
  29  *
  30  * Alternatively, the contents of this file may be used under the terms of
  31  * either the GNU General Public License Version 2 or later (the "GPL"), or
  32  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  33  * in which case the provisions of the GPL or the LGPL are applicable instead
  34  * of those above. If you wish to allow use of your version of this file only
  35  * under the terms of either the GPL or the LGPL, and not to allow others to
  36  * use your version of this file under the terms of the MPL, indicate your
  37  * decision by deleting the provisions above and replace them with the notice
  38  * and other provisions required by the GPL or the LGPL. If you do not delete
  39  * the provisions above, a recipient may use your version of this file under
  40  * the terms of any one of the MPL, the GPL or the LGPL.
  41  *
  42  * ***** END LICENSE BLOCK ***** */
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <string.h>
  46 #include <errno.h>
  47 #ifdef HAVE_GETOPT_H
  48 #include <getopt.h>
  49 #else
  50 extern int  getopt(int argc, char *const *argv, const char *shortopts);
  51 extern char *optarg;
  52 extern int  optind;
  53 #ifdef XP_WIN32
  54 int optind=1;
  55 #endif
  56 #endif
  57 #include <time.h>
  58 #include "nsTraceMalloc.h"
  59 #include "tmreader.h"
  60
  61 static char *program;
  62
  63 static void my_tmevent_handler(tmreader *tmr, tmevent *event)
  64 {
  65     tmcallsite *callsite;
  66
  67     switch (event->type) {
  68       case TM_EVENT_REALLOC:
  69       case TM_EVENT_MALLOC:
  70       case TM_EVENT_CALLOC:
  71         for (callsite = tmreader_callsite(tmr, event->serial);
  72              callsite != &tmr->calltree_root; callsite = callsite->parent) {
  73             fprintf(stdout, "%s +%08X (%s:%d)\n",
  74                     (const char*)callsite->method->graphnode.entry.value,
  75                     callsite->offset,
  76                     callsite->method->sourcefile,
  77                     callsite->method->linenumber);
  78         }
  79         fprintf(stdout, "\n");
  80     }
  81 }
  82
  83 int main(int argc, char **argv)
  84 {
  85     int i, j, rv;
  86     tmreader *tmr;
  87     FILE *fp;
  88     time_t start;
  89
  90     program = *argv;
  91
  92     tmr = tmreader_new(program, NULL);
  93     if (!tmr) {
  94         perror(program);
  95         exit(1);
  96     }
  97
  98     start = time(NULL);
  99     fprintf(stdout, "%s starting at %s", program, ctime(&start));
 100     fflush(stdout);
 101
 102     argc -= optind;
 103     argv += optind;
 104     if (argc == 0) {
 105         if (tmreader_eventloop(tmr, "-", my_tmevent_handler) <= 0)
 106             exit(1);
 107     } else {
 108         for (i = j = 0; i < argc; i++) {
 109             fp = fopen(argv[i], "r");
 110             if (!fp) {
 111                 fprintf(stderr, "%s: can't open %s: %s\n",
 112                         program, argv[i], strerror(errno));
 113                 exit(1);
 114             }
 115             rv = tmreader_eventloop(tmr, argv[i], my_tmevent_handler);
 116             if (rv < 0)
 117                 exit(1);
 118             if (rv > 0)
 119                 j++;
 120             fclose(fp);
 121         }
 122         if (j == 0)
 123             exit(1);
 124     }
 125
 126     tmreader_destroy(tmr);
 127
 128     exit(0);
 129 }