Refactor checking for substring in output

MozillaSecurity · Nov 12, 2024 · 09ca1d1 · 09ca1d1
1 parent abd75e5
commit 09ca1d1
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/lithium/interestingness/outputs.py b/src/lithium/interestingness/outputs.py
@@ -72,11 +72,10 @@ def interesting(
     run_info = timed_run(args.cmd_with_flags, args.timeout, temp_prefix)
 
     if temp_prefix is None:
-        outputs = (run_info.out, run_info.err)
-        for data in outputs:
-            if (args.regex and re.match(args.search, data, flags=re.MULTILINE)) or (
-                args.search.encode("utf-8") in data
-            ):
+        encoded = args.search.encode("utf-8")
+        match = encoded if args.regex else re.escape(encoded)
+        for data in (run_info.out, run_info.err):
+            if re.search(match, data, flags=re.MULTILINE):
                 LOG.info("[Interesting] Match detected!")
                 return True