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