Netzwerkdiagramm mit Phython Script erstellen

Die Dokumentation von Netzwerktopologie kommt immer wieder zu kurz. Ein Netzwerkdiagramm zu erstellen und zu pflegen ist eine Aufgabe, die oftmals nur sehr ungern erledigt wird. Denn sie kann sehr komplex und zeitaufwendig sein. So treffen auch wir in unseren Security Projeken zuweilen auf nicht dokumentierte Netze. Ein eigens erstelltes kleines Python Script ermöglicht es, sich über den routebaren Teil eines IP Netzwerkes einen schnellen Überblick zu verschaffen. Als Aufklärungswerkzeug kann es Strukturen einfach erkennbar machen, die sich aus den Rohdaten nicht direkt erschließen. Von Aussen auf die eigene IP Netze angewendet bietet es einen Überblick über die Systeme, die vom Internet aus sichtbar sind und wie der Paketfluss verläuft. Die Bilder zeigen das Phyton Skript und ein Beispiel für die FQDNs: abc.xyz, google.com und youtube.com aus Sicht eines VERSANET Kunden.

Netzwerkdiagramm erstellen: Phyton Skript

Copy to Clipboard

from scapy.all import traceroute
from graphviz import Digraph
from socket import gethostbyaddr
rev_dns_cache = dict()
def multi_trace(targets=[]):
    
    if type(targets[0]) == dict:
        for target in targets:
            rev_dns_cache[target['ip']] = target['subdomain']
        new_targets = [target['ip'] for target in targets]
        targets = new_targets
    
    def has_edge(graph, v1, v2):
        tail_name = graph._quote_edge(v1)
        head_name = graph._quote_edge(v2)
        return (graph._edge % (tail_name, head_name, '')) in graph.body
    
    dot = Digraph(node_attr={'shape': 'box'})
    for target in targets:
        result, unans = traceroute([target], maxttl=32, verbose=False)
        before_last = str()
        last = str()
        for query, answer in result:
            print('.', end='')
            dns_name = None
            try:
                if answer.src not in rev_dns_cache:
                    dns_name, _, ip = gethostbyaddr(answer.src)
                    rev_dns_cache[ip[0]] = dns_name
                else:
                    dns_name = rev_dns_cache[answer.src]
            except:
                dns_name = '???'
            res_answer_src = answer.src
            dot.node(res_answer_src, label=f'{res_answer_src}\n{dns_name}')
            if query.dst != answer.src and not has_edge(dot, last, answer.src):
                dot.edge(last, answer.src)
            elif last == answer.src:
                dot.edge(before_last, last)
                break
            before_last = last
            last = answer.src
    return dot