use FIre

Author: joyguoguo

fuzz/modify_fuzz_files.py

@@ -1,5 +1,7 @@
+#!/usr/bin/env python3
 import os
 import ast
+import fire
 
 def add_print_to_testoneinput(file_path):
     with open(file_path, 'r') as f:
@@ -36,13 +38,19 @@ def visit_FunctionDef(self, node):
     # 转回代码
     import astor
     new_content = astor.to_source(new_tree)
-
     return new_content
 
-def main():
-    projects_path = "/home/jiayiguo/FuzzAug/fuzz/oss-fuzz/projects"
-    valid_projects_file = "data/valid_projects.txt"
+def main(
+    projects_path="/home/jiayiguo/FuzzAug/fuzz/oss-fuzz/projects",
+    valid_projects_file="data/valid_projects.txt"
+):
+    """
+    给 fuzz target 的 TestOneInput / TestInput 函数开头插入 print(参数名)
 
+    Args:
+        projects_path (str): OSS-Fuzz 项目的根目录
+        valid_projects_file (str): 包含有效项目名的文件路径
+    """
     with open(valid_projects_file, 'r') as f:
         projects = [line.strip() for line in f if line.strip()]
 
@@ -70,4 +78,4 @@ def main():
                         print(f"Error processing {file_path}: {str(e)}")
 
 if __name__ == "__main__":
-    main()
+    fire.Fire(main)