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

Use ~bq / ~bs for messages sent from client / server

Browse Source
This commit is contained in:
Shadab Zafar 2016-07-15 14:35:18 +05:30
parent 5f7d61f864
commit 262a420553
2 changed files with 31 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
mitmproxy/filt.py
View File

@ -219,18 +219,14 @@ class FBod(_Rex):
@only(HTTPFlow, TCPFlow)
def __call__(self, f):
# HTTPFlow
if hasattr(f, 'request'):
if isinstance(f, HTTPFlow):
if f.request and f.request.content:
if self.re.search(f.request.get_decoded_content()):
return True
if f.response and f.response.content:
if self.re.search(f.response.get_decoded_content()):
return True
# TCPFlow
elif hasattr(f, 'messages'):
elif isinstance(f, TCPFlow):
for msg in f.messages:
if self.re.search(msg.content):
return True
@ -242,22 +238,32 @@ class FBodRequest(_Rex):
code = "bq"
help = "Request body"
@only(HTTPFlow)
@only(HTTPFlow, TCPFlow)
def __call__(self, f):
if isinstance(f, HTTPFlow):
if f.request and f.request.content:
if self.re.search(f.request.get_decoded_content()):
return True
elif isinstance(f, TCPFlow):
for msg in f.messages:
if msg.from_client and self.re.search(msg.content):
return True
class FBodResponse(_Rex):
code = "bs"
help = "Response body"
@only(HTTPFlow)
@only(HTTPFlow, TCPFlow)
def __call__(self, f):
if isinstance(f, HTTPFlow):
if f.response and f.response.content:
if self.re.search(f.response.get_decoded_content()):
return True
elif isinstance(f, TCPFlow):
for msg in f.messages:
if not msg.from_client and self.re.search(msg.content):
return True
class FMethod(_Rex):

16
test/mitmproxy/test_filt.py
View File

@ -264,15 +264,21 @@ class TestMatchingTCPFlow:
def test_body(self):
f = self.flow()
assert not self.q("~b nonexistent", f)
# Messages sent by client or server
assert self.q("~b hello", f)
assert self.q("~b me", f)
assert not self.q("~b nonexistent", f)
# Request Body
assert not self.q("~bq whatever", f)
# Messages sent by client
assert self.q("~bq hello", f)
assert not self.q("~bq me", f)
assert not self.q("~bq nonexistent", f)
# Response Body
assert not self.q("~bs whatever", f)
# Messages sent by server
assert self.q("~bs me", f)
assert not self.q("~bs hello", f)
assert not self.q("~bs nonexistent", f)
def test_src(self):
f = self.flow()