Intercept source files in linker_interceptor.py (#68)

AFLplusplus · Apr 26, 2024 · 0a3195e · 0a3195e
1 parent a15c29e
commit 0a3195e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/linker_interceptor.py b/linker_interceptor.py
@@ -17,6 +17,7 @@
 
 out_args = []
 shareds = []
+sources = []
 search = []
 rpath = []
 
@@ -69,12 +70,18 @@ def process_args(args):
 process_args(args)
 
 if is_linking_qemu:
+    with open("compile_commands.json", 'r') as f:
+        compile_commands = json.load(f)
+        for entry in compile_commands:
+            sources.append(entry["file"])
+
     with open(OUT, 'w') as f:
         json.dump({
             'cmd': out_args,
             'libs': shareds,
             'search': search,
             'rpath': rpath,
+            'sources': sources,
         }, f, indent=2)
 
 r = subprocess.run([cc] + args)