AsyncStreams: Refcount ZLibAPI

Author: Pagghiu

Libraries/AsyncStreams/Internal/ZLibAPI.h

@@ -119,7 +119,7 @@ struct SC::ZLibAPI
     Error(SC_ZLIB_API_CC* pInflateInit2)(void* strm, int windowBits, const char* version, int stream_size) = nullptr;
 #undef SC_ZLIB_API_CC
     // Handle for dynamic library
-    void* library = nullptr;
-
+    void* library  = nullptr;
+    int   refCount = 0;
     struct Internal;
 };

Libraries/AsyncStreams/Internal/ZLibAPI.inl

@@ -75,7 +75,10 @@ struct SC::ZLibAPI::Internal
 SC::Result SC::ZLibAPI::load(const char* libPath)
 {
     if (library != nullptr)
+    {
+        refCount++;
         return Result(true);
+    }
 #if _WIN32
     const char* zlib_library_path = "zlib1.dll";
 #elif __APPLE__
@@ -112,11 +115,17 @@ SC::Result SC::ZLibAPI::load(const char* libPath)
     SC_TRY(Internal::loadSymbol(*this, pInflateEnd, "inflateEnd"));
     SC_TRY(Internal::loadSymbol(*this, pDeflateInit2, "deflateInit2_"));
     SC_TRY(Internal::loadSymbol(*this, pInflateInit2, "inflateInit2_"));
+    refCount = 1;
     return Result(true);
 }
 
 void SC::ZLibAPI::unload()
 {
+    --refCount;
+    if (refCount > 0)
+    {
+        return;
+    }
     if (library)
     {
 #ifdef _WIN32

Libraries/AsyncStreams/Internal/ZLibStream.inl

@@ -147,6 +147,7 @@ SC::ZLibStream::~ZLibStream()
         zlib.inflateEnd(stream);
         break;
     }
+    zlib.unload();
 }
 
 SC::Result SC::ZLibStream::init(Algorithm wantedAlgorithm)