2 * Copyright (c) 1983, 1993
3 * The Regents of the University of California. All rights reserved.
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 * must display the following acknowledgement:
15 * This product includes software developed by the University of
16 * California, Berkeley and its contributors.
17 * 4. Neither the name of the University nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36 static char sccsid[] = "@(#)tftp.c 8.1 (Berkeley) 6/6/93";
40 #include <sys/cdefs.h>
41 __FBSDID("$FreeBSD$");
43 /* Many bug fixes are from Jim Guyton <guyton@rand-unix> */
46 * TFTP User Program -- Protocol Machines
48 #include <sys/types.h>
49 #include <sys/socket.h>
52 #include <netinet/in.h>
54 #include <arpa/inet.h>
55 #include <arpa/tftp.h>
69 extern struct sockaddr_storage peeraddr; /* filled in by main */
70 extern int f; /* the opened socket */
74 extern int maxtimeout;
75 extern volatile int txrx_error;
77 #define PKTSIZE SEGSIZE+4
83 static void nak(int, struct sockaddr *);
84 static int makerequest(int, const char *, struct tftphdr *, const char *);
85 static void printstats(const char *, unsigned long);
86 static void startclock(void);
87 static void stopclock(void);
88 static void timer(int);
89 static void tpacket(const char *, struct tftphdr *, int);
90 static int cmpport(struct sockaddr *, struct sockaddr *);
93 * Send the requested file.
96 xmitfile(fd, name, mode)
101 struct tftphdr *ap; /* data and ack packets */
104 volatile unsigned short block;
105 volatile int size, convert;
106 volatile unsigned long amount;
107 struct sockaddr_storage from;
110 struct sockaddr_storage peer;
111 struct sockaddr_storage serv; /* valid server port number */
113 startclock(); /* start stat's clock */
114 dp = r_init(); /* reset fillbuf/read-ahead code */
115 ap = (struct tftphdr *)ackbuf;
116 file = fdopen(fd, "r");
117 convert = !strcmp(mode, "netascii");
120 memcpy(&peer, &peeraddr, peeraddr.ss_len);
121 memset(&serv, 0, sizeof(serv));
123 signal(SIGALRM, timer);
126 size = makerequest(WRQ, name, dp, mode) - 4;
128 /* size = read(fd, dp->th_data, SEGSIZE); */
129 size = readit(file, &dp, convert);
131 nak(errno + 100, (struct sockaddr *)&peer);
134 dp->th_opcode = htons((u_short)DATA);
135 dp->th_block = htons((u_short)block);
138 (void) setjmp(timeoutbuf);
141 tpacket("sent", dp, size + 4);
142 n = sendto(f, dp, size + 4, 0,
143 (struct sockaddr *)&peer, peer.ss_len);
148 read_ahead(file, convert);
152 fromlen = sizeof(from);
153 n = recvfrom(f, ackbuf, sizeof(ackbuf), 0,
154 (struct sockaddr *)&from, &fromlen);
163 else if (!cmpport((struct sockaddr *)&serv,
164 (struct sockaddr *)&from)) {
165 warn("server port mismatch");
170 tpacket("received", ap, n);
171 /* should verify packet came from server */
172 ap->th_opcode = ntohs(ap->th_opcode);
173 ap->th_block = ntohs(ap->th_block);
174 if (ap->th_opcode == ERROR) {
175 printf("Error code %d: %s\n", ap->th_code,
180 if (ap->th_opcode == ACK) {
183 if (ap->th_block == block) {
186 /* On an error, try to synchronize
191 printf("discarded %d packets\n",
194 if (ap->th_block == (block-1)) {
202 } while (size == SEGSIZE || block == 1);
207 printstats("Sent", amount);
215 recvfile(fd, name, mode)
223 volatile unsigned short block;
224 volatile int size, firsttrip;
225 volatile unsigned long amount;
226 struct sockaddr_storage from;
229 volatile int convert; /* true if converting crlf -> lf */
230 struct sockaddr_storage peer;
231 struct sockaddr_storage serv; /* valid server port number */
235 ap = (struct tftphdr *)ackbuf;
236 file = fdopen(fd, "w");
237 convert = !strcmp(mode, "netascii");
241 memcpy(&peer, &peeraddr, peeraddr.ss_len);
242 memset(&serv, 0, sizeof(serv));
244 signal(SIGALRM, timer);
247 size = makerequest(RRQ, name, ap, mode);
250 ap->th_opcode = htons((u_short)ACK);
251 ap->th_block = htons((u_short)(block));
256 (void) setjmp(timeoutbuf);
259 tpacket("sent", ap, size);
260 if (sendto(f, ackbuf, size, 0, (struct sockaddr *)&peer,
261 peer.ss_len) != size) {
266 write_behind(file, convert);
270 fromlen = sizeof(from);
271 n = recvfrom(f, dp, PKTSIZE, 0,
272 (struct sockaddr *)&from, &fromlen);
281 else if (!cmpport((struct sockaddr *)&serv,
282 (struct sockaddr *)&from)) {
283 warn("server port mismatch");
288 tpacket("received", dp, n);
289 /* should verify client address */
290 dp->th_opcode = ntohs(dp->th_opcode);
291 dp->th_block = ntohs(dp->th_block);
292 if (dp->th_opcode == ERROR) {
293 printf("Error code %d: %s\n", dp->th_code,
298 if (dp->th_opcode == DATA) {
301 if (dp->th_block == block) {
302 break; /* have next packet */
304 /* On an error, try to synchronize
309 printf("discarded %d packets\n", j);
311 if (dp->th_block == (block-1)) {
312 goto send_ack; /* resend ack */
316 /* size = write(fd, dp->th_data, n - 4); */
317 size = writeit(file, &dp, n - 4, convert);
319 nak(errno + 100, (struct sockaddr *)&peer);
323 } while (size == SEGSIZE);
324 abort: /* ok to ack, since user */
325 ap->th_opcode = htons((u_short)ACK); /* has seen err msg */
326 ap->th_block = htons((u_short)block);
327 (void) sendto(f, ackbuf, 4, 0, (struct sockaddr *)&peer,
329 write_behind(file, convert); /* flush last buffer */
333 printstats("Received", amount);
338 makerequest(request, name, tp, mode)
346 tp->th_opcode = htons((u_short)request);
354 return (cp - (char *)tp);
361 { EUNDEF, "Undefined error code" },
362 { ENOTFOUND, "File not found" },
363 { EACCESS, "Access violation" },
364 { ENOSPACE, "Disk full or allocation exceeded" },
365 { EBADOP, "Illegal TFTP operation" },
366 { EBADID, "Unknown transfer ID" },
367 { EEXISTS, "File already exists" },
368 { ENOUSER, "No such user" },
373 * Send a nak packet (error message).
374 * Error code passed in is one of the
375 * standard TFTP codes, or a UNIX errno
381 struct sockaddr *peer;
387 tp = (struct tftphdr *)ackbuf;
388 tp->th_opcode = htons((u_short)ERROR);
389 tp->th_code = htons((u_short)error);
390 for (pe = errmsgs; pe->e_code >= 0; pe++)
391 if (pe->e_code == error)
393 if (pe->e_code < 0) {
394 pe->e_msg = strerror(error - 100);
395 tp->th_code = EUNDEF;
397 strcpy(tp->th_msg, pe->e_msg);
398 length = strlen(pe->e_msg) + 4;
400 tpacket("sent", tp, length);
401 if (sendto(f, ackbuf, length, 0, peer, peer->sa_len) != length)
411 static const char *opcodes[] =
412 { "#0", "RRQ", "WRQ", "DATA", "ACK", "ERROR" };
414 u_short op = ntohs(tp->th_opcode);
416 if (op < RRQ || op > ERROR)
417 printf("%s opcode=%x ", s, op);
419 printf("%s %s ", s, opcodes[op]);
425 file = cp = tp->th_stuff;
426 cp = index(cp, '\0');
427 printf("<file=%s, mode=%s>\n", file, cp + 1);
431 printf("<block=%d, %d bytes>\n", ntohs(tp->th_block), n - 4);
435 printf("<block=%d>\n", ntohs(tp->th_block));
439 printf("<code=%d, msg=%s>\n", ntohs(tp->th_code), tp->th_msg);
444 struct timeval tstart;
445 struct timeval tstop;
451 (void)gettimeofday(&tstart, NULL);
458 (void)gettimeofday(&tstop, NULL);
462 printstats(direction, amount)
463 const char *direction;
464 unsigned long amount;
467 /* compute delta in 1/10's second units */
468 delta = ((tstop.tv_sec*10.)+(tstop.tv_usec/100000)) -
469 ((tstart.tv_sec*10.)+(tstart.tv_usec/100000));
470 delta = delta/10.; /* back to seconds */
471 printf("%s %ld bytes in %.1f seconds", direction, amount, delta);
473 printf(" [%.0f bits/sec]", (amount*8.)/delta);
483 if (timeout >= maxtimeout) {
484 printf("Transfer timed out.\n");
485 longjmp(toplevel, -1);
488 longjmp(timeoutbuf, 1);
496 char a[NI_MAXSERV], b[NI_MAXSERV];
498 if (getnameinfo(sa, sa->sa_len, NULL, 0, a, sizeof(a), NI_NUMERICSERV))
500 if (getnameinfo(sb, sb->sa_len, NULL, 0, b, sizeof(b), NI_NUMERICSERV))
502 if (strcmp(a, b) != 0)