2 * SPDX-License-Identifier: BSD-3-Clause
4 * Copyright (c) 1983, 1993
5 * The Regents of the University of California. All rights reserved.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
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.
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
32 #include <sys/cdefs.h>
33 /* Many bug fixes are from Jim Guyton <guyton@rand-unix> */
36 * TFTP User Program -- Protocol Machines
38 #include <sys/socket.h>
41 #include <netinet/in.h>
43 #include <arpa/tftp.h>
54 #include "tftp-file.h"
55 #include "tftp-utils.h"
57 #include "tftp-transfer.h"
58 #include "tftp-options.h"
61 * Send the requested file.
64 xmitfile(int peer, char *port, int fd, char *name, char *mode)
69 struct sockaddr_storage serv; /* valid server port number */
70 char recvbuffer[MAXPKTSIZE];
71 struct tftp_stats tftp_stats;
73 stats_init(&tftp_stats);
75 memset(&serv, 0, sizeof(serv));
76 rp = (struct tftphdr *)recvbuffer;
80 se = getservbyname("tftp", "udp");
82 ((struct sockaddr_in *)&peer_sock)->sin_port = se->s_port;
84 ((struct sockaddr_in *)&peer_sock)->sin_port =
87 for (i = 0; i < 12; i++) {
88 struct sockaddr_storage from;
90 /* Tell the other side what we want to do */
91 if (debug & DEBUG_SIMPLE)
92 printf("Sending %s\n", name);
94 n = send_wrq(peer, name, mode);
96 printf("Cannot send WRQ packet\n");
101 * The first packet we receive has the new destination port
102 * we have to send the next packets to.
104 n = receive_packet(peer, recvbuffer,
105 MAXPKTSIZE, &from, timeoutpacket);
107 /* We got some data! */
109 ((struct sockaddr_in *)&peer_sock)->sin_port =
110 ((struct sockaddr_in *)&from)->sin_port;
114 /* This should be retried */
115 if (n == RP_TIMEOUT) {
116 printf("Try %d, didn't receive answer from remote.\n",
121 /* Everything else is fatal */
125 printf("Transfer timed out.\n");
128 if (rp->th_opcode == ERROR) {
129 printf("Got ERROR, aborted\n");
134 * If the first packet is an OACK instead of an ACK packet,
135 * handle it different.
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);
144 parse_options(peer, rp->th_stuff, n + 2);
147 if (read_init(fd, NULL, mode) < 0) {
153 if (tftp_send(peer, &block, &tftp_stats) != 0)
157 if (tftp_stats.amount > 0)
158 printstats("Sent", verbose, &tftp_stats);
166 recvfile(int peer, char *port, int fd, char *name, char *mode)
170 char recvbuffer[MAXPKTSIZE];
172 struct tftp_stats tftp_stats;
174 stats_init(&tftp_stats);
176 rp = (struct tftphdr *)recvbuffer;
180 se = getservbyname("tftp", "udp");
182 ((struct sockaddr_in *)&peer_sock)->sin_port = se->s_port;
184 ((struct sockaddr_in *)&peer_sock)->sin_port =
187 for (i = 0; i < 12; i++) {
188 struct sockaddr_storage from;
190 /* Tell the other side what we want to do */
191 if (debug & DEBUG_SIMPLE)
192 printf("Requesting %s\n", name);
194 n = send_rrq(peer, name, mode);
196 printf("Cannot send RRQ packet\n");
201 * The first packet we receive has the new destination port
202 * we have to send the next packets to.
204 n = receive_packet(peer, recvbuffer,
205 MAXPKTSIZE, &from, timeoutpacket);
207 /* We got something useful! */
209 ((struct sockaddr_in *)&peer_sock)->sin_port =
210 ((struct sockaddr_in *)&from)->sin_port;
214 /* We should retry if this happens */
215 if (n == RP_TIMEOUT) {
216 printf("Try %d, didn't receive answer from remote.\n",
221 /* Otherwise it is a fatal error */
225 printf("Transfer timed out.\n");
228 if (rp->th_opcode == ERROR) {
229 tftp_log(LOG_ERR, "Error code %d: %s", rp->th_code, rp->th_msg);
233 if (write_init(fd, NULL, mode) < 0) {
239 * If the first packet is an OACK packet instead of an DATA packet,
240 * handle it different.
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);
249 parse_options(peer, rp->th_stuff, n + 2);
251 n = send_ack(peer, 0);
253 printf("Cannot send ACK on OACK.\n");
257 if (tftp_receive(peer, &block, &tftp_stats, NULL, 0) != 0)
261 if (tftp_receive(peer, &block, &tftp_stats, rp, n) != 0)
265 if (tftp_stats.amount > 0)
266 printstats("Received", verbose, &tftp_stats);