AtSpi2: Fix stack overflow caused by LimitPolling.

madewokherd · Oct 7, 2024 · e0a6fbb · e0a6fbb
1 parent c6c75b0
commit e0a6fbb
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/xalia/AtSpi2/AtSpiConnection.cs b/xalia/AtSpi2/AtSpiConnection.cs
@@ -276,7 +276,7 @@ public PollDispoable(AtSpiConnection connection, string peer)
                 this.peer = peer;
             }
 
-            public void Dispose()
+            public void RealDispose()
             {
                 var count = connection.poll_count[peer];
 
@@ -294,6 +294,22 @@ public void Dispose()
                 }
                 connection.poll_count[peer] = count - 1;
             }
+
+            public void Dispose()
+            {
+                if ((connection.poll_known_sources.TryGetValue(peer, out var known_sources) &&
+                     known_sources.Count != 0) ||
+                    (connection.poll_unknown_sources.TryGetValue(peer, out var unknown_sources) &&
+                     unknown_sources.Count != 0))
+                {
+                    // this may resume a task, add some delay so we don't do this recursively
+                    Utils.RunIdle(RealDispose);
+                }
+                else
+                {
+                    connection.poll_count[peer]--;
+                }
+            }
         }
     }
 }