From 4185779a09b5f2d7957cf105837efab68965a7e4 Mon Sep 17 00:00:00 2001 From: Randall Date: Wed, 13 Jul 2022 03:59:53 -0400 Subject: [PATCH] sshyp-mfa - enforce correct syntax --- sshyp-mfa/bin/sshyp-mfa.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/sshyp-mfa/bin/sshyp-mfa.py b/sshyp-mfa/bin/sshyp-mfa.py index ab0b22a..65d7a49 100755 --- a/sshyp-mfa/bin/sshyp-mfa.py +++ b/sshyp-mfa/bin/sshyp-mfa.py @@ -49,6 +49,9 @@ def mfa_read_shortcut(): # reads and extracts MFA info from the user-specified if __name__ == '__main__': # argument fetcher argument_list = argv + if not len(argv) == 1 and not argv[1].strip().startswith('/'): + print(f"\n\u001b[38;5;9merror: invalid argument - run 'man sshyp-mfa' for usage information\u001b[0m\n") + s_exit(1) # user data fetcher ssh_info = get_profile(path.expanduser('~/.config/sshyp/sshyp.sshync'))