Merge from vendor branch LIBPCAP:
[dragonfly.git] / lib / libc / gmon / gmon.c
1 /*-
2  * Copyright (c) 1983, 1992, 1993
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *      This product includes software developed by the University of
16  *      California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * $FreeBSD: src/lib/libc/gmon/gmon.c,v 1.8 2000/01/27 23:06:25 jasone Exp $
34  * $DragonFly: src/lib/libc/gmon/gmon.c,v 1.8 2005/11/13 01:18:20 swildner Exp $
35  *
36  * @(#)gmon.c   8.1 (Berkeley) 6/4/93
37  */
38
39 #include "namespace.h"
40 #include <sys/param.h>
41 #include <sys/time.h>
42 #include <sys/gmon.h>
43 #include <sys/sysctl.h>
44
45 #include <err.h>
46 #include <errno.h>
47 #include <stdio.h>
48 #include <fcntl.h>
49 #include <stdlib.h>
50 #include <string.h>
51 #include <unistd.h>
52 #include "un-namespace.h"
53
54 #if defined(__i386__) || defined(__amd64__)
55 extern char *minbrk asm (".minbrk");
56 #else
57 extern char *minbrk asm ("minbrk");
58 #endif
59
60 struct gmonparam _gmonparam = { GMON_PROF_OFF };
61
62 static int      s_scale;
63 /* see profil(2) where this is describe (incorrectly) */
64 #define         SCALE_1_TO_1    0x10000L
65
66 #define ERR(s) _write(2, s, sizeof(s))
67
68 void    moncontrol (int);
69 static int hertz (void);
70
71 void
72 monstartup(u_long lowpc, u_long highpc)
73 {
74         int o;
75         char *cp;
76         struct gmonparam *p = &_gmonparam;
77
78         /*
79          * round lowpc and highpc to multiples of the density we're using
80          * so the rest of the scaling (here and in gprof) stays in ints.
81          */
82         p->lowpc = ROUNDDOWN(lowpc, HISTFRACTION * sizeof(HISTCOUNTER));
83         p->highpc = ROUNDUP(highpc, HISTFRACTION * sizeof(HISTCOUNTER));
84         p->textsize = p->highpc - p->lowpc;
85         p->kcountsize = p->textsize / HISTFRACTION;
86         p->hashfraction = HASHFRACTION;
87         p->fromssize = p->textsize / HASHFRACTION;
88         p->tolimit = p->textsize * ARCDENSITY / 100;
89         if (p->tolimit < MINARCS)
90                 p->tolimit = MINARCS;
91         else if (p->tolimit > MAXARCS)
92                 p->tolimit = MAXARCS;
93         p->tossize = p->tolimit * sizeof(struct tostruct);
94
95         cp = sbrk(p->kcountsize + p->fromssize + p->tossize);
96         if (cp == (char *)-1) {
97                 ERR("monstartup: out of memory\n");
98                 return;
99         }
100 #ifdef notdef
101         bzero(cp, p->kcountsize + p->fromssize + p->tossize);
102 #endif
103         p->tos = (struct tostruct *)cp;
104         cp += p->tossize;
105         p->kcount = (u_short *)cp;
106         cp += p->kcountsize;
107         p->froms = (u_short *)cp;
108
109         minbrk = sbrk(0);
110         p->tos[0].link = 0;
111
112         o = p->highpc - p->lowpc;
113         if (p->kcountsize < o) {
114 #ifndef hp300
115                 s_scale = ((float)p->kcountsize / o ) * SCALE_1_TO_1;
116 #else /* avoid floating point */
117                 int quot = o / p->kcountsize;
118
119                 if (quot >= 0x10000)
120                         s_scale = 1;
121                 else if (quot >= 0x100)
122                         s_scale = 0x10000 / quot;
123                 else if (o >= 0x800000)
124                         s_scale = 0x1000000 / (o / (p->kcountsize >> 8));
125                 else
126                         s_scale = 0x1000000 / ((o << 8) / p->kcountsize);
127 #endif
128         } else
129                 s_scale = SCALE_1_TO_1;
130
131         moncontrol(1);
132 }
133
134 void
135 _mcleanup(void)
136 {
137         int fd;
138         int fromindex;
139         int endfrom;
140         u_long frompc;
141         int toindex;
142         struct rawarc rawarc;
143         struct gmonparam *p = &_gmonparam;
144         struct gmonhdr gmonhdr, *hdr;
145         struct clockinfo clockinfo;
146         char outname[128];
147         int mib[2];
148         size_t size;
149 #ifdef DEBUG
150         int log, len;
151         char buf[200];
152 #endif
153
154         if (p->state == GMON_PROF_ERROR)
155                 ERR("_mcleanup: tos overflow\n");
156
157         size = sizeof(clockinfo);
158         mib[0] = CTL_KERN;
159         mib[1] = KERN_CLOCKRATE;
160         if (sysctl(mib, 2, &clockinfo, &size, NULL, 0) < 0) {
161                 /*
162                  * Best guess
163                  */
164                 clockinfo.profhz = hertz();
165         } else if (clockinfo.profhz == 0) {
166                 if (clockinfo.hz != 0)
167                         clockinfo.profhz = clockinfo.hz;
168                 else
169                         clockinfo.profhz = hertz();
170         }
171
172         moncontrol(0);
173         snprintf(outname, sizeof(outname), "%s.gmon", getprogname());
174         fd = _open(outname, O_CREAT|O_TRUNC|O_WRONLY, 0666);
175         if (fd < 0) {
176                 warnx("_mcleanup: %s - %s",outname,strerror(errno));
177                 return;
178         }
179 #ifdef DEBUG
180         log = _open("gmon.log", O_CREAT|O_TRUNC|O_WRONLY, 0664);
181         if (log < 0) {
182                 perror("_mcleanup: gmon.log");
183                 return;
184         }
185         len = sprintf(buf, "[mcleanup1] kcount 0x%x ssiz %d\n",
186             p->kcount, p->kcountsize);
187         _write(log, buf, len);
188 #endif
189         hdr = (struct gmonhdr *)&gmonhdr;
190         hdr->lpc = p->lowpc;
191         hdr->hpc = p->highpc;
192         hdr->ncnt = p->kcountsize + sizeof(gmonhdr);
193         hdr->version = GMONVERSION;
194         hdr->profrate = clockinfo.profhz;
195         _write(fd, (char *)hdr, sizeof *hdr);
196         _write(fd, p->kcount, p->kcountsize);
197         endfrom = p->fromssize / sizeof(*p->froms);
198         for (fromindex = 0; fromindex < endfrom; fromindex++) {
199                 if (p->froms[fromindex] == 0)
200                         continue;
201
202                 frompc = p->lowpc;
203                 frompc += fromindex * p->hashfraction * sizeof(*p->froms);
204                 for (toindex = p->froms[fromindex]; toindex != 0;
205                      toindex = p->tos[toindex].link) {
206 #ifdef DEBUG
207                         len = sprintf(buf,
208                         "[mcleanup2] frompc 0x%x selfpc 0x%x count %d\n" ,
209                                 frompc, p->tos[toindex].selfpc,
210                                 p->tos[toindex].count);
211                         _write(log, buf, len);
212 #endif
213                         rawarc.raw_frompc = frompc;
214                         rawarc.raw_selfpc = p->tos[toindex].selfpc;
215                         rawarc.raw_count = p->tos[toindex].count;
216                         _write(fd, &rawarc, sizeof rawarc);
217                 }
218         }
219         _close(fd);
220 }
221
222 /*
223  * Control profiling
224  *      profiling is what mcount checks to see if
225  *      all the data structures are ready.
226  */
227 void
228 moncontrol(int mode)
229 {
230         struct gmonparam *p = &_gmonparam;
231
232         if (mode) {
233                 /* start */
234                 profil((char *)p->kcount, p->kcountsize, p->lowpc, s_scale);
235                 p->state = GMON_PROF_ON;
236         } else {
237                 /* stop */
238                 profil((char *)0, 0, 0, 0);
239                 p->state = GMON_PROF_OFF;
240         }
241 }
242
243 /*
244  * discover the tick frequency of the machine
245  * if something goes wrong, we return 0, an impossible hertz.
246  */
247 static int
248 hertz(void)
249 {
250         struct itimerval tim;
251
252         tim.it_interval.tv_sec = 0;
253         tim.it_interval.tv_usec = 1;
254         tim.it_value.tv_sec = 0;
255         tim.it_value.tv_usec = 0;
256         setitimer(ITIMER_REAL, &tim, 0);
257         setitimer(ITIMER_REAL, 0, &tim);
258         if (tim.it_interval.tv_usec < 2)
259                 return(0);
260         return (1000000 / tim.it_interval.tv_usec);
261 }