From: Guillaume Pellerin <yomguy@parisson.com>
Date: Sat, 27 Jul 2013 10:56:58 +0000 (+0200)
Subject: add recording flag
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=dcba5bf557259fa3d90cdee770fc9baf10031d71;p=deefuzzer.git

add recording flag
---

diff --git a/deefuzzer/station.py b/deefuzzer/station.py
index a0c19fa..be6fe81 100644
--- 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 665632d..f7f3d79 100644
--- 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