mirror of https://github.com/lianthony/NT4.0
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
412 lines
11 KiB
412 lines
11 KiB
/*++
|
|
|
|
Copyright (c) 1991 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
r_query.c
|
|
|
|
Abstract:
|
|
|
|
This module implements routines to formulate and send a DNS resolver
|
|
query.
|
|
|
|
Author:
|
|
|
|
Mike Massa (mikemas) Sept 20, 1991
|
|
|
|
Revision History:
|
|
|
|
Who When What
|
|
-------- -------- ----------------------------------------------
|
|
mikemas 9-20-91 created
|
|
|
|
Notes:
|
|
|
|
Exports:
|
|
res_query()
|
|
res_search()
|
|
|
|
--*/
|
|
|
|
#ident "@(#)res_query.c 5.3 3/8/91"
|
|
|
|
/******************************************************************
|
|
*
|
|
* SpiderTCP BIND
|
|
*
|
|
* Copyright 1990 Spider Systems Limited
|
|
*
|
|
* RES_QUERY.C
|
|
*
|
|
******************************************************************/
|
|
|
|
/*
|
|
* /usr/projects/tcp/SCCS.rel3/rel/src/lib/net/0/s.res_query.c
|
|
* @(#)res_query.c 5.3
|
|
*
|
|
* Last delta created 14:11:46 3/4/91
|
|
* This file extracted 11:20:32 3/8/91
|
|
*
|
|
* Modifications:
|
|
*
|
|
* GSS 24 Jul 90 New File
|
|
*/
|
|
/*
|
|
* Copyright (c) 1988 Regents of the University of California.
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms are permitted
|
|
* provided that: (1) source distributions retain this entire copyright
|
|
* notice and comment, and (2) distributions including binaries display
|
|
* the following acknowledgement: ``This product includes software
|
|
* developed by the University of California, Berkeley and its contributors''
|
|
* in the documentation or other materials provided with the distribution
|
|
* and in all advertising materials mentioning features or use of this
|
|
* software. 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 ``AS IS'' AND WITHOUT ANY EXPRESS OR
|
|
* IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
|
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
|
|
*/
|
|
|
|
#if defined(LIBC_SCCS) && !defined(lint)
|
|
static char sccsid[] = "@(#)res_query.c 5.7 (Berkeley) 6/1/90";
|
|
#endif /* LIBC_SCCS and not lint */
|
|
/****************************************************************************/
|
|
|
|
#include "winsockp.h"
|
|
#include <ctype.h>
|
|
|
|
#if PACKETSZ > 1024
|
|
#define MAXPACKET PACKETSZ
|
|
#else
|
|
#define MAXPACKET 1024
|
|
#endif
|
|
|
|
//
|
|
// Local function prototypes
|
|
//
|
|
|
|
int
|
|
res_querydomain(
|
|
char *name,
|
|
char *domain,
|
|
int class, /* class of query */
|
|
int type, /* type of query */
|
|
unsigned char *answer, /* buffer to put answer */
|
|
int anslen /* size of answer */
|
|
);
|
|
|
|
//
|
|
// Formulate a normal query, send, and await answer. Returned answer is
|
|
// placed in supplied buffer "answer". Perform preliminary check of
|
|
// answer, returning success only if no error is indicated and the
|
|
// answer count is nonzero. Return the size of the response on success,
|
|
// -1 on error. Error number is left in h_errno thru call to
|
|
// SetLastError. Caller must parse answer and determine whether it
|
|
// answers the question.
|
|
//
|
|
|
|
|
|
int
|
|
res_query(
|
|
IN char *name, /* domain name */
|
|
IN int class, /* class of query */
|
|
IN int type, /* type of query */
|
|
OUT unsigned char *answer, /* buffer to put answer */
|
|
IN int anslen /* size of answer buffer */
|
|
)
|
|
{
|
|
char buf[MAXPACKET];
|
|
HEADER *hp;
|
|
int n;
|
|
|
|
IF_DEBUG(RESOLVER) {
|
|
WS_PRINT(("res_query entered\n"));
|
|
}
|
|
|
|
if (res_init() == -1) {
|
|
return (-1);
|
|
}
|
|
|
|
IF_DEBUG(RESOLVER) {
|
|
WS_PRINT(("res_query(%s, %d, %d)\n", name, class, type));
|
|
}
|
|
|
|
n = res_mkquery(QUERY, name, class, type, (char *)NULL, 0, NULL,
|
|
buf, sizeof(buf));
|
|
|
|
if (n <= 0) {
|
|
IF_DEBUG(RESOLVER) {
|
|
WS_PRINT(("res_query: mkquery failed\n"));
|
|
}
|
|
|
|
SetLastError(WSANO_RECOVERY);
|
|
return (n);
|
|
}
|
|
n = res_send(buf, n, answer, anslen);
|
|
if (n < 0) {
|
|
IF_DEBUG(RESOLVER) {
|
|
WS_PRINT(("res_query: send error\n"));
|
|
}
|
|
SockThreadDnrErrorCode = WSATRY_AGAIN;
|
|
SetLastError(WSATRY_AGAIN);
|
|
return(n);
|
|
}
|
|
|
|
hp = (HEADER *) answer;
|
|
|
|
if (hp->rcode != NOERROR || ntohs(hp->ancount) == 0) {
|
|
|
|
IF_DEBUG(RESOLVER) {
|
|
WS_PRINT(("rcode = %d, ancount=%d\n", hp->rcode,
|
|
ntohs(hp->ancount)));
|
|
}
|
|
|
|
switch (hp->rcode) {
|
|
|
|
case NXDOMAIN:
|
|
SockThreadDnrErrorCode = WSAHOST_NOT_FOUND;
|
|
SetLastError(WSAHOST_NOT_FOUND);
|
|
break;
|
|
|
|
case SERVFAIL:
|
|
SockThreadDnrErrorCode = WSATRY_AGAIN;
|
|
SetLastError(WSATRY_AGAIN);
|
|
break;
|
|
|
|
case NOERROR:
|
|
SockThreadDnrErrorCode = WSANO_DATA;
|
|
SetLastError(WSANO_DATA);
|
|
break;
|
|
|
|
case FORMERR:
|
|
case NOTIMP:
|
|
case REFUSED:
|
|
default:
|
|
SockThreadDnrErrorCode = WSANO_RECOVERY;
|
|
SetLastError(WSANO_RECOVERY);
|
|
break;
|
|
}
|
|
|
|
return (-1);
|
|
}
|
|
return(n);
|
|
}
|
|
|
|
//
|
|
// Formulate a normal query, send, and retrieve answer in supplied buffer.
|
|
// Return the size of the response on success, -1 on error.
|
|
// If enabled, implement search rules until answer or unrecoverable failure
|
|
// is detected. Error number is left in h_errno thru call to SetLastError.
|
|
// Only useful for queries in the same name hierarchy as the local host
|
|
// (not, for example, for host address-to-name lookups in domain in-addr.arpa).
|
|
//
|
|
|
|
|
|
int
|
|
res_search(
|
|
IN char *name, /* domain name */
|
|
IN int class, /* class of query */
|
|
IN int type, /* type of query */
|
|
OUT unsigned char *answer, /* buffer to put answer */
|
|
IN int anslen /* size of answer */
|
|
)
|
|
{
|
|
register char *cp, **domain;
|
|
int n, ret, got_nodata = 0;
|
|
char *hostalias();
|
|
|
|
IF_DEBUG(RESOLVER) {
|
|
WS_PRINT(("res_search entered\n"));
|
|
}
|
|
|
|
if (res_init() == -1) {
|
|
return (-1);
|
|
}
|
|
|
|
SetLastError(HOST_NOT_FOUND); /* default, if we never query */
|
|
for (cp = name, n = 0; *cp; cp++) {
|
|
if (*cp == '.') {
|
|
n++;
|
|
}
|
|
}
|
|
|
|
if (n == 0 && (cp = hostalias(name))) {
|
|
return (res_query(cp, class, type, answer, anslen));
|
|
}
|
|
|
|
//
|
|
// If the search/default failed, try the name as fully-qualified,
|
|
// but only if it contained at least one dot (even trailing). This
|
|
// is purely a heuristic; we assume that any reasonable query about
|
|
// a top-level domain (for servers, SOA, etc) will not use
|
|
// res_search.
|
|
//
|
|
|
|
if (!SockThreadGetXByYCancelled && n != 0 &&
|
|
(ret = res_querydomain(name, (char *)NULL, class, type,
|
|
answer, anslen)) > 0) {
|
|
return (ret);
|
|
}
|
|
|
|
//
|
|
// If there was no DNS server available, give up.
|
|
//
|
|
|
|
if (GetLastError() == WSAECONNREFUSED) {
|
|
SetLastError(TRY_AGAIN);
|
|
return (-1);
|
|
}
|
|
|
|
//
|
|
// We do at least one level of search if
|
|
// - there is no dot and RES_DEFNAME is set, or
|
|
// - there is at least one dot, there is no trailing dot,
|
|
// and RES_DNSRCH is set.
|
|
//
|
|
|
|
if ((n == 0 && _res.options & RES_DEFNAMES) ||
|
|
(n != 0 && *--cp != '.' && _res.options & RES_DNSRCH)) {
|
|
|
|
for ( domain = _res.dnsrch;
|
|
*domain != NULL && **domain != '\0' && !SockThreadGetXByYCancelled;
|
|
domain++ ) {
|
|
|
|
ret = res_querydomain(name, *domain, class, type, answer, anslen);
|
|
if (ret > 0) {
|
|
return (ret);
|
|
}
|
|
|
|
//
|
|
// If no server present, give up. If name isn't found in
|
|
// this domain, keep trying higher domains in the search
|
|
// list (if that's enabled). On a NO_DATA error, keep
|
|
// trying, otherwise a wildcard entry of another type could
|
|
// keep us from finding this entry higher in the domain. If
|
|
// we get some other error (negative answer or server
|
|
// failure), then stop searching up, but try the input name
|
|
// below in case it's fully-qualified.
|
|
//
|
|
|
|
if (GetLastError() == WSAECONNREFUSED) {
|
|
SetLastError(TRY_AGAIN);
|
|
return (-1);
|
|
}
|
|
if (GetLastError() == NO_DATA) {
|
|
got_nodata++;
|
|
}
|
|
if ( (GetLastError() != HOST_NOT_FOUND && GetLastError() != NO_DATA
|
|
) ||
|
|
(_res.options & RES_DNSRCH) == 0 ) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// If there were no dots in the name and we still haven't
|
|
// successfully resolved the name, try the raw name.
|
|
//
|
|
|
|
if (!SockThreadGetXByYCancelled && n == 0 &&
|
|
(ret = res_querydomain(name, (char *)NULL, class, type,
|
|
answer, anslen)) > 0) {
|
|
return (ret);
|
|
}
|
|
if (got_nodata) {
|
|
SetLastError(NO_DATA);
|
|
}
|
|
return (-1);
|
|
}
|
|
|
|
//
|
|
// Perform a call on res_query on the concatenation of name and domain,
|
|
// removing a trailing dot from name if domain is NULL.
|
|
//
|
|
|
|
int
|
|
res_querydomain(
|
|
char *name,
|
|
char *domain,
|
|
int class, /* class of query */
|
|
int type, /* type of query */
|
|
unsigned char *answer, /* buffer to put answer */
|
|
int anslen /* size of answer */
|
|
)
|
|
{
|
|
char nbuf[2*MAXDNAME+2];
|
|
char *longname = nbuf;
|
|
int n;
|
|
|
|
IF_DEBUG(RESOLVER) {
|
|
WS_PRINT(("res_querydomain entered\n"));
|
|
WS_PRINT(("res_querydomain(%s, %s, %d, %d)\n",
|
|
name, (domain==NULL)?"NULL":domain, class, type));
|
|
}
|
|
|
|
if (domain == NULL) {
|
|
/*
|
|
* Check for trailing '.';
|
|
* copy without '.' if present.
|
|
*/
|
|
n = strlen(name) - 1;
|
|
if (name[n] == '.' && n < sizeof(nbuf) - 1) {
|
|
bcopy(name, nbuf, n);
|
|
nbuf[n] = '\0';
|
|
} else {
|
|
longname = name;
|
|
}
|
|
} else {
|
|
//
|
|
// Note that we cannot use wsprintf() here, as it does not support
|
|
// the variable field width formatting syntax ("%.*s").
|
|
//
|
|
|
|
(void)sprintf(nbuf, "%.*s.%.*s", MAXDNAME, name, MAXDNAME, domain);
|
|
}
|
|
|
|
return (res_query(longname, class, type, answer, anslen));
|
|
}
|
|
|
|
char *
|
|
hostalias(
|
|
char *name
|
|
)
|
|
{
|
|
register char *C1, *C2;
|
|
FILE *fp;
|
|
char *file;
|
|
char buf[BUFSIZ];
|
|
static char abuf[MAXDNAME];
|
|
|
|
//IF_DEBUG(RESOLVER) {
|
|
// WS_PRINT(("hostalias entered\n"));
|
|
//}
|
|
|
|
file = getenv("HOSTALIASES");
|
|
if (file == NULL || (fp = fopen(file, "r")) == NULL)
|
|
return (NULL);
|
|
buf[sizeof(buf) - 1] = '\0';
|
|
while (fgets(buf, sizeof(buf), fp)) {
|
|
for (C1 = buf; *C1 && !isspace(*C1); ++C1);
|
|
if (!*C1)
|
|
break;
|
|
*C1 = '\0';
|
|
if (!_stricmp(buf, name)) {
|
|
while (isspace(*++C1));
|
|
if (!*C1) {
|
|
break;
|
|
}
|
|
for (C2 = C1 + 1; *C2 && !isspace(*C2); ++C2);
|
|
abuf[sizeof(abuf) - 1] = *C2 = '\0';
|
|
(void)strncpy(abuf, C1, sizeof(abuf) - 1);
|
|
fclose(fp);
|
|
return (abuf);
|
|
}
|
|
}
|
|
fclose(fp);
|
|
return (NULL);
|
|
}
|