use PyMem allocators instead of malloc/free

Author: ngoldbaum

stringdtype/stringdtype/src/casts.c

@@ -1171,7 +1171,7 @@ get_cast_spec(const char *name, NPY_CASTING casting,
               NPY_ARRAYMETHOD_FLAGS flags, PyArray_DTypeMeta **dtypes,
               PyType_Slot *slots)
 {
-    PyArrayMethod_Spec *ret = malloc(sizeof(PyArrayMethod_Spec));
+    PyArrayMethod_Spec *ret = PyMem_Malloc(sizeof(PyArrayMethod_Spec));
 
     ret->name = name;
     ret->nin = 1;
@@ -1187,7 +1187,7 @@ get_cast_spec(const char *name, NPY_CASTING casting,
 PyArray_DTypeMeta **
 get_dtypes(PyArray_DTypeMeta *dt1, PyArray_DTypeMeta *dt2)
 {
-    PyArray_DTypeMeta **ret = malloc(2 * sizeof(PyArray_DTypeMeta *));
+    PyArray_DTypeMeta **ret = PyMem_Malloc(2 * sizeof(PyArray_DTypeMeta *));
 
     ret[0] = dt1;
     ret[1] = dt2;
@@ -1297,7 +1297,7 @@ get_casts()
             dt2s_dtypes, dt2s_slots);
 
     PyArrayMethod_Spec **casts =
-            malloc((num_casts + 1) * sizeof(PyArrayMethod_Spec *));
+            PyMem_Malloc((num_casts + 1) * sizeof(PyArrayMethod_Spec *));
 
     int cast_i = 0;
 

stringdtype/stringdtype/src/dtype.c

@@ -788,10 +788,12 @@ init_string_dtype(void)
     StringDType.base.singleton = singleton;
 
     for (int i = 0; StringDType_casts[i] != NULL; i++) {
-        free(StringDType_casts[i]->dtypes);
-        free(StringDType_casts[i]);
+        PyMem_Free(StringDType_casts[i]->dtypes);
+        PyMem_Free(StringDType_casts[i]);
     }
 
+    PyMem_Free(StringDType_casts);
+
     return 0;
 }
 

stringdtype/stringdtype/src/static_string.c

@@ -1,3 +1,5 @@
+#include "Python.h"
+
 #include "static_string.h"
 
 // defined this way so it has an in-memory representation that is distinct
@@ -16,7 +18,7 @@ ssnewlen(const char *init, size_t len, ss *to_init)
         return 0;
     }
 
-    char *ret_buf = (char *)malloc(sizeof(char) * len);
+    char *ret_buf = (char *)PyMem_RawMalloc(sizeof(char) * len);
 
     if (ret_buf == NULL) {
         return -1;
@@ -35,7 +37,7 @@ void
 ssfree(ss *str)
 {
     if (str->buf != NULL && str->buf != EMPTY_STRING.buf) {
-        free(str->buf);
+        PyMem_RawFree(str->buf);
         str->buf = NULL;
     }
     str->len = 0;
@@ -68,7 +70,7 @@ ssnewemptylen(size_t num_bytes, ss *out)
         return 0;
     }
 
-    char *buf = (char *)malloc(sizeof(char) * num_bytes);
+    char *buf = (char *)PyMem_RawMalloc(sizeof(char) * num_bytes);
 
     if (buf == NULL) {
         return -1;