bench: Add support for creating VLAN interfaces

VLAN interfaces are indexed by their VLAN ID and function like normal
network interfaces (as expected)

Author: ffainelli

dsatest/bench/interface.py

@@ -10,6 +10,7 @@ def __init__(self, name, machine, switch=None, port_id=None):
         self.machine = machine
         self.switch = switch
         self.port_id = port_id
+        self.vlan_interfaces = {}
 
 
     def __repr__(self):
@@ -39,3 +40,19 @@ def ping(self, destination, count=None, deadline=None):
 
     def arp_get(self, address):
         return self.machine.arp_get(address, self.name)
+
+    def add_vlan(self, vid):
+        ret = self.machine.add_vid(self.name, vid)
+        """ Treat existing interfaces as okay """
+        if ret != 0:
+            return ret
+        vlan = Interface("{}.{}".format(self.name, vid),
+                         self.machine, self.switch, self.port_id)
+        self.vlan_interfaces[vid] = vlan
+
+    def del_vlan(self, vid):
+        ret = self.machine.del_vid(self.name, vid)
+        if ret != 0:
+            return ret
+        del self.vlan_interfaces[vid]
+

dsatest/bench/machine.py

@@ -80,3 +80,13 @@ def arp_get(self, address, interface):
                 return mac
 
         return None
+
+    def add_vid(self, interface, vid):
+        command = "ip link add {0}.{1} link {0} type vlan id {1}".format(interface, vid)
+        ret, _, _ = self.control.execute(command)
+        return ret
+
+    def del_vid(self, interface, vid):
+        command = "ip link del {0}.{1}".format(interface, vid)
+        ret, _, _, = self.control.execute(command)
+        return ret