Add ping tester

3 files changed, 116 insertions, 0 deletions

diff --git a/ping_tester/Dockerfile b/ping_tester/Dockerfile
new file mode 100644
index 0000000..702b596
--- /dev/null
+++ b/ping_tester/Dockerfile
@@ -0,0 +1,9 @@
+FROM alpine:latest
+
+RUN set -euxo pipefail; \
+    apk --no-cache add python3; \
+    python3 -m pip install boto3;
+
+COPY ping_test.py /usr/bin/
+
+ENTRYPOINT [ "/usr/bin/ping_test.py" ]
diff --git a/ping_tester/Makefile b/ping_tester/Makefile
new file mode 100644
index 0000000..8e15cbe
--- /dev/null
+++ b/ping_tester/Makefile
@@ -0,0 +1,11 @@
+IMAGE=docker.crute.me/ping_tester:latest
+
+all:
+        docker build -t $(IMAGE) .
+
+all-no-cache:
+        docker build --no-cache -t $(IMAGE) .
+
+publish:
+        docker push $(IMAGE)
+
diff --git a/ping_tester/ping_test.py b/ping_tester/ping_test.py
new file mode 100755
index 0000000..6e19b3f
--- /dev/null
+++ b/ping_tester/ping_test.py
@@ -0,0 +1,96 @@
+#!/usr/bin/env python3
+
+import os
+import re
+import sys
+import boto3
+import subprocess
+from datetime import datetime
+
+
+def main(sample_count=5):
+    try:
+        _, from_location, to_location, hostname = sys.argv
+    except ValueError:
+        print("usage: {} <this_location> <to_location> <hostname>".format(
+            os.path.basename(sys.argv[0])))
+        sys.exit(1)
+
+    client = boto3.client("cloudwatch")
+    now = datetime.now()
+
+    patt = re.compile(
+        "round-trip min/avg/max = "
+        "(?P<min>[0-9]+\.[0-9]+)/(?P<avg>[0-9]+\.[0-9]+)/"
+        "(?P<max>[0-9]+\.[0-9]+) (?P<unit>.*)")
+
+    out = subprocess.run(
+        ["ping", "-c", str(sample_count), hostname],
+        stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+    val = patt.search(out.stdout.decode("us-ascii")).groupdict()
+
+    client.put_metric_data(
+        Namespace="VPNLatency",
+        MetricData=[
+            {
+                "MetricName": "PingRTT",
+                "Dimensions": [
+                    {
+                        "Name": "From Location",
+                        "Value": from_location,
+                    },
+                    {
+                        "Name": "To Location",
+                        "Value": to_location,
+                    }
+                ],
+                "Timestamp": now,
+                "StatisticValues": {
+                    "SampleCount": sample_count,
+                    "Sum": float(val["avg"]) * sample_count,
+                    "Minimum": float(val["min"]),
+                    "Maximum": float(val["max"]),
+                },
+                "Unit": "Milliseconds"
+            },
+            {
+                "MetricName": "PingRTT",
+                "Dimensions": [
+                    {
+                        "Name": "From Location",
+                        "Value": from_location,
+                    },
+                ],
+                "Timestamp": now,
+                "StatisticValues": {
+                    "SampleCount": sample_count,
+                    "Sum": float(val["avg"]) * sample_count,
+                    "Minimum": float(val["min"]),
+                    "Maximum": float(val["max"]),
+                },
+                "Unit": "Milliseconds"
+            },
+            {
+                "MetricName": "PingRTT",
+                "Dimensions": [
+                    {
+                        "Name": "To Location",
+                        "Value": to_location,
+                    }
+                ],
+                "Timestamp": now,
+                "StatisticValues": {
+                    "SampleCount": sample_count,
+                    "Sum": float(val["avg"]) * sample_count,
+                    "Minimum": float(val["min"]),
+                    "Maximum": float(val["max"]),
+                },
+                "Unit": "Milliseconds"
+            },
+        ]
+    )
+
+
+if __name__ == "__main__":
+    main()