mdserver-web/plugins/rsyncd/index.py

998 lines
26 KiB
Python
Raw Permalink Normal View History

2019-03-05 06:20:04 -05:00
# coding: utf-8
import time
import random
import os
import json
import re
import sys
2024-11-23 16:40:30 -05:00
web_dir = os.getcwd() + "/web"
if os.path.exists(web_dir):
sys.path.append(web_dir)
os.chdir(web_dir)
import core.mw as mw
2019-03-05 06:20:04 -05:00
app_debug = False
2020-07-10 03:57:25 -04:00
if mw.isAppleSystem():
2019-03-05 06:20:04 -05:00
app_debug = True
def getPluginName():
return 'rsyncd'
2019-09-12 10:38:34 -04:00
def getInitDTpl():
path = getPluginDir() + "/init.d/" + getPluginName() + ".tpl"
return path
2019-03-05 06:20:04 -05:00
def getPluginDir():
2020-07-10 03:57:25 -04:00
return mw.getPluginDir() + '/' + getPluginName()
2019-03-05 06:20:04 -05:00
def getServerDir():
2020-07-10 03:57:25 -04:00
return mw.getServerDir() + '/' + getPluginName()
2019-03-05 06:20:04 -05:00
def getInitDFile():
if app_debug:
return '/tmp/' + getPluginName()
return '/etc/init.d/' + getPluginName()
def getArgs():
args = sys.argv[2:]
tmp = {}
args_len = len(args)
if args_len == 1:
t = args[0].strip('{').strip('}')
t = t.split(':')
tmp[t[0]] = t[1]
elif args_len > 1:
for i in range(len(args)):
t = args[i].split(':')
tmp[t[0]] = t[1]
return tmp
def checkArgs(data, ck=[]):
for i in range(len(ck)):
if not ck[i] in data:
2020-07-10 03:57:25 -04:00
return (False, mw.returnJson(False, '参数:(' + ck[i] + ')没有!'))
return (True, mw.returnJson(True, 'ok'))
2019-03-05 06:20:04 -05:00
2019-09-12 10:38:34 -04:00
def contentReplace(content):
2020-07-10 03:57:25 -04:00
service_path = mw.getServerDir()
2019-09-12 10:38:34 -04:00
content = content.replace('{$SERVER_PATH}', service_path)
return content
2019-03-05 06:20:04 -05:00
def status():
2020-07-10 03:57:25 -04:00
data = mw.execShell(
2019-03-06 01:33:32 -05:00
"ps -ef|grep rsync |grep -v grep | grep -v python | awk '{print $2}'")
2019-03-05 06:20:04 -05:00
if data[0] == '':
return 'stop'
2022-07-29 23:49:57 -04:00
2022-07-29 23:50:39 -04:00
# data = mw.execShell(
# "ps -ef|grep lsyncd |grep -v grep | grep -v python | awk '{print $2}'")
# if data[0] == '':
# return 'stop'
2022-07-29 23:49:57 -04:00
2019-03-05 06:20:04 -05:00
return 'start'
2019-03-06 02:07:04 -05:00
def appConf():
2022-06-19 22:49:15 -04:00
return getServerDir() + '/rsyncd.conf'
2019-03-06 02:07:04 -05:00
2022-07-27 23:36:23 -04:00
def appAuthPwd(name):
2022-07-28 03:45:00 -04:00
nameDir = getServerDir() + '/receive/' + name
2022-07-27 23:36:23 -04:00
if not os.path.exists(nameDir):
mw.execShell("mkdir -p " + nameDir)
return nameDir + '/auth.db'
2019-03-06 07:31:29 -05:00
2019-03-06 02:07:04 -05:00
def getLog():
conf_path = appConf()
2020-07-10 03:57:25 -04:00
conf = mw.readFile(conf_path)
2024-07-24 01:16:21 -04:00
rep = r'log file\s*=\s*(.*)'
2019-03-06 02:07:04 -05:00
tmp = re.search(rep, conf)
if not tmp:
return ''
return tmp.groups()[0]
2022-07-28 03:45:00 -04:00
def getLsyncdLog():
path = getServerDir() + "/lsyncd.conf"
conf = mw.readFile(path)
2024-07-24 01:16:21 -04:00
rep = r'logfile\s*=\s*\"(.*)\"'
2022-07-28 03:45:00 -04:00
tmp = re.search(rep, conf)
if not tmp:
return ''
return tmp.groups()[0]
2019-04-02 03:13:41 -04:00
2022-11-01 13:04:18 -04:00
def __release_port(port):
2024-12-01 09:16:30 -05:00
from collections import namedtuple
2022-11-01 13:04:18 -04:00
try:
2024-12-01 09:16:30 -05:00
from utils.firewall import Firewall as MwFirewall
MwFirewall.instance().addAcceptPort(port, 'RSYNC同步', 'port')
2022-11-01 13:04:18 -04:00
return port
except Exception as e:
return "Release failed {}".format(e)
def openPort():
for i in ["873"]:
__release_port(i)
return True
2022-07-28 03:45:00 -04:00
def initDReceive():
2022-06-19 22:49:15 -04:00
# conf
conf_path = appConf()
2019-03-06 02:07:04 -05:00
conf_tpl_path = getPluginDir() + '/conf/rsyncd.conf'
2022-06-19 22:58:23 -04:00
if not os.path.exists(conf_path):
2020-07-10 03:57:25 -04:00
content = mw.readFile(conf_tpl_path)
mw.writeFile(conf_path, content)
2022-06-19 22:49:15 -04:00
2019-09-12 10:38:34 -04:00
initD_path = getServerDir() + '/init.d'
if not os.path.exists(initD_path):
os.mkdir(initD_path)
2019-03-06 02:07:04 -05:00
2022-07-28 03:45:00 -04:00
file_bin = initD_path + '/' + getPluginName()
2019-09-12 10:38:34 -04:00
file_tpl = getInitDTpl()
2022-07-28 03:45:00 -04:00
# print(file_bin, file_tpl)
2019-09-12 10:38:34 -04:00
# initd replace
if not os.path.exists(file_bin):
2020-07-10 03:57:25 -04:00
content = mw.readFile(file_tpl)
2019-09-12 10:38:34 -04:00
content = contentReplace(content)
2020-07-10 03:57:25 -04:00
mw.writeFile(file_bin, content)
mw.execShell('chmod +x ' + file_bin)
2019-03-06 02:07:04 -05:00
2022-08-24 03:16:59 -04:00
lock_file = getServerDir() + "/installed_rsyncd.pl"
2022-06-18 22:20:52 -04:00
# systemd
2022-07-11 01:49:25 -04:00
systemDir = mw.systemdCfgDir()
2022-06-18 22:20:52 -04:00
systemService = systemDir + '/rsyncd.service'
systemServiceTpl = getPluginDir() + '/init.d/rsyncd.service.tpl'
2022-08-24 03:31:44 -04:00
if not os.path.exists(lock_file):
2022-11-01 13:04:18 -04:00
2022-06-18 22:20:52 -04:00
rsync_bin = mw.execShell('which rsync')[0].strip()
2022-07-07 23:27:32 -04:00
if rsync_bin == '':
2022-07-27 23:36:23 -04:00
print('rsync missing!')
2022-07-07 23:27:32 -04:00
exit(0)
2022-06-18 22:20:52 -04:00
service_path = mw.getServerDir()
2022-07-28 03:45:00 -04:00
se = mw.readFile(systemServiceTpl)
se = se.replace('{$SERVER_PATH}', service_path)
se = se.replace('{$RSYNC_BIN}', rsync_bin)
mw.writeFile(systemService, se)
2022-06-18 22:20:52 -04:00
mw.execShell('systemctl daemon-reload')
2019-09-12 12:02:55 -04:00
2022-08-24 03:16:59 -04:00
mw.writeFile(lock_file, "ok")
2022-11-01 13:04:18 -04:00
openPort()
2022-08-24 03:16:59 -04:00
2019-09-12 13:42:30 -04:00
rlog = getLog()
2019-09-12 13:43:05 -04:00
if os.path.exists(rlog):
2020-07-10 03:57:25 -04:00
mw.writeFile(rlog, '')
2019-09-12 10:38:34 -04:00
return file_bin
2019-03-05 06:20:04 -05:00
2019-09-12 06:48:57 -04:00
2022-07-28 03:45:00 -04:00
def initDSend():
service_path = mw.getServerDir()
conf_path = getServerDir() + '/lsyncd.conf'
conf_tpl_path = getPluginDir() + '/conf/lsyncd.conf'
if not os.path.exists(conf_path):
content = mw.readFile(conf_tpl_path)
content = content.replace('{$SERVER_PATH}', service_path)
mw.writeFile(conf_path, content)
initD_path = getServerDir() + '/init.d'
if not os.path.exists(initD_path):
os.mkdir(initD_path)
# initd replace
file_bin = initD_path + '/lsyncd'
file_tpl = getPluginDir() + "/init.d/lsyncd.tpl"
if not os.path.exists(file_bin):
content = mw.readFile(file_tpl)
content = contentReplace(content)
mw.writeFile(file_bin, content)
mw.execShell('chmod +x ' + file_bin)
2022-07-28 04:01:17 -04:00
lock_file = getServerDir() + "/installed.pl"
2022-07-28 03:45:00 -04:00
# systemd
systemDir = mw.systemdCfgDir()
systemService = systemDir + '/lsyncd.service'
systemServiceTpl = getPluginDir() + '/init.d/lsyncd.service.tpl'
2022-07-28 04:01:17 -04:00
if not os.path.exists(lock_file):
2022-07-28 03:45:00 -04:00
lsyncd_bin = mw.execShell('which lsyncd')[0].strip()
if lsyncd_bin == '':
print('lsyncd missing!')
exit(0)
content = mw.readFile(systemServiceTpl)
content = content.replace('{$SERVER_PATH}', service_path)
content = content.replace('{$LSYNCD_BIN}', lsyncd_bin)
mw.writeFile(systemService, content)
mw.execShell('systemctl daemon-reload')
2022-08-24 03:16:59 -04:00
mw.writeFile(lock_file, "ok")
2022-07-28 04:01:17 -04:00
2022-07-28 03:45:00 -04:00
lslog = getLsyncdLog()
if os.path.exists(lslog):
mw.writeFile(lslog, '')
return file_bin
2022-07-28 11:08:34 -04:00
def getDefaultConf():
path = getServerDir() + "/config.json"
data = mw.readFile(path)
data = json.loads(data)
return data
def setDefaultConf(data):
path = getServerDir() + "/config.json"
mw.writeFile(path, json.dumps(data))
return True
def initConfigJson():
path = getServerDir() + "/config.json"
tpl = getPluginDir() + "/conf/config.json"
if not os.path.exists(path):
data = mw.readFile(tpl)
data = json.loads(data)
mw.writeFile(path, json.dumps(data))
2022-07-28 03:45:00 -04:00
def initDreplace():
initDSend()
# conf
file_bin = initDReceive()
2022-07-28 11:08:34 -04:00
initConfigJson()
2022-07-28 03:45:00 -04:00
return file_bin
2022-06-19 07:44:41 -04:00
def rsyncOp(method):
2019-09-12 10:51:51 -04:00
file = initDreplace()
2022-06-18 22:20:52 -04:00
if not mw.isAppleSystem():
data = mw.execShell('systemctl ' + method + ' rsyncd')
if data[1] == '':
return 'ok'
return 'fail'
data = mw.execShell(file + ' ' + method)
2019-09-12 10:38:34 -04:00
if data[1] == '':
return 'ok'
return 'fail'
2019-03-05 06:20:04 -05:00
2025-01-09 22:48:20 -05:00
def lsyncdOp(method):
if not mw.isAppleSystem():
data = mw.execShell('systemctl ' + method + ' lsyncd')
if data[1] == '':
return 'ok'
return 'fail'
return 'fail'
2022-06-18 22:20:52 -04:00
2025-01-09 22:48:20 -05:00
def start():
status = rsyncOp('start')
lsyncdOp('start')
return status
2022-06-18 22:20:52 -04:00
2019-03-05 06:20:04 -05:00
def stop():
2025-01-09 22:48:20 -05:00
status = rsyncOp('stop')
lsyncdOp('stop')
return status
2019-03-05 06:20:04 -05:00
def restart():
2025-01-09 22:48:20 -05:00
status = rsyncOp('restart')
lsyncdOp('restart')
return status
2019-03-05 06:20:04 -05:00
def reload():
2025-01-09 22:48:20 -05:00
status = rsyncOp('reload')
lsyncdOp('reload')
return status
2019-03-05 06:20:04 -05:00
def initdStatus():
2022-06-18 22:20:52 -04:00
if mw.isAppleSystem():
return "Apple Computer does not support"
2019-03-06 07:31:29 -05:00
2022-06-18 22:20:52 -04:00
shell_cmd = 'systemctl status rsyncd | grep loaded | grep "enabled;"'
data = mw.execShell(shell_cmd)
if data[0] == '':
return 'fail'
2022-07-30 02:12:18 -04:00
shell_cmd = 'systemctl status lsyncd | grep loaded | grep "enabled;"'
data = mw.execShell(shell_cmd)
if data[0] == '':
return 'fail'
2022-06-18 22:20:52 -04:00
return 'ok'
2019-03-05 06:20:04 -05:00
2019-09-12 06:48:57 -04:00
2019-03-05 06:20:04 -05:00
def initdInstall():
2022-06-18 22:20:52 -04:00
if mw.isAppleSystem():
return "Apple Computer does not support"
2019-09-12 10:38:34 -04:00
2022-07-30 02:12:18 -04:00
mw.execShell('systemctl enable lsyncd')
2022-06-18 22:20:52 -04:00
mw.execShell('systemctl enable rsyncd')
2019-03-06 07:40:11 -05:00
return 'ok'
2019-03-05 06:20:04 -05:00
def initdUinstall():
2019-09-12 10:38:34 -04:00
if not app_debug:
2020-07-10 03:57:25 -04:00
if mw.isAppleSystem():
2019-09-12 10:38:34 -04:00
return "Apple Computer does not support"
2022-06-18 22:20:52 -04:00
2022-07-30 02:12:18 -04:00
mw.execShell('systemctl diable lsyncd')
2022-06-18 22:20:52 -04:00
mw.execShell('systemctl diable rsyncd')
2019-09-12 10:38:34 -04:00
return 'ok'
2019-03-05 06:20:04 -05:00
2019-03-06 05:47:43 -05:00
def getRecListData():
path = appConf()
2020-07-10 03:57:25 -04:00
content = mw.readFile(path)
2019-03-06 05:47:43 -05:00
2024-10-16 08:53:22 -04:00
flist = re.findall("\\[(.*)\\]", content)
2022-07-27 23:36:23 -04:00
2019-03-06 05:47:43 -05:00
flist_len = len(flist)
ret_list = []
for i in range(flist_len):
tmp = {}
tmp['name'] = flist[i]
n = i + 1
reg = ''
if n == flist_len:
2024-12-01 09:19:49 -05:00
reg = r'\[' + flist[i] + r'\](.*)\[?'
2019-03-06 05:47:43 -05:00
else:
2024-12-01 09:19:49 -05:00
reg = r'\[' + flist[i] + r'\](.*)\[' + flist[n] + r'\]'
2019-03-06 05:47:43 -05:00
t1 = re.search(reg, content, re.S)
if t1:
args = t1.groups()[0]
2022-07-27 23:36:23 -04:00
# print('args start', args, 'args_end')
2024-12-01 09:19:49 -05:00
t2 = re.findall(r'\s*(.*)\s*\=\s*?(.*)?', args, re.M | re.I)
2019-03-06 05:47:43 -05:00
for i in range(len(t2)):
2022-07-27 23:36:23 -04:00
tmp[t2[i][0].strip()] = t2[i][1].strip()
2019-03-06 05:47:43 -05:00
ret_list.append(tmp)
2022-07-27 23:36:23 -04:00
2019-03-06 05:47:43 -05:00
return ret_list
2022-07-28 11:08:34 -04:00
def getRecListDataBy(name):
l = getRecListData()
for x in range(len(l)):
if name == l[x]["name"]:
return l[x]
2019-03-06 05:47:43 -05:00
def getRecList():
ret_list = getRecListData()
2020-07-10 03:57:25 -04:00
return mw.returnJson(True, 'ok', ret_list)
2019-03-06 05:47:43 -05:00
def addRec():
args = getArgs()
2019-03-06 07:31:29 -05:00
data = checkArgs(args, ['name', 'path', 'pwd', 'ps'])
2019-03-06 05:47:43 -05:00
if not data[0]:
return data[1]
args_name = args['name']
2019-03-06 07:31:29 -05:00
args_pwd = args['pwd']
2019-03-06 05:47:43 -05:00
args_path = args['path']
args_ps = args['ps']
if not mw.isAppleSystem():
2025-01-08 02:01:48 -05:00
if os.path.exists(args_path):
import utils.file as utils_file
info = utils_file.getAccess(args_path)
file_chown = info['chown']
if file_chown != 'www':
return mw.returnJson(False, '建议手动执行命令: chown -R www:www '+ args_path)
else:
os.system("mkdir -p " + args_path + " &")
os.system("chown -R www:www " + args_path + " &")
os.system("chmod -R 755 " + args_path + " &")
2022-07-28 11:08:34 -04:00
delRecBy(args_name)
2022-07-27 23:36:23 -04:00
auth_path = appAuthPwd(args_name)
pwd_content = args_name + ':' + args_pwd + "\n"
mw.writeFile(auth_path, pwd_content)
mw.execShell("chmod 600 " + auth_path)
2019-03-06 07:31:29 -05:00
2019-03-06 05:47:43 -05:00
path = appConf()
2020-07-10 03:57:25 -04:00
content = mw.readFile(path)
2019-03-06 05:47:43 -05:00
con = "\n\n" + '[' + args_name + ']' + "\n"
con += 'path = ' + args_path + "\n"
con += 'comment = ' + args_ps + "\n"
2019-03-06 07:31:29 -05:00
con += 'auth users = ' + args_name + "\n"
2022-07-27 23:36:23 -04:00
con += 'ignore errors' + "\n"
con += 'secrets file = ' + auth_path + "\n"
2019-03-06 05:47:43 -05:00
con += 'read only = false'
2022-07-28 11:08:34 -04:00
content = content.strip() + "\n" + con
2020-07-10 03:57:25 -04:00
mw.writeFile(path, content)
return mw.returnJson(True, '添加成功')
2019-03-06 05:47:43 -05:00
2022-07-28 11:08:34 -04:00
def getRec():
2019-03-06 05:47:43 -05:00
args = getArgs()
data = checkArgs(args, ['name'])
if not data[0]:
return data[1]
2022-07-28 11:08:34 -04:00
name = args['name']
2019-03-06 07:31:29 -05:00
2022-07-28 11:08:34 -04:00
if name == "":
tmp = {}
tmp["name"] = ""
2022-07-29 14:47:00 -04:00
tmp["comment"] = ""
2022-07-28 11:08:34 -04:00
tmp["path"] = mw.getWwwDir()
tmp["pwd"] = mw.getRandomString(16)
return mw.returnJson(True, 'OK', tmp)
data = getRecListDataBy(name)
content = mw.readFile(data['secrets file'])
pwd = content.strip().split(":")
data['pwd'] = pwd[1]
return mw.returnJson(True, 'OK', data)
2019-03-06 07:31:29 -05:00
2022-07-28 11:08:34 -04:00
def delRecBy(name):
try:
2019-03-06 07:31:29 -05:00
path = appConf()
2020-07-10 03:57:25 -04:00
content = mw.readFile(path)
2019-03-06 07:31:29 -05:00
2022-07-28 11:08:34 -04:00
reclist = getRecListData()
ret_list_len = len(reclist)
2019-03-06 07:31:29 -05:00
is_end = False
next_name = ''
for x in range(ret_list_len):
2022-07-28 11:08:34 -04:00
tmp = reclist[x]
if tmp['name'] == name:
secrets_file = tmp['secrets file']
tp = os.path.dirname(secrets_file)
if os.path.exists(tp):
mw.execShell("rm -rf " + tp)
2019-03-06 07:31:29 -05:00
if x + 1 == ret_list_len:
is_end = True
else:
2022-07-28 11:08:34 -04:00
next_name = reclist[x + 1]['name']
2019-03-06 07:31:29 -05:00
reg = ''
if is_end:
2024-12-01 09:16:30 -05:00
reg = r'\[' + name + r'\]\s*(.*)'
2019-03-06 07:31:29 -05:00
else:
2024-12-01 09:16:30 -05:00
reg = r'\\[' + name + r'\]\s*(.*)\s*\[' + next_name + r'\]'
2019-03-06 05:47:43 -05:00
2019-03-06 07:31:29 -05:00
conre = re.search(reg, content, re.S)
2024-12-01 09:16:30 -05:00
content = content.replace("[" + name + "]\n" + conre.groups()[0], '')
2020-07-10 03:57:25 -04:00
mw.writeFile(path, content)
2019-03-06 07:31:29 -05:00
except Exception as e:
2022-07-28 11:08:34 -04:00
return False
return True
2019-03-06 05:47:43 -05:00
2022-07-28 11:08:34 -04:00
def delRec():
2019-03-06 07:31:29 -05:00
args = getArgs()
data = checkArgs(args, ['name'])
if not data[0]:
return data[1]
2022-07-28 11:08:34 -04:00
name = args['name']
ok = delRecBy(name)
if ok:
return mw.returnJson(True, '删除成功!')
return mw.returnJson(False, '删除失败!')
2019-03-06 07:31:29 -05:00
2022-07-28 11:08:34 -04:00
def cmdRecSecretKey():
import base64
args = getArgs()
data = checkArgs(args, ['name'])
if not data[0]:
return data[1]
name = args['name']
info = getRecListDataBy(name)
2022-07-29 12:06:19 -04:00
secrets_file = info['secrets file']
content = mw.readFile(info['secrets file'])
pwd = content.strip().split(":")
m = {"A": info['name'], "B": pwd[1], "C": "873"}
m = json.dumps(m)
2022-07-28 11:08:34 -04:00
m = m.encode("utf-8")
m = base64.b64encode(m)
cmd = m.decode("utf-8")
return mw.returnJson(True, 'OK!', cmd)
def cmdRecCmd():
args = getArgs()
data = checkArgs(args, ['name'])
if not data[0]:
return data[1]
name = args['name']
info = getRecListDataBy(name)
2020-07-10 03:57:25 -04:00
ip = mw.getLocalIp()
2019-03-06 07:31:29 -05:00
2022-07-28 11:08:34 -04:00
content = mw.readFile(info['secrets file'])
pwd = content.strip().split(":")
tmp_name = '/tmp/' + name + '.pass'
cmd = 'echo "' + pwd[1] + '" > ' + tmp_name + '<br>'
cmd += 'chmod 600 ' + tmp_name + ' <br>'
cmd += 'rsync -arv --password-file=' + tmp_name + \
' --progress --delete /project ' + name + '@' + ip + '::' + name
2020-07-10 03:57:25 -04:00
return mw.returnJson(True, 'OK!', cmd)
2019-03-06 05:47:43 -05:00
2022-07-28 11:08:34 -04:00
# ----------------------------- rsyncdSend start -------------------------
2022-07-30 02:14:40 -04:00
def lsyncdReload():
2022-11-01 10:25:01 -04:00
data = mw.execShell(
"ps -ef|grep lsyncd |grep -v grep | grep -v python | awk '{print $2}'")
if data[0] == '':
mw.execShell('systemctl start lsyncd')
else:
mw.execShell('systemctl restart lsyncd')
2022-07-30 02:14:40 -04:00
2022-07-29 14:28:47 -04:00
def makeLsyncdConf(data):
# print(data)
lsyncd_data = data['send']
lsyncd_setting = lsyncd_data['default']
2022-07-30 02:34:31 -04:00
content = "settings {\n"
2022-07-29 14:28:47 -04:00
for x in lsyncd_setting:
v = lsyncd_setting[x]
# print(v, type(v))
if type(v) == str:
content += "\t" + x + ' = "' + v + "\",\n"
elif type(v) == int:
content += "\t" + x + ' = ' + str(v) + ",\n"
content += "}\n\n"
lsyncd_list = lsyncd_data['list']
rsync_bin = mw.execShell('which rsync')[0].strip()
send_dir = getServerDir() + "/send"
if len(lsyncd_list) > 0:
for x in range(len(lsyncd_list)):
2022-07-30 06:10:31 -04:00
2022-07-29 14:28:47 -04:00
t = lsyncd_list[x]
name_dir = send_dir + "/" + t["name"]
if not os.path.exists(name_dir):
mw.execShell("mkdir -p " + name_dir)
cmd_exclude = name_dir + "/exclude"
2022-07-30 02:02:58 -04:00
cmd_exclude_txt = ""
for x in t['exclude']:
cmd_exclude_txt += x + "\n"
mw.writeFile(cmd_exclude, cmd_exclude_txt)
2022-07-29 14:28:47 -04:00
cmd_pass = name_dir + "/pass"
2025-11-12 12:30:04 -05:00
if os.path.exists(cmd_pass):
mw.execShell("chmod 755 " + cmd_pass)
2022-07-29 14:28:47 -04:00
mw.writeFile(cmd_pass, t['password'])
2022-07-29 23:49:57 -04:00
mw.execShell("chmod 600 " + cmd_pass)
2022-07-29 14:28:47 -04:00
2022-07-30 04:45:25 -04:00
delete_ok = ' '
if t['delete'] == "true":
delete_ok = ' --delete '
2022-07-29 14:28:47 -04:00
remote_addr = t['name'] + '@' + t['ip'] + "::" + t['name']
cmd = rsync_bin + " -avzP " + "--port=" + str(t['rsync']['port']) + " --bwlimit=" + t['rsync'][
2022-07-30 04:45:25 -04:00
'bwlimit'] + delete_ok + " --exclude-from=" + cmd_exclude + " --password-file=" + cmd_pass + " " + t["path"] + " " + remote_addr
2022-07-29 14:28:47 -04:00
mw.writeFile(name_dir + "/cmd", cmd)
mw.execShell("cmod +x " + name_dir + "/cmd")
2022-07-30 06:10:31 -04:00
if t['realtime'] == "false":
continue
2022-07-29 14:28:47 -04:00
# print(x, t)
content += "sync {\n"
content += "\tdefault.rsync,\n"
content += "\tsource = \"" + t['path'] + "\",\n"
content += "\ttarget = \"" + remote_addr + "\",\n"
content += "\tdelete = " + t['delete'] + ",\n"
content += "\tdelay = " + t['delay'] + ",\n"
content += "\tinit = false,\n"
2022-07-30 02:02:58 -04:00
exclude_str = json.dumps(t['exclude'])
exclude_str = exclude_str.replace("[", "{")
exclude_str = exclude_str.replace("]", "}")
# print(exclude_str)
content += "\texclude = " + exclude_str + ",\n"
2022-07-29 14:28:47 -04:00
# rsync
2022-07-30 02:34:47 -04:00
content += "\trsync = {\n"
2022-07-29 14:28:47 -04:00
content += "\t\tbinary = \"" + rsync_bin + "\",\n"
content += "\t\tarchive = true,\n"
content += "\t\tverbose = true,\n"
content += "\t\tcompress = " + t['rsync']['compress'] + ",\n"
2022-07-30 02:31:46 -04:00
content += "\t\tpassword_file = \"" + cmd_pass + "\",\n"
2022-07-30 02:27:00 -04:00
2022-07-29 14:28:47 -04:00
content += "\t\t_extra = {\"--bwlimit=" + t['rsync'][
2022-07-30 02:21:15 -04:00
'bwlimit'] + "\", \"--port=" + str(t['rsync']['port']) + "\"},\n"
2022-07-29 14:28:47 -04:00
content += "\t}\n"
content += "}\n"
path = getServerDir() + "/lsyncd.conf"
mw.writeFile(path, content)
2022-07-30 02:14:40 -04:00
lsyncdReload()
2022-07-30 06:05:21 -04:00
import tool_task
tool_task.createBgTask(lsyncd_list)
2022-07-29 14:28:47 -04:00
2022-07-28 11:08:34 -04:00
def lsyncdListFindIp(slist, ip):
for x in range(len(slist)):
if slist[x]["ip"] == ip:
return (True, x)
return (False, -1)
2022-07-29 12:06:19 -04:00
def lsyncdListFindName(slist, name):
for x in range(len(slist)):
if slist[x]["name"] == name:
return (True, x)
return (False, -1)
2022-07-28 11:08:34 -04:00
def lsyncdList():
data = getDefaultConf()
send = data['send']
return mw.returnJson(True, "设置成功!", send)
2022-07-29 12:06:19 -04:00
def lsyncdGet():
import base64
args = getArgs()
data = checkArgs(args, ['name'])
if not data[0]:
return data[1]
name = args['name']
data = getDefaultConf()
slist = data['send']["list"]
res = lsyncdListFindName(slist, name)
rsync = {
'bwlimit': "1024",
"compress": "true",
"archive": "true",
"verbose": "true"
}
info = {
"secret_key": '',
"ip": '',
2022-07-29 23:57:51 -04:00
"path": mw.getServerDir(),
2022-07-29 12:06:19 -04:00
'rsync': rsync,
'realtime': "true",
2022-07-30 04:03:36 -04:00
'delete': "false",
2022-07-29 12:06:19 -04:00
}
if res[0]:
list_index = res[1]
info = slist[list_index]
m = {"A": info['name'], "B": info["password"], "C": "873"}
m = json.dumps(m)
m = m.encode("utf-8")
m = base64.b64encode(m)
info['secret_key'] = m.decode("utf-8")
return mw.returnJson(True, "OK", info)
def lsyncdDelete():
args = getArgs()
data = checkArgs(args, ['name'])
if not data[0]:
return data[1]
name = args['name']
data = getDefaultConf()
slist = data['send']["list"]
res = lsyncdListFindName(slist, name)
retdata = {}
if res[0]:
list_index = res[1]
slist.pop(list_index)
data['send']["list"] = slist
setDefaultConf(data)
2022-07-29 14:36:29 -04:00
makeLsyncdConf(data)
2022-07-29 12:06:19 -04:00
return mw.returnJson(True, "OK")
2022-07-28 11:08:34 -04:00
def lsyncdAdd():
2022-07-29 12:06:19 -04:00
import base64
2022-07-28 11:08:34 -04:00
args = getArgs()
2025-01-09 23:04:30 -05:00
data = checkArgs(args, ['ip', 'conn_type', 'path', 'delay', 'period', 'bwlimit'])
2022-07-28 11:08:34 -04:00
if not data[0]:
return data[1]
ip = args['ip']
path = args['path']
2022-07-29 12:06:19 -04:00
if not mw.isAppleSystem():
2025-01-08 06:45:32 -05:00
if os.path.exists(path):
import utils.file as utils_file
info = utils_file.getAccess(path)
file_chown = info['chown']
if file_chown != 'www':
return mw.returnJson(False, '建议手动执行命令: chown -R www:www '+ args_path)
else:
os.system("mkdir -p " + path + " &")
os.system("chown -R www:www " + path + " &")
os.system("chmod -R 755 " + path + " &")
2022-07-29 12:06:19 -04:00
conn_type = args['conn_type']
2022-11-25 03:43:13 -05:00
2022-07-29 12:06:19 -04:00
delete = args['delete']
realtime = args['realtime']
delay = args['delay']
bwlimit = args['bwlimit']
compress = args['compress']
period = args['period']
hour = args['hour']
minute = args['minute']
minute_n = args['minute-n']
2022-07-28 11:08:34 -04:00
info = {
"ip": ip,
2022-07-29 12:06:19 -04:00
"path": path,
"delete": delete,
"realtime": realtime,
'delay': delay,
"conn_type": conn_type,
"period": period,
"hour": hour,
"minute": minute,
"minute-n": minute_n,
}
if conn_type == "key":
2022-11-25 03:43:13 -05:00
secret_key_check = checkArgs(args, ['secret_key'])
if not secret_key_check[0]:
return secret_key_check[1]
secret_key = args['secret_key']
2022-07-29 12:06:19 -04:00
try:
m = base64.b64decode(secret_key)
m = json.loads(m)
info['name'] = m['A']
info['password'] = m['B']
info['port'] = m['C']
except Exception as e:
return mw.returnJson(False, "接收密钥格式错误!")
else:
2022-11-25 03:46:25 -05:00
data = checkArgs(args, ['sname', 'password'])
2022-07-29 12:06:19 -04:00
if not data[0]:
return data[1]
2022-11-25 03:46:25 -05:00
info['name'] = args['sname']
info['password'] = args['password']
2022-11-25 03:47:44 -05:00
info['port'] = args['port']
2022-07-29 12:06:19 -04:00
rsync = {
'bwlimit': bwlimit,
"port": info['port'],
"compress": compress,
"archive": "true",
"verbose": "true"
2022-07-28 11:08:34 -04:00
}
2022-07-29 12:06:19 -04:00
info['rsync'] = rsync
2022-07-28 11:08:34 -04:00
data = getDefaultConf()
2023-08-02 03:10:23 -04:00
2022-07-28 11:08:34 -04:00
slist = data['send']["list"]
2022-07-29 12:06:19 -04:00
res = lsyncdListFindName(slist, info['name'])
2023-08-02 03:10:23 -04:00
if not 'exclude' in info:
if res[0]:
info["exclude"] = slist[res[1]]['exclude']
else:
info["exclude"] = [
"/**.upload.tmp", "**/*.log", "**/*.tmp",
"**/*.temp", ".git", ".gitignore", ".user.ini",
]
2022-07-28 11:08:34 -04:00
if res[0]:
list_index = res[1]
slist[list_index] = info
else:
slist.append(info)
2022-07-29 12:06:19 -04:00
2022-07-28 11:08:34 -04:00
data['send']["list"] = slist
setDefaultConf(data)
2022-07-29 14:28:47 -04:00
makeLsyncdConf(data)
2022-07-28 11:08:34 -04:00
return mw.returnJson(True, "设置成功!")
2022-07-29 14:28:47 -04:00
def lsyncdRun():
args = getArgs()
data = checkArgs(args, ['name'])
if not data[0]:
return data[1]
send_dir = getServerDir() + "/send"
name = args['name']
app_dir = send_dir + "/" + name
2022-07-30 02:02:58 -04:00
cmd = "bash " + app_dir + "/cmd >> " + app_dir + "/run.log" + " 2>&1 &"
2022-07-29 14:28:47 -04:00
mw.execShell(cmd)
return mw.returnJson(True, "执行成功!")
2022-07-30 02:38:54 -04:00
def lsyncdConfLog():
logs_path = getServerDir() + "/lsyncd.log"
return logs_path
2022-07-29 14:28:47 -04:00
def lsyncdLog():
args = getArgs()
data = checkArgs(args, ['name'])
if not data[0]:
return data[1]
send_dir = getServerDir() + "/send"
name = args['name']
app_dir = send_dir + "/" + name
return app_dir + "/run.log"
2022-07-30 02:02:58 -04:00
def lsyncdGetExclude():
args = getArgs()
data = checkArgs(args, ['name'])
if not data[0]:
return data[1]
data = getDefaultConf()
slist = data['send']["list"]
res = lsyncdListFindName(slist, args['name'])
i = res[1]
info = slist[i]
return mw.returnJson(True, "OK!", info['exclude'])
def lsyncdRemoveExclude():
args = getArgs()
data = checkArgs(args, ['name', 'exclude'])
if not data[0]:
return data[1]
exclude = args['exclude']
data = getDefaultConf()
slist = data['send']["list"]
res = lsyncdListFindName(slist, args['name'])
i = res[1]
info = slist[i]
exclude_list = info['exclude']
exclude_pop_key = -1
for x in range(len(exclude_list)):
if exclude_list[x] == exclude:
exclude_pop_key = x
if exclude_pop_key > -1:
exclude_list.pop(exclude_pop_key)
data['send']["list"][i]['exclude'] = exclude_list
setDefaultConf(data)
makeLsyncdConf(data)
return mw.returnJson(True, "OK!", exclude_list)
def lsyncdAddExclude():
args = getArgs()
data = checkArgs(args, ['name', 'exclude'])
if not data[0]:
return data[1]
exclude = args['exclude']
data = getDefaultConf()
slist = data['send']["list"]
res = lsyncdListFindName(slist, args['name'])
i = res[1]
info = slist[i]
exclude_list = info['exclude']
exclude_list.append(exclude)
data['send']["list"][i]['exclude'] = exclude_list
setDefaultConf(data)
makeLsyncdConf(data)
return mw.returnJson(True, "OK!", exclude_list)
2019-03-05 06:20:04 -05:00
if __name__ == "__main__":
func = sys.argv[1]
if func == 'status':
2021-10-31 12:33:14 -04:00
print(status())
2019-03-05 06:20:04 -05:00
elif func == 'start':
2021-10-31 12:33:14 -04:00
print(start())
2019-03-05 06:20:04 -05:00
elif func == 'stop':
2021-10-31 12:33:14 -04:00
print(stop())
2019-03-05 06:20:04 -05:00
elif func == 'restart':
2021-10-31 12:33:14 -04:00
print(restart())
2019-03-05 06:20:04 -05:00
elif func == 'reload':
2021-10-31 12:33:14 -04:00
print(reload())
2019-03-05 06:20:04 -05:00
elif func == 'initd_status':
2021-10-31 12:33:14 -04:00
print(initdStatus())
2019-03-05 06:20:04 -05:00
elif func == 'initd_install':
2021-10-31 12:33:14 -04:00
print(initdInstall())
2019-03-05 06:20:04 -05:00
elif func == 'initd_uninstall':
2021-10-31 12:33:14 -04:00
print(initdUinstall())
2019-03-05 06:20:04 -05:00
elif func == 'conf':
2021-10-31 12:33:14 -04:00
print(appConf())
2019-03-06 02:07:04 -05:00
elif func == 'run_log':
2021-10-31 12:33:14 -04:00
print(getLog())
2019-03-06 05:47:43 -05:00
elif func == 'rec_list':
2021-10-31 12:33:14 -04:00
print(getRecList())
2019-03-06 05:47:43 -05:00
elif func == 'add_rec':
2021-10-31 12:33:14 -04:00
print(addRec())
2019-03-06 05:47:43 -05:00
elif func == 'del_rec':
2021-10-31 12:33:14 -04:00
print(delRec())
2022-07-28 11:08:34 -04:00
elif func == 'get_rec':
print(getRec())
elif func == 'cmd_rec_secret_key':
print(cmdRecSecretKey())
elif func == 'cmd_rec_cmd':
print(cmdRecCmd())
elif func == 'lsyncd_list':
print(lsyncdList())
elif func == 'lsyncd_add':
print(lsyncdAdd())
2022-07-29 12:06:19 -04:00
elif func == 'lsyncd_get':
print(lsyncdGet())
elif func == 'lsyncd_delete':
print(lsyncdDelete())
2022-07-29 14:28:47 -04:00
elif func == 'lsyncd_run':
print(lsyncdRun())
elif func == 'lsyncd_log':
print(lsyncdLog())
2022-07-30 02:38:54 -04:00
elif func == 'lsyncd_conf_log':
print(lsyncdConfLog())
2022-07-30 02:02:58 -04:00
elif func == 'lsyncd_get_exclude':
print(lsyncdGetExclude())
elif func == 'lsyncd_remove_exclude':
print(lsyncdRemoveExclude())
elif func == 'lsyncd_add_exclude':
print(lsyncdAddExclude())
2019-03-05 06:20:04 -05:00
else:
2021-10-31 12:33:14 -04:00
print('error')