1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930
|
"""Base and fundamental classes used all over the library.
Besides classes, several constants are defined here. We designed
python-openflow in a manner to make it easy to create new messages and OpenFlow
structs. You can realize that when you see a message class definition.
A **struct** here is a group of basic attributes and/or struct attributes (i.e.
:class:`~pyof.v0x01.common.header.Header`). A **message** here is like a
struct, but all messages have a header attribute (i.e.
:class:`~pyof.v0x01.asynchronous.packet_in.PacketIn`).
The main classes of this module are :class:`GenericStruct`,
:class:`GenericMessage`, :class:`GenericBitMask` and :class:`GenericType`.
These classes are used in all parts of this library.
"""
# System imports
import importlib
import re
import struct
from collections import OrderedDict
from copy import deepcopy
from enum import Enum, IntEnum
from random import randint
from pyof.foundation.constants import UBINT32_MAX_VALUE as MAXID
from pyof.foundation.exceptions import (
BadValueException, PackException, UnpackException, ValidationError)
# Third-party imports
# This will determine the order on sphinx documentation.
__all__ = ('GenericStruct', 'GenericMessage', 'GenericType', 'GenericBitMask',
'MetaStruct', 'MetaBitMask', 'UBIntBase')
# Classes
class GenericType:
"""Foundation class for all custom attributes.
Base class for :class:`~.UBIntBase`, :class:`~.Char`
and others.
"""
_fmt = None
def __init__(self, value=None, enum_ref=None):
"""Create a GenericType with the optional parameters below.
Args:
value: The type's value.
enum_ref (type): If :attr:`value` is from an Enum, specify
its type.
"""
self._value = value
self.enum_ref = enum_ref
def __deepcopy__(self, memo):
"""Improve deepcopy speed."""
return type(self)(value=self._value, enum_ref=self.enum_ref)
def __repr__(self):
return "{}({})".format(type(self).__name__, repr(self._value))
def __str__(self):
return '{}'.format(str(self._value))
def __eq__(self, other):
if isinstance(other, self.__class__):
return self.pack() == other.pack()
if hasattr(other, 'value'):
return self.value == other.value
return self.value == other
def __ne__(self, other):
return self._value != other
def __gt__(self, other):
return self._value > other
def __ge__(self, other):
return self._value >= other
def __lt__(self, other):
return self._value < other
def __le__(self, other):
return self._value <= other
def __add__(self, other):
return self.value + other
def __radd__(self, other):
return self.value + other
def __sub__(self, other):
return self.value - other
def __rsub__(self, other):
return self.value - other
def __or__(self, other):
return self.value | other
def __ror__(self, other):
return self.value | other
def __and__(self, other):
return self.value & other
def __rand__(self, other):
return self.value & other
def __xor__(self, other):
return self.value ^ other
def __rxor__(self, other):
return self.value ^ other
def __lshift__(self, shift):
return self.value << shift
def __rshift__(self, shift):
return self.value >> shift
def __len__(self):
return self.get_size()
@property
def value(self):
"""Return this type's value.
Returns:
object: The value of an enum, bitmask, etc.
"""
if self.is_enum():
if isinstance(self._value, self.enum_ref):
return self._value.value
return self._value
if self.is_bitmask():
return self._value.bitmask
return self._value
def pack(self, value=None):
r"""Pack the value as a binary representation.
Considering an example with UBInt8 class, that inherits from
GenericType:
>>> from pyof.foundation.basic_types import UBInt8
>>> objectA = UBInt8(1)
>>> objectB = 5
>>> objectA.pack()
b'\x01'
>>> objectA.pack(objectB)
b'\x05'
Args:
value: If the value is None, then we will pack the value of the
current instance. Otherwise, if value is an instance of the
same type as the current instance, then we call the pack of the
value object. Otherwise, we will use the current instance pack
method on the passed value.
Returns:
bytes: The binary representation.
Raises:
:exc:`~.exceptions.BadValueException`: If the value does not
fit the binary format.
"""
if isinstance(value, type(self)):
return value.pack()
if value is None:
value = self.value
elif 'value' in dir(value):
# if it is enum or bitmask gets only the 'int' value
value = value.value
try:
return struct.pack(self._fmt, value)
except struct.error:
expected_type = type(self).__name__
actual_type = type(value).__name__
msg_args = expected_type, value, actual_type
msg = 'Expected {}, found value "{}" of type {}'.format(*msg_args)
raise PackException(msg)
def unpack(self, buff, offset=0):
"""Unpack *buff* into this object.
This method will convert a binary data into a readable value according
to the attribute format.
Args:
buff (bytes): Binary buffer.
offset (int): Where to begin unpacking.
Raises:
:exc:`~.exceptions.UnpackException`: If unpack fails.
"""
try:
self._value = struct.unpack_from(self._fmt, buff, offset)[0]
if self.enum_ref:
self._value = self.enum_ref(self._value)
except (struct.error, TypeError, ValueError) as exception:
msg = '{}; fmt = {}, buff = {}, offset = {}.'.format(exception,
self._fmt,
buff, offset)
raise UnpackException(msg)
def get_size(self, value=None): # pylint: disable=unused-argument
"""Return the size in bytes of this type.
Returns:
int: Size in bytes.
"""
return struct.calcsize(self._fmt)
def is_valid(self):
"""Check whether the value fits the binary format.
Assert that :func:`pack` succeeds.
Returns:
bool: Whether the value is valid for this type.
"""
try:
self.pack()
return True
except BadValueException:
return False
def is_enum(self):
"""Test whether it is an :class:`~enum.Enum`.
Returns:
bool: Whether it is an :class:`~enum.Enum`.
"""
return self.enum_ref and issubclass(self.enum_ref, (Enum, IntEnum))
def is_bitmask(self):
"""Test whether it is a :class:`GenericBitMask`.
Returns:
bool: Whether it is a :class:`GenericBitMask`.
"""
return self._value and issubclass(type(self._value), GenericBitMask)
class UBIntBase(GenericType):
"""Base class for UBInt{8,16,32,64,128}."""
def __int__(self):
"""Allow converting an UBInt() back to an int()."""
# Skip GenericType's checking if this is an Enum ou BitMask
# (because it won't be), and convert directly from _value
return int(self._value)
class MetaStruct(type):
"""MetaClass that dynamically handles openflow version of class attributes.
See more about it at:
https://github.com/kytos/python-openflow/wiki/Version-Inheritance
You do not need to use this class. Inherit from :class:`GenericStruct`
instead.
"""
# pylint: disable=unused-argument
@classmethod
def __prepare__(cls, name, bases, **kwargs):
return OrderedDict()
def __new__(cls, name, bases, classdict, **kwargs):
"""Inherit attributes from parent class and update their versions.
Here is the moment that the new class is going to be created. During
this process, two things may be done.
Firstly, we will look if the type of any parent classes is this
MetaStruct. We will inherit from the first parent class that fits this
requirement. If any is found, then we will get all attributes from this
class and place them as class attributes on the class being created.
Secondly, for each class attribute being inherited, we will make sure
that the pyof version of this attribute is the same as the version of
the current class being created. If it is not, then we will find out
which is the class and module of that attribute, look for a version
that matches the version of the current class and replace that
attribute with the correct version.
See this link for more information on why this is being done:
- https://github.com/kytos/python-openflow/wiki/Version-Inheritance
"""
#: Retrieving class attributes management markers
removed_attributes = classdict.pop('_removed_attributes', [])
# renamed_attributes = classdict.pop('_renamed_attributes', [])
# reordered_attributes = classdict.pop('_reordered_attributes', {})
curr_module = classdict.get('__module__')
curr_version = MetaStruct.get_pyof_version(curr_module)
inherited_attributes = None
#: looking for (kytos) class attributes defined on the bases
#: classes so we can copy them into the current class being created
#: so we can "inherit" them as class attributes
for base in bases:
#: Check if we are inheriting from one of our classes.
if isinstance(base, MetaStruct):
inherited_attributes = OrderedDict()
for attr_name, obj in base.get_class_attributes():
#: Get an updated version of this attribute,
#: considering the version of the current class being
#: created.
attr = MetaStruct.get_pyof_obj_new_version(attr_name, obj,
curr_version)
if attr_name == 'header':
attr = cls._header_message_type_update(obj, attr)
inherited_attributes.update([attr])
#: We are going to inherit just from the 'closest parent'
break
#: If we have inherited something, then first we will remove the
#: attributes marked to be removed on the 'removed_attributes' and
#: after that we will update the inherited 'classdict' with the
#: attributes from the current classdict.
if inherited_attributes is not None:
#: removing attributes set to be removed
for attr_name in removed_attributes:
inherited_attributes.pop(attr_name, None)
#: Updating the inherited attributes with those defined on the
#: body of the class being created.
inherited_attributes.update(classdict)
classdict = inherited_attributes
return super().__new__(cls, name, bases, classdict, **kwargs)
@staticmethod
def _header_message_type_update(obj, attr):
"""Update the message type on the header.
Set the message_type of the header according to the message_type of
the parent class.
"""
old_enum = obj.message_type
new_header = attr[1]
new_enum = new_header.__class__.message_type.enum_ref
#: This 'if' will be removed on the future with an
#: improved version of __init_subclass__ method of the
#: GenericMessage class
if old_enum:
msg_type_name = old_enum.name
new_type = new_enum[msg_type_name]
new_header.message_type = new_type
return (attr[0], new_header)
@staticmethod
def get_pyof_version(module_fullname):
"""Get the module pyof version based on the module fullname.
Args:
module_fullname (str): The fullname of the module
(e.g.: pyof.v0x01.common.header)
Returns:
str: openflow version.
The openflow version, on the format 'v0x0?' if any. Or None
if there isn't a version on the fullname.
"""
ver_module_re = re.compile(r'(pyof\.)(v0x\d+)(\..*)')
matched = ver_module_re.match(module_fullname)
if matched:
version = matched.group(2)
return version
return None
@staticmethod
def replace_pyof_version(module_fullname, version):
"""Replace the OF Version of a module fullname.
Get's a module name (eg. 'pyof.v0x01.common.header') and returns it on
a new 'version' (eg. 'pyof.v0x02.common.header').
Args:
module_fullname (str): The fullname of the module
(e.g.: pyof.v0x01.common.header)
version (str): The version to be 'inserted' on the module fullname.
Returns:
str: module fullname
The new module fullname, with the replaced version,
on the format "pyof.v0x01.common.header". If the requested
version is the same as the one of the module_fullname or if
the module_fullname is not a 'OF version' specific module,
returns None.
"""
module_version = MetaStruct.get_pyof_version(module_fullname)
if not module_version or module_version == version:
return None
return module_fullname.replace(module_version, version)
@staticmethod
def get_pyof_obj_new_version(name, obj, new_version):
r"""Return a class attribute on a different pyof version.
This method receives the name of a class attribute, the class attribute
itself (object) and an openflow version.
The attribute will be evaluated and from it we will recover its class
and the module where the class was defined.
If the module is a "python-openflow version specific module" (starts
with "pyof.v0"), then we will get it's version and if it is different
from the 'new_version', then we will get the module on the
'new_version', look for the 'obj' class on the new module and return
an instance of the new version of the 'obj'.
Example:
>>> from pyof.foundation.base import MetaStruct as ms
>>> from pyof.v0x01.common.header import Header
>>> name = 'header'
>>> obj = Header()
>>> new_version = 'v0x04'
>>> header, obj2 = ms.get_pyof_obj_new_version(name, obj, new_version)
>>> header
'header'
>>> obj.version
UBInt8(1)
>>> obj2.version
UBInt8(4)
Args:
name (str): the name of the class attribute being handled.
obj (object): the class attribute itself
new_version (string): the pyof version in which you want the object
'obj'.
Return:
(str, obj): Tuple with class name and object instance.
A tuple in which the first item is the name of the class
attribute (the same that was passed), and the second item is a
instance of the passed class attribute. If the class attribute
is not a pyof versioned attribute, then the same passed object
is returned without any changes. Also, if the obj is a pyof
versioned attribute, but it is already on the right version
(same as new_version), then the passed obj is return.
"""
if new_version is None:
return (name, obj)
cls = obj.__class__
cls_name = cls.__name__
cls_mod = cls.__module__
#: If the module name does not starts with pyof.v0 then it is not a
#: 'pyof versioned' module (OpenFlow specification defined), so we do
#: not have anything to do with it.
new_mod = MetaStruct.replace_pyof_version(cls_mod, new_version)
if new_mod is not None:
# Loads the module
new_mod = importlib.import_module(new_mod)
#: Get the class from the loaded module
new_cls = getattr(new_mod, cls_name)
#: return the tuple with the attribute name and the instance
return (name, new_cls())
return (name, obj)
class GenericStruct(metaclass=MetaStruct):
"""Class inherited by all OpenFlow structs.
If you need to insert a method that will be used by all structs, this is
the place to code it.
.. note:: A struct on this library's context is like a struct in C. It
has a list of attributes and theses attributes can be structs,
too.
"""
def __init__(self):
"""Store attributes' deep copies."""
for name, value in self.get_class_attributes():
setattr(self, name, deepcopy(value))
def __eq__(self, other):
"""Check whether two structures have the same structure and values.
Compare the binary representation of structs to decide whether they
are equal or not.
Args:
other (GenericStruct): The struct to be compared with.
Returns:
bool: Returns the comparison result.
"""
return self.pack() == other.pack()
# def __repr__(self):
# """Generic fallback for __repr__ using the built-in introspection.
#
# For debugging purposes. Disabled to avoid infinite recursion in
# the case that objects reference each other.
#
# """
# # from pprint import pformat
# # attributes = pformat(dict(self._get_instance_attributes()))
# attributes = ', '.join(f'{k}={v!r}' for (k, v)
# in self._get_instance_attributes())
# return f'{type(self).__name__}({attributes})'
@staticmethod
def _attr_fits_into_class(attr, cls):
if not isinstance(attr, cls):
try:
struct.pack(cls._fmt, attr) # pylint: disable=protected-access
except struct.error:
return False
return True
@staticmethod
def _is_pyof_attribute(obj):
"""Return True if the object is a pyof attribute.
To be a pyof attribute the item must be an instance of either
GenericType or GenericStruct.
Returns:
bool: Returns True if the obj is a pyof attribute, otherwise False
"""
return isinstance(obj, (GenericType, GenericStruct))
def _validate_attributes_type(self):
"""Validate the type of each attribute."""
for _attr, _class in self._get_attributes():
if isinstance(_attr, _class):
return True
if issubclass(_class, GenericType):
if GenericStruct._attr_fits_into_class(_attr, _class):
return True
elif not isinstance(_attr, _class):
return False
return True
@classmethod
def get_class_attributes(cls):
"""Return a generator for class attributes' names and value.
This method strict relies on the PEP 520 (Preserving Class Attribute
Definition Order), implemented on Python 3.6. So, if this behaviour
changes this whole lib can lose its functionality (since the
attributes order are a strong requirement.) For the same reason, this
lib will not work on python versions earlier than 3.6.
.. code-block:: python3
for name, value in self.get_class_attributes():
print("attribute name: {}".format(name))
print("attribute type: {}".format(value))
Returns:
generator: tuples with attribute name and value.
"""
#: see this method docstring for a important notice about the use of
#: cls.__dict__
for name, value in cls.__dict__.items():
# gets only our (pyof) attributes. this ignores methods, dunder
# methods and attributes, and common python type attributes.
if GenericStruct._is_pyof_attribute(value):
yield (name, value)
def _get_instance_attributes(self):
"""Return a generator for instance attributes' name and value.
.. code-block:: python3
for _name, _value in self._get_instance_attributes():
print("attribute name: {}".format(_name))
print("attribute value: {}".format(_value))
Returns:
generator: tuples with attribute name and value.
"""
for name, value in self.__dict__.items():
if name in map((lambda x: x[0]), self.get_class_attributes()):
yield (name, value)
def _get_attributes(self):
"""Return a generator for instance and class attribute.
.. code-block:: python3
for instance_attribute, class_attribute in self._get_attributes():
print("Instance Attribute: {}".format(instance_attribute))
print("Class Attribute: {}".format(class_attribute))
Returns:
generator: Tuples with instance attribute and class attribute
"""
return map((lambda i, c: (i[1], c[1])),
self._get_instance_attributes(),
self.get_class_attributes())
def _get_named_attributes(self):
"""Return generator for attribute's name, instance and class values.
Add attribute name to meth:`_get_attributes` for a better debugging
message, so user can find the error easier.
Returns:
generator: Tuple with attribute's name, instance and class values.
"""
for cls, instance in zip(self.get_class_attributes(),
self._get_instance_attributes()):
attr_name, cls_value = cls
instance_value = instance[1]
yield attr_name, instance_value, cls_value
def _unpack_attribute(self, name, obj, buff, begin):
attribute = deepcopy(obj)
setattr(self, name, attribute)
if not buff:
size = 0
else:
try:
attribute.unpack(buff, begin)
size = attribute.get_size()
except UnpackException as exception:
child_cls = type(self).__name__
msg = '{}.{}; {}'.format(child_cls, name, exception)
raise UnpackException(msg)
return size
def get_size(self, value=None):
"""Calculate the total struct size in bytes.
For each struct attribute, sum the result of each one's ``get_size()``
method.
Args:
value: In structs, the user can assign other value instead of a
class' instance.
Returns:
int: Total number of bytes used by the struct.
Raises:
Exception: If the struct is not valid.
"""
if value is None:
return sum(cls_val.get_size(obj_val) for obj_val, cls_val in
self._get_attributes())
if isinstance(value, type(self)):
return value.get_size()
msg = "{} is not an instance of {}".format(value, type(self).__name__)
raise PackException(msg)
def pack(self, value=None):
"""Pack the struct in a binary representation.
Iterate over the class attributes, according to the
order of definition, and then convert each attribute to its byte
representation using its own ``pack`` method.
Returns:
bytes: Binary representation of the struct object.
Raises:
:exc:`~.exceptions.ValidationError`: If validation fails.
"""
if value is None:
if not self.is_valid():
error_msg = "Error on validation prior to pack() on class "
error_msg += "{}.".format(type(self).__name__)
raise ValidationError(error_msg)
message = b''
# pylint: disable=no-member
for attr_info in self._get_named_attributes():
name, instance_value, class_value = attr_info
try:
message += class_value.pack(instance_value)
except PackException as pack_exception:
cls = type(self).__name__
msg = f'{cls}.{name} - {pack_exception}'
raise PackException(msg)
return message
if isinstance(value, type(self)):
return value.pack()
msg = "{} is not an instance of {}".format(value, type(self).__name__)
raise PackException(msg)
def unpack(self, buff, offset=0):
"""Unpack a binary struct into this object's attributes.
Update this object attributes based on the unpacked values of *buff*.
It is an inplace method and it receives the binary data of the struct.
Args:
buff (bytes): Binary data package to be unpacked.
offset (int): Where to begin unpacking.
"""
begin = offset
for name, value in self.get_class_attributes():
size = self._unpack_attribute(name, value, buff, begin)
begin += size
def is_valid(self):
"""Check whether all struct attributes in are valid.
This method will check whether all struct attributes have a proper
value according to the OpenFlow specification. For instance, if you
have a struct with an attribute of type
:class:`~pyof.foundation.basic_types.UBInt8` and you assign a string
value to it, this method will return False.
Returns:
bool: Whether the struct is valid.
"""
return True
# pylint: disable=unreachable
return self._validate_attributes_type()
class GenericMessage(GenericStruct):
"""Base class that is the foundation for all OpenFlow messages.
To add a method that will be used by all messages, write it here.
.. note:: A Message on this library context is like a Struct but has a
also a :attr:`header` attribute.
"""
header = None
def __init__(self, xid=None):
"""Initialize header's xid."""
super().__init__()
self.header.xid = randint(0, MAXID) if xid is None else xid
def __repr__(self):
"""Show a full representation of the object."""
return "%s(xid=%r)" % (self.__class__.__name__,
self.header.xid if self.header else None)
def __init_subclass__(cls, **kwargs):
if cls.header is None or cls.header.__class__.__name__ != 'Header':
msg = "The header attribute must be implemented on the class "
msg += cls.__name__ + "."
raise NotImplementedError(msg)
super().__init_subclass__(**kwargs)
def _validate_message_length(self):
return self.header.length == self.get_size()
def is_valid(self):
"""Check whether a message is valid or not.
This method will validate the Message content. During the validation
process, we check whether the attributes' values are valid according to
the OpenFlow specification. Call this method if you want to verify
whether the message is ready to pack.
Returns:
bool: Whether the message is valid.
"""
return True
# pylint: disable=unreachable
return super().is_valid() and self._validate_message_length()
def pack(self, value=None):
"""Pack the message into a binary data.
One of the basic operations on a Message is the pack operation. During
the packing process, we convert all message attributes to binary
format.
Since that this is usually used before sending the message to a switch,
here we also call :meth:`update_header_length`.
.. seealso:: This method call its parent's :meth:`GenericStruct.pack`
after :meth:`update_header_length`.
Returns:
bytes: A binary data thats represents the Message.
Raises:
Exception: If there are validation errors.
"""
if value is None:
self.update_header_length()
return super().pack()
if isinstance(value, type(self)):
return value.pack()
msg = "{} is not an instance of {}".format(value, type(self).__name__)
raise PackException(msg)
def unpack(self, buff, offset=0):
"""Unpack a binary message into this object's attributes.
Unpack the binary value *buff* and update this object attributes based
on the results. It is an inplace method and it receives the binary data
of the message **without the header**.
Args:
buff (bytes): Binary data package to be unpacked, without the
header.
offset (int): Where to begin unpacking.
"""
begin = offset
for name, value in self.get_class_attributes():
if type(value).__name__ != "Header":
size = self._unpack_attribute(name, value, buff, begin)
begin += size
def update_header_length(self):
"""Update the header length attribute based on current message size.
When sending an OpenFlow message we need to inform the message length
on the header. This is mandatory.
"""
self.header.length = self.get_size()
class MetaBitMask(type):
"""MetaClass to create a special BitMaskEnum type.
You probably do not need to use this class. Inherit from
:class:`GenericBitMask` instead.
This metaclass converts the declared class attributes into elements of an
enum. It also replaces the :meth:`__dir__` and :meth:`__getattr__` methods,
so the resulting class will behave as an :class:`~Enum` class (you can
access object.ELEMENT and recover either values or names).
"""
def __new__(cls, name, bases, classdict):
"""Convert class attributes into enum elements."""
_enum = OrderedDict([(key, value) for key, value in classdict.items()
if key[0] != '_' and not
hasattr(value, '__call__') and not
isinstance(value, property)])
if _enum:
classdict = {key: value for key, value in classdict.items()
if key[0] == '_' or hasattr(value, '__call__') or
isinstance(value, property)}
classdict['_enum'] = _enum
return type.__new__(cls, name, bases, classdict)
def __getattr__(cls, name):
return cls._enum[name]
def __dir__(cls):
res = dir(type(cls)) + list(cls.__dict__.keys())
if cls is not GenericBitMask:
res.extend(cls._enum)
return res
class GenericBitMask(metaclass=MetaBitMask):
"""Base class for enums that use bitmask values."""
def __init__(self, bitmask=None):
"""Create a GenericBitMask with the optional parameter below.
Args:
bitmask: Bitmask value.
"""
self.bitmask = bitmask
self._enum = {}
def __str__(self):
return "{}".format(self.bitmask)
def __repr__(self):
return "{}({})".format(type(self).__name__, self.bitmask)
@property
def names(self):
"""List of selected enum names.
Returns:
list: Enum names.
"""
result = []
for key, value in self.iteritems():
if value & self.bitmask:
result.append(key)
return result
def iteritems(self):
"""Create a generator for attributes' name-value pairs.
Returns:
generator: Attributes' (name, value) tuples.
"""
for key, value in self._enum.items():
yield (key, value)
|