#!/usr/bin/env python3.5.0#-*- coding:utf8 -*-
importos,sys,socket,hashlib,time,select,threading,configparserimportpymssql
rootdir=os.path.abspath(sys.argv[0])
rootdir=os.path.dirname(rootdir) +"/"cf=configparser.ConfigParser()if os.path.exists(rootdir +'srs_server.conf'):
cf.read(rootdir+'srs_server.conf')else:#如果文件不存在
f = open(rootdir + "/srs_server.conf","w")
f.write("")
f.close()#读取配置文件
cf.read(rootdir +'srs_server.conf')
cf.add_section("srs")
cf.set("srs","config","1")#写回配置文件
cf.write(open(rootdir +'srs_server.conf',"w"))def decrypt(s,key=2):"""解密方法
:param key:
:param s:
:return:"""c= bytearray(str(s).encode("gbk"))
n= len(c) #计算 b 的字节数
if n % 2 !=0 :return ""n= n // 2b=bytearray(n)
j=0for i inrange(0, n):
c1=c[j]
c2= c[j+1]
j= j+2c1= c1 - 65c2= c2 - 65b2= c2*16 +c1
b1= b2^key
b[i]=b1try:return b.decode("gbk")except:return "failed"
#实例化数据库类
classMSSQL:def __init__(self,host,user,pwd,db,port =1433):
self.host=host
self.port=port
self.user=user
self.pwd=pwd
self.db=dbdef __GetConnect(self):try:if notself.db:print("没有设置数据库信息")
self.conn= pymssql.connect(host=self.host,port=self.port,user=self.user,password=self.pwd,database=self.db,charset="utf8")
cur=self.conn.cursor()if notcur:print("连接数据库失败!")else:returncurexceptException as e:print("连接数据库失败,%s"%e)defExecQuery(self,sql):
cur= self.__GetConnect()
cur.execute(sql)
resList=cur.fetchall()#查询完毕后必须关闭连接
self.conn.close()returnresListdefExecNonQuery(self,sql):
cur= self.__GetConnect()
cur.execute(sql)
self.conn.commit()
self.conn.close()#查询数据库相关信息情况#更新SQL数据库相关信息情况
def updatesql(load=0,status=0,item=0):#加载配置文件
cf =configparser.ConfigParser()if os.path.exists(rootdir+"srs_server.conf"):try:
cf.read(rootdir+"srs_server.conf")exceptException as c:print(c)else:print("加载srs_server.conf配置文件失败！")#实例化MSSQL操作类
host = cf.get("HOST","host")
ms= MSSQL(host=cf.get("DB","ip"),user=decrypt(cf.get("DB","username")),pwd=decrypt(cf.get("DB","password")),db=cf.get("DB","db"),port=int(cf.get("DB","port")))try:#更新该服务器所有记录
sql = "update sys_load set item = '%s',sysload='%s',status='%s' where servername ='%s'"%(item, load, status, host)
ms.ExecNonQuery(sql)except:pass
defselectdata(conn):
conf= cf.get("srs","config")if conf == "1":
conn.send(bytes("当前为音频模式！","utf-8"))elif conf == "2":
conn.send(bytes("当前为视频模式！","utf-8"))defupdatesound(conn):
with open(rootdir+'srs.conf',"r") as f:
data=f.readlines()for i,line inenumerate(data):#去除制表符，换行符
line = line.replace('\t','').replace('\n','').replace(' ','')if line =="#allowpublish220.168.91.254;":
data[i]="\tallow\t\tpublish\t\t220.168.91.254;\t\n"
elif line == "#forward10.27.68.150;":
data[i]= "\tforward\t\t10.27.68.150;\t\t \n"
elif line == "transcodelive/livestream{":
f.seek(i+1)
data[i+1] = "\tenabled\t\ton;\n"with open(rootdir+'srs.conf',"w") as xf:
xf.writelines(data)
xf.close()
status= os.system("service srs restart")if status ==0:
cf.read(rootdir+'srs_server.conf')
cf.set("srs","config","1")
cf.write(open(rootdir+'srs_server.conf',"w"))
updatesql(item=0,load=1,status=0)
conn.send(bytes("当前已成功切换为音频模式","utf-8"))else:
conn.send(bytes("切换为音频模式失败，请联系IT部！","utf-8"))defupdatevideo(conn):
with open(rootdir+'srs.conf',"r") as f:
data=f.readlines()for i,line inenumerate(data):#去除制表符，换行符
line = line.replace('\t','').replace('\n','').replace(' ','')if line =="allowpublish220.168.91.254;":
data[i]="\t#allow\t\tpublish\t\t220.168.91.254;\t\n"
elif line == "forward10.27.68.150;":
data[i]= "\t#forward\t10.27.68.150;\n"
elif line == "transcodelive/livestream{":#下一行
f.seek(i+1)
data[i+1] = "\tenabled\t\toff;\n"with open(rootdir+'srs.conf',"w") as xf:
xf.writelines(data)
xf.close()
status= os.system("service srs restart")if status ==0:
cf.read(rootdir+'srs_server.conf')
cf.set("srs","config","2")
cf.write(open(rootdir+'srs_server.conf',"w"))
updatesql(item=1,load=3,status=1)
conn.send(bytes("当前已成功切换为视频模式","utf-8"))else:
conn.send(bytes("切换视频模式失败，请联系IT部！","utf-8"))defcrash_recovery(conn):
os.system("service srs stop")#强制还原配置文件
os.system("yes|cp -fr srs_bak.conf srs.conf")
status= os.system("service srs start")if status ==0:
cf.read(rootdir+'srs_server.conf')
cf.set("srs","config","1")
cf.write(open(rootdir+'srs_server.conf',"w"))
updatesql(item=0,load=1,status=0)
conn.send(bytes("修复音频模式成功","utf-8"))else:
conn.send(bytes("修复音频模式失败，请与IT部联系！","utf-8"))#业务执行函数
defoperation(conn):
content= """请输入数字选项进行操作：
1、查询当前直播频道
2、切换频道为音频直播
3、切换频道为视频直播
4、故障修复(默认修复为音频模式)
***********************"""conn.send(bytes(content,"utf-8"))whileTrue:#接收数据
data = conn.recv(1000)
test=data.decode()if int(test) == 1:
selectdata(conn)elif int(test) ==2:
updatesound(conn)elif int(test) ==3:
updatevideo(conn)elif int(test) ==4:
crash_recovery(conn)#SOCKET主进程模块
defprocess(conn,addr):try:
i=0#认证失败允许重试3次
while i < 3:
flage=False#接收客户端连接请求信息
info = conn.recv(1000)#实例化加密函数
hash =hashlib.sha512()
hash.update("a123456789B".encode("utf-8")) #KEY＝a123456789B
hash_pwd =hash.hexdigest()if info.decode() ==hash_pwd:
hash_pwd= ""info= ""
if addr[0] in ["192.168.1.252","192.168.1.190","127.0.0.1"]:
flage=True#接收用户及密码信息
whileflage:
operation(conn)else:#登陆失败，发送给客户端重新验证
i += 1conn.send(bytes("error","utf8"))if i > 2:#主动关闭连接
conn.close()
time.sleep(25)exceptException as e:
conn.close()
time.sleep(15)#SOCKET服务模块
defsock_server():'''启动服务器端，开启线程监听
:return:'''server=socket.socket()
server_ip="localhost"server_port= 1888server.bind((server_ip,server_port))
server.listen(10)whileTrue:
r,w,e= select.select([server,], [], [], 1)for i,server inenumerate(r):
time.sleep(1)
conn,addr=server.accept()#创建线程
t = threading.Thread(target=process, args=(conn, addr))#启动线程
t.start()if __name__ =="__main__":
sock_server()