core/pktfwd_controller.py

   1 # Copyright 2016 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 """Packet forwarder controller for Physical to Physical deployment
  16 """
  17
  18 import logging
  19 from conf import settings
  20
  21 class PktFwdController(object):
  22     """Packet forwarder controller for P2P deployment scenario.
  23
  24     Attributes:
  25         _pktfwd_class: The packet forwarder class to be used.
  26         _pktfwd: The packet forwarder object controlled by this controller
  27     """
  28     def __init__(self, deployment, pktfwd_class):
  29         """Initializes up the prerequisites for the P2P deployment scenario.
  30
  31         :vswitch_class: the vSwitch class to be used.
  32         """
  33         self._deployment = deployment
  34         self._logger = logging.getLogger(__name__)
  35         self._pktfwd_class = pktfwd_class
  36         self._pktfwd = pktfwd_class(guest=True if deployment == "pvp" and
  37                                     settings.getValue('VNF') != "QemuPciPassthrough" else False)
  38         self._logger.debug('Creation using ' + str(self._pktfwd_class))
  39
  40     def setup(self):
  41         """Sets up the packet forwarder for p2p.
  42         """
  43         self._logger.debug('Setup using ' + str(self._pktfwd_class))
  44
  45         try:
  46             self._pktfwd.start()
  47         except:
  48             self._pktfwd.stop()
  49             raise
  50
  51     def setup_for_guest(self):
  52         """Sets up the packet forwarder for pvp.
  53         """
  54         self._logger.debug('Setup using ' + str(self._pktfwd_class))
  55
  56         try:
  57             self._pktfwd.start_for_guest()
  58         except:
  59             self._pktfwd.stop()
  60             raise
  61
  62     def stop(self):
  63         """Tears down the packet forwarder created in setup().
  64         """
  65         self._logger.debug('Stop using ' + str(self._pktfwd_class))
  66         self._pktfwd.stop()
  67
  68     def __enter__(self):
  69         if self._deployment.find("p2p") == 0:
  70             self.setup()
  71         elif self._deployment == "pvp" and settings.getValue('VNF') != "QemuPciPassthrough":
  72             self.setup_for_guest()
  73
  74     def __exit__(self, type_, value, traceback):
  75         if self._deployment.find("p2p") == 0:
  76             self.stop()
  77         elif self._deployment == "pvp" and settings.getValue('VNF') != "QemuPciPassthrough":
  78             self.stop()
  79
  80     def get_pktfwd(self):
  81         """Get the controlled packet forwarder
  82
  83         :return: The controlled IPktFwd
  84         """
  85         return self._pktfwd
  86
  87     def dump_vswitch_flows(self):
  88         """ Dumps flows from vswitch
  89         """
  90         raise NotImplementedError(
  91             "The PktFwdController does not implement the "
  92             "\"dump_vswitch_flows\" function.")