From 274538991b39cd27b1b3ec9db659317507039828 Mon Sep 17 00:00:00 2001 From: Ben Date: Wed, 26 Jul 2017 23:39:06 +1000 Subject: [PATCH] This should fix the logic for ip checking --- HTTPy.py | 39 ++++++++++++++++++++------------------- 1 file changed, 20 insertions(+), 19 deletions(-) diff --git a/HTTPy.py b/HTTPy.py index 4c1a251..53d9354 100755 --- a/HTTPy.py +++ b/HTTPy.py @@ -49,6 +49,8 @@ def compareSOA(inNameservers, inDomain, origSOA): def checkIfIP(in1): count = 0 + if len(in1) != 4: + return 0 for dom in in1: if dom.isdigit(): count += 1 @@ -147,25 +149,24 @@ while running == 1: if running == 0: pass else: - if len(partDomain) == 4: - if checkIfIP(partDomain): + if checkIfIP(partDomain): + print "========================================================" + ipLookup(inInput) + if pingOn == "P": print "========================================================" - ipLookup(inInput) - if pingOn == "P": - print "========================================================" - ipPing(inInput) - else: - if whoOn == "W": - whois(inInput) - + ipPing(inInput) else: + if whoOn == "W": + whois(inInput) + + else: - #Condense dig calls - dig(inInput, "A") - dig(inInput, "MX") - dig(inInput, "TXT") - test = dig(inInput, "NS") - dig(inInput, "SRV") - soaRec = dig(inInput, "SOA") - print compareSOA(test, inInput, soaRec) - print '\n' \ No newline at end of file + #Condense dig calls + dig(inInput, "A") + dig(inInput, "MX") + dig(inInput, "TXT") + test = dig(inInput, "NS") + dig(inInput, "SRV") + soaRec = dig(inInput, "SOA") + print compareSOA(test, inInput, soaRec) + print '\n' \ No newline at end of file