Minor patch (flushing log file output at the end of program run)

stamparm · stamparm · commit d4610890caab · 2014-10-10T10:07:17.000+02:00
diff --git a/lib/core/dump.py b/lib/core/dump.py
@@ -73,6 +73,13 @@ def _write(self, data, newline=True, console=True, content_type=None):
 
         kb.dataOutputFlag = True
 
+    def flush(self):
+        if self._outputFP:
+            try:
+                self._outputFP.flush()
+            except IOError:
+                pass
+
     def setOutputFile(self):
         self._outputFile = os.path.join(conf.outputPath, "log")
         try:
diff --git a/sqlmap.py b/sqlmap.py
@@ -156,6 +156,9 @@ def main():
             except KeyboardInterrupt:
                 pass
 
+        if conf.get("dumper"):
+            conf.dumper.flush()
+
         # Reference: http://stackoverflow.com/questions/1635080/terminate-a-multi-thread-python-program
         if conf.get("threads", 0) > 1 or conf.get("dnsServer"):
             os._exit(0)
