benjamyn/DNS-py
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

hopes and dreams

Browse Source
This commit is contained in:
root 2018-02-15 09:07:24 +11:00
parent 7d20140563
commit 266ba314b3
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
HTTPy.py
View File

@ -148,9 +148,13 @@ def iterate(inArray, pre):
def iterateMX(inArray, pre): def iterateMX(inArray, pre):
global inInput
count = 0 count = 0
while count != len(inArray) - 1: while count != len(inArray) - 1:
tmp = inArray[count].split(' ') tmp = inArray[count].split(' ')
if (len(tmp) == 1):
pass
else:
tmpVal = isCNAME(tmp[1]) tmpVal = isCNAME(tmp[1])
print indent + bcolors.CYAN + pre, bcolors.GREEN + inArray[count] + bcolors.RESET + " -> " + bcolors.GREEN + tmpVal[0] + bcolors.RESET print indent + bcolors.CYAN + pre, bcolors.GREEN + inArray[count] + bcolors.RESET + " -> " + bcolors.GREEN + tmpVal[0] + bcolors.RESET
count = count + 1 count = count + 1
@ -194,7 +198,7 @@ def dig(inDomain, record):
dig = subprocess.check_output(["dig", record, "+short", inDomain]) dig = subprocess.check_output(["dig", record, "+short", inDomain])
except Exception as e: except Exception as e:
dig = "error" dig = "error"
print str(e) #print str(e)
print "========================================================" print "========================================================"
test = dig.split("\n") test = dig.split("\n")
iterate(test, record) iterate(test, record)
@ -205,7 +209,7 @@ def newdig(inDomain, record, sub=""):
dig = subprocess.check_output(["dig", record, "+short", inDomain]) dig = subprocess.check_output(["dig", record, "+short", inDomain])
except Exception as e: except Exception as e:
dig = "error" dig = "error"
print str(e) #print str(e)
#print "========================================================" #print "========================================================"
test = dig.split("\n") test = dig.split("\n")
#iterate(test, record) #iterate(test, record)
@ -238,7 +242,10 @@ def getInput():
if inInput == "quit": if inInput == "quit":
running = 0 running = 0
def isCNAME(CNAME): def isCNAME(CNAME, DOMAIN=""):
if (DOMAIN == CNAME):
return 0
else:
retval = newdig(CNAME, "A") retval = newdig(CNAME, "A")
return retval[0] return retval[0]