| Total Complexity | 173 |
| Total Lines | 930 |
| Duplicated Lines | 1.18 % |
| Changes | 0 | ||
Duplicate code is one of the most pungent code smells. A rule that is often used is to re-structure code once it is duplicated in three or more places.
Common duplication problems, and corresponding solutions are:
Complex classes like kytos.core.interface 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.
| 1 | """Module with main classes related to Interfaces.""" |
||
| 2 | import json |
||
| 3 | import logging |
||
| 4 | import operator |
||
| 5 | from collections import OrderedDict |
||
| 6 | from copy import deepcopy |
||
| 7 | from enum import Enum |
||
| 8 | from functools import reduce |
||
| 9 | from threading import Lock |
||
| 10 | from typing import Union |
||
| 11 | |||
| 12 | from pyof.v0x01.common.phy_port import Port as PortNo01 |
||
| 13 | from pyof.v0x01.common.phy_port import PortFeatures as PortFeatures01 |
||
| 14 | from pyof.v0x04.common.port import PortFeatures as PortFeatures04 |
||
| 15 | from pyof.v0x04.common.port import PortNo as PortNo04 |
||
| 16 | |||
| 17 | from kytos.core.common import EntityStatus, GenericEntity |
||
| 18 | from kytos.core.events import KytosEvent |
||
| 19 | from kytos.core.exceptions import (KytosSetTagRangeError, |
||
| 20 | KytosTagsAreNotAvailable, |
||
| 21 | KytosTagsNotInTagRanges, |
||
| 22 | KytosTagtypeNotSupported) |
||
| 23 | from kytos.core.helpers import now |
||
| 24 | from kytos.core.id import InterfaceID |
||
| 25 | from kytos.core.tag_ranges import (find_index_add, find_index_remove, |
||
| 26 | get_special_tags, get_validated_tags, |
||
| 27 | range_addition, range_difference) |
||
| 28 | |||
| 29 | __all__ = ('Interface',) |
||
| 30 | |||
| 31 | LOG = logging.getLogger(__name__) |
||
| 32 | |||
| 33 | |||
| 34 | class TAGType(Enum): |
||
| 35 | """Class that represents a TAG Type.""" |
||
| 36 | |||
| 37 | VLAN = 'vlan' |
||
| 38 | VLAN_QINQ = 'vlan_qinq' |
||
| 39 | MPLS = 'mpls' |
||
| 40 | |||
| 41 | |||
| 42 | class TAG: |
||
| 43 | """Class that represents a TAG.""" |
||
| 44 | |||
| 45 | def __init__(self, tag_type: str, value: int): |
||
| 46 | self.tag_type = TAGType(tag_type).value |
||
| 47 | self.value = value |
||
| 48 | |||
| 49 | def __eq__(self, other): |
||
| 50 | if not other: |
||
| 51 | return False |
||
| 52 | return self.tag_type == other.tag_type and self.value == other.value |
||
| 53 | |||
| 54 | def as_dict(self): |
||
| 55 | """Return a dictionary representating a tag object.""" |
||
| 56 | return {'tag_type': self.tag_type, 'value': self.value} |
||
| 57 | |||
| 58 | @classmethod |
||
| 59 | def from_dict(cls, tag_dict): |
||
| 60 | """Return a TAG instance from python dictionary.""" |
||
| 61 | return cls(tag_dict.get('tag_type'), tag_dict.get('value')) |
||
| 62 | |||
| 63 | @classmethod |
||
| 64 | def from_json(cls, tag_json): |
||
| 65 | """Return a TAG instance from json.""" |
||
| 66 | return cls.from_dict(json.loads(tag_json)) |
||
| 67 | |||
| 68 | def as_json(self): |
||
| 69 | """Return a json representating a tag object.""" |
||
| 70 | return json.dumps(self.as_dict()) |
||
| 71 | |||
| 72 | def __repr__(self): |
||
| 73 | return f"TAG({self.tag_type!r}, {self.value!r})" |
||
| 74 | |||
| 75 | |||
| 76 | # pylint: disable=super-init-not-called |
||
| 77 | class TAGRange(TAG): |
||
| 78 | """Class that represents an User-to-Network Interface with |
||
| 79 | a tag value as a list.""" |
||
| 80 | |||
| 81 | def __init__( |
||
| 82 | self, |
||
| 83 | tag_type: str, |
||
| 84 | value: list[list[int]], |
||
| 85 | mask_list: list[Union[str, int]] = None |
||
| 86 | ): |
||
| 87 | self.tag_type = TAGType(tag_type).value |
||
| 88 | self.value = value |
||
| 89 | self.mask_list = mask_list or [] |
||
| 90 | |||
| 91 | def as_dict(self): |
||
| 92 | """Return a dictionary representating a tag range object.""" |
||
| 93 | return { |
||
| 94 | 'tag_type': self.tag_type, |
||
| 95 | 'value': self.value, |
||
| 96 | 'mask_list': self.mask_list |
||
| 97 | } |
||
| 98 | |||
| 99 | |||
| 100 | class Interface(GenericEntity): # pylint: disable=too-many-instance-attributes |
||
| 101 | """Interface Class used to abstract the network interfaces.""" |
||
| 102 | |||
| 103 | status_funcs = OrderedDict() |
||
| 104 | status_reason_funcs = OrderedDict() |
||
| 105 | |||
| 106 | # pylint: disable=too-many-arguments, too-many-public-methods |
||
| 107 | def __init__(self, name, port_number, switch, address=None, state=None, |
||
| 108 | features=None, speed=None, config=None): |
||
| 109 | """Assign the parameters to instance attributes. |
||
| 110 | |||
| 111 | Args: |
||
| 112 | name (string): name from this interface. |
||
| 113 | port_number (int): port number from this interface. |
||
| 114 | switch (:class:`~.core.switch.Switch`): Switch with this interface. |
||
| 115 | address (|hw_address|): Port address from this interface. |
||
| 116 | state (|port_stats|): Port Stat from interface. It will be |
||
| 117 | deprecated. |
||
| 118 | features (|port_features|): Port feature used to calculate link |
||
| 119 | utilization from this interface. It will be deprecated. |
||
| 120 | speed (int, float): Interface speed in bytes per second. Defaults |
||
| 121 | to what is informed by the switch. Return ``None`` if not set |
||
| 122 | and switch does not inform the speed. |
||
| 123 | config(|port_config|): Port config used to indicate interface |
||
| 124 | behavior. In general, the port config bits are set by the |
||
| 125 | controller and are not changed by the switch. Options |
||
| 126 | are: administratively down, ignore received packets, drop |
||
| 127 | forwarded packets, and/or do not send packet-in messages. |
||
| 128 | |||
| 129 | Attributes: |
||
| 130 | available_tags (dict[str, list[list[int]]]): Contains the available |
||
| 131 | tags integers in the current interface, the availability is |
||
| 132 | represented as a list of ranges. These ranges are |
||
| 133 | [inclusive, inclusive]. For example, [1, 5] represents |
||
| 134 | [1, 2, 3, 4, 5]. |
||
| 135 | tag_ranges (dict[str, list[list[int]]]): Contains restrictions for |
||
| 136 | available_tags. The list of ranges is the same type as in |
||
| 137 | available_tags. Setting a new tag_ranges will required for |
||
| 138 | available_tags to be resize. |
||
| 139 | |||
| 140 | """ |
||
| 141 | self.name = name |
||
| 142 | self.port_number = int(port_number) |
||
| 143 | self.switch = switch |
||
| 144 | self.address = address |
||
| 145 | self.state = state |
||
| 146 | self.features = features |
||
| 147 | self.config = config |
||
| 148 | self.nni = False |
||
| 149 | self.endpoints = [] |
||
| 150 | self.stats = None |
||
| 151 | self.link = None |
||
| 152 | self.lldp = True |
||
| 153 | self._id = InterfaceID(switch.id, port_number) |
||
| 154 | self._custom_speed = speed |
||
| 155 | self._tag_lock = Lock() |
||
| 156 | self.available_tags = {'vlan': self.default_tag_values['vlan']} |
||
| 157 | self.tag_ranges = {'vlan': self.default_tag_values['vlan']} |
||
| 158 | self.special_available_tags = {'vlan': self.default_special_tags} |
||
| 159 | self.special_tags = {'vlan': self.default_special_tags} |
||
| 160 | self.set_available_tags_tag_ranges( |
||
| 161 | self.available_tags, self.tag_ranges, |
||
| 162 | self.special_available_tags, self.special_tags |
||
| 163 | ) |
||
| 164 | super().__init__() |
||
| 165 | |||
| 166 | def __repr__(self): |
||
| 167 | return f"Interface('{self.name}', {self.port_number}, {self.switch!r})" |
||
| 168 | |||
| 169 | def __eq__(self, other): |
||
| 170 | """Compare Interface class with another instance.""" |
||
| 171 | if isinstance(other, str): |
||
| 172 | return self.address == other |
||
| 173 | if isinstance(other, Interface): |
||
| 174 | return self.port_number == other.port_number and \ |
||
| 175 | self.switch.dpid == other.switch.dpid |
||
| 176 | return False |
||
| 177 | |||
| 178 | @property |
||
| 179 | def id(self): # pylint: disable=invalid-name |
||
| 180 | """Return id from Interface instance. |
||
| 181 | |||
| 182 | Returns: |
||
| 183 | string: Interface id. |
||
| 184 | |||
| 185 | """ |
||
| 186 | return self._id |
||
| 187 | |||
| 188 | @property |
||
| 189 | def uni(self): |
||
| 190 | """Return if an interface is a user-to-network Interface.""" |
||
| 191 | return not self.nni |
||
| 192 | |||
| 193 | View Code Duplication | @property |
|
|
|
|||
| 194 | def status(self): |
||
| 195 | """Return the current status of the Entity.""" |
||
| 196 | state = super().status |
||
| 197 | if state == EntityStatus.DISABLED: |
||
| 198 | return state |
||
| 199 | |||
| 200 | for status_func in self.status_funcs.values(): |
||
| 201 | if status_func(self) == EntityStatus.DOWN: |
||
| 202 | return EntityStatus.DOWN |
||
| 203 | return state |
||
| 204 | |||
| 205 | @classmethod |
||
| 206 | def register_status_func(cls, name: str, func): |
||
| 207 | """Register status func given its name and a callable at setup time.""" |
||
| 208 | cls.status_funcs[name] = func |
||
| 209 | |||
| 210 | @classmethod |
||
| 211 | def register_status_reason_func(cls, name: str, func): |
||
| 212 | """Register status reason func given its name |
||
| 213 | and a callable at setup time.""" |
||
| 214 | cls.status_reason_funcs[name] = func |
||
| 215 | |||
| 216 | @property |
||
| 217 | def status_reason(self): |
||
| 218 | """Return the reason behind the current status of the entity.""" |
||
| 219 | return reduce( |
||
| 220 | operator.or_, |
||
| 221 | map( |
||
| 222 | lambda x: x(self), |
||
| 223 | self.status_reason_funcs.values() |
||
| 224 | ), |
||
| 225 | super().status_reason |
||
| 226 | ) |
||
| 227 | |||
| 228 | @property |
||
| 229 | def default_tag_values(self) -> dict[str, list[list[int]]]: |
||
| 230 | """Return a default list of ranges. Applicable to |
||
| 231 | available_tags and tag_ranges.""" |
||
| 232 | default_values = { |
||
| 233 | "vlan": [[1, 4094]], |
||
| 234 | "vlan_qinq": [[1, 4094]], |
||
| 235 | "mpls": [[1, 1048575]], |
||
| 236 | } |
||
| 237 | return default_values |
||
| 238 | |||
| 239 | @property |
||
| 240 | def default_special_tags(self) -> list[str]: |
||
| 241 | """Reurn a default list of special tags. Applicable to |
||
| 242 | special_available_tags and special_tags.""" |
||
| 243 | return ["untagged", "any"] |
||
| 244 | |||
| 245 | def all_tags_available(self) -> bool: |
||
| 246 | """Return True if all tags are avaiable (no tags used), |
||
| 247 | False otherwise""" |
||
| 248 | with self._tag_lock: |
||
| 249 | if self.available_tags != self.tag_ranges: |
||
| 250 | return False |
||
| 251 | for field, ranges in self.special_available_tags.items(): |
||
| 252 | if set(ranges) != set(self.special_tags[field]): |
||
| 253 | return False |
||
| 254 | return True |
||
| 255 | |||
| 256 | def set_tag_ranges(self, tag_ranges: list[list[int]], tag_type: str): |
||
| 257 | """Set new restriction, tag_ranges.""" |
||
| 258 | if tag_type != TAGType.VLAN.value: |
||
| 259 | msg = f"Tag type {tag_type} is not supported." |
||
| 260 | raise KytosTagtypeNotSupported(msg) |
||
| 261 | with self._tag_lock: |
||
| 262 | used_tags = range_difference( |
||
| 263 | self.tag_ranges[tag_type], self.available_tags[tag_type] |
||
| 264 | ) |
||
| 265 | # Verify new tag_ranges |
||
| 266 | missing = range_difference(used_tags, tag_ranges) |
||
| 267 | if missing: |
||
| 268 | msg = f"Missing tags in tag_range: {missing}" |
||
| 269 | raise KytosSetTagRangeError(msg) |
||
| 270 | |||
| 271 | # Resizing |
||
| 272 | new_available_tags = range_difference( |
||
| 273 | tag_ranges, used_tags |
||
| 274 | ) |
||
| 275 | self.available_tags[tag_type] = new_available_tags |
||
| 276 | self.tag_ranges[tag_type] = tag_ranges |
||
| 277 | |||
| 278 | def remove_tag_ranges(self, tag_type: str): |
||
| 279 | """Set tag_ranges[tag_type] to default value""" |
||
| 280 | if tag_type != TAGType.VLAN.value: |
||
| 281 | msg = f"Tag type {tag_type} is not supported." |
||
| 282 | raise KytosTagtypeNotSupported(msg) |
||
| 283 | with self._tag_lock: |
||
| 284 | used_tags = range_difference( |
||
| 285 | self.tag_ranges[tag_type], self.available_tags[tag_type] |
||
| 286 | ) |
||
| 287 | self.available_tags[tag_type] = range_difference( |
||
| 288 | self.default_tag_values[tag_type], used_tags |
||
| 289 | ) |
||
| 290 | self.tag_ranges[tag_type] = self.default_tag_values[tag_type] |
||
| 291 | |||
| 292 | def set_special_tags( |
||
| 293 | self, |
||
| 294 | tag_type: str, |
||
| 295 | special_tags: list[str], |
||
| 296 | ): |
||
| 297 | """Set new restriction, special_tags""" |
||
| 298 | # Verify values in special_tags |
||
| 299 | tag_range = get_special_tags( |
||
| 300 | special_tags, self.default_special_tags |
||
| 301 | ) |
||
| 302 | |||
| 303 | if tag_type != TAGType.VLAN.value: |
||
| 304 | msg = f"Tag type {tag_type} is not supported." |
||
| 305 | raise KytosTagtypeNotSupported(msg) |
||
| 306 | old_special_set = set(self.special_tags[tag_type]) |
||
| 307 | |||
| 308 | for tag in self.special_available_tags[tag_type]: |
||
| 309 | old_special_set.remove(tag) # Get special used tags |
||
| 310 | used_special = old_special_set.copy() |
||
| 311 | |||
| 312 | for tag in tag_range: |
||
| 313 | used_special.discard(tag) |
||
| 314 | |||
| 315 | # Missing used special used tags |
||
| 316 | if used_special: |
||
| 317 | msg = f"Missing tags in tag_range: {used_special}" |
||
| 318 | raise KytosSetTagRangeError(msg) |
||
| 319 | |||
| 320 | new_special_available = set(tag_range) |
||
| 321 | self.special_available_tags[tag_type] = list( |
||
| 322 | new_special_available - old_special_set |
||
| 323 | ) |
||
| 324 | self.special_tags[tag_type] = tag_range |
||
| 325 | |||
| 326 | def _remove_tags(self, tags: list[int], tag_type: str = 'vlan') -> bool: |
||
| 327 | """Remove tags by resizing available_tags |
||
| 328 | Returns False if nothing was remove, True otherwise""" |
||
| 329 | available = self.available_tags[tag_type] |
||
| 330 | if not available: |
||
| 331 | return False |
||
| 332 | index = find_index_remove(available, tags) |
||
| 333 | if index is None: |
||
| 334 | return False |
||
| 335 | # Resizing |
||
| 336 | if tags[0] == available[index][0]: |
||
| 337 | if tags[1] == available[index][1]: |
||
| 338 | available.pop(index) |
||
| 339 | else: |
||
| 340 | available[index][0] = tags[1] + 1 |
||
| 341 | elif tags[1] == available[index][1]: |
||
| 342 | available[index][1] = tags[0] - 1 |
||
| 343 | else: |
||
| 344 | available[index: index+1] = [ |
||
| 345 | [available[index][0], tags[0]-1], |
||
| 346 | [tags[1]+1, available[index][1]] |
||
| 347 | ] |
||
| 348 | return True |
||
| 349 | |||
| 350 | def use_tags( |
||
| 351 | self, |
||
| 352 | controller, |
||
| 353 | tags: Union[str, int, list[int], list[list[int]]], |
||
| 354 | tag_type: str = 'vlan', |
||
| 355 | use_lock: bool = True, |
||
| 356 | check_order: bool = True, |
||
| 357 | ): |
||
| 358 | """Remove a specific tag from available_tags if it is there. |
||
| 359 | Exception raised in case the tags were not able to be removed. |
||
| 360 | |||
| 361 | Args: |
||
| 362 | controller: Kytos controller |
||
| 363 | tags: value to be removed, multiple types for compatibility: |
||
| 364 | (str): Special vlan, "untagged" or "vlan" |
||
| 365 | (int): Single tag |
||
| 366 | (list[int]): Single range of tags |
||
| 367 | (list[list[int]]): List of ranges of tags |
||
| 368 | tag_type: TAG type value |
||
| 369 | use_lock: Boolean to whether use a lock or not |
||
| 370 | check_order: Boolean to whether validate tags(list). Check order, |
||
| 371 | type and length. Set to false when invocated internally. |
||
| 372 | |||
| 373 | Exceptions: |
||
| 374 | KytosTagsAreNotAvailable from _use_tags() |
||
| 375 | """ |
||
| 376 | if isinstance(tags, int): |
||
| 377 | tags = [tags] * 2 |
||
| 378 | elif check_order and not isinstance(tags, str): |
||
| 379 | tags = get_validated_tags(tags) |
||
| 380 | if use_lock: |
||
| 381 | with self._tag_lock: |
||
| 382 | try: |
||
| 383 | self._use_tags(tags, tag_type) |
||
| 384 | except KeyError as err: |
||
| 385 | msg = f"Tag type {err.args[0]} is not supported" |
||
| 386 | raise KytosTagtypeNotSupported(msg) |
||
| 387 | else: |
||
| 388 | try: |
||
| 389 | self._use_tags(tags, tag_type) |
||
| 390 | except KeyError as err: |
||
| 391 | msg = f"Tag type {err.args[0]} is not supported" |
||
| 392 | raise KytosTagtypeNotSupported(msg) |
||
| 393 | |||
| 394 | self._notify_interface_tags(controller) |
||
| 395 | |||
| 396 | def _use_tags( |
||
| 397 | self, |
||
| 398 | tags: Union[str, list[int], list[list[int]]], |
||
| 399 | tag_type: str |
||
| 400 | ): |
||
| 401 | """Manage available_tags deletion changes.""" |
||
| 402 | if isinstance(tags[0], list): |
||
| 403 | available_copy = deepcopy(self.available_tags[tag_type]) |
||
| 404 | for tag_range in tags: |
||
| 405 | result = self._remove_tags(tag_range, tag_type) |
||
| 406 | if result is False: |
||
| 407 | self.available_tags[tag_type] = available_copy |
||
| 408 | conflict = range_difference(tags, available_copy) |
||
| 409 | raise KytosTagsAreNotAvailable(conflict, self._id) |
||
| 410 | elif isinstance(tags, list): |
||
| 411 | result = self._remove_tags(tags, tag_type) |
||
| 412 | if result is False: |
||
| 413 | raise KytosTagsAreNotAvailable([tags], self._id) |
||
| 414 | else: |
||
| 415 | try: |
||
| 416 | self.special_available_tags[tag_type].remove(tags) |
||
| 417 | except ValueError: |
||
| 418 | raise KytosTagsAreNotAvailable(tags, self.id) |
||
| 419 | |||
| 420 | # pylint: disable=too-many-branches |
||
| 421 | def _add_tags(self, tags: list[int], tag_type: str = 'vlan') -> bool: |
||
| 422 | """Add tags, return True if they were added. |
||
| 423 | Returns False when nothing was added, True otherwise |
||
| 424 | Ensuring that ranges are not unnecessarily divided |
||
| 425 | available_tag e.g [[7, 10], [20, 30], [78, 92], [100, 109], [189, 200]] |
||
| 426 | tags examples are in each if statement. |
||
| 427 | """ |
||
| 428 | if not tags[0] or not tags[1]: |
||
| 429 | return False |
||
| 430 | |||
| 431 | # Check if tags is within self.tag_ranges |
||
| 432 | tag_ranges = self.tag_ranges[tag_type] |
||
| 433 | if find_index_remove(tag_ranges, tags) is None: |
||
| 434 | raise KytosTagsNotInTagRanges([tags], self._id) |
||
| 435 | |||
| 436 | available = self.available_tags[tag_type] |
||
| 437 | if not available: |
||
| 438 | self.available_tags[tag_type] = [tags] |
||
| 439 | return True |
||
| 440 | |||
| 441 | index = find_index_add(available, tags) |
||
| 442 | if index is None: |
||
| 443 | return False |
||
| 444 | if index == 0: |
||
| 445 | # [1, 6] |
||
| 446 | if tags[1] == available[index][0] - 1: |
||
| 447 | available[index][0] = tags[0] |
||
| 448 | # [1, 2] |
||
| 449 | else: |
||
| 450 | available.insert(0, tags) |
||
| 451 | elif index == len(available): |
||
| 452 | # [201, 300] |
||
| 453 | if available[index-1][1] + 1 == tags[0]: |
||
| 454 | available[index-1][1] = tags[1] |
||
| 455 | # [250, 300] |
||
| 456 | else: |
||
| 457 | available.append(tags) |
||
| 458 | else: |
||
| 459 | # [11, 19] |
||
| 460 | if (available[index-1][1] + 1 == tags[0] and |
||
| 461 | available[index][0] - 1 == tags[1]): |
||
| 462 | available[index-1: index+1] = [ |
||
| 463 | [available[index-1][0], available[index][1]] |
||
| 464 | ] |
||
| 465 | # [11, 15] |
||
| 466 | elif available[index-1][1] + 1 == tags[0]: |
||
| 467 | available[index-1][1] = tags[1] |
||
| 468 | # [15, 19] |
||
| 469 | elif available[index][0] - 1 == tags[1]: |
||
| 470 | available[index][0] = tags[0] |
||
| 471 | # [15, 15] |
||
| 472 | else: |
||
| 473 | available.insert(index, tags) |
||
| 474 | return True |
||
| 475 | |||
| 476 | def make_tags_available( |
||
| 477 | self, |
||
| 478 | controller, |
||
| 479 | tags: Union[str, int, list[int], list[list[int]]], |
||
| 480 | tag_type: str = 'vlan', |
||
| 481 | use_lock: bool = True, |
||
| 482 | check_order: bool = True, |
||
| 483 | ) -> list[list[int]]: |
||
| 484 | """Add a tags in available_tags. |
||
| 485 | |||
| 486 | Args: |
||
| 487 | controller: Kytos controller |
||
| 488 | tags: value to be added, multiple types for compatibility: |
||
| 489 | (str): Special vlan, "untagged" or "vlan" |
||
| 490 | (int): Single tag |
||
| 491 | (list[int]): Single range of tags |
||
| 492 | (list[list[int]]): List of ranges of tags |
||
| 493 | tag_type: TAG type value |
||
| 494 | use_lock: Boolean to whether use a lock or not |
||
| 495 | check_order: Boolean to whether validate tags(list). Check order, |
||
| 496 | type and length. Set to false when invocated internally. |
||
| 497 | |||
| 498 | Return: |
||
| 499 | conflict: Return any values that were not added. |
||
| 500 | |||
| 501 | Exeptions: |
||
| 502 | KytosTagsNotInTagRanges from _make_tags_available() |
||
| 503 | """ |
||
| 504 | if isinstance(tags, int): |
||
| 505 | tags = [tags] * 2 |
||
| 506 | elif check_order and not isinstance(tags, str): |
||
| 507 | tags = get_validated_tags(tags) |
||
| 508 | if isinstance(tags[0], int) and tags[0] != tags[1]: |
||
| 509 | tags = [tags] |
||
| 510 | if use_lock: |
||
| 511 | with self._tag_lock: |
||
| 512 | try: |
||
| 513 | conflict = self._make_tags_available(tags, tag_type) |
||
| 514 | except KeyError as err: |
||
| 515 | msg = f"Tag type {err.args[0]} is not supported" |
||
| 516 | raise KytosTagtypeNotSupported(msg) |
||
| 517 | else: |
||
| 518 | try: |
||
| 519 | conflict = self._make_tags_available(tags, tag_type) |
||
| 520 | except KeyError as err: |
||
| 521 | msg = f"Tag type {err.args[0]} is not supported" |
||
| 522 | raise KytosTagtypeNotSupported(msg) |
||
| 523 | self._notify_interface_tags(controller) |
||
| 524 | return conflict |
||
| 525 | |||
| 526 | def _make_tags_available( |
||
| 527 | self, |
||
| 528 | tags: Union[str, list[int], list[list[int]]], |
||
| 529 | tag_type: str, |
||
| 530 | ) -> Union[str, list[list[int]], None]: |
||
| 531 | """Manage available_tags adittion changes |
||
| 532 | |||
| 533 | Exceptions: |
||
| 534 | KytosTagsNotInTagRanges from _add_tags() |
||
| 535 | """ |
||
| 536 | if isinstance(tags[0], list): |
||
| 537 | diff = range_difference(tags, self.tag_ranges[tag_type]) |
||
| 538 | if diff: |
||
| 539 | raise KytosTagsNotInTagRanges(diff, self._id) |
||
| 540 | available_tags = self.available_tags[tag_type] |
||
| 541 | new_tags, conflict = range_addition(tags, available_tags) |
||
| 542 | self.available_tags[tag_type] = new_tags |
||
| 543 | return conflict |
||
| 544 | if isinstance(tags, list): |
||
| 545 | result = self._add_tags(tags, tag_type) |
||
| 546 | if result is False: |
||
| 547 | return [tags] |
||
| 548 | if isinstance(tags, str): |
||
| 549 | if tags not in self.special_tags[tag_type]: |
||
| 550 | raise KytosTagsNotInTagRanges(tags, self._id) |
||
| 551 | if tags not in self.special_available_tags[tag_type]: |
||
| 552 | self.special_available_tags[tag_type].append(tags) |
||
| 553 | return None |
||
| 554 | return tags |
||
| 555 | return None |
||
| 556 | |||
| 557 | def set_available_tags_tag_ranges( |
||
| 558 | self, |
||
| 559 | available_tag: dict[str, list[list[int]]], |
||
| 560 | tag_ranges: dict[str, list[list[int]]], |
||
| 561 | special_available_tags: dict[str, list[str]], |
||
| 562 | special_tags: dict[str, list[str]] |
||
| 563 | ): |
||
| 564 | """Set a range of VLAN tags to be used by this Interface. |
||
| 565 | |||
| 566 | Args: |
||
| 567 | available_tag: Available tags from each tag type |
||
| 568 | tag_ranges: Restriction for each type of available tag |
||
| 569 | """ |
||
| 570 | with self._tag_lock: |
||
| 571 | self.available_tags = available_tag |
||
| 572 | self.tag_ranges = tag_ranges |
||
| 573 | self.special_available_tags = special_available_tags |
||
| 574 | self.special_tags = special_tags |
||
| 575 | |||
| 576 | def is_tag_available(self, tag: int, tag_type: str = 'vlan'): |
||
| 577 | """Check if a tag is available.""" |
||
| 578 | with self._tag_lock: |
||
| 579 | if find_index_remove( |
||
| 580 | self.available_tags[tag_type], [tag, tag] |
||
| 581 | ) is not None: |
||
| 582 | return True |
||
| 583 | return False |
||
| 584 | |||
| 585 | def get_endpoint(self, endpoint): |
||
| 586 | """Return a tuple with existent endpoint, None otherwise. |
||
| 587 | |||
| 588 | Args: |
||
| 589 | endpoint(|hw_address|, :class:`.Interface`): endpoint instance. |
||
| 590 | |||
| 591 | Returns: |
||
| 592 | tuple: A tuple with endpoint and time of last update. |
||
| 593 | |||
| 594 | """ |
||
| 595 | for item in self.endpoints: |
||
| 596 | if endpoint == item[0]: |
||
| 597 | return item |
||
| 598 | return None |
||
| 599 | |||
| 600 | def add_endpoint(self, endpoint): |
||
| 601 | """Create a new endpoint to Interface instance. |
||
| 602 | |||
| 603 | Args: |
||
| 604 | endpoint(|hw_address|, :class:`.Interface`): A target endpoint. |
||
| 605 | """ |
||
| 606 | exists = self.get_endpoint(endpoint) |
||
| 607 | if not exists: |
||
| 608 | self.endpoints.append((endpoint, now())) |
||
| 609 | |||
| 610 | def delete_endpoint(self, endpoint): |
||
| 611 | """Delete a existent endpoint in Interface instance. |
||
| 612 | |||
| 613 | Args: |
||
| 614 | endpoint (|hw_address|, :class:`.Interface`): A target endpoint. |
||
| 615 | """ |
||
| 616 | exists = self.get_endpoint(endpoint) |
||
| 617 | if exists: |
||
| 618 | self.endpoints.remove(exists) |
||
| 619 | |||
| 620 | def update_endpoint(self, endpoint): |
||
| 621 | """Update or create new endpoint to Interface instance. |
||
| 622 | |||
| 623 | Args: |
||
| 624 | endpoint(|hw_address|, :class:`.Interface`): A target endpoint. |
||
| 625 | """ |
||
| 626 | exists = self.get_endpoint(endpoint) |
||
| 627 | if exists: |
||
| 628 | self.delete_endpoint(endpoint) |
||
| 629 | self.add_endpoint(endpoint) |
||
| 630 | |||
| 631 | def update_link(self, link): |
||
| 632 | """Update link for this interface in a consistent way. |
||
| 633 | |||
| 634 | Verify of the other endpoint of the link has the same Link information |
||
| 635 | attached to it, and change it if necessary. |
||
| 636 | |||
| 637 | Warning: This method can potentially change information of other |
||
| 638 | Interface instances. Use it with caution. |
||
| 639 | """ |
||
| 640 | if self not in (link.endpoint_a, link.endpoint_b): |
||
| 641 | return False |
||
| 642 | |||
| 643 | if self.link is None or self.link != link: |
||
| 644 | self.link = link |
||
| 645 | |||
| 646 | if link.endpoint_a == self: |
||
| 647 | endpoint = link.endpoint_b |
||
| 648 | else: |
||
| 649 | endpoint = link.endpoint_a |
||
| 650 | |||
| 651 | if endpoint.link is None or endpoint.link != link: |
||
| 652 | endpoint.link = link |
||
| 653 | |||
| 654 | return True |
||
| 655 | |||
| 656 | @property |
||
| 657 | def speed(self): |
||
| 658 | """Return the link speed in bytes per second, None otherwise. |
||
| 659 | |||
| 660 | If the switch was disconnected, we have :attr:`features` and speed is |
||
| 661 | still returned for common values between v0x01 and v0x04. For specific |
||
| 662 | v0x04 values (40 Gbps, 100 Gbps and 1 Tbps), the connection must be |
||
| 663 | active so we can make sure the protocol version is v0x04. |
||
| 664 | |||
| 665 | Returns: |
||
| 666 | int, None: Link speed in bytes per second or ``None``. |
||
| 667 | |||
| 668 | """ |
||
| 669 | speed = self.get_of_features_speed() |
||
| 670 | |||
| 671 | if speed is not None: |
||
| 672 | return speed |
||
| 673 | |||
| 674 | if self._custom_speed is not None: |
||
| 675 | return self._custom_speed |
||
| 676 | |||
| 677 | if self._is_v0x04() and self.port_number == PortNo04.OFPP_LOCAL: |
||
| 678 | return 0 |
||
| 679 | |||
| 680 | if not self._is_v0x04() and self.port_number == PortNo01.OFPP_LOCAL: |
||
| 681 | return 0 |
||
| 682 | |||
| 683 | # Warn unknown speed |
||
| 684 | # Use shorter switch ID with its beginning and end |
||
| 685 | if isinstance(self.switch.id, str) and len(self.switch.id) > 20: |
||
| 686 | switch_id = self.switch.id[:3] + '...' + self.switch.id[-3:] |
||
| 687 | else: |
||
| 688 | switch_id = self.switch.id |
||
| 689 | LOG.warning("Couldn't get port %s speed, sw %s, feats %s", |
||
| 690 | self.port_number, switch_id, self.features) |
||
| 691 | |||
| 692 | return None |
||
| 693 | |||
| 694 | def set_custom_speed(self, bytes_per_second): |
||
| 695 | """Set a speed that overrides switch OpenFlow information. |
||
| 696 | |||
| 697 | If ``None`` is given, :attr:`speed` becomes the one given by the |
||
| 698 | switch. |
||
| 699 | """ |
||
| 700 | self._custom_speed = bytes_per_second |
||
| 701 | |||
| 702 | def get_custom_speed(self): |
||
| 703 | """Return custom speed or ``None`` if not set.""" |
||
| 704 | return self._custom_speed |
||
| 705 | |||
| 706 | def get_of_features_speed(self): |
||
| 707 | """Return the link speed in bytes per second, None otherwise. |
||
| 708 | |||
| 709 | If the switch was disconnected, we have :attr:`features` and speed is |
||
| 710 | still returned for common values between v0x01 and v0x04. For specific |
||
| 711 | v0x04 values (40 Gbps, 100 Gbps and 1 Tbps), the connection must be |
||
| 712 | active so we can make sure the protocol version is v0x04. |
||
| 713 | |||
| 714 | Returns: |
||
| 715 | int, None: Link speed in bytes per second or ``None``. |
||
| 716 | |||
| 717 | """ |
||
| 718 | speed = self._get_v0x01_v0x04_speed() |
||
| 719 | # Don't use switch.is_connected() because we can have the protocol |
||
| 720 | if speed is None and self._is_v0x04(): |
||
| 721 | speed = self._get_v0x04_speed() |
||
| 722 | return speed |
||
| 723 | |||
| 724 | def _is_v0x04(self): |
||
| 725 | """Whether the switch is connected using OpenFlow 1.3.""" |
||
| 726 | return self.switch.is_connected() and \ |
||
| 727 | self.switch.connection.protocol.version == 0x04 |
||
| 728 | |||
| 729 | def _get_v0x01_v0x04_speed(self): |
||
| 730 | """Check against all values of v0x01. They're part of v0x04.""" |
||
| 731 | fts = self.features |
||
| 732 | pfts = PortFeatures01 |
||
| 733 | if fts and fts & pfts.OFPPF_10GB_FD: |
||
| 734 | return 10 * 10**9 / 8 |
||
| 735 | if fts and fts & (pfts.OFPPF_1GB_HD | pfts.OFPPF_1GB_FD): |
||
| 736 | return 10**9 / 8 |
||
| 737 | if fts and fts & (pfts.OFPPF_100MB_HD | pfts.OFPPF_100MB_FD): |
||
| 738 | return 100 * 10**6 / 8 |
||
| 739 | if fts and fts & (pfts.OFPPF_10MB_HD | pfts.OFPPF_10MB_FD): |
||
| 740 | return 10 * 10**6 / 8 |
||
| 741 | return None |
||
| 742 | |||
| 743 | def _get_v0x04_speed(self): |
||
| 744 | """Check against higher enums of v0x04. |
||
| 745 | |||
| 746 | Must be called after :meth:`get_v0x01_speed` returns ``None``. |
||
| 747 | """ |
||
| 748 | fts = self.features |
||
| 749 | pfts = PortFeatures04 |
||
| 750 | if fts and fts & pfts.OFPPF_1TB_FD: |
||
| 751 | return 10**12 / 8 |
||
| 752 | if fts and fts & pfts.OFPPF_100GB_FD: |
||
| 753 | return 100 * 10**9 / 8 |
||
| 754 | if fts and fts & pfts.OFPPF_40GB_FD: |
||
| 755 | return 40 * 10**9 / 8 |
||
| 756 | return None |
||
| 757 | |||
| 758 | def get_hr_speed(self): |
||
| 759 | """Return Human-Readable string for link speed. |
||
| 760 | |||
| 761 | Returns: |
||
| 762 | string: String with link speed. e.g: '350 Gbps' or '350 Mbps'. |
||
| 763 | |||
| 764 | """ |
||
| 765 | speed = self.speed |
||
| 766 | if speed is None: |
||
| 767 | return '' |
||
| 768 | speed *= 8 |
||
| 769 | if speed == 10**12: |
||
| 770 | return '1 Tbps' |
||
| 771 | if speed >= 10**9: |
||
| 772 | return f"{round(speed / 10**9)} Gbps" |
||
| 773 | return f"{round(speed / 10**6)} Mbps" |
||
| 774 | |||
| 775 | def as_dict(self): |
||
| 776 | """Return a dictionary with Interface attributes. |
||
| 777 | |||
| 778 | Speed is in bytes/sec. Example of output (100 Gbps): |
||
| 779 | |||
| 780 | .. code-block:: python3 |
||
| 781 | |||
| 782 | {'id': '00:00:00:00:00:00:00:01:2', |
||
| 783 | 'name': 'eth01', |
||
| 784 | 'port_number': 2, |
||
| 785 | 'mac': '00:7e:04:3b:c2:a6', |
||
| 786 | 'switch': '00:00:00:00:00:00:00:01', |
||
| 787 | 'type': 'interface', |
||
| 788 | 'nni': False, |
||
| 789 | 'uni': True, |
||
| 790 | 'speed': 12500000000, |
||
| 791 | 'metadata': {}, |
||
| 792 | 'lldp': True, |
||
| 793 | 'active': True, |
||
| 794 | 'enabled': False, |
||
| 795 | 'status': 'DISABLED', |
||
| 796 | 'link': "" |
||
| 797 | } |
||
| 798 | |||
| 799 | Returns: |
||
| 800 | dict: Dictionary filled with interface attributes. |
||
| 801 | |||
| 802 | """ |
||
| 803 | iface_dict = { |
||
| 804 | 'id': self.id, |
||
| 805 | 'name': self.name, |
||
| 806 | 'port_number': self.port_number, |
||
| 807 | 'mac': self.address, |
||
| 808 | 'switch': self.switch.dpid, |
||
| 809 | 'type': 'interface', |
||
| 810 | 'nni': self.nni, |
||
| 811 | 'uni': self.uni, |
||
| 812 | 'speed': self.speed, |
||
| 813 | 'metadata': self.metadata, |
||
| 814 | 'lldp': self.lldp, |
||
| 815 | 'active': self.is_active(), |
||
| 816 | 'enabled': self.is_enabled(), |
||
| 817 | 'status': self.status.value, |
||
| 818 | 'status_reason': sorted(self.status_reason), |
||
| 819 | 'link': self.link.id if self.link else "", |
||
| 820 | } |
||
| 821 | if self.stats: |
||
| 822 | iface_dict['stats'] = self.stats.as_dict() |
||
| 823 | return iface_dict |
||
| 824 | |||
| 825 | @classmethod |
||
| 826 | def from_dict(cls, interface_dict): |
||
| 827 | """Return a Interface instance from python dictionary.""" |
||
| 828 | return cls(interface_dict.get('name'), |
||
| 829 | interface_dict.get('port_number'), |
||
| 830 | interface_dict.get('switch'), |
||
| 831 | interface_dict.get('address'), |
||
| 832 | interface_dict.get('state'), |
||
| 833 | interface_dict.get('features'), |
||
| 834 | interface_dict.get('speed')) |
||
| 835 | |||
| 836 | def as_json(self): |
||
| 837 | """Return a json with Interfaces attributes. |
||
| 838 | |||
| 839 | Example of output: |
||
| 840 | |||
| 841 | .. code-block:: json |
||
| 842 | |||
| 843 | {"mac": "00:7e:04:3b:c2:a6", |
||
| 844 | "switch": "00:00:00:00:00:00:00:01", |
||
| 845 | "type": "interface", |
||
| 846 | "name": "eth01", |
||
| 847 | "id": "00:00:00:00:00:00:00:01:2", |
||
| 848 | "port_number": 2, |
||
| 849 | "speed": "350 Mbps"} |
||
| 850 | |||
| 851 | Returns: |
||
| 852 | string: Json filled with interface attributes. |
||
| 853 | |||
| 854 | """ |
||
| 855 | return json.dumps(self.as_dict()) |
||
| 856 | |||
| 857 | def _notify_interface_tags(self, controller): |
||
| 858 | """Notify link available tags""" |
||
| 859 | name = "kytos/core.interface_tags" |
||
| 860 | content = {"interface": self} |
||
| 861 | event = KytosEvent(name=name, content=content) |
||
| 862 | controller.buffers.app.put(event) |
||
| 863 | |||
| 864 | |||
| 865 | class UNI: |
||
| 866 | """Class that represents an User-to-Network Interface.""" |
||
| 867 | |||
| 868 | def __init__( |
||
| 869 | self, |
||
| 870 | interface: Interface, |
||
| 871 | user_tag: Union[None, TAG, TAGRange] |
||
| 872 | ): |
||
| 873 | self.user_tag = user_tag |
||
| 874 | self.interface = interface |
||
| 875 | |||
| 876 | def __eq__(self, other): |
||
| 877 | """Override the default implementation.""" |
||
| 878 | return (self.user_tag == other.user_tag and |
||
| 879 | self.interface == other.interface) |
||
| 880 | |||
| 881 | def _is_reserved_valid_tag(self) -> bool: |
||
| 882 | """Check if TAG string is possible""" |
||
| 883 | reserved_tag = {"any", "untagged"} |
||
| 884 | if self.user_tag.value in reserved_tag: |
||
| 885 | return True |
||
| 886 | return False |
||
| 887 | |||
| 888 | def is_valid(self): |
||
| 889 | """Check if TAG is possible for this interface TAG pool.""" |
||
| 890 | if self.user_tag: |
||
| 891 | tag = self.user_tag.value |
||
| 892 | if isinstance(tag, str): |
||
| 893 | return self._is_reserved_valid_tag() |
||
| 894 | if isinstance(tag, int): |
||
| 895 | return self.interface.is_tag_available(tag) |
||
| 896 | return True |
||
| 897 | |||
| 898 | def as_dict(self): |
||
| 899 | """Return a dict representating a UNI object.""" |
||
| 900 | return { |
||
| 901 | 'interface_id': self.interface.id, |
||
| 902 | 'tag': self.user_tag.as_dict() if self.user_tag else None |
||
| 903 | } |
||
| 904 | |||
| 905 | @classmethod |
||
| 906 | def from_dict(cls, uni): |
||
| 907 | """Return a Uni instance from python dictionary.""" |
||
| 908 | return cls(uni.get('interface'), |
||
| 909 | uni.get('user_tag')) |
||
| 910 | |||
| 911 | def as_json(self): |
||
| 912 | """Return a json representating a UNI object.""" |
||
| 913 | return json.dumps(self.as_dict()) |
||
| 914 | |||
| 915 | |||
| 916 | class NNI: |
||
| 917 | """Class that represents an Network-to-Network Interface.""" |
||
| 918 | |||
| 919 | def __init__(self, interface): |
||
| 920 | self.interface = interface |
||
| 921 | |||
| 922 | |||
| 923 | class VNNI(NNI): |
||
| 924 | """Class that represents an Virtual Network-to-Network Interface.""" |
||
| 925 | |||
| 926 | def __init__(self, service_tag, *args, **kwargs): |
||
| 927 | self.service_tag = service_tag |
||
| 928 | |||
| 929 | super().__init__(*args, **kwargs) |
||
| 930 |