def update_topology_loop(self):
        hub.sleep(2)
        while True:
            if self.lsdb_updated_flag:
                self.lsdb_updated_flag = False
                ospftopo = {}

                for lsa in (l for l in self.lsdb.itervalues() if l.header.type_ == OSPF_ROUTER_LSA):
                    name = (lsa.header.type_, lsa.header.id_)

                    router = Router(name)
                    if lsa == self.myrouterlsa:
                        self.logger.debug("me: %s" % router)
                        self.me = router

                    ospftopo[name] = router

                    for l in lsa.links:
                        if l.type_ == 2:
                            # l.data is interface address
                            name = (2, l.data)
                            link = Link(name)
                            link.set_start(router, l.data)
                            link.metric = l.metric
                            link.tos = l.tos
                            link.nw = l.id_
                            router.add_link(link)
                            ospftopo[name] = link

                links = [l for l in ospftopo.itervalues() if isinstance(l, Link)]
                while len(links) > 0:
                    link = links.pop(0)
                    if not link.end:
                        el = [el for el in links if link.nw == el.nw]
                        if not len(el) == 1:
                            continue
                        el = el.pop(0)
                        link.set_end(el.start, el.startaddr)
                        el.set_end(link.start, link.startaddr)

                for lsa in (l for l in self.lsdb.itervalues() if l.header.type_ == OSPF_AS_EXTERNAL_LSA):
                    name = (lsa.header.type_, lsa.header.id_)
                    extnw = ExtNetwork(name)
                    ospftopo[name] = extnw
                    adv_r = (1, lsa.header.adv_router)
                    assert ospftopo.has_key(adv_r)

                    extnw.set_router(ospftopo[adv_r])
                    extnw.mask = lsa.extnws[0].mask
                    ospftopo[adv_r].add_extnw(extnw)
                self.ospftopo = ospftopo
                self.install_l3_rules()

            if self.labeldb_updated_flag:
                self.labeldb_updated_flag = False
                self.install_label_rules()

            hub.sleep(5)
Пример #2
0
def getTopology(addr):
    p = subprocess.Popen("/usr/local/sbin/ospfclient %s 0 0 0 0 0 2> /dev/null" % addr, shell=True, stdout = subprocess.PIPE)
    buf = p.stdout.readline()
    try:
        lsalist = json.loads(buf)
    except:
        print "failed to parse: %s" % buf
        return None

    topo = {}
    for lsa in [lsa for lsa in lsalist if lsa[u'type'] == 1]:
        lsa_type = lsa[u'type']
        name = (lsa_type, lsa[u'id'])

        router = Router(name)
        topo[name] = router

        for l in lsa[u'links']:
            if l[u'type'] == 2:
                name = (2, l[u'data'])
                link = Link(name)
                link.set_start(router, l[u'data'])
                link.metric = float(l[u'metric'])
                link.tos = float(l[u'tos'])
                link.nw = l[u'id']
                router.add_link(link)
                topo[name] = link

    links = [link for link in topo.values() if link.__class__ == Link]

    while len(links) > 0:
        link = links.pop(0)
        if not link.end:
            el = [el for el in links if link.nw == el.nw]
            assert len(el) == 1
            el = el.pop(0)
            link.set_end(el.start, el.startaddr)
            el.set_end(link.start, link.startaddr)

    for lsa in [lsa for lsa in lsalist if lsa[u'type'] == 5]:
        lsa_type = lsa[u'type']
        name = (lsa_type, lsa[u'id'])
        extnw = ExtNetwork(name)
        topo[name] = extnw
        adv_r = (1, lsa[u'adv_router'])
        assert topo.has_key(adv_r)
        extnw.set_router(topo[adv_r])
        extnw.mask=lsa[u'mask']
        topo[adv_r].add_extnw(extnw)

    return topo