b2337bd629750d2d9c3e82ebc45d131b88b6be2f
[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, bidir=False):
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._bidir = bidir
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             flow1 = add_ports_to_flow(_FLOW_TEMPLATE, phy1_number,
67                                       vport1_number)
68             flow2 = add_ports_to_flow(_FLOW_TEMPLATE, vport2_number,
69                                       phy2_number)
70             self._vswitch.add_flow(bridge, flow1)
71             self._vswitch.add_flow(bridge, flow2)
72
73             if self._bidir:
74                 flow3 = add_ports_to_flow(_FLOW_TEMPLATE, phy2_number,
75                                           vport2_number)
76                 flow4 = add_ports_to_flow(_FLOW_TEMPLATE, vport1_number,
77                                           phy1_number)
78                 self._vswitch.add_flow(bridge, flow3)
79                 self._vswitch.add_flow(bridge, flow4)
80
81         except:
82             self._vswitch.stop()
83             raise
84
85     def stop(self):
86         """Tears down the switch created in setup().
87         """
88         self._logger.debug('Stop using ' + str(self._vswitch_class))
89         self._vswitch.stop()
90
91     def __enter__(self):
92         self.setup()
93
94     def __exit__(self, type_, value, traceback):
95         self.stop()
96
97     def get_vswitch(self):
98         """See IVswitchController for description
99         """
100         return self._vswitch
101
102     def get_ports_info(self):
103         """See IVswitchController for description
104         """
105         self._logger.debug('get_ports_info  using ' + str(self._vswitch_class))
106         return self._vswitch.get_ports(settings.getValue('VSWITCH_BRIDGE_NAME'))
107
108     def dump_vswitch_flows(self):
109         """See IVswitchController for description
110         """
111         self._vswitch.dump_flows(settings.getValue('VSWITCH_BRIDGE_NAME'))