Display Errors and killed connections in mitmdump.

1 files changed, 3 insertions, 5 deletions

diff --git a/libmproxy/flow.py b/libmproxy/flow.py
index e5f9c35f..60a6ebd2 100644
--- a/libmproxy/flow.py
+++ b/libmproxy/flow.py
@@ -288,6 +288,7 @@ class Flow:
         return False
 
     def kill(self):
+        self.error = proxy.Error(self.request, "Connection killed")
         if self.request and not self.request.acked:
             self.request.ack(None)
         elif self.response and not self.response.acked:
@@ -399,10 +400,6 @@ class State:
         for i in self.flow_list[:]:
             i.accept_intercept()
 
-    def kill_flow(self, f):
-        f.kill()
-        self.delete_flow(f)
-
     def revert(self, f):
         f.revert()
 
@@ -534,7 +531,8 @@ class FlowMaster(controller.Master):
             pb = self.do_server_playback(f)
             if not pb:
                 if self.kill_nonreplay:
-                    self.state.kill_flow(f)
+                    f.kill()
+                    self.handle_error(f.error)
                 else:
                     r.ack()
         return f