freebsd-dev/usr.sbin/named/tools/nstest/nstest.c
1994-09-22 20:45:34 +00:00

424 lines
12 KiB
C

/*
* ++Copyright++ 1986
* -
* Copyright (c) 1986
* The Regents of the University of California. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the University of
* California, Berkeley and its contributors.
* 4. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
* -
* Portions Copyright (c) 1993 by Digital Equipment Corporation.
*
* 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, and that
* the name of Digital Equipment Corporation not be used in advertising or
* publicity pertaining to distribution of the document or software without
* specific, written prior permission.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
* WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL DIGITAL EQUIPMENT
* CORPORATION 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.
* -
* --Copyright--
*/
#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1986 Regents of the University of California.\n\
portions Copyright (c) 1993 Digital Equipment Corporation\n\
All rights reserved.\n";
#endif /* not lint */
#ifndef lint
static char sccsid[] = "@(#)nstest.c 4.15 (Berkeley) 3/21/91";
static char rcsid[] = "$Id: nstest.c,v 4.9.1.6 1994/06/01 21:10:11 vixie Exp $";
#endif /* not lint */
#include <sys/param.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <arpa/nameser.h>
#include <stdio.h>
#include <resolv.h>
char *progname;
FILE *log;
#define MAXDATA 256 /* really should get this from named/db.h */
main(argc, argv)
char **argv;
{
register char *cp;
register u_char *ucp;
struct hostent *hp;
u_short port = htons(NAMESERVER_PORT);
char buf[BUFSIZ];
u_char packet[PACKETSZ], answer[8*1024], OldRRData[MAXDATA];
struct rrec NewRR;
u_int32_t l;
int n, dump_packet;
NewRR.r_data = (char *) malloc(MAXDATA);
NewRR.r_data = (char *) malloc(MAXDATA);
progname = argv[0];
dump_packet = 0;
_res.options |= RES_DEBUG|RES_RECURSE;
(void) res_init();
while (argc > 1 && argv[1][0] == '-') {
argc--;
cp = *++argv;
while (*++cp)
switch (*cp) {
case 'p':
if (--argc <= 0)
usage();
port = htons(atoi(*++argv));
break;
case 'i':
_res.options |= RES_IGNTC;
break;
case 'v':
_res.options |= RES_USEVC|RES_STAYOPEN;
break;
case 'r':
_res.options &= ~RES_RECURSE;
break;
case 'd':
dump_packet++;
break;
default:
usage();
}
}
_res.nsaddr.sin_family = AF_INET;
_res.nsaddr.sin_addr = inet_makeaddr(IN_LOOPBACKNET, 1);
_res.nsaddr.sin_port = port;
if (argc > 1) {
if (!inet_aton(argv[1],
(struct in_addr *)&_res.nsaddr.sin_addr))
usage();
}
if (argc > 2) {
log = fopen(argv[2],"w");
if (log == NULL) perror(argv[2]);
}
for (;;) {
printf("> ");
fflush(stdout);
if ((cp = fgets(buf, sizeof buf, stdin)) == NULL)
break;
switch (*cp++) {
case 'a':
n = res_mkquery(QUERY, cp, C_IN, T_A, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'A':
n = ntohl(inet_addr(cp));
putlong((u_int32_t)n, (u_char*)cp);
n = res_mkquery(IQUERY, "", C_IN, T_A, (u_char *)cp,
INT32SZ, NULL,
packet, sizeof(packet));
break;
case 'f':
n = res_mkquery(QUERY, cp, C_ANY, T_UINFO, NULL,
0, NULL, packet, sizeof packet);
break;
case 'F':
n = res_mkquery(QUERY, cp, C_IN, T_AFSDB, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'g':
n = res_mkquery(QUERY, cp, C_ANY, T_GID, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'G':
*(int *)cp = htonl(atoi(cp));
n = res_mkquery(IQUERY, "", C_ANY, T_GID, (u_char *)cp,
sizeof(int), NULL, packet, sizeof packet);
break;
case 'c':
n = res_mkquery(QUERY, cp, C_IN, T_CNAME, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'h':
n = res_mkquery(QUERY, cp, C_IN, T_HINFO, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'm':
n = res_mkquery(QUERY, cp, C_IN, T_MX, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'M':
n = res_mkquery(QUERY, cp, C_IN, T_MAILB, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'n':
n = res_mkquery(QUERY, cp, C_IN, T_NS, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'p':
n = res_mkquery(QUERY, cp, C_IN, T_PTR, NULL, 0,
NULL, packet, sizeof packet);
break;
case 's':
n = res_mkquery(QUERY, cp, C_IN, T_SOA, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'T':
n = res_mkquery(QUERY, cp, C_IN, T_TXT, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'u':
n = res_mkquery(QUERY, cp, C_ANY, T_UID, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'U':
*(int *)cp = htonl(atoi(cp));
n = res_mkquery(IQUERY, "", C_ANY, T_UID, (u_char *)cp,
sizeof(int), NULL,
packet, sizeof packet);
break;
case 'x':
n = res_mkquery(QUERY, cp, C_IN, T_AXFR, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'w':
n = res_mkquery(QUERY, cp, C_IN, T_WKS, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'b':
n = res_mkquery(QUERY, cp, C_IN, T_MB, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'B':
n = res_mkquery(QUERY, cp, C_IN, T_MG, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'i':
n = res_mkquery(QUERY, cp, C_IN, T_MINFO, NULL, 0,
NULL, packet, sizeof packet);
break;
case 'r':
n = res_mkquery(QUERY, cp, C_IN, T_MR, NULL, 0,
NULL, packet, sizeof packet);
break;
case '*':
n = res_mkquery(QUERY, cp, C_IN, T_ANY, NULL, 0,
NULL, packet, sizeof packet);
break;
#ifdef ALLOW_UPDATES
case '^':
{
char IType[10], TempStr[50];
int Type, oldnbytes, nbytes, i;
#ifdef ALLOW_T_UNSPEC
printf("Data type (a = T_A, u = T_UNSPEC): ");
gets(IType);
if (IType[0] == 'u') {
Type = T_UNSPEC;
printf("How many data bytes? ");
gets(TempStr); /* Throw away CR */
sscanf(TempStr, "%d", &nbytes);
for (i = 0; i < nbytes; i++) {
(NewRR.r_data)[i] = (char) i;
}
} else {
#endif /* ALLOW_T_UNSPEC */
Type = T_A;
nbytes = INT32SZ;
printf(
"Inet addr for new dname (e.g., 192.4.3.2): "
);
gets(TempStr);
putlong(ntohl(inet_addr(TempStr)),
NewRR.r_data);
#ifdef ALLOW_T_UNSPEC
}
#endif
NewRR.r_class = C_IN;
NewRR.r_type = Type;
NewRR.r_size = nbytes;
NewRR.r_ttl = 99999999;
printf("Add, modify, or modify all (a/m/M)? ");
gets(TempStr);
if (TempStr[0] == 'a') {
n = res_mkquery(UPDATEA, cp, C_IN, Type,
OldRRData, nbytes,
&NewRR, packet,
sizeof packet);
} else {
if (TempStr[0] == 'm') {
printf("How many data bytes in old RR? ");
gets(TempStr); /* Throw away CR */
sscanf(TempStr, "%d", &oldnbytes);
for (i = 0; i < oldnbytes; i++) {
OldRRData[i] = (char) i;
}
n = res_mkquery(UPDATEM, cp,
C_IN, Type,
OldRRData, oldnbytes,
&NewRR, packet,
sizeof packet);
} else { /* Modify all */
n = res_mkquery(UPDATEMA, cp,
C_IN, Type, NULL, 0,
&NewRR, packet,
sizeof packet);
}
}
}
break;
#ifdef ALLOW_T_UNSPEC
case 'D':
n = res_mkquery(UPDATEDA, cp, C_IN, T_UNSPEC,
(char *)0, 0, NULL,
packet, sizeof packet);
break;
case 'd':
{
char TempStr[100];
int nbytes, i;
printf("How many data bytes in oldrr data? ");
gets(TempStr); /* Throw away CR */
sscanf(TempStr, "%d", &nbytes);
for (i = 0; i < nbytes; i++) {
OldRRData[i] = (char) i;
}
n = res_mkquery(UPDATED, cp, C_IN, T_UNSPEC,
OldRRData, nbytes, NULL,
packet, sizeof packet);
}
break;
#endif /* ALLOW_T_UNSPEC */
#endif /* ALLOW_UPDATES */
default:
printf("a{host} - query T_A\n");
printf("A{addr} - iquery T_A\n");
printf("b{user} - query T_MB\n");
printf("B{user} - query T_MG\n");
printf("f{host} - query T_UINFO\n");
printf("g{host} - query T_GID\n");
printf("G{gid} - iquery T_GID\n");
printf("h{host} - query T_HINFO\n");
printf("i{host} - query T_MINFO\n");
printf("p{host} - query T_PTR\n");
printf("m{host} - query T_MX\n");
printf("M{host} - query T_MAILB\n");
printf("n{host} - query T_NS\n");
printf("r{host} - query T_MR\n");
printf("s{host} - query T_SOA\n");
printf("T{host} - query T_TXT\n");
printf("u{host} - query T_UID\n");
printf("U{uid} - iquery T_UID\n");
printf("x{host} - query T_AXFR\n");
printf("w{host} - query T_WKS\n");
printf("F{host} - query T_AFSDB\n");
printf("c{host} - query T_CNAME\n");
printf("*{host} - query T_ANY\n");
#ifdef ALLOW_UPDATES
printf("^{host} - add/mod/moda (T_A/T_UNSPEC)\n");
#ifdef ALLOW_T_UNSPEC
printf("D{host} - deletea T_UNSPEC\n");
printf("d{host} - delete T_UNSPEC\n");
#endif /* ALLOW_T_UNSPEC */
#endif /* ALLOW_UPDATES */
continue;
}
if (n < 0) {
printf("res_mkquery: buffer too small\n");
continue;
}
if (log) {
fprintf(log,"SEND QUERY\n");
fp_query(packet, log);
}
n = res_send(packet, n, answer, sizeof(answer));
if (n < 0) {
printf("res_send: send error\n");
if (log) fprintf(log, "res_send: send error\n");
}
else {
if (dump_packet) {
int f;
f = creat("ns_packet.dump", 0644);
write(f, answer, n);
(void) close(f);
}
if (log) {
fprintf(log, "GOT ANSWER\n");
fp_query(answer, log);
}
}
}
}
usage()
{
fprintf(stderr, "Usage: %s [-v] [-i] [-r] [-d] [-p port] hostaddr\n",
progname);
exit(1);
}