From 55bb6b5fd551ca3e758647ed9a2edd2272e38d73 Mon Sep 17 00:00:00 2001
From: tudnai <tamas.rudnai@gmail.com>
Date: Tue, 9 Jun 2020 00:54:06 -0700
Subject: [PATCH] Few bugfixes

---
 src/dev/disk/woz.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/dev/disk/woz.c b/src/dev/disk/woz.c
index 217b74b..6a3dfc8 100644
--- a/src/dev/disk/woz.c
+++ b/src/dev/disk/woz.c
@@ -381,6 +381,7 @@ uint8_t woz_read() {
     const int magicShiftOffset = 50;
     
     uint16_t usedBytes = woz_trks[track].bytes_used < WOZ_TRACK_BYTE_COUNT ? woz_trks[track].bytes_used : WOZ_TRACK_BYTE_COUNT;
+    
     if ( usedBytes ) {
         if ( clkelpased > clkBeforeAdjusting ) {
 //            printf("NEED SYNC : %llu (clkBefRd:%d)\n", clkelpased, clkBeforeSync);
@@ -392,6 +393,7 @@ uint8_t woz_read() {
             // preroll data stream
             WOZread.shift16 = 0;
             WOZread.data = woz_trks[track].data[trackOffset++];
+            trackOffset %= usedBytes;
             
             WOZread.shift16 <<= bitOffset;
 
@@ -399,11 +401,12 @@ uint8_t woz_read() {
                 for ( ; bitOffset < 8; bitOffset++ ) {
                     WOZread.shift16 <<= 1;
                     
-                    if (WOZread.shift & 0x80) {
+                    if ( WOZread.valid ) {
                         WOZread.shift = 0;
                     }
                 }
                 WOZread.data = woz_trks[track].data[trackOffset++];
+                trackOffset %= usedBytes;
                 bitOffset = 0;
             }
         }