When there is no reply from the TFTP server, do_tftpb() should
eventually time out. Add a 10 second timer for that purpose.

Reported-by: Heinrich Schuchardt <heinrich.schucha...@canonical.com>
Signed-off-by: Jerome Forissier <jerome.foriss...@linaro.org>
---

 net/lwip/tftp.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/net/lwip/tftp.c b/net/lwip/tftp.c
index 4f9b2049187..4f88924e47e 100644
--- a/net/lwip/tftp.c
+++ b/net/lwip/tftp.c
@@ -15,6 +15,8 @@
 #include <time.h>
 
 #define PROGRESS_PRINT_STEP_BYTES (10 * 1024)
+/* Max time to wait for first data packet from server */
+#define NO_RSP_TIMEOUT_MS 10000
 
 enum done_state {
        NOT_DONE = 0,
@@ -106,6 +108,17 @@ static const struct tftp_context tftp_context = {
        tftp_error
 };
 
+static void no_response(void *arg)
+{
+       struct tftp_ctx *ctx = (struct tftp_ctx *)arg;
+
+       if (ctx->size)
+               return;
+
+       printf("Timeout!\n");
+       ctx->done = FAILURE;
+}
+
 static int tftp_loop(struct udevice *udev, ulong addr, char *fname,
                     ip_addr_t srvip, uint16_t srvport)
 {
@@ -150,6 +163,7 @@ static int tftp_loop(struct udevice *udev, ulong addr, char 
*fname,
                return -1;
        }
 
+       sys_timeout(NO_RSP_TIMEOUT_MS, no_response, &ctx);
        while (!ctx.done) {
                net_lwip_rx(udev, netif);
                sys_check_timeouts();
@@ -159,6 +173,7 @@ static int tftp_loop(struct udevice *udev, ulong addr, char 
*fname,
                        break;
                }
        }
+       sys_untimeout(no_response, (void *)&ctx);
 
        tftp_cleanup();
 
-- 
2.43.0

base-commit: 233fda6af674736dbc6ff37a9ef003b9fa4b8074
branch: lwip-tftp-fixes

Reply via email to