From 717efdf9fa05d91cac04a9e1a8c3cfcc50043532 Mon Sep 17 00:00:00 2001
From: Romein van Buren <romein@vburen.nl>
Date: Thu, 12 Jun 2025 09:22:32 +0200
Subject: [PATCH] Final chunk reading fixes

---
 fmp/fmp_file.go | 8 ++++----
 fmp/fmp_test.go | 1 -
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/fmp/fmp_file.go b/fmp/fmp_file.go
index 6ebb09e..9b4c1a5 100644
--- a/fmp/fmp_file.go
+++ b/fmp/fmp_file.go
@@ -3,7 +3,6 @@ package fmp
 import (
 	"bytes"
 	"encoding/binary"
-	"encoding/hex"
 	"io"
 	"os"
 	"time"
@@ -58,8 +57,10 @@ func OpenFile(path string) (*FmpFile, error) {
 	ctx.Sectors = make([]*FmpSector, ctx.NumSectors)
 
 	for i := uint(0); i < ctx.NumSectors; i++ {
-		println("reading sector ", i)
 		sector, err := ctx.readSector()
+		if err == io.EOF {
+			break
+		}
 		if err != nil {
 			return nil, err
 		}
@@ -122,7 +123,6 @@ func (ctx *FmpFile) readSector() (*FmpSector, error) {
 	sector.Chunks = make([]*FmpChunk, 0)
 
 	for {
-		println(hex.EncodeToString(payload[0:2]))
 		chunk, err := ctx.readChunk(payload)
 		if err == io.EOF {
 			break
@@ -138,7 +138,7 @@ func (ctx *FmpFile) readSector() (*FmpSector, error) {
 			panic("chunk length not set")
 		}
 		payload = payload[min(chunk.Length, uint32(len(payload))):]
-		if len(payload) == 0 {
+		if len(payload) == 0 || (len(payload) == 1 && payload[0] == 0x00) {
 			break
 		}
 	}
diff --git a/fmp/fmp_test.go b/fmp/fmp_test.go
index 42086fc..b48b73f 100644
--- a/fmp/fmp_test.go
+++ b/fmp/fmp_test.go
@@ -19,5 +19,4 @@ func TestOpenFile(t *testing.T) {
 	if f.VersionDate.Format("2006-01-02") != "2025-01-11" {
 		t.Errorf("expected version date to be '2025-01-11', got '%s'", f.VersionDate.Format("2006-01-02"))
 	}
-	print(f.Sectors[0])
 }