diff options
-rw-r--r-- | networking/tftp.c | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/networking/tftp.c b/networking/tftp.c index 334d03b6a..9b9747785 100644 --- a/networking/tftp.c +++ b/networking/tftp.c @@ -320,7 +320,7 @@ static inline int tftp(const int cmd, const struct hostent *host, FD_ZERO(&rfds); FD_SET(socketfd, &rfds); - switch (select(FD_SETSIZE, &rfds, NULL, NULL, &tv)) { + switch (select(socketfd + 1, &rfds, NULL, NULL, &tv)) { case 1: len = recvfrom(socketfd, buf, tftp_bufsize, 0, (struct sockaddr *) &from, &fromlen); @@ -463,6 +463,11 @@ static inline int tftp(const int cmd, const struct hostent *host, --block_nr; opcode = TFTP_ACK; continue; + } else if (tmp + 1 == block_nr) { + /* Server lost our TFTP_ACK. Resend it */ + block_nr = tmp; + opcode = TFTP_ACK; + continue; } } |