pdb.c revision 114601 
1/*
2 * Copyright (c) 1994 Christopher G. Demetriou
3 * 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 Christopher G. Demetriou.
16 * 4. The name of the author may not be used to endorse or promote products
17 *    derived from this software without specific prior written permission
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
20 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
21 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
22 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
23 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
24 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
28 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30
31#include <sys/cdefs.h>
32__FBSDID("$FreeBSD: head/usr.sbin/sa/pdb.c 114601 2003-05-03 21:06:42Z obrien $");
33
34#include <sys/types.h>
35#include <sys/acct.h>
36#include <err.h>
37#include <errno.h>
38#include <fcntl.h>
39#include <stdint.h>
40#include <stdio.h>
41#include <string.h>
42#include "extern.h"
43#include "pathnames.h"
44
45static int check_junk __P((struct cmdinfo *));
46static void add_ci __P((const struct cmdinfo *, struct cmdinfo *));
47static void print_ci __P((const struct cmdinfo *, const struct cmdinfo *));
48
49static DB	*pacct_db;
50
51int
52pacct_init()
53{
54	DB *saved_pacct_db;
55	int error;
56
57	pacct_db = dbopen(NULL, O_RDWR, 0, DB_BTREE, NULL);
58	if (pacct_db == NULL)
59		return (-1);
60
61	error = 0;
62	if (!iflag) {
63		DBT key, data;
64		int serr, nerr;
65
66		saved_pacct_db = dbopen(_PATH_SAVACCT, O_RDONLY, 0, DB_BTREE,
67		    NULL);
68		if (saved_pacct_db == NULL) {
69			error = errno == ENOENT ? 0 : -1;
70			if (error)
71				warn("retrieving process accounting summary");
72			goto out;
73		}
74
75		serr = DB_SEQ(saved_pacct_db, &key, &data, R_FIRST);
76		if (serr < 0) {
77			warn("retrieving process accounting summary");
78			error = -1;
79			goto closeout;
80		}
81		while (serr == 0) {
82			nerr = DB_PUT(pacct_db, &key, &data, 0);
83			if (nerr < 0) {
84				warn("initializing process accounting stats");
85				error = -1;
86				break;
87			}
88
89			serr = DB_SEQ(saved_pacct_db, &key, &data, R_NEXT);
90			if (serr < 0) {
91				warn("retrieving process accounting summary");
92				error = -1;
93				break;
94			}
95		}
96
97closeout:	if (DB_CLOSE(saved_pacct_db) < 0) {
98			warn("closing process accounting summary");
99			error = -1;
100		}
101	}
102
103out:	if (error != 0)
104		pacct_destroy();
105	return (error);
106}
107
108void
109pacct_destroy()
110{
111	if (DB_CLOSE(pacct_db) < 0)
112		warn("destroying process accounting stats");
113}
114
115int
116pacct_add(ci)
117	const struct cmdinfo *ci;
118{
119	DBT key, data;
120	struct cmdinfo newci;
121	char keydata[sizeof ci->ci_comm];
122	int rv;
123
124	bcopy(ci->ci_comm, &keydata, sizeof keydata);
125	key.data = &keydata;
126	key.size = strlen(keydata);
127
128	rv = DB_GET(pacct_db, &key, &data, 0);
129	if (rv < 0) {
130		warn("get key %s from process accounting stats", ci->ci_comm);
131		return (-1);
132	} else if (rv == 0) {	/* it's there; copy whole thing */
133		/* XXX compare size if paranoid */
134		/* add the old data to the new data */
135		bcopy(data.data, &newci, data.size);
136	} else {		/* it's not there; zero it and copy the key */
137		bzero(&newci, sizeof newci);
138		bcopy(key.data, newci.ci_comm, key.size);
139	}
140
141	add_ci(ci, &newci);
142
143	data.data = &newci;
144	data.size = sizeof newci;
145	rv = DB_PUT(pacct_db, &key, &data, 0);
146	if (rv < 0) {
147		warn("add key %s to process accounting stats", ci->ci_comm);
148		return (-1);
149	} else if (rv == 1) {
150		warnx("duplicate key %s in process accounting stats",
151		    ci->ci_comm);
152		return (-1);
153	}
154
155	return (0);
156}
157
158int
159pacct_update()
160{
161	DB *saved_pacct_db;
162	DBT key, data;
163	int error, serr, nerr;
164
165	saved_pacct_db = dbopen(_PATH_SAVACCT, O_RDWR|O_CREAT|O_TRUNC, 0644,
166	    DB_BTREE, NULL);
167	if (saved_pacct_db == NULL) {
168		warn("creating process accounting summary");
169		return (-1);
170	}
171
172	error = 0;
173
174	serr = DB_SEQ(pacct_db, &key, &data, R_FIRST);
175	if (serr < 0) {
176		warn("retrieving process accounting stats");
177		error = -1;
178	}
179	while (serr == 0) {
180		nerr = DB_PUT(saved_pacct_db, &key, &data, 0);
181		if (nerr < 0) {
182			warn("saving process accounting summary");
183			error = -1;
184			break;
185		}
186
187		serr = DB_SEQ(pacct_db, &key, &data, R_NEXT);
188		if (serr < 0) {
189			warn("retrieving process accounting stats");
190			error = -1;
191			break;
192		}
193	}
194
195	if (DB_SYNC(saved_pacct_db, 0) < 0) {
196		warn("syncing process accounting summary");
197		error = -1;
198	}
199	if (DB_CLOSE(saved_pacct_db) < 0) {
200		warn("closing process accounting summary");
201		error = -1;
202	}
203	return error;
204}
205
206void
207pacct_print()
208{
209	BTREEINFO bti;
210	DBT key, data, ndata;
211	DB *output_pacct_db;
212	struct cmdinfo *cip, ci, ci_total, ci_other, ci_junk;
213	int rv;
214
215	bzero(&ci_total, sizeof ci_total);
216	strcpy(ci_total.ci_comm, "");
217	bzero(&ci_other, sizeof ci_other);
218	strcpy(ci_other.ci_comm, "***other");
219	bzero(&ci_junk, sizeof ci_junk);
220	strcpy(ci_junk.ci_comm, "**junk**");
221
222	/*
223	 * Retrieve them into new DB, sorted by appropriate key.
224	 * At the same time, cull 'other' and 'junk'
225	 */
226	bzero(&bti, sizeof bti);
227	bti.compare = sa_cmp;
228	output_pacct_db = dbopen(NULL, O_RDWR, 0, DB_BTREE, &bti);
229	if (output_pacct_db == NULL) {
230		warn("couldn't sort process accounting stats");
231		return;
232	}
233
234	ndata.data = NULL;
235	ndata.size = 0;
236	rv = DB_SEQ(pacct_db, &key, &data, R_FIRST);
237	if (rv < 0)
238		warn("retrieving process accounting stats");
239	while (rv == 0) {
240		cip = (struct cmdinfo *) data.data;
241		bcopy(cip, &ci, sizeof ci);
242
243		/* add to total */
244		add_ci(&ci, &ci_total);
245
246		if (vflag && ci.ci_calls <= cutoff &&
247		    (fflag || check_junk(&ci))) {
248			/* put it into **junk** */
249			add_ci(&ci, &ci_junk);
250			goto next;
251		}
252		if (!aflag &&
253		    ((ci.ci_flags & CI_UNPRINTABLE) != 0 || ci.ci_calls <= 1)) {
254			/* put into ***other */
255			add_ci(&ci, &ci_other);
256			goto next;
257		}
258		rv = DB_PUT(output_pacct_db, &data, &ndata, 0);
259		if (rv < 0)
260			warn("sorting process accounting stats");
261
262next:		rv = DB_SEQ(pacct_db, &key, &data, R_NEXT);
263		if (rv < 0)
264			warn("retrieving process accounting stats");
265	}
266
267	/* insert **junk** and ***other */
268	if (ci_junk.ci_calls != 0) {
269		data.data = &ci_junk;
270		data.size = sizeof ci_junk;
271		rv = DB_PUT(output_pacct_db, &data, &ndata, 0);
272		if (rv < 0)
273			warn("sorting process accounting stats");
274	}
275	if (ci_other.ci_calls != 0) {
276		data.data = &ci_other;
277		data.size = sizeof ci_other;
278		rv = DB_PUT(output_pacct_db, &data, &ndata, 0);
279		if (rv < 0)
280			warn("sorting process accounting stats");
281	}
282
283	/* print out the total */
284	print_ci(&ci_total, &ci_total);
285
286	/* print out; if reversed, print first (smallest) first */
287	rv = DB_SEQ(output_pacct_db, &data, &ndata, rflag ? R_FIRST : R_LAST);
288	if (rv < 0)
289		warn("retrieving process accounting report");
290	while (rv == 0) {
291		cip = (struct cmdinfo *) data.data;
292		bcopy(cip, &ci, sizeof ci);
293
294		print_ci(&ci, &ci_total);
295
296		rv = DB_SEQ(output_pacct_db, &data, &ndata,
297		    rflag ? R_NEXT : R_PREV);
298		if (rv < 0)
299			warn("retrieving process accounting report");
300	}
301	DB_CLOSE(output_pacct_db);
302}
303
304static int
305check_junk(cip)
306	struct cmdinfo *cip;
307{
308	char *cp;
309	size_t len;
310
311	fprintf(stderr, "%s (%ju) -- ", cip->ci_comm, (uintmax_t)cip->ci_calls);
312	cp = fgetln(stdin, &len);
313
314	return (cp && (cp[0] == 'y' || cp[0] == 'Y')) ? 1 : 0;
315}
316
317static void
318add_ci(fromcip, tocip)
319	const struct cmdinfo *fromcip;
320	struct cmdinfo *tocip;
321{
322	tocip->ci_calls += fromcip->ci_calls;
323	tocip->ci_etime += fromcip->ci_etime;
324	tocip->ci_utime += fromcip->ci_utime;
325	tocip->ci_stime += fromcip->ci_stime;
326	tocip->ci_mem += fromcip->ci_mem;
327	tocip->ci_io += fromcip->ci_io;
328}
329
330static void
331print_ci(cip, totalcip)
332	const struct cmdinfo *cip, *totalcip;
333{
334	double t, c;
335	int uflow;
336
337	c = cip->ci_calls ? cip->ci_calls : 1;
338	t = (cip->ci_utime + cip->ci_stime) / (double) AHZ;
339	if (t < 0.01) {
340		t = 0.01;
341		uflow = 1;
342	} else
343		uflow = 0;
344
345	printf("%8ju ", (uintmax_t)cip->ci_calls);
346	if (cflag) {
347		if (cip != totalcip)
348			printf(" %4.2f%%  ",
349			    cip->ci_calls / (double) totalcip->ci_calls);
350		else
351			printf(" %4s   ", "");
352	}
353
354	if (jflag)
355		printf("%11.2fre ", cip->ci_etime / (double) (AHZ * c));
356	else
357		printf("%11.2fre ", cip->ci_etime / (60.0 * AHZ));
358	if (cflag) {
359		if (cip != totalcip)
360			printf(" %4.2f%%  ",
361			    cip->ci_etime / (double) totalcip->ci_etime);
362		else
363			printf(" %4s   ", "");
364	}
365
366	if (!lflag) {
367		if (jflag)
368			printf("%11.2fcp ", t / (double) cip->ci_calls);
369		else
370			printf("%11.2fcp ", t / 60.0);
371		if (cflag) {
372			if (cip != totalcip)
373				printf(" %4.2f%%  ",
374				    (cip->ci_utime + cip->ci_stime) / (double)
375				    (totalcip->ci_utime + totalcip->ci_stime));
376			else
377				printf(" %4s   ", "");
378		}
379	} else {
380		if (jflag)
381			printf("%11.2fu ", cip->ci_utime / (double) (AHZ * c));
382		else
383			printf("%11.2fu ", cip->ci_utime / (60.0 * AHZ));
384		if (cflag) {
385			if (cip != totalcip)
386				printf(" %4.2f%%  ", cip->ci_utime / (double) totalcip->ci_utime);
387			else
388				printf(" %4s   ", "");
389		}
390		if (jflag)
391			printf("%11.2fs ", cip->ci_stime / (double) (AHZ * c));
392		else
393			printf("%11.2fs ", cip->ci_stime / (60.0 * AHZ));
394		if (cflag) {
395			if (cip != totalcip)
396				printf(" %4.2f%%  ", cip->ci_stime / (double) totalcip->ci_stime);
397			else
398				printf(" %4s   ", "");
399		}
400	}
401
402	if (tflag) {
403		if (!uflow)
404			printf("%8.2fre/cp ",
405			    cip->ci_etime /
406			    (double) (cip->ci_utime + cip->ci_stime));
407		else
408			printf("*ignore*      ");
409	}
410
411	if (Dflag)
412		printf("%10jutio ", (uintmax_t)cip->ci_io);
413	else
414		printf("%8.0favio ", cip->ci_io / c);
415
416	if (Kflag)
417		printf("%10juk*sec ", (uintmax_t)cip->ci_mem);
418	else
419		printf("%8.0fk ", cip->ci_mem / t);
420
421	printf("  %s\n", cip->ci_comm);
422}
423