Add the taint_output flag

Author: 0xbeefd00d

src/crashd/dwarf/dwarf_source_code.py

@@ -326,8 +326,8 @@ def show_tainted_source(zelos, binary_path, trace, taint_path):
             zelos.config.source_code_path,
         )
 
-    # The remaining is debug-only output
-    return
+    if not zelos.config.taint_output == "terminal":
+        return
 
     # print("generated .zcov file")
     crash_line = None
@@ -339,23 +339,22 @@ def show_tainted_source(zelos, binary_path, trace, taint_path):
         source_path[(file, line)] = v
         if file in files:
             if crash_line is None:
-                new_line = colored(
+                files[file][line] = colored(
                     # f"!0x{k:x}" +
                     files[file][line] + str(v) + "\n",
                     color="red",
                 )
                 crash_line = line
                 changed_file_lines[file].append(line)
             else:
-                new_line = colored(
+                files[file][line] = colored(
                     # f"*0x{k:x}" +
                     files[file][line] + str(v) + "\n",
                     color="white",
                     attrs=["bold"],
                 )
                 changed_file_lines[file].append(line)
 
-            files[file][line] = new_line
     # print("Updated source lines with data flow")
     count = 0
     # Get first trace line that is in the source
@@ -405,10 +404,11 @@ def show_tainted_source(zelos, binary_path, trace, taint_path):
                 i += 1
             if lines_to_print[-1] != "...\n":
                 lines_to_print.append("...\n")
-        # print("".join(lines_to_print))
+        print("".join(lines_to_print))
+
     # print("Changed lines: ")
-    # print(changed_lines)
-    # print("Num lines: ", len(changed_lines))
+    # print(changed_file_lines)
+    # print("Num lines: ", len(changed_file_lines))
 
 
 def process_file(filename, zelos_module_base):

src/crashd/taint/taint.py

@@ -54,6 +54,10 @@
     help="Starts taint tracking when the specified ZML condition is met.",
 )
 
+CommandLineOption(
+    "taint_output", default="", help="Specify taint output location.",
+)
+
 CommandLineOption(
     "source_code_path", default="", help="path to search for source code files"
 )