add recording flag

diff --git a/deefuzzer/station.py b/deefuzzer/station.py
index a0c19fac28531d812772aefa09098b4f0913d2d1..be6fe8187218b85856a686a8d68545aaf4c80b5f 100644 (file)
--- a/deefuzzer/station.py
+++ b/deefuzzer/station.py
@@ -288,6 +288,7 @@ class Station(Thread):
             self.recorder.open(self.rec_file)
         else:
             try:
+                self.recorder.recording = False
                 self.recorder.close()
             except:
                 pass

diff --git a/deefuzzer/tools/recorder.py b/deefuzzer/tools/recorder.py
index 665632d9d0050c5e7b84e490a950cbf05ae1d159..f7f3d799f23c78ec9f6fab7980a380695eb8baaf 100644 (file)
--- a/deefuzzer/tools/recorder.py
+++ b/deefuzzer/tools/recorder.py
@@ -43,6 +43,7 @@ class Recorder:
 
     def __init__(self, path):
         self.path = path
+        self.recording = True
 
     def open(self, filename):
         self.filename = filename
@@ -50,8 +51,9 @@ class Recorder:
 
     def write(self, chunk):
         try:
-            self.media.write(chunk)
-            self.media.flush()
+            if self.recording:
+                self.media.write(chunk)
+                self.media.flush()
         except:
             pass