From 41625a0955087cb79c9a7cbca39078f685952e79 Mon Sep 17 00:00:00 2001
From: Dreaded_X <tim.huizinga@gmail.com>
Date: Mon, 18 Jan 2021 17:58:59 +0100
Subject: [PATCH] Fixed progress reporting

---
 src/main.cpp | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/main.cpp b/src/main.cpp
index fa8b2fd..f915b88 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -179,8 +179,11 @@ void program_rom(std::unique_ptr<serial::Serial>& ser, std::string filename, uin
 
 	// Track progress
 	std::cout << "Programming... (base address: 0x" << std::hex << address << std::dec << ")\n";
-	for (uint16_t progress = 0; progress != length; progress = read_length<uint16_t>(ser)) {
+	for (uint16_t progress = 0; ; progress = read_length<uint16_t>(ser)) {
 		std::cout << "Progress: " << progress << '/' << length << '\r';
+		if (progress == length) {
+			break;
+		}
 	}
 	std::cout << '\n';
 
@@ -246,8 +249,11 @@ void program_i2c(std::unique_ptr<serial::Serial>& ser, std::string filename, uin
 
 	// Track progress
 	std::cout << "Programming...\n";
-	for (uint16_t progress = 0; progress != length; progress = read_length<uint16_t>(ser)) {
+	for (uint16_t progress = 0; ; progress = read_length<uint16_t>(ser)) {
 		std::cout << "Progress: " << progress << '/' << length << '\r';
+		if (progress == length) {
+			break;
+		}
 	}
 	std::cout << '\n';