# coding=utf-8 #!/bin/python import imp import os import time import sys import socket import random import psutil import traceback import threading def dectry(k,p): #k = 'djq%5cu#-jeq15abg$z9_i#_w=$o88m!*alpbedlbat8cr74sd' dec_str = "" for i,j in zip(p.split("_")[:-1],k): # i 为加密字符，j为秘钥字符 ttemp = ord(j) temp = int(i) temp = temp - ttemp temp = chr(temp) # 解密字符 = (加密Unicode码字符 - 秘钥字符的Unicode码)的单字节字符 dec_str = dec_str+temp return dec_str def rom(info): i=0 retlist1=[] rom=0 disks=1 '''if info['OS'] == "Linux": romsh = os.popen("LANG=en_US df -h") list = romsh.read().split("\n") ilen = len(list) while i < ilen: if list[i] == "": i += 1 elif list[i].split(" ")[0] == "Filesystem": i += 1 elif list[i].split(" ")[0] == "udev": i += 1 elif list[i].split(" ")[0] == "tmpfs": i += 1 elif list[i].split(" ")[0] == "devtmpfs": i += 1 elif list[i].split(" ")[0] == "overlay": i += 1 elif list[i].split(" ")[0][:-1] == "/dev/zram": i += 1 elif list[i].split(" ")[0][:-1] == "/dev/loop": i += 1 else: diskinfo = int(list[i].split("%")[0].split(" ")[-1]) if rom < diskinfo: rom = diskinfo i=i+1 disks += 1 else:''' if True: list = psutil.disk_partitions(False) ilen = len(list) while i < ilen: if list[i].mountpoint == "/boot": i += 1 elif list[i].device == "/dev/zram1": i += 1 else: try: diskinfo = psutil.disk_usage(list[i].mountpoint) if rom < diskinfo.percent: rom = diskinfo.percent except Exception as e: if(len(e.args) == 2): if e.args[1] != '设备未就绪。': print(e.args) else: print(e.args) i=i+1 disks += 1 return str(rom) def httpserver_void(new_client_socket,RUL,RUL_CS,post_data,Headers,logs,info): global session_ global sent_before global recv_before global ttime global exre strr = {} httpserver = imp.load_source("server/main/httpserver.py","server/main/httpserver.py") try: data = httpserver.catHeaders("Cookie",Headers) #print(data) data = httpserver.catCookie("Cluster_management_Jiang",Headers,session_.split('\n')) #print(data) user = "" tmp4 = session_.split('\n') for y in range(len(tmp4)): tmp5 = tmp4[y].split(',') if len(tmp5) >1: if data == tmp5[0]: user = tmp5[1] if RUL == "/favicon.ico": if post_data == b'': post = RUL_CS else: post = post_data.decode("utf-8").split('&') httpserver.httppostfile(new_client_socket,"200","./web" + RUL,True,Headers,info) else: if user == "": ma = "200" if RUL == "/login": if Headers[0][:4] == "POST": import datetime post = post_data.decode("utf-8").split('&') name = "" password = "" checkbox = "" for p in post: pp = p.split("=") if pp[0] == "name": name = pp[1] if pp[0] == "password": password = pp[1] if pp[0] == "checkbox": checkbox = pp[1] tools = imp.load_source("Tools/Tools.py","Tools/Tools.py") tools = tools.ifuser(name,password) res = "" if tools == "yes": cok = '' for x in range(32): cok = cok + random.choice("0123456789qwertyuiopasdfghjklzxcvbnmQWERTYUIOPASDFGHJKLLZXCVBNM") fs_rr = session_ fs_session = fs_rr.split('\n') for fs_ in fs_session: if fs_ == "sessinon": session_ = (fs_) session_ = session_ + ("\n") else: if fs_ == "": fs_ = fs_ else: fs__ = fs_.split(',') ifstr = (datetime.datetime.now()).strftime("%a, %d %b %Y %H:%M:%S GMT") ifstr = (datetime.datetime.now()).strftime(" %d %b %Y ") fistr = fs__[3][:0-len("23:40:54 GMT")] if fistr == ifstr: session_ = session_ + fs_ + ("\n") ifstr = (datetime.datetime.now()+datetime.timedelta(days=1)).strftime(" %d %b %Y ") if fistr == ifstr: session_ = session_ + fs_ + ("\n") cookie_date = (datetime.datetime.now()+datetime.timedelta(days=1)).strftime("%a, %d %b %Y %H:%M:%S GMT") #time.strftime("%a, %d %b %Y %H:%M:%S GMT", time.localtime(time.strptime("1","%d"))) session_ = session_ + cok + "," + name + "," + cookie_date + "\n" if checkbox == "true": fs_user = open(".config/sessino","r")#sessino fs_r = fs_user.read() fs_rr = fs_r#.decode("utf-8") fs_session = fs_rr.split('\n') fs_user.close() user_write = '' for fs_ in fs_session: if fs_ == "sessinon": fs_rr = fs_rr else: if fs_ == "": fs_ = fs_ else: fs__ = fs_.split(',') ifstr = (datetime.datetime.now()).strftime("%a, %d %b %Y %H:%M:%S GMT") ifstr = (datetime.datetime.now()).strftime(" %d %b %Y ") fistr = fs__[3][:0-len("23:40:54 GMT")] if fistr == ifstr: session_ = session_ + fs_ + ("\n") ifstr = (datetime.datetime.now()+datetime.timedelta(days=1)).strftime(" %d %b %Y ") if fistr == ifstr: session_ = session_ + fs_ + ("\n") res = res + "set-Cookie: Cluster_management_Jiang=" + cok + "; SameSite=Lax; Expires=" + cookie_date fs_user = open(".config/sessino","wb") fs_user.write(fs_rr.decode('utf-8')) fs_user.write((cok + "," + name + "," + cookie_date + "\n").decode('utf-8')) fs_user.close() else: res = res + "set-Cookie: Cluster_management_Jiang=" + cok + "; SameSite=Lax;"# Expires=" + cookie_date + "\r\n" res = "application/json\r\n" + res httpserver.httppostchar(new_client_socket,"200",b"{\"data\":\"login\"}",res,Headers,info) else: res = "application/json" + res httpserver.httppostchar(new_client_socket,"200","{\"data\":\"用户名或密码错误\"}".encode("utf-8"),res,Headers,info) return 0 elif RUL == "/login/dev": if Headers[0][:4] == "POST": post = post_data.decode("utf-8").split('&') a = '' b = '' c = '' for p in post: pp = p.split("=") if pp[0] == "a": a = pp[1] if pp[0] == "b": b = pp[1] if pp[0] == "c": c = pp[1] b = dectry("djq%5cu#-jeq15abg$z9_i#_w=$o88m!*alpbedlbat8cr74sd",b) b = dectry(time.strftime(" %Y-%m-%d %H:%M ", time.localtime()),b) a = dectry("djq%5cu#-jeq15abg$z9_i#_w=$o88m!*alpbedlbat8cr74sd",a) a = dectry(time.strftime(" %Y-%m-%d %H:%M ", time.localtime()),a) tools = imp.load_source("Tools/Tools.py","Tools/Tools.py") tools = tools.ifuser(a,b) res = "" if tools == "yes": import datetime cok = '' for x in range(32): cok = cok + random.choice("0123456789qwertyuiopasdfghjklzxcvbnmQWERTYUIOPASDFGHJKLLZXCVBNM") fs_rr = session_ fs_session = fs_rr.split('\n') for fs_ in fs_session: if fs_ == "sessinon": session_ = (fs_) session_ = session_ + ("\n") else: if fs_ == "": fs_ = fs_ else: fs__ = fs_.split(',') ifstr = (datetime.datetime.now()).strftime("%a, %d %b %Y %H:%M:%S GMT") ifstr = (datetime.datetime.now()).strftime(" %d %b %Y ") fistr = fs__[3][:0-len("23:40:54 GMT")] if fistr == ifstr: session_ = session_ + fs_ + ("\n") ifstr = (datetime.datetime.now()+datetime.timedelta(days=1)).strftime(" %d %b %Y ") if fistr == ifstr: session_ = session_ + fs_ + ("\n") cookie_date = (datetime.datetime.now()+datetime.timedelta(days=1)).strftime("%a, %d %b %Y %H:%M:%S GMT") #time.strftime("%a, %d %b %Y %H:%M:%S GMT", time.localtime(time.strptime("1","%d"))) session_ = session_ + cok + "," + a + "," + cookie_date + "\n" res = res + "set-Cookie: Cluster_management_Jiang=" + cok + "; SameSite=Lax;"# Expires=" + cookie_date + "\r\n" res = "application/json\r\n" + res httpserver.httppostchar(new_client_socket,"200",b"{\"data\":\"login\"}",res,Headers,info) return 0 elif RUL == "/assets/info": res = '' sent_now = psutil.net_io_counters().bytes_sent recv_now = psutil.net_io_counters().bytes_recv sent = (sent_now - sent_before)/1024/(time.time()-ttime) # 算出1秒后的差值 recv = (recv_now - recv_before)/1024/(time.time()-ttime) ttime=time.time() sent_before=sent_now recv_before=recv_now res += '"api": "1",' if info['OS'] == "Linux": res += '"safe": "--",' else: res += '"safe": "--",' res += '"cpu": "'+str(psutil.cpu_percent(None))+'",' res += '"ram": "'+str(psutil.virtual_memory().percent)+'",' res += '"rom": "'+rom(info)+'",' tx = "" rx = "" if sent < 1000: tx = '{0}KB/s'.format("%.2f"%sent) else: sent = sent / 1024 tx = '{0}MB/s'.format("%.2f"%sent) if recv < 1000: rx = '{0}KB/s'.format("%.2f"%recv) else: recv = recv / 1024 rx = '{0}MB/s'.format("%.2f"%recv) res += '"tx": "'+tx+'",' res += '"rx": "'+rx+'"' res += '}' httpserver.httppostchar(new_client_socket,"200",res.encode('utf-8'),"application/json",Headers,info) return 0 if RUL[:len("/assets")] != "/assets": RUL = "/login.html" for h in Headers: if h == 'Connection: Upgrade': httpserver.websockinit(new_client_socket,Headers,info) httpserver.websockend(new_client_socket,b"No Login\r\n") ma = "401" RUL = "./web/" + info["theme"] + RUL httpserver.httppostfile(new_client_socket,ma,RUL,True,Headers,info) else: my_file = "./web/" + info["theme"] + RUL link = '' _link = '' __link = '' for i in RUL_CS: tmp = i.split('=') if tmp[0] == 'link': link = tmp[1] else: if tmp[0] == '_link': __link = tmp[1] _link += "link=" + tmp[1] + '\n' else: _link += i + '\n' if (link != '') and (link != '127.0.0.1'): httpclient = imp.load_source("server/main/httpclient","server/main/httpclient.py") #httpclient.Client(ii[7],int(ii[19]),'/login/dev','') fs = open(".config/main/lits.json", "rb") hosts = fs.read().decode("utf-8").split('\n') for i in range(len(hosts)-1): ii = hosts[i].split('"') if ii[3] == link: dd = '' for ddd in RUL_CS: tmp = ddd.split('=') if tmp[0] == 'link': link = tmp[1] dd += '&_' + ddd else: dd += '&' + ddd data = httpclient.Clientchar(ii[7],int(ii[19]),RUL + "?link=127.0.0.1" + dd,post_data) if data[0] == "401": httpclient.logindev(ii[3]) data = httpclient.Clientchar(ii[7],int(ii[19]),RUL + "?link=127.0.0.1" + dd,post_data) if data[0] == "401": strr = {} strr["err"] = data[1].decode('utf-8') php = imp.load_source("server/main/php.py","server/main/php.py") data = php.php(new_client_socket,"post","/blink.php",user,Headers,info,strr) httpserver.httppostchar(new_client_socket,"401",data.encode("utf-8"),"text/html;charset=UTF-8",Headers,info) return 0 if data[0] == "500": strr = {} strr["err"] = data[1].decode('utf-8') php = imp.load_source("server/main/php.py","server/main/php.py") data = php.php(new_client_socket,"post","/500.php",user,Headers,info,strr) httpserver.httppostchar(new_client_socket,"500",data.encode("utf-8"),"text/html;charset=UTF-8",Headers,info) return 0 if data[0] == "502": strr = {} strr["err"] = data[1].decode('utf-8') php = imp.load_source("server/main/php.py","server/main/php.py") data = php.php(new_client_socket,"post","/500.php",user,Headers,info,strr) httpserver.httppostchar(new_client_socket,"500",data.encode("utf-8"),"text/html;charset=UTF-8",Headers,info) return 0 httpserver.httppostchar(new_client_socket,data[0],data[1],data[2],Headers,info) return 0 #link = _link if __link != '': RUL_CS = _link.split('\n')[:-1] if os.path.isdir(my_file): if os.path.isfile("./web/" + info["theme"] + RUL + "/index.php"): RUL = RUL + "/index.php" else: my_file = "./web/" + info["theme"] + RUL + "/index.html" if os.path.isfile(my_file): RUL = RUL + "/index.html" if os.path.isfile("./web/" + info["theme"] + RUL): tmp2 = RUL.split('.') if tmp2[-1] == 'php': php = imp.load_source("server/main/php.py","server/main/php.py") if post_data == b'': post = RUL_CS else: post = post_data.decode("utf-8").split('&') data = php.php(new_client_socket,post,RUL,user,Headers,info,strr) httpserver.httppostchar(new_client_socket,"200",data.encode("utf-8"),"text/html;charset=UTF-8",Headers,info) else: RUL = "./web/" + info["theme"] + RUL httpserver.httppostfile(new_client_socket,"200",RUL,True,Headers,info) else: if RUL == "/login": httpserver.httppostchar(new_client_socket,"200",b"{\"data\":\"login\"}","application/json",Headers,info) elif RUL == "/login/dev": httpserver.httppostchar(new_client_socket,"200",b"{\"data\":\"login\"}","application/json",Headers,info) elif RUL == "/token": httpserver.httppostchar(new_client_socket,"200",b"{\"token\":\"\"}","application/json",Headers,info) elif RUL == "/assets/info": res = '' sent_now = psutil.net_io_counters().bytes_sent recv_now = psutil.net_io_counters().bytes_recv sent = (sent_now - sent_before)/1024/(time.time()-ttime) # 算出1秒后的差值 recv = (recv_now - recv_before)/1024/(time.time()-ttime) ttime=time.time() sent_before=sent_now recv_before=recv_now res += '"api": "1",' if info['OS'] == "Linux": res += '"safe": "--",' else: res += '"safe": "--",' res += '"cpu": "'+str(psutil.cpu_percent(None))+'",' res += '"ram": "'+str(psutil.virtual_memory().percent)+'",' res += '"rom": "'+rom(info)+'",' tx = "" rx = "" if sent < 1000: tx = '{0}KB/s'.format("%.2f"%sent) else: sent = sent / 1024 tx = '{0}MB/s'.format("%.2f"%sent) if recv < 1000: rx = '{0}KB/s'.format("%.2f"%recv) else: recv = recv / 1024 rx = '{0}MB/s'.format("%.2f"%recv) res += '"tx": "'+tx+'",' res += '"rx": "'+rx+'"' res += '}' httpserver.httppostchar(new_client_socket,"200",res.encode("utf-8"),"application/json",Headers,info) elif RUL == "/main/info": res = '' if os.path.isfile(".config/main/lits.json"): fs = open(".config/main/lits.json", "rb") hosts = fs.read().decode("utf-8").split('\n') res += '{"for": "'+str(len(hosts)-1)+'","server": [' for i in range(len(hosts)-1): ii = hosts[i].split('"') if i == 0 : res += '{"name": "'+ ii[3] +'",' else: res += ',{"name": "'+ ii[3] +'",' if ii[7] == "127.0.0.1": sent_now = psutil.net_io_counters().bytes_sent recv_now = psutil.net_io_counters().bytes_recv sent = (sent_now - sent_before)/1024/(time.time()-ttime) # 算出1秒后的差值 recv = (recv_now - recv_before)/1024/(time.time()-ttime) ttime=time.time() sent_before=sent_now recv_before=recv_now res += '"api": "1",' if info['OS'] == "Linux": res += '"safe": "--",' else: res += '"safe": "--",' res += '"cpu": "'+str(psutil.cpu_percent(None))+'",' res += '"ram": "'+str(psutil.virtual_memory().percent)+'",' res += '"rom": "'+rom(info)+'",' tx = "" rx = "" if sent < 1000: tx = '{0}KB/s'.format("%.2f"%sent) else: sent = sent / 1024 tx = '{0}MB/s'.format("%.2f"%sent) if recv < 1000: rx = '{0}KB/s'.format("%.2f"%recv) else: recv = recv / 1024 rx = '{0}MB/s'.format("%.2f"%recv) res += '"tx": "'+tx+'",' res += '"rx": "'+rx+'"' res += '}' #httpserver.httppostchar(new_client_socket,"200",res.encode('utf-8'),"application/json",Headers,info) else: url = "http://" + ii[7] + ":8889/assets/info" sh = imp.load_source("server/main/httpclient","server/main/httpclient.py") #cat,cdata = sh.Client(ii[7],int(ii[19]),"/assets/info") cat,cdata = sh.Client(ii[7],int(ii[19]),'/assets/info','') if cat == "200": try: #response = urllib.request.urlopen(url, timeout=1) #html = response.read().decode('utf-8') # 获取到页面的源代码 #html = sh.Client("pi.lan",8889,"/assets/info") res += '"api": "1",' data = cdata.decode("utf-8").split('"') res += '"safe": "'+data[7]+'",' res += '"cpu": "'+data[11]+'",' res += '"ram": "'+data[15]+'",' res += '"rom": "'+data[19]+'",' res += '"tx": "'+data[23]+'",' res += '"rx": "'+data[27]+'"' res += '}' except: res += '"api": "0",' res += '"safe": "--",' res += '"cpu": "--",' res += '"ram": "--",' res += '"rom": "--",' res += '"tx": "--KB/s",' res += '"rx": "--KB/s"' res += '}' elif cat =="401": res += '"api": "3",' res += '"safe": "--",' res += '"cpu": "--",' res += '"ram": "--",' res += '"rom": "--",' res += '"tx": "--KB/s",' res += '"rx": "--KB/s"' res += '}' else: data = os.popen('ping ' + ii[7] + ' -c 1 | grep icmp_seq').read() if len(data.split('icmp_seq=1 ttl=64')) == 2: res += '"api": "2",' res += '"safe": "--",' res += '"cpu": "--",' res += '"ram": "--",' res += '"rom": "--",' res += '"tx": "--KB/s",' res += '"rx": "--KB/s"' res += '}' else: res += '"api": "0",' res += '"safe": "--",' res += '"cpu": "--",' res += '"ram": "--",' res += '"rom": "--",' res += '"tx": "--KB/s",' res += '"rx": "--KB/s"' res += '}' res += ']' res += '}' httpserver.httppostchar(new_client_socket,"200",res.encode("utf-8"),"application/json",Headers,info) else: if os.path.isdir('.config') == False: os.mkdir('.config') if os.path.isdir('.config/main') == False: os.mkdir('.config/main') fs = open(".config/main/lits.json", "a") data = '{"name":"'+"127.0.0.1"+'","host":"'+"127.0.0.1"+'"}' + '\n' fs.write(data) fs.close() else: my_file = "./server" + RUL + "/api.py" if os.path.isfile(my_file): #if post_data == b'': # post = RUL_CS #else: #post_data = post_data.split(b'&') #for i in range(0,len(post_data)): # post_data[i] = httpserver.http_to_char(post_data[i].decode("utf-8")) # for i in #post = post_data sh = imp.load_source(RUL + "/api.py",my_file) sh.main(new_client_socket,RUL_CS,post_data,Headers,info,user) else: strr = {} strr["err"] = '404' sh = imp.load_source("server/main/php.py","server/main/php.py") data = sh.php(new_client_socket,"post","/404.php",user,Headers,info,strr) httpserver.httppostchar(new_client_socket,"404",data.encode("utf-8"),"text/html;charset=UTF-8",Headers,info) #res = "HTTP/1.1 404 OK \r\n" except Exception as e: strr = {} strr["err"] = traceback.format_exc() logsprnt("ERROR",info,traceback.format_exc()) sh = imp.load_source("server/main/php.py","server/main/php.py") data = sh.php(new_client_socket,"post","/500.php",user,Headers,info,strr) httpserver.httppostchar(new_client_socket,"500",data.encode("utf-8"),"text/html;charset=UTF-8",Headers,info) if exre == 'reset': return -9 if exre == 'exit': return -2 return 0 def shutdown(data): global exre exre = data logsprnt("info","info",data) def logsprnts(live,info,data): import threading lock = threading.Lock() lock.acquire() #if len(sys.argv) == 5: # if sys.argv[1] == "hass": # os.system("bashio::log." + live + " " + data.decode("utf-8")) # else: # print(live,end=' ') # print(data,end='\r') #else: try: if data[-2:] == b'\r\n': data = data[:-2] except Exception as e: print() if live == "info": if os.path.exists("info.sh"): try: os.system("./info.sh '" + data.decode("utf-8") + "'") except Exception as e: print() print("\033[32m " + live,end=' ') print(data,end='') print('\033[0m') else: if os.path.exists("error.sh"): try: os.system("./error.sh '" + data.decode("utf-8") + "'") except Exception as e: print() print("\033[31m " + live,end=' ') print(data,end='') print('\033[0m') lock.release() def logsprnt(live,info,data): t = threading.Thread(target=logsprnts, args=(live,info,data)) t.start() def main(info): global sent_before global recv_before global ttime global session_ global exre global res exre = '' print("boot") fs = open("./server/server.ini", "rb") pat = fs.read().decode("utf-8") path = pat.split('\n') for p in path: p = "server/" + p.split("\r")[0] if os.path.isdir(p): if os.path.isfile(p + "/boot.py"): print(p) sh = imp.load_source(p + "/boot.py",p + "/boot.py") sh.boot(info) sent_before = psutil.net_io_counters().bytes_sent # 已发送的流量 recv_before = psutil.net_io_counters().bytes_recv # 已接收的流量 ttime = time.time() port = info["port"] python = info["python"] OS = info["OS"] dir = info["dir"] Versino = info["Versino"] fs = open(".config/sessino","rb") session__ = fs.read().decode("utf-8").split('\r\n') session_ = '' for p in session__: session_ += p + '\n' fs.close() tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) a = 1 while a < 10: try: tcp_server_socket.bind(("", port)) a = 16 except Exception as e: print(e.args) time.sleep(1) tcp_server_socket.listen(100) tcp_server_socket_run = tcp_server_socket if os.path.exists(".config/ssl/site.key"): try: import ssl tcp_server_socket_ssl = socket.socket(socket.AF_INET, socket.SOCK_STREAM) tcp_server_socket_ssl = ssl.wrap_socket(tcp_server_socket_ssl,keyfile='.config/ssl/site.key',certfile='.config/ssl/site.crt') tcp_server_socket_ssl.bind(("localhost",0)) ssl_port = tcp_server_socket_ssl.getsockname() tcp_server_socket_ssl.listen(100) tcp_server_socket_run = tcp_server_socket_ssl #t = threading.Thread(target=tcptossl, args=(tcp_server_socket,ssl_port)) #t.start() #tcptossl(tcp_server_socket,ssl_port) except Exception as e: print(e.args) time.sleep(1) tcp_server_socket_run = tcp_server_socket httpserver = imp.load_source("server/main/httpserver.py","server/main/httpserver.py") res = httpserver.server(tcp_server_socket_run,info,logsprnt,httpserver_void) if res == -9: tcp_server_socket.close() run = imp.load_source('run',"server/run.py") run.main(info) #os.system("run.sh || run.bat")