Updated LZ4F_freeDecompressionContext(), to provide stage hint as result

1 files changed, 3 insertions, 1 deletions

diff --git a/lib/lz4frame.c b/lib/lz4frame.c
index eb38390f..a70dbaf0 100644
--- a/lib/lz4frame.c
+++ b/lib/lz4frame.c
@@ -772,14 +772,16 @@ LZ4F_errorCode_t LZ4F_createDecompressionContext(LZ4F_decompressionContext_t* LZ
 
 LZ4F_errorCode_t LZ4F_freeDecompressionContext(LZ4F_decompressionContext_t LZ4F_decompressionContext)
 {
+    LZ4F_errorCode_t result = LZ4F_OK_NoError;
     LZ4F_dctx_internal_t* dctxPtr = (LZ4F_dctx_internal_t*)LZ4F_decompressionContext;
     if (dctxPtr != NULL)   /* can accept NULL input, like free() */
     {
+      result = (LZ4F_errorCode_t)dctxPtr->dStage;
       FREEMEM(dctxPtr->tmpIn);
       FREEMEM(dctxPtr->tmpOutBuffer);
       FREEMEM(dctxPtr);
     }
-    return LZ4F_OK_NoError;
+    return result;
 }