/*
 * 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.
 */

/*
 * Send query to name server and wait for reply.
 */

#include <sys/param.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <sys/uio.h>
#include <sys/stat.h>
#include <netinet/in.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <arpa/nameser.h>
#include <arpa/inet.h>
#include <resolv.h>
#include "crossl.h"

/*
 * Undocumented external function in libsocket
 */
extern int
_socket(int, int, int);

static int s = -1;	/* socket used for communications */
#if	BSD >= 43
static struct sockaddr no_addr;
#endif /* BSD */


#ifndef FD_SET
#define	NFDBITS		32
#define	FD_SETSIZE	32
#define	FD_SET(n, p)	((p)->fds_bits[(n)/NFDBITS] |= (1 << ((n) % NFDBITS)))
#define	FD_CLR(n, p)	((p)->fds_bits[(n)/NFDBITS] &= ~(1 << ((n) % NFDBITS)))
#define	FD_ISSET(n, p)	((p)->fds_bits[(n)/NFDBITS] & (1 << ((n) % NFDBITS)))
#ifdef SYSV
#define	FD_ZERO(p)	(void) memset((void *)(p), 0, sizeof (*(p)))
#else
#define	FD_ZERO(p)	bzero((char *)(p), sizeof (*(p)))
#endif
#endif

/*
 * 1247019: Kludge to time out quickly if there is no /etc/resolv.conf
 * and a TCP connection to the local DNS server fails.
 */

static int _confcheck()
{
	int ns;
	struct stat rc_stat;
	struct sockaddr_in ns_sin;


	/* First, we check to see if /etc/resolv.conf exists.
	 * If it doesn't, then localhost is mostlikely to be
	 * the nameserver.
	 */
	if (stat(_PATH_RESCONF, &rc_stat) == -1 && errno == ENOENT) {

		/* Next, we check to see if _res.nsaddr is set to loopback.
		 * If it isn't, it has been altered by the application
		 * explicitly and we then want to bail with success.
		 */
		if (_res.nsaddr.sin_addr.S_un.S_addr == htonl(INADDR_LOOPBACK)) {

			/* Lastly, we try to connect to the TCP port of the
			 * nameserver.  If this fails, then we know that
			 * DNS is misconfigured and we can quickly exit.
			 */
			ns = socket(AF_INET, SOCK_STREAM, 0);
			IN_SET_LOOPBACK_ADDR(&ns_sin);
			ns_sin.sin_port = htons(NAMESERVER_PORT);
			if (connect(ns, (struct sockaddr *) &ns_sin,
				    sizeof ns_sin) == -1) {
				close(ns);
				return(-1);
			}
			else {
				close(ns);
				return(0);
			}
		}
	
		return(0);
	}
	
	return (0);
}

int
res_send(buf, buflen, answer, anslen)
	char *buf;
	int buflen;
	char *answer;
	int anslen;
{
	register int n;
	int try, v_circuit, resplen, ns;
	int gotsomewhere = 0;
#if	BSD >= 43
	int connected = 0;
#endif /* BSD */
	int connreset = 0;
	u_short id, len;
	char *cp;
	fd_set dsmask;
	struct timeval timeout;
	HEADER *hp = (HEADER *) buf;
	HEADER *anhp = (HEADER *) answer;
	struct iovec iov[2];
	int terrno = ETIMEDOUT;
	char junk[512];

#ifdef DEBUG
	if (_res.options & RES_DEBUG) {
		printf("res_send()\n");
		p_query(buf);
	}
#endif
	if (!(_res.options & RES_INIT))
		if (res_init() == -1) {
			return (-1);
		}

	/* 1247019: Check to see if we can bailout quickly. */
	if (_confcheck() == -1)
	    return(-1);

	v_circuit = (_res.options & RES_USEVC) || buflen > PACKETSZ;
	id = hp->id;
	/*
	 * Send request, RETRY times, or until successful
	 */
	for (try = 0; try < _res.retry; try++) {
		for (ns = 0; ns < _res.nscount; ns++) {
#ifdef DEBUG
			if (_res.options & RES_DEBUG)
				printf("Querying server (# %d) address = %s\n",
				ns+1, inet_ntoa(_res.nsaddr_list[ns].sin_addr));
#endif
		usevc:
			if (v_circuit) {
				int truncated = 0;

				/*
				 * Use virtual circuit;
				 * at most one attempt per server.
				 */
				try = _res.retry;
				if (s < 0) {
					s = _socket(AF_INET, SOCK_STREAM, 0);
					if (s < 0) {
						terrno = errno;
#ifdef DEBUG
						if (_res.options & RES_DEBUG) {
						perror("socket (vc) failed");
						}
#endif
						continue;
					}
					if (connect(s, (struct sockaddr *) &_res.nsaddr_list[ns],
						sizeof (struct sockaddr)) < 0) {
						terrno = errno;
#ifdef DEBUG
						if (_res.options & RES_DEBUG) {
						perror("connect failed");
						}
#endif
						(void) close(s);
						s = -1;
						continue;
					}
				}
				/*
				 * Send length & message
				 */
				len = htons((u_short)buflen);
				iov[0].iov_base = (caddr_t)&len;
				iov[0].iov_len = sizeof (len);
				iov[1].iov_base = buf;
				iov[1].iov_len = buflen;
				if (writev(s, iov, 2) != sizeof (len) +
								buflen) {
					terrno = errno;
#ifdef DEBUG
					if (_res.options & RES_DEBUG)
						perror("write failed");
#endif
					(void) close(s);
					s = -1;
					continue;
				}
				/*
				 * Receive length & response
				 */
				cp = answer;
				len = sizeof (short);
				while (len != 0 && (n = read
					(s, (char *)cp, (int)len)) > 0) {
					cp += n;
					len -= n;
				}
				if (n <= 0) {
					terrno = errno;
#ifdef DEBUG
					if (_res.options & RES_DEBUG)
						perror("read failed");
#endif
					(void) close(s);
					s = -1;
				/*
				 * A long running process might get its TCP
				 * connection reset if the remote server was
				 * restarted.  Requery the server instead of
				 * trying a new one.  When there is only one
				 * server, this means that a query might work
				 * instead of failing.  We only allow one reset
				 * per query to prevent looping.
				 */
					if (terrno == ECONNRESET &&
							!connreset) {
						connreset = 1;
						ns--;
					}
					continue;
				}
				cp = answer;
				if ((resplen = ntohs(*(u_short *)cp)) >
								anslen) {
#ifdef DEBUG
					if (_res.options & RES_DEBUG)
						fprintf(stderr,
							"response truncated\n");
#endif
					len = anslen;
					truncated = 1;
				} else
					len = resplen;
				while (len != 0 &&
					(n = read(s, (char *)cp,
							(int)len)) > 0) {
					cp += n;
					len -= n;
				}
				if (n <= 0) {
					terrno = errno;
#ifdef DEBUG
					if (_res.options & RES_DEBUG)
						perror("read failed");
#endif
					(void) close(s);
					s = -1;
					continue;
				}
				if (truncated) {
					/*
					 * Flush rest of answer
					 * so connection stays in synch.
					 */
					anhp->tc = 1;
					len = resplen - anslen;
					/*
					 * set the value of resplen to anslen,
					 * this is done because the caller
					 * assumes resplen contains the size of
					 * message read into the "answer" buffer
					 * passed in.
					 */
					resplen = anslen;

					while (len != 0) {
						n = (len > sizeof (junk) ?
							sizeof (junk) : len);
						if ((n = read(s, junk, n)) > 0)
							len -= n;
						else
							break;
					}
				}
			} else {
				/*
				 * Use datagrams.
				 */
				if (s < 0) {
					s = _socket(AF_INET, SOCK_DGRAM, 0);
					if (s < 0) {
						terrno = errno;
#ifdef DEBUG
						if (_res.options & RES_DEBUG) {
						perror("socket (dg) failed");
						}
#endif
						continue;
					}
				}
#if	BSD >= 43
			/*
			 * I'm tired of answering this question, so:
			 * On a 4.3BSD+ machine (client and server,
			 * actually), sending to a nameserver datagram
			 * port with no nameserver will cause an
			 * ICMP port unreachable message to be returned.
			 * If our datagram socket is "connected" to the
			 * server, we get an ECONNREFUSED error on the next
			 * socket operation, and select returns if the
			 * error message is received.  We can thus detect
			 * the absence of a nameserver without timing out.
			 * If we have sent queries to at least two servers,
			 * however, we don't want to remain connected,
			 * as we wish to receive answers from the first
			 * server to respond.
			 */
				if (_res.nscount == 1 ||
						(try == 0 && ns == 0)) {
					/*
					 * Don't use connect if we might
					 * still receive a response
					 * from another server.
					 */
					if (connected == 0) {
						if (connect(s,
						(struct sockaddr *) &_res.nsaddr_list[ns],
						sizeof (struct sockaddr)) < 0) {
#ifdef DEBUG
							if (_res.options &
								RES_DEBUG) {
							perror("connect");
							}
#endif
							continue;
						}
						connected = 1;
					}
					if (send(s, buf, buflen, 0) != buflen) {
#ifdef DEBUG
						if (_res.options & RES_DEBUG)
							perror("send");
#endif
						continue;
					}
				} else {
					/*
					 * Disconnect if we want to listen for
					 * responses from more than one server.
					 */
					if (connected) {
						(void) connect(s, &no_addr,
							sizeof (no_addr));
						connected = 0;
					}
#endif /* BSD */
					if (sendto(s, buf, buflen, 0,
						(struct sockaddr *) &_res.nsaddr_list[ns],
					sizeof (struct sockaddr)) != buflen) {
#ifdef DEBUG
						if (_res.options & RES_DEBUG)
							perror("sendto");
#endif
						continue;
					}
#if	BSD >= 43
				}
#endif

				/*
				 * Wait for reply
				 */
				timeout.tv_sec = (_res.retrans << try);
				if (try > 0)
					timeout.tv_sec /= _res.nscount;
				if (timeout.tv_sec <= 0)
					timeout.tv_sec = 1;
				timeout.tv_usec = 0;
wait:
				FD_ZERO(&dsmask);
				FD_SET(s, &dsmask);
				n = select(s+1, &dsmask, (fd_set *)NULL,
						(fd_set *)NULL, &timeout);
				if (n < 0) {
#ifdef DEBUG
					if (_res.options & RES_DEBUG)
						perror("select");
#endif
					continue;
				}
				if (n == 0) {
					/*
					 * timeout
					 */
#ifdef DEBUG
					if (_res.options & RES_DEBUG)
						printf("timeout\n");
#endif
#if BSD >= 43
					gotsomewhere = 1;
#endif
					continue;
				}
				if ((resplen = recv(s, answer, anslen, 0))
									<= 0) {
#ifdef DEBUG
					if (_res.options & RES_DEBUG)
						perror("recvfrom");
#endif
					continue;
				}
				gotsomewhere = 1;
				if (id != anhp->id) {
					/*
					 * response from old query, ignore it
					 */
#ifdef DEBUG
					if (_res.options & RES_DEBUG) {
						printf("old answer:\n");
						p_query(answer);
					}
#endif
					goto wait;
				}
				if (!(_res.options & RES_IGNTC) && anhp->tc) {
					/*
					 * get rest of answer;
					 * use TCP with same server.
					 */
#ifdef DEBUG
					if (_res.options & RES_DEBUG)
						printf("truncated answer\n");
#endif
					(void) close(s);
					s = -1;
					v_circuit = 1;
					goto usevc;
				}
			}
#ifdef DEBUG
			if (_res.options & RES_DEBUG) {
				printf("got answer:\n");
				p_query(answer);
			}
#endif
		/*
		 * If using virtual circuits, we assume that the first server
		 * is preferred * over the rest (i.e. it is on the local
		 * machine) and only keep that one open.
		 * If we have temporarily opened a virtual circuit,
		 * or if we haven't been asked to keep a socket open,
		 * close the socket.
		 */
			if ((v_circuit &&
				((_res.options & RES_USEVC) == 0 || ns != 0)) ||
				(_res.options & RES_STAYOPEN) == 0) {
				(void) close(s);
				s = -1;
			}
			return (resplen);
		}
	}
	if (s >= 0) {
		(void) close(s);
		s = -1;
	}
	if (v_circuit == 0)
		if (gotsomewhere == 0)
			errno = ECONNREFUSED;	/* no nameservers found */
		else
			errno = ETIMEDOUT;	/* no answer obtained */
	else
		errno = terrno;
	return (-1);
}

/*
 * This routine is for closing the socket if a virtual circuit is used and
 * the program wants to close it.  This provides support for endhostent()
 * which expects to close the socket.
 *
 * This routine is not expected to be user visible.
 */
void
_res_close()
{
	if (s != -1) {
		(void) close(s);
		s = -1;
	}
}