Exa-Networks / exabgp

#  ----------------------------

@LINKSTATE.register()

class RsvpBw(object):

    TLV = 1090

    def __init__(self, rsvpbw):

        self.rsvpbw = rsvpbw

    def __repr__(self):

        return "Maximum reservable link bandwidth: %s" % (self.rsvpbw)

    @classmethod

    def unpack(cls, data, length):

        if len(data) != 4:

            raise Notify(3, 5, "Incorrect maximum reservable link bw metric")

        rsvpbw = unpack('!f', data)[0]

        return cls(rsvpbw=rsvpbw)

    def json(self, compact=None):

        return '"maximum-reservable-link-bandwidth": %s' % str(self.rsvpbw)

#     https://tools.ietf.org/html/rfc7752#section-3.3.3.5

@LINKSTATE.register()

class OspfForwardingAddress(object):

    TLV = 1156

    def __init__(self, addr):

        self.addr = addr

    def __repr__(self):

        return "Ospf forwarding address: '%s'" % (self.addr)

    @classmethod

    def unpack(cls, data, length):

        size = len(data)

        if size not in (4, 16):

            raise Notify(3, 5, "Error parsing OSPF Forwarding Address. Wrong size")

        return cls(IP.unpack(data[:size]))

    def json(self, compact=None):

        return '"ospf-forwarding-address": "%s"' % str(self.addr)

#     Source Router Identifier (Source Router-ID) TLV

@LINKSTATE.register()

class SrSourceRouterID(object):

    TLV = 1171

    def __init__(self, srid):

        self.srid = srid

    def __repr__(self):

        return "Source router identifier: %s" % (self.srid)

    @classmethod

    def unpack(cls, data, length):

        size = len(data)

        if size not in (4, 16):

            raise Notify(3, 5, "Error parsing SR Source Router ID. Wrong size")

        return cls(IP.unpack(data[:size]))

    def json(self, compact=None):

        return '"sr-source-router-id": "%s"' % str(self.srid)

#     https://tools.ietf.org/html/rfc7752#section-3.3.3.4

@LINKSTATE.register()

class PrefixMetric(object):

    TLV = 1155

    def __init__(self, prefixmetric):

        self.prefixmetric = prefixmetric

    def __repr__(self):

        return "prefix_metric: %s" % (self.prefixmetric)

    @classmethod

    def unpack(cls, data, length):

        if length != 4:

            raise Notify(3, 5, "Incorrect Prefix Metric size")

        return cls(unpack("!L", data)[0])

    def json(self, compact=None):

        return '"prefix-metric": %d' % int(self.prefixmetric)

#  ----------------------------

@LINKSTATE.register()

class UnRsvpBw(object):

    TLV = 1091

    def __init__(self, unrsvpbw):

        self.unrsvpbw = unrsvpbw

    def __repr__(self):

        return "Maximum link bandwidth: %s" % (self.unrsvpbw)

    @classmethod

    def unpack(cls, data, length):

        if length != 32:

            raise Notify(3, 5, "Wrong Unreservable Bw metric size")

        return cls(unpack('!ffffffff', data))

    def json(self, compact=None):

        return '"unreserved-bandwidth": %s' % str(self.unrsvpbw)

#     https://tools.ietf.org/html/rfc7752 Sec 3.3.1.3.  Node Name TLV

@LINKSTATE.register()

class NodeName(object):

    TLV = 1026

    def __init__(self, nodename):

        self.nodename = nodename

    def __repr__(self):

        return "nodename: %s" % (self.nodename)

    @classmethod

    def unpack(cls, data, length):

        if length > 255:

            raise Notify(3, 5, "Node Name TLV length too large")

        return cls(data[:length].decode('ascii'))

    def json(self, compact=None):

        return '"node-name": "%s"' % str(self.nodename)

#    https://tools.ietf.org/html/rfc7752#section-3.3.2.3 TE Metric

@LINKSTATE.register()

class TeMetric(object):

    TLV = 1092

    def __init__(self, temetric):

        self.temetric = temetric

    def __repr__(self):

        return "TE Default Metric: %s" % (self.temetric)

    @classmethod

    def unpack(cls, data, length):

        if len(data) != 4:

            raise Notify(3, 5, "Incorrect TE Metric Size")

        return cls(unpack('!L', data)[0])

    def json(self, compact=None):

        return '"te-metric": %d' % int(str(self.temetric))

#  ----------------------------

@LINKSTATE.register()

class MaxBw(object):

    TLV = 1089

    def __init__(self, maxbw):

        self.maxbw = maxbw

    def __repr__(self):

        return "Maximum link bandwidth: %s" % (self.maxbw)

    @classmethod

    def unpack(cls, data, length):

        if length != 4:

            raise Notify(3, 5, "Incorrect maximum link bw metric")

        return cls(unpack('!f', data)[0])

    def json(self, compact=None):

        return '"maximum-link-bandwidth": %s' % self.maxbw