update script

script/garble/insert.py

@@ -60,7 +60,7 @@ def process_file_content(path, file_name):
     with open(path, "w", encoding="utf-8") as f_w:
         isReturn = False
         isErrorIf = False
-         = 0
+        return_level = 0
         func_level = 0
         for line in lines:
             #没加{}的if else
@@ -69,15 +69,15 @@ def process_file_content(path, file_name):
             #return后有多行代码
             if "return" in line or "throw" in line and not isReturn:
                 isReturn = True
-                 = 0
+                return_level = 0
 
             if isReturn:
                 if "{" in line:
-                     += 1
+                    return_level += 1
                 elif "}" in line:
-                     -= 1
-                    if  < 0:
-                         = 0
+                    return_level -= 1
+                    if return_level < 0:
+                        return_level = 0
 
 
             if "//interference code" not in line and ("){\n" in line or ") {\n" in line) and func_level == 0:
@@ -88,6 +88,7 @@ def process_file_content(path, file_name):
                 if "//interference code" not in line and "}\n" in line:
                     func_level -=1
 
+
             #需要忽略的行
             if not isReturn and not isErrorIf and \
                     ";\n" in line and \
@@ -102,7 +103,7 @@ def process_file_content(path, file_name):
                     "//" not in line:
 
                 if random.randint(0, 20) % 2 == 0:
-                    line = line.replace(";\n", get_rand_code(func_level>0))
+                    line = line.replace(";\n", get_rand_code(return_level>0))
 
             if ";" in line :
                 isErrorIf = False