Flow Classification extension
[vswitchperf.git] / core / vswitch_controller_pvp.py
1 # Copyright 2015 Intel Corporation.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #   http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 """VSwitch controller for Physical to VM to Physical deployment
16 """
17
18 import logging
19
20 from core.vswitch_controller import IVswitchController
21 from vswitches.utils import add_ports_to_flow
22 from conf import settings
23
24 _FLOW_TEMPLATE = {
25     'idle_timeout': '0'
26 }
27
28 class VswitchControllerPVP(IVswitchController):
29     """VSwitch controller for PVP deployment scenario.
30
31     Attributes:
32         _vswitch_class: The vSwitch class to be used.
33         _vswitch: The vSwitch object controlled by this controller
34         _deployment_scenario: A string describing the scenario to set-up in the
35             constructor.
36     """
37     def __init__(self, vswitch_class, traffic):
38         """Initializes up the prerequisites for the PVP deployment scenario.
39
40         :vswitch_class: the vSwitch class to be used.
41         """
42         self._logger = logging.getLogger(__name__)
43         self._vswitch_class = vswitch_class
44         self._vswitch = vswitch_class()
45         self._deployment_scenario = "PVP"
46         self._traffic = traffic.copy()
47         self._logger.debug('Creation using ' + str(self._vswitch_class))
48
49     def setup(self):
50         """ Sets up the switch for pvp
51         """
52         self._logger.debug('Setup using ' + str(self._vswitch_class))
53
54         try:
55             self._vswitch.start()
56
57             bridge = settings.getValue('VSWITCH_BRIDGE_NAME')
58             self._vswitch.add_switch(bridge)
59
60             (_, phy1_number) = self._vswitch.add_phy_port(bridge)
61             (_, phy2_number) = self._vswitch.add_phy_port(bridge)
62             (_, vport1_number) = self._vswitch.add_vport(bridge)
63             (_, vport2_number) = self._vswitch.add_vport(bridge)
64
65             self._vswitch.del_flow(bridge)
66
67             # configure flows according to the TC definition
68             flow_template = _FLOW_TEMPLATE.copy()
69             if self._traffic['flow_type'] == 'IP':
70                 flow_template.update({'dl_type':'0x0800', 'nw_src':self._traffic['l3']['srcip'],
71                                       'nw_dst':self._traffic['l3']['dstip']})
72
73             flow1 = add_ports_to_flow(flow_template, phy1_number,
74                                       vport1_number)
75             flow2 = add_ports_to_flow(flow_template, vport2_number,
76                                       phy2_number)
77             self._vswitch.add_flow(bridge, flow1)
78             self._vswitch.add_flow(bridge, flow2)
79
80             if self._traffic['bidir']:
81                 flow3 = add_ports_to_flow(flow_template, phy2_number,
82                                           vport2_number)
83                 flow4 = add_ports_to_flow(flow_template, vport1_number,
84                                           phy1_number)
85                 self._vswitch.add_flow(bridge, flow3)
86                 self._vswitch.add_flow(bridge, flow4)
87
88         except:
89             self._vswitch.stop()
90             raise
91
92     def stop(self):
93         """Tears down the switch created in setup().
94         """
95         self._logger.debug('Stop using ' + str(self._vswitch_class))
96         self._vswitch.stop()
97
98     def __enter__(self):
99         self.setup()
100
101     def __exit__(self, type_, value, traceback):
102         self.stop()
103
104     def get_vswitch(self):
105         """See IVswitchController for description
106         """
107         return self._vswitch
108
109     def get_ports_info(self):
110         """See IVswitchController for description
111         """
112         self._logger.debug('get_ports_info  using ' + str(self._vswitch_class))
113         return self._vswitch.get_ports(settings.getValue('VSWITCH_BRIDGE_NAME'))
114
115     def dump_vswitch_flows(self):
116         """See IVswitchController for description
117         """
118         self._vswitch.dump_flows(settings.getValue('VSWITCH_BRIDGE_NAME'))