r00t2/optools
1
0
Fork 0
Code Issues Pull Requests Projects Releases Activity

uhhh... preventing multiple simultaneous runs is important.

Browse Source
This commit is contained in:
brent s 2019-01-17 04:02:40 -05:00
parent 122c366490
commit a79802afa1
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
aif/scripts/post/sshsecure.py
View File

@ -337,12 +337,23 @@ def daemonMgr():
return() return()


def main(): def main():
self_pidfile = '/tmp/sshsecure.pid'
# First, check to see if we're already running.
# This is where I'd put a psutil call... IF I HAD ONE.
if os.path.isfile(self_pidfile):
is_running = subprocess.run(['pgrep', '-F', self_pidfile], stdout = subprocess.PIPE)
if is_running.stdout.decode('utf-8').strip() != '':
# We're still running. Exit gracefully.
print('We seem to still be running from a past execution; exiting')
exit(0)
else:
with open(self_pidfile, 'w') as f:
f.write(str(os.getpid()))
_chkfile = '/etc/ssh/.aif-generated' _chkfile = '/etc/ssh/.aif-generated'
if not os.path.isfile(_chkfile): if not os.path.isfile(_chkfile):
# Warning: The moduli stuff can take a LONG time to run. Hours. # Warning: The moduli stuff can take a LONG time to run. Hours.
buildmoduli = True buildmoduli = True
hostKeys(buildmoduli) hostKeys(buildmoduli)
restart = True
for t in ('sshd', 'ssh'): for t in ('sshd', 'ssh'):
config(conf_options[t], t) config(conf_options[t], t)
clientKeys() clientKeys()
@ -350,6 +361,7 @@ def main():
f.write(('ssh, sshd, and hostkey configurations/keys have been modified by sshsecure.py from OpTools.\n' f.write(('ssh, sshd, and hostkey configurations/keys have been modified by sshsecure.py from OpTools.\n'
'https://git.square-r00t.net/OpTools/\n')) 'https://git.square-r00t.net/OpTools/\n'))
daemonMgr() daemonMgr()
os.remove(self_pidfile)
return() return()


if __name__ == '__main__': if __name__ == '__main__':