Fix NoMethodError on os, mark DCs as 'server'

Author: egypt

modules/post/windows/gather/enum_ad_computers.rb

@@ -66,11 +66,11 @@ def run
 
     # Results table holds raw string data
     results_table = Rex::Ui::Text::Table.new(
-        'Header'     => "Domain Computers",
-        'Indent'     => 1,
-        'SortIndex'  => -1,
-        'Columns'    => fields
-      )
+      'Header'     => "Domain Computers",
+      'Indent'     => 1,
+      'SortIndex'  => -1,
+      'Columns'    => fields
+    )
 
     # Hostnames holds DNS Names to Resolve
     hostnames = []
@@ -81,40 +81,37 @@ def run
 
       report = {}
       0.upto(fields.length-1) do |i|
-        if result[i].nil?
-          field = ""
-        else
-          field = result[i]
-
-          # Only perform these actions if the database is connected and we want
-          # to store in the DB.
-          if db and datastore['STORE_DB']
-            case fields[i]
-            when 'dNSHostName'
-              dns = field
-              report[:name] = dns
-              hostnames << dns
-            when 'operatingSystem'
-              report[:os_name] = os
-            when 'distinguishedName'
-              if field =~ /Domain Controllers/i
-                # TODO: Find another way to mark a host as being a domain controller
-                #       The 'purpose' field should be server, client, device, printer, etc
-                # report[:purpose] = "DC"
-              end
-            when 'operatingSystemServicePack'
-              # XXX: Does this take into account the leading 'SP' string?
-              
-              if field.to_i > 0
-                report[:os_sp] = 'SP' + field
-              end
-              if field =~ /(Service Pack|SP)\s?(\d+)/
-                report[:os_sp] = 'SP' + $2
-              end
-
-            when 'description'
-              report[:info] = field
+        field = result[i] || ""
+
+        # Only perform these actions if the database is connected and we want
+        # to store in the DB.
+        if db && datastore['STORE_DB']
+          case fields[i]
+          when 'dNSHostName'
+            dns = field
+            report[:name] = dns
+            hostnames << dns
+          when 'operatingSystem'
+            report[:os_name] = field
+          when 'distinguishedName'
+            if field =~ /Domain Controllers/i
+              # TODO: Find another way to mark a host as being a domain controller
+              #       The 'purpose' field should be server, client, device, printer, etc
+              #report[:purpose] = "DC"
+              report[:purpose] = "server"
             end
+          when 'operatingSystemServicePack'
+            # XXX: Does this take into account the leading 'SP' string?
+
+            if field.to_i > 0
+              report[:os_sp] = 'SP' + field
+            end
+            if field =~ /(Service Pack|SP)\s?(\d+)/
+              report[:os_sp] = 'SP' + $2
+            end
+
+          when 'description'
+            report[:info] = field
           end
         end
 
@@ -125,7 +122,7 @@ def run
       results_table << row
     end
 
-    if db and datastore['STORE_DB']
+    if db && datastore['STORE_DB']
       print_status("Resolving IP addresses...")
       ip_results = client.net.resolve.resolve_hosts(hostnames, AF_INET)