Fix some memory issues

Author: emmatyping

Modules/_zstd/compressor.c

@@ -304,6 +304,10 @@ _zstd_ZstdCompressor_new(PyTypeObject *type, PyObject *Py_UNUSED(args), PyObject
         goto error;
     }
 
+    self->inited = 0;
+    self->dict = NULL;
+    self->use_multithread = 0;
+
 
     /* Compression context */
     self->cctx = ZSTD_createCCtx();
@@ -322,7 +326,9 @@ _zstd_ZstdCompressor_new(PyTypeObject *type, PyObject *Py_UNUSED(args), PyObject
     return (PyObject*)self;
 
 error:
-    PyObject_GC_UnTrack(self);
+    if (self != NULL) {
+        PyObject_GC_Del(self);
+    }
     return NULL;
 }
 

Modules/_zstd/decompressor.c

@@ -622,6 +622,15 @@ _zstd_ZstdDecompressor_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
         goto error;
     }
 
+    self->inited = 0;
+    self->dict = NULL;
+    self->input_buffer = NULL;
+    self->input_buffer_size = 0;
+    self->in_begin = -1;
+    self->in_end = -1;
+    self->unused_data = NULL;
+    self->eof = 0;
+
     /* needs_input flag */
     self->needs_input = 1;
 
@@ -642,7 +651,9 @@ _zstd_ZstdDecompressor_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
     return (PyObject*)self;
 
 error:
-    PyObject_GC_UnTrack(self);
+    if (self != NULL) {
+        PyObject_GC_Del(self);
+    }
     return NULL;
 }
 

Modules/_zstd/zdict.c

@@ -25,11 +25,15 @@ static PyObject *
 _zstd_ZstdDict_new(PyTypeObject *type, PyObject *Py_UNUSED(args), PyObject *Py_UNUSED(kwargs))
 {
     ZstdDict *self;
-    self = (ZstdDict*)type->tp_alloc(type, 0);
+    self = PyObject_GC_New(ZstdDict, type);
     if (self == NULL) {
         goto error;
     }
 
+    self->dict_content = NULL;
+    self->inited = 0;
+    self->d_dict = NULL;
+
     /* ZSTD_CDict dict */
     self->c_dicts = PyDict_New();
     if (self->c_dicts == NULL) {
@@ -39,7 +43,9 @@ _zstd_ZstdDict_new(PyTypeObject *type, PyObject *Py_UNUSED(args), PyObject *Py_U
     return (PyObject*)self;
 
 error:
-    PyObject_GC_UnTrack(self);
+    if (self != NULL) {
+        PyObject_GC_Del(self);
+    }
     return NULL;
 }