Commit | Line | Data |
---|---|---|
7bfd21d0 GR |
1 | /* |
2 | * Copyright (c) 1994 Christopher G. Demetriou. | |
3 | * @(#)Copyright (c) 1994, Simon J. Gerraty. | |
4 | * | |
5 | * This is free software. It comes with NO WARRANTY. | |
6 | * Permission to use, modify and distribute this source code | |
7 | * is granted subject to the following conditions. | |
8 | * 1/ that the above copyright notice and this notice | |
9 | * are preserved in all copies and that due credit be given | |
10 | * to the author. | |
11 | * 2/ that any changes to this code are clearly commented | |
12 | * as such so that the author does not get blamed for bugs | |
13 | * other than his own. | |
14 | */ | |
15 | ||
16 | #ifndef lint | |
17 | static char rcsid[] = "$Id: ac.c,v 1.3 1994/05/01 04:39:35 cgd Exp $"; | |
18 | #endif | |
19 | ||
20 | #include <sys/types.h> | |
21 | #include <sys/file.h> | |
22 | #include <sys/time.h> | |
23 | #include <err.h> | |
24 | #include <errno.h> | |
25 | #include <pwd.h> | |
26 | #include <stdio.h> | |
27 | #include <stdlib.h> | |
28 | #include <string.h> | |
29 | #include <utmp.h> | |
30 | #include <unistd.h> | |
31 | ||
32 | /* | |
33 | * this is for our list of currently logged in sessions | |
34 | */ | |
35 | struct utmp_list { | |
36 | struct utmp_list *next; | |
37 | struct utmp usr; | |
38 | }; | |
39 | ||
40 | /* | |
41 | * this is for our list of users that are accumulating time. | |
42 | */ | |
43 | struct user_list { | |
44 | struct user_list *next; | |
45 | char name[UT_NAMESIZE+1]; | |
46 | time_t secs; | |
47 | }; | |
48 | ||
49 | /* | |
50 | * this is for chosing whether to ignore a login | |
51 | */ | |
52 | struct tty_list { | |
53 | struct tty_list *next; | |
54 | char name[UT_LINESIZE+3]; | |
55 | int len; | |
56 | int ret; | |
57 | }; | |
58 | ||
59 | /* | |
60 | * globals - yes yuk | |
61 | */ | |
62 | #ifdef CONSOLE_TTY | |
63 | static char *Console = CONSOLE_TTY; | |
64 | #endif | |
65 | static time_t Total = 0; | |
66 | static time_t FirstTime = 0; | |
67 | static int Flags = 0; | |
68 | static struct user_list *Users = NULL; | |
69 | static struct tty_list *Ttys = NULL; | |
70 | ||
71 | #define NEW(type) (type *)malloc(sizeof (type)) | |
72 | ||
73 | #define AC_W 1 /* not _PATH_WTMP */ | |
74 | #define AC_D 2 /* daily totals (ignore -p) */ | |
75 | #define AC_P 4 /* per-user totals */ | |
76 | #define AC_U 8 /* specified users only */ | |
77 | #define AC_T 16 /* specified ttys only */ | |
78 | ||
79 | #ifdef DEBUG | |
80 | static int Debug = 0; | |
81 | #endif | |
82 | ||
83 | int main __P((int, char **)); | |
84 | int ac __P((FILE *)); | |
85 | struct tty_list *add_tty __P((char *)); | |
86 | int do_tty __P((char *)); | |
87 | FILE *file __P((char *)); | |
88 | struct utmp_list *log_in __P((struct utmp_list *, struct utmp *)); | |
89 | struct utmp_list *log_out __P((struct utmp_list *, struct utmp *)); | |
90 | int on_console __P((struct utmp_list *)); | |
91 | void show __P((char *, time_t)); | |
92 | void show_today __P((struct user_list *, struct utmp_list *, | |
93 | time_t)); | |
94 | void show_users __P((struct user_list *)); | |
95 | struct user_list *update_user __P((struct user_list *, char *, time_t)); | |
96 | void usage __P((void)); | |
97 | ||
98 | /* | |
99 | * open wtmp or die | |
100 | */ | |
101 | FILE * | |
102 | file(name) | |
103 | char *name; | |
104 | { | |
105 | FILE *fp; | |
106 | ||
107 | if ((fp = fopen(name, "r")) == NULL) | |
108 | err(1, "%s", name); | |
109 | /* in case we want to discriminate */ | |
110 | if (strcmp(_PATH_WTMP, name)) | |
111 | Flags |= AC_W; | |
112 | return fp; | |
113 | } | |
114 | ||
115 | struct tty_list * | |
116 | add_tty(name) | |
117 | char *name; | |
118 | { | |
119 | struct tty_list *tp; | |
120 | register char *rcp; | |
121 | ||
122 | Flags |= AC_T; | |
123 | ||
124 | if ((tp = NEW(struct tty_list)) == NULL) | |
125 | err(1, "malloc"); | |
126 | tp->len = 0; /* full match */ | |
127 | tp->ret = 1; /* do if match */ | |
128 | if (*name == '!') { /* don't do if match */ | |
129 | tp->ret = 0; | |
130 | name++; | |
131 | } | |
132 | (void)strncpy(tp->name, name, sizeof (tp->name) - 1); | |
133 | tp->name[sizeof (tp->name) - 1] = '\0'; | |
134 | if ((rcp = strchr(tp->name, '*')) != NULL) { /* wild card */ | |
135 | *rcp = '\0'; | |
136 | tp->len = strlen(tp->name); /* match len bytes only */ | |
137 | } | |
138 | tp->next = Ttys; | |
139 | Ttys = tp; | |
140 | return Ttys; | |
141 | } | |
142 | ||
143 | /* | |
144 | * should we process the named tty? | |
145 | */ | |
146 | int | |
147 | do_tty(name) | |
148 | char *name; | |
149 | { | |
150 | struct tty_list *tp; | |
151 | int def_ret = 0; | |
152 | ||
153 | for (tp = Ttys; tp != NULL; tp = tp->next) { | |
154 | if (tp->ret == 0) /* specific don't */ | |
155 | def_ret = 1; /* default do */ | |
156 | if (tp->len != 0) { | |
157 | if (strncmp(name, tp->name, tp->len) == 0) | |
158 | return tp->ret; | |
159 | } else { | |
160 | if (strncmp(name, tp->name, sizeof (tp->name)) == 0) | |
161 | return tp->ret; | |
162 | } | |
163 | } | |
164 | return def_ret; | |
165 | } | |
166 | ||
167 | #ifdef CONSOLE_TTY | |
168 | /* | |
169 | * is someone logged in on Console? | |
170 | */ | |
171 | int | |
172 | on_console(head) | |
173 | struct utmp_list *head; | |
174 | { | |
175 | struct utmp_list *up; | |
176 | ||
177 | for (up = head; up; up = up->next) { | |
178 | if (strncmp(up->usr.ut_line, Console, | |
179 | sizeof (up->usr.ut_line)) == 0) | |
180 | return 1; | |
181 | } | |
182 | return 0; | |
183 | } | |
184 | #endif | |
185 | ||
186 | /* | |
187 | * update user's login time | |
188 | */ | |
189 | struct user_list * | |
190 | update_user(head, name, secs) | |
191 | struct user_list *head; | |
192 | char *name; | |
193 | time_t secs; | |
194 | { | |
195 | struct user_list *up; | |
196 | ||
197 | for (up = head; up != NULL; up = up->next) { | |
198 | if (strncmp(up->name, name, sizeof (up->name)) == 0) { | |
199 | up->secs += secs; | |
200 | Total += secs; | |
201 | return head; | |
202 | } | |
203 | } | |
204 | /* | |
205 | * not found so add new user unless specified users only | |
206 | */ | |
207 | if (Flags & AC_U) | |
208 | return head; | |
209 | ||
210 | if ((up = NEW(struct user_list)) == NULL) | |
211 | err(1, "malloc"); | |
212 | up->next = head; | |
213 | (void)strncpy(up->name, name, sizeof (up->name) - 1); | |
214 | up->name[sizeof (up->name) - 1] = '\0'; /* paranoid! */ | |
215 | up->secs = secs; | |
216 | Total += secs; | |
217 | return up; | |
218 | } | |
219 | ||
220 | int | |
221 | main(argc, argv) | |
222 | int argc; | |
223 | char **argv; | |
224 | { | |
225 | FILE *fp; | |
226 | int c; | |
227 | ||
228 | fp = NULL; | |
229 | while ((c = getopt(argc, argv, "Dc:dpt:w:")) != EOF) { | |
230 | switch (c) { | |
231 | #ifdef DEBUG | |
232 | case 'D': | |
233 | Debug++; | |
234 | break; | |
235 | #endif | |
236 | case 'c': | |
237 | #ifdef CONSOLE_TTY | |
238 | Console = optarg; | |
239 | #else | |
240 | usage(); /* XXX */ | |
241 | #endif | |
242 | break; | |
243 | case 'd': | |
244 | Flags |= AC_D; | |
245 | break; | |
246 | case 'p': | |
247 | Flags |= AC_P; | |
248 | break; | |
249 | case 't': /* only do specified ttys */ | |
250 | add_tty(optarg); | |
251 | break; | |
252 | case 'w': | |
253 | fp = file(optarg); | |
254 | break; | |
255 | case '?': | |
256 | default: | |
257 | usage(); | |
258 | break; | |
259 | } | |
260 | } | |
261 | if (optind < argc) { | |
262 | /* | |
263 | * initialize user list | |
264 | */ | |
265 | for (; optind < argc; optind++) { | |
266 | Users = update_user(Users, argv[optind], 0L); | |
267 | } | |
268 | Flags |= AC_U; /* freeze user list */ | |
269 | } | |
270 | if (Flags & AC_D) | |
271 | Flags &= ~AC_P; | |
272 | if (fp == NULL) { | |
273 | /* | |
274 | * if _PATH_WTMP does not exist, exit quietly | |
275 | */ | |
276 | if (access(_PATH_WTMP, 0) != 0 && errno == ENOENT) | |
277 | return 0; | |
278 | ||
279 | fp = file(_PATH_WTMP); | |
280 | } | |
281 | ac(fp); | |
282 | ||
283 | return 0; | |
284 | } | |
285 | ||
286 | /* | |
287 | * print login time in decimal hours | |
288 | */ | |
289 | void | |
290 | show(name, secs) | |
291 | char *name; | |
292 | time_t secs; | |
293 | { | |
294 | (void)printf("\t%-*s %8.2f\n", UT_NAMESIZE, name, | |
295 | ((double)secs / 3600)); | |
296 | } | |
297 | ||
298 | void | |
299 | show_users(list) | |
300 | struct user_list *list; | |
301 | { | |
302 | struct user_list *lp; | |
303 | ||
304 | for (lp = list; lp; lp = lp->next) | |
305 | show(lp->name, lp->secs); | |
306 | } | |
307 | ||
308 | /* | |
309 | * print total login time for 24hr period in decimal hours | |
310 | */ | |
311 | void | |
312 | show_today(users, logins, secs) | |
313 | struct user_list *users; | |
314 | struct utmp_list *logins; | |
315 | time_t secs; | |
316 | { | |
317 | struct user_list *up; | |
318 | struct utmp_list *lp; | |
319 | char date[64]; | |
320 | time_t yesterday = secs - 1; | |
321 | ||
322 | (void)strftime(date, sizeof (date), "%b %e total", | |
323 | localtime(&yesterday)); | |
324 | ||
325 | /* restore the missing second */ | |
326 | yesterday++; | |
327 | ||
328 | for (lp = logins; lp != NULL; lp = lp->next) { | |
329 | secs = yesterday - lp->usr.ut_time; | |
330 | Users = update_user(Users, lp->usr.ut_name, secs); | |
331 | lp->usr.ut_time = yesterday; /* as if they just logged in */ | |
332 | } | |
333 | secs = 0; | |
334 | for (up = users; up != NULL; up = up->next) { | |
335 | secs += up->secs; | |
336 | up->secs = 0; /* for next day */ | |
337 | } | |
338 | if (secs) | |
339 | (void)printf("%s %11.2f\n", date, ((double)secs / 3600)); | |
340 | } | |
341 | ||
342 | /* | |
343 | * log a user out and update their times. | |
344 | * if ut_line is "~", we log all users out as the system has | |
345 | * been shut down. | |
346 | */ | |
347 | struct utmp_list * | |
348 | log_out(head, up) | |
349 | struct utmp_list *head; | |
350 | struct utmp *up; | |
351 | { | |
352 | struct utmp_list *lp, *lp2, *tlp; | |
353 | time_t secs; | |
354 | ||
355 | for (lp = head, lp2 = NULL; lp != NULL; ) | |
356 | if (*up->ut_line == '~' || strncmp(lp->usr.ut_line, up->ut_line, | |
357 | sizeof (up->ut_line)) == 0) { | |
358 | secs = up->ut_time - lp->usr.ut_time; | |
359 | Users = update_user(Users, lp->usr.ut_name, secs); | |
360 | #ifdef DEBUG | |
361 | if (Debug) | |
362 | printf("%-.*s %-.*s: %-.*s logged out (%2d:%02d:%02d)\n", | |
363 | 19, ctime(&up->ut_time), | |
364 | sizeof (lp->usr.ut_line), lp->usr.ut_line, | |
365 | sizeof (lp->usr.ut_name), lp->usr.ut_name, | |
366 | secs / 3600, (secs % 3600) / 60, secs % 60); | |
367 | #endif | |
368 | /* | |
369 | * now lose it | |
370 | */ | |
371 | tlp = lp; | |
372 | lp = lp->next; | |
373 | if (tlp == head) | |
374 | head = lp; | |
375 | else if (lp2 != NULL) | |
376 | lp2->next = lp; | |
377 | free(tlp); | |
378 | } else { | |
379 | lp2 = lp; | |
380 | lp = lp->next; | |
381 | } | |
382 | return head; | |
383 | } | |
384 | ||
385 | ||
386 | /* | |
387 | * if do_tty says ok, login a user | |
388 | */ | |
389 | struct utmp_list * | |
390 | log_in(head, up) | |
391 | struct utmp_list *head; | |
392 | struct utmp *up; | |
393 | { | |
394 | struct utmp_list *lp; | |
395 | ||
396 | /* | |
397 | * this could be a login. if we're not dealing with | |
398 | * the console name, say it is. | |
399 | * | |
400 | * If we are, and if ut_host==":0.0" we know that it | |
401 | * isn't a real login. _But_ if we have not yet recorded | |
402 | * someone being logged in on Console - due to the wtmp | |
403 | * file starting after they logged in, we'll pretend they | |
404 | * logged in, at the start of the wtmp file. | |
405 | */ | |
406 | ||
407 | #ifdef CONSOLE_TTY | |
408 | if (up->ut_host[0] == ':') { | |
409 | /* | |
410 | * SunOS 4.0.2 does not treat ":0.0" as special but we | |
411 | * do. | |
412 | */ | |
413 | if (on_console(head)) | |
414 | return head; | |
415 | /* | |
416 | * ok, no recorded login, so they were here when wtmp | |
417 | * started! Adjust ut_time! | |
418 | */ | |
419 | up->ut_time = FirstTime; | |
420 | /* | |
421 | * this allows us to pick the right logout | |
422 | */ | |
423 | (void)strncpy(up->ut_line, Console, sizeof (up->ut_line) - 1); | |
424 | up->ut_line[sizeof (up->ut_line) - 1] = '\0'; /* paranoid! */ | |
425 | } | |
426 | #endif | |
427 | /* | |
428 | * If we are doing specified ttys only, we ignore | |
429 | * anything else. | |
430 | */ | |
431 | if (Flags & AC_T) | |
432 | if (!do_tty(up->ut_line)) | |
433 | return head; | |
434 | ||
435 | /* | |
436 | * go ahead and log them in | |
437 | */ | |
438 | if ((lp = NEW(struct utmp_list)) == NULL) | |
439 | err(1, "malloc"); | |
440 | lp->next = head; | |
441 | head = lp; | |
442 | memmove((char *)&lp->usr, (char *)up, sizeof (struct utmp)); | |
443 | #ifdef DEBUG | |
444 | if (Debug) { | |
445 | printf("%-.*s %-.*s: %-.*s logged in", 19, | |
446 | ctime(&lp->usr.ut_time), sizeof (up->ut_line), | |
447 | up->ut_line, sizeof (up->ut_name), up->ut_name); | |
448 | if (*up->ut_host) | |
449 | printf(" (%-.*s)", sizeof (up->ut_host), up->ut_host); | |
450 | putchar('\n'); | |
451 | } | |
452 | #endif | |
453 | return head; | |
454 | } | |
455 | ||
456 | int | |
457 | ac(fp) | |
458 | FILE *fp; | |
459 | { | |
460 | struct utmp_list *lp, *head = NULL; | |
461 | struct utmp usr; | |
462 | struct tm *ltm; | |
463 | time_t secs; | |
464 | int day = -1; | |
465 | ||
466 | while (fread((char *)&usr, sizeof(usr), 1, fp) == 1) { | |
467 | if (!FirstTime) | |
468 | FirstTime = usr.ut_time; | |
469 | if (Flags & AC_D) { | |
470 | ltm = localtime(&usr.ut_time); | |
471 | if (day >= 0 && day != ltm->tm_yday) { | |
472 | day = ltm->tm_yday; | |
473 | /* | |
474 | * print yesterday's total | |
475 | */ | |
476 | secs = usr.ut_time; | |
477 | secs -= ltm->tm_sec; | |
478 | secs -= 60 * ltm->tm_min; | |
479 | secs -= 3600 * ltm->tm_hour; | |
480 | show_today(Users, head, secs); | |
481 | } else | |
482 | day = ltm->tm_yday; | |
483 | } | |
484 | switch(*usr.ut_line) { | |
485 | case '|': | |
486 | secs = usr.ut_time; | |
487 | break; | |
488 | case '{': | |
489 | secs -= usr.ut_time; | |
490 | /* | |
491 | * adjust time for those logged in | |
492 | */ | |
493 | for (lp = head; lp != NULL; lp = lp->next) | |
494 | lp->usr.ut_time -= secs; | |
495 | break; | |
496 | case '~': /* reboot or shutdown */ | |
497 | head = log_out(head, &usr); | |
498 | FirstTime = usr.ut_time; /* shouldn't be needed */ | |
499 | break; | |
500 | default: | |
501 | /* | |
502 | * if they came in on tty[p-y]*, then it is only | |
503 | * a login session if the ut_host field is non-empty | |
504 | */ | |
505 | if (*usr.ut_name) { | |
506 | if (strncmp(usr.ut_line, "tty", 3) != 0 || | |
507 | strchr("pqrstuvwxy", usr.ut_line[3]) == 0 || | |
508 | *usr.ut_host != '\0') | |
509 | head = log_in(head, &usr); | |
510 | } else | |
511 | head = log_out(head, &usr); | |
512 | break; | |
513 | } | |
514 | } | |
515 | (void)fclose(fp); | |
516 | usr.ut_time = time((time_t *)0); | |
517 | (void)strcpy(usr.ut_line, "~"); | |
518 | ||
519 | if (Flags & AC_D) { | |
520 | ltm = localtime(&usr.ut_time); | |
521 | if (day >= 0 && day != ltm->tm_yday) { | |
522 | /* | |
523 | * print yesterday's total | |
524 | */ | |
525 | secs = usr.ut_time; | |
526 | secs -= ltm->tm_sec; | |
527 | secs -= 60 * ltm->tm_min; | |
528 | secs -= 3600 * ltm->tm_hour; | |
529 | show_today(Users, head, secs); | |
530 | } | |
531 | } | |
532 | /* | |
533 | * anyone still logged in gets time up to now | |
534 | */ | |
535 | head = log_out(head, &usr); | |
536 | ||
537 | if (Flags & AC_D) | |
538 | show_today(Users, head, time((time_t *)0)); | |
539 | else { | |
540 | if (Flags & AC_P) | |
541 | show_users(Users); | |
542 | show("total", Total); | |
543 | } | |
544 | return 0; | |
545 | } | |
546 | ||
547 | void | |
548 | usage() | |
549 | { | |
550 | (void)fprintf(stderr, | |
551 | #ifdef CONSOLE_TTY | |
552 | "ac [-dp] [-c console] [-t tty] [-w wtmp] [users ...]\n"); | |
553 | #else | |
554 | "ac [-dp] [-t tty] [-w wtmp] [users ...]\n"); | |
555 | #endif | |
556 | exit(1); | |
557 | } |