1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
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)
|
