Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-27 02:24:18 +00:00
Code Issues Projects Releases Wiki Activity

add simplified connection constructor

Browse Source
This commit is contained in:
Maximilian Hils 2016-02-27 02:17:01 +01:00
parent 679c97ec3e
commit b8b37fdad6
1 changed files with 26 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
mitmproxy/models/connections.py
View File

@ -8,7 +8,6 @@ from .. import stateobject, utils
class ClientConnection(tcp.BaseHandler, stateobject.StateObject): class ClientConnection(tcp.BaseHandler, stateobject.StateObject):
def __init__(self, client_connection, address, server): def __init__(self, client_connection, address, server):
# Eventually, this object is restored from state. We don't have a # Eventually, this object is restored from state. We don't have a
# connection then. # connection then.
@ -65,6 +64,17 @@ class ClientConnection(tcp.BaseHandler, stateobject.StateObject):
f.set_state(state) f.set_state(state)
return f return f
@classmethod
def make_dummy(cls, address):
return cls.from_state(dict(
address=dict(address=address, use_ipv6=False),
clientcert=None,
ssl_established=False,
timestamp_start=None,
timestamp_end=None,
timestamp_ssl_setup=None
))
def convert_to_ssl(self, *args, **kwargs): def convert_to_ssl(self, *args, **kwargs):
super(ClientConnection, self).convert_to_ssl(*args, **kwargs) super(ClientConnection, self).convert_to_ssl(*args, **kwargs)
self.timestamp_ssl_setup = utils.timestamp() self.timestamp_ssl_setup = utils.timestamp()
@ -75,7 +85,6 @@ class ClientConnection(tcp.BaseHandler, stateobject.StateObject):
class ServerConnection(tcp.TCPClient, stateobject.StateObject): class ServerConnection(tcp.TCPClient, stateobject.StateObject):
def __init__(self, address, source_address=None): def __init__(self, address, source_address=None):
tcp.TCPClient.__init__(self, address, source_address) tcp.TCPClient.__init__(self, address, source_address)
@ -123,6 +132,21 @@ class ServerConnection(tcp.TCPClient, stateobject.StateObject):
f.set_state(state) f.set_state(state)
return f return f
@classmethod
def make_dummy(cls, address):
return cls.from_state(dict(
address=dict(address=address, use_ipv6=False),
cert=None,
sni=None,
source_address=dict(address=('', 0), use_ipv6=False),
ssl_established=False,
timestamp_start=None,
timestamp_tcp_setup=None,
timestamp_ssl_setup=None,
timestamp_end=None,
via=None
))
def copy(self): def copy(self):
return copy.copy(self) return copy.copy(self)