Merge from vendor branch AWK:
[dragonfly.git] / usr.bin / lastcomm / lastcomm.c
1 /*
2  * Copyright (c) 1980, 1993
3  *      The Regents of the University of California.  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 the University of
16  *      California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * @(#) Copyright (c) 1980, 1993 The Regents of the University of California.  All rights reserved.
34  * @(#)lastcomm.c       8.1 (Berkeley) 6/6/93
35  * $FreeBSD: src/usr.bin/lastcomm/lastcomm.c,v 1.10.2.3 2001/10/01 12:51:15 dd Exp $
36  * $DragonFly: src/usr.bin/lastcomm/lastcomm.c,v 1.3 2003/10/04 20:36:47 hmp Exp $
37  */
38
39 #include <sys/param.h>
40 #include <sys/stat.h>
41 #include <sys/acct.h>
42
43 #include <ctype.h>
44 #include <err.h>
45 #include <fcntl.h>
46 #include <stdio.h>
47 #include <stdlib.h>
48 #include <string.h>
49 #include <unistd.h>
50 #include <utmp.h>
51 #include "pathnames.h"
52
53 time_t   expand(u_int);
54 char    *flagbits(int);
55 const    char *getdev(dev_t);
56 int      requested(char *[], struct acct *);
57 static   void usage(void);
58
59 #define AC_UTIME 1 /* user */
60 #define AC_STIME 2 /* system */
61 #define AC_ETIME 4 /* elapsed */
62 #define AC_CTIME 8 /* user + system time, default */
63
64 #define AC_BTIME 16 /* starting time */
65 #define AC_FTIME 32 /* exit time (starting time + elapsed time )*/
66
67 #define AC_HZ ((double)AHZ)
68
69 int
70 main(int argc, char **argv)
71 {
72         char *p;
73         struct acct ab;
74         struct stat sb;
75         FILE *fp;
76         off_t size;
77         time_t t;
78         int ch;
79         const char *acctfile;
80         int flags = 0;
81
82         acctfile = _PATH_ACCT;
83         while ((ch = getopt(argc, argv, "f:usecSE")) != -1)
84                 switch((char)ch) {
85                 case 'f':
86                         acctfile = optarg;
87                         break;
88
89                 case 'u': 
90                         flags |= AC_UTIME; /* user time */
91                         break;
92                 case 's':
93                         flags |= AC_STIME; /* system time */
94                         break;
95                 case 'e':
96                         flags |= AC_ETIME; /* elapsed time */
97                         break;
98                 case 'c':
99                         flags |= AC_CTIME; /* user + system time */
100                         break;
101
102                 case 'S':
103                         flags |= AC_BTIME; /* starting time */
104                         break;
105                 case 'E':
106                         /* exit time (starting time + elapsed time )*/
107                         flags |= AC_FTIME; 
108                         break;
109
110                 case '?':
111                 default:
112                         usage();
113                 }
114
115         /* default user + system time and starting time */
116         if (!flags) {
117             flags = AC_CTIME | AC_BTIME;
118         }
119
120         argc -= optind;
121         argv += optind;
122
123         /* Open the file. */
124         if ((fp = fopen(acctfile, "r")) == NULL || fstat(fileno(fp), &sb))
125                 err(1, "%s", acctfile);
126
127         /*
128          * Round off to integral number of accounting records, probably
129          * not necessary, but it doesn't hurt.
130          */
131         size = sb.st_size - sb.st_size % sizeof(struct acct);
132
133         /* Check if any records to display. */
134         if ((unsigned)size < sizeof(struct acct))
135                 exit(0);
136
137         /*
138          * Seek to before the last entry in the file; use lseek(2) in case
139          * the file is bigger than a "long".
140          */
141         size -= sizeof(struct acct);
142         if (lseek(fileno(fp), size, SEEK_SET) == -1)
143                 err(1, "%s", acctfile);
144
145         for (;;) {
146                 if (fread(&ab, sizeof(struct acct), 1, fp) != 1)
147                         err(1, "%s", acctfile);
148
149                 if (fseek(fp, 2 * -(long)sizeof(struct acct), SEEK_CUR) == -1)
150                         err(1, "%s", acctfile);
151
152                 if (size == 0)
153                         break;
154                 size -= sizeof(struct acct);
155
156                 if (ab.ac_comm[0] == '\0') {
157                         ab.ac_comm[0] = '?';
158                         ab.ac_comm[1] = '\0';
159                 } else
160                         for (p = &ab.ac_comm[0];
161                             p < &ab.ac_comm[AC_COMM_LEN] && *p; ++p)
162                                 if (!isprint(*p))
163                                         *p = '?';
164                 if (*argv && !requested(argv, &ab))
165                         continue;
166
167                 (void)printf("%-*.*s %-7s %-*s %-*s",
168                              AC_COMM_LEN, AC_COMM_LEN, ab.ac_comm,
169                              flagbits(ab.ac_flag),
170                              UT_NAMESIZE, user_from_uid(ab.ac_uid, 0),
171                              UT_LINESIZE, getdev(ab.ac_tty));
172                 
173                 
174                 /* user + system time */
175                 if (flags & AC_CTIME) {
176                         (void)printf(" %6.2f secs", 
177                                      (expand(ab.ac_utime) + 
178                                       expand(ab.ac_stime))/AC_HZ);
179                 }
180                 
181                 /* usr time */
182                 if (flags & AC_UTIME) {
183                         (void)printf(" %6.2f us", expand(ab.ac_utime)/AC_HZ);
184                 }
185                 
186                 /* system time */
187                 if (flags & AC_STIME) {
188                         (void)printf(" %6.2f sy", expand(ab.ac_stime)/AC_HZ);
189                 }
190                 
191                 /* elapsed time */
192                 if (flags & AC_ETIME) {
193                         (void)printf(" %8.2f es", expand(ab.ac_etime)/AC_HZ);
194                 }
195                 
196                 /* starting time */
197                 if (flags & AC_BTIME) {
198                         (void)printf(" %.16s", ctime(&ab.ac_btime));
199                 }
200                 
201                 /* exit time (starting time + elapsed time )*/
202                 if (flags & AC_FTIME) {
203                         t = ab.ac_btime;
204                         t += (time_t)(expand(ab.ac_etime)/AC_HZ);
205                         (void)printf(" %.16s", ctime(&t));
206                 }
207                 printf("\n");
208         }
209         exit(0);
210 }
211
212 time_t
213 expand(u_int t)
214 {
215         time_t nt;
216
217         nt = t & 017777;
218         t >>= 13;
219         while (t) {
220                 t--;
221                 nt <<= 3;
222         }
223         return (nt);
224 }
225
226 char *
227 flagbits(int f)
228 {
229         static char flags[20] = "-";
230         char *p;
231
232 #define BIT(flag, ch)   if (f & flag) *p++ = ch
233
234         p = flags + 1;
235         BIT(ASU, 'S');
236         BIT(AFORK, 'F');
237         BIT(ACOMPAT, 'C');
238         BIT(ACORE, 'D');
239         BIT(AXSIG, 'X');
240         *p = '\0';
241         return (flags);
242 }
243
244 int
245 requested(char **argv, struct acct *acp)
246 {
247         const char *p;
248
249         do {
250                 p = user_from_uid(acp->ac_uid, 0);
251                 if (!strcmp(p, *argv))
252                         return (1);
253                 if ((p = getdev(acp->ac_tty)) && !strcmp(p, *argv))
254                         return (1);
255                 if (!strncmp(acp->ac_comm, *argv, AC_COMM_LEN))
256                         return (1);
257         } while (*++argv);
258         return (0);
259 }
260
261 const char *
262 getdev(dev_t dev)
263 {
264         static dev_t lastdev = (dev_t)-1;
265         static const char *lastname;
266
267         if (dev == NODEV)                       /* Special case. */
268                 return ("__");
269         if (dev == lastdev)                     /* One-element cache. */
270                 return (lastname);
271         lastdev = dev;
272         lastname = devname(dev, S_IFCHR);
273         return (lastname);
274 }
275
276 static void
277 usage(void)
278 {
279         (void)fprintf(stderr,
280 "usage: lastcomm [-EScesu] [ -f file ] [command ...] [user ...] [tty ...]\n");
281         exit(1);
282 }