Validate received graph format

web/graph.py

@@ -1,7 +1,15 @@
+import re
+
+
 class Node:
 	def __init__(self, ip, version=None, label=None):
+		if not valid_cjdns_ip(ip):
+			raise ValueError('Invalid IP address')
+		if not valid_version(version):
+			raise ValueError('Invalid version')
+
 		self.ip = ip
-		self.version = version
+		self.version = int(version)
 		self.label = ip[-4:] if label == None else label
 
 	def __lt__(self, b):
@@ -24,3 +32,18 @@ class Edge:
 		return 'Edge(a.ip="%s", b.ip="%s")' % (
 			self.a.ip,
 			self.b.ip)
+
+
+
+_re_cjdns_ip = re.compile(r'^fc[0-9a-f]{2}(:[0-9a-f]{4}){7}$', re.IGNORECASE)
+
+def valid_cjdns_ip(ip):
+	return _re_cjdns_ip.match(ip) != None
+
+def valid_version(version):
+	try:
+		return int(version) < 20
+	except ValueError:
+		return False
+
+		

web/graphData.py

@@ -6,30 +6,37 @@ def insert_graph_data(config, json_str):
 	try:
 		graph_data = json.loads(json_str)
 	except ValueError:
-		return False
+		return 'Invalid JSON'
 
 	nodes = dict()
 	edges = []
 
-	if not 'nodes' in graph_data or not 'edges' in graph_data:
-		return False
-	
-
 	try:
 		for n in graph_data['nodes']:
-			node = Node(n['ip'], version=n['version'])
-			nodes[n['ip']] = node
+			try:
+				node = Node(n['ip'], version=n['version'])
+				nodes[n['ip']] = node
+			except Exception:
+				pass
 
 		for e in graph_data['edges']:
-			edge = Edge(nodes[e['a']], nodes[e['b']])
-			edges.append(edge)
+			try:
+				edge = Edge(nodes[e['a']], nodes[e['b']])
+				edges.append(edge)
+			except Exception:
+				pass
+	except Exception:
+		return 'Invalid JSON nodes'
 
-	except TypeError:
-		return False
+	print "Accepted %d nodes and %d links." % (len(nodes), len(edges))
 
-	print "Received %d nodes and %d links." % (len(nodes), len(edges))
+	if len(nodes) == 0 or len(edges) == 0:
+		return 'No valid nodes or edges'
 
-	with NodeDB(config) as db:
-		db.insert_graph(nodes, edges)
+	try:
+		with NodeDB(config) as db:
+			db.insert_graph(nodes, edges)
+	except Exception:
+		return 'Database failure'
 
-	return True
+	return None

web/updateGraph.py

@@ -6,6 +6,7 @@ import graphPlotter
 
 def generate_graph(time_limit=60*60*3):
 	nodes, edges = load_graph_from_db(time_limit)
+	print '%d nodes, %d edges' % (len(nodes), len(edges))
 
 	graph = graphPlotter.position_nodes(nodes, edges)
 	json = graphPlotter.get_graph_json(graph)

web/web.py

@@ -25,10 +25,10 @@ def page_sendGraph():
 	
 	data = request.form['data']
 	ret = insert_graph_data(app.config, data)
-	if ret:
+	if ret == None:
 		return 'OK'
 	else:
-		return 'FAIL'
+		return 'Error: %s' % ret
 
 if __name__ == '__main__':
 	app.run(host='::')