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
|
"""For the controller to send a packet out through the datapath."""
from copy import deepcopy
from pyof.v0x01.common.header import Header, Type
from pyof.v0x01.common.phy_port import Port
from pyof.v0x01.controller2switch.common import ListOfActions
from pyof.v0x01.foundation.base import GenericMessage
from pyof.v0x01.foundation.basic_types import BinaryData, UBInt16, UBInt32
from pyof.v0x01.foundation.exceptions import ValidationError
__all__ = ('PacketOut',)
# Classes
#: in_port valid virtual port values, for validation
_VIRT_IN_PORTS = (Port.OFPP_LOCAL, Port.OFPP_CONTROLLER, Port.OFPP_NONE)
class PacketOut(GenericMessage):
"""Send packet (controller -> datapath)."""
header = Header(message_type=Type.OFPT_PACKET_OUT)
buffer_id = UBInt32()
in_port = UBInt16(enum_ref=Port)
actions_len = UBInt16()
actions = ListOfActions()
data = BinaryData()
def __init__(self, xid=None, buffer_id=None, in_port=None,
actions_len=None, actions=None, data=b''):
"""The constructor just assings parameters to object attributes.
Args:
xid (int): xid of the message header.
buffer_id (int): ID assigned by datapath (-1 if none).
in_port (:class:`int`, :class:`.Port`): Packet's input port
(:attr:`.Port.OFPP_NONE` if none). Virtual ports OFPP_IN_PORT,
OFPP_TABLE, OFPP_NORMAL, OFPP_FLOOD, and OFPP_ALL cannot be
used as input port.
actions_len (int): Size of action array in bytes.
actions (ListOfActions): Actions (class ActionHeader).
data (bytes): Packet data. The length is inferred from the length
field in the header. (Only meaningful if buffer_id == -1).
"""
super().__init__(xid)
self.buffer_id = buffer_id
self.in_port = in_port
self.actions_len = actions_len
self.actions = [] if actions is None else actions
self.data = data
def validate(self):
"""Validate the entire message."""
if not super().is_valid():
raise ValidationError()
self._validate_in_port()
def is_valid(self):
"""Answer if this message is valid."""
try:
self.validate()
return True
except ValidationError:
return False
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**.
This class' unpack method is like the :meth:`.GenericMessage.unpack`
one, except for the ``actions`` attribute which has a length determined
by the ``actions_len`` attribute.
Args:
buff (bytes): Binary data package to be unpacked, without the
header.
offset (int): Where to begin unpacking.
"""
begin = offset
for attribute_name, class_attribute in self.get_class_attributes():
if type(class_attribute).__name__ != "Header":
attribute = deepcopy(class_attribute)
if attribute_name == 'actions':
length = self.actions_len.value
attribute.unpack(buff[begin:begin+length])
else:
attribute.unpack(buff, begin)
setattr(self, attribute_name, attribute)
begin += attribute.get_size()
def _validate_in_port(self):
port = self.in_port
valid = True
if isinstance(port, int) and (port < 1 or port >= Port.OFPP_MAX.value):
valid = False
elif isinstance(port, Port) and port not in _VIRT_IN_PORTS:
valid = False
if not valid:
raise ValidationError('{} is not a valid input port.'.format(port))
|