Added checks to prevent crash when using unset while no module is sel…

…ected or with an option that does not exist
nccgroup · Apr 3, 2017 · 7357b71 · 7357b71
1 parent 1af1c3a
commit 7357b71
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/featherduster.py b/featherduster.py
@@ -347,7 +347,12 @@ def run(self, line):
          print 'Usage: unset <option>'
          return False
       option = line_split[1]
-      feathermodules.current_options[option] = feathermodules.selected_attack['options'][option]
+      try:
+         feathermodules.current_options[option] = feathermodules.selected_attack['options'][option]
+      except KeyError:
+         print '[*] That option doesn\'t exist, sorry!'
+      except AttributeError:
+         print '[*] Please select an attack first!'
    def args(self):
       return feathermodules.selected_attack['options'].keys()