Passed
Pull Request — master (#553)
by Aldo
04:09
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
                            "owner": "mef_eline",
698
                        }
699
                    ],
700
                    "delete",
701
                    force=force,
702
                )
703
            except FlowModException as err:
704
                log.error(f"Error deleting path in {switch.id}, {err}")
705
                break
706 1
        try:
707 1
            self.failover_path.make_vlans_available(self._controller)
708
        except KytosTagError as err:
709
            log.error(f"Error when removing failover flows: {err}")
710 1
        self.failover_path = Path([])
711 1
        if sync:
712 1
            self.sync()
713
714 1
    def remove_current_flows(self, force=True, sync=True):
715
        """Remove all flows from current path or path intended for
716
         current path if exists."""
717 1
        switches = set()
718
719 1
        if not self.current_path and not self.is_intra_switch():
720 1
            return
721 1
        current_path = self.current_path
722 1
        for link in current_path:
723 1
            switches.add(link.endpoint_a.switch)
724 1
            switches.add(link.endpoint_b.switch)
725 1
        switches.add(self.uni_a.interface.switch)
726 1
        switches.add(self.uni_z.interface.switch)
727 1
        match = {
728
            "cookie": self.get_cookie(),
729
            "cookie_mask": int(0xffffffffffffffff),
730
            "owner": "mef_eline",
731
        }
732
733 1
        for switch in switches:
734 1
            try:
735 1
                self._send_flow_mods(switch.id, [match], 'delete', force=force)
736
            except FlowModException as err:
737
                log.error(f"Error deleting current_path in {switch.id}, {err}")
738
                break
739 1
        try:
740 1
            current_path.make_vlans_available(self._controller)
741
        except KytosTagError as err:
742
            log.error(f"Error when removing current path flows: {err}")
743 1
        self.current_path = Path([])
744 1
        self.deactivate()
745 1
        if sync:
746 1
            self.sync()
747
748 1
    def remove_path_flows(
749
        self, path=None, force=True
750
    ) -> dict[str, list[dict]]:
751
        """Remove all flows from path, and return the removed flows."""
752 1
        dpid_flows_match: dict[str, list[dict]] = {}
753
754 1
        if not path:
755 1
            return dpid_flows_match
756
757 1
        try:
758 1
            nni_flows = self._prepare_nni_flows(path)
759
        # pylint: disable=broad-except
760
        except Exception:
761
            err = traceback.format_exc().replace("\n", ", ")
762
            log.error(f"Fail to remove NNI failover flows for {self}: {err}")
763
            nni_flows = {}
764
765 1
        for dpid, flows in nni_flows.items():
766 1
            dpid_flows_match.setdefault(dpid, [])
767 1
            for flow in flows:
768 1
                dpid_flows_match[dpid].append({
769
                    "cookie": flow["cookie"],
770
                    "match": flow["match"],
771
                    "owner": "mef_eline",
772
                    "cookie_mask": int(0xffffffffffffffff)
773
                })
774
775 1
        try:
776 1
            uni_flows = self._prepare_uni_flows(path, skip_in=True)
777
        # pylint: disable=broad-except
778
        except Exception:
779
            err = traceback.format_exc().replace("\n", ", ")
780
            log.error(f"Fail to remove UNI failover flows for {self}: {err}")
781
            uni_flows = {}
782
783 1
        for dpid, flows in uni_flows.items():
784 1
            dpid_flows_match.setdefault(dpid, [])
785 1
            for flow in flows:
786 1
                dpid_flows_match[dpid].append({
787
                    "cookie": flow["cookie"],
788
                    "match": flow["match"],
789
                    "owner": "mef_eline",
790
                    "cookie_mask": int(0xffffffffffffffff)
791
                })
792
793 1
        for dpid, flows in dpid_flows_match.items():
794 1
            try:
795 1
                self._send_flow_mods(dpid, flows, 'delete', force=force)
796 1
            except FlowModException as err:
797 1
                log.error(f"Error deleting path in {dpid}, {err}")
798 1
                break
799 1
        try:
800 1
            path.make_vlans_available(self._controller)
801
        except KytosTagError as err:
802
            log.error(f"Error when removing path flows: {err}")
803
804 1
        return dpid_flows_match
805
806 1
    @staticmethod
807 1
    def links_zipped(path=None):
808
        """Return an iterator which yields pairs of links in order."""
809 1
        if not path:
810 1
            return []
811 1
        return zip(path[:-1], path[1:])
812
813 1
    def should_deploy(self, path=None):
814
        """Verify if the circuit should be deployed."""
815 1
        if not path:
816 1
            log.debug("Path is empty.")
817 1
            return False
818
819 1
        if not self.is_enabled():
820 1
            log.debug(f"{self} is disabled.")
821 1
            return False
822
823 1
        if not self.is_active():
824 1
            log.debug(f"{self} will be deployed.")
825 1
            return True
826
827 1
        return False
828
829
    # pylint: disable=too-many-branches, too-many-statements
830 1
    def deploy_to_path(self, path=None):
831
        """Install the flows for this circuit.
832
833
        Procedures to deploy:
834
835
        0. Remove current flows installed
836
        1. Decide if will deploy "path" or discover a new path
837
        2. Choose vlan
838
        3. Install NNI flows
839
        4. Install UNI flows
840
        5. Activate
841
        6. Update current_path
842
        7. Update links caches(primary, current, backup)
843
844
        """
845 1
        self.remove_current_flows(sync=False)
846 1
        use_path = path or Path([])
847 1
        tag_errors = []
848 1
        if self.should_deploy(use_path):
849 1
            try:
850 1
                use_path.choose_vlans(self._controller)
851 1
            except KytosNoTagAvailableError as e:
852 1
                tag_errors.append(str(e))
853 1
                use_path = None
854
        else:
855 1
            for use_path in self.discover_new_paths():
856 1
                if use_path is None:
857
                    continue
858 1
                try:
859 1
                    use_path.choose_vlans(self._controller)
860 1
                    break
861 1
                except KytosNoTagAvailableError as e:
862 1
                    tag_errors.append(str(e))
863
            else:
864 1
                use_path = None
865
866 1
        try:
867 1
            if use_path:
868 1
                self._install_nni_flows(use_path)
869 1
                self._install_uni_flows(use_path)
870 1
            elif self.is_intra_switch():
871 1
                use_path = Path()
872 1
                self._install_direct_uni_flows()
873
            else:
874 1
                msg = f"{self} was not deployed. No available path was found."
875 1
                if tag_errors:
876 1
                    msg = self.add_tag_errors(msg, tag_errors)
877 1
                    log.error(msg)
878
                else:
879 1
                    log.warning(msg)
880 1
                return False
881 1
        except EVCPathNotInstalled as err:
882 1
            log.error(
883
                f"Error deploying EVC {self} when calling flow_manager: {err}"
884
            )
885 1
            self.remove_current_flows(use_path, sync=True)
886 1
            return False
887 1
        self.activate()
888 1
        self.current_path = use_path
889 1
        self.sync()
890 1
        log.info(f"{self} was deployed.")
891 1
        return True
892
893 1
    def try_setup_failover_path(self, wait=settings.DEPLOY_EVCS_INTERVAL):
894
        """Try setup failover_path whenever possible."""
895 1
        if (
896
                self.failover_path or not self.current_path
897
                or not self.is_active()
898
                ):
899 1
            return
900 1
        if (now() - self.affected_by_link_at).seconds >= wait:
901 1
            with self.lock:
902 1
                self.setup_failover_path()
903
904
    # pylint: disable=too-many-statements
905 1
    def setup_failover_path(self):
906
        """Install flows for the failover path of this EVC.
907
908
        Procedures to deploy:
909
910
        0. Remove flows currently installed for failover_path (if any)
911
        1. Discover a disjoint path from current_path
912
        2. Choose vlans
913
        3. Install NNI flows
914
        4. Install UNI egress flows
915
        5. Update failover_path
916
        """
917
        # Intra-switch EVCs have no failover_path
918 1
        if self.is_intra_switch():
919 1
            return False
920
921
        # For not only setup failover path for totally dynamic EVCs
922 1
        if not self.is_eligible_for_failover_path():
923 1
            return False
924
925 1
        out_new_flows: dict[str, list[dict]] = {}
926 1
        reason = ""
927 1
        tag_errors = []
928 1
        out_removed_flows = self.remove_path_flows(self.failover_path)
929 1
        self.failover_path = Path([])
930
931 1
        for use_path in self.get_failover_path_candidates():
932 1
            if not use_path:
933 1
                continue
934 1
            try:
935 1
                use_path.choose_vlans(self._controller)
936 1
                break
937 1
            except KytosNoTagAvailableError as e:
938 1
                tag_errors.append(str(e))
939
        else:
940 1
            use_path = Path([])
941 1
            reason = "No available path was found"
942
943 1
        try:
944 1
            if use_path:
945 1
                _nni_flows = self._install_nni_flows(use_path)
946 1
                out_new_flows = merge_flow_dicts(out_new_flows, _nni_flows)
947 1
                _uni_flows = self._install_uni_flows(use_path, skip_in=True)
948 1
                out_new_flows = merge_flow_dicts(out_new_flows, _uni_flows)
949 1
        except EVCPathNotInstalled as err:
950 1
            reason = "Error deploying failover path"
951 1
            log.error(
952
                f"{reason} for {self}. FlowManager error: {err}"
953
            )
954 1
            _rmed_flows = self.remove_path_flows(use_path)
955 1
            out_removed_flows = merge_flow_dicts(
956
                out_removed_flows, _rmed_flows
957
            )
958 1
            use_path = Path([])
959
960 1
        self.failover_path = use_path
961 1
        self.sync()
962
963 1
        if out_new_flows or out_removed_flows:
964 1
            emit_event(self._controller, "failover_deployed", content={
965
                self.id: map_evc_event_content(
966
                    self,
967
                    flows=deepcopy(out_new_flows),
968
                    removed_flows=deepcopy(out_removed_flows),
969
                    error_reason=reason,
970
                    current_path=self.current_path.as_dict(),
971
                )
972
            })
973
974 1
        if not use_path:
975 1
            msg = f"Failover path for {self} was not deployed: {reason}."
976 1
            if tag_errors:
977 1
                msg = self.add_tag_errors(msg, tag_errors)
978 1
                log.error(msg)
979
            else:
980 1
                log.warning(msg)
981 1
            return False
982 1
        log.info(f"Failover path for {self} was deployed.")
983 1
        return True
984
985 1
    @staticmethod
986 1
    def add_tag_errors(msg: str, tag_errors: list):
987
        """Add to msg the tag errors ecountered when chossing path."""
988 1
        path = ['path', 'paths']
989 1
        was = ['was', 'were']
990 1
        message = ['message', 'messages']
991
992
        # Choose either singular(0) or plural(1) words
993 1
        n = 1
994 1
        if len(tag_errors) == 1:
995 1
            n = 0
996
997 1
        msg += f" {len(tag_errors)} {path[n]} {was[n]} rejected"
998 1
        msg += f" with {message[n]}: {tag_errors}"
999 1
        return msg
1000
1001 1
    def get_failover_flows(self):
1002
        """Return the flows needed to make the failover path active, i.e. the
1003
        flows for ingress forwarding.
1004
1005
        Return:
1006
            dict: A dict of flows indexed by the switch_id will be returned, or
1007
                an empty dict if no failover_path is available.
1008
        """
1009 1
        if not self.failover_path:
1010 1
            return {}
1011 1
        return self._prepare_uni_flows(self.failover_path, skip_out=True)
1012
1013
    # pylint: disable=too-many-branches
1014 1
    def _prepare_direct_uni_flows(self):
1015
        """Prepare flows connecting two UNIs for intra-switch EVC."""
1016 1
        vlan_a = self._get_value_from_uni_tag(self.uni_a)
1017 1
        vlan_z = self._get_value_from_uni_tag(self.uni_z)
1018
1019 1
        flow_mod_az = self._prepare_flow_mod(
1020
            self.uni_a.interface, self.uni_z.interface,
1021
            self.queue_id, vlan_a
1022
        )
1023 1
        flow_mod_za = self._prepare_flow_mod(
1024
            self.uni_z.interface, self.uni_a.interface,
1025
            self.queue_id, vlan_z
1026
        )
1027
1028 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...
1029 1
            flow_mod_az["actions"].insert(
1030
                0, {"action_type": "set_vlan", "vlan_id": vlan_z}
1031
            )
1032 1
            if not vlan_a:
1033 1
                flow_mod_az["actions"].insert(
1034
                    0, {"action_type": "push_vlan", "tag_type": "c"}
1035
                )
1036 1
            if vlan_a == 0:
1037 1
                flow_mod_za["actions"].insert(0, {"action_type": "pop_vlan"})
1038 1
        elif vlan_a == 0 and vlan_z == "4096/4096":
1039 1
            flow_mod_za["actions"].insert(0, {"action_type": "pop_vlan"})
1040
1041 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...
1042 1
            flow_mod_za["actions"].insert(
1043
                    0, {"action_type": "set_vlan", "vlan_id": vlan_a}
1044
                )
1045 1
            if not vlan_z:
1046 1
                flow_mod_za["actions"].insert(
1047
                    0, {"action_type": "push_vlan", "tag_type": "c"}
1048
                )
1049 1
            if vlan_z == 0:
1050 1
                flow_mod_az["actions"].insert(0, {"action_type": "pop_vlan"})
1051 1
        elif vlan_a == "4096/4096" and vlan_z == 0:
1052 1
            flow_mod_az["actions"].insert(0, {"action_type": "pop_vlan"})
1053
1054 1
        flows = []
1055 1
        if isinstance(vlan_a, list):
1056 1
            for mask_a in vlan_a:
1057 1
                flow_aux = deepcopy(flow_mod_az)
1058 1
                flow_aux["match"]["dl_vlan"] = mask_a
1059 1
                flows.append(flow_aux)
1060
        else:
1061 1
            if vlan_a is not None:
1062 1
                flow_mod_az["match"]["dl_vlan"] = vlan_a
1063 1
            flows.append(flow_mod_az)
1064
1065 1
        if isinstance(vlan_z, list):
1066 1
            for mask_z in vlan_z:
1067 1
                flow_aux = deepcopy(flow_mod_za)
1068 1
                flow_aux["match"]["dl_vlan"] = mask_z
1069 1
                flows.append(flow_aux)
1070
        else:
1071 1
            if vlan_z is not None:
1072 1
                flow_mod_za["match"]["dl_vlan"] = vlan_z
1073 1
            flows.append(flow_mod_za)
1074 1
        return (
1075
            self.uni_a.interface.switch.id, flows
1076
        )
1077
1078 1
    def _install_direct_uni_flows(self):
1079
        """Install flows connecting two UNIs.
1080
1081
        This case happens when the circuit is between UNIs in the
1082
        same switch.
1083
        """
1084 1
        (dpid, flows) = self._prepare_direct_uni_flows()
1085 1
        try:
1086 1
            self._send_flow_mods(dpid, flows)
1087 1
        except FlowModException as err:
1088 1
            raise EVCPathNotInstalled(f"In {dpid}, {err}") from err
1089
1090 1
    def _prepare_nni_flows(self, path=None):
1091
        """Prepare NNI flows."""
1092 1
        nni_flows = OrderedDict()
1093 1
        previous = self.uni_a.interface.switch.dpid
1094 1
        for incoming, outcoming in self.links_zipped(path):
1095 1
            in_vlan = incoming.get_metadata("s_vlan").value
1096 1
            out_vlan = outcoming.get_metadata("s_vlan").value
1097 1
            in_endpoint = self.get_endpoint_by_id(incoming, previous, ne)
1098 1
            out_endpoint = self.get_endpoint_by_id(
1099
                outcoming, in_endpoint.switch.id, eq
1100
            )
1101
1102 1
            flows = []
1103
            # Flow for one direction
1104 1
            flows.append(
1105
                self._prepare_nni_flow(
1106
                    in_endpoint,
1107
                    out_endpoint,
1108
                    in_vlan,
1109
                    out_vlan,
1110
                    queue_id=self.queue_id,
1111
                )
1112
            )
1113
1114
            # Flow for the other direction
1115 1
            flows.append(
1116
                self._prepare_nni_flow(
1117
                    out_endpoint,
1118
                    in_endpoint,
1119
                    out_vlan,
1120
                    in_vlan,
1121
                    queue_id=self.queue_id,
1122
                )
1123
            )
1124 1
            previous = in_endpoint.switch.id
1125 1
            nni_flows[in_endpoint.switch.id] = flows
1126 1
        return nni_flows
1127
1128 1
    def _install_nni_flows(
1129
        self, path=None
1130
    ) -> dict[str, list[dict]]:
1131
        """Install NNI flows."""
1132 1
        nni_flows = self._prepare_nni_flows(path)
1133 1
        try:
1134 1
            for dpid, flows in nni_flows.items():
1135 1
                self._send_flow_mods(dpid, flows)
1136 1
        except FlowModException as err:
1137 1
            raise EVCPathNotInstalled(f"In {dpid}, {err}") from err
1138 1
        return nni_flows
1139
1140 1
    @staticmethod
1141 1
    def _get_value_from_uni_tag(uni: UNI):
1142
        """Returns the value from tag. In case of any and untagged
1143
        it should return 4096/4096 and 0 respectively"""
1144 1
        special = {"any": "4096/4096", "untagged": 0}
1145 1
        if uni.user_tag:
1146 1
            value = uni.user_tag.value
1147 1
            if isinstance(value, list):
1148 1
                return uni.user_tag.mask_list
1149 1
            return special.get(value, value)
1150 1
        return None
1151
1152
    # pylint: disable=too-many-locals
1153 1
    def _prepare_uni_flows(self, path=None, skip_in=False, skip_out=False):
1154
        """Prepare flows to install UNIs."""
1155 1
        uni_flows = {}
1156 1
        if not path:
1157 1
            log.info("install uni flows without path.")
1158 1
            return uni_flows
1159
1160
        # Determine VLANs
1161 1
        in_vlan_a = self._get_value_from_uni_tag(self.uni_a)
1162 1
        out_vlan_a = path[0].get_metadata("s_vlan").value
1163
1164 1
        in_vlan_z = self._get_value_from_uni_tag(self.uni_z)
1165 1
        out_vlan_z = path[-1].get_metadata("s_vlan").value
1166
1167
        # Get endpoints from path
1168 1
        endpoint_a = self.get_endpoint_by_id(
1169
            path[0], self.uni_a.interface.switch.id, eq
1170
        )
1171 1
        endpoint_z = self.get_endpoint_by_id(
1172
            path[-1], self.uni_z.interface.switch.id, eq
1173
        )
1174
1175
        # Flows for the first UNI
1176 1
        flows_a = []
1177
1178
        # Flow for one direction, pushing the service tag
1179 1
        if not skip_in:
1180 1
            if isinstance(in_vlan_a, list):
1181 1
                for in_mask_a in in_vlan_a:
1182 1
                    push_flow = self._prepare_push_flow(
1183
                        self.uni_a.interface,
1184
                        endpoint_a,
1185
                        in_mask_a,
1186
                        out_vlan_a,
1187
                        in_vlan_z,
1188
                        queue_id=self.queue_id,
1189
                    )
1190 1
                    flows_a.append(push_flow)
1191
            else:
1192 1
                push_flow = self._prepare_push_flow(
1193
                    self.uni_a.interface,
1194
                    endpoint_a,
1195
                    in_vlan_a,
1196
                    out_vlan_a,
1197
                    in_vlan_z,
1198
                    queue_id=self.queue_id,
1199
                )
1200 1
                flows_a.append(push_flow)
1201
1202
        # Flow for the other direction, popping the service tag
1203 1
        if not skip_out:
1204 1
            pop_flow = self._prepare_pop_flow(
1205
                endpoint_a,
1206
                self.uni_a.interface,
1207
                out_vlan_a,
1208
                queue_id=self.queue_id,
1209
            )
1210 1
            flows_a.append(pop_flow)
1211
1212 1
        uni_flows[self.uni_a.interface.switch.id] = flows_a
1213
1214
        # Flows for the second UNI
1215 1
        flows_z = []
1216
1217
        # Flow for one direction, pushing the service tag
1218 1
        if not skip_in:
1219 1
            if isinstance(in_vlan_z, list):
1220 1
                for in_mask_z in in_vlan_z:
1221 1
                    push_flow = self._prepare_push_flow(
1222
                        self.uni_z.interface,
1223
                        endpoint_z,
1224
                        in_mask_z,
1225
                        out_vlan_z,
1226
                        in_vlan_a,
1227
                        queue_id=self.queue_id,
1228
                    )
1229 1
                    flows_z.append(push_flow)
1230
            else:
1231 1
                push_flow = self._prepare_push_flow(
1232
                    self.uni_z.interface,
1233
                    endpoint_z,
1234
                    in_vlan_z,
1235
                    out_vlan_z,
1236
                    in_vlan_a,
1237
                    queue_id=self.queue_id,
1238
                )
1239 1
                flows_z.append(push_flow)
1240
1241
        # Flow for the other direction, popping the service tag
1242 1
        if not skip_out:
1243 1
            pop_flow = self._prepare_pop_flow(
1244
                endpoint_z,
1245
                self.uni_z.interface,
1246
                out_vlan_z,
1247
                queue_id=self.queue_id,
1248
            )
1249 1
            flows_z.append(pop_flow)
1250
1251 1
        uni_flows[self.uni_z.interface.switch.id] = flows_z
1252
1253 1
        return uni_flows
1254
1255 1
    def _install_uni_flows(
1256
        self, path=None, skip_in=False, skip_out=False
1257
    ) -> dict[str, list[dict]]:
1258
        """Install UNI flows."""
1259 1
        uni_flows = self._prepare_uni_flows(path, skip_in, skip_out)
1260 1
        try:
1261 1
            for (dpid, flows) in uni_flows.items():
1262 1
                self._send_flow_mods(dpid, flows)
1263
        except FlowModException as err:
1264
            raise EVCPathNotInstalled(f"In {dpid}, {err}") from err
1265
1266 1
        return uni_flows
1267
1268 1
    @staticmethod
1269 1
    @retry(
1270
        stop=stop_after_attempt(3),
1271
        wait=wait_combine(wait_fixed(3), wait_random(min=2, max=7)),
1272
        retry=retry_if_exception_type(FlowModException),
1273
        before_sleep=before_sleep,
1274
        reraise=True,
1275
    )
1276 1
    def _send_flow_mods(dpid, flow_mods, command='flows', force=False):
1277
        """Send a flow_mod list to a specific switch.
1278
1279
        Args:
1280
            dpid(str): The target of flows (i.e. Switch.id).
1281
            flow_mods(dict): Python dictionary with flow_mods.
1282
            command(str): By default is 'flows'. To remove a flow is 'remove'.
1283
            force(bool): True to send via consistency check in case of errors
1284
1285
        """
1286 1
        endpoint = f"{settings.MANAGER_URL}/{command}/{dpid}"
1287
1288 1
        data = {"flows": flow_mods, "force": force}
1289 1
        try:
1290 1
            res = httpx.post(endpoint, json=data, timeout=30)
1291
        except httpx.RequestError as err:
1292
            raise FlowModException(str(err)) from err
1293 1
        if res.is_server_error or res.status_code >= 400:
1294 1
            raise FlowModException(res.text)
1295
1296 1
    def get_cookie(self):
1297
        """Return the cookie integer from evc id."""
1298 1
        return int(self.id, 16) + (settings.COOKIE_PREFIX << 56)
1299
1300 1
    @staticmethod
1301 1
    def get_id_from_cookie(cookie):
1302
        """Return the evc id given a cookie value."""
1303 1
        evc_id = cookie - (settings.COOKIE_PREFIX << 56)
1304 1
        return f"{evc_id:x}".zfill(14)
1305
1306 1
    def set_flow_table_group_id(self, flow_mod: dict, vlan) -> dict:
1307
        """Set table_group and table_id"""
1308 1
        table_group = "epl" if vlan is None else "evpl"
1309 1
        flow_mod["table_group"] = table_group
1310 1
        flow_mod["table_id"] = self.table_group[table_group]
1311 1
        return flow_mod
1312
1313 1
    @staticmethod
1314 1
    def get_priority(vlan):
1315
        """Return priority value depending on vlan value"""
1316 1
        if isinstance(vlan, list):
1317 1
            return settings.EVPL_SB_PRIORITY
1318 1
        if vlan not in {None, "4096/4096", 0}:
1319 1
            return settings.EVPL_SB_PRIORITY
1320 1
        if vlan == 0:
1321 1
            return settings.UNTAGGED_SB_PRIORITY
1322 1
        if vlan == "4096/4096":
1323 1
            return settings.ANY_SB_PRIORITY
1324 1
        return settings.EPL_SB_PRIORITY
1325
1326 1
    def _prepare_flow_mod(self, in_interface, out_interface,
1327
                          queue_id=None, vlan=True):
1328
        """Prepare a common flow mod."""
1329 1
        default_actions = [
1330
            {"action_type": "output", "port": out_interface.port_number}
1331
        ]
1332 1
        queue_id = settings.QUEUE_ID if queue_id == -1 else queue_id
1333 1
        if queue_id is not None:
1334 1
            default_actions.append(
1335
                {"action_type": "set_queue", "queue_id": queue_id}
1336
            )
1337
1338 1
        flow_mod = {
1339
            "match": {"in_port": in_interface.port_number},
1340
            "cookie": self.get_cookie(),
1341
            "actions": default_actions,
1342
            "owner": "mef_eline",
1343
        }
1344
1345 1
        self.set_flow_table_group_id(flow_mod, vlan)
1346 1
        if self.sb_priority:
1347 1
            flow_mod["priority"] = self.sb_priority
1348
        else:
1349 1
            flow_mod["priority"] = self.get_priority(vlan)
1350 1
        return flow_mod
1351
1352 1
    def _prepare_nni_flow(self, *args, queue_id=None):
1353
        """Create NNI flows."""
1354 1
        in_interface, out_interface, in_vlan, out_vlan = args
1355 1
        flow_mod = self._prepare_flow_mod(
1356
            in_interface, out_interface, queue_id
1357
        )
1358 1
        flow_mod["match"]["dl_vlan"] = in_vlan
1359 1
        new_action = {"action_type": "set_vlan", "vlan_id": out_vlan}
1360 1
        flow_mod["actions"].insert(0, new_action)
1361
1362 1
        return flow_mod
1363
1364 1
    def _prepare_push_flow(self, *args, queue_id=None):
1365
        """Prepare push flow.
1366
1367
        Arguments:
1368
            in_interface(str): Interface input.
1369
            out_interface(str): Interface output.
1370
            in_vlan(int,str,None): Vlan input.
1371
            out_vlan(str): Vlan output.
1372
            new_c_vlan(int,str,list,None): New client vlan.
1373
1374
        Return:
1375
            dict: An python dictionary representing a FlowMod
1376
1377
        """
1378
        # assign all arguments
1379 1
        in_interface, out_interface, in_vlan, out_vlan, new_c_vlan = args
1380 1
        vlan_pri = in_vlan if not isinstance(new_c_vlan, list) else new_c_vlan
1381 1
        flow_mod = self._prepare_flow_mod(
1382
            in_interface, out_interface, queue_id, vlan_pri
1383
        )
1384
        # the service tag must be always pushed
1385 1
        new_action = {"action_type": "set_vlan", "vlan_id": out_vlan}
1386 1
        flow_mod["actions"].insert(0, new_action)
1387
1388 1
        new_action = {"action_type": "push_vlan", "tag_type": "s"}
1389 1
        flow_mod["actions"].insert(0, new_action)
1390
1391 1
        if in_vlan is not None:
1392
            # if in_vlan is set, it must be included in the match
1393 1
            flow_mod["match"]["dl_vlan"] = in_vlan
1394
1395 1
        if (not isinstance(new_c_vlan, list) and in_vlan != new_c_vlan and
1396
                new_c_vlan not in self.special_cases):
1397
            # new_in_vlan is an integer but zero, action to set is required
1398 1
            new_action = {"action_type": "set_vlan", "vlan_id": new_c_vlan}
1399 1
            flow_mod["actions"].insert(0, new_action)
1400
1401 1
        if in_vlan not in self.special_cases and new_c_vlan == 0:
1402
            # # new_in_vlan is an integer but zero and new_c_vlan does not
1403
            # a pop action is required
1404 1
            new_action = {"action_type": "pop_vlan"}
1405 1
            flow_mod["actions"].insert(0, new_action)
1406
1407 1
        elif in_vlan == "4096/4096" and new_c_vlan == 0:
1408
            # if in_vlan match with any tags and new_c_vlan does not
1409
            # a pop action is required
1410 1
            new_action = {"action_type": "pop_vlan"}
1411 1
            flow_mod["actions"].insert(0, new_action)
1412
1413 1
        elif (not in_vlan and
1414
                (not isinstance(new_c_vlan, list) and
1415
                 new_c_vlan not in self.special_cases)):
1416
            # new_in_vlan is an integer but zero and in_vlan is not set
1417
            # then it is set now
1418 1
            new_action = {"action_type": "push_vlan", "tag_type": "c"}
1419 1
            flow_mod["actions"].insert(0, new_action)
1420
1421 1
        return flow_mod
1422
1423 1
    def _prepare_pop_flow(
1424
        self, in_interface, out_interface, out_vlan, queue_id=None
1425
    ):
1426
        # pylint: disable=too-many-arguments
1427
        """Prepare pop flow."""
1428 1
        flow_mod = self._prepare_flow_mod(
1429
            in_interface, out_interface, queue_id
1430
        )
1431 1
        flow_mod["match"]["dl_vlan"] = out_vlan
1432 1
        new_action = {"action_type": "pop_vlan"}
1433 1
        flow_mod["actions"].insert(0, new_action)
1434 1
        return flow_mod
1435
1436 1
    @staticmethod
1437 1
    def run_bulk_sdntraces(
1438
        uni_list: list[tuple[Interface, Union[str, int, None]]]
1439
    ) -> dict:
1440
        """Run SDN traces on control plane starting from EVC UNIs."""
1441 1
        endpoint = f"{settings.SDN_TRACE_CP_URL}/traces"
1442 1
        data = []
1443 1
        for interface, tag_value in uni_list:
1444 1
            data_uni = {
1445
                "trace": {
1446
                            "switch": {
1447
                                "dpid": interface.switch.dpid,
1448
                                "in_port": interface.port_number,
1449
                            }
1450
                        }
1451
                }
1452 1
            if tag_value:
1453 1
                uni_dl_vlan = map_dl_vlan(tag_value)
1454 1
                if uni_dl_vlan:
1455 1
                    data_uni["trace"]["eth"] = {
1456
                                            "dl_type": 0x8100,
1457
                                            "dl_vlan": uni_dl_vlan,
1458
                                            }
1459 1
            data.append(data_uni)
1460 1
        try:
1461 1
            response = httpx.put(endpoint, json=data, timeout=30)
1462 1
        except httpx.TimeoutException as exception:
1463 1
            log.error(f"Request has timed out: {exception}")
1464 1
            return {"result": []}
1465 1
        if response.status_code >= 400:
1466 1
            log.error(f"Failed to run sdntrace-cp: {response.text}")
1467 1
            return {"result": []}
1468 1
        return response.json()
1469
1470
    # pylint: disable=too-many-return-statements, too-many-arguments
1471 1
    @staticmethod
1472 1
    def check_trace(
1473
        evc_id: str,
1474
        evc_name: str,
1475
        tag_a: Union[None, int, str],
1476
        tag_z: Union[None, int, str],
1477
        interface_a: Interface,
1478
        interface_z: Interface,
1479
        current_path: list,
1480
        trace_a: list,
1481
        trace_z: list
1482
    ) -> bool:
1483
        """Auxiliar function to check an individual trace"""
1484 1
        if (
1485
            len(trace_a) != len(current_path) + 1
1486
            or not compare_uni_out_trace(tag_z, interface_z, trace_a[-1])
1487
        ):
1488 1
            log.warning(f"From EVC({evc_id}) named '{evc_name}'. "
1489
                        f"Invalid trace from uni_a: {trace_a}")
1490 1
            return False
1491 1
        if (
1492
            len(trace_z) != len(current_path) + 1
1493
            or not compare_uni_out_trace(tag_a, interface_a, trace_z[-1])
1494
        ):
1495 1
            log.warning(f"From EVC({evc_id}) named '{evc_name}'. "
1496
                        f"Invalid trace from uni_z: {trace_z}")
1497 1
            return False
1498
1499 1
        if not current_path:
1500
            return True
1501
1502 1
        first_link, trace_path_begin, trace_path_end = current_path[0], [], []
1503 1
        if (
1504
            first_link.endpoint_a.switch.id == trace_a[0]["dpid"]
1505
        ):
1506 1
            trace_path_begin, trace_path_end = trace_a, trace_z
1507 1
        elif (
1508
            first_link.endpoint_a.switch.id == trace_z[0]["dpid"]
1509
        ):
1510 1
            trace_path_begin, trace_path_end = trace_z, trace_a
1511
        else:
1512
            msg = (
1513
                f"first link {first_link} endpoint_a didn't match the first "
1514
                f"step of trace_a {trace_a} or trace_z {trace_z}"
1515
            )
1516
            log.warning(msg)
1517
            return False
1518
1519 1
        for link, trace1, trace2 in zip(current_path,
1520
                                        trace_path_begin[1:],
1521
                                        trace_path_end[:0:-1]):
1522 1
            metadata_vlan = None
1523 1
            if link.metadata:
1524 1
                metadata_vlan = glom(link.metadata, 's_vlan.value')
1525 1
            if compare_endpoint_trace(
1526
                                        link.endpoint_a,
1527
                                        metadata_vlan,
1528
                                        trace2
1529
                                    ) is False:
1530 1
                log.warning(f"From EVC({evc_id}) named '{evc_name}'. "
1531
                            f"Invalid trace from uni_a: {trace_a}")
1532 1
                return False
1533 1
            if compare_endpoint_trace(
1534
                                        link.endpoint_b,
1535
                                        metadata_vlan,
1536
                                        trace1
1537
                                    ) is False:
1538 1
                log.warning(f"From EVC({evc_id}) named '{evc_name}'. "
1539
                            f"Invalid trace from uni_z: {trace_z}")
1540 1
                return False
1541
1542 1
        return True
1543
1544 1
    @staticmethod
1545 1
    def check_range(circuit, traces: list) -> bool:
1546
        """Check traces when for UNI with TAGRange"""
1547 1
        check = True
1548 1
        for i, mask in enumerate(circuit.uni_a.user_tag.mask_list):
1549 1
            trace_a = traces[i*2]
1550 1
            trace_z = traces[i*2+1]
1551 1
            check &= EVCDeploy.check_trace(
1552
                circuit.id, circuit.name,
1553
                mask, mask,
1554
                circuit.uni_a.interface,
1555
                circuit.uni_z.interface,
1556
                circuit.current_path,
1557
                trace_a, trace_z,
1558
            )
1559 1
        return check
1560
1561 1
    @staticmethod
1562 1
    def check_list_traces(list_circuits: list) -> dict:
1563
        """Check if current_path is deployed comparing with SDN traces."""
1564 1
        if not list_circuits:
1565 1
            return {}
1566 1
        uni_list = make_uni_list(list_circuits)
1567 1
        traces = EVCDeploy.run_bulk_sdntraces(uni_list)["result"]
1568
1569 1
        if not traces:
1570 1
            return {}
1571
1572 1
        try:
1573 1
            circuits_checked = {}
1574 1
            i = 0
1575 1
            for circuit in list_circuits:
1576 1
                if isinstance(circuit.uni_a.user_tag, TAGRange):
1577 1
                    length = len(circuit.uni_a.user_tag.mask_list)
1578 1
                    circuits_checked[circuit.id] = EVCDeploy.check_range(
1579
                        circuit, traces[i:i+length*2]
1580
                    )
1581 1
                    i += length*2
1582
                else:
1583 1
                    trace_a = traces[i]
1584 1
                    trace_z = traces[i+1]
1585 1
                    tag_a = None
1586 1
                    if circuit.uni_a.user_tag:
1587 1
                        tag_a = circuit.uni_a.user_tag.value
1588 1
                    tag_z = None
1589 1
                    if circuit.uni_z.user_tag:
1590 1
                        tag_z = circuit.uni_z.user_tag.value
1591 1
                    circuits_checked[circuit.id] = EVCDeploy.check_trace(
1592
                        circuit.id, circuit.name,
1593
                        tag_a, tag_z,
1594
                        circuit.uni_a.interface,
1595
                        circuit.uni_z.interface,
1596
                        circuit.current_path,
1597
                        trace_a, trace_z
1598
                    )
1599 1
                    i += 2
1600 1
        except IndexError as err:
1601 1
            log.error(
1602
                f"Bulk sdntraces returned fewer items than expected."
1603
                f"Error = {err}"
1604
            )
1605 1
            return {}
1606
1607 1
        return circuits_checked
1608
1609 1
    @staticmethod
1610 1
    def get_endpoint_by_id(
1611
        link: Link,
1612
        id_: str,
1613
        operator: Union[eq, ne]
1614
    ) -> Interface:
1615
        """Return endpoint from link
1616
        either equal(eq) or not equal(ne) to id"""
1617 1
        if operator(link.endpoint_a.switch.id, id_):
1618 1
            return link.endpoint_a
1619 1
        return link.endpoint_b
1620
1621
1622 1
class LinkProtection(EVCDeploy):
1623
    """Class to handle link protection."""
1624
1625 1
    def is_affected_by_link(self, link=None):
1626
        """Verify if the current path is affected by link down event."""
1627
        return self.current_path.is_affected_by_link(link)
1628
1629 1
    def is_using_primary_path(self):
1630
        """Verify if the current deployed path is self.primary_path."""
1631 1
        return self.current_path == self.primary_path
1632
1633 1
    def is_using_backup_path(self):
1634
        """Verify if the current deployed path is self.backup_path."""
1635 1
        return self.current_path == self.backup_path
1636
1637 1
    def is_using_dynamic_path(self):
1638
        """Verify if the current deployed path is dynamic."""
1639 1
        if (
1640
            self.current_path
1641
            and not self.is_using_primary_path()
1642
            and not self.is_using_backup_path()
1643
            and self.current_path.status is EntityStatus.UP
1644
        ):
1645
            return True
1646 1
        return False
1647
1648 1
    def handle_link_up(self, link):
1649
        """Handle circuit when link up.
1650
1651
        Args:
1652
            link(Link): Link affected by link.up event.
1653
1654
        """
1655 1
        condition_pairs = [
1656
            (
1657
                lambda me: me.is_using_primary_path(),
1658
                lambda _: (True, 'nothing')
1659
            ),
1660
            (
1661
                lambda me: me.is_intra_switch(),
1662
                lambda _: (True, 'nothing')
1663
            ),
1664
            (
1665
                lambda me: me.primary_path.is_affected_by_link(link),
1666
                lambda me: (me.deploy_to_primary_path(), 'redeploy')
1667
            ),
1668
            # We tried to deploy(primary_path) without success.
1669
            # And in this case is up by some how. Nothing to do.
1670
            (
1671
                lambda me: me.is_using_backup_path(),
1672
                lambda _: (True, 'nothing')
1673
            ),
1674
            (
1675
                lambda me:  me.is_using_dynamic_path(),
1676
                lambda _: (True, 'nothing')
1677
            ),
1678
            # In this case, probably the circuit is not being used and
1679
            # we can move to backup
1680
            (
1681
                lambda me: me.backup_path.is_affected_by_link(link),
1682
                lambda me: (me.deploy_to_backup_path(), 'redeploy')
1683
            ),
1684
            # In this case, the circuit is not being used and we should
1685
            # try a dynamic path
1686
            (
1687
                lambda me: me.dynamic_backup_path and not me.is_active(),
1688
                lambda me: (me.deploy_to_path(), 'redeploy')
1689
            )
1690
        ]
1691 1
        for predicate, action in condition_pairs:
1692 1
            if not predicate(self):
1693 1
                continue
1694 1
            success, succcess_type = action(self)
1695 1
            if success:
1696 1
                if succcess_type == 'redeploy':
1697 1
                    emit_event(
1698
                        self._controller,
1699
                        "redeployed_link_up",
1700
                        content=map_evc_event_content(self)
1701
                    )
1702 1
                return True
1703 1
        return False
1704
1705 1
    def handle_link_down(self):
1706
        """Handle circuit when link down.
1707
1708
        Returns:
1709
            bool: True if the re-deploy was successly otherwise False.
1710
1711
        """
1712 1
        success = False
1713 1
        if self.is_using_primary_path():
1714 1
            success = self.deploy_to_backup_path()
1715 1
        elif self.is_using_backup_path():
1716 1
            success = self.deploy_to_primary_path()
1717
1718 1
        if not success and self.dynamic_backup_path:
1719 1
            success = self.deploy_to_path()
1720
1721 1
        if success:
1722 1
            log.debug(f"{self} deployed after link down.")
1723
        else:
1724 1
            self.remove_current_flows(sync=False)
1725 1
            self.deactivate()
1726 1
            self.sync()
1727 1
            log.debug(f"Failed to re-deploy {self} after link down.")
1728
1729 1
        return success
1730
1731 1
    @staticmethod
1732 1
    def get_interface_from_switch(uni: UNI, switches: dict) -> Interface:
1733
        """Get interface from switch by uni"""
1734 1
        switch = switches[uni.interface.switch.dpid]
1735 1
        interface = switch.interfaces[uni.interface.port_number]
1736 1
        return interface
1737
1738 1
    def are_unis_active(self, switches: dict) -> bool:
1739
        """Determine whether this EVC should be active"""
1740 1
        interface_a = self.get_interface_from_switch(self.uni_a, switches)
1741 1
        interface_z = self.get_interface_from_switch(self.uni_z, switches)
1742 1
        active, _ = self.is_uni_interface_active(interface_a, interface_z)
1743 1
        return active
1744
1745 1
    @staticmethod
1746 1
    def is_uni_interface_active(
1747
        *interfaces: Interface
1748
    ) -> tuple[bool, dict]:
1749
        """Determine whether a UNI should be active"""
1750 1
        active = True
1751 1
        bad_interfaces = [
1752
            interface
1753
            for interface in interfaces
1754
            if interface.status != EntityStatus.UP
1755
        ]
1756 1
        if bad_interfaces:
1757 1
            active = False
1758 1
            interfaces = bad_interfaces
1759 1
        return active, {
1760
            interface.id: {
1761
                'status': interface.status.value,
1762
                'status_reason': interface.status_reason,
1763
            }
1764
            for interface in interfaces
1765
        }
1766
1767 1
    def handle_interface_link_up(self, interface: Interface):
1768
        """
1769
        Handler for interface link_up events
1770
        """
1771 1
        if self.is_active():
1772 1
            return
1773 1
        interfaces = (self.uni_a.interface, self.uni_z.interface)
1774 1
        if interface not in interfaces:
1775
            return
1776 1
        down_interfaces = [
1777
            interface
1778
            for interface in interfaces
1779
            if interface.status != EntityStatus.UP
1780
        ]
1781 1
        if down_interfaces:
1782
            return
1783 1
        interface_dicts = {
1784
            interface.id: {
1785
                'status': interface.status.value,
1786
                'status_reason': interface.status_reason,
1787
            }
1788
            for interface in interfaces
1789
        }
1790 1
        self.activate()
1791 1
        log.info(
1792
            f"Activating {self}. Interfaces: "
1793
            f"{interface_dicts}."
1794
        )
1795 1
        emit_event(self._controller, "uni_active_updated",
1796
                   content=map_evc_event_content(self))
1797 1
        self.sync()
1798
1799 1
    def handle_interface_link_down(self, interface):
1800
        """
1801
        Handler for interface link_down events
1802
        """
1803 1
        if not self.is_active():
1804 1
            return
1805 1
        interfaces = (self.uni_a.interface, self.uni_z.interface)
1806 1
        if interface not in interfaces:
1807
            return
1808 1
        down_interfaces = [
1809
            interface
1810
            for interface in interfaces
1811
            if interface.status != EntityStatus.UP
1812
        ]
1813 1
        if not down_interfaces:
1814
            return
1815 1
        interface_dicts = {
1816
            interface.id: {
1817
                'status': interface.status.value,
1818
                'status_reason': interface.status_reason,
1819
            }
1820
            for interface in down_interfaces
1821
        }
1822 1
        self.deactivate()
1823 1
        log.info(
1824
            f"Deactivating {self}. Interfaces: "
1825
            f"{interface_dicts}."
1826
        )
1827 1
        emit_event(self._controller, "uni_active_updated",
1828
                   content=map_evc_event_content(self))
1829 1
        self.sync()
1830
1831
1832 1
class EVC(LinkProtection):
1833
    """Class that represents a E-Line Virtual Connection."""
1834