From 4a80fac56ca122a34de5d03c0876e424cc7c6c56 Mon Sep 17 00:00:00 2001
From: ahothan <ahothan@cisco.com>
Date: Mon, 26 Nov 2018 15:52:42 -0800
Subject: NFVBENCH-113 Add direct support for trex cores as an cli/config
 option

Change-Id: Ic930b605a0eb821487ca4620bc7231794b9add08
Signed-off-by: ahothan <ahothan@cisco.com>
---
 nfvbench/cfg.default.yaml  |  9 ++++++++-
 nfvbench/traffic_client.py |  7 ++++++-
 nfvbench/traffic_server.py | 12 ++++++++----
 3 files changed, 22 insertions(+), 6 deletions(-)

(limited to 'nfvbench')

diff --git a/nfvbench/cfg.default.yaml b/nfvbench/cfg.default.yaml
index 3138420..5c791ab 100755
--- a/nfvbench/cfg.default.yaml
+++ b/nfvbench/cfg.default.yaml
@@ -229,7 +229,7 @@ traffic_generator:
         - name: trex-local
           tool: TRex
           ip: 127.0.0.1
-          cores: 3
+          cores: 4
           software_mode: false
           interfaces:
             - port: 0
@@ -240,6 +240,13 @@ traffic_generator:
               switch_port:
           intf_speed:
 
+# Simpler override for trex core count and mbuf multilier factor
+# if empty defaults to the one specified in generator_profile.cores
+cores:
+
+# mbuffer ratio to use for TRex (see TRex documentation for more details)
+mbuf_factor: 0.2
+
 # -----------------------------------------------------------------------------
 # These variables are not likely to be changed
 
diff --git a/nfvbench/traffic_client.py b/nfvbench/traffic_client.py
index c9daf40..093a02d 100755
--- a/nfvbench/traffic_client.py
+++ b/nfvbench/traffic_client.py
@@ -328,7 +328,12 @@ class GeneratorConfig(object):
         # copy over fields from the dict
         self.tool = gen_config.tool
         self.ip = gen_config.ip
-        self.cores = gen_config.get('cores', 1)
+        # overrides on config.cores and config.mbuf_factor
+        if config.cores:
+            self.cores = config.cores
+        else:
+            self.cores = gen_config.get('cores', 1)
+        self.mbuf_factor = config.mbuf_factor
         if gen_config.intf_speed:
             # interface speed is overriden from config
             self.intf_speed = bitmath.parse_string(gen_config.intf_speed.replace('ps', '')).bits
diff --git a/nfvbench/traffic_server.py b/nfvbench/traffic_server.py
index b2d8367..c3d4d14 100644
--- a/nfvbench/traffic_server.py
+++ b/nfvbench/traffic_server.py
@@ -35,8 +35,7 @@ class TRexTrafficServer(TrafficServer):
         self.trex_dir = os.path.join(trex_base_dir, contents[0])
 
     def run_server(self, generator_config, filename='/etc/trex_cfg.yaml'):
-        """
-        Runs TRex server for specified traffic profile.
+        """Run TRex server for specified traffic profile.
 
         :param traffic_profile: traffic profile object based on config file
         :param filename: path where to save TRex config file
@@ -46,10 +45,15 @@ class TRexTrafficServer(TrafficServer):
         vtep_vlan = generator_config.gen_config.get('vtep_vlan')
         sw_mode = "--software" if generator_config.software_mode else ""
         vlan_opt = "--vlan" if (generator_config.vlan_tagging or vtep_vlan) else ""
+        if generator_config.mbuf_factor:
+            mbuf_opt = "--mbuf-factor " + str(generator_config.mbuf_factor)
+        else:
+            mbuf_opt = ""
         subprocess.Popen(['nohup', '/bin/bash', '-c',
                           './t-rex-64 -i -c {} --iom 0 --no-scapy-server --close-at-end {} '
-                          '{} --cfg {} &> /tmp/trex.log & disown'.format(cores, sw_mode,
-                                                                         vlan_opt, cfg)],
+                          '{} {} --cfg {} &> /tmp/trex.log & disown'.format(cores, sw_mode,
+                                                                            vlan_opt,
+                                                                            mbuf_opt, cfg)],
                          cwd=self.trex_dir)
         LOG.info('TRex server is running...')
 
-- 
cgit 1.2.3-korg