Upload the contribution of vstf as bottleneck network framework.

[bottlenecks.git] / vstf / vstf / agent / perf / netns.py

diff --git a/vstf/vstf/agent/perf/netns.py b/vstf/vstf/agent/perf/netns.py
new file mode 100755 (executable)
index 0000000..d5552fa
--- /dev/null
+++ b/vstf/vstf/agent/perf/netns.py
@@ -0,0 +1,103 @@
+"""
+Created on 2015-8-6
+
+@author: y00228926
+"""
+import logging
+from vstf.common.utils import IPCommandHelper
+from vstf.agent.perf import ethtool
+from vstf.common.utils import check_call, check_output, ns_cmd, my_popen, my_sleep
+
+LOG = logging.getLogger(__name__)
+
+
+class Netns(object):
+    def __init__(self):
+        super(Netns, self).__init__()
+        self.netns_add_str = "ip netns add %s"
+        self.netns_del_str = " ip netns del %s"
+        self.netns_add_device_str = " ip link set %s netns %s"
+        self.set_link_up_str = "ip link set dev %s up"
+        self.set_link_addr_str = "ip addr replace %s dev %s"
+        self.netns_remove_device_str = "ip netns exec %s ip link set %s netns  1"
+        # self.set_link_addr_str = "ifconfig %s %s up"
+        self.ns_devices = {}
+
+    def clean_all_namespace(self):
+        out = check_output("ip netns list", shell=True)
+        for ns in out.splitlines():
+            self.remove_namespace(ns)
+        return True
+
+    def create_namespace(self, name):
+        if name in (None, 'None', 'none'):
+            return True
+        cmd = self.netns_add_str % name
+        check_call(cmd, shell=True)
+        return True
+
+    def remove_namespace(self, ns):
+        if ns in (None, 'None', 'none'):
+            return True
+        ip_helper = IPCommandHelper(ns)
+        for dev in ip_helper.device_mac_map:
+            cmd = self.netns_remove_device_str % (ns, dev)
+            check_call(cmd, shell=True)
+            self.activate_device(None, dev)
+        cmd = self.netns_del_str % ns
+        check_call(cmd, shell=True)
+        return True
+
+    def add_device(self, ns, device):
+        if ns is None:
+            return True
+        cmd = self.netns_add_device_str % (device, ns)
+        check_call(cmd, shell=True)
+        return True
+
+    def config_ip(self, ns, device, ip):
+        self.activate_device(ns, device)
+        cmd = self.set_link_addr_str % (ip, device)
+        cmd = ns_cmd(ns, cmd)
+        check_call(cmd, shell=True)
+        return True
+
+    def activate_device(self, ns, device):
+        cmd = self.set_link_up_str % device
+        cmd = ns_cmd(ns, cmd)
+        check_call(cmd, shell=True)
+        return True
+
+
+class NetnsManager(object):
+    def __init__(self):
+        super(NetnsManager, self).__init__()
+        self._netns = Netns()
+
+    def config_dev(self, netdev):
+        ns, device, ip = netdev["namespace"], netdev["iface"], netdev['ip_setting'] if "ip_setting" in netdev else \
+            netdev['ip']
+        self._netns.create_namespace(ns)
+        self._netns.add_device(ns, device)
+        self._netns.config_ip(ns, device, ip)
+        my_sleep(1)
+        ethtool.autoneg_off(device, ns)
+        return True
+
+    def recover_dev(self, netdev):
+        ns = netdev["namespace"]
+        return self._netns.remove_namespace(ns)
+
+    def clean_all_namespace(self):
+        return self._netns.clean_all_namespace()
+
+    @staticmethod
+    def ping(ns, ip):
+        cmd = "ping -w2 -c1 %s" % ip
+        cmd = ns_cmd(ns, cmd)
+        child = my_popen(cmd, shell=True)
+        return 0 == child.wait()
+
+
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.DEBUG)