← Back to branch summary

~ubuntu-branches/ubuntu/utopic/exabgp/utopic

« back to all changes in this revision

Viewing changes to lib/exabgp/bgp/message/update/nlri/bgp.py

  • Committer: Package Import Robot
  • Author(s): Henry-Nicolas Tourneur
  • Date: 2014-03-08 19:07:00 UTC
  • mfrom: (1.1.8)
  • Revision ID: package-import@ubuntu.com-20140308190700-xjbibpg1g6001c9x
Tags: 3.3.1-1
http://bugs.debian.org/726066
http://bugs.debian.org/703774
* New upstream release
* Bump python minimal required version (2.7)
* Closes: #726066 Debian packaging improvements proposed by Vincent Bernat
* Closes: #703774 not existent rundir (/var/run/exabgp) after reboot

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/exabgp/bgp/message/update/nlri/bgp.py
 
1
# encoding: utf-8
 
2
"""
 
3
bgp.py
 
4
 
 
5
Created by Thomas Mangin on 2012-07-08.
 
6
Copyright (c) 2009-2013 Exa Networks. All rights reserved.
 
7
"""
 
8
 
 
9
from struct import pack,unpack
 
10
from exabgp.protocol.family import AFI,SAFI
 
11
 
 
12
from exabgp.bgp.message.update.nlri.prefix import mask_to_bytes,Prefix
 
13
 
 
14
class PathInfo (object):
 
15
        def __init__ (self,integer=None,ip=None,packed=None):
 
16
                if packed:
 
17
                        self.path_info = packed
 
18
                elif ip:
 
19
                        self.path_info = ''.join([chr(int(_)) for _ in ip.split('.')])
 
20
                elif integer:
 
21
                        self.path_info = ''.join([chr((integer>>offset) & 0xff) for offset in [24,16,8,0]])
 
22
                else:
 
23
                        self.path_info = ''
 
24
                #sum(int(a)<<offset for (a,offset) in zip(ip.split('.'), range(24, -8, -8)))
 
25
 
 
26
        def __len__ (self):
 
27
                return len(self.path_info)
 
28
 
 
29
        def json (self):
 
30
                return '"path-information": "%s"' % '.'.join([str(ord(_)) for _ in self.path_info])
 
31
 
 
32
        def __str__ (self):
 
33
                if self.path_info:
 
34
                        return ' path-information %s' % '.'.join([str(ord(_)) for _ in self.path_info])
 
35
                return ''
 
36
 
 
37
        def pack (self):
 
38
                if self.path_info:
 
39
                        return self.path_info
 
40
                return '\x00\x00\x00\x00'
 
41
 
 
42
_NoPathInfo = PathInfo()
 
43
 
 
44
 
 
45
class Labels (object):
 
46
        biggest = pow(2,20)
 
47
 
 
48
        def __init__ (self,labels):
 
49
                self.labels = labels
 
50
                packed = []
 
51
                for label in labels:
 
52
                        # shift to 20 bits of the label to be at the top of three bytes and then truncate.
 
53
                        packed.append(pack('!L',label << 4)[1:])
 
54
                # Mark the bottom of stack with the bit
 
55
                if packed:
 
56
                        packed.pop()
 
57
                        packed.append(pack('!L',(label << 4)|1)[1:])
 
58
                self.packed = ''.join(packed)
 
59
                self._len = len(self.packed)
 
60
 
 
61
        def pack (self):
 
62
                return self.packed
 
63
 
 
64
        def __len__ (self):
 
65
                return self._len
 
66
 
 
67
        def json (self):
 
68
                if self._len > 1:
 
69
                        return '"label": [ %s ]' % ', '.join([str(_) for _ in self.labels])
 
70
                else:
 
71
                        return ''
 
72
 
 
73
        def __str__ (self):
 
74
                if self._len > 1:
 
75
                        return ' label [ %s ]' % ' '.join([str(_) for _ in self.labels])
 
76
                elif self._len == 1:
 
77
                        return ' label %s' % self.labels[0]
 
78
                else:
 
79
                        return ''
 
80
 
 
81
_NoLabels = Labels([])
 
82
 
 
83
class RouteDistinguisher (object):
 
84
        def __init__ (self,rd):
 
85
                self.rd = rd
 
86
                self._len = len(self.rd)
 
87
 
 
88
        def pack (self):
 
89
                return self.rd
 
90
 
 
91
        def __len__ (self):
 
92
                return self._len
 
93
 
 
94
        def _str (self):
 
95
                t,c1,c2,c3 = unpack('!HHHH',self.rd)
 
96
                if t == 0:
 
97
                        rd = '%d:%d' % (c1,(c2<<16)+c3)
 
98
                elif t == 1:
 
99
                        rd = '%d.%d.%d.%d:%d' % (c1>>8,c1&0xFF,c2>>8,c2&0xFF,c3)
 
100
                elif t == 2:
 
101
                        rd = '%d:%d' % ((c1<<16)+c2,c3)
 
102
                else:
 
103
                        rd = str(self.rd)
 
104
                return rd
 
105
 
 
106
        def json (self):
 
107
                if not self.rd:
 
108
                        return ''
 
109
                return '"route-distinguisher": "%s"' % self._str()
 
110
 
 
111
        def __str__ (self):
 
112
                if not self.rd:
 
113
                        return ''
 
114
                return ' route-distinguisher %s' % self._str()
 
115
 
 
116
_NoRD = RouteDistinguisher('')
 
117
 
 
118
 
 
119
class NLRI (Prefix):
 
120
        def __init__(self,afi,safi,packed,mask,nexthop,action):
 
121
                self.path_info = _NoPathInfo
 
122
                self.labels = _NoLabels
 
123
                self.rd = _NoRD
 
124
                self.nexthop = nexthop
 
125
                self.action = action
 
126
 
 
127
                Prefix.__init__(self,afi,safi,packed,mask)
 
128
 
 
129
        def has_label (self):
 
130
                if self.afi == AFI.ipv4 and self.safi in (SAFI.nlri_mpls,SAFI.mpls_vpn):
 
131
                        return True
 
132
                if self.afi == AFI.ipv6 and self.safi == SAFI.mpls_vpn:
 
133
                        return True
 
134
                return False
 
135
 
 
136
        def nlri (self):
 
137
                return "%s%s%s%s" % (self.prefix(),str(self.labels),str(self.path_info),str(self.rd))
 
138
 
 
139
        def __len__ (self):
 
140
                prefix_len = len(self.path_info) + len(self.labels) + len(self.rd)
 
141
                return 1 + prefix_len + mask_to_bytes[self.mask]
 
142
 
 
143
        def __str__ (self):
 
144
                nexthop = ' next-hop %s' % self.nexthop.inet() if self.nexthop else ''
 
145
                return "%s%s" % (self.nlri(),nexthop)
 
146
 
 
147
        def __eq__ (self,other):
 
148
                return str(self) == str(other)
 
149
 
 
150
        def __ne__ (self,other):
 
151
                return not self.__eq__(other)
 
152
 
 
153
        def json (self,announced=True):
 
154
                label = self.labels.json()
 
155
                pinfo = self.path_info.json()
 
156
                rdist = self.rd.json()
 
157
 
 
158
                r = []
 
159
                if announced:
 
160
                        if self.labels: r.append(label)
 
161
                        if self.rd: r.append(rdist)
 
162
                if self.path_info: r.append(pinfo)
 
163
                return '"%s": { %s }' % (self.prefix(),", ".join(r))
 
164
 
 
165
        def pack (self,addpath):
 
166
                if addpath:
 
167
                        path_info = self.path_info.pack()
 
168
                else:
 
169
                        path_info = ''
 
170
 
 
171
                if self.has_label():
 
172
                        length = len(self.labels)*8 + len(self.rd)*8 + self.mask
 
173
                        return path_info + chr(length) + self.labels.pack() + self.rd.pack() + self.packed[:mask_to_bytes[self.mask]]
 
174
                else:
 
175
                        return path_info + Prefix.pack(self)
 
176
 
 
177
        def index (self):
 
178
                return self.pack(True)+self.rd.rd+self.path_info.path_info