1 files changed, 119 insertions, 0 deletions
diff --git a/test/vppctl.py b/test/vppctl.py
new file mode 100644
index 0000000..1f5cc99
--- /dev/null
+++ b/test/vppctl.py
@@ -0,0 +1,119 @@
+#
+# Copyright (c) 2019 Cisco and/or its affiliates.
+#
+# 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.
+#
+
+import subprocess
+import re
+
+
+class VppInterface:
+    def __init__(self, str):
+        self.str = str
+        self.name = None
+        self.Idx = None
+        self.State = None
+        self.MTU = None
+        self.Counter = None
+        self.Count = None
+        self._parse()
+
+    def _parse(self):
+        columns = re.split('\s+', self.str)
+        j = 0
+        for column in columns:
+            j += 1
+            if 1 == j:
+                self.name = column
+            elif 2 == j:
+                self.Idx = column
+            elif 3 == j:
+                if 'up' == column:
+                    self.State = True
+                elif 'down' == column:
+                    self.State = False
+            elif 4 == j:
+                self.MTU = column.split("/")[0]
+            elif 5 == j:
+                self.Counter = column
+            elif 6 == j:
+                self.Count = column
+
+
+class VppIpv4Address:
+    def __init__(self, name):
+        self.name = name
+        self.str = str
+        self.addr = list()
+
+    def addAddress(self, str):
+        tmp = re.split('\s+', str)
+        self.addr.append(tmp[2])
+
+
+class Vppctl:
+    def __init__(self):
+        self.cmd = "vppctl"
+
+    def show_interface(self, name=None):
+        interfaces = list()
+        p = subprocess.run(self.cmd + " show int", shell=True,
+                           stdout=subprocess.PIPE)
+        str = p.stdout.decode("utf-8")
+        lines = str.split("\n")
+        i = 0
+        for line in lines:
+            i += 1
+            if 1 == i:
+                continue
+
+            interfaces.append(VppInterface(line))
+
+        if name is None:
+            return interfaces
+        else:
+            for intf in interfaces:
+                if intf.name == name:
+                    return intf
+
+        return None
+
+    def show_address(self, ifName=None):
+        interfaces = dict()
+        p = subprocess.run(self.cmd + " show int addr", shell=True,
+                           stdout=subprocess.PIPE)
+        str = p.stdout.decode("utf-8")
+        lines = str.split("\n")
+        for line in lines:
+            if re.match('^\s+', line) is None:
+                tmp = line.split(" ")
+                name = tmp[0]
+            else:
+                if name is None:
+                    continue
+
+                if name in interfaces:
+                    interface = interfaces[name]
+                else:
+                    interface = VppIpv4Address(name)
+                    interfaces[name] = interface
+
+                interface.addAddress(line)
+
+        if ifName is None:
+            return interfaces
+        elif ifName in interfaces:
+            return interfaces[ifName]
+        else:
+            return None