#!/usr/bin/env python3.5.0
# -*- coding:utf8 -*-
import os,sys,socket,hashlib,time,select,threading,configparser
import pymssql
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","")
# 写回配置文件
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 // 2
b = bytearray(n)
j = 0
for i in range(0, n):
c1 = c[j]
c2 = c[j+1]
j = j+2
c1 = c1 - 65
c2 = c2 - 65
b2 = c2*16 + c1
b1 = b2^ key
b[i]= b1
try:
return b.decode("gbk")
except:
return "failed"
# 实例化数据库类
class MSSQL:
def __init__(self,host,user,pwd,db,port =1433):
self.host = host
self.port = port
self.user = user
self.pwd = pwd
self.db = db
def __GetConnect(self):
try:
if not self.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 not cur:
print("连接数据库失败!")
else:
return cur
except Exception as e:
print("连接数据库失败,%s"%e)
def ExecQuery(self,sql):
cur = self.__GetConnect()
cur.execute(sql)
resList = cur.fetchall()
# 查询完毕后必须关闭连接
self.conn.close()
return resList def ExecNonQuery(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")
except Exception 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 def selectdata(conn):
conf = cf.get("srs","config")
if conf == "":
conn.send(bytes("当前为音频模式！","utf-8"))
elif conf == "":
conn.send(bytes("当前为视频模式！","utf-8"))
def updatesound(conn):
with open(rootdir +'srs.conf',"r") as f:
data = f.readlines()
for i,line in enumerate(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","")
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"))
def updatevideo(conn):
with open(rootdir +'srs.conf',"r") as f:
data = f.readlines()
for i,line in enumerate(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","")
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"))
def crash_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","")
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"))
# 业务执行函数
def operation(conn):
content = """
请输入数字选项进行操作：
1、查询当前直播频道
2、切换频道为音频直播
3、切换频道为视频直播
4、故障修复（默认修复为音频模式）
***********************
"""
conn.send(bytes(content,"utf-8"))
while True:
# 接收数据
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主进程模块
def process(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
# 接收用户及密码信息
while flage:
operation(conn)
else:
# 登陆失败，发送给客户端重新验证
i += 1
conn.send(bytes("error","utf8"))
if i > 2:
# 主动关闭连接
conn.close()
time.sleep(25)
except Exception as e:
conn.close()
time.sleep(15)
# SOCKET服务模块
def sock_server():
'''
启动服务器端，开启线程监听
:return:
'''
server = socket.socket()
server_ip ="localhost"
server_port = 1888
server.bind((server_ip,server_port))
server.listen(10)
while True:
r,w,e = select.select([server,], [], [], 1)
for i,server in enumerate(r):
time.sleep(1)
conn,addr = server.accept()
# 创建线程
t = threading.Thread(target=process, args=(conn, addr))
# 启动线程
t.start()
if __name__ =="__main__":
sock_server()