fix the memory leak

python · erlend-aasland · Apr 17, 2023 · Apr 4, 2023 · Apr 6, 2023 · Apr 7, 2023
commit a2017d1ac04ba2777c90722a2c3c5cf80c39723a
@@ -2203,13 +2203,29 @@ static PyModuleDef_Slot winreg_slots[] = {
     {0, NULL}
 };
 
+static winreg_traverse(PyObject* module, visitproc visit, void* arg)
+{
+    winreg_state *state = PyModule_GetState(module);
+    Py_VISIT(state->PyHKEY_Type);
+    return 0;
+}
+
+static winreg_clear(PyObject *module)
+{
+    winreg_state *state = PyModule_GetState(module);
+    Py_CLEAR(state->PyHKEY_Type);
+    return 0;
+}
+
 static struct PyModuleDef winregmodule = {
     .m_base = PyModuleDef_HEAD_INIT,
     .m_name = "winreg",
     .m_doc = module_doc,
     .m_size = sizeof(winreg_state),
     .m_methods = winreg_methods,
     .m_slots = winreg_slots,
+    .m_traverse = winreg_traverse,
+    .m_clear = winreg_clear,
 };
 
 PyMODINIT_FUNC PyInit_winreg(void)