use flow.id to link websocket + handshake flows

mhils · Mar 10, 2017 · 49e0f23 · 49e0f23
1 parent b1dd86d
commit 49e0f23
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mitmproxy/proxy/protocol/websocket.py b/mitmproxy/proxy/protocol/websocket.py
@@ -140,7 +140,8 @@ def _handle_unknown_frame(self, frame, source_conn, other_conn, is_server):
 
     def __call__(self):
         self.flow = WebSocketFlow(self.client_conn, self.server_conn, self.handshake_flow, self)
-        self.handshake_flow.metadata['websocket_flow'] = self.flow
+        self.flow.metadata['websocket_handshake'] = self.handshake_flow.id
+        self.handshake_flow.metadata['websocket_flow'] = self.flow.id
         self.channel.ask("websocket_start", self.flow)
 
         client = self.client_conn.connection