def make_graph_from_hosts(hosts):
#hosts = parser.get_root().search_children('host', deep=True)
graph = Graph()
nodes = list()
node_cache = {}
# Setting initial reference host
main_node = NetNode()
nodes.append(main_node)
localhost = TracerouteHostInfo()
localhost.ip = {"addr": "127.0.0.1/8", "type": "ipv4"}
localhost.hostname = "localhost"
main_node.set_host(localhost)
main_node.set_draw_info({
"valid": True,
"color": (0, 0, 0),
"radius": NONE_RADIUS
})
#Save endpoints for attaching scanned hosts to
endpoints = {}
# For each host in hosts just mount the graph
for host in hosts:
trace = host.trace
endpoints[host] = nodes[0]
hops = trace.get("hops")
# If host has traceroute information mount graph
if hops is not None and len(hops) > 0:
prev_node = nodes[0]
hops = trace.get("hops", [])
ttls = [int(hop["ttl"]) for hop in hops]
# Getting nodes of host by ttl
for ttl in range(1, max(ttls) + 1):
if ttl in ttls:
# Find a hop by ttl
hop = None
for h in hops:
if ttl == int(h["ttl"]):
hop = h
break
node = node_cache.get(hop["ipaddr"])
if node is None:
node = NetNode()
nodes.append(node)
hop_host = TracerouteHostInfo()
hop_host.ip = {
"addr": hop["ipaddr"],
"type": "",
"vendor": ""
}
node.set_draw_info({"valid": True})
node.set_draw_info({
"color": (1, 1, 1),
"radius": NONE_RADIUS
})
if hop["host"] != "":
hop_host.hostname = hop["host"]
node.set_host(hop_host)
node_cache[node.get_info("ip")] = node
rtt = hop["rtt"]
if rtt != "--":
graph.set_connection(node, prev_node, float(rtt))
else:
graph.set_connection(node, prev_node)
else:
node = NetNode()
nodes.append(node)
node.set_draw_info({"valid": False})
node.set_draw_info({
"color": (1, 1, 1),
"radius": NONE_RADIUS
})
graph.set_connection(node, prev_node)
prev_node = node
endpoints[host] = node
# For each fully scanned host
for host in hosts:
ip = host.ip
if ip is None:
ip = host.ipv6
node = node_cache.get(ip["addr"])
if node is None:
node = NetNode()
nodes.append(node)
node.set_draw_info({"no_route": True})
graph.set_connection(node, endpoints[host])
node.set_draw_info({"valid": True})
node.set_draw_info({"scanned": True})
set_node_info(node, host)
node_cache[node.get_info("ip")] = node
graph.set_nodes(nodes)
graph.set_main_node(main_node)
return graph
def make_graph_from_hosts(hosts):
#hosts = parser.get_root().search_children('host', deep=True)
graph = Graph()
nodes = list()
node_cache = {}
# Setting initial reference host
main_node = NetNode()
nodes.append(main_node)
localhost = TracerouteHostInfo()
localhost.ip = {"addr": "127.0.0.1/8", "type": "ipv4"}
localhost.hostname = "localhost"
main_node.set_host(localhost)
main_node.set_draw_info(
{"valid": True, "color": (0, 0, 0), "radius": NONE_RADIUS})
#Save endpoints for attaching scanned hosts to
endpoints = {}
# For each host in hosts just mount the graph
for host in hosts:
trace = host.trace
endpoints[host] = nodes[0]
hops = trace.get("hops")
# If host has traceroute information mount graph
if hops is not None and len(hops) > 0:
prev_node = nodes[0]
hops = trace.get("hops", [])
ttls = [int(hop["ttl"]) for hop in hops]
# Getting nodes of host by ttl
for ttl in range(1, max(ttls) + 1):
if ttl in ttls:
# Find a hop by ttl
hop = None
for h in hops:
if ttl == int(h["ttl"]):
hop = h
break
node = node_cache.get(hop["ipaddr"])
if node is None:
node = NetNode()
nodes.append(node)
hop_host = TracerouteHostInfo()
hop_host.ip = {
"addr": hop["ipaddr"],
"type": "",
"vendor": ""
}
node.set_draw_info({"valid": True})
node.set_draw_info({"color": (1, 1, 1),
"radius": NONE_RADIUS})
if hop["host"] != "":
hop_host.hostname = hop["host"]
node.set_host(hop_host)
node_cache[node.get_info("ip")] = node
rtt = hop["rtt"]
if rtt != "--":
graph.set_connection(node, prev_node, float(rtt))
else:
graph.set_connection(node, prev_node)
else:
node = NetNode()
nodes.append(node)
node.set_draw_info({"valid": False})
node.set_draw_info(
{"color": (1, 1, 1), "radius": NONE_RADIUS})
graph.set_connection(node, prev_node)
prev_node = node
endpoints[host] = node
# For each fully scanned host
for host in hosts:
ip = host.ip
if ip is None:
ip = host.ipv6
node = node_cache.get(ip["addr"])
if node is None:
node = NetNode()
nodes.append(node)
node.set_draw_info({"no_route": True})
graph.set_connection(node, endpoints[host])
node.set_draw_info({"valid": True})
node.set_draw_info({"scanned": True})
set_node_info(node, host)
node_cache[node.get_info("ip")] = node
graph.set_nodes(nodes)
graph.set_main_node(main_node)
return graph
def make_graph_from_hosts(hosts):
#hosts = parser.get_root().search_children('host', deep=True)
graph = Graph()
nodes = list()
node_cache = {}
ancestor_node_cache = {}
descendant_node_cache = {}
# Setting initial reference host
main_node = NetNode()
nodes.append(main_node)
localhost = TracerouteHostInfo()
localhost.ip = {"addr": "127.0.0.1/8", "type": "ipv4"}
localhost.hostname = "localhost"
main_node.set_host(localhost)
main_node.set_draw_info({
"valid": True,
"color": (0, 0, 0),
"radius": NONE_RADIUS
})
#Save endpoints for attaching scanned hosts to
endpoints = {}
# For each host in hosts just mount the graph
for host in hosts:
trace = host.trace
endpoints[host] = nodes[0]
hops = trace.get("hops")
# If host has traceroute information mount graph
if hops is not None and len(hops) > 0:
prev_node = nodes[0]
hops = trace.get("hops", [])
ttls = [int(hop["ttl"]) for hop in hops]
# Getting nodes of host by ttl
for ttl in range(1, max(ttls) + 1):
if ttl in ttls:
hop = find_hop_by_ttl(hops, ttl)
node = node_cache.get(hop["ipaddr"])
if node is None:
node = NetNode()
nodes.append(node)
hop_host = TracerouteHostInfo()
hop_host.ip = {
"addr": hop["ipaddr"],
"type": "",
"vendor": ""
}
node.set_draw_info({"valid": True})
node.set_draw_info({
"color": (1, 1, 1),
"radius": NONE_RADIUS
})
if hop["host"] != "":
hop_host.hostname = hop["host"]
node.set_host(hop_host)
node_cache[node.get_info("ip")] = node
rtt = hop["rtt"]
if rtt != "--":
graph.set_connection(node, prev_node, float(rtt))
else:
graph.set_connection(node, prev_node)
else:
# Add an "anonymous" node only if there isn't already a node
# equivalent to it (i.e. at same distance from the previous
# "real" node)
pre_hop = None
pre_hop_distance = 0
for i in range(1, ttl + 1):
pre_hop = find_hop_by_ttl(hops, ttl - i)
if pre_hop is not None:
pre_hop_distance = i
break
post_hop = None
post_hop_distance = 0
for i in range(1, max(ttls) - ttl):
post_hop = find_hop_by_ttl(hops, ttl + i)
if post_hop is not None:
post_hop_distance = i
break
assert pre_hop is not None, "pre_hop should have become localhost if nothing else"
ancestor_key = (pre_hop["ipaddr"], pre_hop_distance)
descendant_key = None
if post_hop is not None:
descendant_key = (post_hop["ipaddr"],
post_hop_distance)
if ancestor_key in ancestor_node_cache:
node = ancestor_node_cache[ancestor_key]
elif descendant_key is not None and descendant_key in descendant_node_cache:
node = descendant_node_cache[descendant_key]
graph.set_connection(node, prev_node)
else:
node = NetNode()
nodes.append(node)
node.set_draw_info({"valid": False})
node.set_draw_info({
"color": (1, 1, 1),
"radius": NONE_RADIUS
})
graph.set_connection(node, prev_node)
ancestor_node_cache[ancestor_key] = node
if descendant_key is not None:
descendant_node_cache[descendant_key] = node
prev_node = node
endpoints[host] = node
# For each fully scanned host
for host in hosts:
ip = host.ip
if ip is None:
ip = host.ipv6
node = node_cache.get(ip["addr"])
if node is None:
node = NetNode()
nodes.append(node)
node.set_draw_info({"no_route": True})
graph.set_connection(node, endpoints[host])
node.set_draw_info({"valid": True})
node.set_draw_info({"scanned": True})
set_node_info(node, host)
node_cache[node.get_info("ip")] = node
graph.set_nodes(nodes)
graph.set_main_node(main_node)
return graph
def make_graph_from_hosts(hosts):
#hosts = parser.get_root().search_children('host', deep=True)
graph = Graph()
nodes = list()
node_cache = {}
ancestor_node_cache = {}
descendant_node_cache = {}
# Setting initial reference host
main_node = NetNode()
nodes.append(main_node)
localhost = TracerouteHostInfo()
localhost.ip = {"addr": "127.0.0.1/8", "type": "ipv4"}
localhost.hostname = "localhost"
main_node.set_host(localhost)
main_node.set_draw_info(
{"valid": True, "color": (0, 0, 0), "radius": NONE_RADIUS})
#Save endpoints for attaching scanned hosts to
endpoints = {}
# For each host in hosts just mount the graph
for host in hosts:
trace = host.trace
endpoints[host] = nodes[0]
hops = trace.get("hops")
# If host has traceroute information mount graph
if hops is not None and len(hops) > 0:
prev_node = nodes[0]
hops = trace.get("hops", [])
ttls = [int(hop["ttl"]) for hop in hops]
# Getting nodes of host by ttl
for ttl in range(1, max(ttls) + 1):
if ttl in ttls:
hop = find_hop_by_ttl(hops, ttl)
node = node_cache.get(hop["ipaddr"])
if node is None:
node = NetNode()
nodes.append(node)
hop_host = TracerouteHostInfo()
hop_host.ip = {
"addr": hop["ipaddr"],
"type": "",
"vendor": ""
}
node.set_draw_info({"valid": True})
node.set_draw_info({"color": (1, 1, 1),
"radius": NONE_RADIUS})
if hop["host"] != "":
hop_host.hostname = hop["host"]
node.set_host(hop_host)
node_cache[node.get_info("ip")] = node
rtt = hop["rtt"]
if rtt != "--":
graph.set_connection(node, prev_node, float(rtt))
else:
graph.set_connection(node, prev_node)
else:
# Add an "anonymous" node only if there isn't already a node
# equivalent to it (i.e. at same distance from the previous
# "real" node)
pre_hop = None
pre_hop_distance = 0
for i in range(1, ttl + 1):
pre_hop = find_hop_by_ttl(hops, ttl-i)
if pre_hop is not None:
pre_hop_distance = i
break
post_hop = None
post_hop_distance = 0
for i in range(1, max(ttls) - ttl):
post_hop = find_hop_by_ttl(hops, ttl+i)
if post_hop is not None:
post_hop_distance = i
break
assert pre_hop is not None, "pre_hop should have become localhost if nothing else"
ancestor_key = (pre_hop["ipaddr"], pre_hop_distance)
descendant_key = None
if post_hop is not None:
descendant_key = (post_hop["ipaddr"], post_hop_distance)
if ancestor_key in ancestor_node_cache:
node = ancestor_node_cache[ancestor_key]
elif descendant_key is not None and descendant_key in descendant_node_cache:
node = descendant_node_cache[descendant_key]
graph.set_connection(node, prev_node)
else:
node = NetNode()
nodes.append(node)
node.set_draw_info({"valid":False})
node.set_draw_info({"color":(1,1,1), "radius":NONE_RADIUS})
graph.set_connection(node, prev_node)
ancestor_node_cache[ancestor_key] = node
if descendant_key is not None:
descendant_node_cache[descendant_key] = node
prev_node = node
endpoints[host] = node
# For each fully scanned host
for host in hosts:
ip = host.ip
if ip is None:
ip = host.ipv6
node = node_cache.get(ip["addr"])
if node is None:
node = NetNode()
nodes.append(node)
node.set_draw_info({"no_route": True})
graph.set_connection(node, endpoints[host])
node.set_draw_info({"valid": True})
node.set_draw_info({"scanned": True})
set_node_info(node, host)
node_cache[node.get_info("ip")] = node
graph.set_nodes(nodes)
graph.set_main_node(main_node)
return graph