#!/usr/bin/env python#-*- coding:utf-8 -*-

importos,refrom .base importBasePluginfrom lib.response importBaseResponseimporttracebackfrom lib.log importloggerclassNetwork(BasePlugin):defwin(self,handler,hostname):'''执行命令拿到结果-网卡

:return:'''

print("执行win方法")

ret= handler.cmd('ipconfig',hostname)[0:10]returnret#def linux(self,handler,hostname):

#'''

#执行命令拿到结果-网卡

#:return:

#'''

#print("执行Linux方法")

#ret = handler.cmd('ifconfig',hostname)[0:10]

#return ret

deflinux(self, handler, hostname):

reponse=BaseResponse()try:ifself.debug:

output= open(os.path.join(self.base_dir, 'files', 'nic.out'), 'r').read()

interfaces_info=self._interfaces_ip(output)else:

interfaces_info=self.linux_interfaces(handler)

self.standard(interfaces_info)

reponse.data=interfaces_info#reponse.data = self.standard(interfaces_info)

#print(reponse.data)

exceptException as e:

error_msg=traceback.format_exc()

reponse.status=False

reponse.error=error_msg#记录错误日志

logger.error(error_msg)returnreponse.dictdeflinux_interfaces(self, handler):'''Obtain interface information for *NIX/BSD variants'''ifaces=dict()

ip_path= 'ip'

ifip_path:

cmd1= handler.cmd('sudo {0} link show'.format(ip_path))

cmd2= handler.cmd('sudo {0} addr show'.format(ip_path))

ifaces= self._interfaces_ip(cmd1 + '\n' +cmd2)returnifacesdefwhich(self, exe):def_is_executable_file_or_link(exe):#check for os.X_OK doesn't suffice because directory may executable

return (os.access(exe, os.X_OK) and(os.path.isfile(exe)oros.path.islink(exe)))ifexe:if_is_executable_file_or_link(exe):#executable in cwd or fullpath

returnexe#default path based on busybox's default

default_path = '/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin'search_path= os.environ.get('PATH', default_path)

path_ext= os.environ.get('PATHEXT', '.EXE')

ext_list= path_ext.split(';')

search_path=search_path.split(os.pathsep)ifTrue:#Add any dirs in the default_path which are not in search_path. If

#there was no PATH variable found in os.environ, then this will be

#a no-op. This ensures that all dirs in the default_path are

#searched, which lets salt.utils.which() work well when invoked by

#salt-call running from cron (which, depending on platform, may

#have a severely limited PATH).

search_path.extend(

[

xfor x indefault_path.split(os.pathsep)if x not insearch_path

]

)for path insearch_path:

full_path=os.path.join(path, exe)if_is_executable_file_or_link(full_path):returnfull_pathreturnNonedef _number_of_set_bits_to_ipv4_netmask(self, set_bits): #pylint: disable=C0103

'''Returns an IPv4 netmask from the integer representation of that mask.

Ex. 0xffffff00 -> '255.255.255.0''''

returnself.cidr_to_ipv4_netmask(self._number_of_set_bits(set_bits))defcidr_to_ipv4_netmask(self, cidr_bits):'''Returns an IPv4 netmask'''

try:

cidr_bits=int(cidr_bits)if not 1 <= cidr_bits <= 32:return ''

exceptValueError:return ''netmask= ''

for idx in range(4):ifidx:

netmask+= '.'

if cidr_bits >= 8:

netmask+= '255'cidr_bits-= 8

else:

netmask+= '{0:d}'.format(256 - (2 ** (8 -cidr_bits)))

cidr_bits=0returnnetmaskdef_number_of_set_bits(self, x):'''Returns the number of bits that are set in a 32bit int'''

#Taken from http://stackoverflow.com/a/4912729. Many thanks!

x -= (x >> 1) & 0x55555555x= ((x >> 2) & 0x33333333) + (x & 0x33333333)

x= ((x >> 4) + x) & 0x0f0f0f0fx+= x >> 8x+= x >> 16

return x & 0x0000003f

def_interfaces_ip(self, out):'''Uses ip to return a dictionary of interfaces with various information about

each (up/down state, ip address, netmask, and hwaddr)'''ret=dict()

right_keys= ['name', 'hwaddr', 'up', 'netmask', 'ipaddrs']defparse_network(value, cols):'''Return a tuple of ip, netmask, broadcast

based on the current set of cols'''brd=Noneif '/' in value: #we have a CIDR in this address

ip, cidr = value.split('/') #pylint: disable=C0103

else:

ip= value #pylint: disable=C0103

cidr = 32

if type_ == 'inet':

mask=self.cidr_to_ipv4_netmask(int(cidr))if 'brd' incols:

brd= cols[cols.index('brd') + 1]return(ip, mask, brd)

groups= re.compile('\r?\n\\d').split(out)for group ingroups:

iface=None

data=dict()for line ingroup.splitlines():if ' ' not inline:continuematch= re.match(r'^\d*:\s+([\w.\-]+)(?:@)?([\w.\-]+)?:\s+', line)ifmatch:

iface, parent, attrs=match.groups()if 'UP' in attrs.split(','):

data['up'] =Trueelse:

data['up'] =Falseif parent and parent inright_keys:

data[parent]=parentcontinuecols=line.split()if len(cols) >= 2:

type_, value= tuple(cols[0:2])

iflabel= cols[-1:][0]if type_ in ('inet',):if 'secondary' not incols:

ipaddr, netmask, broadcast=parse_network(value, cols)if type_ == 'inet':if 'inet' not indata:

data['inet'] =list()

addr_obj=dict()

addr_obj['address'] =ipaddr

addr_obj['netmask'] =netmask

addr_obj['broadcast'] =broadcast

data['inet'].append(addr_obj)else:if 'secondary' not indata:

data['secondary'] =list()

ip_, mask, brd=parse_network(value, cols)

data['secondary'].append({'type': type_,'address': ip_,'netmask': mask,'broadcast': brd,

})delip_, mask, brdelif type_.startswith('link'):

data['hwaddr'] =valueififace:if iface.startswith('pan') or iface.startswith('lo') or iface.startswith('v'):deliface, dataelse:

ret[iface]=datadeliface, datareturnretdefstandard(self, interfaces_info):for key, value ininterfaces_info.items():

ipaddrs=set()

netmask=set()if not 'inet' invalue:

value['ipaddrs'] = ''value['netmask'] = ''

else:for item in value['inet']:

ipaddrs.add(item['address'])

netmask.add(item['netmask'])

value['ipaddrs'] = '/'.join(ipaddrs)

value['netmask'] = '/'.join(netmask)del value['inet']