IXIA traffic generator

Change-Id: I09bcb3f2c4b945283070d442589d3bf00468abbc
Signed-off-by: Deepak S <deepak.s@linux.intel.com>
Signed-off-by: Edward MacGillivray <edward.s.macgillivray@intel.com>
Signed-off-by: Ross Brattain <ross.b.brattain@intel.com>

2 files changed, 449 insertions, 0 deletions

diff --git a/yardstick/network_services/traffic_profile/http_ixload.py b/yardstick/network_services/traffic_profile/http_ixload.py
new file mode 100644
index 000000000..8a4f97f04
--- /dev/null
+++ b/yardstick/network_services/traffic_profile/http_ixload.py
@@ -0,0 +1,294 @@
+# 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
+from __future__ import print_function
+
+import sys
+import os
+import logging
+
+# ixload uses its own py2. So importing jsonutils fails. So adding below
+# workaround to support call from yardstick
+try:
+    from oslo_serialization import jsonutils
+except ImportError:
+    import json as jsonutils
+
+from yardstick.common.utils import join_non_strings
+from yardstick.common.utils import ErrorClass
+
+try:
+    from IxLoad import IxLoad, StatCollectorUtils
+except ImportError:
+    IxLoad = ErrorClass
+    StatCollectorUtils = ErrorClass
+
+LOG = logging.getLogger(__name__)
+CSV_FILEPATH_NAME = 'IxL_statResults.csv'
+
+STATS_TO_GET = (
+    'HTTP_Client.csv',
+    'HTTP_Server.csv',
+    'L2-3 Stats for Client Ports.csv',
+    'L2-3 Stats for Server Ports.csv',
+    'IxLoad Detailed Report.html',
+    'IxLoad Detailed Report.pdf'
+)
+
+HTTP_CLIENT_STATS = [
+    ["HTTP Client", "TCP Connections Established", "kSum"],
+    ["HTTP Client", "TCP Connection Requests Failed", "kSum"],
+    ["HTTP Client", "HTTP Simulated Users", "kSum"],
+    ["HTTP Client", "HTTP Concurrent Connections", "kSum"],
+    ["HTTP Client", "HTTP Connections", "kSum"],
+    ["HTTP Client", "HTTP Transactions", "kSum"],
+    ["HTTP Client", "HTTP Connection Attempts", "kSum"]
+]
+
+HTTP_SERVER_STATS = [
+    ["HTTP Server", "TCP Connections Established", "kSum"],
+    ["HTTP Server", "TCP Connection Requests Failed", "kSum"]
+]
+
+
+INCOMING_STAT_RECORD_TEMPLATE = """
+=====================================
+INCOMING STAT RECORD >>> %s
+Len = %s
+%s
+%s
+=====================================
+"""
+
+INCOMING_STAT_INTERVAL_TEMPLATE = """
+=====================================
+Incoming stats: Time interval: %s
+Incoming stats: Time interval: %s
+=====================================
+"""
+
+
+class IXLOADHttpTest(object):
+
+    def __init__(self, test_input):
+        self.test_input = jsonutils.loads(test_input)
+        self.parse_run_test()
+        self.ix_load = None
+        self.stat_utils = None
+        self.remote_server = None
+        self.config_file = None
+        self.results_on_windows = None
+        self.result_dir = None
+        self.chassis = None
+        self.card = None
+        self.ports_to_reassign = None
+
+    @staticmethod
+    def format_ports_for_reassignment(ports):
+        formatted = [join_non_strings(';', p) for p in ports if len(p) == 3]
+        LOG.debug('for client ports:%s', os.linesep.join(formatted))
+        return formatted
+
+    def reassign_ports(self, test, repository, ports_to_reassign):
+        LOG.debug('ReassignPorts: %s %s', test, repository)
+
+        chassis_chain = repository.cget('chassisChain')
+        LOG.debug('chassischain: %s', chassis_chain)
+        client_ports = ports_to_reassign[0::2]
+        server_ports = ports_to_reassign[1::2]
+
+        client_ports = self.format_ports_for_reassignment(client_ports)
+        LOG.debug('Reassigning client ports: %s', client_ports)
+        server_ports = self.format_ports_for_reassignment(server_ports)
+        LOG.debug('Reassigning server ports: %s', server_ports)
+        ports_to_set = client_ports + server_ports
+
+        try:
+            LOG.debug('Reassigning ports: %s', ports_to_set)
+            test.setPorts(ports_to_set)
+        except Exception:
+            LOG.error('Error: Could not remap port assignment for: %s',
+                      ports_to_set)
+            self.ix_load.delete(repository)
+            self.ix_load.disconnect()
+            raise
+
+    @staticmethod
+    def stat_collector(*args):
+        LOG.debug(INCOMING_STAT_RECORD_TEMPLATE, args, len(args), args[0], args[1])
+
+    @staticmethod
+    def IxL_StatCollectorCommand(*args):
+        stats = args[1][3]
+        timestamp = args[1][1]
+        LOG.debug(INCOMING_STAT_INTERVAL_TEMPLATE, timestamp, stats)
+
+    @staticmethod
+    def set_results_dir(test_controller, results_on_windows):
+        """
+        If the folder doesn't exists on the Windows Client PC,
+        IxLoad will automatically create it.
+        """
+        try:
+            test_controller.setResultDir(results_on_windows)
+        except Exception:
+            LOG.error('Error creating results dir on Win: %s',
+                      results_on_windows)
+            raise
+
+    def load_config_file(self, config_file):
+        try:
+            LOG.debug(config_file)
+            repository = self.ix_load.new("ixRepository", name=config_file)
+            return repository
+        except Exception:
+            LOG.error('Error: IxLoad config file not found: %s', config_file)
+            raise
+
+    def start_http_test(self):
+        self.ix_load = IxLoad()
+
+        LOG.debug('--- ixLoad obj: %s', self.ix_load)
+        try:
+            self.ix_load.connect(self.remote_server)
+        except Exception:
+            raise
+
+        log_tag = "IxLoad-api"
+        log_name = "reprun"
+        logger = self.ix_load.new("ixLogger", log_tag, 1)
+        log_engine = logger.getEngine()
+        log_engine.setLevels(self.ix_load.ixLogger.kLevelDebug,
+                             self.ix_load.ixLogger.kLevelInfo)
+        log_engine.setFile(log_name, 2, 256, 1)
+
+        # Initialize stat collection utilities
+        self.stat_utils = StatCollectorUtils()
+
+        test_controller = self.ix_load.new("ixTestController", outputDir=1)
+
+        repository = self.load_config_file(self.config_file)
+
+        # Get the first test on the testList
+        test_name = repository.testList[0].cget("name")
+        test = repository.testList.getItem(test_name)
+
+        self.set_results_dir(test_controller, self.results_on_windows)
+
+        test.config(statsRequired=1, enableResetPorts=1, csvInterval=2,
+                    enableForceOwnership=True)
+
+        #  ---- Remap ports ----
+        try:
+            self.reassign_ports(test, repository, self.ports_to_reassign)
+        except Exception:
+            LOG.exception("Exception occurred during reassign_ports")
+
+        # -----------------------------------------------------------------------
+        # Set up stat Collection
+        # -----------------------------------------------------------------------
+        test_server_handle = test_controller.getTestServerHandle()
+        self.stat_utils.Initialize(test_server_handle)
+
+        # Clear any stats that may have been registered previously
+        self.stat_utils.ClearStats()
+
+        # Define the stats we would like to collect
+        self.stat_utils.AddStat(caption="Watch_Stat_1",
+                                statSourceType="HTTP Client",
+                                statName="TCP Connections Established",
+                                aggregationType="kSum",
+                                filterList={})
+
+        self.stat_utils.AddStat(caption="Watch_Stat_2",
+                                statSourceType="HTTP Client",
+                                statName="TCP Connection Requests Failed",
+                                aggregationType="kSum",
+                                filterList={})
+
+        self.stat_utils.AddStat(caption="Watch_Stat_3",
+                                statSourceType="HTTP Server",
+                                statName="TCP Connections Established",
+                                aggregationType="kSum",
+                                filterList={})
+
+        self.stat_utils.AddStat(caption="Watch_Stat_4",
+                                statSourceType="HTTP Server",
+                                statName="TCP Connection Requests Failed",
+                                aggregationType="kSum",
+                                filterList={})
+
+        self.stat_utils.StartCollector(self.IxL_StatCollectorCommand)
+
+        test_controller.run(test)
+        self.ix_load.waitForTestFinish()
+
+        test_controller.releaseConfigWaitFinish()
+
+        # Stop the collector (running in the tcl event loop)
+        self.stat_utils.StopCollector()
+
+        # Cleanup
+        test_controller.generateReport(detailedReport=1, format="PDF;HTML")
+        test_controller.releaseConfigWaitFinish()
+
+        self.ix_load.delete(test)
+        self.ix_load.delete(test_controller)
+        self.ix_load.delete(logger)
+        self.ix_load.delete(log_engine)
+
+        LOG.debug('Retrieving CSV stats from Windows Client PC ...')
+        for stat_file in STATS_TO_GET:
+            enhanced_stat_file = stat_file.replace('-', '')
+            enhanced_stat_file = enhanced_stat_file.replace(' ', '_')
+            enhanced_stat_file = enhanced_stat_file.replace('__', '_')
+
+            LOG.debug('Getting csv stat file: %s', stat_file)
+            src_file = os.path.join(self.results_on_windows, stat_file)
+            dst_file = os.path.join(self.result_dir, '_'.join(['ixLoad', enhanced_stat_file]))
+            self.ix_load.retrieveFileCopy(src_file, dst_file)
+
+        self.ix_load.disconnect()
+
+    def parse_run_test(self):
+        self.remote_server = self.test_input["remote_server"]
+        LOG.debug("remote tcl server: %s", self.remote_server)
+
+        self.config_file = self.test_input["ixload_cfg"]
+        LOG.debug("ixload config: %s", self.remote_server)
+
+        self.results_on_windows = 'C:/Results'
+        self.result_dir = self.test_input["result_dir"]
+        self.chassis = self.test_input["ixia_chassis"]
+        LOG.debug("remote ixia chassis: %s", self.chassis)
+
+        self.card = self.test_input["IXIA"]["card"]
+        self.ports_to_reassign = [
+            [self.chassis, self.card, port] for port in
+            self.test_input["IXIA"]["ports"]
+        ]
+
+        LOG.debug("Ports to be reassigned: %s", self.ports_to_reassign)
+
+
+def main(args):
+    # Get the args from cmdline and parse and run the test
+    test_input = "".join(args[1:])
+    if test_input:
+        ixload_obj = IXLOADHttpTest(test_input)
+        ixload_obj.start_http_test()
+
+if __name__ == '__main__':
+    main(sys.argv)
diff --git a/yardstick/network_services/traffic_profile/ixia_rfc2544.py b/yardstick/network_services/traffic_profile/ixia_rfc2544.py
new file mode 100644
index 000000000..5ba00180b
--- /dev/null
+++ b/yardstick/network_services/traffic_profile/ixia_rfc2544.py
@@ -0,0 +1,155 @@
+# 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 json
+
+from yardstick.network_services.traffic_profile.traffic_profile import \
+    TrexProfile
+
+LOG = logging.getLogger(__name__)
+
+
+class IXIARFC2544Profile(TrexProfile):
+    def _get_ixia_traffic_profile(self, profile_data, mac={},
+                                  xfile=None, static_traffic={}):
+        result = {}
+        if xfile:
+            with open(xfile, 'r') as stream:
+                try:
+                    static_traffic = json.load(stream)
+                except Exception as exc:
+                    LOG.debug(exc)
+
+        for traffickey, trafficvalue in static_traffic.items():
+            traffic = static_traffic[traffickey]
+            # outer_l2
+            index = 0
+            for key, value in profile_data[traffickey].items():
+                framesize = value['outer_l2']['framesize']
+                traffic['outer_l2']['framesize'] = framesize
+                traffic['framesPerSecond'] = True
+                traffic['bidir'] = False
+                traffic['outer_l2']['srcmac'] = \
+                    mac["src_mac_{}".format(traffic['id'])]
+                traffic['outer_l2']['dstmac'] = \
+                    mac["dst_mac_{}".format(traffic['id'])]
+                # outer_l3
+                if "outer_l3v6" in list(value.keys()):
+                    traffic['outer_l3'] = value['outer_l3v6']
+                    srcip4 = value['outer_l3v6']['srcip6']
+                    traffic['outer_l3']['srcip4'] = srcip4.split("-")[0]
+                    dstip4 = value['outer_l3v6']['dstip6']
+                    traffic['outer_l3']['dstip4'] = dstip4.split("-")[0]
+                else:
+                    traffic['outer_l3'] = value['outer_l3v4']
+                    srcip4 = value['outer_l3v4']['srcip4']
+                    traffic['outer_l3']['srcip4'] = srcip4.split("-")[0]
+                    dstip4 = value['outer_l3v4']['dstip4']
+                    traffic['outer_l3']['dstip4'] = dstip4.split("-")[0]
+
+                traffic['outer_l3']['type'] = key
+                # outer_l4
+                traffic['outer_l4'] = value['outer_l4']
+                index = index + 1
+            result.update({traffickey: traffic})
+
+        return result
+
+    def _ixia_traffic_generate(self, traffic_generator, traffic, ixia_obj):
+        for key, value in traffic.items():
+            if "public" in key or "private" in key:
+                traffic[key]["iload"] = str(self.rate)
+        ixia_obj.ix_update_frame(traffic)
+        ixia_obj.ix_update_ether(traffic)
+        # ixia_obj.ix_update_ipv4(traffic)
+        ixia_obj.ix_start_traffic()
+        self.tmp_drop = 0
+        self.tmp_throughput = 0
+
+    def execute(self, traffic_generator, ixia_obj, mac={}, xfile=None):
+        if self.first_run:
+            self.full_profile = {}
+            self.pg_id = 0
+            self.profile = 'private_1'
+            for key, value in self.params.items():
+                if "private" in key or "public" in key:
+                    self.profile_data = self.params[key]
+                    self.get_streams(self.profile_data)
+                    self.full_profile.update({key: self.profile_data})
+            traffic = \
+                self._get_ixia_traffic_profile(self.full_profile, mac, xfile)
+            self.max_rate = self.rate
+            self.min_rate = 0
+            self.get_multiplier()
+            self._ixia_traffic_generate(traffic_generator, traffic, ixia_obj)
+
+    def get_multiplier(self):
+        self.rate = round((self.max_rate + self.min_rate) / 2.0, 2)
+        multiplier = round(self.rate / self.pps, 2)
+        return str(multiplier)
+
+    def start_ixia_latency(self, traffic_generator, ixia_obj,
+                           mac={}, xfile=None):
+        traffic = self._get_ixia_traffic_profile(self.full_profile, mac)
+        self._ixia_traffic_generate(traffic_generator, traffic,
+                                    ixia_obj, xfile)
+
+    def get_drop_percentage(self, traffic_generator, samples, tol_min,
+                            tolerance, ixia_obj, mac={}, xfile=None):
+        status = 'Running'
+        drop_percent = 100
+        in_packets = sum([samples[iface]['in_packets'] for iface in samples])
+        out_packets = sum([samples[iface]['out_packets'] for iface in samples])
+        rx_throughput = \
+            sum([samples[iface]['RxThroughput'] for iface in samples])
+        tx_throughput = \
+            sum([samples[iface]['TxThroughput'] for iface in samples])
+        packet_drop = abs(out_packets - in_packets)
+        try:
+            drop_percent = round((packet_drop / float(out_packets)) * 100, 2)
+        except ZeroDivisionError:
+            LOG.info('No traffic is flowing')
+        samples['TxThroughput'] = round(tx_throughput / 1.0, 2)
+        samples['RxThroughput'] = round(rx_throughput / 1.0, 2)
+        samples['CurrentDropPercentage'] = drop_percent
+        samples['Throughput'] = self.tmp_throughput
+        samples['DropPercentage'] = self.tmp_drop
+        if drop_percent > tolerance and self.tmp_throughput == 0:
+            samples['Throughput'] = round(rx_throughput / 1.0, 2)
+            samples['DropPercentage'] = drop_percent
+        if self.first_run:
+            max_supported_rate = out_packets / 30.0
+            self.rate = max_supported_rate
+            self.first_run = False
+            if drop_percent <= tolerance:
+                status = 'Completed'
+        if drop_percent > tolerance:
+            self.max_rate = self.rate
+        elif drop_percent < tol_min:
+            self.min_rate = self.rate
+            if drop_percent >= self.tmp_drop:
+                self.tmp_drop = drop_percent
+                self.tmp_throughput = round((rx_throughput / 1.0), 2)
+                samples['Throughput'] = round(rx_throughput / 1.0, 2)
+                samples['DropPercentage'] = drop_percent
+        else:
+            samples['Throughput'] = round(rx_throughput / 1.0, 2)
+            samples['DropPercentage'] = drop_percent
+            return status, samples
+        self.get_multiplier()
+        traffic = self._get_ixia_traffic_profile(self.full_profile, mac, xfile)
+        self._ixia_traffic_generate(traffic_generator, traffic, ixia_obj)
+        return status, samples