#!/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","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 // 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 == "1": conn.send(bytes("当前为音频模式！","utf-8")) elif conf == "2": 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(' ','').replace(' ','').replace(' ','') if line =="#allowpublish220.168.91.254;": data[i] =" allow publish 220.168.91.254; " elif line == "#forward10.27.68.150;": data[i] = " forward 10.27.68.150; " elif line == "transcodelive/livestream{": f.seek(i+1) data[i+1] = " enabled on; " 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")) def updatevideo(conn): with open(rootdir +'srs.conf',"r") as f: data = f.readlines() for i,line in enumerate(data): # 去除制表符，换行符 line = line.replace(' ','').replace(' ','').replace(' ','') if line =="allowpublish220.168.91.254;": data[i] =" #allow publish 220.168.91.254; " elif line == "forward10.27.68.150;": data[i] = " #forward 10.27.68.150; " elif line == "transcodelive/livestream{": # 下一行 f.seek(i+1) data[i+1] = " enabled off; " 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")) 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","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")) # 业务执行函数 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()