aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries/python/LoadBalancerUtil.py
blob: 77f641297339e6d0ff0206d0967b3242e0bd035a (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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
# Copyright (c) 2019 Intel and/or its affiliates.
# 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.

"""Loadbalancer util library."""

from socket import htonl
from ipaddress import ip_address
from resources.libraries.python.topology import NodeType, Topology
from resources.libraries.python.PapiExecutor import PapiSocketExecutor

class LoadBalancerUtil(object):
    """Basic Loadbalancer parameter configuration."""

    @staticmethod
    def vpp_lb_conf(node, **kwargs):
        """Config global parameters for loadbalancer.

        :param node: Node where the interface is.
        :param kwargs: Optional key-value arguments:

            ip4_src_addr: IPv4 address to be used as source for IPv4 traffic.
                          (str)
            ip6_src_addr: IPv6 address to be used as source for IPv6 traffic.
                          (str)
            flow_timeout: Time in seconds after which, if no packet is received
                          for a given flow, the flow is removed from the
                          established flow table. (int)
            buckets_per_core: Number of buckets *per worker thread* in the
                              established flow table (int)

        :type node: dict
        :type kwargs: dict
        :returns: Nothing.
        :raises ValueError: If the node has an unknown node type.
        """
        if node['type'] == NodeType.DUT:
            ip4_src_addr = ip_address(unicode(kwargs.pop('ip4_src_addr',
                                                         '255.255.255.255')))
            ip6_src_addr = ip_address(unicode(kwargs.pop('ip6_src_addr',\
                    'ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff')))
            flow_timeout = kwargs.pop('flow_timeout', 40)
            sticky_buckets_per_core = kwargs.pop('buckets_per_core', 1024)

            cmd = 'lb_conf'
            err_msg = 'Failed to set lb conf on host {host}'.format(
                host=node['host'])

            args = dict(ip4_src_address=str(ip4_src_addr),
                        ip6_src_address=str(ip6_src_addr),
                        sticky_buckets_per_core=sticky_buckets_per_core,
                        flow_timeout=flow_timeout)

            with PapiSocketExecutor(node) as papi_exec:
                papi_exec.add(cmd, **args).get_reply(err_msg)
        else:
            raise ValueError('Node {host} has unknown NodeType: "{type}"'
                             .format(host=node['host'], type=node['type']))

    @staticmethod
    def vpp_lb_add_del_vip(node, **kwargs):
        """Config vip for loadbalancer.

        :param node: Node where the interface is.
        :param kwargs: Optional key-value arguments:

            vip_addr: IPv4 address to be used as source for IPv4 traffic. (str)
            protocol: tcp or udp. (int)
            port: destination port. (int)
            encap: encap is ip4 GRE(0) or ip6 (1GRE) or L3DSR(2) or NAT4(3) or
                   NAT6(4). (int)
            dscp: dscp bit corresponding to VIP
            type: service type
            target_port: Pod's port corresponding to specific service
            node_port: Node's port
            new_len: Size of the new connections flow table used
                     for this VIP
            is_del: 1 if the VIP should be removed otherwise 0.

        :type node: dict
        :type kwargs: dict
        :returns: Nothing.
        :raises ValueError: If the node has an unknown node type.
        """
        if node['type'] == NodeType.DUT:
            vip_addr = kwargs.pop('vip_addr', '0.0.0.0')
            protocol = kwargs.pop('protocol', 255)
            port = kwargs.pop('port', 0)
            encap = kwargs.pop('encap', 0)
            dscp = kwargs.pop('dscp', 0)
            srv_type = kwargs.pop('srv_type', 0)
            target_port = kwargs.pop('target_port', 0)
            node_port = kwargs.pop('node_port', 0)
            new_len = kwargs.pop('new_len', 1024)
            is_del = kwargs.pop('is_del', 0)

            cmd = 'lb_add_del_vip'
            err_msg = 'Failed to add vip on host {host}'.format(
                host=node['host'])

            vip_addr = ip_address(unicode(vip_addr)).packed
            args = dict(pfx={'len': 128,
                             'address': {'un': {'ip4': vip_addr}, 'af': 0}},
                        protocol=protocol,
                        port=port,
                        encap=htonl(encap),
                        dscp=dscp,
                        type=srv_type,
                        target_port=target_port,
                        node_port=node_port,
                        new_flows_table_length=int(new_len),
                        is_del=is_del)

            with PapiSocketExecutor(node) as papi_exec:
                papi_exec.add(cmd, **args).get_reply(err_msg)
        else:
            raise ValueError('Node {host} has unknown NodeType: "{type}"'
                             .format(host=node['host'], type=node['type']))

    @staticmethod
    def vpp_lb_add_del_as(node, **kwargs):
        """Config AS for Loadbalancer.

        :param node: Node where the interface is.
        :param kwargs: Optional key-value arguments:

            vip_addr: IPv4 address to be used as source for IPv4 traffic. (str)
            protocol: tcp or udp. (int)
            port: destination port. (int)
            as_addr: The application server address. (str)
            is_del: 1 if the VIP should be removed otherwise 0. (int)
            is_flush: 1 if the sessions related to this AS should be flushed
                      otherwise 0. (int)

        :type node: dict
        :type kwargs: dict
        :returns: Nothing.
        :raises ValueError: If the node has an unknown node type.
        """
        if node['type'] == NodeType.DUT:
            cmd = 'lb_add_del_as'
            err_msg = 'Failed to add lb as on host {host}'.format(
                host=node['host'])

            vip_addr = kwargs.pop('vip_addr', '0.0.0.0')
            protocol = kwargs.pop('protocol', 255)
            port = kwargs.pop('port', 0)
            as_addr = kwargs.pop('as_addr', '0.0.0.0')
            is_del = kwargs.pop('is_del', 0)
            is_flush = kwargs.pop('is_flush', 0)

            vip_addr = ip_address(unicode(vip_addr)).packed
            as_addr = ip_address(unicode(as_addr)).packed

            args = dict(pfx={'len': 128,
                             'address': {'un': {'ip4': vip_addr}, 'af': 0}},
                        protocol=protocol,
                        port=port,
                        as_address={'un': {'ip4': as_addr}, 'af': 0},
                        is_del=is_del,
                        is_flush=is_flush)

            with PapiSocketExecutor(node) as papi_exec:
                papi_exec.add(cmd, **args).get_reply(err_msg)
        else:
            raise ValueError('Node {host} has unknown NodeType: "{type}"'
                             .format(host=node['host'], type=node['type']))

    @staticmethod
    def vpp_lb_add_del_intf_nat4(node, **kwargs):
        """Enable/disable NAT4 feature on the interface.

        :param node: Node where the interface is.
        :param kwargs: Optional key-value arguments:

            is_add: true if add, false if delete. (bool)
            interface: software index of the interface. (int)

        :type node: dict
        :type kwargs: dict
        :returns: Nothing.
        :raises ValueError: If the node has an unknown node type.
        """
        if node['type'] == NodeType.DUT:
            cmd = 'lb_add_del_intf_nat4'
            err_msg = 'Failed to add interface nat4 on host {host}'.format(
                host=node['host'])

            is_add = kwargs.pop('is_add', True)
            interface = kwargs.pop('interface', 0)
            sw_if_index = Topology.get_interface_sw_index(node, interface)
            args = dict(is_add=is_add, sw_if_index=sw_if_index)

            with PapiSocketExecutor(node) as papi_exec:
                papi_exec.add(cmd, **args).get_reply(err_msg)
        else:
            raise ValueError('Node {host} has unknown NodeType: "{type}"'
                             .format(host=node['host'], type=node['type']))