Passed
Push — master ( c72751...2e9109 )
by Aldo
05:05 queued 02:27
created

build.models.evc.EVCDeploy.check_trace()   D

Complexity

Conditions 12

Size

Total Lines 72
Code Lines 55

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 25
CRAP Score 12.3775

Importance

Changes 0
Metric Value
cc 12
eloc 55
nop 9
dl 0
loc 72
ccs 25
cts 29
cp 0.8621
crap 12.3775
rs 4.8
c 0
b 0
f 0

How to fix   Long Method    Complexity    Many Parameters   

Long Method

Small methods make your code easier to understand, in particular if combined with a good name. Besides, if your method is small, finding a good name is usually much easier.

For example, if you find yourself adding comments to a method's body, this is usually a good sign to extract the commented part to a new method, and use the comment as a starting point when coming up with a good name for this new method.

Commonly applied refactorings include:

Complexity

Complex classes like build.models.evc.EVCDeploy.check_trace() often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.

Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.

Many Parameters

Methods with many parameters are not only hard to understand, but their parameters also often become inconsistent when you need more, or different data.

There are several approaches to avoid long parameter lists:

1
"""Classes used in the main application."""  # pylint: disable=too-many-lines
2 1
import traceback
3 1
from collections import OrderedDict
4 1
from copy import deepcopy
5 1
from datetime import datetime
6 1
from operator import eq, ne
7 1
from threading import Lock
8 1
from typing import Union
9 1
from uuid import uuid4
10
11 1
import httpx
12 1
from glom import glom
13 1
from tenacity import (retry, retry_if_exception_type, stop_after_attempt,
14
                      wait_combine, wait_fixed, wait_random)
15
16 1
from kytos.core import log
17 1
from kytos.core.common import EntityStatus, GenericEntity
18 1
from kytos.core.exceptions import KytosNoTagAvailableError, KytosTagError
19 1
from kytos.core.helpers import get_time, now
20 1
from kytos.core.interface import UNI, Interface, TAGRange
21 1
from kytos.core.link import Link
22 1
from kytos.core.retry import before_sleep
23 1
from kytos.core.tag_ranges import range_difference
24 1
from napps.kytos.mef_eline import controllers, settings
25 1
from napps.kytos.mef_eline.exceptions import (DuplicatedNoTagUNI,
26
                                              EVCPathNotInstalled,
27
                                              FlowModException, InvalidPath)
28 1
from napps.kytos.mef_eline.utils import (check_disabled_component,
29
                                         compare_endpoint_trace,
30
                                         compare_uni_out_trace, emit_event,
31
                                         make_uni_list, map_dl_vlan,
32
                                         map_evc_event_content,
33
                                         merge_flow_dicts)
34
35 1
from .path import DynamicPathManager, Path
36
37
38 1
class EVCBase(GenericEntity):
39
    """Class to represent a circuit."""
40
41 1
    attributes_requiring_redeploy = [
42
        "primary_path",
43
        "backup_path",
44
        "dynamic_backup_path",
45
        "queue_id",
46
        "sb_priority",
47
        "primary_constraints",
48
        "secondary_constraints",
49
        "uni_a",
50
        "uni_z",
51
    ]
52 1
    required_attributes = ["name", "uni_a", "uni_z"]
53
54 1
    updatable_attributes = {
55
        "uni_a",
56
        "uni_z",
57
        "name",
58
        "start_date",
59
        "end_date",
60
        "queue_id",
61
        "bandwidth",
62
        "primary_path",
63
        "backup_path",
64
        "dynamic_backup_path",
65
        "primary_constraints",
66
        "secondary_constraints",
67
        "owner",
68
        "sb_priority",
69
        "service_level",
70
        "circuit_scheduler",
71
        "metadata",
72
        "enabled"
73
    }
74
75
    # pylint: disable=too-many-statements
76 1
    def __init__(self, controller, **kwargs):
77
        """Create an EVC instance with the provided parameters.
78
79
        Args:
80
            id(str): EVC identifier. Whether it's None an ID will be genereted.
81
                     Only the first 14 bytes passed will be used.
82
            name: represents an EVC name.(Required)
83
            uni_a (UNI): Endpoint A for User Network Interface.(Required)
84
            uni_z (UNI): Endpoint Z for User Network Interface.(Required)
85
            start_date(datetime|str): Date when the EVC was registred.
86
                                      Default is now().
87
            end_date(datetime|str): Final date that the EVC will be fineshed.
88
                                    Default is None.
89
            bandwidth(int): Bandwidth used by EVC instance. Default is 0.
90
            primary_links(list): Primary links used by evc. Default is []
91
            backup_links(list): Backups links used by evc. Default is []
92
            current_path(list): Circuit being used at the moment if this is an
93
                                active circuit. Default is [].
94
            failover_path(list): Path being used to provide EVC protection via
95
                                failover during link failures. Default is [].
96
            primary_path(list): primary circuit offered to user IF one or more
97
                                links were provided. Default is [].
98
            backup_path(list): backup circuit offered to the user IF one or
99
                               more links were provided. Default is [].
100
            dynamic_backup_path(bool): Enable computer backup path dynamically.
101
                                       Dafault is False.
102
            creation_time(datetime|str): datetime when the circuit should be
103
                                         activated. default is now().
104
            enabled(Boolean): attribute to indicate the administrative state;
105
                              default is False.
106
            active(Boolean): attribute to indicate the operational state;
107
                             default is False.
108
            archived(Boolean): indicate the EVC has been deleted and is
109
                               archived; default is False.
110
            owner(str): The EVC owner. Default is None.
111
            sb_priority(int): Service level provided in the request.
112
                              Default is None.
113
            service_level(int): Service level provided. The higher the better.
114
                                Default is 0.
115
116
        Raises:
117
            ValueError: raised when object attributes are invalid.
118
119
        """
120 1
        self._controller = controller
121 1
        self._validate(**kwargs)
122 1
        super().__init__()
123
124
        # required attributes
125 1
        self._id = kwargs.get("id", uuid4().hex)[:14]
126 1
        self.uni_a: UNI = kwargs.get("uni_a")
127 1
        self.uni_z: UNI = kwargs.get("uni_z")
128 1
        self.name = kwargs.get("name")
129
130
        # optional attributes
131 1
        self.start_date = get_time(kwargs.get("start_date")) or now()
132 1
        self.end_date = get_time(kwargs.get("end_date")) or None
133 1
        self.queue_id = kwargs.get("queue_id", -1)
134
135 1
        self.bandwidth = kwargs.get("bandwidth", 0)
136 1
        self.primary_links = Path(kwargs.get("primary_links", []))
137 1
        self.backup_links = Path(kwargs.get("backup_links", []))
138 1
        self.current_path = Path(kwargs.get("current_path", []))
139 1
        self.failover_path = Path(kwargs.get("failover_path", []))
140 1
        self.primary_path = Path(kwargs.get("primary_path", []))
141 1
        self.backup_path = Path(kwargs.get("backup_path", []))
142 1
        self.dynamic_backup_path = kwargs.get("dynamic_backup_path", False)
143 1
        self.primary_constraints = kwargs.get("primary_constraints", {})
144 1
        self.secondary_constraints = kwargs.get("secondary_constraints", {})
145 1
        self.creation_time = get_time(kwargs.get("creation_time")) or now()
146 1
        self.owner = kwargs.get("owner", None)
147 1
        self.sb_priority = kwargs.get("sb_priority", None) or kwargs.get(
148
            "priority", None
149
        )
150 1
        self.service_level = kwargs.get("service_level", 0)
151 1
        self.circuit_scheduler = kwargs.get("circuit_scheduler", [])
152 1
        self.flow_removed_at = get_time(kwargs.get("flow_removed_at")) or None
153 1
        self.updated_at = get_time(kwargs.get("updated_at")) or now()
154 1
        self.execution_rounds = kwargs.get("execution_rounds", 0)
155 1
        self.current_links_cache = set()
156 1
        self.primary_links_cache = set()
157 1
        self.backup_links_cache = set()
158 1
        self.affected_by_link_at = get_time("0001-01-01T00:00:00")
159 1
        self.old_path = Path([])
160
161 1
        self.lock = Lock()
162
163 1
        self.archived = kwargs.get("archived", False)
164
165 1
        self.metadata = kwargs.get("metadata", {})
166
167 1
        self._mongo_controller = controllers.ELineController()
168
169 1
        if kwargs.get("active", False):
170 1
            self.activate()
171
        else:
172 1
            self.deactivate()
173
174 1
        if kwargs.get("enabled", False):
175 1
            self.enable()
176
        else:
177 1
            self.disable()
178
179
        # datetime of user request for a EVC (or datetime when object was
180
        # created)
181 1
        self.request_time = kwargs.get("request_time", now())
182
        # dict with the user original request (input)
183 1
        self._requested = kwargs
184
185
        # Special cases: No tag, any, untagged
186 1
        self.special_cases = {None, "4096/4096", 0}
187 1
        self.table_group = kwargs.get("table_group")
188
189 1
    def sync(self, keys: set = None):
190
        """Sync this EVC in the MongoDB."""
191 1
        self.updated_at = now()
192 1
        if keys:
193 1
            self._mongo_controller.update_evc(self.as_dict(keys))
194 1
            return
195 1
        self._mongo_controller.upsert_evc(self.as_dict())
196
197 1
    def _get_unis_use_tags(self, **kwargs) -> tuple[UNI, UNI]:
198
        """Obtain both UNIs (uni_a, uni_z).
199
        If a UNI is changing, verify tags"""
200 1
        uni_a = kwargs.get("uni_a", None)
201 1
        uni_a_flag = False
202 1
        if uni_a and uni_a != self.uni_a:
203 1
            uni_a_flag = True
204 1
            self._use_uni_vlan(uni_a, uni_dif=self.uni_a)
205
206 1
        uni_z = kwargs.get("uni_z", None)
207 1
        if uni_z and uni_z != self.uni_z:
208 1
            try:
209 1
                self._use_uni_vlan(uni_z, uni_dif=self.uni_z)
210 1
                self.make_uni_vlan_available(self.uni_z, uni_dif=uni_z)
211 1
            except KytosTagError as err:
212 1
                if uni_a_flag:
213 1
                    self.make_uni_vlan_available(uni_a, uni_dif=self.uni_a)
214 1
                raise err
215
        else:
216 1
            uni_z = self.uni_z
217
218 1
        if uni_a_flag:
219 1
            self.make_uni_vlan_available(self.uni_a, uni_dif=uni_a)
220
        else:
221 1
            uni_a = self.uni_a
222 1
        return uni_a, uni_z
223
224 1
    def update(self, **kwargs):
225
        """Update evc attributes.
226
227
        This method will raises an error trying to change the following
228
        attributes: [creation_time, active, current_path, failover_path,
229
        _id, archived]
230
        [name, uni_a and uni_z]
231
232
        Returns:
233
            the values for enable and a redeploy attribute, if exists and None
234
            otherwise
235
        Raises:
236
            ValueError: message with error detail.
237
238
        """
239 1
        enable, redeploy = (None, None)
240 1
        if not self._tag_lists_equal(**kwargs):
241 1
            raise ValueError(
242
                "UNI_A and UNI_Z tag lists should be the same."
243
            )
244 1
        uni_a, uni_z = self._get_unis_use_tags(**kwargs)
245 1
        check_disabled_component(uni_a, uni_z)
246 1
        self._validate_has_primary_or_dynamic(
247
            primary_path=kwargs.get("primary_path"),
248
            dynamic_backup_path=kwargs.get("dynamic_backup_path"),
249
            uni_a=uni_a,
250
            uni_z=uni_z,
251
        )
252 1
        for attribute, value in kwargs.items():
253 1
            if attribute not in self.updatable_attributes:
254 1
                raise ValueError(f"{attribute} can't be updated.")
255 1
            if attribute in ("primary_path", "backup_path"):
256 1
                try:
257 1
                    value.is_valid(
258
                        uni_a.interface.switch, uni_z.interface.switch
259
                    )
260 1
                except InvalidPath as exception:
261 1
                    raise ValueError(  # pylint: disable=raise-missing-from
262
                        f"{attribute} is not a " f"valid path: {exception}"
263
                    )
264 1
        for attribute, value in kwargs.items():
265 1
            if attribute == "enabled":
266 1
                if value:
267 1
                    self.enable()
268
                else:
269 1
                    self.disable()
270 1
                enable = value
271
            else:
272 1
                setattr(self, attribute, value)
273 1
                if attribute in self.attributes_requiring_redeploy:
274 1
                    redeploy = True
275 1
        self.sync(set(kwargs.keys()))
276 1
        return enable, redeploy
277
278 1
    def set_flow_removed_at(self):
279
        """Update flow_removed_at attribute."""
280
        self.flow_removed_at = now()
281
282 1
    def has_recent_removed_flow(self, setting=settings):
283
        """Check if any flow has been removed from the evc"""
284
        if self.flow_removed_at is None:
285
            return False
286
        res_seconds = (now() - self.flow_removed_at).seconds
287
        return res_seconds < setting.TIME_RECENT_DELETED_FLOWS
288
289 1
    def is_recent_updated(self, setting=settings):
290
        """Check if the evc has been updated recently"""
291
        res_seconds = (now() - self.updated_at).seconds
292
        return res_seconds < setting.TIME_RECENT_UPDATED
293
294 1
    def __repr__(self):
295
        """Repr method."""
296 1
        return f"EVC({self._id}, {self.name})"
297
298 1
    def _validate(self, **kwargs):
299
        """Do Basic validations.
300
301
        Verify required attributes: name, uni_a, uni_z
302
303
        Raises:
304
            ValueError: message with error detail.
305
306
        """
307 1
        for attribute in self.required_attributes:
308
309 1
            if attribute not in kwargs:
310 1
                raise ValueError(f"{attribute} is required.")
311
312 1
            if "uni" in attribute:
313 1
                uni = kwargs.get(attribute)
314 1
                if not isinstance(uni, UNI):
315
                    raise ValueError(f"{attribute} is an invalid UNI.")
316
317 1
    def _tag_lists_equal(self, **kwargs):
318
        """Verify that tag lists are the same."""
319 1
        uni_a = kwargs.get("uni_a") or self.uni_a
320 1
        uni_z = kwargs.get("uni_z") or self.uni_z
321 1
        uni_a_list = uni_z_list = False
322 1
        if (uni_a.user_tag and isinstance(uni_a.user_tag, TAGRange)):
323 1
            uni_a_list = True
324 1
        if (uni_z.user_tag and isinstance(uni_z.user_tag, TAGRange)):
325 1
            uni_z_list = True
326 1
        if uni_a_list and uni_z_list:
327 1
            return uni_a.user_tag.value == uni_z.user_tag.value
328 1
        return uni_a_list == uni_z_list
329
330 1
    def _validate_has_primary_or_dynamic(
331
        self,
332
        primary_path=None,
333
        dynamic_backup_path=None,
334
        uni_a=None,
335
        uni_z=None,
336
    ) -> None:
337
        """Validate that it must have a primary path or allow dynamic paths."""
338 1
        primary_path = (
339
            primary_path
340
            if primary_path is not None
341
            else self.primary_path
342
        )
343 1
        dynamic_backup_path = (
344
            dynamic_backup_path
345
            if dynamic_backup_path is not None
346
            else self.dynamic_backup_path
347
        )
348 1
        uni_a = uni_a if uni_a is not None else self.uni_a
349 1
        uni_z = uni_z if uni_z is not None else self.uni_z
350 1
        if (
351
            not primary_path
352
            and not dynamic_backup_path
353
            and uni_a and uni_z
354
            and uni_a.interface.switch != uni_z.interface.switch
355
        ):
356 1
            msg = "The EVC must have a primary path or allow dynamic paths."
357 1
            raise ValueError(msg)
358
359 1
    def __eq__(self, other):
360
        """Override the default implementation."""
361 1
        if not isinstance(other, EVC):
362
            return False
363
364 1
        attrs_to_compare = ["name", "uni_a", "uni_z", "owner", "bandwidth"]
365 1
        for attribute in attrs_to_compare:
366 1
            if getattr(other, attribute) != getattr(self, attribute):
367 1
                return False
368 1
        return True
369
370 1
    def is_intra_switch(self):
371
        """Check if the UNIs are in the same switch."""
372 1
        return self.uni_a.interface.switch == self.uni_z.interface.switch
373
374 1
    def check_no_tag_duplicate(self, other_uni: UNI):
375
        """Check if a no tag UNI is duplicated."""
376 1
        if other_uni in (self.uni_a, self.uni_z):
377 1
            msg = f"UNI with interface {other_uni.interface.id} is"\
378
                  f" duplicated with {self}."
379 1
            raise DuplicatedNoTagUNI(msg)
380
381 1
    def as_dict(self, keys: set = None):
382
        """Return a dictionary representing an EVC object.
383
            keys: Only fields on this variable will be
384
                  returned in the dictionary"""
385 1
        evc_dict = {
386
            "id": self.id,
387
            "name": self.name,
388
            "uni_a": self.uni_a.as_dict(),
389
            "uni_z": self.uni_z.as_dict(),
390
        }
391
392 1
        time_fmt = "%Y-%m-%dT%H:%M:%S"
393
394 1
        evc_dict["start_date"] = self.start_date
395 1
        if isinstance(self.start_date, datetime):
396 1
            evc_dict["start_date"] = self.start_date.strftime(time_fmt)
397
398 1
        evc_dict["end_date"] = self.end_date
399 1
        if isinstance(self.end_date, datetime):
400 1
            evc_dict["end_date"] = self.end_date.strftime(time_fmt)
401
402 1
        evc_dict["queue_id"] = self.queue_id
403 1
        evc_dict["bandwidth"] = self.bandwidth
404 1
        evc_dict["primary_links"] = self.primary_links.as_dict()
405 1
        evc_dict["backup_links"] = self.backup_links.as_dict()
406 1
        evc_dict["current_path"] = self.current_path.as_dict()
407 1
        evc_dict["failover_path"] = self.failover_path.as_dict()
408 1
        evc_dict["primary_path"] = self.primary_path.as_dict()
409 1
        evc_dict["backup_path"] = self.backup_path.as_dict()
410 1
        evc_dict["dynamic_backup_path"] = self.dynamic_backup_path
411 1
        evc_dict["metadata"] = self.metadata
412
413 1
        evc_dict["request_time"] = self.request_time
414 1
        if isinstance(self.request_time, datetime):
415 1
            evc_dict["request_time"] = self.request_time.strftime(time_fmt)
416
417 1
        time = self.creation_time.strftime(time_fmt)
418 1
        evc_dict["creation_time"] = time
419
420 1
        evc_dict["owner"] = self.owner
421 1
        evc_dict["circuit_scheduler"] = [
422
            sc.as_dict() for sc in self.circuit_scheduler
423
        ]
424
425 1
        evc_dict["active"] = self.is_active()
426 1
        evc_dict["enabled"] = self.is_enabled()
427 1
        evc_dict["archived"] = self.archived
428 1
        evc_dict["sb_priority"] = self.sb_priority
429 1
        evc_dict["service_level"] = self.service_level
430 1
        evc_dict["primary_constraints"] = self.primary_constraints
431 1
        evc_dict["secondary_constraints"] = self.secondary_constraints
432 1
        evc_dict["flow_removed_at"] = self.flow_removed_at
433 1
        evc_dict["updated_at"] = self.updated_at
434
435 1
        if keys:
436 1
            selected = {}
437 1
            for key in keys:
438 1
                selected[key] = evc_dict[key]
439 1
            selected["id"] = evc_dict["id"]
440 1
            return selected
441 1
        return evc_dict
442
443 1
    @property
444 1
    def id(self):  # pylint: disable=invalid-name
445
        """Return this EVC's ID."""
446 1
        return self._id
447
448 1
    def archive(self):
449
        """Archive this EVC on deletion."""
450 1
        self.archived = True
451
452 1
    def _use_uni_vlan(
453
        self,
454
        uni: UNI,
455
        uni_dif: Union[None, UNI] = None
456
    ):
457
        """Use tags from UNI"""
458 1
        if uni.user_tag is None:
459 1
            return
460 1
        tag = uni.user_tag.value
461 1
        tag_type = uni.user_tag.tag_type
462 1
        if (uni_dif and isinstance(tag, list) and
463
                isinstance(uni_dif.user_tag.value, list)):
464 1
            tag = range_difference(tag, uni_dif.user_tag.value)
465 1
            if not tag:
466 1
                return
467 1
        uni.interface.use_tags(
468
            self._controller, tag, tag_type, use_lock=True, check_order=False
469
        )
470
471 1
    def make_uni_vlan_available(
472
        self,
473
        uni: UNI,
474
        uni_dif: Union[None, UNI] = None,
475
    ):
476
        """Make available tag from UNI"""
477 1
        if uni.user_tag is None:
478 1
            return
479 1
        tag = uni.user_tag.value
480 1
        tag_type = uni.user_tag.tag_type
481 1
        if (uni_dif and isinstance(tag, list) and
482
                isinstance(uni_dif.user_tag.value, list)):
483 1
            tag = range_difference(tag, uni_dif.user_tag.value)
484 1
            if not tag:
485
                return
486 1
        try:
487 1
            conflict = uni.interface.make_tags_available(
488
                self._controller, tag, tag_type, use_lock=True,
489
                check_order=False
490
            )
491 1
        except KytosTagError as err:
492 1
            log.error(f"Error in {self}: {err}")
493 1
            return
494 1
        if conflict:
495 1
            intf = uni.interface.id
496 1
            log.warning(f"Tags {conflict} was already available in {intf}")
497
498 1
    def remove_uni_tags(self):
499
        """Remove both UNI usage of a tag"""
500 1
        self.make_uni_vlan_available(self.uni_a)
501 1
        self.make_uni_vlan_available(self.uni_z)
502
503
504
# pylint: disable=fixme, too-many-public-methods
505 1
class EVCDeploy(EVCBase):
506
    """Class to handle the deploy procedures."""
507
508 1
    def create(self):
509
        """Create a EVC."""
510
511 1
    def discover_new_paths(self):
512
        """Discover new paths to satisfy this circuit and deploy it."""
513
        return DynamicPathManager.get_best_paths(self,
514
                                                 **self.primary_constraints)
515
516 1
    def get_failover_path_candidates(self):
517
        """Get failover paths to satisfy this EVC."""
518
        # in the future we can return primary/backup paths as well
519
        # we just have to properly handle link_up and failover paths
520
        # if (
521
        #     self.is_using_primary_path() and
522
        #     self.backup_path.status is EntityStatus.UP
523
        # ):
524
        #     yield self.backup_path
525 1
        return DynamicPathManager.get_disjoint_paths(self, self.current_path)
526
527 1
    def change_path(self):
528
        """Change EVC path."""
529
530 1
    def reprovision(self):
531
        """Force the EVC (re-)provisioning."""
532
533 1
    def is_affected_by_link(self, link):
534
        """Return True if this EVC has the given link on its current path."""
535 1
        return link in self.current_path
536
537 1
    def link_affected_by_interface(self, interface):
538
        """Return True if this EVC has the given link on its current path."""
539
        return self.current_path.link_affected_by_interface(interface)
540
541 1
    def is_backup_path_affected_by_link(self, link):
542
        """Return True if the backup path of this EVC uses the given link."""
543 1
        return link in self.backup_path
544
545
    # pylint: disable=invalid-name
546 1
    def is_primary_path_affected_by_link(self, link):
547
        """Return True if the primary path of this EVC uses the given link."""
548 1
        return link in self.primary_path
549
550 1
    def is_failover_path_affected_by_link(self, link):
551
        """Return True if this EVC has the given link on its failover path."""
552 1
        return link in self.failover_path
553
554 1
    def is_eligible_for_failover_path(self):
555
        """Verify if this EVC is eligible for failover path (EP029)"""
556
        # In the future this function can be augmented to consider
557
        # primary/backup, primary/dynamic, and other path combinations
558 1
        return (
559
            self.dynamic_backup_path and
560
            not self.primary_path and not self.backup_path
561
        )
562
563 1
    def is_using_primary_path(self):
564
        """Verify if the current deployed path is self.primary_path."""
565 1
        return self.primary_path and (self.current_path == self.primary_path)
566
567 1
    def is_using_backup_path(self):
568
        """Verify if the current deployed path is self.backup_path."""
569 1
        return self.backup_path and (self.current_path == self.backup_path)
570
571 1
    def is_using_dynamic_path(self):
572
        """Verify if the current deployed path is a dynamic path."""
573 1
        if (
574
            self.current_path
575
            and not self.is_using_primary_path()
576
            and not self.is_using_backup_path()
577
            and self.current_path.status == EntityStatus.UP
578
        ):
579
            return True
580 1
        return False
581
582 1
    def deploy_to_backup_path(self):
583
        """Deploy the backup path into the datapaths of this circuit.
584
585
        If the backup_path attribute is valid and up, this method will try to
586
        deploy this backup_path.
587
588
        If everything fails and dynamic_backup_path is True, then tries to
589
        deploy a dynamic path.
590
        """
591
        # TODO: Remove flows from current (cookies)
592 1
        if self.is_using_backup_path():
593
            # TODO: Log to say that cannot move backup to backup
594
            return True
595
596 1
        success = False
597 1
        if self.backup_path.status is EntityStatus.UP:
598 1
            success = self.deploy_to_path(self.backup_path)
599
600 1
        if success:
601 1
            return True
602
603 1
        if self.dynamic_backup_path or self.is_intra_switch():
604 1
            return self.deploy_to_path()
605
606
        return False
607
608 1
    def deploy_to_primary_path(self):
609
        """Deploy the primary path into the datapaths of this circuit.
610
611
        If the primary_path attribute is valid and up, this method will try to
612
        deploy this primary_path.
613
        """
614
        # TODO: Remove flows from current (cookies)
615 1
        if self.is_using_primary_path():
616
            # TODO: Log to say that cannot move primary to primary
617
            return True
618
619 1
        if self.primary_path.status is EntityStatus.UP:
620 1
            return self.deploy_to_path(self.primary_path)
621
        return False
622
623 1
    def deploy(self):
624
        """Deploy EVC to best path.
625
626
        Best path can be the primary path, if available. If not, the backup
627
        path, and, if it is also not available, a dynamic path.
628
        """
629 1
        if self.archived:
630 1
            return False
631 1
        self.enable()
632 1
        success = self.deploy_to_primary_path()
633 1
        if not success:
634 1
            success = self.deploy_to_backup_path()
635
636 1
        if success:
637 1
            emit_event(self._controller, "deployed",
638
                       content=map_evc_event_content(self))
639 1
        return success
640
641 1
    @staticmethod
642 1
    def get_path_status(path):
643
        """Check for the current status of a path.
644
645
        If any link in this path is down, the path is considered down.
646
        """
647 1
        if not path:
648 1
            return EntityStatus.DISABLED
649
650 1
        for link in path:
651 1
            if link.status is not EntityStatus.UP:
652 1
                return link.status
653 1
        return EntityStatus.UP
654
655
    #    def discover_new_path(self):
656
    #        # TODO: discover a new path to satisfy this circuit and deploy
657
658 1
    def remove(self):
659
        """Remove EVC path and disable it."""
660 1
        self.remove_current_flows(sync=False)
661 1
        self.remove_failover_flows(sync=False)
662 1
        self.disable()
663 1
        self.sync()
664 1
        emit_event(self._controller, "undeployed",
665
                   content=map_evc_event_content(self))
666
667 1
    def remove_failover_flows(self, exclude_uni_switches=True,
668
                              force=True, sync=True) -> None:
669
        """Remove failover_flows.
670
671
        By default, it'll exclude UNI switches, if mef_eline has already
672
        called remove_current_flows before then this minimizes the number
673
        of FlowMods and IO.
674
        """
675 1
        if not self.failover_path:
676 1
            return
677 1
        switches, cookie, excluded = OrderedDict(), self.get_cookie(), set()
678 1
        links = set()
679 1
        if exclude_uni_switches:
680 1
            excluded.add(self.uni_a.interface.switch.id)
681 1
            excluded.add(self.uni_z.interface.switch.id)
682 1
        for link in self.failover_path:
683 1
            if link.endpoint_a.switch.id not in excluded:
684 1
                switches[link.endpoint_a.switch.id] = link.endpoint_a.switch
685 1
                links.add(link)
686 1
            if link.endpoint_b.switch.id not in excluded:
687 1
                switches[link.endpoint_b.switch.id] = link.endpoint_b.switch
688 1
                links.add(link)
689 1
        for switch in switches.values():
690 1
            try:
691 1
                self._send_flow_mods(
692
                    switch.id,
693
                    [
694
                        {
695
                            "cookie": cookie,
696
                            "cookie_mask": int(0xffffffffffffffff),
697
                        }
698
                    ],
699
                    "delete",
700
                    force=force,
701
                )
702
            except FlowModException as err:
703
                log.error(f"Error deleting path in {switch.id}, {err}")
704
                break
705 1
        try:
706 1
            self.failover_path.make_vlans_available(self._controller)
707
        except KytosTagError as err:
708
            log.error(f"Error when removing failover flows: {err}")
709 1
        self.failover_path = Path([])
710 1
        if sync:
711 1
            self.sync()
712
713 1
    def remove_current_flows(self, force=True, sync=True):
714
        """Remove all flows from current path or path intended for
715
         current path if exists."""
716 1
        switches = set()
717
718 1
        if not self.current_path and not self.is_intra_switch():
719 1
            return
720 1
        current_path = self.current_path
721 1
        for link in current_path:
722 1
            switches.add(link.endpoint_a.switch)
723 1
            switches.add(link.endpoint_b.switch)
724 1
        switches.add(self.uni_a.interface.switch)
725 1
        switches.add(self.uni_z.interface.switch)
726 1
        match = {
727
            "cookie": self.get_cookie(),
728
            "cookie_mask": int(0xffffffffffffffff)
729
        }
730
731 1
        for switch in switches:
732 1
            try:
733 1
                self._send_flow_mods(switch.id, [match], 'delete', force=force)
734
            except FlowModException as err:
735
                log.error(f"Error deleting current_path in {switch.id}, {err}")
736
                break
737 1
        try:
738 1
            current_path.make_vlans_available(self._controller)
739
        except KytosTagError as err:
740
            log.error(f"Error when removing current path flows: {err}")
741 1
        self.current_path = Path([])
742 1
        self.deactivate()
743 1
        if sync:
744 1
            self.sync()
745
746 1
    def remove_path_flows(
747
        self, path=None, force=True
748
    ) -> dict[str, list[dict]]:
749
        """Remove all flows from path, and return the removed flows."""
750 1
        dpid_flows_match: dict[str, list[dict]] = {}
751
752 1
        if not path:
753 1
            return dpid_flows_match
754
755 1
        try:
756 1
            nni_flows = self._prepare_nni_flows(path)
757
        # pylint: disable=broad-except
758
        except Exception:
759
            err = traceback.format_exc().replace("\n", ", ")
760
            log.error(f"Fail to remove NNI failover flows for {self}: {err}")
761
            nni_flows = {}
762
763 1
        for dpid, flows in nni_flows.items():
764 1
            dpid_flows_match.setdefault(dpid, [])
765 1
            for flow in flows:
766 1
                dpid_flows_match[dpid].append({
767
                    "cookie": flow["cookie"],
768
                    "match": flow["match"],
769
                    "cookie_mask": int(0xffffffffffffffff)
770
                })
771
772 1
        try:
773 1
            uni_flows = self._prepare_uni_flows(path, skip_in=True)
774
        # pylint: disable=broad-except
775
        except Exception:
776
            err = traceback.format_exc().replace("\n", ", ")
777
            log.error(f"Fail to remove UNI failover flows for {self}: {err}")
778
            uni_flows = {}
779
780 1
        for dpid, flows in uni_flows.items():
781 1
            dpid_flows_match.setdefault(dpid, [])
782 1
            for flow in flows:
783 1
                dpid_flows_match[dpid].append({
784
                    "cookie": flow["cookie"],
785
                    "match": flow["match"],
786
                    "cookie_mask": int(0xffffffffffffffff)
787
                })
788
789 1
        for dpid, flows in dpid_flows_match.items():
790 1
            try:
791 1
                self._send_flow_mods(dpid, flows, 'delete', force=force)
792 1
            except FlowModException as err:
793 1
                log.error(f"Error deleting path in {dpid}, {err}")
794 1
                break
795 1
        try:
796 1
            path.make_vlans_available(self._controller)
797
        except KytosTagError as err:
798
            log.error(f"Error when removing path flows: {err}")
799
800 1
        return dpid_flows_match
801
802 1
    @staticmethod
803 1
    def links_zipped(path=None):
804
        """Return an iterator which yields pairs of links in order."""
805 1
        if not path:
806 1
            return []
807 1
        return zip(path[:-1], path[1:])
808
809 1
    def should_deploy(self, path=None):
810
        """Verify if the circuit should be deployed."""
811 1
        if not path:
812 1
            log.debug("Path is empty.")
813 1
            return False
814
815 1
        if not self.is_enabled():
816 1
            log.debug(f"{self} is disabled.")
817 1
            return False
818
819 1
        if not self.is_active():
820 1
            log.debug(f"{self} will be deployed.")
821 1
            return True
822
823 1
        return False
824
825
    # pylint: disable=too-many-branches, too-many-statements
826 1
    def deploy_to_path(self, path=None):
827
        """Install the flows for this circuit.
828
829
        Procedures to deploy:
830
831
        0. Remove current flows installed
832
        1. Decide if will deploy "path" or discover a new path
833
        2. Choose vlan
834
        3. Install NNI flows
835
        4. Install UNI flows
836
        5. Activate
837
        6. Update current_path
838
        7. Update links caches(primary, current, backup)
839
840
        """
841 1
        self.remove_current_flows(sync=False)
842 1
        use_path = path or Path([])
843 1
        tag_errors = []
844 1
        if self.should_deploy(use_path):
845 1
            try:
846 1
                use_path.choose_vlans(self._controller)
847 1
            except KytosNoTagAvailableError as e:
848 1
                tag_errors.append(str(e))
849 1
                use_path = None
850
        else:
851 1
            for use_path in self.discover_new_paths():
852 1
                if use_path is None:
853
                    continue
854 1
                try:
855 1
                    use_path.choose_vlans(self._controller)
856 1
                    break
857 1
                except KytosNoTagAvailableError as e:
858 1
                    tag_errors.append(str(e))
859
            else:
860 1
                use_path = None
861
862 1
        try:
863 1
            if use_path:
864 1
                self._install_nni_flows(use_path)
865 1
                self._install_uni_flows(use_path)
866 1
            elif self.is_intra_switch():
867 1
                use_path = Path()
868 1
                self._install_direct_uni_flows()
869
            else:
870 1
                msg = f"{self} was not deployed. No available path was found."
871 1
                if tag_errors:
872 1
                    msg = self.add_tag_errors(msg, tag_errors)
873 1
                    log.error(msg)
874
                else:
875 1
                    log.warning(msg)
876 1
                return False
877 1
        except EVCPathNotInstalled as err:
878 1
            log.error(
879
                f"Error deploying EVC {self} when calling flow_manager: {err}"
880
            )
881 1
            self.remove_current_flows(use_path, sync=True)
882 1
            return False
883 1
        self.activate()
884 1
        self.current_path = use_path
885 1
        self.sync()
886 1
        log.info(f"{self} was deployed.")
887 1
        return True
888
889 1
    def try_setup_failover_path(self, wait=settings.DEPLOY_EVCS_INTERVAL):
890
        """Try setup failover_path whenever possible."""
891 1
        if (
892
                self.failover_path or not self.current_path
893
                or not self.is_active()
894
                ):
895 1
            return
896 1
        if (now() - self.affected_by_link_at).seconds >= wait:
897 1
            with self.lock:
898 1
                self.setup_failover_path()
899
900
    # pylint: disable=too-many-statements
901 1
    def setup_failover_path(self):
902
        """Install flows for the failover path of this EVC.
903
904
        Procedures to deploy:
905
906
        0. Remove flows currently installed for failover_path (if any)
907
        1. Discover a disjoint path from current_path
908
        2. Choose vlans
909
        3. Install NNI flows
910
        4. Install UNI egress flows
911
        5. Update failover_path
912
        """
913
        # Intra-switch EVCs have no failover_path
914 1
        if self.is_intra_switch():
915 1
            return False
916
917
        # For not only setup failover path for totally dynamic EVCs
918 1
        if not self.is_eligible_for_failover_path():
919 1
            return False
920
921 1
        out_new_flows: dict[str, list[dict]] = {}
922 1
        reason = ""
923 1
        tag_errors = []
924 1
        out_removed_flows = self.remove_path_flows(self.failover_path)
925 1
        self.failover_path = Path([])
926
927 1
        for use_path in self.get_failover_path_candidates():
928 1
            if not use_path:
929 1
                continue
930 1
            try:
931 1
                use_path.choose_vlans(self._controller)
932 1
                break
933 1
            except KytosNoTagAvailableError as e:
934 1
                tag_errors.append(str(e))
935
        else:
936 1
            use_path = Path([])
937 1
            reason = "No available path was found"
938
939 1
        try:
940 1
            if use_path:
941 1
                _nni_flows = self._install_nni_flows(use_path)
942 1
                out_new_flows = merge_flow_dicts(out_new_flows, _nni_flows)
943 1
                _uni_flows = self._install_uni_flows(use_path, skip_in=True)
944 1
                out_new_flows = merge_flow_dicts(out_new_flows, _uni_flows)
945 1
        except EVCPathNotInstalled as err:
946 1
            reason = "Error deploying failover path"
947 1
            log.error(
948
                f"{reason} for {self}. FlowManager error: {err}"
949
            )
950 1
            _rmed_flows = self.remove_path_flows(use_path)
951 1
            out_removed_flows = merge_flow_dicts(
952
                out_removed_flows, _rmed_flows
953
            )
954 1
            use_path = Path([])
955
956 1
        self.failover_path = use_path
957 1
        self.sync()
958
959 1
        if out_new_flows or out_removed_flows:
960 1
            emit_event(self._controller, "failover_deployed", content={
961
                self.id: map_evc_event_content(
962
                    self,
963
                    flows=out_new_flows,
964
                    removed_flows=out_removed_flows,
965
                    error_reason=reason,
966
                    current_path=self.current_path.as_dict(),
967
                )
968
            })
969
970 1
        if not use_path:
971 1
            msg = f"Failover path for {self} was not deployed: {reason}."
972 1
            if tag_errors:
973 1
                msg = self.add_tag_errors(msg, tag_errors)
974 1
                log.error(msg)
975
            else:
976 1
                log.warning(msg)
977 1
            return False
978 1
        log.info(f"Failover path for {self} was deployed.")
979 1
        return True
980
981 1
    @staticmethod
982 1
    def add_tag_errors(msg: str, tag_errors: list):
983
        """Add to msg the tag errors ecountered when chossing path."""
984 1
        path = ['path', 'paths']
985 1
        was = ['was', 'were']
986 1
        message = ['message', 'messages']
987
988
        # Choose either singular(0) or plural(1) words
989 1
        n = 1
990 1
        if len(tag_errors) == 1:
991 1
            n = 0
992
993 1
        msg += f" {len(tag_errors)} {path[n]} {was[n]} rejected"
994 1
        msg += f" with {message[n]}: {tag_errors}"
995 1
        return msg
996
997 1
    def get_failover_flows(self):
998
        """Return the flows needed to make the failover path active, i.e. the
999
        flows for ingress forwarding.
1000
1001
        Return:
1002
            dict: A dict of flows indexed by the switch_id will be returned, or
1003
                an empty dict if no failover_path is available.
1004
        """
1005 1
        if not self.failover_path:
1006 1
            return {}
1007 1
        return self._prepare_uni_flows(self.failover_path, skip_out=True)
1008
1009
    # pylint: disable=too-many-branches
1010 1
    def _prepare_direct_uni_flows(self):
1011
        """Prepare flows connecting two UNIs for intra-switch EVC."""
1012 1
        vlan_a = self._get_value_from_uni_tag(self.uni_a)
1013 1
        vlan_z = self._get_value_from_uni_tag(self.uni_z)
1014
1015 1
        flow_mod_az = self._prepare_flow_mod(
1016
            self.uni_a.interface, self.uni_z.interface,
1017
            self.queue_id, vlan_a
1018
        )
1019 1
        flow_mod_za = self._prepare_flow_mod(
1020
            self.uni_z.interface, self.uni_a.interface,
1021
            self.queue_id, vlan_z
1022
        )
1023
1024 1 View Code Duplication
        if not isinstance(vlan_z, list) and vlan_z not in self.special_cases:
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1025 1
            flow_mod_az["actions"].insert(
1026
                0, {"action_type": "set_vlan", "vlan_id": vlan_z}
1027
            )
1028 1
            if not vlan_a:
1029 1
                flow_mod_az["actions"].insert(
1030
                    0, {"action_type": "push_vlan", "tag_type": "c"}
1031
                )
1032 1
            if vlan_a == 0:
1033 1
                flow_mod_za["actions"].insert(0, {"action_type": "pop_vlan"})
1034 1
        elif vlan_a == 0 and vlan_z == "4096/4096":
1035 1
            flow_mod_za["actions"].insert(0, {"action_type": "pop_vlan"})
1036
1037 1 View Code Duplication
        if not isinstance(vlan_a, list) and vlan_a not in self.special_cases:
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
1038 1
            flow_mod_za["actions"].insert(
1039
                    0, {"action_type": "set_vlan", "vlan_id": vlan_a}
1040
                )
1041 1
            if not vlan_z:
1042 1
                flow_mod_za["actions"].insert(
1043
                    0, {"action_type": "push_vlan", "tag_type": "c"}
1044
                )
1045 1
            if vlan_z == 0:
1046 1
                flow_mod_az["actions"].insert(0, {"action_type": "pop_vlan"})
1047 1
        elif vlan_a == "4096/4096" and vlan_z == 0:
1048 1
            flow_mod_az["actions"].insert(0, {"action_type": "pop_vlan"})
1049
1050 1
        flows = []
1051 1
        if isinstance(vlan_a, list):
1052 1
            for mask_a in vlan_a:
1053 1
                flow_aux = deepcopy(flow_mod_az)
1054 1
                flow_aux["match"]["dl_vlan"] = mask_a
1055 1
                flows.append(flow_aux)
1056
        else:
1057 1
            if vlan_a is not None:
1058 1
                flow_mod_az["match"]["dl_vlan"] = vlan_a
1059 1
            flows.append(flow_mod_az)
1060
1061 1
        if isinstance(vlan_z, list):
1062 1
            for mask_z in vlan_z:
1063 1
                flow_aux = deepcopy(flow_mod_za)
1064 1
                flow_aux["match"]["dl_vlan"] = mask_z
1065 1
                flows.append(flow_aux)
1066
        else:
1067 1
            if vlan_z is not None:
1068 1
                flow_mod_za["match"]["dl_vlan"] = vlan_z
1069 1
            flows.append(flow_mod_za)
1070 1
        return (
1071
            self.uni_a.interface.switch.id, flows
1072
        )
1073
1074 1
    def _install_direct_uni_flows(self):
1075
        """Install flows connecting two UNIs.
1076
1077
        This case happens when the circuit is between UNIs in the
1078
        same switch.
1079
        """
1080 1
        (dpid, flows) = self._prepare_direct_uni_flows()
1081 1
        try:
1082 1
            self._send_flow_mods(dpid, flows)
1083 1
        except FlowModException as err:
1084 1
            raise EVCPathNotInstalled(f"In {dpid}, {err}") from err
1085
1086 1
    def _prepare_nni_flows(self, path=None):
1087
        """Prepare NNI flows."""
1088 1
        nni_flows = OrderedDict()
1089 1
        previous = self.uni_a.interface.switch.dpid
1090 1
        for incoming, outcoming in self.links_zipped(path):
1091 1
            in_vlan = incoming.get_metadata("s_vlan").value
1092 1
            out_vlan = outcoming.get_metadata("s_vlan").value
1093 1
            in_endpoint = self.get_endpoint_by_id(incoming, previous, ne)
1094 1
            out_endpoint = self.get_endpoint_by_id(
1095
                outcoming, in_endpoint.switch.id, eq
1096
            )
1097
1098 1
            flows = []
1099
            # Flow for one direction
1100 1
            flows.append(
1101
                self._prepare_nni_flow(
1102
                    in_endpoint,
1103
                    out_endpoint,
1104
                    in_vlan,
1105
                    out_vlan,
1106
                    queue_id=self.queue_id,
1107
                )
1108
            )
1109
1110
            # Flow for the other direction
1111 1
            flows.append(
1112
                self._prepare_nni_flow(
1113
                    out_endpoint,
1114
                    in_endpoint,
1115
                    out_vlan,
1116
                    in_vlan,
1117
                    queue_id=self.queue_id,
1118
                )
1119
            )
1120 1
            previous = in_endpoint.switch.id
1121 1
            nni_flows[in_endpoint.switch.id] = flows
1122 1
        return nni_flows
1123
1124 1
    def _install_nni_flows(
1125
        self, path=None
1126
    ) -> dict[str, list[dict]]:
1127
        """Install NNI flows."""
1128 1
        nni_flows = self._prepare_nni_flows(path)
1129 1
        try:
1130 1
            for dpid, flows in nni_flows.items():
1131 1
                self._send_flow_mods(dpid, flows)
1132 1
        except FlowModException as err:
1133 1
            raise EVCPathNotInstalled(f"In {dpid}, {err}") from err
1134 1
        return nni_flows
1135
1136 1
    @staticmethod
1137 1
    def _get_value_from_uni_tag(uni: UNI):
1138
        """Returns the value from tag. In case of any and untagged
1139
        it should return 4096/4096 and 0 respectively"""
1140 1
        special = {"any": "4096/4096", "untagged": 0}
1141 1
        if uni.user_tag:
1142 1
            value = uni.user_tag.value
1143 1
            if isinstance(value, list):
1144 1
                return uni.user_tag.mask_list
1145 1
            return special.get(value, value)
1146 1
        return None
1147
1148
    # pylint: disable=too-many-locals
1149 1
    def _prepare_uni_flows(self, path=None, skip_in=False, skip_out=False):
1150
        """Prepare flows to install UNIs."""
1151 1
        uni_flows = {}
1152 1
        if not path:
1153 1
            log.info("install uni flows without path.")
1154 1
            return uni_flows
1155
1156
        # Determine VLANs
1157 1
        in_vlan_a = self._get_value_from_uni_tag(self.uni_a)
1158 1
        out_vlan_a = path[0].get_metadata("s_vlan").value
1159
1160 1
        in_vlan_z = self._get_value_from_uni_tag(self.uni_z)
1161 1
        out_vlan_z = path[-1].get_metadata("s_vlan").value
1162
1163
        # Get endpoints from path
1164 1
        endpoint_a = self.get_endpoint_by_id(
1165
            path[0], self.uni_a.interface.switch.id, eq
1166
        )
1167 1
        endpoint_z = self.get_endpoint_by_id(
1168
            path[-1], self.uni_z.interface.switch.id, eq
1169
        )
1170
1171
        # Flows for the first UNI
1172 1
        flows_a = []
1173
1174
        # Flow for one direction, pushing the service tag
1175 1
        if not skip_in:
1176 1
            if isinstance(in_vlan_a, list):
1177 1
                for in_mask_a in in_vlan_a:
1178 1
                    push_flow = self._prepare_push_flow(
1179
                        self.uni_a.interface,
1180
                        endpoint_a,
1181
                        in_mask_a,
1182
                        out_vlan_a,
1183
                        in_vlan_z,
1184
                        queue_id=self.queue_id,
1185
                    )
1186 1
                    flows_a.append(push_flow)
1187
            else:
1188 1
                push_flow = self._prepare_push_flow(
1189
                    self.uni_a.interface,
1190
                    endpoint_a,
1191
                    in_vlan_a,
1192
                    out_vlan_a,
1193
                    in_vlan_z,
1194
                    queue_id=self.queue_id,
1195
                )
1196 1
                flows_a.append(push_flow)
1197
1198
        # Flow for the other direction, popping the service tag
1199 1
        if not skip_out:
1200 1
            pop_flow = self._prepare_pop_flow(
1201
                endpoint_a,
1202
                self.uni_a.interface,
1203
                out_vlan_a,
1204
                queue_id=self.queue_id,
1205
            )
1206 1
            flows_a.append(pop_flow)
1207
1208 1
        uni_flows[self.uni_a.interface.switch.id] = flows_a
1209
1210
        # Flows for the second UNI
1211 1
        flows_z = []
1212
1213
        # Flow for one direction, pushing the service tag
1214 1
        if not skip_in:
1215 1
            if isinstance(in_vlan_z, list):
1216 1
                for in_mask_z in in_vlan_z:
1217 1
                    push_flow = self._prepare_push_flow(
1218
                        self.uni_z.interface,
1219
                        endpoint_z,
1220
                        in_mask_z,
1221
                        out_vlan_z,
1222
                        in_vlan_a,
1223
                        queue_id=self.queue_id,
1224
                    )
1225 1
                    flows_z.append(push_flow)
1226
            else:
1227 1
                push_flow = self._prepare_push_flow(
1228
                    self.uni_z.interface,
1229
                    endpoint_z,
1230
                    in_vlan_z,
1231
                    out_vlan_z,
1232
                    in_vlan_a,
1233
                    queue_id=self.queue_id,
1234
                )
1235 1
                flows_z.append(push_flow)
1236
1237
        # Flow for the other direction, popping the service tag
1238 1
        if not skip_out:
1239 1
            pop_flow = self._prepare_pop_flow(
1240
                endpoint_z,
1241
                self.uni_z.interface,
1242
                out_vlan_z,
1243
                queue_id=self.queue_id,
1244
            )
1245 1
            flows_z.append(pop_flow)
1246
1247 1
        uni_flows[self.uni_z.interface.switch.id] = flows_z
1248
1249 1
        return uni_flows
1250
1251 1
    def _install_uni_flows(
1252
        self, path=None, skip_in=False, skip_out=False
1253
    ) -> dict[str, list[dict]]:
1254
        """Install UNI flows."""
1255 1
        uni_flows = self._prepare_uni_flows(path, skip_in, skip_out)
1256 1
        try:
1257 1
            for (dpid, flows) in uni_flows.items():
1258 1
                self._send_flow_mods(dpid, flows)
1259
        except FlowModException as err:
1260
            raise EVCPathNotInstalled(f"In {dpid}, {err}") from err
1261
1262 1
        return uni_flows
1263
1264 1
    @staticmethod
1265 1
    @retry(
1266
        stop=stop_after_attempt(3),
1267
        wait=wait_combine(wait_fixed(3), wait_random(min=2, max=7)),
1268
        retry=retry_if_exception_type(FlowModException),
1269
        before_sleep=before_sleep,
1270
        reraise=True,
1271
    )
1272 1
    def _send_flow_mods(dpid, flow_mods, command='flows', force=False):
1273
        """Send a flow_mod list to a specific switch.
1274
1275
        Args:
1276
            dpid(str): The target of flows (i.e. Switch.id).
1277
            flow_mods(dict): Python dictionary with flow_mods.
1278
            command(str): By default is 'flows'. To remove a flow is 'remove'.
1279
            force(bool): True to send via consistency check in case of errors
1280
1281
        """
1282 1
        endpoint = f"{settings.MANAGER_URL}/{command}/{dpid}"
1283
1284 1
        data = {"flows": flow_mods, "force": force}
1285 1
        try:
1286 1
            res = httpx.post(endpoint, json=data, timeout=30)
1287
        except httpx.RequestError as err:
1288
            raise FlowModException(str(err)) from err
1289 1
        if res.is_server_error or res.status_code >= 400:
1290 1
            raise FlowModException(res.text)
1291
1292 1
    def get_cookie(self):
1293
        """Return the cookie integer from evc id."""
1294 1
        return int(self.id, 16) + (settings.COOKIE_PREFIX << 56)
1295
1296 1
    @staticmethod
1297 1
    def get_id_from_cookie(cookie):
1298
        """Return the evc id given a cookie value."""
1299 1
        evc_id = cookie - (settings.COOKIE_PREFIX << 56)
1300 1
        return f"{evc_id:x}".zfill(14)
1301
1302 1
    def set_flow_table_group_id(self, flow_mod: dict, vlan) -> dict:
1303
        """Set table_group and table_id"""
1304 1
        table_group = "epl" if vlan is None else "evpl"
1305 1
        flow_mod["table_group"] = table_group
1306 1
        flow_mod["table_id"] = self.table_group[table_group]
1307 1
        return flow_mod
1308
1309 1
    @staticmethod
1310 1
    def get_priority(vlan):
1311
        """Return priority value depending on vlan value"""
1312 1
        if isinstance(vlan, list):
1313 1
            return settings.EVPL_SB_PRIORITY
1314 1
        if vlan not in {None, "4096/4096", 0}:
1315 1
            return settings.EVPL_SB_PRIORITY
1316 1
        if vlan == 0:
1317 1
            return settings.UNTAGGED_SB_PRIORITY
1318 1
        if vlan == "4096/4096":
1319 1
            return settings.ANY_SB_PRIORITY
1320 1
        return settings.EPL_SB_PRIORITY
1321
1322 1
    def _prepare_flow_mod(self, in_interface, out_interface,
1323
                          queue_id=None, vlan=True):
1324
        """Prepare a common flow mod."""
1325 1
        default_actions = [
1326
            {"action_type": "output", "port": out_interface.port_number}
1327
        ]
1328 1
        queue_id = settings.QUEUE_ID if queue_id == -1 else queue_id
1329 1
        if queue_id is not None:
1330 1
            default_actions.append(
1331
                {"action_type": "set_queue", "queue_id": queue_id}
1332
            )
1333
1334 1
        flow_mod = {
1335
            "match": {"in_port": in_interface.port_number},
1336
            "cookie": self.get_cookie(),
1337
            "actions": default_actions,
1338
            "owner": "mef_eline",
1339
        }
1340
1341 1
        self.set_flow_table_group_id(flow_mod, vlan)
1342 1
        if self.sb_priority:
1343 1
            flow_mod["priority"] = self.sb_priority
1344
        else:
1345 1
            flow_mod["priority"] = self.get_priority(vlan)
1346 1
        return flow_mod
1347
1348 1
    def _prepare_nni_flow(self, *args, queue_id=None):
1349
        """Create NNI flows."""
1350 1
        in_interface, out_interface, in_vlan, out_vlan = args
1351 1
        flow_mod = self._prepare_flow_mod(
1352
            in_interface, out_interface, queue_id
1353
        )
1354 1
        flow_mod["match"]["dl_vlan"] = in_vlan
1355 1
        new_action = {"action_type": "set_vlan", "vlan_id": out_vlan}
1356 1
        flow_mod["actions"].insert(0, new_action)
1357
1358 1
        return flow_mod
1359
1360 1
    def _prepare_push_flow(self, *args, queue_id=None):
1361
        """Prepare push flow.
1362
1363
        Arguments:
1364
            in_interface(str): Interface input.
1365
            out_interface(str): Interface output.
1366
            in_vlan(int,str,None): Vlan input.
1367
            out_vlan(str): Vlan output.
1368
            new_c_vlan(int,str,list,None): New client vlan.
1369
1370
        Return:
1371
            dict: An python dictionary representing a FlowMod
1372
1373
        """
1374
        # assign all arguments
1375 1
        in_interface, out_interface, in_vlan, out_vlan, new_c_vlan = args
1376 1
        vlan_pri = in_vlan if not isinstance(new_c_vlan, list) else new_c_vlan
1377 1
        flow_mod = self._prepare_flow_mod(
1378
            in_interface, out_interface, queue_id, vlan_pri
1379
        )
1380
        # the service tag must be always pushed
1381 1
        new_action = {"action_type": "set_vlan", "vlan_id": out_vlan}
1382 1
        flow_mod["actions"].insert(0, new_action)
1383
1384 1
        new_action = {"action_type": "push_vlan", "tag_type": "s"}
1385 1
        flow_mod["actions"].insert(0, new_action)
1386
1387 1
        if in_vlan is not None:
1388
            # if in_vlan is set, it must be included in the match
1389 1
            flow_mod["match"]["dl_vlan"] = in_vlan
1390
1391 1
        if (not isinstance(new_c_vlan, list) and in_vlan != new_c_vlan and
1392
                new_c_vlan not in self.special_cases):
1393
            # new_in_vlan is an integer but zero, action to set is required
1394 1
            new_action = {"action_type": "set_vlan", "vlan_id": new_c_vlan}
1395 1
            flow_mod["actions"].insert(0, new_action)
1396
1397 1
        if in_vlan not in self.special_cases and new_c_vlan == 0:
1398
            # # new_in_vlan is an integer but zero and new_c_vlan does not
1399
            # a pop action is required
1400 1
            new_action = {"action_type": "pop_vlan"}
1401 1
            flow_mod["actions"].insert(0, new_action)
1402
1403 1
        elif in_vlan == "4096/4096" and new_c_vlan == 0:
1404
            # if in_vlan match with any tags and new_c_vlan does not
1405
            # a pop action is required
1406 1
            new_action = {"action_type": "pop_vlan"}
1407 1
            flow_mod["actions"].insert(0, new_action)
1408
1409 1
        elif (not in_vlan and
1410
                (not isinstance(new_c_vlan, list) and
1411
                 new_c_vlan not in self.special_cases)):
1412
            # new_in_vlan is an integer but zero and in_vlan is not set
1413
            # then it is set now
1414 1
            new_action = {"action_type": "push_vlan", "tag_type": "c"}
1415 1
            flow_mod["actions"].insert(0, new_action)
1416
1417 1
        return flow_mod
1418
1419 1
    def _prepare_pop_flow(
1420
        self, in_interface, out_interface, out_vlan, queue_id=None
1421
    ):
1422
        # pylint: disable=too-many-arguments
1423
        """Prepare pop flow."""
1424 1
        flow_mod = self._prepare_flow_mod(
1425
            in_interface, out_interface, queue_id
1426
        )
1427 1
        flow_mod["match"]["dl_vlan"] = out_vlan
1428 1
        new_action = {"action_type": "pop_vlan"}
1429 1
        flow_mod["actions"].insert(0, new_action)
1430 1
        return flow_mod
1431
1432 1
    @staticmethod
1433 1
    def run_bulk_sdntraces(
1434
        uni_list: list[tuple[Interface, Union[str, int, None]]]
1435
    ) -> dict:
1436
        """Run SDN traces on control plane starting from EVC UNIs."""
1437 1
        endpoint = f"{settings.SDN_TRACE_CP_URL}/traces"
1438 1
        data = []
1439 1
        for interface, tag_value in uni_list:
1440 1
            data_uni = {
1441
                "trace": {
1442
                            "switch": {
1443
                                "dpid": interface.switch.dpid,
1444
                                "in_port": interface.port_number,
1445
                            }
1446
                        }
1447
                }
1448 1
            if tag_value:
1449 1
                uni_dl_vlan = map_dl_vlan(tag_value)
1450 1
                if uni_dl_vlan:
1451 1
                    data_uni["trace"]["eth"] = {
1452
                                            "dl_type": 0x8100,
1453
                                            "dl_vlan": uni_dl_vlan,
1454
                                            }
1455 1
            data.append(data_uni)
1456 1
        try:
1457 1
            response = httpx.put(endpoint, json=data, timeout=30)
1458 1
        except httpx.TimeoutException as exception:
1459 1
            log.error(f"Request has timed out: {exception}")
1460 1
            return {"result": []}
1461 1
        if response.status_code >= 400:
1462 1
            log.error(f"Failed to run sdntrace-cp: {response.text}")
1463 1
            return {"result": []}
1464 1
        return response.json()
1465
1466
    # pylint: disable=too-many-return-statements, too-many-arguments
1467 1
    @staticmethod
1468 1
    def check_trace(
1469
        evc_id: str,
1470
        evc_name: str,
1471
        tag_a: Union[None, int, str],
1472
        tag_z: Union[None, int, str],
1473
        interface_a: Interface,
1474
        interface_z: Interface,
1475
        current_path: list,
1476
        trace_a: list,
1477
        trace_z: list
1478
    ) -> bool:
1479
        """Auxiliar function to check an individual trace"""
1480 1
        if (
1481
            len(trace_a) != len(current_path) + 1
1482
            or not compare_uni_out_trace(tag_z, interface_z, trace_a[-1])
1483
        ):
1484 1
            log.warning(f"From EVC({evc_id}) named '{evc_name}'. "
1485
                        f"Invalid trace from uni_a: {trace_a}")
1486 1
            return False
1487 1
        if (
1488
            len(trace_z) != len(current_path) + 1
1489
            or not compare_uni_out_trace(tag_a, interface_a, trace_z[-1])
1490
        ):
1491 1
            log.warning(f"From EVC({evc_id}) named '{evc_name}'. "
1492
                        f"Invalid trace from uni_z: {trace_z}")
1493 1
            return False
1494
1495 1
        if not current_path:
1496
            return True
1497
1498 1
        first_link, trace_path_begin, trace_path_end = current_path[0], [], []
1499 1
        if (
1500
            first_link.endpoint_a.switch.id == trace_a[0]["dpid"]
1501
        ):
1502 1
            trace_path_begin, trace_path_end = trace_a, trace_z
1503 1
        elif (
1504
            first_link.endpoint_a.switch.id == trace_z[0]["dpid"]
1505
        ):
1506 1
            trace_path_begin, trace_path_end = trace_z, trace_a
1507
        else:
1508
            msg = (
1509
                f"first link {first_link} endpoint_a didn't match the first "
1510
                f"step of trace_a {trace_a} or trace_z {trace_z}"
1511
            )
1512
            log.warning(msg)
1513
            return False
1514
1515 1
        for link, trace1, trace2 in zip(current_path,
1516
                                        trace_path_begin[1:],
1517
                                        trace_path_end[:0:-1]):
1518 1
            metadata_vlan = None
1519 1
            if link.metadata:
1520 1
                metadata_vlan = glom(link.metadata, 's_vlan.value')
1521 1
            if compare_endpoint_trace(
1522
                                        link.endpoint_a,
1523
                                        metadata_vlan,
1524
                                        trace2
1525
                                    ) is False:
1526 1
                log.warning(f"From EVC({evc_id}) named '{evc_name}'. "
1527
                            f"Invalid trace from uni_a: {trace_a}")
1528 1
                return False
1529 1
            if compare_endpoint_trace(
1530
                                        link.endpoint_b,
1531
                                        metadata_vlan,
1532
                                        trace1
1533
                                    ) is False:
1534 1
                log.warning(f"From EVC({evc_id}) named '{evc_name}'. "
1535
                            f"Invalid trace from uni_z: {trace_z}")
1536 1
                return False
1537
1538 1
        return True
1539
1540 1
    @staticmethod
1541 1
    def check_range(circuit, traces: list) -> bool:
1542
        """Check traces when for UNI with TAGRange"""
1543 1
        check = True
1544 1
        for i, mask in enumerate(circuit.uni_a.user_tag.mask_list):
1545 1
            trace_a = traces[i*2]
1546 1
            trace_z = traces[i*2+1]
1547 1
            check &= EVCDeploy.check_trace(
1548
                circuit.id, circuit.name,
1549
                mask, mask,
1550
                circuit.uni_a.interface,
1551
                circuit.uni_z.interface,
1552
                circuit.current_path,
1553
                trace_a, trace_z,
1554
            )
1555 1
        return check
1556
1557 1
    @staticmethod
1558 1
    def check_list_traces(list_circuits: list) -> dict:
1559
        """Check if current_path is deployed comparing with SDN traces."""
1560 1
        if not list_circuits:
1561 1
            return {}
1562 1
        uni_list = make_uni_list(list_circuits)
1563 1
        traces = EVCDeploy.run_bulk_sdntraces(uni_list)["result"]
1564
1565 1
        if not traces:
1566 1
            return {}
1567
1568 1
        try:
1569 1
            circuits_checked = {}
1570 1
            i = 0
1571 1
            for circuit in list_circuits:
1572 1
                if isinstance(circuit.uni_a.user_tag, TAGRange):
1573 1
                    length = len(circuit.uni_a.user_tag.mask_list)
1574 1
                    circuits_checked[circuit.id] = EVCDeploy.check_range(
1575
                        circuit, traces[i:i+length*2]
1576
                    )
1577 1
                    i += length*2
1578
                else:
1579 1
                    trace_a = traces[i]
1580 1
                    trace_z = traces[i+1]
1581 1
                    tag_a = None
1582 1
                    if circuit.uni_a.user_tag:
1583 1
                        tag_a = circuit.uni_a.user_tag.value
1584 1
                    tag_z = None
1585 1
                    if circuit.uni_z.user_tag:
1586 1
                        tag_z = circuit.uni_z.user_tag.value
1587 1
                    circuits_checked[circuit.id] = EVCDeploy.check_trace(
1588
                        circuit.id, circuit.name,
1589
                        tag_a, tag_z,
1590
                        circuit.uni_a.interface,
1591
                        circuit.uni_z.interface,
1592
                        circuit.current_path,
1593
                        trace_a, trace_z
1594
                    )
1595 1
                    i += 2
1596 1
        except IndexError as err:
1597 1
            log.error(
1598
                f"Bulk sdntraces returned fewer items than expected."
1599
                f"Error = {err}"
1600
            )
1601 1
            return {}
1602
1603 1
        return circuits_checked
1604
1605 1
    @staticmethod
1606 1
    def get_endpoint_by_id(
1607
        link: Link,
1608
        id_: str,
1609
        operator: Union[eq, ne]
1610
    ) -> Interface:
1611
        """Return endpoint from link
1612
        either equal(eq) or not equal(ne) to id"""
1613 1
        if operator(link.endpoint_a.switch.id, id_):
1614 1
            return link.endpoint_a
1615 1
        return link.endpoint_b
1616
1617
1618 1
class LinkProtection(EVCDeploy):
1619
    """Class to handle link protection."""
1620
1621 1
    def is_affected_by_link(self, link=None):
1622
        """Verify if the current path is affected by link down event."""
1623
        return self.current_path.is_affected_by_link(link)
1624
1625 1
    def is_using_primary_path(self):
1626
        """Verify if the current deployed path is self.primary_path."""
1627 1
        return self.current_path == self.primary_path
1628
1629 1
    def is_using_backup_path(self):
1630
        """Verify if the current deployed path is self.backup_path."""
1631 1
        return self.current_path == self.backup_path
1632
1633 1
    def is_using_dynamic_path(self):
1634
        """Verify if the current deployed path is dynamic."""
1635 1
        if (
1636
            self.current_path
1637
            and not self.is_using_primary_path()
1638
            and not self.is_using_backup_path()
1639
            and self.current_path.status is EntityStatus.UP
1640
        ):
1641
            return True
1642 1
        return False
1643
1644 1
    def handle_link_up(self, link):
1645
        """Handle circuit when link up.
1646
1647
        Args:
1648
            link(Link): Link affected by link.up event.
1649
1650
        """
1651 1
        condition_pairs = [
1652
            (
1653
                lambda me: me.is_using_primary_path(),
1654
                lambda _: (True, 'nothing')
1655
            ),
1656
            (
1657
                lambda me: me.is_intra_switch(),
1658
                lambda _: (True, 'nothing')
1659
            ),
1660
            (
1661
                lambda me: me.primary_path.is_affected_by_link(link),
1662
                lambda me: (me.deploy_to_primary_path(), 'redeploy')
1663
            ),
1664
            # We tried to deploy(primary_path) without success.
1665
            # And in this case is up by some how. Nothing to do.
1666
            (
1667
                lambda me: me.is_using_backup_path(),
1668
                lambda _: (True, 'nothing')
1669
            ),
1670
            (
1671
                lambda me:  me.is_using_dynamic_path(),
1672
                lambda _: (True, 'nothing')
1673
            ),
1674
            # In this case, probably the circuit is not being used and
1675
            # we can move to backup
1676
            (
1677
                lambda me: me.backup_path.is_affected_by_link(link),
1678
                lambda me: (me.deploy_to_backup_path(), 'redeploy')
1679
            ),
1680
            # In this case, the circuit is not being used and we should
1681
            # try a dynamic path
1682
            (
1683
                lambda me: me.dynamic_backup_path and not me.is_active(),
1684
                lambda me: (me.deploy_to_path(), 'redeploy')
1685
            )
1686
        ]
1687 1
        for predicate, action in condition_pairs:
1688 1
            if not predicate(self):
1689 1
                continue
1690 1
            success, succcess_type = action(self)
1691 1
            if success:
1692 1
                if succcess_type == 'redeploy':
1693 1
                    emit_event(
1694
                        self._controller,
1695
                        "redeployed_link_up",
1696
                        content=map_evc_event_content(self)
1697
                    )
1698 1
                return True
1699 1
        return False
1700
1701 1
    def handle_link_down(self):
1702
        """Handle circuit when link down.
1703
1704
        Returns:
1705
            bool: True if the re-deploy was successly otherwise False.
1706
1707
        """
1708 1
        success = False
1709 1
        if self.is_using_primary_path():
1710 1
            success = self.deploy_to_backup_path()
1711 1
        elif self.is_using_backup_path():
1712 1
            success = self.deploy_to_primary_path()
1713
1714 1
        if not success and self.dynamic_backup_path:
1715 1
            success = self.deploy_to_path()
1716
1717 1
        if success:
1718 1
            log.debug(f"{self} deployed after link down.")
1719
        else:
1720 1
            self.remove_current_flows(sync=False)
1721 1
            self.deactivate()
1722 1
            self.sync()
1723 1
            log.debug(f"Failed to re-deploy {self} after link down.")
1724
1725 1
        return success
1726
1727 1
    @staticmethod
1728 1
    def get_interface_from_switch(uni: UNI, switches: dict) -> Interface:
1729
        """Get interface from switch by uni"""
1730 1
        switch = switches[uni.interface.switch.dpid]
1731 1
        interface = switch.interfaces[uni.interface.port_number]
1732 1
        return interface
1733
1734 1
    def are_unis_active(self, switches: dict) -> bool:
1735
        """Determine whether this EVC should be active"""
1736 1
        interface_a = self.get_interface_from_switch(self.uni_a, switches)
1737 1
        interface_z = self.get_interface_from_switch(self.uni_z, switches)
1738 1
        active, _ = self.is_uni_interface_active(interface_a, interface_z)
1739 1
        return active
1740
1741 1
    @staticmethod
1742 1
    def is_uni_interface_active(
1743
        *interfaces: Interface
1744
    ) -> tuple[bool, dict]:
1745
        """Determine whether a UNI should be active"""
1746 1
        active = True
1747 1
        bad_interfaces = [
1748
            interface
1749
            for interface in interfaces
1750
            if interface.status != EntityStatus.UP
1751
        ]
1752 1
        if bad_interfaces:
1753 1
            active = False
1754 1
            interfaces = bad_interfaces
1755 1
        return active, {
1756
            interface.id: {
1757
                'status': interface.status.value,
1758
                'status_reason': interface.status_reason,
1759
            }
1760
            for interface in interfaces
1761
        }
1762
1763 1
    def handle_interface_link_up(self, interface: Interface):
1764
        """
1765
        Handler for interface link_up events
1766
        """
1767 1
        if self.is_active():
1768 1
            return
1769 1
        interfaces = (self.uni_a.interface, self.uni_z.interface)
1770 1
        if interface not in interfaces:
1771
            return
1772 1
        down_interfaces = [
1773
            interface
1774
            for interface in interfaces
1775
            if interface.status != EntityStatus.UP
1776
        ]
1777 1
        if down_interfaces:
1778
            return
1779 1
        interface_dicts = {
1780
            interface.id: {
1781
                'status': interface.status.value,
1782
                'status_reason': interface.status_reason,
1783
            }
1784
            for interface in interfaces
1785
        }
1786 1
        self.activate()
1787 1
        log.info(
1788
            f"Activating {self}. Interfaces: "
1789
            f"{interface_dicts}."
1790
        )
1791 1
        emit_event(self._controller, "uni_active_updated",
1792
                   content=map_evc_event_content(self))
1793 1
        self.sync()
1794
1795 1
    def handle_interface_link_down(self, interface):
1796
        """
1797
        Handler for interface link_down events
1798
        """
1799 1
        if not self.is_active():
1800 1
            return
1801 1
        interfaces = (self.uni_a.interface, self.uni_z.interface)
1802 1
        if interface not in interfaces:
1803
            return
1804 1
        down_interfaces = [
1805
            interface
1806
            for interface in interfaces
1807
            if interface.status != EntityStatus.UP
1808
        ]
1809 1
        if not down_interfaces:
1810
            return
1811 1
        interface_dicts = {
1812
            interface.id: {
1813
                'status': interface.status.value,
1814
                'status_reason': interface.status_reason,
1815
            }
1816
            for interface in down_interfaces
1817
        }
1818 1
        self.deactivate()
1819 1
        log.info(
1820
            f"Deactivating {self}. Interfaces: "
1821
            f"{interface_dicts}."
1822
        )
1823 1
        emit_event(self._controller, "uni_active_updated",
1824
                   content=map_evc_event_content(self))
1825 1
        self.sync()
1826
1827
1828 1
class EVC(LinkProtection):
1829
    """Class that represents a E-Line Virtual Connection."""
1830