protect potential overflow for x86_64
[oweals/busybox.git] / miscutils / last.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * last implementation for busybox
4  *
5  * Copyright (C) 2003-2004 by Erik Andersen <andersen@codepoet.org>
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20  *
21  */
22
23 #include <sys/types.h>
24 #include <fcntl.h>
25 #include <unistd.h>
26 #include <stdlib.h>
27 #include <utmp.h>
28 #include <sys/stat.h>
29 #include <errno.h>
30 #include <string.h>
31 #include <time.h>
32 #include "busybox.h"
33
34 #ifndef SHUTDOWN_TIME
35 #  define SHUTDOWN_TIME 254
36 #endif
37
38 /* Grr... utmp char[] members  do not have to be nul-terminated.
39  * Do what we can while still keeping this reasonably small.
40  * Note: We are assuming the ut_id[] size is fixed at 4. */
41
42 #if (UT_LINESIZE != 32) || (UT_NAMESIZE != 32) || (UT_HOSTSIZE != 256)
43 #error struct utmp member char[] size(s) have changed!
44 #endif
45
46 extern int last_main(int argc, char **argv)
47 {
48         struct utmp ut;
49         int n, file = STDIN_FILENO;
50         time_t t_tmp;
51
52         if (argc > 1) {
53                 bb_show_usage();
54         }
55         file = bb_xopen(_PATH_WTMP, O_RDONLY);
56
57         printf("%-10s %-14s %-18s %-12.12s %s\n", "USER", "TTY", "HOST", "LOGIN", "TIME");
58         while ((n = safe_read(file, (void*)&ut, sizeof(struct utmp))) != 0) {
59
60                 if (n != sizeof(struct utmp)) {
61                         bb_perror_msg_and_die("short read");
62                 }
63
64                 if (strncmp(ut.ut_line, "~", 1) == 0) {
65                         if (strncmp(ut.ut_user, "shutdown", 8) == 0)
66                                 ut.ut_type = SHUTDOWN_TIME;
67                         else if (strncmp(ut.ut_user, "reboot", 6) == 0)
68                                 ut.ut_type = BOOT_TIME;
69                         else if (strncmp(ut.ut_user, "runlevel", 7) == 0)
70                                 ut.ut_type = RUN_LVL;
71                 } else {
72                         if (!ut.ut_name[0] || strcmp(ut.ut_name, "LOGIN") == 0 ||
73                                         ut.ut_name[0] == 0)
74                         {
75                                 /* Don't bother.  This means we can't find how long
76                                  * someone was logged in for.  Oh well. */
77                                 continue;
78                         }
79                         if (ut.ut_type != DEAD_PROCESS &&
80                                         ut.ut_name[0] && ut.ut_line[0])
81                         {
82                                 ut.ut_type = USER_PROCESS;
83                         }
84                         if (strcmp(ut.ut_name, "date") == 0) {
85                                 if (ut.ut_line[0] == '|') ut.ut_type = OLD_TIME;
86                                 if (ut.ut_line[0] == '{') ut.ut_type = NEW_TIME;
87                         }
88                 }
89
90                 if (ut.ut_type!=USER_PROCESS) {
91                         switch (ut.ut_type) {
92                                 case OLD_TIME:
93                                 case NEW_TIME:
94                                 case RUN_LVL:
95                                 case SHUTDOWN_TIME:
96                                         continue;
97                                 case BOOT_TIME:
98                                         strcpy(ut.ut_line, "system boot");
99                                         break;
100                         }
101                 }
102                 t_tmp = (time_t)ut.ut_tv.tv_sec;
103                 printf("%-10s %-14s %-18s %-12.12s\n", ut.ut_user, ut.ut_line, ut.ut_host,
104                                 ctime(&t_tmp) + 4);
105         }
106
107         bb_fflush_stdout_and_exit(EXIT_SUCCESS);
108 }