1 /* $NetBSD: rwho.c,v 1.19 2009/04/13 07:11:37 lukem Exp $ */ 2 3 /* 4 * Copyright (c) 1983, 1993 5 * The Regents of the University of California. All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. Neither the name of the University nor the names of its contributors 16 * may be used to endorse or promote products derived from this software 17 * without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 29 * SUCH DAMAGE. 30 */ 31 32 #include <sys/cdefs.h> 33 #ifndef lint 34 __COPYRIGHT("@(#) Copyright (c) 1983, 1993\ 35 The Regents of the University of California. All rights reserved."); 36 #endif /* not lint */ 37 38 #ifndef lint 39 /*static char sccsid[] = "from: @(#)rwho.c 8.1 (Berkeley) 6/6/93";*/ 40 __RCSID("$NetBSD: rwho.c,v 1.19 2009/04/13 07:11:37 lukem Exp $"); 41 #endif /* not lint */ 42 43 #include <sys/param.h> 44 #include <sys/file.h> 45 46 #include <protocols/rwhod.h> 47 48 #include <dirent.h> 49 #include <err.h> 50 #include <locale.h> 51 #include <stdio.h> 52 #include <stdlib.h> 53 #include <string.h> 54 #include <time.h> 55 #include <fcntl.h> 56 #include <unistd.h> 57 #include <utmp.h> 58 #include <sysexits.h> 59 60 static void usage(void) __dead; 61 static int utmpcmp(const void *, const void *); 62 63 /* 64 * this macro should be shared with ruptime. 65 */ 66 #define DOWN(w,now) ((now) - (w).wd_recvtime > 11 * 60) 67 #define WHDRSIZE (sizeof(struct whod) - \ 68 sizeof (((struct whod *)NULL)->wd_we)) 69 70 struct my_utmp { 71 char myhost[MAXHOSTNAMELEN]; 72 int myidle; 73 struct outmp myutmp; 74 }; 75 76 int 77 main(int argc, char **argv) 78 { 79 DIR *dirp; 80 struct dirent *dp; 81 struct whoent *we; 82 struct whod wd; 83 struct my_utmp *mp, *start; 84 time_t now; 85 int ch; 86 size_t width, n, i, nhosts, nusers, namount; 87 int aflg, qflg, hflg; 88 89 setprogname(argv[0]); 90 (void)setlocale(LC_TIME, ""); 91 aflg = nusers = nhosts = qflg = hflg = 0; 92 namount = 40; 93 start = NULL; 94 now = 0; 95 96 while ((ch = getopt(argc, argv, "aHq")) != -1) 97 switch(ch) { 98 case 'a': 99 aflg = 1; 100 break; 101 case 'q': 102 qflg = 1; 103 break; 104 case 'H': 105 hflg = 1; 106 break; 107 default: 108 usage(); 109 } 110 111 if(optind != argc) 112 usage(); 113 114 if (qflg) { 115 aflg = 0; 116 hflg = 0; 117 } 118 119 if (chdir(_PATH_RWHODIR) || (dirp = opendir(".")) == NULL) 120 err(EXIT_FAILURE, "Cannot access `%s'", _PATH_RWHODIR); 121 122 (void)time(&now); 123 124 if ((start = malloc(sizeof(*start) * namount)) == NULL) 125 err(EXIT_FAILURE, "malloc"); 126 127 while ((dp = readdir(dirp)) != NULL) { 128 int f; 129 ssize_t cc; 130 131 if (dp->d_ino == 0 || strncmp(dp->d_name, "whod.", 5)) 132 continue; 133 134 f = open(dp->d_name, O_RDONLY); 135 if (f < 0) 136 continue; 137 cc = read(f, &wd, sizeof (wd)); 138 if (cc < (ssize_t)WHDRSIZE) { 139 (void)close(f); 140 continue; 141 } 142 nhosts++; 143 if (DOWN(wd, now)) { 144 (void)close(f); 145 continue; 146 } 147 cc -= WHDRSIZE; 148 we = wd.wd_we; 149 for (n = cc / sizeof (struct whoent); n > 0; n--) { 150 if (aflg == 0 && we->we_idle >= 60*60) { 151 we++; 152 continue; 153 } 154 155 if (nusers == namount) { 156 namount = namount + 40; 157 if ((start = realloc(start, 158 sizeof(struct my_utmp) * namount)) == NULL) 159 err(1, "realloc"); 160 } 161 162 mp = start + nusers; 163 164 mp->myutmp = we->we_utmp; 165 mp->myidle = we->we_idle; 166 (void)strcpy(mp->myhost, wd.wd_hostname); 167 nusers++; we++; mp++; 168 } 169 (void)close(f); 170 } 171 172 if (nhosts == 0) 173 errx(EX_OK, "No hosts in `%s'.", _PATH_RWHODIR); 174 175 mp = start; 176 qsort(start, nusers, sizeof(*start), utmpcmp); 177 width = 0; 178 for (i = 0; i < nusers; i++) { 179 size_t j = strlen(mp->myhost) + 1 + strlen(mp->myutmp.out_line); 180 if (j > width) 181 width = j; 182 mp++; 183 } 184 mp = start; 185 if (hflg) { 186 (void)printf("%-*.*s %-*s %-12s %-*s\n", UT_NAMESIZE, 187 UT_NAMESIZE, "USER", (int)width, "LINE", "WHEN", 188 (int)width, "IDLE"); 189 } 190 for (i = 0; i < nusers; i++) { 191 char buf[BUFSIZ], cbuf[80]; 192 time_t t; 193 194 if (qflg) { 195 (void)printf("%-*.*s\n", UT_NAMESIZE, UT_NAMESIZE, 196 mp->myutmp.out_name); 197 mp++; 198 continue; 199 } 200 t = mp->myutmp.out_time; 201 (void)strftime(cbuf, sizeof(cbuf), "%c", localtime(&t)); 202 (void)snprintf(buf, sizeof(buf), "%s:%s", mp->myhost, 203 mp->myutmp.out_line); 204 (void)printf("%-*.*s %-*s %.12s", UT_NAMESIZE, UT_NAMESIZE, 205 mp->myutmp.out_name, (int)width, buf, cbuf + 4); 206 mp->myidle /= 60; 207 if (mp->myidle) { 208 if (aflg) { 209 if (mp->myidle >= 100 * 60) 210 mp->myidle = 100 * 60 - 1; 211 if (mp->myidle >= 60) 212 (void)printf(" %2d", mp->myidle / 60); 213 else 214 (void)printf(" "); 215 } else 216 (void)printf(" "); 217 (void)printf(":%02d", mp->myidle % 60); 218 } 219 (void)printf("\n"); 220 mp++; 221 } 222 223 if (qflg) 224 (void)printf("# users = %zd\n", nusers); 225 226 return EX_OK; 227 } 228 229 static int 230 utmpcmp(const void *v1, const void *v2) 231 { 232 const struct my_utmp *u1, *u2; 233 int rc; 234 235 u1 = v1; 236 u2 = v2; 237 rc = strncmp(u1->myutmp.out_name, u2->myutmp.out_name, 8); 238 if (rc) 239 return rc; 240 rc = strcmp(u1->myhost, u2->myhost); 241 if (rc) 242 return rc; 243 return strncmp(u1->myutmp.out_line, u2->myutmp.out_line, 8); 244 } 245 246 static void 247 usage(void) 248 { 249 (void)fprintf(stderr, "Usage: %s [-aHq]\n", getprogname()); 250 exit(1); 251 } 252