diff options
Diffstat (limited to 'contrib/tftp/tftp.c')
| -rw-r--r-- | contrib/tftp/tftp.c | 536 |
1 files changed, 536 insertions, 0 deletions
diff --git a/contrib/tftp/tftp.c b/contrib/tftp/tftp.c new file mode 100644 index 00000000..894e535e --- /dev/null +++ b/contrib/tftp/tftp.c @@ -0,0 +1,536 @@ +/* + * Copyright (c) 1983 Regents of the University of California. + * All rights reserved. + * + * Redistribution and use in source and binary forms are permitted + * provided that the above copyright notice and this paragraph are + * duplicated in all such forms and that any documentation, + * advertising materials, and other materials related to such + * distribution and use acknowledge that the software was developed + * by the University of California, Berkeley. The name of the + * University may not be used to endorse or promote products derived + * from this software without specific prior written permission. + * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR + * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED + * WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE. + */ + +#ifndef lint +static char sccsid[] = "@(#)tftp.c 5.7 (Berkeley) 6/29/88"; +#endif /* not lint */ + +/* Many bug fixes are from Jim Guyton <guyton@rand-unix> */ + +/* + * TFTP User Program -- Protocol Machines + */ +#include <sys/types.h> +#include <sys/socket.h> +#include <sys/time.h> + +#include <netinet/in.h> + +#include <arpa/tftp.h> + +#include <signal.h> +#include <stdio.h> +#include <errno.h> +#include <setjmp.h> + +extern int errno; + +extern struct sockaddr_in sin; /* filled in by main */ +extern int f; /* the opened socket */ +extern int trace; +extern int verbose; +extern int rexmtval; +extern int maxtimeout; +extern int segsize; + +#define PKTSIZE (1432+4) /* SEGSIZE+4 */ +char ackbuf[PKTSIZE]; +int timeout; +jmp_buf toplevel; +jmp_buf timeoutbuf; + +#ifndef OACK +#define OACK 6 +#endif + +void timer(int sig) +{ + + signal(SIGALRM, timer); + timeout += rexmtval; + if (timeout >= maxtimeout) { + printf("Transfer timed out.\n"); + longjmp(toplevel, -1); + } + longjmp(timeoutbuf, 1); +} + +strnlen(s, n) + char *s; + int n; +{ + int i = 0; + + while (n-- > 0 && *s++) i++; + return(i); +} + +/* + * Parse an OACK package and set blocksize accordingly + */ +parseoack(cp, sz) + char *cp; + int sz; +{ + int n; + + segsize = 512; + while (sz > 0 && *cp) { + n = strnlen(cp, sz); + if (n == 7 && !strncmp("blksize", cp, 7)) { + cp += 8; + sz -= 8; + if (sz <= 0) + break; + for (segsize = 0, n = strnlen(cp, sz); n > 0; + n--, cp++, sz--) { + if (*cp < '0' || *cp > '9') + break; + segsize = 10*segsize + *cp - '0'; } + } + cp += n + 1; + sz -= n + 1; + } + if (segsize < 8 || segsize > 1432) { + printf("Remote host negotiated illegal blocksize %d\n", + segsize); + segsize = 512; + longjmp(timeoutbuf, -1); + } +} + +/* + * Send the requested file. + */ +sendfile(fd, name, mode) + int fd; + char *name; + char *mode; +{ + register struct tftphdr *ap; /* data and ack packets */ + struct tftphdr *r_init(), *dp; + register int size, n; + u_short block = 0; + register unsigned long amount = 0; + struct sockaddr_in from; + int fromlen; + int convert; /* true if doing nl->crlf conversion */ + FILE *file; + + startclock(); /* start stat's clock */ + dp = r_init(); /* reset fillbuf/read-ahead code */ + ap = (struct tftphdr *)ackbuf; + file = fdopen(fd, "r"); + convert = !strcmp(mode, "netascii"); + + signal(SIGALRM, timer); + do { + if (block == 0) + size = makerequest(WRQ, name, dp, mode) - 4; + else { + /* size = read(fd, dp->th_data, SEGSIZE); */ + size = readit(file, &dp, convert); + if (size < 0) { + nak(errno + 100); + break; + } + dp->th_opcode = htons((u_short)DATA); + dp->th_block = htons(block); + } + timeout = 0; + (void) setjmp(timeoutbuf); +send_data: + if (trace) + tpacket("sent", dp, size + 4); + n = sendto(f, dp, size + 4, 0, (struct sockaddr *)&sin, + sizeof (sin)); + if (n != size + 4) { + perror("tftp: sendto"); + goto abort; + } + if (block) /* do not start reading until the blocksize + has been negotiated */ + read_ahead(file, convert); + for ( ; ; ) { + alarm(rexmtval); + do { + fromlen = sizeof (from); + n = recvfrom(f, ackbuf, sizeof (ackbuf), 0, + (struct sockaddr *)&from, + &fromlen); + } while (n <= 0); + alarm(0); + if (n < 0) { + perror("tftp: recvfrom"); + goto abort; + } + sin.sin_port = from.sin_port; /* added */ + if (trace) + tpacket("received", ap, n); + /* should verify packet came from server */ + ap->th_opcode = ntohs(ap->th_opcode); + if (ap->th_opcode == ERROR) { + printf("Error code %d: %s\n", ap->th_code, + ap->th_msg); + goto abort; + } + if (ap->th_opcode == ACK) { + int j; + + ap->th_block = ntohs(ap->th_block); + + if (block == 0) { + if (trace) + printf("server does not know " + "about RFC1782; reset" + "ting blocksize\n"); + segsize = 512; + } + if (ap->th_block == block) { + break; + } + /* On an error, try to synchronize + * both sides. + */ + j = synchnet(f); + if (j && trace) { + printf("discarded %d packets\n", + j); + } + if (ap->th_block == (block-1)) { + goto send_data; + } + } + else if (ap->th_opcode == OACK) { + if (block) { + printf("protocol violation\n"); + longjmp(toplevel, -1); + } + parseoack(&ap->th_stuff, n - 2); + break; + } + } + if (block > 0) + amount += size; + else + read_ahead(file, convert); + block++; + } while (size == segsize || block == 1); +abort: + fclose(file); + stopclock(); + if (amount > 0) + printstats("Sent", amount); +} + +/* + * Receive a file. + */ +recvfile(fd, name, mode) + int fd; + char *name; + char *mode; +{ + register struct tftphdr *ap; + struct tftphdr *dp, *w_init(); + register int n, size; + u_short block = 1; + unsigned long amount = 0; + struct sockaddr_in from; + int fromlen, firsttrip = 1; + FILE *file; + int convert; /* true if converting crlf -> lf */ + int waitforoack = 1; + + startclock(); + dp = w_init(); + ap = (struct tftphdr *)ackbuf; + file = fdopen(fd, "w"); + convert = !strcmp(mode, "netascii"); + + signal(SIGALRM, timer); + do { + if (firsttrip) { + size = makerequest(RRQ, name, ap, mode); + firsttrip = 0; + } else { + ap->th_opcode = htons((u_short)ACK); + ap->th_block = htons(block); + size = 4; + block++; + } + timeout = 0; + (void) setjmp(timeoutbuf); +send_ack: + if (trace) + tpacket("sent", ap, size); + if (sendto(f, ackbuf, size, 0, (struct sockaddr *)&sin, + sizeof (sin)) != size) { + alarm(0); + perror("tftp: sendto"); + goto abort; + } + if (!waitforoack) + write_behind(file, convert); + for ( ; ; ) { + alarm(rexmtval); + do { + fromlen = sizeof (from); + n = recvfrom(f, dp, PKTSIZE, 0, + (struct sockaddr *)&from, &fromlen); + } while (n <= 0); + alarm(0); + if (n < 0) { + perror("tftp: recvfrom"); + goto abort; + } + sin.sin_port = from.sin_port; /* added */ + if (trace) + tpacket("received", dp, n); + /* should verify client address */ + dp->th_opcode = ntohs(dp->th_opcode); + if (dp->th_opcode == ERROR) { + printf("Error code %d: %s\n", dp->th_code, + dp->th_msg); + goto abort; + } + if (dp->th_opcode == DATA) { + int j; + + if (waitforoack) { + if (trace) + printf("server does not know " + "about RFC1782; reset" + "ting blocksize\n"); + waitforoack = 0; + segsize = 512; + } + dp->th_block = ntohs(dp->th_block); + if (dp->th_block == block) { + break; /* have next packet */ + } + /* On an error, try to synchronize + * both sides. + */ + j = synchnet(f); + if (j && trace) { + printf("discarded %d packets\n", j); + } + if (dp->th_block == (block-1)) { + goto send_ack; /* resend ack */ + } + } + else if (dp->th_opcode == OACK) { + if (block != 1 || !waitforoack) { + printf("protocol violation\n"); + longjmp(toplevel, -1); + } + waitforoack = 0; + parseoack(&dp->th_stuff, n - 2); + ap->th_opcode = htons((u_short)ACK); + ap->th_block = htons(0); + size = 4; + goto send_ack; + } + } + /* size = write(fd, dp->th_data, n - 4); */ + size = writeit(file, &dp, n - 4, convert); + if (size < 0) { + nak(errno + 100); + break; + } + amount += size; + } while (size == segsize); +abort: /* ok to ack, since user */ + ap->th_opcode = htons((u_short)ACK); /* has seen err msg */ + ap->th_block = htons(block); + (void) sendto(f, ackbuf, 4, 0, (struct sockaddr *)&sin, sizeof (sin)); + write_behind(file, convert); /* flush last buffer */ + fclose(file); + stopclock(); + if (amount > 0) + printstats("Received", amount); +} + +makerequest(request, name, tp, mode) + int request; + char *name, *mode; + struct tftphdr *tp; +{ + register char *cp; + + tp->th_opcode = htons((u_short)request); + cp = tp->th_stuff; + strcpy(cp, name); + cp += strlen(name); + *cp++ = '\0'; + strcpy(cp, mode); + cp += strlen(mode); + *cp++ = '\0'; + strcpy(cp, "blksize"); + cp += 7; + *cp++ = '\0'; + sprintf(cp, "%d", segsize); + cp += strlen(cp) + 1; + return (cp - (char *)tp); +} + +struct errmsg { + int e_code; + const char *e_msg; +} errmsgs[] = { + { EUNDEF, "Undefined error code" }, + { ENOTFOUND, "File not found" }, + { EACCESS, "Access violation" }, + { ENOSPACE, "Disk full or allocation exceeded" }, + { EBADOP, "Illegal TFTP operation" }, + { EBADID, "Unknown transfer ID" }, + { EEXISTS, "File already exists" }, + { ENOUSER, "No such user" }, + { -1, 0 } +}; + +/* + * Send a nak packet (error message). + * Error code passed in is one of the + * standard TFTP codes, or a UNIX errno + * offset by 100. + */ +nak(error) + int error; +{ + register struct tftphdr *tp; + int length; + register struct errmsg *pe; +/* extern char *sys_errlist[]; */ + + tp = (struct tftphdr *)ackbuf; + tp->th_opcode = htons((u_short)ERROR); + tp->th_code = htons((u_short)error); + for (pe = errmsgs; pe->e_code >= 0; pe++) + if (pe->e_code == error) + break; + if (pe->e_code < 0) { + pe->e_msg = sys_errlist[error - 100]; + tp->th_code = EUNDEF; + } + strcpy(tp->th_msg, pe->e_msg); + length = strlen(pe->e_msg) + 4; + if (trace) + tpacket("sent", tp, length); + if (sendto(f, ackbuf, length, 0, (struct sockaddr *)&sin, sizeof (sin)) + != length) + perror("nak"); +} + +topts(cp, sz) + char *cp; + int sz; +{ + int n, i = 0; + + while (sz > 0 && *cp) { + n = strnlen(cp, sz); + if (n > 0) { + printf("%s%s=", i++ ? ", " : "", cp); + cp += n + 1; + sz -= n + 1; + if (sz <= 0) + break; + n = strnlen(cp, sz); + if (n > 0) + printf("%s", cp); + } + cp += n + 1; + sz -= n + 1; + } +} + +tpacket(s, tp, n) + char *s; + struct tftphdr *tp; + int n; +{ + static char *opcodes[] = + { "#0", "RRQ", "WRQ", "DATA", "ACK", "ERROR", "OACK" }; + register char *cp, *file; + u_short op = ntohs(tp->th_opcode); + char *index(); + + if (op < RRQ || op > OACK) + printf("%s opcode=%x ", s, op); + else + printf("%s %s ", s, opcodes[op]); + switch (op) { + + case RRQ: + case WRQ: + n -= 2; + file = cp = tp->th_stuff; + cp = index(cp, '\0'); + printf("<file=%s, mode=%s, opts: ", file, cp + 1); + topts(index(cp + 1, '\000') + 1, n - strlen(file) + - strlen(cp + 1) - 2); + printf(">\n"); + break; + + case DATA: + printf("<block=%d, %d bytes>\n", ntohs(tp->th_block), n - 4); + break; + + case ACK: + printf("<block=%d>\n", ntohs(tp->th_block)); + break; + + case ERROR: + printf("<code=%d, msg=%s>\n", ntohs(tp->th_code), tp->th_msg); + break; + case OACK: + printf("<"); + topts(tp->th_stuff, n - 2); + printf(">\n"); + break; + } +} + +struct timeval tstart; +struct timeval tstop; +struct timezone zone; + +startclock() { + gettimeofday(&tstart, &zone); +} + +stopclock() { + gettimeofday(&tstop, &zone); +} + +printstats(direction, amount) +char *direction; +unsigned long amount; +{ + double delta; + /* compute delta in 1/10's second units */ + delta = ((tstop.tv_sec*10.)+(tstop.tv_usec/100000)) - + ((tstart.tv_sec*10.)+(tstart.tv_usec/100000)); + delta = delta/10.; /* back to seconds */ + printf("%s %ld bytes in %.1f seconds", direction, amount, delta); + if ((verbose) && (delta >= 0.1)) + printf(" [%.0f bits/sec]", (amount*8.)/delta); + putchar('\n'); +} + |