yardstick/network_services/vnf_generic/vnf/tg_rfc2544_ixia.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.common import utils
  18 from yardstick.network_services.libs.ixia_libs.ixnet import ixnet_api
  19 from yardstick.network_services.vnf_generic.vnf.sample_vnf import SampleVNFTrafficGen
  20 from yardstick.network_services.vnf_generic.vnf.sample_vnf import ClientResourceHelper
  21 from yardstick.network_services.vnf_generic.vnf.sample_vnf import Rfc2544ResourceHelper
  22
  23
  24 LOG = logging.getLogger(__name__)
  25
  26 WAIT_AFTER_CFG_LOAD = 10
  27 WAIT_FOR_TRAFFIC = 30
  28
  29
  30 class IxiaRfc2544Helper(Rfc2544ResourceHelper):
  31
  32     def is_done(self):
  33         return self.latency and self.iteration.value > 10
  34
  35
  36 class IxiaResourceHelper(ClientResourceHelper):
  37
  38     LATENCY_TIME_SLEEP = 120
  39
  40     def __init__(self, setup_helper, rfc_helper_type=None):
  41         super(IxiaResourceHelper, self).__init__(setup_helper)
  42         self.scenario_helper = setup_helper.scenario_helper
  43
  44         self.client = ixnet_api.IxNextgen()
  45
  46         if rfc_helper_type is None:
  47             rfc_helper_type = IxiaRfc2544Helper
  48
  49         self.rfc_helper = rfc_helper_type(self.scenario_helper)
  50         self.uplink_ports = None
  51         self.downlink_ports = None
  52         self._connect()
  53
  54     def _connect(self, client=None):
  55         self.client.connect(self.vnfd_helper)
  56
  57     def get_stats(self, *args, **kwargs):
  58         return self.client.get_statistics()
  59
  60     def stop_collect(self):
  61         self._terminated.value = 1
  62
  63     def generate_samples(self, ports, duration):
  64         stats = self.get_stats()
  65
  66         samples = {}
  67         # this is not DPDK port num, but this is whatever number we gave
  68         # when we selected ports and programmed the profile
  69         for port_num in ports:
  70             try:
  71                 # reverse lookup port name from port_num so the stats dict is descriptive
  72                 intf = self.vnfd_helper.find_interface_by_port(port_num)
  73                 port_name = intf['name']
  74                 avg_latency = stats['Store-Forward_Avg_latency_ns'][port_num]
  75                 min_latency = stats['Store-Forward_Min_latency_ns'][port_num]
  76                 max_latency = stats['Store-Forward_Max_latency_ns'][port_num]
  77                 samples[port_name] = {
  78                     'rx_throughput_kps': float(stats['Rx_Rate_Kbps'][port_num]),
  79                     'tx_throughput_kps': float(stats['Tx_Rate_Kbps'][port_num]),
  80                     'rx_throughput_mbps': float(stats['Rx_Rate_Mbps'][port_num]),
  81                     'tx_throughput_mbps': float(stats['Tx_Rate_Mbps'][port_num]),
  82                     'in_packets': int(stats['Valid_Frames_Rx'][port_num]),
  83                     'out_packets': int(stats['Frames_Tx'][port_num]),
  84                     'RxThroughput': float(stats['Valid_Frames_Rx'][port_num]) / duration,
  85                     'TxThroughput': float(stats['Frames_Tx'][port_num]) / duration,
  86                     'Store-Forward_Avg_latency_ns': utils.safe_cast(avg_latency, int, 0),
  87                     'Store-Forward_Min_latency_ns': utils.safe_cast(min_latency, int, 0),
  88                     'Store-Forward_Max_latency_ns': utils.safe_cast(max_latency, int, 0)
  89                 }
  90             except IndexError:
  91                 pass
  92
  93         return samples
  94
  95     def _initialize_client(self):
  96         """Initialize the IXIA IxNetwork client and configure the server"""
  97         self.client.clear_config()
  98         self.client.assign_ports()
  99         self.client.create_traffic_model()
 100
 101     def run_traffic(self, traffic_profile, *args):
 102         if self._terminated.value:
 103             return
 104
 105         min_tol = self.rfc_helper.tolerance_low
 106         max_tol = self.rfc_helper.tolerance_high
 107         default = "00:00:00:00:00:00"
 108
 109         self._build_ports()
 110         self._initialize_client()
 111
 112         mac = {}
 113         for port_name in self.vnfd_helper.port_pairs.all_ports:
 114             intf = self.vnfd_helper.find_interface(name=port_name)
 115             virt_intf = intf["virtual-interface"]
 116             # we only know static traffic id by reading the json
 117             # this is used by _get_ixia_trafficrofile
 118             port_num = self.vnfd_helper.port_num(intf)
 119             mac["src_mac_{}".format(port_num)] = virt_intf.get("local_mac", default)
 120             mac["dst_mac_{}".format(port_num)] = virt_intf.get("dst_mac", default)
 121
 122         try:
 123             while not self._terminated.value:
 124                 first_run = traffic_profile.execute_traffic(
 125                     self, self.client, mac)
 126                 self.client_started.value = 1
 127                 # pylint: disable=unnecessary-lambda
 128                 utils.wait_until_true(lambda: self.client.is_traffic_stopped(),
 129                                       timeout=traffic_profile.config.duration * 2)
 130                 samples = self.generate_samples(traffic_profile.ports,
 131                                                 traffic_profile.config.duration)
 132
 133                 completed, samples = traffic_profile.get_drop_percentage(
 134                     samples, min_tol, max_tol, first_run=first_run)
 135                 self._queue.put(samples)
 136
 137                 if completed:
 138                     self._terminated.value = 1
 139
 140         except Exception:  # pylint: disable=broad-except
 141             LOG.exception('Run Traffic terminated')
 142
 143         self._terminated.value = 1
 144
 145     def collect_kpi(self):
 146         self.rfc_helper.iteration.value += 1
 147         return super(IxiaResourceHelper, self).collect_kpi()
 148
 149
 150 class IxiaTrafficGen(SampleVNFTrafficGen):
 151
 152     APP_NAME = 'Ixia'
 153
 154     def __init__(self, name, vnfd, task_id, setup_env_helper_type=None,
 155                  resource_helper_type=None):
 156         if resource_helper_type is None:
 157             resource_helper_type = IxiaResourceHelper
 158         super(IxiaTrafficGen, self).__init__(
 159             name, vnfd, task_id, setup_env_helper_type, resource_helper_type)
 160         self._ixia_traffic_gen = None
 161         self.ixia_file_name = ''
 162         self.vnf_port_pairs = []
 163
 164     def _check_status(self):
 165         pass
 166
 167     def terminate(self):
 168         self.resource_helper.stop_collect()
 169         super(IxiaTrafficGen, self).terminate()