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
104
105
106
107
108
109
110
|
"""
Created on 2015-7-8
@author: y00228926
"""
import subprocess
from StringIO import StringIO
import re
import logging
LOG = logging.getLogger(__name__)
def call(cmd, shell=False):
if shell:
LOG.info(cmd)
else:
LOG.info(' '.join(cmd))
return subprocess.call(cmd, shell=shell)
def check_call(cmd, shell=False):
if shell:
LOG.info(cmd)
else:
LOG.info(' '.join(cmd))
subprocess.check_call(cmd, shell=shell)
def check_output(cmd, shell=False):
if shell:
LOG.info(cmd)
else:
LOG.info(' '.join(cmd))
return subprocess.check_output(cmd, shell=shell)
def check_and_kill(process):
cmd = "ps -ef | grep -v grep | awk '{print $8}' | grep -w %s | wc -l" % process
out = check_output(cmd, shell=True)
if int(out):
check_call(['killall', process])
def check_and_rmmod(mod):
cmd = "lsmod | awk '{print $1}' | grep -w %s | wc -l" % mod
out = check_output(cmd, shell=True)
if int(out):
check_call(['rmmod', mod])
def umount(path):
mount_path_set = set()
out = check_output("cat /proc/mounts", shell=True)
f = StringIO(out)
line = f.readline()
while line:
line = f.readline()
if line:
mpath = line.split()[1]
mount_path_set.add(mpath)
if path in mount_path_set:
ret = call("umount %s" % path, shell=True)
return ret == 0
return True
class IPCommandHelper(object):
def __init__(self):
self.devices = []
self.macs = []
self.device_mac_map = {}
self.mac_device_map = {}
self.bdf_device_map = {}
self.device_bdf_map = {}
self.mac_bdf_map = {}
self.bdf_mac_map = {}
buf = check_output("ip link", shell=True)
macs = re.compile("[A-F0-9]{2}(?::[A-F0-9]{2}){5}", re.IGNORECASE | re.MULTILINE)
for mac in macs.findall(buf):
if mac.lower() in ('00:00:00:00:00:00', 'ff:ff:ff:ff:ff:ff'):
continue
self.macs.append(mac)
sio = StringIO(buf)
for line in sio:
m = re.match(r'^\d+:(.*):.*', line)
if m and m.group(1).strip() != 'lo':
self.devices.append(m.group(1).strip())
for device, mac in zip(self.devices, self.macs):
self.device_mac_map[device] = mac
self.mac_device_map[mac] = device
for device in self.devices:
buf = check_output("ethtool -i %s" % device, shell=True)
bdfs = re.findall(r'^bus-info: \d{4}:(\d{2}:\d{2}\.\d*)$', buf, re.MULTILINE)
if bdfs:
self.bdf_device_map[bdfs[0]] = device
self.device_bdf_map[device] = bdfs[0]
mac = self.device_mac_map[device]
self.mac_bdf_map[mac] = bdfs[0]
self.bdf_mac_map[bdfs[0]] = mac
if __name__ == '__main__':
ip_helper = IPCommandHelper()
print ip_helper.device_mac_map
print ip_helper.mac_device_map
print ip_helper.bdf_device_map
print ip_helper.device_bdf_map
print ip_helper.mac_bdf_map
print ip_helper.bdf_mac_map
|
