fix assert position within ZSTD_compress2()

diff --git a/lib/compress/zstd_compress.c b/lib/compress/zstd_compress.c
index 3d6ed47eb643622508ea2ade3e14ef716e85a73c..f05b5e1072df6b45645039b627315bb516543636 100644 (file)
--- a/lib/compress/zstd_compress.c
+++ b/lib/compress/zstd_compress.c
@@ -4068,11 +4068,11 @@ size_t ZSTD_compress2(ZSTD_CCtx* cctx,
                                         src, srcSize, &iPos,
                                         ZSTD_e_end);
         if (ZSTD_isError(result)) return result;
-        assert(iPos == srcSize);
         if (result != 0) {  /* compression not completed, due to lack of output space */
             assert(oPos == dstCapacity);
             return ERROR(dstSize_tooSmall);
         }
+        assert(iPos == srcSize);   /* all input is expected consumed */
         return oPos;
     }
 }