Loop parsing: expression -> condition, rather than expression -> begin loop

6 years ago · d7ebe32361
parent bdca73056c
commit d7ebe32361
1 changed files with 3 additions and 3 deletions
--- a/slither/solc_parsing/declarations/function.py
+++ b/slither/solc_parsing/declarations/function.py
@ -191,9 +191,9 @@ class FunctionSolc(Function):

        if loop_expression:
            node_LoopExpression = self._parse_statement(loop_expression, node_body)
-            link_nodes(node_LoopExpression, node_startLoop)
+            link_nodes(node_LoopExpression, node_condition)
        else:
-            link_nodes(node_body, node_startLoop)
+            link_nodes(node_body, node_condition)

        if not condition:
            if not loop_expression:
@ -286,7 +286,7 @@ class FunctionSolc(Function):
        if not hasCondition and not hasLoopExpression:
            link_nodes(node, node_endLoop)

-        link_nodes(node_LoopExpression, node_startLoop)
+        link_nodes(node_LoopExpression, node_condition)

        return node_endLoop