From b25f98a417233f8c470ad61a6e191ff3aa8bd633 Mon Sep 17 00:00:00 2001 From: Bernhard Reutner-Fischer Date: Sat, 10 Jun 2006 14:15:03 +0000 Subject: - fix two segfaults (reported by Horst Kronstorfer) - remove dangling file if get fails (spotted and fixed by Jason Schoon) - shrink it (Bernhard Fischer) Thanks, all! text data bss dec hex filename 2684 0 0 2684 a7c networking/tftp.o.orig 2748 0 0 2748 abc networking/tftp.o.allfixed 2666 0 0 2666 a6a networking/tftp.o.+shrink --- networking/tftp.c | 286 ++++++++++++++++++++++++++---------------------------- 1 file changed, 139 insertions(+), 147 deletions(-) diff --git a/networking/tftp.c b/networking/tftp.c index 527e3dc11..e32c5de38 100644 --- a/networking/tftp.c +++ b/networking/tftp.c @@ -33,13 +33,22 @@ #include "busybox.h" -//#define CONFIG_FEATURE_TFTP_DEBUG -#define TFTP_BLOCKSIZE_DEFAULT 512 /* according to RFC 1350, don't change */ -#define TFTP_TIMEOUT 5 /* seconds */ +#define TFTP_BLOCKSIZE_DEFAULT 512 /* according to RFC 1350, don't change */ +#define TFTP_TIMEOUT 5 /* seconds */ +#define TFTP_NUM_RETRIES 5 /* number of retries */ -/* opcodes we support */ +/* RFC2348 says between 8 and 65464 */ +#define TFTP_OCTECTS_MIN 8 +#define TFTP_OCTECTS_MAX 65464 + +static const char * const MODE_OCTET = "octet"; +#define MODE_OCTET_LEN 6 /* sizeof(MODE_OCTET)*/ +static const char * const OPTION_BLOCKSIZE = "blksize"; +#define OPTION_BLOCKSIZE_LEN 8 /* sizeof(OPTION_BLOCKSIZE) */ + +/* opcodes we support */ #define TFTP_RRQ 1 #define TFTP_WRQ 2 #define TFTP_DATA 3 @@ -47,7 +56,7 @@ #define TFTP_ERROR 5 #define TFTP_OACK 6 -static const char * const tftp_bb_error_msg[] = { +static const char *const tftp_bb_error_msg[] = { "Undefined error", "File not found", "Access violation", @@ -58,13 +67,10 @@ static const char * const tftp_bb_error_msg[] = { "No such user" }; -#ifdef CONFIG_FEATURE_TFTP_GET -# define tftp_cmd_get 1 -#else -# define tftp_cmd_get 0 -#endif -#ifdef CONFIG_FEATURE_TFTP_PUT -# define tftp_cmd_put (tftp_cmd_get+1) +#define tftp_cmd_get ENABLE_FEATURE_TFTP_GET + +#if ENABLE_FEATURE_TFTP_PUT +# define tftp_cmd_put (tftp_cmd_get+ENABLE_FEATURE_TFTP_PUT) #else # define tftp_cmd_put 0 #endif @@ -81,15 +87,15 @@ static int tftp_blocksize_check(int blocksize, int bufsize) */ if ((bufsize && (blocksize > bufsize)) || - (blocksize < 8) || (blocksize > 65464)) { - bb_error_msg("bad blocksize"); - return 0; + (blocksize < TFTP_OCTECTS_MIN) || (blocksize > TFTP_OCTECTS_MAX)) { + bb_error_msg("bad blocksize"); + return 0; } return blocksize; } -static char *tftp_option_get(char *buf, int len, char *option) +static char *tftp_option_get(char *buf, int len, const char const *option) { int opt_val = 0; int opt_found = 0; @@ -97,25 +103,24 @@ static char *tftp_option_get(char *buf, int len, char *option) while (len > 0) { - /* Make sure the options are terminated correctly */ + /* Make sure the options are terminated correctly */ - for (k = 0; k < len; k++) { - if (buf[k] == '\0') { - break; + for (k = 0; k < len; k++) { + if (buf[k] == '\0') { + break; } } if (k >= len) { - break; + break; } if (opt_val == 0) { if (strcasecmp(buf, option) == 0) { - opt_found = 1; + opt_found = 1; } - } - else { - if (opt_found) { + } else { + if (opt_found) { return buf; } } @@ -133,38 +138,34 @@ static char *tftp_option_get(char *buf, int len, char *option) #endif -static inline int tftp(const int cmd, const struct hostent *host, - const char *remotefile, int localfd, const unsigned short port, int tftp_bufsize) +static int tftp(const int cmd, const struct hostent *host, + const char *remotefile, const int localfd, + const unsigned short port, int tftp_bufsize) { - const int cmd_get = cmd & tftp_cmd_get; - const int cmd_put = cmd & tftp_cmd_put; - const int bb_tftp_num_retries = 5; - +#define cmd_get cmd & tftp_cmd_get +#define cmd_put cmd & tftp_cmd_put struct sockaddr_in sa; struct sockaddr_in from; struct timeval tv; socklen_t fromlen; fd_set rfds; - char *cp; - unsigned short tmp; int socketfd; - int len; + int len, itmp; int opcode = 0; int finished = 0; - int timeout = bb_tftp_num_retries; + int timeout = TFTP_NUM_RETRIES; unsigned short block_nr = 1; + unsigned short tmp; + char *cp; -#ifdef CONFIG_FEATURE_TFTP_BLOCKSIZE - int want_option_ack = 0; -#endif + USE_FEATURE_TFTP_BLOCKSIZE(int want_option_ack = 0;) /* Can't use RESERVE_CONFIG_BUFFER here since the allocation * size varies meaning BUFFERS_GO_ON_STACK would fail */ - char *buf=xmalloc(tftp_bufsize + 4); - - tftp_bufsize += 4; + char *buf=xmalloc(tftp_bufsize += 4); - if ((socketfd = socket(PF_INET, SOCK_DGRAM, 0)) < 0) { /* bb_xsocket? */ + if ((socketfd = socket(PF_INET, SOCK_DGRAM, 0)) < 0) { + /* need to unlink the localfile, so don't use bb_xsocket here. */ bb_perror_msg("socket"); return EXIT_FAILURE; } @@ -180,11 +181,9 @@ static inline int tftp(const int cmd, const struct hostent *host, sizeof(sa.sin_addr)); /* build opcode */ - if (cmd_get) { opcode = TFTP_RRQ; } - if (cmd_put) { opcode = TFTP_WRQ; } @@ -194,56 +193,49 @@ static inline int tftp(const int cmd, const struct hostent *host, cp = buf; /* first create the opcode part */ - *((unsigned short *) cp) = htons(opcode); - cp += 2; /* add filename and mode */ - - if ((cmd_get && (opcode == TFTP_RRQ)) || - (cmd_put && (opcode == TFTP_WRQ))) { + if (((cmd_get) && (opcode == TFTP_RRQ)) || + ((cmd_put) && (opcode == TFTP_WRQ))) + { int too_long = 0; - /* see if the filename fits into buf */ - /* and fill in packet */ - + /* see if the filename fits into buf + * and fill in packet. */ len = strlen(remotefile) + 1; if ((cp + len) >= &buf[tftp_bufsize - 1]) { - too_long = 1; - } - else { - safe_strncpy(cp, remotefile, len); + too_long = 1; + } else { + safe_strncpy(cp, remotefile, len); cp += len; } - if (too_long || ((&buf[tftp_bufsize - 1] - cp) < 6)) { - bb_error_msg("too long remote-filename"); + if (too_long || ((&buf[tftp_bufsize - 1] - cp) < MODE_OCTET_LEN)) { + bb_error_msg("remote filename too long"); break; } /* add "mode" part of the package */ - - memcpy(cp, "octet", 6); - cp += 6; + memcpy(cp, MODE_OCTET, MODE_OCTET_LEN); + cp += MODE_OCTET_LEN; #ifdef CONFIG_FEATURE_TFTP_BLOCKSIZE - len = tftp_bufsize - 4; /* data block size */ + len = tftp_bufsize - 4; /* data block size */ if (len != TFTP_BLOCKSIZE_DEFAULT) { - if ((&buf[tftp_bufsize - 1] - cp) < 15) { - bb_error_msg("too long remote-filename"); + if ((&buf[tftp_bufsize - 1] - cp) < 15) { + bb_error_msg("remote filename too long"); break; } /* add "blksize" + number of blocks */ - - memcpy(cp, "blksize", 8); - cp += 8; - + memcpy(cp, OPTION_BLOCKSIZE, OPTION_BLOCKSIZE_LEN); + cp += OPTION_BLOCKSIZE_LEN; cp += snprintf(cp, 6, "%d", len) + 1; want_option_ack = 1; @@ -253,8 +245,8 @@ static inline int tftp(const int cmd, const struct hostent *host, /* add ack and data */ - if ((cmd_get && (opcode == TFTP_ACK)) || - (cmd_put && (opcode == TFTP_DATA))) { + if (((cmd_get) && (opcode == TFTP_ACK)) || + ((cmd_put) && (opcode == TFTP_DATA))) { *((unsigned short *) cp) = htons(block_nr); @@ -262,7 +254,7 @@ static inline int tftp(const int cmd, const struct hostent *host, block_nr++; - if (cmd_put && (opcode == TFTP_DATA)) { + if ((cmd_put) && (opcode == TFTP_DATA)) { len = bb_full_read(localfd, cp, tftp_bufsize - 4); if (len < 0) { @@ -282,7 +274,7 @@ static inline int tftp(const int cmd, const struct hostent *host, /* send packet */ - timeout = bb_tftp_num_retries; /* re-initialize */ + timeout = TFTP_NUM_RETRIES; /* re-initialize */ do { len = cp - buf; @@ -290,11 +282,11 @@ static inline int tftp(const int cmd, const struct hostent *host, #ifdef CONFIG_FEATURE_TFTP_DEBUG fprintf(stderr, "sending %u bytes\n", len); for (cp = buf; cp < &buf[len]; cp++) - fprintf(stderr, "%02x ", (unsigned char)*cp); + fprintf(stderr, "%02x ", (unsigned char) *cp); fprintf(stderr, "\n"); #endif if (sendto(socketfd, buf, len, 0, - (struct sockaddr *) &sa, sizeof(sa)) < 0) { + (struct sockaddr *) &sa, sizeof(sa)) < 0) { bb_perror_msg("send"); len = -1; break; @@ -316,10 +308,10 @@ static inline int tftp(const int cmd, const struct hostent *host, FD_ZERO(&rfds); FD_SET(socketfd, &rfds); - switch (select(socketfd + 1, &rfds, NULL, NULL, &tv)) { - case 1: + itmp = select(socketfd + 1, &rfds, NULL, NULL, &tv); + if (itmp == 1) { len = recvfrom(socketfd, buf, tftp_bufsize, 0, - (struct sockaddr *) &from, &fromlen); + (struct sockaddr *) &from, &fromlen); if (len < 0) { bb_perror_msg("recvfrom"); @@ -337,9 +329,9 @@ static inline int tftp(const int cmd, const struct hostent *host, /* fall-through for bad packets! */ /* discard the packet - treat as timeout */ - timeout = bb_tftp_num_retries; + timeout = TFTP_NUM_RETRIES; - case 0: + } else if (itmp == 0) { bb_error_msg("timeout"); timeout--; @@ -349,7 +341,7 @@ static inline int tftp(const int cmd, const struct hostent *host, } break; - default: + } else { bb_perror_msg("select"); len = -1; } @@ -362,7 +354,6 @@ static inline int tftp(const int cmd, const struct hostent *host, /* process received packet */ - opcode = ntohs(*((unsigned short *) buf)); tmp = ntohs(*((unsigned short *) &buf[2])); @@ -377,7 +368,7 @@ static inline int tftp(const int cmd, const struct hostent *host, msg = &buf[4]; buf[tftp_bufsize - 1] = '\0'; } else if (tmp < (sizeof(tftp_bb_error_msg) - / sizeof(char *))) { + / sizeof(char *))) { msg = tftp_bb_error_msg[tmp]; } @@ -388,55 +379,52 @@ static inline int tftp(const int cmd, const struct hostent *host, break; } - #ifdef CONFIG_FEATURE_TFTP_BLOCKSIZE if (want_option_ack) { - want_option_ack = 0; + want_option_ack = 0; - if (opcode == TFTP_OACK) { + if (opcode == TFTP_OACK) { - /* server seems to support options */ + /* server seems to support options */ - char *res; + char *res; - res = tftp_option_get(&buf[2], len-2, - "blksize"); + res = tftp_option_get(&buf[2], len - 2, OPTION_BLOCKSIZE); - if (res) { - int blksize = atoi(res); + if (res) { + int blksize = atoi(res); - if (tftp_blocksize_check(blksize, - tftp_bufsize - 4)) { + if (tftp_blocksize_check(blksize, tftp_bufsize - 4)) { - if (cmd_put) { - opcode = TFTP_DATA; - } - else { - opcode = TFTP_ACK; - } + if (cmd_put) { + opcode = TFTP_DATA; + } else { + opcode = TFTP_ACK; + } #ifdef CONFIG_FEATURE_TFTP_DEBUG - fprintf(stderr, "using blksize %u\n", blksize); + fprintf(stderr, "using %s %u\n", OPTION_BLOCKSIZE, + blksize); #endif - tftp_bufsize = blksize + 4; - block_nr = 0; - continue; - } - } - /* FIXME: - * we should send ERROR 8 */ - bb_error_msg("bad server option"); - break; - } - - bb_error_msg("warning: blksize not supported by server" - " - reverting to 512"); - - tftp_bufsize = TFTP_BLOCKSIZE_DEFAULT + 4; + tftp_bufsize = blksize + 4; + block_nr = 0; + continue; + } + } + /* FIXME: + * we should send ERROR 8 */ + bb_error_msg("bad server option"); + break; + } + + bb_error_msg("warning: blksize not supported by server" + " - reverting to 512"); + + tftp_bufsize = TFTP_BLOCKSIZE_DEFAULT + 4; } #endif - if (cmd_get && (opcode == TFTP_DATA)) { + if ((cmd_get) && (opcode == TFTP_DATA)) { if (tmp == block_nr) { @@ -455,7 +443,7 @@ static inline int tftp(const int cmd, const struct hostent *host, continue; } /* in case the last ack disappeared into the ether */ - if ( tmp == (block_nr - 1) ) { + if (tmp == (block_nr - 1)) { --block_nr; opcode = TFTP_ACK; continue; @@ -467,9 +455,9 @@ static inline int tftp(const int cmd, const struct hostent *host, } } - if (cmd_put && (opcode == TFTP_ACK)) { + if ((cmd_put) && (opcode == TFTP_ACK)) { - if (tmp == (unsigned short)(block_nr - 1)) { + if (tmp == (unsigned short) (block_nr - 1)) { if (finished) { break; } @@ -482,7 +470,6 @@ static inline int tftp(const int cmd, const struct hostent *host, #ifdef CONFIG_FEATURE_CLEAN_UP close(socketfd); - free(buf); #endif @@ -505,6 +492,7 @@ int tftp_main(int argc, char **argv) #ifdef CONFIG_FEATURE_TFTP_BLOCKSIZE char *sblocksize = NULL; + #define BS "b:" #define BS_ARG , &sblocksize #else @@ -533,62 +521,66 @@ int tftp_main(int argc, char **argv) #elif defined(CONFIG_FEATURE_TFTP_GET) || defined(CONFIG_FEATURE_TFTP_PUT) bb_opt_complementally = GET_COMPL PUT_COMPL; #else - /* XXX: may be should #error ? */ +#error "Either CONFIG_FEATURE_TFTP_GET or CONFIG_FEATURE_TFTP_PUT must be defined" #endif cmd = bb_getopt_ulflags(argc, argv, GET PUT "l:r:" BS, - &localfile, &remotefile BS_ARG); -#ifdef CONFIG_FEATURE_TFTP_BLOCKSIZE - if(sblocksize) { - blocksize = atoi(sblocksize); - if (!tftp_blocksize_check(blocksize, 0)) { - return EXIT_FAILURE; - } - } -#endif + &localfile, &remotefile BS_ARG); cmd &= (tftp_cmd_get | tftp_cmd_put); #ifdef CONFIG_FEATURE_TFTP_GET - if(cmd == tftp_cmd_get) + if (cmd == tftp_cmd_get) flags = O_WRONLY | O_CREAT | O_TRUNC; #endif #ifdef CONFIG_FEATURE_TFTP_PUT - if(cmd == tftp_cmd_put) + if (cmd == tftp_cmd_put) flags = O_RDONLY; #endif - if(localfile == NULL) - localfile = remotefile; - if(remotefile == NULL) - remotefile = localfile; - /* XXX: I corrected this, but may be wrong too. vodz */ - if(localfile==NULL || strcmp(localfile, "-") == 0) { - fd = fileno((cmd==tftp_cmd_get)? stdout : stdin); - } else if (fd==-1) { - fd = open(localfile, flags, 0644); +#ifdef CONFIG_FEATURE_TFTP_BLOCKSIZE + if (sblocksize) { + blocksize = atoi(sblocksize); + if (!tftp_blocksize_check(blocksize, 0)) { + return EXIT_FAILURE; + } + } +#endif + + if (localfile == NULL) + localfile = remotefile; + if (remotefile == NULL) + remotefile = localfile; + if ((localfile == NULL && remotefile == NULL) || (argv[optind] == NULL)) + bb_show_usage(); + + if (localfile == NULL || strcmp(localfile, "-") == 0) { + fd = (cmd == tftp_cmd_get) ? STDOUT_FILENO : STDIN_FILENO; + } else { + fd = open(localfile, flags, 0644); /* fail below */ } if (fd < 0) { bb_perror_msg_and_die("local file"); } - /* XXX: argv[optind] and/or argv[optind + 1] may be NULL! */ host = xgethostbyname(argv[optind]); port = bb_lookup_port(argv[optind + 1], "udp", 69); #ifdef CONFIG_FEATURE_TFTP_DEBUG fprintf(stderr, "using server \"%s\", remotefile \"%s\", " - "localfile \"%s\".\n", - inet_ntoa(*((struct in_addr *) host->h_addr)), - remotefile, localfile); + "localfile \"%s\".\n", + inet_ntoa(*((struct in_addr *) host->h_addr)), + remotefile, localfile); #endif result = tftp(cmd, host, remotefile, fd, port, blocksize); #ifdef CONFIG_FEATURE_CLEAN_UP if (!(fd == STDOUT_FILENO || fd == STDIN_FILENO)) { - close(fd); + close(fd); } #endif - return(result); + if (cmd == tftp_cmd_get && result != EXIT_SUCCESS) + unlink(localfile); + return (result); } -- cgit v1.2.3