From 65a8992f10af1c8ad327ca9dee970c4bc44586dc Mon Sep 17 00:00:00 2001
From: tomascz <nestorovic@seznam.cz>
Date: Mon, 19 Feb 2024 09:41:00 +0100
Subject: [PATCH] [#57] Support for KryoFlux and associated images (Part 192:
 Fixed buffer overrun in track normalization)

---
 Main/src/Image.h           |  2 +-
 Main/src/Image_TrackRW.cpp | 25 ++++++++++++++-----------
 Main/src/stdafx.h          |  2 +-
 3 files changed, 16 insertions(+), 13 deletions(-)

diff --git a/Main/src/Image.h b/Main/src/Image.h
index a0370f16..b56ac9cf 100644
--- a/Main/src/Image.h
+++ b/Main/src/Image.h
@@ -412,7 +412,7 @@
 				void RemoveConsecutiveBeforeEnd(TLogTime tEndMax);
 			};
 		protected:
-			const PLogTime logTimes; // absolute logical times since the start of recording
+			PLogTime logTimes; // absolute logical times since the start of recording
 			const TDecoderMethod method;
 			bool resetDecoderOnIndex;
 			DWORD iNextTime,nLogTimes;
diff --git a/Main/src/Image_TrackRW.cpp b/Main/src/Image_TrackRW.cpp
index 2ddb7a74..32881529 100644
--- a/Main/src/Image_TrackRW.cpp
+++ b/Main/src/Image_TrackRW.cpp
@@ -1525,10 +1525,11 @@
 		// - ignoring what's before the first Index
 		TLogTime tCurrIndexOrg=RewindToIndex(0);
 		// - normalization
-		const TLogTime tLastIndex=GetIndexTime(nIndexPulses-1);
+		const TLogTime tLastIndex=GetLastIndexTime();
 		const DWORD iModifStart=iNextTime;
 		DWORD iTime=iModifStart;
-		const Utils::CCallocPtr<TLogTime> buffer(GetBufferCapacity());
+		const DWORD nLogTimesMaxNew=std::max( nIndexPulses*mp->nCells, GetBufferCapacity() );
+		const Utils::CCallocPtr<TLogTime> buffer( nLogTimesMaxNew, 0 ); // pessimistic estimation for FM encoding
 		const PLogTime ptModified=buffer.get();
 		for( BYTE nextIndex=1; nextIndex<nIndexPulses; nextIndex++ ){
 			// . resetting inspection conditions
@@ -1541,7 +1542,7 @@
 				// alignment wanted
 				for( ; *this&&logTimes[iNextTime]<tNextIndexOrg; nAlignedCells++ )
 					if (ReadBit())
-						if (iTime<GetBufferCapacity())
+						if (iTime<nLogTimesMaxNew)
 							ptModified[iTime++] = tCurrIndexOrg + nAlignedCells*profile.iwTimeDefault;
 						else
 							return ERROR_INSUFFICIENT_BUFFER; // mustn't overrun the Buffer
@@ -1597,15 +1598,17 @@
 			iTime=iModifRevEnd;
 		}
 		// - copying Modified LogicalTimes to the Track
-		if (nLogTimes+iTime-iNextTime>=GetBufferCapacity())
-			return ERROR_INSUFFICIENT_BUFFER; // mustn't overrun the Buffer
-		::memmove( logTimes+iTime, logTimes+iNextTime, (nLogTimes-iNextTime)*sizeof(TLogTime) );
-		nLogTimes+=iTime-iNextTime;
-		if (const TLogTime dt=indexPulses[nIndexPulses-1]-tLastIndex)
-			for( DWORD i=iTime; i<nLogTimes; logTimes[i++]+=dt );
-		::memcpy( logTimes+iModifStart, ptModified+iModifStart, (iTime-iModifStart)*sizeof(TLogTime) );
+		const DWORD nNewLogTimes=nLogTimes+iTime-iNextTime;
+		CTrackReaderWriter tmp( nLogTimesMaxNew, method, resetDecoderOnIndex );
+			::memcpy( tmp.logTimes, logTimes, iModifStart*sizeof(TLogTime) ); // Times before first Index
+			::memcpy( tmp.logTimes+iModifStart, ptModified+iModifStart, (iTime-iModifStart)*sizeof(TLogTime) ); // Times in full Revolutions
+			::memcpy( tmp.logTimes+iTime, logTimes+iNextTime, (nLogTimes-iNextTime)*sizeof(TLogTime) ); // Times after last Index
+			if (const TLogTime dt=GetLastIndexTime()-tLastIndex)
+				for( DWORD i=iTime; i<nNewLogTimes; tmp.logTimes[i++]+=dt );
+		std::swap( tmp.logTimes, logTimes );
+		nLogTimes=nNewLogTimes;
 		SetCurrentTime(0); // setting valid state
-		// - successfully aligned
+		// - successfully normalized
 		return ERROR_SUCCESS;
 	}
 
diff --git a/Main/src/stdafx.h b/Main/src/stdafx.h
index 77729da8..214c749d 100644
--- a/Main/src/stdafx.h
+++ b/Main/src/stdafx.h
@@ -207,7 +207,7 @@ struct TLogTimeInterval{
 
 #define APP_FULLNAME	_T("Real and Imaginary Disk Editor")
 #define APP_ABBREVIATION "RIDE"
-#define APP_VERSION		"1.7.5 debug special"
+#define APP_VERSION		"1.7.5.1"
 //#define APP_SPECIAL_VER
 #define APP_IDENTIFIER	APP_ABBREVIATION APP_VERSION
 #define APP_CLASSNAME	_T("Afx:tomascz.") _T(APP_ABBREVIATION)