update script

script/garble/insert.py

@@ -18,10 +18,18 @@ def gen_rand_str():
     return rand_str
 
 
-def get_rand_code(index):
+def get_rand_code():
     code = ""
-    if index == 0:
+    num = random.randint(0, 5);
+    if num == 0:
         code += ";\nString " + gen_rand_str() + "=\"" + gen_rand_str() + "\";\n"
+    elif num == 1:
+        code += ";\nint " + gen_rand_str() + "=\"" + random.randint(0, 100000) + "\";\n"
+    elif num == 3:
+        code += ";\nif(\"" + gen_rand_str() + "\".equals(\"" + gen_rand_str() + "\")){}\n"
+    elif num == 4:
+        index = gen_rand_str()
+        code += ";\nfor(int " + index + "="+random.randint(0,10)+";"+index+"<"+random.randint(11,20)+";"+index+"++){}\n"
     return code
 
 
@@ -34,11 +42,12 @@ def process_file_content(path, file_name):
     with open(path, "w", encoding="utf-8") as f_w:
         for line in lines:
             if ";\n" in line and "return" not in line and "break" not in line and "continue" not in line and "import" not in line and "package" not in line:
-                line = line.replace(";\n", get_rand_code(0))
+                line = line.replace(";\n", get_rand_code())
             f_w.write(line)
         print("done\n")
     return
 
+
 # read configs
 with open('obact.json', "rt") as f:
     json_root = json.load(f)