Take options to control memprofstat.
[kopensolaris-gnu/glibc.git] / 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 $"Try \`memprof --help' for more information."
27   exit 1
28 }
29
30 do_help() {
31   echo $"Usage: memprof [OPTION]... PROGRAM [PROGRAMOPTION...]
32 Profile memory usage of PROGRAM.
33
34       --help              print this help and exit
35       --version           print version information and exit
36       --progname          name of the program file to profile
37       --png=FILE          generate PNG graphic and store it in FILE
38       --data=FILE         generate binary data file and store it in FILE
39       --unbuffered        don't buffer output
40       --buffer=SIZE       collect SIZE entries before writing them out
41       --no-timer          don't collect additional information though timer
42
43    The following options only apply when generating graphical output:
44       --time-based        make graph linear in time
45       --total             also draw graph of total memory use
46       --title=STRING      use STRING as title of the graph
47       --x-size=SIZE       make graphic SIZE pixels wide
48       --y-size=SIZE       make graphic SIZE pixels high
49 Report bugs using the \`glibcbug' script to <bugs@gnu.org>."
50   exit 0
51 }
52
53 do_version() {
54   echo 'memprof (GNU libc) @VERSION@'
55   echo $"Copyright (C) 1999 Free Software Foundation, Inc.
56 This is free software; see the source for copying conditions.  There is NO
57 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
58 Written by Ulrich Drepper."
59   exit 0
60 }
61
62 # Process arguments.  But stop as soon as the program name is found.
63 while test $# -gt 0; do
64   case "$1" in
65   --v | --ve | --ver | --vers | --versi | --versio | --version)
66     do_version
67     ;;
68   --h | --he | --hel | --help)
69     do_help
70     ;;
71   --pr | --pro | --prog | --progn | --progna | --prognam | --progname)
72     if test $# -eq 1; then
73       usage
74     fi
75     shift
76     progname="$1"
77     ;;
78   --pr=* | --pro=* | --prog=* | --progn=* | --progna=* | --prognam=* | --progname=*)
79     progname=${1##*=}
80     ;;
81   --pn | --png)
82     if test $# -eq 1; then
83       usage
84     fi
85     shift
86     png="$1"
87     ;;
88   --pn=* | --png=*)
89     png=${1##*=}
90     ;;
91   --d | --da | --dat | --data)
92     if test $# -eq 1; then
93       usage
94     fi
95     shift
96     data="$1"
97     ;;
98   --d=* | --da=* | --dat=* | --data=*)
99     data=${1##*=}
100     ;;
101   --u | --un | --unb | --unbu | --unbuf | --unbuff | --unbuffe | --unbuffer | --unbuffere | --unbuffered)
102     buffer=1
103     ;;
104   --b | --bu | --buf | --buff | --buffe | --buffer)
105     if test $# -eq 1; then
106       usage
107     fi
108     shift
109     buffer="$1"
110     ;;
111   --b=* | --bu=* | --buf=* | --buff=* | --buffe=* | --buffer=*)
112     buffer=${1##*=}
113     ;;
114   --n | --no | --no- | --no-t | --no-ti | --no-tim | --no-time | --no-timer)
115     notimer=yes
116     ;;
117   --tim | --time | --time- | --time-b | --time-ba | --time-bas | --time-base | --time-based)
118     memprofstat_args="$memprofstat_args -t"
119     ;;
120   --to | --tot | --tota | --total)
121     memprofstat_args="$memprofstat_args -T"
122     ;;
123   --tit | --titl | --title)
124     if test $# -eq 1; then
125       usage
126     fi
127     shift
128     memprofstat_args="$memprofstat_args -s $1"
129     ;;
130   --tit=* | --titl=* | --title=*)
131     memprofstat_args="$memprofstat_args -s ${1##*=}"
132     ;;
133   --x | --x- | --x-s | --x-si | --x-siz | --x-size)
134     if test $# -eq 1; then
135       usage
136     fi
137     shift
138     memprofstat_args="$memprofstat_args -x $1"
139     ;;
140   --x=* | --x-=* | --x-s=* | --x-si=* | --x-siz=* | --x-size=*)
141     memprofstat_args="$memprofstat_args -x ${1##*=}"
142     ;;
143   --y | --y- | --y-s | --y-si | --y-siz | --y-size)
144     if test $# -eq 1; then
145       usage
146     fi
147     shift
148     memprofstat_args="$memprofstat_args -y $1"
149     ;;
150   --y=* | --y-=* | --y-s=* | --y-si=* | --y-siz=* | --y-size=*)
151     memprofstat_args="$memprofstat_args -y ${1##*=}"
152     ;;
153   --p | --p=* | --t | --t=* | --ti | --ti=*)
154     echo >&2 $"memprof: option \`${1##*=}' is ambiguous"
155     usage
156     ;;
157   --)
158     # Stop processing arguments.
159     shift
160     break
161     ;;
162   *)
163     # Unknown option.  This means the rest is the program name and parameters.
164     break
165     ;;
166   esac
167   shift
168 done
169
170 # See whether any arguments are left.
171 if test $# -le 0; then
172   usage
173 fi
174
175 # This will be in the environment.
176 add_env="LD_PRELOAD=$memprofso"
177
178 # Generate data file name.
179 if test -n "$data"; then
180   datafile="$data"
181 elif test -n "$png"; then
182   datafile=$(mktemp ${TMPDIR:-/tmp}/memprof.XXXXXX 2> /dev/null)
183   if test $? -ne 0; then
184     # Lame, but if there is no `mktemp' program the user cannot expect more.
185     datafile=${TMPDIR:-/tmp}/memprof.$$
186   fi
187 fi
188 if test -n "$datafile"; then
189   add_env="$add_env MEMPROF_OUTPUT=$datafile"
190 fi
191
192 # Set buffer size.
193 if test -n "$buffer"; then
194   add_env="$add_env MEMPROF_BUFFER_SIZE=$buffer"
195 fi
196
197 # Disable timers.
198 if test -n "$notimer"; then
199   add_env="$add_env MEMPROF_NO_TIMER=yes"
200 fi
201
202 # Execute the program itself.
203 eval $add_env $*
204 result=$?
205
206 # Generate the PNG data file is wanted and there is something to generate
207 # it from.
208 if test -n "$png" -a -s "$datafile"; then
209   # Append extension .png if it isn't already there.
210   if test $png = ${png%*.png}; then
211     png="$png.png"
212   fi
213   eval $memprofstat $memprofstat_args $datafile $png
214 fi
215
216 if test -z "$data" -a -n $datafile; then
217   rm -f $datafile
218 fi
219
220 exit $result
221 # Local Variables:
222 #  mode:ksh
223 # End: