Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-22 15:37:45 +00:00
Code Issues Projects Releases Wiki Activity

Change "connection" to the less confusing "client_conn" throughout.

Browse Source
This commit is contained in:
Aldo Cortesi 2011-02-03 14:51:32 +13:00
parent 7924f07971
commit 2cb7429d38
8 changed files with 57 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libmproxy/console.py
View File

@ -47,7 +47,7 @@ def format_keyvals(lst, key="key", val="text", space=5, indent=0):
def format_flow(f, focus, extended=False, padding=3):
if not f.request and not f.response:
txt = [
("title", " Connection from %s..."%(f.connection.address)),
("title", " Connection from %s..."%(f.client_conn.address)),
]
else:
if extended:

54
libmproxy/flow.py
View File

@ -29,14 +29,14 @@ class ReplayThread(threading.Thread):
response = server.read_response()
response.send(self.masterq)
except proxy.ProxyError, v:
err = proxy.Error(self.flow.connection, v.msg)
err = proxy.Error(self.flow.client_conn, v.msg)
err.send(self.masterq)
# end nocover
class Flow:
def __init__(self, connection):
self.connection = connection
def __init__(self, client_conn):
self.client_conn = client_conn
self.request, self.response, self.error = None, None, None
self.intercepting = False
self._backup = None
@ -131,7 +131,7 @@ class Flow:
def backup(self):
if not self._backup:
self._backup = [
self.connection.copy() if self.connection else None,
self.client_conn.copy() if self.client_conn else None,
self.request.copy() if self.request else None,
self.response.copy() if self.response else None,
self.error.copy() if self.error else None,
@ -140,7 +140,7 @@ class Flow:
def revert(self):
if self._backup:
restore = [i.copy() if i else None for i in self._backup]
self.connection, self.request, self.response, self.error = restore
self.client_conn, self.request, self.response, self.error = restore
self._backup = None
def match(self, pattern):
@ -152,7 +152,7 @@ class Flow:
return False
def is_replay(self):
return isinstance(self.connection, ReplayConnection)
return isinstance(self.client_conn, ReplayConnection)
def kill(self):
if self.request and not self.request.acked:
@ -188,13 +188,13 @@ class State:
Start a browser connection.
"""
self.flow_list.insert(0, f)
self.flow_map[f.connection] = f
self.flow_map[f.client_conn] = f
def add_request(self, req):
"""
Add a request to the state. Returns the matching flow.
"""
f = self.flow_map.get(req.connection)
f = self.flow_map.get(req.client_conn)
if not f:
return False
f.request = req
@ -204,7 +204,7 @@ class State:
"""
Add a response to the state. Returns the matching flow.
"""
f = self.flow_map.get(resp.request.connection)
f = self.flow_map.get(resp.request.client_conn)
if not f:
return False
f.response = resp
@ -215,7 +215,7 @@ class State:
Add an error response to the state. Returns the matching flow, or
None if there isn't one.
"""
f = self.flow_map.get(err.connection)
f = self.flow_map.get(err.client_conn)
if not f:
return None
f.error = err
@ -245,26 +245,26 @@ class State:
else:
return tuple(self.flow_list[:])
def get_connection(self, itm):
if isinstance(itm, (proxy.BrowserConnection, ReplayConnection)):
def get_client_conn(self, itm):
if isinstance(itm, (proxy.ClientConnection, ReplayConnection)):
return itm
elif hasattr(itm, "connection"):
return itm.connection
elif hasattr(itm, "client_conn"):
return itm.client_conn
elif hasattr(itm, "request"):
return itm.request.connection
return itm.request.client_conn
def lookup(self, itm):
"""
Checks for matching connection, using a Flow, Replay Connection,
BrowserConnection, Request, Response or Error object. Returns None
Checks for matching client_conn, using a Flow, Replay Connection,
ClientConnection, Request, Response or Error object. Returns None
if not found.
"""
connection = self.get_connection(itm)
return self.flow_map.get(connection)
client_conn = self.get_client_conn(itm)
return self.flow_map.get(client_conn)
def delete_flow(self, f):
if not f.intercepting:
c = self.get_connection(f)
c = self.get_client_conn(f)
if c in self.flow_map:
del self.flow_map[c]
self.flow_list.remove(f)
@ -285,17 +285,17 @@ class State:
def revert(self, f):
"""
Replaces the matching connection object with a ReplayConnection object.
Replaces the matching client_conn object with a ReplayConnection object.
"""
conn = self.get_connection(f)
conn = self.get_client_conn(f)
if conn in self.flow_map:
del self.flow_map[conn]
f.revert()
self.flow_map[f.connection] = f
self.flow_map[f.client_conn] = f
def replay(self, f, masterq):
"""
Replaces the matching connection object with a ReplayConnection object.
Replaces the matching client_conn object with a ReplayConnection object.
Returns None if successful, or error message if not.
"""
@ -304,12 +304,12 @@ class State:
return "Can't replay while intercepting..."
if f.request:
f.backup()
conn = self.get_connection(f)
conn = self.get_client_conn(f)
if conn in self.flow_map:
del self.flow_map[conn]
rp = ReplayConnection()
f.connection = rp
f.request.connection = rp
f.client_conn = rp
f.request.client_conn = rp
if f.request.content:
f.request.headers["content-length"] = [str(len(f.request.content))]
f.response = None

16
libmproxy/proxy.py
View File

@ -84,8 +84,8 @@ def parse_proxy_request(request):
class Request(controller.Msg):
FMT = '%s %s HTTP/1.0\r\n%s\r\n%s'
def __init__(self, connection, host, port, scheme, method, path, headers, content, timestamp=None):
self.connection = connection
def __init__(self, client_conn, host, port, scheme, method, path, headers, content, timestamp=None):
self.client_conn = client_conn
self.host, self.port, self.scheme = host, port, scheme
self.method, self.path, self.headers, self.content = method, path, headers, content
self.timestamp = timestamp or time.time()
@ -221,7 +221,7 @@ class Response(controller.Msg):
return self.FMT%data
class BrowserConnection(controller.Msg):
class ClientConnection(controller.Msg):
def __init__(self, address, port):
self.address, self.port = address, port
controller.Msg.__init__(self)
@ -231,8 +231,8 @@ class BrowserConnection(controller.Msg):
class Error(controller.Msg):
def __init__(self, connection, msg, timestamp=None):
self.connection, self.msg = connection, msg
def __init__(self, client_conn, msg, timestamp=None):
self.client_conn, self.msg = client_conn, msg
self.timestamp = timestamp or time.time()
controller.Msg.__init__(self)
@ -350,7 +350,7 @@ class ProxyHandler(SocketServer.StreamRequestHandler):
def handle(self):
server = None
bc = BrowserConnection(*self.client_address)
bc = ClientConnection(*self.client_address)
bc.send(self.mqueue)
try:
request = self.read_request(bc)
@ -376,7 +376,7 @@ class ProxyHandler(SocketServer.StreamRequestHandler):
server.terminate()
self.finish()
def read_request(self, connection):
def read_request(self, client_conn):
request = self.rfile.readline()
method, scheme, host, port, path = parse_proxy_request(request)
if not host:
@ -412,7 +412,7 @@ class ProxyHandler(SocketServer.StreamRequestHandler):
content = self.rfile.read(int(headers["content-length"][0]))
else:
content = ""
return Request(connection, host, port, scheme, method, path, headers, content)
return Request(client_conn, host, port, scheme, method, path, headers, content)
def send_response(self, response):
self.wfile.write(response.assemble())

12
test/test_console.py
View File

@ -10,7 +10,7 @@ class uState(libpry.AutoTree):
connect -> request -> response
"""
bc = proxy.BrowserConnection("address", 22)
bc = proxy.ClientConnection("address", 22)
c = console.ConsoleState()
f = flow.Flow(bc)
c.add_browserconnect(f)
@ -25,7 +25,7 @@ class uState(libpry.AutoTree):
"""
c = console.ConsoleState()
bc = proxy.BrowserConnection("address", 22)
bc = proxy.ClientConnection("address", 22)
f = flow.Flow(bc)
c.add_browserconnect(f)
assert c.get_focus() == (f, 0)
@ -33,7 +33,7 @@ class uState(libpry.AutoTree):
assert c.get_from_pos(1) == (None, None)
assert c.get_next(0) == (None, None)
bc2 = proxy.BrowserConnection("address", 22)
bc2 = proxy.ClientConnection("address", 22)
f2 = flow.Flow(bc2)
c.add_browserconnect(f2)
assert c.get_focus() == (f, 1)
@ -54,7 +54,7 @@ class uState(libpry.AutoTree):
def _add_request(self, state):
f = utils.tflow()
state.add_browserconnect(f)
q = utils.treq(f.connection)
q = utils.treq(f.client_conn)
state.add_request(q)
return f
@ -67,7 +67,7 @@ class uState(libpry.AutoTree):
c = console.ConsoleState()
f = utils.tflow()
c.add_browserconnect(f)
q = utils.treq(f.connection)
q = utils.treq(f.client_conn)
c.focus = None
assert c.add_request(q)
@ -117,7 +117,7 @@ class uformat_flow(libpry.AutoTree):
assert ('method', '[edited] ') in console.format_flow(f, True)
assert ('method', '[edited] ') in console.format_flow(f, True, True)
f.connection = flow.ReplayConnection()
f.client_conn = flow.ReplayConnection()
assert ('method', '[replay] ') in console.format_flow(f, True)
assert ('method', '[replay] ') in console.format_flow(f, True, True)

2
test/test_filt.py
View File

@ -72,7 +72,7 @@ class uParsing(libpry.AutoTree):
class uMatching(libpry.AutoTree):
def req(self):
conn = proxy.BrowserConnection("one", 2222)
conn = proxy.ClientConnection("one", 2222)
headers = utils.Headers()
headers["header"] = ["qvalue"]
return proxy.Request(

16
test/test_flow.py
View File

@ -66,7 +66,7 @@ class uFlow(libpry.AutoTree):
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.connection = flow.ReplayConnection()
f.client_conn = flow.ReplayConnection()
assert console.format_flow(f, True)
assert console.format_flow(f, False)
@ -115,7 +115,7 @@ class uFlow(libpry.AutoTree):
class uState(libpry.AutoTree):
def test_backup(self):
bc = proxy.BrowserConnection("address", 22)
bc = proxy.ClientConnection("address", 22)
c = flow.State()
f = flow.Flow(bc)
c.add_browserconnect(f)
@ -129,7 +129,7 @@ class uState(libpry.AutoTree):
connect -> request -> response
"""
bc = proxy.BrowserConnection("address", 22)
bc = proxy.ClientConnection("address", 22)
c = flow.State()
f = flow.Flow(bc)
c.add_browserconnect(f)
@ -154,14 +154,14 @@ class uState(libpry.AutoTree):
assert not c.lookup(newresp)
def test_err(self):
bc = proxy.BrowserConnection("address", 22)
bc = proxy.ClientConnection("address", 22)
c = flow.State()
f = flow.Flow(bc)
c.add_browserconnect(f)
e = proxy.Error(bc, "message")
assert c.add_error(e)
e = proxy.Error(proxy.BrowserConnection("address", 22), "message")
e = proxy.Error(proxy.ClientConnection("address", 22), "message")
assert not c.add_error(e)
def test_view(self):
@ -176,7 +176,7 @@ class uState(libpry.AutoTree):
f = utils.tflow()
req = utils.treq(f.connection)
req = utils.treq(f.client_conn)
c.add_browserconnect(f)
c.add_request(req)
assert len(c.view) == 2
@ -188,7 +188,7 @@ class uState(libpry.AutoTree):
def _add_request(self, state):
f = utils.tflow()
state.add_browserconnect(f)
q = utils.treq(f.connection)
q = utils.treq(f.client_conn)
state.add_request(q)
return f
@ -201,7 +201,7 @@ class uState(libpry.AutoTree):
f = utils.tflow()
f.error = proxy.Error(None, "msg")
state.add_browserconnect(f)
q = utils.treq(f.connection)
q = utils.treq(f.client_conn)
state.add_request(q)
def test_kill_flow(self):

8
test/test_proxy.py
View File

@ -221,7 +221,7 @@ class uRequest(libpry.AutoTree):
def test_simple(self):
h = utils.Headers()
h["test"] = ["test"]
c = proxy.BrowserConnection("addr", 2222)
c = proxy.ClientConnection("addr", 2222)
r = proxy.Request(c, "host", 22, "https", "GET", "/", h, "content")
u = r.url()
assert r.set_url(u)
@ -233,7 +233,7 @@ class uRequest(libpry.AutoTree):
def test_getset_state(self):
h = utils.Headers()
h["test"] = ["test"]
c = proxy.BrowserConnection("addr", 2222)
c = proxy.ClientConnection("addr", 2222)
r = proxy.Request(c, "host", 22, "https", "GET", "/", h, "content")
state = r.get_state()
assert proxy.Request.from_state(state) == r
@ -243,7 +243,7 @@ class uResponse(libpry.AutoTree):
def test_simple(self):
h = utils.Headers()
h["test"] = ["test"]
c = proxy.BrowserConnection("addr", 2222)
c = proxy.ClientConnection("addr", 2222)
req = proxy.Request(c, "host", 22, "https", "GET", "/", h, "content")
resp = proxy.Response(req, 200, "HTTP", "msg", h.copy(), "content")
assert resp.short()
@ -252,7 +252,7 @@ class uResponse(libpry.AutoTree):
def test_getset_state(self):
h = utils.Headers()
h["test"] = ["test"]
c = proxy.BrowserConnection("addr", 2222)
c = proxy.ClientConnection("addr", 2222)
r = proxy.Request(c, "host", 22, "https", "GET", "/", h, "content")
req = proxy.Request(c, "host", 22, "https", "GET", "/", h, "content")
resp = proxy.Response(req, 200, "HTTP", "msg", h.copy(), "content")

4
test/utils.py
View File

@ -2,7 +2,7 @@ from libmproxy import proxy, utils, filt, flow
def treq(conn=None):
if not conn:
conn = proxy.BrowserConnection("address", 22)
conn = proxy.ClientConnection("address", 22)
headers = utils.Headers()
headers["header"] = ["qvalue"]
return proxy.Request(conn, "host", 80, "http", "GET", "/path", headers, "content")
@ -17,6 +17,6 @@ def tresp(req=None):
def tflow():
bc = proxy.BrowserConnection("address", 22)
bc = proxy.ClientConnection("address", 22)
return flow.Flow(bc)