#!/usr/bin/env python

import os,sys,time,re,shutil

import socket

import fcntl

import struct

import traceback

import commands

#Get interface name

interface_path = '/etc/sysconfig/network-scripts/'

def LOG(info):

""" Log files ...."""

logfile = '/root/pxe_install.log'

files = open(logfile,'a')

try:

files.write('%s : %s \n'%(time.ctime(),info))

except IOError:

files.close()

files.close()

def get_interface ():

os.chdir(interface_path)

eth  = em = list()

for inter in os.listdir(interface_path):

if inter[0:-1] == 'ifcfg-em':

if inter == 'ifcfg-em1' or inter == 'ifcfg-em2':

em.append(inter)

elif inter[0:-1] == 'ifcfg-eth':

if inter == 'ifcfg-eth0' or inter == 'ifcfg-eth1':

eth.append(inter)

if eth:

LOG("Getting interface file name %s is Ok" %eth)

return eth

else:

LOG("Getting interface file name %s is Ok" %em)

return em

def main():

net_name = get_interface()

ipaddr = str()

for inter in net_name:

try:

shutil.move(inter,'/opt/' + inter+'.bak')

_interface_config(inter)

new_interface = inter.split('-')[-1]

if _configure_bond(new_interface):

_configure_bond(new_interface)

LOG("bond script init is Ok")

except Exception,e:

LOG(traceback.format_exc())

if _interface_modprobe():

_interface_modprobe()

if _rester_network():

_rester_network()

# Set interface eth* or em*

def _interface_config(interface):

"""

DEVICE=eth0

BOOTPROTO=static

NOBOOT=yes

NM_CONTROLLED=no

MASTER=bond0

SLAVE=yes

"""

fp = open(interface,'w')

new_interface = interface.split('-')[-1]

if interface == 'ifcfg-em1':

fp.write('DEVICE=%s \n'%new_interface)

fp.write('BOOTPROTO=static \n')

fp.write('ONBOOT=yes \n')

fp.write('NM_CONTROLLED=no \n')

fp.write('MASTER=bond0 \n')

fp.write('SLAVE=yes \n')

elif interface == 'ifcfg-em2':

fp.write('DEVICE=%s \n'%new_interface)

fp.write('BOOTPROTO=static \n')

fp.write('ONBOOT=yes \n')

fp.write('NM_CONTROLLED=no \n')

fp.write('MASTER=bond0 \n')

fp.write('SLAVE=yes \n')

elif interface == 'ifcfg-eth0':

fp.write('DEVICE=%s \n'%new_interface)

fp.write('BOOTPROTO=static \n')

fp.write('ONBOOT=yes \n')

fp.write('NM_CONTROLLED=no \n')

fp.write('MASTER=bond0 \n')

fp.write('SLAVE=yes \n')

elif interface == 'ifcfg-eth1':

fp.write('DEVICE=%s \n'%new_interface)

fp.write('BOOTPROTO=static \n')

fp.write('ONBOOT=yes \n')

fp.write('NM_CONTROLLED=no \n')

fp.write('MASTER=bond0 \n')

fp.write('SLAVE=yes \n')

fp.close()

def _configure_bond(inter):

"""

DEVICE=bond0

BOOTPROTO=static

ONBOOT=yes

IPADDR=192.168.0.100

NETMASK=255.255.255.0

NETWORK=192.168.0.0

BROADCAST=192.168.0.255

"""

#bond name message

if inter == 'eth0':

bond_name = 'ifcfg-bond0'

elif inter == 'em1':

bond_name =  'ifcfg-bond0'

elif inter == 'eth1':

bond_name = 'ifcfg-bond0'

elif inter == 'em2':

bond_name = 'ifcfg-bond0'

# ip address message

if _interface_get_ip(inter):

ipaddr  = _interface_get_ip(inter)

else:

ipaddr = '0.0.0.0'

# ip net mask info

try:

net_mk  = os.popen('ip a |grep %s|grep inet' %inter).readlines()[0]

res = net_mk.split()[1]

net_masklen = res.split('/')[-1]

except:

net_masklen = 18

net_mask = _interface_sum_master(net_masklen)

# default gateway is ....

try:

net_gate = os.popen('ip route |grep default').readlines()[0]

net_gateway = net_gate = net_gate.split(' ')[2]

except:

net_gateway = '0.0.0.0'

try:

if ipaddr == '0.0.0.0':

return ''

fp = open(bond_name,'w')

bond = bond_name.split('-')[-1]

fp.write("DEVICE=%s \n" %bond)

fp.write("BOOTPROTO=static \n")

fp.write("ONBOOT=yes \n")

fp.write("IPADDR=%s \n" %ipaddr)

fp.write("NETMASK=%s \n" % net_mask)

if bond == 'bond0':

fp.write("GATEWAY=%s \n" % net_gateway)

fp.write("DNS1=202.106.0.20 \n")

fp.write("DNS2=8.8.8.8 \n")

LOG("ifcfg-bond* configure is Ok")

return True

except Exception,e:

return False

def _interface_get_ip(inter):

try:

s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

IP = socket.inet_ntoa(fcntl.ioctl(

s.fileno(),

0x8915,  # SIOCGIFADDR

struct.pack('24s',inter))[20:24])

return  IP

except Exception,e:

pass

return

# Add modprobe bonding model

def _interface_modprobe():

try:

fp = open('/etc/modprobe.d/bonding.conf','w')

fp.write("#Module options and blacklists written by bonding \n")

fp.write("alias bond0 bonding \n")

fp.write("options bond0 miimon=100 mode=1 \n")

fp.close()

x,y  = commands.getstatusoutput('modprobe bonding')

if x != 0:

LOG("modprobe bonding is failed")

return True

except:

LOG(traceback.format_exc())

return

# Restart Network

def _rester_network():

x,y  = commands.getstatusoutput('service network restart')

if x == 0:

LOG("restart netowrk is Ok ")

return True

else:

LOG("restart netowrk is Faild ")

return

# According to the CIDR calculation.net master

def _interface_sum_master(net_master):

mask =  (2** 8) - 2 ** (24 - int(net_master))

return '255.255.%s.0' % mask

if __name__  == "__main__":

sc = main()