1 files changed, 113 insertions, 0 deletions
diff --git a/yardstick/network_services/pipeline.py b/yardstick/network_services/pipeline.py
new file mode 100644
index 000000000..d781ba0cd
--- /dev/null
+++ b/yardstick/network_services/pipeline.py
@@ -0,0 +1,113 @@
+# Copyright (c) 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 itertools
+
+from six.moves import zip
+
+FIREWALL_ADD_DEFAULT = "p {0} firewall add default 1"
+FIREWALL_ADD_PRIO = """\
+p {0} firewall add priority 1 ipv4  {1} 24 0.0.0.0 0 0 65535 0 65535 6 0xFF port 0"""
+
+FLOW_ADD_QINQ_RULES = """\
+p {0} flow add qinq 128 512 port 0 id 1
+p {0} flow add default 1"""
+
+ACTION_FLOW_BULK = "p {0} action flow bulk /tmp/action_bulk_512.txt"
+ACTION_DSCP_CLASS_COLOR = "p {0} action dscp {1} class {2} color {3}"
+ROUTE_ADD_DEFAULT = "p {0} route add default 1"
+ROUTE_ADD_ETHER_QINQ = 'p {0} route add {1} {2} port 0 ether {3} qinq 0 {4}'
+ROUTE_ADD_ETHER_MPLS = "p {0} route add {1} 21 port 0 ether {2} mpls 0:{3}"
+
+
+class PipelineRules(object):
+
+    def __init__(self, pipeline_id=0):
+        super(PipelineRules, self).__init__()
+        self.rule_list = []
+        self.pipeline_id = pipeline_id
+
+    def __str__(self):
+        return '\n'.join(self.rule_list)
+
+    def get_string(self):
+        return str(self)
+
+    def next_pipeline(self, num=1):
+        self.pipeline_id += num
+
+    def add_newline(self):
+        self.rule_list.append('')
+
+    def add_rule(self, base, *args):
+        self.rule_list.append(base.format(self.pipeline_id, *args))
+
+    def add_firewall_prio(self, ip):
+        self.add_rule(FIREWALL_ADD_PRIO, ip)
+
+    def add_firewall_script(self, ip):
+        ip_addr = ip.split('.')
+        assert len(ip_addr) == 4
+        ip_addr[-1] = '0'
+        for i in range(256):
+            ip_addr[-2] = str(i)
+            ip = '.'.join(ip_addr)
+            self.add_firewall_prio(ip)
+        self.add_rule(FIREWALL_ADD_DEFAULT)
+        self.add_newline()
+
+    def add_flow_classification_script(self):
+        self.add_rule(FLOW_ADD_QINQ_RULES)
+
+    def add_flow_action(self):
+        self.add_rule(ACTION_FLOW_BULK)
+
+    def add_dscp_class_color(self, dscp, color):
+        self.add_rule(ACTION_DSCP_CLASS_COLOR, dscp, dscp % 4, color)
+
+    def add_flow_action2(self):
+        self.add_rule(ACTION_FLOW_BULK)
+        for dscp, color in zip(range(64), itertools.cycle('GYR')):
+            self.add_dscp_class_color(dscp, color)
+
+    def add_route_ether_mpls(self, ip, mac_addr, index):
+        self.add_rule(ROUTE_ADD_ETHER_MPLS, ip, mac_addr, index)
+
+    def add_route_script(self, ip, mac_addr):
+        ip_addr = ip.split('.')
+        assert len(ip_addr) == 4
+        ip_addr[-1] = '0'
+        for index in range(0, 256, 8):
+            ip_addr[-2] = str(index)
+            ip = '.'.join(ip_addr)
+            self.add_route_ether_mpls(ip, mac_addr, index)
+        self.add_rule(ROUTE_ADD_DEFAULT)
+        self.add_newline()
+
+    def add_ether_qinq(self, ip, mask, mac_addr, index):
+        self.add_rule(ROUTE_ADD_ETHER_QINQ, ip, mask, mac_addr, index)
+
+    def add_route_script2(self, ip, mac_addr):
+        ip_addr = ip.split('.')
+        assert len(ip_addr) == 4
+        ip_addr[-1] = '0'
+        mask = 24
+        for i in range(0, 256):
+            ip_addr[-2] = str(i)
+            ip = '.'.join(ip_addr)
+            self.add_ether_qinq(ip, mask, mac_addr, i)
+        self.add_rule(ROUTE_ADD_DEFAULT)
+        self.add_newline()