l autowifi: move into package, make more robust

3 files changed, 239 insertions, 83 deletions

diff --git a/lass/3modules/autowifi.nix b/lass/3modules/autowifi.nix
index b84569df8..9aa1a2d28 100644
--- a/lass/3modules/autowifi.nix
+++ b/lass/3modules/autowifi.nix
@@ -11,101 +11,28 @@ in {
       type = types.str;
       default = "/etc/wifis";
     };
+    enablePrisonBreak = mkOption {
+      type = types.bool;
+      default = false;
+    };
   };
 
   config = lib.mkIf cfg.enable {
     systemd.services.autowifi = {
       description = "Automatic wifi connector";
       wantedBy = [ "multi-user.target" ];
+      path = [ pkgs.networkmanager ];
       serviceConfig = {
         Type = "simple";
         Restart = "always";
         RestartSec = "10s";
-        ExecStart = pkgs.writers.writePython3 "autowifi" {} /* python3 */ ''
-          import subprocess
-          import time
-          import urllib.request
-
-
-          def connect(ssid, psk=None):
-              subprocess.run(["${pkgs.networkmanager}/bin/nmcli", "connection", "delete", "autowifi"])
-              print("connecting to {}".format(ssid))
-              if psk is None:
-                  subprocess.run(["${pkgs.networkmanager}/bin/nmcli", "device", "wifi", "connect", ssid, "name", "autowifi"])
-              else:
-                  subprocess.run(["${pkgs.networkmanager}/bin/nmcli", "device", "wifi", "connect", ssid, "name", "autowifi", "password", psk])
-
-
-          def scan():
-              wifis_raw = subprocess.check_output(["${pkgs.networkmanager}/bin/nmcli", "-t", "device", "wifi", "list", "--rescan", "yes"])
-              wifis_list = wifis_raw.split(b'\n')
-              wifis = []
-              for line in wifis_list:
-                  ls = line.split(b':')
-                  if len(ls) == 8:
-                      wifis.append({"ssid": ls[1], "signal": int(ls[5]), "crypto": ls[7]})
-              return wifis
-
-
-          def get_known_wifis():
-              wifis_lines = []
-              with open('${cfg.knownWifisFile}') as f:
-                  wifis_lines = f.read().splitlines()
-              wifis = []
-              for line in wifis_lines:
-                  ls = line.split(':')
-                  wifis.append({"ssid": ls[0].encode(), "psk": ls[1].encode()})
-              return wifis
-
-
-          def check_internet():
-              try:
-                  beacon = urllib.request.urlopen('http://krebsco.de/secret')
-              except:  # noqa
-                  print("no internet")
-                  return False
-              if beacon.read() == b'1337\n':
-                  return True
-              print("no internet")
-              return False
-
-
-          def is_wifi_open(wifi):
-              if wifi['crypto'] == ${"b''"}:
-                  return True
-              else:
-                  return False
-
-
-          def is_wifi_seen(wifi, seen_wifis):
-              for seen_wifi in seen_wifis:
-                  if seen_wifi["ssid"] == wifi["ssid"]:
-                      return True
-              return False
-
-
-          def bloop():
-              while True:
-                  if not check_internet():
-                      wifis = scan()
-                      known_wifis = get_known_wifis()
-                      known_seen_wifis = [wifi for wifi in known_wifis if is_wifi_seen(wifi, wifis)]
-                      for wifi in known_seen_wifis:
-                          connect(wifi['ssid'], wifi['psk'])
-                          if check_internet():
-                              continue
-                      open_wifis = filter(is_wifi_open, wifis)
-                      for wifi in open_wifis:
-                          connect(wifi['ssid'])
-                          if check_internet():
-                              continue
-                  time.sleep(10)
-
-
-          bloop()
-        '';
+        ExecStart = "${autowifi}/bin/autowifi";
       };
     };
+
+    networking.networkmanager.dispatcherScripts = mkIf cfg.enablePrisonBreak [
+      { source = "${pkgs.callPackage <stockholm/makefu/5pkgs/prison-break}/bin/prison-break"; }
+    ];
   };
 }
 
diff --git a/lass/5pkgs/autowifi/autowifi.py b/lass/5pkgs/autowifi/autowifi.py
new file mode 100644
index 000000000..fa3d007e7
--- /dev/null
+++ b/lass/5pkgs/autowifi/autowifi.py
@@ -0,0 +1,228 @@
+import subprocess
+import time
+import urllib.request
+import logging
+import argparse
+import socket
+import struct
+import signal
+import os
+
+wifiDB = ''
+logger = logging.getLogger()
+got_signal = False
+
+
+def signal_handler(signum, frame):
+    global got_signal
+    got_signal = True
+
+
+def get_default_gateway() -> str:
+    """Read the default gateway directly from /proc."""
+    with open("/proc/net/route") as fh:
+        for line in fh:
+            fields = line.strip().split()
+            if fields[1] != '00000000' or not int(fields[3], 16) & 2:
+                continue
+
+            return socket.inet_ntoa(struct.pack("<L", int(fields[2], 16)))
+
+
+def connect(ssid, psk=None):
+    subprocess.run(
+        ["nmcli", "connection", "delete", "autowifi"],
+        stdout=subprocess.PIPE,
+    )
+    logging.info('connecting to %s', ssid)
+    if psk is None:
+        subprocess.run(
+            [
+                "nmcli",
+                "device",
+                "wifi",
+                "connect",
+                ssid,
+                "name",
+                "autowifi",
+            ],
+            stdout=subprocess.PIPE,
+        )
+    else:
+        subprocess.run(
+            [
+                "nmcli",
+                "device",
+                "wifi",
+                "connect",
+                ssid,
+                "name",
+                "autowifi",
+                "password",
+                psk,
+            ],
+            stdout=subprocess.PIPE,
+        )
+    time.sleep(5)
+
+
+def scan():
+    logging.debug('scanning wifis')
+    wifis_raw = subprocess.check_output([
+        "nmcli",
+        "-t",
+        "device",
+        "wifi",
+        "list",
+        "--rescan",
+        "yes",
+    ])
+    wifis_list = wifis_raw.split(b'\n')
+    logging.debug('scanning wifis finished')
+    wifis = []
+    for line in wifis_list:
+        logging.debug(line)
+        ls = line.split(b':')
+        if len(ls) == 8:
+            wifis.append({
+                "ssid": ls[1],
+                "signal": int(ls[5]),
+                "crypto": ls[7]
+            })
+    return wifis
+
+
+def get_known_wifis():
+    wifis_lines = []
+    with open(wifiDB) as f:
+        wifis_lines = f.read().splitlines()
+    wifis = []
+    for line in wifis_lines:
+        ls = line.split('/')
+        wifis.append({"ssid": ls[0].encode(), "psk": ls[1].encode()})
+    return wifis
+
+
+def check_network():
+    logging.debug('checking network')
+
+    global got_signal
+    if got_signal:
+        logging.info('got disconnect signal')
+        got_signal = False
+        return False
+    else:
+        gateway = get_default_gateway()
+        if gateway:
+            response = subprocess.run(
+                [
+                    'ping',
+                    '-q',
+                    '-c',
+                    '1',
+                    gateway,
+                ],
+                stdout=subprocess.PIPE,
+            )
+            if response.returncode == 0:
+                logging.debug('host %s is up', gateway)
+                return True
+            else:
+                logging.debug('host %s is down', gateway)
+                return False
+        else:
+            logging.debug('no gateway')
+            return False
+
+
+def check_internet():
+    logging.debug('checking internet')
+
+    try:
+        with open('./dummy_internet') as f:
+            dummy_content = f.read()
+            if dummy_content == 'xxx\n':
+                return True
+        beacon = urllib.request.urlopen('http://krebsco.de/secret')
+    except Exception as e:  # noqa
+        logging.debug(e)
+        logging.info('no internet exc')
+        return False
+    if beacon.read() == b'1337\n':
+        return True
+    logging.info('no internet oh')
+    return False
+
+
+def is_wifi_open(wifi):
+    if wifi['crypto'] == b'':
+        return True
+    else:
+        return False
+
+
+def is_wifi_seen(wifi, seen_wifis):
+    for seen_wifi in seen_wifis:
+        if seen_wifi["ssid"] == wifi["ssid"]:
+            return True
+    return False
+
+
+def main():
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument(
+        '-c', '--config',
+        dest='config',
+        help='wifi config file to use',
+        default='/etc/wifis',
+    )
+
+    parser.add_argument(
+        '-l', '--loglevel',
+        dest='loglevel',
+        help='loglevel to use',
+        default=logging.INFO,
+    )
+
+    parser.add_argument(
+        '-p', '--pidfile',
+        dest='pidfile',
+        help='file to write the pid to',
+        default=None,
+    )
+
+    args = parser.parse_args()
+
+    global wifiDB
+    wifiDB = args.config
+    logger.setLevel(args.loglevel)
+
+    signal.signal(signal.SIGUSR1, signal_handler)
+
+    if args.pidfile:
+        with open(args.pidfile, 'w+') as f:
+            f.write(str(os.getpid()))
+
+    while True:
+        if not check_network():
+            wifis = scan()
+            known_wifis = get_known_wifis()
+            known_seen_wifis = [
+                wifi for wifi in known_wifis if is_wifi_seen(wifi, wifis)
+            ]
+            for wifi in known_seen_wifis:
+                connect(wifi['ssid'], wifi['psk'])
+                if check_network():
+                    break
+            open_wifis = filter(is_wifi_open, wifis)
+            for wifi in open_wifis:
+                connect(wifi['ssid'])
+
+                if check_network():
+                    break
+        time.sleep(10)
+
+
+if __name__ == '__main__':
+    main()
diff --git a/lass/5pkgs/autowifi/default.nix b/lass/5pkgs/autowifi/default.nix
new file mode 100644
index 000000000..d565a6bb6
--- /dev/null
+++ b/lass/5pkgs/autowifi/default.nix
@@ -0,0 +1 @@
+pkgs.writers.writePython3Bin "autowifi" {} ./autowifi.py