1c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* $Id: evdns.c 6979 2006-08-04 18:31:13Z nickm $ */ 2c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* The original version of this module was written by Adam Langley; for 4c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * a history of modifications, check out the subversion logs. 5c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * 6c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * When editing this module, try to keep it re-mergeable by Adam. Don't 7c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * reformat the whitespace, add Tor dependencies, or so on. 8c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * 9c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * TODO: 10c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * - Support IPv6 and PTR records. 11c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * - Replace all externally visible magic numbers with #defined constants. 12c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * - Write doccumentation for APIs of all external functions. 13c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott */ 14c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 15c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Async DNS Library 16c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * Adam Langley <agl@imperialviolet.org> 17c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * http://www.imperialviolet.org/eventdns.html 18c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * Public Domain code 19c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * 20c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * This software is Public Domain. To view a copy of the public domain dedication, 21c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * visit http://creativecommons.org/licenses/publicdomain/ or send a letter to 22c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * Creative Commons, 559 Nathan Abbott Way, Stanford, California 94305, USA. 23c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * 24c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * I ask and expect, but do not require, that all derivative works contain an 25c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * attribution similar to: 26c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * Parts developed by Adam Langley <agl@imperialviolet.org> 27c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * 28c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * You may wish to replace the word "Parts" with something else depending on 29c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * the amount of original code. 30c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * 31c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * (Derivative works does not include programs which link against, run or include 32c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * the source verbatim in their source distributions) 33c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * 34c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * Version: 0.1b 35c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott */ 36c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 37c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <sys/types.h> 38c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef HAVE_CONFIG_H 39c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "config.h" 40c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 41c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 42c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_FTIME_FOR_ID 43c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <sys/timeb.h> 44c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 45c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 46c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifndef DNS_USE_CPU_CLOCK_FOR_ID 47c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifndef DNS_USE_GETTIMEOFDAY_FOR_ID 48c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifndef DNS_USE_OPENSSL_FOR_ID 49c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifndef DNS_USE_FTIME_FOR_ID 50c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#error Must configure at least one id generation method. 51c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#error Please see the documentation. 52c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 53c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 54c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 55c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 56c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 57c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* #define _POSIX_C_SOURCE 200507 */ 58c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define _GNU_SOURCE 59c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 60c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_CPU_CLOCK_FOR_ID 61c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_OPENSSL_FOR_ID 62c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#error Multiple id options selected 63c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 64c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_GETTIMEOFDAY_FOR_ID 65c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#error Multiple id options selected 66c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 67c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <time.h> 68c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 69c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 70c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_OPENSSL_FOR_ID 71c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_GETTIMEOFDAY_FOR_ID 72c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#error Multiple id options selected 73c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 74c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <openssl/rand.h> 75c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 76c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 77c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifndef _FORTIFY_SOURCE 78c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define _FORTIFY_SOURCE 3 79c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 80c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 81c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <string.h> 82c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <fcntl.h> 83c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef HAVE_SYS_TIME_H 84c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <sys/time.h> 85c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 86c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef HAVE_STDINT_H 87c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <stdint.h> 88c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 89c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <stdlib.h> 90c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <string.h> 91c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <errno.h> 92c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <assert.h> 93c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef HAVE_UNISTD_H 94c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <unistd.h> 95c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 96c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <limits.h> 97c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <sys/stat.h> 98c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <ctype.h> 99c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <stdio.h> 100c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <stdarg.h> 101c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 102c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "evdns.h" 103c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "evutil.h" 104c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "log.h" 105c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef WIN32 106c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <winsock2.h> 107c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <windows.h> 108c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <iphlpapi.h> 109c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <io.h> 110c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#else 111c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <sys/socket.h> 112c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <netinet/in.h> 113c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <arpa/inet.h> 114c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 115c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 116c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef HAVE_NETINET_IN6_H 117c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <netinet/in6.h> 118c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 119c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 120c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define EVDNS_LOG_DEBUG 0 121c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define EVDNS_LOG_WARN 1 122c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 123c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifndef HOST_NAME_MAX 124c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define HOST_NAME_MAX 255 125c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 126c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 127c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <stdio.h> 128c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 129c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#undef MIN 130c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define MIN(a,b) ((a)<(b)?(a):(b)) 131c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 132c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef __USE_ISOC99B 133c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* libevent doesn't work without this */ 134c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scotttypedef ev_uint8_t u_char; 135c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scotttypedef unsigned int uint; 136c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 137c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include <event.h> 138c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 139c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define u64 ev_uint64_t 140c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define u32 ev_uint32_t 141c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define u16 ev_uint16_t 142c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define u8 ev_uint8_t 143c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 144c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef WIN32 145c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define open _open 146c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define read _read 147c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define close _close 148c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define strdup _strdup 149c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 150c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 151c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define MAX_ADDRS 32 /* maximum number of addresses from a single packet */ 152c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* which we bother recording */ 153c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 154c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define TYPE_A EVDNS_TYPE_A 155c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define TYPE_CNAME 5 156c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define TYPE_PTR EVDNS_TYPE_PTR 157c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define TYPE_AAAA EVDNS_TYPE_AAAA 158c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 159c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define CLASS_INET EVDNS_CLASS_INET 160c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 161c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct request { 162c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u8 *request; /* the dns packet data */ 163c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott unsigned int request_len; 164c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int reissue_count; 165c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int tx_count; /* the number of times that this packet has been sent */ 166c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott unsigned int request_type; /* TYPE_PTR or TYPE_A */ 167c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott void *user_pointer; /* the pointer given to us for this request */ 168c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_callback_type user_callback; 169c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *ns; /* the server which we last sent it */ 170c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 171c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* elements used by the searching code */ 172c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int search_index; 173c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct search_state *search_state; 174c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *search_origname; /* needs to be free()ed */ 175c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int search_flags; 176c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 177c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* these objects are kept in a circular list */ 178c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *next, *prev; 179c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 180c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct event timeout_event; 181c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 182c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 trans_id; /* the transaction id */ 183c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char request_appended; /* true if the request pointer is data which follows this struct */ 184c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char transmit_me; /* needs to be transmitted */ 185c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 186c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 187c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifndef HAVE_STRUCT_IN6_ADDR 188c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct in6_addr { 189c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u8 s6_addr[16]; 190c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 191c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 192c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 193c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct reply { 194c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott unsigned int type; 195c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott unsigned int have_answer; 196c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott union { 197c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct { 198c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 addrcount; 199c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 addresses[MAX_ADDRS]; 200c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } a; 201c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct { 202c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 addrcount; 203c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct in6_addr addresses[MAX_ADDRS]; 204c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } aaaa; 205c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct { 206c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char name[HOST_NAME_MAX]; 207c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } ptr; 208c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } data; 209c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 210c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 211c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct nameserver { 212c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int socket; /* a connected UDP socket */ 213c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 address; 214c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 port; 215c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int failed_times; /* number of times which we have given this server a chance */ 216c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int timedout; /* number of times in a row a request has timed out */ 217c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct event event; 218c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* these objects are kept in a circular list */ 219c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *next, *prev; 220c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct event timeout_event; /* used to keep the timeout for */ 221c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* when we next probe this server. */ 222c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Valid if state == 0 */ 223c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char state; /* zero if we think that this server is down */ 224c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char choked; /* true if we have an EAGAIN from this server's socket */ 225c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char write_waiting; /* true if we are waiting for EV_WRITE events */ 226c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 227c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 228c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct request *req_head = NULL, *req_waiting_head = NULL; 229c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct nameserver *server_head = NULL; 230c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 231c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Represents a local port where we're listening for DNS requests. Right now, */ 232c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* only UDP is supported. */ 233c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct evdns_server_port { 234c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int socket; /* socket we use to read queries and write replies. */ 235c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int refcnt; /* reference count. */ 236c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char choked; /* Are we currently blocked from writing? */ 237c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char closing; /* Are we trying to close this port, pending writes? */ 238c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_request_callback_fn_type user_callback; /* Fn to handle requests */ 239c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott void *user_data; /* Opaque pointer passed to user_callback */ 240c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct event event; /* Read/write event */ 241c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* circular list of replies that we want to write. */ 242c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_request *pending_replies; 243c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 244c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 245c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Represents part of a reply being built. (That is, a single RR.) */ 246c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct server_reply_item { 247c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_reply_item *next; /* next item in sequence. */ 248c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *name; /* name part of the RR */ 249c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 type : 16; /* The RR type */ 250c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 class : 16; /* The RR class (usually CLASS_INET) */ 251c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 ttl; /* The RR TTL */ 252c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char is_name; /* True iff data is a label */ 253c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 datalen; /* Length of data; -1 if data is a label */ 254c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott void *data; /* The contents of the RR */ 255c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 256c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 257c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Represents a request that we've received as a DNS server, and holds */ 258c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* the components of the reply as we're constructing it. */ 259c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct server_request { 260c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Pointers to the next and previous entries on the list of replies */ 261c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* that we're waiting to write. Only set if we have tried to respond */ 262c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* and gotten EAGAIN. */ 263c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_request *next_pending; 264c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_request *prev_pending; 265c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 266c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 trans_id; /* Transaction id. */ 267c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct evdns_server_port *port; /* Which port received this request on? */ 268c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct sockaddr_storage addr; /* Where to send the response */ 269c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott socklen_t addrlen; /* length of addr */ 270c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 271c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int n_answer; /* how many answer RRs have been set? */ 272c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int n_authority; /* how many authority RRs have been set? */ 273c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int n_additional; /* how many additional RRs have been set? */ 274c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 275c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_reply_item *answer; /* linked list of answer RRs */ 276c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_reply_item *authority; /* linked list of authority RRs */ 277c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_reply_item *additional; /* linked list of additional RRs */ 278c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 279c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Constructed response. Only set once we're ready to send a reply. */ 280c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Once this is set, the RR fields are cleared, and no more should be set. */ 281c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *response; 282c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott size_t response_len; 283c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 284c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Caller-visible fields: flags, questions. */ 285c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct evdns_server_request base; 286c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 287c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 288c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* helper macro */ 289c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define OFFSET_OF(st, member) ((off_t) (((char*)&((st*)0)->member)-(char*)0)) 290c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 291c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Given a pointer to an evdns_server_request, get the corresponding */ 292c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* server_request. */ 293c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define TO_SERVER_REQUEST(base_ptr) \ 294c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ((struct server_request*) \ 295c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (((char*)(base_ptr) - OFFSET_OF(struct server_request, base)))) 296c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 297c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* The number of good nameservers that we have */ 298c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int global_good_nameservers = 0; 299c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 300c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* inflight requests are contained in the req_head list */ 301c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* and are actually going out across the network */ 302c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int global_requests_inflight = 0; 303c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* requests which aren't inflight are in the waiting list */ 304c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* and are counted here */ 305c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int global_requests_waiting = 0; 306c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 307c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int global_max_requests_inflight = 64; 308c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 309c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct timeval global_timeout = {5, 0}; /* 5 seconds */ 310c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int global_max_reissues = 1; /* a reissue occurs when we get some errors from the server */ 311c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int global_max_retransmits = 3; /* number of times we'll retransmit a request which timed out */ 312c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* number of timeouts in a row before we consider this server to be down */ 313c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int global_max_nameserver_timeout = 3; 314c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 315c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* These are the timeout values for nameservers. If we find a nameserver is down */ 316c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* we try to probe it at intervals as given below. Values are in seconds. */ 317c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic const struct timeval global_nameserver_timeouts[] = {{10, 0}, {60, 0}, {300, 0}, {900, 0}, {3600, 0}}; 318c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic const int global_nameserver_timeouts_length = sizeof(global_nameserver_timeouts)/sizeof(struct timeval); 319c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 320c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct nameserver *nameserver_pick(void); 321c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void evdns_request_insert(struct request *req, struct request **head); 322c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void nameserver_ready_callback(int fd, short events, void *arg); 323c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int evdns_transmit(void); 324c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int evdns_request_transmit(struct request *req); 325c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void nameserver_send_probe(struct nameserver *const ns); 326c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void search_request_finished(struct request *const); 327c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int search_try_next(struct request *const req); 328c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int search_request_new(int type, const char *const name, int flags, evdns_callback_type user_callback, void *user_arg); 329c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void evdns_requests_pump_waiting_queue(void); 330c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic u16 transaction_id_pick(void); 331c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct request *request_new(int type, const char *name, int flags, evdns_callback_type callback, void *ptr); 332c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void request_submit(struct request *const req); 333c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 334c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int server_request_free(struct server_request *req); 335c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void server_request_free_answers(struct server_request *req); 336c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void server_port_free(struct evdns_server_port *port); 337c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void server_port_ready_callback(int fd, short events, void *arg); 338c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 339c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int strtoint(const char *const str); 340c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 341c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef WIN32 342c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 343c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottlast_error(int sock) 344c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 345c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int optval, optvallen=sizeof(optval); 346c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int err = WSAGetLastError(); 347c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (err == WSAEWOULDBLOCK && sock >= 0) { 348c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (getsockopt(sock, SOL_SOCKET, SO_ERROR, (void*)&optval, 349c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott &optvallen)) 350c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return err; 351c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (optval) 352c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return optval; 353c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 354c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return err; 355c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 356c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 357c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 358c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scotterror_is_eagain(int err) 359c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 360c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return err == EAGAIN || err == WSAEWOULDBLOCK; 361c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 362c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 363c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottinet_aton(const char *c, struct in_addr *addr) 364c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 365c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ev_uint32_t r; 366c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (strcmp(c, "255.255.255.255") == 0) { 367c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott addr->s_addr = 0xffffffffu; 368c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 369c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = inet_addr(c); 370c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r == INADDR_NONE) 371c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 372c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott addr->s_addr = r; 373c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 374c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 375c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 376c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#else 377c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define last_error(sock) (errno) 378c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define error_is_eagain(err) ((err) == EAGAIN) 379c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 380c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define CLOSE_SOCKET(s) EVUTIL_CLOSESOCKET(s) 381c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 382c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define ISSPACE(c) isspace((int)(unsigned char)(c)) 383c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define ISDIGIT(c) isdigit((int)(unsigned char)(c)) 384c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 385c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic const char * 386c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottdebug_ntoa(u32 address) 387c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 388c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott static char buf[32]; 389c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 a = ntohl(address); 390c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evutil_snprintf(buf, sizeof(buf), "%d.%d.%d.%d", 391c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a>>24)&0xff), 392c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a>>16)&0xff), 393c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a>>8 )&0xff), 394c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a )&0xff)); 395c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return buf; 396c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 397c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 398c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic evdns_debug_log_fn_type evdns_log_fn = NULL; 399c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 400c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 401c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_set_log_fn(evdns_debug_log_fn_type fn) 402c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 403c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_log_fn = fn; 404c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 405c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 406c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef __GNUC__ 407c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define EVDNS_LOG_CHECK __attribute__ ((format(printf, 2, 3))) 408c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#else 409c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define EVDNS_LOG_CHECK 410c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 411c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 412c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void _evdns_log(int warn, const char *fmt, ...) EVDNS_LOG_CHECK; 413c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 414c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott_evdns_log(int warn, const char *fmt, ...) 415c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 416c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott va_list args; 417c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott static char buf[512]; 418c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!evdns_log_fn) 419c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 420c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott va_start(args,fmt); 421c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evutil_vsnprintf(buf, sizeof(buf), fmt, args); 422c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf[sizeof(buf)-1] = '\0'; 423c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_log_fn(warn, buf); 424c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott va_end(args); 425c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 426c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 427c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define log _evdns_log 428c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 429c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* This walks the list of inflight requests to find the */ 430c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* one with a matching transaction id. Returns NULL on */ 431c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* failure */ 432c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct request * 433c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottrequest_find_from_trans_id(u16 trans_id) { 434c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *req = req_head, *const started_at = req_head; 435c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 436c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req) { 437c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott do { 438c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->trans_id == trans_id) return req; 439c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = req->next; 440c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while (req != started_at); 441c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 442c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 443c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return NULL; 444c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 445c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 446c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* a libevent callback function which is called when a nameserver */ 447c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* has gone down and we want to test if it has came back to life yet */ 448c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 449c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_prod_callback(int fd, short events, void *arg) { 450c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *const ns = (struct nameserver *) arg; 451c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void)fd; 452c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void)events; 453c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 454c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_send_probe(ns); 455c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 456c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 457c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* a libevent callback which is called when a nameserver probe (to see if */ 458c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* it has come back to life) times out. We increment the count of failed_times */ 459c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* and wait longer to send the next probe packet. */ 460c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 461c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_probe_failed(struct nameserver *const ns) { 462c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const struct timeval * timeout; 463c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) evtimer_del(&ns->timeout_event); 464c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (ns->state == 1) { 465c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* This can happen if the nameserver acts in a way which makes us mark */ 466c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* it as bad and then starts sending good replies. */ 467c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 468c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 469c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 470c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott timeout = 471c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott &global_nameserver_timeouts[MIN(ns->failed_times, 472c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_nameserver_timeouts_length - 1)]; 473c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->failed_times++; 474c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 475c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (evtimer_add(&ns->timeout_event, (struct timeval *) timeout) < 0) { 476c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, 477c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott "Error from libevent when adding timer event for %s", 478c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott debug_ntoa(ns->address)); 479c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* ???? Do more? */ 480c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 481c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 482c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 483c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* called when a nameserver has been deemed to have failed. For example, too */ 484c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* many packets have timed out etc */ 485c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 486c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_failed(struct nameserver *const ns, const char *msg) { 487c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *req, *started_at; 488c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* if this nameserver has already been marked as failed */ 489c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* then don't do anything */ 490c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!ns->state) return; 491c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 492c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Nameserver %s has failed: %s", 493c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott debug_ntoa(ns->address), msg); 494c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_good_nameservers--; 495c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(global_good_nameservers >= 0); 496c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (global_good_nameservers == 0) { 497c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "All nameservers have failed"); 498c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 499c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 500c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->state = 0; 501c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->failed_times = 1; 502c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 503c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (evtimer_add(&ns->timeout_event, (struct timeval *) &global_nameserver_timeouts[0]) < 0) { 504c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, 505c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott "Error from libevent when adding timer event for %s", 506c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott debug_ntoa(ns->address)); 507c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* ???? Do more? */ 508c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 509c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 510c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* walk the list of inflight requests to see if any can be reassigned to */ 511c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* a different server. Requests in the waiting queue don't have a */ 512c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* nameserver assigned yet */ 513c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 514c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* if we don't have *any* good nameservers then there's no point */ 515c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* trying to reassign requests to one */ 516c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!global_good_nameservers) return; 517c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 518c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = req_head; 519c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott started_at = req_head; 520c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req) { 521c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott do { 522c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->tx_count == 0 && req->ns == ns) { 523c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* still waiting to go out, can be moved */ 524c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* to another server */ 525c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->ns = nameserver_pick(); 526c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 527c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = req->next; 528c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while (req != started_at); 529c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 530c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 531c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 532c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 533c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_up(struct nameserver *const ns) { 534c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (ns->state) return; 535c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Nameserver %s is back up", 536c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott debug_ntoa(ns->address)); 537c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evtimer_del(&ns->timeout_event); 538c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->state = 1; 539c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->failed_times = 0; 540c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->timedout = 0; 541c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_good_nameservers++; 542c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 543c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 544c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 545c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottrequest_trans_id_set(struct request *const req, const u16 trans_id) { 546c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->trans_id = trans_id; 547c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *((u16 *) req->request) = htons(trans_id); 548c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 549c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 550c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Called to remove a request from a list and dealloc it. */ 551c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* head is a pointer to the head of the list it should be */ 552c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* removed from or NULL if the request isn't in a list. */ 553c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 554c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottrequest_finished(struct request *const req, struct request **head) { 555c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (head) { 556c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->next == req) { 557c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* only item in the list */ 558c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *head = NULL; 559c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 560c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->next->prev = req->prev; 561c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->prev->next = req->next; 562c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (*head == req) *head = req->next; 563c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 564c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 565c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 566c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Removing timeout for request %lx", 567c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (unsigned long) req); 568c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evtimer_del(&req->timeout_event); 569c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 570c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_request_finished(req); 571c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_requests_inflight--; 572c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 573c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req->request_appended) { 574c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* need to free the request data on it's own */ 575c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(req->request); 576c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 577c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* the request data is appended onto the header */ 578c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* so everything gets free()ed when we: */ 579c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 580c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 581c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(req); 582c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 583c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_requests_pump_waiting_queue(); 584c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 585c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 586c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* This is called when a server returns a funny error code. */ 587c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* We try the request again with another server. */ 588c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 589c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* return: */ 590c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 0 ok */ 591c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 1 failed/reissue is pointless */ 592c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 593c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottrequest_reissue(struct request *req) { 594c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const struct nameserver *const last_ns = req->ns; 595c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* the last nameserver should have been marked as failing */ 596c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* by the caller of this function, therefore pick will try */ 597c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* not to return it */ 598c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->ns = nameserver_pick(); 599c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->ns == last_ns) { 600c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* ... but pick did return it */ 601c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* not a lot of point in trying again with the */ 602c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* same server */ 603c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 604c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 605c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 606c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->reissue_count++; 607c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->tx_count = 0; 608c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->transmit_me = 1; 609c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 610c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 611c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 612c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 613c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* this function looks for space on the inflight queue and promotes */ 614c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* requests from the waiting queue if it can. */ 615c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 616c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_requests_pump_waiting_queue(void) { 617c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (global_requests_inflight < global_max_requests_inflight && 618c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_requests_waiting) { 619c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *req; 620c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* move a request from the waiting queue to the inflight queue */ 621c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(req_waiting_head); 622c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req_waiting_head->next == req_waiting_head) { 623c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* only one item in the queue */ 624c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = req_waiting_head; 625c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req_waiting_head = NULL; 626c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 627c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = req_waiting_head; 628c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->next->prev = req->prev; 629c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->prev->next = req->next; 630c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req_waiting_head = req->next; 631c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 632c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 633c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_requests_waiting--; 634c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_requests_inflight++; 635c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 636c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->ns = nameserver_pick(); 637c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_trans_id_set(req, transaction_id_pick()); 638c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 639c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_request_insert(req, &req_head); 640c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_request_transmit(req); 641c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_transmit(); 642c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 643c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 644c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 645c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 646c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottreply_callback(struct request *const req, u32 ttl, u32 err, struct reply *reply) { 647c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott switch (req->request_type) { 648c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case TYPE_A: 649c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (reply) 650c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_callback(DNS_ERR_NONE, DNS_IPv4_A, 651c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply->data.a.addrcount, ttl, 652c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply->data.a.addresses, 653c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_pointer); 654c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else 655c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_callback(err, 0, 0, 0, NULL, req->user_pointer); 656c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 657c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case TYPE_PTR: 658c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (reply) { 659c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *name = reply->data.ptr.name; 660c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_callback(DNS_ERR_NONE, DNS_PTR, 1, ttl, 661c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott &name, req->user_pointer); 662c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 663c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_callback(err, 0, 0, 0, NULL, 664c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_pointer); 665c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 666c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 667c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case TYPE_AAAA: 668c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (reply) 669c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_callback(DNS_ERR_NONE, DNS_IPv6_AAAA, 670c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply->data.aaaa.addrcount, ttl, 671c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply->data.aaaa.addresses, 672c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_pointer); 673c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else 674c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_callback(err, 0, 0, 0, NULL, req->user_pointer); 675c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 676c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 677c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(0); 678c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 679c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 680c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* this processes a parsed reply packet */ 681c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 682c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottreply_handle(struct request *const req, u16 flags, u32 ttl, struct reply *reply) { 683c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int error; 684c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott static const int error_codes[] = { 685c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott DNS_ERR_FORMAT, DNS_ERR_SERVERFAILED, DNS_ERR_NOTEXIST, 686c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott DNS_ERR_NOTIMPL, DNS_ERR_REFUSED 687c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott }; 688c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 689c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & 0x020f || !reply || !reply->have_answer) { 690c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* there was an error */ 691c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & 0x0200) { 692c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott error = DNS_ERR_TRUNCATED; 693c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 694c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 error_code = (flags & 0x000f) - 1; 695c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (error_code > 4) { 696c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott error = DNS_ERR_UNKNOWN; 697c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 698c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott error = error_codes[error_code]; 699c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 700c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 701c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 702c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott switch(error) { 703c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_NOTIMPL: 704c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_REFUSED: 705c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we regard these errors as marking a bad nameserver */ 706c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->reissue_count < global_max_reissues) { 707c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char msg[64]; 708c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evutil_snprintf(msg, sizeof(msg), 709c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott "Bad response %d (%s)", 710c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott error, evdns_err_to_string(error)); 711c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_failed(req->ns, msg); 712c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!request_reissue(req)) return; 713c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 714c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 715c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_SERVERFAILED: 716c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* rcode 2 (servfailed) sometimes means "we 717c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * are broken" and sometimes (with some binds) 718c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * means "that request was very confusing." 719c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * Treat this as a timeout, not a failure. 720c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott */ 721c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Got a SERVERFAILED from nameserver %s; " 722c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott "will allow the request to time out.", 723c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott debug_ntoa(req->ns->address)); 724c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 725c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott default: 726c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we got a good reply from the nameserver */ 727c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_up(req->ns); 728c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 729c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 730c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->search_state && req->request_type != TYPE_PTR) { 731c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* if we have a list of domains to search in, 732c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * try the next one */ 733c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!search_try_next(req)) { 734c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* a new request was issued so this 735c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * request is finished and */ 736c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* the user callback will be made when 737c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * that request (or a */ 738c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* child of it) finishes. */ 739c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_finished(req, &req_head); 740c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 741c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 742c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 743c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 744c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* all else failed. Pass the failure up */ 745c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply_callback(req, 0, error, NULL); 746c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_finished(req, &req_head); 747c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 748c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* all ok, tell the user */ 749c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply_callback(req, ttl, 0, reply); 750c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_up(req->ns); 751c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_finished(req, &req_head); 752c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 753c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 754c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 755c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 756c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottname_parse(u8 *packet, int length, int *idx, char *name_out, int name_out_len) { 757c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int name_end = -1; 758c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int j = *idx; 759c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int ptr_count = 0; 760c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define GET32(x) do { if (j + 4 > length) goto err; memcpy(&_t32, packet + j, 4); j += 4; x = ntohl(_t32); } while(0) 761c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define GET16(x) do { if (j + 2 > length) goto err; memcpy(&_t, packet + j, 2); j += 2; x = ntohs(_t); } while(0) 762c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define GET8(x) do { if (j >= length) goto err; x = packet[j++]; } while(0) 763c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 764c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *cp = name_out; 765c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *const end = name_out + name_out_len; 766c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 767c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Normally, names are a series of length prefixed strings terminated */ 768c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* with a length of 0 (the lengths are u8's < 63). */ 769c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* However, the length can start with a pair of 1 bits and that */ 770c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* means that the next 14 bits are a pointer within the current */ 771c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* packet. */ 772c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 773c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for(;;) { 774c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u8 label_len; 775c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j >= length) return -1; 776c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET8(label_len); 777c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!label_len) break; 778c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (label_len & 0xc0) { 779c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u8 ptr_low; 780c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET8(ptr_low); 781c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (name_end < 0) name_end = j; 782c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j = (((int)label_len & 0x3f) << 8) + ptr_low; 783c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Make sure that the target offset is in-bounds. */ 784c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j < 0 || j >= length) return -1; 785c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* If we've jumped more times than there are characters in the 786c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * message, we must have a loop. */ 787c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (++ptr_count > length) return -1; 788c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott continue; 789c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 790c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (label_len > 63) return -1; 791c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (cp != name_out) { 792c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (cp + 1 >= end) return -1; 793c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *cp++ = '.'; 794c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 795c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (cp + label_len >= end) return -1; 796c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(cp, packet + j, label_len); 797c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cp += label_len; 798c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += label_len; 799c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 800c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (cp >= end) return -1; 801c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *cp = '\0'; 802c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (name_end < 0) 803c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *idx = j; 804c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else 805c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *idx = name_end; 806c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 807c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott err: 808c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 809c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 810c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 811c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* parses a raw request from a nameserver */ 812c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 813c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottreply_parse(u8 *packet, int length) { 814c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int j = 0, k = 0; /* index into packet */ 815c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 _t; /* used by the macros */ 816c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 _t32; /* used by the macros */ 817c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char tmp_name[256], cmp_name[256]; /* used by the macros */ 818c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 819c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 trans_id, questions, answers, authority, additional, datalength; 820c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 flags = 0; 821c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 ttl, ttl_r = 0xffffffff; 822c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct reply reply; 823c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *req = NULL; 824c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott unsigned int i; 825c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 826c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(trans_id); 827c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(flags); 828c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(questions); 829c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(answers); 830c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(authority); 831c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(additional); 832c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) authority; /* suppress "unused variable" warnings. */ 833c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) additional; /* suppress "unused variable" warnings. */ 834c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 835c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = request_find_from_trans_id(trans_id); 836c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req) return -1; 837c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 838c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memset(&reply, 0, sizeof(reply)); 839c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 840c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* If it's not an answer, it doesn't correspond to any request. */ 841c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(flags & 0x8000)) return -1; /* must be an answer */ 842c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & 0x020f) { 843c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* there was an error */ 844c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; 845c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 846c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* if (!answers) return; */ /* must have an answer of some form */ 847c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 848c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* This macro skips a name in the DNS reply. */ 849c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define SKIP_NAME \ 850c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott do { tmp_name[0] = '\0'; \ 851c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (name_parse(packet, length, &j, tmp_name, sizeof(tmp_name))<0)\ 852c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; \ 853c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while(0) 854c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define TEST_NAME \ 855c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott do { tmp_name[0] = '\0'; \ 856c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cmp_name[0] = '\0'; \ 857c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott k = j; \ 858c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (name_parse(packet, length, &j, tmp_name, sizeof(tmp_name))<0)\ 859c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; \ 860c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (name_parse(req->request, req->request_len, &k, cmp_name, sizeof(cmp_name))<0) \ 861c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; \ 862c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (memcmp(tmp_name, cmp_name, strlen (tmp_name)) != 0) \ 863c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (-1); /* we ignore mismatching names */ \ 864c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while(0) 865c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 866c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply.type = req->request_type; 867c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 868c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* skip over each question in the reply */ 869c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < questions; ++i) { 870c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* the question looks like 871c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * <label:name><u16:type><u16:class> 872c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott */ 873c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott TEST_NAME; 874c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += 4; 875c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j > length) goto err; 876c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 877c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 878c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* now we have the answer section which looks like 879c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * <label:name><u16:type><u16:class><u32:ttl><u16:len><data...> 880c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott */ 881c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 882c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < answers; ++i) { 883c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 type, class; 884c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 885c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott SKIP_NAME; 886c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(type); 887c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(class); 888c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET32(ttl); 889c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(datalength); 890c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 891c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (type == TYPE_A && class == CLASS_INET) { 892c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int addrcount, addrtocopy; 893c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->request_type != TYPE_A) { 894c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += datalength; continue; 895c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 896c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if ((datalength & 3) != 0) /* not an even number of As. */ 897c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; 898c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott addrcount = datalength >> 2; 899c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott addrtocopy = MIN(MAX_ADDRS - reply.data.a.addrcount, (unsigned)addrcount); 900c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 901c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ttl_r = MIN(ttl_r, ttl); 902c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we only bother with the first four addresses. */ 903c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j + 4*addrtocopy > length) goto err; 904c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(&reply.data.a.addresses[reply.data.a.addrcount], 905c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott packet + j, 4*addrtocopy); 906c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += 4*addrtocopy; 907c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply.data.a.addrcount += addrtocopy; 908c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply.have_answer = 1; 909c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (reply.data.a.addrcount == MAX_ADDRS) break; 910c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (type == TYPE_PTR && class == CLASS_INET) { 911c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->request_type != TYPE_PTR) { 912c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += datalength; continue; 913c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 914c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (name_parse(packet, length, &j, reply.data.ptr.name, 915c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott sizeof(reply.data.ptr.name))<0) 916c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; 917c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ttl_r = MIN(ttl_r, ttl); 918c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply.have_answer = 1; 919c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 920c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (type == TYPE_AAAA && class == CLASS_INET) { 921c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int addrcount, addrtocopy; 922c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->request_type != TYPE_AAAA) { 923c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += datalength; continue; 924c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 925c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if ((datalength & 15) != 0) /* not an even number of AAAAs. */ 926c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; 927c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott addrcount = datalength >> 4; /* each address is 16 bytes long */ 928c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott addrtocopy = MIN(MAX_ADDRS - reply.data.aaaa.addrcount, (unsigned)addrcount); 929c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ttl_r = MIN(ttl_r, ttl); 930c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 931c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we only bother with the first four addresses. */ 932c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j + 16*addrtocopy > length) goto err; 933c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(&reply.data.aaaa.addresses[reply.data.aaaa.addrcount], 934c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott packet + j, 16*addrtocopy); 935c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply.data.aaaa.addrcount += addrtocopy; 936c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += 16*addrtocopy; 937c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply.have_answer = 1; 938c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (reply.data.aaaa.addrcount == MAX_ADDRS) break; 939c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 940c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* skip over any other type of resource */ 941c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += datalength; 942c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 943c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 944c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 945c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply_handle(req, flags, ttl_r, &reply); 946c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 947c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott err: 948c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req) 949c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply_handle(req, flags, 0, NULL); 950c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 951c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 952c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 953c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Parse a raw request (packet,length) sent to a nameserver port (port) from */ 954c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* a DNS client (addr,addrlen), and if it's well-formed, call the corresponding */ 955c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* callback. */ 956c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 957c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottrequest_parse(u8 *packet, int length, struct evdns_server_port *port, struct sockaddr *addr, socklen_t addrlen) 958c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 959c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int j = 0; /* index into packet */ 960c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 _t; /* used by the macros */ 961c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char tmp_name[256]; /* used by the macros */ 962c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 963c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int i; 964c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 trans_id, flags, questions, answers, authority, additional; 965c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_request *server_req = NULL; 966c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 967c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Get the header fields */ 968c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(trans_id); 969c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(flags); 970c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(questions); 971c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(answers); 972c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(authority); 973c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(additional); 974c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 975c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & 0x8000) return -1; /* Must not be an answer. */ 976c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott flags &= 0x0110; /* Only RD and CD get preserved. */ 977c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 978c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_req = malloc(sizeof(struct server_request)); 979c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_req == NULL) return -1; 980c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memset(server_req, 0, sizeof(struct server_request)); 981c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 982c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_req->trans_id = trans_id; 983c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(&server_req->addr, addr, addrlen); 984c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_req->addrlen = addrlen; 985c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 986c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_req->base.flags = flags; 987c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_req->base.nquestions = 0; 988c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_req->base.questions = malloc(sizeof(struct evdns_server_question *) * questions); 989c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_req->base.questions == NULL) 990c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; 991c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 992c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < questions; ++i) { 993c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 type, class; 994c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct evdns_server_question *q; 995c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int namelen; 996c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (name_parse(packet, length, &j, tmp_name, sizeof(tmp_name))<0) 997c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; 998c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(type); 999c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GET16(class); 1000c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott namelen = strlen(tmp_name); 1001c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott q = malloc(sizeof(struct evdns_server_question) + namelen); 1002c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!q) 1003c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err; 1004c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott q->type = type; 1005c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott q->dns_question_class = class; 1006c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(q->name, tmp_name, namelen+1); 1007c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_req->base.questions[server_req->base.nquestions++] = q; 1008c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1009c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1010c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Ignore answers, authority, and additional. */ 1011c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1012c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_req->port = port; 1013c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->refcnt++; 1014c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1015c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Only standard queries are supported. */ 1016c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & 0x7800) { 1017c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_server_request_respond(&(server_req->base), DNS_ERR_NOTIMPL); 1018c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 1019c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1020c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1021c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->user_callback(&(server_req->base), port->user_data); 1022c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1023c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 1024c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scotterr: 1025c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_req) { 1026c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_req->base.questions) { 1027c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < server_req->base.nquestions; ++i) 1028c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(server_req->base.questions[i]); 1029c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(server_req->base.questions); 1030c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1031c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(server_req); 1032c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1033c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 1034c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1035c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#undef SKIP_NAME 1036c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#undef GET32 1037c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#undef GET16 1038c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#undef GET8 1039c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1040c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1041c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic u16 1042c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottdefault_transaction_id_fn(void) 1043c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1044c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 trans_id; 1045c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_CPU_CLOCK_FOR_ID 1046c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct timespec ts; 1047c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott static int clkid = -1; 1048c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (clkid == -1) { 1049c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott clkid = CLOCK_REALTIME; 1050c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef CLOCK_MONOTONIC 1051c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (clock_gettime(CLOCK_MONOTONIC, &ts) != -1) 1052c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott clkid = CLOCK_MONOTONIC; 1053c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 1054c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1055c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (clock_gettime(clkid, &ts) == -1) 1056c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott event_err(1, "clock_gettime"); 1057c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott trans_id = ts.tv_nsec & 0xffff; 1058c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 1059c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1060c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_FTIME_FOR_ID 1061c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct _timeb tb; 1062c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott _ftime(&tb); 1063c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott trans_id = tb.millitm & 0xffff; 1064c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 1065c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1066c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_GETTIMEOFDAY_FOR_ID 1067c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct timeval tv; 1068c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evutil_gettimeofday(&tv, NULL); 1069c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott trans_id = tv.tv_usec & 0xffff; 1070c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 1071c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1072c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef DNS_USE_OPENSSL_FOR_ID 1073c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (RAND_pseudo_bytes((u8 *) &trans_id, 2) == -1) { 1074c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* in the case that the RAND call fails we back */ 1075c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* down to using gettimeofday. */ 1076c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* 1077c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct timeval tv; 1078c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evutil_gettimeofday(&tv, NULL); 1079c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott trans_id = tv.tv_usec & 0xffff; 1080c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott */ 1081c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott abort(); 1082c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1083c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 1084c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return trans_id; 1085c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1086c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1087c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic ev_uint16_t (*trans_id_function)(void) = default_transaction_id_fn; 1088c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1089c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 1090c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_set_transaction_id_fn(ev_uint16_t (*fn)(void)) 1091c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1092c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (fn) 1093c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott trans_id_function = fn; 1094c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else 1095c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott trans_id_function = default_transaction_id_fn; 1096c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1097c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1098c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Try to choose a strong transaction id which isn't already in flight */ 1099c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic u16 1100c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scotttransaction_id_pick(void) { 1101c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (;;) { 1102c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const struct request *req = req_head, *started_at; 1103c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 trans_id = trans_id_function(); 1104c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1105c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (trans_id == 0xffff) continue; 1106c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* now check to see if that id is already inflight */ 1107c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = started_at = req_head; 1108c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req) { 1109c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott do { 1110c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->trans_id == trans_id) break; 1111c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = req->next; 1112c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while (req != started_at); 1113c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1114c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we didn't find it, so this is a good id */ 1115c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req == started_at) return trans_id; 1116c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1117c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1118c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1119c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* choose a namesever to use. This function will try to ignore */ 1120c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* nameservers which we think are down and load balance across the rest */ 1121c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* by updating the server_head global each time. */ 1122c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct nameserver * 1123c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_pick(void) { 1124c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *started_at = server_head, *picked; 1125c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!server_head) return NULL; 1126c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1127c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* if we don't have any good nameservers then there's no */ 1128c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* point in trying to find one. */ 1129c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!global_good_nameservers) { 1130c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_head = server_head->next; 1131c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return server_head; 1132c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1133c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1134c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* remember that nameservers are in a circular list */ 1135c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (;;) { 1136c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_head->state) { 1137c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we think this server is currently good */ 1138c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott picked = server_head; 1139c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_head = server_head->next; 1140c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return picked; 1141c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1142c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1143c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_head = server_head->next; 1144c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_head == started_at) { 1145c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* all the nameservers seem to be down */ 1146c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* so we just return this one and hope for the */ 1147c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* best */ 1148c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(global_good_nameservers == 0); 1149c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott picked = server_head; 1150c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_head = server_head->next; 1151c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return picked; 1152c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1153c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1154c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1155c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1156c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 1157c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottaddress_is_correct(struct nameserver *ns, struct sockaddr *sa, socklen_t slen) 1158c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1159c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct sockaddr_in *sin = (struct sockaddr_in*) sa; 1160c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (sa->sa_family != AF_INET || slen != sizeof(struct sockaddr_in)) 1161c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 1162c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (sin->sin_addr.s_addr != ns->address) 1163c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 1164c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 1165c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1166c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1167c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* this is called when a namesever socket is ready for reading */ 1168c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1169c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_read(struct nameserver *ns) { 1170c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u8 packet[1500]; 1171c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct sockaddr_storage ss; 1172c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott socklen_t addrlen = sizeof(ss); 1173c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1174c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (;;) { 1175c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int r = recvfrom(ns->socket, packet, sizeof(packet), 0, 1176c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (struct sockaddr*)&ss, &addrlen); 1177c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r < 0) { 1178c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int err = last_error(ns->socket); 1179c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (error_is_eagain(err)) return; 1180c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_failed(ns, strerror(err)); 1181c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 1182c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1183c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!address_is_correct(ns, (struct sockaddr*)&ss, addrlen)) { 1184c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Address mismatch on received " 1185c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott "DNS packet."); 1186c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 1187c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1188c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->timedout = 0; 1189c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply_parse(packet, r); 1190c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1191c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1192c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1193c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Read a packet from a DNS client on a server port s, parse it, and */ 1194c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* act accordingly. */ 1195c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1196c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottserver_port_read(struct evdns_server_port *s) { 1197c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u8 packet[1500]; 1198c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct sockaddr_storage addr; 1199c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott socklen_t addrlen; 1200c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int r; 1201c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1202c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (;;) { 1203c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott addrlen = sizeof(struct sockaddr_storage); 1204c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = recvfrom(s->socket, packet, sizeof(packet), 0, 1205c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (struct sockaddr*) &addr, &addrlen); 1206c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r < 0) { 1207c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int err = last_error(s->socket); 1208c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (error_is_eagain(err)) return; 1209c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Error %s (%d) while reading request.", 1210c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott strerror(err), err); 1211c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 1212c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1213c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_parse(packet, r, s, (struct sockaddr*) &addr, addrlen); 1214c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1215c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1216c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1217c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Try to write all pending replies on a given DNS server port. */ 1218c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1219c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottserver_port_flush(struct evdns_server_port *port) 1220c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1221c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (port->pending_replies) { 1222c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_request *req = port->pending_replies; 1223c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int r = sendto(port->socket, req->response, req->response_len, 0, 1224c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (struct sockaddr*) &req->addr, req->addrlen); 1225c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r < 0) { 1226c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int err = last_error(port->socket); 1227c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (error_is_eagain(err)) 1228c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 1229c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Error %s (%d) while writing response to port; dropping", strerror(err), err); 1230c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1231c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_request_free(req)) { 1232c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we released the last reference to req->port. */ 1233c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 1234c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1235c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1236c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1237c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* We have no more pending requests; stop listening for 'writeable' events. */ 1238c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) event_del(&port->event); 1239c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott event_set(&port->event, port->socket, EV_READ | EV_PERSIST, 1240c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_port_ready_callback, port); 1241c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (event_add(&port->event, NULL) < 0) { 1242c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Error from libevent when adding event for DNS server."); 1243c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* ???? Do more? */ 1244c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1245c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1246c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1247c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* set if we are waiting for the ability to write to this server. */ 1248c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* if waiting is true then we ask libevent for EV_WRITE events, otherwise */ 1249c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* we stop these events. */ 1250c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1251c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_write_waiting(struct nameserver *ns, char waiting) { 1252c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (ns->write_waiting == waiting) return; 1253c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1254c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->write_waiting = waiting; 1255c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) event_del(&ns->event); 1256c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott event_set(&ns->event, ns->socket, EV_READ | (waiting ? EV_WRITE : 0) | EV_PERSIST, 1257c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_ready_callback, ns); 1258c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (event_add(&ns->event, NULL) < 0) { 1259c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Error from libevent when adding event for %s", 1260c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott debug_ntoa(ns->address)); 1261c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* ???? Do more? */ 1262c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1263c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1264c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1265c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* a callback function. Called by libevent when the kernel says that */ 1266c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* a nameserver socket is ready for writing or reading */ 1267c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1268c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_ready_callback(int fd, short events, void *arg) { 1269c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *ns = (struct nameserver *) arg; 1270c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void)fd; 1271c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1272c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (events & EV_WRITE) { 1273c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->choked = 0; 1274c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!evdns_transmit()) { 1275c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_write_waiting(ns, 0); 1276c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1277c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1278c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (events & EV_READ) { 1279c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_read(ns); 1280c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1281c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1282c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1283c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* a callback function. Called by libevent when the kernel says that */ 1284c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* a server socket is ready for writing or reading. */ 1285c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1286c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottserver_port_ready_callback(int fd, short events, void *arg) { 1287c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct evdns_server_port *port = (struct evdns_server_port *) arg; 1288c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) fd; 1289c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1290c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (events & EV_WRITE) { 1291c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->choked = 0; 1292c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_port_flush(port); 1293c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1294c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (events & EV_READ) { 1295c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_port_read(port); 1296c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1297c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1298c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1299c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* This is an inefficient representation; only use it via the dnslabel_table_* 1300c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * functions, so that is can be safely replaced with something smarter later. */ 1301c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define MAX_LABELS 128 1302c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Structures used to implement name compression */ 1303c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct dnslabel_entry { char *v; off_t pos; }; 1304c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct dnslabel_table { 1305c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int n_labels; /* number of current entries */ 1306c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* map from name to position in message */ 1307c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct dnslabel_entry labels[MAX_LABELS]; 1308c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 1309c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1310c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Initialize dnslabel_table. */ 1311c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1312c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottdnslabel_table_init(struct dnslabel_table *table) 1313c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1314c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott table->n_labels = 0; 1315c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1316c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1317c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Free all storage held by table, but not the table itself. */ 1318c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1319c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottdnslabel_clear(struct dnslabel_table *table) 1320c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1321c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int i; 1322c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < table->n_labels; ++i) 1323c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(table->labels[i].v); 1324c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott table->n_labels = 0; 1325c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1326c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1327c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* return the position of the label in the current message, or -1 if the label */ 1328c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* hasn't been used yet. */ 1329c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 1330c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottdnslabel_table_get_pos(const struct dnslabel_table *table, const char *label) 1331c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1332c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int i; 1333c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < table->n_labels; ++i) { 1334c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!strcmp(label, table->labels[i].v)) 1335c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return table->labels[i].pos; 1336c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1337c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 1338c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1339c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1340c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* remember that we've used the label at position pos */ 1341c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 1342c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottdnslabel_table_add(struct dnslabel_table *table, const char *label, off_t pos) 1343c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1344c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *v; 1345c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int p; 1346c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (table->n_labels == MAX_LABELS) 1347c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (-1); 1348c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott v = strdup(label); 1349c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (v == NULL) 1350c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (-1); 1351c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott p = table->n_labels++; 1352c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott table->labels[p].v = v; 1353c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott table->labels[p].pos = pos; 1354c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1355c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (0); 1356c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1357c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1358c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Converts a string to a length-prefixed set of DNS labels, starting */ 1359c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* at buf[j]. name and buf must not overlap. name_len should be the length */ 1360c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* of name. table is optional, and is used for compression. */ 1361c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 1362c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Input: abc.def */ 1363c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Output: <3>abc<3>def<0> */ 1364c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 1365c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Returns the first index after the encoded name, or negative on error. */ 1366c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* -1 label was > 63 bytes */ 1367c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* -2 name too long to fit in buffer. */ 1368c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 1369c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic off_t 1370c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottdnsname_to_labels(u8 *const buf, size_t buf_len, off_t j, 1371c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *name, const int name_len, 1372c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct dnslabel_table *table) { 1373c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *end = name + name_len; 1374c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int ref = 0; 1375c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 _t; 1376c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1377c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define APPEND16(x) do { \ 1378c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j + 2 > (off_t)buf_len) \ 1379c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto overflow; \ 1380c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott _t = htons(x); \ 1381c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(buf + j, &_t, 2); \ 1382c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += 2; \ 1383c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while (0) 1384c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define APPEND32(x) do { \ 1385c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j + 4 > (off_t)buf_len) \ 1386c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto overflow; \ 1387c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott _t32 = htonl(x); \ 1388c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(buf + j, &_t32, 4); \ 1389c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += 4; \ 1390c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while (0) 1391c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1392c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (name_len > 255) return -2; 1393c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1394c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (;;) { 1395c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *const start = name; 1396c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (table && (ref = dnslabel_table_get_pos(table, name)) >= 0) { 1397c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(ref | 0xc000); 1398c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return j; 1399c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1400c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott name = strchr(name, '.'); 1401c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!name) { 1402c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const unsigned int label_len = end - start; 1403c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (label_len > 63) return -1; 1404c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if ((size_t)(j+label_len+1) > buf_len) return -2; 1405c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (table) dnslabel_table_add(table, start, j); 1406c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf[j++] = label_len; 1407c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1408c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(buf + j, start, end - start); 1409c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += end - start; 1410c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 1411c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 1412c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* append length of the label. */ 1413c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const unsigned int label_len = name - start; 1414c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (label_len > 63) return -1; 1415c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if ((size_t)(j+label_len+1) > buf_len) return -2; 1416c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (table) dnslabel_table_add(table, start, j); 1417c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf[j++] = label_len; 1418c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1419c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(buf + j, start, name - start); 1420c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += name - start; 1421c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* hop over the '.' */ 1422c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott name++; 1423c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1424c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1425c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1426c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* the labels must be terminated by a 0. */ 1427c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* It's possible that the name ended in a . */ 1428c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* in which case the zero is already there */ 1429c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!j || buf[j-1]) buf[j++] = 0; 1430c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return j; 1431c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott overflow: 1432c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (-2); 1433c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1434c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1435c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Finds the length of a dns request for a DNS name of the given */ 1436c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* length. The actual request may be smaller than the value returned */ 1437c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* here */ 1438c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 1439c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_request_len(const int name_len) { 1440c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 96 + /* length of the DNS standard header */ 1441c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott name_len + 2 + 1442c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 4; /* space for the resource type */ 1443c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1444c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1445c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* build a dns request packet into buf. buf should be at least as long */ 1446c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* as evdns_request_len told you it should be. */ 1447c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 1448c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Returns the amount of space used. Negative on error. */ 1449c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 1450c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_request_data_build(const char *const name, const int name_len, 1451c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const u16 trans_id, const u16 type, const u16 class, 1452c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u8 *const buf, size_t buf_len) { 1453c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott off_t j = 0; /* current offset into buf */ 1454c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 _t; /* used by the macros */ 1455c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1456c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(trans_id); 1457c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(0x0100); /* standard query, recusion needed */ 1458c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(1); /* one question */ 1459c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(0); /* no answers */ 1460c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(0); /* no authority */ 1461c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(0); /* no additional */ 1462c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1463c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j = dnsname_to_labels(buf, buf_len, j, name, name_len, NULL); 1464c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j < 0) { 1465c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (int)j; 1466c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1467c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1468c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(type); 1469c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(class); 1470c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1471c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (int)j; 1472c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott overflow: 1473c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (-1); 1474c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1475c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1476c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1477c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct evdns_server_port * 1478c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_add_server_port(int socket, int is_tcp, evdns_request_callback_fn_type cb, void *user_data) 1479c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1480c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct evdns_server_port *port; 1481c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(port = malloc(sizeof(struct evdns_server_port)))) 1482c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return NULL; 1483c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memset(port, 0, sizeof(struct evdns_server_port)); 1484c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1485c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(!is_tcp); /* TCP sockets not yet implemented */ 1486c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->socket = socket; 1487c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->refcnt = 1; 1488c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->choked = 0; 1489c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->closing = 0; 1490c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->user_callback = cb; 1491c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->user_data = user_data; 1492c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->pending_replies = NULL; 1493c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1494c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott event_set(&port->event, port->socket, EV_READ | EV_PERSIST, 1495c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_port_ready_callback, port); 1496c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott event_add(&port->event, NULL); /* check return. */ 1497c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return port; 1498c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1499c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1500c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1501c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 1502c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_close_server_port(struct evdns_server_port *port) 1503c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1504c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (--port->refcnt == 0) 1505c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_port_free(port); 1506c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->closing = 1; 1507c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1508c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1509c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1510c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 1511c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_request_add_reply(struct evdns_server_request *_req, int section, const char *name, int type, int class, int ttl, int datalen, int is_name, const char *data) 1512c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1513c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_request *req = TO_SERVER_REQUEST(_req); 1514c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_reply_item **itemp, *item; 1515c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int *countp; 1516c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1517c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->response) /* have we already answered? */ 1518c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (-1); 1519c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1520c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott switch (section) { 1521c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case EVDNS_ANSWER_SECTION: 1522c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott itemp = &req->answer; 1523c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott countp = &req->n_answer; 1524c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 1525c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case EVDNS_AUTHORITY_SECTION: 1526c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott itemp = &req->authority; 1527c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott countp = &req->n_authority; 1528c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 1529c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case EVDNS_ADDITIONAL_SECTION: 1530c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott itemp = &req->additional; 1531c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott countp = &req->n_additional; 1532c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 1533c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott default: 1534c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (-1); 1535c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1536c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (*itemp) { 1537c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott itemp = &((*itemp)->next); 1538c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1539c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item = malloc(sizeof(struct server_reply_item)); 1540c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!item) 1541c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 1542c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item->next = NULL; 1543c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(item->name = strdup(name))) { 1544c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(item); 1545c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 1546c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1547c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item->type = type; 1548c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item->dns_question_class = class; 1549c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item->ttl = ttl; 1550c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item->is_name = is_name != 0; 1551c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item->datalen = 0; 1552c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item->data = NULL; 1553c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (data) { 1554c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (item->is_name) { 1555c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(item->data = strdup(data))) { 1556c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(item->name); 1557c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(item); 1558c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 1559c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1560c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item->datalen = (u16)-1; 1561c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 1562c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(item->data = malloc(datalen))) { 1563c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(item->name); 1564c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(item); 1565c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 1566c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1567c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item->datalen = datalen; 1568c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(item->data, data, datalen); 1569c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1570c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1571c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1572c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *itemp = item; 1573c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ++(*countp); 1574c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 1575c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1576c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1577c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1578c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 1579c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_request_add_a_reply(struct evdns_server_request *req, const char *name, int n, void *addrs, int ttl) 1580c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1581c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return evdns_server_request_add_reply( 1582c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req, EVDNS_ANSWER_SECTION, name, TYPE_A, CLASS_INET, 1583c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ttl, n*4, 0, addrs); 1584c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1585c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1586c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1587c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 1588c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_request_add_aaaa_reply(struct evdns_server_request *req, const char *name, int n, void *addrs, int ttl) 1589c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1590c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return evdns_server_request_add_reply( 1591c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req, EVDNS_ANSWER_SECTION, name, TYPE_AAAA, CLASS_INET, 1592c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ttl, n*16, 0, addrs); 1593c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1594c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1595c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1596c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 1597c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_request_add_ptr_reply(struct evdns_server_request *req, struct in_addr *in, const char *inaddr_name, const char *hostname, int ttl) 1598c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1599c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 a; 1600c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char buf[32]; 1601c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(in || inaddr_name); 1602c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(!(in && inaddr_name)); 1603c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (in) { 1604c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott a = ntohl(in->s_addr); 1605c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evutil_snprintf(buf, sizeof(buf), "%d.%d.%d.%d.in-addr.arpa", 1606c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a )&0xff), 1607c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a>>8 )&0xff), 1608c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a>>16)&0xff), 1609c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a>>24)&0xff)); 1610c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott inaddr_name = buf; 1611c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1612c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return evdns_server_request_add_reply( 1613c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req, EVDNS_ANSWER_SECTION, inaddr_name, TYPE_PTR, CLASS_INET, 1614c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ttl, -1, 1, hostname); 1615c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1616c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1617c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1618c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 1619c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_request_add_cname_reply(struct evdns_server_request *req, const char *name, const char *cname, int ttl) 1620c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1621c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return evdns_server_request_add_reply( 1622c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req, EVDNS_ANSWER_SECTION, name, TYPE_CNAME, CLASS_INET, 1623c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ttl, -1, 1, cname); 1624c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1625c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1626c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1627c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 1628c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_request_format_response(struct server_request *req, int err) 1629c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1630c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott unsigned char buf[1500]; 1631c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott size_t buf_len = sizeof(buf); 1632c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott off_t j = 0, r; 1633c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 _t; 1634c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 _t32; 1635c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int i; 1636c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u16 flags; 1637c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct dnslabel_table table; 1638c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1639c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (err < 0 || err > 15) return -1; 1640c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1641c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Set response bit and error code; copy OPCODE and RD fields from 1642c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * question; copy RA and AA if set by caller. */ 1643c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott flags = req->base.flags; 1644c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott flags |= (0x8000 | err); 1645c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1646c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott dnslabel_table_init(&table); 1647c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(req->trans_id); 1648c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(flags); 1649c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(req->base.nquestions); 1650c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(req->n_answer); 1651c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(req->n_authority); 1652c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(req->n_additional); 1653c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1654c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Add questions. */ 1655c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i=0; i < req->base.nquestions; ++i) { 1656c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *s = req->base.questions[i]->name; 1657c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j = dnsname_to_labels(buf, buf_len, j, s, strlen(s), &table); 1658c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j < 0) { 1659c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott dnslabel_clear(&table); 1660c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (int) j; 1661c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1662c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(req->base.questions[i]->type); 1663c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(req->base.questions[i]->dns_question_class); 1664c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1665c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1666c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Add answer, authority, and additional sections. */ 1667c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i=0; i<3; ++i) { 1668c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_reply_item *item; 1669c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (i==0) 1670c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item = req->answer; 1671c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else if (i==1) 1672c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item = req->authority; 1673c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else 1674c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item = req->additional; 1675c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (item) { 1676c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = dnsname_to_labels(buf, buf_len, j, item->name, strlen(item->name), &table); 1677c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r < 0) 1678c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto overflow; 1679c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j = r; 1680c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1681c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(item->type); 1682c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(item->dns_question_class); 1683c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND32(item->ttl); 1684c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (item->is_name) { 1685c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott off_t len_idx = j, name_start; 1686c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += 2; 1687c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott name_start = j; 1688c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = dnsname_to_labels(buf, buf_len, j, item->data, strlen(item->data), &table); 1689c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r < 0) 1690c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto overflow; 1691c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j = r; 1692c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott _t = htons( (short) (j-name_start) ); 1693c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(buf+len_idx, &_t, 2); 1694c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 1695c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott APPEND16(item->datalen); 1696c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j+item->datalen > (off_t)buf_len) 1697c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto overflow; 1698c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(buf+j, item->data, item->datalen); 1699c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j += item->datalen; 1700c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1701c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott item = item->next; 1702c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1703c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1704c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1705c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (j > 512) { 1706c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottoverflow: 1707c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott j = 512; 1708c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf[2] |= 0x02; /* set the truncated bit. */ 1709c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1710c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1711c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->response_len = j; 1712c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1713c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(req->response = malloc(req->response_len))) { 1714c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_request_free_answers(req); 1715c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott dnslabel_clear(&table); 1716c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (-1); 1717c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1718c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(req->response, buf, req->response_len); 1719c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_request_free_answers(req); 1720c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott dnslabel_clear(&table); 1721c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (0); 1722c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1723c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1724c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1725c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 1726c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_request_respond(struct evdns_server_request *_req, int err) 1727c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1728c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_request *req = TO_SERVER_REQUEST(_req); 1729c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct evdns_server_port *port = req->port; 1730c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int r; 1731c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req->response) { 1732c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if ((r = evdns_server_request_format_response(req, err))<0) 1733c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return r; 1734c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1735c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1736c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = sendto(port->socket, req->response, req->response_len, 0, 1737c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (struct sockaddr*) &req->addr, req->addrlen); 1738c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r<0) { 1739c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int sock_err = last_error(port->socket); 1740c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (! error_is_eagain(sock_err)) 1741c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 1742c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1743c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (port->pending_replies) { 1744c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->prev_pending = port->pending_replies->prev_pending; 1745c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->next_pending = port->pending_replies; 1746c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->prev_pending->next_pending = 1747c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->next_pending->prev_pending = req; 1748c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 1749c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->prev_pending = req->next_pending = req; 1750c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->pending_replies = req; 1751c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->choked = 1; 1752c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1753c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) event_del(&port->event); 1754c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott event_set(&port->event, port->socket, (port->closing?0:EV_READ) | EV_WRITE | EV_PERSIST, server_port_ready_callback, port); 1755c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1756c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (event_add(&port->event, NULL) < 0) { 1757c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Error from libevent when adding event for DNS server"); 1758c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1759c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1760c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1761c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1762c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 1763c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1764c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_request_free(req)) 1765c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 1766c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1767c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (port->pending_replies) 1768c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_port_flush(port); 1769c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1770c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 1771c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1772c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1773c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Free all storage held by RRs in req. */ 1774c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1775c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottserver_request_free_answers(struct server_request *req) 1776c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1777c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_reply_item *victim, *next, **list; 1778c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int i; 1779c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < 3; ++i) { 1780c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (i==0) 1781c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott list = &req->answer; 1782c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else if (i==1) 1783c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott list = &req->authority; 1784c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else 1785c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott list = &req->additional; 1786c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1787c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott victim = *list; 1788c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (victim) { 1789c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott next = victim->next; 1790c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(victim->name); 1791c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (victim->data) 1792c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(victim->data); 1793c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(victim); 1794c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott victim = next; 1795c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1796c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *list = NULL; 1797c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1798c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1799c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1800c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Free all storage held by req, and remove links to it. */ 1801c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* return true iff we just wound up freeing the server_port. */ 1802c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 1803c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottserver_request_free(struct server_request *req) 1804c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1805c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int i, rc=1; 1806c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->base.questions) { 1807c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < req->base.nquestions; ++i) 1808c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(req->base.questions[i]); 1809c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(req->base.questions); 1810c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1811c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1812c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->port) { 1813c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->port->pending_replies == req) { 1814c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->next_pending) 1815c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->port->pending_replies = req->next_pending; 1816c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else 1817c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->port->pending_replies = NULL; 1818c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1819c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott rc = --req->port->refcnt; 1820c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1821c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1822c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->response) { 1823c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(req->response); 1824c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1825c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1826c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_request_free_answers(req); 1827c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1828c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->next_pending && req->next_pending != req) { 1829c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->next_pending->prev_pending = req->prev_pending; 1830c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->prev_pending->next_pending = req->next_pending; 1831c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1832c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1833c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (rc == 0) { 1834c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_port_free(req->port); 1835c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(req); 1836c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (1); 1837c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1838c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(req); 1839c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (0); 1840c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1841c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1842c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Free all storage held by an evdns_server_port. Only called when */ 1843c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1844c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottserver_port_free(struct evdns_server_port *port) 1845c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1846c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(port); 1847c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(!port->refcnt); 1848c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(!port->pending_replies); 1849c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (port->socket > 0) { 1850c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott CLOSE_SOCKET(port->socket); 1851c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port->socket = -1; 1852c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1853c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) event_del(&port->event); 1854c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* XXXX actually free the port? -NM */ 1855c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1856c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1857c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1858c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 1859c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_request_drop(struct evdns_server_request *_req) 1860c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1861c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_request *req = TO_SERVER_REQUEST(_req); 1862c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_request_free(req); 1863c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 1864c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1865c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1866c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 1867c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 1868c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_request_get_requesting_addr(struct evdns_server_request *_req, struct sockaddr *sa, int addr_len) 1869c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 1870c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct server_request *req = TO_SERVER_REQUEST(_req); 1871c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (addr_len < (int)req->addrlen) 1872c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 1873c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(sa, &(req->addr), req->addrlen); 1874c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return req->addrlen; 1875c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1876c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1877c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#undef APPEND16 1878c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#undef APPEND32 1879c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1880c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* this is a libevent callback function which is called when a request */ 1881c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* has timed out. */ 1882c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1883c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_request_timeout_callback(int fd, short events, void *arg) { 1884c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *const req = (struct request *) arg; 1885c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) fd; 1886c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) events; 1887c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1888c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Request %lx timed out", (unsigned long) arg); 1889c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1890c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->ns->timedout++; 1891c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->ns->timedout > global_max_nameserver_timeout) { 1892c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->ns->timedout = 0; 1893c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_failed(req->ns, "request timed out."); 1894c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1895c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1896c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) evtimer_del(&req->timeout_event); 1897c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->tx_count >= global_max_retransmits) { 1898c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* this request has failed */ 1899c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply_callback(req, 0, DNS_ERR_TIMEOUT, NULL); 1900c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_finished(req, &req_head); 1901c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 1902c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* retransmit it */ 1903c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_request_transmit(req); 1904c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1905c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1906c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1907c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* try to send a request to a given server. */ 1908c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 1909c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* return: */ 1910c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 0 ok */ 1911c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 1 temporary failure */ 1912c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 2 other failure */ 1913c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 1914c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_request_transmit_to(struct request *req, struct nameserver *server) { 1915c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct sockaddr_in sin; 1916c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int r; 1917c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memset(&sin, 0, sizeof(sin)); 1918c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott sin.sin_addr.s_addr = req->ns->address; 1919c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott sin.sin_port = req->ns->port; 1920c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott sin.sin_family = AF_INET; 1921c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1922c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = sendto(server->socket, req->request, req->request_len, 0, 1923c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (struct sockaddr*)&sin, sizeof(sin)); 1924c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r < 0) { 1925c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int err = last_error(server->socket); 1926c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (error_is_eagain(err)) return 1; 1927c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_failed(req->ns, strerror(err)); 1928c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 2; 1929c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (r != (int)req->request_len) { 1930c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; /* short write */ 1931c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 1932c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 1933c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1934c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1935c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1936c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* try to send a request, updating the fields of the request */ 1937c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* as needed */ 1938c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 1939c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* return: */ 1940c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 0 ok */ 1941c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 1 failed */ 1942c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 1943c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_request_transmit(struct request *req) { 1944c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int retcode = 0, r; 1945c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1946c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* if we fail to send this packet then this flag marks it */ 1947c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* for evdns_transmit */ 1948c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->transmit_me = 1; 1949c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->trans_id == 0xffff) abort(); 1950c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1951c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->ns->choked) { 1952c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* don't bother trying to write to a socket */ 1953c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* which we have had EAGAIN from */ 1954c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 1955c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1956c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1957c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = evdns_request_transmit_to(req, req->ns); 1958c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott switch (r) { 1959c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case 1: 1960c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* temp failure */ 1961c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->ns->choked = 1; 1962c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_write_waiting(req->ns, 1); 1963c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 1964c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case 2: 1965c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* failed in some other way */ 1966c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott retcode = 1; 1967c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* fall through */ 1968c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott default: 1969c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* all ok */ 1970c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, 1971c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott "Setting timeout for request %lx", (unsigned long) req); 1972c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (evtimer_add(&req->timeout_event, &global_timeout) < 0) { 1973c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, 1974c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott "Error from libevent when adding timer for request %lx", 1975c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (unsigned long) req); 1976c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* ???? Do more? */ 1977c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1978c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->tx_count++; 1979c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->transmit_me = 0; 1980c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return retcode; 1981c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 1982c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1983c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1984c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1985c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_probe_callback(int result, char type, int count, int ttl, void *addresses, void *arg) { 1986c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *const ns = (struct nameserver *) arg; 1987c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) type; 1988c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) count; 1989c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) ttl; 1990c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) addresses; 1991c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1992c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (result == DNS_ERR_NONE || result == DNS_ERR_NOTEXIST) { 1993c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* this is a good reply */ 1994c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott nameserver_up(ns); 1995c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else nameserver_probe_failed(ns); 1996c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 1997c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1998c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 1999c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnameserver_send_probe(struct nameserver *const ns) { 2000c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *req; 2001c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* here we need to send a probe to a given nameserver */ 2002c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* in the hope that it is up now. */ 2003c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2004c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Sending probe to %s", debug_ntoa(ns->address)); 2005c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2006c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = request_new(TYPE_A, "www.google.com", DNS_QUERY_NO_SEARCH, nameserver_probe_callback, ns); 2007c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req) return; 2008c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we force this into the inflight queue no matter what */ 2009c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_trans_id_set(req, transaction_id_pick()); 2010c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->ns = ns; 2011c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_submit(req); 2012c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2013c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2014c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* returns: */ 2015c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 0 didn't try to transmit anything */ 2016c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 1 tried to transmit something */ 2017c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2018c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_transmit(void) { 2019c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char did_try_to_transmit = 0; 2020c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2021c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req_head) { 2022c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *const started_at = req_head, *req = req_head; 2023c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* first transmit all the requests which are currently waiting */ 2024c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott do { 2025c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->transmit_me) { 2026c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott did_try_to_transmit = 1; 2027c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_request_transmit(req); 2028c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2029c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2030c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = req->next; 2031c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while (req != started_at); 2032c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2033c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2034c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return did_try_to_transmit; 2035c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2036c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2037c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2038c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 2039c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_count_nameservers(void) 2040c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 2041c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const struct nameserver *server = server_head; 2042c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int n = 0; 2043c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!server) 2044c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2045c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott do { 2046c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ++n; 2047c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server = server->next; 2048c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while (server != server_head); 2049c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return n; 2050c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2051c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2052c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2053c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 2054c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_clear_nameservers_and_suspend(void) 2055c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 2056c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *server = server_head, *started_at = server_head; 2057c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *req = req_head, *req_started_at = req_head; 2058c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2059c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!server) 2060c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2061c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (1) { 2062c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *next = server->next; 2063c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) event_del(&server->event); 2064c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (evtimer_initialized(&server->timeout_event)) 2065c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) evtimer_del(&server->timeout_event); 2066c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server->socket >= 0) 2067c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott CLOSE_SOCKET(server->socket); 2068c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(server); 2069c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (next == started_at) 2070c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 2071c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server = next; 2072c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2073c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_head = NULL; 2074c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_good_nameservers = 0; 2075c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2076c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (req) { 2077c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *next = req->next; 2078c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->tx_count = req->reissue_count = 0; 2079c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->ns = NULL; 2080c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* ???? What to do about searches? */ 2081c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) evtimer_del(&req->timeout_event); 2082c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->trans_id = 0; 2083c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->transmit_me = 0; 2084c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2085c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_requests_waiting++; 2086c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_request_insert(req, &req_waiting_head); 2087c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* We want to insert these suspended elements at the front of 2088c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * the waiting queue, since they were pending before any of 2089c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * the waiting entries were added. This is a circular list, 2090c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * so we can just shift the start back by one.*/ 2091c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req_waiting_head = req_waiting_head->prev; 2092c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2093c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (next == req_started_at) 2094c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 2095c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = next; 2096c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2097c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req_head = NULL; 2098c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_requests_inflight = 0; 2099c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2100c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2101c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2102c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2103c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2104c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2105c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 2106c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_resume(void) 2107c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 2108c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_requests_pump_waiting_queue(); 2109c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2110c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2111c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2112c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2113c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott_evdns_nameserver_add_impl(unsigned long int address, int port) { 2114c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* first check to see if we already have this nameserver */ 2115c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2116c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const struct nameserver *server = server_head, *const started_at = server_head; 2117c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *ns; 2118c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int err = 0; 2119c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server) { 2120c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott do { 2121c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server->address == address) return 3; 2122c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server = server->next; 2123c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } while (server != started_at); 2124c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2125c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2126c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns = (struct nameserver *) malloc(sizeof(struct nameserver)); 2127c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!ns) return -1; 2128c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2129c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memset(ns, 0, sizeof(struct nameserver)); 2130c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2131c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evtimer_set(&ns->timeout_event, nameserver_prod_callback, ns); 2132c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2133c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->socket = socket(PF_INET, SOCK_DGRAM, 0); 2134c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (ns->socket < 0) { err = 1; goto out1; } 2135c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evutil_make_socket_nonblocking(ns->socket); 2136c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2137c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->address = address; 2138c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->port = htons(port); 2139c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->state = 1; 2140c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott event_set(&ns->event, ns->socket, EV_READ | EV_PERSIST, nameserver_ready_callback, ns); 2141c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (event_add(&ns->event, NULL) < 0) { 2142c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott err = 2; 2143c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto out2; 2144c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2145c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2146c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Added nameserver %s", debug_ntoa(address)); 2147c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2148c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* insert this nameserver into the list of them */ 2149c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!server_head) { 2150c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->next = ns->prev = ns; 2151c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_head = ns; 2152c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2153c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->next = server_head->next; 2154c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns->prev = server_head; 2155c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_head->next = ns; 2156c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_head->prev == server_head) { 2157c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_head->prev = ns; 2158c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2159c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2160c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2161c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_good_nameservers++; 2162c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2163c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2164c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2165c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottout2: 2166c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott CLOSE_SOCKET(ns->socket); 2167c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottout1: 2168c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(ns); 2169c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Unable to add nameserver %s: error %d", debug_ntoa(address), err); 2170c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return err; 2171c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2172c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2173c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2174c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 2175c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_nameserver_add(unsigned long int address) { 2176c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return _evdns_nameserver_add_impl(address, 53); 2177c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2178c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2179c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2180c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 2181c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_nameserver_ip_add(const char *ip_as_string) { 2182c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct in_addr ina; 2183c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int port; 2184c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char buf[20]; 2185c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *cp; 2186c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cp = strchr(ip_as_string, ':'); 2187c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (! cp) { 2188c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cp = ip_as_string; 2189c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port = 53; 2190c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2191c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott port = strtoint(cp+1); 2192c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (port < 0 || port > 65535) { 2193c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 4; 2194c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2195c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if ((cp-ip_as_string) >= (int)sizeof(buf)) { 2196c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 4; 2197c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2198c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(buf, ip_as_string, cp-ip_as_string); 2199c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf[cp-ip_as_string] = '\0'; 2200c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cp = buf; 2201c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2202c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!inet_aton(cp, &ina)) { 2203c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 4; 2204c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2205c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return _evdns_nameserver_add_impl(ina.s_addr, port); 2206c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2207c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2208c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* insert into the tail of the queue */ 2209c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2210c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_request_insert(struct request *req, struct request **head) { 2211c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!*head) { 2212c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *head = req; 2213c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->next = req->prev = req; 2214c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return; 2215c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2216c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2217c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->prev = (*head)->prev; 2218c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->prev->next = req; 2219c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->next = *head; 2220c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (*head)->prev = req; 2221c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2222c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2223c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2224c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstring_num_dots(const char *s) { 2225c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int count = 0; 2226c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while ((s = strchr(s, '.'))) { 2227c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott s++; 2228c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott count++; 2229c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2230c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return count; 2231c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2232c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2233c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct request * 2234c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottrequest_new(int type, const char *name, int flags, 2235c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_callback_type callback, void *user_ptr) { 2236c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char issuing_now = 2237c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (global_requests_inflight < global_max_requests_inflight) ? 1 : 0; 2238c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2239c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int name_len = strlen(name); 2240c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int request_max_len = evdns_request_len(name_len); 2241c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const u16 trans_id = issuing_now ? transaction_id_pick() : 0xffff; 2242c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* the request data is alloced in a single block with the header */ 2243c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *const req = 2244c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (struct request *) malloc(sizeof(struct request) + request_max_len); 2245c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int rlen; 2246c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) flags; 2247c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2248c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req) return NULL; 2249c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memset(req, 0, sizeof(struct request)); 2250c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2251c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evtimer_set(&req->timeout_event, evdns_request_timeout_callback, req); 2252c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2253c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* request data lives just after the header */ 2254c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->request = ((u8 *) req) + sizeof(struct request); 2255c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* denotes that the request data shouldn't be free()ed */ 2256c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->request_appended = 1; 2257c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott rlen = evdns_request_data_build(name, name_len, trans_id, 2258c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott type, CLASS_INET, req->request, request_max_len); 2259c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (rlen < 0) 2260c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto err1; 2261c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->request_len = rlen; 2262c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->trans_id = trans_id; 2263c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->tx_count = 0; 2264c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->request_type = type; 2265c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_pointer = user_ptr; 2266c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->user_callback = callback; 2267c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->ns = issuing_now ? nameserver_pick() : NULL; 2268c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->next = req->prev = NULL; 2269c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2270c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return req; 2271c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scotterr1: 2272c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(req); 2273c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return NULL; 2274c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2275c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2276c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2277c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottrequest_submit(struct request *const req) { 2278c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->ns) { 2279c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* if it has a nameserver assigned then this is going */ 2280c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* straight into the inflight queue */ 2281c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_request_insert(req, &req_head); 2282c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_requests_inflight++; 2283c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_request_transmit(req); 2284c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2285c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_request_insert(req, &req_waiting_head); 2286c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_requests_waiting++; 2287c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2288c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2289c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2290c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2291c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint evdns_resolve_ipv4(const char *name, int flags, 2292c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_callback_type callback, void *ptr) { 2293c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Resolve requested for %s", name); 2294c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & DNS_QUERY_NO_SEARCH) { 2295c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *const req = 2296c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_new(TYPE_A, name, flags, callback, ptr); 2297c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req == NULL) 2298c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (1); 2299c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_submit(req); 2300c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (0); 2301c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2302c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (search_request_new(TYPE_A, name, flags, callback, ptr)); 2303c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2304c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2305c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2306c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2307c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint evdns_resolve_ipv6(const char *name, int flags, 2308c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_callback_type callback, void *ptr) { 2309c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Resolve requested for %s", name); 2310c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & DNS_QUERY_NO_SEARCH) { 2311c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *const req = 2312c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_new(TYPE_AAAA, name, flags, callback, ptr); 2313c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req == NULL) 2314c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (1); 2315c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_submit(req); 2316c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (0); 2317c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2318c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (search_request_new(TYPE_AAAA, name, flags, callback, ptr)); 2319c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2320c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2321c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2322c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint evdns_resolve_reverse(const struct in_addr *in, int flags, evdns_callback_type callback, void *ptr) { 2323c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char buf[32]; 2324c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *req; 2325c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 a; 2326c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(in); 2327c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott a = ntohl(in->s_addr); 2328c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evutil_snprintf(buf, sizeof(buf), "%d.%d.%d.%d.in-addr.arpa", 2329c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a )&0xff), 2330c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a>>8 )&0xff), 2331c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a>>16)&0xff), 2332c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (int)(u8)((a>>24)&0xff)); 2333c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Resolve requested for %s (reverse)", buf); 2334c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = request_new(TYPE_PTR, buf, flags, callback, ptr); 2335c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req) return 1; 2336c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_submit(req); 2337c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2338c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2339c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2340c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint evdns_resolve_reverse_ipv6(const struct in6_addr *in, int flags, evdns_callback_type callback, void *ptr) { 2341c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* 32 nybbles, 32 periods, "ip6.arpa", NUL. */ 2342c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char buf[73]; 2343c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *cp; 2344c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *req; 2345c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int i; 2346c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(in); 2347c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cp = buf; 2348c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i=15; i >= 0; --i) { 2349c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u8 byte = in->s6_addr[i]; 2350c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *cp++ = "0123456789abcdef"[byte & 0x0f]; 2351c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *cp++ = '.'; 2352c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *cp++ = "0123456789abcdef"[byte >> 4]; 2353c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *cp++ = '.'; 2354c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2355c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(cp + strlen("ip6.arpa") < buf+sizeof(buf)); 2356c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(cp, "ip6.arpa", strlen("ip6.arpa")+1); 2357c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Resolve requested for %s (reverse)", buf); 2358c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = request_new(TYPE_PTR, buf, flags, callback, ptr); 2359c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req) return 1; 2360c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_submit(req); 2361c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2362c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2363c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2364c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/*/////////////////////////////////////////////////////////////////// */ 2365c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Search support */ 2366c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 2367c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* the libc resolver has support for searching a number of domains */ 2368c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* to find a name. If nothing else then it takes the single domain */ 2369c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* from the gethostname() call. */ 2370c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 2371c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* It can also be configured via the domain and search options in a */ 2372c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* resolv.conf. */ 2373c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* */ 2374c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* The ndots option controls how many dots it takes for the resolver */ 2375c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* to decide that a name is non-local and so try a raw lookup first. */ 2376c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2377c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct search_domain { 2378c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int len; 2379c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct search_domain *next; 2380c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* the text string is appended to this structure */ 2381c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 2382c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2383c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstruct search_state { 2384c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int refcount; 2385c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int ndots; 2386c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int num_domains; 2387c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct search_domain *head; 2388c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott}; 2389c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2390c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct search_state *global_search_state = NULL; 2391c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2392c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2393c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_state_decref(struct search_state *const state) { 2394c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!state) return; 2395c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott state->refcount--; 2396c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!state->refcount) { 2397c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct search_domain *next, *dom; 2398c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (dom = state->head; dom; dom = next) { 2399c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott next = dom->next; 2400c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(dom); 2401c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2402c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(state); 2403c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2404c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2405c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2406c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic struct search_state * 2407c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_state_new(void) { 2408c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct search_state *state = (struct search_state *) malloc(sizeof(struct search_state)); 2409c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!state) return NULL; 2410c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memset(state, 0, sizeof(struct search_state)); 2411c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott state->refcount = 1; 2412c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott state->ndots = 1; 2413c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2414c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return state; 2415c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2416c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2417c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2418c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_postfix_clear(void) { 2419c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_state_decref(global_search_state); 2420c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2421c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state = search_state_new(); 2422c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2423c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2424c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2425c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 2426c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_search_clear(void) { 2427c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_postfix_clear(); 2428c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2429c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2430c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2431c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_postfix_add(const char *domain) { 2432c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int domain_len; 2433c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct search_domain *sdomain; 2434c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (domain[0] == '.') domain++; 2435c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott domain_len = strlen(domain); 2436c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2437c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!global_search_state) global_search_state = search_state_new(); 2438c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!global_search_state) return; 2439c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state->num_domains++; 2440c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2441c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott sdomain = (struct search_domain *) malloc(sizeof(struct search_domain) + domain_len); 2442c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!sdomain) return; 2443c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy( ((u8 *) sdomain) + sizeof(struct search_domain), domain, domain_len); 2444c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott sdomain->next = global_search_state->head; 2445c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott sdomain->len = domain_len; 2446c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2447c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state->head = sdomain; 2448c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2449c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2450c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* reverse the order of members in the postfix list. This is needed because, */ 2451c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* when parsing resolv.conf we push elements in the wrong order */ 2452c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2453c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_reverse(void) { 2454c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct search_domain *cur, *prev = NULL, *next; 2455c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cur = global_search_state->head; 2456c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (cur) { 2457c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott next = cur->next; 2458c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cur->next = prev; 2459c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott prev = cur; 2460c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cur = next; 2461c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2462c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2463c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state->head = prev; 2464c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2465c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2466c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2467c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 2468c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_search_add(const char *domain) { 2469c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_postfix_add(domain); 2470c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2471c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2472c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2473c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 2474c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_search_ndots_set(const int ndots) { 2475c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!global_search_state) global_search_state = search_state_new(); 2476c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!global_search_state) return; 2477c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state->ndots = ndots; 2478c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2479c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2480c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2481c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_set_from_hostname(void) { 2482c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char hostname[HOST_NAME_MAX + 1], *domainname; 2483c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2484c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_postfix_clear(); 2485c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (gethostname(hostname, sizeof(hostname))) return; 2486c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott domainname = strchr(hostname, '.'); 2487c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!domainname) return; 2488c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_postfix_add(domainname); 2489c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2490c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2491c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* warning: returns malloced string */ 2492c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic char * 2493c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_make_new(const struct search_state *const state, int n, const char *const base_name) { 2494c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int base_len = strlen(base_name); 2495c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char need_to_append_dot = base_name[base_len - 1] == '.' ? 0 : 1; 2496c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct search_domain *dom; 2497c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2498c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (dom = state->head; dom; dom = dom->next) { 2499c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!n--) { 2500c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* this is the postfix we want */ 2501c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* the actual postfix string is kept at the end of the structure */ 2502c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const u8 *const postfix = ((u8 *) dom) + sizeof(struct search_domain); 2503c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int postfix_len = dom->len; 2504c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *const newname = (char *) malloc(base_len + need_to_append_dot + postfix_len + 1); 2505c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!newname) return NULL; 2506c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(newname, base_name, base_len); 2507c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (need_to_append_dot) newname[base_len] = '.'; 2508c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(newname + base_len + need_to_append_dot, postfix, postfix_len); 2509c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott newname[base_len + need_to_append_dot + postfix_len] = 0; 2510c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return newname; 2511c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2512c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2513c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2514c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we ran off the end of the list and still didn't find the requested string */ 2515c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott abort(); 2516c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return NULL; /* unreachable; stops warnings in some compilers. */ 2517c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2518c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2519c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2520c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_request_new(int type, const char *const name, int flags, evdns_callback_type user_callback, void *user_arg) { 2521c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(type == TYPE_A || type == TYPE_AAAA); 2522c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if ( ((flags & DNS_QUERY_NO_SEARCH) == 0) && 2523c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state && 2524c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state->num_domains) { 2525c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* we have some domains to search */ 2526c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *req; 2527c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (string_num_dots(name) >= global_search_state->ndots) { 2528c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = request_new(type, name, flags, user_callback, user_arg); 2529c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req) return 1; 2530c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->search_index = -1; 2531c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2532c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *const new_name = search_make_new(global_search_state, 0, name); 2533c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!new_name) return 1; 2534c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req = request_new(type, new_name, flags, user_callback, user_arg); 2535c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(new_name); 2536c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req) return 1; 2537c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->search_index = 0; 2538c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2539c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->search_origname = strdup(name); 2540c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->search_state = global_search_state; 2541c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->search_flags = flags; 2542c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state->refcount++; 2543c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_submit(req); 2544c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2545c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2546c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *const req = request_new(type, name, flags, user_callback, user_arg); 2547c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!req) return 1; 2548c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_submit(req); 2549c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2550c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2551c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2552c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2553c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* this is called when a request has failed to find a name. We need to check */ 2554c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* if it is part of a search and, if so, try the next name in the list */ 2555c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* returns: */ 2556c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 0 another request has been submitted */ 2557c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 1 no more requests needed */ 2558c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2559c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_try_next(struct request *const req) { 2560c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->search_state) { 2561c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* it is part of a search */ 2562c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *new_name; 2563c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct request *newreq; 2564c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->search_index++; 2565c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->search_index >= req->search_state->num_domains) { 2566c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* no more postfixes to try, however we may need to try */ 2567c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* this name without a postfix */ 2568c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (string_num_dots(req->search_origname) < req->search_state->ndots) { 2569c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* yep, we need to try it raw */ 2570c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott newreq = request_new(req->request_type, req->search_origname, req->search_flags, req->user_callback, req->user_pointer); 2571c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Search: trying raw query %s", req->search_origname); 2572c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (newreq) { 2573c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_submit(newreq); 2574c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2575c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2576c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2577c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 2578c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2579c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2580c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott new_name = search_make_new(req->search_state, req->search_index, req->search_origname); 2581c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!new_name) return 1; 2582c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Search: now trying %s (%d)", new_name, req->search_index); 2583c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott newreq = request_new(req->request_type, new_name, req->search_flags, req->user_callback, req->user_pointer); 2584c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(new_name); 2585c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!newreq) return 1; 2586c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott newreq->search_origname = req->search_origname; 2587c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->search_origname = NULL; 2588c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott newreq->search_state = req->search_state; 2589c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott newreq->search_flags = req->search_flags; 2590c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott newreq->search_index = req->search_index; 2591c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott newreq->search_state->refcount++; 2592c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_submit(newreq); 2593c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2594c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2595c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 2596c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2597c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2598c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2599c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottsearch_request_finished(struct request *const req) { 2600c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->search_state) { 2601c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_state_decref(req->search_state); 2602c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->search_state = NULL; 2603c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2604c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->search_origname) { 2605c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(req->search_origname); 2606c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->search_origname = NULL; 2607c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2608c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2609c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2610c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/*/////////////////////////////////////////////////////////////////// */ 2611c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Parsing resolv.conf files */ 2612c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2613c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2614c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_resolv_set_defaults(int flags) { 2615c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* if the file isn't found then we assume a local resolver */ 2616c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & DNS_OPTION_SEARCH) search_set_from_hostname(); 2617c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & DNS_OPTION_NAMESERVERS) evdns_nameserver_ip_add("127.0.0.1"); 2618c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2619c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2620c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifndef HAVE_STRTOK_R 2621c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic char * 2622c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstrtok_r(char *s, const char *delim, char **state) { 2623c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return strtok(s, delim); 2624c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2625c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 2626c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2627c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* helper version of atoi which returns -1 on error */ 2628c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2629c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstrtoint(const char *const str) { 2630c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *endptr; 2631c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int r = strtol(str, &endptr, 10); 2632c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (*endptr) return -1; 2633c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return r; 2634c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2635c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2636c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* helper version of atoi that returns -1 on error and clips to bounds. */ 2637c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2638c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstrtoint_clipped(const char *const str, int min, int max) 2639c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 2640c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int r = strtoint(str); 2641c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r == -1) 2642c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return r; 2643c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else if (r<min) 2644c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return min; 2645c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else if (r>max) 2646c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return max; 2647c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else 2648c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return r; 2649c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2650c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2651c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2652c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 2653c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_set_option(const char *option, const char *val, int flags) 2654c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 2655c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!strncmp(option, "ndots:", 6)) { 2656c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int ndots = strtoint(val); 2657c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (ndots == -1) return -1; 2658c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(flags & DNS_OPTION_SEARCH)) return 0; 2659c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Setting ndots to %d", ndots); 2660c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!global_search_state) global_search_state = search_state_new(); 2661c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!global_search_state) return -1; 2662c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state->ndots = ndots; 2663c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (!strncmp(option, "timeout:", 8)) { 2664c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int timeout = strtoint(val); 2665c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (timeout == -1) return -1; 2666c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(flags & DNS_OPTION_MISC)) return 0; 2667c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Setting timeout to %d", timeout); 2668c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_timeout.tv_sec = timeout; 2669c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (!strncmp(option, "max-timeouts:", 12)) { 2670c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int maxtimeout = strtoint_clipped(val, 1, 255); 2671c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (maxtimeout == -1) return -1; 2672c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(flags & DNS_OPTION_MISC)) return 0; 2673c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Setting maximum allowed timeouts to %d", 2674c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott maxtimeout); 2675c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_max_nameserver_timeout = maxtimeout; 2676c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (!strncmp(option, "max-inflight:", 13)) { 2677c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const int maxinflight = strtoint_clipped(val, 1, 65000); 2678c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (maxinflight == -1) return -1; 2679c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(flags & DNS_OPTION_MISC)) return 0; 2680c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Setting maximum inflight requests to %d", 2681c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott maxinflight); 2682c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_max_requests_inflight = maxinflight; 2683c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (!strncmp(option, "attempts:", 9)) { 2684c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int retries = strtoint(val); 2685c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (retries == -1) return -1; 2686c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (retries > 255) retries = 255; 2687c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(flags & DNS_OPTION_MISC)) return 0; 2688c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Setting retries to %d", retries); 2689c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_max_retransmits = retries; 2690c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2691c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2692c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2693c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2694c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic void 2695c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottresolv_conf_parse_line(char *const start, int flags) { 2696c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *strtok_state; 2697c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott static const char *const delims = " \t"; 2698c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define NEXT_TOKEN strtok_r(NULL, delims, &strtok_state) 2699c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2700c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *const first_token = strtok_r(start, delims, &strtok_state); 2701c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!first_token) return; 2702c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2703c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!strcmp(first_token, "nameserver") && (flags & DNS_OPTION_NAMESERVERS)) { 2704c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *const nameserver = NEXT_TOKEN; 2705c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct in_addr ina; 2706c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2707c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (inet_aton(nameserver, &ina)) { 2708c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* address is valid */ 2709c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_nameserver_add(ina.s_addr); 2710c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2711c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (!strcmp(first_token, "domain") && (flags & DNS_OPTION_SEARCH)) { 2712c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *const domain = NEXT_TOKEN; 2713c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (domain) { 2714c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_postfix_clear(); 2715c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_postfix_add(domain); 2716c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2717c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (!strcmp(first_token, "search") && (flags & DNS_OPTION_SEARCH)) { 2718c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *domain; 2719c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_postfix_clear(); 2720c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2721c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while ((domain = NEXT_TOKEN)) { 2722c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_postfix_add(domain); 2723c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2724c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_reverse(); 2725c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (!strcmp(first_token, "options")) { 2726c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *option; 2727c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while ((option = NEXT_TOKEN)) { 2728c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *val = strchr(option, ':'); 2729c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_set_option(option, val ? val+1 : "", flags); 2730c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2731c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2732c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#undef NEXT_TOKEN 2733c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2734c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2735c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* exported function */ 2736c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* returns: */ 2737c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 0 no errors */ 2738c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 1 failed to open file */ 2739c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 2 failed to stat file */ 2740c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 3 file too large */ 2741c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 4 out of memory */ 2742c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* 5 short read from file */ 2743c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 2744c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_resolv_conf_parse(int flags, const char *const filename) { 2745c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct stat st; 2746c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int fd, n, r; 2747c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u8 *resolv; 2748c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *start; 2749c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int err = 0; 2750c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2751c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Parsing resolv.conf file %s", filename); 2752c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2753c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fd = open(filename, O_RDONLY); 2754c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (fd < 0) { 2755c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_resolv_set_defaults(flags); 2756c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 2757c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2758c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2759c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (fstat(fd, &st)) { err = 2; goto out1; } 2760c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!st.st_size) { 2761c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_resolv_set_defaults(flags); 2762c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott err = (flags & DNS_OPTION_NAMESERVERS) ? 6 : 0; 2763c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto out1; 2764c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2765c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (st.st_size > 65535) { err = 3; goto out1; } /* no resolv.conf should be any bigger */ 2766c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2767c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott resolv = (u8 *) malloc((size_t)st.st_size + 1); 2768c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!resolv) { err = 4; goto out1; } 2769c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2770c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott n = 0; 2771c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while ((r = read(fd, resolv+n, (size_t)st.st_size-n)) > 0) { 2772c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott n += r; 2773c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (n == st.st_size) 2774c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 2775c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(n < st.st_size); 2776c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2777c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r < 0) { err = 5; goto out2; } 2778c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott resolv[n] = 0; /* we malloced an extra byte; this should be fine. */ 2779c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2780c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott start = (char *) resolv; 2781c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (;;) { 2782c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *const newline = strchr(start, '\n'); 2783c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!newline) { 2784c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott resolv_conf_parse_line(start, flags); 2785c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 2786c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2787c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott *newline = 0; 2788c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott resolv_conf_parse_line(start, flags); 2789c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott start = newline + 1; 2790c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2791c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2792c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2793c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!server_head && (flags & DNS_OPTION_NAMESERVERS)) { 2794c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* no nameservers were configured. */ 2795c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_nameserver_ip_add("127.0.0.1"); 2796c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott err = 6; 2797c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2798c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (flags & DNS_OPTION_SEARCH && (!global_search_state || global_search_state->num_domains == 0)) { 2799c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott search_set_from_hostname(); 2800c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2801c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2802c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottout2: 2803c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(resolv); 2804c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottout1: 2805c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott close(fd); 2806c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return err; 2807c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2808c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2809c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef WIN32 2810c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Add multiple nameservers from a space-or-comma-separated list. */ 2811c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2812c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_nameserver_ip_add_line(const char *ips) { 2813c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott const char *addr; 2814c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *buf; 2815c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int r; 2816c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (*ips) { 2817c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (ISSPACE(*ips) || *ips == ',' || *ips == '\t') 2818c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ++ips; 2819c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott addr = ips; 2820c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (ISDIGIT(*ips) || *ips == '.' || *ips == ':') 2821c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ++ips; 2822c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf = malloc(ips-addr+1); 2823c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!buf) return 4; 2824c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott memcpy(buf, addr, ips-addr); 2825c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf[ips-addr] = '\0'; 2826c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = evdns_nameserver_ip_add(buf); 2827c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(buf); 2828c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r) return r; 2829c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2830c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2831c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2832c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2833c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scotttypedef DWORD(WINAPI *GetNetworkParams_fn_t)(FIXED_INFO *, DWORD*); 2834c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2835c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* Use the windows GetNetworkParams interface in iphlpapi.dll to */ 2836c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott/* figure out what our nameservers are. */ 2837c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2838c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottload_nameservers_with_getnetworkparams(void) 2839c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 2840c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* Based on MSDN examples and inspection of c-ares code. */ 2841c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott FIXED_INFO *fixed; 2842c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott HMODULE handle = 0; 2843c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ULONG size = sizeof(FIXED_INFO); 2844c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott void *buf = NULL; 2845c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int status = 0, r, added_any; 2846c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott IP_ADDR_STRING *ns; 2847c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott GetNetworkParams_fn_t fn; 2848c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2849c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(handle = LoadLibrary("iphlpapi.dll"))) { 2850c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Could not open iphlpapi.dll"); 2851c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott status = -1; 2852c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto done; 2853c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2854c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(fn = (GetNetworkParams_fn_t) GetProcAddress(handle, "GetNetworkParams"))) { 2855c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN, "Could not get address of function."); 2856c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott status = -1; 2857c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto done; 2858c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2859c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2860c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf = malloc(size); 2861c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!buf) { status = 4; goto done; } 2862c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fixed = buf; 2863c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = fn(fixed, &size); 2864c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r != ERROR_SUCCESS && r != ERROR_BUFFER_OVERFLOW) { 2865c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott status = -1; 2866c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto done; 2867c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2868c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r != ERROR_SUCCESS) { 2869c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(buf); 2870c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott buf = malloc(size); 2871c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!buf) { status = 4; goto done; } 2872c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fixed = buf; 2873c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = fn(fixed, &size); 2874c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r != ERROR_SUCCESS) { 2875c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "fn() failed."); 2876c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott status = -1; 2877c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto done; 2878c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2879c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2880c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2881c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott assert(fixed); 2882c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott added_any = 0; 2883c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns = &(fixed->DnsServerList); 2884c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (ns) { 2885c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = evdns_nameserver_ip_add_line(ns->IpAddress.String); 2886c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r) { 2887c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG,"Could not add nameserver %s to list,error: %d", 2888c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (ns->IpAddress.String),(int)GetLastError()); 2889c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott status = r; 2890c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott goto done; 2891c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2892c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG,"Succesfully added %s as nameserver",ns->IpAddress.String); 2893c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2894c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2895c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott added_any++; 2896c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ns = ns->Next; 2897c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2898c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2899c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!added_any) { 2900c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "No nameservers added."); 2901c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott status = -1; 2902c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2903c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2904c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott done: 2905c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (buf) 2906c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(buf); 2907c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (handle) 2908c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott FreeLibrary(handle); 2909c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return status; 2910c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2911c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2912c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2913c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottconfig_nameserver_from_reg_key(HKEY key, const char *subkey) 2914c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 2915c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *buf; 2916c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott DWORD bufsz = 0, type = 0; 2917c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int status = 0; 2918c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2919c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (RegQueryValueEx(key, subkey, 0, &type, NULL, &bufsz) 2920c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott != ERROR_MORE_DATA) 2921c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 2922c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!(buf = malloc(bufsz))) 2923c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 2924c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2925c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (RegQueryValueEx(key, subkey, 0, &type, (LPBYTE)buf, &bufsz) 2926c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott == ERROR_SUCCESS && bufsz > 1) { 2927c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott status = evdns_nameserver_ip_add_line(buf); 2928c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2929c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2930c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(buf); 2931c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return status; 2932c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2933c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2934c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define SERVICES_KEY "System\\CurrentControlSet\\Services\\" 2935c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define WIN_NS_9X_KEY SERVICES_KEY "VxD\\MSTCP" 2936c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define WIN_NS_NT_KEY SERVICES_KEY "Tcpip\\Parameters" 2937c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2938c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic int 2939c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottload_nameservers_from_registry(void) 2940c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 2941c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int found = 0; 2942c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int r; 2943c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#define TRY(k, name) \ 2944c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!found && config_nameserver_from_reg_key(k,name) == 0) { \ 2945c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG,"Found nameservers in %s/%s",#k,name); \ 2946c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott found = 1; \ 2947c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (!found) { \ 2948c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG,"Didn't find nameservers in %s/%s", \ 2949c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott #k,#name); \ 2950c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2951c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2952c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (((int)GetVersion()) > 0) { /* NT */ 2953c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott HKEY nt_key = 0, interfaces_key = 0; 2954c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2955c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, WIN_NS_NT_KEY, 0, 2956c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott KEY_READ, &nt_key) != ERROR_SUCCESS) { 2957c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG,"Couldn't open nt key, %d",(int)GetLastError()); 2958c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 2959c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2960c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = RegOpenKeyEx(nt_key, "Interfaces", 0, 2961c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott KEY_QUERY_VALUE|KEY_ENUMERATE_SUB_KEYS, 2962c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott &interfaces_key); 2963c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r != ERROR_SUCCESS) { 2964c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG,"Couldn't open interfaces key, %d",(int)GetLastError()); 2965c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 2966c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2967c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott TRY(nt_key, "NameServer"); 2968c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott TRY(nt_key, "DhcpNameServer"); 2969c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott TRY(interfaces_key, "NameServer"); 2970c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott TRY(interfaces_key, "DhcpNameServer"); 2971c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott RegCloseKey(interfaces_key); 2972c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott RegCloseKey(nt_key); 2973c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 2974c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott HKEY win_key = 0; 2975c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, WIN_NS_9X_KEY, 0, 2976c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott KEY_READ, &win_key) != ERROR_SUCCESS) { 2977c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_DEBUG, "Couldn't open registry key, %d", (int)GetLastError()); 2978c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return -1; 2979c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2980c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott TRY(win_key, "NameServer"); 2981c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott RegCloseKey(win_key); 2982c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2983c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2984c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (found == 0) { 2985c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott log(EVDNS_LOG_WARN,"Didn't find any nameservers."); 2986c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 2987c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2988c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return found ? 0 : -1; 2989c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#undef TRY 2990c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2991c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 2992c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 2993c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_config_windows_nameservers(void) 2994c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 2995c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (load_nameservers_with_getnetworkparams() == 0) 2996c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 2997c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return load_nameservers_from_registry(); 2998c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 2999c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 3000c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3001c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 3002c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_init(void) 3003c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 3004c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int res = 0; 3005c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef WIN32 3006c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott res = evdns_config_windows_nameservers(); 3007c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#else 3008c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott res = evdns_resolv_conf_parse(DNS_OPTIONS_ALL, "/etc/resolv.conf"); 3009c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 3010c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3011c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return (res); 3012c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 3013c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3014c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottconst char * 3015c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_err_to_string(int err) 3016c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 3017c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott switch (err) { 3018c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_NONE: return "no error"; 3019c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_FORMAT: return "misformatted query"; 3020c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_SERVERFAILED: return "server failed"; 3021c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_NOTEXIST: return "name does not exist"; 3022c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_NOTIMPL: return "query not implemented"; 3023c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_REFUSED: return "refused"; 3024c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3025c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_TRUNCATED: return "reply truncated or ill-formed"; 3026c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_UNKNOWN: return "unknown"; 3027c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_TIMEOUT: return "request timed out"; 3028c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott case DNS_ERR_SHUTDOWN: return "dns subsystem shut down"; 3029c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott default: return "[Unknown error code]"; 3030c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3031c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 3032c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3033c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 3034c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_shutdown(int fail_requests) 3035c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 3036c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct nameserver *server, *server_next; 3037c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct search_domain *dom, *dom_next; 3038c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3039c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (req_head) { 3040c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (fail_requests) 3041c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply_callback(req_head, 0, DNS_ERR_SHUTDOWN, NULL); 3042c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_finished(req_head, &req_head); 3043c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3044c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (req_waiting_head) { 3045c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (fail_requests) 3046c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reply_callback(req_waiting_head, 0, DNS_ERR_SHUTDOWN, NULL); 3047c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott request_finished(req_waiting_head, &req_waiting_head); 3048c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3049c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_requests_inflight = global_requests_waiting = 0; 3050c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3051c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (server = server_head; server; server = server_next) { 3052c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_next = server->next; 3053c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server->socket >= 0) 3054c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott CLOSE_SOCKET(server->socket); 3055c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) event_del(&server->event); 3056c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server->state == 0) 3057c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) event_del(&server->timeout_event); 3058c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(server); 3059c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (server_next == server_head) 3060c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott break; 3061c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3062c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott server_head = NULL; 3063c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_good_nameservers = 0; 3064c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3065c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (global_search_state) { 3066c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (dom = global_search_state->head; dom; dom = dom_next) { 3067c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott dom_next = dom->next; 3068c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(dom); 3069c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3070c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott free(global_search_state); 3071c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott global_search_state = NULL; 3072c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3073c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_log_fn = NULL; 3074c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 3075c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3076c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#ifdef EVDNS_MAIN 3077c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 3078c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottmain_callback(int result, char type, int count, int ttl, 3079c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott void *addrs, void *orig) { 3080c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott char *n = (char*)orig; 3081c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int i; 3082c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < count; ++i) { 3083c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (type == DNS_IPv4_A) { 3084c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott printf("%s: %s\n", n, debug_ntoa(((u32*)addrs)[i])); 3085c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (type == DNS_PTR) { 3086c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott printf("%s: %s\n", n, ((char**)addrs)[i]); 3087c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3088c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3089c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!count) { 3090c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott printf("%s: No answer (%d)\n", n, result); 3091c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3092c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fflush(stdout); 3093c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 3094c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 3095c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottevdns_server_callback(struct evdns_server_request *req, void *data) 3096c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott{ 3097c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int i, r; 3098c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void)data; 3099c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott /* dummy; give 192.168.11.11 as an answer for all A questions, 3100c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott * give foo.bar.example.com as an answer for all PTR questions. */ 3101c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (i = 0; i < req->nquestions; ++i) { 3102c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott u32 ans = htonl(0xc0a80b0bUL); 3103c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (req->questions[i]->type == EVDNS_TYPE_A && 3104c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->questions[i]->dns_question_class == EVDNS_CLASS_INET) { 3105c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott printf(" -- replying for %s (A)\n", req->questions[i]->name); 3106c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = evdns_server_request_add_a_reply(req, req->questions[i]->name, 3107c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 1, &ans, 10); 3108c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r<0) 3109c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott printf("eeep, didn't work.\n"); 3110c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else if (req->questions[i]->type == EVDNS_TYPE_PTR && 3111c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->questions[i]->dns_question_class == EVDNS_CLASS_INET) { 3112c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott printf(" -- replying for %s (PTR)\n", req->questions[i]->name); 3113c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = evdns_server_request_add_ptr_reply(req, NULL, req->questions[i]->name, 3114c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott "foo.bar.example.com", 10); 3115c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 3116c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott printf(" -- skipping %s [%d %d]\n", req->questions[i]->name, 3117c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott req->questions[i]->type, req->questions[i]->dns_question_class); 3118c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3119c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3120c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3121c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott r = evdns_request_respond(req, 0); 3122c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (r<0) 3123c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott printf("eeek, couldn't send reply.\n"); 3124c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 3125c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 3126c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottvoid 3127c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottlogfn(int is_warn, const char *msg) { 3128c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott (void) is_warn; 3129c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fprintf(stderr, "%s\n", msg); 3130c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 3131c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottint 3132c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottmain(int c, char **v) { 3133c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int idx; 3134c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int reverse = 0, verbose = 1, servertest = 0; 3135c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (c<2) { 3136c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fprintf(stderr, "syntax: %s [-x] [-v] hostname\n", v[0]); 3137c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fprintf(stderr, "syntax: %s [-servertest]\n", v[0]); 3138c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 1; 3139c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3140c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott idx = 1; 3141c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott while (idx < c && v[idx][0] == '-') { 3142c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!strcmp(v[idx], "-x")) 3143c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott reverse = 1; 3144c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else if (!strcmp(v[idx], "-v")) 3145c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott verbose = 1; 3146c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else if (!strcmp(v[idx], "-servertest")) 3147c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott servertest = 1; 3148c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott else 3149c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fprintf(stderr, "Unknown option %s\n", v[idx]); 3150c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott ++idx; 3151c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3152c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott event_init(); 3153c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (verbose) 3154c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_set_log_fn(logfn); 3155c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_resolv_conf_parse(DNS_OPTION_NAMESERVERS, "/etc/resolv.conf"); 3156c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (servertest) { 3157c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott int sock; 3158c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct sockaddr_in my_addr; 3159c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott sock = socket(PF_INET, SOCK_DGRAM, 0); 3160c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evutil_make_socket_nonblocking(sock); 3161c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott my_addr.sin_family = AF_INET; 3162c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott my_addr.sin_port = htons(10053); 3163c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott my_addr.sin_addr.s_addr = INADDR_ANY; 3164c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (bind(sock, (struct sockaddr*)&my_addr, sizeof(my_addr))<0) { 3165c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott perror("bind"); 3166c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott exit(1); 3167c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3168c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_add_server_port(sock, 0, evdns_server_callback, NULL); 3169c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3170c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (; idx < c; ++idx) { 3171c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (reverse) { 3172c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott struct in_addr addr; 3173c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott if (!inet_aton(v[idx], &addr)) { 3174c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fprintf(stderr, "Skipping non-IP %s\n", v[idx]); 3175c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott continue; 3176c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3177c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fprintf(stderr, "resolving %s...\n",v[idx]); 3178c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_resolve_reverse(&addr, 0, main_callback, v[idx]); 3179c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } else { 3180c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fprintf(stderr, "resolving (fwd) %s...\n",v[idx]); 3181c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott evdns_resolve_ipv4(v[idx], 0, main_callback, v[idx]); 3182c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3183c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 3184c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott fflush(stdout); 3185c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott event_dispatch(); 3186c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott return 0; 3187c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 3188c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#endif 3189