xref: /freebsd/usr.bin/tftp/tftp.c (revision a1ec94b88d3db03ca744027b8281ffcbff7c22f0)
19b50d902SRodney W. Grimes /*
29b50d902SRodney W. Grimes  * Copyright (c) 1983, 1993
39b50d902SRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
49b50d902SRodney W. Grimes  *
59b50d902SRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
69b50d902SRodney W. Grimes  * modification, are permitted provided that the following conditions
79b50d902SRodney W. Grimes  * are met:
89b50d902SRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
99b50d902SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
109b50d902SRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
119b50d902SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
129b50d902SRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
139b50d902SRodney W. Grimes  * 3. All advertising materials mentioning features or use of this software
149b50d902SRodney W. Grimes  *    must display the following acknowledgement:
159b50d902SRodney W. Grimes  *	This product includes software developed by the University of
169b50d902SRodney W. Grimes  *	California, Berkeley and its contributors.
179b50d902SRodney W. Grimes  * 4. Neither the name of the University nor the names of its contributors
189b50d902SRodney W. Grimes  *    may be used to endorse or promote products derived from this software
199b50d902SRodney W. Grimes  *    without specific prior written permission.
209b50d902SRodney W. Grimes  *
219b50d902SRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
229b50d902SRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
239b50d902SRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
249b50d902SRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
259b50d902SRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
269b50d902SRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
279b50d902SRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
289b50d902SRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
299b50d902SRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
309b50d902SRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
319b50d902SRodney W. Grimes  * SUCH DAMAGE.
329b50d902SRodney W. Grimes  */
339b50d902SRodney W. Grimes 
346dfc2060SDavid Malone #if 0
359b50d902SRodney W. Grimes #ifndef lint
366dfc2060SDavid Malone static char sccsid[] = "@(#)tftp.c	8.1 (Berkeley) 6/6/93";
376dfc2060SDavid Malone #endif /* not lint */
38fd129a02SPhilippe Charnier #endif
399b50d902SRodney W. Grimes 
406dfc2060SDavid Malone #include <sys/cdefs.h>
416dfc2060SDavid Malone __FBSDID("$FreeBSD$");
426dfc2060SDavid Malone 
439b50d902SRodney W. Grimes /* Many bug fixes are from Jim Guyton <guyton@rand-unix> */
449b50d902SRodney W. Grimes 
459b50d902SRodney W. Grimes /*
469b50d902SRodney W. Grimes  * TFTP User Program -- Protocol Machines
479b50d902SRodney W. Grimes  */
489b50d902SRodney W. Grimes #include <sys/socket.h>
49752fa694SWarner Losh #include <sys/stat.h>
509b50d902SRodney W. Grimes 
519b50d902SRodney W. Grimes #include <netinet/in.h>
529b50d902SRodney W. Grimes 
539b50d902SRodney W. Grimes #include <arpa/tftp.h>
549b50d902SRodney W. Grimes 
55fd129a02SPhilippe Charnier #include <err.h>
564dac6235SHajimu UMEMOTO #include <netdb.h>
57752fa694SWarner Losh #include <stdio.h>
58752fa694SWarner Losh #include <stdlib.h>
59752fa694SWarner Losh #include <string.h>
60752fa694SWarner Losh #include <syslog.h>
619b50d902SRodney W. Grimes 
62752fa694SWarner Losh #include "tftp.h"
63752fa694SWarner Losh #include "tftp-file.h"
64752fa694SWarner Losh #include "tftp-utils.h"
65752fa694SWarner Losh #include "tftp-io.h"
66752fa694SWarner Losh #include "tftp-transfer.h"
67752fa694SWarner Losh #include "tftp-options.h"
689b50d902SRodney W. Grimes 
699b50d902SRodney W. Grimes /*
709b50d902SRodney W. Grimes  * Send the requested file.
719b50d902SRodney W. Grimes  */
729b50d902SRodney W. Grimes void
73752fa694SWarner Losh xmitfile(int peer, char *port, int fd, char *name, char *mode)
749b50d902SRodney W. Grimes {
75752fa694SWarner Losh 	struct tftphdr *rp;
76752fa694SWarner Losh 	int n, i;
77752fa694SWarner Losh 	uint16_t block;
78752fa694SWarner Losh 	uint32_t amount;
794dac6235SHajimu UMEMOTO 	struct sockaddr_storage serv;	/* valid server port number */
80752fa694SWarner Losh 	char recvbuffer[MAXPKTSIZE];
81752fa694SWarner Losh 	struct tftp_stats tftp_stats;
829b50d902SRodney W. Grimes 
83752fa694SWarner Losh 	stats_init(&tftp_stats);
84752fa694SWarner Losh 
854dac6235SHajimu UMEMOTO 	memset(&serv, 0, sizeof(serv));
86752fa694SWarner Losh 	rp = (struct tftphdr *)recvbuffer;
879b50d902SRodney W. Grimes 
88752fa694SWarner Losh 	if (port == NULL) {
89752fa694SWarner Losh 		struct servent *se;
90752fa694SWarner Losh 		se = getservbyname("tftp", "udp");
91752fa694SWarner Losh 		((struct sockaddr_in *)&peer_sock)->sin_port = se->s_port;
92752fa694SWarner Losh 	} else
93752fa694SWarner Losh 		((struct sockaddr_in *)&peer_sock)->sin_port =
94752fa694SWarner Losh 		    htons(atoi(port));
959b50d902SRodney W. Grimes 
96752fa694SWarner Losh 	for (i = 0; i < 12; i++) {
97752fa694SWarner Losh 		struct sockaddr_storage from;
98752fa694SWarner Losh 
99752fa694SWarner Losh 		/* Tell the other side what we want to do */
100752fa694SWarner Losh 		if (debug&DEBUG_SIMPLE)
101752fa694SWarner Losh 			printf("Sending %s\n", name);
102752fa694SWarner Losh 
103752fa694SWarner Losh 		n = send_wrq(peer, name, mode);
104752fa694SWarner Losh 		if (n > 0) {
105752fa694SWarner Losh 			printf("Cannot send WRQ packet\n");
106752fa694SWarner Losh 			return;
1079b50d902SRodney W. Grimes 		}
108752fa694SWarner Losh 
109752fa694SWarner Losh 		/*
110752fa694SWarner Losh 		 * The first packet we receive has the new destination port
111752fa694SWarner Losh 		 * we have to send the next packets to.
1129b50d902SRodney W. Grimes 		 */
113752fa694SWarner Losh 		n = receive_packet(peer, recvbuffer,
114752fa694SWarner Losh 		    MAXPKTSIZE, &from, timeoutpacket);
115752fa694SWarner Losh 
116752fa694SWarner Losh 		/* We got some data! */
117752fa694SWarner Losh 		if (n >= 0) {
118752fa694SWarner Losh 			((struct sockaddr_in *)&peer_sock)->sin_port =
119752fa694SWarner Losh 			    ((struct sockaddr_in *)&from)->sin_port;
120752fa694SWarner Losh 			break;
1219b50d902SRodney W. Grimes 		}
122752fa694SWarner Losh 
123752fa694SWarner Losh 		/* This should be retried */
124752fa694SWarner Losh 		if (n == RP_TIMEOUT) {
125752fa694SWarner Losh 			printf("Try %d, didn't receive answer from remote.\n",
126752fa694SWarner Losh 			    i + 1);
127752fa694SWarner Losh 			continue;
1289b50d902SRodney W. Grimes 		}
129752fa694SWarner Losh 
130752fa694SWarner Losh 		/* Everything else is fatal */
131752fa694SWarner Losh 		break;
1329b50d902SRodney W. Grimes 	}
133752fa694SWarner Losh 	if (i == 12) {
134752fa694SWarner Losh 		printf("Transfer timed out.\n");
135752fa694SWarner Losh 		return;
1369b50d902SRodney W. Grimes 	}
137752fa694SWarner Losh 	if (rp->th_opcode == ERROR) {
138752fa694SWarner Losh 		printf("Got ERROR, aborted\n");
139752fa694SWarner Losh 		return;
140752fa694SWarner Losh 	}
141752fa694SWarner Losh 
142752fa694SWarner Losh 	/*
143752fa694SWarner Losh 	 * If the first packet is an OACK instead of an ACK packet,
144752fa694SWarner Losh 	 * handle it different.
145752fa694SWarner Losh 	 */
146752fa694SWarner Losh 	if (rp->th_opcode == OACK) {
147752fa694SWarner Losh 		if (!options_rfc_enabled) {
148752fa694SWarner Losh 			printf("Got OACK while options are not enabled!\n");
149752fa694SWarner Losh 			send_error(peer, EBADOP);
150752fa694SWarner Losh 			return;
151752fa694SWarner Losh 		}
152752fa694SWarner Losh 
153752fa694SWarner Losh 		parse_options(peer, rp->th_stuff, n + 2);
154752fa694SWarner Losh 	}
155752fa694SWarner Losh 
156752fa694SWarner Losh 	if (read_init(fd, NULL, mode) < 0) {
157752fa694SWarner Losh 		warn("read_init()");
158752fa694SWarner Losh 		return;
159752fa694SWarner Losh 	}
160752fa694SWarner Losh 
161752fa694SWarner Losh 	block = 1;
162752fa694SWarner Losh 	tftp_send(peer, &block, &tftp_stats);
163752fa694SWarner Losh 
164752fa694SWarner Losh 	read_close();
1659b50d902SRodney W. Grimes 	if (amount > 0)
166752fa694SWarner Losh 		printstats("Sent", verbose, &tftp_stats);
167752fa694SWarner Losh 
168752fa694SWarner Losh 	txrx_error = 1;
1699b50d902SRodney W. Grimes }
1709b50d902SRodney W. Grimes 
1719b50d902SRodney W. Grimes /*
1729b50d902SRodney W. Grimes  * Receive a file.
1739b50d902SRodney W. Grimes  */
1749b50d902SRodney W. Grimes void
175752fa694SWarner Losh recvfile(int peer, char *port, int fd, char *name, char *mode)
1769b50d902SRodney W. Grimes {
177752fa694SWarner Losh 	struct tftphdr *rp;
178752fa694SWarner Losh 	uint16_t block;
179752fa694SWarner Losh 	char recvbuffer[MAXPKTSIZE];
180752fa694SWarner Losh 	int n, i;
181752fa694SWarner Losh 	struct tftp_stats tftp_stats;
182752fa694SWarner Losh 
183752fa694SWarner Losh 	stats_init(&tftp_stats);
184752fa694SWarner Losh 
185752fa694SWarner Losh 	rp = (struct tftphdr *)recvbuffer;
186752fa694SWarner Losh 
187752fa694SWarner Losh 	if (port == NULL) {
188752fa694SWarner Losh 		struct servent *se;
189752fa694SWarner Losh 		se = getservbyname("tftp", "udp");
190752fa694SWarner Losh 		((struct sockaddr_in *)&peer_sock)->sin_port = se->s_port;
191752fa694SWarner Losh 	} else
192752fa694SWarner Losh 		((struct sockaddr_in *)&peer_sock)->sin_port =
193752fa694SWarner Losh 		    htons(atoi(port));
194752fa694SWarner Losh 
195752fa694SWarner Losh 	for (i = 0; i < 12; i++) {
1964dac6235SHajimu UMEMOTO 		struct sockaddr_storage from;
1979b50d902SRodney W. Grimes 
198752fa694SWarner Losh 		/* Tell the other side what we want to do */
199752fa694SWarner Losh 		if (debug&DEBUG_SIMPLE)
200752fa694SWarner Losh 			printf("Requesting %s\n", name);
2019b50d902SRodney W. Grimes 
202752fa694SWarner Losh 		n = send_rrq(peer, name, mode);
203752fa694SWarner Losh 		if (n > 0) {
204752fa694SWarner Losh 			printf("Cannot send RRQ packet\n");
205752fa694SWarner Losh 			return;
2069b50d902SRodney W. Grimes 		}
2079b50d902SRodney W. Grimes 
2089b50d902SRodney W. Grimes 		/*
209752fa694SWarner Losh 		 * The first packet we receive has the new destination port
210752fa694SWarner Losh 		 * we have to send the next packets to.
2119b50d902SRodney W. Grimes 		 */
212752fa694SWarner Losh 		n = receive_packet(peer, recvbuffer,
213752fa694SWarner Losh 		    MAXPKTSIZE, &from, timeoutpacket);
2149b50d902SRodney W. Grimes 
215752fa694SWarner Losh 		/* We got something useful! */
216752fa694SWarner Losh 		if (n >= 0) {
217752fa694SWarner Losh 			((struct sockaddr_in *)&peer_sock)->sin_port =
218752fa694SWarner Losh 			    ((struct sockaddr_in *)&from)->sin_port;
2199b50d902SRodney W. Grimes 			break;
2209b50d902SRodney W. Grimes 		}
221752fa694SWarner Losh 
222752fa694SWarner Losh 		/* We should retry if this happens */
223752fa694SWarner Losh 		if (n == RP_TIMEOUT) {
224752fa694SWarner Losh 			printf("Try %d, didn't receive answer from remote.\n",
225752fa694SWarner Losh 			    i + 1);
226752fa694SWarner Losh 			continue;
2279b50d902SRodney W. Grimes 		}
2289b50d902SRodney W. Grimes 
229752fa694SWarner Losh 		/* Otherwise it is a fatal error */
230752fa694SWarner Losh 		break;
2319b50d902SRodney W. Grimes 	}
232*a1ec94b8SGavin Atkinson 	if (i == 12) {
233*a1ec94b8SGavin Atkinson 		printf("Transfer timed out.\n");
234*a1ec94b8SGavin Atkinson 		return;
235*a1ec94b8SGavin Atkinson 	}
236752fa694SWarner Losh 	if (rp->th_opcode == ERROR) {
237752fa694SWarner Losh 		tftp_log(LOG_ERR, "Error code %d: %s", rp->th_code, rp->th_msg);
238752fa694SWarner Losh 		return;
2399b50d902SRodney W. Grimes 	}
2409b50d902SRodney W. Grimes 
241752fa694SWarner Losh 	if (write_init(fd, NULL, mode) < 0) {
242752fa694SWarner Losh 		warn("write_init");
243752fa694SWarner Losh 		return;
2449b50d902SRodney W. Grimes 	}
2459b50d902SRodney W. Grimes 
246752fa694SWarner Losh 	stats_init(&tftp_stats);
2479b50d902SRodney W. Grimes 
248752fa694SWarner Losh 	/*
249752fa694SWarner Losh 	 * If the first packet is an OACK packet instead of an DATA packet,
250752fa694SWarner Losh 	 * handle it different.
251752fa694SWarner Losh 	 */
252752fa694SWarner Losh 	if (rp->th_opcode == OACK) {
253752fa694SWarner Losh 		if (!options_rfc_enabled) {
254752fa694SWarner Losh 			printf("Got OACK while options are not enabled!\n");
255752fa694SWarner Losh 			send_error(peer, EBADOP);
256752fa694SWarner Losh 			return;
2579b50d902SRodney W. Grimes 		}
2584dac6235SHajimu UMEMOTO 
259752fa694SWarner Losh 		parse_options(peer, rp->th_stuff, n + 2);
2604dac6235SHajimu UMEMOTO 
261752fa694SWarner Losh 		n = send_ack(peer, 0);
262752fa694SWarner Losh 		if (n > 0) {
263752fa694SWarner Losh 			printf("Cannot send ACK on OACK.\n");
264752fa694SWarner Losh 			return;
265752fa694SWarner Losh 		}
266752fa694SWarner Losh 		block = 0;
267752fa694SWarner Losh 		tftp_receive(peer, &block, &tftp_stats, NULL, 0);
268752fa694SWarner Losh 	} else {
269752fa694SWarner Losh 		block = 1;
270752fa694SWarner Losh 		tftp_receive(peer, &block, &tftp_stats, rp, n);
271752fa694SWarner Losh 	}
2724dac6235SHajimu UMEMOTO 
273752fa694SWarner Losh 	write_close();
274752fa694SWarner Losh 	if (tftp_stats.amount > 0)
275752fa694SWarner Losh 		printstats("Received", verbose, &tftp_stats);
276752fa694SWarner Losh 	return;
2774dac6235SHajimu UMEMOTO }
278