From 266ba314b3829fe8e501df8de860ad39860905a7 Mon Sep 17 00:00:00 2001
From: root <root@vps.benjamyn-testing.com>
Date: Thu, 15 Feb 2018 09:07:24 +1100
Subject: [PATCH] hopes and dreams

---
 HTTPy.py | 23 +++++++++++++++--------
 1 file changed, 15 insertions(+), 8 deletions(-)

diff --git a/HTTPy.py b/HTTPy.py
index 7c9be2b..6e9f102 100755
--- a/HTTPy.py
+++ b/HTTPy.py
@@ -148,11 +148,15 @@ def iterate(inArray, pre):
 
 
 def iterateMX(inArray, pre):
+    global inInput
     count = 0
     while count != len(inArray) - 1:
         tmp = inArray[count].split(' ')
-        tmpVal = isCNAME(tmp[1])
-        print indent +  bcolors.CYAN + pre, bcolors.GREEN + inArray[count] + bcolors.RESET + " -> " + bcolors.GREEN + tmpVal[0] + bcolors.RESET
+        if (len(tmp) == 1):
+	    pass
+	else:
+            tmpVal = isCNAME(tmp[1])
+            print indent +  bcolors.CYAN + pre, bcolors.GREEN + inArray[count] + bcolors.RESET + " -> " + bcolors.GREEN + tmpVal[0] + bcolors.RESET
         count = count + 1
         
 def parseWhois(whioisInfo):
@@ -194,7 +198,7 @@ def dig(inDomain, record):
         dig = subprocess.check_output(["dig", record, "+short", inDomain])
     except Exception as e:
         dig = "error"
-        print str(e)
+        #print str(e)
     print "========================================================" 
     test = dig.split("\n")
     iterate(test, record)
@@ -205,7 +209,7 @@ def newdig(inDomain, record, sub=""):
         dig = subprocess.check_output(["dig", record, "+short", inDomain])
     except Exception as e:
         dig = "error"
-        print str(e)
+        #print str(e)
     #print "========================================================" 
     test = dig.split("\n")
     #iterate(test, record)
@@ -238,9 +242,12 @@ def getInput():
     if inInput == "quit":
         running = 0
 
-def isCNAME(CNAME):
-    retval = newdig(CNAME, "A")
-    return retval[0]
+def isCNAME(CNAME, DOMAIN=""):
+    if (DOMAIN == CNAME):
+    	return 0
+    else:
+	retval = newdig(CNAME, "A")
+    	return retval[0]
 
 
 #Main Loop
@@ -281,4 +288,4 @@ while running == 1:
                         else:
                             dnsRec, dnsType, blank = newdig(inInput, rec)
                             iterate(dnsRec, dnsType)
-                print '\n'
\ No newline at end of file
+                print '\n'