19#endif /* not lint */
20
21#include <sys/types.h>
22#include <sys/file.h>
23#include <sys/time.h>
24#include <err.h>
25#include <errno.h>
26#include <langinfo.h>
27#include <locale.h>
28#include <pwd.h>
29#include <stdio.h>
30#include <stdlib.h>
31#include <string.h>
32#include <unistd.h>
33#include <utmp.h>
34
35/*
36 * this is for our list of currently logged in sessions
37 */
38struct utmp_list {
39 struct utmp_list *next;
40 struct utmp usr;
41};
42
43/*
44 * this is for our list of users that are accumulating time.
45 */
46struct user_list {
47 struct user_list *next;
48 char name[UT_NAMESIZE+1];
49 time_t secs;
50};
51
52/*
53 * this is for chosing whether to ignore a login
54 */
55struct tty_list {
56 struct tty_list *next;
57 char name[UT_LINESIZE+3];
58 int len;
59 int ret;
60};
61
62/*
63 * globals - yes yuk
64 */
65#ifdef CONSOLE_TTY
66static char *Console = CONSOLE_TTY;
67#endif
68static time_t Total = 0;
69static time_t FirstTime = 0;
70static int Flags = 0;
71static struct user_list *Users = NULL;
72static struct tty_list *Ttys = NULL;
73
74#define NEW(type) (type *)malloc(sizeof (type))
75
76#define AC_W 1 /* not _PATH_WTMP */
77#define AC_D 2 /* daily totals (ignore -p) */
78#define AC_P 4 /* per-user totals */
79#define AC_U 8 /* specified users only */
80#define AC_T 16 /* specified ttys only */
81
82#ifdef DEBUG
83static int Debug = 0;
84#endif
85
86int main __P((int, char **));
87int ac __P((FILE *));
88struct tty_list *add_tty __P((char *));
89int do_tty __P((char *));
90FILE *file __P((const char *));
91struct utmp_list *log_in __P((struct utmp_list *, struct utmp *));
92struct utmp_list *log_out __P((struct utmp_list *, struct utmp *));
93int on_console __P((struct utmp_list *));
94void show __P((const char *, time_t));
95void show_today __P((struct user_list *, struct utmp_list *,
96 time_t));
97void show_users __P((struct user_list *));
98struct user_list *update_user __P((struct user_list *, char *, time_t));
99void usage __P((void));
100
101/*
102 * open wtmp or die
103 */
104FILE *
105file(name)
106 const char *name;
107{
108 FILE *fp;
109
110 if ((fp = fopen(name, "r")) == NULL)
111 err(1, "%s", name);
112 /* in case we want to discriminate */
113 if (strcmp(_PATH_WTMP, name))
114 Flags |= AC_W;
115 return fp;
116}
117
118struct tty_list *
119add_tty(name)
120 char *name;
121{
122 struct tty_list *tp;
123 register char *rcp;
124
125 Flags |= AC_T;
126
127 if ((tp = NEW(struct tty_list)) == NULL)
128 errx(1, "malloc failed");
129 tp->len = 0; /* full match */
130 tp->ret = 1; /* do if match */
131 if (*name == '!') { /* don't do if match */
132 tp->ret = 0;
133 name++;
134 }
135 strlcpy(tp->name, name, sizeof (tp->name));
136 if ((rcp = strchr(tp->name, '*')) != NULL) { /* wild card */
137 *rcp = '\0';
138 tp->len = strlen(tp->name); /* match len bytes only */
139 }
140 tp->next = Ttys;
141 Ttys = tp;
142 return Ttys;
143}
144
145/*
146 * should we process the named tty?
147 */
148int
149do_tty(name)
150 char *name;
151{
152 struct tty_list *tp;
153 int def_ret = 0;
154
155 for (tp = Ttys; tp != NULL; tp = tp->next) {
156 if (tp->ret == 0) /* specific don't */
157 def_ret = 1; /* default do */
158 if (tp->len != 0) {
159 if (strncmp(name, tp->name, tp->len) == 0)
160 return tp->ret;
161 } else {
162 if (strncmp(name, tp->name, sizeof (tp->name)) == 0)
163 return tp->ret;
164 }
165 }
166 return def_ret;
167}
168
169#ifdef CONSOLE_TTY
170/*
171 * is someone logged in on Console?
172 */
173int
174on_console(head)
175 struct utmp_list *head;
176{
177 struct utmp_list *up;
178
179 for (up = head; up; up = up->next) {
180 if (strncmp(up->usr.ut_line, Console,
181 sizeof (up->usr.ut_line)) == 0)
182 return 1;
183 }
184 return 0;
185}
186#endif
187
188/*
189 * update user's login time
190 */
191struct user_list *
192update_user(head, name, secs)
193 struct user_list *head;
194 char *name;
195 time_t secs;
196{
197 struct user_list *up;
198
199 for (up = head; up != NULL; up = up->next) {
200 if (strncmp(up->name, name, UT_NAMESIZE) == 0) {
201 up->secs += secs;
202 Total += secs;
203 return head;
204 }
205 }
206 /*
207 * not found so add new user unless specified users only
208 */
209 if (Flags & AC_U)
210 return head;
211
212 if ((up = NEW(struct user_list)) == NULL)
213 errx(1, "malloc failed");
214 up->next = head;
215 strlcpy(up->name, name, sizeof (up->name));
216 up->secs = secs;
217 Total += secs;
218 return up;
219}
220
221int
222main(argc, argv)
223 int argc;
224 char **argv;
225{
226 FILE *fp;
227 int c;
228
229 (void) setlocale(LC_TIME, "");
230
231 fp = NULL;
232 while ((c = getopt(argc, argv, "Dc:dpt:w:")) != -1) {
233 switch (c) {
234#ifdef DEBUG
235 case 'D':
236 Debug++;
237 break;
238#endif
239 case 'c':
240#ifdef CONSOLE_TTY
241 Console = optarg;
242#else
243 usage(); /* XXX */
244#endif
245 break;
246 case 'd':
247 Flags |= AC_D;
248 break;
249 case 'p':
250 Flags |= AC_P;
251 break;
252 case 't': /* only do specified ttys */
253 add_tty(optarg);
254 break;
255 case 'w':
256 fp = file(optarg);
257 break;
258 case '?':
259 default:
260 usage();
261 break;
262 }
263 }
264 if (optind < argc) {
265 /*
266 * initialize user list
267 */
268 for (; optind < argc; optind++) {
269 Users = update_user(Users, argv[optind], 0L);
270 }
271 Flags |= AC_U; /* freeze user list */
272 }
273 if (Flags & AC_D)
274 Flags &= ~AC_P;
275 if (fp == NULL) {
276 /*
277 * if _PATH_WTMP does not exist, exit quietly
278 */
279 if (access(_PATH_WTMP, 0) != 0 && errno == ENOENT)
280 return 0;
281
282 fp = file(_PATH_WTMP);
283 }
284 ac(fp);
285
286 return 0;
287}
288
289/*
290 * print login time in decimal hours
291 */
292void
293show(name, secs)
294 const char *name;
295 time_t secs;
296{
297 (void)printf("\t%-*s %8.2f\n", UT_NAMESIZE, name,
298 ((double)secs / 3600));
299}
300
301void
302show_users(list)
303 struct user_list *list;
304{
305 struct user_list *lp;
306
307 for (lp = list; lp; lp = lp->next)
308 show(lp->name, lp->secs);
309}
310
311/*
312 * print total login time for 24hr period in decimal hours
313 */
314void
315show_today(users, logins, secs)
316 struct user_list *users;
317 struct utmp_list *logins;
318 time_t secs;
319{
320 struct user_list *up;
321 struct utmp_list *lp;
322 char date[64];
323 time_t yesterday = secs - 1;
324 static int d_first = -1;
325
326 if (d_first < 0)
327 d_first = (*nl_langinfo(D_MD_ORDER) == 'd');
328 (void)strftime(date, sizeof (date),
329 d_first ? "%e %b total" : "%b %e total",
330 localtime(&yesterday));
331
332 /* restore the missing second */
333 yesterday++;
334
335 for (lp = logins; lp != NULL; lp = lp->next) {
336 secs = yesterday - lp->usr.ut_time;
337 Users = update_user(Users, lp->usr.ut_name, secs);
338 lp->usr.ut_time = yesterday; /* as if they just logged in */
339 }
340 secs = 0;
341 for (up = users; up != NULL; up = up->next) {
342 secs += up->secs;
343 up->secs = 0; /* for next day */
344 }
345 if (secs)
346 (void)printf("%s %11.2f\n", date, ((double)secs / 3600));
347}
348
349/*
350 * log a user out and update their times.
351 * if ut_line is "~", we log all users out as the system has
352 * been shut down.
353 */
354struct utmp_list *
355log_out(head, up)
356 struct utmp_list *head;
357 struct utmp *up;
358{
359 struct utmp_list *lp, *lp2, *tlp;
360 time_t secs;
361
362 for (lp = head, lp2 = NULL; lp != NULL; )
363 if (*up->ut_line == '~' || strncmp(lp->usr.ut_line, up->ut_line,
364 sizeof (up->ut_line)) == 0) {
365 secs = up->ut_time - lp->usr.ut_time;
366 Users = update_user(Users, lp->usr.ut_name, secs);
367#ifdef DEBUG
368 if (Debug)
369 printf("%-.*s %-.*s: %-.*s logged out (%2d:%02d:%02d)\n",
370 19, ctime(&up->ut_time),
371 sizeof (lp->usr.ut_line), lp->usr.ut_line,
372 sizeof (lp->usr.ut_name), lp->usr.ut_name,
373 secs / 3600, (secs % 3600) / 60, secs % 60);
374#endif
375 /*
376 * now lose it
377 */
378 tlp = lp;
379 lp = lp->next;
380 if (tlp == head)
381 head = lp;
382 else if (lp2 != NULL)
383 lp2->next = lp;
384 free(tlp);
385 } else {
386 lp2 = lp;
387 lp = lp->next;
388 }
389 return head;
390}
391
392
393/*
394 * if do_tty says ok, login a user
395 */
396struct utmp_list *
397log_in(head, up)
398 struct utmp_list *head;
399 struct utmp *up;
400{
401 struct utmp_list *lp;
402
403 /*
404 * this could be a login. if we're not dealing with
405 * the console name, say it is.
406 *
407 * If we are, and if ut_host==":0.0" we know that it
408 * isn't a real login. _But_ if we have not yet recorded
409 * someone being logged in on Console - due to the wtmp
410 * file starting after they logged in, we'll pretend they
411 * logged in, at the start of the wtmp file.
412 */
413
414#ifdef CONSOLE_TTY
415 if (up->ut_host[0] == ':') {
416 /*
417 * SunOS 4.0.2 does not treat ":0.0" as special but we
418 * do.
419 */
420 if (on_console(head))
421 return head;
422 /*
423 * ok, no recorded login, so they were here when wtmp
424 * started! Adjust ut_time!
425 */
426 up->ut_time = FirstTime;
427 /*
428 * this allows us to pick the right logout
429 */
430 strlcpy(up->ut_line, Console, sizeof (up->ut_line));
431 }
432#endif
433 /*
434 * If we are doing specified ttys only, we ignore
435 * anything else.
436 */
437 if (Flags & AC_T)
438 if (!do_tty(up->ut_line))
439 return head;
440
441 /*
442 * go ahead and log them in
443 */
444 if ((lp = NEW(struct utmp_list)) == NULL)
445 errx(1, "malloc failed");
446 lp->next = head;
447 head = lp;
448 memmove((char *)&lp->usr, (char *)up, sizeof (struct utmp));
449#ifdef DEBUG
450 if (Debug) {
451 printf("%-.*s %-.*s: %-.*s logged in", 19,
452 ctime(&lp->usr.ut_time), sizeof (up->ut_line),
453 up->ut_line, sizeof (up->ut_name), up->ut_name);
454 if (*up->ut_host)
455 printf(" (%-.*s)", sizeof (up->ut_host), up->ut_host);
456 putchar('\n');
457 }
458#endif
459 return head;
460}
461
462int
463ac(fp)
464 FILE *fp;
465{
466 struct utmp_list *lp, *head = NULL;
467 struct utmp usr;
468 struct tm *ltm;
469 time_t secs;
470 int day = -1;
471
472 while (fread((char *)&usr, sizeof(usr), 1, fp) == 1) {
473 if (!FirstTime)
474 FirstTime = usr.ut_time;
475 if (Flags & AC_D) {
| 19#endif /* not lint */
20
21#include <sys/types.h>
22#include <sys/file.h>
23#include <sys/time.h>
24#include <err.h>
25#include <errno.h>
26#include <langinfo.h>
27#include <locale.h>
28#include <pwd.h>
29#include <stdio.h>
30#include <stdlib.h>
31#include <string.h>
32#include <unistd.h>
33#include <utmp.h>
34
35/*
36 * this is for our list of currently logged in sessions
37 */
38struct utmp_list {
39 struct utmp_list *next;
40 struct utmp usr;
41};
42
43/*
44 * this is for our list of users that are accumulating time.
45 */
46struct user_list {
47 struct user_list *next;
48 char name[UT_NAMESIZE+1];
49 time_t secs;
50};
51
52/*
53 * this is for chosing whether to ignore a login
54 */
55struct tty_list {
56 struct tty_list *next;
57 char name[UT_LINESIZE+3];
58 int len;
59 int ret;
60};
61
62/*
63 * globals - yes yuk
64 */
65#ifdef CONSOLE_TTY
66static char *Console = CONSOLE_TTY;
67#endif
68static time_t Total = 0;
69static time_t FirstTime = 0;
70static int Flags = 0;
71static struct user_list *Users = NULL;
72static struct tty_list *Ttys = NULL;
73
74#define NEW(type) (type *)malloc(sizeof (type))
75
76#define AC_W 1 /* not _PATH_WTMP */
77#define AC_D 2 /* daily totals (ignore -p) */
78#define AC_P 4 /* per-user totals */
79#define AC_U 8 /* specified users only */
80#define AC_T 16 /* specified ttys only */
81
82#ifdef DEBUG
83static int Debug = 0;
84#endif
85
86int main __P((int, char **));
87int ac __P((FILE *));
88struct tty_list *add_tty __P((char *));
89int do_tty __P((char *));
90FILE *file __P((const char *));
91struct utmp_list *log_in __P((struct utmp_list *, struct utmp *));
92struct utmp_list *log_out __P((struct utmp_list *, struct utmp *));
93int on_console __P((struct utmp_list *));
94void show __P((const char *, time_t));
95void show_today __P((struct user_list *, struct utmp_list *,
96 time_t));
97void show_users __P((struct user_list *));
98struct user_list *update_user __P((struct user_list *, char *, time_t));
99void usage __P((void));
100
101/*
102 * open wtmp or die
103 */
104FILE *
105file(name)
106 const char *name;
107{
108 FILE *fp;
109
110 if ((fp = fopen(name, "r")) == NULL)
111 err(1, "%s", name);
112 /* in case we want to discriminate */
113 if (strcmp(_PATH_WTMP, name))
114 Flags |= AC_W;
115 return fp;
116}
117
118struct tty_list *
119add_tty(name)
120 char *name;
121{
122 struct tty_list *tp;
123 register char *rcp;
124
125 Flags |= AC_T;
126
127 if ((tp = NEW(struct tty_list)) == NULL)
128 errx(1, "malloc failed");
129 tp->len = 0; /* full match */
130 tp->ret = 1; /* do if match */
131 if (*name == '!') { /* don't do if match */
132 tp->ret = 0;
133 name++;
134 }
135 strlcpy(tp->name, name, sizeof (tp->name));
136 if ((rcp = strchr(tp->name, '*')) != NULL) { /* wild card */
137 *rcp = '\0';
138 tp->len = strlen(tp->name); /* match len bytes only */
139 }
140 tp->next = Ttys;
141 Ttys = tp;
142 return Ttys;
143}
144
145/*
146 * should we process the named tty?
147 */
148int
149do_tty(name)
150 char *name;
151{
152 struct tty_list *tp;
153 int def_ret = 0;
154
155 for (tp = Ttys; tp != NULL; tp = tp->next) {
156 if (tp->ret == 0) /* specific don't */
157 def_ret = 1; /* default do */
158 if (tp->len != 0) {
159 if (strncmp(name, tp->name, tp->len) == 0)
160 return tp->ret;
161 } else {
162 if (strncmp(name, tp->name, sizeof (tp->name)) == 0)
163 return tp->ret;
164 }
165 }
166 return def_ret;
167}
168
169#ifdef CONSOLE_TTY
170/*
171 * is someone logged in on Console?
172 */
173int
174on_console(head)
175 struct utmp_list *head;
176{
177 struct utmp_list *up;
178
179 for (up = head; up; up = up->next) {
180 if (strncmp(up->usr.ut_line, Console,
181 sizeof (up->usr.ut_line)) == 0)
182 return 1;
183 }
184 return 0;
185}
186#endif
187
188/*
189 * update user's login time
190 */
191struct user_list *
192update_user(head, name, secs)
193 struct user_list *head;
194 char *name;
195 time_t secs;
196{
197 struct user_list *up;
198
199 for (up = head; up != NULL; up = up->next) {
200 if (strncmp(up->name, name, UT_NAMESIZE) == 0) {
201 up->secs += secs;
202 Total += secs;
203 return head;
204 }
205 }
206 /*
207 * not found so add new user unless specified users only
208 */
209 if (Flags & AC_U)
210 return head;
211
212 if ((up = NEW(struct user_list)) == NULL)
213 errx(1, "malloc failed");
214 up->next = head;
215 strlcpy(up->name, name, sizeof (up->name));
216 up->secs = secs;
217 Total += secs;
218 return up;
219}
220
221int
222main(argc, argv)
223 int argc;
224 char **argv;
225{
226 FILE *fp;
227 int c;
228
229 (void) setlocale(LC_TIME, "");
230
231 fp = NULL;
232 while ((c = getopt(argc, argv, "Dc:dpt:w:")) != -1) {
233 switch (c) {
234#ifdef DEBUG
235 case 'D':
236 Debug++;
237 break;
238#endif
239 case 'c':
240#ifdef CONSOLE_TTY
241 Console = optarg;
242#else
243 usage(); /* XXX */
244#endif
245 break;
246 case 'd':
247 Flags |= AC_D;
248 break;
249 case 'p':
250 Flags |= AC_P;
251 break;
252 case 't': /* only do specified ttys */
253 add_tty(optarg);
254 break;
255 case 'w':
256 fp = file(optarg);
257 break;
258 case '?':
259 default:
260 usage();
261 break;
262 }
263 }
264 if (optind < argc) {
265 /*
266 * initialize user list
267 */
268 for (; optind < argc; optind++) {
269 Users = update_user(Users, argv[optind], 0L);
270 }
271 Flags |= AC_U; /* freeze user list */
272 }
273 if (Flags & AC_D)
274 Flags &= ~AC_P;
275 if (fp == NULL) {
276 /*
277 * if _PATH_WTMP does not exist, exit quietly
278 */
279 if (access(_PATH_WTMP, 0) != 0 && errno == ENOENT)
280 return 0;
281
282 fp = file(_PATH_WTMP);
283 }
284 ac(fp);
285
286 return 0;
287}
288
289/*
290 * print login time in decimal hours
291 */
292void
293show(name, secs)
294 const char *name;
295 time_t secs;
296{
297 (void)printf("\t%-*s %8.2f\n", UT_NAMESIZE, name,
298 ((double)secs / 3600));
299}
300
301void
302show_users(list)
303 struct user_list *list;
304{
305 struct user_list *lp;
306
307 for (lp = list; lp; lp = lp->next)
308 show(lp->name, lp->secs);
309}
310
311/*
312 * print total login time for 24hr period in decimal hours
313 */
314void
315show_today(users, logins, secs)
316 struct user_list *users;
317 struct utmp_list *logins;
318 time_t secs;
319{
320 struct user_list *up;
321 struct utmp_list *lp;
322 char date[64];
323 time_t yesterday = secs - 1;
324 static int d_first = -1;
325
326 if (d_first < 0)
327 d_first = (*nl_langinfo(D_MD_ORDER) == 'd');
328 (void)strftime(date, sizeof (date),
329 d_first ? "%e %b total" : "%b %e total",
330 localtime(&yesterday));
331
332 /* restore the missing second */
333 yesterday++;
334
335 for (lp = logins; lp != NULL; lp = lp->next) {
336 secs = yesterday - lp->usr.ut_time;
337 Users = update_user(Users, lp->usr.ut_name, secs);
338 lp->usr.ut_time = yesterday; /* as if they just logged in */
339 }
340 secs = 0;
341 for (up = users; up != NULL; up = up->next) {
342 secs += up->secs;
343 up->secs = 0; /* for next day */
344 }
345 if (secs)
346 (void)printf("%s %11.2f\n", date, ((double)secs / 3600));
347}
348
349/*
350 * log a user out and update their times.
351 * if ut_line is "~", we log all users out as the system has
352 * been shut down.
353 */
354struct utmp_list *
355log_out(head, up)
356 struct utmp_list *head;
357 struct utmp *up;
358{
359 struct utmp_list *lp, *lp2, *tlp;
360 time_t secs;
361
362 for (lp = head, lp2 = NULL; lp != NULL; )
363 if (*up->ut_line == '~' || strncmp(lp->usr.ut_line, up->ut_line,
364 sizeof (up->ut_line)) == 0) {
365 secs = up->ut_time - lp->usr.ut_time;
366 Users = update_user(Users, lp->usr.ut_name, secs);
367#ifdef DEBUG
368 if (Debug)
369 printf("%-.*s %-.*s: %-.*s logged out (%2d:%02d:%02d)\n",
370 19, ctime(&up->ut_time),
371 sizeof (lp->usr.ut_line), lp->usr.ut_line,
372 sizeof (lp->usr.ut_name), lp->usr.ut_name,
373 secs / 3600, (secs % 3600) / 60, secs % 60);
374#endif
375 /*
376 * now lose it
377 */
378 tlp = lp;
379 lp = lp->next;
380 if (tlp == head)
381 head = lp;
382 else if (lp2 != NULL)
383 lp2->next = lp;
384 free(tlp);
385 } else {
386 lp2 = lp;
387 lp = lp->next;
388 }
389 return head;
390}
391
392
393/*
394 * if do_tty says ok, login a user
395 */
396struct utmp_list *
397log_in(head, up)
398 struct utmp_list *head;
399 struct utmp *up;
400{
401 struct utmp_list *lp;
402
403 /*
404 * this could be a login. if we're not dealing with
405 * the console name, say it is.
406 *
407 * If we are, and if ut_host==":0.0" we know that it
408 * isn't a real login. _But_ if we have not yet recorded
409 * someone being logged in on Console - due to the wtmp
410 * file starting after they logged in, we'll pretend they
411 * logged in, at the start of the wtmp file.
412 */
413
414#ifdef CONSOLE_TTY
415 if (up->ut_host[0] == ':') {
416 /*
417 * SunOS 4.0.2 does not treat ":0.0" as special but we
418 * do.
419 */
420 if (on_console(head))
421 return head;
422 /*
423 * ok, no recorded login, so they were here when wtmp
424 * started! Adjust ut_time!
425 */
426 up->ut_time = FirstTime;
427 /*
428 * this allows us to pick the right logout
429 */
430 strlcpy(up->ut_line, Console, sizeof (up->ut_line));
431 }
432#endif
433 /*
434 * If we are doing specified ttys only, we ignore
435 * anything else.
436 */
437 if (Flags & AC_T)
438 if (!do_tty(up->ut_line))
439 return head;
440
441 /*
442 * go ahead and log them in
443 */
444 if ((lp = NEW(struct utmp_list)) == NULL)
445 errx(1, "malloc failed");
446 lp->next = head;
447 head = lp;
448 memmove((char *)&lp->usr, (char *)up, sizeof (struct utmp));
449#ifdef DEBUG
450 if (Debug) {
451 printf("%-.*s %-.*s: %-.*s logged in", 19,
452 ctime(&lp->usr.ut_time), sizeof (up->ut_line),
453 up->ut_line, sizeof (up->ut_name), up->ut_name);
454 if (*up->ut_host)
455 printf(" (%-.*s)", sizeof (up->ut_host), up->ut_host);
456 putchar('\n');
457 }
458#endif
459 return head;
460}
461
462int
463ac(fp)
464 FILE *fp;
465{
466 struct utmp_list *lp, *head = NULL;
467 struct utmp usr;
468 struct tm *ltm;
469 time_t secs;
470 int day = -1;
471
472 while (fread((char *)&usr, sizeof(usr), 1, fp) == 1) {
473 if (!FirstTime)
474 FirstTime = usr.ut_time;
475 if (Flags & AC_D) {
|