Corrected bad import of DNS for exception handling

Author: codingo

VHostScan.py

@@ -2,8 +2,8 @@
 
 import os
 import sys
+import dns.resolver
 from argparse import ArgumentParser
-from dns.resolver import Resolver
 from socket import gethostbyaddr
 from lib.core.virtual_host_scanner import *
 from lib.helpers.output_helper import *
@@ -88,14 +88,14 @@ def main():
     if not arguments.no_lookup:
         try:
             print("[+] Resolving DNS for additional wordlist entries")
-            for ip in Resolver().query(arguments.target_hosts, 'A'):
+            for ip in dns.resolver.query(arguments.target_hosts, 'A'):
                 host, aliases, ips = gethostbyaddr(str(ip))
                 wordlist.append(str(ip))
                 wordlist.append(host)
                 wordlist.extend(aliases)
-        except (NXDOMAIN):
+        except (dns.resolver.NXDOMAIN):
             print("[!] Couldn't find any records (NXDOMAIN)")
-        except (NoAnswer):
+        except (dns.resolver.NoAnswer):
             print("[!] Couldn't find any records (NoAnswer)")
 
     scanner_args = vars(arguments)