emo/busybox
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tftpd: make it emit error packets

Browse Source 
telnetd: use login always, not "sometimes login, sometimes shell"

   text    data     bss     dec     hex filename
 797612     641    7380  805633   c4b01 busybox_old
 797695     641    7380  805716   c4b54 busybox_unstripped
This commit is contained in:
Denis Vlasenko 2008-03-19 05:00:05 +00:00
parent c0183e6e0d
commit dd9228b861
2 changed files with 183 additions and 123 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
networking/telnetd.c
View File

@ -56,11 +56,7 @@ enum { BUFSIZE = (4 * 1024 - sizeof(struct tsession)) / 2 };
/* Globals */ /* Globals */
static int maxfd; static int maxfd;
static struct tsession *sessions; static struct tsession *sessions;
#if ENABLE_LOGIN
static const char *loginpath = "/bin/login"; static const char *loginpath = "/bin/login";
#else
static const char *loginpath = DEFAULT_SHELL;
#endif
static const char *issuefile = "/etc/issue.net"; static const char *issuefile = "/etc/issue.net";

302
networking/tftp.c
View File

@ -25,7 +25,7 @@
#if ENABLE_FEATURE_TFTP_GET || ENABLE_FEATURE_TFTP_PUT #if ENABLE_FEATURE_TFTP_GET || ENABLE_FEATURE_TFTP_PUT
#define TFTP_BLOCKSIZE_DEFAULT 512 /* according to RFC 1350, don't change */ #define TFTP_BLKSIZE_DEFAULT 512 /* according to RFC 1350, don't change */
#define TFTP_TIMEOUT_MS 50 #define TFTP_TIMEOUT_MS 50
#define TFTP_MAXTIMEOUT_MS 2000 #define TFTP_MAXTIMEOUT_MS 2000
#define TFTP_NUM_RETRIES 12 /* number of backed-off retries */ #define TFTP_NUM_RETRIES 12 /* number of backed-off retries */
@ -38,6 +38,17 @@
#define TFTP_ERROR 5 #define TFTP_ERROR 5
#define TFTP_OACK 6 #define TFTP_OACK 6
/* error codes sent over network */
#define ERR_UNSPEC 0
#define ERR_NOFILE 1
#define ERR_ACCESS 2
#define ERR_WRITE 3
#define ERR_OP 4
#define ERR_BAD_ID 5
#define ERR_EXIST 6
#define ERR_BAD_USER 7
#define ERR_BAD_OPT 8
#if ENABLE_FEATURE_TFTP_GET && !ENABLE_FEATURE_TFTP_PUT #if ENABLE_FEATURE_TFTP_GET && !ENABLE_FEATURE_TFTP_PUT
#define USE_GETPUT(...) #define USE_GETPUT(...)
#define CMD_GET(cmd) 1 #define CMD_GET(cmd) 1
@ -56,29 +67,48 @@
* CMD_GET(cmd) and CMD_PUT(cmd) are mutually exclusive * CMD_GET(cmd) and CMD_PUT(cmd) are mutually exclusive
*/ */
// TODO: emit error packets before dying
struct globals {
/* u16 TFTP_ERROR; u16 reason; both network-endian, then error text: */
uint8_t error_pkt[4 + 32];
/* used in tftpd_main(), a bit big fro stack: */
char block_buf[TFTP_BLKSIZE_DEFAULT];
};
#define G (*(struct globals*)&bb_common_bufsiz1)
#define block_buf (G.block_buf )
#define error_pkt (G.error_pkt )
#define INIT_G() \
do { \
} while (0)
#define error_pkt_reason (error_pkt[3])
#define error_pkt_str (error_pkt + 4)
#if ENABLE_FEATURE_TFTP_BLOCKSIZE #if ENABLE_FEATURE_TFTP_BLOCKSIZE
static int tftp_blocksize_check(int blocksize, int bufsize) static int tftp_blksize_check(const char *blksize_str, int maxsize)
{ {
/* Check if the blocksize is valid: /* Check if the blksize is valid:
* RFC2348 says between 8 and 65464, * RFC2348 says between 8 and 65464,
* but our implementation makes it impossible * but our implementation makes it impossible
* to use blocksizes smaller than 22 octets. * to use blksizes smaller than 22 octets. */
*/ unsigned blksize = bb_strtou(blksize_str, NULL, 10);
if ((bufsize && (blocksize > bufsize)) if (errno
|| (blocksize < 24) || (blocksize > 65564) || (blksize < 24) || (blksize > maxsize)
) { ) {
bb_error_msg("bad blocksize"); bb_error_msg("bad blocksize '%s'", blksize_str);
return 0; return -1;
} }
return blocksize; #if ENABLE_DEBUG_TFTP
bb_error_msg("using blksize %u", blksize);
#endif
return blksize;
} }
static char *tftp_option_get(char *buf, int len, const char *option) static char *tftp_get_blksize(char *buf, int len)
{ {
#define option "blksize"
int opt_val = 0; int opt_val = 0;
int opt_found = 0; int opt_found = 0;
int k; int k;
@ -110,6 +140,7 @@ static char *tftp_option_get(char *buf, int len, const char *option)
} }
return NULL; return NULL;
#undef option
} }
#endif #endif
@ -120,7 +151,7 @@ static int tftp_protocol(
len_and_sockaddr *peer_lsa, len_and_sockaddr *peer_lsa,
USE_TFTP(const char *remote_file,) USE_TFTP(const char *remote_file,)
int local_fd, int local_fd,
int blocksize) int blksize)
{ {
#if !ENABLE_TFTP #if !ENABLE_TFTP
#define remote_file NULL #define remote_file NULL
@ -135,14 +166,14 @@ static int tftp_protocol(
uint16_t block_nr; uint16_t block_nr;
uint16_t recv_blk; uint16_t recv_blk;
int retries, waittime_ms; int retries, waittime_ms;
int tftp_bufsize = blocksize + 4; int io_bufsize = blksize + 4;
char *cp; char *cp;
/* Can't use RESERVE_CONFIG_BUFFER here since the allocation /* Can't use RESERVE_CONFIG_BUFFER here since the allocation
* size varies meaning BUFFERS_GO_ON_STACK would fail */ * size varies meaning BUFFERS_GO_ON_STACK would fail */
/* We must keep the transmit and receive buffers seperate */ /* We must keep the transmit and receive buffers seperate */
/* In case we rcv a garbage pkt and we need to rexmit the last pkt */ /* In case we rcv a garbage pkt and we need to rexmit the last pkt */
char *xbuf = xmalloc(tftp_bufsize); char *xbuf = xmalloc(io_bufsize);
char *rbuf = xmalloc(tftp_bufsize); char *rbuf = xmalloc(io_bufsize);
socket_fd = xsocket(peer_lsa->u.sa.sa_family, SOCK_DGRAM, 0); socket_fd = xsocket(peer_lsa->u.sa.sa_family, SOCK_DGRAM, 0);
setsockopt_reuseaddr(socket_fd); setsockopt_reuseaddr(socket_fd);
@ -160,6 +191,10 @@ static int tftp_protocol(
xbind(socket_fd, &our_lsa->u.sa, our_lsa->len); xbind(socket_fd, &our_lsa->u.sa, our_lsa->len);
xconnect(socket_fd, &peer_lsa->u.sa, peer_lsa->len); xconnect(socket_fd, &peer_lsa->u.sa, peer_lsa->len);
/* Is there an error already? Send pkt and bail out */
if (error_pkt_reason || error_pkt_str[0])
goto send_err_pkt;
if (CMD_GET(cmd)) { if (CMD_GET(cmd)) {
/* it's upload - we must ACK 1st packet (with filename) /* it's upload - we must ACK 1st packet (with filename)
* as if it's "block 0" */ * as if it's "block 0" */
@ -167,10 +202,11 @@ static int tftp_protocol(
} }
#if ENABLE_FEATURE_TFTP_BLOCKSIZE #if ENABLE_FEATURE_TFTP_BLOCKSIZE
if (blocksize != TFTP_BLOCKSIZE_DEFAULT) { if (blksize != TFTP_BLKSIZE_DEFAULT) {
/* Create and send OACK packet */ /* Create and send OACK packet */
/* block_nr is still 1, we expect ACK to (block_nr-1), /* For the download case, block_nr is still 1 -
* that is, to "block 0" */ * we expect 1st ACK from peer to be for (block_nr-1),
* that is, for "block 0" which is our OACK pkt */
opcode = TFTP_OACK; opcode = TFTP_OACK;
cp = xbuf + 2; cp = xbuf + 2;
goto add_blksize_opt; goto add_blksize_opt;
@ -200,7 +236,7 @@ static int tftp_protocol(
/* add filename and mode */ /* add filename and mode */
/* fill in packet if the filename fits into xbuf */ /* fill in packet if the filename fits into xbuf */
len = strlen(remote_file) + 1; len = strlen(remote_file) + 1;
if (2 + len + sizeof("octet") >= tftp_bufsize) { if (2 + len + sizeof("octet") >= io_bufsize) {
bb_error_msg("remote filename is too long"); bb_error_msg("remote filename is too long");
goto ret; goto ret;
} }
@ -211,18 +247,18 @@ static int tftp_protocol(
cp += sizeof("octet"); cp += sizeof("octet");
#if ENABLE_FEATURE_TFTP_BLOCKSIZE #if ENABLE_FEATURE_TFTP_BLOCKSIZE
if (blocksize != TFTP_BLOCKSIZE_DEFAULT) { if (blksize != TFTP_BLKSIZE_DEFAULT) {
/* rfc2348 says that 65464 is a max allowed value */ /* rfc2348 says that 65464 is a max allowed value */
if ((&xbuf[tftp_bufsize - 1] - cp) < sizeof("blksize NNNNN")) { if ((&xbuf[io_bufsize - 1] - cp) < sizeof("blksize NNNNN")) {
bb_error_msg("remote filename is too long"); bb_error_msg("remote filename is too long");
goto ret; goto ret;
} }
want_option_ack = 1; want_option_ack = 1;
add_blksize_opt: add_blksize_opt:
/* add "blksize", <nul>, blocksize, <nul> */ /* add "blksize", <nul>, blksize, <nul> */
strcpy(cp, "blksize"); strcpy(cp, "blksize");
cp += sizeof("blksize"); cp += sizeof("blksize");
cp += snprintf(cp, 6, "%d", blocksize) + 1; cp += snprintf(cp, 6, "%d", blksize) + 1;
} }
#endif #endif
/* First packet is built, so skip packet generation */ /* First packet is built, so skip packet generation */
@ -240,12 +276,11 @@ static int tftp_protocol(
opcode = TFTP_ACK; opcode = TFTP_ACK;
if (CMD_PUT(cmd)) { if (CMD_PUT(cmd)) {
opcode = TFTP_DATA; opcode = TFTP_DATA;
len = full_read(local_fd, cp, tftp_bufsize - 4); len = full_read(local_fd, cp, blksize);
if (len < 0) { if (len < 0) {
bb_perror_msg(bb_msg_read_error); goto send_read_err_pkt;
goto ret;
} }
if (len != (tftp_bufsize - 4)) { if (len != blksize) {
finished = 1; finished = 1;
} }
cp += len; cp += len;
@ -277,34 +312,14 @@ static int tftp_protocol(
/*pfd[0].fd = socket_fd;*/ /*pfd[0].fd = socket_fd;*/
pfd[0].events = POLLIN; pfd[0].events = POLLIN;
switch (safe_poll(pfd, 1, waittime_ms)) { switch (safe_poll(pfd, 1, waittime_ms)) {
case 1: default:
if (!our_lsa) { /*bb_perror_msg("poll"); - done in safe_poll */
/* tftp (not tftpd!) receiving 1st packet */ goto ret;
our_lsa = ((void*)(ptrdiff_t)-1); /* not NULL */
len = recvfrom(socket_fd, rbuf, tftp_bufsize, 0,
&peer_lsa->u.sa, &peer_lsa->len);
/* Our first dgram went to port 69
* but reply may come from different one.
* Remember and use this new port (and IP) */
if (len >= 0)
xconnect(socket_fd, &peer_lsa->u.sa, peer_lsa->len);
} else {
/* tftpd, or not the very first packet:
* socket is connect()ed, can just read from it. */
len = safe_read(socket_fd, rbuf, tftp_bufsize);
}
if (len < 0) {
bb_perror_msg("read");
goto ret;
}
if (len < 4) /* too small? */
goto recv_again;
goto process_pkt;
case 0: case 0:
retries--; retries--;
if (retries == 0) { if (retries == 0) {
bb_error_msg("timeout"); bb_error_msg("timeout");
goto ret; goto ret; /* no err packet sent */
} }
/* exponential backoff with limit */ /* exponential backoff with limit */
@ -314,39 +329,58 @@ static int tftp_protocol(
} }
goto send_again; /* resend last sent pkt */ goto send_again; /* resend last sent pkt */
default: case 1:
/*bb_perror_msg("poll"); - done in safe_poll */ if (!our_lsa) {
goto ret; /* tftp (not tftpd!) receiving 1st packet */
our_lsa = ((void*)(ptrdiff_t)-1); /* not NULL */
len = recvfrom(socket_fd, rbuf, io_bufsize, 0,
&peer_lsa->u.sa, &peer_lsa->len);
/* Our first dgram went to port 69
* but reply may come from different one.
* Remember and use this new port (and IP) */
if (len >= 0)
xconnect(socket_fd, &peer_lsa->u.sa, peer_lsa->len);
} else {
/* tftpd, or not the very first packet:
* socket is connect()ed, can just read from it. */
/* Don't full_read()!
* This is not TCP, one read == one pkt! */
len = safe_read(socket_fd, rbuf, io_bufsize);
}
if (len < 0) {
goto send_read_err_pkt;
}
if (len < 4) { /* too small? */
goto recv_again;
}
} }
process_pkt:
/* Process recv'ed packet */ /* Process recv'ed packet */
opcode = ntohs( ((uint16_t*)rbuf)[0] ); opcode = ntohs( ((uint16_t*)rbuf)[0] );
recv_blk = ntohs( ((uint16_t*)rbuf)[1] ); recv_blk = ntohs( ((uint16_t*)rbuf)[1] );
#if ENABLE_DEBUG_TFTP #if ENABLE_DEBUG_TFTP
fprintf(stderr, "received %d bytes: %04x %04x\n", len, opcode, recv_blk); fprintf(stderr, "received %d bytes: %04x %04x\n", len, opcode, recv_blk);
#endif #endif
if (opcode == TFTP_ERROR) { if (opcode == TFTP_ERROR) {
static const char *const errcode_str[] = { static const char errcode_str[] =
"", "\0"
"file not found", "file not found\0"
"access violation", "access violation\0"
"disk full", "disk full\0"
"illegal TFTP operation", "bad operation\0"
"unknown transfer id", "unknown transfer id\0"
"file already exists", "file already exists\0"
"no such user", "no such user\0"
"bad option" "bad option";
};
const char *msg = ""; const char *msg = "";
if (len > 4 && rbuf[4] != '\0') { if (len > 4 && rbuf[4] != '\0') {
msg = &rbuf[4]; msg = &rbuf[4];
rbuf[tftp_bufsize - 1] = '\0'; rbuf[io_bufsize - 1] = '\0'; /* paranoia */
} else if (recv_blk < ARRAY_SIZE(errcode_str)) { } else if (recv_blk <= 8) {
msg = errcode_str[recv_blk]; msg = nth_string(errcode_str, recv_blk);
} }
bb_error_msg("server error: (%u) %s", recv_blk, msg); bb_error_msg("server error: (%u) %s", recv_blk, msg);
goto ret; goto ret;
@ -355,29 +389,18 @@ static int tftp_protocol(
#if ENABLE_FEATURE_TFTP_BLOCKSIZE #if ENABLE_FEATURE_TFTP_BLOCKSIZE
if (want_option_ack) { 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(&rbuf[2], len - 2, "blksize"); res = tftp_get_blksize(&rbuf[2], len - 2);
if (res) { if (res) {
int blksize = xatoi_u(res); blksize = tftp_blksize_check(res, blksize);
if (!tftp_blocksize_check(blksize, tftp_bufsize - 4)) { if (blksize < 0) {
/* send ERROR 8 to server... */ error_pkt_reason = ERR_BAD_OPT;
/* htons can be impossible to use in const initializer: */ goto send_err_pkt;
/*static const uint16_t error_8[2] = { htons(TFTP_ERROR), htons(8) };*/
/* thus we open-code big-endian layout */
static const uint8_t error_8[4] = { 0,TFTP_ERROR, 0,8 };
xsendto(socket_fd, error_8, 4, &peer_lsa->u.sa, peer_lsa->len);
bb_error_msg("server proposes bad blksize %d, exiting", blksize);
goto ret;
} }
#if ENABLE_DEBUG_TFTP io_bufsize = blksize + 4;
fprintf(stderr, "using blksize %u\n",
blksize);
#endif
tftp_bufsize = blksize + 4;
/* Send ACK for OACK ("block" no: 0) */ /* Send ACK for OACK ("block" no: 0) */
block_nr = 0; block_nr = 0;
continue; continue;
@ -387,10 +410,9 @@ static int tftp_protocol(
* must be ignored by the client and server * must be ignored by the client and server
* as if it were never requested." */ * as if it were never requested." */
} }
bb_error_msg("server only supports blocksize of 512");
bb_error_msg("blksize is not supported by server" blksize = TFTP_BLKSIZE_DEFAULT;
" - reverting to 512"); io_bufsize = TFTP_BLKSIZE_DEFAULT + 4;
tftp_bufsize = TFTP_BLOCKSIZE_DEFAULT + 4;
} }
#endif #endif
/* block_nr is already advanced to next block# we expect /* block_nr is already advanced to next block# we expect
@ -401,9 +423,10 @@ static int tftp_protocol(
len = full_write(local_fd, &rbuf[4], len - 4); len = full_write(local_fd, &rbuf[4], len - 4);
if (len < 0) { if (len < 0) {
bb_perror_msg(bb_msg_write_error); bb_perror_msg(bb_msg_write_error);
goto ret; error_pkt_reason = ERR_WRITE;
goto send_err_pkt;
} }
if (len != (tftp_bufsize - 4)) { if (len != blksize) {
finished = 1; finished = 1;
} }
continue; /* send ACK */ continue; /* send ACK */
@ -442,6 +465,15 @@ static int tftp_protocol(
free(rbuf); free(rbuf);
} }
return finished == 0; /* returns 1 on failure */ return finished == 0; /* returns 1 on failure */
send_read_err_pkt:
bb_perror_msg(bb_msg_read_error);
strcpy(error_pkt_str, bb_msg_read_error);
send_err_pkt:
error_pkt[1] = TFTP_ERROR;
xsendto(socket_fd, error_pkt, 4 + 1 + strlen(error_pkt_str),
&peer_lsa->u.sa, peer_lsa->len);
return EXIT_FAILURE;
} }
#if ENABLE_TFTP #if ENABLE_TFTP
@ -457,18 +489,20 @@ int tftp_main(int argc ATTRIBUTE_UNUSED, char **argv)
int local_fd; int local_fd;
int flags = 0; int flags = 0;
int result; int result;
int blocksize = TFTP_BLOCKSIZE_DEFAULT; int blksize;
const char *blksize_str = "512";
INIT_G();
/* -p or -g is mandatory, and they are mutually exclusive */ /* -p or -g is mandatory, and they are mutually exclusive */
opt_complementary = "" USE_FEATURE_TFTP_GET("g:") USE_FEATURE_TFTP_PUT("p:") opt_complementary = "" USE_FEATURE_TFTP_GET("g:") USE_FEATURE_TFTP_PUT("p:")
USE_GETPUT("g--p:p--g:") USE_GETPUT("g--p:p--g:");
USE_FEATURE_TFTP_BLOCKSIZE("b+");
USE_GETPUT(cmd =) getopt32(argv, USE_GETPUT(cmd =) getopt32(argv,
USE_FEATURE_TFTP_GET("g") USE_FEATURE_TFTP_PUT("p") USE_FEATURE_TFTP_GET("g") USE_FEATURE_TFTP_PUT("p")
"l:r:" USE_FEATURE_TFTP_BLOCKSIZE("b:"), "l:r:" USE_FEATURE_TFTP_BLOCKSIZE("b:"),
&local_file, &remote_file &local_file, &remote_file
USE_FEATURE_TFTP_BLOCKSIZE(, &blocksize)); USE_FEATURE_TFTP_BLOCKSIZE(, &blksize_str));
argv += optind; argv += optind;
flags = O_RDONLY; flags = O_RDONLY;
@ -476,8 +510,13 @@ int tftp_main(int argc ATTRIBUTE_UNUSED, char **argv)
flags = O_WRONLY | O_CREAT | O_TRUNC; flags = O_WRONLY | O_CREAT | O_TRUNC;
#if ENABLE_FEATURE_TFTP_BLOCKSIZE #if ENABLE_FEATURE_TFTP_BLOCKSIZE
if (!tftp_blocksize_check(blocksize, 0)) /* Check if the blksize is valid:
* RFC2348 says between 8 and 65464 */
blksize = tftp_blksize_check(blksize_str, 65564);
if (blksize < 0) {
//bb_error_msg("bad block size");
return EXIT_FAILURE; return EXIT_FAILURE;
}
#endif #endif
if (!local_file) if (!local_file)
@ -506,7 +545,7 @@ int tftp_main(int argc ATTRIBUTE_UNUSED, char **argv)
USE_GETPUT(cmd,) USE_GETPUT(cmd,)
NULL /* our_lsa*/, NULL /* our_lsa*/,
peer_lsa, peer_lsa,
remote_file, local_fd, blocksize); remote_file, local_fd, blksize);
if (ENABLE_FEATURE_CLEAN_UP) if (ENABLE_FEATURE_CLEAN_UP)
close(local_fd); close(local_fd);
@ -537,12 +576,16 @@ static len_and_sockaddr *get_sock_lsa(int s)
int tftpd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int tftpd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int tftpd_main(int argc ATTRIBUTE_UNUSED, char **argv) int tftpd_main(int argc ATTRIBUTE_UNUSED, char **argv)
{ {
char block_buf[TFTP_BLOCKSIZE_DEFAULT];
len_and_sockaddr *our_lsa; len_and_sockaddr *our_lsa;
len_and_sockaddr *peer_lsa; len_and_sockaddr *peer_lsa;
char *filename, *mode, *opt_str; char *filename, *mode;
int opt_r, result, opcode, open_mode, local_fd, blksize; const char *error_msg;
USE_GETPUT(int cmd;) int opt_r, result, opcode, open_mode;
int local_fd = local_fd; /* for compiler */
int blksize = blksize;
USE_GETPUT(int cmd = cmd;)
INIT_G();
our_lsa = get_sock_lsa(STDIN_FILENO); our_lsa = get_sock_lsa(STDIN_FILENO);
if (!our_lsa) if (!our_lsa)
@ -559,6 +602,7 @@ int tftpd_main(int argc ATTRIBUTE_UNUSED, char **argv)
0 /* flags */, 0 /* flags */,
&peer_lsa->u.sa, &our_lsa->u.sa, our_lsa->len); &peer_lsa->u.sa, &our_lsa->u.sa, our_lsa->len);
error_msg = "malformed packet";
opcode = ntohs(*(uint16_t*)block_buf); opcode = ntohs(*(uint16_t*)block_buf);
if (result < 4 || result >= sizeof(block_buf) if (result < 4 || result >= sizeof(block_buf)
|| block_buf[result-1] != '\0' || block_buf[result-1] != '\0'
@ -567,27 +611,35 @@ int tftpd_main(int argc ATTRIBUTE_UNUSED, char **argv)
USE_FEATURE_TFTP_GET(opcode != TFTP_WRQ) /* not upload */ USE_FEATURE_TFTP_GET(opcode != TFTP_WRQ) /* not upload */
) )
) { ) {
bb_error_msg_and_die("malformed packet"); goto err;
} }
filename = block_buf + 2; filename = block_buf + 2;
if (filename[0] == '.' || strstr(filename, "/.")) { if (filename[0] == '.' || strstr(filename, "/.")) {
bb_error_msg_and_die("dot in filename"); error_msg = "dot in filename";
goto err;
} }
mode = filename + strlen(filename) + 1; mode = filename + strlen(filename) + 1;
if (mode >= block_buf + sizeof(block_buf) if (mode >= block_buf + result
|| strcmp(mode, "octet") != 0 || strcmp(mode, "octet") != 0
) { ) {
bb_error_msg_and_die("malformed packet"); goto err;
} }
blksize = TFTP_BLOCKSIZE_DEFAULT; blksize = TFTP_BLKSIZE_DEFAULT;
#if ENABLE_FEATURE_TFTP_BLOCKSIZE #if ENABLE_FEATURE_TFTP_BLOCKSIZE
opt_str = mode + 6; {
if (opt_str < block_buf + sizeof(block_buf)) { char *res;
char *res = tftp_option_get(opt_str, block_buf + sizeof(block_buf) - opt_str, "blksize"); char *opt_str = mode + sizeof("octet");
if (res) { int opt_len = block_buf + result - opt_str;
int sz = xatoi_u(res); if (opt_len > 0) {
if (tftp_blocksize_check(sz, 0)) res = tftp_get_blksize(opt_str, opt_len);
blksize = sz; if (res) {
blksize = tftp_blksize_check(res, 65564);
if (blksize < 0) {
error_pkt_reason = ERR_BAD_OPT;
/* will just send error pkt */
goto do_proto;
}
}
} }
} }
#endif #endif
@ -599,16 +651,28 @@ int tftpd_main(int argc ATTRIBUTE_UNUSED, char **argv)
#endif #endif
#if ENABLE_FEATURE_TFTP_GET #if ENABLE_FEATURE_TFTP_GET
if (!ENABLE_FEATURE_TFTP_PUT || opcode == TFTP_WRQ) { if (!ENABLE_FEATURE_TFTP_PUT || opcode == TFTP_WRQ) {
if (opt_r) if (opt_r) {
bb_error_msg_and_die("upload is prohibited"); error_pkt_reason = ERR_WRITE;
/* will just send error pkt */
goto do_proto;
}
USE_GETPUT(cmd = 1;) /* CMD_GET: we will receive file's data */ USE_GETPUT(cmd = 1;) /* CMD_GET: we will receive file's data */
open_mode = O_WRONLY | O_TRUNC; open_mode = O_WRONLY | O_TRUNC;
} }
#endif #endif
local_fd = xopen(filename, open_mode); local_fd = open(filename, open_mode);
if (local_fd < 0) {
error_msg = "can't open file";
err:
strcpy(error_pkt_str, error_msg);
}
close(STDIN_FILENO); /* close old, possibly wildcard socket */ close(STDIN_FILENO); /* close old, possibly wildcard socket */
/* tftp_protocol() will create new one, bound to particular local IP */ /* tftp_protocol() will create new one, bound to particular local IP */
/* NB: if error_pkt_str or error_pkt_reason is set up,
* tftp_protocol() just sends one error pkt and returns */
do_proto:
result = tftp_protocol( result = tftp_protocol(
USE_GETPUT(cmd,) USE_GETPUT(cmd,)
our_lsa, peer_lsa, our_lsa, peer_lsa,