Fix dereferencing of end iterator in PPMQ decompressor

temisu · Jul 15, 2024 · b34989f · b34989f
1 parent ab9704a
commit b34989f
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/PPMQDecompressor.cpp b/src/PPMQDecompressor.cpp
@@ -267,13 +267,15 @@ void PPMQDecompressor::decompressImpl(Buffer &rawData,const Buffer &previousData
 
 		void scale() noexcept
 		{
-
 			for (auto it=_nodes.begin();it!=_nodes.end();)
 			{
-				auto next=it++;
 				it->freq>>=1U;
-				if (!it->freq) _nodes.erase(it);
-				it=next;
+				if (!it->freq)
+				{
+					it = _nodes.erase(it);
+				} else {
+					it++;
+				}
 			}
 		}