[dts] [PATCH V1 08/27] framework/pktgen: enable ixNetwork

[yufengmx <yufengx.mo@intel.com>]

create ixia_network package.

Signed-off-by: yufengmx <yufengx.mo@intel.com>
---
 framework/ixia_network/__init__.py | 182 +++++++++++++++++++++++++++++
 1 file changed, 182 insertions(+)
 create mode 100644 framework/ixia_network/__init__.py

diff --git a/framework/ixia_network/__init__.py b/framework/ixia_network/__init__.py
new file mode 100644
index 00000000..086fb5af
--- /dev/null
+++ b/framework/ixia_network/__init__.py
@@ -0,0 +1,182 @@
+# BSD LICENSE
+#
+# Copyright(c) 2010-2021 Intel Corporation. All rights reserved.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+#
+#   * Redistributions of source code must retain the above copyright
+#     notice, this list of conditions and the following disclaimer.
+#   * Redistributions in binary form must reproduce the above copyright
+#     notice, this list of conditions and the following disclaimer in
+#     the documentation and/or other materials provided with the
+#     distribution.
+#   * Neither the name of Intel Corporation nor the names of its
+#     contributors may be used to endorse or promote products derived
+#     from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+"""
+ixNetwork package
+"""
+import os
+import time
+import traceback
+from pprint import pformat
+
+from .ixnet import IxnetTrafficGenerator
+from .ixnet_config import IxiaNetworkConfig
+
+__all__ = [
+    "IxNetwork",
+]
+
+
+class IxNetwork(IxnetTrafficGenerator):
+    """
+    ixNetwork performance measurement class.
+    """
+
+    def __init__(self, name, config, logger):
+        self.NAME = name
+        self.logger = logger
+        ixiaRef = self.NAME
+        if ixiaRef not in config:
+            return
+        _config = config.get(ixiaRef, {})
+        self.ixiaVersion = _config.get("Version")
+        self.ports = _config.get("Ports")
+        ixia_ip = _config.get("IP")
+        rest_server_ip = _config.get("ixnet_api_server_ip")
+        self.max_retry = int(_config.get("max_retry") or '5') # times
+        self.logger.debug(locals())
+        rest_config = IxiaNetworkConfig(
+            ixia_ip,
+            rest_server_ip,
+            '11009',
+            [[ixia_ip, p.get('card'), p.get('port')] for p in self.ports],
+            )
+        super(IxNetwork, self).__init__(rest_config, logger)
+        self._traffic_list = []
+        self._result = None
+
+    @property
+    def OUTPUT_DIR(self):
+        # get dts output folder path
+        if self.logger.log_path.startswith(os.sep):
+            output_path = self.logger.log_path
+        else:
+            cur_path = os.sep.join(
+                os.path.realpath(__file__).split(os.sep)[:-2])
+            output_path = os.path.join(cur_path, self.logger.log_path)
+        if not os.path.exists(output_path):
+            os.makedirs(output_path)
+
+        return output_path
+
+    def get_ports(self):
+        """
+        get ixNetwork ports for dts `ports_info`
+        """
+        plist = []
+        for p in self.ports:
+            plist.append({
+                'type': 'ixia',
+                'pci': 'IXIA:%d.%d' % (p['card'], p['port']),
+                })
+        return plist
+
+    def send_ping6(self, pci, mac, ipv6):
+        return '64 bytes from'
+
+    def disconnect(self):
+        ''' quit from ixNetwork api server '''
+        self.tear_down()
+        msg = 'close ixNetwork session done !'
+        self.logger.info(msg)
+
+    def prepare_ixia_network_stream(self, traffic_list):
+        self._traffic_list = []
+        for txPort, rxPort, pcapFile, option in traffic_list:
+            stream = self.configure_streams(pcapFile, option.get('fields_config'))
+            tx_p = self.tg_vports[txPort]
+            rx_p = self.tg_vports[rxPort]
+            self._traffic_list.append((tx_p, rx_p, stream))
+
+    def start(self, options):
+        ''' start ixNetwork measurement '''
+        test_mode = options.get('method')
+        options['traffic_list'] = self._traffic_list
+        self.logger.debug(pformat(options))
+        if test_mode == 'rfc2544_dichotomy':
+            cnt = 0
+            while cnt < self.max_retry:
+                try:
+                    result = self.send_rfc2544_throughput(options)
+                    if result:
+                        break
+                except Exception as e:
+                    msg = "failed to run rfc2544".format(cnt)
+                    self.logger.error(msg)
+                    self.logger.error(traceback.format_exc())
+                cnt += 1
+                msg = "No.{} rerun ixNetwork rfc2544".format(cnt)
+                self.logger.warning(msg)
+                time.sleep(10)
+            else:
+                result = []
+        else:
+            msg = "not support measurement {}".format(test_mode)
+            self.logger.error(msg)
+            self._result = None
+            return None
+        self.logger.info('measure <{}> completed'.format(test_mode))
+        self.logger.info(result)
+        self._result = result
+        return result
+
+    def get_rfc2544_stat(self, port_list):
+        """
+        Get RX/TX packet statistics.
+        """
+        if not self._result:
+            return [0] * 3
+
+        result = self._result
+        _ixnet_stats = {}
+        for item in result:
+            port_id = int(item.get('Trial')) - 1
+            _ixnet_stats[port_id] = dict(item)
+        port_stat = _ixnet_stats.get(0, {})
+        rx_packets = float(port_stat.get('Agg Rx Count (frames)') or '0.0')
+        tx_packets = float(port_stat.get('Agg Tx Count (frames)') or '0.0')
+        rx_pps = float(port_stat.get('Agg Rx Throughput (fps)') or '0.0')
+        return tx_packets, rx_packets, rx_pps
+
+    def get_stats(self, ports, mode):
+        '''
+        get statistics of custom mode
+        '''
+        methods = {
+            'rfc2544': self.get_rfc2544_stat,
+        }
+        if mode not in list(methods.keys()):
+            msg = "not support mode <{0}>".format(mode)
+            raise Exception(msg)
+        # get custom mode stat
+        func = methods.get(mode)
+        stats = func(ports)
+
+        return stats
-- 
2.21.0