Add internetcheck.py.

internetcheck.py

@@ -0,0 +1,143 @@
+import os
+import socket
+import sqlite3
+import subprocess
+import sys
+import time
+
+from voussoirkit import hms
+
+sql = sqlite3.connect('internetcheck.db')
+cur = sql.cursor()
+
+DB_INIT = '''
+BEGIN;
+CREATE TABLE IF NOT EXISTS outages(
+    timestamp INT PRIMARY KEY,
+    human TEXT,
+    lan_ok INT,
+    dns_ok INT,
+    ip_ok INT,
+    duration INT
+);
+COMMIT;
+'''
+cur.executescript(DB_INIT)
+
+down = False
+outage_started = None
+
+def percentage(items):
+    trues = sum(bool(i) for i in items)
+    return trues / len(items)
+
+def ping_lan():
+    try:
+        socket.setdefaulttimeout(1)
+        socket.socket(socket.AF_INET, socket.SOCK_STREAM).connect(('192.168.1.1', 80))
+        return True
+    except socket.error:
+        return False
+
+def check_dns():
+    def check(domain):
+        # print(domain)
+        try:
+            socket.getaddrinfo(domain, 53)
+            return True
+        except socket.gaierror:
+            return False
+    names = [
+        'amazon.com',
+        'bing.com',
+        'comcast.com',
+        'frontier.com',
+        'google.com',
+        'microsoft.com',
+        'netflix.com',
+        'reddit.com',
+        'spectrum.com',
+        'youtube.com',
+    ]
+    checks = [check(name) for name in names]
+    return percentage(checks)
+
+def check_ip():
+    def check(ip):
+        # print(ip)
+        try:
+            socket.setdefaulttimeout(2)
+            socket.socket(socket.AF_INET, socket.SOCK_STREAM).connect(('104.43.253.214', 80))
+            return True
+        except socket.error:
+            return False
+    ips = [
+        '104.43.253.214',
+        '13.107.21.200',
+        '13.77.161.179',
+        '142.136.81.136',
+        '151.101.1.140',
+        '172.217.11.174',
+        '172.217.11.78',
+        '176.32.103.205',
+        '35.165.194.49',
+        '69.252.80.75',
+    ]
+    checks = [check(ip) for ip in ips]
+    return percentage(checks)
+
+def set_down(lan_ok, dns_ok, ip_ok):
+    global down
+    global outage_started
+    if down:
+        duration = int(time.time() - outage_started)
+        duration = hms.seconds_to_hms(duration)
+        print(f'Still down for {duration}.')
+        return
+    print('Starting down timer.')
+    down = True
+    outage_started = int(time.time())
+    human = time.strftime('%Y-%m-%d %H:%M:%S')
+    cur.execute(
+        'INSERT INTO outages VALUES(?, ?, ?, ?, ?, ?)',
+        [outage_started, human, lan_ok, dns_ok, ip_ok, None]
+    )
+    sql.commit()
+
+def end_down():
+    global down
+    global outage_started
+    if not down:
+        return
+    duration = int(time.time() - outage_started)
+    cur.execute('UPDATE outages SET duration = ? WHERE timestamp == ?', [duration, outage_started])
+    sql.commit()
+    down = False
+    outage_started = None
+
+def check_forever():
+    while True:
+        now = time.strftime('%Y-%m-%d %H:%M:%S')
+        lan_ok = int(ping_lan())
+        dns_stat = check_dns()
+        ip_stat = check_ip()
+        message = f'{now}, LAN={lan_ok}, DNS={dns_stat:0.2f}, IP={ip_stat:0.2f}'
+        dns_ok = dns_stat > 0.75
+        ip_ok = ip_stat > 0.75
+        if not (lan_ok and dns_ok and ip_ok):
+            print(message, end=', ')
+            set_down(lan_ok, dns_ok, ip_ok)
+        else:
+            print(message + '\r', end='', flush=True)
+            end_down()
+
+        if down:
+            time.sleep(1)
+        else:
+            time.sleep(20)
+
+def main(argv):
+    check_forever()
+
+if __name__ == '__main__':
+    raise SystemExit(main(sys.argv[1:]))