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 */
63 int
xmitfile(int peer,char * port,int fd,char * name,char * mode)64 xmitfile(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 */
165 int
recvfile(int peer,char * port,int fd,char * name,char * mode)166 recvfile(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