Updated miscellaneous python example programs.

Author: Shumon Huque

examples/get-ip-many.py

@@ -1,16 +1,66 @@
 #!/usr/bin/env python
+#
 
-import getdns, sys
+"""
+get-ip.py: resolve given DNS names into IP addresses. The -s switch
+constains answers to only ones secured by DNSSEC. The -4 switch only
+returns IPv4 addresses, the -6 switch only IPv6 addresses.
 
-hostname = sys.argv[1]
+An example run:
 
-ctx = getdns.Context()
-extensions = { "return_both_v4_and_v6" : getdns.GETDNS_EXTENSION_TRUE }
-results = ctx.address(name=hostname, extensions=extensions)
+    $ get-ip.py -s www.huque.com www.google.com
+    www.huque.com: IPv4  50.116.63.23
+    www.huque.com: IPv6  2600:3c03:e000:81::a
+    www.google.com: No DNSSEC secured responses found
+
+"""
+
+import getdns, sys, getopt
+
+def usage():
+    print """\
+Usage: get-ip.py [-s] [-4|-6] <domain1> <domain2> ...
+
+    -s: only return DNSSEC secured answers
+    -4: only return IPv4 address answers
+    -6: only return IPv6 address answers
+
+-4 and -6 are mutually exclusive. If both are specified, IPv6 wins.
+"""
+    sys.exit(1)
 
-if results["status"] == getdns.GETDNS_RESPSTATUS_GOOD:
-    for addr in results["just_address_answers"]:
-        print addr["IPSTRING"]
+try:
+    (options, args) = getopt.getopt(sys.argv[1:], 's46')
+except getopt.GetoptError:
+    usage()
 else:
-    print "getdns.address() returned an error: %d" % results["status"]
+    if not args:
+        usage()
+
+extensions = { "return_both_v4_and_v6" : getdns.GETDNS_EXTENSION_TRUE }
+desired_addr_type = None
+
+for (opt, optval) in options:
+    if opt == "-s":
+        extensions["dnssec_return_only_secure"] = getdns.GETDNS_EXTENSION_TRUE
+    elif opt == "-4":
+        desired_addr_type = "IPv4"
+    elif opt == "-6":
+        desired_addr_type = "IPv6"
+
+ctx = getdns.Context()
+
+for hostname in args:
+    results = ctx.address(name=hostname, extensions=extensions)
+    status = results['status']
+    if status == getdns.GETDNS_RESPSTATUS_GOOD:
+        for addr in results['just_address_answers']:
+            addr_type = addr['address_type']
+            addr_data = addr['address_data']
+            if (desired_addr_type == None) or (addr_type == desired_addr_type):
+                print "%s: %s  %s" % (hostname, addr_type, addr_data)
+    elif status == getdns.GETDNS_RESPSTATUS_NO_SECURE_ANSWERS:
+        print "%s: No DNSSEC secured responses found" % hostname
+    else:
+        print "%s: getdns.address() returned error: %d" % (hostname, status)
 

examples/get-ip-only-secure.py

@@ -15,8 +15,8 @@ def get_ip(ctx, qname):
     iplist = []
     results = ctx.address(name=qname, extensions=extensions)
     if results['status'] == getdns.GETDNS_RESPSTATUS_GOOD:
-        for addr in results["just_address_answers"]:
-            iplist.append(addr['IPSTRING'])
+        for addr in results['just_address_answers']:
+            iplist.append(addr['address_data'])
     else:
         print "getdns.address() returned an error: %d" % results['status']
     return iplist

examples/get-ip.py

@@ -0,0 +1,62 @@
+#!/usr/bin/env python
+#
+
+"""
+Lookup an NS record and printout all the hostnames and associated IP
+addresses of the listed nameservers.
+"""
+
+import getdns, pprint, sys
+
+extensions = { "return_both_v4_and_v6" : getdns.GETDNS_EXTENSION_TRUE }
+
+
+def usage():
+    print """Usage: get-ns-ip.py <zone>
+
+where <zone> is a DNS zone (domain).
+"""
+    sys.exit(1)
+
+
+def get_ip(ctx, qname):
+    iplist = []
+    results = ctx.address(name=qname, extensions=extensions)
+    if results['status'] == getdns.GETDNS_RESPSTATUS_GOOD:
+        for addr in results['just_address_answers']:
+            iplist.append(addr['address_data'])
+    else:
+        print "getdns.address() returned an error: %d" % results['status']
+    return iplist
+
+
+if __name__ == '__main__':
+
+    if len(sys.argv) != 2:
+        usage()
+
+    qname = sys.argv[1]
+
+    ctx = getdns.Context()
+    results = ctx.general(name=qname, request_type=getdns.GETDNS_RRTYPE_NS)
+    status = results['status']
+
+    hostlist = []
+    if status == getdns.GETDNS_RESPSTATUS_GOOD:
+        for reply in results['replies_tree']:
+            answers = reply['answer']
+            for answer in answers:
+                if answer['type'] == getdns.GETDNS_RRTYPE_NS:
+                    iplist = get_ip(ctx, answer['rdata']['nsdname'])
+                    for ip in iplist:
+                        hostlist.append( (answer['rdata']['nsdname'], ip) )
+    elif status == getdns.GETDNS_RESPSTATUS_NO_NAME:
+        print "%s: no such DNS zone" % qname
+    elif status == getdns.GETDNS_RESPSTATUS_ALL_TIMEOUT:
+        print "%s, NS: query timed out" % qname
+    else:
+        print "%s, %s: unknown return code: %d" % results["status"]
+
+    # Print out each NS server name and IP address
+    for (nsdname, addr) in sorted(hostlist):
+        print nsdname, addr

examples/get-mx-ip.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python
+# 
+# Use stub mode with google public DNS as recursive resolvers
+# to query addresses associated with a host.
+#
+
+import getdns, sys, pprint
+
+google_public_dns = [
+    {'address_data': '8.8.8.8', 'address_type': 'IPv4'},
+    {'address_data': '8.8.4.4', 'address_type': 'IPv4'},
+    {'address_data': '2001:4860:4860::8888', 'address_type': 'IPv6'},
+    {'address_data': '2001:4860:4860::8844', 'address_type': 'IPv6'},
+]
+
+hostname = sys.argv[1]
+
+ctx = getdns.Context()
+ctx.resolution_type = getdns.GETDNS_RESOLUTION_STUB
+ctx.upstream_recursive_servers = google_public_dns
+
+extensions = { "return_both_v4_and_v6" : getdns.GETDNS_EXTENSION_TRUE }
+
+results = ctx.address(name=hostname, extensions=extensions)
+
+if results["status"] == getdns.GETDNS_RESPSTATUS_GOOD:
+    for addr in results["just_address_answers"]:
+        print addr["address_data"]
+else:
+    print "getdns.address() returned an error: %d" % results["status"]
+

examples/get-ns-ip.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python
+
+import getdns, sys
+
+hostname = sys.argv[1]
+
+ctx = getdns.Context()
+ctx.resolution_type = getdns.GETDNS_RESOLUTION_STUB
+
+extensions = { "return_both_v4_and_v6" : getdns.GETDNS_EXTENSION_TRUE }
+ctx.resolver_type = getdns.GETDNS_RESOLUTION_STUB
+
+results = ctx.address(name=hostname, extensions=extensions)
+
+if results["status"] == getdns.GETDNS_RESPSTATUS_GOOD:
+    for addr in results["just_address_answers"]:
+        print addr["address_data"]
+else:
+    print "getdns.address() returned an error: %d" % results["status"]
+

examples/query-gdns.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+#
+
+"""
+simply.py
+
+A simple example to query a domain name and print out addresses
+associated with it.
+"""
+
+import sys, getdns
+
+hostname = sys.argv[1]
+
+ctx = getdns.Context()
+extensions = { "return_both_v4_and_v6" : getdns.GETDNS_EXTENSION_TRUE }
+
+results = ctx.address(name=hostname, extensions=extensions)
+status = results['status']
+
+if status == getdns.GETDNS_RESPSTATUS_GOOD:
+    for addr in results['just_address_answers']:
+        print addr['address_data']
+else:
+    print "%s: getdns.address() returned error: %d" % (hostname, status)