handle LZW decompression edge case (tlnagy#148)

* handle LZW decompression edge case * better * add test --------- Co-authored-by: Tamas Nagy <[email protected]>
chrstphrbrns · Jan 22, 2024 · 24605fa · 24605fa
1 parent 6bfae1c
commit 24605fa
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/src/compression.jl b/src/compression.jl
@@ -188,12 +188,14 @@ function lzw_decode!(io, arr)
                     end
                 end
 
-                if table_count == 511
-                    codesize = 10
-                elseif table_count == 1023
-                    codesize = 11
-                elseif table_count == 2047
-                    codesize = 12
+                if out_position < output_size
+                    if table_count == 511
+                        codesize = 10
+                    elseif table_count == 1023
+                        codesize = 11
+                    elseif table_count == 2047
+                        codesize = 12
+                    end
                 end
             end
 

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -308,3 +308,8 @@ end
     encoded = get_example("shapes_lzw_predictor3.tif")
     @test TiffImages.load(original) == TiffImages.load(encoded)
 end
+
+@testset "Issue #148" begin
+    im = get_example("earthlab.tif")
+    @test size(TiffImages.load(im)) == (2400, 2400) # no error
+end