fix enumeration handling

snotsbubble · Dec 26, 2019 · 3dac95e · 3dac95e
1 parent 8576a78
commit 3dac95e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/modules/auxiliary/dos/http/metasploit_httphandler_dos.rb b/modules/auxiliary/dos/http/metasploit_httphandler_dos.rb
@@ -50,7 +50,7 @@ def initialize(info = {})
 
     register_options(
         [
-            OptString.new('DOSTYPE', [true, 'GENTLE|SOFT|HARD', 'HARD'])
+            OptEnum.new('DOSTYPE', [true, 'Type of DoS to trigger', 'HARD', %w[GENTLE SOFT HARD]])
         ])
   end
 
@@ -126,7 +126,7 @@ def dos
       end
 
     else
-      bla = ""
+      fail_with Failure::BadConfig, 'Invalid DOSTYPE selected'
     end
 
     print_status("DOS request sent")
@@ -146,7 +146,6 @@ def is_alive?
   def run
     print_status("#{rhost}:#{rport} - Sending DoS packet...")
     dos
-
   end
 
 end