restore default enum is first value behavior

2017-09-21 09:20:12 -05:00 · 2017-09-21 09:20:12 -05:00 · 967b459ff1
parent 6615c6efc7
commit 967b459ff1
1 changed files with 6 additions and 2 deletions
--- a/lib/msf/core/opt_base.rb
+++ b/lib/msf/core/opt_base.rb
@ -36,8 +36,12 @@ module Msf
        self.required = required
        self.desc     = attrs.is_a?(String) ? attrs : desc
        self.enums    = [ *(enums) ].map { |x| x.to_s }
-        self.default  = default
-        regex_temp    = regex
+        if default.nil? && enums.length > 0
+          self.default = enums[0]
+        else
+          self.default = default
+        end
+        regex_temp = regex
      else
        if attrs[0].nil?
          self.required = required