Check for required argument

1 files changed, 4 insertions, 0 deletions

diff --git a/contrib/digviz b/contrib/digviz
index 08d2914..8828f07 100755
--- a/contrib/digviz
+++ b/contrib/digviz
@@ -257,6 +257,8 @@ class DigCommandLineQuery:
                     (len(arg) <= 12 or arg[12] == '='):
                 try:
                     opt, arg = arg.split('=')
+                    if not arg:
+                        raise ValueError()
                 except ValueError:
                     raise CommandLineException('+trusted-key requires a filename argument.')
                 else:
@@ -276,6 +278,8 @@ class DigCommandLineQuery:
                     (len(arg) <= 3 or arg[3] == '='):
                 try:
                     opt, arg = arg.split('=')
+                    if not arg:
+                        raise ValueError()
                 except ValueError:
                     raise CommandLineException('+lg requires a filename argument.')
                 else: