aboutsummaryrefslogtreecommitdiffstats
path: root/core/pktfwd_controller.py
blob: 2b6a008c291533fc1a4f2a265c7e4596ad989df3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
# Copyright 2016-2018 Intel Corporation., Tieto
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Packet forwarder controller for Physical to Physical deployment
"""

import logging
from conf import settings

class PktFwdController(object):
    """Packet forwarder controller for P2P deployment scenario.

    Attributes:
        _pktfwd_class: The packet forwarder class to be used.
        _pktfwd: The packet forwarder object controlled by this controller
    """
    def __init__(self, deployment, pktfwd_class):
        """Initializes up the prerequisites for the P2P deployment scenario.

        :vswitch_class: the vSwitch class to be used.
        """
        self._deployment = deployment
        self._logger = logging.getLogger(__name__)
        self._pktfwd_class = pktfwd_class
        if 'DummyFWD' in settings.getValue("PKTFWD") or 'pc' in deployment:
            self._pktfwd = pktfwd_class()
        else:
            self._pktfwd = pktfwd_class(guest=True if deployment == "pvp" and
                                        settings.getValue('VNF') != "QemuPciPassthrough" else False)
        self._logger.debug('Creation using %s', str(self._pktfwd_class))

    def setup(self):
        """Sets up the packet forwarder for p2p.
        """
        self._logger.debug('Setup using %s', str(self._pktfwd_class))

        try:
            self._pktfwd.start()
        except:
            self._pktfwd.stop()
            raise

    def get_vswitch(self):
        """See IVswitchController for description
        """
        return self._pktfwd

    def setup_for_guest(self):
        """Sets up the packet forwarder for pvp.
        """
        self._logger.debug('Setup using %s', str(self._pktfwd_class))

        try:
            self._pktfwd.start_for_guest()
        except:
            self._pktfwd.stop()
            raise

    def stop(self):
        """Tears down the packet forwarder created in setup().
        """
        self._logger.debug('Stop using %s', str(self._pktfwd_class))
        self._pktfwd.stop()

    def __enter__(self):
        if self._deployment.find("p2p") == 0:
            self.setup()
        elif self._deployment == "pvp" and settings.getValue('VNF') != "QemuPciPassthrough":
            self.setup_for_guest()

    def __exit__(self, type_, value, traceback):
        if self._deployment.find("p2p") == 0:
            self.stop()
        elif self._deployment == "pvp" and settings.getValue('VNF') != "QemuPciPassthrough":
            self.stop()

    def get_pktfwd(self):
        """Get the controlled packet forwarder

        :return: The controlled IPktFwd
        """
        return self._pktfwd

    def dump_vswitch_connections(self):
        """ Dumps connections from vswitch
        """
        raise NotImplementedError(
            "The PktFwdController does not implement the "
            "\"dump_vswitch_connections\" function.")