Add upper and lower safety margins

1 files changed, 9 insertions, 0 deletions

diff --git a/src/afl-fuzz.c b/src/afl-fuzz.c
index 803a1acc..08f716fa 100644
--- a/src/afl-fuzz.c
+++ b/src/afl-fuzz.c
@@ -2496,6 +2496,15 @@ int main(int argc, char **argv_orig, char **envp) {
           if ((afl->queue_buf[entry]->exec_us/1000) > max_ms)
             max_ms = afl->queue_buf[entry]->exec_us/1000;
       
+      // Add 20% as a safety margin, capped to exec_tmout given in -t option
+      max_ms *= 1.2;
+      if(max_ms > afl->fsrv.exec_tmout)
+        max_ms = afl->fsrv.exec_tmout;
+      
+      // Ensure that there is a sensible timeout even for very fast binaries
+      if(max_ms < 5)
+        max_ms = 5;
+
       afl->fsrv.exec_tmout = max_ms;
       afl->timeout_given = 1;