mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
245 lines
6.5 KiB
Python
245 lines
6.5 KiB
Python
5 years ago
|
#!/usr/bin/env python3
|
||
5 years ago
|
import pyllarp
|
||
5 years ago
|
from time import sleep
|
||
5 years ago
|
from signal import signal, SIGINT
|
||
5 years ago
|
from shutil import rmtree
|
||
|
from os import makedirs
|
||
|
from socket import AF_INET, htons, inet_aton
|
||
|
from pprint import pprint
|
||
5 years ago
|
import sys
|
||
5 years ago
|
from argparse import ArgumentParser as ap
|
||
5 years ago
|
import threading
|
||
|
from collections import deque
|
||
5 years ago
|
|
||
5 years ago
|
class RouterHive(object):
|
||
5 years ago
|
|
||
5 years ago
|
def __init__(self, n_relays=10, n_clients=10, netid="hive"):
|
||
5 years ago
|
|
||
5 years ago
|
self.endpointName = "pyllarp"
|
||
|
self.tmpdir = "/tmp/lokinet_hive"
|
||
|
self.netid = netid
|
||
5 years ago
|
|
||
5 years ago
|
self.n_relays = n_relays
|
||
|
self.n_clients = n_clients
|
||
5 years ago
|
|
||
5 years ago
|
self.addrs = []
|
||
|
self.events = deque()
|
||
5 years ago
|
|
||
5 years ago
|
self.hive = None
|
||
5 years ago
|
|
||
5 years ago
|
pyllarp.EnableDebug()
|
||
|
if not self.RemoveTmpDir():
|
||
|
raise RuntimeError("Failed to initialize Router Hive")
|
||
5 years ago
|
|
||
5 years ago
|
def RemoveTmpDir(self):
|
||
|
if self.tmpdir.startswith("/tmp/") and len(self.tmpdir) > 5:
|
||
|
print("calling rmdir -r %s" % self.tmpdir)
|
||
5 years ago
|
rmtree(self.tmpdir, ignore_errors=True)
|
||
|
return True
|
||
5 years ago
|
else:
|
||
|
print("not removing dir %s because it doesn't start with /tmp/" % self.tmpdir)
|
||
5 years ago
|
|
||
5 years ago
|
return False
|
||
5 years ago
|
|
||
5 years ago
|
def MakeEndpoint(self, router, after):
|
||
|
if router.IsRelay():
|
||
|
return
|
||
|
ep = pyllarp.Endpoint(self.endpointName, router)
|
||
|
router.AddEndpoint(ep)
|
||
|
if after is not None:
|
||
|
router.CallSafe(lambda : after(ep))
|
||
5 years ago
|
|
||
5 years ago
|
def AddRelay(self, index):
|
||
|
dirname = "%s/relays/%d" % (self.tmpdir, index)
|
||
|
makedirs("%s/netdb" % dirname, exist_ok=True)
|
||
5 years ago
|
|
||
5 years ago
|
config = pyllarp.Config()
|
||
5 years ago
|
|
||
5 years ago
|
port = index + 30000
|
||
|
tunname = "lokihive%d" % index
|
||
5 years ago
|
|
||
5 years ago
|
config.router.encryptionKeyfile = "%s/encryption.key" % dirname
|
||
|
config.router.transportKeyfile = "%s/transport.key" % dirname
|
||
|
config.router.identKeyfile = "%s/identity.key" % dirname
|
||
|
config.router.ourRcFile = "%s/rc.signed" % dirname
|
||
|
config.router.netid = self.netid
|
||
|
config.router.nickname = "Router%d" % index
|
||
|
config.router.publicOverride = True
|
||
|
config.router.overrideAddress("127.0.0.1", '{}'.format(port))
|
||
|
"""
|
||
|
config.router.ip4addr.sin_family = AF_INET
|
||
|
config.router.ip4addr.sin_port = htons(port)
|
||
|
config.router.ip4addr.sin_addr.set("127.0.0.1")
|
||
|
"""
|
||
|
config.router.blockBogons = False
|
||
5 years ago
|
|
||
5 years ago
|
config.network.enableProfiling = False
|
||
|
config.network.routerProfilesFile = "%s/profiles.dat" % dirname
|
||
|
config.network.netConfig = {"type": "null"}
|
||
5 years ago
|
|
||
5 years ago
|
config.netdb.nodedbDir = "%s/netdb" % dirname
|
||
5 years ago
|
|
||
5 years ago
|
config.links.InboundLinks = [("lo", AF_INET, port, set())]
|
||
5 years ago
|
|
||
5 years ago
|
config.system.pidfile = "%s/lokinet.pid" % dirname
|
||
5 years ago
|
|
||
5 years ago
|
config.dns.netConfig = {"local-dns": ("127.3.2.1:%d" % port)}
|
||
5 years ago
|
|
||
5 years ago
|
if index != 1:
|
||
|
config.bootstrap.routers = ["%s/relays/1/rc.signed" % self.tmpdir]
|
||
5 years ago
|
|
||
5 years ago
|
self.hive.AddRelay(config)
|
||
5 years ago
|
|
||
|
|
||
5 years ago
|
def AddClient(self, index):
|
||
|
dirname = "%s/clients/%d" % (self.tmpdir, index)
|
||
|
makedirs("%s/netdb" % dirname, exist_ok=True)
|
||
5 years ago
|
|
||
5 years ago
|
config = pyllarp.Config()
|
||
5 years ago
|
|
||
5 years ago
|
port = index + 40000
|
||
|
tunname = "lokihive%d" % index
|
||
5 years ago
|
|
||
5 years ago
|
config.router.encryptionKeyfile = "%s/encryption.key" % dirname
|
||
|
config.router.transportKeyfile = "%s/transport.key" % dirname
|
||
|
config.router.identKeyfile = "%s/identity.key" % dirname
|
||
|
config.router.ourRcFile = "%s/rc.signed" % dirname
|
||
|
config.router.netid = self.netid
|
||
|
config.router.blockBogons = False
|
||
5 years ago
|
|
||
5 years ago
|
config.network.enableProfiling = False
|
||
|
config.network.routerProfilesFile = "%s/profiles.dat" % dirname
|
||
|
config.network.netConfig = {"type": "null"}
|
||
5 years ago
|
|
||
5 years ago
|
config.netdb.nodedbDir = "%s/netdb" % dirname
|
||
5 years ago
|
|
||
5 years ago
|
config.system.pidfile = "%s/lokinet.pid" % dirname
|
||
5 years ago
|
|
||
5 years ago
|
config.dns.netConfig = {"local-dns": ("127.3.2.1:%d" % port)}
|
||
5 years ago
|
|
||
5 years ago
|
config.bootstrap.routers = ["%s/relays/1/rc.signed" % self.tmpdir]
|
||
5 years ago
|
|
||
5 years ago
|
self.hive.AddClient(config)
|
||
5 years ago
|
|
||
5 years ago
|
def onGotEndpoint(self, ep):
|
||
5 years ago
|
addr = ep.OurAddress()
|
||
5 years ago
|
self.addrs.append(pyllarp.ServiceAddress(addr))
|
||
5 years ago
|
|
||
5 years ago
|
def sendToAddress(self, router, toaddr, pkt):
|
||
5 years ago
|
if router.IsRelay():
|
||
|
return
|
||
|
if router.TrySendPacket("default", toaddr, pkt):
|
||
|
print("sending {} bytes to {}".format(len(pkt), toaddr))
|
||
5 years ago
|
|
||
5 years ago
|
def broadcastTo(self, addr, pkt):
|
||
|
self.hive.ForEachRouter(lambda r : sendToAddress(r, addr, pkt))
|
||
|
|
||
|
def InitFirstRC(self):
|
||
|
print("Starting first router to init its RC for bootstrap")
|
||
|
self.hive = pyllarp.RouterHive()
|
||
|
self.AddRelay(1)
|
||
|
self.hive.StartRelays()
|
||
|
print("sleeping 2 sec to give plenty of time to save bootstrap rc")
|
||
|
sleep(2)
|
||
5 years ago
|
|
||
5 years ago
|
self.hive.StopAll()
|
||
5 years ago
|
|
||
5 years ago
|
def Start(self):
|
||
5 years ago
|
|
||
5 years ago
|
self.InitFirstRC()
|
||
5 years ago
|
|
||
5 years ago
|
print("Resetting hive. Creating %d relays and %d clients" % (self.n_relays, self.n_clients))
|
||
5 years ago
|
|
||
5 years ago
|
self.hive = pyllarp.RouterHive()
|
||
5 years ago
|
|
||
5 years ago
|
for i in range(1, self.n_relays + 1):
|
||
|
self.AddRelay(i)
|
||
5 years ago
|
|
||
5 years ago
|
for i in range(1, self.n_clients + 1):
|
||
|
self.AddClient(i)
|
||
5 years ago
|
|
||
5 years ago
|
print("Starting relays")
|
||
|
self.hive.StartRelays()
|
||
|
|
||
|
sleep(1)
|
||
|
self.hive.ForEachRelay(lambda r: self.MakeEndpoint(r, self.onGotEndpoint))
|
||
|
|
||
|
print("Sleeping 5 seconds before starting clients")
|
||
|
sleep(5)
|
||
|
|
||
|
self.hive.StartClients()
|
||
|
|
||
|
sleep(1)
|
||
|
self.hive.ForEachClient(lambda r: self.MakeEndpoint(r, self.onGotEndpoint))
|
||
5 years ago
|
|
||
5 years ago
|
def Stop(self):
|
||
|
self.hive.StopAll()
|
||
|
|
||
|
def CollectNextEvent(self):
|
||
|
self.events.append(self.hive.GetNextEvent())
|
||
|
|
||
|
def CollectAllEvents(self):
|
||
|
self.events.extend(self.hive.GetAllEvents())
|
||
|
|
||
|
def PopEvent(self):
|
||
|
self.CollectAllEvents()
|
||
|
if len(self.events):
|
||
|
return self.events.popleft()
|
||
|
return None
|
||
|
|
||
|
def main(n_relays=10, n_clients=10, print_each_event=True):
|
||
|
|
||
|
running = True
|
||
|
|
||
|
def handle_sigint(sig, frame):
|
||
|
nonlocal running
|
||
|
running = False
|
||
|
print("SIGINT received, attempting to stop all routers")
|
||
|
|
||
|
signal(SIGINT, handle_sigint)
|
||
|
|
||
|
try:
|
||
|
hive = RouterHive(n_relays, n_clients)
|
||
|
hive.Start()
|
||
|
|
||
|
except Exception as err:
|
||
|
print(err)
|
||
|
return 1
|
||
5 years ago
|
|
||
5 years ago
|
total_events = 0
|
||
|
event_counts = dict()
|
||
5 years ago
|
while running:
|
||
5 years ago
|
event = hive.PopEvent()
|
||
5 years ago
|
event_name = event.__class__.__name__
|
||
5 years ago
|
if event:
|
||
5 years ago
|
if print_each_event:
|
||
|
print("Event: %s -- Triggered: %s" % (event_name, event.triggered))
|
||
|
print(event)
|
||
|
hops = getattr(event, "hops", None)
|
||
|
if hops:
|
||
|
for hop in hops:
|
||
|
print(hop)
|
||
5 years ago
|
|
||
5 years ago
|
total_events = total_events + 1
|
||
|
if event_name in event_counts:
|
||
|
event_counts[event_name] = event_counts[event_name] + 1
|
||
|
else:
|
||
|
event_counts[event_name] = 1
|
||
|
|
||
|
if total_events % 10 == 0:
|
||
|
pprint(event_counts)
|
||
|
|
||
5 years ago
|
sleep(.01)
|
||
|
|
||
5 years ago
|
print('stopping')
|
||
5 years ago
|
hive.Stop()
|
||
5 years ago
|
print('stopped')
|
||
5 years ago
|
del hive
|
||
5 years ago
|
|
||
|
if __name__ == '__main__':
|
||
5 years ago
|
parser = ap()
|
||
|
print_events = False
|
||
|
parser.add_argument('--print-events', dest="print_events", action='store_true')
|
||
|
args = parser.parse_args()
|
||
|
main(print_each_event = args.print_events)
|