Free our wchar strings after opening our file.

NOAA-ORR-ERD · Mar 13, 2024 · 29cb87f · 29cb87f
1 parent 049444d
commit 29cb87f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/py_gd/py_gd.pyx b/py_gd/py_gd.pyx
@@ -16,6 +16,7 @@ from cython cimport view
 
 from py_gd cimport *
 
+from cpython.mem cimport PyMem_Free
 from libc.stdio cimport FILE, fopen, fclose
 from libc.string cimport memcpy, strlen
 from libc.stdlib cimport malloc, free
@@ -73,12 +74,10 @@ cdef FILE* open_file(file_path) except *:
     Note: On Windows, it uses a wchar, UTC-16 encoded
           On other platforms (Mac and Linux), it assumes utf-8
     """
-    cdef FILE* fp
+    cdef FILE* fp = NULL
 
     file_path = os.fspath(file_path)
 
-    fp = NULL
-
     IF UNAME_SYSNAME == 'Windows':
         #cdef bytes bytes_flag = "wb".encode('utf-16')
         #cdef bytes bytes_filepath = file_path.encode('utf-16').bytes
@@ -87,6 +86,9 @@ cdef FILE* open_file(file_path) except *:
         cdef wchar_t *wchar_filepath = PyUnicode_AsWideCharString(file_path, &length)
 
         fp = _wfopen(wchar_filepath, wchar_flag)
+
+        PyMem_Free(<void *>wchar_filepath)
+        PyMem_Free(<void *>wchar_flag)
     ELSE:
         cdef bytes bytes_flag = "wb".encode('ascii')
         fp = fopen(file_path.encode('utf-8'), bytes_flag)