mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-17 15:25:35 +00:00
272 lines
8.3 KiB
Python
Executable File
272 lines
8.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import requests
|
|
import json
|
|
import time
|
|
import curses
|
|
import math
|
|
import traceback
|
|
|
|
|
|
class Monitor:
|
|
|
|
_speedSamples = 8
|
|
_globalspeed = []
|
|
|
|
def __init__(self, url):
|
|
self.data = dict()
|
|
self.win = curses.initscr()
|
|
self._url = url
|
|
while len(self._globalspeed) < self._speedSamples:
|
|
self._globalspeed.append((0, 0))
|
|
|
|
def __del__(self):
|
|
curses.endwin()
|
|
|
|
def on_timer(self, event):
|
|
"""called on timer event"""
|
|
self.update_data()
|
|
|
|
def jsonrpc(self, meth, params):
|
|
r = requests.post(
|
|
self._url,
|
|
headers={"Content-Type": "application/json", "Host": "localhost"},
|
|
json={
|
|
"jsonrpc": "2.0",
|
|
"id": "0",
|
|
"method": "{}".format(meth),
|
|
"params": params,
|
|
},
|
|
)
|
|
return r.json()
|
|
|
|
def update_data(self):
|
|
"""update data from lokinet"""
|
|
try:
|
|
j = self.jsonrpc("llarp.admin.dumpstate", {})
|
|
self.data = j["result"]
|
|
except Exception as ex:
|
|
self.data = None
|
|
|
|
def _render_path(self, y, path, name):
|
|
"""render a path at current position"""
|
|
self.win.move(y, 1)
|
|
self.win.addstr("({}) ".format(name))
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
y += 1
|
|
self.win.addstr("[tx:\t{}]\t[rx:\t{}]".format(self.speedOf(path['txRateCurrent']), self.speedOf(path['rxRateCurrent'])))
|
|
self.win.move(y, 1)
|
|
y += 1
|
|
self.win.addstr("me -> ")
|
|
for hop in path["hops"]:
|
|
self.win.addstr(" {} ->".format(hop["router"][:4]))
|
|
self.win.addstr(" [{} ms latency]".format(path["intro"]["latency"]))
|
|
self.win.addstr(" [{} until expire]".format(self.timeTo(path["expiresAt"])))
|
|
if path["expiresSoon"]:
|
|
self.win.addstr("(expiring)")
|
|
elif path["expired"]:
|
|
self.win.addstr("(expired)")
|
|
return y
|
|
|
|
def timeTo(self, ts):
|
|
""" return time until timestamp in seconds formatted"""
|
|
now = time.time() * 1000
|
|
return "{} seconds".format(int((ts - now) / 1000))
|
|
|
|
def speedOf(self, rate):
|
|
"""turn int speed into string formatted"""
|
|
units = ["B", "KB", "MB", "GB"]
|
|
idx = 0
|
|
while rate > 1000 and idx < len(units):
|
|
rate /= 1000.0
|
|
idx += 1
|
|
return "{} {}ps".format("%.2f" % rate, units[idx])
|
|
|
|
def display_service(self, y, name, status):
|
|
"""display a service at current position"""
|
|
self.win.move(y, 1)
|
|
self.win.addstr("service [{}]".format(name))
|
|
build = status["buildStats"]
|
|
ratio = build["success"] / (build["attempts"] or 1)
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
self.win.addstr("build success: {} %".format(int(100 * ratio)))
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
paths = status["paths"]
|
|
self.win.addstr("paths: {}".format(len(paths)))
|
|
for path in paths:
|
|
y = self._render_path(y, path, "inbound")
|
|
for session in (status["remoteSessions"] or []):
|
|
for path in session["paths"]:
|
|
y = self._render_path(
|
|
y, path, "[active] {}".format(session["currentConvoTag"])
|
|
)
|
|
for session in (status["snodeSessions"] or []):
|
|
for path in session["paths"]:
|
|
y = self._render_path(y, path, "[snode]")
|
|
return y
|
|
|
|
# for k in status:
|
|
# self.win.move(y + 1, 1)
|
|
# y += 1
|
|
# self.win.addstr('{}: {}'.format(k, json.dumps(status[k])))
|
|
|
|
def display_links(self, y, data):
|
|
self.txrate = 0
|
|
self.rxrate = 0
|
|
for link in data["outbound"]:
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
self.win.addstr("outbound sessions:")
|
|
y = self.display_link(y, link)
|
|
for link in data["inbound"]:
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
self.win.addstr("inbound sessions:")
|
|
y = self.display_link(y, link)
|
|
y += 2
|
|
self.win.move(y, 1)
|
|
self.win.addstr(
|
|
"global speed:\t\t[{}\ttx]\t[{}\trx]".format(
|
|
self.speedOf(self.txrate), self.speedOf(self.rxrate)
|
|
)
|
|
)
|
|
|
|
self._globalspeed.append((self.txrate, self.rxrate))
|
|
while len(self._globalspeed) > self._speedSamples:
|
|
self._globalspeed.pop(0)
|
|
return self.display_speedgraph(y + 2, self._globalspeed)
|
|
|
|
def display_speedgraph(self, y, samps, maxsz=20):
|
|
""" display global speed graph """
|
|
|
|
def scale(x, n):
|
|
while n > 0:
|
|
x /= 2
|
|
n -= 1
|
|
return int(x)
|
|
|
|
txmax, rxmax = 1000, 1000
|
|
for tx, rx in samps:
|
|
if tx > txmax:
|
|
txmax = tx
|
|
if rx > rxmax:
|
|
rxmax = rx
|
|
|
|
rxscale = 0
|
|
while rxmax > maxsz:
|
|
rxscale += 1
|
|
rxmax /= 2
|
|
|
|
txscale = 0
|
|
while txmax > maxsz:
|
|
txscale += 1
|
|
txmax /= 2
|
|
|
|
def makebar(samp, max):
|
|
bar = "#" * samp
|
|
pad = " " * (max - samp)
|
|
return pad, bar
|
|
|
|
txlabelpad = int(txmax / 2) - 1
|
|
rxlabelpad = int(rxmax / 2) - 1
|
|
if txlabelpad <= 0:
|
|
txlabelpad = 1
|
|
if rxlabelpad <= 0:
|
|
rxlabelpad = 1
|
|
txlabelpad = " " * txlabelpad
|
|
rxlabelpad = " " * rxlabelpad
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
self.win.addstr(
|
|
"{}tx{}{}rx{}".format(txlabelpad, txlabelpad, rxlabelpad, rxlabelpad)
|
|
)
|
|
for tx, rx in samps:
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
txpad, txbar = makebar(scale(tx, txscale), int(txmax))
|
|
rxpad, rxbar = makebar(scale(rx, rxscale), int(rxmax))
|
|
self.win.addstr("{}{}|{}{}".format(txpad, txbar, rxbar, rxpad))
|
|
return y + 2
|
|
|
|
def display_link(self, y, link):
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
sessions = link["sessions"]["established"]
|
|
for s in sessions:
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
self.txrate += s["txRateCurrent"]
|
|
self.rxrate += s["rxRateCurrent"]
|
|
self.win.addstr(
|
|
"{}\t[{}\ttx]\t[{}\trx]".format(
|
|
s["remoteAddr"], self.speedOf(s["txRateCurrent"]), self.speedOf(s["rxRateCurrent"])
|
|
)
|
|
)
|
|
if (s['txMsgQueueSize'] or 0) > 1:
|
|
self.win.addstr(" [out window:\t{}]".format(s['txMsgQueueSize']))
|
|
if (s['rxMsgQueueSize'] or 0) > 1:
|
|
self.win.addstr(" [in window:\t{}]".format(s['rxMsgQueueSize']))
|
|
return y
|
|
|
|
def display_dht(self, y, data):
|
|
y += 2
|
|
self.win.move(y, 1)
|
|
self.win.addstr("DHT:")
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
self.win.addstr("introset lookups")
|
|
y = self.display_bucket(y, data["pendingIntrosetLookups"])
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
self.win.addstr("router lookups")
|
|
return self.display_bucket(y, data["pendingRouterLookups"])
|
|
|
|
def display_bucket(self, y, data):
|
|
txs = data["tx"]
|
|
self.win.addstr(" ({} lookups)".format(len(txs)))
|
|
for tx in txs:
|
|
y += 1
|
|
self.win.move(y, 1)
|
|
self.win.addstr("search for {}".format(tx["tx"]["target"]))
|
|
return y
|
|
|
|
def display_data(self):
|
|
"""draw main window"""
|
|
if self.data is not None:
|
|
self.win.addstr(1, 1, "lokinet online")
|
|
# print(self.data)
|
|
services = self.data["services"] or {}
|
|
y = 3
|
|
try:
|
|
y = self.display_links(y, self.data["links"])
|
|
for k in services:
|
|
y = self.display_service(y, k, services[k])
|
|
y = self.display_dht(y, self.data["dht"])
|
|
except Exception as exc:
|
|
pass
|
|
else:
|
|
self.win.move(1, 1)
|
|
self.win.addstr("lokinet offline")
|
|
|
|
def run(self):
|
|
while True:
|
|
self.win.clear()
|
|
self.win.box()
|
|
self.update_data()
|
|
self.display_data()
|
|
self.win.refresh()
|
|
time.sleep(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
|
|
mon = Monitor(
|
|
"http://{}/jsonrpc".format(
|
|
len(sys.argv) > 1 and sys.argv[1] or "127.0.0.1:1190"
|
|
)
|
|
)
|
|
mon.run()
|