|
1
|
|
|
"""Deal with OpenFlow 1.0 specificities related to flows.""" |
|
2
|
1 |
|
from pyof.v0x01.common.action import ActionOutput as OFActionOutput |
|
3
|
1 |
|
from pyof.v0x01.common.action import ActionVlanVid as OFActionVlanVid |
|
4
|
1 |
|
from pyof.v0x01.common.flow_match import Match as OFMatch |
|
5
|
1 |
|
from pyof.v0x01.controller2switch.flow_mod import FlowMod |
|
6
|
|
|
|
|
7
|
1 |
|
from napps.kytos.of_core.flow import (ActionBase, ActionFactoryBase, FlowBase, |
|
8
|
|
|
FlowStats, MatchBase, PortStats) |
|
9
|
|
|
|
|
10
|
1 |
|
__all__ = ('ActionOutput', 'ActionSetVlan', 'Action', 'Flow', 'FlowStats', |
|
11
|
|
|
'PortStats') |
|
12
|
|
|
|
|
13
|
|
|
|
|
14
|
1 |
|
class Match(MatchBase): |
|
15
|
|
|
"""High-level Match for OpenFlow 1.0.""" |
|
16
|
|
|
|
|
17
|
1 |
|
@classmethod |
|
18
|
|
|
def from_of_match(cls, of_match): |
|
19
|
|
|
"""Return an instance from a pyof Match.""" |
|
20
|
|
|
match = cls(in_port=of_match.in_port, |
|
21
|
|
|
dl_src=of_match.dl_src.value, |
|
22
|
|
|
dl_dst=of_match.dl_dst.value, |
|
23
|
|
|
dl_vlan=of_match.dl_vlan, |
|
24
|
|
|
dl_vlan_pcp=of_match.dl_vlan_pcp, |
|
25
|
|
|
dl_type=of_match.dl_type, |
|
26
|
|
|
nw_proto=of_match.nw_proto, |
|
27
|
|
|
nw_src=of_match.nw_src.value, |
|
28
|
|
|
nw_dst=of_match.nw_dst.value, |
|
29
|
|
|
tp_src=of_match.tp_src, |
|
30
|
|
|
tp_dst=of_match.tp_dst) |
|
31
|
|
|
return match |
|
32
|
|
|
|
|
33
|
1 |
|
def as_of_match(self): |
|
34
|
|
|
"""Return a pyof Match instance with current contents.""" |
|
35
|
1 |
|
match = OFMatch() |
|
36
|
1 |
|
for field, value in self.__dict__.items(): |
|
37
|
1 |
|
if value is not None: |
|
38
|
1 |
|
setattr(match, field, value) |
|
39
|
1 |
|
return match |
|
40
|
|
|
|
|
41
|
|
|
|
|
42
|
1 |
View Code Duplication |
class ActionOutput(ActionBase): |
|
|
|
|
|
|
43
|
|
|
"""Action with an output port.""" |
|
44
|
|
|
|
|
45
|
1 |
|
def __init__(self, port): |
|
46
|
|
|
"""Require an output port. |
|
47
|
|
|
|
|
48
|
|
|
Args: |
|
49
|
|
|
port (int): Specific port number. |
|
50
|
|
|
""" |
|
51
|
|
|
self.port = port |
|
52
|
|
|
self.action_type = 'output' |
|
53
|
|
|
|
|
54
|
1 |
|
@classmethod |
|
55
|
|
|
def from_of_action(cls, of_action): |
|
56
|
|
|
"""Return a high-level ActionOuput instance from pyof ActionOutput.""" |
|
57
|
|
|
return ActionOutput(port=of_action.port.value) |
|
58
|
|
|
|
|
59
|
1 |
|
def as_of_action(self): |
|
60
|
|
|
"""Return a pyof ActionOuput instance.""" |
|
61
|
|
|
return OFActionOutput(port=self.port) |
|
62
|
|
|
|
|
63
|
|
|
|
|
64
|
1 |
|
class ActionSetVlan(ActionBase): |
|
65
|
|
|
"""Action to set VLAN ID.""" |
|
66
|
|
|
|
|
67
|
1 |
|
def __init__(self, vlan_id): |
|
68
|
|
|
"""Require a VLAN ID.""" |
|
69
|
1 |
|
self.vlan_id = vlan_id |
|
70
|
1 |
|
self.action_type = 'set_vlan' |
|
71
|
|
|
|
|
72
|
1 |
|
@classmethod |
|
73
|
|
|
def from_of_action(cls, of_action): |
|
74
|
|
|
"""Return a high-level ActionSetVlan object from pyof ActionVlanVid.""" |
|
75
|
|
|
return cls(vlan_id=of_action.vlan_id.value) |
|
76
|
|
|
|
|
77
|
1 |
|
def as_of_action(self): |
|
78
|
|
|
"""Return a pyof ActionVlanVid instance.""" |
|
79
|
1 |
|
return OFActionVlanVid(vlan_id=self.vlan_id) |
|
80
|
|
|
|
|
81
|
|
|
|
|
82
|
1 |
|
class Action(ActionFactoryBase): |
|
83
|
|
|
"""An action to be executed once a flow is activated. |
|
84
|
|
|
|
|
85
|
|
|
This class behavies like a factory but has no "Factory" suffix for end-user |
|
86
|
|
|
usability issues. |
|
87
|
|
|
""" |
|
88
|
|
|
|
|
89
|
|
|
# Set v0x01 classes for action types and pyof classes |
|
90
|
1 |
|
_action_class = { |
|
91
|
|
|
'output': ActionOutput, |
|
92
|
|
|
'set_vlan': ActionSetVlan, |
|
93
|
|
|
OFActionOutput: ActionOutput, |
|
94
|
|
|
OFActionVlanVid: ActionSetVlan |
|
95
|
|
|
} |
|
96
|
|
|
|
|
97
|
|
|
|
|
98
|
1 |
|
class Flow(FlowBase): |
|
99
|
|
|
"""High-level flow representation for OpenFlow 1.0. |
|
100
|
|
|
|
|
101
|
|
|
This is a subclass that only deals with 1.0 flow actions. |
|
102
|
|
|
""" |
|
103
|
|
|
|
|
104
|
1 |
|
_action_factory = Action |
|
105
|
1 |
|
_flow_mod_class = FlowMod |
|
106
|
1 |
|
_match_class = Match |
|
107
|
|
|
|
|
108
|
1 |
|
@staticmethod |
|
109
|
|
|
def _get_of_actions(of_flow_stats): |
|
110
|
|
|
"""Return the pyof actions from pyof ``FlowStats.actions``.""" |
|
111
|
|
|
return of_flow_stats.actions |
|
112
|
|
|
|
|
113
|
1 |
|
def _as_of_flow_mod(self, command): |
|
114
|
|
|
"""Return pyof FlowMod with a ``command`` to add or delete a flow.""" |
|
115
|
1 |
|
flow_mod = super()._as_of_flow_mod(command) |
|
116
|
1 |
|
flow_mod.actions = [action.as_of_action() for action in self.actions] |
|
117
|
|
|
return flow_mod |
|
118
|
|
|
|