0
0
mirror of https://gitlab.nic.cz/labs/bird.git synced 2024-12-25 11:11:53 +00:00
bird/python/BIRD/Protocol/__init__.py

79 lines
2.5 KiB
Python
Raw Normal View History

import asyncio
from BIRD.Basic import Basic, BIRDException, Code
class ProtocolException(Exception):
def __init__(self, msg):
Exception.__init__(self, f"Failed to parse protocol {self.protocol_name}: {msg}")
class ProtocolListException(Exception):
def __init__(self, msg):
Exception.__init__(self, f"Failed to parse protocol list: {msg}")
class ProtocolList(Basic):
match = {}
# def __init__(self, **kwargs):
# super().__init__(**kwargs)
def register(sub):
if sub.match in ProtocolList.match:
raise BIRDException(f"Protocol match {sub.match} already registered for {ProtocolList.match[sub.match]}")
ProtocolList.match[sub.match] = sub
async def update(self):
self.data = {}
await self.bird.cli.open()
data = await self.bird.cli.socket.command("show protocols all")
# Get header
if data[0]["code"] != Code.ProtocolListHeader:
raise ProtocolListException(f"First line is not protocol list header, got {data[0]}")
if data[0]["data"].split() != ['Name', 'Proto', 'Table', 'State', 'Since', 'Info']:
raise ProtocolListException(f"Strange protocol list header: {data[0]['data']}")
data.pop(0)
for line in data:
if line["code"] == Code.ProtocolInfo:
kwargs = Protocol.parse_info(line["data"])
if (name := kwargs["name"]) in self.data:
raise ProtocolListException(f"Duplicate protocol {name}")
if (m := kwargs["match"]) in self.match:
del kwargs["match"]
kwargs["bird"] = self.bird
self.data[name] = self.match[m](**kwargs)
else:
raise ProtocolListException(f"Unknown protocol kind {m}")
class Protocol(Basic):
def __init__(self, name, state, last_change, info, **kwargs):
super().__init__(**kwargs)
self.name = name
self.state = state
self.last_change = last_change
self.info = info
def parse_info(data):
s = data.split(maxsplit=5) + [None]
assert(len(s) <= 7)
if len(s) < 6:
raise ProtocolListException(f"Strange protocol info: {data}")
s.append(None)
s.pop(2) # drop the default table name, it's a BIRD 1 anachronism
return dict(zip(
["name", "match", "state", "last_change", "info"],
s
))
import BIRD.Protocol.Kernel
import BIRD.Protocol.Babel
import BIRD.Protocol.RAdv