Avoid RecursionError on bad python code

fl0-O · Dec 24, 2020 · 2b3b7c0 · 2b3b7c0
1 parent d677a13
commit 2b3b7c0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/regexploit/bin/regexploit_python_ast.py b/regexploit/bin/regexploit_python_ast.py
@@ -22,14 +22,14 @@ def handle_file(filename: str, output: TextOutput):
         code = f.read()
     try:
         code_ast = ast.parse(code)
+        pnv = PythonNodeVisitor()
+        pnv.visit(code_ast)
     except RecursionError:
         print(f"RecursionError parsing AST for {filename}")
         return
     except SyntaxError:
         print(f"Bad Python3 syntax in {filename}")
         return
-    pnv = PythonNodeVisitor()
-    pnv.visit(code_ast)
     for regex in pnv.patterns:
         first_for_regex = True
         try: