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