ceb29aa906a39fdef0638c02727b55930aaa3236
[vswitchperf.git] / core / vswitch_controller_p2p.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 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
23 _FLOW_TEMPLATE = {
24     'idle_timeout': '0'
25 }
26 BRIDGE_NAME = 'br0'
27
28 class VswitchControllerP2P(IVswitchController):
29     """VSwitch controller for P2P 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):
38         """Initializes up the prerequisites for the P2P 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 = "P2P"
46         self._logger.debug('Creation using ' + str(self._vswitch_class))
47
48     def setup(self):
49         """Sets up the switch for p2p.
50         """
51         self._logger.debug('Setup using ' + str(self._vswitch_class))
52
53         try:
54             self._vswitch.start()
55
56             self._vswitch.add_switch(BRIDGE_NAME)
57
58             (_, phy1_number) = self._vswitch.add_phy_port(BRIDGE_NAME)
59             (_, phy2_number) = self._vswitch.add_phy_port(BRIDGE_NAME)
60
61             self._vswitch.del_flow(BRIDGE_NAME)
62             flow = add_ports_to_flow(_FLOW_TEMPLATE, phy1_number, phy2_number)
63             self._vswitch.add_flow(BRIDGE_NAME, flow)
64
65         except:
66             self._vswitch.stop()
67             raise
68
69     def stop(self):
70         """Tears down the switch created in setup().
71         """
72         self._logger.debug('Stop using ' + str(self._vswitch_class))
73         self._vswitch.stop()
74
75     def __enter__(self):
76         self.setup()
77
78     def __exit__(self, type_, value, traceback):
79         self.stop()
80
81     def get_vswitch(self):
82         """See IVswitchController for description
83         """
84         return self._vswitch
85
86     def get_ports_info(self):
87         """See IVswitchController for description
88         """
89         self._logger.debug('get_ports_info  using ' + str(self._vswitch_class))
90         return self._vswitch.get_ports(BRIDGE_NAME)