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