mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
deb79a9c5a
We use a loose hash to match incoming requests with recorded flows. At the moment, this hash is over the host, port, scheme, method, path and content of the request. Note that headers are not included here - if we do want to include headers, we would have to do some work to normalize them to remove variations between user agents, header order, etc. etc.
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
from libmproxy import netstring
|
|
from cStringIO import StringIO
|
|
import libpry
|
|
|
|
|
|
class uNetstring(libpry.AutoTree):
|
|
def setUp(self):
|
|
self.test_data = "Netstring module by Will McGugan"
|
|
self.encoded_data = "9:Netstring,6:module,2:by,4:Will,7:McGugan,"
|
|
|
|
def test_header(self):
|
|
tests = [ ("netstring", "9:"),
|
|
("Will McGugan", "12:"),
|
|
("", "0:") ]
|
|
for test, result in tests:
|
|
assert netstring.header(test) == result
|
|
|
|
def test_file_encoder(self):
|
|
file_out = StringIO()
|
|
data = self.test_data.split()
|
|
encoder = netstring.FileEncoder(file_out)
|
|
for s in data:
|
|
encoder.write(s)
|
|
encoded_data = file_out.getvalue()
|
|
assert encoded_data == self.encoded_data
|
|
|
|
def test_decode_file(self):
|
|
data = self.test_data.split()
|
|
for buffer_size in range(1, len(self.encoded_data)):
|
|
file_in = StringIO(self.encoded_data[:])
|
|
decoded_data = list(netstring.decode_file(file_in, buffer_size = buffer_size))
|
|
assert decoded_data == data
|
|
|
|
def test_decoder(self):
|
|
encoded_data = self.encoded_data
|
|
for step in range(1, len(encoded_data)):
|
|
i = 0
|
|
chunks = []
|
|
while i < len(encoded_data):
|
|
chunks.append(encoded_data[i:i+step])
|
|
i += step
|
|
decoder = netstring.Decoder()
|
|
decoded_data = []
|
|
for chunk in chunks:
|
|
for s in decoder.feed(chunk):
|
|
decoded_data.append(s)
|
|
assert decoded_data == self.test_data.split()
|
|
|
|
|
|
|
|
|
|
tests = [
|
|
uNetstring()
|
|
]
|
|
|