Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	usr.sbin/sa/usrdb.c	Lines:	76	127	59.8 %
Date:	2017-11-07	Branches:	33	65	50.8 %


/*	$OpenBSD: usrdb.c,v 1.10 2016/08/14 22:29:01 krw Exp $	*/
/*
 * Copyright (c) 1994 Christopher G. Demetriou
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *      This product includes software developed by Christopher G. Demetriou.
 * 4. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/types.h>
#include <sys/acct.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <pwd.h>
#include <stdio.h>
#include <string.h>
#include "extern.h"
#include "pathnames.h"

static int uid_compare(const DBT *, const DBT *);

static DB	*usracct_db;

int
usracct_init(void)
{
	DB *saved_usracct_db;
	BTREEINFO bti;
	int error;

	memset(&bti, 0, sizeof(bti));
	bti.compare = uid_compare;

	usracct_db = dbopen(NULL, O_RDWR, 0, DB_BTREE, &bti);
	if (usracct_db == NULL)
		return (-1);

	error = 0;
	if (!iflag) {
		DBT key, data;
		int serr, nerr;

		saved_usracct_db = dbopen(_PATH_USRACCT, O_RDONLY, 0, DB_BTREE,
		    &bti);
		if (saved_usracct_db == NULL) {
			error = (errno == ENOENT) ? 0 : -1;
			if (error)
				warn("retrieving user accounting summary");
			goto out;
		}

		serr = DB_SEQ(saved_usracct_db, &key, &data, R_FIRST);
		if (serr < 0) {
			warn("retrieving user accounting summary");
			error = -1;
			goto closeout;
		}
		while (serr == 0) {
			nerr = DB_PUT(usracct_db, &key, &data, 0);
			if (nerr < 0) {
				warn("initializing user accounting stats");
				error = -1;
				break;
			}

			serr = DB_SEQ(saved_usracct_db, &key, &data, R_NEXT);
			if (serr < 0) {
				warn("retrieving user accounting summary");
				error = -1;
				break;
			}
		}

closeout:
		if (DB_CLOSE(saved_usracct_db) < 0) {
			warn("closing user accounting summary");
			error = -1;
		}
	}

out:
	if (error != 0)
		usracct_destroy();
	return (error);
}

void
usracct_destroy(void)
{
	if (DB_CLOSE(usracct_db) < 0)
		warn("destroying user accounting stats");
}

int
usracct_add(const struct cmdinfo *ci)
{
	DBT key, data;
	struct userinfo newui;
	uid_t uid;
	int rv;

	uid = ci->ci_uid;
	key.data = &uid;
	key.size = sizeof(uid);

	rv = DB_GET(usracct_db, &key, &data, 0);
	if (rv < 0) {
		warn("get key %u from user accounting stats", uid);
		return (-1);
	} else if (rv == 0) {	/* it's there; copy whole thing */
		/* add the old data to the new data */
		memcpy(&newui, data.data, data.size);
		if (newui.ui_uid != uid) {
			warnx("key %u != expected record number %u",
			    newui.ui_uid, uid);
			warnx("inconsistent user accounting stats");
			return (-1);
		}
	} else {		/* it's not there; zero it and copy the key */
		memset(&newui, 0, sizeof(newui));
		newui.ui_uid = ci->ci_uid;
	}

	newui.ui_calls += ci->ci_calls;
	newui.ui_utime += ci->ci_utime;
	newui.ui_stime += ci->ci_stime;
	newui.ui_mem += ci->ci_mem;
	newui.ui_io += ci->ci_io;

	data.data = &newui;
	data.size = sizeof(newui);
	rv = DB_PUT(usracct_db, &key, &data, 0);
	if (rv < 0) {
		warn("add key %u to user accounting stats", uid);
		return (-1);
	} else if (rv != 0) {
		warnx("DB_PUT returned 1");
		return (-1);
	}

	return (0);
}

int
usracct_update(void)
{
	DB *saved_usracct_db;
	DBT key, data;
	BTREEINFO bti;
	int error, serr, nerr;

	memset(&bti, 0, sizeof(bti));
	bti.compare = uid_compare;

	saved_usracct_db = dbopen(_PATH_USRACCT, O_RDWR|O_CREAT|O_TRUNC, 0644,
	    DB_BTREE, &bti);
	if (saved_usracct_db == NULL) {
		warn("creating user accounting summary");
		return (-1);
	}

	error = 0;

	serr = DB_SEQ(usracct_db, &key, &data, R_FIRST);
	if (serr < 0) {
		warn("retrieving user accounting stats");
		error = -1;
	}
	while (serr == 0) {
		nerr = DB_PUT(saved_usracct_db, &key, &data, 0);
		if (nerr < 0) {
			warn("saving user accounting summary");
			error = -1;
			break;
		}

		serr = DB_SEQ(usracct_db, &key, &data, R_NEXT);
		if (serr < 0) {
			warn("retrieving user accounting stats");
			error = -1;
			break;
		}
	}

	if (DB_SYNC(saved_usracct_db, 0) < 0) {
		warn("syncing process accounting summary");
		error = -1;
	}
	if (DB_CLOSE(saved_usracct_db) < 0) {
		warn("closing process accounting summary");
		error = -1;
	}
	return error;
}

void
usracct_print(void)
{
	DBT key, data;
	struct userinfo uistore, *ui = &uistore;
	double t;
	int rv;

	rv = DB_SEQ(usracct_db, &key, &data, R_FIRST);
	if (rv < 0)
		warn("retrieving user accounting stats");

	while (rv == 0) {
		memcpy(ui, data.data, sizeof(struct userinfo));

		printf("%-8s %9llu ",
		    user_from_uid(ui->ui_uid, 0), ui->ui_calls);

		t = (double) (ui->ui_utime + ui->ui_stime) /
		    (double) AHZ;
		if (t < 0.0001)		/* kill divide by zero */
			t = 0.0001;

		printf("%12.2f%s ", t / 60.0, "cpu");

		/* ui->ui_calls is always != 0 */
		if (dflag)
			printf("%12llu%s", ui->ui_io / ui->ui_calls, "avio");
		else
			printf("%12llu%s", ui->ui_io, "tio");

		/* t is always >= 0.0001; see above */
		if (kflag)
			printf("%12.0f%s", ui->ui_mem / t, "k");
		else
			printf("%12llu%s", ui->ui_mem, "k*sec");

		printf("\n");

		rv = DB_SEQ(usracct_db, &key, &data, R_NEXT);
		if (rv < 0)
			warn("retrieving user accounting stats");
	}
}

static int
uid_compare(const DBT *k1, const DBT *k2)
{
	uid_t d1, d2;

	memcpy(&d1, k1->data, sizeof(d1));
	memcpy(&d2, k2->data, sizeof(d2));

	if (d1 < d2)
		return -1;
	else if (d1 == d2)
		return 0;
	else
		return 1;
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: usrdb.c,v 1.10 2016/08/14 22:29:01 krw Exp $ */
2			/*
3			* Copyright (c) 1994 Christopher G. Demetriou
4			* All rights reserved.
5			*
6			* Redistribution and use in source and binary forms, with or without
7			* modification, are permitted provided that the following conditions
8			* are met:
9			* 1. Redistributions of source code must retain the above copyright
10			* notice, this list of conditions and the following disclaimer.
11			* 2. Redistributions in binary form must reproduce the above copyright
12			* notice, this list of conditions and the following disclaimer in the
13			* documentation and/or other materials provided with the distribution.
14			* 3. All advertising materials mentioning features or use of this software
15			* must display the following acknowledgement:
16			* This product includes software developed by Christopher G. Demetriou.
17			* 4. The name of the author may not be used to endorse or promote products
18			* derived from this software without specific prior written permission
19			*
20			* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
21			* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
22			* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
23			* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
24			* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
25			* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26			* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27			* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28			* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
29			* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30			*/
31
32			#include <sys/types.h>
33			#include <sys/acct.h>
34			#include <err.h>
35			#include <errno.h>
36			#include <fcntl.h>
37			#include <pwd.h>
38			#include <stdio.h>
39			#include <string.h>
40			#include "extern.h"
41			#include "pathnames.h"
42
43			static int uid_compare(const DBT , const DBT );
44
45			static DB *usracct_db;
46
47			int
48			usracct_init(void)
49			{
50			DB *saved_usracct_db;
51		20	BTREEINFO bti;
52			int error;
53
54		10	memset(&bti, 0, sizeof(bti));
55		10	bti.compare = uid_compare;
56
57		10	usracct_db = dbopen(NULL, O_RDWR, 0, DB_BTREE, &bti);
58	✗✓	10	if (usracct_db == NULL)
59			return (-1);
60
61			error = 0;
62	✓✗	10	if (!iflag) {
63		10	DBT key, data;
64			int serr, nerr;
65
66		10	saved_usracct_db = dbopen(_PATH_USRACCT, O_RDONLY, 0, DB_BTREE,
67			&bti);
68	✓✓	10	if (saved_usracct_db == NULL) {
69		1	error = (errno == ENOENT) ? 0 : -1;
70	✗✓	1	if (error)
71			warn("retrieving user accounting summary");
72		1	goto out;
73			}
74
75		9	serr = DB_SEQ(saved_usracct_db, &key, &data, R_FIRST);
76	✗✓	9	if (serr < 0) {
77			warn("retrieving user accounting summary");
78			error = -1;
79			goto closeout;
80			}
81	✓✓	112	while (serr == 0) {
82		103	nerr = DB_PUT(usracct_db, &key, &data, 0);
83	✗✓	103	if (nerr < 0) {
84			warn("initializing user accounting stats");
85			error = -1;
86			break;
87			}
88
89		103	serr = DB_SEQ(saved_usracct_db, &key, &data, R_NEXT);
90	✓✗	103	if (serr < 0) {
91			warn("retrieving user accounting summary");
92			error = -1;
93			break;
94			}
95			}
96
97			closeout:
98	✗✓	9	if (DB_CLOSE(saved_usracct_db) < 0) {
99			warn("closing user accounting summary");
100			error = -1;
101			}
102	✗✓✓	29	}
103
104			out:
105	✗✓	10	if (error != 0)
106			usracct_destroy();
107		10	return (error);
108		10	}
109
110			void
111			usracct_destroy(void)
112			{
113	✗✓	20	if (DB_CLOSE(usracct_db) < 0)
114			warn("destroying user accounting stats");
115		10	}
116
117			int
118			usracct_add(const struct cmdinfo *ci)
119			{
120		1363220	DBT key, data;
121		681610	struct userinfo newui;
122		681610	uid_t uid;
123			int rv;
124
125		681610	uid = ci->ci_uid;
126		681610	key.data = &uid;
127		681610	key.size = sizeof(uid);
128
129		681610	rv = DB_GET(usracct_db, &key, &data, 0);
130	✗✓	681610	if (rv < 0) {
131			warn("get key %u from user accounting stats", uid);
132			return (-1);
133	✓✓	681610	} else if (rv == 0) { /* it's there; copy whole thing */
134			/* add the old data to the new data */
135		681595	memcpy(&newui, data.data, data.size);
136	✗✓	681595	if (newui.ui_uid != uid) {
137			warnx("key %u != expected record number %u",
138			newui.ui_uid, uid);
139			warnx("inconsistent user accounting stats");
140			return (-1);
141			}
142			} else { /* it's not there; zero it and copy the key */
143		15	memset(&newui, 0, sizeof(newui));
144		15	newui.ui_uid = ci->ci_uid;
145			}
146
147		681610	newui.ui_calls += ci->ci_calls;
148		681610	newui.ui_utime += ci->ci_utime;
149		681610	newui.ui_stime += ci->ci_stime;
150		681610	newui.ui_mem += ci->ci_mem;
151		681610	newui.ui_io += ci->ci_io;
152
153		681610	data.data = &newui;
154		681610	data.size = sizeof(newui);
155		681610	rv = DB_PUT(usracct_db, &key, &data, 0);
156	✗✓	681610	if (rv < 0) {
157			warn("add key %u to user accounting stats", uid);
158			return (-1);
159	✗✓	681610	} else if (rv != 0) {
160			warnx("DB_PUT returned 1");
161			return (-1);
162			}
163
164		681610	return (0);
165		681610	}
166
167			int
168			usracct_update(void)
169			{
170			DB *saved_usracct_db;
171		20	DBT key, data;
172		10	BTREEINFO bti;
173			int error, serr, nerr;
174
175		10	memset(&bti, 0, sizeof(bti));
176		10	bti.compare = uid_compare;
177
178		10	saved_usracct_db = dbopen(_PATH_USRACCT, O_RDWR\|O_CREAT\|O_TRUNC, 0644,
179			DB_BTREE, &bti);
180	✗✓	10	if (saved_usracct_db == NULL) {
181			warn("creating user accounting summary");
182			return (-1);
183			}
184
185			error = 0;
186
187		10	serr = DB_SEQ(usracct_db, &key, &data, R_FIRST);
188	✗✓	10	if (serr < 0) {
189			warn("retrieving user accounting stats");
190			error = -1;
191			}
192	✓✓	128	while (serr == 0) {
193		118	nerr = DB_PUT(saved_usracct_db, &key, &data, 0);
194	✗✓	118	if (nerr < 0) {
195			warn("saving user accounting summary");
196			error = -1;
197			break;
198			}
199
200		118	serr = DB_SEQ(usracct_db, &key, &data, R_NEXT);
201	✓✗	118	if (serr < 0) {
202			warn("retrieving user accounting stats");
203			error = -1;
204			break;
205			}
206			}
207
208	✗✓	10	if (DB_SYNC(saved_usracct_db, 0) < 0) {
209			warn("syncing process accounting summary");
210			error = -1;
211			}
212	✗✓	10	if (DB_CLOSE(saved_usracct_db) < 0) {
213			warn("closing process accounting summary");
214			error = -1;
215			}
216		10	return error;
217		10	}
218
219			void
220			usracct_print(void)
221			{
222			DBT key, data;
223			struct userinfo uistore, *ui = &uistore;
224			double t;
225			int rv;
226
227			rv = DB_SEQ(usracct_db, &key, &data, R_FIRST);
228			if (rv < 0)
229			warn("retrieving user accounting stats");
230
231			while (rv == 0) {
232			memcpy(ui, data.data, sizeof(struct userinfo));
233
234			printf("%-8s %9llu ",
235			user_from_uid(ui->ui_uid, 0), ui->ui_calls);
236
237			t = (double) (ui->ui_utime + ui->ui_stime) /
238			(double) AHZ;
239			if (t < 0.0001) /* kill divide by zero */
240			t = 0.0001;
241
242			printf("%12.2f%s ", t / 60.0, "cpu");
243
244			/* ui->ui_calls is always != 0 */
245			if (dflag)
246			printf("%12llu%s", ui->ui_io / ui->ui_calls, "avio");
247			else
248			printf("%12llu%s", ui->ui_io, "tio");
249
250			/* t is always >= 0.0001; see above */
251			if (kflag)
252			printf("%12.0f%s", ui->ui_mem / t, "k");
253			else
254			printf("%12llu%s", ui->ui_mem, "k*sec");
255
256			printf("\n");
257
258			rv = DB_SEQ(usracct_db, &key, &data, R_NEXT);
259			if (rv < 0)
260			warn("retrieving user accounting stats");
261			}
262			}
263
264			static int
265			uid_compare(const DBT k1, const DBT k2)
266			{
267			uid_t d1, d2;
268
269		10601734	memcpy(&d1, k1->data, sizeof(d1));
270		5300867	memcpy(&d2, k2->data, sizeof(d2));
271
272	✓✓	5300867	if (d1 < d2)
273		3923009	return -1;
274	✓✓	1377858	else if (d1 == d2)
275		1363190	return 0;
276			else
277		14668	return 1;
278		5300867	}