Report results of ping

clients/report_ping

@@ -0,0 +1,82 @@
+#!/usr/bin/python3
+
+import argparse
+import hmac
+import json
+import os
+import pprint
+import random
+import socket
+import subprocess
+import time
+import re
+
+import requests
+
+ap = argparse.ArgumentParser()
+ap.add_argument("hostname")
+args = ap.parse_args()
+
+# It's a bit weird that this works.
+myhostname = socket.gethostbyaddr(socket.gethostname())[0]
+
+now = time.time()
+p = subprocess.run(["/bin/ping", "-c", "20", "-i", "0.2", "-n", args.hostname], capture_output=True, text=True)
+report0 = []
+for ln in p.stdout.split("\n")[:-1]:
+    if m := re.match(r"(\d+) packets transmitted, (\d+) received, (\d+)% packet loss", ln):
+        report0.append({ "measure": "ping_transmitted", "unit": "packets", "value": int(m.group(1)) })
+        report0.append({ "measure": "ping_received", "unit": "packets", "value": int(m.group(2)) })
+        report0.append({ "measure": "ping_loss", "unit": "%", "value": int(m.group(3)) })
+    elif m := re.match(r"rtt min/avg/max/mdev = ([0-9.]+)/([0-9.]+)/([0-9.]+)/([0-9.]+) ms", ln):
+        report0.append({ "measure": "ping_min_rtt", "unit": "s", "value": float(m.group(1)) / 1000 })
+        report0.append({ "measure": "ping_avg_rtt", "unit": "s", "value": float(m.group(2)) / 1000 })
+        report0.append({ "measure": "ping_max_rtt", "unit": "s", "value": float(m.group(3)) / 1000 })
+        report0.append({ "measure": "ping_mdev_rtt", "unit": "s", "value": float(m.group(4)) / 1000 })
+
+if len(report0) != 7:
+    print("Partial results:", report0)
+
+report = [
+    {
+        "description": {
+            "hostname": args.hostname,
+            "measure": r["measure"],
+            "unit": r["unit"]
+        },
+        "data": [
+            [now, r["value"]]
+        ]
+    }
+    for r in report0
+]
+
+for dir in (".", os.environ["HOME"] + "/.config/ltsdb", "/etc/ltsdb"):
+    try:
+        with open(dir + "/config.json") as fh:
+            client_config = json.load(fh)
+        baseurl = client_config["server"]
+        break
+    except FileNotFoundError:
+        pass
+while True:
+    for r in report:
+        node = myhostname
+        timestamp = time.time()
+        msg = (node + " " + str(timestamp)).encode("UTF-8")
+        digest = hmac.new(client_config["key"].encode("UTF-8"), msg, "SHA256").hexdigest()
+        r["auth"] = {
+            "node": node,
+            "timestamp": timestamp,
+            "hmac": digest,
+        }
+    #pprint.pp(report)
+    r = requests.post(baseurl + "report", json=report)
+    print(r)
+    if r.status_code == 200:
+        exit(0)
+    elif r.status_code == 409:
+        time.sleep(0.5 + random.random())
+        continue
+    else:
+        exit(1)