From f399913a32f4c81a2b6db86fe220451179881a5c Mon Sep 17 00:00:00 2001
From: nhz2 <nhz2@cornell.edu>
Date: Sun, 15 Sep 2024 22:33:37 -0400
Subject: [PATCH] add tests

---
 test/runtests.jl | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/test/runtests.jl b/test/runtests.jl
index a111d9a..9488ec8 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -156,6 +156,41 @@ include("utils.jl")
         @test CodecZstd.find_decompressed_size(v) == CodecZstd.ZSTD_CONTENTSIZE_ERROR
     end
 
+    @testset "use after free doesn't segfault" begin
+        @testset "$(Codec)" for Codec in (ZstdCompressor, ZstdDecompressor)
+            codec = Codec()
+            TranscodingStreams.initialize(codec)
+            TranscodingStreams.finalize(codec)
+            data = [0x00,0x01]
+            GC.@preserve data let m = TranscodingStreams.Memory(pointer(data), length(data))
+                try
+                    TranscodingStreams.expectedsize(codec, m)
+                catch
+                end
+                try
+                    TranscodingStreams.minoutsize(codec, m)
+                catch
+                end
+                try
+                    TranscodingStreams.initialize(codec)
+                catch
+                end
+                try
+                    TranscodingStreams.startproc(codec, :read, TranscodingStreams.Error())
+                catch
+                end
+                try
+                    TranscodingStreams.process(codec, m, m, TranscodingStreams.Error())
+                catch
+                end
+                try
+                    TranscodingStreams.finalize(codec)
+                catch
+                end
+            end
+        end
+    end
+
     include("compress_endOp.jl")
     include("static_only_tests.jl")
 end