IJ-CR-108265 [java-decompiler] IDEA-198397 add cancelled checks for j…

…ava-decompiler. add 2 more points GitOrigin-RevId: 6aae82cd10be851fadfe9dcd6402d4bc941e1efb
posledniypoet · Jun 15, 2023 · f3cab81 · f3cab81
1 parent 34d476f
commit f3cab81
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/org/jetbrains/java/decompiler/modules/decompiler/sforms/SSAUConstructorSparseEx.java b/src/org/jetbrains/java/decompiler/modules/decompiler/sforms/SSAUConstructorSparseEx.java
@@ -101,7 +101,7 @@ public void splitVariables(RootStatement root, StructMethod mt) {
   }
 
   private void ssaStatements(DirectGraph dgraph, HashSet<String> updated, boolean calcLiveVars) {
-
+    CancellationManager cancellationManager = DecompilerContext.getCancellationManager();
     for (DirectNode node : dgraph.nodes) {
 
       updated.remove(node.id);
@@ -113,6 +113,7 @@ private void ssaStatements(DirectGraph dgraph, HashSet<String> updated, boolean
 
       if (node.exprents != null) {
         for (Exprent expr : node.exprents) {
+          cancellationManager.checkCanceled();
           processExprent(expr, varmaparr, node.statement, calcLiveVars);
         }
       }

diff --git a/src/org/jetbrains/java/decompiler/modules/decompiler/stats/Statement.java b/src/org/jetbrains/java/decompiler/modules/decompiler/stats/Statement.java
@@ -382,6 +382,7 @@ public List<Statement> getReversePostOrderList() {
   }
 
   public List<Statement> getReversePostOrderList(Statement stat) {
+    cancellationManager.checkCanceled();
     List<Statement> res = new ArrayList<>();
 
     addToReversePostOrderListIterative(stat, res);
@@ -711,6 +712,7 @@ public Statement getPost() {
   }
 
   public VBStyleCollection<Statement, Integer> getStats() {
+    cancellationManager.checkCanceled();
     return stats;
   }