mirror of
https://github.com/flucont/btcloud.git
synced 2025-01-23 13:18:13 -05:00
1040 lines
35 KiB
Python
1040 lines
35 KiB
Python
#coding: utf-8
|
||
# +-------------------------------------------------------------------
|
||
# | 宝塔Windows面板
|
||
# +-------------------------------------------------------------------
|
||
# | Copyright (c) 2015-2099 宝塔软件(http://bt.cn) All rights reserved.
|
||
# +-------------------------------------------------------------------
|
||
# | Author: 沐落 <cjx@bt.cn>
|
||
# +-------------------------------------------------------------------
|
||
|
||
import os,chardet,time,sys,re
|
||
import win32net, win32api, win32netcon,win32security,win32serviceutil
|
||
import traceback,shlex,datetime,subprocess,platform
|
||
import sqlite3,shutil
|
||
|
||
def readReg(path,key):
|
||
import winreg
|
||
try:
|
||
newKey = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE ,path)
|
||
value,type = winreg.QueryValueEx(newKey, key)
|
||
return value
|
||
except :
|
||
return False
|
||
|
||
panelPath = readReg(r'SOFTWARE\Wow6432Node\Microsoft\Windows\CurrentVersion\Uninstall\宝塔面板','PanelPath')
|
||
if not panelPath:
|
||
panelPath = os.getenv('BT_PANEL')
|
||
if not panelPath: exit();
|
||
|
||
setupPath = os.path.dirname(panelPath)
|
||
|
||
error_path = '{}/error.log'.format(setupPath)
|
||
logPath = panelPath + '/data/panelExec.log'
|
||
|
||
class Sql():
|
||
#------------------------------
|
||
# 数据库操作类 For sqlite3
|
||
#------------------------------
|
||
__DB_FILE = None # 数据库文件
|
||
__DB_CONN = None # 数据库连接对象
|
||
__DB_TABLE = "" # 被操作的表名称
|
||
__OPT_WHERE = "" # where条件
|
||
__OPT_LIMIT = "" # limit条件
|
||
__OPT_ORDER = "" # order条件
|
||
__OPT_FIELD = "*" # field条件
|
||
__OPT_PARAM = () # where值
|
||
__LOCK = panelPath + '/data/sqlite_lock.pl'
|
||
|
||
def __init__(self):
|
||
self.__DB_FILE = panelPath + '/data/default.db'
|
||
|
||
def __GetConn(self):
|
||
#取数据库对象
|
||
try:
|
||
if self.__DB_CONN == None:
|
||
self.__DB_CONN = sqlite3.connect(self.__DB_FILE)
|
||
self.__DB_CONN.text_factory = str
|
||
except Exception as ex:
|
||
print(str(ex))
|
||
return "error: " + str(ex)
|
||
|
||
def table(self,table):
|
||
#设置表名
|
||
self.__DB_TABLE = table
|
||
return self
|
||
|
||
|
||
def where(self,where,param):
|
||
#WHERE条件
|
||
if where:
|
||
self.__OPT_WHERE = " WHERE " + where
|
||
self.__OPT_PARAM = self.__to_tuple(param)
|
||
return self
|
||
|
||
def __to_tuple(self,param):
|
||
#将参数转换为tuple
|
||
if type(param) != tuple:
|
||
if type(param) == list:
|
||
param = tuple(param)
|
||
else:
|
||
param = (param,)
|
||
return param
|
||
|
||
#更新数据
|
||
def update(self,pdata):
|
||
if not pdata: return False
|
||
keys,param = self.__format_pdata(pdata)
|
||
return self.save(keys,param)
|
||
|
||
#构造数据
|
||
def __format_pdata(self,pdata):
|
||
keys = pdata.keys()
|
||
keys_str = ','.join(keys)
|
||
param = []
|
||
for k in keys: param.append(pdata[k])
|
||
return keys_str,tuple(param)
|
||
|
||
def field(self,field):
|
||
#FIELD条件
|
||
if len(field):
|
||
self.__OPT_FIELD = field
|
||
return self
|
||
|
||
def getField(self,keyName):
|
||
#取回指定字段
|
||
|
||
result = self.field(keyName).select()
|
||
print(result)
|
||
if len(result) != 0:
|
||
return result[0][keyName]
|
||
return result
|
||
|
||
def __format_field(self,field):
|
||
import re
|
||
fields = []
|
||
for key in field:
|
||
s_as = re.search(r'\s+as\s+',key,flags=re.IGNORECASE)
|
||
if s_as:
|
||
as_tip = s_as.group()
|
||
key = key.split(as_tip)[1]
|
||
fields.append(key)
|
||
return fields
|
||
|
||
def __get_columns(self):
|
||
if self.__OPT_FIELD == '*':
|
||
tmp_cols = self.query('PRAGMA table_info('+self.__DB_TABLE+')',())
|
||
cols = []
|
||
for col in tmp_cols:
|
||
if len(col) > 2: cols.append('`' + col[1] + '`')
|
||
if len(cols) > 0: self.__OPT_FIELD = ','.join(cols)
|
||
|
||
def select(self):
|
||
#查询数据集
|
||
self.__GetConn()
|
||
try:
|
||
self.__get_columns()
|
||
sql = "SELECT " + self.__OPT_FIELD + " FROM " + self.__DB_TABLE + self.__OPT_WHERE + self.__OPT_ORDER + self.__OPT_LIMIT
|
||
result = self.__DB_CONN.execute(sql,self.__OPT_PARAM)
|
||
data = result.fetchall()
|
||
#构造字典系列
|
||
if self.__OPT_FIELD != "*":
|
||
fields = self.__format_field(self.__OPT_FIELD.split(','))
|
||
tmp = []
|
||
for row in data:
|
||
i=0
|
||
tmp1 = {}
|
||
for key in fields:
|
||
tmp1[key.strip('`')] = row[i]
|
||
i += 1
|
||
tmp.append(tmp1)
|
||
del(tmp1)
|
||
data = tmp
|
||
del(tmp)
|
||
else:
|
||
#将元组转换成列表
|
||
tmp = list(map(list,data))
|
||
data = tmp
|
||
del(tmp)
|
||
self.__close()
|
||
return data
|
||
except Exception as ex:
|
||
return "error: " + str(ex)
|
||
|
||
def setField(self,keyName,keyValue):
|
||
#更新指定字段
|
||
return self.save(keyName,(keyValue,))
|
||
|
||
def commit(self):
|
||
self.__close()
|
||
self.__DB_CONN.commit()
|
||
|
||
|
||
def save(self,keys,param):
|
||
#更新数据
|
||
self.write_lock()
|
||
self.__GetConn()
|
||
self.__DB_CONN.text_factory = str
|
||
try:
|
||
opt = ""
|
||
for key in keys.split(','):
|
||
opt += key + "=?,"
|
||
opt = opt[0:len(opt)-1]
|
||
sql = "UPDATE " + self.__DB_TABLE + " SET " + opt+self.__OPT_WHERE
|
||
|
||
#处理拼接WHERE与UPDATE参数
|
||
tmp = list(self.__to_tuple(param))
|
||
for arg in self.__OPT_PARAM:
|
||
tmp.append(arg)
|
||
self.__OPT_PARAM = tuple(tmp)
|
||
result = self.__DB_CONN.execute(sql,self.__OPT_PARAM)
|
||
self.__close()
|
||
self.__DB_CONN.commit()
|
||
self.rm_lock()
|
||
return result.rowcount
|
||
except Exception as ex:
|
||
return "error: " + str(ex)
|
||
|
||
|
||
def execute(self,sql,param = ()):
|
||
#执行SQL语句返回受影响行
|
||
self.write_lock()
|
||
self.__GetConn()
|
||
try:
|
||
result = self.__DB_CONN.execute(sql,self.__to_tuple(param))
|
||
self.__DB_CONN.commit()
|
||
self.rm_lock()
|
||
return result.rowcount
|
||
except Exception as ex:
|
||
return "error: " + str(ex)
|
||
|
||
#是否有锁
|
||
def is_lock(self):
|
||
n = 0
|
||
while os.path.exists(self.__LOCK):
|
||
n+=1
|
||
if n > 100:
|
||
self.rm_lock()
|
||
break
|
||
time.sleep(0.01)
|
||
#写锁
|
||
def write_lock(self):
|
||
self.is_lock()
|
||
open(self.__LOCK,'wb+').close()
|
||
|
||
#解锁
|
||
def rm_lock(self):
|
||
if os.path.exists(self.__LOCK):
|
||
os.remove(self.__LOCK)
|
||
|
||
def query(self,sql,param = ()):
|
||
#执行SQL语句返回数据集
|
||
self.__GetConn()
|
||
try:
|
||
result = self.__DB_CONN.execute(sql,self.__to_tuple(param))
|
||
#将元组转换成列表
|
||
data = list(map(list,result))
|
||
return data
|
||
except Exception as ex:
|
||
return "error: " + str(ex)
|
||
|
||
def __close(self):
|
||
#清理条件属性
|
||
self.__OPT_WHERE = ""
|
||
self.__OPT_FIELD = "*"
|
||
self.__OPT_ORDER = ""
|
||
self.__OPT_LIMIT = ""
|
||
self.__OPT_PARAM = ()
|
||
|
||
|
||
def close(self):
|
||
#释放资源
|
||
try:
|
||
self.__DB_CONN.close()
|
||
self.__DB_CONN = None
|
||
except:
|
||
pass
|
||
|
||
|
||
def GetLocalIp():
|
||
"""
|
||
取本地外网IP
|
||
|
||
"""
|
||
try:
|
||
filename = panelPath + '/data/iplist.txt'
|
||
ipaddress = readFile(filename)
|
||
if not ipaddress:
|
||
|
||
url = 'http://www.example.com/api/getIpAddress';
|
||
str = httpGet(url)
|
||
writeFile(filename,ipaddress)
|
||
|
||
ipaddress = re.search('\d+.\d+.\d+.\d+',ipaddress).group(0);
|
||
return ipaddress
|
||
except:
|
||
try:
|
||
url = 'https://www.bt.cn/Api/getIpAddress';
|
||
str = httpGet(url)
|
||
writeFile(filename,ipaddress)
|
||
return str
|
||
except:
|
||
pass
|
||
|
||
def get_error_info():
|
||
errorMsg = traceback.format_exc();
|
||
return errorMsg
|
||
|
||
|
||
def get_server_status(name):
|
||
try:
|
||
serviceStatus = win32serviceutil.QueryServiceStatus(name)
|
||
if serviceStatus[1] == 4:
|
||
return 1
|
||
return 0
|
||
except :
|
||
return -1
|
||
|
||
def start_service(name):
|
||
|
||
try:
|
||
timeout = 0;
|
||
while get_server_status(name) == 0:
|
||
try:
|
||
win32serviceutil.StartService(name)
|
||
time.sleep(1);
|
||
except : time.sleep(1);
|
||
timeout += 1
|
||
if timeout > 10:break
|
||
|
||
if get_server_status(name) != 0:
|
||
return True,None
|
||
return False,'操作失败，10秒内未完成启动服务【{}】'.format(name)
|
||
except :
|
||
return False,get_error_info()
|
||
|
||
def stop_service(name):
|
||
try:
|
||
timeout = 0;
|
||
while get_server_status(name) == 1:
|
||
try:
|
||
win32serviceutil.StopService(name)
|
||
time.sleep(1);
|
||
except : time.sleep(1);
|
||
timeout += 1
|
||
if timeout > 10:break
|
||
|
||
if get_server_status(name) != 1:
|
||
return True,None
|
||
return False,'操作失败，10秒内未完成启动服务【{}】'.format(name)
|
||
except :
|
||
return False,get_error_info()
|
||
|
||
def delete_server(name):
|
||
try:
|
||
stop_service(name)
|
||
win32serviceutil.RemoveService(name)
|
||
return True,''
|
||
except :
|
||
return False,get_error_info()
|
||
|
||
def get_requests_headers():
|
||
return {"Content-type":"application/x-www-form-urlencoded","User-Agent":"BT-Panel"}
|
||
|
||
def downloadFile(url,filename):
|
||
try:
|
||
import requests
|
||
res = requests.get(url,verify=False)
|
||
with open(filename,"wb") as f:
|
||
f.write(res.content)
|
||
except:
|
||
import requests
|
||
res = requests.get(url,verify=False)
|
||
with open(filename,"wb") as f:
|
||
f.write(res.content)
|
||
|
||
|
||
def downloadFileByWget(url,filename):
|
||
"""
|
||
wget下载文件
|
||
@url 下载地址
|
||
@filename 本地文件路径
|
||
"""
|
||
try:
|
||
if os.path.exists(logPath): os.remove(logPath)
|
||
except : pass
|
||
loacl_path = '{}/script/wget.exe'.format(panelPath)
|
||
if not os.path.exists(loacl_path): downloadFile(get_url()+'/win/panel/data/wget.exe',loacl_path)
|
||
|
||
if os.path.getsize(loacl_path) < 10:
|
||
os.remove(loacl_path)
|
||
downloadFile(url,filename)
|
||
else:
|
||
shell = "{} {} -O {} -t 5 -T 60 --no-check-certificate --auth-no-challenge --force-directorie > {} 2>&1".format(loacl_path,url,filename,logPath)
|
||
os.system(shell)
|
||
|
||
num = 0
|
||
re_size = 0
|
||
while num <= 5:
|
||
if os.path.exists(filename):
|
||
cr_size = os.path.getsize(filename)
|
||
if re_size > 0 and re_size == cr_size:
|
||
break;
|
||
else:
|
||
re_size = cr_size
|
||
time.sleep(0.5)
|
||
num += 1
|
||
|
||
if os.path.exists(filename):
|
||
if os.path.getsize(filename) < 1:
|
||
os.remove(filename)
|
||
downloadFile(url,filename)
|
||
else:
|
||
downloadFile(url,filename)
|
||
|
||
def writeFile(filename,s_body,mode='w+',encoding = 'utf-8'):
|
||
try:
|
||
fp = open(filename, mode,encoding = encoding);
|
||
fp.write(s_body)
|
||
fp.close()
|
||
return True
|
||
except:
|
||
return False
|
||
|
||
def readFile(filename,mode = 'r'):
|
||
|
||
import os,chardet
|
||
if not os.path.exists(filename): return False
|
||
if not os.path.isfile(filename): return False
|
||
|
||
encoding = 'utf-8'
|
||
f_body = '';
|
||
try:
|
||
fp = open(filename, mode,encoding = encoding)
|
||
f_body = fp.read()
|
||
except :
|
||
fp.close()
|
||
|
||
try:
|
||
encoding = 'gbk'
|
||
fp = open(filename, mode,encoding = encoding)
|
||
f_body = fp.read()
|
||
except :
|
||
fp.close()
|
||
|
||
encoding = 'ansi'
|
||
fp = open(filename, mode,encoding = encoding)
|
||
f_body = fp.read()
|
||
|
||
try:
|
||
if f_body[0] == '\ufeff':
|
||
#处理带bom格式
|
||
new_code = chardet.detect(f_body.encode(encoding))["encoding"]
|
||
f_body = f_body.encode(encoding).decode(new_code);
|
||
except : pass
|
||
|
||
fp.close()
|
||
return f_body
|
||
|
||
def httpGet(url,timeout = 60,headers = {}):
|
||
try:
|
||
import urllib.request,ssl
|
||
try:
|
||
ssl._create_default_https_context = ssl._create_unverified_context
|
||
except:pass;
|
||
req = urllib.request.Request(url,headers = headers)
|
||
response = urllib.request.urlopen(req,timeout = timeout)
|
||
result = response.read()
|
||
if type(result) == bytes:
|
||
try:
|
||
result = result.decode('utf-8')
|
||
except :
|
||
result = result.decode('gb2312')
|
||
return result
|
||
except Exception as ex:
|
||
if headers: return False
|
||
return str(ex)
|
||
|
||
def httpPost(url, data, timeout=60, headers={}):
|
||
|
||
try:
|
||
import urllib.request,ssl
|
||
try:
|
||
ssl._create_default_https_context = ssl._create_unverified_context
|
||
except:pass;
|
||
data2 = urllib.parse.urlencode(data).encode('utf-8')
|
||
req = urllib.request.Request(url, data2,headers = headers)
|
||
response = urllib.request.urlopen(req,timeout = timeout)
|
||
result = response.read()
|
||
if type(result) == bytes: result = result.decode('utf-8')
|
||
|
||
return result
|
||
except Exception as ex:
|
||
|
||
return str(ex);
|
||
|
||
|
||
def get_timeout(url,timeout=3):
|
||
|
||
try:
|
||
start = time.time()
|
||
result = int(httpGet(url,timeout))
|
||
return result,int((time.time() - start) * 1000 - 500)
|
||
except: return 0,False
|
||
|
||
def get_url(timeout = 0.5):
|
||
import json
|
||
try:
|
||
#
|
||
node_list = [{"protocol":"http://","address":"dg2.bt.cn","port":"80","ping":500},{"protocol":"http://","address":"dg1.bt.cn","port":"80","ping":500},{"protocol":"http://","address":"download.bt.cn","port":"80","ping":500},{"protocol":"http://","address":"hk1-node.bt.cn","port":"80","ping":500},{"protocol":"http://","address":"na1-node.bt.cn","port":"80","ping":500},{"protocol":"http://","address":"jp1-node.bt.cn","port":"80","ping":500}]
|
||
|
||
mnode1 = []
|
||
mnode2 = []
|
||
mnode3 = []
|
||
for node in node_list:
|
||
node['net'],node['ping'] = get_timeout(node['protocol'] + node['address'] + ':' + node['port'] + '/net_test',1)
|
||
if not node['ping']: continue
|
||
if node['ping'] < 100: #当响应时间<100ms且可用带宽大于1500KB时
|
||
if node['net'] > 1500:
|
||
mnode1.append(node)
|
||
elif node['net'] > 1000:
|
||
mnode3.append(node)
|
||
else:
|
||
if node['net'] > 1000: #当响应时间>=100ms且可用带宽大于1000KB时
|
||
mnode2.append(node)
|
||
if node['ping'] < 100:
|
||
if node['net'] > 3000: break #有节点可用带宽大于3000时，不再检查其它节点
|
||
if mnode1: #优选低延迟高带宽
|
||
mnode = sorted(mnode1,key= lambda x:x['net'],reverse=True)
|
||
elif mnode3: #备选低延迟，中等带宽
|
||
mnode = sorted(mnode3,key= lambda x:x['net'],reverse=True)
|
||
else: #终选中等延迟，中等带宽
|
||
mnode = sorted(mnode2,key= lambda x:x['ping'],reverse=False)
|
||
|
||
if not mnode: return 'https://download.bt.cn'
|
||
#return mnode[0]['protocol'] + mnode[0]['address'] + ':' + mnode[0]['port']
|
||
return "https://" + mnode[0]['address']
|
||
except:
|
||
return 'https://download.bt.cn'
|
||
|
||
|
||
|
||
#删除文件权限
|
||
def del_file_access(filename,user):
|
||
try:
|
||
|
||
if filename.lower() in ["c:/","c:","c:\\","c"]:
|
||
return True
|
||
import win32security
|
||
sd = win32security.GetFileSecurity(filename, win32security.DACL_SECURITY_INFORMATION)
|
||
dacl = sd.GetSecurityDescriptorDacl()
|
||
ace_count = dacl.GetAceCount()
|
||
|
||
for i in range(ace_count ,0 ,-1):
|
||
try:
|
||
data = {}
|
||
data['rev'], data['access'], usersid = dacl.GetAce(i-1)
|
||
data['user'],data['group'], data['type'] = win32security.LookupAccountSid('', usersid)
|
||
if data['user'].lower() == user.lower(): dacl.DeleteAce(i-1) #删除旧的dacl
|
||
if data['user'].lower() == 'users': dacl.DeleteAce(i-1) #删除旧的dacl
|
||
|
||
except :
|
||
try:
|
||
#处理拒绝访问
|
||
dacl.DeleteAce(i-1)
|
||
except : pass
|
||
sd.SetSecurityDescriptorDacl(1, dacl, 0)
|
||
win32security.SetFileSecurity(filename, win32security.DACL_SECURITY_INFORMATION, sd)
|
||
except :
|
||
pass
|
||
return True
|
||
|
||
def set_file_access(filename,user,access):
|
||
try:
|
||
sd = win32security.GetFileSecurity(filename, win32security.DACL_SECURITY_INFORMATION)
|
||
dacl = sd.GetSecurityDescriptorDacl()
|
||
ace_count = dacl.GetAceCount()
|
||
|
||
for i in range(ace_count, 0,-1):
|
||
try:
|
||
data = {}
|
||
data['rev'], data['access'], usersid = dacl.GetAce(i-1)
|
||
data['user'],data['group'], data['type'] = win32security.LookupAccountSid('', usersid)
|
||
if data['user'].lower() == user.lower(): dacl.DeleteAce(i-1) #删除旧的dacl
|
||
if data['user'].lower() == 'users': dacl.DeleteAce(i-1) #删除旧的dacl
|
||
|
||
except :
|
||
pass
|
||
try:
|
||
userx, domain, type = win32security.LookupAccountName("", user)
|
||
except :
|
||
userx, domain, type = win32security.LookupAccountName("", 'IIS APPPOOL\\' + user)
|
||
if access > 0: dacl.AddAccessAllowedAceEx(win32security.ACL_REVISION, 3, access, userx)
|
||
|
||
sd.SetSecurityDescriptorDacl(1, dacl, 0)
|
||
win32security.SetFileSecurity(filename, win32security.DACL_SECURITY_INFORMATION, sd)
|
||
return True,None
|
||
except :
|
||
return False,get_error_info()
|
||
|
||
def ExecShell(cmdstring, cwd=None, timeout=None, shell=True):
|
||
if shell:
|
||
cmdstring_list = cmdstring
|
||
else:
|
||
cmdstring_list = shlex.split(cmdstring)
|
||
|
||
if timeout:
|
||
end_time = datetime.datetime.now() + datetime.timedelta(seconds=timeout)
|
||
|
||
sub = subprocess.Popen(cmdstring_list, cwd=cwd, stdin=subprocess.PIPE,shell=shell,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
|
||
while sub.poll() is None:
|
||
time.sleep(0.1)
|
||
if timeout:
|
||
if end_time <= datetime.datetime.now():
|
||
raise Exception("Timeout：%s"%cmdstring)
|
||
a,e = sub.communicate()
|
||
if type(a) == bytes:
|
||
try:
|
||
a = a.decode('utf-8')
|
||
except :
|
||
a = a.decode('gb2312','ignore')
|
||
|
||
if type(e) == bytes:
|
||
try:
|
||
e = e.decode('utf-8')
|
||
except :
|
||
e = e.decode('gb2312','ignore')
|
||
return a,e
|
||
|
||
def GetRandomString(length):
|
||
from random import Random
|
||
strings = ''
|
||
chars = 'AaBbCcDdEeFfGgHhIiJjKkLlMmNnOoPpQqRrSsTtUuVvWwXxYyZz0123456789'
|
||
chrlen = len(chars) - 1
|
||
random = Random()
|
||
for i in range(length):
|
||
strings += chars[random.randint(0, chrlen)]
|
||
return strings
|
||
|
||
def GetRandomString1(length):
|
||
from random import Random
|
||
strings = ''
|
||
chars = '0123456789'
|
||
chrlen = len(chars) - 1
|
||
random = Random()
|
||
for i in range(length):
|
||
strings += chars[random.randint(0, chrlen)]
|
||
return strings
|
||
|
||
def GetRandomString2(length):
|
||
from random import Random
|
||
strings = ''
|
||
chars = '!@#$%^&*()_+.,?[]-='
|
||
chrlen = len(chars) - 1
|
||
random = Random()
|
||
for i in range(length):
|
||
strings += chars[random.randint(0, chrlen)]
|
||
return strings
|
||
|
||
def chdck_salt():
|
||
|
||
sql = Sql()
|
||
sql.table('users').execute("ALTER TABLE 'users' ADD 'salt' TEXT",())
|
||
|
||
u_list = sql.table('users').field('id,username,password,salt').select()
|
||
for u_info in u_list:
|
||
salt = GetRandomString(12) #12位随机
|
||
pdata = {}
|
||
pdata['password'] = md5(md5(u_info['password']+'_bt.cn') + salt)
|
||
pdata['salt'] = salt
|
||
sql.table('users').where('id=?',(u_info['id'],)).update(pdata)
|
||
|
||
def md5(strings):
|
||
"""
|
||
生成MD5
|
||
@strings 要被处理的字符串
|
||
return string(32)
|
||
"""
|
||
import hashlib
|
||
m = hashlib.md5()
|
||
|
||
m.update(strings.encode('utf-8'))
|
||
return m.hexdigest()
|
||
|
||
def password_salt(password,username=None,uid=None):
|
||
|
||
chdck_salt()
|
||
sql = Sql()
|
||
|
||
if not uid:
|
||
if not username:
|
||
raise Exception('username或uid必需传一项')
|
||
uid = sql.table('users').where('username=?',(username,)).getField('id')
|
||
salt = sql.table('users').where('id=?',(uid,)).getField('salt')
|
||
return md5(md5(password+'_bt.cn')+salt)
|
||
|
||
def check_user(username):
|
||
resume = 0
|
||
while True:
|
||
data, total, resume = win32net.NetUserEnum(None, 3, win32netcon.FILTER_NORMAL_ACCOUNT, resume)
|
||
for user in data:
|
||
if user['name'] == username: return True
|
||
if not resume: break
|
||
return False
|
||
|
||
def add_user(username,password,ps):
|
||
try:
|
||
if not check_user(username):
|
||
d = {}
|
||
d['name'] = username
|
||
d['password'] = password
|
||
d['comment'] = ps
|
||
d['flags'] = win32netcon.UF_NORMAL_ACCOUNT | win32netcon.UF_SCRIPT
|
||
d['priv'] = win32netcon.USER_PRIV_USER
|
||
win32net.NetUserAdd(None, 1, d)
|
||
|
||
#设置用户允许登录服务
|
||
handle = win32security.LsaOpenPolicy(None, win32security.POLICY_ALL_ACCESS)
|
||
sid_obj, domain, tmp = win32security.LookupAccountName(None, username)
|
||
win32security.LsaAddAccountRights(handle, sid_obj, ('SeServiceLogonRight',) )
|
||
win32security.LsaClose( handle)
|
||
|
||
if not check_user(username): return False, '添加用户[{}]失败.'.format(username)
|
||
writeFile('{}/data/{}'.format(panelPath,username),password)
|
||
return True , None
|
||
else:
|
||
ExecShell('net user "{}" "{}"'.format(username,password))
|
||
writeFile('{}/data/{}'.format(panelPath,username),password)
|
||
return True , None
|
||
except :
|
||
return False,get_error_info()
|
||
|
||
def add_user_bywww():
|
||
|
||
pwd = GetRandomString(64) + GetRandomString1(32) + GetRandomString2(32)
|
||
status,error = add_user('www',pwd,'用于启动宝塔安装的程序,删除后会导致部分软件无法启动,请勿删除')
|
||
if not status:
|
||
writeFile(error_path,error)
|
||
return False
|
||
return True
|
||
|
||
def add_user_bymysql():
|
||
|
||
pwd = GetRandomString(64) + GetRandomString1(32) + GetRandomString2(32)
|
||
status,error = add_user('mysql',pwd,'用于启动宝塔安装的程序,删除后会导致部分软件无法启动,请勿删除')
|
||
if not status:
|
||
writeFile(error_path,error)
|
||
return False
|
||
return True
|
||
|
||
def getIP(url):
|
||
import socket,re
|
||
|
||
tmp = re.search('http://(.+)\:\d*',url)
|
||
if tmp:
|
||
domain = tmp.groups()[0]
|
||
myaddr = socket.getaddrinfo(domain, 'http')
|
||
return myaddr[0][4][0]
|
||
return ''
|
||
|
||
|
||
def add_panel_dir():
|
||
try:
|
||
slist = [
|
||
[panelPath , [] ],
|
||
['{}/data'.format(panelPath) , [] ],
|
||
['{}/script'.format(panelPath) , [] ],
|
||
['{}/backup'.format(panelPath) , [] ],
|
||
['{}/backup/database/sqlserver'.format(setupPath[:2]) , [ 'Authenticated Users']],
|
||
['{}/wwwroot'.format(setupPath[:2]) , [ 'IIS_IUSRS','www'] ],
|
||
['{}/wwwlogs'.format(setupPath) , [ 'IIS_IUSRS','www'] ],
|
||
['{}/php'.format(setupPath) , [ 'IIS_IUSRS','www'] ],
|
||
['{}/mysql'.format(setupPath) , [ 'mysql'] ],
|
||
['{}/temp'.format(setupPath) , [ 'IIS_IUSRS','www'] ],
|
||
['{}/temp/session'.format(setupPath) , [ 'IIS_IUSRS','www'] ],
|
||
['C:/Temp' , [ 'IIS_IUSRS','www'] ],
|
||
]
|
||
|
||
is_break = False
|
||
for sobj in slist:
|
||
if not os.path.exists(sobj[0]):
|
||
os.makedirs(sobj[0])
|
||
n = 0
|
||
while n < 5:
|
||
if os.path.exists(sobj[0]): break
|
||
|
||
os.makedirs(sobj[0])
|
||
time.sleep(0.5)
|
||
n += 1
|
||
|
||
if not os.path.exists(sobj[0]):
|
||
writeFile(error_path,"自动创建目录【{}】失败，已重试最大次数 5 次，请手动创建该目录后重新安装".format(sobj[0]))
|
||
return False
|
||
|
||
del_file_access(sobj[0],'users')
|
||
|
||
for user in sobj[1]:
|
||
n = 0
|
||
while n < 3:
|
||
status,error = set_file_access(sobj[0],user,2032127)
|
||
if status: break
|
||
time.sleep(0.5)
|
||
|
||
if not status:
|
||
writeFile(error_path,"目录{}设置{}权限设置错误 -> {}".format(sobj[0],user,error))
|
||
break
|
||
|
||
del_file_access(setupPath,'users')
|
||
url = get_url()
|
||
|
||
files = ['default.db','session.db','system.db','phplib.win','defaultDoc.html','404.html']
|
||
for f_name in files:
|
||
local_path = '{}/data/{}'.format(panelPath,f_name)
|
||
download_url = '{}/win/panel/data/{}'.format(url,f_name)
|
||
|
||
n = 0
|
||
while n < 10:
|
||
n += 1;
|
||
|
||
try:
|
||
if os.path.exists(local_path) and os.path.getsize(local_path) < 10: os.remove(local_path)
|
||
if not os.path.exists(local_path): downloadFileByWget(download_url,local_path)
|
||
if os.path.getsize(local_path) and os.path.getsize(local_path) > 10: break;
|
||
|
||
writeFile(error_path,'download {} error ->> {} \r\n {}'.format(f_name,download_url,""))
|
||
except :
|
||
ip = getIP(url)
|
||
writeFile(error_path,'download {} error ->> {} \r\n connect {} \r\n {}'.format(ip,f_name,download_url,get_error_info()))
|
||
|
||
if n > 5: return False
|
||
time.sleep(0.2)
|
||
|
||
return True
|
||
except :
|
||
writeFile(error_path,get_error_info())
|
||
return False
|
||
|
||
def unzip(src_path,dst_path):
|
||
import zipfile
|
||
zip_file = zipfile.ZipFile(src_path)
|
||
for names in zip_file.namelist():
|
||
zip_file.extract(names,dst_path)
|
||
zip_file.close()
|
||
return True
|
||
|
||
def to_path(path):
|
||
return path.replace('/','\\')
|
||
|
||
def download_panel(file_list = []):
|
||
try:
|
||
url = 'http://www.example.com'
|
||
|
||
ExecShell("taskkill /f /t /im BtTools.exe")
|
||
|
||
#下载面板
|
||
loacl_path = setupPath + '/panel.zip'
|
||
tmpPath = "{}/temp/panel".format(setupPath)
|
||
if os.path.exists(loacl_path): os.remove(loacl_path)
|
||
if os.path.exists(tmpPath): shutil.rmtree(tmpPath,True)
|
||
if not os.path.exists(tmpPath): os.makedirs(tmpPath)
|
||
|
||
p_ver = sys.argv[2]
|
||
downUrl = url + '/win/panel/panel_' + p_ver + '.zip';
|
||
downloadFileByWget(downUrl,loacl_path);
|
||
unzip(loacl_path,tmpPath)
|
||
|
||
for ff_path in file_list:
|
||
if os.path.exists(tmpPath + '/' + ff_path): os.remove(tmpPath + '/' + ff_path)
|
||
|
||
tcPath = '{}\class'.format(tmpPath)
|
||
for name in os.listdir(tcPath):
|
||
try:
|
||
if name.find('win_amd64.pyd') >=0:
|
||
oldName = os.path.join(tcPath,name);
|
||
lName = name.split('.')[0] + '.pyd'
|
||
newName = os.path.join(tcPath,lName)
|
||
if not os.path.exists(newName):os.rename(oldName,newName)
|
||
except :pass
|
||
|
||
cPath = '{}/panel/class'.format(setupPath)
|
||
|
||
if os.path.exists(cPath):
|
||
os.system("del /s {}\*.pyc".format(to_path(cPath)))
|
||
os.system("del /s {}\*.pyt".format(to_path(cPath)))
|
||
for name in os.listdir(cPath):
|
||
try:
|
||
if name.find('.pyd') >=0:
|
||
oldName = os.path.join(cPath,name)
|
||
newName = os.path.join(cPath,GetRandomString(8) + '.pyt')
|
||
os.rename(oldName,newName)
|
||
except : pass
|
||
os.system("del /s {}\*.pyc".format(to_path(cPath)))
|
||
os.system("del /s {}\*.pyt".format(to_path(cPath)))
|
||
|
||
os.system("xcopy /s /c /e /y /r {} {}".format(to_path(tmpPath),to_path(panelPath)))
|
||
try:
|
||
os.remove(loacl_path)
|
||
except : pass
|
||
|
||
try:
|
||
shutil.rmtree(tmpPath,True)
|
||
except : pass
|
||
|
||
s_ver = platform.platform()
|
||
net_v = '45'
|
||
if s_ver.find('2008') >= 0: net_v = '20'
|
||
writeFile('{}/data/net'.format(setupPath),net_v)
|
||
|
||
not_workorder_path = '{}/data/not_workorder.pl'.format(panelPath)
|
||
if not os.path.exists(not_workorder_path):
|
||
writeFile(not_workorder_path,'True')
|
||
bind_path = '{}/data/bind_path.pl'.format(panelPath)
|
||
if os.path.exists(bind_path):
|
||
os.remove(bind_path)
|
||
userinfo_path = '{}/data/userInfo.json'.format(panelPath)
|
||
if not os.path.exists(userinfo_path):
|
||
writeFile(userinfo_path,'{"uid":1,"username":"Administrator","address":"127.0.0.1","serverid":"1","access_key":"test","secret_key":"123456","ukey":"123456","state":1}')
|
||
|
||
local_path = '{}/temp/api.py'.format(setupPath)
|
||
downloadFileByWget('{}/win/panel/data/api.py'.format(url),local_path)
|
||
if os.path.exists(local_path):
|
||
os.remove('C:/Program Files/python/Lib/site-packages/requests/api.py')
|
||
shutil.move(local_path,'C:/Program Files/python/Lib/site-packages/requests')
|
||
|
||
local_path = '{}/script/BtTools.exe'.format(panelPath)
|
||
downloadFileByWget('{}/win/panel/BtTools{}.exe'.format(url,net_v),local_path)
|
||
if os.path.getsize(local_path) > 128:
|
||
return True
|
||
return False
|
||
downloadFileByWget('{}/win/panel/data/softList.conf'.format(url),'{}/data/softList.conf'.format(panelPath))
|
||
try:
|
||
from gevent import monkey
|
||
except :
|
||
os.system('"C:\Program Files\python\python.exe" -m pip install gevent')
|
||
except :
|
||
writeFile(error_path,get_error_info())
|
||
|
||
def update_panel():
|
||
|
||
file_list = ['config/config.json','config/index.json','data/libList.conf','data/plugin.json']
|
||
download_panel(file_list)
|
||
|
||
py_path = 'C:/Program Files/python/python.exe'
|
||
|
||
ExecShell("\"{}\" {}/panel/runserver.py --startup auto install".format(py_path,setupPath))
|
||
ExecShell("\"{}\" {}/panel/task.py --startup auto install".format(py_path,setupPath))
|
||
|
||
print("升级成功，重启面板后生效..")
|
||
|
||
def init_panel_data():
|
||
try:
|
||
sql = Sql()
|
||
username = sql.table('users').where('id=?',(1,)).getField('username')
|
||
if username == 'admin':
|
||
username = GetRandomString(8)
|
||
password = GetRandomString(8)
|
||
writeFile(panelPath + '/data/default.pl',password)
|
||
|
||
sql.table('users').where('id=?',(1,)).setField('username',username)
|
||
pwd = password_salt(md5(password),uid=1)
|
||
|
||
result = sql.table('users').where('id=?',(1,)).setField('password',pwd)
|
||
|
||
backup_path = panelPath[:2] + '/backup'
|
||
www_path = panelPath[:2] + '/wwwroot'
|
||
|
||
if not os.path.exists(backup_path): os.makedirs(backup_path)
|
||
if not os.path.exists(www_path): os.makedirs(www_path)
|
||
|
||
sql.table('config').where('id=?',(1,)).setField('backup_path',backup_path)
|
||
sql.table('config').where('id=?',(1,)).setField('sites_path',www_path)
|
||
|
||
bind_path = panelPath+ '/data/bind_path.pl'
|
||
if not os.path.exists(bind_path): writeFile(bind_path,'True')
|
||
|
||
admin_path = panelPath+ '/data/admin_path.pl'
|
||
if not os.path.exists(admin_path): writeFile(admin_path,"/" + GetRandomString(8))
|
||
|
||
port_path = panelPath+ '/data/port.pl'
|
||
if not os.path.exists(port_path): writeFile(port_path,'8888')
|
||
|
||
recycle_bin_db = panelPath+ '/data/recycle_bin_db.pl'
|
||
if not os.path.exists(recycle_bin_db): writeFile(recycle_bin_db,'True')
|
||
|
||
recycle_bin = panelPath+ '/data/recycle_bin.pl'
|
||
if not os.path.exists(recycle_bin): writeFile(recycle_bin,'True')
|
||
|
||
conf_path = panelPath + '/config/config.json'
|
||
if os.path.exists(conf_path):
|
||
conf = readFile(conf_path).replace('[PATH]',setupPath.replace('\\','/'))
|
||
writeFile(conf_path,conf)
|
||
|
||
GetLocalIp()
|
||
|
||
return True
|
||
except :
|
||
writeFile(error_path,get_error_info())
|
||
return False
|
||
|
||
def add_panel_services(num = 0):
|
||
try:
|
||
py_path = 'C:/Program Files/python/python.exe'
|
||
|
||
delete_server('btPanel')
|
||
ret = ExecShell("\"{}\" {}/panel/runserver.py --startup auto install".format(py_path,setupPath))
|
||
|
||
delete_server('btTask')
|
||
ret1 = ExecShell("\"{}\" {}/panel/task.py --startup auto install".format(py_path,setupPath))
|
||
|
||
if get_server_status('btPanel') < 0 or get_server_status('btTask') < 0:
|
||
if num <= 0 :
|
||
localPath = setupPath + "/temp/Time_Zones.reg";
|
||
downloadFileByWget(get_url() + '/win/panel/data/Time_Zones.reg',localPath)
|
||
ExecShell("regedit /s " + localPath)
|
||
|
||
add_panel_services(1)
|
||
else:
|
||
writeFile(error_path,ret[0] + ret[1] + ret1[0] + ret1[1])
|
||
else:
|
||
os.system('sc failure btPanel reset=1800 actions=restart/60000/restart/120000/restart/30000')
|
||
os.system('sc failure btTask reset=1800 actions=restart/60000/restart/120000/restart/30000')
|
||
start_service('btPanel')
|
||
start_service('btTask')
|
||
except :
|
||
writeFile(error_path,get_error_info())
|
||
|
||
|
||
def add_firewall_byport():
|
||
|
||
conf = ExecShell('netsh advfirewall firewall show rule "宝塔面板"')[0]
|
||
if conf.lower().find('tcp') == -1:
|
||
ExecShell("netsh advfirewall firewall add rule name=宝塔面板 dir=in action=allow protocol=tcp localport=8888");
|
||
ExecShell("netsh advfirewall firewall add rule name=网站访问端口 dir=in action=allow protocol=tcp localport=80");
|
||
ExecShell("netsh advfirewall firewall add rule name=远程桌面 dir=in action=allow protocol=tcp localport=3389");
|
||
ExecShell("netsh advfirewall firewall add rule name=HTTPS端口 dir=in action=allow protocol=tcp localport=443");
|
||
ExecShell("netsh advfirewall firewall add rule name=FTP主动端口 dir=in action=allow protocol=tcp localport=21");
|
||
ExecShell("netsh advfirewall firewall add rule name=FTP被动端口 dir=in action=allow protocol=tcp localport=3000-4000");
|
||
|
||
def get_error_log():
|
||
error = readFile(error_path)
|
||
try:
|
||
data = {}
|
||
data['msg'] = 'setup'
|
||
data['os'] = 'Windows'
|
||
data['error'] = error
|
||
data['version'] = ''
|
||
httpPost('http://www.example.com/api/wpanel/PanelBug',data)
|
||
except :
|
||
pass
|
||
return error
|
||
|
||
if __name__ == "__main__":
|
||
stype = sys.argv[1];
|
||
if not stype in ['get_error_log']:
|
||
if os.path.exists(error_path): os.remove(error_path)
|
||
result = eval('{}()'.format(stype))
|
||
print(result)
|
||
|
||
|
||
|
||
|