Home | History | Annotate | Line # | Download | only in last
      1 /*	$NetBSD: want.c,v 1.17 2012/03/15 03:04:05 dholland Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 1987, 1993, 1994
      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 static struct utmp *buf;
     32 static time_t seentime;
     33 
     34 static void onintr(int);
     35 static int want(struct utmp *, int);
     36 static const char *gethost(struct utmp *, const char *, int);
     37 
     38 static const char *
     39 /*ARGSUSED*/
     40 gethost(struct utmp *ut, const char *host, int numeric)
     41 {
     42 #if HAS_UT_SS == 0
     43 	return numeric ? "" : host;
     44 #else
     45 	if (numeric) {
     46 		static char hbuf[512];
     47 		hbuf[0] = '\0';
     48 		(void)sockaddr_snprintf(hbuf, sizeof(hbuf), "%a",
     49 		    (struct sockaddr *)&ut->ut_ss);
     50 		return hbuf;
     51 	} else
     52 		return host;
     53 #endif
     54 }
     55 
     56 #define NULTERM(what) \
     57 	if (check ## what) \
     58 		(void)strlcpy(what ## p = what ## buf, bp->ut_ ## what, \
     59 		    sizeof(what ## buf)); \
     60 	else \
     61 		what ## p = bp->ut_ ## what
     62 
     63 /*
     64  * wtmp --
     65  *	read through the wtmp file
     66  */
     67 static void
     68 wtmp(const char *file, int namesz, int linesz, int hostsz, int numeric)
     69 {
     70 	struct utmp	*bp;		/* current structure */
     71 	TTY	*T;			/* tty list entry */
     72 	struct stat	stb;		/* stat of file for sz */
     73 	off_t	offset;
     74 	int	wfd;
     75 	char	*ct;
     76 	const char *crmsg;
     77 	size_t  len = sizeof(*buf) * MAXUTMP;
     78 	char namebuf[sizeof(bp->ut_name) + 1], *namep;
     79 	char linebuf[sizeof(bp->ut_line) + 1], *linep;
     80 	char hostbuf[sizeof(bp->ut_host) + 1], *hostp;
     81 	int checkname = namesz > (int)sizeof(bp->ut_name);
     82 	int checkline = linesz > (int)sizeof(bp->ut_line);
     83 	int checkhost = hostsz > (int)sizeof(bp->ut_host);
     84 
     85 	if ((buf = malloc(len)) == NULL)
     86 		err(EXIT_FAILURE, "Cannot allocate utmp buffer");
     87 
     88 	crmsg = NULL;
     89 
     90 	if (!strcmp(file, "-")) {
     91 		wfd = STDIN_FILENO;
     92 		file = "<stdin>";
     93 	} else if ((wfd = open(file, O_RDONLY, 0)) < 0) {
     94 		err(EXIT_FAILURE, "%s", file);
     95 	}
     96 
     97 	if (lseek(wfd, 0, SEEK_CUR) < 0) {
     98 		const char *dir;
     99 		char *tfile;
    100 		int tempfd;
    101 		ssize_t tlen;
    102 
    103 		if (ESPIPE != errno) {
    104 			err(EXIT_FAILURE, "lseek");
    105 		}
    106 		dir = getenv("TMPDIR");
    107 		if (asprintf(&tfile, "%s/last.XXXXXX", dir ? dir : _PATH_TMP) == -1)
    108 			err(EXIT_FAILURE, "asprintf");
    109 		tempfd = mkstemp(tfile);
    110 		if (tempfd < 0) {
    111 			err(EXIT_FAILURE, "mkstemp");
    112 		}
    113 		unlink(tfile);
    114 		for (;;) {
    115 		   	tlen = read(wfd, buf, len);
    116 			if (tlen < 0) {
    117 				err(1, "%s: read", file);
    118 			}
    119 			if (tlen == 0) {
    120 				break;
    121 			}
    122 			if (write(tempfd, buf, tlen) != tlen) {
    123 				err(1, "%s: write", tfile);
    124 			}
    125 		}
    126 		wfd = tempfd;
    127 	}
    128 
    129 	if (fstat(wfd, &stb) == -1)
    130 		err(EXIT_FAILURE, "%s: fstat", file);
    131 	if (!S_ISREG(stb.st_mode))
    132 		errx(EXIT_FAILURE, "%s: Not a regular file", file);
    133 
    134 	seentime = stb.st_mtime;
    135 	(void)signal(SIGINT, onintr);
    136 	(void)signal(SIGQUIT, onintr);
    137 
    138 	offset = stb.st_size;
    139 	/* Ignore trailing garbage or partial record */
    140 	offset -= offset % (off_t) sizeof(*buf);
    141 
    142 	while (offset >= (off_t) sizeof(*buf)) {
    143 		ssize_t ret, i;
    144 		size_t size;
    145 
    146 		size = MIN((off_t)len, offset);
    147 		offset -= size; /* Always a multiple of sizeof(*buf) */
    148 		ret = pread(wfd, buf, size, offset);
    149 		if (ret < 0) {
    150 			err(EXIT_FAILURE, "%s: pread", file);
    151 		} else if ((size_t) ret < size) {
    152 			err(EXIT_FAILURE, "%s: Unexpected end of file", file);
    153 		}
    154 
    155 		for (i = ret / sizeof(*buf) - 1; i >= 0; i--) {
    156 			bp = &buf[i];
    157 
    158 			NULTERM(name);
    159 			NULTERM(line);
    160 			NULTERM(host);
    161 
    162 			seentime = bp->ut_timefld;
    163 
    164 			/*
    165 			 * if the terminal line is '~', the machine stopped.
    166 			 * see utmp(5) for more info.
    167 			 */
    168 			if (linep[0] == '~' && !linep[1]) {
    169 				/* everybody just logged out */
    170 				for (T = ttylist; T; T = T->next)
    171 					T->logout = -bp->ut_timefld;
    172 				currentout = -bp->ut_timefld;
    173 				crmsg = strncmp(namep, "shutdown",
    174 				    namesz) ? "crash" : "shutdown";
    175 				if (want(bp, NO)) {
    176 					ct = fmttime(bp->ut_timefld, fulltime);
    177 					printf("%-*.*s  %-*.*s %-*.*s %s\n",
    178 					    namesz, namesz, namep,
    179 					    linesz, linesz, linep,
    180 					    hostsz, hostsz,
    181 					    gethost(bp, hostp, numeric), ct);
    182 					if (maxrec != -1 && !--maxrec)
    183 						return;
    184 				}
    185 				continue;
    186 			}
    187 			/*
    188 			 * if the line is '{' or '|', date got set; see
    189 			 * utmp(5) for more info.
    190 			 */
    191 			if ((linep[0] == '{' || linep[0] == '|') && !linep[1]) {
    192 				if (want(bp, NO)) {
    193 					ct = fmttime(bp->ut_timefld, fulltime);
    194 					printf("%-*.*s  %-*.*s %-*.*s %s\n",
    195 					    namesz, namesz, namep,
    196 					    linesz, linesz, linep,
    197 					    hostsz, hostsz,
    198 					    gethost(bp, hostp, numeric),
    199 					    ct);
    200 					if (maxrec && !--maxrec)
    201 						return;
    202 				}
    203 				continue;
    204 			}
    205 			/* find associated tty */
    206 			for (T = ttylist;; T = T->next) {
    207 				if (!T) {
    208 					/* add new one */
    209 					T = addtty(linep);
    210 					break;
    211 				}
    212 				if (!strncmp(T->tty, linep, LINESIZE))
    213 					break;
    214 			}
    215 			if (TYPE(bp) == SIGNATURE)
    216 				continue;
    217 			if (namep[0] && want(bp, YES)) {
    218 				ct = fmttime(bp->ut_timefld, fulltime);
    219 				printf("%-*.*s  %-*.*s %-*.*s %s ",
    220 				    namesz, namesz, namep,
    221 				    linesz, linesz, linep,
    222 				    hostsz, hostsz,
    223 				    gethost(bp, hostp, numeric),
    224 				    ct);
    225 				if (!T->logout)
    226 					puts("  still logged in");
    227 				else {
    228 					time_t	delta;			/* time difference */
    229 
    230 					if (T->logout < 0) {
    231 						T->logout = -T->logout;
    232 						printf("- %s", crmsg);
    233 					}
    234 					else
    235 						printf("- %s",
    236 						    fmttime(T->logout,
    237 						    fulltime | TIMEONLY));
    238 					delta = T->logout - bp->ut_timefld;
    239 					if (delta < SECSPERDAY)
    240 						printf("  (%s)\n",
    241 						    fmttime(delta,
    242 						    fulltime | TIMEONLY | GMT));
    243 					else
    244 						printf(" (%lld+%s)\n",
    245 						    (long long)
    246 						    delta / SECSPERDAY,
    247 						    fmttime(delta,
    248 						    fulltime | TIMEONLY | GMT));
    249 				}
    250 				if (maxrec != -1 && !--maxrec)
    251 					return;
    252 			}
    253 			T->logout = bp->ut_timefld;
    254 		}
    255 	}
    256 	fulltime = 1;	/* show full time */
    257 	crmsg = fmttime(seentime, FULLTIME);
    258 	if ((ct = strrchr(file, '/')) != NULL)
    259 		ct++;
    260 	printf("\n%s begins %s\n", ct ? ct : file, crmsg);
    261 }
    262 
    263 /*
    264  * want --
    265  *	see if want this entry
    266  */
    267 static int
    268 want(struct utmp *bp, int check)
    269 {
    270 	ARG *step;
    271 
    272 	if (check) {
    273 		/*
    274 		 * when uucp and ftp log in over a network, the entry in
    275 		 * the utmp file is the name plus their process id.  See
    276 		 * etc/ftpd.c and usr.bin/uucp/uucpd.c for more information.
    277 		 */
    278 		if (!strncmp(bp->ut_line, "ftp", sizeof("ftp") - 1))
    279 			bp->ut_line[3] = '\0';
    280 		else if (!strncmp(bp->ut_line, "uucp", sizeof("uucp") - 1))
    281 			bp->ut_line[4] = '\0';
    282 	}
    283 	if (!arglist)
    284 		return (YES);
    285 
    286 	for (step = arglist; step; step = step->next)
    287 		switch(step->type) {
    288 		case HOST_TYPE:
    289 			if (!strncasecmp(step->name, bp->ut_host, HOSTSIZE))
    290 				return (YES);
    291 			break;
    292 		case TTY_TYPE:
    293 			if (!strncmp(step->name, bp->ut_line, LINESIZE))
    294 				return (YES);
    295 			break;
    296 		case USER_TYPE:
    297 			if (!strncmp(step->name, bp->ut_name, NAMESIZE))
    298 				return (YES);
    299 			break;
    300 	}
    301 	return (NO);
    302 }
    303 
    304 /*
    305  * onintr --
    306  *	on interrupt, we inform the user how far we've gotten
    307  */
    308 static void
    309 onintr(int signo)
    310 {
    311 	/* FIXME: None of this is allowed in a signal handler */
    312 	printf("\ninterrupted %s\n", fmttime(seentime, FULLTIME));
    313 	if (signo == SIGINT) {
    314 		(void)raise_default_signal(signo);
    315 		exit(EXIT_FAILURE);
    316 	}
    317 	(void)fflush(stdout);		/* fix required for rsh */
    318 }
    319