def test_stream_changed_with_detach(self):
"Change a stream-id mid-stream, but with a DETACHED message"
self.circuits[123] = FakeCircuit(123)
self.circuits[456] = FakeCircuit(456)
listener = Listener([('new', {
'target_host': 'www.yahoo.com',
'target_port': 80
}), ('attach', {}),
('detach', {
'kwargs':
dict(reason='END', remote_reason='MISC')
}), ('attach', {})])
stream = Stream(self)
stream.listen(listener)
stream.update(
"999 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER"
.split())
stream.update("999 SENTCONNECT 123 1.2.3.4:80".split())
self.assertEqual(len(self.circuits[123].streams), 1)
self.assertEqual(self.circuits[123].streams[0], stream)
stream.update(
"999 DETACHED 123 1.2.3.4:80 REASON=END REMOTE_REASON=MISC".split(
))
self.assertEqual(len(self.circuits[123].streams), 0)
stream.update("999 SENTCONNECT 456 1.2.3.4:80 SOURCE=EXIT".split())
self.assertEqual(len(self.circuits[456].streams), 1)
self.assertEqual(self.circuits[456].streams[0], stream)
def test_ipv6(self):
listener = Listener([('new', {'target_host': '::1',
'target_port': 80})])
stream = Stream(self)
stream.listen(listener)
stream.update("1234 NEW 0 ::1:80 SOURCE_ADDR=127.0.0.1:57349 PURPOSE=USER".split())
def test_stream_changed_with_detach(self):
"Change a stream-id mid-stream, but with a DETACHED message"
self.circuits[123] = FakeCircuit(123)
self.circuits[456] = FakeCircuit(456)
listener = Listener(
[
('new', {'target_host': 'www.yahoo.com', 'target_port': 80}),
('attach', {}),
('detach', {'kwargs': dict(reason='END', remote_reason='MISC')}),
('attach', {})
]
)
stream = Stream(self)
stream.listen(listener)
stream.update("999 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER".split())
stream.update("999 SENTCONNECT 123 1.2.3.4:80".split())
self.assertEqual(len(self.circuits[123].streams), 1)
self.assertEqual(self.circuits[123].streams[0], stream)
stream.update("999 DETACHED 123 1.2.3.4:80 REASON=END REMOTE_REASON=MISC".split())
self.assertEqual(len(self.circuits[123].streams), 0)
stream.update("999 SENTCONNECT 456 1.2.3.4:80 SOURCE=EXIT".split())
self.assertEqual(len(self.circuits[456].streams), 1)
self.assertEqual(self.circuits[456].streams[0], stream)
def test_listener_new(self):
listener = Listener([('new', {'target_port': 9001})])
stream = Stream(self)
stream.listen(listener)
stream.update(
"1 NEW 0 94.23.164.42.$43ED8310EB968746970896E8835C2F1991E50B69.exit:9001 SOURCE_ADDR=(Tor_internal):0 PURPOSE=DIR_FETCH"
.split())
def test_listen_unlisten(self):
self.circuits[186] = FakeCircuit(186)
listener = Listener([])
stream = Stream(self)
stream.listen(listener)
stream.listen(listener)
self.assertEqual(len(stream.listeners), 1)
stream.unlisten(listener)
self.assertEqual(len(stream.listeners), 0)
def test_ipv6_source(self):
listener = Listener(
[
('new', {'source_addr': maybe_ip_addr('::1'),
'source_port': 12345})
]
)
stream = Stream(self)
stream.listen(listener)
stream.update("1234 NEW 0 127.0.0.1:80 SOURCE_ADDR=::1:12345 PURPOSE=USER".split())
def test_ipv6(self):
listener = Listener([('new', {
'target_host': '::1',
'target_port': 80
})])
stream = Stream(self)
stream.listen(listener)
stream.update(
"1234 NEW 0 ::1:80 SOURCE_ADDR=127.0.0.1:57349 PURPOSE=USER".split(
))
def test_listen_unlisten(self):
self.circuits[186] = FakeCircuit(186)
listener = Listener([])
stream = Stream(self)
stream.listen(listener)
stream.listen(listener)
self.assertEqual(len(stream.listeners), 1)
stream.unlisten(listener)
self.assertEqual(len(stream.listeners), 0)
def test_ipv6_source(self):
listener = Listener([('new', {
'source_addr': maybe_ip_addr('::1'),
'source_port': 12345
})])
stream = Stream(self)
stream.listen(listener)
stream.update(
"1234 NEW 0 127.0.0.1:80 SOURCE_ADDR=::1:12345 PURPOSE=USER".split(
))
def test_listener_fail(self):
listener = Listener(
[
('new', {'target_host': 'www.yahoo.com', 'target_port': 80}),
('attach', {'target_addr': maybe_ip_addr('1.2.3.4')}),
('failed', {'kwargs': dict(REASON='TIMEOUT', REMOTE_REASON='DESTROYED')})
]
)
stream = Stream(self)
stream.listen(listener)
stream.update("316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER".split())
self.circuits[186] = FakeCircuit(186)
stream.update("316 REMAP 186 1.2.3.4:80 SOURCE=EXIT".split())
stream.update("316 FAILED 0 1.2.3.4:80 REASON=TIMEOUT REMOTE_REASON=DESTROYED".split())
def test_listener_attach_no_remap(self):
"Attachment is via SENTCONNECT on .onion addresses (for example)"
self.circuits[186] = FakeCircuit(186)
listener = Listener([('new', {'target_host': 'www.yahoo.com',
'target_port': 80}),
('attach', {})])
stream = Stream(self)
stream.listen(listener)
stream.update("316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER".split())
stream.update("316 SENTCONNECT 186 1.2.3.4:80 SOURCE=EXIT".split())
self.assertEqual(self.circuits[186].streams[0], stream)
def test_listener_attach(self):
self.circuits[186] = FakeCircuit(186)
listener = Listener(
[
('new', {'target_host': 'www.yahoo.com', 'target_port': 80}),
('attach', {'target_addr': maybe_ip_addr('1.2.3.4')})
]
)
stream = Stream(self)
stream.listen(listener)
stream.update("316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER".split())
stream.update("316 REMAP 186 1.2.3.4:80 SOURCE=EXIT".split())
self.assertEqual(self.circuits[186].streams[0], stream)
def test_listener_close(self):
self.circuits[186] = FakeCircuit(186)
listener = Listener(
[
('new', {'target_host': 'www.yahoo.com', 'target_port': 80}),
('attach', {'target_addr': maybe_ip_addr('1.2.3.4')}),
('closed', {'kwargs': dict(REASON='END', REMOTE_REASON='DONE')})
]
)
stream = Stream(self)
stream.listen(listener)
stream.update("316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER".split())
stream.update("316 REMAP 186 1.2.3.4:80 SOURCE=EXIT".split())
stream.update("316 CLOSED 186 1.2.3.4:80 REASON=END REMOTE_REASON=DONE".split())
self.assertEqual(len(self.circuits[186].streams), 0)
def test_listener_attach_no_remap(self):
"Attachment is via SENTCONNECT on .onion addresses (for example)"
self.circuits[186] = FakeCircuit(186)
listener = Listener([('new', {
'target_host': 'www.yahoo.com',
'target_port': 80
}), ('attach', {})])
stream = Stream(self)
stream.listen(listener)
stream.update(
"316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER"
.split())
stream.update("316 SENTCONNECT 186 1.2.3.4:80 SOURCE=EXIT".split())
self.assertEqual(self.circuits[186].streams[0], stream)
def test_listener_exception(self):
"""A listener throws an exception during notify"""
exc = Exception("the bad stuff happened")
class Bad(StreamListenerMixin):
def stream_new(*args, **kw):
raise exc
listener = Bad()
stream = Stream(self)
stream.listen(listener)
stream.update("1 NEW 0 94.23.164.42.$43ED8310EB968746970896E8835C2F1991E50B69.exit:9001 SOURCE_ADDR=(Tor_internal):0 PURPOSE=DIR_FETCH".split())
errors = self.flushLoggedErrors()
self.assertEqual(1, len(errors))
self.assertEqual(errors[0].value, exc)
def test_listener_attach(self):
self.circuits[186] = FakeCircuit(186)
listener = Listener([('new', {
'target_host': 'www.yahoo.com',
'target_port': 80
}), ('attach', {
'target_addr': maybe_ip_addr('1.2.3.4')
})])
stream = Stream(self)
stream.listen(listener)
stream.update(
"316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER"
.split())
stream.update("316 REMAP 186 1.2.3.4:80 SOURCE=EXIT".split())
self.assertEqual(self.circuits[186].streams[0], stream)
def test_listener_exception(self):
"""A listener throws an exception during notify"""
exc = Exception("the bad stuff happened")
class Bad(StreamListenerMixin):
def stream_new(*args, **kw):
raise exc
listener = Bad()
stream = Stream(self)
stream.listen(listener)
stream.update(
"1 NEW 0 94.23.164.42.$43ED8310EB968746970896E8835C2F1991E50B69.exit:9001 SOURCE_ADDR=(Tor_internal):0 PURPOSE=DIR_FETCH"
.split())
errors = self.flushLoggedErrors()
self.assertEqual(1, len(errors))
self.assertEqual(errors[0].value, exc)
def test_stream_changed(self):
"Change a stream-id mid-stream."
self.circuits[186] = FakeCircuit(186)
listener = Listener([('new', {'target_host': 'www.yahoo.com',
'target_port': 80}),
('attach', {}),
('succeeded', {})])
stream = Stream(self)
stream.listen(listener)
stream.update("316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER".split())
stream.update("316 SENTCONNECT 186 1.2.3.4:80 SOURCE=EXIT".split())
self.assertEqual(self.circuits[186].streams[0], stream)
# magically change circuit ID without a DETACHED, should fail
stream.update("316 SUCCEEDED 999 1.2.3.4:80 SOURCE=EXIT".split())
errs = self.flushLoggedErrors()
self.assertEqual(len(errs), 1)
# kind of fragile to look at strings, but...
self.assertTrue('186 to 999' in str(errs[0]))
def test_listener_close(self):
self.circuits[186] = FakeCircuit(186)
listener = Listener([('new', {
'target_host': 'www.yahoo.com',
'target_port': 80
}), ('attach', {
'target_addr': maybe_ip_addr('1.2.3.4')
}), ('closed', {
'kwargs': dict(REASON='END', REMOTE_REASON='DONE')
})])
stream = Stream(self)
stream.listen(listener)
stream.update(
"316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER"
.split())
stream.update("316 REMAP 186 1.2.3.4:80 SOURCE=EXIT".split())
stream.update(
"316 CLOSED 186 1.2.3.4:80 REASON=END REMOTE_REASON=DONE".split())
self.assertEqual(len(self.circuits[186].streams), 0)
def test_listener_fail(self):
listener = Listener([('new', {
'target_host': 'www.yahoo.com',
'target_port': 80
}), ('attach', {
'target_addr': maybe_ip_addr('1.2.3.4')
}),
('failed', {
'kwargs':
dict(REASON='TIMEOUT',
REMOTE_REASON='DESTROYED')
})])
stream = Stream(self)
stream.listen(listener)
stream.update(
"316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER"
.split())
self.circuits[186] = FakeCircuit(186)
stream.update("316 REMAP 186 1.2.3.4:80 SOURCE=EXIT".split())
stream.update(
"316 FAILED 0 1.2.3.4:80 REASON=TIMEOUT REMOTE_REASON=DESTROYED".
split())
def test_stream_changed(self):
"Change a stream-id mid-stream."
self.circuits[186] = FakeCircuit(186)
listener = Listener([('new', {
'target_host': 'www.yahoo.com',
'target_port': 80
}), ('attach', {}), ('succeeded', {})])
stream = Stream(self)
stream.listen(listener)
stream.update(
"316 NEW 0 www.yahoo.com:80 SOURCE_ADDR=127.0.0.1:55877 PURPOSE=USER"
.split())
stream.update("316 SENTCONNECT 186 1.2.3.4:80 SOURCE=EXIT".split())
self.assertEqual(self.circuits[186].streams[0], stream)
# magically change circuit ID without a DETACHED, should fail
stream.update("316 SUCCEEDED 999 1.2.3.4:80 SOURCE=EXIT".split())
errs = self.flushLoggedErrors()
self.assertEqual(len(errs), 1)
# kind of fragile to look at strings, but...
self.assertTrue('186 to 999' in str(errs[0]))
def test_listener_new(self):
listener = Listener([('new', {'target_port': 9001})])
stream = Stream(self)
stream.listen(listener)
stream.update("1 NEW 0 94.23.164.42.$43ED8310EB968746970896E8835C2F1991E50B69.exit:9001 SOURCE_ADDR=(Tor_internal):0 PURPOSE=DIR_FETCH".split())
