Merge pull request #46 from ffac/legacy

Fixes a bug that will add a node with upercase letters in its id to the ...

.gitignore

@@ -1,8 +1,3 @@
-# script-generated
-aliases*.json
-nodedb/
-
-# python bytecode / cache
 *.pyc
-pycache/
+aliases.json
-__pycache__/
+nodedb/

.travis.yml

@@ -1,6 +0,0 @@
-sudo: false
-language: python
-python:
-  - "3.4"
-install: "pip install pep8"
-script: "pep8 --ignore=E501 *.py lib/*.py"

GlobalRRD.py

@@ -1,8 +1,6 @@
 import os
 import subprocess
-
+from RRD import RRD, DS, RRA
-from lib.RRD import DS, RRA, RRD
-
 
 class GlobalRRD(RRD):
     ds_list = [
@@ -12,21 +10,17 @@ class GlobalRRD(RRD):
         DS('clients', 'GAUGE', 120, 0, float('NaN')),
     ]
     rra_list = [
-        # 2 hours of 1 minute samples
+        RRA('AVERAGE', 0.5, 1, 120),    #  2 hours of 1 minute samples
-        RRA('AVERAGE', 0.5, 1, 120),
+        RRA('AVERAGE', 0.5, 60, 744),   # 31 days  of 1 hour   samples
-        # 31 days  of 1 hour samples
+        RRA('AVERAGE', 0.5, 1440, 1780),# ~5 years of 1 day    samples
-        RRA('AVERAGE', 0.5, 60, 744),
-        # ~5 years of 1 day samples
-        RRA('AVERAGE', 0.5, 1440, 1780),
     ]
 
     def __init__(self, directory):
         super().__init__(os.path.join(directory, "nodes.rrd"))
-        self.ensure_sanity(self.ds_list, self.rra_list, step=60)
+        self.ensureSanity(self.ds_list, self.rra_list, step=60)
 
-    # TODO: fix this, python does not support function overloading
+    def update(self, nodeCount, clientCount):
-    def update(self, node_count, client_count):
+        super().update({'nodes': nodeCount, 'clients': clientCount})
-        super().update({'nodes': node_count, 'clients': client_count})
 
     def graph(self, filename, timeframe):
         args = ["rrdtool", 'graph', filename,
@@ -36,5 +30,6 @@ class GlobalRRD(RRD):
                 'DEF:nodes=' + self.filename + ':nodes:AVERAGE',
                 'LINE1:nodes#F00:nodes\\l',
                 'DEF:clients=' + self.filename + ':clients:AVERAGE',
-                'LINE2:clients#00F:clients']
+                'LINE2:clients#00F:clients',
+        ]
         subprocess.check_output(args)

NodeRRD.py

@@ -1,8 +1,7 @@
 import os
 import subprocess
-
+from node import Node
-from lib.RRD import DS, RRA, RRD
+from RRD import RRD, DS, RRA
-
 
 class NodeRRD(RRD):
     ds_list = [
@@ -10,14 +9,10 @@ class NodeRRD(RRD):
         DS('clients', 'GAUGE', 120, 0, float('NaN')),
     ]
     rra_list = [
-        # 2 hours of  1 minute samples
+        RRA('AVERAGE', 0.5, 1, 120),    #  2 hours of  1 minute samples
-        RRA('AVERAGE', 0.5, 1, 120),
+        RRA('AVERAGE', 0.5, 5, 1440),   #  5 days  of  5 minute samples
-        #  5 days  of  5 minute samples
+        RRA('AVERAGE', 0.5, 60, 720),   # 30 days  of  1 hour   samples
-        RRA('AVERAGE', 0.5, 5, 1440),
+        RRA('AVERAGE', 0.5, 720, 730),  #  1 year  of 12 hour   samples
-        # 30 days  of  1 hour   samples
-        RRA('AVERAGE', 0.5, 60, 720),
-        #  1 year  of 12 hour   samples
-        RRA('AVERAGE', 0.5, 720, 730),
     ]
 
     def __init__(self, filename, node = None):
@@ -28,17 +23,14 @@ class NodeRRD(RRD):
         """
         self.node = node
         super().__init__(filename)
-        self.ensure_sanity(self.ds_list, self.rra_list, step=60)
+        self.ensureSanity(self.ds_list, self.rra_list, step=60)
 
     @property
     def imagename(self):
-        return "{basename}.png".format(
+        return os.path.basename(self.filename).rsplit('.', 2)[0] + ".png"
-            basename=os.path.basename(self.filename).rsplit('.', 2)[0])
 
-    # TODO: fix this, python does not support function overloading
     def update(self):
-        super().update({'upstate': int(self.node['flags']['online']),
+        super().update({'upstate': 1, 'clients': self.node.clientcount})
-                        'clients': self.node['statistics']['clients']})
 
     def graph(self, directory, timeframe):
         """
@@ -57,5 +49,6 @@ class NodeRRD(RRD):
                 'CDEF:d=clients,UN,maxc,UN,1,maxc,IF,*',
                 'AREA:c#0F0:up\\l',
                 'AREA:d#F00:down\\l',
-                'LINE1:c#00F:clients connected\\l']
+                'LINE1:c#00F:clients connected\\l',
+                ]
         subprocess.check_output(args)

README.md

@@ -1,117 +1,53 @@
 # Data for Freifunk Map, Graph and Node List
 
-[![Build Status](https://travis-ci.org/ffnord/ffmap-backend.svg?branch=master)](https://travis-ci.org/ffnord/ffmap-backend)
+ffmap-backend gathers information on the batman network by invoking 
+   batctl
+and
+   batadv-vis
+as root (via sudo) and has this information placed into a target directory
+as the file "nodes.json" and also updates the directory "nodes" with graphical
+representations of uptimes and the number of clients connecting.
 
-ffmap-backend gathers information on the batman network by invoking :
+The target directory is suggested to host all information for interpreting those
-
+node descriptions, e.g. as provided by https://github.com/ffnord/ffmap-d3.git .
- * batctl (might require root),
+When executed without root privileges, we suggest to grant sudo permissions
- * alfred-json and
+within wrappers of those binaries, so no further changes are required in other
- * batadv-vis
+scripts:
-
-The output will be written to a directory (`-d output`).
-
-Run `backend.py --help` for a quick overview of all available options.
-
-For the script's regular execution add the following to the crontab:
 
 <pre>
-* * * * * backend.py -d /path/to/output -a /path/to/aliases.json --vpn ae:7f:58:7d:6c:2a d2:d0:93:63:f7:da
+$ cat <<EOCAT > $HOME/batctl
+#!/bin/sh
+exec sudo /usr/sbin/batctl $*
+EOCAT
 </pre>
 
-# Dependencies
+and analogously for batadv-vis. The entry for /etc/sudoers could be
+whateveruser   ALL=(ALL:ALL) NOPASSWD: /usr/sbin/batctl,/usr/sbin/batadv-vis,/usr/sbin/alfred-json
 
-- Python 3
+The destination directory can be made directly available through apache:
-- Python 3 Package [Networkx](https://networkx.github.io/)
+<pre>
-- [alfred-json](https://github.com/tcatm/alfred-json)
+$ cat /etc/apache2/site-enabled/000-default
-- rrdtool (if run with `--with-rrd`)
+...
-
+        <Directory /home/whateverusername/www/>
-# Running as unprivileged user
+                Options Indexes FollowSymLinks MultiViews
-
+                AllowOverride None
-Some information collected by ffmap-backend requires access to specific system resources.
+                Order allow,deny
-
+                allow from all
-Make sure the user you are running this under is part of the group that owns the alfred socket, so
+        </Directory>
-alfred-json can access the alfred daemon.
-
-    # ls -al /var/run/alfred.sock
-    srw-rw---- 1 root alfred 0 Mar 19 22:00 /var/run/alfred.sock=
-    # adduser map alfred
-    Adding user `map' to group `alfred' ...
-    Adding user map to group alfred
-    Done.
-    $ groups
-    map alfred
-
-Running batctl requires passwordless sudo access, because it needs to access the debugfs to retrive
-the gateway list.
-
-    # echo 'map ALL = NOPASSWD: /usr/sbin/batctl' | tee /etc/sudoers.d/map
-    map ALL = NOPASSWD: /usr/sbin/batctl
-    # chmod 0440 /etc/sudoers.d/map
-
-That should be everything. The script automatically detects if it is run in unprivileged mode and
-will prefix `sudo` where necessary.
-
-# Data format
-
-## nodes.json
-
-    { 'nodes': {
-        node_id: { 'flags': { flags },
-                   'firstseen': isoformat,
-                   'lastseen': isoformat,
-                   'nodeinfo': {...},         # copied from alfred type 158
-                   'statistics': {
-                      'uptime': double,       # seconds
-                      'memory_usage': double, # 0..1
-                      'clients': double,
-                      'rootfs_usage': double, # 0..1
-                      'loadavg': double,
-                      'gateway': mac
-                    }
-                 },
 ...
-      }
-      'timestamp': isoformat
-    }
 
-### flags (bool)
+$ cat /etc/apache2/conf.d/freifunk
+Alias /map /home/ffmap/www/
+Alias /firmware /home/freifunk/autoupdates/
+</pre>
 
-- online
+To execute, run 
-- gateway
+ ./mkmap.sh ../www
+The script expects above described sudo-wrappers in the $HOME directory of the user executing
+the script. If those are not available, an error will occurr if not executed as root. Also,
+the tool realpath optionally allows to execute the script from anywhere in the directory tree.
 
-## Old data format
+For the script's regular execution add the following to the crontab:
-
+<pre>
-If you want to still use the old [ffmap-d3](https://github.com/ffnord/ffmap-d3)
+*/5 * * * * /home/ffmap/ffmap-backend/mkmap.sh /home/ffmap/www
-front end, you can use the file `ffmap-d3.jq` to convert the new output to the
+</pre>
-old one:
-
-```
-jq -n -f ffmap-d3.jq \
-    --argfile nodes nodedb/nodes.json \
-    --argfile graph nodedb/graph.json \
-    > nodedb/ffmap-d3.json
-```
-
-Then point your ffmap-d3 instance to the `ffmap-d3.json` file.
-
-# Removing owner information
-
-If you'd like to redact information about the node owner from `nodes.json`,
-you may use a filter like [jq]. In this case, specify an output directory
-different from your webserver directory, e.g.:
-
-    ./backend.py -d /ffmap-data
-
-Don't write to files generated in there. ffmap-backend uses them as its
-database.
-
-After running ffmap-backend, copy `graph.json` to your webserver. Then,
-filter `nodes.json` using `jq` like this:
-
-     jq '.nodes = (.nodes | with_entries(del(.value.nodeinfo.owner)))' \
-       < /ffmap-data/nodes.json > /var/www/data/nodes.json
-
-This will remove owner information from nodes.json before copying the data
-to your webserver.
-
-[jq]: https://stedolan.github.io/jq/

RRD.py

@@ -1,20 +1,19 @@
 import subprocess
 import re
+import io
 import os
+from tempfile import TemporaryFile
 from operator import xor, eq
 from functools import reduce
 from itertools import starmap
 import math
 
-
 class RRDIncompatibleException(Exception):
     """
     Is raised when an RRD doesn't have the desired definition and cannot be
     upgraded to it.
     """
     pass
-
-
 class RRDOutdatedException(Exception):
     """
     Is raised when an RRD doesn't have the desired definition, but can be
@@ -26,8 +25,7 @@ if not hasattr(__builtins__, "FileNotFoundError"):
     class FileNotFoundError(Exception):
         pass
 
-
+class RRD:
-class RRD(object):
     """
     An RRD is a Round Robin Database, a database which forgets old data and
     aggregates multiple records into new ones.
@@ -59,7 +57,7 @@ class RRD(object):
     def __init__(self, filename):
         self.filename = filename
 
-    def ensure_sanity(self, ds_list, rra_list, **kwargs):
+    def ensureSanity(self, ds_list, rra_list, **kwargs):
         """
         Create or upgrade the RRD file if necessary to contain all DS in
         ds_list. If it needs to be created, the RRAs in rra_list and any kwargs
@@ -67,13 +65,13 @@ class RRD(object):
         database are NOT modified!
         """
         try:
-            self.check_sanity(ds_list)
+            self.checkSanity(ds_list)
         except FileNotFoundError:
             self.create(ds_list, rra_list, **kwargs)
         except RRDOutdatedException:
             self.upgrade(ds_list)
 
-    def check_sanity(self, ds_list=()):
+    def checkSanity(self, ds_list=()):
         """
         Check if the RRD file exists and contains (at least) the DS listed in
         ds_list.
@@ -83,11 +81,8 @@ class RRD(object):
         info = self.info()
         if set(ds_list) - set(info['ds'].values()) != set():
             for ds in ds_list:
-                if ds.name in info['ds'] and\
+                if ds.name in info['ds'] and ds.type != info['ds'][ds.name].type:
-                   ds.type != info['ds'][ds.name].type:
+                    raise RRDIncompatibleException("%s is %s but should be %s" % (ds.name, ds.type, info['ds'][ds.name].type))
-                    raise RRDIncompatibleException(
-                        "{} is {} but should be {}".format(
-                            ds.name, ds.type, info['ds'][ds.name].type))
             else:
                 raise RRDOutdatedException()
 
@@ -110,10 +105,8 @@ class RRD(object):
             if ds.name in info['ds']:
                 old_ds = info['ds'][ds.name]
                 if info['ds'][ds.name].type != ds.type:
-                    raise RuntimeError(
+                    raise RuntimeError('Cannot convert existing DS "%s" from type "%s" to "%s"' %
-                        "Cannot convert existing DS '{}'"
+                        (ds.name, old_ds.type, ds.type))
-                        "from type '{}' to '{}'".format(
-                            ds.name, old_ds.type, ds.type))
                 ds.index = old_ds.index
                 new_ds[ds.index] = ds
             else:
@@ -123,11 +116,12 @@ class RRD(object):
 
         dump = subprocess.Popen(
             ["rrdtool", "dump", self.filename],
-            stdout=subprocess.PIPE)
+            stdout=subprocess.PIPE
-
+        )
         restore = subprocess.Popen(
             ["rrdtool", "restore", "-", self.filename + ".new"],
-            stdin=subprocess.PIPE)
+            stdin=subprocess.PIPE
+        )
         echo = True
         ds_definitions = True
         for line in dump.stdout:
@@ -149,14 +143,16 @@ class RRD(object):
                            <value>%s</value>
                            <unknown_sec> %i </unknown_sec>
                         </ds>
-                        """ % (ds.name,
+                        """ % (
+                            ds.name,
                             ds.type,
                             ds.args[0],
                             ds.args[1],
                             ds.args[2],
                             ds.last_ds,
                             ds.value,
-                               ds.unknown_sec), "utf-8"))
+                            ds.unknown_sec)
+                    , "utf-8"))
 
             if b'</cdp_prep>' in line:
                 restore.stdin.write(added_ds_num*b"""
@@ -241,8 +237,7 @@ class RRD(object):
         for line in out.splitlines():
             base = info
             for match in self._info_regex.finditer(line):
-                section, key, name, value = match.group(
+                section, key, name, value = match.group("section", "key", "name", "value")
-                    "section", "key", "name", "value")
                 if section and key:
                     try:
                         key = int(key)
@@ -263,8 +258,7 @@ class RRD(object):
                             base[name] = value
         dss = {}
         for name, ds in info['ds'].items():
-            ds_obj = DS(name, ds['type'], ds['minimal_heartbeat'],
+            ds_obj = DS(name, ds['type'], ds['minimal_heartbeat'], ds['min'], ds['max'])
-                        ds['min'], ds['max'])
             ds_obj.index = ds['index']
             ds_obj.last_ds = ds['last_ds']
             ds_obj.value = ds['value']
@@ -273,14 +267,12 @@ class RRD(object):
         info['ds'] = dss
         rras = []
         for rra in info['rra'].values():
-            rras.append(RRA(rra['cf'], rra['xff'],
+            rras.append(RRA(rra['cf'], rra['xff'], rra['pdp_per_row'], rra['rows']))
-                            rra['pdp_per_row'], rra['rows']))
         info['rra'] = rras
         self._cached_info = info
         return info
 
-
+class DS:
-class DS(object):
     """
     DS stands for Data Source and represents one line of data points in a Round
     Robin Database (RRD).
@@ -292,7 +284,6 @@ class DS(object):
     last_ds = 'U'
     value = 0
     unknown_sec = 0
-
     def __init__(self, name, dst, *args):
         self.name = name
         self.type = dst
@@ -302,7 +293,7 @@ class DS(object):
         return "DS:%s:%s:%s" % (
             self.name,
             self.type,
-            ":".join(map(str, self._nan_to_u_args()))
+            ":".join(map(str, self._nan_to_U_args()))
         )
 
     def __repr__(self):
@@ -314,23 +305,22 @@ class DS(object):
         )
 
     def __eq__(self, other):
-        return all(starmap(eq, zip(self.compare_keys(), other.compare_keys())))
+        return all(starmap(eq, zip(self._compare_keys(), other._compare_keys())))
 
     def __hash__(self):
-        return reduce(xor, map(hash, self.compare_keys()))
+        return reduce(xor, map(hash, self._compare_keys()))
 
-    def _nan_to_u_args(self):
+    def _nan_to_U_args(self):
         return tuple(
             'U' if type(arg) is float and math.isnan(arg)
             else arg
             for arg in self.args
         )
 
-    def compare_keys(self):
+    def _compare_keys(self):
-        return self.name, self.type, self._nan_to_u_args()
+        return (self.name, self.type, self._nan_to_U_args())
 
-
+class RRA:
-class RRA(object):
     def __init__(self, cf, *args):
         self.cf = cf
         self.args = args

alfred.py

@@ -0,0 +1,42 @@
+#!/usr/bin/env python3
+import subprocess
+import json
+
+class alfred:
+  def __init__(self,request_data_type = 158):
+    self.request_data_type = request_data_type
+
+  def aliases(self):
+    output = subprocess.check_output(["alfred-json","-r",str(self.request_data_type),"-f","json","-z"])
+    alfred_data = json.loads(output.decode("utf-8"))
+    alias = {}
+    for mac,node in alfred_data.items():
+      node_alias = {}
+      if 'location' in node:
+        try:
+          node_alias['gps'] = str(node['location']['latitude']) + ' ' + str(node['location']['longitude'])
+        except:
+          pass
+
+      try:
+        node_alias['firmware'] = node['software']['firmware']['release']
+      except KeyError:
+        pass
+
+      try:
+        node_alias['id'] = node['network']['mac']
+      except KeyError:
+        pass
+
+      if 'hostname' in node:
+        node_alias['name'] = node['hostname']
+      elif 'name' in node:
+        node_alias['name'] = node['name']
+      if len(node_alias):
+        alias[mac] = node_alias
+    return alias
+
+if __name__ == "__main__":
+  ad = alfred()
+  al = ad.aliases()
+  print(al)

alfred_merge.py

@@ -1,42 +0,0 @@
-#!/usr/bin/env python3
-import subprocess
-import json
-
-from collections import MutableMapping
-
-def rec_merge(d1, d2):
-    '''
-    Update two dicts of dicts recursively, 
-    if either mapping has leaves that are non-dicts, 
-    the second's leaf overwrites the first's.
-    '''
-    for k, v in d1.items(): # in Python 2, use .iteritems()!
-        if k in d2:
-            # this next check is the only difference!
-            if all(isinstance(e, MutableMapping) for e in (v, d2[k])):
-                d2[k] = rec_merge(v, d2[k])
-            # we could further check types and merge as appropriate here.
-    d3 = d1.copy()
-    d3.update(d2)
-    return d3
-
-
-class alfred_merge:
-  def __init__(self,request_data_type_1 = 158, request_data_type_2 = 159):
-    self.request_data_type_1 = request_data_type_1
-    self.request_data_type_2 = request_data_type_2
-
-  def aliases(self):
-    output = subprocess.check_output(["/usr/local/bin/alfred-json","-z", "-r",str(self.request_data_type_1),"-f","json"])
-    alfred_data_1 = json.loads(output.decode("utf-8"))
-    output = subprocess.check_output(["/usr/local/bin/alfred-json","-z", "-r",str(self.request_data_type_2),"-f","json"])
-    alfred_data_2 = json.loads(output.decode("utf-8"))
-    
-    return json.dumps(rec_merge(alfred_data_1, alfred_data_2))
-
-
-if __name__ == "__main__":
-  ad = alfred_merge()
-  al = ad.aliases()
-  print(al)
-

aliases.json_sample

@@ -1,36 +1,9 @@
-[
 {
-    "node_id": "krtek",
+   "b0:48:7a:e7:d3:64" : {
-    "hostname": "krtek",
+      "name" : "Meute-AP"
-    "location": {
-      "longitude": 10.74,
-      "latitude": 53.86
    },
-    "network": {
+   "8e:3d:c2:10:10:28" : {
-      "mesh": {
+      "name" : "holstentor",
-        "bat0": {
+      "vpn" : true
-          "interfaces": {
-            "tunnel": [
-              "00:25:86:e6:f1:bf"
-            ]
    }
 }
-      }
-    }
-  },
-  {
-    "node_id": "gw1",
-    "hostname": "burgtor",
-    "network": {
-      "mesh": {
-        "bat0": {
-          "interfaces": {
-            "tunnel": [
-              "52:54:00:f3:62:d9"
-            ]
-          }
-        }
-      }
-    }
-  }
-]

backend.py

@@ -1,194 +0,0 @@
-#!/usr/bin/env python3
-"""
-backend.py - ffmap-backend runner
-https://github.com/ffnord/ffmap-backend
-"""
-import argparse
-import json
-import os
-import sys
-from datetime import datetime
-
-import networkx as nx
-from networkx.readwrite import json_graph
-
-from lib import graph, nodes
-from lib.alfred import Alfred
-from lib.batman import Batman
-from lib.rrddb import RRD
-from lib.nodelist import export_nodelist
-from lib.validate import validate_nodeinfos
-
-NODES_VERSION = 1
-GRAPH_VERSION = 1
-
-
-def main(params):
-    os.makedirs(params['dest_dir'], exist_ok=True)
-
-    nodes_fn = os.path.join(params['dest_dir'], 'nodes.json')
-    tmp_nodes_fn = os.path.join(params['dest_dir'], 'nodes.json.tmp')
-
-    graph_fn = os.path.join(params['dest_dir'], 'graph.json')
-    tmp_graph_fn = os.path.join(params['dest_dir'], 'graph.json.tmp')
-
-    nodelist_fn = os.path.join(params['dest_dir'], 'nodelist.json')
-    tmp_nodelist_fn = os.path.join(params['dest_dir'], 'nodelist.json.tmp')
-
-    now = datetime.utcnow().replace(microsecond=0)
-
-    # parse mesh param and instantiate Alfred/Batman instances
-    alfred_instances = []
-    batman_instances = []
-    for value in params['mesh']:
-        # (1) only batman-adv if, no alfred sock
-        if ':' not in value:
-            if len(params['mesh']) > 1:
-                raise ValueError(
-                    'Multiple mesh interfaces require the use of '
-                    'alfred socket paths.')
-            alfred_instances.append(Alfred(unix_sockpath=None))
-            batman_instances.append(Batman(mesh_interface=value))
-        else:
-            # (2) batman-adv if + alfred socket
-            try:
-                batif, alfredsock = value.split(':')
-                alfred_instances.append(Alfred(unix_sockpath=alfredsock))
-                batman_instances.append(Batman(mesh_interface=batif,
-                                               alfred_sockpath=alfredsock))
-            except ValueError:
-                raise ValueError(
-                    'Unparseable value "{0}" in --mesh parameter.'.
-                    format(value))
-
-    # read nodedb state from node.json
-    try:
-        with open(nodes_fn, 'r') as nodedb_handle:
-            nodedb = json.load(nodedb_handle)
-    except IOError:
-        nodedb = {'nodes': dict()}
-
-    # flush nodedb if it uses the old format
-    if 'links' in nodedb:
-        nodedb = {'nodes': dict()}
-
-    # set version we're going to output
-    nodedb['version'] = NODES_VERSION
-
-    # update timestamp and assume all nodes are offline
-    nodedb['timestamp'] = now.isoformat()
-    for node_id, node in nodedb['nodes'].items():
-        node['flags']['online'] = False
-
-    # integrate alfred nodeinfo
-    for alfred in alfred_instances:
-        nodeinfo = validate_nodeinfos(alfred.nodeinfo())
-        nodes.import_nodeinfo(nodedb['nodes'], nodeinfo,
-                              now, assume_online=True)
-
-    # integrate static aliases data
-    for aliases in params['aliases']:
-        with open(aliases, 'r') as f:
-#           nodeinfo = validate_nodeinfos(json.load(f))
-            nodes.import_nodeinfo(nodedb['nodes'], json.load(f),
-                                  now, assume_online=False, statics=True)
-
-    nodes.reset_statistics(nodedb['nodes'])
-    for alfred in alfred_instances:
-        nodes.import_statistics(nodedb['nodes'], alfred.statistics())
-
-    # acquire gwl and visdata for each batman instance
-    mesh_info = []
-    for batman in batman_instances:
-        vd = batman.vis_data()
-        gwl = batman.gateway_list()
-        mesh_info.append((vd, gwl))
-
-    # update nodedb from batman-adv data
-    for vd, gwl in mesh_info:
-        nodes.import_mesh_ifs_vis_data(nodedb['nodes'], vd)
-        nodes.import_vis_clientcount(nodedb['nodes'], vd)
-        nodes.mark_vis_data_online(nodedb['nodes'], vd, now)
-        nodes.mark_gateways(nodedb['nodes'], gwl)
-
-    # clear the nodedb from nodes that have not been online in $prune days
-    if params['prune']:
-        nodes.prune_nodes(nodedb['nodes'], now, params['prune'])
-
-    # build nxnetworks graph from nodedb and visdata
-    batadv_graph = nx.DiGraph()
-    for vd, gwl in mesh_info:
-        graph.import_vis_data(batadv_graph, nodedb['nodes'], vd)
-
-    # force mac addresses to be vpn-link only (like gateways for example)
-    if params['vpn']:
-        graph.mark_vpn(batadv_graph, frozenset(params['vpn']))
-
-    def extract_tunnel(nodes):
-        macs = set()
-        for id, node in nodes.items():
-            try:
-                for mac in node["nodeinfo"]["network"]["mesh"]["bat0"]["interfaces"]["tunnel"]:
-                    macs.add(mac)
-                for mac in node["nodeinfo"]["network"]["mesh"]["bat-ffhh"]["interfaces"]["tunnel"]:
-                    macs.add(mac)
-            except KeyError:
-                pass
-
-        return macs
-
-    graph.mark_vpn(batadv_graph, extract_tunnel(nodedb['nodes']))
-
-    batadv_graph = graph.merge_nodes(batadv_graph)
-    batadv_graph = graph.to_undirected(batadv_graph)
-
-    # write processed data to dest dir
-    with open(tmp_nodes_fn, 'w') as f:
-        json.dump(nodedb, f)
-
-    graph_out = {'batadv': json_graph.node_link_data(batadv_graph),
-                 'version': GRAPH_VERSION}
-
-    with open(tmp_graph_fn, 'w') as f:
-        json.dump(graph_out, f)
-
-    with open(tmp_nodelist_fn, 'w') as f:
-        json.dump(export_nodelist(now, nodedb), f)
-
-    os.rename(tmp_nodes_fn, nodes_fn)
-    os.rename(tmp_graph_fn, graph_fn)
-    os.rename(tmp_nodelist_fn, nodelist_fn)
-
-    # optional rrd graphs (trigger with --rrd)
-    if params['rrd']:
-        script_directory = os.path.dirname(os.path.realpath(__file__))
-        rrd = RRD(os.path.join(script_directory, 'nodedb'),
-                  os.path.join(params['dest_dir'], 'nodes'))
-        rrd.update_database(nodedb['nodes'])
-        rrd.update_images()
-
-if __name__ == '__main__':
-    parser = argparse.ArgumentParser()
-
-    parser.add_argument('-a', '--aliases',
-                        help='Read aliases from FILE',
-                        nargs='+', default=[], metavar='FILE')
-    parser.add_argument('-m', '--mesh',
-                        default=['bat0'], nargs='+',
-                        help='Use given batman-adv mesh interface(s) (defaults'
-                             'to bat0); specify alfred unix socket like '
-                             'bat0:/run/alfred0.sock.')
-    parser.add_argument('-d', '--dest-dir', action='store',
-                        help='Write output to destination directory',
-                        required=True)
-    parser.add_argument('-V', '--vpn', nargs='+', metavar='MAC',
-                        help='Assume MAC addresses are part of vpn')
-    parser.add_argument('-p', '--prune', metavar='DAYS', type=int,
-                        help='forget nodes offline for at least DAYS')
-    parser.add_argument('--with-rrd', dest='rrd', action='store_true',
-                        default=False,
-                        help='enable the rendering of RRD graphs (cpu '
-                             'intensive)')
-
-    options = vars(parser.parse_args())
-    main(options)

bat2nodes.py

@@ -0,0 +1,82 @@
+#!/usr/bin/env python3
+
+import json
+import fileinput
+import argparse
+import os
+import time
+
+from batman import batman
+from alfred import alfred
+from rrddb import rrd
+from nodedb import NodeDB
+from d3mapbuilder import D3MapBuilder
+
+# Force encoding to UTF-8
+import locale                                  # Ensures that subsequent open()s
+locale.getpreferredencoding = lambda _=None: 'UTF-8'  # are UTF-8 encoded.
+
+import sys
+#sys.stdin = open('/dev/stdin', 'r')
+#sys.stdout = open('/dev/stdout', 'w')
+#sys.stderr = open('/dev/stderr', 'w')
+
+parser = argparse.ArgumentParser()
+
+parser.add_argument('-a', '--aliases',
+                  help='read aliases from FILE',
+                  action='append',
+                  metavar='FILE')
+
+parser.add_argument('-m', '--mesh', action='append',
+                  default=["bat0"],
+                  help='batman mesh interface')
+
+parser.add_argument('-A', '--alfred', action='store_true',
+                  help='retrieve aliases from alfred')
+
+parser.add_argument('-d', '--destination-directory', action='store',
+                  help='destination directory for generated files',required=True)
+
+args = parser.parse_args()
+
+options = vars(args)
+
+db = NodeDB(int(time.time()))
+
+for mesh_interface in options['mesh']:
+  bm = batman(mesh_interface)
+  db.parse_vis_data(bm.vis_data(options['alfred']))
+  for gw in bm.gateway_list():
+    db.mark_gateway(gw)
+
+if options['aliases']:
+  for aliases in options['aliases']:
+    db.import_aliases(json.load(open(aliases)))
+
+if options['alfred']:
+  af = alfred()
+  db.import_aliases(af.aliases())
+
+db.load_state("state.json")
+
+# remove nodes that have been offline for more than 30 days
+db.prune_offline(time.time() - 30*86400)
+
+db.dump_state("state.json")
+
+scriptdir = os.path.dirname(os.path.realpath(__file__))
+
+m = D3MapBuilder(db)
+
+#Write nodes json
+nodes_json = open(options['destination_directory'] + '/nodes.json.new','w')
+nodes_json.write(m.build())
+nodes_json.close()
+
+#Move to destination
+os.rename(options['destination_directory'] + '/nodes.json.new',options['destination_directory'] + '/nodes.json')
+
+rrd = rrd(scriptdir +  "/nodedb/", options['destination_directory'] + "/nodes")
+rrd.update_database(db)
+rrd.update_images()

batman.py

@@ -0,0 +1,82 @@
+#!/usr/bin/env python3
+import subprocess
+import json
+import re
+
+class batman:
+  """ Bindings for B.A.T.M.A.N. advanced batctl tool
+  """
+  def __init__(self, mesh_interface = "bat0"):
+    self.mesh_interface = mesh_interface
+
+  def vis_data(self,batadv_vis=False):
+    vds = self.vis_data_batctl_legacy()
+    if batadv_vis:
+        vds += self.vis_data_batadv_vis()
+    return vds
+
+  def vis_data_helper(self,lines):
+    vd = []
+    for line in lines:
+      try:
+        utf8_line = line.decode("utf-8")
+        vd.append(json.loads(utf8_line))
+      except e:
+        pass
+    return vd
+
+  def vis_data_batctl_legacy(self):
+    """ Parse "batctl -m <mesh_interface> vd json -n" into an array of dictionaries.
+    """
+    output = subprocess.check_output(["batctl","-m",self.mesh_interface,"vd","json","-n"])
+    lines = output.splitlines()
+    vds = self.vis_data_helper(lines)
+    return vds
+
+  def vis_data_batadv_vis(self):
+    """ Parse "batadv-vis -i <mesh_interface> -f json" into an array of dictionaries.
+    """
+    output = subprocess.check_output(["batadv-vis","-i",self.mesh_interface,"-f","json"])
+    lines = output.splitlines()
+    return self.vis_data_helper(lines)
+
+  def gateway_list(self):
+    """ Parse "batctl -m <mesh_interface> gwl -n" into an array of dictionaries.
+    """
+    output = subprocess.check_output(["batctl","-m",self.mesh_interface,"gwl","-n"])
+    output_utf8 = output.decode("utf-8")
+    lines = output_utf8.splitlines()
+
+    own_mac = re.match(r"^.*MainIF/MAC: [^/]+/([0-9a-f:]+).*$", lines[0]).group(1)
+
+    gw = []
+    gw_mode = self.gateway_mode()
+    if gw_mode['mode'] == 'server':
+      gw.append(own_mac)
+
+    for line in lines:
+      gw_line = re.match(r"^(?:=>)? +([0-9a-f:]+) ", line)
+      if gw_line:
+        gw.append(gw_line.group(1))
+
+    return gw
+
+  def gateway_mode(self):
+    """ Parse "batctl -m <mesh_interface> gw"
+    """
+    output = subprocess.check_output(["batctl","-m",self.mesh_interface,"gw"])
+    elements = output.decode("utf-8").split()
+    mode = elements[0]
+    if mode == "server":
+        return {'mode': 'server', 'bandwidth': elements[3]}
+    else:
+        return {'mode': mode}
+
+if __name__ == "__main__":
+  bc = batman()
+  vd = bc.vis_data()
+  gw = bc.gateway_list()
+  for x in vd:
+    print(x)
+  print(gw)
+  print(bc.gateway_mode())

d3mapbuilder.py

@@ -0,0 +1,35 @@
+import json
+import datetime
+
+class D3MapBuilder:
+  def __init__(self, db):
+    self._db = db
+
+  def build(self):
+    output = dict()
+
+    now = datetime.datetime.utcnow().replace(microsecond=0)
+
+    nodes = self._db.get_nodes()
+
+    output['nodes'] = [{'name': x.name, 'id': x.id,
+                        'geo': [float(x) for x in x.gps.split(" ")] if x.gps else None,
+                        'firmware': x.firmware,
+                        'flags': x.flags,
+                        'clientcount': x.clientcount
+                       } for x in nodes]
+
+    links = self._db.get_links()
+
+    output['links'] = [{'source': x.source.id, 'target': x.target.id,
+                        'quality': x.quality,
+                        'type': x.type,
+                        'id': x.id
+                       } for x in links]
+
+    output['meta'] = {
+                      'timestamp': now.isoformat()
+                     }
+
+    return json.dumps(output)
+

ffhlwiki.py

@@ -0,0 +1,93 @@
+#!/usr/bin/env python3
+
+import json
+import argparse
+from itertools import zip_longest
+from urllib.request import urlopen
+from bs4 import BeautifulSoup
+
+def import_wikigps(url):
+  def fetch_wikitable(url):
+    f = urlopen(url)
+
+    soup = BeautifulSoup(f)
+
+    table = soup.find_all("table")[0]
+
+    rows = table.find_all("tr")
+
+    headers = []
+
+    data = []
+
+    def maybe_strip(x):
+      if isinstance(x.string, str):
+        return x.string.strip()
+      else:
+        return ""
+
+    for row in rows:
+      tds = list([maybe_strip(x) for x in row.find_all("td")])
+      ths = list([maybe_strip(x) for x in row.find_all("th")])
+
+      if any(tds):
+        data.append(tds)
+
+      if any(ths):
+        headers = ths
+
+    nodes = []
+
+    for d in data:
+      nodes.append(dict(zip(headers, d)))
+
+    return nodes
+
+  nodes = fetch_wikitable(url)
+
+  aliases = {}
+
+  for node in nodes:
+    try:
+      node['MAC'] = node['MAC'].split(',')
+    except KeyError:
+      pass
+
+    try:
+      node['GPS'] = node['GPS'].split(',')
+    except KeyError:
+      pass
+
+    try:
+      node['Knotenname'] = node['Knotenname'].split(',')
+    except KeyError:
+      pass
+
+    nodes = zip_longest(node['MAC'], node['GPS'], node['Knotenname'])
+
+    for data in nodes:
+      alias = {}
+
+      mac = data[0].strip()
+
+      if data[1]:
+        alias['gps'] = data[1].strip()
+
+      if data[2]:
+        alias['name'] = data[2].strip()
+
+      aliases[mac] = alias
+
+  return aliases
+
+parser = argparse.ArgumentParser()
+
+parser.add_argument('url', help='wiki URL')
+
+args = parser.parse_args()
+
+options = vars(args)
+
+aliases = import_wikigps(options['url'])
+
+print(json.dumps(aliases))

ffmap-d3.jq

@@ -1,52 +0,0 @@
-{
-    "meta": {
-        "timestamp": $nodes.timestamp
-    },
-    "nodes": (
-        $graph.batadv.nodes
-        | map(
-            if has("node_id") and .node_id
-            then (
-                $nodes.nodes[.node_id] as $node
-                | {
-                    "id": .id,
-                    "uptime": $node.statistics.uptime,
-                    "flags": ($node.flags + {"client": false}),
-                    "name": $node.nodeinfo.hostname,
-                    "clientcount": (if $node.statistics.clients >= 0 then $node.statistics.clients else 0 end),
-                    "hardware": $node.nodeinfo.hardware.model,
-                    "firmware": $node.nodeinfo.software.firmware.release,
-                    "geo": (if $node.nodeinfo.location then [$node.nodeinfo.location.latitude, $node.nodeinfo.location.longitude] else null end),
-                    #"lastseen": $node.lastseen,
-                    "network": $node.nodeinfo.network
-                }
-            )
-            else
-                {
-                    "flags": {},
-                    "id": .id,
-                    "geo": null,
-                    "clientcount": 0
-                }
-            end
-        )
-    ),
-    "links": (
-        $graph.batadv.links
-        | map(
-            $graph.batadv.nodes[.source].node_id as $source_id
-            | $graph.batadv.nodes[.target].node_id as $target_id
-            | select(
-                $source_id and $target_id and
-                ($nodes.nodes | (has($source_id) and has($target_id)))
-            )
-            | {
-                "target": .target,
-                "source": .source,
-                "quality": "\(.tq), \(.tq)",
-                "id": ($source_id + "-" + $target_id),
-                "type": (if .vpn then "vpn" else null end)
-            }
-        )
-    )
-}

gateway.json

@@ -1,29 +0,0 @@
-[
-  {
-    "node_id": "deadbfff0101",
-    "hostname": "gw01"
-  },
-  {
-    "node_id": "deadbeef0505",
-    "hostname": "gw02.hamburg.freifunk.net",
-    "network": {
-        "mac": "de:ad:be:ef:05:05",
-        "mesh": {
-            "bat0": {
-                "interfaces": {
-                    "tunnel": [
-                        "de:ad:be:ff:05:05",
-                        "de:ad:be:fc:05:05",
-                        "de:ad:bf:ff:05:05"
-                    ]
-                }
-            }
-        }
-    }
-  },
-  {
-    "node_id": "00163efb9d8d",
-    "hostname": "gw03"
-  }
-]
-

generate_aliases.py

@@ -1,110 +0,0 @@
-#!/usr/bin/env python2
-
-from __future__ import print_function
-
-import json
-import os
-import sys
-
-if len(sys.argv) != 2:
-    print('usage: ' + sys.argv[0] + ' /path/to/peers')
-    sys.exit(1)
-
-peersDir = sys.argv[1]
-
-def normalizeMac(mac):
-    mac = mac.lower()
-    normalized = ''
-
-    n = 0
-
-    for c in mac:
-        if c != ':':
-            if n > 0 and n % 2 == 0:
-                normalized = normalized + ':'
-            normalized = normalized + c
-            n += 1
-
-    return normalized
-
-def toAlias(peer):
-    alias = {}
-
-    if not (peer.has_key('name') and peer.has_key('mac')):
-        return None
-
-    name = peer['name']
-    mac = peer['mac']
-
-    alias['node_id'] = mac.replace(':', '')
-    alias['hostname'] = name
-
-    if peer.has_key('geo'):
-        geo = peer['geo']
-
-        location = {}
-
-        if geo.has_key('lon'): location['longitude'] = geo['lon']
-        if geo.has_key('lat'): location['latitude'] = geo['lat']
-
-        alias['location'] = location
-
-    #alias['network'] = {}
-    #alias['network']['mesh_interfaces'] = [mac]
-
-    return alias
-
-aliases = []
-
-for filename in os.listdir(peersDir):
-    if len(filename) == 0 or filename[0] == '.':
-        continue
-
-    isGateway = False
-
-    absFilename = peersDir + '/' + filename
-    if os.path.isfile(absFilename):
-        peerFile = open(absFilename, 'r')
-        try:
-            peerLines = peerFile.readlines()
-            peer = {}
-
-            for line in peerLines:
-                parts = line.split()
-
-                if len(parts) > 2:
-                    if parts[1] == 'Knotenname:':
-                        peer['name'] = parts[2]
-
-                    elif parts[0] == 'remote':
-                        isGateway = True
-
-                    elif parts[1] == 'MAC:':
-                        peer['mac'] = normalizeMac(parts[2])
-
-                    elif parts[1] == 'Koordinaten:' and len(parts) > 3:
-                        try:
-                            peer['geo'] = {'lat': float(parts[2]), 'lon': float(parts[3])}
-
-                        except ValueError:
-                            print('Error in %s: Invalid coordinates: %s' % (absFilename, parts[2:4]), file = sys.stderr)
-
-                elif len(parts) == 2 and parts[0] == 'key':
-                    keyParts = parts[1].split('"')
-                    if len(keyParts) > 1:
-                        peer['vpn'] = keyParts[1].lower()
-
-            if isGateway:
-                continue
-
-            alias = toAlias(peer)
-            if alias:
-                aliases.append(alias)
-
-        except Exception as e:
-            print('Error in %s, ignoring peer: %s' % (absFilename, e), file = sys.stderr)
-
-        finally:
-            peerFile.close()
-
-print(json.dumps(aliases))

generate_aliases_v2.py

@@ -1,112 +0,0 @@
-#!/usr/bin/env python2
-
-from __future__ import print_function
-
-import json
-import os
-import sys
-
-if len(sys.argv) != 2:
-    print('usage: ' + sys.argv[0] + ' /path/to/peers')
-    sys.exit(1)
-
-peersDir = sys.argv[1]
-
-def normalizeMac(mac):
-    mac = mac.lower()
-    normalized = ''
-
-    n = 0
-
-    for c in mac:
-        if c != ':':
-            if n > 0 and n % 2 == 0:
-                normalized = normalized + ':'
-            normalized = normalized + c
-            n += 1
-
-    return normalized
-
-def toAlias(peer):
-    alias = {}
-
-    if not (peer.has_key('name') and peer.has_key('mac')):
-        return None
-
-    name = peer['name']
-    mac = peer['mac']
-
-    alias['node_id'] = mac.replace(':', '')
-    alias['hostname'] = name
-
-    if peer.has_key('geo'):
-        geo = peer['geo']
-
-        location = {}
-
-        if geo.has_key('lon'): location['longitude'] = geo['lon']
-        if geo.has_key('lat'): location['latitude'] = geo['lat']
-
-        alias['location'] = location
-
-    #alias['network'] = {}
-    #alias['network']['mesh_interfaces'] = [mac]
-
-    return {'nodeinfo':alias}
-
-aliases = {}
-
-for filename in os.listdir(peersDir):
-    if len(filename) == 0 or filename[0] == '.':
-        continue
-
-    isGateway = False
-
-    absFilename = peersDir + '/' + filename
-    if os.path.isfile(absFilename):
-        peerFile = open(absFilename, 'r')
-        try:
-            peerLines = peerFile.readlines()
-            peer = {}
-
-            for line in peerLines:
-                parts = line.split()
-
-                if len(parts) > 2:
-                    if parts[1] == 'Knotenname:':
-                        peer['name'] = parts[2]
-
-                    elif parts[0] == 'remote':
-                        isGateway = True
-
-                    elif parts[1] == 'MAC:':
-                        peer['mac'] = normalizeMac(parts[2])
-
-                    elif parts[1] == 'Koordinaten:' and len(parts) > 3:
-                        try:
-                            peer['geo'] = {'lat': float(parts[2]), 'lon': float(parts[3])}
-
-                        except ValueError:
-                            print('Error in %s: Invalid coordinates: %s' % (absFilename, parts[2:4]), file = sys.stderr)
-
-                elif len(parts) == 2 and parts[0] == 'key':
-                    keyParts = parts[1].split('"')
-                    if len(keyParts) > 1:
-                        peer['vpn'] = keyParts[1].lower()
-
-            if isGateway:
-                continue
-
-            alias = toAlias(peer)
-            if alias:
-                tmpid = alias['nodeinfo']['node_id']
-#                alias['nodeinfo'].pop('node_id')
-                aliases[tmpid] = alias
-
-        except Exception as e:
-            print('Error in %s, ignoring peer: %s' % (absFilename, e), file = sys.stderr)
-
-        finally:
-            peerFile.close()
-
-print(json.dumps(aliases))

lib/__init__.py

@@ -1 +0,0 @@
-__author__ = 'hexa'

lib/alfred.py

@@ -1,33 +0,0 @@
-import subprocess
-import json
-import os
-
-
-class Alfred(object):
-    """
-    Bindings for the alfred-json utility
-    """
-    def __init__(self, unix_sockpath=None):
-        self.unix_sock = unix_sockpath
-        if unix_sockpath is not None and not os.path.exists(unix_sockpath):
-            raise RuntimeError('alfred: invalid unix socket path given')
-
-    def _fetch(self, data_type):
-        cmd = ['/usr/local/bin/alfred-json',
-               '-z',
-               '-f', 'json',
-               '-r', str(data_type)]
-        if self.unix_sock:
-            cmd.extend(['-s', self.unix_sock])
-
-        output = subprocess.check_output(cmd)
-        return json.loads(output.decode("utf-8")).values()
-
-    def nodeinfo(self):
-        return self._fetch(158)
-
-    def statistics(self):
-        return self._fetch(159)
-
-    def vis(self):
-        return self._fetch(160)

lib/batman.py

@@ -1,98 +0,0 @@
-import subprocess
-import json
-import os
-import re
-
-
-class Batman(object):
-    """
-    Bindings for B.A.T.M.A.N. Advanced
-    commandline interface "batctl"
-    """
-    def __init__(self, mesh_interface='bat0', alfred_sockpath=None):
-        self.mesh_interface = mesh_interface
-        self.alfred_sock = alfred_sockpath
-
-        # ensure /usr/sbin and /usr/local/sbin are in PATH
-        env = os.environ
-        path = set(env['PATH'].split(':'))
-        path.add('/usr/sbin/')
-        path.add('/usr/local/sbin')
-        env['PATH'] = ':'.join(path)
-        self.environ = env
-
-        # compile regular expressions only once on startup
-        self.mac_addr_pattern = re.compile(r'(([a-z0-9]{2}:){5}[a-z0-9]{2})')
-
-    def vis_data(self):
-        return self.vis_data_batadv_vis()
-
-    @staticmethod
-    def vis_data_helper(lines):
-        vd_tmp = []
-        for line in lines:
-            try:
-                utf8_line = line.decode('utf-8')
-                vd_tmp.append(json.loads(utf8_line))
-            except UnicodeDecodeError:
-                pass
-        return vd_tmp
-
-    def vis_data_batadv_vis(self):
-        """
-        Parse "batadv-vis -i <mesh_interface> -f json"
-        into an array of dictionaries.
-        """
-        cmd = ['batadv-vis', '-i', self.mesh_interface, '-f', 'json']
-        if self.alfred_sock:
-            cmd.extend(['-u', self.alfred_sock])
-        output = subprocess.check_output(cmd, env=self.environ)
-        lines = output.splitlines()
-        return self.vis_data_helper(lines)
-
-    def gateway_list(self):
-        """
-        Parse "batctl -m <mesh_interface> gwl -n"
-        into an array of dictionaries.
-        """
-        cmd = ['batctl', '-m', self.mesh_interface, 'gwl', '-n']
-        if os.geteuid() > 0:
-            cmd.insert(0, 'sudo')
-        output = subprocess.check_output(cmd, env=self.environ)
-        output_utf8 = output.decode('utf-8')
-        rows = output_utf8.splitlines()
-
-        gateways = []
-
-        # local gateway
-        header = rows.pop(0)
-        mode, bandwidth = self.gateway_mode()
-        if mode == 'server':
-            local_gw_mac = self.mac_addr_pattern.search(header).group(0)
-            gateways.append(local_gw_mac)
-
-        # remote gateway(s)
-        for row in rows:
-            match = self.mac_addr_pattern.search(row)
-            if match:
-                gateways.append(match.group(1))
-
-        return gateways
-
-    def gateway_mode(self):
-        """
-        Parse "batctl -m <mesh_interface> gw"
-        return: tuple mode, bandwidth, if mode != server then bandwidth is None
-        """
-        cmd = ['batctl', '-m', self.mesh_interface, 'gw']
-        if os.geteuid() > 0:
-            cmd.insert(0, 'sudo')
-        output = subprocess.check_output(cmd, env=self.environ)
-        chunks = output.decode("utf-8").split()
-
-        return chunks[0], chunks[3] if 3 in chunks else None
-
-if __name__ == "__main__":
-    bc = Batman()
-    vd = bc.vis_data()
-    gw = bc.gateway_list()

lib/graph.py

@@ -1,84 +0,0 @@
-from functools import reduce
-from itertools import chain
-
-import networkx as nx
-
-from lib.nodes import build_mac_table
-
-
-def import_vis_data(graph, nodes, vis_data):
-    macs = build_mac_table(nodes)
-    nodes_a = map(lambda d: 2 * [d['primary']],
-                  filter(lambda d: 'primary' in d, vis_data))
-    nodes_b = map(lambda d: [d['secondary'], d['of']],
-                  filter(lambda d: 'secondary' in d, vis_data))
-    graph.add_nodes_from(map(lambda a, b:
-                             (a, dict(primary=b, node_id=macs.get(b))),
-                             *zip(*chain(nodes_a, nodes_b))))
-
-    edges = filter(lambda d: 'neighbor' in d, vis_data)
-    graph.add_edges_from(map(lambda d: (d['router'], d['neighbor'],
-                                        dict(tq=float(d['label']))), edges))
-
-
-def mark_vpn(graph, vpn_macs):
-    components = map(frozenset, nx.weakly_connected_components(graph))
-    components = filter(vpn_macs.intersection, components)
-    nodes = reduce(lambda a, b: a | b, components, set())
-    for node in nodes:
-        for k, v in graph[node].items():
-            v['vpn'] = True
-
-
-def to_multigraph(graph):
-    def f(a):
-        node = graph.node[a]
-        return node['primary'] if node else a
-
-    def map_node(node, data):
-        return (data['primary'],
-                dict(node_id=data['node_id'])) if data else (node, dict())
-
-    digraph = nx.MultiDiGraph()
-    digraph.add_nodes_from(map(map_node, *zip(*graph.nodes_iter(data=True))))
-    digraph.add_edges_from(map(lambda a, b, data: (f(a), f(b), data),
-                               *zip(*graph.edges_iter(data=True))))
-
-    return digraph
-
-
-def merge_nodes(graph):
-    def merge_edges(data):
-        tq = min(map(lambda d: d['tq'], data))
-        vpn = all(map(lambda d: d.get('vpn', False), data))
-        return dict(tq=tq, vpn=vpn)
-
-    multigraph = to_multigraph(graph)
-    digraph = nx.DiGraph()
-    digraph.add_nodes_from(multigraph.nodes_iter(data=True))
-    edges = chain.from_iterable([[(e, d, merge_edges(
-        multigraph[e][d].values()))
-        for d in multigraph[e]] for e in multigraph])
-    digraph.add_edges_from(edges)
-
-    return digraph
-
-
-def to_undirected(graph):
-    multigraph = nx.MultiGraph()
-    multigraph.add_nodes_from(graph.nodes_iter(data=True))
-    multigraph.add_edges_from(graph.edges_iter(data=True))
-
-    def merge_edges(data):
-        tq = max(map(lambda d: d['tq'], data))
-        vpn = all(map(lambda d: d.get('vpn', False), data))
-        return dict(tq=tq, vpn=vpn, bidirect=len(data) == 2)
-
-    graph = nx.Graph()
-    graph.add_nodes_from(multigraph.nodes_iter(data=True))
-    edges = chain.from_iterable([[(e, d, merge_edges(
-        multigraph[e][d].values()))
-        for d in multigraph[e]] for e in multigraph])
-    graph.add_edges_from(edges)
-
-    return graph

lib/nodelist.py

@@ -1,27 +0,0 @@
-def export_nodelist(now, nodedb):
-    nodelist = list()
-
-    for node_id, node in nodedb["nodes"].items():
-        node_out = dict()
-        node_out["id"] = node_id
-        node_out["name"] = node["nodeinfo"]["hostname"]
-
-        if "location" in node["nodeinfo"]:
-            node_out["position"] = {"lat": node["nodeinfo"]["location"]["latitude"],
-                                    "long": node["nodeinfo"]["location"]["longitude"]}
-
-        node_out["status"] = dict()
-        node_out["status"]["online"] = node["flags"]["online"]
-
-        if "firstseen" in node:
-            node_out["status"]["firstcontact"] = node["firstseen"]
-
-        if "lastseen" in node:
-            node_out["status"]["lastcontact"] = node["lastseen"]
-
-        if "clients" in node["statistics"]:
-            node_out["status"]["clients"] = node["statistics"]["clients"]
-
-        nodelist.append(node_out)
-
-    return {"version": "1.0.1", "nodes": nodelist, "updated_at": now.isoformat()}

lib/nodes.py

@@ -1,202 +0,0 @@
-from collections import Counter, defaultdict
-from datetime import datetime
-from functools import reduce
-
-
-def build_mac_table(nodes):
-    macs = dict()
-    for node_id, node in nodes.items():
-        try:
-            macs[node['network']['mac']] = node_id
-        except KeyError:
-            pass
-        try:
-            for mac in node['nodeinfo']['network']['mesh_interfaces']:
-                macs[mac] = node_id
-        except KeyError:
-            pass
-
-        try:
-            for mac in node['nodeinfo']['network']['mesh']['bat0']['interfaces']['wireless']:
-                macs[mac] = node_id
-        except KeyError:
-            pass
-
-        try:
-            for mac in node['nodeinfo']['network']['mesh']['bat0']['interfaces']['tunnel']:
-                macs[mac] = node_id
-        except KeyError:
-            pass
-        try:
-            for mac in node['nodeinfo']['network']['mesh']['bat-ffhh']['interfaces']['tunnel']:
-                macs[mac] = node_id
-        except KeyError:
-            pass
-
-        try:
-            for mac in node['nodeinfo']['network']['mesh']['bat0']['interfaces']['other']:
-                macs[mac] = node_id
-        except KeyError:
-            pass
-
-    return macs
-
-
-def prune_nodes(nodes, now, days):
-    prune = []
-    for node_id, node in nodes.items():
-        if 'lastseen' not in node:
-            prune.append(node_id)
-            continue
-
-        lastseen = datetime.strptime(node['lastseen'], '%Y-%m-%dT%H:%M:%S')
-        delta = (now - lastseen).days
-
-        if delta >= days:
-            prune.append(node_id)
-
-    for node_id in prune:
-        del nodes[node_id]
-
-
-def mark_online(node, now):
-    node['lastseen'] = now.isoformat()
-    node.setdefault('firstseen', now.isoformat())
-    node['flags']['online'] = True
-
-
-def overrideFields(dest, src, fields):
-    for field in fields:
-        if field in src:
-            dest[field] = src[field]
-        else:
-            dest.pop(field, None)
-
-
-def import_nodeinfo(nodes, nodeinfos, now, assume_online=False, statics=False):
-    for nodeinfo in filter(lambda d: 'node_id' in d, nodeinfos):
-        node = nodes.setdefault(nodeinfo['node_id'], {'flags': {'online': False, 'gateway': False}})
-
-        if statics:
-            node['nodeinfo'] = node.setdefault('nodeinfo', {})
-            overrideFields(node['nodeinfo'], nodeinfo, ['hostname', 'location', 'node_id'])
-        else:
-            node['nodeinfo'] = nodeinfo
-
-        if assume_online:
-            mark_online(node, now)
-
-
-def reset_statistics(nodes):
-    for node in nodes.values():
-        node['statistics'] = {'clients': 0}
-
-
-def import_statistics(nodes, stats):
-    def add(node, statistics, target, source, f=lambda d: d):
-        try:
-            node['statistics'][target] = f(reduce(dict.__getitem__,
-                                                  source,
-                                                  statistics))
-        except (KeyError, TypeError, ZeroDivisionError):
-            pass
-
-    macs = build_mac_table(nodes)
-    stats = filter(lambda d: 'node_id' in d, stats)
-    stats = filter(lambda d: d['node_id'] in nodes, stats)
-    for node, stats in map(lambda d: (nodes[d['node_id']], d), stats):
-        add(node, stats, 'clients', ['clients', 'total'])
-        add(node, stats, 'gateway', ['gateway'], lambda d: macs.get(d, d))
-        add(node, stats, 'uptime', ['uptime'])
-        add(node, stats, 'loadavg', ['loadavg'])
-        add(node, stats, 'memory_usage', ['memory'],
-            lambda d: 1 - d['free'] / d['total'])
-        add(node, stats, 'rootfs_usage', ['rootfs_usage'])
-        add(node, stats, 'traffic', ['traffic'])
-
-
-def import_mesh_ifs_vis_data(nodes, vis_data):
-    macs = build_mac_table(nodes)
-
-    mesh_ifs = defaultdict(lambda: set())
-    for line in filter(lambda d: 'secondary' in d, vis_data):
-        primary = line['of']
-        mesh_ifs[primary].add(primary)
-        mesh_ifs[primary].add(line['secondary'])
-
-    def if_to_node(ifs):
-        a = filter(lambda d: d in macs, ifs)
-        a = map(lambda d: nodes[macs[d]], a)
-        try:
-            return next(a), ifs
-        except StopIteration:
-            return None
-
-    mesh_nodes = filter(lambda d: d, map(if_to_node, mesh_ifs.values()))
-
-    for v in mesh_nodes:
-        node = v[0]
-
-        ifs = set()
-
-        try:
-            ifs = ifs.union(set(node['nodeinfo']['network']['mesh_interfaces']))
-        except KeyError:
-            pass
-
-        try:
-            ifs = ifs.union(set(node['nodeinfo']['network']['mesh']['bat0']['interfaces']['wireless']))
-        except KeyError:
-            pass
-
-        try:
-            ifs = ifs.union(set(node['nodeinfo']['network']['mesh']['bat0']['interfaces']['tunnel']))
-        except KeyError:
-            pass
-
-        try:
-            ifs = ifs.union(set(node['nodeinfo']['network']['mesh']['bat-ffhh']['interfaces']['tunnel']))
-        except KeyError:
-            pass
-
-        try:
-            ifs = ifs.union(set(node['nodeinfo']['network']['mesh']['bat0']['interfaces']['other']))
-        except KeyError:
-            pass
-
-        node['nodeinfo']['network']['mesh_interfaces'] = list(ifs | v[1])
-
-
-def import_vis_clientcount(nodes, vis_data):
-    macs = build_mac_table(nodes)
-    data = filter(lambda d: d.get('label', None) == 'TT', vis_data)
-    data = filter(lambda d: d['router'] in macs, data)
-    data = map(lambda d: macs[d['router']], data)
-
-    for node_id, clientcount in Counter(data).items():
-        nodes[node_id]['statistics'].setdefault('clients', clientcount)
-
-
-def mark_gateways(nodes, gateways):
-    macs = build_mac_table(nodes)
-    gateways = filter(lambda d: d in macs, gateways)
-    for node in map(lambda d: nodes[macs[d]], gateways):
-        node['flags']['gateway'] = True
-
-
-def mark_vis_data_online(nodes, vis_data, now):
-    macs = build_mac_table(nodes)
-
-    online = set()
-    for line in vis_data:
-        if 'primary' in line:
-            online.add(line['primary'])
-        elif 'secondary' in line:
-            online.add(line['secondary'])
-        elif 'gateway' in line:
-            # This matches clients' MACs.
-            # On pre-Gluon nodes the primary MAC will be one of it.
-            online.add(line['gateway'])
-
-    for mac in filter(lambda d: d in macs, online):
-        mark_online(nodes[macs[mac]], now)

lib/rrddb.py

@@ -1,54 +0,0 @@
-#!/usr/bin/env python3
-import time
-import os
-
-from lib.GlobalRRD import GlobalRRD
-from lib.NodeRRD import NodeRRD
-
-
-class RRD(object):
-    def __init__(self,
-                 database_directory,
-                 image_path,
-                 display_time_global="7d",
-                 display_time_node="1d"):
-
-        self.dbPath = database_directory
-        self.globalDb = GlobalRRD(self.dbPath)
-        self.imagePath = image_path
-        self.displayTimeGlobal = display_time_global
-        self.displayTimeNode = display_time_node
-
-        self.currentTimeInt = (int(time.time()) / 60) * 60
-        self.currentTime = str(self.currentTimeInt)
-
-        try:
-            os.stat(self.imagePath)
-        except OSError:
-            os.mkdir(self.imagePath)
-
-    def update_database(self, nodes):
-        online_nodes = dict(filter(
-            lambda d: d[1]['flags']['online'], nodes.items()))
-        client_count = sum(map(
-            lambda d: d['statistics']['clients'], online_nodes.values()))
-
-        self.globalDb.update(len(online_nodes), client_count)
-        for node_id, node in online_nodes.items():
-            rrd = NodeRRD(os.path.join(self.dbPath, node_id + '.rrd'), node)
-            rrd.update()
-
-    def update_images(self):
-        self.globalDb.graph(os.path.join(self.imagePath, "globalGraph.png"),
-                            self.displayTimeGlobal)
-
-        nodedb_files = os.listdir(self.dbPath)
-
-        for file_name in nodedb_files:
-            if not os.path.isfile(os.path.join(self.dbPath, file_name)):
-                continue
-
-            node_name = os.path.basename(file_name).split('.')
-            if node_name[1] == 'rrd' and not node_name[0] == "nodes":
-                rrd = NodeRRD(os.path.join(self.dbPath, file_name))
-                rrd.graph(self.imagePath, self.displayTimeNode)

lib/validate.py

@@ -1,19 +0,0 @@
-import json
-
-
-def validate_nodeinfos(nodeinfos):
-    result = []
-
-    for nodeinfo in nodeinfos:
-        if validate_nodeinfo(nodeinfo):
-            result.append(nodeinfo)
-
-    return result
-
-
-def validate_nodeinfo(nodeinfo):
-    if 'location' in nodeinfo:
-        if 'latitude' not in nodeinfo['location'] or 'longitude' not in nodeinfo['location']:
-            return False
-
-    return True

link.py

@@ -0,0 +1,15 @@
+class Link():
+  def __init__(self):
+    self.id = None
+    self.source = None
+    self.target = None
+    self.quality = None
+    self.type = None
+
+class LinkConnector():
+  def __init__(self):
+    self.id = None
+    self.interface = None
+
+  def __repr__(self):
+    return "LinkConnector(%d, %s)" % (self.id, self.interface)

mkmap.sh

@@ -1,7 +1,15 @@
 #!/bin/bash
-FFMAPPATH='/opt/ffmap-backend/'
-PEERS="/etc/fastd/ffhh-mesh-vpn/peers"
 
-python2 $FFMAPPATH/generate_aliases.py $PEERS > $FFMAPPATH/aliases.json
+set -e
-#python3 $FFMAPPATH/backend.py -d /var/www/meshviewer/ --aliases $FFMAPPATH/aliases.json $FFMAPPATH/gateway.json -m bat0:/var/run/alfred.sock -p 30 --vpn de:ad:be:ff:01:01 --vpn de:ad:be:ff:05:05 --vpn de:ad:be:ff:05:06 --vpn de:ad:be:ff:03:03 --vpn de:ad:be:ff:22:22 --vpn de:ad:be:ff:22:23 --vpn de:ad:be:ff:88:88 --vpn de:ad:be:ff:88:89 --vpn de:ad:bf:ff:88:88 --vpn de:ad:bf:ff:22:22 --vpn de:ad:bf:ff:03:03 --vpn de:ad:bf:ff:05:05 --vpn de:ad:bf:ff:01:01 --vpn de:ad:be:fc:03:03 --vpn 00:16:3e:53:75:0d --vpn de:ad:be:fc:05:05 --vpn de:ad:be:fc:01:01 --vpn de:ad:be:ef:03:03 --vpn de:ad:be:ef:01:01 --vpn de:ad:be:ef:05:05 --vpn 00:16:3e:fb:9d:8d --vpn 00:16:3e:fb:9d:9d
+
-python3 $FFMAPPATH/backend.py -d /var/www/meshviewer/ --aliases $FFMAPPATH/aliases.json $FFMAPPATH/gateway.json -m bat0:/var/run/alfred.sock -p 30 --vpn de:ad:be:ff:01:01 de:ad:be:ff:05:05 de:ad:be:ff:05:06 de:ad:be:ff:03:03 de:ad:be:ff:22:22 de:ad:be:ff:22:23 de:ad:be:ff:88:88 de:ad:be:ff:88:89 de:ad:bf:ff:88:88 de:ad:bf:ff:22:22 de:ad:bf:ff:03:03 de:ad:bf:ff:05:05 de:ad:bf:ff:01:01 de:ad:be:fc:03:03 00:16:3e:53:75:0d de:ad:be:fc:05:05 de:ad:be:fc:01:01 de:ad:be:ef:03:03 de:ad:be:ef:01:01 de:ad:be:ef:05:05 00:16:3e:fb:9d:8d 00:16:3e:fb:9d:9d
+DEST=$1
+
+
+[ "$DEST" ] || exit 1
+
+cd "$(dirname "$0")"/
+
+./ffhlwiki.py http://freifunk.metameute.de/wiki/Knoten > aliases_hl.json
+./ffhlwiki.py http://freifunk.metameute.de/wiki/Moelln:Knoten > aliases_moelln.json
+
+./bat2nodes.py -A -a aliases.json -a aliases_hl.json -a aliases_moelln.json -d $DEST

node.py

@@ -0,0 +1,32 @@
+class Node():
+  def __init__(self):
+    self.name = ""
+    self.id = ""
+    self.macs = set()
+    self.interfaces = dict()
+    self.flags = dict({
+      "online": False,
+      "gateway": False,
+    })
+    self.gps = None
+    self.firmware = None
+    self.clientcount = 0
+    self.lastseen = 0
+    self.firstseen = 0
+
+  def add_mac(self, mac):
+    mac = mac.lower()
+    if len(self.macs) == 0:
+      self.id = mac
+
+    self.macs.add(mac)
+
+    self.interfaces[mac] = Interface()
+
+  def __repr__(self):
+    return '<Node %s>' % self.macs.__repr__()
+
+class Interface():
+  def __init__(self):
+    self.vpn = False
+

node_number.py

@@ -1,32 +0,0 @@
-#!/usr/bin/env python
-#Bibliotheken importieren
-import time
-import datetime
-import json
-import urllib2
-
-#Datei oeffnen
-Datei = urllib2.urlopen('https://map.hamburg.freifunk.net/nodes.json')
-Datei_Sued = urllib2.urlopen('https://map.hamburg.freifunk.net/hhsued/mv1/nodes.json')
-
-Text = Datei.read()
-Knotenzahl = Text.count('"online": true')
-Text = Datei_Sued.read()
-Knotenzahl = Knotenzahl + Text.count('"online":true')
-
-#Zeit holen
-thetime = datetime.datetime.now().isoformat()
-
-ffhh = None
-
-#Freifunk API-Datei einladen und JSON lesen
-with open('/var/www/meta/ffhh.json', 'r') as fp:
-        ffhh = json.load(fp)
-
-#Attribute Zeitstempel und Knotenanzahl setzen
-ffhh['state']['lastchange'] = thetime
-ffhh['state']['nodes'] = Knotenzahl
-
-#Freifunk API-Datein mit geaenderten werten schreiben
-with open('/var/www/meta/ffhh.json', 'w') as fp:
-        json.dump(ffhh, fp, indent=2, separators=(',', ': '))

nodedb.py

@@ -0,0 +1,251 @@
+import json
+from functools import reduce
+from collections import defaultdict
+from node import Node, Interface
+from link import Link, LinkConnector
+
+class NodeDB:
+  def __init__(self, time=0):
+    self.time = time
+    self._nodes = []
+    self._links = []
+
+  # fetch list of links
+  def get_links(self):
+    self.update_vpn_links()
+    return self.reduce_links()
+
+  # fetch list of nodes
+  def get_nodes(self):
+    return self._nodes
+
+  # remove all offlines nodes with lastseen < timestamp
+  def prune_offline(self, timestamp):
+    self._nodes = list(filter(lambda x: x.lastseen >= timestamp, self._nodes))
+
+  # write persistent state to file
+  def dump_state(self, filename):
+    obj = []
+
+    for node in self._nodes:
+      obj.append({ 'id': node.id
+                 , 'name': node.name
+                 , 'lastseen': node.lastseen
+                 , 'firstseen': node.firstseen
+                 , 'geo': node.gps
+                 })
+
+    with open(filename, "w") as f:
+      json.dump(obj, f)
+
+  # load persistent state from file
+  def load_state(self, filename):
+    try:
+      with open(filename, "r") as f:
+        obj = json.load(f)
+        for n in obj:
+          try:
+            node = self.maybe_node_by_id(n['id'])
+          except KeyError:
+            node = Node()
+            node.id = n['id']
+            node.name = n['name']
+            node.lastseen = n['lastseen']
+            node.gps = n['geo']
+            self._nodes.append(node)
+
+          if 'firstseen' in n:
+            node.firstseen = n['firstseen']
+
+    except:
+      pass
+
+  def maybe_node_by_mac(self, macs):
+    for node in self._nodes:
+      for mac in macs:
+        if mac.lower() in node.macs:
+          return node
+
+    raise KeyError
+
+  def maybe_node_by_id(self, mac):
+    for node in self._nodes:
+      if mac.lower() == node.id.lower():
+        return node
+
+    raise KeyError
+
+  def parse_vis_data(self,vis_data):
+    for x in vis_data:
+
+      if 'of' in x:
+        try:
+          node = self.maybe_node_by_mac((x['of'], x['secondary']))
+        except KeyError:
+          node = Node()
+          node.lastseen = self.time
+          node.firstseen = self.time
+          node.flags['online'] = True
+          self._nodes.append(node)
+
+        node.add_mac(x['of'])
+        node.add_mac(x['secondary'])
+
+    for x in vis_data:
+      if 'router' in x:
+        # TTs will be processed later
+        if x['label'] == "TT":
+          continue
+
+        try:
+          node = self.maybe_node_by_mac((x['router'], ))
+        except KeyError:
+          node = Node()
+          node.lastseen = self.time
+          node.firstseen = self.time
+          node.flags['online'] = True
+          node.add_mac(x['router'])
+          self._nodes.append(node)
+
+        try:
+          if 'neighbor' in x:
+            try:
+              node = self.maybe_node_by_mac((x['neighbor'], ))
+            except KeyError:
+              continue
+
+          if 'gateway' in x:
+            x['neighbor'] = x['gateway']
+
+          node = self.maybe_node_by_mac((x['neighbor'], ))
+        except KeyError:
+          node = Node()
+          node.lastseen = self.time
+          node.firstseen = self.time
+          node.flags['online'] = True
+          node.add_mac(x['neighbor'])
+          self._nodes.append(node)
+
+    for x in vis_data:
+      if 'router' in x:
+        # TTs will be processed later
+        if x['label'] == "TT":
+          continue
+
+        try:
+          if 'gateway' in x:
+            x['neighbor'] = x['gateway']
+
+          router = self.maybe_node_by_mac((x['router'], ))
+          neighbor = self.maybe_node_by_mac((x['neighbor'], ))
+        except KeyError:
+          continue
+
+        # filter TT links merged in previous step
+        if router == neighbor:
+          continue
+
+        link = Link()
+        link.source = LinkConnector()
+        link.source.interface = x['router']
+        link.source.id = self._nodes.index(router)
+        link.target = LinkConnector()
+        link.target.interface = x['neighbor']
+        link.target.id = self._nodes.index(neighbor)
+        link.quality = x['label']
+        link.id = "-".join(sorted((link.source.interface, link.target.interface)))
+
+        self._links.append(link)
+
+    for x in vis_data:
+      if 'primary' in x:
+        try:
+          node = self.maybe_node_by_mac((x['primary'], ))
+        except KeyError:
+          continue
+
+        node.id = x['primary']
+
+    for x in vis_data:
+      if 'router' in x and x['label'] == 'TT':
+        try:
+          node = self.maybe_node_by_mac((x['router'], ))
+          node.add_mac(x['gateway'])
+          node.clientcount += 1
+        except KeyError:
+          pass
+ 
+    # don't count node as its own client
+    for node in self._nodes:
+      if node.clientcount > 0:
+        node.clientcount -= 1
+
+  def reduce_links(self):
+    tmp_links = defaultdict(list)
+
+    for link in self._links:
+      tmp_links[link.id].append(link)
+
+    links = []
+
+    def reduce_link(a, b):
+      a.id = b.id
+      a.source = b.source
+      a.target = b.target
+      a.type = b.type
+      a.quality = ", ".join([x for x in (a.quality, b.quality) if x])
+
+      return a
+
+    for k, v in tmp_links.items():
+      new_link = reduce(reduce_link, v, Link())
+      links.append(new_link)
+
+    return links
+
+  def import_aliases(self, aliases):
+    for mac, alias in aliases.items():
+      try:
+        node = self.maybe_node_by_mac([mac])
+      except KeyError:
+        # create an offline node
+        node = Node()
+        node.add_mac(mac)
+        self._nodes.append(node)
+
+      if 'name' in alias:
+        node.name = alias['name']
+
+      if 'vpn' in alias and alias['vpn'] and mac and node.interfaces and mac in node.interfaces:
+        node.interfaces[mac].vpn = True
+
+      if 'gps' in alias:
+        node.gps = alias['gps']
+
+      if 'firmware' in alias:
+        node.firmware = alias['firmware']
+
+      if 'id' in alias:
+        node.id = alias['id']
+
+  def mark_gateway(self, gateway):
+    try:
+      node = self.maybe_node_by_mac((gateway, ))
+      node.flags['gateway'] = True
+    except KeyError:
+      print("WARNING: did not find gateway ", gateway, " in node list")
+
+  def update_vpn_links(self):
+    changes = 1
+    while changes > 0:
+      changes = 0
+      for link in self._links:
+        source_interface = self._nodes[link.source.id].interfaces[link.source.interface]
+        target_interface = self._nodes[link.target.id].interfaces[link.target.interface]
+        if source_interface.vpn or target_interface.vpn:
+          source_interface.vpn = True
+          target_interface.vpn = True
+          if link.type != "vpn":
+            changes += 1
+
+          link.type = "vpn"

rrddb.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python3
+import subprocess
+import time
+import os
+from GlobalRRD import GlobalRRD
+from NodeRRD import NodeRRD
+
+class rrd:
+  def __init__( self
+              , databaseDirectory
+              , imagePath
+              , displayTimeGlobal = "7d"
+              , displayTimeNode = "1d"
+              ):
+    self.dbPath = databaseDirectory
+    self.globalDb = GlobalRRD(self.dbPath)
+    self.imagePath = imagePath
+    self.displayTimeGlobal = displayTimeGlobal
+    self.displayTimeNode = displayTimeNode
+
+    self.currentTimeInt = (int(time.time())/60)*60
+    self.currentTime    = str(self.currentTimeInt)
+
+    try:
+      os.stat(self.imagePath)
+    except:
+      os.mkdir(self.imagePath)
+
+  def update_database(self,db):
+    nodes = db.get_nodes()
+    clientCount = sum(map(lambda d: d.clientcount, nodes))
+
+    curtime = time.time() - 60
+    self.globalDb.update(len(list(filter(lambda x: x.lastseen >= curtime, nodes))), clientCount)
+    for node in nodes:
+      rrd = NodeRRD(
+        os.path.join(self.dbPath, str(node.id).replace(':', '') + '.rrd'),
+        node
+      )
+      rrd.update()
+
+  def update_images(self):
+    """ Creates an image for every rrd file in the database directory.
+    """
+
+    self.globalDb.graph(os.path.join(self.imagePath, "globalGraph.png"), self.displayTimeGlobal)
+
+    nodeDbFiles = os.listdir(self.dbPath)
+
+    for fileName in nodeDbFiles:
+      if not os.path.isfile(os.path.join(self.dbPath, fileName)):
+        continue
+
+      nodeName = os.path.basename(fileName).split('.')
+      if nodeName[1] == 'rrd' and not nodeName[0] == "nodes":
+        rrd = NodeRRD(os.path.join(self.dbPath, fileName))
+        rrd.graph(self.imagePath, self.displayTimeNode)