From 7f11e2404477cbdc8cd38fb974ba3864f2c92b5b Mon Sep 17 00:00:00 2001 From: VirtuBox Date: Mon, 23 Sep 2019 12:35:11 +0200 Subject: [PATCH] Another fix for secure --- wo/cli/plugins/secure.py | 20 ++++++++------------ 1 file changed, 8 insertions(+), 12 deletions(-) diff --git a/wo/cli/plugins/secure.py b/wo/cli/plugins/secure.py index fe50e28..e71ef5e 100644 --- a/wo/cli/plugins/secure.py +++ b/wo/cli/plugins/secure.py @@ -148,20 +148,16 @@ class WOSecureController(CementBaseController): @expose(hide=True) def secure_ssh(self): """Harden ssh security""" - pargs = self.app.pargs - if pargs.user_input: - current_ssh_port = pargs.user_input - else: - if os.path.isfile('/etc/ssh/sshd_config'): - for line in open('/etc/ssh/sshd_config', encoding='utf-8'): - if 'Port' in line: - ssh_line = line.strip() + if os.path.isfile('/etc/ssh/sshd_config'): + for line in open('/etc/ssh/sshd_config', encoding='utf-8'): + if 'Port' in line: + ssh_line = line.strip() break - else: - port = (ssh_line).split(' ') - current_ssh_port = port[1] else: - Log.error(self, "SSH config file not found") + port = (ssh_line).split(' ') + current_ssh_port = port[1] + else: + Log.error(self, "SSH config file not found") if not current_ssh_port: current_ssh_port = '22' data = dict(sshport=current_ssh_port, allowpass='no')