1 /*- 2 * Copyright (c) 2002 Tim J. Robbins. 3 * 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 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 */ 26 27 #include <sys/cdefs.h> 28 __FBSDID("$FreeBSD$"); 29 30 #include <sys/types.h> 31 #include <sys/ioctl.h> 32 #include <sys/stat.h> 33 34 #include <err.h> 35 #include <langinfo.h> 36 #include <locale.h> 37 #include <paths.h> 38 #include <pwd.h> 39 #include <stdio.h> 40 #include <stdlib.h> 41 #include <string.h> 42 #include <time.h> 43 #include <unistd.h> 44 #include <utmp.h> 45 46 static void heading(void); 47 static void process_utmp(FILE *); 48 static void quick(FILE *); 49 static void row(struct utmp *); 50 static int ttywidth(void); 51 static void usage(void); 52 static void whoami(FILE *); 53 54 static int Hflag; /* Write column headings */ 55 static int mflag; /* Show info about current terminal */ 56 static int qflag; /* "Quick" mode */ 57 static int sflag; /* Show name, line, time */ 58 static int Tflag; /* Show terminal state */ 59 static int uflag; /* Show idle time */ 60 61 int 62 main(int argc, char *argv[]) 63 { 64 int ch; 65 const char *file; 66 FILE *fp; 67 68 setlocale(LC_TIME, ""); 69 70 while ((ch = getopt(argc, argv, "HTabdlmpqrstu")) != -1) { 71 switch (ch) { 72 case 'H': /* Write column headings */ 73 Hflag = 1; 74 break; 75 case 'T': /* Show terminal state */ 76 Tflag = 1; 77 break; 78 case 'm': /* Show info about current terminal */ 79 mflag = 1; 80 break; 81 case 'q': /* "Quick" mode */ 82 qflag = 1; 83 break; 84 case 's': /* Show name, line, time */ 85 sflag = 1; 86 break; 87 case 'u': /* Show idle time */ 88 uflag = 1; 89 break; 90 default: 91 usage(); 92 /*NOTREACHED*/ 93 } 94 } 95 argc -= optind; 96 argv += optind; 97 98 if (argc >= 2 && strcmp(argv[0], "am") == 0 && 99 (strcmp(argv[1], "i") == 0 || strcmp(argv[1], "I") == 0)) { 100 /* "who am i" or "who am I", equivalent to -m */ 101 mflag = 1; 102 argc -= 2; 103 argv += 2; 104 } 105 if (argc > 1) 106 usage(); 107 108 if (*argv != NULL) 109 file = *argv; 110 else 111 file = _PATH_UTMP; 112 if ((fp = fopen(file, "r")) == NULL) 113 err(1, "%s", file); 114 115 if (qflag) 116 quick(fp); 117 else { 118 if (sflag) 119 Tflag = uflag = 0; 120 if (Hflag) 121 heading(); 122 if (mflag) 123 whoami(fp); 124 else 125 process_utmp(fp); 126 } 127 128 fclose(fp); 129 130 exit(0); 131 } 132 133 void 134 usage(void) 135 { 136 137 fprintf(stderr, "usage: who [-HmqsTu] [am I] [file]\n"); 138 exit(1); 139 } 140 141 void 142 heading(void) 143 { 144 145 printf("%-*s ", UT_NAMESIZE, "NAME"); 146 if (Tflag) 147 printf("S "); 148 printf("%-*s ", UT_LINESIZE, "LINE"); 149 printf("%-*s ", 12, "TIME"); 150 if (uflag) 151 printf("IDLE "); 152 putchar('\n'); 153 } 154 155 void 156 row(struct utmp *ut) 157 { 158 char buf[80], tty[sizeof(_PATH_DEV) + UT_LINESIZE]; 159 struct stat sb; 160 time_t idle; 161 static int d_first = -1; 162 struct tm *tm; 163 char state; 164 165 if (d_first < 0) 166 d_first = (*nl_langinfo(D_MD_ORDER) == 'd'); 167 168 if (Tflag || uflag) { 169 snprintf(tty, sizeof(tty), "%s%.*s", _PATH_DEV, 170 UT_LINESIZE, ut->ut_line); 171 state = '?'; 172 idle = 0; 173 if (stat(tty, &sb) == 0) { 174 state = sb.st_mode & (S_IWOTH|S_IWGRP) ? 175 '+' : '-'; 176 idle = time(NULL) - sb.st_mtime; 177 } 178 } 179 180 printf("%-*.*s ", UT_NAMESIZE, UT_NAMESIZE, ut->ut_name); 181 if (Tflag) 182 printf("%c ", state); 183 printf("%-*.*s ", UT_LINESIZE, UT_LINESIZE, ut->ut_line); 184 tm = localtime(&ut->ut_time); 185 strftime(buf, sizeof(buf), d_first ? "%e %b %R" : "%b %e %R", tm); 186 printf("%-*s ", 12, buf); 187 if (uflag) { 188 if (idle < 60) 189 printf(" . "); 190 else if (idle < 24 * 60 * 60) 191 printf("%02d:%02d ", (int)(idle / 60 / 60), 192 (int)(idle / 60 % 60)); 193 else 194 printf(" old "); 195 } 196 if (*ut->ut_host != '\0') 197 printf("(%.*s)", UT_HOSTSIZE, ut->ut_host); 198 putchar('\n'); 199 } 200 201 void 202 process_utmp(FILE *fp) 203 { 204 struct utmp ut; 205 206 while (fread(&ut, sizeof(ut), 1, fp) == 1) 207 if (*ut.ut_name != '\0') 208 row(&ut); 209 } 210 211 void 212 quick(FILE *fp) 213 { 214 struct utmp ut; 215 int col, ncols, num; 216 217 ncols = ttywidth(); 218 col = num = 0; 219 while (fread(&ut, sizeof(ut), 1, fp) == 1) { 220 if (*ut.ut_name == '\0') 221 continue; 222 printf("%-*.*s", UT_NAMESIZE, UT_NAMESIZE, ut.ut_name); 223 if (++col < ncols / (UT_NAMESIZE + 1)) 224 putchar(' '); 225 else { 226 col = 0; 227 putchar('\n'); 228 } 229 num++; 230 } 231 if (col != 0) 232 putchar('\n'); 233 234 printf("# users = %d\n", num); 235 } 236 237 void 238 whoami(FILE *fp) 239 { 240 struct utmp ut; 241 struct passwd *pwd; 242 const char *name, *p, *tty; 243 244 if ((tty = ttyname(STDIN_FILENO)) == NULL) 245 tty = "tty??"; 246 else if ((p = strrchr(tty, '/')) != NULL) 247 tty = p + 1; 248 249 /* Search utmp for our tty, dump first matching record. */ 250 while (fread(&ut, sizeof(ut), 1, fp) == 1) 251 if (*ut.ut_name != '\0' && strncmp(ut.ut_line, tty, 252 UT_LINESIZE) == 0) { 253 row(&ut); 254 return; 255 } 256 257 /* Not found; fill the utmp structure with the information we have. */ 258 memset(&ut, 0, sizeof(ut)); 259 if ((pwd = getpwuid(getuid())) != NULL) 260 name = pwd->pw_name; 261 else 262 name = "?"; 263 strncpy(ut.ut_name, name, UT_NAMESIZE); 264 strncpy(ut.ut_line, tty, UT_LINESIZE); 265 time(&ut.ut_time); 266 row(&ut); 267 } 268 269 int 270 ttywidth(void) 271 { 272 struct winsize ws; 273 int width; 274 275 if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) != -1) 276 width = ws.ws_col; 277 else 278 width = 80; 279 280 return (width); 281 } 282