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
|
# Copyright (c) 2016-2017 Intel Corporation
#
# 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.
from __future__ import absolute_import
import logging
import time
from yardstick.network_services.vnf_generic.vnf.sample_vnf import SampleVNF, DpdkVnfSetupEnvHelper
LOG = logging.getLogger(__name__)
# CGNAPT should work the same on all systems, we can provide the binary
CGNAPT_PIPELINE_COMMAND = 'sudo {tool_path} -p {port_mask_hex} -f {cfg_file} -s {script} {hwlb}'
WAIT_FOR_STATIC_NAPT = 4
CGNAPT_COLLECT_KPI = r"""\
CG-NAPT(.*\n)*\
Received\s(\d+),\
Missed\s(\d+),\
Dropped\s(\d+),\
Translated\s(\d+),\
ingress\
"""
class CgnaptApproxSetupEnvHelper(DpdkVnfSetupEnvHelper):
APP_NAME = "vCGNAPT"
CFG_CONFIG = "/tmp/cgnapt_config"
CFG_SCRIPT = "/tmp/cgnapt_script"
DEFAULT_CONFIG_TPL_CFG = "cgnat.cfg"
PIPELINE_COMMAND = CGNAPT_PIPELINE_COMMAND
SW_DEFAULT_CORE = 6
HW_DEFAULT_CORE = 3
VNF_TYPE = "CGNAPT"
@staticmethod
def _generate_ip_from_pool(ip):
ip_parts = ip.split('.')
assert len(ip_parts) == 4
iter1 = (str(n) for n in range(int(ip_parts[2]), 256))
for ip_parts[2] in iter1:
yield '.'.join(ip_parts)
@staticmethod
def _update_cgnat_script_file(ip_pipeline_cfg, mcpi):
pipeline_config_str = str(ip_pipeline_cfg)
input_cmds = '\n'.join(mcpi)
icmp_flag = 'link 0 down' in input_cmds
if icmp_flag:
pipeline_config_str = ''
return '\n'.join([pipeline_config_str, input_cmds])
def scale(self, flavor=""):
raise NotImplementedError
def _get_cgnapt_config(self):
# fixme: Get private port and gateway from port list
uplink_ports = self.vnfd_helper.port_pairs.uplink_ports
return \
[self.vnfd_helper.find_interface(name=intf)["virtual-interface"]['dst_ip']
for intf in uplink_ports]
class CgnaptApproxVnf(SampleVNF):
APP_NAME = "vCGNAPT"
APP_WORD = 'cgnapt'
COLLECT_KPI = CGNAPT_COLLECT_KPI
COLLECT_MAP = {
"packets_in": 2,
"packets_fwd": 5,
"packets_dropped": 4,
}
def __init__(self, name, vnfd, setup_env_helper_type=None, resource_helper_type=None):
if setup_env_helper_type is None:
setup_env_helper_type = CgnaptApproxSetupEnvHelper
super(CgnaptApproxVnf, self).__init__(name, vnfd, setup_env_helper_type,
resource_helper_type)
def _vnf_up_post(self):
super(CgnaptApproxVnf, self)._vnf_up_post()
if self.scenario_helper.options.get('napt', 'static') != 'static':
return
flow = self.scenario_helper.all_options.get('flow', {})
public_ip = flow.get('public_ip', ['152.16.40.10']).pop()
ip_iter = self.setup_helper._generate_ip_from_pool(public_ip)
gw_ips = self.setup_helper._get_cgnapt_config()
if self.scenario_helper.vnf_cfg.get("lb_config", "SW") == 'HW':
pipeline = self.setup_helper.HW_DEFAULT_CORE
offset = 3
else:
pipeline = self.setup_helper.SW_DEFAULT_CORE - 1
offset = 0
worker_threads = int(self.scenario_helper.vnf_cfg["worker_threads"])
# p <pipeline id> entry addm <prv_ipv4/6> prvport> <pub_ip> <pub_port> <phy_port> <ttl>
# <no_of_entries> <end_prv_port> <end_pub_port>
cmd_template = "p {0} entry addm {1} 1 {2} 1 0 32 65535 65535 65535"
for gw, ip in zip(gw_ips, ip_iter):
cmd = cmd_template.format(pipeline, gw, ip)
pipeline += worker_threads
pipeline += offset
LOG.info(cmd)
self.vnf_execute(cmd)
time.sleep(WAIT_FOR_STATIC_NAPT)
|
