malloc/memusage.sh

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