mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-17 15:25:35 +00:00
166 lines
4.9 KiB
Python
166 lines
4.9 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# lokinet kubernetes runtime wrapper
|
|
#
|
|
|
|
from ctypes import *
|
|
import configparser
|
|
import signal
|
|
import time
|
|
import threading
|
|
import os
|
|
import sys
|
|
import requests
|
|
|
|
lib_file = os.path.join(os.path.realpath('.'), 'liblokinet-shared.so')
|
|
|
|
|
|
def log(msg):
|
|
sys.stderr.write("lokinet: {}\n".format(msg))
|
|
sys.stderr.flush()
|
|
|
|
class LokiNET(threading.Thread):
|
|
|
|
lib = None
|
|
ctx = 0
|
|
failed = False
|
|
up = False
|
|
|
|
asRouter = True
|
|
|
|
def configure(self, lib, conf, ip=None, port=None, ifname=None):
|
|
log("configure lib={} conf={}".format(lib, conf))
|
|
try:
|
|
self.lib = CDLL(lib)
|
|
except OSError as ex:
|
|
log("failed to load library: {}".format(ex))
|
|
return False
|
|
if self.lib.llarp_ensure_config(conf.encode('utf-8'), os.path.dirname(conf).encode('utf-8'), True, self.asRouter):
|
|
config = configparser.ConfigParser()
|
|
config.read(conf)
|
|
log('overwrite ip="{}" port="{}" ifname="{}"'.format(ip, port, ifname))
|
|
if ip:
|
|
config['router']['public-address'] = '{}'.format(ip)
|
|
if port:
|
|
config['router']['public-port'] = '{}'.format(port)
|
|
if ifname and port:
|
|
config['bind'] = {
|
|
ifname: '{}'.format(port)
|
|
}
|
|
with open(conf, "w") as f:
|
|
config.write(f)
|
|
self.ctx = self.lib.llarp_main_init(conf.encode('utf-8'))
|
|
else:
|
|
return False
|
|
return self.lib.llarp_main_setup(self.ctx) == 0
|
|
|
|
|
|
def inform_fail(self):
|
|
"""
|
|
inform lokinet crashed
|
|
"""
|
|
self.failed = True
|
|
self._inform()
|
|
|
|
def inform_up(self):
|
|
self.up = True
|
|
self._inform()
|
|
|
|
def _inform(self):
|
|
"""
|
|
inform waiter
|
|
"""
|
|
|
|
def wait_for_up(self, timeout):
|
|
"""
|
|
wait for lokinet to go up for :timeout: seconds
|
|
:return True if we are up and running otherwise False:
|
|
"""
|
|
# return self._up.wait(timeout)
|
|
|
|
def signal(self, sig):
|
|
if self.ctx and self.lib:
|
|
self.lib.llarp_main_signal(self.ctx, int(sig))
|
|
|
|
def run(self):
|
|
# self._up.acquire()
|
|
self.up = True
|
|
code = self.lib.llarp_main_run(self.ctx)
|
|
log("llarp_main_run exited with status {}".format(code))
|
|
if code:
|
|
self.inform_fail()
|
|
self.up = False
|
|
# self._up.release()
|
|
|
|
def close(self):
|
|
if self.lib and self.ctx:
|
|
self.lib.llarp_main_free(self.ctx)
|
|
|
|
def getconf(name, fallback=None):
|
|
return name in os.environ and os.environ[name] or fallback
|
|
|
|
def main(args):
|
|
log("going up")
|
|
root = getconf("LOKINET_ROOT")
|
|
if root is None:
|
|
print("LOKINET_ROOT was not set")
|
|
return
|
|
|
|
rc_callback = getconf("LOKINET_SUBMIT_URL")
|
|
if rc_callback is None:
|
|
print("LOKINET_SUBMIT_URL was not set")
|
|
return
|
|
|
|
lib = getconf("LOKINET_LIB", lib_file)
|
|
timeout = int(getconf("LOKINET_TIMEOUT", "5"))
|
|
ping_interval = int(getconf("LOKINET_PING_INTERVAL", "60"))
|
|
ping_callback = getconf("LOKINET_PING_URL")
|
|
ip = getconf("LOKINET_IP")
|
|
port = getconf("LOKINET_PORT")
|
|
ifname = getconf("LOKINET_IFNAME")
|
|
if ping_callback is None:
|
|
print("LOKINET_PING_URL was not set")
|
|
return
|
|
conf = os.path.join(root, "daemon.ini")
|
|
loki = LokiNET()
|
|
if loki.configure(lib, conf, ip, port, ifname):
|
|
log("configured")
|
|
loki.start()
|
|
try:
|
|
log("waiting for spawn")
|
|
while timeout > 0:
|
|
time.sleep(1)
|
|
if loki.failed:
|
|
log("failed")
|
|
break
|
|
log("waiting {}".format(timeout))
|
|
timeout -= 1
|
|
if loki.up:
|
|
log("submitting rc")
|
|
try:
|
|
with open(os.path.join(root, 'self.signed'), 'rb') as f:
|
|
r = requests.put(rc_callback, data=f.read(), headers={"content-type": "application/octect-stream"})
|
|
log('submit rc reply: HTTP {}'.format(r.status_code))
|
|
except Exception as ex:
|
|
log("failed to submit rc: {}".format(ex))
|
|
while loki.up:
|
|
time.sleep(ping_interval)
|
|
try:
|
|
r = requests.get(ping_callback)
|
|
log("ping reply: HTTP {}".format(r.status_code))
|
|
except Exception as ex:
|
|
log("failed to submit ping: {}".format(ex))
|
|
else:
|
|
log("failed to go up")
|
|
loki.signal(signal.SIGINT)
|
|
except KeyboardInterrupt:
|
|
loki.signal(signal.SIGINT)
|
|
time.sleep(2)
|
|
finally:
|
|
loki.close()
|
|
else:
|
|
loki.close()
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv[1:])
|