Connect to same address family as source address

diff --git a/drill/drill.c b/drill/drill.c
index 2fa645500dfb719430b57bc622e120b6daa2062c..3ded1a6a8473a58e76a5bc0948a185a23ba0baf0 100644 (file)
--- a/drill/drill.c
+++ b/drill/drill.c
@@ -497,9 +497,15 @@ main(int argc, char *argv[])
        if (src) {
                src_rdf = ldns_rdf_new_addr_frm_str(src);
                if(!src_rdf) {
-                       fprintf(stderr, "-I must be (or resolve) to a valid IP[v6] address.\n");
+                       fprintf(stderr, "-I must be a valid IP[v6] address.\n");
                        exit(EXIT_FAILURE);
                }
+               if (ldns_rdf_size(src_rdf) == 4) {
+                       qfamily = LDNS_RESOLV_INET;
+
+               } else if (ldns_rdf_size(src_rdf) == 16) {
+                       qfamily = LDNS_RESOLV_INET6;
+               }
        }
 
        /* set the nameserver to use */