From 0c81b639fb842469acbd616704e2673952bd1a26 Mon Sep 17 00:00:00 2001
From: N00byEdge <hannesbredberg@gmail.com>
Date: Tue, 1 Dec 2020 23:50:49 +0100
Subject: [PATCH] Don't stop download if tftp server sends packets smaller than
 the negotiated mtu

---
 stage2/pxe/tftp.c | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/stage2/pxe/tftp.c b/stage2/pxe/tftp.c
index 9e546599..83435652 100644
--- a/stage2/pxe/tftp.c
+++ b/stage2/pxe/tftp.c
@@ -61,6 +61,8 @@ int tftp_open(struct tftp_file_handle* handle, uint32_t server_ip, uint16_t serv
     size_t to_transfer = handle->file_size;
     size_t progress = 0;
 
+    bool slow = false;
+
     while (to_transfer > 0) {
         volatile struct pxenv_read read = {
             .boff = ((uint16_t)rm_off(buf)),
@@ -72,8 +74,9 @@ int tftp_open(struct tftp_file_handle* handle, uint32_t server_ip, uint16_t serv
         }
         memcpy(handle->data + progress, buf, read.bsize);
 
-        if (read.bsize < mtu) {
-            break;
+        if (read.bsize < mtu && !slow) {
+            slow = true;
+            print("Server is sending the file in smaller packets (it sent %d bytes), download might take longer.\n", read.bsize);
         }
         to_transfer -= read.bsize;
         progress += read.bsize;