exabgp.bgp.message.update.attribute.community.extended.traffic

Complexity

Total Complexity

44

Size/Duplication

Total Lines	306
Duplicated Lines	42.81 %

Importance

Changes

0

27 Methods

Rating	Name	Duplication	Size	Complexity
A	TrafficNextHopIPv6IETF.__init__()	10	10	3
A	TrafficRate.unpack()	4	4	1
A	TrafficMark.__repr__()	0	2	1
A	TrafficAction.__init__()	0	10	2
A	TrafficRedirectASN4.unpack()	0	4	1
A	TrafficNextHopIPv6IETF.unpack()	4	4	1
A	TrafficNextHopIPv4IETF.__init__()	10	10	3
A	TrafficMark.unpack()	0	4	1
A	TrafficRedirect.__init__()	9	9	2
A	TrafficRedirect.unpack()	4	4	1
A	TrafficRedirectASN4.__init__()	0	9	2
A	TrafficMark.__init__()	0	8	2
A	TrafficRate.__repr__()	2	2	1
A	TrafficRedirectIPv6.unpack()	0	4	1
A	TrafficNextHopIPv4IETF.__repr__()	2	2	2
A	TrafficRedirectASN4.__str__()	0	2	1
A	TrafficNextHopSimpson.__init__()	8	8	3
A	TrafficNextHopSimpson.unpack()	4	4	1
A	TrafficAction.__repr__()	0	7	3
A	TrafficRedirectIPv6.__init__()	0	5	2
A	TrafficRate.__init__()	9	9	2
A	TrafficNextHopSimpson.__repr__()	2	2	2
A	TrafficRedirectIPv6.__str__()	0	2	1
A	TrafficRedirect.__repr__()	2	2	1
A	TrafficAction.unpack()	0	6	1
A	TrafficNextHopIPv6IETF.__repr__()	2	2	2
A	TrafficNextHopIPv4IETF.unpack()	4	4	1

# encoding: utf-8
"""
traffic.py

Created by Thomas Mangin on 2014-06-21.
Copyright (c) 2014-2017 Exa Networks. All rights reserved.
License: 3-clause BSD. (See the COPYRIGHT file)
"""

import socket

from struct import pack
from struct import unpack

from exabgp.protocol.ip import IPv4
from exabgp.protocol.ip import IPv6
from exabgp.bgp.message.open.asn import ASN
from exabgp.bgp.message.open.capability.asn4 import ASN4
from exabgp.bgp.message.update.attribute.community.extended import ExtendedCommunity
from exabgp.bgp.message.update.attribute.community.extended import ExtendedCommunityIPv6


# ================================================================== TrafficRate
# RFC 5575

@ExtendedCommunity.register

This code seems to be duplicated in your project.

class TrafficRate (ExtendedCommunity):
	COMMUNITY_TYPE = 0x80
	COMMUNITY_SUBTYPE = 0x06

	__slots__ = ['asn','rate']

	def __init__ (self, asn, rate, community=None):
		self.asn = asn
		self.rate = rate
		ExtendedCommunity.__init__(
			self,
			community if community is not None else pack(
				"!2sHf",
				self._subtype(),
				asn,rate
			)
		)

	def __repr__ (self):
		return "rate-limit:%d" % self.rate

	@staticmethod
	def unpack (data):
		asn,rate = unpack('!Hf',data[2:8])
		return TrafficRate(ASN(asn),rate,data[:8])


# ================================================================ TrafficAction
# RFC 5575

@ExtendedCommunity.register
class TrafficAction (ExtendedCommunity):
	COMMUNITY_TYPE = 0x80
	COMMUNITY_SUBTYPE = 0x07

	_sample = {
		False: 0x0,
		True:  0x2,
	}

	_terminal = {
		False: 0x0,
		True:  0x1,
	}

	__slots__ = ['sample','terminal']

	def __init__ (self, sample, terminal, community=None):
		self.sample = sample
		self.terminal = terminal
		bitmask = self._sample[sample] | self._terminal[terminal]
		ExtendedCommunity.__init__(
			self,
			community if community is not None else pack(
				'!2sLBB',
				self._subtype(),
				0,0,bitmask
			)
		)

	def __repr__ (self):
		s = []
		if self.sample:
			s.append('sample')
		if self.terminal:
			s.append('terminal')
		return 'action %s' % '-'.join(s)

	@staticmethod
	def unpack (data):
		bit, = unpack('!B',data[7:8])
		sample = bool(bit & 0x02)
		terminal = bool(bit & 0x01)
		return TrafficAction(sample,terminal,data[:8])


# ============================================================== TrafficRedirect
# RFC 5575 and 7674

@ExtendedCommunity.register

This code seems to be duplicated in your project.

class TrafficRedirect (ExtendedCommunity):
	COMMUNITY_TYPE = 0x80
	COMMUNITY_SUBTYPE = 0x08

	__slots__ = ['asn','target']

	def __init__ (self, asn, target, community=None):
		self.asn = asn
		self.target = target
		ExtendedCommunity.__init__(
			self,
			community if community is not None else pack(
				"!2sHL",
				self._subtype(),
				asn,target
			)
		)

	def __repr__ (self):
		return "redirect:%s:%s" % (self.asn,self.target)

	@staticmethod
	def unpack (data):
		asn,target = unpack('!HL',data[2:8])
		return TrafficRedirect(ASN(asn),target,data[:8])


@ExtendedCommunity.register
class TrafficRedirectASN4 (ExtendedCommunity):
	COMMUNITY_TYPE = 0x82
	COMMUNITY_SUBTYPE = 0x08

	__slots__ = ['asn', 'target']

	def __init__(self, asn, target, community=None):
		self.asn = asn
		self.target = target
		ExtendedCommunity.__init__(
			self,
			community if community is not None else pack(
				"!2sLH",
				self._subtype(),
					asn, target
			)
		)

	def __str__(self):
		return "redirect:%s:%s" % (self.asn, self.target)

	@staticmethod
	def unpack(data):
		asn, target = unpack('!LH', data[2:8])
		return TrafficRedirectASN4(ASN4(asn), target, data[:8])

# ================================================================== TrafficMark
# RFC 5575

@ExtendedCommunity.register
class TrafficMark (ExtendedCommunity):
	COMMUNITY_TYPE = 0x80
	COMMUNITY_SUBTYPE = 0x09

	__slots__ = ['dscp']

	def __init__ (self, dscp, community=None):
		self.dscp = dscp
		ExtendedCommunity.__init__(
			self,
			community if community is not None else pack(
				"!2sLBB",
				self._subtype(),
				0,0,dscp
			)
		)

	def __repr__ (self):
		return "mark %d" % self.dscp

	@staticmethod
	def unpack (data):
		dscp, = unpack('!B',data[7:8])
		return TrafficMark(dscp,data[:8])

# =============================================================== TrafficNextHopIPv4IETF
# draft-ietf-idr-flowspec-redirect-02
# see RFC 4360 for ipv4 address specific extended community format

@ExtendedCommunity.register

This code seems to be duplicated in your project.

class TrafficNextHopIPv4IETF (ExtendedCommunity):
	COMMUNITY_TYPE = 0x01
	COMMUNITY_SUBTYPE = 0x0C

	__slots__ = ['ip', 'copy']

	def __init__ (self, ip, copy, community=None):
		self.ip = ip
		self.copy = copy
		ExtendedCommunity.__init__(
			self,
			community if community is not None else pack(
				"!2s4sH",
				self._subtype(),
				ip.pack(),
				1 if copy else 0
			)
		)

	def __repr__ (self):
		return "copy-to-nexthop-ietf" if self.copy else "redirect-to-nexthop-ietf"

	@staticmethod
	def unpack (data):
		ip, bit = unpack('!4sH', data[2:8])
		return TrafficNextHopIPv4IETF(IPv4.ntop(ip), bool(bit & 0x01), data[:8])

# =============================================================== TrafficNextHopIPv6IETF
# draft-ietf-idr-flowspec-redirect-02
# see RFC 5701 for ipv6 address specific extended community format

@ExtendedCommunityIPv6.register

This code seems to be duplicated in your project.

class TrafficNextHopIPv6IETF (	ExtendedCommunityIPv6):
	COMMUNITY_TYPE = 0x00
	COMMUNITY_SUBTYPE = 0x0C

	__slots__ = ['ip', 'copy']

	def __init__ (self, ip, copy, community=None):
		self.ip = ip
		self.copy = copy
		ExtendedCommunityIPv6.__init__(
			self,
			community if community is not None else pack(
				"!2s16sH",
				self._subtype(),
				ip.pack(),
				1 if copy else 0
			)
		)

	def __repr__ (self):
		return "copy-to-nexthop-ietf" if self.copy else "redirect-to-nexthop-ietf"

	@staticmethod
	def unpack (data):
		ip, bit = unpack('!16sH', data[2:20])
		return TrafficNextHopIPv6IETF(IPv6.ntop(ip), bool(bit & 0x01), data[:20])

# =============================================================== TrafficNextHopSimpson
# draft-simpson-idr-flowspec-redirect-02

# XXX: FIXME: I guess this should be a subclass of NextHop or IP ..

@ExtendedCommunity.register

This code seems to be duplicated in your project.

class TrafficNextHopSimpson (ExtendedCommunity):
	COMMUNITY_TYPE = 0x08
	COMMUNITY_SUBTYPE = 0x00

	__slots__ = ['copy']

	def __init__ (self, copy, community=None):
		self.copy = copy
		ExtendedCommunity.__init__(
			self,
			community if community is not None else pack(
				"!2sLH",
				self._subtype(),
				0,1 if copy else 0
			)
		)

	def __repr__ (self):
		return "copy-to-nexthop" if self.copy else "redirect-to-nexthop"

	@staticmethod
	def unpack (data):
		bit, = unpack('!B',data[7:8])
		return TrafficNextHopSimpson(bool(bit & 0x01), data[:8])

# ============================================================ TrafficRedirectIPv6
# https://tools.ietf.org/html/rfc5701

@ExtendedCommunityIPv6.register
class TrafficRedirectIPv6 (ExtendedCommunityIPv6):
	COMMUNITY_TYPE = 0x00
	COMMUNITY_SUBTYPE = 0x02

	def __init__(self, ip, asn, community=None):
		self.ip = ip
		self.asn = asn
		ExtendedCommunityIPv6.__init__(self, community if community is not None else pack(
			"!BB16sH", 0x00, 0x02, socket.inet_aton(socket.AF_INET6, ip), asn))

	def __str__(self):
		return "redirect %s:%d" % (self.ip, self.asn)

	@staticmethod
	def unpack(data):
		ip, asn = unpack('!16sH', data[2:11])
		return TrafficRedirectIPv6(socket.inet_ntoa(socket.AF_INET6, ip), asn, data[:11])


# ============================================================ TrafficRedirectIP
# RFC 5575
# If we need to provide the <IP>:<ASN> form for the FlowSpec Redirect ...

# import socket
# Do not use socket, use IPv4.ntop or pton

# TrafficRedirectASN = TrafficRedirect

# class TrafficRedirectIP (ExtendedCommunity):
# 	COMMUNITY_TYPE = 0x80
# 	COMMUNITY_SUBTYPE = 0x08

# 	def __init__ (self, ip, target, community=None):
# 		self.ip = ip
# 		self.target = target
# 		ExtendedCommunity.__init__(self,community if community is not None else pack("!BB4sH",0x80,0x08,socket.inet_pton(socket.AF_INET,ip),target))

# 	def __str__ (self):
# 		return "redirect %s:%d" % (self.ip,self.target)

# 	@staticmethod
# 	def unpack (data):
# 		ip,target = unpack('!4sH',data[2:8])
# 		return TrafficRedirectIP(socket.inet_ntop(socket.AF_INET,ip),target,data[:8])