107 lines
3.6 KiB
Python
107 lines
3.6 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# -*- coding: utf-8 -
|
||
|
|
#Imports:
|
||
|
|
import urllib
|
||
|
|
import json
|
||
|
|
from pprint import pprint
|
||
|
|
from node import Node
|
||
|
|
from geocode import Geocode
|
||
|
|
|
||
|
|
class Graph:
|
||
|
|
def __init__(self, nodesData, graphData):
|
||
|
|
self.coder = Geocode(geocoderCache = True, printStatus = True)
|
||
|
|
self.data = graphData
|
||
|
|
self.nodes = nodesData
|
||
|
|
self.nodes_list = {}
|
||
|
|
self.parseNodes()
|
||
|
|
self.parseLinks()
|
||
|
|
self.calculateStepsToVpn()
|
||
|
|
self.findMissingGeo()
|
||
|
|
|
||
|
|
def parseNodes(self):
|
||
|
|
for k,v in self.nodes['nodes'].iteritems():
|
||
|
|
lat, lon = self.getGeo(k)
|
||
|
|
node = Node(k, ipv6 = self.getPublicAddress(k), hostname = self.getHostname(k), isOnline = self.getOnlineState(k), lat=lat, lon=lon, coder = self.coder)
|
||
|
|
self.nodes_list[k] = node
|
||
|
|
|
||
|
|
def parseLinks(self):
|
||
|
|
link_nodes = self.data['batadv']['nodes']
|
||
|
|
for link in self.data['batadv']['links']:
|
||
|
|
if 'node_id' in link_nodes[link['source']].keys() and 'node_id' in link_nodes[link['target']].keys():#else it is a vpn link
|
||
|
|
self.setLinkBetween(link_nodes[link['source']]['node_id'], link_nodes[link['target']]['node_id'])
|
||
|
|
else:
|
||
|
|
self.setVpnLink(link['source'], link['target'])
|
||
|
|
|
||
|
|
def setLinkBetween(self, src, dst, stateOnline = True, lastSeen = None):
|
||
|
|
if src and dst:
|
||
|
|
self.nodes_list[src].links[dst] = {
|
||
|
|
'node' : self.nodes_list[dst],
|
||
|
|
'state_online' : stateOnline,
|
||
|
|
'last_seen' : lastSeen
|
||
|
|
}
|
||
|
|
self.nodes_list[dst].links[src] = {
|
||
|
|
'node' : self.nodes_list[src],
|
||
|
|
'state_online' : stateOnline,
|
||
|
|
'last_seen' : lastSeen
|
||
|
|
}
|
||
|
|
|
||
|
|
def setVpnLink(self, src, dst):
|
||
|
|
if 'node_id' not in self.data['batadv']['nodes'][src].keys():
|
||
|
|
if self.data['batadv']['nodes'][dst]['node_id']:
|
||
|
|
self.nodes_list[self.data['batadv']['nodes'][dst]['node_id']].stepsToVpn = 0
|
||
|
|
elif 'node_id' not in self.data['batadv']['nodes'][dst].keys():
|
||
|
|
if self.data['batadv']['nodes'][src]['node_id']:
|
||
|
|
self.nodes_list[self.data['batadv']['nodes'][src]['node_id']].stepsToVpn = 0
|
||
|
|
|
||
|
|
def calculateStepsToVpn(self):
|
||
|
|
for node in self.nodes_list.itervalues():
|
||
|
|
node.calculateStepsToVpn()
|
||
|
|
|
||
|
|
def findMissingGeo(self):
|
||
|
|
for node in self.nodes_list.itervalues():
|
||
|
|
node.findMissingGeo()
|
||
|
|
|
||
|
|
def getAllLevelXNodes(self, level, online = True):
|
||
|
|
zmap = {}
|
||
|
|
for k,v in self.nodes_list.iteritems():
|
||
|
|
if v.isOnline or online == False:
|
||
|
|
if v.stepsToVpn == level:
|
||
|
|
zmap[k] = v
|
||
|
|
return zmap
|
||
|
|
|
||
|
|
|
||
|
|
def getHostname(self,node_id):
|
||
|
|
return self.nodes['nodes'][node_id]['nodeinfo']['hostname']
|
||
|
|
|
||
|
|
def getGeo(self, node_id):
|
||
|
|
if 'location' in self.nodes['nodes'][node_id]['nodeinfo'] and 'latitude' in self.nodes['nodes'][node_id]['nodeinfo']['location'] and 'longitude' in self.nodes['nodes'][node_id]['nodeinfo']['location']:
|
||
|
|
return self.nodes['nodes'][node_id]['nodeinfo']['location']['latitude'], self.nodes['nodes'][node_id]['nodeinfo']['location']['longitude']
|
||
|
|
return None, None
|
||
|
|
|
||
|
|
def getPublicAddress(self,node_id):
|
||
|
|
if node_id in self.nodes['nodes']:
|
||
|
|
if 'addresses' in self.nodes['nodes'][node_id]['nodeinfo']['network']:
|
||
|
|
for address in self.nodes['nodes'][node_id]['nodeinfo']['network']['addresses']:
|
||
|
|
if address.startswith('2a03'):
|
||
|
|
return address
|
||
|
|
return None
|
||
|
|
|
||
|
|
def getOnlineState(self,node_id):
|
||
|
|
return self.nodes['nodes'][node_id]['flags']['online']
|
||
|
|
|
||
|
|
|
||
|
|
def getNodeCloudsIn(self, region):
|
||
|
|
results = {}
|
||
|
|
for k,v in self.getAllLevelXNodes(0).iteritems():
|
||
|
|
if v.geodata != None and v.isOnline == True:
|
||
|
|
if v.isInRegion(region):
|
||
|
|
results.update(v.getNodeCloud({}))
|
||
|
|
print "Result:",len(results), region
|
||
|
|
return results
|
||
|
|
|
||
|
|
def maxDepth(self):
|
||
|
|
maxDepth = 0
|
||
|
|
for v in self.nodes_list.itervalues():
|
||
|
|
if v.stepsToVpn > maxDepth:
|
||
|
|
maxDepth = v.stepsToVpn
|
||
|
|
return maxDepth+1
|