Merge pull request #4050 from Adenilson/fix_legacy_nullptr01

[fix] Add check on failed allocation in legacy/zstd_v06
facebook · May 21, 2024 · 0e2ceb2 · 0e2ceb2
2 parents 78955f5 + 1872688
commit 0e2ceb2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/legacy/zstd_v06.c b/lib/legacy/zstd_v06.c
@@ -3919,6 +3919,10 @@ ZBUFFv06_DCtx* ZBUFFv06_createDCtx(void)
     if (zbd==NULL) return NULL;
     memset(zbd, 0, sizeof(*zbd));
     zbd->zd = ZSTDv06_createDCtx();
+    if (zbd->zd==NULL) {
+        ZBUFFv06_freeDCtx(zbd); /* avoid leaking the context */
+        return NULL;
+    }
     zbd->stage = ZBUFFds_init;
     return zbd;
 }