mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2024-12-19 04:57:54 +00:00
124 lines
3.7 KiB
Python
124 lines
3.7 KiB
Python
from __future__ import print_function
|
|
|
|
import sys
|
|
import time
|
|
import shutil
|
|
from os import mkdir, unlink, listdir
|
|
from os.path import join, exists
|
|
from six.moves import StringIO
|
|
|
|
from twisted.internet.protocol import ProcessProtocol
|
|
from twisted.internet.error import ProcessExitedAlready, ProcessDone
|
|
from twisted.internet.defer import inlineCallbacks, Deferred
|
|
import pytest_twisted
|
|
|
|
import util
|
|
|
|
# see "conftest.py" for the fixtures (e.g. "magic_folder")
|
|
|
|
@pytest_twisted.inlineCallbacks
|
|
def test_onion_service_storage(reactor, request, temp_dir, flog_gatherer, tor_network, tor_introducer_furl):
|
|
yield _create_anonymous_node(reactor, 'carol', 8008, request, temp_dir, flog_gatherer, tor_network, tor_introducer_furl)
|
|
yield _create_anonymous_node(reactor, 'dave', 8009, request, temp_dir, flog_gatherer, tor_network, tor_introducer_furl)
|
|
# ensure both nodes are connected to "a grid" by uploading
|
|
# something via carol, and retrieve it using dave.
|
|
gold_path = join(temp_dir, "gold")
|
|
with open(gold_path, "w") as f:
|
|
f.write(
|
|
"The object-capability model is a computer security model. A "
|
|
"capability describes a transferable right to perform one (or "
|
|
"more) operations on a given object."
|
|
)
|
|
# XXX could use treq or similar to POST these to their respective
|
|
# WUIs instead ...
|
|
|
|
proto = util._CollectOutputProtocol()
|
|
reactor.spawnProcess(
|
|
proto,
|
|
sys.executable,
|
|
(
|
|
sys.executable, '-m', 'allmydata.scripts.runner',
|
|
'-d', join(temp_dir, 'carol'),
|
|
'put', gold_path,
|
|
)
|
|
)
|
|
yield proto.done
|
|
cap = proto.output.getvalue().strip().split()[-1]
|
|
print("TEH CAP!", cap)
|
|
|
|
proto = util._CollectOutputProtocol()
|
|
reactor.spawnProcess(
|
|
proto,
|
|
sys.executable,
|
|
(
|
|
sys.executable, '-m', 'allmydata.scripts.runner',
|
|
'-d', join(temp_dir, 'dave'),
|
|
'get', cap,
|
|
)
|
|
)
|
|
yield proto.done
|
|
|
|
dave_got = proto.output.getvalue().strip()
|
|
assert dave_got == open(gold_path, 'r').read().strip()
|
|
|
|
|
|
@pytest_twisted.inlineCallbacks
|
|
def _create_anonymous_node(reactor, name, control_port, request, temp_dir, flog_gatherer, tor_network, introducer_furl):
|
|
node_dir = join(temp_dir, name)
|
|
web_port = "tcp:{}:interface=localhost".format(control_port + 2000)
|
|
|
|
if True:
|
|
print("creating", node_dir)
|
|
mkdir(node_dir)
|
|
proto = util._DumpOutputProtocol(None)
|
|
reactor.spawnProcess(
|
|
proto,
|
|
sys.executable,
|
|
(
|
|
sys.executable, '-m', 'allmydata.scripts.runner',
|
|
'create-node',
|
|
'--nickname', name,
|
|
'--introducer', introducer_furl,
|
|
'--hide-ip',
|
|
'--tor-control-port', 'tcp:localhost:{}'.format(control_port),
|
|
'--listen', 'tor',
|
|
node_dir,
|
|
)
|
|
)
|
|
yield proto.done
|
|
|
|
with open(join(node_dir, 'tahoe.cfg'), 'w') as f:
|
|
f.write('''
|
|
[node]
|
|
nickname = %(name)s
|
|
web.port = %(web_port)s
|
|
web.static = public_html
|
|
log_gatherer.furl = %(log_furl)s
|
|
|
|
[tor]
|
|
control.port = tcp:localhost:%(control_port)d
|
|
onion.external_port = 3457
|
|
onion.local_port = %(local_port)d
|
|
onion = true
|
|
onion.private_key_file = private/tor_onion.privkey
|
|
|
|
[client]
|
|
# Which services should this client connect to?
|
|
introducer.furl = %(furl)s
|
|
shares.needed = 1
|
|
shares.happy = 1
|
|
shares.total = 2
|
|
|
|
''' % {
|
|
'name': name,
|
|
'furl': introducer_furl,
|
|
'web_port': web_port,
|
|
'log_furl': flog_gatherer,
|
|
'control_port': control_port,
|
|
'local_port': control_port + 1000,
|
|
})
|
|
|
|
print("running")
|
|
yield util._run_node(reactor, node_dir, request, None)
|
|
print("okay, launched")
|