def test_set_save_modify(self):
self.protocol.answers.append('')
conf = TorConfig(self.protocol)
conf.hiddenservices = [HiddenService(conf, '/fake/path', ['80 127.0.0.1:1234'], '', 3)]
self.assertTrue(conf.needs_save())
conf.save()
self.assertEqual(len(conf.hiddenservices), 1)
self.assertEqual(conf.hiddenservices[0].dir, '/fake/path')
self.assertEqual(conf.hiddenservices[0].version, 3)
self.assertEqual(conf.hiddenservices[0].authorize_client, '')
conf.hiddenservices[0].ports = ['123 127.0.0.1:4321']
conf.save()
self.assertTrue(not conf.needs_save())
conf.hiddenservices[0].ports.append('90 127.0.0.1:2345')
self.assertTrue(conf.needs_save())
def test_create_torrc(self):
config = TorConfig()
config.SocksPort = 1234
config.hiddenservices = [
HiddenService(config, '/some/dir', '80 127.0.0.1:1234',
'auth', 2, True)
]
config.Log = ['80 127.0.0.1:80', '90 127.0.0.1:90']
config.save()
torrc = config.create_torrc()
lines = torrc.split('\n')
lines.sort()
torrc = '\n'.join(lines).strip()
self.assertEqual(torrc, '''HiddenServiceAuthorizeClient auth
HiddenServiceDir /some/dir
HiddenServicePort 80 127.0.0.1:1234
HiddenServiceVersion 2
Log 80 127.0.0.1:80
Log 90 127.0.0.1:90
SocksPort 1234''')
def test_options_hidden(self):
self.protocol.answers.append('HiddenServiceDir=/fake/path\nHiddenServicePort=80 127.0.0.1:1234\n')
conf = TorConfig(self.protocol)
self.assertTrue('HiddenServiceOptions' not in conf.config)
self.assertEqual(len(conf.HiddenServices), 1)
self.assertTrue(not conf.needs_save())
conf.hiddenservices.append(HiddenService(conf, '/some/dir', '80 127.0.0.1:2345', 'auth', 2))
conf.hiddenservices[0].ports.append('443 127.0.0.1:443')
self.assertTrue(conf.needs_save())
conf.save()
self.assertEqual(conf.get_type('HiddenServices'), HiddenService)
self.assertEqual(len(self.protocol.sets), 7)
self.assertEqual(self.protocol.sets[0], ('HiddenServiceDir', '/fake/path'))
self.assertEqual(self.protocol.sets[1], ('HiddenServicePort', '80 127.0.0.1:1234'))
self.assertEqual(self.protocol.sets[2], ('HiddenServicePort', '443 127.0.0.1:443'))
self.assertEqual(self.protocol.sets[3], ('HiddenServiceDir', '/some/dir'))
self.assertEqual(self.protocol.sets[4], ('HiddenServicePort', '80 127.0.0.1:2345'))
self.assertEqual(self.protocol.sets[5], ('HiddenServiceVersion', '2'))
self.assertEqual(self.protocol.sets[6], ('HiddenServiceAuthorizeClient', 'auth'))
def test_stealth_clients(self):
# FIXME test without crapping on filesystem
self.protocol.answers.append('HiddenServiceDir=/fake/path\n')
d = tempfile.mkdtemp()
try:
with open(os.path.join(d, 'hostname'), 'w') as f:
f.write('oniona cookiea\n')
f.write('onionb cookieb\n')
conf = TorConfig(self.protocol)
hs = HiddenService(conf, d, [])
self.assertEqual(2, len(hs.clients))
self.assertEqual('oniona', hs.clients[0][0])
self.assertEqual('cookiea', hs.clients[0][1])
self.assertEqual('onionb', hs.clients[1][0])
self.assertEqual('cookieb', hs.clients[1][1])
self.assertRaises(RuntimeError, getattr, hs, 'hostname')
finally:
shutil.rmtree(d, ignore_errors=True)
def test_two_hidden_services_before_save(self):
conf = TorConfig()
conf.HiddenServices = [HiddenService(conf, '/fake/path', ['80 127.0.0.1:1234'])]
conf.HiddenServices.append(HiddenService(conf, '/fake/path/two', ['1234 127.0.0.1:1234']))
conf.save()
self.assertEqual(2, len(conf.HiddenServices))
def test_save_no_protocol(self):
conf = TorConfig()
conf.HiddenServices = [HiddenService(conf, '/fake/path', ['80 127.0.0.1:1234'])]
conf.save()