yan/mdserver-web
1
0
Fork 0
mirror of https://github.com/midoks/mdserver-web.git synced 2026-10-18 23:45:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Mr Chen
2024-11-24 04:17:05 +08:00
parent 9f6cd75893
commit 92f9a2b9a2
3 changed files with 71 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/utils/firewall.py
+29 -1
View File
@@ -393,7 +393,35 @@ class Firewall(object):
mw.writeFile(file, conf)
mw.execShell("systemctl restart sshd.service")
mw.writeLog("SSH管理", msg)
return mw.returnJson(True, msg)
return mw.returnData(True, msg)
def setSshPubkeyStatus(self, status):
msg = '禁止密钥登陆成功'
if status == "1":
msg = '开启密钥登陆成功'
file = '/etc/ssh/sshd_config'
if not os.path.exists(file):
return mw.returnJson(False, '无法设置!')
content = mw.readFile(file)
pubkey_rep = r"PubkeyAuthentication\s+(\w*)\s*\n"
pubkey_status = re.search(pubkey_rep, content)
if not pubkey_status:
rep = r"(#)?PubkeyAuthentication\s+(\w*)\s*\n"
content = re.sub(rep, "PubkeyAuthentication yes\n", content)
if status == '1':
rep = r"PubkeyAuthentication\s+(\w*)\s*\n"
content = re.sub(rep, "PubkeyAuthentication yes\n", content)
else:
rep = r"PubkeyAuthentication\s+(\w*)\s*\n"
content = re.sub(rep, "PubkeyAuthentication no\n", content)
mw.writeFile(file, content)
mw.execShell("systemctl restart sshd.service")
mw.writeLog("SSH管理", msg)
return mw.returnData(True, msg)