[apple/system_cmds.git] / sa.tproj / usrdb.c

/*
 * 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.
 */

#ifndef lint
static const char rcsid[] =
  "$FreeBSD: src/usr.sbin/sa/usrdb.c,v 1.12 2002/07/15 16:05:15 des Exp $";
#endif /* not lint */

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

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

static DB	*usracct_db;

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

	bzero(&bti, 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()
{
	if (DB_CLOSE(usracct_db) < 0)
		warn("destroying user accounting stats");
}

int
usracct_add(ci)
	const struct cmdinfo *ci;
{
	DBT key, data;
	struct userinfo newui;
	u_long 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 %lu 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 */
		bcopy(data.data, &newui, data.size);
		if (newui.ui_uid != uid) {
			warnx("key %lu != expected record number %lu",
			    newui.ui_uid, uid);
			warnx("inconsistent user accounting stats");
			return (-1);
		}
	} else {		/* it's not there; zero it and copy the key */
		bzero(&newui, 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 %lu to user accounting stats", uid);
		return (-1);
	} else if (rv != 0) {
		warnx("DB_PUT returned 1");
		return (-1);
	}

	return (0);
}

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

	bzero(&bti, 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()
{
	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("%-*s %9ju ", MAXLOGNAME - 1,
		    user_from_uid(ui->ui_uid, 0), (uintmax_t)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("%12ju%s",
			    (uintmax_t)(ui->ui_io / ui->ui_calls), "avio");
		else
			printf("%12ju%s", (uintmax_t)ui->ui_io, "tio");

		/* t is always >= 0.0001; see above */
		if (kflag)
			printf("%12.0f%s", ui->ui_mem / t, "k");
		else
			printf("%12ju%s", (uintmax_t)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(k1, k2)
	const DBT *k1, *k2;
{
	u_long d1, d2;

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

	if (d1 < d2)
		return -1;
	else if (d1 == d2)
		return 0;
	else
		return 1;
}
Commit	Line	Data
2fc1e207 A	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	#ifndef lint
	32	static const char rcsid[] =
	33	"$FreeBSD: src/usr.sbin/sa/usrdb.c,v 1.12 2002/07/15 16:05:15 des Exp $";
	34	#endif /* not lint */
	35
	36	#include <sys/param.h>
	37	#include <sys/types.h>
	38	#include <sys/acct.h>
	39	#include <err.h>
	40	#include <errno.h>
	41	#include <fcntl.h>
	42	#include <pwd.h>
	43	#include <stdint.h>
	44	#include <stdio.h>
	45	#include <stdlib.h>
	46	#include <string.h>
	47	#include "extern.h"
	48	#include "pathnames.h"
	49
	50	static int uid_compare __P((const DBT , const DBT ));
	51
	52	static DB *usracct_db;
	53
	54	int
	55	usracct_init()
	56	{
	57	DB *saved_usracct_db;
	58	BTREEINFO bti;
	59	int error;
	60
	61	bzero(&bti, sizeof bti);
	62	bti.compare = uid_compare;
	63
	64	usracct_db = dbopen(NULL, O_RDWR, 0, DB_BTREE, &bti);
65	if (usracct_db == NULL)
66	return (-1);
67
68	error = 0;
69	if (!iflag) {
70	DBT key, data;
71	int serr, nerr;
72
73	saved_usracct_db = dbopen(_PATH_USRACCT, O_RDONLY, 0, DB_BTREE,
74	&bti);
75	if (saved_usracct_db == NULL) {
76	error = (errno == ENOENT) ? 0 : -1;
77	if (error)
78	warn("retrieving user accounting summary");
79	goto out;
80	}
81
82	serr = DB_SEQ(saved_usracct_db, &key, &data, R_FIRST);
83	if (serr < 0) {
84	warn("retrieving user accounting summary");
85	error = -1;
86	goto closeout;
87	}
88	while (serr == 0) {
89	nerr = DB_PUT(usracct_db, &key, &data, 0);
90	if (nerr < 0) {
91	warn("initializing user accounting stats");
92	error = -1;
93	break;
94	}
95
96	serr = DB_SEQ(saved_usracct_db, &key, &data, R_NEXT);
97	if (serr < 0) {
98	warn("retrieving user accounting summary");
99	error = -1;
100	break;
101	}
102	}
103
104	closeout:
105	if (DB_CLOSE(saved_usracct_db) < 0) {
106	warn("closing user accounting summary");
107	error = -1;
108	}
109	}
110
111	out:
112	if (error != 0)
113	usracct_destroy();
114	return (error);
115	}
116
117	void
118	usracct_destroy()
119	{
120	if (DB_CLOSE(usracct_db) < 0)
121	warn("destroying user accounting stats");
122	}
123
124	int
125	usracct_add(ci)
126	const struct cmdinfo *ci;
127	{
128	DBT key, data;
129	struct userinfo newui;
130	u_long uid;
131	int rv;
132
133	uid = ci->ci_uid;
134	key.data = &uid;
135	key.size = sizeof uid;
136
137	rv = DB_GET(usracct_db, &key, &data, 0);
138	if (rv < 0) {
139	warn("get key %lu from user accounting stats", uid);
140	return (-1);
141	} else if (rv == 0) { /* it's there; copy whole thing */
142	/* add the old data to the new data */
143	bcopy(data.data, &newui, data.size);
144	if (newui.ui_uid != uid) {
145	warnx("key %lu != expected record number %lu",
146	newui.ui_uid, uid);
147	warnx("inconsistent user accounting stats");
148	return (-1);
149	}
150	} else { /* it's not there; zero it and copy the key */
151	bzero(&newui, sizeof newui);
152	newui.ui_uid = ci->ci_uid;
153	}
154
155	newui.ui_calls += ci->ci_calls;
156	newui.ui_utime += ci->ci_utime;
157	newui.ui_stime += ci->ci_stime;
158	newui.ui_mem += ci->ci_mem;
159	newui.ui_io += ci->ci_io;
160
161	data.data = &newui;
162	data.size = sizeof newui;
163	rv = DB_PUT(usracct_db, &key, &data, 0);
164	if (rv < 0) {
165	warn("add key %lu to user accounting stats", uid);
166	return (-1);
167	} else if (rv != 0) {
168	warnx("DB_PUT returned 1");
169	return (-1);
170	}
171
172	return (0);
173	}
174
175	int
176	usracct_update()
177	{
178	DB *saved_usracct_db;
179	DBT key, data;
180	BTREEINFO bti;
181	int error, serr, nerr;
182
183	bzero(&bti, sizeof bti);
184	bti.compare = uid_compare;
185
186	saved_usracct_db = dbopen(_PATH_USRACCT, O_RDWR\|O_CREAT\|O_TRUNC, 0644,
187	DB_BTREE, &bti);
188	if (saved_usracct_db == NULL) {
189	warn("creating user accounting summary");
190	return (-1);
191	}
192
193	error = 0;
194
195	serr = DB_SEQ(usracct_db, &key, &data, R_FIRST);
196	if (serr < 0) {
197	warn("retrieving user accounting stats");
198	error = -1;
199	}
200	while (serr == 0) {
201	nerr = DB_PUT(saved_usracct_db, &key, &data, 0);
202	if (nerr < 0) {
203	warn("saving user accounting summary");
204	error = -1;
205	break;
206	}
207
208	serr = DB_SEQ(usracct_db, &key, &data, R_NEXT);
209	if (serr < 0) {
210	warn("retrieving user accounting stats");
211	error = -1;
212	break;
213	}
214	}
215
216	if (DB_SYNC(saved_usracct_db, 0) < 0) {
217	warn("syncing process accounting summary");
218	error = -1;
219	}
220	if (DB_CLOSE(saved_usracct_db) < 0) {
221	warn("closing process accounting summary");
222	error = -1;
223	}
224	return error;
225	}
226
227	void
228	usracct_print()
229	{
230	DBT key, data;
231	struct userinfo uistore, *ui = &uistore;
232	double t;
233	int rv;
234
235	rv = DB_SEQ(usracct_db, &key, &data, R_FIRST);
236	if (rv < 0)
237	warn("retrieving user accounting stats");
238
239	while (rv == 0) {
240	memcpy(ui, data.data, sizeof(struct userinfo));
241
242	printf("%-*s %9ju ", MAXLOGNAME - 1,
243	user_from_uid(ui->ui_uid, 0), (uintmax_t)ui->ui_calls);
244
245	t = (double) (ui->ui_utime + ui->ui_stime) /
246	(double) AHZ;
247	if (t < 0.0001) /* kill divide by zero */
248	t = 0.0001;
249
250	printf("%12.2f%s ", t / 60.0, "cpu");
251
252	/* ui->ui_calls is always != 0 */
253	if (dflag)
254	printf("%12ju%s",
255	(uintmax_t)(ui->ui_io / ui->ui_calls), "avio");
256	else
257	printf("%12ju%s", (uintmax_t)ui->ui_io, "tio");
258
259	/* t is always >= 0.0001; see above */
260	if (kflag)
261	printf("%12.0f%s", ui->ui_mem / t, "k");
262	else
263	printf("%12ju%s", (uintmax_t)ui->ui_mem, "k*sec");
264
265	printf("\n");
266
267	rv = DB_SEQ(usracct_db, &key, &data, R_NEXT);
268	if (rv < 0)
269	warn("retrieving user accounting stats");
270	}
271	}
272
273	static int
274	uid_compare(k1, k2)
275	const DBT k1, k2;
276	{
277	u_long d1, d2;
278
279	bcopy(k1->data, &d1, sizeof d1);
280	bcopy(k2->data, &d2, sizeof d2);
281
282	if (d1 < d2)
283	return -1;
284	else if (d1 == d2)
285	return 0;
286	else
287	return 1;
288	}