1 /* $NetBSD: who.c,v 1.22 2008/07/21 14:19:28 lukem Exp $ */ 2 3 /* 4 * Copyright (c) 1989, 1993 5 * The Regents of the University of California. All rights reserved. 6 * 7 * This code is derived from software contributed to Berkeley by 8 * Michael Fischbein. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 3. Neither the name of the University nor the names of its contributors 19 * may be used to endorse or promote products derived from this software 20 * without specific prior written permission. 21 * 22 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 25 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 32 * SUCH DAMAGE. 33 */ 34 35 #include <sys/cdefs.h> 36 #include <sys/sysctl.h> 37 #include <sys/types.h> 38 #include <sys/stat.h> 39 40 #include <err.h> 41 #include <locale.h> 42 #include <pwd.h> 43 #include <stdio.h> 44 #include <stdlib.h> 45 #include <string.h> 46 #include <time.h> 47 #include <unistd.h> 48 #ifdef SUPPORT_UTMP 49 #include <utmp.h> 50 #endif 51 #ifdef SUPPORT_UTMPX 52 #include <utmpx.h> 53 #endif 54 55 #include "utmpentry.h" 56 57 static void print_boottime(void); 58 static void output_labels(void); 59 static void who_am_i(const char *, int); 60 static void usage(void); 61 static void process(const char *, int); 62 static void eprint(const struct utmpentry *); 63 static void print(const char *, const char *, time_t, const char *, pid_t pid, 64 uint16_t term, uint16_t xit, uint16_t sess, uint16_t type); 65 static void quick(const char *); 66 67 static int show_term; /* show term state */ 68 static int show_idle; /* show idle time */ 69 static int show_details; /* show exit status etc. */ 70 static int bflag; /* Show date and time of last reboot */ 71 72 struct ut_type_names { 73 int type; 74 const char *name; 75 } ut_type_names[] = { 76 #ifdef SUPPORT_UTMPX 77 { EMPTY, "empty" }, 78 { RUN_LVL, "run level" }, 79 { BOOT_TIME, "boot time" }, 80 { OLD_TIME, "old time" }, 81 { NEW_TIME, "new time" }, 82 { INIT_PROCESS, "init process" }, 83 { LOGIN_PROCESS, "login process" }, 84 { USER_PROCESS, "user process" }, 85 { DEAD_PROCESS, "dead process" }, 86 { ACCOUNTING, "accounting" }, 87 { SIGNATURE, "signature" }, 88 { DOWN_TIME, "down time" }, 89 #endif /* SUPPORT_UTMPX */ 90 { -1, "unknown" } 91 }; 92 93 int 94 main(int argc, char *argv[]) 95 { 96 int c, only_current_term, show_labels, quick_mode, default_mode; 97 int et = 0; 98 99 setlocale(LC_ALL, ""); 100 101 only_current_term = show_term = show_idle = show_labels = 0; 102 quick_mode = default_mode = 0; 103 104 while ((c = getopt(argc, argv, "abdHlmpqrsTtuv")) != -1) { 105 switch (c) { 106 case 'a': 107 et = -1; 108 show_idle = show_details = 1; 109 break; 110 case 'b': 111 #if 0 112 et |= (1 << BOOT_TIME); 113 #endif 114 bflag = 1; 115 break; 116 case 'd': 117 et |= (1 << DEAD_PROCESS); 118 break; 119 case 'H': 120 show_labels = 1; 121 break; 122 case 'l': 123 et |= (1 << LOGIN_PROCESS); 124 break; 125 case 'm': 126 only_current_term = 1; 127 break; 128 case 'p': 129 et |= (1 << INIT_PROCESS); 130 break; 131 case 'q': 132 quick_mode = 1; 133 break; 134 case 'r': 135 et |= (1 << RUN_LVL); 136 break; 137 case 's': 138 default_mode = 1; 139 break; 140 case 'T': 141 show_term = 1; 142 break; 143 case 't': 144 et |= (1 << NEW_TIME); 145 break; 146 case 'u': 147 show_idle = 1; 148 break; 149 case 'v': 150 show_details = 1; 151 break; 152 default: 153 usage(); 154 /* NOTREACHED */ 155 } 156 } 157 argc -= optind; 158 argv += optind; 159 160 if (et != 0) 161 etype = et; 162 163 if (chdir("/dev")) { 164 err(EXIT_FAILURE, "cannot change directory to /dev"); 165 /* NOTREACHED */ 166 } 167 168 if (default_mode) 169 only_current_term = show_term = show_idle = 0; 170 171 if (!quick_mode && bflag) 172 print_boottime(); 173 174 switch (argc) { 175 case 0: /* who */ 176 if (quick_mode) { 177 quick(NULL); 178 } else if (only_current_term) { 179 who_am_i(NULL, show_labels); 180 } else { 181 process(NULL, show_labels); 182 } 183 break; 184 case 1: /* who utmp_file */ 185 if (quick_mode) { 186 quick(*argv); 187 } else if (only_current_term) { 188 who_am_i(*argv, show_labels); 189 } else { 190 process(*argv, show_labels); 191 } 192 break; 193 case 2: /* who am i */ 194 who_am_i(NULL, show_labels); 195 break; 196 default: 197 usage(); 198 /* NOTREACHED */ 199 } 200 201 return 0; 202 } 203 204 static void 205 print_boottime(void) 206 { 207 struct timeval boottime; 208 size_t size; 209 210 size = sizeof(boottime); 211 if (sysctlbyname("kern.boottime", &boottime, &size, NULL, 0) != -1 && 212 boottime.tv_sec != 0) { 213 printf("%s", ctime(&boottime.tv_sec)); 214 } 215 } 216 217 218 static char * 219 strrstr(const char *str, const char *pat) 220 { 221 const char *estr; 222 size_t len; 223 if (*pat == '\0') 224 return __DECONST(char *, str); 225 226 len = strlen(pat); 227 228 for (estr = str + strlen(str); str < estr; estr--) 229 if (strncmp(estr, pat, len) == 0) 230 return __DECONST(char *, estr); 231 return NULL; 232 } 233 234 static void 235 who_am_i(const char *fname, int show_labels) 236 { 237 struct passwd *pw; 238 const char *p; 239 char *t; 240 time_t now; 241 struct utmpentry *ehead, *ep; 242 243 /* search through the utmp and find an entry for this tty */ 244 if ((p = ttyname(STDIN_FILENO)) != NULL) { 245 246 /* strip directory prefixes for ttys */ 247 if ((t = strrstr(p, "/pts/")) != NULL || 248 (t = strrchr(p, '/')) != NULL) 249 p = t + 1; 250 251 (void)getutentries(fname, &ehead); 252 for (ep = ehead; ep; ep = ep->next) 253 if (strcmp(ep->line, p) == 0) { 254 if (show_labels) 255 output_labels(); 256 eprint(ep); 257 return; 258 } 259 } else 260 p = "tty??"; 261 262 (void)time(&now); 263 pw = getpwuid(getuid()); 264 if (show_labels) 265 output_labels(); 266 print(pw ? pw->pw_name : "?", p, now, "", getpid(), 0, 0, 0, 0); 267 } 268 269 static void 270 process(const char *fname, int show_labels) 271 { 272 struct utmpentry *ehead, *ep; 273 (void)getutentries(fname, &ehead); 274 if (show_labels) 275 output_labels(); 276 for (ep = ehead; ep != NULL; ep = ep->next) 277 eprint(ep); 278 } 279 280 static void 281 eprint(const struct utmpentry *ep) 282 { 283 print(ep->name, ep->line, (time_t)ep->tv.tv_sec, ep->host, ep->pid, 284 ep->term, ep->exit, ep->sess, ep->type); 285 } 286 287 static void 288 print(const char *name, const char *line, time_t t, const char *host, 289 pid_t pid, uint16_t term, uint16_t xit, uint16_t sess, uint16_t type) 290 { 291 struct stat sb; 292 char state; 293 static time_t now = 0; 294 time_t idle; 295 const char *types = NULL; 296 size_t i; 297 298 state = '?'; 299 idle = 0; 300 301 for (i = 0; ut_type_names[i].type >= 0; i++) { 302 types = ut_type_names[i].name; 303 if (ut_type_names[i].type == type) 304 break; 305 } 306 307 if (show_term || show_idle) { 308 if (now == 0) 309 time(&now); 310 311 if (stat(line, &sb) == 0) { 312 state = (sb.st_mode & 020) ? '+' : '-'; 313 idle = now - sb.st_atime; 314 } 315 316 } 317 318 (void)printf("%-*.*s ", maxname, maxname, name); 319 320 if (show_term) 321 (void)printf("%c ", state); 322 323 (void)printf("%-*.*s ", maxline, maxline, line); 324 (void)printf("%.12s ", ctime(&t) + 4); 325 326 if (show_idle) { 327 if (idle < 60) 328 (void)printf(" . "); 329 else if (idle < (24 * 60 * 60)) 330 (void)printf("%02ld:%02ld ", 331 (long)(idle / (60 * 60)), 332 (long)(idle % (60 * 60)) / 60); 333 else 334 (void)printf(" old "); 335 336 (void)printf("\t%6d", pid); 337 338 if (show_details) { 339 if (type == RUN_LVL) 340 (void)printf("\tnew=%c old=%c", term, xit); 341 else 342 (void)printf("\tterm=%d exit=%d", term, xit); 343 (void)printf(" sess=%d", sess); 344 (void)printf(" type=%s ", types); 345 } 346 } 347 348 if (*host) 349 (void)printf("\t(%.*s)", maxhost, host); 350 (void)putchar('\n'); 351 } 352 353 static void 354 output_labels(void) 355 { 356 (void)printf("%-*.*s ", maxname, maxname, "USER"); 357 358 if (show_term) 359 (void)printf("S "); 360 361 (void)printf("%-*.*s ", maxline, maxline, "LINE"); 362 (void)printf("WHEN "); 363 364 if (show_idle) { 365 (void)printf("IDLE "); 366 (void)printf("\t PID"); 367 368 (void)printf("\tCOMMENT"); 369 } 370 371 (void)putchar('\n'); 372 } 373 374 static void 375 quick(const char *fname) 376 { 377 struct utmpentry *ehead, *ep; 378 int num = 0; 379 380 (void)getutentries(fname, &ehead); 381 for (ep = ehead; ep != NULL; ep = ep->next) { 382 (void)printf("%-*s ", maxname, ep->name); 383 if ((++num % 8) == 0) 384 (void)putchar('\n'); 385 } 386 if (num % 8) 387 (void)putchar('\n'); 388 389 (void)printf("# users = %d\n", num); 390 } 391 392 static void 393 usage(void) 394 { 395 (void)fprintf(stderr, "Usage: %s [-abdHlmqrsTtuv] [file]\n\t%s am i\n", 396 getprogname(), getprogname()); 397 exit(EXIT_FAILURE); 398 } 399