1 files changed, 62 insertions, 0 deletions
diff --git a/nfvbench/network.py b/nfvbench/network.py
new file mode 100644
index 0000000..e097c2b
--- /dev/null
+++ b/nfvbench/network.py
@@ -0,0 +1,62 @@
+# 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.
+#
+
+
+class Interface(object):
+
+    def __init__(self, name, device, tx_packets, rx_packets):
+        self.name = name
+        self.device = device
+        self.packets = {
+            'tx': tx_packets,
+            'rx': rx_packets
+        }
+
+    def set_packets(self, tx, rx):
+        self.packets = {
+            'tx': tx,
+            'rx': rx
+        }
+
+    def set_packets_diff(self, tx, rx):
+        self.packets = {
+            'tx': tx - self.packets['tx'],
+            'rx': rx - self.packets['rx'],
+        }
+
+    def is_no_op(self):
+        return self.name is None
+
+    def get_packet_count(self, traffic_type):
+        return self.packets.get(traffic_type, 0)
+
+    @staticmethod
+    def no_op():
+        return Interface(None, None, 0, 0)
+
+
+class Network(object):
+
+    def __init__(self, interfaces=None, reverse=False):
+        if interfaces is None:
+            interfaces = []
+        self.interfaces = interfaces
+        self.reverse = reverse
+
+    def add_interface(self, interface):
+        self.interfaces.append(interface)
+
+    def get_interfaces(self):
+        return self.interfaces[::-1] if self.reverse else self.interfaces