pythongh-132983: Fix refleak in zstd dictionary functions (pythongh-134459)

emmatyping · lkollar · commit 29d616c62c6f · 2025-05-26T16:29:44.000+03:00
diff --git a/Modules/_zstd/_zstdmodule.c b/Modules/_zstd/_zstdmodule.c
@@ -251,7 +251,7 @@ _zstd_train_dict_impl(PyObject *module, PyBytesObject *samples_bytes,
                                             &chunk_sizes);
     if (chunks_number < 0)
     {
-        return NULL;
+        goto error;
     }
 
     /* Allocate dict buffer */
@@ -333,7 +333,7 @@ _zstd_finalize_dict_impl(PyObject *module, PyBytesObject *custom_dict_bytes,
                                             &chunk_sizes);
     if (chunks_number < 0)
     {
-        return NULL;
+        goto error;
     }
 
     /* Allocate dict buffer */

Original file line number	Diff line number	Diff line change
`@@ -251,7 +251,7 @@ _zstd_train_dict_impl(PyObject module, PyBytesObject samples_bytes,`
`251`	`251`	`&chunk_sizes);`
`252`	`252`	`if (chunks_number < 0)`
`253`	`253`	`{`
`254`		`- return NULL;`
	`254`	`+ goto error;`
`255`	`255`	`}`
`256`	`256`
`257`	`257`	`/* Allocate dict buffer */`
`@@ -333,7 +333,7 @@ _zstd_finalize_dict_impl(PyObject module, PyBytesObject custom_dict_bytes,`
`333`	`333`	`&chunk_sizes);`
`334`	`334`	`if (chunks_number < 0)`
`335`	`335`	`{`
`336`		`- return NULL;`
	`336`	`+ goto error;`
`337`	`337`	`}`
`338`	`338`
`339`	`339`	`/* Allocate dict buffer */`