[dragonfly.git] / usr.bin / who / who.c

/*-
 * Copyright (c) 2002 Tim J. Robbins.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * $FreeBSD: src/usr.bin/who/who.c,v 1.9.2.4 2002/12/21 00:44:58 tjr Exp $
 * $DragonFly: src/usr.bin/who/who.c,v 1.4 2004/08/25 01:56:49 dillon Exp $
 */

#include <sys/types.h>
#include <sys/ioctl.h>
#include <sys/stat.h>

#include <err.h>
#include <errno.h>
#include <langinfo.h>
#include <limits.h>
#include <locale.h>
#include <paths.h>
#include <pwd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>
#include <utmp.h>

static void	heading(void);
static void	process_utmp(FILE *);
static void	quick(FILE *);
static void	row(struct utmp *);
static int	ttywidth(void);
static void	usage(void);
static void	whoami(FILE *);

static int	Hflag;			/* Write column headings */
static int	mflag;			/* Show info about current terminal */
static int	qflag;			/* "Quick" mode */
static int	sflag;			/* Show name, line, time */
static int	Tflag;			/* Show terminal state */
static int	uflag;			/* Show idle time */

int
main(int argc, char *argv[])
{
	int ch;
	const char *file;
	FILE *fp;

	setlocale(LC_TIME, "");

	while ((ch = getopt(argc, argv, "HTmqsu")) != -1) {
		switch (ch) {
		case 'H':		/* Write column headings */
			Hflag = 1;
			break;
		case 'T':		/* Show terminal state */
			Tflag = 1;
			break;
		case 'm':		/* Show info about current terminal */
			mflag = 1;
			break;
		case 'q':		/* "Quick" mode */
			qflag = 1;
			break;
		case 's':		/* Show name, line, time */
			sflag = 1;
			break;
		case 'u':		/* Show idle time */
			uflag = 1;
			break;
		default:
			usage();
			/*NOTREACHED*/
		}
	}
	argc -= optind;
	argv += optind;

	if (argc >= 2 && strcmp(argv[0], "am") == 0 &&
	    (strcmp(argv[1], "i") == 0 || strcmp(argv[1], "I") == 0)) {
		/* "who am i" or "who am I", equivalent to -m */
		mflag = 1;
		argc -= 2;
		argv += 2;
	}
	if (argc > 1)
		usage();

	if (*argv != NULL)
		file = *argv;
	else
		file = _PATH_UTMP;
	if ((fp = fopen(file, "r")) == NULL)
		err(1, "%s", file);

	if (qflag)
		quick(fp);
	else {
		if (sflag)
			Tflag = uflag = 0;
		if (Hflag)
			heading();
		if (mflag)
			whoami(fp);
		else
			process_utmp(fp);
	}

	fclose(fp);

	exit(0);
}

static void
usage(void)
{

	fprintf(stderr, "usage: who [-HmqsTu] [am I] [file]\n");
	exit(1);
}

static void
heading(void)
{

	printf("%-*s ", UT_NAMESIZE, "NAME");
	if (Tflag)
		printf("S ");
	printf("%-*s ", UT_LINESIZE, "LINE");
	printf("%-*s ", 12, "TIME");
	if (uflag)
		printf("IDLE  ");
	printf("%-*s", UT_HOSTSIZE, "FROM");
	putchar('\n');
}

static void
row(struct utmp *ut)
{
	char buf[80], tty[sizeof(_PATH_DEV) + UT_LINESIZE];
	struct stat sb;
	time_t idle;
	static int d_first = -1;
	struct tm *tm;
	char state;

	if (d_first < 0)
		d_first = (*nl_langinfo(D_MD_ORDER) == 'd');

	if (Tflag || uflag) {
		snprintf(tty, sizeof(tty), "%s%.*s", _PATH_DEV,
			UT_LINESIZE, ut->ut_line);
		state = '?';
		idle = 0;
		if (stat(tty, &sb) == 0) {
			state = sb.st_mode & (S_IWOTH|S_IWGRP) ?
			    '+' : '-';
			idle = time(NULL) - sb.st_mtime;
		} else {
			err(1, "Cannot open %s", tty);
		}
	}

	printf("%-*.*s ", UT_NAMESIZE, UT_NAMESIZE, ut->ut_name);
	if (Tflag)
		printf("%c ", state);
	printf("%-*.*s ", UT_LINESIZE, UT_LINESIZE, ut->ut_line);
	tm = localtime(&ut->ut_time);
	strftime(buf, sizeof(buf), d_first ? "%e %b %R" : "%b %e %R", tm);
	printf("%-*s ", 12, buf);
	if (uflag) {
		if (idle < 60)
			printf("  .   ");
		else if (idle < 24 * 60 * 60)
			printf("%02d:%02d ", (int)(idle / 60 / 60),
			    (int)(idle / 60 % 60));
		else
			printf(" old  ");
	}
	if (*ut->ut_host != '\0')
		printf("(%.*s)", UT_HOSTSIZE, ut->ut_host);
	putchar('\n');
}

static void
process_utmp(FILE *fp)
{
	struct utmp ut;

	while (fread(&ut, sizeof(ut), 1, fp) == 1)
		if (*ut.ut_name != '\0')
			row(&ut);
}

static void
quick(FILE *fp)
{
	struct utmp ut;
	int col, ncols, num;

	ncols = ttywidth();
	col = num = 0;
	while (fread(&ut, sizeof(ut), 1, fp) == 1) { 
		if (*ut.ut_name == '\0')
			continue;
		printf("%-*.*s", UT_NAMESIZE, UT_NAMESIZE, ut.ut_name);
		if (++col < ncols / (UT_NAMESIZE + 1))
			putchar(' ');
		else {
			col = 0;
			putchar('\n');
		}
		num++;
	}
	if (col != 0)
		putchar('\n');

	printf("# users = %d\n", num);
}

static void
whoami(FILE *fp)
{
	struct utmp ut;
	struct passwd *pwd;
	const char *name, *p, *tty;

	if ((tty = ttyname(STDIN_FILENO)) == NULL)
		tty = "tty??";
	else if ((p = strrchr(tty, '/')) != NULL)
		tty = p + 1;

	/* Search utmp for our tty, dump first matching record. */
	while (fread(&ut, sizeof(ut), 1, fp) == 1)
		if (*ut.ut_name != '\0' && strncmp(ut.ut_line, tty,
		    UT_LINESIZE) == 0) {
			row(&ut);
			return;
		}

	/* Not found; fill the utmp structure with the information we have. */
	memset(&ut, 0, sizeof(ut));
	if ((pwd = getpwuid(getuid())) != NULL)
		name = pwd->pw_name;
	else
		name = "?";
	strncpy(ut.ut_name, name, UT_NAMESIZE);
	strncpy(ut.ut_line, tty, UT_LINESIZE);
	time(&ut.ut_time);
	row(&ut);
}

static int
ttywidth(void)
{
	struct winsize ws;
	long width;
	char *cols, *ep;

	if ((cols = getenv("COLUMNS")) != NULL && *cols != '\0') {
		errno = 0;
		width = strtol(cols, &ep, 10);
		if (errno || width <= 0 || width > INT_MAX || ep == cols ||
		    *ep != '\0')
			warnx("invalid COLUMNS environment variable ignored");
		else
			return (width);
	}
	if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) != -1)
		return (ws.ws_col);

	return (80);
}
Commit	Line	Data
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	*
1de703da MD	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	*/
984263bc MD	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
	48	static void heading(void);
	49	static void process_utmp(FILE *);
	50	static void quick(FILE *);
	51	static void row(struct utmp *);
	52	static int ttywidth(void);
	53	static void usage(void);
	54	static void whoami(FILE *);
	55
	56	static int Hflag; /* Write column headings */
	57	static int mflag; /* Show info about current terminal */
	58	static int qflag; /* "Quick" mode */
	59	static int sflag; /* Show name, line, time */
	60	static int Tflag; /* Show terminal state */
	61	static int uflag; /* Show idle time */
	62
	63	int
	64	main(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
135	static void
136	usage(void)
137	{
138
139	fprintf(stderr, "usage: who [-HmqsTu] [am I] [file]\n");
140	exit(1);
141	}
142
143	static void
144	heading(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
158	static void
159	row(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 {
0f5078d3 MD	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
	206	static void
	207	process_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
	216	static void
	217	quick(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
	242	static void
	243	whoami(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
274	static int
275	ttywidth(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	}