1/*-
2 * SPDX-License-Identifier: BSD-3-Clause
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/* Many bug fixes are from Jim Guyton <guyton@rand-unix> */
34
35/*
36 * TFTP User Program -- Protocol Machines
37 */
38#include <sys/socket.h>
39#include <sys/stat.h>
40
41#include <netinet/in.h>
42
43#include <arpa/tftp.h>
44
45#include <assert.h>
46#include <err.h>
47#include <netdb.h>
48#include <stdio.h>
49#include <stdlib.h>
50#include <string.h>
51#include <syslog.h>
52
53#include "tftp.h"
54#include "tftp-file.h"
55#include "tftp-utils.h"
56#include "tftp-io.h"
57#include "tftp-transfer.h"
58#include "tftp-options.h"
59
60/*
61 * Send the requested file.
62 */
63int
64xmitfile(int peer, char *port, int fd, char *name, char *mode)
65{
66	struct tftphdr *rp;
67	int n, i, ret = 0;
68	uint16_t block;
69	struct sockaddr_storage serv;	/* valid server port number */
70	char recvbuffer[MAXPKTSIZE];
71	struct tftp_stats tftp_stats;
72
73	stats_init(&tftp_stats);
74
75	memset(&serv, 0, sizeof(serv));
76	rp = (struct tftphdr *)recvbuffer;
77
78	if (port == NULL) {
79		struct servent *se;
80		se = getservbyname("tftp", "udp");
81		assert(se != NULL);
82		((struct sockaddr_in *)&peer_sock)->sin_port = se->s_port;
83	} else
84		((struct sockaddr_in *)&peer_sock)->sin_port =
85		    htons(atoi(port));
86
87	for (i = 0; i < 12; i++) {
88		struct sockaddr_storage from;
89
90		/* Tell the other side what we want to do */
91		if (debug & DEBUG_SIMPLE)
92			printf("Sending %s\n", name);
93
94		n = send_wrq(peer, name, mode);
95		if (n > 0) {
96			printf("Cannot send WRQ packet\n");
97			return -1;
98		}
99
100		/*
101		 * The first packet we receive has the new destination port
102		 * we have to send the next packets to.
103		 */
104		n = receive_packet(peer, recvbuffer,
105		    MAXPKTSIZE, &from, timeoutpacket);
106
107		/* We got some data! */
108		if (n >= 0) {
109			((struct sockaddr_in *)&peer_sock)->sin_port =
110			    ((struct sockaddr_in *)&from)->sin_port;
111			break;
112		}
113
114		/* This should be retried */
115		if (n == RP_TIMEOUT) {
116			printf("Try %d, didn't receive answer from remote.\n",
117			    i + 1);
118			continue;
119		}
120
121		/* Everything else is fatal */
122		break;
123	}
124	if (i == 12) {
125		printf("Transfer timed out.\n");
126		return -1;
127	}
128	if (rp->th_opcode == ERROR) {
129		printf("Got ERROR, aborted\n");
130		return -1;
131	}
132
133	/*
134	 * If the first packet is an OACK instead of an ACK packet,
135	 * handle it different.
136	 */
137	if (rp->th_opcode == OACK) {
138		if (!options_rfc_enabled) {
139			printf("Got OACK while options are not enabled!\n");
140			send_error(peer, EBADOP);
141			return -1;
142		}
143
144		parse_options(peer, rp->th_stuff, n + 2);
145	}
146
147	if (read_init(fd, NULL, mode) < 0) {
148		warn("read_init()");
149		return -1;
150	}
151
152	block = 1;
153	if (tftp_send(peer, &block, &tftp_stats) != 0)
154		ret = -1;
155
156	read_close();
157	if (tftp_stats.amount > 0)
158		printstats("Sent", verbose, &tftp_stats);
159	return ret;
160}
161
162/*
163 * Receive a file.
164 */
165int
166recvfile(int peer, char *port, int fd, char *name, char *mode)
167{
168	struct tftphdr *rp;
169	uint16_t block;
170	char recvbuffer[MAXPKTSIZE];
171	int n, i, ret = 0;
172	struct tftp_stats tftp_stats;
173
174	stats_init(&tftp_stats);
175
176	rp = (struct tftphdr *)recvbuffer;
177
178	if (port == NULL) {
179		struct servent *se;
180		se = getservbyname("tftp", "udp");
181		assert(se != NULL);
182		((struct sockaddr_in *)&peer_sock)->sin_port = se->s_port;
183	} else
184		((struct sockaddr_in *)&peer_sock)->sin_port =
185		    htons(atoi(port));
186
187	for (i = 0; i < 12; i++) {
188		struct sockaddr_storage from;
189
190		/* Tell the other side what we want to do */
191		if (debug & DEBUG_SIMPLE)
192			printf("Requesting %s\n", name);
193
194		n = send_rrq(peer, name, mode);
195		if (n > 0) {
196			printf("Cannot send RRQ packet\n");
197			return -1;
198		}
199
200		/*
201		 * The first packet we receive has the new destination port
202		 * we have to send the next packets to.
203		 */
204		n = receive_packet(peer, recvbuffer,
205		    MAXPKTSIZE, &from, timeoutpacket);
206
207		/* We got something useful! */
208		if (n >= 0) {
209			((struct sockaddr_in *)&peer_sock)->sin_port =
210			    ((struct sockaddr_in *)&from)->sin_port;
211			break;
212		}
213
214		/* We should retry if this happens */
215		if (n == RP_TIMEOUT) {
216			printf("Try %d, didn't receive answer from remote.\n",
217			    i + 1);
218			continue;
219		}
220
221		/* Otherwise it is a fatal error */
222		break;
223	}
224	if (i == 12) {
225		printf("Transfer timed out.\n");
226		return -1;
227	}
228	if (rp->th_opcode == ERROR) {
229		tftp_log(LOG_ERR, "Error code %d: %s", rp->th_code, rp->th_msg);
230		return -1;
231	}
232
233	if (write_init(fd, NULL, mode) < 0) {
234		warn("write_init");
235		return -1;
236	}
237
238	/*
239	 * If the first packet is an OACK packet instead of an DATA packet,
240	 * handle it different.
241	 */
242	if (rp->th_opcode == OACK) {
243		if (!options_rfc_enabled) {
244			printf("Got OACK while options are not enabled!\n");
245			send_error(peer, EBADOP);
246			return -1;
247		}
248
249		parse_options(peer, rp->th_stuff, n + 2);
250
251		n = send_ack(peer, 0);
252		if (n > 0) {
253			printf("Cannot send ACK on OACK.\n");
254			return -1;
255		}
256		block = 0;
257		if (tftp_receive(peer, &block, &tftp_stats, NULL, 0) != 0)
258			ret = -1;
259	} else {
260		block = 1;
261		if (tftp_receive(peer, &block, &tftp_stats, rp, n) != 0)
262			ret = -1;
263	}
264
265	if (tftp_stats.amount > 0)
266		printstats("Received", verbose, &tftp_stats);
267	return ret;
268}
269