| Total Complexity | 151 |
| Total Lines | 849 |
| Duplicated Lines | 1.3 % |
| 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_validated_tags, range_addition, |
||
| 27 | 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.set_available_tags_tag_ranges( |
||
| 159 | self.available_tags, self.tag_ranges |
||
| 160 | ) |
||
| 161 | super().__init__() |
||
| 162 | |||
| 163 | def __repr__(self): |
||
| 164 | return f"Interface('{self.name}', {self.port_number}, {self.switch!r})" |
||
| 165 | |||
| 166 | def __eq__(self, other): |
||
| 167 | """Compare Interface class with another instance.""" |
||
| 168 | if isinstance(other, str): |
||
| 169 | return self.address == other |
||
| 170 | if isinstance(other, Interface): |
||
| 171 | return self.port_number == other.port_number and \ |
||
| 172 | self.switch.dpid == other.switch.dpid |
||
| 173 | return False |
||
| 174 | |||
| 175 | @property |
||
| 176 | def id(self): # pylint: disable=invalid-name |
||
| 177 | """Return id from Interface instance. |
||
| 178 | |||
| 179 | Returns: |
||
| 180 | string: Interface id. |
||
| 181 | |||
| 182 | """ |
||
| 183 | return self._id |
||
| 184 | |||
| 185 | @property |
||
| 186 | def uni(self): |
||
| 187 | """Return if an interface is a user-to-network Interface.""" |
||
| 188 | return not self.nni |
||
| 189 | |||
| 190 | View Code Duplication | @property |
|
|
|
|||
| 191 | def status(self): |
||
| 192 | """Return the current status of the Entity.""" |
||
| 193 | state = super().status |
||
| 194 | if state == EntityStatus.DISABLED: |
||
| 195 | return state |
||
| 196 | |||
| 197 | for status_func in self.status_funcs.values(): |
||
| 198 | if status_func(self) == EntityStatus.DOWN: |
||
| 199 | return EntityStatus.DOWN |
||
| 200 | return state |
||
| 201 | |||
| 202 | @classmethod |
||
| 203 | def register_status_func(cls, name: str, func): |
||
| 204 | """Register status func given its name and a callable at setup time.""" |
||
| 205 | cls.status_funcs[name] = func |
||
| 206 | |||
| 207 | @classmethod |
||
| 208 | def register_status_reason_func(cls, name: str, func): |
||
| 209 | """Register status reason func given its name |
||
| 210 | and a callable at setup time.""" |
||
| 211 | cls.status_reason_funcs[name] = func |
||
| 212 | |||
| 213 | @property |
||
| 214 | def status_reason(self): |
||
| 215 | """Return the reason behind the current status of the entity.""" |
||
| 216 | return reduce( |
||
| 217 | operator.or_, |
||
| 218 | map( |
||
| 219 | lambda x: x(self), |
||
| 220 | self.status_reason_funcs.values() |
||
| 221 | ), |
||
| 222 | super().status_reason |
||
| 223 | ) |
||
| 224 | |||
| 225 | @property |
||
| 226 | def default_tag_values(self) -> dict[str, list[list[int]]]: |
||
| 227 | """Return a default list of ranges. Applicable to |
||
| 228 | available_tags and tag_ranges.""" |
||
| 229 | default_values = { |
||
| 230 | "vlan": [[1, 4095]], |
||
| 231 | "vlan_qinq": [[1, 4095]], |
||
| 232 | "mpls": [[1, 1048575]], |
||
| 233 | } |
||
| 234 | return default_values |
||
| 235 | |||
| 236 | def set_tag_ranges(self, tag_ranges: list[list[int]], tag_type: str): |
||
| 237 | """Set new restriction, tag_ranges.""" |
||
| 238 | if tag_type != TAGType.VLAN.value: |
||
| 239 | msg = f"Tag type {tag_type} is not supported." |
||
| 240 | raise KytosTagtypeNotSupported(msg) |
||
| 241 | with self._tag_lock: |
||
| 242 | used_tags = range_difference( |
||
| 243 | self.tag_ranges[tag_type], self.available_tags[tag_type] |
||
| 244 | ) |
||
| 245 | # Verify new tag_ranges |
||
| 246 | missing = range_difference(used_tags, tag_ranges) |
||
| 247 | if missing: |
||
| 248 | msg = f"Missing tags in tag_range: {missing}" |
||
| 249 | raise KytosSetTagRangeError(msg) |
||
| 250 | |||
| 251 | # Resizing |
||
| 252 | new_available_tags = range_difference( |
||
| 253 | tag_ranges, used_tags |
||
| 254 | ) |
||
| 255 | self.available_tags[tag_type] = new_available_tags |
||
| 256 | self.tag_ranges[tag_type] = tag_ranges |
||
| 257 | |||
| 258 | def remove_tag_ranges(self, tag_type: str): |
||
| 259 | """Set tag_ranges[tag_type] to default value""" |
||
| 260 | if tag_type != TAGType.VLAN.value: |
||
| 261 | msg = f"Tag type {tag_type} is not supported." |
||
| 262 | raise KytosTagtypeNotSupported(msg) |
||
| 263 | with self._tag_lock: |
||
| 264 | used_tags = range_difference( |
||
| 265 | self.tag_ranges[tag_type], self.available_tags[tag_type] |
||
| 266 | ) |
||
| 267 | self.available_tags[tag_type] = range_difference( |
||
| 268 | self.default_tag_values[tag_type], used_tags |
||
| 269 | ) |
||
| 270 | self.tag_ranges[tag_type] = self.default_tag_values[tag_type] |
||
| 271 | |||
| 272 | def _remove_tags(self, tags: list[int], tag_type: str = 'vlan') -> bool: |
||
| 273 | """Remove tags by resizing available_tags |
||
| 274 | Returns False if nothing was remove, True otherwise""" |
||
| 275 | available = self.available_tags[tag_type] |
||
| 276 | if not available: |
||
| 277 | return False |
||
| 278 | index = find_index_remove(available, tags) |
||
| 279 | if index is None: |
||
| 280 | return False |
||
| 281 | # Resizing |
||
| 282 | if tags[0] == available[index][0]: |
||
| 283 | if tags[1] == available[index][1]: |
||
| 284 | available.pop(index) |
||
| 285 | else: |
||
| 286 | available[index][0] = tags[1] + 1 |
||
| 287 | elif tags[1] == available[index][1]: |
||
| 288 | available[index][1] = tags[0] - 1 |
||
| 289 | else: |
||
| 290 | available[index: index+1] = [ |
||
| 291 | [available[index][0], tags[0]-1], |
||
| 292 | [tags[1]+1, available[index][1]] |
||
| 293 | ] |
||
| 294 | return True |
||
| 295 | |||
| 296 | def use_tags( |
||
| 297 | self, |
||
| 298 | controller, |
||
| 299 | tags: Union[int, list[int], list[list[int]]], |
||
| 300 | tag_type: str = 'vlan', |
||
| 301 | use_lock: bool = True, |
||
| 302 | check_order: bool = True, |
||
| 303 | ): |
||
| 304 | """Remove a specific tag from available_tags if it is there. |
||
| 305 | Exception raised in case the tags were not able to be removed. |
||
| 306 | |||
| 307 | Args: |
||
| 308 | controller: Kytos controller |
||
| 309 | tags: value to be removed, multiple types for compatibility: |
||
| 310 | (int): Single tag |
||
| 311 | (list[int]): Single range of tags |
||
| 312 | (list[list[int]]): List of ranges of tags |
||
| 313 | tag_type: TAG type value |
||
| 314 | use_lock: Boolean to whether use a lock or not |
||
| 315 | check_order: Boolean to whether validate tags(list). Check order, |
||
| 316 | type and length. Set to false when invocated internally. |
||
| 317 | |||
| 318 | Exceptions: |
||
| 319 | KytosTagsAreNotAvailable from _use_tags() |
||
| 320 | """ |
||
| 321 | if isinstance(tags, int): |
||
| 322 | tags = [tags] * 2 |
||
| 323 | elif check_order: |
||
| 324 | tags = get_validated_tags(tags) |
||
| 325 | if use_lock: |
||
| 326 | with self._tag_lock: |
||
| 327 | self._use_tags(tags, tag_type) |
||
| 328 | else: |
||
| 329 | self._use_tags(tags, tag_type) |
||
| 330 | |||
| 331 | self._notify_interface_tags(controller) |
||
| 332 | |||
| 333 | def _use_tags( |
||
| 334 | self, |
||
| 335 | tags: Union[list[int], list[list[int]]], |
||
| 336 | tag_type: str |
||
| 337 | ): |
||
| 338 | """Manage available_tags deletion changes.""" |
||
| 339 | if isinstance(tags[0], list): |
||
| 340 | available_copy = deepcopy(self.available_tags[tag_type]) |
||
| 341 | for tag_range in tags: |
||
| 342 | result = self._remove_tags(tag_range, tag_type) |
||
| 343 | if result is False: |
||
| 344 | self.available_tags[tag_type] = available_copy |
||
| 345 | conflict = range_difference(tags, available_copy) |
||
| 346 | raise KytosTagsAreNotAvailable(conflict, self._id) |
||
| 347 | else: |
||
| 348 | result = self._remove_tags(tags, tag_type) |
||
| 349 | if result is False: |
||
| 350 | raise KytosTagsAreNotAvailable([tags], self._id) |
||
| 351 | |||
| 352 | # pylint: disable=too-many-branches |
||
| 353 | def _add_tags(self, tags: list[int], tag_type: str = 'vlan') -> bool: |
||
| 354 | """Add tags, return True if they were added. |
||
| 355 | Returns False when nothing was added, True otherwise |
||
| 356 | Ensuring that ranges are not unnecessarily divided |
||
| 357 | available_tag e.g [[7, 10], [20, 30], [78, 92], [100, 109], [189, 200]] |
||
| 358 | tags examples are in each if statement. |
||
| 359 | """ |
||
| 360 | if not tags[0] or not tags[1]: |
||
| 361 | return False |
||
| 362 | |||
| 363 | # Check if tags is within self.tag_ranges |
||
| 364 | tag_ranges = self.tag_ranges[tag_type] |
||
| 365 | if find_index_remove(tag_ranges, tags) is None: |
||
| 366 | raise KytosTagsNotInTagRanges([tags], self._id) |
||
| 367 | |||
| 368 | available = self.available_tags[tag_type] |
||
| 369 | if not available: |
||
| 370 | self.available_tags[tag_type] = [tags] |
||
| 371 | return True |
||
| 372 | |||
| 373 | index = find_index_add(available, tags) |
||
| 374 | if index is None: |
||
| 375 | return False |
||
| 376 | if index == 0: |
||
| 377 | # [1, 6] |
||
| 378 | if tags[1] == available[index][0] - 1: |
||
| 379 | available[index][0] = tags[0] |
||
| 380 | # [1, 2] |
||
| 381 | else: |
||
| 382 | available.insert(0, tags) |
||
| 383 | elif index == len(available): |
||
| 384 | # [201, 300] |
||
| 385 | if available[index-1][1] + 1 == tags[0]: |
||
| 386 | available[index-1][1] = tags[1] |
||
| 387 | # [250, 300] |
||
| 388 | else: |
||
| 389 | available.append(tags) |
||
| 390 | else: |
||
| 391 | # [11, 19] |
||
| 392 | if (available[index-1][1] + 1 == tags[0] and |
||
| 393 | available[index][0] - 1 == tags[1]): |
||
| 394 | available[index-1: index+1] = [ |
||
| 395 | [available[index-1][0], available[index][1]] |
||
| 396 | ] |
||
| 397 | # [11, 15] |
||
| 398 | elif available[index-1][1] + 1 == tags[0]: |
||
| 399 | available[index-1][1] = tags[1] |
||
| 400 | # [15, 19] |
||
| 401 | elif available[index][0] - 1 == tags[1]: |
||
| 402 | available[index][0] = tags[0] |
||
| 403 | # [15, 15] |
||
| 404 | else: |
||
| 405 | available.insert(index, tags) |
||
| 406 | return True |
||
| 407 | |||
| 408 | def make_tags_available( |
||
| 409 | self, |
||
| 410 | controller, |
||
| 411 | tags: Union[int, list[int], list[list[int]]], |
||
| 412 | tag_type: str = 'vlan', |
||
| 413 | use_lock: bool = True, |
||
| 414 | check_order: bool = True, |
||
| 415 | ) -> list[list[int]]: |
||
| 416 | """Add a tags in available_tags. |
||
| 417 | |||
| 418 | Args: |
||
| 419 | controller: Kytos controller |
||
| 420 | tags: value to be added, multiple types for compatibility: |
||
| 421 | (int): Single tag |
||
| 422 | (list[int]): Single range of tags |
||
| 423 | (list[list[int]]): List of ranges of tags |
||
| 424 | tag_type: TAG type value |
||
| 425 | use_lock: Boolean to whether use a lock or not |
||
| 426 | check_order: Boolean to whether validate tags(list). Check order, |
||
| 427 | type and length. Set to false when invocated internally. |
||
| 428 | |||
| 429 | Return: |
||
| 430 | conflict: Return any values that were not added. |
||
| 431 | |||
| 432 | Exeptions: |
||
| 433 | KytosTagsNotInTagRanges from _make_tags_available() |
||
| 434 | """ |
||
| 435 | if isinstance(tags, int): |
||
| 436 | tags = [tags] * 2 |
||
| 437 | elif check_order: |
||
| 438 | tags = get_validated_tags(tags) |
||
| 439 | if isinstance(tags[0], int) and tags[0] != tags[1]: |
||
| 440 | tags = [tags] |
||
| 441 | if use_lock: |
||
| 442 | with self._tag_lock: |
||
| 443 | conflict = self._make_tags_available(tags, tag_type) |
||
| 444 | else: |
||
| 445 | conflict = self._make_tags_available(tags, tag_type) |
||
| 446 | self._notify_interface_tags(controller) |
||
| 447 | return conflict |
||
| 448 | |||
| 449 | def _make_tags_available( |
||
| 450 | self, |
||
| 451 | tags: Union[list[int], list[list[int]]], |
||
| 452 | tag_type: str, |
||
| 453 | ) -> list[list[int]]: |
||
| 454 | """Manage available_tags adittion changes |
||
| 455 | |||
| 456 | Exceptions: |
||
| 457 | KytosTagsNotInTagRanges from _add_tags() |
||
| 458 | """ |
||
| 459 | if isinstance(tags[0], list): |
||
| 460 | diff = range_difference(tags, self.tag_ranges[tag_type]) |
||
| 461 | if diff: |
||
| 462 | raise KytosTagsNotInTagRanges(diff, self._id) |
||
| 463 | available_tags = self.available_tags[tag_type] |
||
| 464 | new_tags, conflict = range_addition(tags, available_tags) |
||
| 465 | self.available_tags[tag_type] = new_tags |
||
| 466 | return conflict |
||
| 467 | result = self._add_tags(tags, tag_type) |
||
| 468 | if result is False: |
||
| 469 | return [tags] |
||
| 470 | return [] |
||
| 471 | |||
| 472 | def set_available_tags_tag_ranges( |
||
| 473 | self, |
||
| 474 | available_tag: dict[str, list[list[int]]], |
||
| 475 | tag_ranges: dict[str, list[list[int]]] |
||
| 476 | ): |
||
| 477 | """Set a range of VLAN tags to be used by this Interface. |
||
| 478 | |||
| 479 | Args: |
||
| 480 | available_tag: Available tags from each tag type |
||
| 481 | tag_ranges: Restriction for each type of available tag |
||
| 482 | """ |
||
| 483 | with self._tag_lock: |
||
| 484 | self.available_tags = available_tag |
||
| 485 | self.tag_ranges = tag_ranges |
||
| 486 | |||
| 487 | def enable(self): |
||
| 488 | """Enable this interface instance. |
||
| 489 | |||
| 490 | Also enable the switch instance this interface is attached to. |
||
| 491 | """ |
||
| 492 | self.switch.enable() |
||
| 493 | self._enabled = True |
||
| 494 | |||
| 495 | def is_tag_available(self, tag: int, tag_type: str = 'vlan'): |
||
| 496 | """Check if a tag is available.""" |
||
| 497 | with self._tag_lock: |
||
| 498 | if find_index_remove( |
||
| 499 | self.available_tags[tag_type], [tag, tag] |
||
| 500 | ) is not None: |
||
| 501 | return True |
||
| 502 | return False |
||
| 503 | |||
| 504 | def get_endpoint(self, endpoint): |
||
| 505 | """Return a tuple with existent endpoint, None otherwise. |
||
| 506 | |||
| 507 | Args: |
||
| 508 | endpoint(|hw_address|, :class:`.Interface`): endpoint instance. |
||
| 509 | |||
| 510 | Returns: |
||
| 511 | tuple: A tuple with endpoint and time of last update. |
||
| 512 | |||
| 513 | """ |
||
| 514 | for item in self.endpoints: |
||
| 515 | if endpoint == item[0]: |
||
| 516 | return item |
||
| 517 | return None |
||
| 518 | |||
| 519 | def add_endpoint(self, endpoint): |
||
| 520 | """Create a new endpoint to Interface instance. |
||
| 521 | |||
| 522 | Args: |
||
| 523 | endpoint(|hw_address|, :class:`.Interface`): A target endpoint. |
||
| 524 | """ |
||
| 525 | exists = self.get_endpoint(endpoint) |
||
| 526 | if not exists: |
||
| 527 | self.endpoints.append((endpoint, now())) |
||
| 528 | |||
| 529 | def delete_endpoint(self, endpoint): |
||
| 530 | """Delete a existent endpoint in Interface instance. |
||
| 531 | |||
| 532 | Args: |
||
| 533 | endpoint (|hw_address|, :class:`.Interface`): A target endpoint. |
||
| 534 | """ |
||
| 535 | exists = self.get_endpoint(endpoint) |
||
| 536 | if exists: |
||
| 537 | self.endpoints.remove(exists) |
||
| 538 | |||
| 539 | def update_endpoint(self, endpoint): |
||
| 540 | """Update or create new endpoint to Interface instance. |
||
| 541 | |||
| 542 | Args: |
||
| 543 | endpoint(|hw_address|, :class:`.Interface`): A target endpoint. |
||
| 544 | """ |
||
| 545 | exists = self.get_endpoint(endpoint) |
||
| 546 | if exists: |
||
| 547 | self.delete_endpoint(endpoint) |
||
| 548 | self.add_endpoint(endpoint) |
||
| 549 | |||
| 550 | def update_link(self, link): |
||
| 551 | """Update link for this interface in a consistent way. |
||
| 552 | |||
| 553 | Verify of the other endpoint of the link has the same Link information |
||
| 554 | attached to it, and change it if necessary. |
||
| 555 | |||
| 556 | Warning: This method can potentially change information of other |
||
| 557 | Interface instances. Use it with caution. |
||
| 558 | """ |
||
| 559 | if self not in (link.endpoint_a, link.endpoint_b): |
||
| 560 | return False |
||
| 561 | |||
| 562 | if self.link is None or self.link != link: |
||
| 563 | self.link = link |
||
| 564 | |||
| 565 | if link.endpoint_a == self: |
||
| 566 | endpoint = link.endpoint_b |
||
| 567 | else: |
||
| 568 | endpoint = link.endpoint_a |
||
| 569 | |||
| 570 | if endpoint.link is None or endpoint.link != link: |
||
| 571 | endpoint.link = link |
||
| 572 | |||
| 573 | return True |
||
| 574 | |||
| 575 | @property |
||
| 576 | def speed(self): |
||
| 577 | """Return the link speed in bytes per second, None otherwise. |
||
| 578 | |||
| 579 | If the switch was disconnected, we have :attr:`features` and speed is |
||
| 580 | still returned for common values between v0x01 and v0x04. For specific |
||
| 581 | v0x04 values (40 Gbps, 100 Gbps and 1 Tbps), the connection must be |
||
| 582 | active so we can make sure the protocol version is v0x04. |
||
| 583 | |||
| 584 | Returns: |
||
| 585 | int, None: Link speed in bytes per second or ``None``. |
||
| 586 | |||
| 587 | """ |
||
| 588 | speed = self.get_of_features_speed() |
||
| 589 | |||
| 590 | if speed is not None: |
||
| 591 | return speed |
||
| 592 | |||
| 593 | if self._custom_speed is not None: |
||
| 594 | return self._custom_speed |
||
| 595 | |||
| 596 | if self._is_v0x04() and self.port_number == PortNo04.OFPP_LOCAL: |
||
| 597 | return 0 |
||
| 598 | |||
| 599 | if not self._is_v0x04() and self.port_number == PortNo01.OFPP_LOCAL: |
||
| 600 | return 0 |
||
| 601 | |||
| 602 | # Warn unknown speed |
||
| 603 | # Use shorter switch ID with its beginning and end |
||
| 604 | if isinstance(self.switch.id, str) and len(self.switch.id) > 20: |
||
| 605 | switch_id = self.switch.id[:3] + '...' + self.switch.id[-3:] |
||
| 606 | else: |
||
| 607 | switch_id = self.switch.id |
||
| 608 | LOG.warning("Couldn't get port %s speed, sw %s, feats %s", |
||
| 609 | self.port_number, switch_id, self.features) |
||
| 610 | |||
| 611 | return None |
||
| 612 | |||
| 613 | def set_custom_speed(self, bytes_per_second): |
||
| 614 | """Set a speed that overrides switch OpenFlow information. |
||
| 615 | |||
| 616 | If ``None`` is given, :attr:`speed` becomes the one given by the |
||
| 617 | switch. |
||
| 618 | """ |
||
| 619 | self._custom_speed = bytes_per_second |
||
| 620 | |||
| 621 | def get_custom_speed(self): |
||
| 622 | """Return custom speed or ``None`` if not set.""" |
||
| 623 | return self._custom_speed |
||
| 624 | |||
| 625 | def get_of_features_speed(self): |
||
| 626 | """Return the link speed in bytes per second, None otherwise. |
||
| 627 | |||
| 628 | If the switch was disconnected, we have :attr:`features` and speed is |
||
| 629 | still returned for common values between v0x01 and v0x04. For specific |
||
| 630 | v0x04 values (40 Gbps, 100 Gbps and 1 Tbps), the connection must be |
||
| 631 | active so we can make sure the protocol version is v0x04. |
||
| 632 | |||
| 633 | Returns: |
||
| 634 | int, None: Link speed in bytes per second or ``None``. |
||
| 635 | |||
| 636 | """ |
||
| 637 | speed = self._get_v0x01_v0x04_speed() |
||
| 638 | # Don't use switch.is_connected() because we can have the protocol |
||
| 639 | if speed is None and self._is_v0x04(): |
||
| 640 | speed = self._get_v0x04_speed() |
||
| 641 | return speed |
||
| 642 | |||
| 643 | def _is_v0x04(self): |
||
| 644 | """Whether the switch is connected using OpenFlow 1.3.""" |
||
| 645 | return self.switch.is_connected() and \ |
||
| 646 | self.switch.connection.protocol.version == 0x04 |
||
| 647 | |||
| 648 | def _get_v0x01_v0x04_speed(self): |
||
| 649 | """Check against all values of v0x01. They're part of v0x04.""" |
||
| 650 | fts = self.features |
||
| 651 | pfts = PortFeatures01 |
||
| 652 | if fts and fts & pfts.OFPPF_10GB_FD: |
||
| 653 | return 10 * 10**9 / 8 |
||
| 654 | if fts and fts & (pfts.OFPPF_1GB_HD | pfts.OFPPF_1GB_FD): |
||
| 655 | return 10**9 / 8 |
||
| 656 | if fts and fts & (pfts.OFPPF_100MB_HD | pfts.OFPPF_100MB_FD): |
||
| 657 | return 100 * 10**6 / 8 |
||
| 658 | if fts and fts & (pfts.OFPPF_10MB_HD | pfts.OFPPF_10MB_FD): |
||
| 659 | return 10 * 10**6 / 8 |
||
| 660 | return None |
||
| 661 | |||
| 662 | def _get_v0x04_speed(self): |
||
| 663 | """Check against higher enums of v0x04. |
||
| 664 | |||
| 665 | Must be called after :meth:`get_v0x01_speed` returns ``None``. |
||
| 666 | """ |
||
| 667 | fts = self.features |
||
| 668 | pfts = PortFeatures04 |
||
| 669 | if fts and fts & pfts.OFPPF_1TB_FD: |
||
| 670 | return 10**12 / 8 |
||
| 671 | if fts and fts & pfts.OFPPF_100GB_FD: |
||
| 672 | return 100 * 10**9 / 8 |
||
| 673 | if fts and fts & pfts.OFPPF_40GB_FD: |
||
| 674 | return 40 * 10**9 / 8 |
||
| 675 | return None |
||
| 676 | |||
| 677 | def get_hr_speed(self): |
||
| 678 | """Return Human-Readable string for link speed. |
||
| 679 | |||
| 680 | Returns: |
||
| 681 | string: String with link speed. e.g: '350 Gbps' or '350 Mbps'. |
||
| 682 | |||
| 683 | """ |
||
| 684 | speed = self.speed |
||
| 685 | if speed is None: |
||
| 686 | return '' |
||
| 687 | speed *= 8 |
||
| 688 | if speed == 10**12: |
||
| 689 | return '1 Tbps' |
||
| 690 | if speed >= 10**9: |
||
| 691 | return f"{round(speed / 10**9)} Gbps" |
||
| 692 | return f"{round(speed / 10**6)} Mbps" |
||
| 693 | |||
| 694 | def as_dict(self): |
||
| 695 | """Return a dictionary with Interface attributes. |
||
| 696 | |||
| 697 | Speed is in bytes/sec. Example of output (100 Gbps): |
||
| 698 | |||
| 699 | .. code-block:: python3 |
||
| 700 | |||
| 701 | {'id': '00:00:00:00:00:00:00:01:2', |
||
| 702 | 'name': 'eth01', |
||
| 703 | 'port_number': 2, |
||
| 704 | 'mac': '00:7e:04:3b:c2:a6', |
||
| 705 | 'switch': '00:00:00:00:00:00:00:01', |
||
| 706 | 'type': 'interface', |
||
| 707 | 'nni': False, |
||
| 708 | 'uni': True, |
||
| 709 | 'speed': 12500000000, |
||
| 710 | 'metadata': {}, |
||
| 711 | 'lldp': True, |
||
| 712 | 'active': True, |
||
| 713 | 'enabled': False, |
||
| 714 | 'status': 'DISABLED', |
||
| 715 | 'link': "" |
||
| 716 | } |
||
| 717 | |||
| 718 | Returns: |
||
| 719 | dict: Dictionary filled with interface attributes. |
||
| 720 | |||
| 721 | """ |
||
| 722 | iface_dict = { |
||
| 723 | 'id': self.id, |
||
| 724 | 'name': self.name, |
||
| 725 | 'port_number': self.port_number, |
||
| 726 | 'mac': self.address, |
||
| 727 | 'switch': self.switch.dpid, |
||
| 728 | 'type': 'interface', |
||
| 729 | 'nni': self.nni, |
||
| 730 | 'uni': self.uni, |
||
| 731 | 'speed': self.speed, |
||
| 732 | 'metadata': self.metadata, |
||
| 733 | 'lldp': self.lldp, |
||
| 734 | 'active': self.is_active(), |
||
| 735 | 'enabled': self.is_enabled(), |
||
| 736 | 'status': self.status.value, |
||
| 737 | 'status_reason': sorted(self.status_reason), |
||
| 738 | 'link': self.link.id if self.link else "", |
||
| 739 | } |
||
| 740 | if self.stats: |
||
| 741 | iface_dict['stats'] = self.stats.as_dict() |
||
| 742 | return iface_dict |
||
| 743 | |||
| 744 | @classmethod |
||
| 745 | def from_dict(cls, interface_dict): |
||
| 746 | """Return a Interface instance from python dictionary.""" |
||
| 747 | return cls(interface_dict.get('name'), |
||
| 748 | interface_dict.get('port_number'), |
||
| 749 | interface_dict.get('switch'), |
||
| 750 | interface_dict.get('address'), |
||
| 751 | interface_dict.get('state'), |
||
| 752 | interface_dict.get('features'), |
||
| 753 | interface_dict.get('speed')) |
||
| 754 | |||
| 755 | def as_json(self): |
||
| 756 | """Return a json with Interfaces attributes. |
||
| 757 | |||
| 758 | Example of output: |
||
| 759 | |||
| 760 | .. code-block:: json |
||
| 761 | |||
| 762 | {"mac": "00:7e:04:3b:c2:a6", |
||
| 763 | "switch": "00:00:00:00:00:00:00:01", |
||
| 764 | "type": "interface", |
||
| 765 | "name": "eth01", |
||
| 766 | "id": "00:00:00:00:00:00:00:01:2", |
||
| 767 | "port_number": 2, |
||
| 768 | "speed": "350 Mbps"} |
||
| 769 | |||
| 770 | Returns: |
||
| 771 | string: Json filled with interface attributes. |
||
| 772 | |||
| 773 | """ |
||
| 774 | return json.dumps(self.as_dict()) |
||
| 775 | |||
| 776 | def _notify_interface_tags(self, controller): |
||
| 777 | """Notify link available tags""" |
||
| 778 | name = "kytos/core.interface_tags" |
||
| 779 | content = {"interface": self} |
||
| 780 | event = KytosEvent(name=name, content=content) |
||
| 781 | controller.buffers.app.put(event) |
||
| 782 | |||
| 783 | |||
| 784 | class UNI: |
||
| 785 | """Class that represents an User-to-Network Interface.""" |
||
| 786 | |||
| 787 | def __init__( |
||
| 788 | self, |
||
| 789 | interface: Interface, |
||
| 790 | user_tag: Union[None, TAG, TAGRange] |
||
| 791 | ): |
||
| 792 | self.user_tag = user_tag |
||
| 793 | self.interface = interface |
||
| 794 | |||
| 795 | def __eq__(self, other): |
||
| 796 | """Override the default implementation.""" |
||
| 797 | return (self.user_tag == other.user_tag and |
||
| 798 | self.interface == other.interface) |
||
| 799 | |||
| 800 | def _is_reserved_valid_tag(self) -> bool: |
||
| 801 | """Check if TAG string is possible""" |
||
| 802 | reserved_tag = {"any", "untagged"} |
||
| 803 | if self.user_tag.value in reserved_tag: |
||
| 804 | return True |
||
| 805 | return False |
||
| 806 | |||
| 807 | def is_valid(self): |
||
| 808 | """Check if TAG is possible for this interface TAG pool.""" |
||
| 809 | if self.user_tag: |
||
| 810 | tag = self.user_tag.value |
||
| 811 | if isinstance(tag, str): |
||
| 812 | return self._is_reserved_valid_tag() |
||
| 813 | if isinstance(tag, int): |
||
| 814 | return self.interface.is_tag_available(tag) |
||
| 815 | return True |
||
| 816 | |||
| 817 | def as_dict(self): |
||
| 818 | """Return a dict representating a UNI object.""" |
||
| 819 | return { |
||
| 820 | 'interface_id': self.interface.id, |
||
| 821 | 'tag': self.user_tag.as_dict() if self.user_tag else None |
||
| 822 | } |
||
| 823 | |||
| 824 | @classmethod |
||
| 825 | def from_dict(cls, uni): |
||
| 826 | """Return a Uni instance from python dictionary.""" |
||
| 827 | return cls(uni.get('interface'), |
||
| 828 | uni.get('user_tag')) |
||
| 829 | |||
| 830 | def as_json(self): |
||
| 831 | """Return a json representating a UNI object.""" |
||
| 832 | return json.dumps(self.as_dict()) |
||
| 833 | |||
| 834 | |||
| 835 | class NNI: |
||
| 836 | """Class that represents an Network-to-Network Interface.""" |
||
| 837 | |||
| 838 | def __init__(self, interface): |
||
| 839 | self.interface = interface |
||
| 840 | |||
| 841 | |||
| 842 | class VNNI(NNI): |
||
| 843 | """Class that represents an Virtual Network-to-Network Interface.""" |
||
| 844 | |||
| 845 | def __init__(self, service_tag, *args, **kwargs): |
||
| 846 | self.service_tag = service_tag |
||
| 847 | |||
| 848 | super().__init__(*args, **kwargs) |
||
| 849 |