]> granicus.if.org Git - strace/blob - count.c
Do not include <limits.h>
[strace] / count.c
1 /*
2  * Copyright (c) 1991, 1992 Paul Kranenburg <pk@cs.few.eur.nl>
3  * Copyright (c) 1993 Branko Lankester <branko@hacktic.nl>
4  * Copyright (c) 1993, 1994, 1995, 1996 Rick Sladkey <jrs@world.std.com>
5  * Copyright (c) 1996-1999 Wichert Akkerman <wichert@cistron.nl>
6  * Copyright (c) 1999 IBM Deutschland Entwicklung GmbH, IBM Corporation
7  *                     Linux for s390 port by D.J. Barrow
8  *                    <barrow_dj@mail.yahoo.com,djbarrow@de.ibm.com>
9  * Copyright (c) 2004 Roland McGrath <roland@redhat.com>
10  * Copyright (c) 2006 Dmitry V. Levin <ldv@altlinux.org>
11  * All rights reserved.
12  *
13  * Redistribution and use in source and binary forms, with or without
14  * modification, are permitted provided that the following conditions
15  * are met:
16  * 1. Redistributions of source code must retain the above copyright
17  *    notice, this list of conditions and the following disclaimer.
18  * 2. Redistributions in binary form must reproduce the above copyright
19  *    notice, this list of conditions and the following disclaimer in the
20  *    documentation and/or other materials provided with the distribution.
21  * 3. The name of the author may not be used to endorse or promote products
22  *    derived from this software without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
25  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
26  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
27  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
28  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
29  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
33  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34  *
35  *      $Id$
36  */
37
38 #include "defs.h"
39
40 struct call_counts {
41         struct timeval time;
42         int calls, errors;
43 };
44
45 static struct call_counts *countv[SUPPORTED_PERSONALITIES];
46 #define counts (countv[current_personality])
47
48 static struct timeval shortest = { 1000000, 0 };
49
50 void
51 count_syscall(struct tcb *tcp, struct timeval *tv)
52 {
53         if (!SCNO_IN_RANGE(tcp->scno))
54                 return;
55
56         if (!counts) {
57                 counts = calloc(nsyscalls, sizeof(*counts));
58                 if (!counts) {
59                         fprintf(stderr,
60                                 "strace: out of memory for call counts\n");
61                         exit(1);
62                 }
63         }
64
65         counts[tcp->scno].calls++;
66         if (tcp->u_error)
67                 counts[tcp->scno].errors++;
68
69         tv_sub(tv, tv, &tcp->etime);
70 #ifdef LINUX
71         if (tv_cmp(tv, &tcp->dtime) > 0) {
72                 static struct timeval one_tick;
73
74                 if (one_tick.tv_usec == 0) {
75                         /* Initialize it.  */
76                         struct itimerval it;
77
78                         memset(&it, 0, sizeof it);
79                         it.it_interval.tv_usec = 1;
80                         setitimer(ITIMER_REAL, &it, NULL);
81                         getitimer(ITIMER_REAL, &it);
82                         one_tick = it.it_interval;
83                 }
84
85                 if (tv_nz(&tcp->dtime))
86                         *tv = tcp->dtime;
87                 else if (tv_cmp(tv, &one_tick) > 0) {
88                         if (tv_cmp(&shortest, &one_tick) < 0)
89                                 *tv = shortest;
90                         else
91                                 *tv = one_tick;
92                 }
93         }
94 #endif /* LINUX */
95         if (tv_cmp(tv, &shortest) < 0)
96                 shortest = *tv;
97         tv_add(&counts[tcp->scno].time, &counts[tcp->scno].time, tv);
98 }
99
100 static int
101 time_cmp(void *a, void *b)
102 {
103         return -tv_cmp(&counts[*((int *) a)].time,
104                        &counts[*((int *) b)].time);
105 }
106
107 static int
108 syscall_cmp(void *a, void *b)
109 {
110         return strcmp(sysent[*((int *) a)].sys_name,
111                       sysent[*((int *) b)].sys_name);
112 }
113
114 static int
115 count_cmp(void *a, void *b)
116 {
117         int     m = counts[*((int *) a)].calls;
118         int     n = counts[*((int *) b)].calls;
119
120         return (m < n) ? 1 : (m > n) ? -1 : 0;
121 }
122
123 static int (*sortfun)();
124 static struct timeval overhead = { -1, -1 };
125
126 void
127 set_sortby(const char *sortby)
128 {
129         if (strcmp(sortby, "time") == 0)
130                 sortfun = time_cmp;
131         else if (strcmp(sortby, "calls") == 0)
132                 sortfun = count_cmp;
133         else if (strcmp(sortby, "name") == 0)
134                 sortfun = syscall_cmp;
135         else if (strcmp(sortby, "nothing") == 0)
136                 sortfun = NULL;
137         else {
138                 fprintf(stderr, "invalid sortby: `%s'\n", sortby);
139                 exit(1);
140         }
141 }
142
143 void set_overhead(int n)
144 {
145         overhead.tv_sec = n / 1000000;
146         overhead.tv_usec = n % 1000000;
147 }
148
149 static void
150 call_summary_pers(FILE *outf)
151 {
152         int     i, j;
153         int     call_cum, error_cum;
154         struct timeval tv_cum, dtv;
155         double  percent;
156         const char *dashes = "-------------------------";
157         char    error_str[16];
158         int    *sorted_count = calloc(sizeof(int), nsyscalls);
159
160         if (!sorted_count) {
161                 fprintf(stderr, "strace: out of memory for call summary\n");
162                 return;
163         }
164
165         call_cum = error_cum = tv_cum.tv_sec = tv_cum.tv_usec = 0;
166         if (overhead.tv_sec == -1) {
167                 tv_mul(&overhead, &shortest, 8);
168                 tv_div(&overhead, &overhead, 10);
169         }
170         for (i = 0; i < nsyscalls; i++) {
171                 sorted_count[i] = i;
172                 if (counts == NULL || counts[i].calls == 0)
173                         continue;
174                 tv_mul(&dtv, &overhead, counts[i].calls);
175                 tv_sub(&counts[i].time, &counts[i].time, &dtv);
176                 call_cum += counts[i].calls;
177                 error_cum += counts[i].errors;
178                 tv_add(&tv_cum, &tv_cum, &counts[i].time);
179         }
180         if (counts && sortfun)
181                 qsort((void *) sorted_count, nsyscalls, sizeof(int), sortfun);
182         fprintf(outf, "%6.6s %11.11s %11.11s %9.9s %9.9s %s\n",
183                 "% time", "seconds", "usecs/call",
184                 "calls", "errors", "syscall");
185         fprintf(outf, "%6.6s %11.11s %11.11s %9.9s %9.9s %-16.16s\n",
186                 dashes, dashes, dashes, dashes, dashes, dashes);
187         if (counts) {
188                 for (i = 0; i < nsyscalls; i++) {
189                         j = sorted_count[i];
190                         if (counts[j].calls == 0)
191                                 continue;
192                         tv_div(&dtv, &counts[j].time, counts[j].calls);
193                         if (counts[j].errors)
194                                 sprintf(error_str, "%d", counts[j].errors);
195                         else
196                                 error_str[0] = '\0';
197                         percent = (100.0 * tv_float(&counts[j].time)
198                                    / tv_float(&tv_cum));
199                         fprintf(outf, "%6.2f %11.6f %11ld %9d %9.9s %s\n",
200                                 percent, tv_float(&counts[j].time),
201                                 (long) 1000000 * dtv.tv_sec + dtv.tv_usec,
202                                 counts[j].calls,
203                                 error_str, sysent[j].sys_name);
204                 }
205         }
206         free(sorted_count);
207
208         fprintf(outf, "%6.6s %11.11s %11.11s %9.9s %9.9s %-16.16s\n",
209                 dashes, dashes, dashes, dashes, dashes, dashes);
210         if (error_cum)
211                 sprintf(error_str, "%d", error_cum);
212         else
213                 error_str[0] = '\0';
214         fprintf(outf, "%6.6s %11.6f %11.11s %9d %9.9s %s\n",
215                 "100.00", tv_float(&tv_cum), "",
216                 call_cum, error_str, "total");
217 }
218
219 void
220 call_summary(FILE *outf)
221 {
222         int i, old_pers = current_personality;
223
224         for (i = 0; i < SUPPORTED_PERSONALITIES; ++i) {
225                 if (!countv[i])
226                         continue;
227
228                 if (current_personality != i)
229                         set_personality(i);
230                 if (i)
231                         fprintf(outf,
232                                 "System call usage summary for %u bit mode:\n",
233                                 personality_wordsize[current_personality] * 8);
234                 call_summary_pers(outf);
235         }
236
237         if (old_pers != current_personality)
238                 set_personality(old_pers);
239 }