1
|
|
|
"""Classes used in the main application.""" |
2
|
1 |
|
from datetime import datetime |
3
|
1 |
|
from uuid import uuid4 |
4
|
|
|
|
5
|
1 |
|
import requests |
6
|
|
|
|
7
|
1 |
|
from kytos.core import log |
8
|
1 |
|
from kytos.core.common import EntityStatus, GenericEntity |
9
|
1 |
|
from kytos.core.exceptions import KytosNoTagAvailableError |
10
|
1 |
|
from kytos.core.helpers import get_time, now |
11
|
1 |
|
from kytos.core.interface import UNI |
12
|
1 |
|
from kytos.core.link import Link |
13
|
1 |
|
from napps.kytos.mef_eline import settings |
14
|
1 |
|
from napps.kytos.mef_eline.storehouse import StoreHouse |
15
|
|
|
|
16
|
|
|
|
17
|
1 |
|
class Path(list, GenericEntity): |
18
|
|
|
"""Class to represent a Path.""" |
19
|
|
|
|
20
|
1 |
|
def __eq__(self, other=None): |
21
|
|
|
"""Compare paths.""" |
22
|
1 |
|
if not other or not isinstance(other, Path): |
23
|
1 |
|
return False |
24
|
1 |
|
return super().__eq__(other) |
25
|
|
|
|
26
|
1 |
|
def is_affected_by_link(self, link=None): |
27
|
|
|
"""Verify if the current path is affected by link.""" |
28
|
1 |
|
if not link: |
29
|
|
|
return False |
30
|
1 |
|
return link in self |
31
|
|
|
|
32
|
1 |
|
def link_affected_by_interface(self, interface=None): |
33
|
|
|
"""Return the link using this interface, if any, or None otherwise.""" |
34
|
|
|
if not interface: |
35
|
|
|
return None |
36
|
|
|
for link in self: |
37
|
|
|
if interface in (link.endpoint_a, link.endpoint_b): |
38
|
|
|
return link |
39
|
|
|
return None |
40
|
|
|
|
41
|
1 |
|
def choose_vlans(self): |
42
|
|
|
"""Choose the VLANs to be used for the circuit.""" |
43
|
1 |
|
for link in self: |
44
|
1 |
|
tag = link.get_next_available_tag() |
45
|
1 |
|
link.use_tag(tag) |
46
|
1 |
|
link.add_metadata('s_vlan', tag) |
47
|
|
|
|
48
|
1 |
|
def make_vlans_available(self): |
49
|
|
|
"""Make the VLANs used in a path available when undeployed.""" |
50
|
1 |
|
for link in self: |
51
|
1 |
|
link.make_tag_available(link.get_metadata('s_vlan')) |
52
|
1 |
|
link.remove_metadata('s_vlan') |
53
|
|
|
|
54
|
1 |
|
@property |
55
|
|
|
def status(self): |
56
|
|
|
"""Check for the status of a path. |
57
|
|
|
|
58
|
|
|
If any link in this path is down, the path is considered down. |
59
|
|
|
""" |
60
|
1 |
|
if not self: |
61
|
1 |
|
return EntityStatus.DISABLED |
62
|
|
|
|
63
|
1 |
|
endpoint = '%s/%s' % (settings.TOPOLOGY_URL, 'links') |
64
|
1 |
|
api_reply = requests.get(endpoint) |
65
|
1 |
|
if api_reply.status_code != getattr(requests.codes, 'ok'): |
66
|
|
|
log.error('Failed to get links at %s. Returned %s', |
67
|
|
|
endpoint, api_reply.status_code) |
68
|
|
|
return None |
69
|
1 |
|
links = api_reply.json()['links'] |
70
|
1 |
|
return_status = EntityStatus.UP |
71
|
1 |
|
for path_link in self: |
72
|
1 |
|
try: |
73
|
1 |
|
link = links[path_link.id] |
74
|
|
|
except KeyError: |
75
|
|
|
return EntityStatus.DISABLED |
76
|
1 |
|
if link['enabled'] is False: |
77
|
1 |
|
return EntityStatus.DISABLED |
78
|
1 |
|
if link['active'] is False: |
79
|
1 |
|
return_status = EntityStatus.DOWN |
80
|
1 |
|
return return_status |
81
|
|
|
|
82
|
1 |
|
def as_dict(self): |
83
|
|
|
"""Return list comprehension of links as_dict.""" |
84
|
1 |
|
return [link.as_dict() for link in self if link] |
85
|
|
|
|
86
|
|
|
|
87
|
1 |
|
class DynamicPathManager: |
88
|
|
|
"""Class to handle and create paths.""" |
89
|
|
|
|
90
|
1 |
|
controller = None |
91
|
|
|
|
92
|
1 |
|
@classmethod |
93
|
1 |
|
def set_controller(cls, controller=None): |
94
|
|
|
"""Set the controller to discovery news paths.""" |
95
|
1 |
|
cls.controller = controller |
96
|
|
|
|
97
|
1 |
|
@staticmethod |
98
|
|
|
def get_paths(circuit): |
99
|
|
|
"""Get a valid path for the circuit from the Pathfinder.""" |
100
|
|
|
endpoint = settings.PATHFINDER_URL |
101
|
|
|
request_data = {"source": circuit.uni_a.interface.id, |
102
|
|
|
"destination": circuit.uni_z.interface.id} |
103
|
|
|
api_reply = requests.post(endpoint, json=request_data) |
104
|
|
|
|
105
|
|
|
if api_reply.status_code != getattr(requests.codes, 'ok'): |
106
|
|
|
log.error("Failed to get paths at %s. Returned %s", |
107
|
|
|
endpoint, api_reply.status_code) |
108
|
|
|
return None |
109
|
|
|
reply_data = api_reply.json() |
110
|
|
|
return reply_data.get('paths') |
111
|
|
|
|
112
|
1 |
|
@staticmethod |
113
|
|
|
def _clear_path(path): |
114
|
|
|
"""Remove switches from a path, returning only interfaces.""" |
115
|
|
|
return [endpoint for endpoint in path if len(endpoint) > 23] |
116
|
|
|
|
117
|
1 |
|
@classmethod |
118
|
|
|
def get_best_path(cls, circuit): |
119
|
|
|
"""Return the best path available for a circuit, if exists.""" |
120
|
|
|
paths = cls.get_paths(circuit) |
121
|
|
|
if paths: |
122
|
|
|
return cls.create_path(cls.get_paths(circuit)[0]['hops']) |
123
|
|
|
return None |
124
|
|
|
|
125
|
1 |
|
@classmethod |
126
|
|
|
def get_best_paths(cls, circuit): |
127
|
|
|
"""Return the best paths available for a circuit, if they exist.""" |
128
|
|
|
for path in cls.get_paths(circuit): |
129
|
|
|
yield cls.create_path(path['hops']) |
130
|
|
|
|
131
|
1 |
|
@classmethod |
132
|
|
|
def create_path(cls, path): |
133
|
|
|
"""Return the path containing only the interfaces.""" |
134
|
|
|
new_path = Path() |
135
|
|
|
clean_path = cls._clear_path(path) |
136
|
|
|
|
137
|
|
|
if len(clean_path) % 2: |
138
|
|
|
return None |
139
|
|
|
|
140
|
|
|
for link in zip(clean_path[1:-1:2], clean_path[2::2]): |
141
|
|
|
interface_a = cls.controller.get_interface_by_id(link[0]) |
142
|
|
|
interface_b = cls.controller.get_interface_by_id(link[1]) |
143
|
|
|
if interface_a is None or interface_b is None: |
144
|
|
|
return None |
145
|
|
|
new_path.append(Link(interface_a, interface_b)) |
146
|
|
|
|
147
|
|
|
return new_path |
148
|
|
|
|
149
|
|
|
|
150
|
1 |
|
class EVCBase(GenericEntity): |
151
|
|
|
"""Class to represent a circuit.""" |
152
|
|
|
|
153
|
1 |
|
unique_attributes = ['name', 'uni_a', 'uni_z'] |
154
|
|
|
|
155
|
1 |
|
def __init__(self, controller, **kwargs): |
156
|
|
|
"""Create an EVC instance with the provided parameters. |
157
|
|
|
|
158
|
|
|
Args: |
159
|
|
|
id(str): EVC identifier. Whether it's None an ID will be genereted. |
160
|
|
|
name: represents an EVC name.(Required) |
161
|
|
|
uni_a (UNI): Endpoint A for User Network Interface.(Required) |
162
|
|
|
uni_z (UNI): Endpoint Z for User Network Interface.(Required) |
163
|
|
|
start_date(datetime|str): Date when the EVC was registred. |
164
|
|
|
Default is now(). |
165
|
|
|
end_date(datetime|str): Final date that the EVC will be fineshed. |
166
|
|
|
Default is None. |
167
|
|
|
bandwidth(int): Bandwidth used by EVC instance. Default is 0. |
168
|
|
|
primary_links(list): Primary links used by evc. Default is [] |
169
|
|
|
backup_links(list): Backups links used by evc. Default is [] |
170
|
|
|
current_path(list): Circuit being used at the moment if this is an |
171
|
|
|
active circuit. Default is []. |
172
|
|
|
primary_path(list): primary circuit offered to user IF one or more |
173
|
|
|
links were provided. Default is []. |
174
|
|
|
backup_path(list): backup circuit offered to the user IF one or |
175
|
|
|
more links were provided. Default is []. |
176
|
|
|
dynamic_backup_path(bool): Enable computer backup path dynamically. |
177
|
|
|
Dafault is False. |
178
|
|
|
creation_time(datetime|str): datetime when the circuit should be |
179
|
|
|
activated. default is now(). |
180
|
|
|
enabled(Boolean): attribute to indicate the administrative state; |
181
|
|
|
default is False. |
182
|
|
|
active(Boolean): attribute to indicate the operational state; |
183
|
|
|
default is False. |
184
|
|
|
archived(Boolean): indicate the EVC has been deleted and is |
185
|
|
|
archived; default is False. |
186
|
|
|
owner(str): The EVC owner. Default is None. |
187
|
|
|
priority(int): Service level provided in the request. Default is 0. |
188
|
|
|
|
189
|
|
|
Raises: |
190
|
|
|
ValueError: raised when object attributes are invalid. |
191
|
|
|
|
192
|
|
|
""" |
193
|
1 |
|
self._validate(**kwargs) |
194
|
1 |
|
super().__init__() |
195
|
|
|
|
196
|
|
|
# required attributes |
197
|
1 |
|
self._id = kwargs.get('id', uuid4().hex[:16]) |
198
|
1 |
|
self.uni_a = kwargs.get('uni_a') |
199
|
1 |
|
self.uni_z = kwargs.get('uni_z') |
200
|
1 |
|
self.name = kwargs.get('name') |
201
|
|
|
|
202
|
|
|
# optional attributes |
203
|
1 |
|
self.start_date = get_time(kwargs.get('start_date')) or now() |
204
|
1 |
|
self.end_date = get_time(kwargs.get('end_date')) or None |
205
|
|
|
|
206
|
1 |
|
self.bandwidth = kwargs.get('bandwidth', 0) |
207
|
1 |
|
self.primary_links = Path(kwargs.get('primary_links', [])) |
208
|
1 |
|
self.backup_links = Path(kwargs.get('backup_links', [])) |
209
|
1 |
|
self.current_path = Path(kwargs.get('current_path', [])) |
210
|
1 |
|
self.primary_path = Path(kwargs.get('primary_path', [])) |
211
|
1 |
|
self.backup_path = Path(kwargs.get('backup_path', [])) |
212
|
1 |
|
self.dynamic_backup_path = kwargs.get('dynamic_backup_path', False) |
213
|
1 |
|
self.creation_time = get_time(kwargs.get('creation_time')) or now() |
214
|
1 |
|
self.owner = kwargs.get('owner', None) |
215
|
1 |
|
self.priority = kwargs.get('priority', 0) |
216
|
1 |
|
self.circuit_scheduler = kwargs.get('circuit_scheduler', []) |
217
|
|
|
|
218
|
1 |
|
self.current_links_cache = set() |
219
|
1 |
|
self.primary_links_cache = set() |
220
|
1 |
|
self.backup_links_cache = set() |
221
|
|
|
|
222
|
1 |
|
self.archived = kwargs.get('archived', False) |
223
|
|
|
|
224
|
1 |
|
self._storehouse = StoreHouse(controller) |
225
|
|
|
|
226
|
1 |
|
if kwargs.get('active', False): |
227
|
1 |
|
self.activate() |
228
|
|
|
else: |
229
|
1 |
|
self.deactivate() |
230
|
|
|
|
231
|
1 |
|
if kwargs.get('enabled', False): |
232
|
1 |
|
self.enable() |
233
|
|
|
else: |
234
|
1 |
|
self.disable() |
235
|
|
|
|
236
|
|
|
# datetime of user request for a EVC (or datetime when object was |
237
|
|
|
# created) |
238
|
1 |
|
self.request_time = kwargs.get('request_time', now()) |
239
|
|
|
# dict with the user original request (input) |
240
|
1 |
|
self._requested = kwargs |
241
|
|
|
|
242
|
1 |
|
def sync(self): |
243
|
|
|
"""Sync this EVC in the storehouse.""" |
244
|
1 |
|
self._storehouse.save_evc(self) |
245
|
1 |
|
log.info(f'EVC {self.id} was synced to the storehouse.') |
246
|
|
|
|
247
|
1 |
|
def update(self, **kwargs): |
248
|
|
|
"""Update evc attributes. |
249
|
|
|
|
250
|
|
|
This method will raises an error trying to change the following |
251
|
|
|
attributes: [name, uni_a and uni_z] |
252
|
|
|
|
253
|
|
|
Returns: |
254
|
|
|
the values for enable and a path attribute, if exists and None |
255
|
|
|
otherwise |
256
|
|
|
Raises: |
257
|
|
|
ValueError: message with error detail. |
258
|
|
|
|
259
|
|
|
""" |
260
|
1 |
|
enable, path = (None, None) |
261
|
1 |
|
for attribute, value in kwargs.items(): |
262
|
1 |
|
if attribute in self.unique_attributes: |
263
|
1 |
|
raise ValueError(f'{attribute} can\'t be be updated.') |
264
|
1 |
|
if hasattr(self, attribute): |
265
|
1 |
|
if attribute in ('enable', 'enabled'): |
266
|
1 |
|
if value: |
267
|
1 |
|
self.enable() |
268
|
|
|
else: |
269
|
|
|
self.disable() |
270
|
1 |
|
enable = value |
271
|
1 |
|
elif attribute in ('active', 'activate'): |
272
|
|
|
if value: |
273
|
|
|
self.activate() |
274
|
|
|
else: |
275
|
|
|
self.deactivate() |
276
|
|
|
else: |
277
|
1 |
|
setattr(self, attribute, value) |
278
|
1 |
|
if 'path' in attribute: |
279
|
1 |
|
path = value |
280
|
|
|
else: |
281
|
|
|
raise ValueError(f'The attribute "{attribute}" is invalid.') |
282
|
1 |
|
self.sync() |
283
|
1 |
|
return enable, path |
284
|
|
|
|
285
|
1 |
|
def __repr__(self): |
286
|
|
|
"""Repr method.""" |
287
|
1 |
|
return f"EVC({self._id}, {self.name})" |
288
|
|
|
|
289
|
1 |
|
def _validate(self, **kwargs): |
290
|
|
|
"""Do Basic validations. |
291
|
|
|
|
292
|
|
|
Verify required attributes: name, uni_a, uni_z |
293
|
|
|
Verify if the attributes uni_a and uni_z are valid. |
294
|
|
|
|
295
|
|
|
Raises: |
296
|
|
|
ValueError: message with error detail. |
297
|
|
|
|
298
|
|
|
""" |
299
|
1 |
|
for attribute in self.unique_attributes: |
300
|
|
|
|
301
|
1 |
|
if attribute not in kwargs: |
302
|
1 |
|
raise ValueError(f'{attribute} is required.') |
303
|
|
|
|
304
|
1 |
|
if 'uni' in attribute: |
305
|
1 |
|
uni = kwargs.get(attribute) |
306
|
1 |
|
if not isinstance(uni, UNI): |
307
|
|
|
raise ValueError(f'{attribute} is an invalid UNI.') |
308
|
|
|
|
309
|
1 |
|
if not uni.is_valid(): |
310
|
1 |
|
tag = uni.user_tag.value |
311
|
1 |
|
message = f'VLAN tag {tag} is not available in {attribute}' |
312
|
1 |
|
raise ValueError(message) |
313
|
|
|
|
314
|
1 |
|
def __eq__(self, other): |
315
|
|
|
"""Override the default implementation.""" |
316
|
1 |
|
if not isinstance(other, EVC): |
317
|
|
|
return False |
318
|
|
|
|
319
|
1 |
|
attrs_to_compare = ['name', 'uni_a', 'uni_z', 'owner', 'bandwidth'] |
320
|
1 |
|
for attribute in attrs_to_compare: |
321
|
1 |
|
if getattr(other, attribute) != getattr(self, attribute): |
322
|
1 |
|
return False |
323
|
1 |
|
return True |
324
|
|
|
|
325
|
1 |
|
def as_dict(self): |
326
|
|
|
"""Return a dictionary representing an EVC object.""" |
327
|
1 |
|
evc_dict = {"id": self.id, "name": self.name, |
328
|
|
|
"uni_a": self.uni_a.as_dict(), |
329
|
|
|
"uni_z": self.uni_z.as_dict()} |
330
|
|
|
|
331
|
1 |
|
time_fmt = "%Y-%m-%dT%H:%M:%S" |
332
|
|
|
|
333
|
1 |
|
evc_dict["start_date"] = self.start_date |
334
|
1 |
|
if isinstance(self.start_date, datetime): |
335
|
1 |
|
evc_dict["start_date"] = self.start_date.strftime(time_fmt) |
336
|
|
|
|
337
|
1 |
|
evc_dict["end_date"] = self.end_date |
338
|
1 |
|
if isinstance(self.end_date, datetime): |
339
|
1 |
|
evc_dict["end_date"] = self.end_date.strftime(time_fmt) |
340
|
|
|
|
341
|
1 |
|
evc_dict['bandwidth'] = self.bandwidth |
342
|
1 |
|
evc_dict['primary_links'] = self.primary_links.as_dict() |
343
|
1 |
|
evc_dict['backup_links'] = self.backup_links.as_dict() |
344
|
1 |
|
evc_dict['current_path'] = self.current_path.as_dict() |
345
|
1 |
|
evc_dict['primary_path'] = self.primary_path.as_dict() |
346
|
1 |
|
evc_dict['backup_path'] = self.backup_path.as_dict() |
347
|
1 |
|
evc_dict['dynamic_backup_path'] = self.dynamic_backup_path |
348
|
|
|
|
349
|
|
|
# if self._requested: |
350
|
|
|
# request_dict = self._requested.copy() |
351
|
|
|
# request_dict['uni_a'] = request_dict['uni_a'].as_dict() |
352
|
|
|
# request_dict['uni_z'] = request_dict['uni_z'].as_dict() |
353
|
|
|
# request_dict['circuit_scheduler'] = self.circuit_scheduler |
354
|
|
|
# evc_dict['_requested'] = request_dict |
355
|
|
|
|
356
|
1 |
|
evc_dict["request_time"] = self.request_time |
357
|
1 |
|
if isinstance(self.request_time, datetime): |
358
|
1 |
|
evc_dict["request_time"] = self.request_time.strftime(time_fmt) |
359
|
|
|
|
360
|
1 |
|
time = self.creation_time.strftime(time_fmt) |
361
|
1 |
|
evc_dict['creation_time'] = time |
362
|
|
|
|
363
|
1 |
|
evc_dict['owner'] = self.owner |
364
|
1 |
|
evc_dict['circuit_scheduler'] = [sc.as_dict() |
365
|
|
|
for sc in self.circuit_scheduler] |
366
|
|
|
|
367
|
1 |
|
evc_dict['active'] = self.is_active() |
368
|
1 |
|
evc_dict['enabled'] = self.is_enabled() |
369
|
1 |
|
evc_dict['archived'] = self.archived |
370
|
1 |
|
evc_dict['priority'] = self.priority |
371
|
|
|
|
372
|
1 |
|
return evc_dict |
373
|
|
|
|
374
|
1 |
|
@property |
375
|
|
|
def id(self): # pylint: disable=invalid-name |
376
|
|
|
"""Return this EVC's ID.""" |
377
|
1 |
|
return self._id |
378
|
|
|
|
379
|
1 |
|
def archive(self): |
380
|
|
|
"""Archive this EVC on deletion.""" |
381
|
1 |
|
self.archived = True |
382
|
|
|
|
383
|
|
|
|
384
|
|
|
# pylint: disable=fixme, too-many-public-methods |
385
|
1 |
|
class EVCDeploy(EVCBase): |
386
|
|
|
"""Class to handle the deploy procedures.""" |
387
|
|
|
|
388
|
1 |
|
def create(self): |
389
|
|
|
"""Create a EVC.""" |
390
|
|
|
|
391
|
1 |
|
def discover_new_paths(self): |
392
|
|
|
"""Discover new paths to satisfy this circuit and deploy it.""" |
393
|
|
|
return DynamicPathManager.get_best_paths(self) |
394
|
|
|
|
395
|
1 |
|
def change_path(self): |
396
|
|
|
"""Change EVC path.""" |
397
|
|
|
|
398
|
1 |
|
def reprovision(self): |
399
|
|
|
"""Force the EVC (re-)provisioning.""" |
400
|
|
|
|
401
|
1 |
|
def is_affected_by_link(self, link): |
402
|
|
|
"""Return True if this EVC has the given link on its current path.""" |
403
|
|
|
return link in self.current_path |
404
|
|
|
|
405
|
1 |
|
def link_affected_by_interface(self, interface): |
406
|
|
|
"""Return True if this EVC has the given link on its current path.""" |
407
|
|
|
return self.current_path.link_affected_by_interface(interface) |
408
|
|
|
|
409
|
1 |
|
def is_backup_path_affected_by_link(self, link): |
410
|
|
|
"""Return True if the backup path of this EVC uses the given link.""" |
411
|
|
|
return link in self.backup_path |
412
|
|
|
|
413
|
|
|
# pylint: disable=invalid-name |
414
|
1 |
|
def is_primary_path_affected_by_link(self, link): |
415
|
|
|
"""Return True if the primary path of this EVC uses the given link.""" |
416
|
|
|
return link in self.primary_path |
417
|
|
|
|
418
|
1 |
|
def is_using_primary_path(self): |
419
|
|
|
"""Verify if the current deployed path is self.primary_path.""" |
420
|
|
|
return self.primary_path and (self.current_path == self.primary_path) |
421
|
|
|
|
422
|
1 |
|
def is_using_backup_path(self): |
423
|
|
|
"""Verify if the current deployed path is self.backup_path.""" |
424
|
|
|
return self.backup_path and (self.current_path == self.backup_path) |
425
|
|
|
|
426
|
1 |
|
def is_using_dynamic_path(self): |
427
|
|
|
"""Verify if the current deployed path is a dynamic path.""" |
428
|
|
|
if self.current_path and \ |
429
|
|
|
not self.is_using_primary_path() and \ |
430
|
|
|
not self.is_using_backup_path() and \ |
431
|
|
|
self.current_path.status == EntityStatus.UP: |
432
|
|
|
return True |
433
|
|
|
return False |
434
|
|
|
|
435
|
1 |
|
def deploy_to_backup_path(self): |
436
|
|
|
"""Deploy the backup path into the datapaths of this circuit. |
437
|
|
|
|
438
|
|
|
If the backup_path attribute is valid and up, this method will try to |
439
|
|
|
deploy this backup_path. |
440
|
|
|
|
441
|
|
|
If everything fails and dynamic_backup_path is True, then tries to |
442
|
|
|
deploy a dynamic path. |
443
|
|
|
""" |
444
|
|
|
# TODO: Remove flows from current (cookies) |
445
|
1 |
|
if self.is_using_backup_path(): |
446
|
|
|
# TODO: Log to say that cannot move backup to backup |
447
|
|
|
return True |
448
|
|
|
|
449
|
1 |
|
success = False |
450
|
1 |
|
if self.backup_path.status is EntityStatus.UP: |
451
|
1 |
|
success = self.deploy_to_path(self.backup_path) |
452
|
|
|
|
453
|
1 |
|
if success: |
454
|
1 |
|
return True |
455
|
|
|
|
456
|
1 |
|
if self.dynamic_backup_path: |
457
|
1 |
|
return self.deploy_to_path() |
458
|
|
|
|
459
|
1 |
|
return False |
460
|
|
|
|
461
|
1 |
|
def deploy_to_primary_path(self): |
462
|
|
|
"""Deploy the primary path into the datapaths of this circuit. |
463
|
|
|
|
464
|
|
|
If the primary_path attribute is valid and up, this method will try to |
465
|
|
|
deploy this primary_path. |
466
|
|
|
""" |
467
|
|
|
# TODO: Remove flows from current (cookies) |
468
|
1 |
|
if self.is_using_primary_path(): |
469
|
|
|
# TODO: Log to say that cannot move primary to primary |
470
|
|
|
return True |
471
|
|
|
|
472
|
1 |
|
if self.primary_path.status is EntityStatus.UP: |
473
|
1 |
|
return self.deploy_to_path(self.primary_path) |
474
|
1 |
|
return False |
475
|
|
|
|
476
|
1 |
|
def deploy(self): |
477
|
|
|
"""Deploy EVC to best path. |
478
|
|
|
|
479
|
|
|
Best path can be the primary path, if available. If not, the backup |
480
|
|
|
path, and, if it is also not available, a dynamic path. |
481
|
|
|
""" |
482
|
1 |
|
if self.archived: |
483
|
|
|
return False |
484
|
1 |
|
self.enable() |
485
|
1 |
|
success = self.deploy_to_primary_path() |
486
|
1 |
|
if not success: |
487
|
1 |
|
success = self.deploy_to_backup_path() |
488
|
|
|
|
489
|
1 |
|
return success |
490
|
|
|
|
491
|
1 |
|
@staticmethod |
492
|
|
|
def get_path_status(path): |
493
|
|
|
"""Check for the current status of a path. |
494
|
|
|
|
495
|
|
|
If any link in this path is down, the path is considered down. |
496
|
|
|
""" |
497
|
|
|
if not path: |
498
|
|
|
return EntityStatus.DISABLED |
499
|
|
|
|
500
|
|
|
for link in path: |
501
|
|
|
if link.status is not EntityStatus.UP: |
502
|
|
|
return link.status |
503
|
|
|
return EntityStatus.UP |
504
|
|
|
|
505
|
|
|
# def discover_new_path(self): |
506
|
|
|
# # TODO: discover a new path to satisfy this circuit and deploy |
507
|
|
|
|
508
|
1 |
|
def remove(self): |
509
|
|
|
"""Remove EVC path and disable it.""" |
510
|
1 |
|
self.remove_current_flows() |
511
|
1 |
|
self.disable() |
512
|
1 |
|
self.sync() |
513
|
|
|
|
514
|
1 |
|
def remove_current_flows(self): |
515
|
|
|
"""Remove all flows from current path.""" |
516
|
1 |
|
switches = set() |
517
|
|
|
|
518
|
1 |
|
switches.add(self.uni_a.interface.switch) |
519
|
1 |
|
switches.add(self.uni_z.interface.switch) |
520
|
1 |
|
for link in self.current_path: |
521
|
1 |
|
switches.add(link.endpoint_a.switch) |
522
|
1 |
|
switches.add(link.endpoint_b.switch) |
523
|
|
|
|
524
|
1 |
|
match = {'cookie': self.get_cookie(), |
525
|
|
|
'cookie_mask': 18446744073709551615} |
526
|
|
|
|
527
|
1 |
|
for switch in switches: |
528
|
1 |
|
self._send_flow_mods(switch, [match], 'delete') |
529
|
|
|
|
530
|
1 |
|
self.current_path.make_vlans_available() |
531
|
1 |
|
self.current_path = Path([]) |
532
|
1 |
|
self.deactivate() |
533
|
1 |
|
self.sync() |
534
|
|
|
|
535
|
1 |
|
@staticmethod |
536
|
1 |
|
def links_zipped(path=None): |
537
|
|
|
"""Return an iterator which yields pairs of links in order.""" |
538
|
1 |
|
if not path: |
539
|
|
|
return [] |
540
|
1 |
|
return zip(path[:-1], path[1:]) |
541
|
|
|
|
542
|
1 |
|
def should_deploy(self, path=None): |
543
|
|
|
"""Verify if the circuit should be deployed.""" |
544
|
1 |
|
if not path: |
545
|
1 |
|
log.debug("Path is empty.") |
546
|
1 |
|
return False |
547
|
|
|
|
548
|
1 |
|
if not self.is_enabled(): |
549
|
1 |
|
log.debug(f'{self} is disabled.') |
550
|
1 |
|
return False |
551
|
|
|
|
552
|
1 |
|
if not self.is_active(): |
553
|
1 |
|
log.debug(f'{self} will be deployed.') |
554
|
1 |
|
return True |
555
|
|
|
|
556
|
1 |
|
return False |
557
|
|
|
|
558
|
1 |
|
def deploy_to_path(self, path=None): |
559
|
|
|
"""Install the flows for this circuit. |
560
|
|
|
|
561
|
|
|
Procedures to deploy: |
562
|
|
|
|
563
|
|
|
0. Remove current flows installed |
564
|
|
|
1. Decide if will deploy "path" or discover a new path |
565
|
|
|
2. Choose vlan |
566
|
|
|
3. Install NNI flows |
567
|
|
|
4. Install UNI flows |
568
|
|
|
5. Activate |
569
|
|
|
6. Update current_path |
570
|
|
|
7. Update links caches(primary, current, backup) |
571
|
|
|
|
572
|
|
|
""" |
573
|
1 |
|
self.remove_current_flows() |
574
|
1 |
|
use_path = path |
575
|
1 |
|
if self.should_deploy(use_path): |
576
|
1 |
|
try: |
577
|
1 |
|
use_path.choose_vlans() |
578
|
|
|
except KytosNoTagAvailableError: |
579
|
|
|
use_path = None |
580
|
|
|
else: |
581
|
1 |
|
for use_path in self.discover_new_paths(): |
582
|
1 |
|
try: |
583
|
1 |
|
use_path.choose_vlans() |
584
|
1 |
|
break |
585
|
|
|
except KytosNoTagAvailableError: |
586
|
|
|
pass |
587
|
|
|
else: |
588
|
1 |
|
use_path = None |
589
|
|
|
|
590
|
1 |
|
if use_path: |
591
|
1 |
|
self._install_nni_flows(use_path) |
592
|
1 |
|
self._install_uni_flows(use_path) |
593
|
1 |
|
self.activate() |
594
|
1 |
|
self.current_path = use_path |
595
|
1 |
|
self.sync() |
596
|
1 |
|
log.info(f"{self} was deployed.") |
597
|
1 |
|
return True |
598
|
1 |
|
return False |
599
|
|
|
|
600
|
1 |
|
def _install_direct_uni_flows(self): |
601
|
|
|
"""Install flows connecting two UNIs. |
602
|
|
|
|
603
|
|
|
This case happens when the circuit is between UNIs in the |
604
|
|
|
same switch. |
605
|
|
|
""" |
606
|
|
|
vlan_a = self.uni_a.user_tag.value if self.uni_a.user_tag else None |
607
|
|
|
vlan_z = self.uni_z.user_tag.value if self.uni_z.user_tag else None |
608
|
|
|
|
609
|
|
|
flow_mod_az = self._prepare_flow_mod(self.uni_a.interface, |
610
|
|
|
self.uni_z.interface) |
611
|
|
|
flow_mod_za = self._prepare_flow_mod(self.uni_z.interface, |
612
|
|
|
self.uni_a.interface) |
613
|
|
|
|
614
|
|
|
if vlan_a and vlan_z: |
615
|
|
|
flow_mod_az['match']['dl_vlan'] = vlan_a |
616
|
|
|
flow_mod_za['match']['dl_vlan'] = vlan_z |
617
|
|
|
flow_mod_az['actions'].insert(0, {'action_type': 'set_vlan', |
618
|
|
|
'vlan_id': vlan_z}) |
619
|
|
|
flow_mod_za['actions'].insert(0, {'action_type': 'set_vlan', |
620
|
|
|
'vlan_id': vlan_a}) |
621
|
|
|
elif vlan_a: |
622
|
|
|
flow_mod_az['match']['dl_vlan'] = vlan_a |
623
|
|
|
flow_mod_az['actions'].insert(0, {'action_type': 'pop_vlan'}) |
624
|
|
|
flow_mod_za['actions'].insert(0, {'action_type': 'set_vlan', |
625
|
|
|
'vlan_id': vlan_a}) |
626
|
|
|
elif vlan_z: |
627
|
|
|
flow_mod_za['match']['dl_vlan'] = vlan_z |
628
|
|
|
flow_mod_za['actions'].insert(0, {'action_type': 'pop_vlan'}) |
629
|
|
|
flow_mod_az['actions'].insert(0, {'action_type': 'set_vlan', |
630
|
|
|
'vlan_id': vlan_z}) |
631
|
|
|
self._send_flow_mods(self.uni_a.interface.switch, |
632
|
|
|
[flow_mod_az, flow_mod_za]) |
633
|
|
|
|
634
|
1 |
|
def _install_nni_flows(self, path=None): |
635
|
|
|
"""Install NNI flows.""" |
636
|
1 |
|
for incoming, outcoming in self.links_zipped(path): |
637
|
1 |
|
in_vlan = incoming.get_metadata('s_vlan').value |
638
|
1 |
|
out_vlan = outcoming.get_metadata('s_vlan').value |
639
|
|
|
|
640
|
1 |
|
flows = [] |
641
|
|
|
# Flow for one direction |
642
|
1 |
|
flows.append(self._prepare_nni_flow(incoming.endpoint_b, |
643
|
|
|
outcoming.endpoint_a, |
644
|
|
|
in_vlan, out_vlan)) |
645
|
|
|
|
646
|
|
|
# Flow for the other direction |
647
|
1 |
|
flows.append(self._prepare_nni_flow(outcoming.endpoint_a, |
648
|
|
|
incoming.endpoint_b, |
649
|
|
|
out_vlan, in_vlan)) |
650
|
1 |
|
self._send_flow_mods(incoming.endpoint_b.switch, flows) |
651
|
|
|
|
652
|
1 |
|
def _install_uni_flows(self, path=None): |
653
|
|
|
"""Install UNI flows.""" |
654
|
1 |
|
if not path: |
655
|
|
|
log.info('install uni flows without path.') |
656
|
|
|
return |
657
|
|
|
|
658
|
|
|
# Determine VLANs |
659
|
1 |
|
in_vlan_a = self.uni_a.user_tag.value if self.uni_a.user_tag else None |
660
|
1 |
|
out_vlan_a = path[0].get_metadata('s_vlan').value |
661
|
|
|
|
662
|
1 |
|
in_vlan_z = self.uni_z.user_tag.value if self.uni_z.user_tag else None |
663
|
1 |
|
out_vlan_z = path[-1].get_metadata('s_vlan').value |
664
|
|
|
|
665
|
|
|
# Flows for the first UNI |
666
|
1 |
|
flows_a = [] |
667
|
|
|
|
668
|
|
|
# Flow for one direction, pushing the service tag |
669
|
1 |
|
push_flow = self._prepare_push_flow(self.uni_a.interface, |
670
|
|
|
path[0].endpoint_a, |
671
|
|
|
in_vlan_a, out_vlan_a, in_vlan_z) |
672
|
1 |
|
flows_a.append(push_flow) |
673
|
|
|
|
674
|
|
|
# Flow for the other direction, popping the service tag |
675
|
1 |
|
pop_flow = self._prepare_pop_flow(path[0].endpoint_a, |
676
|
|
|
self.uni_a.interface, out_vlan_a) |
677
|
1 |
|
flows_a.append(pop_flow) |
678
|
|
|
|
679
|
1 |
|
self._send_flow_mods(self.uni_a.interface.switch, flows_a) |
680
|
|
|
|
681
|
|
|
# Flows for the second UNI |
682
|
1 |
|
flows_z = [] |
683
|
|
|
|
684
|
|
|
# Flow for one direction, pushing the service tag |
685
|
1 |
|
push_flow = self._prepare_push_flow(self.uni_z.interface, |
686
|
|
|
path[-1].endpoint_b, |
687
|
|
|
in_vlan_z, out_vlan_z, in_vlan_a) |
688
|
1 |
|
flows_z.append(push_flow) |
689
|
|
|
|
690
|
|
|
# Flow for the other direction, popping the service tag |
691
|
1 |
|
pop_flow = self._prepare_pop_flow(path[-1].endpoint_b, |
692
|
|
|
self.uni_z.interface, out_vlan_z) |
693
|
1 |
|
flows_z.append(pop_flow) |
694
|
|
|
|
695
|
1 |
|
self._send_flow_mods(self.uni_z.interface.switch, flows_z) |
696
|
|
|
|
697
|
1 |
|
@staticmethod |
698
|
1 |
|
def _send_flow_mods(switch, flow_mods, command='flows'): |
699
|
|
|
"""Send a flow_mod list to a specific switch. |
700
|
|
|
|
701
|
|
|
Args: |
702
|
|
|
switch(Switch): The target of flows. |
703
|
|
|
flow_mods(dict): Python dictionary with flow_mods. |
704
|
|
|
command(str): By default is 'flows'. To remove a flow is 'remove'. |
705
|
|
|
|
706
|
|
|
""" |
707
|
1 |
|
endpoint = f'{settings.MANAGER_URL}/{command}/{switch.id}' |
708
|
|
|
|
709
|
1 |
|
data = {"flows": flow_mods} |
710
|
1 |
|
requests.post(endpoint, json=data) |
711
|
|
|
|
712
|
1 |
|
def get_cookie(self): |
713
|
|
|
"""Return the cookie integer from evc id.""" |
714
|
1 |
|
return int(self.id, 16) |
715
|
|
|
|
716
|
1 |
|
def _prepare_flow_mod(self, in_interface, out_interface): |
717
|
|
|
"""Prepare a common flow mod.""" |
718
|
1 |
|
default_action = {"action_type": "output", |
719
|
|
|
"port": out_interface.port_number} |
720
|
|
|
|
721
|
1 |
|
flow_mod = {"match": {"in_port": in_interface.port_number}, |
722
|
|
|
"cookie": self.get_cookie(), |
723
|
|
|
"actions": [default_action]} |
724
|
|
|
|
725
|
1 |
|
return flow_mod |
726
|
|
|
|
727
|
1 |
|
def _prepare_nni_flow(self, |
728
|
|
|
in_interface, out_interface, in_vlan, out_vlan): |
729
|
|
|
"""Create NNI flows.""" |
730
|
1 |
|
flow_mod = self._prepare_flow_mod(in_interface, out_interface) |
731
|
1 |
|
flow_mod['match']['dl_vlan'] = in_vlan |
732
|
|
|
|
733
|
1 |
|
new_action = {"action_type": "set_vlan", |
734
|
|
|
"vlan_id": out_vlan} |
735
|
1 |
|
flow_mod["actions"].insert(0, new_action) |
736
|
|
|
|
737
|
1 |
|
return flow_mod |
738
|
|
|
|
739
|
1 |
|
def _prepare_push_flow(self, *args): |
740
|
|
|
"""Prepare push flow. |
741
|
|
|
|
742
|
|
|
Arguments: |
743
|
|
|
in_interface(str): Interface input. |
744
|
|
|
out_interface(str): Interface output. |
745
|
|
|
in_vlan(str): Vlan input. |
746
|
|
|
out_vlan(str): Vlan output. |
747
|
|
|
new_in_vlan(str): Interface input. |
748
|
|
|
|
749
|
|
|
Return: |
750
|
|
|
dict: An python dictionary representing a FlowMod |
751
|
|
|
|
752
|
|
|
""" |
753
|
|
|
# assign all arguments |
754
|
1 |
|
in_interface, out_interface, in_vlan, out_vlan, new_in_vlan = args |
755
|
|
|
|
756
|
1 |
|
flow_mod = self._prepare_flow_mod(in_interface, out_interface) |
757
|
|
|
|
758
|
|
|
# the service tag must be always pushed |
759
|
1 |
|
new_action = {"action_type": "set_vlan", "vlan_id": out_vlan} |
760
|
1 |
|
flow_mod["actions"].insert(0, new_action) |
761
|
|
|
|
762
|
1 |
|
new_action = {"action_type": "push_vlan", "tag_type": "s"} |
763
|
1 |
|
flow_mod["actions"].insert(0, new_action) |
764
|
|
|
|
765
|
1 |
|
if in_vlan: |
766
|
|
|
# if in_vlan is set, it must be included in the match |
767
|
1 |
|
flow_mod['match']['dl_vlan'] = in_vlan |
768
|
1 |
|
if new_in_vlan: |
769
|
|
|
# new_in_vlan is set, so an action to set it is necessary |
770
|
1 |
|
new_action = {"action_type": "set_vlan", "vlan_id": new_in_vlan} |
771
|
1 |
|
flow_mod["actions"].insert(0, new_action) |
772
|
1 |
|
if not in_vlan: |
773
|
|
|
# new_in_vlan is set, but in_vlan is not, so there was no |
774
|
|
|
# vlan set; then it is set now |
775
|
1 |
|
new_action = {"action_type": "push_vlan", "tag_type": "c"} |
776
|
1 |
|
flow_mod["actions"].insert(0, new_action) |
777
|
1 |
|
elif in_vlan: |
778
|
|
|
# in_vlan is set, but new_in_vlan is not, so the existing vlan |
779
|
|
|
# must be removed |
780
|
1 |
|
new_action = {"action_type": "pop_vlan"} |
781
|
1 |
|
flow_mod["actions"].insert(0, new_action) |
782
|
1 |
|
return flow_mod |
783
|
|
|
|
784
|
1 |
|
def _prepare_pop_flow(self, in_interface, out_interface, out_vlan): |
785
|
|
|
"""Prepare pop flow.""" |
786
|
1 |
|
flow_mod = self._prepare_flow_mod(in_interface, out_interface) |
787
|
1 |
|
flow_mod['match']['dl_vlan'] = out_vlan |
788
|
1 |
|
new_action = {"action_type": "pop_vlan"} |
789
|
1 |
|
flow_mod["actions"].insert(0, new_action) |
790
|
1 |
|
return flow_mod |
791
|
|
|
|
792
|
|
|
|
793
|
1 |
|
class LinkProtection(EVCDeploy): |
794
|
|
|
"""Class to handle link protection.""" |
795
|
|
|
|
796
|
1 |
|
def is_affected_by_link(self, link=None): |
797
|
|
|
"""Verify if the current path is affected by link down event.""" |
798
|
|
|
return self.current_path.is_affected_by_link(link) |
799
|
|
|
|
800
|
1 |
|
def is_using_primary_path(self): |
801
|
|
|
"""Verify if the current deployed path is self.primary_path.""" |
802
|
1 |
|
return self.current_path == self.primary_path |
803
|
|
|
|
804
|
1 |
|
def is_using_backup_path(self): |
805
|
|
|
"""Verify if the current deployed path is self.backup_path.""" |
806
|
1 |
|
return self.current_path == self.backup_path |
807
|
|
|
|
808
|
1 |
|
def is_using_dynamic_path(self): |
809
|
|
|
"""Verify if the current deployed path is dynamic.""" |
810
|
1 |
|
if self.current_path and \ |
811
|
|
|
not self.is_using_primary_path() and \ |
812
|
|
|
not self.is_using_backup_path() and \ |
813
|
|
|
self.current_path.status is EntityStatus.UP: |
814
|
|
|
return True |
815
|
1 |
|
return False |
816
|
|
|
|
817
|
1 |
|
def deploy_to(self, path_name=None, path=None): |
818
|
|
|
"""Create a deploy to path.""" |
819
|
1 |
|
if self.current_path == path: |
820
|
1 |
|
log.debug(f'{path_name} is equal to current_path.') |
821
|
1 |
|
return True |
822
|
|
|
|
823
|
1 |
|
if path.status is EntityStatus.UP: |
824
|
1 |
|
return self.deploy_to_path(path) |
825
|
|
|
|
826
|
1 |
|
return False |
827
|
|
|
|
828
|
1 |
|
def handle_link_up(self, link): |
829
|
|
|
"""Handle circuit when link down. |
830
|
|
|
|
831
|
|
|
Args: |
832
|
|
|
link(Link): Link affected by link.down event. |
833
|
|
|
|
834
|
|
|
""" |
835
|
1 |
|
if self.is_using_primary_path(): |
836
|
1 |
|
return True |
837
|
|
|
|
838
|
1 |
|
success = False |
839
|
1 |
|
if self.primary_path.is_affected_by_link(link): |
840
|
1 |
|
success = self.deploy_to_primary_path() |
841
|
|
|
|
842
|
1 |
|
if success: |
843
|
1 |
|
return True |
844
|
|
|
|
845
|
|
|
# We tried to deploy(primary_path) without success. |
846
|
|
|
# And in this case is up by some how. Nothing to do. |
847
|
1 |
|
if self.is_using_backup_path() or self.is_using_dynamic_path(): |
848
|
|
|
return True |
849
|
|
|
|
850
|
|
|
# In this case, probably the circuit is not being used and |
851
|
|
|
# we can move to backup |
852
|
1 |
|
if self.backup_path.is_affected_by_link(link): |
853
|
1 |
|
success = self.deploy_to_backup_path() |
854
|
|
|
|
855
|
1 |
|
if success: |
856
|
1 |
|
return True |
857
|
|
|
|
858
|
|
|
# In this case, the circuit is not being used and we should |
859
|
|
|
# try a dynamic path |
860
|
|
|
if self.dynamic_backup_path: |
861
|
|
|
return self.deploy_to_path() |
862
|
|
|
|
863
|
|
|
return True |
864
|
|
|
|
865
|
1 |
|
def handle_link_down(self): |
866
|
|
|
"""Handle circuit when link down. |
867
|
|
|
|
868
|
|
|
Returns: |
869
|
|
|
bool: True if the re-deploy was successly otherwise False. |
870
|
|
|
|
871
|
|
|
""" |
872
|
1 |
|
success = False |
873
|
1 |
|
if self.is_using_primary_path(): |
874
|
1 |
|
success = self.deploy_to_backup_path() |
875
|
1 |
|
elif self.is_using_backup_path(): |
876
|
1 |
|
success = self.deploy_to_primary_path() |
877
|
|
|
|
878
|
1 |
|
if not success and self.dynamic_backup_path: |
879
|
1 |
|
success = self.deploy_to_path() |
880
|
|
|
|
881
|
1 |
|
if success: |
882
|
1 |
|
log.debug(f"{self} deployed after link down.") |
883
|
|
|
else: |
884
|
1 |
|
self.deactivate() |
885
|
1 |
|
self.current_path = Path([]) |
886
|
1 |
|
self.sync() |
887
|
1 |
|
log.debug(f'Failed to re-deploy {self} after link down.') |
888
|
|
|
|
889
|
1 |
|
return success |
890
|
|
|
|
891
|
|
|
|
892
|
1 |
|
class EVC(LinkProtection): |
893
|
|
|
"""Class that represents a E-Line Virtual Connection.""" |
894
|
|
|
|