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 from __future__ import absolute_import
  16 import logging
  17
  18 from yardstick.benchmark.scenarios.networking.vnf_generic import find_relative_file
  19 from yardstick.network_services.vnf_generic.vnf.sample_vnf import SampleVNF, DpdkVnfSetupEnvHelper
  20 from yardstick.network_services.yang_model import YangModel
  21
  22 LOG = logging.getLogger(__name__)
  23
  24 # vFW should work the same on all systems, we can provide the binary
  25 FW_PIPELINE_COMMAND = """sudo {tool_path} -p {port_mask_hex} -f {cfg_file} -s {script}"""
  26
  27 FW_COLLECT_KPI = (r"""VFW TOTAL:[^p]+pkts_received"?:\s(\d+),[^p]+pkts_fw_forwarded"?:\s(\d+),"""
  28                   r"""[^p]+pkts_drop_fw"?:\s(\d+),\s""")
  29
  30
  31 class FWApproxSetupEnvHelper(DpdkVnfSetupEnvHelper):
  32
  33     APP_NAME = "vFW"
  34     CFG_CONFIG = "/tmp/vfw_config"
  35     CFG_SCRIPT = "/tmp/vfw_script"
  36     DEFAULT_CONFIG_TPL_CFG = "vfw.cfg"
  37     PIPELINE_COMMAND = FW_PIPELINE_COMMAND
  38     SW_DEFAULT_CORE = 5
  39     HW_DEFAULT_CORE = 2
  40     VNF_TYPE = "VFW"
  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)
  60         self.vfw_rules = None
  61
  62     def _start_vnf(self):
  63         yang_model_path = find_relative_file(self.scenario_helper.options['rules'],
  64                                              self.scenario_helper.task_path)
  65         yang_model = YangModel(yang_model_path)
  66         self.vfw_rules = yang_model.get_rules()
  67         super(FWApproxVnf, self)._start_vnf()