[gentoo-commits] proj/security:ajak-cvetool commit in: bin/

[John Helmert III]

commit:     b599466bea8a657b2716a96d9c0a355728af4f7c
Author:     John Helmert III <ajak <AT> gentoo <DOT> org>
AuthorDate: Mon Jul  5 01:40:16 2021 +0000
Commit:     John Helmert III <ajak <AT> gentoo <DOT> org>
CommitDate: Mon Jul  5 01:43:32 2021 +0000
URL:        https://gitweb.gentoo.org/proj/security.git/commit/?id=b599466b

cvetool: sanity check CVETool args rather than sys.argv

Signed-off-by: John Helmert III <ajak <AT> gentoo.org>

 bin/cvetool.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/bin/cvetool.py b/bin/cvetool.py
index 233375a..744e2a5 100644
--- a/bin/cvetool.py
+++ b/bin/cvetool.py
@@ -73,14 +73,14 @@ class CVETool:
 
             self.nfu(self.cleanup_cve(args[0]))
         elif command == 'pw':
-            if len(sys.argv) != 4:
+            if len(args) != 2:
                 print('Usage: pw <user> <password>')
                 print('Generates a base64-encoded credential for storing')
                 sys.exit(1)
 
             self.pw(sys.argv[2], sys.argv[3])
         elif command == 'dobug':
-            if len(sys.argv) != 3:
+            if len(args) != 1:
                 print('Usage: dobug <bug>')
                 print('Adds and assigns a bug\'s CVEs')
                 sys.exit(1)