Merge "Bugfix: HA kill process recovery has a conflict"
[yardstick.git] / yardstick / network_services / vnf_generic / vnf / vfw_vnf.py
1 # Copyright (c) 2016-2017 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 import logging
16
17 from yardstick.network_services.vnf_generic.vnf.sample_vnf import SampleVNF
18 from yardstick.network_services.vnf_generic.vnf.acl_vnf import AclApproxSetupEnvSetupEnvHelper
19
20 LOG = logging.getLogger(__name__)
21
22 # vFW should work the same on all systems, we can provide the binary
23 FW_PIPELINE_COMMAND = "sudo {tool_path} -p {port_mask_hex} -f {cfg_file} -s {script} {hwlb}"
24
25 FW_COLLECT_KPI = (r"""VFW TOTAL:[^p]+pkts_received"?:\s(\d+),[^p]+pkts_fw_forwarded"?:\s(\d+),"""
26                   r"""[^p]+pkts_drop_fw"?:\s(\d+),\s""")
27
28
29 class FWApproxSetupEnvHelper(AclApproxSetupEnvSetupEnvHelper):
30
31     APP_NAME = "vFW"
32     CFG_CONFIG = "/tmp/vfw_config"
33     CFG_SCRIPT = "/tmp/vfw_script"
34     DEFAULT_CONFIG_TPL_CFG = "vfw.cfg"
35     PIPELINE_COMMAND = FW_PIPELINE_COMMAND
36     SW_DEFAULT_CORE = 5
37     HW_DEFAULT_CORE = 2
38     VNF_TYPE = "VFW"
39     RULE_CMD = "vfw"
40     DEFAULT_FWD_ACTIONS = ["accept", "count", "conntrack"]
41
42
43 class FWApproxVnf(SampleVNF):
44
45     APP_NAME = "vFW"
46     APP_WORD = 'vfw'
47     COLLECT_KPI = FW_COLLECT_KPI
48
49     COLLECT_MAP = {
50         'packets_in': 1,
51         'packets_fwd': 2,
52         'packets_dropped': 3,
53     }
54
55     def __init__(self, name, vnfd, setup_env_helper_type=None, resource_helper_type=None):
56         if setup_env_helper_type is None:
57             setup_env_helper_type = FWApproxSetupEnvHelper
58
59         super(FWApproxVnf, self).__init__(name, vnfd, setup_env_helper_type, resource_helper_type)