python-ldap
diff --git a/‎Modules/constants.c‎
Lines changed: 38 additions & 0 deletions b/‎Modules/constants.c‎
Lines changed: 38 additions & 0 deletions
diff --git a/‎Modules/ldapmodule.c‎
Lines changed: 3 additions & 0 deletions b/‎Modules/ldapmodule.c‎
Lines changed: 3 additions & 0 deletions
diff --git a/‎Modules/pythonldap.h‎
Lines changed: 5 additions & 0 deletions b/‎Modules/pythonldap.h‎
Lines changed: 5 additions & 0 deletions
@@ -179,6 +179,44 @@ LDAPerror(PyObject *m, LDAP *l)
     return LDAPraise_for_message(m, l, NULL);
 }
 
+/* GC protocol for heap types */
+int
+LDAPMod_traverse(PyObject *m, visitproc visit, void *arg)
+{
+    LDAPModState *state = PyModule_GetState(m);
+    size_t i = 0;
+
+    Py_VISIT(state->ldap_type);
+    Py_VISIT(state->exception_class);
+    for ( ; i < sizeof(state->errobjects)/sizeof(state->errobjects[0]); i++ ) {
+        Py_VISIT(state->errobjects[i]);
+    }
+
+    return 0;
+}
+
+int
+LDAPMod_clear(PyObject *m)
+{
+    LDAPModState *state = PyModule_GetState(m);
+    size_t i = 0;
+
+    Py_CLEAR(state->ldap_type);
+    Py_CLEAR(state->exception_class);
+    for ( ; i < sizeof(state->errobjects)/sizeof(state->errobjects[0]); i++ ) {
+        Py_CLEAR(state->errobjects[i]);
+    }
+
+    return 0;
+}
+
+/* destructor */
+void
+LDAPMod_free(void *m)
+{
+    LDAPMod_clear(m);
+}
+
 /* initialise the module constants */
 
 int
 
@@ -67,6 +67,9 @@ static struct PyModuleDef ldap_moduledef = {
     .m_size = sizeof(LDAPModState),
     .m_methods = ldap_functions,
     .m_slots = ldap_slots,
+    .m_traverse = LDAPMod_traverse,
+    .m_clear = LDAPMod_clear,
+    .m_free = LDAPMod_free,
 };
 struct PyModuleDef *LDAPMod_moduledef;
 
 
@@ -119,6 +119,11 @@ typedef struct LDAPModState {
     PyObject *errobjects[LDAP_ERROR_MAX - LDAP_ERROR_MIN + 1];
 } LDAPModState;
 
+/* *** module level state *** */
+PYLDAP_FUNC(int) LDAPMod_traverse(PyObject *, visitproc, void *);
+PYLDAP_FUNC(int) LDAPMod_clear(PyObject *);
+PYLDAP_FUNC(void) LDAPMod_free(void *);
+
 /* *** ldapcontrol *** */
 PYLDAP_FUNC(void) LDAPControl_List_DEL(LDAPControl **);
 PYLDAP_FUNC(int) LDAPControls_from_object(PyObject *, LDAPControl ***);