Files
jcm_test/server/run.py

685 lines
34 KiB
Python
Raw Normal View History

2023-08-22 04:34:35 +08:00
# coding=utf-8
#!/bin/python
import imp
import os
import time
import sys
2023-11-26 20:20:29 +00:00
import socket
2023-08-22 04:34:35 +08:00
import random
2023-11-26 20:20:29 +00:00
import psutil
import traceback
import threading
2024-03-03 19:43:51 +00:00
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
2023-11-26 20:20:29 +00:00
def rom(info):
i=0
retlist1=[]
rom=0
disks=1
2024-03-03 19:43:51 +00:00
'''if info['OS'] == "Linux":
2023-11-26 20:20:29 +00:00
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:
2023-08-22 04:34:35 +08:00
2023-11-26 20:20:29 +00:00
diskinfo = int(list[i].split("%")[0].split(" ")[-1])
2023-08-22 04:34:35 +08:00
2023-11-26 20:20:29 +00:00
if rom < diskinfo:
rom = diskinfo
i=i+1
disks += 1
2024-03-03 19:43:51 +00:00
else:'''
if True:
list = psutil.disk_partitions(False)
2023-11-26 20:20:29 +00:00
ilen = len(list)
while i < ilen:
if list[i].mountpoint == "/boot":
i += 1
2024-03-03 19:43:51 +00:00
elif list[i].device == "/dev/zram1":
i += 1
2023-11-26 20:20:29 +00:00
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):
2023-08-22 04:34:35 +08:00
global session_
global sent_before
global recv_before
2023-11-26 20:20:29 +00:00
global ttime
global exre
2023-08-22 04:34:35 +08:00
2023-11-26 20:20:29 +00:00
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)
2023-11-26 20:20:29 +00:00
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'':
2024-03-03 19:43:51 +00:00
post = RUL_CS
2023-08-22 04:34:35 +08:00
else:
post = post_data.decode("utf-8").split('&')
2023-11-26 20:20:29 +00:00
httpserver.httppostfile(new_client_socket,"200","./web" + RUL,True,Headers,info)
else:
if user == "":
ma = "200"
if RUL == "/login":
if Headers[0][:4] == "POST":
2024-03-03 19:43:51 +00:00
import datetime
2023-11-26 20:20:29 +00:00
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(',')
2024-03-03 19:43:51 +00:00
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")
2023-11-26 20:20:29 +00:00
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":
2024-03-03 19:43:51 +00:00
fs_rr = fs_rr
2023-11-26 20:20:29 +00:00
else:
if fs_ == "":
fs_ = fs_
else:
2024-03-03 19:43:51 +00:00
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")
2023-11-26 20:20:29 +00:00
res = res + "set-Cookie: Cluster_management_Jiang=" + cok + "; SameSite=Lax; Expires=" + cookie_date
2024-03-03 19:43:51 +00:00
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'))
2023-11-26 20:20:29 +00:00
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)
2023-08-22 04:34:35 +08:00
else:
2023-11-26 20:20:29 +00:00
res = "application/json" + res
httpserver.httppostchar(new_client_socket,"200","{\"data\":\"用户名或密码错误\"}".encode("utf-8"),res,Headers,info)
2024-03-03 19:43:51 +00:00
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
2023-11-26 20:20:29 +00:00
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")
2023-11-26 20:20:29 +00:00
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 = ''
2024-03-03 19:43:51 +00:00
_link = ''
__link = ''
for i in RUL_CS:
2023-11-26 20:20:29 +00:00
tmp = i.split('=')
2024-03-03 19:43:51 +00:00
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]
2023-11-26 20:20:29 +00:00
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'':
2024-03-03 19:43:51 +00:00
post = RUL_CS
2023-08-22 04:34:35 +08:00
else:
2023-11-26 20:20:29 +00:00
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)
2023-08-22 04:34:35 +08:00
else:
2023-11-26 20:20:29 +00:00
if RUL == "/login":
httpserver.httppostchar(new_client_socket,"200",b"{\"data\":\"login\"}","application/json",Headers,info)
2024-03-03 19:43:51 +00:00
elif RUL == "/login/dev":
httpserver.httppostchar(new_client_socket,"200",b"{\"data\":\"login\"}","application/json",Headers,info)
2023-11-26 20:20:29 +00:00
elif RUL == "/token":
httpserver.httppostchar(new_client_socket,"200",b"{\"token\":\"\"}","application/json",Headers,info)
2024-03-03 19:43:51 +00:00
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)
2023-11-26 20:20:29 +00:00
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 += '}'
2024-03-03 19:43:51 +00:00
#httpserver.httppostchar(new_client_socket,"200",res.encode('utf-8'),"application/json",Headers,info)
2023-11-26 20:20:29 +00:00
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")
2024-03-03 19:43:51 +00:00
cat,cdata = sh.Client(ii[7],int(ii[19]),'/assets/info','')
2023-11-26 20:20:29 +00:00
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('"')
2024-03-03 19:43:51 +00:00
res += '"safe": "'+data[7]+'",'
res += '"cpu": "'+data[11]+'",'
res += '"ram": "'+data[15]+'",'
res += '"rom": "'+data[19]+'",'
res += '"tx": "'+data[23]+'",'
res += '"rx": "'+data[27]+'"'
2023-11-26 20:20:29 +00:00
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 += '}'
2023-11-26 20:20:29 +00:00
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()
2023-08-22 04:34:35 +08:00
else:
2023-11-26 20:20:29 +00:00
my_file = "./server" + RUL + "/api.py"
if os.path.isfile(my_file):
2024-03-03 19:43:51 +00:00
#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
2023-11-26 20:20:29 +00:00
sh = imp.load_source(RUL + "/api.py",my_file)
2024-03-03 19:43:51 +00:00
sh.main(new_client_socket,RUL_CS,post_data,Headers,info,user)
2023-11-26 20:20:29 +00:00
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"
2023-08-22 04:34:35 +08:00
except Exception as e:
2023-11-26 20:20:29 +00:00
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
2023-08-22 04:34:35 +08:00
2023-11-26 20:20:29 +00:00
def shutdown(data):
global exre
exre = data
logsprnt("info","info",data)
2023-08-22 04:34:35 +08:00
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()
2024-03-03 19:43:51 +00:00
print("\033[31m " + live,end=' ')
print(data,end='')
2024-03-03 19:43:51 +00:00
print('\033[0m')
lock.release()
2023-11-26 20:20:29 +00:00
def logsprnt(live,info,data):
t = threading.Thread(target=logsprnts, args=(live,info,data))
t.start()
2023-08-22 04:34:35 +08:00
def main(info):
global sent_before
global recv_before
2023-11-26 20:20:29 +00:00
global ttime
global session_
global exre
global res
2023-11-26 20:20:29 +00:00
exre = ''
2023-08-22 04:34:35 +08:00
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()
2023-08-22 04:34:35 +08:00
#tcptossl(tcp_server_socket,ssl_port)
except Exception as e:
print(e.args)
time.sleep(1)
tcp_server_socket_run = tcp_server_socket
2023-11-26 20:20:29 +00:00
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")
2024-03-03 19:43:51 +00:00
run.main(info)
#os.system("run.sh || run.bat")