2 * Copyright (c) 1994 Christopher G. Demetriou
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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
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.
30 * $FreeBSD: src/usr.sbin/sa/usrdb.c,v 1.8.2.2 2000/10/28 02:28:20 gallatin Exp $
31 * $DragonFly: src/usr.sbin/sa/usrdb.c,v 1.3 2003/11/03 19:31:43 eirikn Exp $
34 #include <sys/param.h>
35 #include <sys/types.h>
44 #include "pathnames.h"
46 static int uid_compare(const DBT
*, const DBT
*);
48 static DB
*usracct_db
;
57 bzero(&bti
, sizeof bti
);
58 bti
.compare
= uid_compare
;
60 usracct_db
= dbopen(NULL
, O_RDWR
, 0, DB_BTREE
, &bti
);
61 if (usracct_db
== NULL
)
69 saved_usracct_db
= dbopen(_PATH_USRACCT
, O_RDONLY
, 0, DB_BTREE
,
71 if (saved_usracct_db
== NULL
) {
72 error
= (errno
== ENOENT
) ? 0 : -1;
74 warn("retrieving user accounting summary");
78 serr
= DB_SEQ(saved_usracct_db
, &key
, &data
, R_FIRST
);
80 warn("retrieving user accounting summary");
85 nerr
= DB_PUT(usracct_db
, &key
, &data
, 0);
87 warn("initializing user accounting stats");
92 serr
= DB_SEQ(saved_usracct_db
, &key
, &data
, R_NEXT
);
94 warn("retrieving user accounting summary");
101 if (DB_CLOSE(saved_usracct_db
) < 0) {
102 warn("closing user accounting summary");
116 if (DB_CLOSE(usracct_db
) < 0)
117 warn("destroying user accounting stats");
122 const struct cmdinfo
*ci
;
125 struct userinfo newui
;
131 key
.size
= sizeof uid
;
133 rv
= DB_GET(usracct_db
, &key
, &data
, 0);
135 warn("get key %lu from user accounting stats", uid
);
137 } else if (rv
== 0) { /* it's there; copy whole thing */
138 /* add the old data to the new data */
139 bcopy(data
.data
, &newui
, data
.size
);
140 if (newui
.ui_uid
!= uid
) {
141 warnx("key %lu != expected record number %lu",
143 warnx("inconsistent user accounting stats");
146 } else { /* it's not there; zero it and copy the key */
147 bzero(&newui
, sizeof newui
);
148 newui
.ui_uid
= ci
->ci_uid
;
151 newui
.ui_calls
+= ci
->ci_calls
;
152 newui
.ui_utime
+= ci
->ci_utime
;
153 newui
.ui_stime
+= ci
->ci_stime
;
154 newui
.ui_mem
+= ci
->ci_mem
;
155 newui
.ui_io
+= ci
->ci_io
;
158 data
.size
= sizeof newui
;
159 rv
= DB_PUT(usracct_db
, &key
, &data
, 0);
161 warn("add key %lu to user accounting stats", uid
);
163 } else if (rv
!= 0) {
164 warnx("DB_PUT returned 1");
174 DB
*saved_usracct_db
;
177 int error
, serr
, nerr
;
179 bzero(&bti
, sizeof bti
);
180 bti
.compare
= uid_compare
;
182 saved_usracct_db
= dbopen(_PATH_USRACCT
, O_RDWR
|O_CREAT
|O_TRUNC
, 0644,
184 if (saved_usracct_db
== NULL
) {
185 warn("creating user accounting summary");
191 serr
= DB_SEQ(usracct_db
, &key
, &data
, R_FIRST
);
193 warn("retrieving user accounting stats");
197 nerr
= DB_PUT(saved_usracct_db
, &key
, &data
, 0);
199 warn("saving user accounting summary");
204 serr
= DB_SEQ(usracct_db
, &key
, &data
, R_NEXT
);
206 warn("retrieving user accounting stats");
212 if (DB_SYNC(saved_usracct_db
, 0) < 0) {
213 warn("syncing process accounting summary");
216 if (DB_CLOSE(saved_usracct_db
) < 0) {
217 warn("closing process accounting summary");
227 struct userinfo uistore
, *ui
= &uistore
;
231 rv
= DB_SEQ(usracct_db
, &key
, &data
, R_FIRST
);
233 warn("retrieving user accounting stats");
236 memcpy(ui
, data
.data
, sizeof(struct userinfo
));
238 printf("%-*s %9qu ", MAXLOGNAME
- 1,
239 user_from_uid(ui
->ui_uid
, 0), ui
->ui_calls
);
241 t
= (double) (ui
->ui_utime
+ ui
->ui_stime
) /
243 if (t
< 0.0001) /* kill divide by zero */
246 printf("%12.2f%s ", t
/ 60.0, "cpu");
248 /* ui->ui_calls is always != 0 */
250 printf("%12qu%s", ui
->ui_io
/ ui
->ui_calls
, "avio");
252 printf("%12qu%s", ui
->ui_io
, "tio");
254 /* t is always >= 0.0001; see above */
256 printf("%12.0f%s", ui
->ui_mem
/ t
, "k");
258 printf("%12qu%s", ui
->ui_mem
, "k*sec");
262 rv
= DB_SEQ(usracct_db
, &key
, &data
, R_NEXT
);
264 warn("retrieving user accounting stats");
274 bcopy(k1
->data
, &d1
, sizeof d1
);
275 bcopy(k2
->data
, &d2
, sizeof d2
);