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.
124 lines
3.1 KiB
Python
124 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
import pyllarp
|
|
from time import sleep
|
|
from signal import signal, SIGINT
|
|
from shutil import rmtree
|
|
from os import makedirs
|
|
from socket import AF_INET, htons, inet_aton
|
|
from pprint import pprint
|
|
|
|
tmpdir = "/tmp/lokinet_hive"
|
|
|
|
def RemoveTmpDir(dirname):
|
|
if dirname.startswith("/tmp/") and len(dirname) > 5:
|
|
print("calling rmdir -r %s" % dirname)
|
|
if (input("Is this ok? (y/n): ").lower().strip()[:1] == "y"):
|
|
rmtree(dirname, ignore_errors=True)
|
|
else:
|
|
print("not removing dir %s because it doesn't start with /tmp/" % dirname)
|
|
|
|
def MakeEndpoint(router, idx, after):
|
|
ep = pyllarp.Endpoint("pyllarp-{}".format(idx), router)
|
|
if after is not None:
|
|
router.SafeCall(lambda : after(ep))
|
|
|
|
def AddRouter(hive, index, gotEndpoint=None, netid="hive"):
|
|
dirname = "%s/routers/%d" % (tmpdir, index)
|
|
makedirs("%s/netdb" % dirname, exist_ok=True)
|
|
|
|
config = pyllarp.Config()
|
|
|
|
port = index + 30000
|
|
tunname = "lokihive%d" % index
|
|
|
|
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 = netid
|
|
config.router.nickname = "Router%d" % index
|
|
config.router.publicOverride = True
|
|
"""
|
|
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
|
|
|
|
config.network.enableProfiling = False
|
|
config.network.routerProfilesFile = "%s/profiles.dat" % dirname
|
|
config.network.netConfig = {"type": "null"}
|
|
|
|
config.netdb.nodedbDir = "%s/netdb" % dirname
|
|
|
|
config.links.InboundLinks = [("lo", AF_INET, port, set())]
|
|
|
|
config.system.pidfile = "%s/lokinet.pid" % dirname
|
|
|
|
config.dns.netConfig = {"local-dns": ("127.3.2.1:%d" % port)}
|
|
|
|
if index != 1:
|
|
config.bootstrap.routers = ["%s/routers/1/rc.signed" % tmpdir]
|
|
|
|
hive.AddRouter(config)
|
|
if index != 1:
|
|
hive.VisitRouter(index, lambda r : MakeEndpoint(r, index, gotEndpoint))
|
|
|
|
|
|
|
|
def main():
|
|
|
|
running = True
|
|
RemoveTmpDir(tmpdir)
|
|
|
|
def handle_sigint(sig, frame):
|
|
nonlocal running
|
|
running = False
|
|
print("SIGINT received, attempting to stop all routers")
|
|
|
|
signal(SIGINT, handle_sigint)
|
|
|
|
hive = pyllarp.RouterHive()
|
|
AddRouter(hive, 1)
|
|
hive.StartAll()
|
|
sleep(5)
|
|
|
|
hive.StopAll()
|
|
|
|
r = pyllarp.RouterContact()
|
|
r.ReadFile("/tmp/lokinet_hive/routers/1/rc.signed")
|
|
print(r.ToString())
|
|
|
|
hive = pyllarp.RouterHive()
|
|
|
|
addrs = []
|
|
|
|
def onGotEndpoint(ep):
|
|
addr = ep.OurAddress()
|
|
print("got endpoint: {}".format(addr))
|
|
addrs.append(addr)
|
|
|
|
def sendToAddress(router, toaddr, pkt):
|
|
print("sending {} bytes to {}".format(len(pkt), toaddr))
|
|
router.TrySendPacket(toaddr, pkt)
|
|
|
|
def broadcastTo(addr, pkt):
|
|
hive.ForEach(lambda r : sendToAddress(r, addr, pkt))
|
|
|
|
for i in range(1, 11):
|
|
AddRouter(hive, i, onGotEndpoint)
|
|
|
|
hive.StartAll()
|
|
|
|
while running:
|
|
event = hive.GetNextEvent()
|
|
if event:
|
|
print(event)
|
|
for addr in addrs:
|
|
broadcastTo(addr, "packet lol")
|
|
|
|
hive.StopAll()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|