Merge pull request #4050 from Adenilson/fix_legacy_nullptr01

[fix] Add check on failed allocation in legacy/zstd_v06
2024-05-21 10:30:45 -07:00
parent 78955f5f9d 1872688e0a
commit 0e2ceb2d50
1 changed files with 4 additions and 0 deletions
@@ -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;
 }