From db30f02ffe0c957c58137fb4279a2e59380379bc Mon Sep 17 00:00:00 2001 From: Benjamyn Love Date: Sat, 10 Mar 2018 19:10:34 +1100 Subject: [PATCH] Implemented DNS lookups based on domain and not test data --- main.py | 29 ++++++++++++----------------- 1 file changed, 12 insertions(+), 17 deletions(-) diff --git a/main.py b/main.py index b7b2d28..fd4bee1 100755 --- a/main.py +++ b/main.py @@ -31,13 +31,20 @@ def printWhois(resultData): def parseInput(inData): if inData.lower() == "quit": quit() - #Assume the input is a domain name result = domainWhois(inData) -# for results in result: -# for item in result[results]: -# print('%s: %s' % (results, item)) + print("\nWhos Info") printWhois(result) -#print(result) + print("\nDNS Records") + recLookup(inData) + +def recLookup(inData): + for dnsrec in dnsrecords: + try: + for rdata in dns.resolver.query(inData, dnsrec): + print( indent + "%s:\t%s" % (dnsrec, rdata)) + except Exception as e: + pass + signal.signal(signal.SIGINT, signal_handler) @@ -46,17 +53,5 @@ jsonData = config.readConf() subdomains = jsonData["subdomains"] dnsrecords = jsonData["dnsrecords"] - -print "root" -for rdata in dns.resolver.query('yahoo.com', 'A') : - print rdata -for sub in subdomains: - print sub - try: - for rdata in dns.resolver.query(sub + '.yahoo.com', 'CNAME') : - print( rdata.target ) - except: - pass - while running == 1: parseInput(raw_input("\nPlease Enter a domain" + prompt) )