Fix incorrect do while CFG recovery (fix 97)

6 years ago · 39d88e119a
parent a4032dc000
commit 39d88e119a
1 changed files with 1 additions and 2 deletions
--- a/slither/solc_parsing/declarations/function.py
+++ b/slither/solc_parsing/declarations/function.py
@ -304,10 +304,9 @@ class FunctionSolc(Function):
        node_endDoWhile = self._new_node(NodeType.ENDLOOP, doWhilestatement['src'])

        link_nodes(node, node_startDoWhile)
-        link_nodes(node_startDoWhile, statement)
+        link_nodes(node_startDoWhile, node_condition.sons[0])
        link_nodes(statement, node_condition)
        link_nodes(node_condition, node_endDoWhile)
-
        return node_endDoWhile

    def _parse_variable_definition(self, statement, node):