summaryrefslogtreecommitdiffstats
path: root/nfvbench/traffic_server.py
diff options
context:
space:
mode:
authorahothan <ahothan@cisco.com>2017-07-28 17:08:46 -0700
committerahothan <ahothan@cisco.com>2017-07-31 12:34:00 -0700
commit04a7de082bd221eae3c7004f4e0b99dfa4f8be91 (patch)
treec9fb7beaedc80479772ba24c3b47c85d49c22f76 /nfvbench/traffic_server.py
parentb8f02ed4e72399840a93aceb02b8c53831bbe68a (diff)
Initial code drop from Cisco1.0.0
Change-Id: Ie2993886dc8e95c5f73ccdb871add8b96ffcc849 Signed-off-by: ahothan <ahothan@cisco.com>
Diffstat (limited to 'nfvbench/traffic_server.py')
-rw-r--r--nfvbench/traffic_server.py64
1 files changed, 64 insertions, 0 deletions
diff --git a/nfvbench/traffic_server.py b/nfvbench/traffic_server.py
new file mode 100644
index 0000000..05f20e5
--- /dev/null
+++ b/nfvbench/traffic_server.py
@@ -0,0 +1,64 @@
+# Copyright 2016 Cisco Systems, Inc. All rights reserved.
+#
+# 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 log import LOG
+import os
+import subprocess
+import yaml
+
+class TrafficServerException(Exception):
+ pass
+
+class TrafficServer(object):
+ """Base class for traffic servers."""
+
+class TRexTrafficServer(TrafficServer):
+ """Creates configuration file for TRex and runs server."""
+
+ def __init__(self, trex_base_dir='/opt/trex'):
+ contents = os.listdir(trex_base_dir)
+ # only one version of TRex should be supported in container
+ assert(len(contents) == 1)
+ self.trex_dir = os.path.join(trex_base_dir, contents[0])
+
+ def run_server(self, traffic_profile, filename='/etc/trex_cfg.yaml'):
+ """
+ Runs 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
+ """
+ cfg = self.__save_config(traffic_profile, filename)
+ cores = traffic_profile.cores
+ subprocess.Popen(['nohup', '/bin/bash', '-c',
+ './t-rex-64 -i -c {} --iom 0 --no-scapy-server --close-at-end --vlan'
+ ' --cfg {} &> /tmp/trex.log & disown'.format(cores, cfg)],
+ cwd=self.trex_dir)
+ LOG.info('TRex server is running...')
+
+ def __save_config(self, traffic_profile, filename):
+ ifs = ",".join([repr(pci) for pci in traffic_profile.pcis])
+
+ result = """# Config generated by NFVBench tool
+ - port_limit : 2
+ version : 2
+ interfaces : [{ifs}]""".format(ifs=ifs)
+
+ yaml.safe_load(result)
+ if os.path.exists(filename):
+ os.remove(filename)
+ with open(filename, 'w') as f:
+ f.write(result)
+
+ return filename