packet lib: add ospf
Signed-off-by: Wataru ISHIDA <ishida.wataru@lab.ntt.co.jp> Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
This commit is contained in:
parent
23b1226cd2
commit
dd7a9db0d9
@ -22,6 +22,7 @@ from . import igmp
|
|||||||
from . import udp
|
from . import udp
|
||||||
from . import tcp
|
from . import tcp
|
||||||
from . import sctp
|
from . import sctp
|
||||||
|
from . import ospf
|
||||||
from ryu.ofproto import inet
|
from ryu.ofproto import inet
|
||||||
from ryu.lib import addrconv
|
from ryu.lib import addrconv
|
||||||
|
|
||||||
@ -140,3 +141,4 @@ ipv4.register_packet_type(igmp.igmp, inet.IPPROTO_IGMP)
|
|||||||
ipv4.register_packet_type(tcp.tcp, inet.IPPROTO_TCP)
|
ipv4.register_packet_type(tcp.tcp, inet.IPPROTO_TCP)
|
||||||
ipv4.register_packet_type(udp.udp, inet.IPPROTO_UDP)
|
ipv4.register_packet_type(udp.udp, inet.IPPROTO_UDP)
|
||||||
ipv4.register_packet_type(sctp.sctp, inet.IPPROTO_SCTP)
|
ipv4.register_packet_type(sctp.sctp, inet.IPPROTO_SCTP)
|
||||||
|
ipv4.register_packet_type(ospf.ospf, inet.IPPROTO_OSPF)
|
||||||
|
691
ryu/lib/packet/ospf.py
Normal file
691
ryu/lib/packet/ospf.py
Normal file
@ -0,0 +1,691 @@
|
|||||||
|
"""
|
||||||
|
RFC 2328 OSPF version 2
|
||||||
|
"""
|
||||||
|
|
||||||
|
import struct
|
||||||
|
|
||||||
|
from ryu.lib.stringify import StringifyMixin
|
||||||
|
from ryu.lib.packet import packet_base
|
||||||
|
from ryu.lib.packet import packet_utils
|
||||||
|
from ryu.lib.packet import stream_parser
|
||||||
|
|
||||||
|
from ryu.lib import addrconv
|
||||||
|
|
||||||
|
_VERSION = 2
|
||||||
|
|
||||||
|
OSPF_MSG_UNKNOWN = 0
|
||||||
|
OSPF_MSG_HELLO = 1
|
||||||
|
OSPF_MSG_DB_DESC = 2
|
||||||
|
OSPF_MSG_LS_REQ = 3
|
||||||
|
OSPF_MSG_LS_UPD = 4
|
||||||
|
OSPF_MSG_LS_ACK = 5
|
||||||
|
|
||||||
|
OSPF_UNKNOWN_LSA = 0
|
||||||
|
OSPF_ROUTER_LSA = 1
|
||||||
|
OSPF_NETWORK_LSA = 2
|
||||||
|
OSPF_SUMMARY_LSA = 3
|
||||||
|
OSPF_ASBR_SUMMARY_LSA = 4
|
||||||
|
OSPF_AS_EXTERNAL_LSA = 5
|
||||||
|
OSPF_AS_NSSA_LSA = 7 # RFC 3101
|
||||||
|
OSPF_OPAQUE_LINK_LSA = 9 # RFC 5250
|
||||||
|
OSPF_OPAQUE_AREA_LSA = 10 # RFC 5250
|
||||||
|
OSPF_OPAQUE_AS_LSA = 11 # RFC 5250
|
||||||
|
|
||||||
|
OSPF_OPTION_T = 1 # Obsolete
|
||||||
|
OSPF_OPTION_E = 1 << 1 # RFC 2328
|
||||||
|
OSPF_OPTION_MC = 1 << 2 # RFC 1584
|
||||||
|
OSPF_OPTION_NP = 1 << 3 # RFC 3101
|
||||||
|
OSPF_OPTION_EA = 1 << 4 # Obsolete
|
||||||
|
OSPF_OPTION_DC = 1 << 5 # RFC 2370
|
||||||
|
OSPF_OPTION_DN = 1 << 7 # RFC 2567
|
||||||
|
|
||||||
|
LSA_LINK_TYPE_P2P = 1
|
||||||
|
LSA_LINK_TYPE_TRANSIT = 2
|
||||||
|
LSA_LINK_TYPE_STUB = 3
|
||||||
|
LSA_LINK_TYPE_VL = 4
|
||||||
|
|
||||||
|
ROUTER_LSA_BORDER = 0x01 # The router is an ABR
|
||||||
|
ROUTER_LSA_EXTERNAL = 0x02 # The router is an ASBR
|
||||||
|
ROUTER_LSA_VIRTUAL = 0x04 # The router has a VL in this area
|
||||||
|
ROUTER_LSA_NT = 0x10 # The router always translates Type-7
|
||||||
|
ROUTER_LSA_SHORTCUT = 0x20 # Shortcut-ABR specific flag
|
||||||
|
|
||||||
|
AS_EXTERNAL_METRIC = 0x80
|
||||||
|
|
||||||
|
|
||||||
|
class InvalidChecksum(Exception):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class _TypeDisp(object):
|
||||||
|
_TYPES = {}
|
||||||
|
_REV_TYPES = None
|
||||||
|
_UNKNOWN_TYPE = None
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def register_unknown_type(cls):
|
||||||
|
def _register_type(subcls):
|
||||||
|
cls._UNKNOWN_TYPE = subcls
|
||||||
|
return subcls
|
||||||
|
return _register_type
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def register_type(cls, type_):
|
||||||
|
cls._TYPES = cls._TYPES.copy()
|
||||||
|
|
||||||
|
def _register_type(subcls):
|
||||||
|
cls._TYPES[type_] = subcls
|
||||||
|
cls._REV_TYPES = None
|
||||||
|
return subcls
|
||||||
|
return _register_type
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _lookup_type(cls, type_):
|
||||||
|
try:
|
||||||
|
return cls._TYPES[type_]
|
||||||
|
except KeyError:
|
||||||
|
return cls._UNKNOWN_TYPE
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _rev_lookup_type(cls, targ_cls):
|
||||||
|
if cls._REV_TYPES is None:
|
||||||
|
rev = dict((v, k) for k, v in cls._TYPES.iteritems())
|
||||||
|
cls._REV_TYPES = rev
|
||||||
|
return cls._REV_TYPES[targ_cls]
|
||||||
|
|
||||||
|
|
||||||
|
class LSAHeader(StringifyMixin):
|
||||||
|
_HDR_PACK_STR = '!HBB4s4sIHH'
|
||||||
|
_HDR_LEN = struct.calcsize(_HDR_PACK_STR)
|
||||||
|
|
||||||
|
def __init__(self, ls_age=0, options=0, type_=OSPF_UNKNOWN_LSA,
|
||||||
|
id_='0.0.0.0', adv_router='0.0.0.0', ls_seqnum=0,
|
||||||
|
checksum=None, length=None):
|
||||||
|
self.ls_age = ls_age
|
||||||
|
self.options = options
|
||||||
|
self.type_ = type_
|
||||||
|
self.id_ = id_
|
||||||
|
self.adv_router = adv_router
|
||||||
|
self.ls_seqnum = ls_seqnum
|
||||||
|
self.checksum = checksum
|
||||||
|
self.length = length
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
if len(buf) < cls._HDR_LEN:
|
||||||
|
raise stream_parser.StreamParser.TooSmallException(
|
||||||
|
'%d < %d' % (len(buf), cls._HDR_LEN))
|
||||||
|
(ls_age, options, type_, id_, adv_router, ls_seqnum, checksum,
|
||||||
|
length,) = struct.unpack_from(cls._HDR_PACK_STR, buffer(buf))
|
||||||
|
id_ = addrconv.ipv4.bin_to_text(id_)
|
||||||
|
adv_router = addrconv.ipv4.bin_to_text(adv_router)
|
||||||
|
rest = buf[cls._HDR_LEN:]
|
||||||
|
|
||||||
|
return {
|
||||||
|
"ls_age": ls_age,
|
||||||
|
"options": options,
|
||||||
|
"type_": type_,
|
||||||
|
"id_": id_,
|
||||||
|
"adv_router": adv_router,
|
||||||
|
"ls_seqnum": ls_seqnum,
|
||||||
|
"checksum": checksum,
|
||||||
|
"length": length,
|
||||||
|
}, rest
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
id_ = addrconv.ipv4.text_to_bin(self.id_)
|
||||||
|
adv_router = addrconv.ipv4.text_to_bin(self.adv_router)
|
||||||
|
return bytearray(struct.pack(self._HDR_PACK_STR, self.ls_age,
|
||||||
|
self.options, self.type_, id_, adv_router,
|
||||||
|
self.ls_seqnum, self.checksum, self.length))
|
||||||
|
|
||||||
|
|
||||||
|
class LSA(_TypeDisp, StringifyMixin):
|
||||||
|
def __init__(self, ls_age=0, options=0, type_=OSPF_UNKNOWN_LSA,
|
||||||
|
id_='0.0.0.0', adv_router='0.0.0.0', ls_seqnum=0,
|
||||||
|
checksum=None, length=None):
|
||||||
|
self.header = LSAHeader(ls_age, options, type_, id_, adv_router,
|
||||||
|
ls_seqnum, 0, 0)
|
||||||
|
if not (checksum or length):
|
||||||
|
tail = self.serialize_tail()
|
||||||
|
length = self.header._HDR_LEN + len(tail)
|
||||||
|
if not checksum:
|
||||||
|
head = self.header.serialize()
|
||||||
|
checksum = packet_utils.fletcher_checksum(head[2:], 14)
|
||||||
|
self.header.length = length
|
||||||
|
self.header.checksum = checksum
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
hdr, rest = LSAHeader.parser(buf)
|
||||||
|
#exclude ls_age for checksum calculation
|
||||||
|
csum = packet_utils.fletcher_checksum(buf[2:hdr['length']], 14)
|
||||||
|
if csum != hdr['checksum']:
|
||||||
|
raise InvalidChecksum("header has %d, but calculated value is %d"
|
||||||
|
% (hdr['checksum'], csum))
|
||||||
|
subcls = cls._lookup_type(hdr['type_'])
|
||||||
|
body = rest[:hdr['length']-LSAHeader._HDR_LEN]
|
||||||
|
rest = rest[hdr['length']-LSAHeader._HDR_LEN:]
|
||||||
|
kwargs = subcls.parser(body)
|
||||||
|
kwargs.update(hdr)
|
||||||
|
return subcls(**kwargs), subcls, rest
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
tail = self.serialize_tail()
|
||||||
|
self.header.length = self.header._HDR_LEN + len(tail)
|
||||||
|
head = self.header.serialize()
|
||||||
|
#exclude ls_age for checksum calculation
|
||||||
|
csum = packet_utils.fletcher_checksum(head[2:]+tail, 14)
|
||||||
|
self.header.checksum = csum
|
||||||
|
struct.pack_into("!H", head, 16, csum)
|
||||||
|
return head + tail
|
||||||
|
|
||||||
|
|
||||||
|
@LSA.register_type(OSPF_ROUTER_LSA)
|
||||||
|
class RouterLSA(LSA):
|
||||||
|
_PACK_STR = '!BBH'
|
||||||
|
_PACK_LEN = struct.calcsize(_PACK_STR) # 4bytes
|
||||||
|
|
||||||
|
class Link(StringifyMixin):
|
||||||
|
_PACK_STR = '!4s4sBBH'
|
||||||
|
_PACK_LEN = struct.calcsize(_PACK_STR) # 12bytes
|
||||||
|
|
||||||
|
def __init__(self, id_='0.0.0.0', data='0.0.0.0',
|
||||||
|
type_=LSA_LINK_TYPE_STUB, tos=0, metric=10):
|
||||||
|
self.id_ = id_
|
||||||
|
self.data = data
|
||||||
|
self.type_ = type_
|
||||||
|
self.tos = tos
|
||||||
|
self.metric = metric
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
if len(buf) < cls._PACK_LEN:
|
||||||
|
raise stream_parser.StreamParser.TooSmallException(
|
||||||
|
'%d < %d' % (len(buf), cls._PACK_LEN))
|
||||||
|
link = buf[:cls._PACK_LEN]
|
||||||
|
rest = buf[cls._PACK_LEN:]
|
||||||
|
(id_, data, type_, tos, metric) = \
|
||||||
|
struct.unpack_from(cls._PACK_STR, buffer(link))
|
||||||
|
id_ = addrconv.ipv4.bin_to_text(id_)
|
||||||
|
data = addrconv.ipv4.bin_to_text(data)
|
||||||
|
return cls(id_, data, type_, tos, metric), rest
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
id_ = addrconv.ipv4.text_to_bin(self.id_)
|
||||||
|
data = addrconv.ipv4.text_to_bin(self.data)
|
||||||
|
return bytearray(struct.pack(self._PACK_STR, id_, data, self.type_,
|
||||||
|
self.tos, self.metric))
|
||||||
|
|
||||||
|
def __init__(self, ls_age=0, options=0, type_=OSPF_ROUTER_LSA,
|
||||||
|
id_='0.0.0.0', adv_router='0.0.0.0', ls_seqnum=0,
|
||||||
|
checksum=None, length=None, flags=0, links=[]):
|
||||||
|
self.flags = flags
|
||||||
|
self.links = links
|
||||||
|
super(RouterLSA, self).__init__(ls_age, options, type_, id_,
|
||||||
|
adv_router, ls_seqnum, checksum,
|
||||||
|
length)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
links = []
|
||||||
|
hdr = buf[:cls._PACK_LEN]
|
||||||
|
buf = buf[cls._PACK_LEN:]
|
||||||
|
(flags, padding, num) = struct.unpack_from(cls._PACK_STR, buffer(hdr))
|
||||||
|
while buf:
|
||||||
|
link, buf = cls.Link.parser(buf)
|
||||||
|
links.append(link)
|
||||||
|
assert(len(links) == num)
|
||||||
|
return {
|
||||||
|
"flags": flags,
|
||||||
|
"links": links,
|
||||||
|
}
|
||||||
|
|
||||||
|
def serialize_tail(self):
|
||||||
|
head = bytearray(struct.pack(self._PACK_STR, self.flags, 0,
|
||||||
|
len(self.links)))
|
||||||
|
try:
|
||||||
|
return head + reduce(lambda a, b: a+b,
|
||||||
|
(link.serialize() for link in self.links))
|
||||||
|
except TypeError:
|
||||||
|
return head
|
||||||
|
|
||||||
|
|
||||||
|
@LSA.register_type(OSPF_NETWORK_LSA)
|
||||||
|
class NetworkLSA(LSA):
|
||||||
|
_PACK_STR = '!4s'
|
||||||
|
_PACK_LEN = struct.calcsize(_PACK_STR)
|
||||||
|
|
||||||
|
def __init__(self, ls_age=0, options=0, type_=OSPF_NETWORK_LSA,
|
||||||
|
id_='0.0.0.0', adv_router='0.0.0.0', ls_seqnum=0,
|
||||||
|
checksum=None, length=None, mask='0.0.0.0', routers=[]):
|
||||||
|
self.mask = mask
|
||||||
|
self.routers = routers
|
||||||
|
super(NetworkLSA, self).__init__(ls_age, options, type_, id_,
|
||||||
|
adv_router, ls_seqnum, checksum,
|
||||||
|
length)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
if len(buf) < cls._PACK_LEN:
|
||||||
|
raise stream_parser.StreamParser.TooSmallException(
|
||||||
|
'%d < %d' % (len(buf), cls._PACK_LEN))
|
||||||
|
binmask = buf[:cls._PACK_LEN]
|
||||||
|
(mask,) = struct.unpack_from(cls._PACK_STR, buffer(binmask))
|
||||||
|
mask = addrconv.ipv4.bin_to_text(mask)
|
||||||
|
buf = buf[cls._PACK_LEN:]
|
||||||
|
routers = []
|
||||||
|
while buf:
|
||||||
|
binrouter = buf[:cls._PACK_LEN]
|
||||||
|
(router,) = struct.unpack_from(cls._PACK_STR, buffer(binrouter))
|
||||||
|
router = addrconv.ipv4.bin_to_text(router)
|
||||||
|
routers.append(router)
|
||||||
|
buf = buf[cls._PACK_LEN:]
|
||||||
|
return {
|
||||||
|
"mask": mask,
|
||||||
|
"routers": routers,
|
||||||
|
}
|
||||||
|
|
||||||
|
def serialize_tail(self):
|
||||||
|
mask = addrconv.ipv4.text_to_bin(self.mask)
|
||||||
|
routers = [addrconv.ipv4.text_to_bin(
|
||||||
|
router) for router in self.routers]
|
||||||
|
return bytearray(struct.pack("!"+"4s"*(1+len(routers)), mask,
|
||||||
|
*routers))
|
||||||
|
|
||||||
|
|
||||||
|
@LSA.register_type(OSPF_SUMMARY_LSA)
|
||||||
|
class SummaryLSA(LSA):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@LSA.register_type(OSPF_ASBR_SUMMARY_LSA)
|
||||||
|
class ASBRSummaryLSA(LSA):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@LSA.register_type(OSPF_AS_EXTERNAL_LSA)
|
||||||
|
class ASExternalLSA(LSA):
|
||||||
|
class ExternalNetwork(StringifyMixin):
|
||||||
|
_PACK_STR = '!4sBBH4sI'
|
||||||
|
_PACK_LEN = struct.calcsize(_PACK_STR)
|
||||||
|
|
||||||
|
def __init__(self, mask='0.0.0.0', flags=0, metric=0,
|
||||||
|
fwd_addr='0.0.0.0', tag=0):
|
||||||
|
self.mask = mask
|
||||||
|
self.flags = flags
|
||||||
|
self.metric = metric
|
||||||
|
self.fwd_addr = fwd_addr
|
||||||
|
self.tag = tag
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
if len(buf) < cls._PACK_LEN:
|
||||||
|
raise stream_parser.StreamParser.TooSmallException(
|
||||||
|
'%d < %d' % (len(buf), cls._PACK_LEN))
|
||||||
|
ext_nw = buf[:cls._PACK_LEN]
|
||||||
|
rest = buf[cls._PACK_LEN:]
|
||||||
|
(mask, flags, metric_fst, metric_lst, fwd_addr,
|
||||||
|
tag) = struct.unpack_from(cls._PACK_STR, buffer(ext_nw))
|
||||||
|
mask = addrconv.ipv4.bin_to_text(mask)
|
||||||
|
metric = metric_fst << 16 | (metric_lst & 0xffff)
|
||||||
|
fwd_addr = addrconv.ipv4.bin_to_text(fwd_addr)
|
||||||
|
return cls(mask, flags, metric, fwd_addr, tag), rest
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
mask = addrconv.ipv4.text_to_bin(self.mask)
|
||||||
|
metric_fst = (self.metric >> 16) & 0xff
|
||||||
|
metric_lst = self.metric & 0xffff
|
||||||
|
fwd_addr = addrconv.ipv4.text_to_bin(self.fwd_addr)
|
||||||
|
return bytearray(struct.pack(self._PACK_STR, mask, self.flags,
|
||||||
|
metric_fst, metric_lst, fwd_addr,
|
||||||
|
self.tag))
|
||||||
|
|
||||||
|
def __init__(self, ls_age=0, options=0, type_=OSPF_AS_EXTERNAL_LSA,
|
||||||
|
id_='0.0.0.0', adv_router='0.0.0.0', ls_seqnum=0,
|
||||||
|
checksum=None, length=None, extnws=[]):
|
||||||
|
self.extnws = extnws
|
||||||
|
super(ASExternalLSA, self).__init__(ls_age, options, type_, id_,
|
||||||
|
adv_router, ls_seqnum, checksum,
|
||||||
|
length)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
extnws = []
|
||||||
|
while buf:
|
||||||
|
extnw, buf = cls.ExternalNetwork.parser(buf)
|
||||||
|
extnws.append(extnw)
|
||||||
|
return {
|
||||||
|
"extnws": extnws,
|
||||||
|
}
|
||||||
|
|
||||||
|
def serialize_tail(self):
|
||||||
|
return reduce(lambda a, b: a+b,
|
||||||
|
(extnw.serialize() for extnw in self.extnws))
|
||||||
|
|
||||||
|
|
||||||
|
@LSA.register_type(OSPF_AS_NSSA_LSA)
|
||||||
|
class NSSAExternalLSA(LSA):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@LSA.register_type(OSPF_OPAQUE_LINK_LSA)
|
||||||
|
class LocalOpaqueLSA(LSA):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@LSA.register_type(OSPF_OPAQUE_AREA_LSA)
|
||||||
|
class AreaOpaqueLSA(LSA):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@LSA.register_type(OSPF_OPAQUE_AS_LSA)
|
||||||
|
class ASOpaqueLSA(LSA):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class OSPFMessage(packet_base.PacketBase, _TypeDisp):
|
||||||
|
"""Base class for OSPF version 2 messages.
|
||||||
|
"""
|
||||||
|
|
||||||
|
_HDR_PACK_STR = '!BBH4s4sHHQ'
|
||||||
|
_HDR_LEN = struct.calcsize(_HDR_PACK_STR)
|
||||||
|
|
||||||
|
def __init__(self, type_, length=None, router_id='0.0.0.0',
|
||||||
|
area_id='0.0.0.0', au_type=1, authentication=0, checksum=None,
|
||||||
|
version=_VERSION):
|
||||||
|
self.version = version
|
||||||
|
self.type_ = type_
|
||||||
|
self.length = length
|
||||||
|
self.router_id = router_id
|
||||||
|
self.area_id = area_id
|
||||||
|
self.checksum = checksum
|
||||||
|
self.au_type = au_type
|
||||||
|
self.authentication = authentication
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
if len(buf) < cls._HDR_LEN:
|
||||||
|
raise stream_parser.StreamParser.TooSmallException(
|
||||||
|
'%d < %d' % (len(buf), cls._HDR_LEN))
|
||||||
|
(version, type_, length, router_id, area_id, checksum, au_type,
|
||||||
|
authentication) = struct.unpack_from(cls._HDR_PACK_STR, buffer(buf))
|
||||||
|
|
||||||
|
#Exclude checksum and authentication field for checksum validation.
|
||||||
|
if packet_utils.checksum(buf[:12]+buf[14:16]+buf[cls._HDR_LEN:]) \
|
||||||
|
!= checksum:
|
||||||
|
raise InvalidChecksum
|
||||||
|
|
||||||
|
if len(buf) < length:
|
||||||
|
raise stream_parser.StreamParser.TooSmallException(
|
||||||
|
'%d < %d' % (len(buf), length))
|
||||||
|
|
||||||
|
router_id = addrconv.ipv4.bin_to_text(router_id)
|
||||||
|
area_id = addrconv.ipv4.bin_to_text(area_id)
|
||||||
|
binmsg = buf[cls._HDR_LEN:length]
|
||||||
|
rest = buf[length:]
|
||||||
|
subcls = cls._lookup_type(type_)
|
||||||
|
kwargs = subcls.parser(binmsg)
|
||||||
|
return subcls(length, router_id, area_id, au_type, authentication,
|
||||||
|
checksum, version, **kwargs), None, rest
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
tail = self.serialize_tail()
|
||||||
|
self.length = self._HDR_LEN + len(tail)
|
||||||
|
head = bytearray(struct.pack(self._HDR_PACK_STR, self.version,
|
||||||
|
self.type_, self.length,
|
||||||
|
addrconv.ipv4.text_to_bin(self.router_id),
|
||||||
|
addrconv.ipv4.text_to_bin(self.area_id), 0,
|
||||||
|
self.au_type, self.authentication))
|
||||||
|
buf = head + tail
|
||||||
|
csum = packet_utils.checksum(buf[:12]+buf[14:16]+buf[self._HDR_LEN:])
|
||||||
|
self.checksum = csum
|
||||||
|
struct.pack_into("!H", buf, 12, csum)
|
||||||
|
return buf
|
||||||
|
|
||||||
|
#alias
|
||||||
|
ospf = OSPFMessage
|
||||||
|
|
||||||
|
|
||||||
|
@OSPFMessage.register_type(OSPF_MSG_HELLO)
|
||||||
|
class OSPFHello(OSPFMessage):
|
||||||
|
|
||||||
|
_PACK_STR = '!4sHBBI4s4s' # + neighbors
|
||||||
|
_PACK_LEN = struct.calcsize(_PACK_STR)
|
||||||
|
_MIN_LEN = OSPFMessage._HDR_LEN + _PACK_LEN
|
||||||
|
|
||||||
|
def __init__(self, length=None, router_id='0.0.0.0', area_id='0.0.0.0',
|
||||||
|
au_type=1, authentication=0, checksum=None, version=_VERSION,
|
||||||
|
mask='0.0.0.0', hello_interval=10, options=0, priority=1,
|
||||||
|
dead_interval=40, designated_router='0.0.0.0',
|
||||||
|
backup_router='0.0.0.0', neighbors=[]):
|
||||||
|
super(OSPFHello, self).__init__(OSPF_MSG_HELLO, length, router_id,
|
||||||
|
area_id, au_type, authentication,
|
||||||
|
checksum, version)
|
||||||
|
self.mask = mask
|
||||||
|
self.hello_interval = hello_interval
|
||||||
|
self.options = options
|
||||||
|
self.priority = priority
|
||||||
|
self.dead_interval = dead_interval
|
||||||
|
self.designated_router = designated_router
|
||||||
|
self.backup_router = backup_router
|
||||||
|
self.neighbors = neighbors
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
(mask, hello_interval, options, priority, dead_interval,
|
||||||
|
designated_router, backup_router) = struct.unpack_from(cls._PACK_STR,
|
||||||
|
buffer(buf))
|
||||||
|
mask = addrconv.ipv4.bin_to_text(mask)
|
||||||
|
designated_router = addrconv.ipv4.bin_to_text(designated_router)
|
||||||
|
backup_router = addrconv.ipv4.bin_to_text(backup_router)
|
||||||
|
neighbors = []
|
||||||
|
binneighbors = buf[cls._PACK_LEN:len(buf)]
|
||||||
|
while binneighbors:
|
||||||
|
n = binneighbors[:4]
|
||||||
|
n = addrconv.ipv4.bin_to_text(n)
|
||||||
|
binneighbors = binneighbors[4:]
|
||||||
|
neighbors.append(n)
|
||||||
|
return {
|
||||||
|
"mask": mask,
|
||||||
|
"hello_interval": hello_interval,
|
||||||
|
"options": options,
|
||||||
|
"priority": priority,
|
||||||
|
"dead_interval": dead_interval,
|
||||||
|
"designated_router": designated_router,
|
||||||
|
"backup_router": backup_router,
|
||||||
|
"neighbors": neighbors,
|
||||||
|
}
|
||||||
|
|
||||||
|
def serialize_tail(self):
|
||||||
|
head = bytearray(struct.pack(self._PACK_STR,
|
||||||
|
addrconv.ipv4.text_to_bin(self.mask),
|
||||||
|
self.hello_interval, self.options, self.priority,
|
||||||
|
self.dead_interval,
|
||||||
|
addrconv.ipv4.text_to_bin(self.designated_router),
|
||||||
|
addrconv.ipv4.text_to_bin(self.backup_router)))
|
||||||
|
try:
|
||||||
|
return head + reduce(lambda a, b: a+b,
|
||||||
|
(addrconv.ipv4.text_to_bin(
|
||||||
|
n) for n in self.neighbors))
|
||||||
|
except TypeError:
|
||||||
|
return head
|
||||||
|
|
||||||
|
|
||||||
|
@OSPFMessage.register_type(OSPF_MSG_DB_DESC)
|
||||||
|
class OSPFDBDesc(OSPFMessage):
|
||||||
|
|
||||||
|
_PACK_STR = '!HBBI' # + LSA_HEADERS
|
||||||
|
_PACK_LEN = struct.calcsize(_PACK_STR)
|
||||||
|
_MIN_LEN = OSPFMessage._HDR_LEN + _PACK_LEN
|
||||||
|
|
||||||
|
def __init__(self, length=None, router_id='0.0.0.0', area_id='0.0.0.0',
|
||||||
|
au_type=1, authentication=0, checksum=None, version=_VERSION,
|
||||||
|
mtu=1500, options=0, i_flag=0, m_flag=0, ms_flag=0,
|
||||||
|
sequence_number=0, lsa_headers=[]):
|
||||||
|
super(OSPFDBDesc, self).__init__(OSPF_MSG_DB_DESC, length, router_id,
|
||||||
|
area_id, au_type, authentication,
|
||||||
|
checksum, version)
|
||||||
|
self.mtu = mtu
|
||||||
|
self.options = options
|
||||||
|
self.i_flag = i_flag
|
||||||
|
self.m_flag = m_flag
|
||||||
|
self.ms_flag = ms_flag
|
||||||
|
self.sequence_number = sequence_number
|
||||||
|
self.lsa_headers = lsa_headers
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
(mtu, options, flags,
|
||||||
|
sequence_number) = struct.unpack_from(cls._PACK_STR, buffer(buf))
|
||||||
|
i_flag = (flags >> 2) & 0x1
|
||||||
|
m_flag = (flags >> 1) & 0x1
|
||||||
|
ms_flag = flags & 0x1
|
||||||
|
lsahdrs = []
|
||||||
|
buf = buf[cls._PACK_LEN:]
|
||||||
|
while buf:
|
||||||
|
kwargs, buf = LSAHeader.parser(buf)
|
||||||
|
lsahdrs.append(LSAHeader(**kwargs))
|
||||||
|
return {
|
||||||
|
"mtu": mtu,
|
||||||
|
"options": options,
|
||||||
|
"i_flag": i_flag,
|
||||||
|
"m_flag": m_flag,
|
||||||
|
"ms_flag": ms_flag,
|
||||||
|
"sequence_number": sequence_number,
|
||||||
|
"lsa_headers": lsahdrs,
|
||||||
|
}
|
||||||
|
|
||||||
|
def serialize_tail(self):
|
||||||
|
flags = ((self.i_flag & 0x1) << 2) ^ \
|
||||||
|
((self.m_flag & 0x1) << 1) ^ \
|
||||||
|
(self.ms_flag & 0x1)
|
||||||
|
head = bytearray(struct.pack(self._PACK_STR, self.mtu,
|
||||||
|
self.options, flags,
|
||||||
|
self.sequence_number))
|
||||||
|
try:
|
||||||
|
return head + reduce(lambda a, b: a+b,
|
||||||
|
(hdr.serialize() for hdr in self.lsa_headers))
|
||||||
|
except TypeError:
|
||||||
|
return head
|
||||||
|
|
||||||
|
|
||||||
|
@OSPFMessage.register_type(OSPF_MSG_LS_REQ)
|
||||||
|
class OSPFLSReq(OSPFMessage):
|
||||||
|
_MIN_LEN = OSPFMessage._HDR_LEN
|
||||||
|
|
||||||
|
class Request(StringifyMixin):
|
||||||
|
_PACK_STR = '!I4s4s'
|
||||||
|
_PACK_LEN = struct.calcsize(_PACK_STR)
|
||||||
|
|
||||||
|
def __init__(self, type_=OSPF_UNKNOWN_LSA, id_='0.0.0.0',
|
||||||
|
adv_router='0.0.0.0'):
|
||||||
|
self.type_ = type_
|
||||||
|
self.id = id_
|
||||||
|
self.adv_router = adv_router
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
if len(buf) < cls._PACK_LEN:
|
||||||
|
raise stream_parser.StreamParser.TooSmallException(
|
||||||
|
'%d < %d' % (len(buf), cls._PACK_LEN))
|
||||||
|
link = buf[:cls._PACK_LEN]
|
||||||
|
rest = buf[cls._PACK_LEN:]
|
||||||
|
(type_, id_, adv_router) = struct.unpack_from(cls._PACK_STR,
|
||||||
|
buffer(link))
|
||||||
|
id_ = addrconv.ipv4.bin_to_text(id_)
|
||||||
|
adv_router = addrconv.ipv4.bin_to_text(adv_router)
|
||||||
|
return cls(type_, id_, adv_router), rest
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
id_ = addrconv.ipv4.text_to_bin(self.id)
|
||||||
|
adv_router = addrconv.ipv4.text_to_bin(self.adv_router)
|
||||||
|
return bytearray(struct.pack(self._PACK_STR, self.type_,
|
||||||
|
id_, adv_router))
|
||||||
|
|
||||||
|
def __init__(self, length=None, router_id='0.0.0.0', area_id='0.0.0.0',
|
||||||
|
au_type=1, authentication=0, checksum=None, version=_VERSION,
|
||||||
|
lsa_requests=[]):
|
||||||
|
super(OSPFLSReq, self).__init__(OSPF_MSG_LS_REQ, length, router_id,
|
||||||
|
area_id, au_type, authentication,
|
||||||
|
checksum, version)
|
||||||
|
self.lsa_requests = lsa_requests
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
reqs = []
|
||||||
|
while buf:
|
||||||
|
req, buf = cls.Request.parser(buf)
|
||||||
|
reqs.append(req)
|
||||||
|
return {
|
||||||
|
"lsa_requests": reqs,
|
||||||
|
}
|
||||||
|
|
||||||
|
def serialize_tail(self):
|
||||||
|
return reduce(lambda a, b: a+b,
|
||||||
|
(req.serialize() for req in self.lsa_requests))
|
||||||
|
|
||||||
|
|
||||||
|
@OSPFMessage.register_type(OSPF_MSG_LS_UPD)
|
||||||
|
class OSPFLSUpd(OSPFMessage):
|
||||||
|
_PACK_STR = '!I'
|
||||||
|
_PACK_LEN = struct.calcsize(_PACK_STR)
|
||||||
|
_MIN_LEN = OSPFMessage._HDR_LEN + _PACK_LEN
|
||||||
|
|
||||||
|
def __init__(self, length=None, router_id='0.0.0.0', area_id='0.0.0.0',
|
||||||
|
au_type=1, authentication=0, checksum=None, version=_VERSION,
|
||||||
|
lsas=[]):
|
||||||
|
super(OSPFLSUpd, self).__init__(OSPF_MSG_LS_UPD, length, router_id,
|
||||||
|
area_id, au_type, authentication,
|
||||||
|
checksum, version)
|
||||||
|
self.lsas = lsas
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
binnum = buf[:cls._PACK_LEN]
|
||||||
|
(num,) = struct.unpack_from(cls._PACK_STR, buffer(binnum))
|
||||||
|
|
||||||
|
buf = buf[cls._PACK_LEN:]
|
||||||
|
lsas = []
|
||||||
|
while buf:
|
||||||
|
lsa, cls, buf = LSA.parser(buf)
|
||||||
|
lsas.append(lsa)
|
||||||
|
assert(len(lsas) == num)
|
||||||
|
return {
|
||||||
|
"lsas": lsas,
|
||||||
|
}
|
||||||
|
|
||||||
|
def serialize_tail(self):
|
||||||
|
head = bytearray(struct.pack(self._PACK_STR, len(self.lsas)))
|
||||||
|
try:
|
||||||
|
return head + reduce(lambda a, b: a+b,
|
||||||
|
(lsa.serialize() for lsa in self.lsas))
|
||||||
|
except TypeError:
|
||||||
|
return head
|
||||||
|
|
||||||
|
|
||||||
|
@OSPFMessage.register_type(OSPF_MSG_LS_ACK)
|
||||||
|
class OSPFLSAck(OSPFMessage):
|
||||||
|
_MIN_LEN = OSPFMessage._HDR_LEN
|
||||||
|
|
||||||
|
def __init__(self, length=None, router_id='0.0.0.0', area_id='0.0.0.0',
|
||||||
|
au_type=1, authentication=0, checksum=None, version=_VERSION,
|
||||||
|
lsa_headers=[]):
|
||||||
|
super(OSPFLSAck, self).__init__(OSPF_MSG_LS_ACK, length, router_id,
|
||||||
|
area_id, au_type, authentication,
|
||||||
|
checksum, version)
|
||||||
|
self.lsa_headers = lsa_headers
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parser(cls, buf):
|
||||||
|
lsahdrs = []
|
||||||
|
while buf:
|
||||||
|
kwargs, buf = LSAHeader.parser(buf)
|
||||||
|
lsahdrs.append(LSAHeader(**kwargs))
|
||||||
|
return {
|
||||||
|
"lsa_headers": lsahdrs,
|
||||||
|
}
|
||||||
|
|
||||||
|
def serialize_tail(self):
|
||||||
|
return reduce(lambda a, b: a+b,
|
||||||
|
(hdr.serialize() for hdr in self.lsa_headers))
|
@ -26,5 +26,6 @@ IPPROTO_AH = 51
|
|||||||
IPPROTO_ICMPV6 = 58
|
IPPROTO_ICMPV6 = 58
|
||||||
IPPROTO_NONE = 59
|
IPPROTO_NONE = 59
|
||||||
IPPROTO_DSTOPTS = 60
|
IPPROTO_DSTOPTS = 60
|
||||||
|
IPPROTO_OSPF = 89
|
||||||
IPPROTO_VRRP = 112
|
IPPROTO_VRRP = 112
|
||||||
IPPROTO_SCTP = 132
|
IPPROTO_SCTP = 132
|
||||||
|
Loading…
Reference in New Issue
Block a user