summaryrefslogtreecommitdiffstats
path: root/retiolum/scripts/adv_graphgen/tinc_graphs/Log2JSON.py
diff options
context:
space:
mode:
authormakefu <github@syntax-fehler.de>2015-09-16 08:30:04 +0200
committermakefu <github@syntax-fehler.de>2015-09-16 08:30:04 +0200
commit147044be891c92b3c0c1bcc3a4e53e2d0eef9963 (patch)
tree13ed3ac36d514347f503043b454dddaee51b03ac /retiolum/scripts/adv_graphgen/tinc_graphs/Log2JSON.py
parent9bf9f8d045801b17956d12f599bb166e608ed6dd (diff)
parent219fab970c7fe455d3dd9bc48e909d96a234046b (diff)
Merge branch 'master' of github.com:krebscode/painload
Diffstat (limited to 'retiolum/scripts/adv_graphgen/tinc_graphs/Log2JSON.py')
-rwxr-xr-xretiolum/scripts/adv_graphgen/tinc_graphs/Log2JSON.py119
1 files changed, 119 insertions, 0 deletions
diff --git a/retiolum/scripts/adv_graphgen/tinc_graphs/Log2JSON.py b/retiolum/scripts/adv_graphgen/tinc_graphs/Log2JSON.py
new file mode 100755
index 00000000..b0bc209b
--- /dev/null
+++ b/retiolum/scripts/adv_graphgen/tinc_graphs/Log2JSON.py
@@ -0,0 +1,119 @@
+#!/usr/bin/python
+import subprocess
+import os
+import re
+import sys
+import json
+
+
+TINC_NETWORK =os.environ.get("TINC_NETWORK","retiolum")
+
+# is_legacy is the parameter which defines if the tinc config files are handled old fashioned (parse from syslog),
+# or if the new and hip tincctl should be used
+
+
+# Tags and Delimiters
+TINC_TAG="tinc.%s" % TINC_NETWORK
+BEGIN_NODES = "Nodes:"
+END_NODES = "End of nodes."
+BEGIN_SUBNET = "Subnet list:"
+END_SUBNET = "End of subnet list"
+BEGIN_EDGES = "Edges:"
+END_EDGES = "End of edges."
+def usage():
+ from sys import argv,exit
+ print("""usage: %s
+This tool dumps all tinc node informations as json
+
+ENVIRONMENT VARIABLES:
+ TINC_NETWORK The tinc network to dump
+ (default: retiolum)
+""" % argv[0])
+ exit(1)
+
+def debug(func):
+ from functools import wraps
+ @wraps(func)
+ def with_debug(*args,**kwargs):
+ print( func.__name__ + " (args: %s | kwargs %s)"% (args,kwargs))
+ return func(*args,**kwargs)
+ return with_debug
+
+
+def parse_tinc_stats():
+ import subprocess
+ from time import sleep
+ from distutils.spawn import find_executable as which
+ #newest tinc
+ if which("tinc"):
+ return parse_new_input("tinc")
+ #new tinc
+ elif which("tincctl"):
+ return parse_new_input("tincctl")
+ #old tinc
+ elif which("tincd"):
+ print("old tincd not supported")
+ sys.exit(1)
+ else:
+ raise Exception("no tinc executable found!")
+
+
+def parse_new_input(tinc_bin):
+ nodes = {}
+ pnodes = subprocess.Popen(
+ [tinc_bin,"-n",TINC_NETWORK,"dump","reachable","nodes"],
+ stdout=subprocess.PIPE).communicate()[0].decode()
+ for line in pnodes.split('\n'):
+ if not line: continue
+ l = line.split()
+ n = l[0]
+
+ token = l[1]
+ if token == 'id':
+ # new format
+ # <name> id <ident> at <ip> port <port>
+ ident = l[1]
+ l = l[2:] #shift over 'id <ident>'
+ # else: # token = 'at'
+ # old format:
+ # <name> at <ip> port <port>
+ _,_,ip,_,port = l[:5]
+ nodes[n]= { 'external-ip': ip, 'external-port' : l[4] }
+
+ psubnets = subprocess.check_output(
+ [tinc_bin,"-n",TINC_NETWORK,"dump","subnets"]).decode()
+ for line in psubnets.split('\n'):
+ if not line: continue
+ l = line.split()
+ try:
+ if not nodes[l[2]].get('internal-ip',False):
+ nodes[l[2]]['internal-ip'] = []
+ nodes[l[2]]['internal-ip'].append(l[0].split('#')[0])
+ except KeyError:
+ pass # node does not exist (presumably)
+
+ pedges = subprocess.check_output(
+ [tinc_bin,"-n",TINC_NETWORK,"dump","edges"]).decode()
+ for line in pedges.split('\n'):
+ if not line: continue
+ l = line.split()
+ # TODO: tokenize this and parse the line
+ n = l[0]
+ try:
+ if not 'to' in nodes[l[0]] :
+ nodes[n]['to'] = []
+ nodes[n]['to'].append(
+ {'name':l[2],'addr':l[4],'port':l[6],'weight' : l[-1] })
+ except KeyError:
+ pass #node does not exist
+ return nodes
+
+def main():
+ from sys import argv
+ if len(argv) > 1:
+ usage()
+ else:
+ print (json.dumps(parse_tinc_stats()))
+
+if __name__ == '__main__':
+ main()