def test_connect_tls_context(self):
socks_ep = Mock()
transport = proto_helpers.StringTransport()
def connect(factory):
factory.startFactory()
proto = factory.buildProtocol("addr")
proto.makeConnection(transport)
self.assertEqual(b'\x05\x01\x00', transport.value())
proto.dataReceived(b'\x05\x00')
proto.dataReceived(b'\x05\x00\x00\x01\x00\x00\x00\x00\x00\x00')
return proto
socks_ep.connect = connect
protocol = Mock()
factory = Mock()
factory.buildProtocol = Mock(return_value=protocol)
from OpenSSL import SSL
class CertificateOptions(object):
def getContext(self, *args, **kw):
return SSL.Context(SSL.TLSv1_METHOD)
ep = socks.TorSocksEndpoint(socks_ep, u'meejah.ca', 443, tls=CertificateOptions())
proto = yield ep.connect(factory)
self.assertEqual(proto, protocol)
def test_get_address_endpoint(self):
socks_ep = Mock()
transport = proto_helpers.StringTransport()
delayed_addr = []
def connect(factory):
delayed_addr.append(factory._get_address())
delayed_addr.append(factory._get_address())
factory.startFactory()
proto = factory.buildProtocol("addr")
proto.makeConnection(transport)
self.assertEqual(b'\x05\x01\x00', transport.value())
proto.dataReceived(b'\x05\x00')
proto.dataReceived(b'\x05\x00\x00\x01\x00\x00\x00\x00\x00\x00')
return proto
socks_ep.connect = connect
protocol = Mock()
factory = Mock()
factory.buildProtocol = Mock(return_value=protocol)
ep = socks.TorSocksEndpoint(socks_ep, u'meejah.ca', 443, tls=True)
yield ep.connect(factory)
addr = yield ep._get_address()
self.assertEqual(addr, IPv4Address('TCP', '10.0.0.1', 12345))
self.assertEqual(2, len(delayed_addr))
self.assertTrue(delayed_addr[0] is not delayed_addr[1])
self.assertTrue(all([d.called for d in delayed_addr]))
def test_connect_deferred_proxy_wrong_return(self):
class NotAnEndpoint(object):
"definitely doesn't implement IStreamClientEndpoint"
not_an_endpoint = NotAnEndpoint()
factory = Mock()
ep = socks.TorSocksEndpoint(
socks_endpoint=defer.succeed(not_an_endpoint),
host=u'meejah.ca',
port=443,
)
with self.assertRaises(ValueError) as ctx:
yield ep.connect(factory)
self.assertIn(
"must resolve to an IStreamClientEndpoint provider",
str(ctx.exception),
)
def test_connect_tls(self):
socks_ep = Mock()
transport = proto_helpers.StringTransport()
def connect(factory):
factory.startFactory()
proto = factory.buildProtocol("addr")
proto.makeConnection(transport)
self.assertEqual(b'\x05\x01\x00', transport.value())
proto.dataReceived(b'\x05\x00')
proto.dataReceived(b'\x05\x00\x00\x01\x00\x00\x00\x00\x00\x00')
return proto
socks_ep.connect = connect
protocol = Mock()
factory = Mock()
factory.buildProtocol = Mock(return_value=protocol)
ep = socks.TorSocksEndpoint(socks_ep, u'meejah.ca', 443, tls=True)
proto = yield ep.connect(factory)
self.assertEqual(proto, protocol)
def test_connect_socks_error_unknown(self):
socks_ep = Mock()
transport = proto_helpers.StringTransport()
def connect(factory):
factory.startFactory()
proto = factory.buildProtocol("addr")
proto.makeConnection(transport)
self.assertEqual(b'\x05\x01\x00', transport.value())
proto.dataReceived(b'\x05\x00')
proto.dataReceived(b'\x05\xff\x00\x01\x00\x00\x00\x00')
return proto
socks_ep.connect = connect
protocol = Mock()
factory = Mock()
factory.buildProtocol = Mock(return_value=protocol)
ep = socks.TorSocksEndpoint(socks_ep, u'meejah.ca', 443, tls=True)
with self.assertRaises(Exception) as ctx:
yield ep.connect(factory)
self.assertTrue('Unknown SOCKS error-code' in str(ctx.exception))
def test_connect_socks_illegal_byte(self):
socks_ep = Mock()
transport = proto_helpers.StringTransport()
def connect(factory):
factory.startFactory()
proto = factory.buildProtocol("addr")
proto.makeConnection(transport)
self.assertEqual(b'\x05\x01\x00', transport.value())
proto.dataReceived(b'\x05\x00')
proto.dataReceived(b'\x05\x01\x00\x01\x00\x00\x00\x00')
return proto
socks_ep.connect = connect
protocol = Mock()
factory = Mock()
factory.buildProtocol = Mock(return_value=protocol)
ep = socks.TorSocksEndpoint(socks_ep, u'meejah.ca', 443, tls=True)
with self.assertRaises(Exception) as ctx:
yield ep.connect(factory)
self.assertTrue(isinstance(ctx.exception,
socks.GeneralServerFailureError))
def test_connect_deferred_proxy(self):
socks_ep = Mock()
directlyProvides(socks_ep, IStreamClientEndpoint)
transport = proto_helpers.StringTransport()
def connect(factory):
factory.startFactory()
proto = factory.buildProtocol("addr")
proto.makeConnection(transport)
self.assertEqual(b'\x05\x01\x00', transport.value())
proto.dataReceived(b'\x05\x00')
proto.dataReceived(b'\x05\x00\x00\x01\x00\x00\x00\x00\x00\x00')
return proto
socks_ep.connect = connect
protocol = Mock()
factory = Mock()
factory.buildProtocol = Mock(return_value=protocol)
ep = socks.TorSocksEndpoint(
socks_endpoint=defer.succeed(socks_ep),
host=u'meejah.ca',
port=443,
)
proto = yield ep.connect(factory)
self.assertEqual(proto, protocol)
