From 2162ce1ae349d5a39364cc7e074a5b30ff12785c Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sun, 15 Apr 2012 10:35:19 +1200
Subject: Shut mitmdump down gracefully on SIGTERM.

This is a hack at the moment, but needs must.
---
 libmproxy/dump.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'libmproxy/dump.py')

diff --git a/libmproxy/dump.py b/libmproxy/dump.py
index 253a22d7..a797efbf 100644
--- a/libmproxy/dump.py
+++ b/libmproxy/dump.py
@@ -214,6 +214,11 @@ class DumpMaster(flow.FlowMaster):
             self._process_flow(f)
         return f
 
+    def shutdown(self):
+        if self.o.wfile:
+            self.wfile.fo.close()
+        return flow.FlowMaster.shutdown(self)
+
 # begin nocover
     def run(self):
         if self.o.rfile and not self.o.keepserving:
-- 
cgit v1.2.3