Add special case for the german whois nameserver. Without the
[dragonfly.git] / usr.bin / who / who.c
CommitLineData
984263bc
MD
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.
1de703da
MD
25 *
26 * $FreeBSD: src/usr.bin/who/who.c,v 1.9.2.4 2002/12/21 00:44:58 tjr Exp $
0f5078d3 27 * $DragonFly: src/usr.bin/who/who.c,v 1.4 2004/08/25 01:56:49 dillon Exp $
984263bc
MD
28 */
29
984263bc
MD
30#include <sys/types.h>
31#include <sys/ioctl.h>
32#include <sys/stat.h>
33
34#include <err.h>
35#include <errno.h>
36#include <langinfo.h>
37#include <limits.h>
38#include <locale.h>
39#include <paths.h>
40#include <pwd.h>
41#include <stdio.h>
42#include <stdlib.h>
43#include <string.h>
44#include <time.h>
45#include <unistd.h>
46#include <utmp.h>
47
48static void heading(void);
49static void process_utmp(FILE *);
50static void quick(FILE *);
51static void row(struct utmp *);
52static int ttywidth(void);
53static void usage(void);
54static void whoami(FILE *);
55
56static int Hflag; /* Write column headings */
57static int mflag; /* Show info about current terminal */
58static int qflag; /* "Quick" mode */
59static int sflag; /* Show name, line, time */
60static int Tflag; /* Show terminal state */
61static int uflag; /* Show idle time */
62
63int
64main(int argc, char *argv[])
65{
66 int ch;
67 const char *file;
68 FILE *fp;
69
70 setlocale(LC_TIME, "");
71
72 while ((ch = getopt(argc, argv, "HTmqsu")) != -1) {
73 switch (ch) {
74 case 'H': /* Write column headings */
75 Hflag = 1;
76 break;
77 case 'T': /* Show terminal state */
78 Tflag = 1;
79 break;
80 case 'm': /* Show info about current terminal */
81 mflag = 1;
82 break;
83 case 'q': /* "Quick" mode */
84 qflag = 1;
85 break;
86 case 's': /* Show name, line, time */
87 sflag = 1;
88 break;
89 case 'u': /* Show idle time */
90 uflag = 1;
91 break;
92 default:
93 usage();
94 /*NOTREACHED*/
95 }
96 }
97 argc -= optind;
98 argv += optind;
99
100 if (argc >= 2 && strcmp(argv[0], "am") == 0 &&
101 (strcmp(argv[1], "i") == 0 || strcmp(argv[1], "I") == 0)) {
102 /* "who am i" or "who am I", equivalent to -m */
103 mflag = 1;
104 argc -= 2;
105 argv += 2;
106 }
107 if (argc > 1)
108 usage();
109
110 if (*argv != NULL)
111 file = *argv;
112 else
113 file = _PATH_UTMP;
114 if ((fp = fopen(file, "r")) == NULL)
115 err(1, "%s", file);
116
117 if (qflag)
118 quick(fp);
119 else {
120 if (sflag)
121 Tflag = uflag = 0;
122 if (Hflag)
123 heading();
124 if (mflag)
125 whoami(fp);
126 else
127 process_utmp(fp);
128 }
129
130 fclose(fp);
131
132 exit(0);
133}
134
135static void
136usage(void)
137{
138
139 fprintf(stderr, "usage: who [-HmqsTu] [am I] [file]\n");
140 exit(1);
141}
142
143static void
144heading(void)
145{
146
147 printf("%-*s ", UT_NAMESIZE, "NAME");
148 if (Tflag)
149 printf("S ");
150 printf("%-*s ", UT_LINESIZE, "LINE");
151 printf("%-*s ", 12, "TIME");
152 if (uflag)
153 printf("IDLE ");
154 printf("%-*s", UT_HOSTSIZE, "FROM");
155 putchar('\n');
156}
157
158static void
159row(struct utmp *ut)
160{
161 char buf[80], tty[sizeof(_PATH_DEV) + UT_LINESIZE];
162 struct stat sb;
163 time_t idle;
164 static int d_first = -1;
165 struct tm *tm;
166 char state;
167
168 if (d_first < 0)
169 d_first = (*nl_langinfo(D_MD_ORDER) == 'd');
170
171 if (Tflag || uflag) {
172 snprintf(tty, sizeof(tty), "%s%.*s", _PATH_DEV,
173 UT_LINESIZE, ut->ut_line);
174 state = '?';
175 idle = 0;
176 if (stat(tty, &sb) == 0) {
177 state = sb.st_mode & (S_IWOTH|S_IWGRP) ?
178 '+' : '-';
179 idle = time(NULL) - sb.st_mtime;
0f5078d3
MD
180 } else {
181 err(1, "Cannot open %s", tty);
984263bc
MD
182 }
183 }
184
185 printf("%-*.*s ", UT_NAMESIZE, UT_NAMESIZE, ut->ut_name);
186 if (Tflag)
187 printf("%c ", state);
188 printf("%-*.*s ", UT_LINESIZE, UT_LINESIZE, ut->ut_line);
189 tm = localtime(&ut->ut_time);
190 strftime(buf, sizeof(buf), d_first ? "%e %b %R" : "%b %e %R", tm);
191 printf("%-*s ", 12, buf);
192 if (uflag) {
193 if (idle < 60)
194 printf(" . ");
195 else if (idle < 24 * 60 * 60)
196 printf("%02d:%02d ", (int)(idle / 60 / 60),
197 (int)(idle / 60 % 60));
198 else
199 printf(" old ");
200 }
201 if (*ut->ut_host != '\0')
202 printf("(%.*s)", UT_HOSTSIZE, ut->ut_host);
203 putchar('\n');
204}
205
206static void
207process_utmp(FILE *fp)
208{
209 struct utmp ut;
210
211 while (fread(&ut, sizeof(ut), 1, fp) == 1)
212 if (*ut.ut_name != '\0')
213 row(&ut);
214}
215
216static void
217quick(FILE *fp)
218{
219 struct utmp ut;
220 int col, ncols, num;
221
222 ncols = ttywidth();
223 col = num = 0;
224 while (fread(&ut, sizeof(ut), 1, fp) == 1) {
225 if (*ut.ut_name == '\0')
226 continue;
227 printf("%-*.*s", UT_NAMESIZE, UT_NAMESIZE, ut.ut_name);
228 if (++col < ncols / (UT_NAMESIZE + 1))
229 putchar(' ');
230 else {
231 col = 0;
232 putchar('\n');
233 }
234 num++;
235 }
236 if (col != 0)
237 putchar('\n');
238
239 printf("# users = %d\n", num);
240}
241
242static void
243whoami(FILE *fp)
244{
245 struct utmp ut;
246 struct passwd *pwd;
247 const char *name, *p, *tty;
248
249 if ((tty = ttyname(STDIN_FILENO)) == NULL)
250 tty = "tty??";
251 else if ((p = strrchr(tty, '/')) != NULL)
252 tty = p + 1;
253
254 /* Search utmp for our tty, dump first matching record. */
255 while (fread(&ut, sizeof(ut), 1, fp) == 1)
256 if (*ut.ut_name != '\0' && strncmp(ut.ut_line, tty,
257 UT_LINESIZE) == 0) {
258 row(&ut);
259 return;
260 }
261
262 /* Not found; fill the utmp structure with the information we have. */
263 memset(&ut, 0, sizeof(ut));
264 if ((pwd = getpwuid(getuid())) != NULL)
265 name = pwd->pw_name;
266 else
267 name = "?";
268 strncpy(ut.ut_name, name, UT_NAMESIZE);
269 strncpy(ut.ut_line, tty, UT_LINESIZE);
270 time(&ut.ut_time);
271 row(&ut);
272}
273
274static int
275ttywidth(void)
276{
277 struct winsize ws;
278 long width;
279 char *cols, *ep;
280
281 if ((cols = getenv("COLUMNS")) != NULL && *cols != '\0') {
282 errno = 0;
283 width = strtol(cols, &ep, 10);
284 if (errno || width <= 0 || width > INT_MAX || ep == cols ||
285 *ep != '\0')
286 warnx("invalid COLUMNS environment variable ignored");
287 else
5b4dd598 288 return (width);
984263bc
MD
289 }
290 if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) != -1)
291 return (ws.ws_col);
292
293 return (80);
294}