From 804a6b148519d78a75b9cd45cf005bd8e4cb0fae Mon Sep 17 00:00:00 2001
From: Imbus <aa>
Date: Sat, 7 Feb 2026 13:52:13 +0100
Subject: [PATCH] Include the file size in the write job struct for future
 reference in data rate and progress bar

---
 writeimg.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/writeimg.c b/writeimg.c
index 8a445ce..9d3c47d 100644
--- a/writeimg.c
+++ b/writeimg.c
@@ -27,6 +27,8 @@
 #define BLOCKSIZE (4 * 1024 * 1024)
 #endif
 
+#define BYTES_TO_MIB(bts) ((double)bts / (1024 * 1024))
+
 // clang-format off
 const char help[] =
     "Usage:\n"
@@ -50,6 +52,7 @@ struct write_job {
     char  *buffer2; /* For memcmp integrity checks */
     size_t bufsize;
     size_t block_size;
+    size_t total_bytes;
     char   verify_only;
 } wjob = {0};
 
@@ -207,6 +210,7 @@ int main(int argc, char *argv[]) {
     wjob.bufsize = BLOCKSIZE;
     wjob.block_size = BLOCKSIZE;
 
+    wjob.total_bytes = file_stat.st_size;
     assert(file_stat.st_size >= 0);
 
     perform_write(&wjob);
@@ -216,7 +220,7 @@ int main(int argc, char *argv[]) {
     if (wjob.buffer2)
         free(wjob.buffer2);
 
-    printf("\nOkay!\n");
+    printf("\n%.1f MiB's written and verified.\nAll good!\n", BYTES_TO_MIB(wjob.total_bytes));
     exit(0);
     return 0;
 }