Clean up PR

Author: wvu

lib/metasploit/framework/login_scanner/snmp.rb

@@ -17,12 +17,11 @@ class SNMP
         PRIVATE_TYPES        = [ :password ]
         REALM_KEY            = nil
 
-        # The number of retries
+        # The number of retries per community string
         # @return [Fixnum]
         attr_accessor :retries
 
-        # The SNMAP version to scan
-        #
+        # The SNMP version to scan
         # @return [String]
         attr_accessor :version
 
@@ -35,15 +34,20 @@ class SNMP
 
         validates :version,
                   presence: true,
-                  inclusion: { in: ['1', '2c', 'all'] }
+                  inclusion: {
+                    in: ['1', '2c', 'all']
+                  }
 
-        # This method returns an array of versions to scan
+        # This method returns an array of versions to scan for
         # @return [Array] An array of versions
         def versions
           case version
-          when '1'; [:SNMPv1]
-          when '2c'; [:SNMPv2c]
-          when 'all'; [:SNMPv1,:SNMPv2c]
+          when '1'
+            [:SNMPv1]
+          when '2c'
+            [:SNMPv2c]
+          when 'all'
+            [:SNMPv1, :SNMPv2c]
           end
         end
 

modules/auxiliary/scanner/snmp/snmp_login.rb

@@ -33,7 +33,7 @@ def initialize
       OptInt.new('CONNECTION_TIMEOUT', [true, 'The timeout value for each probe', 2]),
       OptInt.new('RETRIES', [true, 'The number of retries per community string', 0]),
       OptInt.new('BATCHSIZE', [true, 'The number of hosts to probe in each set', 256]),
-      OptEnum.new('VERSION', [true, 'The SNMP version to scan', 'all', ['1','2c','all']]),
+      OptEnum.new('VERSION', [true, 'The SNMP version to scan', 'all', ['1', '2c', 'all']]),
       OptString.new('PASSWORD', [ false, 'The password to test' ]),
       OptPath.new('PASS_FILE',  [ false, "File containing communities, one per line",
         File.join(Msf::Config.data_directory, "wordlists", "snmp_default_pass.txt")