/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2015 Gary Mills
 * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
/*	  All Rights Reserved  	*/

/*
 * University Copyright- Copyright (c) 1982, 1986, 1988
 * The Regents of the University of California
 * All Rights Reserved
 *
 * University Acknowledgment- Portions of this document are derived from
 * software developed by the University of California, Berkeley, and its
 * contributors.
 */

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <string.h>
#include <arpa/nameser.h>
#include <resolv.h>
#include "crossl.h"

void fp_query(char *msg, FILE *file);

char *_res_opcodes[] = {
	"QUERY",
	"IQUERY",
	"CQUERYM",
	"CQUERYU",
	"4",
	"5",
	"6",
	"7",
	"8",
	"UPDATEA",
	"UPDATED",
	"UPDATEDA",
	"UPDATEM",
	"UPDATEMA",
	"ZONEINIT",
	"ZONEREF",
};

char *_res_resultcodes[] = {
	"NOERROR",
	"FORMERR",
	"SERVFAIL",
	"NXDOMAIN",
	"NOTIMP",
	"REFUSED",
	"6",
	"7",
	"8",
	"9",
	"10",
	"11",
	"12",
	"13",
	"14",
	"NOCHANGE",
};

void
p_query(msg)
	char *msg;
{
	fp_query(msg, stdout);
}

/*
 * Print the contents of a query.
 * This is intended to be primarily a debugging routine.
 */
void
fp_query(msg, file)
	char *msg;
	FILE *file;
{
	register char *cp;
	register HEADER *hp;
	register int n;

	/*
	 * Print header fields.
	 */
	hp = (HEADER *)msg;
	cp = msg + sizeof (HEADER);
	fprintf(file, "HEADER:\n");
	fprintf(file, "\topcode = %s", _res_opcodes[hp->opcode]);
	fprintf(file, ", id = %d", ntohs(hp->id));
	fprintf(file, ", rcode = %s\n", _res_resultcodes[hp->rcode]);
	fprintf(file, "\theader flags: ");
	if (hp->qr)
		fprintf(file, " qr");
	if (hp->aa)
		fprintf(file, " aa");
	if (hp->tc)
		fprintf(file, " tc");
	if (hp->rd)
		fprintf(file, " rd");
	if (hp->ra)
		fprintf(file, " ra");
	if (hp->pr)
		fprintf(file, " pr");
	fprintf(file, "\n\tqdcount = %d", ntohs(hp->qdcount));
	fprintf(file, ", ancount = %d", ntohs(hp->ancount));
	fprintf(file, ", nscount = %d", ntohs(hp->nscount));
	fprintf(file, ", arcount = %d\n\n", ntohs(hp->arcount));
	/*
	 * Print question records.
	 */
	if (n = ntohs(hp->qdcount)) {
		fprintf(file, "QUESTIONS:\n");
		while (--n >= 0) {
			fprintf(file, "\t");
			cp = p_cdname(cp, msg, file);
			if (cp == NULL)
				return;
			fprintf(file, ", type = %s", p_type(_getshort(cp)));
			cp += sizeof (u_short);
			fprintf(file, ", class = %s\n\n",
						p_class(_getshort(cp)));
			cp += sizeof (u_short);
		}
	}
	/*
	 * Print authoritative answer records
	 */
	if (n = ntohs(hp->ancount)) {
		fprintf(file, "ANSWERS:\n");
		while (--n >= 0) {
			fprintf(file, "\t");
			cp = p_rr(cp, msg, file);
			if (cp == NULL)
				return;
		}
	}
	/*
	 * print name server records
	 */
	if (n = ntohs(hp->nscount)) {
		fprintf(file, "NAME SERVERS:\n");
		while (--n >= 0) {
			fprintf(file, "\t");
			cp = p_rr(cp, msg, file);
			if (cp == NULL)
				return;
		}
	}
	/*
	 * print additional records
	 */
	if (n = ntohs(hp->arcount)) {
		fprintf(file, "ADDITIONAL RECORDS:\n");
		while (--n >= 0) {
			fprintf(file, "\t");
			cp = p_rr(cp, msg, file);
			if (cp == NULL)
				return;
		}
	}
}

char *
p_cdname(cp, msg, file)
	char *cp, *msg;
	FILE *file;
{
	char name[MAXDNAME];
	int n;

	if ((n = dn_expand((u_char *)msg, (u_char *)(msg + 512), (u_char *)cp,
	    (u_char *)name, sizeof (name))) < 0)
		return (NULL);
	if (name[0] == '\0') {
		name[0] = '.';
		name[1] = '\0';
	}
	fputs(name, file);
	return (cp + n);
}

/*
 * Print resource record fields in human readable form.
 */
char *
p_rr(cp, msg, file)
	char *cp, *msg;
	FILE *file;
{
	int type, class, dlen, n, c;
	struct in_addr inaddr;
	char *cp1, *cp2;

	if ((cp = p_cdname(cp, msg, file)) == NULL)
		return (NULL);			/* compression error */
	fprintf(file, "\n\ttype = %s", p_type(type = _getshort(cp)));
	cp += sizeof (u_short);
	fprintf(file, ", class = %s", p_class(class = _getshort(cp)));
	cp += sizeof (u_short);
	fprintf(file, ", ttl = %s", p_time(_getlong(cp)));
	cp += sizeof (u_long);
	fprintf(file, ", dlen = %d\n", dlen = _getshort(cp));
	cp += sizeof (u_short);
	cp1 = cp;
	/*
	 * Print type specific data, if appropriate
	 */
	switch (type) {
	case T_A:
		switch (class) {
		case C_IN:
		case C_HS:
#ifdef SYSV
			memcpy((void *)&inaddr, (void *)cp, sizeof (inaddr));
#else
			bcopy(cp, (char *)&inaddr, sizeof (inaddr));
#endif
			if (dlen == 4) {
				fprintf(file, "\tinternet address = %s\n",
					inet_ntoa(inaddr));
				cp += dlen;
			} else if (dlen == 7) {
				fprintf(file, "\tinternet address = %s",
					inet_ntoa(inaddr));
				fprintf(file, ", protocol = %d", cp[4]);
				fprintf(file, ", port = %d\n",
					(cp[5] << 8) + cp[6]);
				cp += dlen;
			}
			break;
		default:
			cp += dlen;
		}
		break;
	case T_CNAME:
	case T_MB:
	case T_MG:
	case T_MR:
	case T_NS:
	case T_PTR:
		fprintf(file, "\tdomain name = ");
		cp = p_cdname(cp, msg, file);
		fprintf(file, "\n");
		break;

	case T_HINFO:
		if (n = *cp++) {
			fprintf(file, "\tCPU=%.*s\n", n, cp);
			cp += n;
		}
		if (n = *cp++) {
			fprintf(file, "\tOS=%.*s\n", n, cp);
			cp += n;
		}
		break;

	case T_SOA:
		fprintf(file, "\torigin = ");
		cp = p_cdname(cp, msg, file);
		fprintf(file, "\n\tmail addr = ");
		cp = p_cdname(cp, msg, file);
		fprintf(file, "\n\tserial = %ld", _getlong(cp));
		cp += sizeof (u_long);
		fprintf(file, "\n\trefresh = %s", p_time(_getlong(cp)));
		cp += sizeof (u_long);
		fprintf(file, "\n\tretry = %s", p_time(_getlong(cp)));
		cp += sizeof (u_long);
		fprintf(file, "\n\texpire = %s", p_time(_getlong(cp)));
		cp += sizeof (u_long);
		fprintf(file, "\n\tmin = %s\n", p_time(_getlong(cp)));
		cp += sizeof (u_long);
		break;

	case T_MX:
		fprintf(file, "\tpreference = %ld,", _getshort(cp));
		cp += sizeof (u_short);
		fprintf(file, " name = ");
		cp = p_cdname(cp, msg, file);
		break;

	case T_TXT:
		(void) fputs("\t\"", file);
		cp2 = cp1 + dlen;
		while (cp < cp2) {
			if (n = (unsigned char) *cp++) {
				for (c = n; c > 0 && cp < cp2; c--)
					if (*cp == '\n') {
					    (void) putc('\\', file);
					    (void) putc(*cp++, file);
					} else
					    (void) putc(*cp++, file);
			}
		}
		(void) fputs("\"\n", file);
		break;

	case T_MINFO:
		fprintf(file, "\trequests = ");
		cp = p_cdname(cp, msg, file);
		fprintf(file, "\n\terrors = ");
		cp = p_cdname(cp, msg, file);
		break;

	case T_UINFO:
		fprintf(file, "\t%s\n", cp);
		cp += dlen;
		break;

	case T_UID:
	case T_GID:
		if (dlen == 4) {
			fprintf(file, "\t%ld\n", _getlong(cp));
			cp += sizeof (int);
		}
		break;

	case T_WKS:
		if (dlen < sizeof (u_long) + 1)
			break;
#ifdef SYSV
		memcpy((void *)&inaddr, (void *)cp, sizeof (inaddr));
#else
		bcopy(cp, (char *)&inaddr, sizeof (inaddr));
#endif
		cp += sizeof (u_long);
		fprintf(file, "\tinternet address = %s, protocol = %d\n\t",
			inet_ntoa(inaddr), *cp++);
		n = 0;
		while (cp < cp1 + dlen) {
			c = *cp++;
			do {
				if (c & 0200)
					fprintf(file, " %d", n);
				c <<= 1;
			} while (++n & 07);
		}
		putc('\n', file);
		break;

#ifdef ALLOW_T_UNSPEC
	case T_UNSPEC:
		{
			int NumBytes = 8;
			char *DataPtr;
			int i;

			if (dlen < NumBytes) NumBytes = dlen;
			fprintf(file, "\tFirst %d bytes of hex data:",
				NumBytes);
			for (i = 0, DataPtr = cp; i < NumBytes; i++, DataPtr++)
				fprintf(file, " %x", *DataPtr);
			fputs("\n", file);
			cp += dlen;
		}
		break;
#endif /* ALLOW_T_UNSPEC */

	default:
		fprintf(file, "\t???\n");
		cp += dlen;
	}
	if (cp != cp1 + dlen) {
		fprintf(file, "packet size error (%#x != %#x)\n", cp, cp1+dlen);
		cp = NULL;
	}
	fprintf(file, "\n");
	return (cp);
}

static	char nbuf[40];

/*
 * Return a string for the type
 */
char *
p_type(type)
	int type;
{
	switch (type) {
	case T_A:
		return ("A");
	case T_NS:		/* authoritative server */
		return ("NS");
	case T_CNAME:		/* canonical name */
		return ("CNAME");
	case T_SOA:		/* start of authority zone */
		return ("SOA");
	case T_MB:		/* mailbox domain name */
		return ("MB");
	case T_MG:		/* mail group member */
		return ("MG");
	case T_MR:		/* mail rename name */
		return ("MR");
	case T_NULL:		/* null resource record */
		return ("NULL");
	case T_WKS:		/* well known service */
		return ("WKS");
	case T_PTR:		/* domain name pointer */
		return ("PTR");
	case T_HINFO:		/* host information */
		return ("HINFO");
	case T_MINFO:		/* mailbox information */
		return ("MINFO");
	case T_MX:		/* mail routing info */
		return ("MX");
	case T_TXT:		/* text */
		return ("TXT");
	case T_AXFR:		/* zone transfer */
		return ("AXFR");
	case T_MAILB:		/* mail box */
		return ("MAILB");
	case T_MAILA:		/* mail address */
		return ("MAILA");
	case T_ANY:		/* matches any type */
		return ("ANY");
	case T_UINFO:
		return ("UINFO");
	case T_UID:
		return ("UID");
	case T_GID:
		return ("GID");
#ifdef ALLOW_T_UNSPEC
	case T_UNSPEC:
		return ("UNSPEC");
#endif /* ALLOW_T_UNSPEC */
	default:
		(void) sprintf(nbuf, "%d", type);
		return (nbuf);
	}
}

/*
 * Return a mnemonic for class
 */
char *
p_class(class)
	int class;
{

	switch (class) {
	case C_IN:		/* internet class */
		return ("IN");
	case C_HS:		/* hesiod class */
		return ("HS");
	case C_ANY:		/* matches any class */
		return ("ANY");
	default:
		(void) sprintf(nbuf, "%d", class);
		return (nbuf);
	}
}

/*
 * Return a mnemonic for a time to live
 */
char *
p_time(value)
	u_long value;
{
	int secs, mins, hours;
	register char *p;

	if (value == 0) {
		strcpy(nbuf, "0 secs");
		return (nbuf);
	}

	secs = value % 60;
	value /= 60;
	mins = value % 60;
	value /= 60;
	hours = value % 24;
	value /= 24;

#define	PLURALIZE(x)	x, (x == 1) ? "" : "s"
	p = nbuf;
	if (value) {
		(void) sprintf(p, "%d day%s", PLURALIZE(value));
		while (*++p);
	}
	if (hours) {
		if (value)
			*p++ = ' ';
		(void) sprintf(p, "%d hour%s", PLURALIZE(hours));
		while (*++p);
	}
	if (mins) {
		if (value || hours)
			*p++ = ' ';
		(void) sprintf(p, "%d min%s", PLURALIZE(mins));
		while (*++p);
	}
	if (secs || ! (value || hours || mins)) {
		if (value || hours || mins)
			*p++ = ' ';
		(void) sprintf(p, "%d sec%s", PLURALIZE(secs));
	}
	return (nbuf);
}