1 /*
2  * Copyright (c) 1983, 1993
3  *	The Regents of the University of California.  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  * 3. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 #ifndef lint
31 static const char copyright[] =
32 "@(#) Copyright (c) 1983, 1993\n\
33 	The Regents of the University of California.  All rights reserved.\n";
34 #endif /* not lint */
35 
36 #ifndef lint
37 #if 0
38 static char sccsid[] = "@(#)tftpd.c	8.1 (Berkeley) 6/4/93";
39 #endif
40 #endif /* not lint */
41 #include <sys/cdefs.h>
42 __FBSDID("$FreeBSD: stable/10/libexec/tftpd/tftpd.c 339059 2018-10-01 16:08:27Z asomers $");
43 
44 /*
45  * Trivial file transfer protocol server.
46  *
47  * This version includes many modifications by Jim Guyton
48  * <guyton@rand-unix>.
49  */
50 
51 #include <sys/param.h>
52 #include <sys/ioctl.h>
53 #include <sys/stat.h>
54 #include <sys/socket.h>
55 
56 #include <netinet/in.h>
57 #include <arpa/tftp.h>
58 
59 #include <ctype.h>
60 #include <errno.h>
61 #include <fcntl.h>
62 #include <netdb.h>
63 #include <pwd.h>
64 #include <stdint.h>
65 #include <stdio.h>
66 #include <stdlib.h>
67 #include <string.h>
68 #include <syslog.h>
69 #include <unistd.h>
70 
71 #include "tftp-file.h"
72 #include "tftp-io.h"
73 #include "tftp-utils.h"
74 #include "tftp-transfer.h"
75 #include "tftp-options.h"
76 
77 #ifdef	LIBWRAP
78 #include <tcpd.h>
79 #endif
80 
81 static void	tftp_wrq(int peer, char *, ssize_t);
82 static void	tftp_rrq(int peer, char *, ssize_t);
83 
84 /*
85  * Null-terminated directory prefix list for absolute pathname requests and
86  * search list for relative pathname requests.
87  *
88  * MAXDIRS should be at least as large as the number of arguments that
89  * inetd allows (currently 20).
90  */
91 #define MAXDIRS	20
92 static struct dirlist {
93 	const char	*name;
94 	int	len;
95 } dirs[MAXDIRS+1];
96 static int	suppress_naks;
97 static int	logging;
98 static int	ipchroot;
99 static int	create_new = 0;
100 static const char *newfile_format = "%Y%m%d";
101 static int	increase_name = 0;
102 static mode_t	mask = S_IWGRP | S_IWOTH;
103 
104 struct formats;
105 static void	tftp_recvfile(int peer, const char *mode);
106 static void	tftp_xmitfile(int peer, const char *mode);
107 static int	validate_access(int peer, char **, int);
108 static char	peername[NI_MAXHOST];
109 
110 static FILE *file;
111 
112 static struct formats {
113 	const char	*f_mode;
114 	int	f_convert;
115 } formats[] = {
116 	{ "netascii",	1 },
117 	{ "octet",	0 },
118 	{ NULL,		0 }
119 };
120 
121 int
main(int argc,char * argv[])122 main(int argc, char *argv[])
123 {
124 	struct tftphdr *tp;
125 	int		peer;
126 	socklen_t	peerlen, len;
127 	ssize_t		n;
128 	int		ch;
129 	char		*chroot_dir = NULL;
130 	struct passwd	*nobody;
131 	const char	*chuser = "nobody";
132 	char		recvbuffer[MAXPKTSIZE];
133 	int		allow_ro = 1, allow_wo = 1;
134 
135 	tzset();			/* syslog in localtime */
136 	acting_as_client = 0;
137 
138 	tftp_openlog("tftpd", LOG_PID | LOG_NDELAY, LOG_FTP);
139 	while ((ch = getopt(argc, argv, "cCd:F:lnoOp:s:u:U:wW")) != -1) {
140 		switch (ch) {
141 		case 'c':
142 			ipchroot = 1;
143 			break;
144 		case 'C':
145 			ipchroot = 2;
146 			break;
147 		case 'd':
148 			if (atoi(optarg) != 0)
149 				debug += atoi(optarg);
150 			else
151 				debug |= debug_finds(optarg);
152 			break;
153 		case 'F':
154 			newfile_format = optarg;
155 			break;
156 		case 'l':
157 			logging = 1;
158 			break;
159 		case 'n':
160 			suppress_naks = 1;
161 			break;
162 		case 'o':
163 			options_rfc_enabled = 0;
164 			break;
165 		case 'O':
166 			options_extra_enabled = 0;
167 			break;
168 		case 'p':
169 			packetdroppercentage = atoi(optarg);
170 			tftp_log(LOG_INFO,
171 			    "Randomly dropping %d out of 100 packets",
172 			    packetdroppercentage);
173 			break;
174 		case 's':
175 			chroot_dir = optarg;
176 			break;
177 		case 'u':
178 			chuser = optarg;
179 			break;
180 		case 'U':
181 			mask = strtol(optarg, NULL, 0);
182 			break;
183 		case 'w':
184 			create_new = 1;
185 			break;
186 		case 'W':
187 			create_new = 1;
188 			increase_name = 1;
189 			break;
190 		default:
191 			tftp_log(LOG_WARNING,
192 				"ignoring unknown option -%c", ch);
193 		}
194 	}
195 	if (optind < argc) {
196 		struct dirlist *dirp;
197 
198 		/* Get list of directory prefixes. Skip relative pathnames. */
199 		for (dirp = dirs; optind < argc && dirp < &dirs[MAXDIRS];
200 		     optind++) {
201 			if (argv[optind][0] == '/') {
202 				dirp->name = argv[optind];
203 				dirp->len  = strlen(dirp->name);
204 				dirp++;
205 			}
206 		}
207 	}
208 	else if (chroot_dir) {
209 		dirs->name = "/";
210 		dirs->len = 1;
211 	}
212 	if (ipchroot > 0 && chroot_dir == NULL) {
213 		tftp_log(LOG_ERR, "-c requires -s");
214 		exit(1);
215 	}
216 
217 	umask(mask);
218 
219 	{
220 		int on = 1;
221 		if (ioctl(0, FIONBIO, &on) < 0) {
222 			tftp_log(LOG_ERR, "ioctl(FIONBIO): %s", strerror(errno));
223 			exit(1);
224 		}
225 	}
226 
227 	/* Find out who we are talking to and what we are going to do */
228 	peerlen = sizeof(peer_sock);
229 	n = recvfrom(0, recvbuffer, MAXPKTSIZE, 0,
230 	    (struct sockaddr *)&peer_sock, &peerlen);
231 	if (n < 0) {
232 		tftp_log(LOG_ERR, "recvfrom: %s", strerror(errno));
233 		exit(1);
234 	}
235 	getnameinfo((struct sockaddr *)&peer_sock, peer_sock.ss_len,
236 	    peername, sizeof(peername), NULL, 0, NI_NUMERICHOST);
237 
238 	/*
239 	 * Now that we have read the message out of the UDP
240 	 * socket, we fork and exit.  Thus, inetd will go back
241 	 * to listening to the tftp port, and the next request
242 	 * to come in will start up a new instance of tftpd.
243 	 *
244 	 * We do this so that inetd can run tftpd in "wait" mode.
245 	 * The problem with tftpd running in "nowait" mode is that
246 	 * inetd may get one or more successful "selects" on the
247 	 * tftp port before we do our receive, so more than one
248 	 * instance of tftpd may be started up.  Worse, if tftpd
249 	 * break before doing the above "recvfrom", inetd would
250 	 * spawn endless instances, clogging the system.
251 	 */
252 	{
253 		int i, pid;
254 
255 		for (i = 1; i < 20; i++) {
256 		    pid = fork();
257 		    if (pid < 0) {
258 				sleep(i);
259 				/*
260 				 * flush out to most recently sent request.
261 				 *
262 				 * This may drop some request, but those
263 				 * will be resent by the clients when
264 				 * they timeout.  The positive effect of
265 				 * this flush is to (try to) prevent more
266 				 * than one tftpd being started up to service
267 				 * a single request from a single client.
268 				 */
269 				peerlen = sizeof peer_sock;
270 				i = recvfrom(0, recvbuffer, MAXPKTSIZE, 0,
271 				    (struct sockaddr *)&peer_sock, &peerlen);
272 				if (i > 0) {
273 					n = i;
274 				}
275 		    } else {
276 				break;
277 		    }
278 		}
279 		if (pid < 0) {
280 			tftp_log(LOG_ERR, "fork: %s", strerror(errno));
281 			exit(1);
282 		} else if (pid != 0) {
283 			exit(0);
284 		}
285 	}
286 
287 #ifdef	LIBWRAP
288 	/*
289 	 * See if the client is allowed to talk to me.
290 	 * (This needs to be done before the chroot())
291 	 */
292 	{
293 		struct request_info req;
294 
295 		request_init(&req, RQ_CLIENT_ADDR, peername, 0);
296 		request_set(&req, RQ_DAEMON, "tftpd", 0);
297 
298 		if (hosts_access(&req) == 0) {
299 			if (debug&DEBUG_ACCESS)
300 				tftp_log(LOG_WARNING,
301 				    "Access denied by 'tftpd' entry "
302 				    "in /etc/hosts.allow");
303 
304 			/*
305 			 * Full access might be disabled, but maybe the
306 			 * client is allowed to do read-only access.
307 			 */
308 			request_set(&req, RQ_DAEMON, "tftpd-ro", 0);
309 			allow_ro = hosts_access(&req);
310 
311 			request_set(&req, RQ_DAEMON, "tftpd-wo", 0);
312 			allow_wo = hosts_access(&req);
313 
314 			if (allow_ro == 0 && allow_wo == 0) {
315 				tftp_log(LOG_WARNING,
316 				    "Unauthorized access from %s", peername);
317 				exit(1);
318 			}
319 
320 			if (debug&DEBUG_ACCESS) {
321 				if (allow_ro)
322 					tftp_log(LOG_WARNING,
323 					    "But allowed readonly access "
324 					    "via 'tftpd-ro' entry");
325 				if (allow_wo)
326 					tftp_log(LOG_WARNING,
327 					    "But allowed writeonly access "
328 					    "via 'tftpd-wo' entry");
329 			}
330 		} else
331 			if (debug&DEBUG_ACCESS)
332 				tftp_log(LOG_WARNING,
333 				    "Full access allowed"
334 				    "in /etc/hosts.allow");
335 	}
336 #endif
337 
338 	/*
339 	 * Since we exit here, we should do that only after the above
340 	 * recvfrom to keep inetd from constantly forking should there
341 	 * be a problem.  See the above comment about system clogging.
342 	 */
343 	if (chroot_dir) {
344 		if (ipchroot > 0) {
345 			char *tempchroot;
346 			struct stat sb;
347 			int statret;
348 			struct sockaddr_storage ss;
349 			char hbuf[NI_MAXHOST];
350 
351 			statret = -1;
352 			memcpy(&ss, &peer_sock, peer_sock.ss_len);
353 			unmappedaddr((struct sockaddr_in6 *)&ss);
354 			getnameinfo((struct sockaddr *)&ss, ss.ss_len,
355 				    hbuf, sizeof(hbuf), NULL, 0,
356 				    NI_NUMERICHOST);
357 			asprintf(&tempchroot, "%s/%s", chroot_dir, hbuf);
358 			if (ipchroot == 2)
359 				statret = stat(tempchroot, &sb);
360 			if (ipchroot == 1 ||
361 			    (statret == 0 && (sb.st_mode & S_IFDIR)))
362 				chroot_dir = tempchroot;
363 		}
364 		/* Must get this before chroot because /etc might go away */
365 		if ((nobody = getpwnam(chuser)) == NULL) {
366 			tftp_log(LOG_ERR, "%s: no such user", chuser);
367 			exit(1);
368 		}
369 		if (chroot(chroot_dir)) {
370 			tftp_log(LOG_ERR, "chroot: %s: %s",
371 			    chroot_dir, strerror(errno));
372 			exit(1);
373 		}
374 		chdir("/");
375 		if (setgroups(1, &nobody->pw_gid) != 0) {
376 			tftp_log(LOG_ERR, "setgroups failed");
377 			exit(1);
378 		}
379 		if (setuid(nobody->pw_uid) != 0) {
380 			tftp_log(LOG_ERR, "setuid failed");
381 			exit(1);
382 		}
383 	}
384 
385 	len = sizeof(me_sock);
386 	if (getsockname(0, (struct sockaddr *)&me_sock, &len) == 0) {
387 		switch (me_sock.ss_family) {
388 		case AF_INET:
389 			((struct sockaddr_in *)&me_sock)->sin_port = 0;
390 			break;
391 		case AF_INET6:
392 			((struct sockaddr_in6 *)&me_sock)->sin6_port = 0;
393 			break;
394 		default:
395 			/* unsupported */
396 			break;
397 		}
398 	} else {
399 		memset(&me_sock, 0, sizeof(me_sock));
400 		me_sock.ss_family = peer_sock.ss_family;
401 		me_sock.ss_len = peer_sock.ss_len;
402 	}
403 	close(0);
404 	close(1);
405 	peer = socket(peer_sock.ss_family, SOCK_DGRAM, 0);
406 	if (peer < 0) {
407 		tftp_log(LOG_ERR, "socket: %s", strerror(errno));
408 		exit(1);
409 	}
410 	if (bind(peer, (struct sockaddr *)&me_sock, me_sock.ss_len) < 0) {
411 		tftp_log(LOG_ERR, "bind: %s", strerror(errno));
412 		exit(1);
413 	}
414 
415 	tp = (struct tftphdr *)recvbuffer;
416 	tp->th_opcode = ntohs(tp->th_opcode);
417 	if (tp->th_opcode == RRQ) {
418 		if (allow_ro)
419 			tftp_rrq(peer, tp->th_stuff, n - 1);
420 		else {
421 			tftp_log(LOG_WARNING,
422 			    "%s read access denied", peername);
423 			exit(1);
424 		}
425 	} else if (tp->th_opcode == WRQ) {
426 		if (allow_wo)
427 			tftp_wrq(peer, tp->th_stuff, n - 1);
428 		else {
429 			tftp_log(LOG_WARNING,
430 			    "%s write access denied", peername);
431 			exit(1);
432 		}
433 	} else
434 		send_error(peer, EBADOP);
435 	exit(1);
436 }
437 
438 static void
reduce_path(char * fn)439 reduce_path(char *fn)
440 {
441 	char *slash, *ptr;
442 
443 	/* Reduce all "/+./" to "/" (just in case we've got "/./../" later */
444 	while ((slash = strstr(fn, "/./")) != NULL) {
445 		for (ptr = slash; ptr > fn && ptr[-1] == '/'; ptr--)
446 			;
447 		slash += 2;
448 		while (*slash)
449 			*++ptr = *++slash;
450 	}
451 
452 	/* Now reduce all "/something/+../" to "/" */
453 	while ((slash = strstr(fn, "/../")) != NULL) {
454 		if (slash == fn)
455 			break;
456 		for (ptr = slash; ptr > fn && ptr[-1] == '/'; ptr--)
457 			;
458 		for (ptr--; ptr >= fn; ptr--)
459 			if (*ptr == '/')
460 				break;
461 		if (ptr < fn)
462 			break;
463 		slash += 3;
464 		while (*slash)
465 			*++ptr = *++slash;
466 	}
467 }
468 
469 static char *
parse_header(int peer,char * recvbuffer,ssize_t size,char ** filename,char ** mode)470 parse_header(int peer, char *recvbuffer, ssize_t size,
471 	char **filename, char **mode)
472 {
473 	char	*cp;
474 	int	i;
475 	struct formats *pf;
476 
477 	*mode = NULL;
478 	cp = recvbuffer;
479 
480 	i = get_field(peer, recvbuffer, size);
481 	if (i >= PATH_MAX) {
482 		tftp_log(LOG_ERR, "Bad option - filename too long");
483 		send_error(peer, EBADOP);
484 		exit(1);
485 	}
486 	*filename = recvbuffer;
487 	tftp_log(LOG_INFO, "Filename: '%s'", *filename);
488 	cp += i;
489 
490 	i = get_field(peer, cp, size);
491 	*mode = cp;
492 	cp += i;
493 
494 	/* Find the file transfer mode */
495 	for (cp = *mode; *cp; cp++)
496 		if (isupper(*cp))
497 			*cp = tolower(*cp);
498 	for (pf = formats; pf->f_mode; pf++)
499 		if (strcmp(pf->f_mode, *mode) == 0)
500 			break;
501 	if (pf->f_mode == NULL) {
502 		tftp_log(LOG_ERR,
503 		    "Bad option - Unknown transfer mode (%s)", *mode);
504 		send_error(peer, EBADOP);
505 		exit(1);
506 	}
507 	tftp_log(LOG_INFO, "Mode: '%s'", *mode);
508 
509 	return (cp + 1);
510 }
511 
512 /*
513  * WRQ - receive a file from the client
514  */
515 void
tftp_wrq(int peer,char * recvbuffer,ssize_t size)516 tftp_wrq(int peer, char *recvbuffer, ssize_t size)
517 {
518 	char *cp;
519 	int has_options = 0, ecode;
520 	char *filename, *mode;
521 	char fnbuf[PATH_MAX];
522 
523 	cp = parse_header(peer, recvbuffer, size, &filename, &mode);
524 	size -= (cp - recvbuffer) + 1;
525 
526 	strlcpy(fnbuf, filename, sizeof(fnbuf));
527 	reduce_path(fnbuf);
528 	filename = fnbuf;
529 
530 	if (size > 0) {
531 		if (options_rfc_enabled)
532 			has_options = !parse_options(peer, cp, size);
533 		else
534 			tftp_log(LOG_INFO, "Options found but not enabled");
535 	}
536 
537 	ecode = validate_access(peer, &filename, WRQ);
538 	if (ecode == 0) {
539 		if (has_options)
540 			send_oack(peer);
541 		else
542 			send_ack(peer, 0);
543 	}
544 	if (logging) {
545 		tftp_log(LOG_INFO, "%s: write request for %s: %s", peername,
546 			    filename, errtomsg(ecode));
547 	}
548 
549 	if (ecode) {
550 		send_error(peer, ecode);
551 		exit(1);
552 	}
553 	tftp_recvfile(peer, mode);
554 	exit(0);
555 }
556 
557 /*
558  * RRQ - send a file to the client
559  */
560 void
tftp_rrq(int peer,char * recvbuffer,ssize_t size)561 tftp_rrq(int peer, char *recvbuffer, ssize_t size)
562 {
563 	char *cp;
564 	int has_options = 0, ecode;
565 	char *filename, *mode;
566 	char	fnbuf[PATH_MAX];
567 
568 	cp = parse_header(peer, recvbuffer, size, &filename, &mode);
569 	size -= (cp - recvbuffer) + 1;
570 
571 	strlcpy(fnbuf, filename, sizeof(fnbuf));
572 	reduce_path(fnbuf);
573 	filename = fnbuf;
574 
575 	if (size > 0) {
576 		if (options_rfc_enabled)
577 			has_options = !parse_options(peer, cp, size);
578 		else
579 			tftp_log(LOG_INFO, "Options found but not enabled");
580 	}
581 
582 	ecode = validate_access(peer, &filename, RRQ);
583 	if (ecode == 0) {
584 		if (has_options) {
585 			int n;
586 			char lrecvbuffer[MAXPKTSIZE];
587 			struct tftphdr *rp = (struct tftphdr *)lrecvbuffer;
588 
589 			send_oack(peer);
590 			n = receive_packet(peer, lrecvbuffer, MAXPKTSIZE,
591 				NULL, timeoutpacket);
592 			if (n < 0) {
593 				if (debug&DEBUG_SIMPLE)
594 					tftp_log(LOG_DEBUG, "Aborting: %s",
595 					    rp_strerror(n));
596 				return;
597 			}
598 			if (rp->th_opcode != ACK) {
599 				if (debug&DEBUG_SIMPLE)
600 					tftp_log(LOG_DEBUG,
601 					    "Expected ACK, got %s on OACK",
602 					    packettype(rp->th_opcode));
603 				return;
604 			}
605 		}
606 	}
607 
608 	if (logging)
609 		tftp_log(LOG_INFO, "%s: read request for %s: %s", peername,
610 			    filename, errtomsg(ecode));
611 
612 	if (ecode) {
613 		/*
614 		 * Avoid storms of naks to a RRQ broadcast for a relative
615 		 * bootfile pathname from a diskless Sun.
616 		 */
617 		if (suppress_naks && *filename != '/' && ecode == ENOTFOUND)
618 			exit(0);
619 		send_error(peer, ecode);
620 		exit(1);
621 	}
622 	tftp_xmitfile(peer, mode);
623 }
624 
625 /*
626  * Find the next value for YYYYMMDD.nn when the file to be written should
627  * be unique. Due to the limitations of nn, we will fail if nn reaches 100.
628  * Besides, that is four updates per hour on a file, which is kind of
629  * execessive anyway.
630  */
631 static int
find_next_name(char * filename,int * fd)632 find_next_name(char *filename, int *fd)
633 {
634 	int i;
635 	time_t tval;
636 	size_t len;
637 	struct tm lt;
638 	char yyyymmdd[MAXPATHLEN];
639 	char newname[MAXPATHLEN];
640 
641 	/* Create the YYYYMMDD part of the filename */
642 	time(&tval);
643 	lt = *localtime(&tval);
644 	len = strftime(yyyymmdd, sizeof(yyyymmdd), newfile_format, &lt);
645 	if (len == 0) {
646 		syslog(LOG_WARNING,
647 			"Filename suffix too long (%d characters maximum)",
648 			MAXPATHLEN);
649 		return (EACCESS);
650 	}
651 
652 	/* Make sure the new filename is not too long */
653 	if (strlen(filename) > MAXPATHLEN - len - 5) {
654 		syslog(LOG_WARNING,
655 			"Filename too long (%zd characters, %zd maximum)",
656 			strlen(filename), MAXPATHLEN - len - 5);
657 		return (EACCESS);
658 	}
659 
660 	/* Find the first file which doesn't exist */
661 	for (i = 0; i < 100; i++) {
662 		sprintf(newname, "%s.%s.%02d", filename, yyyymmdd, i);
663 		*fd = open(newname,
664 		    O_WRONLY | O_CREAT | O_EXCL,
665 		    S_IRUSR | S_IWUSR | S_IRGRP |
666 		    S_IWGRP | S_IROTH | S_IWOTH);
667 		if (*fd > 0)
668 			return 0;
669 	}
670 
671 	return (EEXIST);
672 }
673 
674 /*
675  * Validate file access.  Since we
676  * have no uid or gid, for now require
677  * file to exist and be publicly
678  * readable/writable.
679  * If we were invoked with arguments
680  * from inetd then the file must also be
681  * in one of the given directory prefixes.
682  * Note also, full path name must be
683  * given as we have no login directory.
684  */
685 int
validate_access(int peer,char ** filep,int mode)686 validate_access(int peer, char **filep, int mode)
687 {
688 	struct stat stbuf;
689 	int	fd;
690 	int	error;
691 	struct dirlist *dirp;
692 	static char pathname[MAXPATHLEN];
693 	char *filename = *filep;
694 
695 	/*
696 	 * Prevent tricksters from getting around the directory restrictions
697 	 */
698 	if (strstr(filename, "/../"))
699 		return (EACCESS);
700 
701 	if (*filename == '/') {
702 		/*
703 		 * Allow the request if it's in one of the approved locations.
704 		 * Special case: check the null prefix ("/") by looking
705 		 * for length = 1 and relying on the arg. processing that
706 		 * it's a /.
707 		 */
708 		for (dirp = dirs; dirp->name != NULL; dirp++) {
709 			if (dirp->len == 1 ||
710 			    (!strncmp(filename, dirp->name, dirp->len) &&
711 			     filename[dirp->len] == '/'))
712 				    break;
713 		}
714 		/* If directory list is empty, allow access to any file */
715 		if (dirp->name == NULL && dirp != dirs)
716 			return (EACCESS);
717 		if (stat(filename, &stbuf) < 0)
718 			return (errno == ENOENT ? ENOTFOUND : EACCESS);
719 		if ((stbuf.st_mode & S_IFMT) != S_IFREG)
720 			return (ENOTFOUND);
721 		if (mode == RRQ) {
722 			if ((stbuf.st_mode & S_IROTH) == 0)
723 				return (EACCESS);
724 		} else {
725 			if ((stbuf.st_mode & S_IWOTH) == 0)
726 				return (EACCESS);
727 		}
728 	} else {
729 		int err;
730 
731 		/*
732 		 * Relative file name: search the approved locations for it.
733 		 * Don't allow write requests that avoid directory
734 		 * restrictions.
735 		 */
736 
737 		if (!strncmp(filename, "../", 3))
738 			return (EACCESS);
739 
740 		/*
741 		 * If the file exists in one of the directories and isn't
742 		 * readable, continue looking. However, change the error code
743 		 * to give an indication that the file exists.
744 		 */
745 		err = ENOTFOUND;
746 		for (dirp = dirs; dirp->name != NULL; dirp++) {
747 			snprintf(pathname, sizeof(pathname), "%s/%s",
748 				dirp->name, filename);
749 			if (stat(pathname, &stbuf) == 0 &&
750 			    (stbuf.st_mode & S_IFMT) == S_IFREG) {
751 				if (mode == RRQ) {
752 					if ((stbuf.st_mode & S_IROTH) != 0)
753 						break;
754 				} else {
755 					if ((stbuf.st_mode & S_IWOTH) != 0)
756 						break;
757 				}
758 				err = EACCESS;
759 			}
760 		}
761 		if (dirp->name != NULL)
762 			*filep = filename = pathname;
763 		else if (mode == RRQ)
764 			return (err);
765 		else if (err != ENOTFOUND || !create_new)
766 			return (err);
767 	}
768 
769 	/*
770 	 * This option is handled here because it (might) require(s) the
771 	 * size of the file.
772 	 */
773 	option_tsize(peer, NULL, mode, &stbuf);
774 
775 	if (mode == RRQ)
776 		fd = open(filename, O_RDONLY);
777 	else {
778 		if (create_new) {
779 			if (increase_name) {
780 				error = find_next_name(filename, &fd);
781 				if (error > 0)
782 					return (error + 100);
783 			} else
784 				fd = open(filename,
785 				    O_WRONLY | O_TRUNC | O_CREAT,
786 				    S_IRUSR | S_IWUSR | S_IRGRP |
787 				    S_IWGRP | S_IROTH | S_IWOTH );
788 		} else
789 			fd = open(filename, O_WRONLY | O_TRUNC);
790 	}
791 	if (fd < 0)
792 		return (errno + 100);
793 	file = fdopen(fd, (mode == RRQ)? "r":"w");
794 	if (file == NULL) {
795 		close(fd);
796 		return (errno + 100);
797 	}
798 	return (0);
799 }
800 
801 static void
tftp_xmitfile(int peer,const char * mode)802 tftp_xmitfile(int peer, const char *mode)
803 {
804 	uint16_t block;
805 	time_t now;
806 	struct tftp_stats ts;
807 
808 	memset(&ts, 0, sizeof(ts));
809 	now = time(NULL);
810 	if (debug&DEBUG_SIMPLE)
811 		tftp_log(LOG_DEBUG, "Transmitting file");
812 
813 	read_init(0, file, mode);
814 	block = 1;
815 	tftp_send(peer, &block, &ts);
816 	read_close();
817 	if (debug&DEBUG_SIMPLE)
818 		tftp_log(LOG_INFO, "Sent %jd bytes in %jd seconds",
819 		    (intmax_t)ts.amount, (intmax_t)time(NULL) - now);
820 }
821 
822 static void
tftp_recvfile(int peer,const char * mode)823 tftp_recvfile(int peer, const char *mode)
824 {
825 	uint16_t block;
826 	struct timeval now1, now2;
827 	struct tftp_stats ts;
828 
829 	gettimeofday(&now1, NULL);
830 	if (debug&DEBUG_SIMPLE)
831 		tftp_log(LOG_DEBUG, "Receiving file");
832 
833 	write_init(0, file, mode);
834 
835 	block = 0;
836 	tftp_receive(peer, &block, &ts, NULL, 0);
837 
838 	gettimeofday(&now2, NULL);
839 
840 	if (debug&DEBUG_SIMPLE) {
841 		double f;
842 		if (now1.tv_usec > now2.tv_usec) {
843 			now2.tv_usec += 1000000;
844 			now2.tv_sec--;
845 		}
846 
847 		f = now2.tv_sec - now1.tv_sec +
848 		    (now2.tv_usec - now1.tv_usec) / 100000.0;
849 		tftp_log(LOG_INFO,
850 		    "Download of %jd bytes in %d blocks completed after %0.1f seconds\n",
851 		    (intmax_t)ts.amount, block, f);
852 	}
853 
854 	return;
855 }
856