Rob Glew
8 years ago
parent
76d20774a5
commit
cbc0b4be4c
8 changed files with 980 additions and 431 deletions
@ -1 +1 @@ |
|||||||
__version__ = '0.2.13' |
__version__ = '0.2.14' |
||||||
|
|||||||
@ -0,0 +1,421 @@ |
|||||||
|
import pytest |
||||||
|
import mock |
||||||
|
import random |
||||||
|
import datetime |
||||||
|
import pappyproxy |
||||||
|
import base64 |
||||||
|
import collections |
||||||
|
|
||||||
|
from pappyproxy import http |
||||||
|
from pappyproxy.proxy import ProxyClientFactory, ProxyServerFactory, UpstreamHTTPProxyClient |
||||||
|
from pappyproxy.http import Request, Response |
||||||
|
from pappyproxy.macros import InterceptMacro |
||||||
|
from testutil import mock_deferred, func_deleted, TLSStringTransport, freeze, mock_int_macro, no_tcp |
||||||
|
from twisted.internet import defer |
||||||
|
|
||||||
|
class InterceptMacroTest(InterceptMacro): |
||||||
|
|
||||||
|
def __init__(self, new_req=None, new_rsp=None): |
||||||
|
InterceptMacro.__init__(self) |
||||||
|
|
||||||
|
self.new_req = None |
||||||
|
self.new_rsp = None |
||||||
|
if new_req: |
||||||
|
self.intercept_requests = True |
||||||
|
self.new_req = new_req |
||||||
|
if new_rsp: |
||||||
|
self.intercept_responses = True |
||||||
|
self.new_rsp = new_rsp |
||||||
|
|
||||||
|
def mangle_request(self, request): |
||||||
|
if self.intercept_requests: |
||||||
|
return self.new_req |
||||||
|
else: |
||||||
|
return request |
||||||
|
|
||||||
|
def mangle_response(self, request): |
||||||
|
if self.intercept_responses: |
||||||
|
return self.new_rsp |
||||||
|
else: |
||||||
|
return request.response |
||||||
|
|
||||||
|
class TestProxyConnection(object): |
||||||
|
|
||||||
|
@property |
||||||
|
def client_protocol(self): |
||||||
|
if 'protocol' not in self.conn_info: |
||||||
|
raise Exception('Connection to server not made. Cannot write data as server.') |
||||||
|
return self.conn_info['protocol'] |
||||||
|
|
||||||
|
@property |
||||||
|
def client_factory(self): |
||||||
|
if 'protocol' not in self.conn_info: |
||||||
|
raise Exception('Connection to server not made. Cannot write data as server.') |
||||||
|
return self.conn_info['factory'] |
||||||
|
|
||||||
|
def setUp(self, mocker, int_macros={}, socks_config=None, http_config=None, in_scope=True): |
||||||
|
self.mocker = mocker |
||||||
|
self.conn_info = {} |
||||||
|
|
||||||
|
# Mock config |
||||||
|
self.mock_config = pappyproxy.config.PappyConfig() |
||||||
|
self.mock_config.socks_proxy = socks_config |
||||||
|
self.mock_config.http_proxy = http_config |
||||||
|
self.mock_session = pappyproxy.pappy.PappySession(self.mock_config) |
||||||
|
mocker.patch.object(pappyproxy.pappy, 'session', new=self.mock_session) |
||||||
|
mocker.patch("pappyproxy.proxy.load_certs_from_dir", new=mock_generate_cert) |
||||||
|
|
||||||
|
# Listening server |
||||||
|
## self.server_factory = ProxyServerFactory() |
||||||
|
## self.server_factory.save_all = True |
||||||
|
## self.server_factory.intercepting_macros = int_macros |
||||||
|
## self.server_protocol = self.server_factory.buildProtocol(('127.0.0.1', 0)) |
||||||
|
## self.server_transport = TLSStringTransport() |
||||||
|
## self.server_protocol.makeConnection(self.server_transport) |
||||||
|
|
||||||
|
# Other mocks |
||||||
|
self.req_save = mocker.patch.object(pappyproxy.http.Request, 'async_deep_save', autospec=True, side_effect=mock_req_async_save) |
||||||
|
self.submit_request = mocker.patch('pappyproxy.http.Request.submit_request', |
||||||
|
new=self.gen_mock_submit_request()) |
||||||
|
self.make_proxied_connection = mocker.patch('pappyproxy.proxy.make_proxied_connection') |
||||||
|
self.in_scope = mocker.patch('pappyproxy.context.in_scope').return_value = in_scope |
||||||
|
|
||||||
|
def gen_mock_submit_request(self): |
||||||
|
orig = Request.submit_request |
||||||
|
def f(request, save_request=False, intercepting_macros={}, stream_transport=None): |
||||||
|
return orig(request, save_request=save_request, |
||||||
|
intercepting_macros=intercepting_macros, |
||||||
|
stream_transport=stream_transport, |
||||||
|
_factory_string_transport=True, |
||||||
|
_conn_info=self.conn_info) |
||||||
|
return f |
||||||
|
|
||||||
|
def perform_connect_request(self): |
||||||
|
self.write_as_browser('CONNECT https://www.AAAA.BBBB:443 HTTP/1.1\r\n\r\n') |
||||||
|
assert self.read_as_browser() == 'HTTP/1.1 200 Connection established\r\n\r\n' |
||||||
|
|
||||||
|
def write_as_browser(self, data): |
||||||
|
self.server_protocol.dataReceived(data) |
||||||
|
|
||||||
|
def read_as_browser(self): |
||||||
|
s = self.server_protocol.transport.value() |
||||||
|
self.server_protocol.transport.clear() |
||||||
|
return s |
||||||
|
|
||||||
|
def write_as_server(self, data): |
||||||
|
self.client_protocol.dataReceived(data) |
||||||
|
|
||||||
|
def read_as_server(self): |
||||||
|
s = self.client_protocol.transport.value() |
||||||
|
self.client_protocol.transport.clear() |
||||||
|
return s |
||||||
|
|
||||||
|
|
||||||
|
def mock_req_async_save(req): |
||||||
|
req.reqid = str(random.randint(1,1000000)) |
||||||
|
return mock_deferred() |
||||||
|
|
||||||
|
def mock_mangle_response_side_effect(new_rsp): |
||||||
|
def f(request, mangle_macros): |
||||||
|
request.response = new_rsp |
||||||
|
return mock_deferred(True) |
||||||
|
return f |
||||||
|
|
||||||
|
def mock_generate_cert(cert_dir): |
||||||
|
private_key = ('-----BEGIN PRIVATE KEY-----\n' |
||||||
|
'MIIEvwIBADANBgkqhkiG9w0BAQEFAASCBKkwggSlAgEAAoIBAQDAoClrYUEB7lM0\n' |
||||||
|
'zQaKkXZVG2d1Bu9hV8urpx0gNXMbyZ2m3xb+sKZju/FHPuWenA4KaN5gRUT+oLfv\n' |
||||||
|
'tnF6Ia0jpRNWnX0Fyn/irdg1BWGJn7k7mJ2D0NXZQczn2+xxY05599NfGWqNKCYy\n' |
||||||
|
'jhSwPsUK+sGJqi7aSDdlS97ZTjrQVTTFsC0+kSu4lS5fsWXxqrKLa6Ao8W7abVRO\n' |
||||||
|
'JHazh/cxM4UKpgWU+E6yD4o4ZgHY+SMTVUh/IOM8DuOVyLEWtx4oLNiLMlpWT3qy\n' |
||||||
|
'4IMpOF6VuU6JF2HGV13SoJfhsLXsPRbLVTAnZvJcZwtgDm6NfKapU8W8olkDV1Bf\n' |
||||||
|
'YQEMSNX7AgMBAAECggEBAII0wUrAdrzjaIMsg9tu8FofKBPHGFDok9f4Iov/FUwX\n' |
||||||
|
'QOXnrxeCOTb5d+L89SH9ws/ui0LwD+8+nJcA8DvqP6r0jtnhov0jIMcNVDSi6oeo\n' |
||||||
|
'3AEY7ICJzcQJ4oRn+K+8vPNdPhfuikPYe9l4iSuJgpAlaGWyD/GlFyz12DFz2/Wu\n' |
||||||
|
'NIcqR1ucvezRHn3eGMtvDv2WGaN4ifUc30k8XgSUesmwSI6beb5+hxq7wXfsurnP\n' |
||||||
|
'EUrPY9ts3lfiAgxzTKOuj1VR5hn7cJyLN8jF0mZs4D6eSSHorIddhmaNiCq5ZbMd\n' |
||||||
|
'QdlDiPvnXHT41OoXOb7tDEt7SGoiRh2noCZ1aZiSziECgYEA+tuPPLYWU6JRB6EW\n' |
||||||
|
'PhbcXQbh3vML7eT1q7DOz0jYCojgT2+k7EWSI8T830oQyjbpe3Z86XEgH7UBjUgq\n' |
||||||
|
'27nJ4E6dQDYGbYCKEklOoCGLE7A60i1feIz8otOQRrbQ4jcpibEgscA6gzHmunYf\n' |
||||||
|
'De5euUgYW+Rq2Vmr6/NzUaUgui8CgYEAxJMDwPOGgiLM1cczlaSIU9Obz+cVnwWn\n' |
||||||
|
'nsdKYMto2V3yKLydDfjsgOgzxHOxxy+5L645TPxK6CkiISuhJ93kAFFtx+1sCBCT\n' |
||||||
|
'tVzY5robVAekxA9tlPIxtsn3+/axx3n6HnV0oA/XtxkuOS5JImgEdXqFwJZkerGE\n' |
||||||
|
'waftIU2FCfUCgYEArl8+ErJzlJEIiCgWIPSdGuD00pfZW/TCPCT7rKRy3+fDHBR7\n' |
||||||
|
'7Gxzp/9+0utV/mnrJBH5w/8JmGCmgoF+oRtk01FyBzdGgolN8GYajD6kwPvH917o\n' |
||||||
|
'tRAzcC9lY3IigoxbiEWid0wqoBVoz4XaEkH2gA44OG/vQcQOOEYSi9cfh6sCgYBg\n' |
||||||
|
'KLaOXdJvuIxRCzgNvMW/k+VFh3pJJx//COg2f2qT4mQCT3nYiutOh8hDEoFluc+y\n' |
||||||
|
'Jlz7bvNJrE14wnn8IYxWJ383bMoLC+jlsDyeaW3S5kZQbmehk/SDwTrg86W1udKD\n' |
||||||
|
'sdtSLU3N0LCO4jh+bzm3Ki9hrXALoOkbPoU+ZEhvPQKBgQDf79XQ3RNxZSk+eFyq\n' |
||||||
|
'qD8ytVqxEoD+smPDflXXseVH6o+pNWrF8+A0KqmO8c+8KVzWj/OfULO6UbKd3E+x\n' |
||||||
|
'4JGkWu9yF1lEgtHgibF2ER8zCSIL4ikOEasPCkrKj5SrS4Q+j4u5ha76dIc2CVu1\n' |
||||||
|
'hkX2PQ1xU4ocu06k373sf73A4Q==\n' |
||||||
|
'-----END PRIVATE KEY-----') |
||||||
|
ca_key = ('-----BEGIN CERTIFICATE-----\n' |
||||||
|
'MIIDjzCCAncCFQCjC8r+I4xa7JoGUJYGOTcqDROA0DANBgkqhkiG9w0BAQsFADBg\n' |
||||||
|
'MQswCQYDVQQGEwJVUzERMA8GA1UECBMITWljaGlnYW4xEjAQBgNVBAcTCUFubiBB\n' |
||||||
|
'cmJvcjEUMBIGA1UEChMLUGFwcHkgUHJveHkxFDASBgNVBAMTC1BhcHB5IFByb3h5\n' |
||||||
|
'MB4XDTE1MTEyMDIxMTEzOVoXDTI1MTExNzIxMTEzOVowYDELMAkGA1UEBhMCVVMx\n' |
||||||
|
'ETAPBgNVBAgTCE1pY2hpZ2FuMRIwEAYDVQQHEwlBbm4gQXJib3IxFDASBgNVBAoT\n' |
||||||
|
'C1BhcHB5IFByb3h5MRQwEgYDVQQDEwtQYXBweSBQcm94eTCCASIwDQYJKoZIhvcN\n' |
||||||
|
'AQEBBQADggEPADCCAQoCggEBAMCgKWthQQHuUzTNBoqRdlUbZ3UG72FXy6unHSA1\n' |
||||||
|
'cxvJnabfFv6wpmO78Uc+5Z6cDgpo3mBFRP6gt++2cXohrSOlE1adfQXKf+Kt2DUF\n' |
||||||
|
'YYmfuTuYnYPQ1dlBzOfb7HFjTnn3018Zao0oJjKOFLA+xQr6wYmqLtpIN2VL3tlO\n' |
||||||
|
'OtBVNMWwLT6RK7iVLl+xZfGqsotroCjxbtptVE4kdrOH9zEzhQqmBZT4TrIPijhm\n' |
||||||
|
'Adj5IxNVSH8g4zwO45XIsRa3Higs2IsyWlZPerLggyk4XpW5TokXYcZXXdKgl+Gw\n' |
||||||
|
'tew9FstVMCdm8lxnC2AObo18pqlTxbyiWQNXUF9hAQxI1fsCAwEAAaNFMEMwEgYD\n' |
||||||
|
'VR0TAQH/BAgwBgEB/wIBADAOBgNVHQ8BAf8EBAMCAQYwHQYDVR0OBBYEFNo5o+5e\n' |
||||||
|
'a0sNMlW/75VgGJCv2AcJMA0GCSqGSIb3DQEBCwUAA4IBAQBdJDhxbmoEe27bD8me\n' |
||||||
|
'YTcLGjs/StKkSil7rLbX+tBCwtkm5UEEejBuAhKk2FuAXW8yR1FqKJSZwVCAocBT\n' |
||||||
|
'Bo/+97Ee+h7ywrRFhATEr9D/TbbHKOjCjDzOMl9yLZa2DKErZjbI30ZD6NafWS/X\n' |
||||||
|
'hx5X1cGohHcVVzT4jIgUEU70vvYfNn8CTZm4oJ7qqRe/uQPUYy0rwvbd60oprtGg\n' |
||||||
|
'jNv1H5R4ODHUMBXAI9H7ft9cWrd0fBQjxhoj8pvgJXEZ52flXSqQc7qHLg1wO/zC\n' |
||||||
|
'RUgpTcNAb2qCssBKbj+c1vKEPRUJfw6UYb0s1462rQNc8BgZiKaNbwokFmkAnjUg\n' |
||||||
|
'AvnX\n' |
||||||
|
'-----END CERTIFICATE-----') |
||||||
|
return (ca_key, private_key) |
||||||
|
|
||||||
|
######## |
||||||
|
## Tests |
||||||
|
|
||||||
|
def test_no_tcp(): |
||||||
|
from twisted.internet.endpoints import SSL4ClientEndpoint, TCP4ClientEndpoint |
||||||
|
from txsocksx.client import SOCKS5ClientEndpoint |
||||||
|
from txsocksx.tls import TLSWrapClientEndpoint |
||||||
|
with pytest.raises(NotImplementedError): |
||||||
|
SSL4ClientEndpoint('aasdfasdf.sdfwerqwer') |
||||||
|
with pytest.raises(NotImplementedError): |
||||||
|
TCP4ClientEndpoint('aasdfasdf.sdfwerqwer') |
||||||
|
with pytest.raises(NotImplementedError): |
||||||
|
SOCKS5ClientEndpoint('aasdfasdf.sdfwerqwer') |
||||||
|
with pytest.raises(NotImplementedError): |
||||||
|
TLSWrapClientEndpoint('asdf.2341') |
||||||
|
|
||||||
|
def test_proxy_server_connect(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
proxy.setUp(mocker) |
||||||
|
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:443 HTTP/1.1\r\n\r\n') |
||||||
|
rsp = proxy.read_as_browser() |
||||||
|
print rsp |
||||||
|
assert rsp == 'HTTP/1.1 200 Connection established\r\n\r\n' |
||||||
|
|
||||||
|
def test_proxy_server_forward_basic(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
proxy.setUp(mocker) |
||||||
|
req_contents = ('POST /fooo HTTP/1.1\r\n' |
||||||
|
'Test-Header: foo\r\n' |
||||||
|
'Content-Length: 4\r\n' |
||||||
|
'Host: www.AAAA.BBBB\r\n' |
||||||
|
'\r\n' |
||||||
|
'ABCD') |
||||||
|
rsp_contents = ('HTTP/1.1 200 OK\r\n\r\n') |
||||||
|
proxy.write_as_browser(req_contents) |
||||||
|
assert proxy.read_as_server() == req_contents |
||||||
|
proxy.write_as_server(rsp_contents) |
||||||
|
assert proxy.read_as_browser() == rsp_contents |
||||||
|
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80, False, socks_config=None, use_http_proxy=True) |
||||||
|
assert proxy.req_save.called |
||||||
|
|
||||||
|
def test_proxy_server_forward_basic_ssl(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
proxy.setUp(mocker) |
||||||
|
proxy.perform_connect_request() |
||||||
|
req_contents = ('POST /fooo HTTP/1.1\r\n' |
||||||
|
'Test-Header: foo\r\n' |
||||||
|
'Content-Length: 4\r\n' |
||||||
|
'\r\n' |
||||||
|
'ABCD') |
||||||
|
rsp_contents = ('HTTP/1.1 200 OK\r\n\r\n') |
||||||
|
proxy.write_as_browser(req_contents) |
||||||
|
assert proxy.read_as_server() == req_contents |
||||||
|
proxy.write_as_server(rsp_contents) |
||||||
|
assert proxy.read_as_browser() == rsp_contents |
||||||
|
assert proxy.req_save.called |
||||||
|
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 443, True, socks_config=None, use_http_proxy=True) |
||||||
|
|
||||||
|
def test_proxy_server_connect_uri(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
proxy.setUp(mocker) |
||||||
|
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:443 HTTP/1.1\r\n\r\n') |
||||||
|
proxy.read_as_browser() |
||||||
|
req_contents = ('POST /fooo HTTP/1.1\r\n' |
||||||
|
'Test-Header: foo\r\n' |
||||||
|
'Content-Length: 4\r\n' |
||||||
|
'\r\n' |
||||||
|
'ABCD') |
||||||
|
proxy.write_as_browser(req_contents) |
||||||
|
assert proxy.client_protocol.transport.startTLS.called |
||||||
|
assert proxy.client_factory.request.host == 'www.AAAA.BBBB' |
||||||
|
assert proxy.client_factory.request.port == 443 |
||||||
|
assert proxy.client_factory.request.is_ssl == True |
||||||
|
assert proxy.read_as_server() == req_contents |
||||||
|
assert proxy.client_protocol.transport.startTLS.called |
||||||
|
assert proxy.req_save.called |
||||||
|
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 443, True, socks_config=None, use_http_proxy=True) |
||||||
|
|
||||||
|
def test_proxy_server_connect_uri_alt_port(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
proxy.setUp(mocker) |
||||||
|
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:80085 HTTP/1.1\r\n\r\n') |
||||||
|
proxy.read_as_browser() |
||||||
|
req_contents = ('POST /fooo HTTP/1.1\r\n' |
||||||
|
'Test-Header: foo\r\n' |
||||||
|
'Content-Length: 4\r\n' |
||||||
|
'\r\n' |
||||||
|
'ABCD') |
||||||
|
proxy.write_as_browser(req_contents) |
||||||
|
assert proxy.client_factory.request.host == 'www.AAAA.BBBB' |
||||||
|
assert proxy.client_factory.request.port == 80085 |
||||||
|
assert proxy.client_factory.request.is_ssl == True |
||||||
|
assert proxy.req_save.called |
||||||
|
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80085, True, socks_config=None, use_http_proxy=True) |
||||||
|
|
||||||
|
def test_proxy_server_socks_basic(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
proxy.setUp(mocker, socks_config={'host': 'www.banana.faketld', 'port': 1337}) |
||||||
|
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:80085 HTTP/1.1\r\n\r\n') |
||||||
|
proxy.read_as_browser() |
||||||
|
req_contents = ('POST /fooo HTTP/1.1\r\n' |
||||||
|
'Test-Header: foo\r\n' |
||||||
|
'Content-Length: 4\r\n' |
||||||
|
'\r\n' |
||||||
|
'ABCD') |
||||||
|
proxy.write_as_browser(req_contents) |
||||||
|
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80085, True, |
||||||
|
socks_config={'host':'www.banana.faketld', 'port':1337}, |
||||||
|
use_http_proxy=True) |
||||||
|
|
||||||
|
def test_proxy_server_http_basic(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
proxy.setUp(mocker, http_config={'host': 'www.banana.faketld', 'port': 1337}) |
||||||
|
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:80085 HTTP/1.1\r\n\r\n') |
||||||
|
proxy.read_as_browser() |
||||||
|
req_contents = ('POST /fooo HTTP/1.1\r\n' |
||||||
|
'Test-Header: foo\r\n' |
||||||
|
'Content-Length: 4\r\n' |
||||||
|
'\r\n' |
||||||
|
'ABCD') |
||||||
|
proxy.write_as_browser(req_contents) |
||||||
|
assert proxy.req_save.called |
||||||
|
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80085, True, |
||||||
|
socks_config=None, |
||||||
|
use_http_proxy=True) |
||||||
|
|
||||||
|
def test_proxy_server_360_noscope(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
proxy.setUp(mocker, in_scope=False, socks_config={'host': 'www.banana.faketld', 'port': 1337}) |
||||||
|
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:80085 HTTP/1.1\r\n\r\n') |
||||||
|
proxy.read_as_browser() |
||||||
|
req_contents = ('POST /fooo HTTP/1.1\r\n' |
||||||
|
'Test-Header: foo\r\n' |
||||||
|
'Content-Length: 4\r\n' |
||||||
|
'\r\n' |
||||||
|
'ABCD') |
||||||
|
proxy.write_as_browser(req_contents) |
||||||
|
assert not proxy.req_save.called |
||||||
|
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80085, True, |
||||||
|
socks_config=None, |
||||||
|
use_http_proxy=False) |
||||||
|
|
||||||
|
def test_proxy_server_macro_simple(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
|
||||||
|
new_req_contents = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
||||||
|
new_rsp_contents = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
||||||
|
new_req = Request(new_req_contents) |
||||||
|
new_rsp = Response(new_rsp_contents) |
||||||
|
test_macro = InterceptMacroTest(new_req=new_req, new_rsp=new_rsp) |
||||||
|
proxy.setUp(mocker, int_macros={'test_macro': test_macro}) |
||||||
|
proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
||||||
|
assert proxy.read_as_server() == new_req_contents |
||||||
|
proxy.write_as_server('HTTP/1.1 404 NOT FOUND\r\n\r\n') |
||||||
|
assert proxy.read_as_browser() == new_rsp_contents |
||||||
|
|
||||||
|
def test_proxy_server_macro_multiple(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
|
||||||
|
new_req_contents1 = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
||||||
|
new_rsp_contents1 = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
||||||
|
new_req1 = Request(new_req_contents1) |
||||||
|
new_rsp1 = Response(new_rsp_contents1) |
||||||
|
|
||||||
|
new_req_contents2 = 'GET / HTTP/1.1\r\nMangled: Very very yes\r\n\r\n' |
||||||
|
new_rsp_contents2 = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very very yes\r\n\r\n' |
||||||
|
new_req2 = Request(new_req_contents2) |
||||||
|
new_rsp2 = Response(new_rsp_contents2) |
||||||
|
|
||||||
|
test_macro1 = InterceptMacroTest(new_req=new_req1, new_rsp=new_rsp1) |
||||||
|
test_macro2 = InterceptMacroTest(new_req=new_req2, new_rsp=new_rsp2) |
||||||
|
|
||||||
|
macros = collections.OrderedDict() |
||||||
|
macros['macro1'] = test_macro1 |
||||||
|
macros['macro2'] = test_macro2 |
||||||
|
|
||||||
|
proxy.setUp(mocker, int_macros=macros) |
||||||
|
proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
||||||
|
assert proxy.read_as_server() == new_req_contents2 |
||||||
|
proxy.write_as_server('HTTP/1.1 404 NOT FOUND\r\n\r\n') |
||||||
|
assert proxy.read_as_browser() == new_rsp_contents2 |
||||||
|
|
||||||
|
def test_proxy_server_macro_360_noscope(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
|
||||||
|
new_req_contents = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
||||||
|
new_rsp_contents = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
||||||
|
new_req = Request(new_req_contents) |
||||||
|
new_rsp = Response(new_rsp_contents) |
||||||
|
test_macro = InterceptMacroTest(new_req=new_req, new_rsp=new_rsp) |
||||||
|
proxy.setUp(mocker, int_macros={'test_macro': test_macro}, in_scope=False) |
||||||
|
proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
||||||
|
assert proxy.read_as_server() == 'GET /serious.php HTTP/1.1\r\n\r\n' |
||||||
|
proxy.write_as_server('HTTP/1.1 404 NOT FOUND\r\n\r\n') |
||||||
|
assert proxy.read_as_browser() == 'HTTP/1.1 404 NOT FOUND\r\n\r\n' |
||||||
|
|
||||||
|
def test_proxy_server_stream_simple(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
proxy.setUp(mocker) |
||||||
|
req_contents = ('POST /fooo HTTP/1.1\r\n' |
||||||
|
'Test-Header: foo\r\n' |
||||||
|
'Content-Length: 4\r\n' |
||||||
|
'Host: www.AAAA.BBBB\r\n' |
||||||
|
'\r\n' |
||||||
|
'ABCD') |
||||||
|
rsp_contents = ('HTTP/1.1 200 OK\r\n\r\n') |
||||||
|
proxy.write_as_browser(req_contents) |
||||||
|
assert proxy.read_as_server() == req_contents |
||||||
|
proxy.write_as_server(rsp_contents[:20]) |
||||||
|
assert proxy.read_as_browser() == rsp_contents[:20] |
||||||
|
proxy.write_as_server(rsp_contents[20:]) |
||||||
|
assert proxy.read_as_browser() == rsp_contents[20:] |
||||||
|
|
||||||
|
def test_proxy_server_macro_stream(mocker): |
||||||
|
proxy = TestProxyConnection() |
||||||
|
|
||||||
|
new_req_contents = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
||||||
|
new_rsp_contents = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
||||||
|
new_req = Request(new_req_contents) |
||||||
|
new_rsp = Response(new_rsp_contents) |
||||||
|
test_macro = InterceptMacroTest(new_req=new_req, new_rsp=new_rsp) |
||||||
|
proxy.setUp(mocker, int_macros={'test_macro': test_macro}) |
||||||
|
proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
||||||
|
assert proxy.read_as_server() == new_req_contents |
||||||
|
proxy.write_as_server('HTTP/1.1 404 ') |
||||||
|
assert proxy.read_as_browser() == '' |
||||||
|
proxy.write_as_server('NOT FOUND\r\n\r\n') |
||||||
|
assert proxy.read_as_browser() == new_rsp_contents |
||||||
|
|
||||||
|
# It doesn't stream if out of scope and macros are active, but whatever. |
||||||
|
# def test_proxy_server_macro_stream_360_noscope(mocker): |
||||||
|
# proxy = TestProxyConnection() |
||||||
|
|
||||||
|
# new_req_contents = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
||||||
|
# new_rsp_contents = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
||||||
|
# new_req = Request(new_req_contents) |
||||||
|
# new_rsp = Response(new_rsp_contents) |
||||||
|
# test_macro = InterceptMacroTest(new_req=new_req, new_rsp=new_rsp) |
||||||
|
# proxy.setUp(mocker, int_macros={'test_macro': test_macro}, in_scope=False) |
||||||
|
# proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
||||||
|
# assert proxy.read_as_server() == 'GET /serious.php HTTP/1.1\r\n\r\n' |
||||||
|
# proxy.write_as_server('HTTP/1.1 404 ') |
||||||
|
# assert proxy.read_as_browser() == 'HTTP/1.1 404 ' |
||||||
|
# proxy.write_as_server('NOT FOUND\r\n\r\n') |
||||||
|
# assert proxy.read_as_browser() == 'NOT FOUND\r\n\r\n' |
||||||
@ -1,422 +1,370 @@ |
|||||||
import pytest |
|
||||||
import mock |
import mock |
||||||
import random |
import pytest |
||||||
import datetime |
import twisted.internet.endpoints |
||||||
import pappyproxy |
|
||||||
import base64 |
|
||||||
import collections |
|
||||||
|
|
||||||
from pappyproxy import http |
from pappyproxy import http |
||||||
from pappyproxy.proxy import ProxyClientFactory, ProxyServerFactory, UpstreamHTTPProxyClient |
from pappyproxy.proxy import MaybeTLSProtocol, start_maybe_tls, PassthroughProtocolFactory, make_proxied_connection |
||||||
from pappyproxy.http import Request, Response |
from testutil import mock_deferred, func_deleted, TLSStringTransport, freeze, mock_int_macro, no_tcp, mock_config |
||||||
from pappyproxy.macros import InterceptMacro |
from pappyproxy.util import PappyStringTransport |
||||||
from testutil import mock_deferred, func_deleted, TLSStringTransport, freeze, mock_int_macro, no_tcp |
from twisted.internet import defer, ssl |
||||||
from twisted.internet import defer |
|
||||||
|
############################### |
||||||
class InterceptMacroTest(InterceptMacro): |
## Helper functions and classes |
||||||
|
|
||||||
def __init__(self, new_req=None, new_rsp=None): |
def gen_debug_protocol(mocker): |
||||||
InterceptMacro.__init__(self) |
def fake_start_tls(transport, context): |
||||||
|
transport.protocol = mock.MagicMock() |
||||||
self.new_req = None |
mocker.patch("pappyproxy.util.PappyStringTransport.startTLS", new=fake_start_tls) |
||||||
self.new_rsp = None |
mocker.patch("pappyproxy.proxy.generate_tls_context") |
||||||
if new_req: |
|
||||||
self.intercept_requests = True |
t = PappyStringTransport() |
||||||
self.new_req = new_req |
def server_data_received(s): |
||||||
if new_rsp: |
t.write(s) |
||||||
self.intercept_responses = True |
factory = PassthroughProtocolFactory(server_data_received, None, None) |
||||||
self.new_rsp = new_rsp |
p = factory.buildProtocol(None) |
||||||
|
p.transport = t |
||||||
def mangle_request(self, request): |
t.protocol= p |
||||||
if self.intercept_requests: |
return p, t |
||||||
return self.new_req |
|
||||||
else: |
def mock_protocol_proxy(mocker): |
||||||
return request |
from pappyproxy import proxy |
||||||
|
mock_make_proxied_connection = mocker.patch("pappyproxy.proxy.make_proxied_connection") |
||||||
def mangle_response(self, request): |
p = proxy.ProtocolProxy() |
||||||
if self.intercept_responses: |
p.client_transport = PappyStringTransport() |
||||||
return self.new_rsp |
p.server_transport = PappyStringTransport() |
||||||
else: |
|
||||||
return request.response |
client_protocol, _ = gen_debug_protocol(mocker) |
||||||
|
server_protocol, _ = gen_debug_protocol(mocker) |
||||||
class TestProxyConnection(object): |
|
||||||
|
return p, client_protocol, server_protocol, mock_make_proxied_connection |
||||||
@property |
|
||||||
def client_protocol(self): |
########################## |
||||||
if 'protocol' not in self.conn_info: |
## Tests for ProtocolProxy |
||||||
raise Exception('Connection to server not made. Cannot write data as server.') |
|
||||||
return self.conn_info['protocol'] |
def test_simple(mocker): |
||||||
|
mock_config(mocker) |
||||||
@property |
proxy, _, _, _ = mock_protocol_proxy(mocker) |
||||||
def client_factory(self): |
proxy.send_client_data("foobar") |
||||||
if 'protocol' not in self.conn_info: |
assert proxy.client_buffer == "foobar" |
||||||
raise Exception('Connection to server not made. Cannot write data as server.') |
proxy.send_server_data("barfoo") |
||||||
return self.conn_info['factory'] |
assert proxy.server_buffer == "barfoo" |
||||||
|
|
||||||
def setUp(self, mocker, int_macros={}, socks_config=None, http_config=None, in_scope=True): |
def test_connect(mocker): |
||||||
self.mocker = mocker |
mock_config(mocker) |
||||||
self.conn_info = {} |
proxy, _, _, mock_make_proxied_connection = mock_protocol_proxy(mocker) |
||||||
|
proxy.send_client_data("foobar") |
||||||
# Mock config |
proxy.send_server_data("barfoo") |
||||||
self.mock_config = pappyproxy.config.PappyConfig() |
proxy.connect("fakehost", 1337, False) |
||||||
self.mock_config.socks_proxy = socks_config |
assert len(mock_make_proxied_connection.mock_calls) == 1 |
||||||
self.mock_config.http_proxy = http_config |
callargs = mock_make_proxied_connection.mock_calls[0][1] |
||||||
self.mock_session = pappyproxy.pappy.PappySession(self.mock_config) |
assert callargs[1] == "fakehost" |
||||||
mocker.patch.object(pappyproxy.pappy, 'session', new=self.mock_session) |
assert callargs[2] == 1337 |
||||||
mocker.patch("pappyproxy.proxy.load_certs_from_dir", new=mock_generate_cert) |
assert callargs[3] == False |
||||||
|
|
||||||
# Listening server |
def test_send_before_connect(mocker): |
||||||
self.server_factory = ProxyServerFactory() |
mock_config(mocker) |
||||||
self.server_factory.save_all = True |
proxy, client_protocol, server_protocol, mock_make_proxied_connection = mock_protocol_proxy(mocker) |
||||||
self.server_factory.intercepting_macros = int_macros |
proxy.send_client_data("foobar") |
||||||
|
proxy.send_server_data("barfoo") |
||||||
self.server_protocol = self.server_factory.buildProtocol(('127.0.0.1', 0)) |
|
||||||
self.server_transport = TLSStringTransport() |
proxy.connect("fakehost", 1337, False) |
||||||
self.server_protocol.makeConnection(self.server_transport) |
|
||||||
|
proxy.client_connection_made(client_protocol) |
||||||
# Other mocks |
assert proxy.client_buffer == '' |
||||||
self.req_save = mocker.patch.object(pappyproxy.http.Request, 'async_deep_save', autospec=True, side_effect=mock_req_async_save) |
assert client_protocol.transport.pop_value() == 'foobar' |
||||||
self.submit_request = mocker.patch('pappyproxy.http.Request.submit_request', |
|
||||||
new=self.gen_mock_submit_request()) |
assert proxy.server_buffer == 'barfoo' |
||||||
self.get_endpoint = mocker.patch('pappyproxy.proxy.get_endpoint') |
|
||||||
self.in_scope = mocker.patch('pappyproxy.context.in_scope').return_value = in_scope |
proxy.server_connection_made(server_protocol) |
||||||
|
assert proxy.server_buffer == '' |
||||||
def gen_mock_submit_request(self): |
assert server_protocol.transport.pop_value() == 'barfoo' |
||||||
orig = Request.submit_request |
|
||||||
def f(request, save_request=False, intercepting_macros={}, stream_transport=None): |
def test_send_after_connect(mocker): |
||||||
return orig(request, save_request=save_request, |
mock_config(mocker) |
||||||
intercepting_macros=intercepting_macros, |
proxy, client_protocol, server_protocol, mock_make_proxied_connection = mock_protocol_proxy(mocker) |
||||||
stream_transport=stream_transport, |
|
||||||
_factory_string_transport=True, |
proxy.connect("fakehost", 1337, False) |
||||||
_conn_info=self.conn_info) |
|
||||||
return f |
proxy.client_connection_made(client_protocol) |
||||||
|
proxy.send_client_data("foobar") |
||||||
def perform_connect_request(self): |
assert client_protocol.transport.pop_value() == 'foobar' |
||||||
self.write_as_browser('CONNECT https://www.AAAA.BBBB:443 HTTP/1.1\r\n\r\n') |
|
||||||
assert self.read_as_browser() == 'HTTP/1.1 200 Connection established\r\n\r\n' |
proxy.server_connection_made(server_protocol) |
||||||
|
proxy.send_server_data("barfoo") |
||||||
def write_as_browser(self, data): |
assert server_protocol.transport.pop_value() == 'barfoo' |
||||||
self.server_protocol.dataReceived(data) |
|
||||||
|
def test_start_tls_before_connect(mocker): |
||||||
def read_as_browser(self): |
mock_config(mocker) |
||||||
s = self.server_protocol.transport.value() |
proxy, client_protocol, server_protocol, mock_make_proxied_connection = mock_protocol_proxy(mocker) |
||||||
self.server_protocol.transport.clear() |
client_protocol.transport.startTLS = mock.MagicMock() |
||||||
return s |
server_protocol.transport.startTLS = mock.MagicMock() |
||||||
|
|
||||||
def write_as_server(self, data): |
server_protocol.transport.startTLS.assert_not_called() |
||||||
self.client_protocol.dataReceived(data) |
client_protocol.transport.startTLS.assert_not_called() |
||||||
|
proxy.start_server_tls() |
||||||
def read_as_server(self): |
proxy.start_client_tls("fakehost") |
||||||
s = self.client_protocol.transport.value() |
|
||||||
self.client_protocol.transport.clear() |
server_protocol.transport.startTLS.assert_not_called() |
||||||
return s |
client_protocol.transport.startTLS.assert_not_called() |
||||||
|
proxy.connect("fakehost", 1337, False) |
||||||
|
server_protocol.transport.startTLS.assert_not_called() |
||||||
def mock_req_async_save(req): |
client_protocol.transport.startTLS.assert_not_called() |
||||||
req.reqid = str(random.randint(1,1000000)) |
|
||||||
return mock_deferred() |
proxy.server_connection_made(server_protocol) |
||||||
|
assert len(server_protocol.transport.startTLS.mock_calls) == 1 |
||||||
def mock_mangle_response_side_effect(new_rsp): |
client_protocol.transport.startTLS.assert_not_called() |
||||||
def f(request, mangle_macros): |
|
||||||
request.response = new_rsp |
proxy.client_connection_made(client_protocol) |
||||||
return mock_deferred(True) |
assert len(client_protocol.transport.startTLS.mock_calls) == 1 |
||||||
return f |
|
||||||
|
def test_start_tls_after_connect(mocker): |
||||||
def mock_generate_cert(cert_dir): |
mock_config(mocker) |
||||||
private_key = ('-----BEGIN PRIVATE KEY-----\n' |
proxy, client_protocol, server_protocol, mock_make_proxied_connection = mock_protocol_proxy(mocker) |
||||||
'MIIEvwIBADANBgkqhkiG9w0BAQEFAASCBKkwggSlAgEAAoIBAQDAoClrYUEB7lM0\n' |
client_protocol.transport.startTLS = mock.MagicMock() |
||||||
'zQaKkXZVG2d1Bu9hV8urpx0gNXMbyZ2m3xb+sKZju/FHPuWenA4KaN5gRUT+oLfv\n' |
server_protocol.transport.startTLS = mock.MagicMock() |
||||||
'tnF6Ia0jpRNWnX0Fyn/irdg1BWGJn7k7mJ2D0NXZQczn2+xxY05599NfGWqNKCYy\n' |
|
||||||
'jhSwPsUK+sGJqi7aSDdlS97ZTjrQVTTFsC0+kSu4lS5fsWXxqrKLa6Ao8W7abVRO\n' |
server_protocol.transport.startTLS.assert_not_called() |
||||||
'JHazh/cxM4UKpgWU+E6yD4o4ZgHY+SMTVUh/IOM8DuOVyLEWtx4oLNiLMlpWT3qy\n' |
client_protocol.transport.startTLS.assert_not_called() |
||||||
'4IMpOF6VuU6JF2HGV13SoJfhsLXsPRbLVTAnZvJcZwtgDm6NfKapU8W8olkDV1Bf\n' |
|
||||||
'YQEMSNX7AgMBAAECggEBAII0wUrAdrzjaIMsg9tu8FofKBPHGFDok9f4Iov/FUwX\n' |
proxy.connect("fakehost", 1337, False) |
||||||
'QOXnrxeCOTb5d+L89SH9ws/ui0LwD+8+nJcA8DvqP6r0jtnhov0jIMcNVDSi6oeo\n' |
|
||||||
'3AEY7ICJzcQJ4oRn+K+8vPNdPhfuikPYe9l4iSuJgpAlaGWyD/GlFyz12DFz2/Wu\n' |
server_protocol.transport.startTLS.assert_not_called() |
||||||
'NIcqR1ucvezRHn3eGMtvDv2WGaN4ifUc30k8XgSUesmwSI6beb5+hxq7wXfsurnP\n' |
client_protocol.transport.startTLS.assert_not_called() |
||||||
'EUrPY9ts3lfiAgxzTKOuj1VR5hn7cJyLN8jF0mZs4D6eSSHorIddhmaNiCq5ZbMd\n' |
|
||||||
'QdlDiPvnXHT41OoXOb7tDEt7SGoiRh2noCZ1aZiSziECgYEA+tuPPLYWU6JRB6EW\n' |
proxy.server_connection_made(server_protocol) |
||||||
'PhbcXQbh3vML7eT1q7DOz0jYCojgT2+k7EWSI8T830oQyjbpe3Z86XEgH7UBjUgq\n' |
proxy.start_server_tls() |
||||||
'27nJ4E6dQDYGbYCKEklOoCGLE7A60i1feIz8otOQRrbQ4jcpibEgscA6gzHmunYf\n' |
assert len(server_protocol.transport.startTLS.mock_calls) == 1 |
||||||
'De5euUgYW+Rq2Vmr6/NzUaUgui8CgYEAxJMDwPOGgiLM1cczlaSIU9Obz+cVnwWn\n' |
|
||||||
'nsdKYMto2V3yKLydDfjsgOgzxHOxxy+5L645TPxK6CkiISuhJ93kAFFtx+1sCBCT\n' |
proxy.client_connection_made(client_protocol) |
||||||
'tVzY5robVAekxA9tlPIxtsn3+/axx3n6HnV0oA/XtxkuOS5JImgEdXqFwJZkerGE\n' |
proxy.start_client_tls("fakehost") |
||||||
'waftIU2FCfUCgYEArl8+ErJzlJEIiCgWIPSdGuD00pfZW/TCPCT7rKRy3+fDHBR7\n' |
assert len(client_protocol.transport.startTLS.mock_calls) == 1 |
||||||
'7Gxzp/9+0utV/mnrJBH5w/8JmGCmgoF+oRtk01FyBzdGgolN8GYajD6kwPvH917o\n' |
|
||||||
'tRAzcC9lY3IigoxbiEWid0wqoBVoz4XaEkH2gA44OG/vQcQOOEYSi9cfh6sCgYBg\n' |
def test_start_maybe_tls_before_connect(mocker): |
||||||
'KLaOXdJvuIxRCzgNvMW/k+VFh3pJJx//COg2f2qT4mQCT3nYiutOh8hDEoFluc+y\n' |
mock_config(mocker) |
||||||
'Jlz7bvNJrE14wnn8IYxWJ383bMoLC+jlsDyeaW3S5kZQbmehk/SDwTrg86W1udKD\n' |
proxy, client_protocol, server_protocol, mock_make_proxied_connection = mock_protocol_proxy(mocker) |
||||||
'sdtSLU3N0LCO4jh+bzm3Ki9hrXALoOkbPoU+ZEhvPQKBgQDf79XQ3RNxZSk+eFyq\n' |
mock_maybe_tls = mocker.patch("pappyproxy.proxy.start_maybe_tls") |
||||||
'qD8ytVqxEoD+smPDflXXseVH6o+pNWrF8+A0KqmO8c+8KVzWj/OfULO6UbKd3E+x\n' |
client_protocol.transport.startTLS = mock.MagicMock() |
||||||
'4JGkWu9yF1lEgtHgibF2ER8zCSIL4ikOEasPCkrKj5SrS4Q+j4u5ha76dIc2CVu1\n' |
server_protocol.transport.startTLS = mock.MagicMock() |
||||||
'hkX2PQ1xU4ocu06k373sf73A4Q==\n' |
|
||||||
'-----END PRIVATE KEY-----') |
proxy.start_client_maybe_tls("fakehost") |
||||||
ca_key = ('-----BEGIN CERTIFICATE-----\n' |
|
||||||
'MIIDjzCCAncCFQCjC8r+I4xa7JoGUJYGOTcqDROA0DANBgkqhkiG9w0BAQsFADBg\n' |
client_protocol.transport.startTLS.assert_not_called() |
||||||
'MQswCQYDVQQGEwJVUzERMA8GA1UECBMITWljaGlnYW4xEjAQBgNVBAcTCUFubiBB\n' |
proxy.connect("fakehost", 1337, False) |
||||||
'cmJvcjEUMBIGA1UEChMLUGFwcHkgUHJveHkxFDASBgNVBAMTC1BhcHB5IFByb3h5\n' |
client_protocol.transport.startTLS.assert_not_called() |
||||||
'MB4XDTE1MTEyMDIxMTEzOVoXDTI1MTExNzIxMTEzOVowYDELMAkGA1UEBhMCVVMx\n' |
|
||||||
'ETAPBgNVBAgTCE1pY2hpZ2FuMRIwEAYDVQQHEwlBbm4gQXJib3IxFDASBgNVBAoT\n' |
proxy.client_connection_made(client_protocol) |
||||||
'C1BhcHB5IFByb3h5MRQwEgYDVQQDEwtQYXBweSBQcm94eTCCASIwDQYJKoZIhvcN\n' |
client_protocol.transport.startTLS.assert_not_called() |
||||||
'AQEBBQADggEPADCCAQoCggEBAMCgKWthQQHuUzTNBoqRdlUbZ3UG72FXy6unHSA1\n' |
assert len(mock_maybe_tls.mock_calls) == 1 |
||||||
'cxvJnabfFv6wpmO78Uc+5Z6cDgpo3mBFRP6gt++2cXohrSOlE1adfQXKf+Kt2DUF\n' |
|
||||||
'YYmfuTuYnYPQ1dlBzOfb7HFjTnn3018Zao0oJjKOFLA+xQr6wYmqLtpIN2VL3tlO\n' |
callargs = mock_maybe_tls.mock_calls[0][2] |
||||||
'OtBVNMWwLT6RK7iVLl+xZfGqsotroCjxbtptVE4kdrOH9zEzhQqmBZT4TrIPijhm\n' |
assert callargs['tls_host'] == 'fakehost' |
||||||
'Adj5IxNVSH8g4zwO45XIsRa3Higs2IsyWlZPerLggyk4XpW5TokXYcZXXdKgl+Gw\n' |
assert callargs['start_tls_callback'] == proxy.start_server_tls |
||||||
'tew9FstVMCdm8lxnC2AObo18pqlTxbyiWQNXUF9hAQxI1fsCAwEAAaNFMEMwEgYD\n' |
|
||||||
'VR0TAQH/BAgwBgEB/wIBADAOBgNVHQ8BAf8EBAMCAQYwHQYDVR0OBBYEFNo5o+5e\n' |
def test_start_maybe_tls_before_connect(mocker): |
||||||
'a0sNMlW/75VgGJCv2AcJMA0GCSqGSIb3DQEBCwUAA4IBAQBdJDhxbmoEe27bD8me\n' |
mock_config(mocker) |
||||||
'YTcLGjs/StKkSil7rLbX+tBCwtkm5UEEejBuAhKk2FuAXW8yR1FqKJSZwVCAocBT\n' |
proxy, client_protocol, server_protocol, mock_make_proxied_connection = mock_protocol_proxy(mocker) |
||||||
'Bo/+97Ee+h7ywrRFhATEr9D/TbbHKOjCjDzOMl9yLZa2DKErZjbI30ZD6NafWS/X\n' |
mock_maybe_tls = mocker.patch("pappyproxy.proxy.start_maybe_tls") |
||||||
'hx5X1cGohHcVVzT4jIgUEU70vvYfNn8CTZm4oJ7qqRe/uQPUYy0rwvbd60oprtGg\n' |
client_protocol.transport.startTLS = mock.MagicMock() |
||||||
'jNv1H5R4ODHUMBXAI9H7ft9cWrd0fBQjxhoj8pvgJXEZ52flXSqQc7qHLg1wO/zC\n' |
server_protocol.transport.startTLS = mock.MagicMock() |
||||||
'RUgpTcNAb2qCssBKbj+c1vKEPRUJfw6UYb0s1462rQNc8BgZiKaNbwokFmkAnjUg\n' |
|
||||||
'AvnX\n' |
|
||||||
'-----END CERTIFICATE-----') |
client_protocol.transport.startTLS.assert_not_called() |
||||||
return (ca_key, private_key) |
proxy.connect("fakehost", 1337, False) |
||||||
|
client_protocol.transport.startTLS.assert_not_called() |
||||||
######## |
|
||||||
## Tests |
proxy.client_connection_made(client_protocol) |
||||||
|
proxy.start_client_maybe_tls("fakehost") |
||||||
def test_no_tcp(): |
client_protocol.transport.startTLS.assert_not_called() |
||||||
from twisted.internet.endpoints import SSL4ClientEndpoint, TCP4ClientEndpoint |
assert len(mock_maybe_tls.mock_calls) == 1 |
||||||
from txsocksx.client import SOCKS5ClientEndpoint |
|
||||||
from txsocksx.tls import TLSWrapClientEndpoint |
callargs = mock_maybe_tls.mock_calls[0][2] |
||||||
with pytest.raises(NotImplementedError): |
assert callargs['tls_host'] == 'fakehost' |
||||||
SSL4ClientEndpoint('aasdfasdf.sdfwerqwer') |
assert callargs['start_tls_callback'] == proxy.start_server_tls |
||||||
with pytest.raises(NotImplementedError): |
|
||||||
TCP4ClientEndpoint('aasdfasdf.sdfwerqwer') |
|
||||||
with pytest.raises(NotImplementedError): |
############################# |
||||||
SOCKS5ClientEndpoint('aasdfasdf.sdfwerqwer') |
## Tests for MaybeTLSProtocol |
||||||
with pytest.raises(NotImplementedError): |
def test_maybe_tls_plaintext(mocker): |
||||||
TLSWrapClientEndpoint('asdf.2341') |
mock_config(mocker) |
||||||
|
tls_callback = mock.MagicMock() |
||||||
def test_proxy_server_connect(mocker): |
p, t = gen_debug_protocol(mocker) |
||||||
proxy = TestProxyConnection() |
start_maybe_tls(p.transport, 'www.foo.faketld') |
||||||
proxy.setUp(mocker) |
p.dataReceived("Hello world!") |
||||||
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:443 HTTP/1.1\r\n\r\n') |
assert p.transport.pop_value() == "Hello world!" |
||||||
rsp = proxy.read_as_browser() |
|
||||||
print rsp |
def test_maybe_tls_use_tls(mocker): |
||||||
assert rsp == 'HTTP/1.1 200 Connection established\r\n\r\n' |
mock_config(mocker) |
||||||
|
tls_callback = mock.MagicMock() |
||||||
def test_proxy_server_forward_basic(mocker): |
p, t = gen_debug_protocol(mocker) |
||||||
proxy = TestProxyConnection() |
start_maybe_tls(p.transport, 'www.foo.faketld') |
||||||
proxy.setUp(mocker) |
maybe_tls_prot = t.protocol |
||||||
req_contents = ('POST /fooo HTTP/1.1\r\n' |
assert isinstance(maybe_tls_prot, MaybeTLSProtocol) |
||||||
'Test-Header: foo\r\n' |
assert maybe_tls_prot.state == MaybeTLSProtocol.STATE_DECIDING |
||||||
'Content-Length: 4\r\n' |
t.protocol.dataReceived("\x16") |
||||||
'Host: www.AAAA.BBBB\r\n' |
assert not isinstance(t.protocol, MaybeTLSProtocol) |
||||||
'\r\n' |
assert maybe_tls_prot.state == MaybeTLSProtocol.STATE_PASSTHROUGH |
||||||
'ABCD') |
|
||||||
rsp_contents = ('HTTP/1.1 200 OK\r\n\r\n') |
#################################### |
||||||
proxy.write_as_browser(req_contents) |
## Tests for make_proxied_connection |
||||||
assert proxy.read_as_server() == req_contents |
def test_mpc_simple(mocker): |
||||||
proxy.write_as_server(rsp_contents) |
mock_config(mocker) |
||||||
assert proxy.read_as_browser() == rsp_contents |
endpoint_instance = mock.MagicMock() |
||||||
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80, False, socks_config=None, use_http_proxy=True) |
endpoint = mocker.patch('twisted.internet.endpoints.TCP4ClientEndpoint', return_value=endpoint_instance) |
||||||
assert proxy.req_save.called |
|
||||||
|
make_proxied_connection('fakefactory', 'fakehost', 1337, False) |
||||||
def test_proxy_server_forward_basic_ssl(mocker): |
|
||||||
proxy = TestProxyConnection() |
endpointcalls = endpoint.mock_calls[0] |
||||||
proxy.setUp(mocker) |
assert endpointcalls[1][1] == 'fakehost' |
||||||
proxy.perform_connect_request() |
assert endpointcalls[1][2] == 1337 |
||||||
req_contents = ('POST /fooo HTTP/1.1\r\n' |
|
||||||
'Test-Header: foo\r\n' |
connectcall = endpoint_instance.connect |
||||||
'Content-Length: 4\r\n' |
assert len(connectcall.mock_calls) == 1 |
||||||
'\r\n' |
|
||||||
'ABCD') |
def test_mpc_ssl(mocker): |
||||||
rsp_contents = ('HTTP/1.1 200 OK\r\n\r\n') |
mock_config(mocker) |
||||||
proxy.write_as_browser(req_contents) |
endpoint_instance = mock.MagicMock() |
||||||
assert proxy.read_as_server() == req_contents |
endpoint = mocker.patch('twisted.internet.endpoints.SSL4ClientEndpoint', return_value=endpoint_instance) |
||||||
proxy.write_as_server(rsp_contents) |
|
||||||
assert proxy.read_as_browser() == rsp_contents |
make_proxied_connection('fakefactory', 'fakehost', 1337, True) |
||||||
assert proxy.req_save.called |
|
||||||
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 443, True, socks_config=None, use_http_proxy=True) |
endpointcalls = endpoint.mock_calls[0] |
||||||
|
assert endpointcalls[1][1] == 'fakehost' |
||||||
def test_proxy_server_connect_uri(mocker): |
assert endpointcalls[1][2] == 1337 |
||||||
proxy = TestProxyConnection() |
assert isinstance(endpointcalls[1][3], ssl.ClientContextFactory) |
||||||
proxy.setUp(mocker) |
|
||||||
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:443 HTTP/1.1\r\n\r\n') |
connectcall = endpoint_instance.connect |
||||||
proxy.read_as_browser() |
assert len(connectcall.mock_calls) == 1 |
||||||
req_contents = ('POST /fooo HTTP/1.1\r\n' |
|
||||||
'Test-Header: foo\r\n' |
def test_mpc_socks(mocker): |
||||||
'Content-Length: 4\r\n' |
mock_config(mocker) |
||||||
'\r\n' |
tcp_endpoint_instance = mock.MagicMock() |
||||||
'ABCD') |
socks_endpoint_instance = mock.MagicMock() |
||||||
proxy.write_as_browser(req_contents) |
tcp_endpoint = mocker.patch('twisted.internet.endpoints.TCP4ClientEndpoint', return_value=tcp_endpoint_instance) |
||||||
assert proxy.client_protocol.transport.startTLS.called |
socks_endpoint = mocker.patch('txsocksx.client.SOCKS5ClientEndpoint', return_value=socks_endpoint_instance) |
||||||
assert proxy.client_factory.request.host == 'www.AAAA.BBBB' |
|
||||||
assert proxy.client_factory.request.port == 443 |
target_host = 'fakehost' |
||||||
assert proxy.client_factory.request.is_ssl == True |
target_port = 1337 |
||||||
assert proxy.read_as_server() == req_contents |
|
||||||
assert proxy.client_protocol.transport.startTLS.called |
socks_host = 'fakesockshost' |
||||||
assert proxy.req_save.called |
socks_port = 1447 |
||||||
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 443, True, socks_config=None, use_http_proxy=True) |
socks_config = {'host':socks_host, 'port':socks_port} |
||||||
|
|
||||||
def test_proxy_server_connect_uri_alt_port(mocker): |
make_proxied_connection('fakefactory', target_host, target_port, False, socks_config=socks_config) |
||||||
proxy = TestProxyConnection() |
|
||||||
proxy.setUp(mocker) |
tcp_ep_calls = tcp_endpoint.mock_calls[0] |
||||||
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:80085 HTTP/1.1\r\n\r\n') |
assert tcp_ep_calls[1][1] == socks_host |
||||||
proxy.read_as_browser() |
assert tcp_ep_calls[1][2] == socks_port |
||||||
req_contents = ('POST /fooo HTTP/1.1\r\n' |
|
||||||
'Test-Header: foo\r\n' |
socks_ep_calls = socks_endpoint.mock_calls[0] |
||||||
'Content-Length: 4\r\n' |
assert socks_ep_calls[1][0] == target_host |
||||||
'\r\n' |
assert socks_ep_calls[1][1] == target_port |
||||||
'ABCD') |
assert socks_ep_calls[1][2] is tcp_endpoint_instance |
||||||
proxy.write_as_browser(req_contents) |
assert socks_ep_calls[2]['methods'] == {'anonymous': ()} |
||||||
assert proxy.client_factory.request.host == 'www.AAAA.BBBB' |
|
||||||
assert proxy.client_factory.request.port == 80085 |
connectcall = socks_endpoint_instance.connect |
||||||
assert proxy.client_factory.request.is_ssl == True |
assert len(connectcall.mock_calls) == 1 |
||||||
assert proxy.req_save.called |
|
||||||
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80085, True, socks_config=None, use_http_proxy=True) |
def test_mpc_socks_creds(mocker): |
||||||
|
mock_config(mocker) |
||||||
def test_proxy_server_socks_basic(mocker): |
tcp_endpoint_instance = mock.MagicMock() |
||||||
proxy = TestProxyConnection() |
socks_endpoint_instance = mock.MagicMock() |
||||||
proxy.setUp(mocker, socks_config={'host': 'www.banana.faketld', 'port': 1337}) |
tcp_endpoint = mocker.patch('twisted.internet.endpoints.TCP4ClientEndpoint', return_value=tcp_endpoint_instance) |
||||||
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:80085 HTTP/1.1\r\n\r\n') |
socks_endpoint = mocker.patch('txsocksx.client.SOCKS5ClientEndpoint', return_value=socks_endpoint_instance) |
||||||
proxy.read_as_browser() |
|
||||||
req_contents = ('POST /fooo HTTP/1.1\r\n' |
target_host = 'fakehost' |
||||||
'Test-Header: foo\r\n' |
target_port = 1337 |
||||||
'Content-Length: 4\r\n' |
|
||||||
'\r\n' |
socks_host = 'fakesockshost' |
||||||
'ABCD') |
socks_port = 1447 |
||||||
proxy.write_as_browser(req_contents) |
socks_user = 'username' |
||||||
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80085, True, |
socks_password = 'password' |
||||||
socks_config={'host':'www.banana.faketld', 'port':1337}, |
socks_config = {'host':socks_host, 'port':socks_port, |
||||||
use_http_proxy=True) |
'username':socks_user, 'password':socks_password} |
||||||
|
|
||||||
def test_proxy_server_http_basic(mocker): |
make_proxied_connection('fakefactory', target_host, target_port, False, socks_config=socks_config) |
||||||
proxy = TestProxyConnection() |
|
||||||
proxy.setUp(mocker, http_config={'host': 'www.banana.faketld', 'port': 1337}) |
tcp_ep_calls = tcp_endpoint.mock_calls[0] |
||||||
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:80085 HTTP/1.1\r\n\r\n') |
assert tcp_ep_calls[1][1] == socks_host |
||||||
proxy.read_as_browser() |
assert tcp_ep_calls[1][2] == socks_port |
||||||
req_contents = ('POST /fooo HTTP/1.1\r\n' |
|
||||||
'Test-Header: foo\r\n' |
socks_ep_calls = socks_endpoint.mock_calls[0] |
||||||
'Content-Length: 4\r\n' |
assert socks_ep_calls[1][0] == target_host |
||||||
'\r\n' |
assert socks_ep_calls[1][1] == target_port |
||||||
'ABCD') |
assert socks_ep_calls[1][2] is tcp_endpoint_instance |
||||||
proxy.write_as_browser(req_contents) |
assert socks_ep_calls[2]['methods'] == {'login': (socks_user, socks_password), 'anonymous': ()} |
||||||
assert proxy.req_save.called |
|
||||||
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80085, True, |
connectcall = socks_endpoint_instance.connect |
||||||
socks_config=None, |
assert len(connectcall.mock_calls) == 1 |
||||||
use_http_proxy=True) |
|
||||||
|
def test_mpc_socks_ssl(mocker): |
||||||
def test_proxy_server_360_noscope(mocker): |
mock_config(mocker) |
||||||
proxy = TestProxyConnection() |
tcp_endpoint_instance = mock.MagicMock() |
||||||
proxy.setUp(mocker, in_scope=False, socks_config={'host': 'www.banana.faketld', 'port': 1337}) |
socks_endpoint_instance = mock.MagicMock() |
||||||
proxy.write_as_browser('CONNECT https://www.AAAA.BBBB:80085 HTTP/1.1\r\n\r\n') |
wrapper_instance = mock.MagicMock() |
||||||
proxy.read_as_browser() |
tcp_endpoint = mocker.patch('twisted.internet.endpoints.TCP4ClientEndpoint', return_value=tcp_endpoint_instance) |
||||||
req_contents = ('POST /fooo HTTP/1.1\r\n' |
socks_endpoint = mocker.patch('txsocksx.client.SOCKS5ClientEndpoint', return_value=socks_endpoint_instance) |
||||||
'Test-Header: foo\r\n' |
wrapper_endpoint = mocker.patch('txsocksx.tls.TLSWrapClientEndpoint', return_value=wrapper_instance) |
||||||
'Content-Length: 4\r\n' |
|
||||||
'\r\n' |
target_host = 'fakehost' |
||||||
'ABCD') |
target_port = 1337 |
||||||
proxy.write_as_browser(req_contents) |
|
||||||
assert not proxy.req_save.called |
socks_host = 'fakesockshost' |
||||||
proxy.get_endpoint.assert_called_with('www.AAAA.BBBB', 80085, True, |
socks_port = 1447 |
||||||
socks_config=None, |
socks_config = {'host':socks_host, 'port':socks_port} |
||||||
use_http_proxy=False) |
|
||||||
|
make_proxied_connection('fakefactory', target_host, target_port, True, socks_config=socks_config) |
||||||
def test_proxy_server_macro_simple(mocker): |
|
||||||
proxy = TestProxyConnection() |
tcp_ep_calls = tcp_endpoint.mock_calls[0] |
||||||
|
assert tcp_ep_calls[1][1] == socks_host |
||||||
new_req_contents = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
assert tcp_ep_calls[1][2] == socks_port |
||||||
new_rsp_contents = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
|
||||||
new_req = Request(new_req_contents) |
socks_ep_calls = socks_endpoint.mock_calls[0] |
||||||
new_rsp = Response(new_rsp_contents) |
assert socks_ep_calls[1][0] == target_host |
||||||
test_macro = InterceptMacroTest(new_req=new_req, new_rsp=new_rsp) |
assert socks_ep_calls[1][1] == target_port |
||||||
proxy.setUp(mocker, int_macros={'test_macro': test_macro}) |
assert socks_ep_calls[1][2] is tcp_endpoint_instance |
||||||
proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
assert socks_ep_calls[2]['methods'] == {'anonymous': ()} |
||||||
assert proxy.read_as_server() == new_req_contents |
|
||||||
proxy.write_as_server('HTTP/1.1 404 NOT FOUND\r\n\r\n') |
wrapper_calls = wrapper_endpoint.mock_calls[0] |
||||||
assert proxy.read_as_browser() == new_rsp_contents |
assert isinstance(wrapper_calls[1][0], ssl.ClientContextFactory) |
||||||
|
assert wrapper_calls[1][1] is socks_endpoint_instance |
||||||
def test_proxy_server_macro_multiple(mocker): |
|
||||||
proxy = TestProxyConnection() |
connectcall = wrapper_instance.connect |
||||||
|
assert len(connectcall.mock_calls) == 1 |
||||||
new_req_contents1 = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
|
||||||
new_rsp_contents1 = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
def test_mpc_socks_ssl_creds(mocker): |
||||||
new_req1 = Request(new_req_contents1) |
mock_config(mocker) |
||||||
new_rsp1 = Response(new_rsp_contents1) |
tcp_endpoint_instance = mock.MagicMock() |
||||||
|
socks_endpoint_instance = mock.MagicMock() |
||||||
new_req_contents2 = 'GET / HTTP/1.1\r\nMangled: Very very yes\r\n\r\n' |
wrapper_instance = mock.MagicMock() |
||||||
new_rsp_contents2 = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very very yes\r\n\r\n' |
tcp_endpoint = mocker.patch('twisted.internet.endpoints.TCP4ClientEndpoint', return_value=tcp_endpoint_instance) |
||||||
new_req2 = Request(new_req_contents2) |
socks_endpoint = mocker.patch('txsocksx.client.SOCKS5ClientEndpoint', return_value=socks_endpoint_instance) |
||||||
new_rsp2 = Response(new_rsp_contents2) |
wrapper_endpoint = mocker.patch('txsocksx.tls.TLSWrapClientEndpoint', return_value=wrapper_instance) |
||||||
|
|
||||||
test_macro1 = InterceptMacroTest(new_req=new_req1, new_rsp=new_rsp1) |
target_host = 'fakehost' |
||||||
test_macro2 = InterceptMacroTest(new_req=new_req2, new_rsp=new_rsp2) |
target_port = 1337 |
||||||
|
|
||||||
macros = collections.OrderedDict() |
socks_host = 'fakesockshost' |
||||||
macros['macro1'] = test_macro1 |
socks_port = 1447 |
||||||
macros['macro2'] = test_macro2 |
socks_user = 'username' |
||||||
|
socks_password = 'password' |
||||||
proxy.setUp(mocker, int_macros=macros) |
socks_config = {'host':socks_host, 'port':socks_port, |
||||||
proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
'username':socks_user, 'password':socks_password} |
||||||
assert proxy.read_as_server() == new_req_contents2 |
|
||||||
proxy.write_as_server('HTTP/1.1 404 NOT FOUND\r\n\r\n') |
make_proxied_connection('fakefactory', target_host, target_port, True, socks_config=socks_config) |
||||||
assert proxy.read_as_browser() == new_rsp_contents2 |
|
||||||
|
tcp_ep_calls = tcp_endpoint.mock_calls[0] |
||||||
def test_proxy_server_macro_360_noscope(mocker): |
assert tcp_ep_calls[1][1] == socks_host |
||||||
proxy = TestProxyConnection() |
assert tcp_ep_calls[1][2] == socks_port |
||||||
|
|
||||||
new_req_contents = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
socks_ep_calls = socks_endpoint.mock_calls[0] |
||||||
new_rsp_contents = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
assert socks_ep_calls[1][0] == target_host |
||||||
new_req = Request(new_req_contents) |
assert socks_ep_calls[1][1] == target_port |
||||||
new_rsp = Response(new_rsp_contents) |
assert socks_ep_calls[1][2] is tcp_endpoint_instance |
||||||
test_macro = InterceptMacroTest(new_req=new_req, new_rsp=new_rsp) |
assert socks_ep_calls[2]['methods'] == {'login': (socks_user, socks_password), 'anonymous': ()} |
||||||
proxy.setUp(mocker, int_macros={'test_macro': test_macro}, in_scope=False) |
|
||||||
proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
wrapper_calls = wrapper_endpoint.mock_calls[0] |
||||||
assert proxy.read_as_server() == 'GET /serious.php HTTP/1.1\r\n\r\n' |
assert isinstance(wrapper_calls[1][0], ssl.ClientContextFactory) |
||||||
proxy.write_as_server('HTTP/1.1 404 NOT FOUND\r\n\r\n') |
assert wrapper_calls[1][1] is socks_endpoint_instance |
||||||
assert proxy.read_as_browser() == 'HTTP/1.1 404 NOT FOUND\r\n\r\n' |
|
||||||
|
connectcall = wrapper_instance.connect |
||||||
def test_proxy_server_stream_simple(mocker): |
assert len(connectcall.mock_calls) == 1 |
||||||
proxy = TestProxyConnection() |
|
||||||
proxy.setUp(mocker) |
|
||||||
req_contents = ('POST /fooo HTTP/1.1\r\n' |
|
||||||
'Test-Header: foo\r\n' |
|
||||||
'Content-Length: 4\r\n' |
|
||||||
'Host: www.AAAA.BBBB\r\n' |
|
||||||
'\r\n' |
|
||||||
'ABCD') |
|
||||||
rsp_contents = ('HTTP/1.1 200 OK\r\n\r\n') |
|
||||||
proxy.write_as_browser(req_contents) |
|
||||||
assert proxy.read_as_server() == req_contents |
|
||||||
proxy.write_as_server(rsp_contents[:20]) |
|
||||||
assert proxy.read_as_browser() == rsp_contents[:20] |
|
||||||
proxy.write_as_server(rsp_contents[20:]) |
|
||||||
assert proxy.read_as_browser() == rsp_contents[20:] |
|
||||||
|
|
||||||
def test_proxy_server_macro_stream(mocker): |
|
||||||
proxy = TestProxyConnection() |
|
||||||
|
|
||||||
new_req_contents = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
|
||||||
new_rsp_contents = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
|
||||||
new_req = Request(new_req_contents) |
|
||||||
new_rsp = Response(new_rsp_contents) |
|
||||||
test_macro = InterceptMacroTest(new_req=new_req, new_rsp=new_rsp) |
|
||||||
proxy.setUp(mocker, int_macros={'test_macro': test_macro}) |
|
||||||
proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
|
||||||
assert proxy.read_as_server() == new_req_contents |
|
||||||
proxy.write_as_server('HTTP/1.1 404 ') |
|
||||||
assert proxy.read_as_browser() == '' |
|
||||||
proxy.write_as_server('NOT FOUND\r\n\r\n') |
|
||||||
assert proxy.read_as_browser() == new_rsp_contents |
|
||||||
|
|
||||||
# It doesn't stream if out of scope and macros are active, but whatever. |
|
||||||
# def test_proxy_server_macro_stream_360_noscope(mocker): |
|
||||||
# proxy = TestProxyConnection() |
|
||||||
|
|
||||||
# new_req_contents = 'GET / HTTP/1.1\r\nMangled: Very yes\r\n\r\n' |
|
||||||
# new_rsp_contents = 'HTTP/1.1 200 OKILIE DOKILIE\r\nMangled: Very yes\r\n\r\n' |
|
||||||
# new_req = Request(new_req_contents) |
|
||||||
# new_rsp = Response(new_rsp_contents) |
|
||||||
# test_macro = InterceptMacroTest(new_req=new_req, new_rsp=new_rsp) |
|
||||||
# proxy.setUp(mocker, int_macros={'test_macro': test_macro}, in_scope=False) |
|
||||||
# proxy.write_as_browser('GET /serious.php HTTP/1.1\r\n\r\n') |
|
||||||
# assert proxy.read_as_server() == 'GET /serious.php HTTP/1.1\r\n\r\n' |
|
||||||
# proxy.write_as_server('HTTP/1.1 404 ') |
|
||||||
# assert proxy.read_as_browser() == 'HTTP/1.1 404 ' |
|
||||||
# proxy.write_as_server('NOT FOUND\r\n\r\n') |
|
||||||
# assert proxy.read_as_browser() == 'NOT FOUND\r\n\r\n' |
|
||||||
|
|||||||
Loading…
Reference in new issue