aboutsummaryrefslogtreecommitdiff
path: root/networks/hs_instance_start.py
blob: 19cb3ad009d01d649391a5eac7df481afc48877d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
from chutney.Testing import *
from chutney.TorNet import *

from stem.control import EventType

def hs_instance_start(network):
    result = True

    hs_nodes = network.get("h")
    clients = network.get("c")

    if len(clients) % len(hs_nodes) != 0:
        logging.error("There must be a equal number of clients per hs_node")
        return False

    clients_per_node = len(clients) / len(hs_nodes)

    if clients_per_node < 1:
        logging.error("Not enough clients")
        return False

    logging.info("Clients per node %d" % clients_per_node)

    # To ensure the ..
    client_groups = zip(*(iter(clients),) * clients_per_node)

    initial_nodes = network.get("a") + network.get("r") + network.get("c")

    if not all([ n.getController().start() for n in initial_nodes ]):
        return False

    nodes_by_fingerprint = get_node_fingerprints(network.get("a") + network.get("r"))

    logging.info("All initial nodes running")

    logging.info("There are %d nodes to start" % len(hs_nodes))

    for i, hs_node in enumerate(hs_nodes):
        logging.info("Starting node %d" % i)

        if not hs_node.getController().start():
            return False # Stop on failure

        track_introduction_points(hs_node)

        node_published_descriptor = threading.Event()

        def hs_node_listener(logevent):
            if "Successfully uploaded v2 rend descriptors" in logevent.message:
                node_published_descriptor.set()

        hs_node.getStemController().add_event_listener(hs_node_listener, EventType.INFO)

        node_published_descriptor.wait()

        hs_node.getStemController().remove_event_listener(hs_node_listener)

        # Now begin testing

        for c_group in range(i + 1):
            logging.info("Testing with the %d client group" % c_group)
            expected_nodes = min(c_group, i)
            result &= connection_test(client_groups[c_group], expected_nodes)

        result &= check_same_intro_points()

    return result