Made processing timeout for sample verification phase configurable in…

… afl-collect (fixes #38)

afl_utils/__init__.py

@@ -17,4 +17,4 @@
 __author__ = 'rc0r <hlt99@blinkenshell.org> # @_rc0r'
 __author_name__ = 'rc0r'
 __author_email__ = 'hlt99@blinkenshell.org'
-__version__ = "1.32a"
+__version__ = "1.33a"

afl_utils/afl_collect.py

@@ -358,6 +358,10 @@ def main(argv):
                         default=False, help="Remove crash samples that have an exploitable classification of \
 'NOT_EXPLOITABLE' or 'PROBABLY_NOT_EXPLOITABLE'. Sample file removal will take place after gdb+exploitable \
 script execution. Has no effect without '-e'.")
+    parser.add_argument("-rt", "--remove-timeout", dest="remove_timeout", default=10,
+                        help="Specifies the maximum processing time in seconds for each sample during verification \
+phase. Samples that cause the target to run longer are marked as timeouts and are removed from the index. Has no \
+effect without '-r'.")
     parser.add_argument("target_cmd", nargs="+", help="Path to the target binary and its command line arguments. \
 Use '@@' to specify crash sample input file position (see afl-fuzz usage).")
 
@@ -410,7 +414,7 @@ def main(argv):
     if args.remove_invalid:
         from afl_utils import afl_vcrash
         invalid_samples, timeout_samples = afl_vcrash.verify_samples(int(args.num_threads), sample_index.inputs(),
-                                                                     args.target_cmd, timeout_secs=10)
+                                                                     args.target_cmd, timeout_secs=args.remove_timeout)
 
         # store invalid samples in db
         if args.gdb_expl_script_file and db_file:

docs/CHANGELOG.md

@@ -1,5 +1,9 @@
 # afl-utils Changelog
 
+Version 1.33a
+
+  - Made processing timeout for sample verification phase configurable in afl-collect.
+
 Version 1.32a
 
   - SQLite db connector improved. Gives a huge afl-collect speed-up