malloc/memprof.sh

   1 #! @BASH@
   2 # Copyright (C) 1999 Free Software Foundation, Inc.
   3 # This file is part of the GNU C Library.
   4 # Contributed by Ulrich Drepper <drepper@gnu.org>, 1999.
   5
   6 # The GNU C Library is free software; you can redistribute it and/or
   7 # modify it under the terms of the GNU Library General Public License as
   8 # published by the Free Software Foundation; either version 2 of the
   9 # License, or (at your option) any later version.
  10
  11 # The GNU C Library is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 # Library General Public License for more details.
  15
  16 # You should have received a copy of the GNU Library General Public
  17 # License along with the GNU C Library; see the file COPYING.LIB.  If not,
  18 # write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19 # Boston, MA 02111-1307, USA.
  20
  21 memprofso=@LIBDIR@/libmemprof.so
  22 memprofstat=@BINDIR@/memprofstat
  23
  24 # Print usage message.
  25 do_usage() {
  26   echo >&2 $"Try \`memprof --help' for more information."
  27   exit 1
  28 }
  29
  30 # Message for missing argument.
  31 do_missing_arg() {
  32   echo >&2 $"memprof: option \`$1' requires an argument"
  33   do_usage
  34 }
  35
  36 # Print help message
  37 do_help() {
  38   echo $"Usage: memprof [OPTION]... PROGRAM [PROGRAMOPTION]...
  39 Profile memory usage of PROGRAM.
  40
  41    -n,--progname=NAME     Name of the program file to profile
  42    -p,--png=FILE          Generate PNG graphic and store it in FILE
  43    -d,--data=FILE         Generate binary data file and store it in FILE
  44    -u,--unbuffered        Don't buffer output
  45    -b,--buffer=SIZE       Collect SIZE entries before writing them out
  46       --no-timer          Don't collect additional information though timer
  47
  48    -?,--help              Print this help and exit
  49       --usage             Give a short usage message
  50    -V,--version           Print version information and exit
  51
  52  The following options only apply when generating graphical output:
  53    -t,--time-based        Make graph linear in time
  54    -T,--total             Also draw graph of total memory use
  55       --title=STRING      Use STRING as title of the graph
  56    -x,--x-size=SIZE       Make graphic SIZE pixels wide
  57    -y,--y-size=SIZE       Make graphic SIZE pixels high
  58
  59 Mandatory arguments to long options are also mandatory for any corresponding
  60 short options.
  61
  62 Report bugs using the \`glibcbug' script to <bugs@gnu.org>."
  63   exit 0
  64 }
  65
  66 do_version() {
  67   echo 'memprof (GNU libc) @VERSION@'
  68   echo $"Copyright (C) 1999 Free Software Foundation, Inc.
  69 This is free software; see the source for copying conditions.  There is NO
  70 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  71 Written by Ulrich Drepper."
  72   exit 0
  73 }
  74
  75 # Process arguments.  But stop as soon as the program name is found.
  76 while test $# -gt 0; do
  77   case "$1" in
  78   -V | --v | --ve | --ver | --vers | --versi | --versio | --version)
  79     do_version
  80     ;;
  81   -\? | --h | --he | --hel | --help)
  82     do_help
  83     ;;
  84   --us | --usa | --usag | --usage)
  85     echo $"Syntax: memprof [--data=FILE] [--progname=NAME] [--png=FILE] [--unbuffered]
  86             [--buffer=SIZE] [--no-timer] [--time-based] [--total]
  87             [--title=STRING] [--x-size=SIZE] [--y-size=SIZE]
  88             PROGRAM [PROGRAMOPTION]..."
  89     exit 0
  90     ;;
  91   -n | --pr | --pro | --prog | --progn | --progna | --prognam | --progname)
  92     if test $# -eq 1; then
  93       do_missing_arg $1
  94     fi
  95     shift
  96     progname="$1"
  97     ;;
  98   --pr=* | --pro=* | --prog=* | --progn=* | --progna=* | --prognam=* | --progname=*)
  99     progname=${1##*=}
 100     ;;
 101   -p | --pn | --png)
 102     if test $# -eq 1; then
 103       do_missing_arg $1
 104     fi
 105     shift
 106     png="$1"
 107     ;;
 108   --pn=* | --png=*)
 109     png=${1##*=}
 110     ;;
 111   -d | --d | --da | --dat | --data)
 112     if test $# -eq 1; then
 113       do_missing_arg $1
 114     fi
 115     shift
 116     data="$1"
 117     ;;
 118   --d=* | --da=* | --dat=* | --data=*)
 119     data=${1##*=}
 120     ;;
 121   -u | --un | --unb | --unbu | --unbuf | --unbuff | --unbuffe | --unbuffer | --unbuffere | --unbuffered)
 122     buffer=1
 123     ;;
 124   -b | --b | --bu | --buf | --buff | --buffe | --buffer)
 125     if test $# -eq 1; then
 126       do_missing_arg $1
 127     fi
 128     shift
 129     buffer="$1"
 130     ;;
 131   --b=* | --bu=* | --buf=* | --buff=* | --buffe=* | --buffer=*)
 132     buffer=${1##*=}
 133     ;;
 134   --n | --no | --no- | --no-t | --no-ti | --no-tim | --no-time | --no-timer)
 135     notimer=yes
 136     ;;
 137   -t | --tim | --time | --time- | --time-b | --time-ba | --time-bas | --time-base | --time-based)
 138     memprofstat_args="$memprofstat_args -t"
 139     ;;
 140   -T | --to | --tot | --tota | --total)
 141     memprofstat_args="$memprofstat_args -T"
 142     ;;
 143   --tit | --titl | --title)
 144     if test $# -eq 1; then
 145       do_missing_arg $1
 146     fi
 147     shift
 148     memprofstat_args="$memprofstat_args -s $1"
 149     ;;
 150   --tit=* | --titl=* | --title=*)
 151     memprofstat_args="$memprofstat_args -s ${1##*=}"
 152     ;;
 153   -x | --x | --x- | --x-s | --x-si | --x-siz | --x-size)
 154     if test $# -eq 1; then
 155       do_missing_arg $1
 156     fi
 157     shift
 158     memprofstat_args="$memprofstat_args -x $1"
 159     ;;
 160   --x=* | --x-=* | --x-s=* | --x-si=* | --x-siz=* | --x-size=*)
 161     memprofstat_args="$memprofstat_args -x ${1##*=}"
 162     ;;
 163   -y | --y | --y- | --y-s | --y-si | --y-siz | --y-size)
 164     if test $# -eq 1; then
 165       do_missing_arg $1
 166     fi
 167     shift
 168     memprofstat_args="$memprofstat_args -y $1"
 169     ;;
 170   --y=* | --y-=* | --y-s=* | --y-si=* | --y-siz=* | --y-size=*)
 171     memprofstat_args="$memprofstat_args -y ${1##*=}"
 172     ;;
 173   --p | --p=* | --t | --t=* | --ti | --ti=* | --u)
 174     echo >&2 $"memprof: option \`${1##*=}' is ambiguous"
 175     do_usage
 176     ;;
 177   --)
 178     # Stop processing arguments.
 179     shift
 180     break
 181     ;;
 182   --*)
 183     echo >&2 $"memprof: unrecognized option \`$1'"
 184     do_usage
 185     ;;
 186   *)
 187     # Unknown option.  This means the rest is the program name and parameters.
 188     break
 189     ;;
 190   esac
 191   shift
 192 done
 193
 194 # See whether any arguments are left.
 195 if test $# -eq 0; then
 196   echo >&2 $"No program name given"
 197   do_usage
 198 fi
 199
 200 # This will be in the environment.
 201 add_env="LD_PRELOAD=$memprofso"
 202
 203 # Generate data file name.
 204 datafile=
 205 if test -n "$data"; then
 206   datafile="$data"
 207 elif test -n "$png"; then
 208   datafile=$(mktemp ${TMPDIR:-/tmp}/memprof.XXXXXX 2> /dev/null)
 209   if test $? -ne 0; then
 210     # Lame, but if there is no `mktemp' program the user cannot expect more.
 211     if test "$RANDOM" != "$RANDOM"; then
 212       datafile=${TMPDIR:-/tmp}/memprof.$RANDOM
 213     else
 214       datafile=${TMPDIR:-/tmp}/memprof.$$
 215     fi
 216   fi
 217 fi
 218 if test -n "$datafile"; then
 219   add_env="$add_env MEMPROF_OUTPUT=$datafile"
 220 fi
 221
 222 # Set buffer size.
 223 if test -n "$buffer"; then
 224   add_env="$add_env MEMPROF_BUFFER_SIZE=$buffer"
 225 fi
 226
 227 # Disable timers.
 228 if test -n "$notimer"; then
 229   add_env="$add_env MEMPROF_NO_TIMER=yes"
 230 fi
 231
 232 # Execute the program itself.
 233 eval $add_env '"$@"'
 234 result=$?
 235
 236 # Generate the PNG data file if wanted and there is something to generate
 237 # it from.
 238 if test -n "$png" -a -n "$datafile" -a -s "$datafile"; then
 239   # Append extension .png if it isn't already there.
 240   case $png in
 241   *.png) ;;
 242   *) png="$png.png" ;;
 243   esac
 244   $memprofstat $memprofstat_args "$datafile" "$png"
 245 fi
 246
 247 if test -z "$data" -a -n "$datafile"; then
 248   rm -f "$datafile"
 249 fi
 250
 251 exit $result
 252 # Local Variables:
 253 #  mode:ksh
 254 # End: