/* * Copyright (C) 2000 Internet Software Consortium. * * Permission to use, copy, modify, and distribute this software for any * purpose with or without fee is hereby granted, provided that the above * copyright notice and this permission notice appear in all copies. * * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS * SOFTWARE. */ /* $Id: dig.c,v 1.78 2000/07/25 00:51:14 mws Exp $ */ #include #include #include #include #include #include #include #include #include #include #include #include #include extern ISC_LIST(dig_lookup_t) lookup_list; extern dig_serverlist_t server_list; extern ISC_LIST(dig_searchlist_t) search_list; #define ADD_STRING(b, s) { \ if (strlen(s) >= isc_buffer_availablelength(b)) \ return (ISC_R_NOSPACE); \ else \ isc_buffer_putstr(b, s); \ } extern isc_boolean_t have_ipv6, show_details, specified_source, usesearch, qr; extern in_port_t port; extern unsigned int timeout; extern isc_mem_t *mctx; extern dns_messageid_t id; extern char *rootspace[BUFSIZE]; extern isc_buffer_t rootbuf; extern int sendcount; extern int ndots; extern int tries; extern int lookup_counter; extern char fixeddomain[MXNAME]; extern int exitcode; extern isc_sockaddr_t bind_address; extern char keynametext[MXNAME]; extern char keyfile[MXNAME]; extern char keysecret[MXNAME]; extern dns_tsigkey_t *key; extern isc_boolean_t validated; extern isc_taskmgr_t *taskmgr; extern isc_task_t *global_task; extern isc_boolean_t free_now; dig_lookup_t *default_lookup = NULL; extern isc_uint32_t name_limit; extern isc_uint32_t rr_limit; extern isc_boolean_t debugging; extern isc_boolean_t isc_mem_debugging; char *batchname = NULL; FILE *batchfp = NULL; char *argv0; isc_boolean_t short_form = ISC_FALSE, printcmd = ISC_TRUE; isc_uint16_t bufsize = 0; isc_boolean_t forcecomment = ISC_FALSE; static const char *opcodetext[] = { "QUERY", "IQUERY", "STATUS", "RESERVED3", "NOTIFY", "UPDATE", "RESERVED6", "RESERVED7", "RESERVED8", "RESERVED9", "RESERVED10", "RESERVED11", "RESERVED12", "RESERVED13", "RESERVED14", "RESERVED15" }; static const char *rcodetext[] = { "NOERROR", "FORMERR", "SERVFAIL", "NXDOMAIN", "NOTIMPL", "REFUSED", "YXDOMAIN", "YXRRSET", "NXRRSET", "NOTAUTH", "NOTZONE", "RESERVED11", "RESERVED12", "RESERVED13", "RESERVED14", "RESERVED15", "BADVERS" }; extern char *progname; static void show_usage(void) { fputs( "Usage: dig [@global-server] [domain] [q-type] [q-class] {q-opt}\n" " {global-d-opt} host [@local-server] {local-d-opt}\n" " [ host [@local-server] {local-d-opt} [...]]\n" "Where: domain are in the Domain Name System\n" " q-class is one of (in,chaos,...) [default: in]\n" " q-type is one of (a,any,mx,ns,soa,hinfo,axfr,txt,...) [default:a]\n" " (Use ixfr=version for type ixfr)\n" " q-opt is one of:\n" " -x dot-notation (shortcut for in-addr lookups)\n" " -f filename (batch mode)\n" " -p port (specify port number)\n" " -t type (specify query type)\n" " -c class (specify query class)\n" " -y name:key (specify named base64 tsig key)\n" " d-opt is of the form +keyword[=value], where keyword is:\n" " +[no]vc (TCP mode)\n" " +[no]tcp (TCP mode, alternate syntax)\n" " +time=### (Set query timeout) [5]\n" " +tries=### (Set number of UDP attempts) [3]\n" " +domain=### (Set default domainname)\n" " +bufsize=### (Set EDNS0 Max UDP packet size)\n" " +[no]search (Set whether to use searchlist)\n" " +[no]defname (Set whether to use default domain)\n" " +[no]recursive (Recursive mode)\n" " +[no]aaonly (Set AA flag in query)\n" " +[no]adflag (Set AD flag in query)\n" " +[no]cdflag (Set CD flag in query)\n" " +[no]details (Show details of all requests)\n" " +ndots=### (Set NDOTS value)\n" " +[no]comments (Control display of comment lines)\n" " +[no]question (Control display of question)\n" " +[no]answer (Control display of answer)\n" " +[no]authority (Control display of authority)\n" " +[no]additional (Control display of additional)\n" " +[no]short (Disable everything except short\n" " form of answer)\n" " +[no]all (Set or clear all display flags)\n" " +qr (Print question before sending)\n" " +[no]cdflag (Set or clear CD flag in query)\n" " +[no]adflag (Set or clear AD flag in query)\n" " +[no]nssearch (Search all authorative nameservers)\n" " +[no]identify (ID responders in short answers)\n" " +[no]trace (Trace delegation down from root)\n" " +rrlimit=### (Limit number of rr's in xfr)\n" " +namelimit=### (Limit number of names in xfr)\n" " global d-opts and servers (before host name) affect all queries.\n" " local d-opts and servers (after host name) affect only that lookup.\n" , stderr); } /* * Callback from dighost.c to print the received message. */ void received(int bytes, int frmsize, char *frm, dig_query_t *query) { isc_uint64_t diff; isc_time_t now; isc_result_t result; time_t tnow; result = isc_time_now(&now); check_result(result, "isc_time_now"); if (query->lookup->stats) { diff = isc_time_microdiff(&now, &query->time_sent); printf(";; Query time: %ld msec\n", (long int)diff/1000); printf(";; SERVER: %.*s(%s)\n", frmsize, frm, query->servname); time(&tnow); printf(";; WHEN: %s", ctime(&tnow)); if (query->lookup->doing_xfr) { printf(";; XFR size: %d names, %d rrs\n", query->name_count, query->rr_count); } else { printf(";; MSG SIZE rcvd: %d\n", bytes); } if (key != NULL) { if (!validated) puts(";; WARNING -- Some TSIG could not " "be validated"); } if ((key == NULL) && (keysecret[0] != 0)) { puts(";; WARNING -- TSIG key was not used."); } puts(""); } else if (query->lookup->identify && !short_form) { diff = isc_time_microdiff(&now, &query->time_sent); printf(";; Received %u bytes from %.*s(%s) in %d ms\n\n", bytes, frmsize, frm, query->servname, (int)diff/1000); } } /* * Callback from dighost.c to print that it is trying a server. * Not used in dig. */ void trying(int frmsize, char *frm, dig_lookup_t *lookup) { UNUSED(frmsize); UNUSED(frm); UNUSED(lookup); } /* * Internal print routine used to print short form replies. */ static isc_result_t say_message(dns_rdata_t *rdata, dig_query_t *query, isc_buffer_t *buf) { isc_result_t result; isc_uint64_t diff; isc_time_t now; char store[sizeof("12345678901234567890")]; if (query->lookup->trace || query->lookup->ns_search_only) { result = dns_rdatatype_totext(rdata->type, buf); if (result != ISC_R_SUCCESS) return (result); ADD_STRING(buf, " "); } result = dns_rdata_totext(rdata, NULL, buf); check_result(result, "dns_rdata_totext"); if (query->lookup->identify) { result = isc_time_now(&now); if (result != ISC_R_SUCCESS) return (result); diff = isc_time_microdiff(&now, &query->time_sent); ADD_STRING(buf, " from server "); ADD_STRING(buf, query->servname); snprintf(store, 19, " in %d ms.", (int)diff/1000); ADD_STRING(buf, store); } ADD_STRING(buf, "\n"); return (ISC_R_SUCCESS); } /* * short_form message print handler. Calls above say_message() */ static isc_result_t short_answer(dns_message_t *msg, dns_messagetextflag_t flags, isc_buffer_t *buf, dig_query_t *query) { dns_name_t *name; dns_rdataset_t *rdataset; isc_buffer_t target; isc_result_t result, loopresult; dns_name_t empty_name; char t[4096]; dns_rdata_t rdata; UNUSED(flags); dns_name_init(&empty_name, NULL); result = dns_message_firstname(msg, DNS_SECTION_ANSWER); if (result == ISC_R_NOMORE) return (ISC_R_SUCCESS); else if (result != ISC_R_SUCCESS) return (result); for (;;) { name = NULL; dns_message_currentname(msg, DNS_SECTION_ANSWER, &name); isc_buffer_init(&target, t, sizeof(t)); for (rdataset = ISC_LIST_HEAD(name->list); rdataset != NULL; rdataset = ISC_LIST_NEXT(rdataset, link)) { loopresult = dns_rdataset_first(rdataset); while (loopresult == ISC_R_SUCCESS) { dns_rdataset_current(rdataset, &rdata); result = say_message(&rdata, query, buf); check_result(result, "say_message"); loopresult = dns_rdataset_next(rdataset); } } result = dns_message_nextname(msg, DNS_SECTION_ANSWER); if (result == ISC_R_NOMORE) break; else if (result != ISC_R_SUCCESS) return (result); } return (ISC_R_SUCCESS); } /* * Callback from dighost.c to print the reply from a server */ isc_result_t printmessage(dig_query_t *query, dns_message_t *msg, isc_boolean_t headers) { isc_boolean_t did_flag = ISC_FALSE; isc_result_t result; dns_messagetextflag_t flags; isc_buffer_t *buf = NULL; unsigned int len = OUTPUTBUF; UNUSED(query); debug("printmessage(%s)", headers ? "headers" : "noheaders"); /* * Exitcode 9 means we timed out, but if we're printing a message, * we must have recovered. Go ahead and reset it to code 0, and * call this a success. */ if (exitcode == 9) exitcode = 0; flags = 0; if (!headers) { flags |= DNS_MESSAGETEXTFLAG_NOHEADERS; flags |= DNS_MESSAGETEXTFLAG_NOCOMMENTS; } if (!query->lookup->comments) flags |= DNS_MESSAGETEXTFLAG_NOCOMMENTS; result = ISC_R_SUCCESS; result = isc_buffer_allocate(mctx, &buf, len); check_result(result, "isc_buffer_allocate"); if (query->lookup->comments && !short_form) { if (!query->lookup->doing_xfr) { if (msg == query->lookup->sendmsg) printf(";; Sending:\n"); else printf(";; Got answer:\n"); } if (headers) { printf(";; ->>HEADER<<- opcode: %s, status: %s, " "id: %u\n", opcodetext[msg->opcode], rcodetext[msg->rcode], msg->id); printf(";; flags: "); if ((msg->flags & DNS_MESSAGEFLAG_QR) != 0) { printf("qr"); did_flag = ISC_TRUE; } if ((msg->flags & DNS_MESSAGEFLAG_AA) != 0) { printf("%saa", did_flag ? " " : ""); did_flag = ISC_TRUE; } if ((msg->flags & DNS_MESSAGEFLAG_TC) != 0) { printf("%stc", did_flag ? " " : ""); did_flag = ISC_TRUE; } if ((msg->flags & DNS_MESSAGEFLAG_RD) != 0) { printf("%srd", did_flag ? " " : ""); did_flag = ISC_TRUE; } if ((msg->flags & DNS_MESSAGEFLAG_RA) != 0) { printf("%sra", did_flag ? " " : ""); did_flag = ISC_TRUE; } if ((msg->flags & DNS_MESSAGEFLAG_AD) != 0) { printf("%sad", did_flag ? " " : ""); did_flag = ISC_TRUE; } if ((msg->flags & DNS_MESSAGEFLAG_CD) != 0) { printf("%scd", did_flag ? " " : ""); did_flag = ISC_TRUE; } printf("; QUERY: %u, ANSWER: %u, " "AUTHORITY: %u, ADDITIONAL: %u\n", msg->counts[DNS_SECTION_QUESTION], msg->counts[DNS_SECTION_ANSWER], msg->counts[DNS_SECTION_AUTHORITY], msg->counts[DNS_SECTION_ADDITIONAL]); result = dns_message_pseudosectiontotext(msg, DNS_PSEUDOSECTION_OPT, flags, buf); check_result(result, "dns_message_pseudosectiontotext"); } } if (query->lookup->section_question && headers) { if (!short_form) { question_again: result = dns_message_sectiontotext(msg, DNS_SECTION_QUESTION, flags, buf); if (result == ISC_R_NOSPACE) { len += OUTPUTBUF; isc_buffer_free(&buf); result = isc_buffer_allocate(mctx, &buf, len); if (result == ISC_R_SUCCESS) goto question_again; } check_result(result, "dns_message_sectiontotext"); } } if (query->lookup->section_answer) { if (!short_form) { answer_again: result = dns_message_sectiontotext(msg, DNS_SECTION_ANSWER, flags, buf); if (result == ISC_R_NOSPACE) { len += OUTPUTBUF; isc_buffer_free(&buf); result = isc_buffer_allocate(mctx, &buf, len); if (result == ISC_R_SUCCESS) goto answer_again; } check_result(result, "dns_message_sectiontotext"); } else { result = short_answer(msg, flags, buf, query); check_result(result, "short_answer"); } } if (query->lookup->section_authority) { if (!short_form) { authority_again: result = dns_message_sectiontotext(msg, DNS_SECTION_AUTHORITY, flags, buf); if (result == ISC_R_NOSPACE) { len += OUTPUTBUF; isc_buffer_free(&buf); result = isc_buffer_allocate(mctx, &buf, len); if (result == ISC_R_SUCCESS) goto authority_again; } check_result(result, "dns_message_sectiontotext"); } } if (query->lookup->section_additional) { if (!short_form) { additional_again: result = dns_message_sectiontotext(msg, DNS_SECTION_ADDITIONAL, flags, buf); if (result == ISC_R_NOSPACE) { len += OUTPUTBUF; isc_buffer_free(&buf); result = isc_buffer_allocate(mctx, &buf, len); if (result == ISC_R_SUCCESS) goto additional_again; } check_result(result, "dns_message_sectiontotext"); /* * Only print the signature on the first record. */ if (headers) { result = dns_message_pseudosectiontotext( msg, DNS_PSEUDOSECTION_TSIG, flags, buf); check_result(result, "dns_message_pseudosectiontotext"); result = dns_message_pseudosectiontotext( msg, DNS_PSEUDOSECTION_SIG0, flags, buf); check_result(result, "dns_message_pseudosectiontotext"); } } } if (headers && query->lookup->comments && !short_form) printf("\n"); printf("%.*s", (int)isc_buffer_usedlength(buf), (char *)isc_buffer_base(buf)); isc_buffer_free(&buf); return (result); } /* * print the greeting message when the program first starts up. */ static void printgreeting(int argc, char **argv) { int i = 1; if (printcmd) { puts(""); printf("; <<>> DiG 9.0 <<>>"); while (i < argc) { printf(" %s", argv[i++]); } puts(""); printf(";; global options: %s %s\n", short_form ? "short_form" : "", printcmd ? "printcmd" : ""); } } /* * Reorder an argument list so that server names all come at the end. * This is a bit of a hack, to allow batch-mode processing to properly * handle the server options. */ static void reorder_args(int argc, char *argv[]) { int i, j; char *ptr; int end; debug("reorder_args()"); end = argc - 1; while (argv[end][0] == '@') { end--; if (end == 0) return; } debug("arg[end]=%s", argv[end]); for (i = 1; i < end - 1; i++) { if (argv[i][0] == '@') { debug("arg[%d]=%s", i, argv[i]); ptr = argv[i]; for (j = i + 1; j < end; j++) { debug("Moving %s to %d", argv[j], j - 1); argv[j - 1] = argv[j]; } debug("moving %s to end, %d", ptr, end - 1); argv[end - 1] = ptr; end--; if (end < 1) return; } } } /* * We're not using isc_commandline_parse() here since the command line * syntax of dig is quite a bit different from that which can be described * that routine. */ static void parse_args(isc_boolean_t is_batchfile, isc_boolean_t config_only, int argc, char **argv) { isc_boolean_t have_host = ISC_FALSE; isc_result_t result; isc_textregion_t tr; dig_server_t *srv = NULL; dig_lookup_t *lookup = NULL; dns_rdatatype_t rdtype; dns_rdataclass_t rdclass; char batchline[MXNAME]; char address[MXNAME]; int bargc; char *bargv[16]; int i, n; int adrs[4]; int rc; char **rv; char *ptr; #ifndef NOPOSIX char *homedir; char rcfile[132]; #endif /* * The semantics for parsing the args is a bit complex; if * we don't have a host yet, make the arg apply globally, * otherwise make it apply to the latest host. This is * a bit different than the previous versions, but should * form a consistent user interface. * * First, create a "default lookup" which won't actually be used * anywhere, except for cloning into new lookups */ debug("parse_args()"); if (!is_batchfile) { debug("making new lookup"); default_lookup = make_empty_lookup(); #ifndef NOPOSIX /* * Treat .digrc as a special batchfile */ homedir = getenv("HOME"); if (homedir != NULL) snprintf(rcfile, 132, "%s/.digrc", homedir); else strcpy(rcfile, ".digrc"); batchfp = fopen(rcfile, "r"); if (batchfp != NULL) { while (fgets(batchline, sizeof(batchline), batchfp) != 0) { debug("config line %s", batchline); bargc = 1; bargv[bargc] = strtok(batchline, " \t\r\n"); while ((bargv[bargc] != NULL) && (bargc < 14)) { bargc++; bargv[bargc] = strtok(NULL, " \t\r\n"); } bargv[0] = argv[0]; argv0 = argv[0]; reorder_args(bargc, (char **)bargv); parse_args(ISC_TRUE, ISC_TRUE, bargc, (char **)bargv); } fclose(batchfp); } #endif } lookup = default_lookup; rc = argc; rv = argv; for (rc--, rv++; rc > 0; rc--, rv++) { debug("main parsing %s", rv[0]); if (strncmp(rv[0], "%", 1) == 0) break; if (strncmp(rv[0], "@", 1) == 0) { srv = make_server(&rv[0][1]); ISC_LIST_APPEND(lookup->my_server_list, srv, link); } else if ((strcmp(rv[0], "+vc") == 0) && (!is_batchfile)) { lookup->tcp_mode = ISC_TRUE; } else if ((strcmp(rv[0], "+novc") == 0) && (!is_batchfile)) { lookup->tcp_mode = ISC_FALSE; } else if ((strcmp(rv[0], "+tcp") == 0) && (!is_batchfile)) { lookup->tcp_mode = ISC_TRUE; } else if ((strcmp(rv[0], "+notcp") == 0) && (!is_batchfile)) { lookup->tcp_mode = ISC_FALSE; } else if (strncmp(rv[0], "+domain=", 8) == 0) { /* Global option always */ strncpy(fixeddomain, &rv[0][8], MXNAME); } else if (strncmp(rv[0], "+sea", 4) == 0) { /* Global option always */ usesearch = ISC_TRUE; } else if (strncmp(rv[0], "+nosea", 6) == 0) { usesearch = ISC_FALSE; } else if (strncmp(rv[0], "+defn", 5) == 0) { lookup->defname = ISC_TRUE; } else if (strncmp(rv[0], "+nodefn", 7) == 0) { lookup->defname = ISC_FALSE; } else if (strncmp(rv[0], "+time=", 6) == 0) { /* Global option always */ timeout = atoi(&rv[0][6]); if (timeout <= 0) timeout = 1; debug("timeout set to %d", timeout); } else if (strncmp(rv[0], "+timeout=", 9) == 0) { /* Global option always */ timeout = atoi(&rv[0][9]); if (timeout <= 0) timeout = 1; debug("timeout set to %d", timeout); } else if (strncmp(rv[0], "+tries=", 7) == 0) { lookup->retries = atoi(&rv[0][7]); if (lookup->retries <= 0) lookup->retries = 1; } else if (strncmp(rv[0], "+namelimit=", 11) == 0) { name_limit = atoi(&rv[0][11]); } else if (strncmp(rv[0], "+rrlimit=", 9) == 0) { rr_limit = atoi(&rv[0][9]); } else if (strncmp(rv[0], "+buf=", 5) == 0) { lookup->udpsize = atoi(&rv[0][5]); if (lookup->udpsize <= 0) lookup->udpsize = 0; if (lookup->udpsize > COMMSIZE) lookup->udpsize = COMMSIZE; } else if (strncmp(rv[0], "+buf=", 5) == 0) { lookup->udpsize = atoi(&rv[0][5]); if (lookup->udpsize <= 0) lookup->udpsize = 0; if (lookup->udpsize > COMMSIZE) lookup->udpsize = COMMSIZE; } else if (strncmp(rv[0], "+bufsize=", 9) == 0) { lookup->udpsize = atoi(&rv[0][9]); if (lookup->udpsize <= 0) lookup->udpsize = 0; if (lookup->udpsize > COMMSIZE) lookup->udpsize = COMMSIZE; } else if (strncmp(rv[0], "+ndots=", 7) == 0) { /* Global option always */ ndots = atoi(&rv[0][7]); if (ndots < 0) ndots = 0; } else if (strncmp(rv[0], "+rec", 4) == 0) { lookup->recurse = ISC_TRUE; } else if (strncmp(rv[0], "+norec", 6) == 0) { lookup->recurse = ISC_FALSE; } else if (strncmp(rv[0], "+aa", 3) == 0) { lookup->aaonly = ISC_TRUE; } else if (strncmp(rv[0], "+noaa", 5) == 0) { lookup->aaonly = ISC_FALSE; } else if (strncmp(rv[0], "+adf", 4) == 0) { lookup->adflag = ISC_TRUE; } else if (strncmp(rv[0], "+noadf", 6) == 0) { lookup->adflag = ISC_FALSE; } else if (strncmp(rv[0], "+cd", 3) == 0) { lookup->cdflag = ISC_TRUE; } else if (strncmp(rv[0], "+nocd", 5) == 0) { lookup->cdflag = ISC_FALSE; } else if (strncmp(rv[0], "+ns", 3) == 0) { lookup->ns_search_only = ISC_TRUE; lookup->trace_root = ISC_TRUE; lookup->recurse = ISC_FALSE; lookup->identify = ISC_TRUE; lookup->stats = ISC_FALSE; if (!forcecomment) lookup->comments = ISC_FALSE; lookup->section_additional = ISC_FALSE; lookup->section_authority = ISC_FALSE; lookup->section_question = ISC_FALSE; lookup->rdtype = dns_rdatatype_soa; short_form = ISC_TRUE; } else if (strncmp(rv[0], "+nons", 6) == 0) { lookup->ns_search_only = ISC_FALSE; } else if (strncmp(rv[0], "+tr", 3) == 0) { lookup->trace = ISC_TRUE; lookup->trace_root = ISC_TRUE; lookup->recurse = ISC_FALSE; lookup->identify = ISC_TRUE; if (!forcecomment) { lookup->comments = ISC_FALSE; lookup->stats = ISC_FALSE; } lookup->section_additional = ISC_FALSE; lookup->section_authority = ISC_TRUE; lookup->section_question = ISC_FALSE; show_details = ISC_TRUE; } else if (strncmp(rv[0], "+notr", 6) == 0) { lookup->trace = ISC_FALSE; lookup->trace_root = ISC_FALSE; } else if (strncmp(rv[0], "+det", 4) == 0) { show_details = ISC_TRUE; } else if (strncmp(rv[0], "+nodet", 6) == 0) { show_details = ISC_FALSE; } else if (strncmp(rv[0], "+cmd", 4) == 0) { printcmd = ISC_TRUE; } else if (strncmp(rv[0], "+nocmd", 6) == 0) { printcmd = ISC_FALSE; } else if (strncmp(rv[0], "+sho", 4) == 0) { short_form = ISC_TRUE; printcmd = ISC_FALSE; lookup->section_additional = ISC_FALSE; lookup->section_authority = ISC_FALSE; lookup->section_question = ISC_FALSE; if (!forcecomment) { lookup->comments = ISC_FALSE; lookup->stats = ISC_FALSE; } } else if (strncmp(rv[0], "+nosho", 6) == 0) { short_form = ISC_FALSE; } else if (strncmp(rv[0], "+id", 3) == 0) { lookup->identify = ISC_TRUE; } else if (strncmp(rv[0], "+noid", 5) == 0) { lookup->identify = ISC_FALSE; } else if (strncmp(rv[0], "+com", 4) == 0) { lookup->comments = ISC_TRUE; forcecomment = ISC_TRUE; } else if (strncmp(rv[0], "+nocom", 6) == 0) { lookup->comments = ISC_FALSE; lookup->stats = ISC_FALSE; forcecomment = ISC_FALSE; } else if (strncmp(rv[0], "+sta", 4) == 0) { lookup->stats = ISC_TRUE; } else if (strncmp(rv[0], "+nosta", 6) == 0) { lookup->stats = ISC_FALSE; } else if (strncmp(rv[0], "+qr", 3) == 0) { qr = ISC_TRUE; } else if (strncmp(rv[0], "+noqr", 5) == 0) { qr = ISC_FALSE; } else if (strncmp(rv[0], "+que", 4) == 0) { lookup->section_question = ISC_TRUE; } else if (strncmp(rv[0], "+noque", 6) == 0) { lookup->section_question = ISC_FALSE; } else if (strncmp(rv[0], "+ans", 4) == 0) { lookup->section_answer = ISC_TRUE; } else if (strncmp(rv[0], "+noans", 6) == 0) { lookup->section_answer = ISC_FALSE; } else if (strncmp(rv[0], "+add", 4) == 0) { lookup->section_additional = ISC_TRUE; } else if (strncmp(rv[0], "+noadd", 6) == 0) { lookup->section_additional = ISC_FALSE; } else if (strncmp(rv[0], "+aut", 4) == 0) { lookup->section_authority = ISC_TRUE; } else if (strncmp(rv[0], "+noaut", 6) == 0) { lookup->section_authority = ISC_FALSE; } else if (strncmp(rv[0], "+all", 4) == 0) { lookup->section_question = ISC_TRUE; lookup->section_authority = ISC_TRUE; lookup->section_answer = ISC_TRUE; lookup->section_additional = ISC_TRUE; lookup->comments = ISC_TRUE; } else if (strncmp(rv[0], "+noall", 6) == 0) { lookup->section_question = ISC_FALSE; lookup->section_authority = ISC_FALSE; lookup->section_answer = ISC_FALSE; lookup->section_additional = ISC_FALSE; lookup->comments = ISC_FALSE; } else if (strncmp(rv[0], "-c", 2) == 0) { if (rv[0][2] != 0) { ptr = &rv[0][2]; } else { if (rc <= 1) { show_usage(); exit (exitcode); } ptr = rv[1]; rv++; rc--; } tr.base = ptr; tr.length = strlen(ptr); result = dns_rdataclass_fromtext(&rdclass, (isc_textregion_t *)&tr); if (result == ISC_R_SUCCESS) lookup->rdclass = rdclass; else fprintf (stderr, ";; Warning, ignoring " "invalid class %s\n", ptr); } else if (strncmp(rv[0], "-t", 2) == 0) { if (rv[0][2] != 0) { ptr = &rv[0][2]; } else { if (rc <= 1) { show_usage(); exit (exitcode); } ptr = rv[1]; rv++; rc--; } tr.base = ptr; tr.length = strlen(ptr); if (strncmp(rv[0], "ixfr=", 5) == 0) { lookup->rdtype = dns_rdatatype_ixfr; lookup->ixfr_serial = atoi(&rv[0][5]); } else { result = dns_rdatatype_fromtext(&rdtype, (isc_textregion_t *)&tr); if ((result == ISC_R_SUCCESS) && (rdtype != dns_rdatatype_ixfr)) lookup->rdtype = rdtype; else fprintf (stderr, ";; Warning, " "ignoring invalid type %s\n", ptr); } } else if (strncmp(rv[0], "-f", 2) == 0) { if (rv[0][2] != 0) { batchname = &rv[0][2]; } else { if (rc <= 1) { show_usage(); exit (exitcode); } batchname = rv[1]; rv++; rc--; } } else if (strncmp(rv[0], "-y", 2) == 0) { if (rv[0][2] != 0) ptr = &rv[0][2]; else { if (rc <= 1) { show_usage(); exit (exitcode); } ptr = rv[1]; rv++; rc--; } ptr = strtok(ptr,":"); if (ptr == NULL) { show_usage(); exit(exitcode); } strncpy(keynametext, ptr, MXNAME); ptr = strtok(NULL, ""); if (ptr == NULL) { show_usage(); exit(exitcode); } strncpy(keysecret, ptr, MXNAME); } else if (strncmp(rv[0], "-k", 2) == 0) { if (rv[0][2] != 0) ptr = &rv[0][2]; else { if (rc <= 1) { show_usage(); exit (exitcode); } ptr = rv[1]; rv++; rc--; } strncpy(keyfile, ptr, MXNAME); } else if (strncmp(rv[0], "-p", 2) == 0) { if (rv[0][2] != 0) { port = atoi(&rv[0][2]); } else { if (rc <= 1) { show_usage(); exit (exitcode); } port = atoi(rv[1]); rv++; rc--; } } else if (strncmp(rv[0], "-b", 2) == 0) { if (rv[0][2] != 0) { strncpy(address, &rv[0][2], MXRD); } else { if (rc <= 1) { show_usage(); exit (exitcode); } strncpy(address, rv[1], MXRD); rv++; rc--; } get_address(address, 0, &bind_address); specified_source = ISC_TRUE; } else if (strncmp(rv[0], "-h", 2) == 0) { show_usage(); exit(exitcode); } else if (strcmp(rv[0], "-memdebug") == 0) { isc_mem_debugging = 1; } else if (strcmp(rv[0], "-debug") == 0) { debugging = ISC_TRUE; } else if ((strncmp(rv[0], "-x", 2) == 0) && !config_only) { /* * XXXMWS Only works for ipv4 now. * Can't use inet_pton here, since we allow * partial addresses. */ if (rc == 1) { show_usage(); exit(exitcode); } n = sscanf(rv[1], "%d.%d.%d.%d", &adrs[0], &adrs[1], &adrs[2], &adrs[3]); if (n == 0) show_usage(); lookup = clone_lookup(default_lookup, ISC_TRUE); for (i = n - 1; i >= 0; i--) { snprintf(batchline, MXNAME/8, "%d.", adrs[i]); strncat(lookup->textname, batchline, MXNAME); } strncat(lookup->textname, "in-addr.arpa.", MXNAME); debug("looking up %s", lookup->textname); lookup->trace_root = ISC_TF(lookup->trace || lookup->ns_search_only); lookup->rdtype = dns_rdatatype_ptr; lookup->rdclass = dns_rdataclass_in; lookup->new_search = ISC_TRUE; ISC_LIST_APPEND(lookup_list, lookup, link); have_host = ISC_TRUE; rv++; rc--; } else { tr.base = rv[0]; tr.length = strlen(rv[0]); if (strncmp(rv[0], "ixfr=", 5) == 0) { lookup->rdtype = dns_rdatatype_ixfr; lookup->ixfr_serial = atoi(&rv[0][5]); continue; } result = dns_rdatatype_fromtext(&rdtype, (isc_textregion_t *)&tr); if ((result == ISC_R_SUCCESS) && (rdtype != dns_rdatatype_ixfr)) { lookup->rdtype = rdtype; continue; } result = dns_rdataclass_fromtext(&rdclass, (isc_textregion_t *)&tr); if (result == ISC_R_SUCCESS) { lookup->rdclass = rdclass; continue; } if (!config_only) { lookup=clone_lookup(default_lookup, ISC_TRUE); strncpy(lookup->textname, rv[0], MXNAME-1); lookup->trace_root = ISC_TF(lookup->trace || lookup->ns_search_only); lookup->new_search = ISC_TRUE; ISC_LIST_APPEND(lookup_list, lookup, link); have_host = ISC_TRUE; debug("looking up %s", lookup->textname); } } } /* * If we have a batchfile, seed the lookup list with the * first entry, then trust the callback in dighost_shutdown * to get the rest */ if ((batchname != NULL) && !(is_batchfile)) { batchfp = fopen(batchname, "r"); if (batchfp == NULL) { perror(batchname); if (exitcode < 10) exitcode = 10; fatal("Couldn't open specified batch file"); } if (fgets(batchline, sizeof(batchline), batchfp) != 0) { debug("batch line %s", batchline); bargc = 1; bargv[bargc] = strtok(batchline, " \t\r\n"); while ((bargv[bargc] != NULL) && (bargc < 14)) { bargc++; bargv[bargc] = strtok(NULL, " \t\r\n"); } bargv[0] = argv[0]; argv0 = argv[0]; reorder_args(bargc, (char **)bargv); parse_args(ISC_TRUE, ISC_FALSE, bargc, (char **)bargv); } } if ((lookup_list.head == NULL) && !config_only) { lookup=clone_lookup(default_lookup, ISC_TRUE); lookup->trace_root = ISC_TF(lookup->trace || lookup->ns_search_only); lookup->new_search = ISC_TRUE; strcpy(lookup->textname, "."); lookup->rdtype = dns_rdatatype_ns; ISC_LIST_APPEND(lookup_list, lookup, link); } if (!config_only) printgreeting(argc, argv); } /* * Callback from dighost.c to allow program-specific shutdown code. Here, * Here, we're possibly reading from a batch file, then shutting down for * real if there's nothing in the batch file to read. */ void dighost_shutdown(void) { char batchline[MXNAME]; int bargc; char *bargv[16]; if (batchname == NULL) { isc_app_shutdown(); return; } if (feof(batchfp)) { batchname = NULL; isc_app_shutdown(); fclose(batchfp); return; } if (fgets(batchline, sizeof(batchline), batchfp) != 0) { debug("batch line %s", batchline); bargc = 1; bargv[bargc] = strtok(batchline, " \t\r\n"); while ((bargv[bargc] != NULL) && (bargc < 14)) { bargc++; bargv[bargc] = strtok(NULL, " \t\r\n"); } bargv[0] = argv0; reorder_args(bargc, (char **)bargv); parse_args(ISC_TRUE, ISC_FALSE, bargc, (char **)bargv); start_lookup(); } else { batchname = NULL; fclose(batchfp); isc_app_shutdown(); return; } } int main(int argc, char **argv) { isc_result_t result; dig_server_t *s, *s2; ISC_LIST_INIT(lookup_list); ISC_LIST_INIT(server_list); ISC_LIST_INIT(search_list); debug("main()"); progname = argv[0]; result = isc_app_start(); check_result(result, "isc_app_start"); setup_libs(); parse_args(ISC_FALSE, ISC_FALSE, argc, argv); setup_system(); result = isc_app_onrun(mctx, global_task, onrun_callback, NULL); check_result(result, "isc_app_onrun"); isc_app_run(); s = ISC_LIST_HEAD(default_lookup->my_server_list); while (s != NULL) { debug("freeing server %p belonging to %p", s, default_lookup); s2 = s; s = ISC_LIST_NEXT(s, link); ISC_LIST_DEQUEUE(default_lookup->my_server_list, (dig_server_t *)s2, link); isc_mem_free(mctx, s2); } isc_mem_free(mctx, default_lookup); if (batchname != NULL) { fclose(batchfp); batchname = NULL; } cancel_all(); destroy_libs(); isc_app_finish(); return (exitcode); }