#!/usr/bin/env python
# -*- coding: utf-8 -*-
import subprocess
import xml.etree.cElementTree as ET
#from binascii import hexlify, unhexlify
class eGW_config():
def __init__(self):
self.tree = ET.ElementTree(file='/home/baicells/eGW/config.xml')
self.root = self.tree.getroot()
self.firewall_enable = self.getnode('system/firewall/enable')[0]
self.firewall_on = self.getnode('system/firewall/on')[0]
self.selinux_enable = self.getnode('system/selinux/enable')[0]
self.selinux_on = self.getnode('system/selinux/on')[0]
self.ipsec_service = self.getnode('network/ipsec/service')[0]
self.ipsec_ul = self.getnode('network/ipsec/ipsec_ul')[0]
self.ipsec_dl = self.getnode('network/ipsec/ipsec_dl')[0]
self.ports_name = self.getnode('network/ports/port/name')
self.ports_addr = self.getnode('network/ports/port/addr')
self.ports_num = len(self.ports_name)
self.routes_dst = self.getnode('network/routes/route/dst')
self.routes_gw = self.getnode('network/routes/route/gw')
self.routes_num = len(self.routes_dst)
self.iptables_enable = self.getnode('network/iptables/iptable/enable')
self.iptables_forward_port = self.getnode('network/iptables/iptable/forward_port')
self.iptables_sourceip = self.getnode('network/iptabels/iptable/sourceip')
self.iptables_num = len(self.iptables_enable)
self.plmn = self.getnode('eGW/plmn')[0]
self.plmn = self.initPLMN(self.plmn)
self.charge_local_ip = self.getnode('eGW/charge_service/local_ip')[0]
self.charge_remote_ip = self.getnode('eGW/charge_service/remote_ip')[0]
self.charge_num = self.getnode('eGW/charge_service/charge_num')[0]
self.enb_index = self.getnode('eGW/eNBs/eNB/index')
self.enb_id = self.getnode('eGW/eNBs/eNB/id')
self.enb_tac = self.getnode('eGW/eNBs/eNB/tac')
self.enb_local_c = self.getnode('eGW/eNBs/eNB/local_c')
self.enb_remote_c = self.getnode('eGW/eNBs/eNB/remote_c')
self.enb_link_enable = self.getnode('eGW/eNBs/eNB/link_enable')
self.enb_gtpu = self.getnode('eGW/eNBs/eNB/gtpu')
self.enb_num = len(self.enb_index)
self.epc_local_c = self.getnode('eGW/epcs/epc/local_c')
self.epc_gtpu = self.getnode('eGW/epcs/epc/gtpu')
self.epc_num = len(self.epc_local_c)
self.forwarding_policy = self.getnode('eGW/forwarding_policys/forwarding_policy')
self.forwarding_policy_num = len(self.forwarding_policy)
self.config_path = '/home/baicells/eGW/config.txt'
#»ñÈ¡½ÚµãÐÅÏ¢
def getnode(self,xpath):
self.node = []
for elem in self.tree.iterfind(xpath):
self.node.append(elem.text)
return self.node
#Ö´ÐÐshellÃüÁî
def sendcmd(self,cmd):
self.cmd = cmd
self.resp = []
self.p = subprocess.Popen(self.cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
for line in self.p.stdout.readlines():
self.resp.append(line)
return self.resp
def initPLMN(self,plmn):
self.plmn = plmn
if len(self.plmn) == 5:
self.mcc = self.plmn[0:3]
self.mnc = self.plmn[3:5]
self.plmn = ''.join([self.mcc[1],self.mcc[0],'f',self.mcc[2],self.mnc[1],self.mnc[0]])
elif len(self.plmn) == 6:
self.mcc = self.plmn[0:3]
self.mnc = self.plmn[3:6]
self.plmn = ''.join([self.mcc[1],self.mnc[0],self.mnc[0],self.mcc[2],self.mnc[2],self.mnc[1]])
return self.plmn
def install_gtp_module(self):
self.sendcmd('insmod /home/baicells/eGW/gtp-relay.ko')
def config_firewall(self):
if self.firewall_enable=='0':
self.sendcmd('systemctl disable firewalld.service')
elif self.firewall_enable=='1':
self.sendcmd('systemctl enable firewalld.service')
else:
print "ERROR!"
if self.firewall_on=='0':
self.sendcmd('systemctl stop firewalld.service')
elif self.firewall_on=='1':
self.sendcmd('systemctl start firewalld.service')
else:
print "ERROR!"
def config_selinux(self):
if self.selinux_enable=='0':
self.sendcmd('sed -i "s/SELINUX=enforcing/SELINUX=disabled/g" /etc/selinux/config')
elif self.selinux_enable=='1':
self.sendcmd('sed -i "s/SELINUX=disabled/SELINUX=enforcing/g" /etc/selinux/config')
else:
print "ERROR!"
if self.selinux_on=='0':
self.sendcmd('setenforce 0')
elif self.selinux_on=='1':
self.sendcmd('setenforce 1')
else:
print "ERROR!"
def config_ipsec(self):
if self.ipsec_service=='0':
self.sendcmd('ipsec stop')
elif self.ipsec_service=='1':
self.sendcmd('ipsec start')
else:
print "ERROR!"
if self.ipsec_ul=='0':
self.sendcmd('echo 0 > /sys/module/gtp_relay/parameters/gtp_ipsec_ul')
elif self.ipsec_ul=='1':
self.sendcmd('echo 1 > /sys/module/gtp_relay/parameters/gtp_ipsec_ul')
else:
print "ERROR!"
if self.ipsec_dl=='0':
self.sendcmd('echo 0 > /sys/module/gtp_relay/parameters/gtp_ipsec_dl')
elif self.ipsec_dl=='1':
self.sendcmd('echo 1 > /sys/module/gtp_relay/parameters/gtp_ipsec_dl')
else:
print "ERROR!"
def config_port(self):
self.sendcmd('systemctl restart network')
for i in range(self.ports_num):
self.sendcmd('ifconfig %s %s' %(self.ports_name[i],self.ports_addr[i]))
if self.ports_name[i]=='gtp1_1':
self.gtp_a = self.ports_addr[i].split('.')[0]
self.gtp_b = self.ports_addr[i].split('.')[1]
self.var = int(self.gtp_a)*256+int(self.gtp_b)
self.sendcmd('echo %s > /sys/module/gtp_relay/parameters/gtp_lip' %self.var)
def config_route(self):
for i in range(self.routes_num):
self.sendcmd('route add -net %s gw %s' %(self.routes_dst[i],self.routes_gw[i]))
def config_iptable(self):
for i in range(self.iptables_num):
if self.iptables_enable[i]=='1':
self.sendcmd('iptables -t nat -A POSTROUTING -s %s.%s.0.0/16 -o %s -j SNAT --to-source %s' %(self.gtp_a,self.gtp_b,self.iptables_forward_port[i],self.iptables_sourceip[i]))
else:
print "PASS"
def config_egw(self):
self.sendcmd('echo ./lccmd set_gw_plmnid %s > %s' %(self.plmn,self.config_path))
if self.charge_local_ip is not None and self.charge_remote_ip is not None:
if self.charge_num == None:
self.charge_num = '1'
self.sendcmd('echo ./lccmd set_charge_service %s %s %s >> %s' %(self.charge_local_ip,self.charge_num,self.charge_remote_ip,self.config_path))
for i in range(self.forwarding_policy_num):
if i > 4:
print "Max num is 4!"
elif i >0:
self.sendcmd('echo ./lccmd set_local_forwarding_policy add %s >> %s' %(self.forwarding_policy[i],self.config_path))
else:
print "forwarding_policy not config!"
for i in range(self.enb_num):
self.sendcmd('echo ./lccmd add_gwenb %s %s %s >> %s' %(self.enb_index[i],self.enb_id[i],self.enb_tac[i],self.config_path))
self.sendcmd('echo ./lccmd set_gwenb_link %s 1 %s %s >> %s' %(self.enb_index[i],self.enb_local_c[i],self.enb_remote_c[i],self.config_path))
if self.enb_link_enable[i]=='1':
self.sendcmd('echo ./lccmd set_gwenb_link_enable %s 1 enable >> %s' %(self.enb_index[i],self.config_path))
elif self.enb_link_enable[i]=='0':
self.sendcmd('echo ./lccmd set_gwenb_link_enable %s 1 disable >> %s' %(self.enb_index[i],self.config_path))
else:
print "ERROR!"
self.sendcmd('echo ./lccmd set_gtp_ip uplink add %s >> %s' %(self.enb_gtpu[i],self.config_path))
for i in range(self.epc_num):
self.sendcmd('echo ./lccmd set_enb_access_ip add %s >> %s' %(self.epc_local_c[i],self.config_path))
self.sendcmd('echo ./lccmd set_gtp_ip downlink add %s >> %s' %(self.epc_gtpu[i],self.config_path))
def main():
myeGW = eGW_config()
myeGW.config_firewall()
myeGW.config_selinux()
myeGW.config_ipsec()
myeGW.config_port()
myeGW.config_route()
myeGW.config_iptable()
myeGW.install_gtp_module()
myeGW.config_egw()
if __name__ == '__main__':
main()
