python · brandtbucher · Apr 19, 2024 · Apr 9, 2024 · Apr 10, 2024 · Apr 11, 2024
@@ -1601,11 +1601,21 @@ static void
 FutureIter_dealloc(futureiterobject *it)
 {
    PyTypeObject *tp = Py_TYPE(it);
-    asyncio_state *state = get_asyncio_state_by_def((PyObject *)it);
+
+    assert(_PyType_HasFeature((PyTypeObject *)tp, Py_TPFLAGS_HEAPTYPE));
+
+    PyHeapTypeObject *ht = (PyHeapTypeObject*)tp;
+    PyObject *module = ht->ht_module;
+    asyncio_state *state = NULL;
+
    PyObject_GC_UnTrack(it);
    tp->tp_clear((PyObject *)it);

-    if (state->fi_freelist_len < FI_FREELIST_MAXLEN) {
+    if (module && _PyModule_GetDef(module) == &_asynciomodule) {
+        state = get_asyncio_state(module);
+    }
+
+    if (state && state->fi_freelist_len < FI_FREELIST_MAXLEN) {
        state->fi_freelist_len++;
        it->future = (FutureObj*) state->fi_freelist;
        state->fi_freelist = it;
-Original file line number
+Diff line change
@@ -1601,11 +1601,21 @@ static void
     FutureIter_dealloc(futureiterobject *it)
     {
         PyTypeObject *tp = Py_TYPE(it);
-        asyncio_state *state = get_asyncio_state_by_def((PyObject *)it);
+        assert(_PyType_HasFeature((PyTypeObject *)tp, Py_TPFLAGS_HEAPTYPE));
                   savannahostrowski marked this conversation as resolved.
              

          
            Show resolved
            Hide resolved
        

                
  
            
  

  
    
  
    
      

              Uh oh!

              
There was an error while loading. Please reload this page.


  
  


  


    
    
      
  
    
    
              
          
              
                  savannahostrowski marked this conversation as resolved.
              

          
            Show resolved
            Hide resolved
        

                
  
            
  

  
    
  
    
      

              Uh oh!

              
There was an error while loading. Please reload this page.
+        PyHeapTypeObject *ht = (PyHeapTypeObject*)tp;
+        PyObject *module = ht->ht_module;
                   savannahostrowski marked this conversation as resolved.
              

          
            Show resolved
            Hide resolved
        

                
  
            
  

  
    
  
    
      

              Uh oh!

              
There was an error while loading. Please reload this page.
+        asyncio_state *state = NULL;
         PyObject_GC_UnTrack(it);
         tp->tp_clear((PyObject *)it);
-        if (state->fi_freelist_len < FI_FREELIST_MAXLEN) {
+        if (module && _PyModule_GetDef(module) == &_asynciomodule) {
                   brandtbucher marked this conversation as resolved.
              

          
            Show resolved
            Hide resolved
        

                
  
            
  

  
    
  
    
      

              Uh oh!

              
There was an error while loading. Please reload this page.
+            state = get_asyncio_state(module);
+        }
+        if (state && state->fi_freelist_len < FI_FREELIST_MAXLEN) {
             state->fi_freelist_len++;
             it->future = (FutureObj*) state->fi_freelist;
             state->fi_freelist = it;
-          Expand Down