xref: /netbsd-src/libexec/tftpd/tftpd.c (revision 6a493d6bc668897c91594964a732d38505b70cbb)
1 /*	$NetBSD: tftpd.c,v 1.43 2013/10/04 07:51:48 jnemeth 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 #if 0
37 static char sccsid[] = "@(#)tftpd.c	8.1 (Berkeley) 6/4/93";
38 #else
39 __RCSID("$NetBSD: tftpd.c,v 1.43 2013/10/04 07:51:48 jnemeth Exp $");
40 #endif
41 #endif /* not lint */
42 
43 /*
44  * Trivial file transfer protocol server.
45  *
46  * This version includes many modifications by Jim Guyton
47  * <guyton@rand-unix>.
48  */
49 
50 #include <sys/param.h>
51 #include <sys/ioctl.h>
52 #include <sys/stat.h>
53 #include <sys/socket.h>
54 
55 #include <netinet/in.h>
56 #include <arpa/tftp.h>
57 #include <arpa/inet.h>
58 
59 #include <ctype.h>
60 #include <errno.h>
61 #include <fcntl.h>
62 #include <grp.h>
63 #include <netdb.h>
64 #include <pwd.h>
65 #include <setjmp.h>
66 #include <signal.h>
67 #include <stdio.h>
68 #include <stdlib.h>
69 #include <string.h>
70 #include <syslog.h>
71 #include <time.h>
72 #include <unistd.h>
73 
74 #include "tftpsubs.h"
75 
76 #define	DEFAULTUSER	"nobody"
77 
78 #define	TIMEOUT		5
79 
80 static int	peer;
81 static int	rexmtval = TIMEOUT;
82 static int	maxtimeout = 5*TIMEOUT;
83 
84 static char	buf[MAXPKTSIZE];
85 static char	ackbuf[PKTSIZE];
86 static char	oackbuf[PKTSIZE];
87 static struct	sockaddr_storage from;
88 static socklen_t	fromlen;
89 static int	debug;
90 
91 static int	tftp_opt_tsize = 0;
92 static int	tftp_blksize = SEGSIZE;
93 static int	tftp_tsize = 0;
94 
95 /*
96  * Null-terminated directory prefix list for absolute pathname requests and
97  * search list for relative pathname requests.
98  *
99  * MAXDIRS should be at least as large as the number of arguments that
100  * inetd allows (currently 20).
101  */
102 #define MAXDIRS	20
103 static struct dirlist {
104 	char	*name;
105 	int	len;
106 } dirs[MAXDIRS+1];
107 static int	suppress_naks;
108 static int	logging;
109 static int	secure;
110 static char	pathsep = '\0';
111 static char	*securedir;
112 static int	unrestricted_writes;    /* uploaded files don't have to exist */
113 
114 struct formats;
115 
116 static const char *errtomsg(int);
117 static void	nak(int);
118 __dead static void	tftp(struct tftphdr *, int);
119 __dead static void	usage(void);
120 static char	*verifyhost(struct sockaddr *);
121 __dead static void	justquit(int);
122 static void	recvfile(struct formats *, int, int);
123 static void	sendfile(struct formats *, int, int);
124 __dead static void	timer(int);
125 static const char *opcode(int);
126 static int	validate_access(char **, int);
127 
128 static struct formats {
129 	const char	*f_mode;
130 	int		(*f_validate)(char **, int);
131 	void		(*f_send)(struct formats *, int, int);
132 	void		(*f_recv)(struct formats *, int, int);
133 	int		f_convert;
134 } formats[] = {
135 	{ "netascii",	validate_access,	sendfile,	recvfile, 1 },
136 	{ "octet",	validate_access,	sendfile,	recvfile, 0 },
137 	{ .f_mode = NULL }
138 };
139 
140 static void
141 usage(void)
142 {
143 
144 	syslog(LOG_ERR,
145     "Usage: %s [-cdln] [-g group] [-p pathsep] [-s directory] [-u user] [directory ...]",
146 		    getprogname());
147 	exit(1);
148 }
149 
150 int
151 main(int argc, char *argv[])
152 {
153 	struct sockaddr_storage me;
154 	struct passwd	*pwent;
155 	struct group	*grent;
156 	struct tftphdr	*tp;
157 	const char	*tgtuser, *tgtgroup;
158 	char *ep;
159 	int	n, ch, on, fd;
160 	int	soopt;
161 	socklen_t len;
162 	uid_t	curuid, tgtuid;
163 	gid_t	curgid, tgtgid;
164 	long	nid;
165 
166 	n = 0;
167 	fd = 0;
168 	tzset();
169 	openlog("tftpd", LOG_PID | LOG_NDELAY, LOG_DAEMON);
170 	tgtuser = DEFAULTUSER;
171 	tgtgroup = NULL;
172 	curuid = getuid();
173 	curgid = getgid();
174 
175 	while ((ch = getopt(argc, argv, "cdg:lnp:s:u:")) != -1)
176 		switch (ch) {
177 		case 'c':
178 			unrestricted_writes = 1;
179 			break;
180 
181 		case 'd':
182 			debug++;
183 			break;
184 
185 		case 'g':
186 			tgtgroup = optarg;
187 			break;
188 
189 		case 'l':
190 			logging = 1;
191 			break;
192 
193 		case 'n':
194 			suppress_naks = 1;
195 			break;
196 
197 		case 'p':
198 			if (optarg[0] == '\0' || optarg[1] != '\0')
199 				usage();
200 			pathsep = optarg[0];
201 			break;
202 
203 		case 's':
204 			secure = 1;
205 			securedir = optarg;
206 			break;
207 
208 		case 'u':
209 			tgtuser = optarg;
210 			break;
211 
212 		default:
213 			usage();
214 			break;
215 		}
216 
217 	if (optind < argc) {
218 		struct dirlist *dirp;
219 
220 		/* Get list of directory prefixes. Skip relative pathnames. */
221 		for (dirp = dirs; optind < argc && dirp < &dirs[MAXDIRS];
222 		     optind++) {
223 			if (argv[optind][0] == '/') {
224 				dirp->name = argv[optind];
225 				dirp->len  = strlen(dirp->name);
226 				dirp++;
227 			}
228 		}
229 	}
230 
231 	if (*tgtuser == '\0' || (tgtgroup != NULL && *tgtgroup == '\0'))
232 		usage();
233 
234 	nid = (strtol(tgtuser, &ep, 10));
235 	if (*ep == '\0') {
236 		if ((uid_t)nid > UID_MAX) {
237 			syslog(LOG_ERR, "uid %ld is too large", nid);
238 			exit(1);
239 		}
240 		pwent = getpwuid((uid_t)nid);
241 	} else
242 		pwent = getpwnam(tgtuser);
243 	if (pwent == NULL) {
244 		syslog(LOG_ERR, "unknown user `%s'", tgtuser);
245 		exit(1);
246 	}
247 	tgtuid = pwent->pw_uid;
248 	tgtgid = pwent->pw_gid;
249 
250 	if (tgtgroup != NULL) {
251 		nid = (strtol(tgtgroup, &ep, 10));
252 		if (*ep == '\0') {
253 			if ((uid_t)nid > GID_MAX) {
254 				syslog(LOG_ERR, "gid %ld is too large", nid);
255 				exit(1);
256 			}
257 			grent = getgrgid((gid_t)nid);
258 		} else
259 			grent = getgrnam(tgtgroup);
260 		if (grent != NULL)
261 			tgtgid = grent->gr_gid;
262 		else {
263 			syslog(LOG_ERR, "unknown group `%s'", tgtgroup);
264 			exit(1);
265 		}
266 	}
267 
268 	if (secure) {
269 		if (chdir(securedir) < 0) {
270 			syslog(LOG_ERR, "chdir %s: %m", securedir);
271 			exit(1);
272 		}
273 		if (chroot(".")) {
274 			syslog(LOG_ERR, "chroot: %m");
275 			exit(1);
276 		}
277 	}
278 
279 	if (logging)
280 		syslog(LOG_DEBUG, "running as user `%s' (%d), group `%s' (%d)",
281 		    tgtuser, tgtuid, tgtgroup ? tgtgroup : "(unspecified)",
282 		    tgtgid);
283 	if (curgid != tgtgid) {
284 		if (setgid(tgtgid)) {
285 			syslog(LOG_ERR, "setgid to %d: %m", (int)tgtgid);
286 			exit(1);
287 		}
288 		if (setgroups(0, NULL)) {
289 			syslog(LOG_ERR, "setgroups: %m");
290 			exit(1);
291 		}
292 	}
293 
294 	if (curuid != tgtuid) {
295 		if (setuid(tgtuid)) {
296 			syslog(LOG_ERR, "setuid to %d: %m", (int)tgtuid);
297 			exit(1);
298 		}
299 	}
300 
301 	on = 1;
302 	if (ioctl(fd, FIONBIO, &on) < 0) {
303 		syslog(LOG_ERR, "ioctl(FIONBIO): %m");
304 		exit(1);
305 	}
306 	fromlen = sizeof (from);
307 	n = recvfrom(fd, buf, sizeof (buf), 0,
308 	    (struct sockaddr *)&from, &fromlen);
309 	if (n < 0) {
310 		syslog(LOG_ERR, "recvfrom: %m");
311 		exit(1);
312 	}
313 	/*
314 	 * Now that we have read the message out of the UDP
315 	 * socket, we fork and exit.  Thus, inetd will go back
316 	 * to listening to the tftp port, and the next request
317 	 * to come in will start up a new instance of tftpd.
318 	 *
319 	 * We do this so that inetd can run tftpd in "wait" mode.
320 	 * The problem with tftpd running in "nowait" mode is that
321 	 * inetd may get one or more successful "selects" on the
322 	 * tftp port before we do our receive, so more than one
323 	 * instance of tftpd may be started up.  Worse, if tftpd
324 	 * break before doing the above "recvfrom", inetd would
325 	 * spawn endless instances, clogging the system.
326 	 */
327 	{
328 		int pid;
329 		int i;
330 		socklen_t j;
331 
332 		for (i = 1; i < 20; i++) {
333 		    pid = fork();
334 		    if (pid < 0) {
335 				sleep(i);
336 				/*
337 				 * flush out to most recently sent request.
338 				 *
339 				 * This may drop some request, but those
340 				 * will be resent by the clients when
341 				 * they timeout.  The positive effect of
342 				 * this flush is to (try to) prevent more
343 				 * than one tftpd being started up to service
344 				 * a single request from a single client.
345 				 */
346 				j = sizeof from;
347 				i = recvfrom(fd, buf, sizeof (buf), 0,
348 				    (struct sockaddr *)&from, &j);
349 				if (i > 0) {
350 					n = i;
351 					fromlen = j;
352 				}
353 		    } else {
354 				break;
355 		    }
356 		}
357 		if (pid < 0) {
358 			syslog(LOG_ERR, "fork: %m");
359 			exit(1);
360 		} else if (pid != 0) {
361 			exit(0);
362 		}
363 	}
364 
365 	/*
366 	 * remember what address this was sent to, so we can respond on the
367 	 * same interface
368 	 */
369 	len = sizeof(me);
370 	if (getsockname(fd, (struct sockaddr *)&me, &len) == 0) {
371 		switch (me.ss_family) {
372 		case AF_INET:
373 			((struct sockaddr_in *)&me)->sin_port = 0;
374 			break;
375 		case AF_INET6:
376 			((struct sockaddr_in6 *)&me)->sin6_port = 0;
377 			break;
378 		default:
379 			/* unsupported */
380 			break;
381 		}
382 	} else {
383 		memset(&me, 0, sizeof(me));
384 		me.ss_family = from.ss_family;
385 		me.ss_len = from.ss_len;
386 	}
387 
388 	alarm(0);
389 	close(fd);
390 	close(1);
391 	peer = socket(from.ss_family, SOCK_DGRAM, 0);
392 	if (peer < 0) {
393 		syslog(LOG_ERR, "socket: %m");
394 		exit(1);
395 	}
396 	if (bind(peer, (struct sockaddr *)&me, me.ss_len) < 0) {
397 		syslog(LOG_ERR, "bind: %m");
398 		exit(1);
399 	}
400 	if (connect(peer, (struct sockaddr *)&from, from.ss_len) < 0) {
401 		syslog(LOG_ERR, "connect: %m");
402 		exit(1);
403 	}
404 	soopt = 65536;	/* larger than we'll ever need */
405 	if (setsockopt(peer, SOL_SOCKET, SO_SNDBUF, (void *) &soopt, sizeof(soopt)) < 0) {
406 		syslog(LOG_ERR, "set SNDBUF: %m");
407 		exit(1);
408 	}
409 	if (setsockopt(peer, SOL_SOCKET, SO_RCVBUF, (void *) &soopt, sizeof(soopt)) < 0) {
410 		syslog(LOG_ERR, "set RCVBUF: %m");
411 		exit(1);
412 	}
413 
414 	tp = (struct tftphdr *)buf;
415 	tp->th_opcode = ntohs(tp->th_opcode);
416 	if (tp->th_opcode == RRQ || tp->th_opcode == WRQ)
417 		tftp(tp, n);
418 	exit(1);
419 }
420 
421 static int
422 blk_handler(struct tftphdr *tp, const char *val, char *ack, size_t asize,
423     size_t *ackl, int *ecode)
424 {
425 	unsigned long bsize;
426 	char *endp;
427 	int l;
428 
429 	/*
430 	 * On these failures, we could just ignore the blocksize option.
431 	 * Perhaps that should be a command-line option.
432 	 */
433 	errno = 0;
434 	bsize = strtoul(val, &endp, 10);
435 	if ((bsize == ULONG_MAX && errno == ERANGE) || *endp) {
436 		syslog(LOG_NOTICE, "%s: %s request for %s: "
437 			"illegal value %s for blksize option",
438 			verifyhost((struct sockaddr *)&from),
439 			tp->th_opcode == WRQ ? "write" : "read",
440 			tp->th_stuff, val);
441 		*ecode = EBADOP;
442 		return -1;
443 	}
444 	if (bsize < 8 || bsize > 65464) {
445 		syslog(LOG_NOTICE, "%s: %s request for %s: "
446 			"out of range value %s for blksize option",
447 			verifyhost((struct sockaddr *)&from),
448 			tp->th_opcode == WRQ ? "write" : "read",
449 			tp->th_stuff, val);
450 		*ecode = EBADOP;
451 		return -1;
452 	}
453 
454 	tftp_blksize = bsize;
455 	if (asize > *ackl && (l = snprintf(ack + *ackl, asize - *ackl,
456 	    "blksize%c%lu%c", 0, bsize, 0)) > 0) {
457 		*ackl += l;
458 	} else {
459 		*ecode = EBADOP;
460 		return -1;
461 	}
462 
463 	return 0;
464 }
465 
466 static int
467 timeout_handler(struct tftphdr *tp, const char *val, char *ack, size_t asize,
468 		size_t *ackl, int *ecode)
469 {
470 	unsigned long tout;
471 	char *endp;
472 	int l;
473 
474 	errno = 0;
475 	tout = strtoul(val, &endp, 10);
476 	if ((tout == ULONG_MAX && errno == ERANGE) || *endp) {
477 		syslog(LOG_NOTICE, "%s: %s request for %s: "
478 			"illegal value %s for timeout option",
479 			verifyhost((struct sockaddr *)&from),
480 			tp->th_opcode == WRQ ? "write" : "read",
481 			tp->th_stuff, val);
482 		*ecode = EBADOP;
483 		return -1;
484 	}
485 	if (tout < 1 || tout > 255) {
486 		syslog(LOG_NOTICE, "%s: %s request for %s: "
487 			"out of range value %s for timeout option",
488 			verifyhost((struct sockaddr *)&from),
489 			tp->th_opcode == WRQ ? "write" : "read",
490 			tp->th_stuff, val);
491 		return 0;
492 	}
493 
494 	rexmtval = tout;
495 	if (asize > *ackl && (l = snprintf(ack + *ackl, asize - *ackl,
496 	    "timeout%c%lu%c", 0, tout, 0)) > 0)
497 		*ackl += l;
498 	else
499 		return -1;
500 	/*
501 	 * Arbitrarily pick a maximum timeout on a request to 3
502 	 * retransmissions if the interval timeout is more than
503 	 * one minute.  Longest possible timeout is therefore
504 	 * 3 * 255 - 1, or 764 seconds.
505 	 */
506 	if (rexmtval > 60) {
507 		maxtimeout = rexmtval * 3;
508 	} else {
509 		maxtimeout = rexmtval * 5;
510 	}
511 
512 	return 0;
513 }
514 
515 static int
516 tsize_handler(struct tftphdr *tp, const char *val, char *ack, size_t asize,
517     size_t *ackl, int *ecode)
518 {
519 	unsigned long fsize;
520 	char *endp;
521 
522 	/*
523 	 * Maximum file even with extended tftp is 65535 blocks of
524 	 * length 65464, or 4290183240 octets (4784056 less than 2^32).
525 	 * unsigned long is at least 32 bits on all NetBSD archs.
526 	 */
527 
528 	errno = 0;
529 	fsize = strtoul(val, &endp, 10);
530 	if ((fsize == ULONG_MAX && errno == ERANGE) || *endp) {
531 		syslog(LOG_NOTICE, "%s: %s request for %s: "
532 			"illegal value %s for tsize option",
533 			verifyhost((struct sockaddr *)&from),
534 			tp->th_opcode == WRQ ? "write" : "read",
535 			tp->th_stuff, val);
536 		*ecode = EBADOP;
537 		return -1;
538 	}
539 	if (fsize > (unsigned long) 65535 * 65464) {
540 		syslog(LOG_NOTICE, "%s: %s request for %s: "
541 			"out of range value %s for tsize option",
542 			verifyhost((struct sockaddr *)&from),
543 			tp->th_opcode == WRQ ? "write" : "read",
544 			tp->th_stuff, val);
545 		*ecode = EBADOP;
546 		return -1;
547 	}
548 
549 	tftp_opt_tsize = 1;
550 	tftp_tsize = fsize;
551 	/*
552 	 * We will report this later -- either replying with the fsize (WRQ)
553 	 * or replying with the actual filesize (RRQ).
554 	 */
555 
556 	return 0;
557 }
558 
559 static const struct tftp_options {
560 	const char *o_name;
561 	int (*o_handler)(struct tftphdr *, const char *, char *, size_t,
562 			 size_t *, int *);
563 } options[] = {
564 	{ "blksize", blk_handler },
565 	{ "timeout", timeout_handler },
566 	{ "tsize", tsize_handler },
567 	{ .o_name = NULL }
568 };
569 
570 /*
571  * Get options for an extended tftp session.  Stuff the ones we
572  * recognize in oackbuf.
573  */
574 static int
575 get_options(struct tftphdr *tp, char *cp, int size, char *ackb, size_t asize,
576     size_t *alen, int *ecode)
577 {
578 	const struct tftp_options *op;
579 	char *option, *value, *endp;
580 	int r, rv=0;
581 
582 	endp = cp + size;
583 	while (cp < endp) {
584 		option = cp;
585 		while (*cp && cp < endp) {
586 			*cp = tolower((unsigned char)*cp);
587 			cp++;
588 		}
589 		if (*cp) {
590 			/* if we have garbage at the end, just ignore it */
591 			break;
592 		}
593 		cp++;	/* skip over NUL */
594 		value = cp;
595 		while (*cp && cp < endp) {
596 			cp++;
597 		}
598 		if (*cp) {
599 			/* if we have garbage at the end, just ignore it */
600 			break;
601 		}
602 		cp++;
603 		for (op = options; op->o_name; op++) {
604 			if (strcmp(op->o_name, option) == 0)
605 				break;
606 		}
607 		if (op->o_name) {
608 			r = op->o_handler(tp, value, ackb, asize, alen, ecode);
609 			if (r < 0) {
610 				rv = -1;
611 				break;
612 			}
613 			rv++;
614 		} /* else ignore unknown options */
615 	}
616 
617 	return rv;
618 }
619 
620 /*
621  * Handle initial connection protocol.
622  */
623 static void
624 tftp(struct tftphdr *tp, int size)
625 {
626 	struct formats *pf;
627 	char	*cp;
628 	char	*filename, *mode;
629 	int	 first, ecode, etftp = 0, r;
630 	size_t alen;
631 
632 	ecode = 0;	/* XXX gcc */
633 	first = 1;
634 	mode = NULL;
635 
636 	filename = cp = tp->th_stuff;
637 again:
638 	while (cp < buf + size) {
639 		if (*cp == '\0')
640 			break;
641 		cp++;
642 	}
643 	if (*cp != '\0') {
644 		nak(EBADOP);
645 		exit(1);
646 	}
647 	if (first) {
648 		mode = ++cp;
649 		first = 0;
650 		goto again;
651 	}
652 	for (cp = mode; *cp; cp++)
653 		*cp = tolower((unsigned char)*cp);
654 	for (pf = formats; pf->f_mode; pf++)
655 		if (strcmp(pf->f_mode, mode) == 0)
656 			break;
657 	if (pf->f_mode == 0) {
658 		nak(EBADOP);
659 		exit(1);
660 	}
661 	/*
662 	 * cp currently points to the NUL byte following the mode.
663 	 *
664 	 * If we have some valid options, then let's assume that we're
665 	 * now dealing with an extended tftp session.  Note that if we
666 	 * don't get any options, then we *must* assume that we do not
667 	 * have an extended tftp session.  If we get options, we fill
668 	 * in the ack buf to acknowledge them.  If we skip that, then
669 	 * the client *must* assume that we are not using an extended
670 	 * session.
671 	 */
672 	size -= (++cp - (char *) tp);
673 	if (size > 0 && *cp) {
674 		alen = 2; /* Skip over opcode */
675 		r = get_options(tp, cp, size, oackbuf, sizeof(oackbuf),
676 		    &alen, &ecode);
677 		if (r > 0) {
678 			etftp = 1;
679 		} else if (r < 0) {
680 			nak(ecode);
681 			exit(1);
682 		}
683 	}
684 	/*
685 	 * Globally replace the path separator given in the -p option
686 	 * with / to cope with clients expecting a non-unix path separator.
687 	 */
688 	if (pathsep != '\0') {
689 		for (cp = filename; *cp != '\0'; ++cp) {
690 			if (*cp == pathsep)
691 				*cp = '/';
692 		}
693 	}
694 	ecode = (*pf->f_validate)(&filename, tp->th_opcode);
695 	if (logging) {
696 		syslog(LOG_INFO, "%s: %s request for %s: %s",
697 			verifyhost((struct sockaddr *)&from),
698 			tp->th_opcode == WRQ ? "write" : "read",
699 			filename, errtomsg(ecode));
700 	}
701 	if (ecode) {
702 		/*
703 		 * Avoid storms of naks to a RRQ broadcast for a relative
704 		 * bootfile pathname from a diskless Sun.
705 		 */
706 		if (suppress_naks && *filename != '/' && ecode == ENOTFOUND)
707 			exit(0);
708 		nak(ecode);
709 		exit(1);
710 	}
711 
712 	if (etftp) {
713 		struct tftphdr *oack_h;
714 
715 		if (tftp_opt_tsize) {
716 			int l;
717 
718 			if (sizeof(oackbuf) > alen &&
719 			    (l = snprintf(oackbuf + alen,
720 			    sizeof(oackbuf) - alen, "tsize%c%u%c", 0,
721 			    tftp_tsize, 0)) > 0)
722 				alen += l;
723 		}
724 		oack_h = (struct tftphdr *) oackbuf;
725 		oack_h->th_opcode = htons(OACK);
726 	}
727 
728 	if (tp->th_opcode == WRQ)
729 		(*pf->f_recv)(pf, etftp, alen);
730 	else
731 		(*pf->f_send)(pf, etftp, alen);
732 	exit(0);
733 }
734 
735 
736 FILE *file;
737 
738 /*
739  * Validate file access.  Since we
740  * have no uid or gid, for now require
741  * file to exist and be publicly
742  * readable/writable.
743  * If we were invoked with arguments
744  * from inetd then the file must also be
745  * in one of the given directory prefixes.
746  */
747 int
748 validate_access(char **filep, int mode)
749 {
750 	struct stat	 stbuf;
751 	struct dirlist	*dirp;
752 	static char	 pathname[MAXPATHLEN];
753 	char		*filename;
754 	int		 fd;
755 	int		 create = 0;
756 	int		 trunc = 0;
757 
758 	filename = *filep;
759 
760 	/*
761 	 * Prevent tricksters from getting around the directory restrictions
762 	 */
763 	if (strstr(filename, "/../"))
764 		return (EACCESS);
765 
766 	if (*filename == '/') {
767 		/*
768 		 * Allow the request if it's in one of the approved locations.
769 		 * Special case: check the null prefix ("/") by looking
770 		 * for length = 1 and relying on the arg. processing that
771 		 * it's a /.
772 		 */
773 		for (dirp = dirs; dirp->name != NULL; dirp++) {
774 			if (dirp->len == 1 ||
775 			    (!strncmp(filename, dirp->name, dirp->len) &&
776 			     filename[dirp->len] == '/'))
777 				    break;
778 		}
779 		/* If directory list is empty, allow access to any file */
780 		if (dirp->name == NULL && dirp != dirs)
781 			return (EACCESS);
782 		if (stat(filename, &stbuf) < 0)
783 			return (errno == ENOENT ? ENOTFOUND : EACCESS);
784 		if (!S_ISREG(stbuf.st_mode))
785 			return (ENOTFOUND);
786 		if (mode == RRQ) {
787 			if ((stbuf.st_mode & S_IROTH) == 0)
788 				return (EACCESS);
789 		} else {
790 			if ((stbuf.st_mode & S_IWOTH) == 0)
791 				return (EACCESS);
792 		}
793 	} else {
794 		/*
795 		 * Relative file name: search the approved locations for it.
796 		 */
797 
798 		if (!strncmp(filename, "../", 3))
799 			return (EACCESS);
800 
801 		/*
802 		 * Find the first file that exists in any of the directories,
803 		 * check access on it.
804 		 */
805 		if (dirs[0].name != NULL) {
806 			for (dirp = dirs; dirp->name != NULL; dirp++) {
807 				snprintf(pathname, sizeof pathname, "%s/%s",
808 				    dirp->name, filename);
809 				if (stat(pathname, &stbuf) == 0 &&
810 				    (stbuf.st_mode & S_IFMT) == S_IFREG) {
811 					break;
812 				}
813 			}
814 			if (dirp->name == NULL)
815 				return (ENOTFOUND);
816 			if (mode == RRQ && !(stbuf.st_mode & S_IROTH))
817 				return (EACCESS);
818 			if (mode == WRQ && !(stbuf.st_mode & S_IWOTH))
819 				return (EACCESS);
820 			*filep = filename = pathname;
821 		} else {
822 			int stat_rc;
823 
824 			/*
825 			 * If there's no directory list, take our cue from the
826 			 * absolute file request check above (*filename == '/'),
827 			 * and allow access to anything.
828 			 */
829 			stat_rc = stat(filename, &stbuf);
830 			if (mode == RRQ) {
831 				/* Read request */
832 				if (stat_rc < 0)
833 				       return (errno == ENOENT ? ENOTFOUND : EACCESS);
834 				if (!S_ISREG(stbuf.st_mode))
835 				       return (ENOTFOUND);
836 				if ((stbuf.st_mode & S_IROTH) == 0)
837 					return (EACCESS);
838 			} else {
839 				if (stat_rc < 0) {
840 				       /* Can't stat */
841 				       if (errno == EACCES) {
842 					       /* Permission denied */
843 					       return EACCESS;
844 				       } else {
845 					       /* Not there */
846 					       if (unrestricted_writes) {
847 						       /* need to creat new file! */
848 						       create = O_CREAT;
849 					       } else {
850 						       /* Permission denied */
851 						       return EACCESS;
852 					       }
853 				       }
854 				} else {
855 				       /* Can stat */
856 				       if ((stbuf.st_mode & S_IWOTH) == 0) {
857 					       return (EACCESS);
858 				       }
859 				       trunc = O_TRUNC;
860 				}
861 			}
862 			*filep = filename;
863 		}
864 	}
865 
866 	if (tftp_opt_tsize && mode == RRQ)
867 		tftp_tsize = (unsigned long) stbuf.st_size;
868 
869 	fd = open(filename, mode == RRQ ? O_RDONLY : O_WRONLY | trunc | create,
870 			0644); /* debatable */
871 	if (fd < 0)
872 		return (errno + 100);
873 	file = fdopen(fd, (mode == RRQ)? "r":"w");
874 	if (file == NULL) {
875 		close(fd);
876 		return (errno + 100);
877 	}
878 	return (0);
879 }
880 
881 static int	timeout;
882 static jmp_buf	timeoutbuf;
883 
884 static void
885 timer(int dummy)
886 {
887 
888 	timeout += rexmtval;
889 	if (timeout >= maxtimeout)
890 		exit(1);
891 	longjmp(timeoutbuf, 1);
892 }
893 
894 static const char *
895 opcode(int code)
896 {
897 	static char obuf[64];
898 
899 	switch (code) {
900 	case RRQ:
901 		return "RRQ";
902 	case WRQ:
903 		return "WRQ";
904 	case DATA:
905 		return "DATA";
906 	case ACK:
907 		return "ACK";
908 	case ERROR:
909 		return "ERROR";
910 	case OACK:
911 		return "OACK";
912 	default:
913 		(void)snprintf(obuf, sizeof(obuf), "*code 0x%x*", code);
914 		return obuf;
915 	}
916 }
917 
918 /*
919  * Send the requested file.
920  */
921 static void
922 sendfile(struct formats *pf, volatile int etftp, int acklength)
923 {
924 	volatile unsigned int block;
925 	struct tftphdr	*dp;
926 	struct tftphdr	*ap;    /* ack packet */
927 	volatile int	 size;
928 	int n;
929 
930 	signal(SIGALRM, timer);
931 	ap = (struct tftphdr *)ackbuf;
932 	if (etftp) {
933 		dp = (struct tftphdr *)oackbuf;
934 		size = acklength - 4;
935 		block = 0;
936 	} else {
937 		dp = r_init();
938 		size = 0;
939 		block = 1;
940 	}
941 
942 	do {
943 		if (block > 0) {
944 			size = readit(file, &dp, tftp_blksize, pf->f_convert);
945 			if (size < 0) {
946 				nak(errno + 100);
947 				goto abort;
948 			}
949 			dp->th_opcode = htons((u_short)DATA);
950 			dp->th_block = htons((u_short)block);
951 		}
952 		timeout = 0;
953 		(void)setjmp(timeoutbuf);
954 
955 send_data:
956 		if (!etftp && debug)
957 			syslog(LOG_DEBUG, "Send DATA %u", block);
958 		if ((n = send(peer, dp, size + 4, 0)) != size + 4) {
959 			syslog(LOG_ERR, "tftpd: write: %m");
960 			goto abort;
961 		}
962 		if (block)
963 			read_ahead(file, tftp_blksize, pf->f_convert);
964 		for ( ; ; ) {
965 			alarm(rexmtval);        /* read the ack */
966 			n = recv(peer, ackbuf, tftp_blksize, 0);
967 			alarm(0);
968 			if (n < 0) {
969 				syslog(LOG_ERR, "tftpd: read: %m");
970 				goto abort;
971 			}
972 			ap->th_opcode = ntohs((u_short)ap->th_opcode);
973 			ap->th_block = ntohs((u_short)ap->th_block);
974 			switch (ap->th_opcode) {
975 			case ERROR:
976 				goto abort;
977 
978 			case ACK:
979 				if (etftp && ap->th_block == 0) {
980 					etftp = 0;
981 					acklength = 0;
982 					dp = r_init();
983 					goto done;
984 				}
985 				if (ap->th_block == (u_short)block)
986 					goto done;
987 				if (debug)
988 					syslog(LOG_DEBUG, "Resync ACK %u != %u",
989 					    (unsigned int)ap->th_block, block);
990 				/* Re-synchronize with the other side */
991 				(void) synchnet(peer, tftp_blksize);
992 				if (ap->th_block == (u_short)(block - 1))
993 					goto send_data;
994 			default:
995 				syslog(LOG_INFO, "Received %s in sendfile\n",
996 				    opcode(dp->th_opcode));
997 			}
998 
999 		}
1000 done:
1001 		if (debug)
1002 			syslog(LOG_DEBUG, "Received ACK for block %u", block);
1003 		if (block == UINT16_MAX && size == tftp_blksize)
1004 			syslog(LOG_WARNING,
1005 			    "Block number wrapped (hint: increase block size)");
1006 		block++;
1007 	} while (size == tftp_blksize || block == 1);
1008 abort:
1009 	(void) fclose(file);
1010 }
1011 
1012 static void
1013 justquit(int dummy)
1014 {
1015 
1016 	exit(0);
1017 }
1018 
1019 /*
1020  * Receive a file.
1021  */
1022 static void
1023 recvfile(struct formats *pf, volatile int etftp, volatile int acklength)
1024 {
1025 	volatile unsigned int block;
1026 	struct tftphdr	*dp;
1027 	struct tftphdr	*ap;    /* ack buffer */
1028 	volatile int size;
1029 	int n;
1030 
1031 	signal(SIGALRM, timer);
1032 	dp = w_init();
1033 	ap = (struct tftphdr *)oackbuf;
1034 	block = 0;
1035 	do {
1036 		timeout = 0;
1037 		if (etftp == 0) {
1038 			ap = (struct tftphdr *)ackbuf;
1039 			ap->th_opcode = htons((u_short)ACK);
1040 			ap->th_block = htons((u_short)block);
1041 			acklength = 4;
1042 		}
1043 		if (debug)
1044 			syslog(LOG_DEBUG, "Sending ACK for block %u\n", block);
1045 		if (block == UINT16_MAX)
1046 			syslog(LOG_WARNING,
1047 			    "Block number wrapped (hint: increase block size)");
1048 		block++;
1049 		(void) setjmp(timeoutbuf);
1050 send_ack:
1051 		ap = (struct tftphdr *) (etftp ? oackbuf : ackbuf);
1052 		if (send(peer, ap, acklength, 0) != acklength) {
1053 			syslog(LOG_ERR, "tftpd: write: %m");
1054 			goto abort;
1055 		}
1056 		write_behind(file, pf->f_convert);
1057 		for ( ; ; ) {
1058 			alarm(rexmtval);
1059 			n = recv(peer, dp, tftp_blksize + 4, 0);
1060 			alarm(0);
1061 			if (n < 0) {            /* really? */
1062 				syslog(LOG_ERR, "tftpd: read: %m");
1063 				goto abort;
1064 			}
1065 			etftp = 0;
1066 			dp->th_opcode = ntohs((u_short)dp->th_opcode);
1067 			dp->th_block = ntohs((u_short)dp->th_block);
1068 			if (debug)
1069 				syslog(LOG_DEBUG, "Received %s for block %u",
1070 				    opcode(dp->th_opcode),
1071 				    (unsigned int)dp->th_block);
1072 
1073 			switch (dp->th_opcode) {
1074 			case ERROR:
1075 				goto abort;
1076 			case DATA:
1077 				if (dp->th_block == block)
1078 					goto done;   /* normal */
1079 				if (debug)
1080 					syslog(LOG_DEBUG, "Resync %u != %u",
1081 					    (unsigned int)dp->th_block, block);
1082 				/* Re-synchronize with the other side */
1083 				(void) synchnet(peer, tftp_blksize);
1084 				if (dp->th_block == (block-1))
1085 					goto send_ack;          /* rexmit */
1086 				break;
1087 			default:
1088 				syslog(LOG_INFO, "Received %s in recvfile\n",
1089 				    opcode(dp->th_opcode));
1090 				break;
1091 			}
1092 		}
1093 done:
1094 		if (debug)
1095 			syslog(LOG_DEBUG, "Got block %u", block);
1096 		/*  size = write(file, dp->th_data, n - 4); */
1097 		size = writeit(file, &dp, n - 4, pf->f_convert);
1098 		if (size != (n-4)) {                    /* ahem */
1099 			if (size < 0) nak(errno + 100);
1100 			else nak(ENOSPACE);
1101 			goto abort;
1102 		}
1103 	} while (size == tftp_blksize);
1104 	write_behind(file, pf->f_convert);
1105 	(void) fclose(file);            /* close data file */
1106 
1107 	ap->th_opcode = htons((u_short)ACK);    /* send the "final" ack */
1108 	ap->th_block = htons((u_short)(block));
1109 	if (debug)
1110 		syslog(LOG_DEBUG, "Send final ACK %u", block);
1111 	(void) send(peer, ackbuf, 4, 0);
1112 
1113 	signal(SIGALRM, justquit);      /* just quit on timeout */
1114 	alarm(rexmtval);
1115 	n = recv(peer, buf, sizeof (buf), 0); /* normally times out and quits */
1116 	alarm(0);
1117 	if (n >= 4 &&                   /* if read some data */
1118 	    dp->th_opcode == DATA &&    /* and got a data block */
1119 	    block == dp->th_block) {	/* then my last ack was lost */
1120 		(void) send(peer, ackbuf, 4, 0);     /* resend final ack */
1121 	}
1122 abort:
1123 	return;
1124 }
1125 
1126 const struct errmsg {
1127 	int		 e_code;
1128 	const char	*e_msg;
1129 } errmsgs[] = {
1130 	{ EUNDEF,	"Undefined error code" },
1131 	{ ENOTFOUND,	"File not found" },
1132 	{ EACCESS,	"Access violation" },
1133 	{ ENOSPACE,	"Disk full or allocation exceeded" },
1134 	{ EBADOP,	"Illegal TFTP operation" },
1135 	{ EBADID,	"Unknown transfer ID" },
1136 	{ EEXISTS,	"File already exists" },
1137 	{ ENOUSER,	"No such user" },
1138 	{ EOPTNEG,	"Option negotiation failed" },
1139 	{ -1,		0 }
1140 };
1141 
1142 static const char *
1143 errtomsg(int error)
1144 {
1145 	static char ebuf[20];
1146 	const struct errmsg *pe;
1147 
1148 	if (error == 0)
1149 		return ("success");
1150 	for (pe = errmsgs; pe->e_code >= 0; pe++)
1151 		if (pe->e_code == error)
1152 			return (pe->e_msg);
1153 	snprintf(ebuf, sizeof(ebuf), "error %d", error);
1154 	return (ebuf);
1155 }
1156 
1157 /*
1158  * Send a nak packet (error message).
1159  * Error code passed in is one of the
1160  * standard TFTP codes, or a UNIX errno
1161  * offset by 100.
1162  */
1163 static void
1164 nak(int error)
1165 {
1166 	const struct errmsg *pe;
1167 	struct tftphdr *tp;
1168 	int	length;
1169 	size_t	msglen;
1170 
1171 	tp = (struct tftphdr *)buf;
1172 	tp->th_opcode = htons((u_short)ERROR);
1173 	msglen = sizeof(buf) - (&tp->th_msg[0] - buf);
1174 	for (pe = errmsgs; pe->e_code >= 0; pe++)
1175 		if (pe->e_code == error)
1176 			break;
1177 	if (pe->e_code < 0) {
1178 		tp->th_code = EUNDEF;   /* set 'undef' errorcode */
1179 		strlcpy(tp->th_msg, strerror(error - 100), msglen);
1180 	} else {
1181 		tp->th_code = htons((u_short)error);
1182 		strlcpy(tp->th_msg, pe->e_msg, msglen);
1183 	}
1184 	if (debug)
1185 		syslog(LOG_DEBUG, "Send NACK %s", tp->th_msg);
1186 	length = strlen(tp->th_msg);
1187 	msglen = &tp->th_msg[length + 1] - buf;
1188 	if (send(peer, buf, msglen, 0) != (ssize_t)msglen)
1189 		syslog(LOG_ERR, "nak: %m");
1190 }
1191 
1192 static char *
1193 verifyhost(struct sockaddr *fromp)
1194 {
1195 	static char hbuf[MAXHOSTNAMELEN];
1196 
1197 	if (getnameinfo(fromp, fromp->sa_len, hbuf, sizeof(hbuf), NULL, 0, 0))
1198 		strlcpy(hbuf, "?", sizeof(hbuf));
1199 	return (hbuf);
1200 }
1201