Bonjour - First step: AutoIP compiled in and working

Users » darran » Code » Bonjour

Darran Shepherd / Mbed 2 deprecated Bonjour

First step: AutoIP compiled in and working

lwip/core/dns.c@1:4218cacaf696, 2010-06-18 (annotated)

Committer:: darran
Date:: Fri Jun 18 15:54:21 2010 +0000
Revision:: 1:4218cacaf696
Parent:: 0:55a05330f8cc

Who changed what in which revision?

User	Revision	Line number	New contents of line
darran	1:4218cacaf696	1	/**
darran	1:4218cacaf696	2	* @file
darran	1:4218cacaf696	3	* DNS - host name to IP address resolver.
darran	1:4218cacaf696	4	*
darran	1:4218cacaf696	5	*/
darran	1:4218cacaf696	6
darran	1:4218cacaf696	7	/**
darran	1:4218cacaf696	8
darran	1:4218cacaf696	9	* This file implements a DNS host name to IP address resolver.
darran	1:4218cacaf696	10
darran	1:4218cacaf696	11	* Port to lwIP from uIP
darran	1:4218cacaf696	12	* by Jim Pettinato April 2007
darran	1:4218cacaf696	13
darran	1:4218cacaf696	14	* uIP version Copyright (c) 2002-2003, Adam Dunkels.
darran	1:4218cacaf696	15	* All rights reserved.
darran	1:4218cacaf696	16	*
darran	1:4218cacaf696	17	* Redistribution and use in source and binary forms, with or without
darran	1:4218cacaf696	18	* modification, are permitted provided that the following conditions
darran	1:4218cacaf696	19	* are met:
darran	1:4218cacaf696	20	* 1. Redistributions of source code must retain the above copyright
darran	1:4218cacaf696	21	* notice, this list of conditions and the following disclaimer.
darran	1:4218cacaf696	22	* 2. Redistributions in binary form must reproduce the above copyright
darran	1:4218cacaf696	23	* notice, this list of conditions and the following disclaimer in the
darran	1:4218cacaf696	24	* documentation and/or other materials provided with the distribution.
darran	1:4218cacaf696	25	* 3. The name of the author may not be used to endorse or promote
darran	1:4218cacaf696	26	* products derived from this software without specific prior
darran	1:4218cacaf696	27	* written permission.
darran	1:4218cacaf696	28	*
darran	1:4218cacaf696	29	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
darran	1:4218cacaf696	30	* OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
darran	1:4218cacaf696	31	* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
darran	1:4218cacaf696	32	* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
darran	1:4218cacaf696	33	* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
darran	1:4218cacaf696	34	* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
darran	1:4218cacaf696	35	* GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
darran	1:4218cacaf696	36	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
darran	1:4218cacaf696	37	* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
darran	1:4218cacaf696	38	* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
darran	1:4218cacaf696	39	* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
darran	1:4218cacaf696	40	*
darran	1:4218cacaf696	41	*
darran	1:4218cacaf696	42	* DNS.C
darran	1:4218cacaf696	43	*
darran	1:4218cacaf696	44	* The lwIP DNS resolver functions are used to lookup a host name and
darran	1:4218cacaf696	45	* map it to a numerical IP address. It maintains a list of resolved
darran	1:4218cacaf696	46	* hostnames that can be queried with the dns_lookup() function.
darran	1:4218cacaf696	47	* New hostnames can be resolved using the dns_query() function.
darran	1:4218cacaf696	48	*
darran	1:4218cacaf696	49	* The lwIP version of the resolver also adds a non-blocking version of
darran	1:4218cacaf696	50	* gethostbyname() that will work with a raw API application. This function
darran	1:4218cacaf696	51	* checks for an IP address string first and converts it if it is valid.
darran	1:4218cacaf696	52	* gethostbyname() then does a dns_lookup() to see if the name is
darran	1:4218cacaf696	53	* already in the table. If so, the IP is returned. If not, a query is
darran	1:4218cacaf696	54	* issued and the function returns with a ERR_INPROGRESS status. The app
darran	1:4218cacaf696	55	* using the dns client must then go into a waiting state.
darran	1:4218cacaf696	56	*
darran	1:4218cacaf696	57	* Once a hostname has been resolved (or found to be non-existent),
darran	1:4218cacaf696	58	* the resolver code calls a specified callback function (which
darran	1:4218cacaf696	59	* must be implemented by the module that uses the resolver).
darran	1:4218cacaf696	60	*/
darran	1:4218cacaf696	61
darran	1:4218cacaf696	62	/*-----------------------------------------------------------------------------
darran	1:4218cacaf696	63	* RFC 1035 - Domain names - implementation and specification
darran	1:4218cacaf696	64	* RFC 2181 - Clarifications to the DNS Specification
darran	1:4218cacaf696	65	----------------------------------------------------------------------------/
darran	1:4218cacaf696	66
darran	1:4218cacaf696	67	/** @todo: define good default values (rfc compliance) */
darran	1:4218cacaf696	68	/** @todo: improve answer parsing, more checkings... */
darran	1:4218cacaf696	69	/** @todo: check RFC1035 - 7.3. Processing responses */
darran	1:4218cacaf696	70
darran	1:4218cacaf696	71	/*-----------------------------------------------------------------------------
darran	1:4218cacaf696	72	* Includes
darran	1:4218cacaf696	73	----------------------------------------------------------------------------/
darran	1:4218cacaf696	74
darran	1:4218cacaf696	75	#include "lwip/opt.h"
darran	1:4218cacaf696	76
darran	1:4218cacaf696	77	#if LWIP_DNS /* don't build if not configured for use in lwipopts.h */
darran	1:4218cacaf696	78
darran	1:4218cacaf696	79	#include "lwip/udp.h"
darran	1:4218cacaf696	80	#include "lwip/mem.h"
darran	1:4218cacaf696	81	#include "lwip/memp.h"
darran	1:4218cacaf696	82	#include "lwip/dns.h"
darran	1:4218cacaf696	83	#include "lwip/debug.h"
darran	1:4218cacaf696	84
darran	1:4218cacaf696	85	#include <string.h>
darran	1:4218cacaf696	86
darran	1:4218cacaf696	87	/** DNS server IP address */
darran	1:4218cacaf696	88	#ifndef DNS_SERVER_ADDRESS
darran	1:4218cacaf696	89	#define DNS_SERVER_ADDRESS(ipaddr) (ip4_addr_set_u32(ipaddr, ipaddr_addr("208.67.222.222"))) /* resolver1.opendns.com */
darran	1:4218cacaf696	90	#endif
darran	1:4218cacaf696	91
darran	1:4218cacaf696	92	/** DNS server port address */
darran	1:4218cacaf696	93	#ifndef DNS_SERVER_PORT
darran	1:4218cacaf696	94	#define DNS_SERVER_PORT 53
darran	1:4218cacaf696	95	#endif
darran	1:4218cacaf696	96
darran	1:4218cacaf696	97	/** DNS maximum number of retries when asking for a name, before "timeout". */
darran	1:4218cacaf696	98	#ifndef DNS_MAX_RETRIES
darran	1:4218cacaf696	99	#define DNS_MAX_RETRIES 4
darran	1:4218cacaf696	100	#endif
darran	1:4218cacaf696	101
darran	1:4218cacaf696	102	/** DNS resource record max. TTL (one week as default) */
darran	1:4218cacaf696	103	#ifndef DNS_MAX_TTL
darran	1:4218cacaf696	104	#define DNS_MAX_TTL 604800
darran	1:4218cacaf696	105	#endif
darran	1:4218cacaf696	106
darran	1:4218cacaf696	107	/* DNS protocol flags */
darran	1:4218cacaf696	108	#define DNS_FLAG1_RESPONSE 0x80
darran	1:4218cacaf696	109	#define DNS_FLAG1_OPCODE_STATUS 0x10
darran	1:4218cacaf696	110	#define DNS_FLAG1_OPCODE_INVERSE 0x08
darran	1:4218cacaf696	111	#define DNS_FLAG1_OPCODE_STANDARD 0x00
darran	1:4218cacaf696	112	#define DNS_FLAG1_AUTHORATIVE 0x04
darran	1:4218cacaf696	113	#define DNS_FLAG1_TRUNC 0x02
darran	1:4218cacaf696	114	#define DNS_FLAG1_RD 0x01
darran	1:4218cacaf696	115	#define DNS_FLAG2_RA 0x80
darran	1:4218cacaf696	116	#define DNS_FLAG2_ERR_MASK 0x0f
darran	1:4218cacaf696	117	#define DNS_FLAG2_ERR_NONE 0x00
darran	1:4218cacaf696	118	#define DNS_FLAG2_ERR_NAME 0x03
darran	1:4218cacaf696	119
darran	1:4218cacaf696	120	/* DNS protocol states */
darran	1:4218cacaf696	121	#define DNS_STATE_UNUSED 0
darran	1:4218cacaf696	122	#define DNS_STATE_NEW 1
darran	1:4218cacaf696	123	#define DNS_STATE_ASKING 2
darran	1:4218cacaf696	124	#define DNS_STATE_DONE 3
darran	1:4218cacaf696	125
darran	1:4218cacaf696	126	#ifdef PACK_STRUCT_USE_INCLUDES
darran	1:4218cacaf696	127	# include "arch/bpstruct.h"
darran	1:4218cacaf696	128	#endif
darran	1:4218cacaf696	129	PACK_STRUCT_BEGIN
darran	1:4218cacaf696	130	/** DNS message header */
darran	1:4218cacaf696	131	struct dns_hdr {
darran	1:4218cacaf696	132	PACK_STRUCT_FIELD(u16_t id);
darran	1:4218cacaf696	133	PACK_STRUCT_FIELD(u8_t flags1);
darran	1:4218cacaf696	134	PACK_STRUCT_FIELD(u8_t flags2);
darran	1:4218cacaf696	135	PACK_STRUCT_FIELD(u16_t numquestions);
darran	1:4218cacaf696	136	PACK_STRUCT_FIELD(u16_t numanswers);
darran	1:4218cacaf696	137	PACK_STRUCT_FIELD(u16_t numauthrr);
darran	1:4218cacaf696	138	PACK_STRUCT_FIELD(u16_t numextrarr);
darran	1:4218cacaf696	139	} PACK_STRUCT_STRUCT;
darran	1:4218cacaf696	140	PACK_STRUCT_END
darran	1:4218cacaf696	141	#ifdef PACK_STRUCT_USE_INCLUDES
darran	1:4218cacaf696	142	# include "arch/epstruct.h"
darran	1:4218cacaf696	143	#endif
darran	1:4218cacaf696	144	#define SIZEOF_DNS_HDR 12
darran	1:4218cacaf696	145
darran	1:4218cacaf696	146	/** DNS query message structure.
darran	1:4218cacaf696	147	No packing needed: only used locally on the stack. */
darran	1:4218cacaf696	148	struct dns_query {
darran	1:4218cacaf696	149	/* DNS query record starts with either a domain name or a pointer
darran	1:4218cacaf696	150	to a name already present somewhere in the packet. */
darran	1:4218cacaf696	151	u16_t type;
darran	1:4218cacaf696	152	u16_t cls;
darran	1:4218cacaf696	153	};
darran	1:4218cacaf696	154	#define SIZEOF_DNS_QUERY 4
darran	1:4218cacaf696	155
darran	1:4218cacaf696	156	/** DNS answer message structure.
darran	1:4218cacaf696	157	No packing needed: only used locally on the stack. */
darran	1:4218cacaf696	158	struct dns_answer {
darran	1:4218cacaf696	159	/* DNS answer record starts with either a domain name or a pointer
darran	1:4218cacaf696	160	to a name already present somewhere in the packet. */
darran	1:4218cacaf696	161	u16_t type;
darran	1:4218cacaf696	162	u16_t cls;
darran	1:4218cacaf696	163	u32_t ttl;
darran	1:4218cacaf696	164	u16_t len;
darran	1:4218cacaf696	165	};
darran	1:4218cacaf696	166	#define SIZEOF_DNS_ANSWER 10
darran	1:4218cacaf696	167
darran	1:4218cacaf696	168	/** DNS table entry */
darran	1:4218cacaf696	169	struct dns_table_entry {
darran	1:4218cacaf696	170	u8_t state;
darran	1:4218cacaf696	171	u8_t numdns;
darran	1:4218cacaf696	172	u8_t tmr;
darran	1:4218cacaf696	173	u8_t retries;
darran	1:4218cacaf696	174	u8_t seqno;
darran	1:4218cacaf696	175	u8_t err;
darran	1:4218cacaf696	176	u32_t ttl;
darran	1:4218cacaf696	177	char name[DNS_MAX_NAME_LENGTH];
darran	1:4218cacaf696	178	ip_addr_t ipaddr;
darran	1:4218cacaf696	179	/* pointer to callback on DNS query done */
darran	1:4218cacaf696	180	dns_found_callback found;
darran	1:4218cacaf696	181	void *arg;
darran	1:4218cacaf696	182	};
darran	1:4218cacaf696	183
darran	1:4218cacaf696	184	#if DNS_LOCAL_HOSTLIST
darran	1:4218cacaf696	185
darran	1:4218cacaf696	186	#if DNS_LOCAL_HOSTLIST_IS_DYNAMIC
darran	1:4218cacaf696	187	/** Local host-list. For hostnames in this list, no
darran	1:4218cacaf696	188	* external name resolution is performed */
darran	1:4218cacaf696	189	static struct local_hostlist_entry *local_hostlist_dynamic;
darran	1:4218cacaf696	190	#else /* DNS_LOCAL_HOSTLIST_IS_DYNAMIC */
darran	1:4218cacaf696	191
darran	1:4218cacaf696	192	/** Defining this allows the local_hostlist_static to be placed in a different
darran	1:4218cacaf696	193	* linker section (e.g. FLASH) */
darran	1:4218cacaf696	194	#ifndef DNS_LOCAL_HOSTLIST_STORAGE_PRE
darran	1:4218cacaf696	195	#define DNS_LOCAL_HOSTLIST_STORAGE_PRE static
darran	1:4218cacaf696	196	#endif /* DNS_LOCAL_HOSTLIST_STORAGE_PRE */
darran	1:4218cacaf696	197	/** Defining this allows the local_hostlist_static to be placed in a different
darran	1:4218cacaf696	198	* linker section (e.g. FLASH) */
darran	1:4218cacaf696	199	#ifndef DNS_LOCAL_HOSTLIST_STORAGE_POST
darran	1:4218cacaf696	200	#define DNS_LOCAL_HOSTLIST_STORAGE_POST
darran	1:4218cacaf696	201	#endif /* DNS_LOCAL_HOSTLIST_STORAGE_POST */
darran	1:4218cacaf696	202	DNS_LOCAL_HOSTLIST_STORAGE_PRE struct local_hostlist_entry local_hostlist_static[]
darran	1:4218cacaf696	203	DNS_LOCAL_HOSTLIST_STORAGE_POST = DNS_LOCAL_HOSTLIST_INIT;
darran	1:4218cacaf696	204
darran	1:4218cacaf696	205	#endif /* DNS_LOCAL_HOSTLIST_IS_DYNAMIC */
darran	1:4218cacaf696	206
darran	1:4218cacaf696	207	static void dns_init_local();
darran	1:4218cacaf696	208	#endif /* DNS_LOCAL_HOSTLIST */
darran	1:4218cacaf696	209
darran	1:4218cacaf696	210
darran	1:4218cacaf696	211	/* forward declarations */
darran	1:4218cacaf696	212	static void dns_recv(void s, struct udp_pcb pcb, struct pbuf p, ip_addr_t addr, u16_t port);
darran	1:4218cacaf696	213	static void dns_check_entries(void);
darran	1:4218cacaf696	214
darran	1:4218cacaf696	215	/*-----------------------------------------------------------------------------
darran	1:4218cacaf696	216	* Globales
darran	1:4218cacaf696	217	----------------------------------------------------------------------------/
darran	1:4218cacaf696	218
darran	1:4218cacaf696	219	/* DNS variables */
darran	1:4218cacaf696	220	static struct udp_pcb *dns_pcb;
darran	1:4218cacaf696	221	static u8_t dns_seqno;
darran	1:4218cacaf696	222	static struct dns_table_entry dns_table[DNS_TABLE_SIZE];
darran	1:4218cacaf696	223	static ip_addr_t dns_servers[DNS_MAX_SERVERS];
darran	1:4218cacaf696	224	/** Contiguous buffer for processing responses */
darran	1:4218cacaf696	225	static u8_t dns_payload_buffer[LWIP_MEM_ALIGN_BUFFER(DNS_MSG_SIZE)];
darran	1:4218cacaf696	226	static u8_t* dns_payload;
darran	1:4218cacaf696	227
darran	1:4218cacaf696	228	/**
darran	1:4218cacaf696	229	* Initialize the resolver: set up the UDP pcb and configure the default server
darran	1:4218cacaf696	230	* (DNS_SERVER_ADDRESS).
darran	1:4218cacaf696	231	*/
darran	1:4218cacaf696	232	void
darran	1:4218cacaf696	233	dns_init()
darran	1:4218cacaf696	234	{
darran	1:4218cacaf696	235	ip_addr_t dnsserver;
darran	1:4218cacaf696	236
darran	1:4218cacaf696	237	dns_payload = (u8_t *)LWIP_MEM_ALIGN(dns_payload_buffer);
darran	1:4218cacaf696	238
darran	1:4218cacaf696	239	/* initialize default DNS server address */
darran	1:4218cacaf696	240	DNS_SERVER_ADDRESS(&dnsserver);
darran	1:4218cacaf696	241
darran	1:4218cacaf696	242	LWIP_DEBUGF(DNS_DEBUG, ("dns_init: initializing\n"));
darran	1:4218cacaf696	243
darran	1:4218cacaf696	244	/* if dns client not yet initialized... */
darran	1:4218cacaf696	245	if (dns_pcb == NULL) {
darran	1:4218cacaf696	246	dns_pcb = udp_new();
darran	1:4218cacaf696	247
darran	1:4218cacaf696	248	if (dns_pcb != NULL) {
darran	1:4218cacaf696	249	/* initialize DNS table not needed (initialized to zero since it is a
darran	1:4218cacaf696	250	* global variable) */
darran	1:4218cacaf696	251	LWIP_ASSERT("For implicit initialization to work, DNS_STATE_UNUSED needs to be 0",
darran	1:4218cacaf696	252	DNS_STATE_UNUSED == 0);
darran	1:4218cacaf696	253
darran	1:4218cacaf696	254	/* initialize DNS client */
darran	1:4218cacaf696	255	udp_bind(dns_pcb, IP_ADDR_ANY, 0);
darran	1:4218cacaf696	256	udp_recv(dns_pcb, dns_recv, NULL);
darran	1:4218cacaf696	257
darran	1:4218cacaf696	258	/* initialize default DNS primary server */
darran	1:4218cacaf696	259	dns_setserver(0, &dnsserver);
darran	1:4218cacaf696	260	}
darran	1:4218cacaf696	261	}
darran	1:4218cacaf696	262	#if DNS_LOCAL_HOSTLIST
darran	1:4218cacaf696	263	dns_init_local();
darran	1:4218cacaf696	264	#endif
darran	1:4218cacaf696	265	}
darran	1:4218cacaf696	266
darran	1:4218cacaf696	267	/**
darran	1:4218cacaf696	268	* Initialize one of the DNS servers.
darran	1:4218cacaf696	269	*
darran	1:4218cacaf696	270	* @param numdns the index of the DNS server to set must be < DNS_MAX_SERVERS
darran	1:4218cacaf696	271	* @param dnsserver IP address of the DNS server to set
darran	1:4218cacaf696	272	*/
darran	1:4218cacaf696	273	void
darran	1:4218cacaf696	274	dns_setserver(u8_t numdns, ip_addr_t *dnsserver)
darran	1:4218cacaf696	275	{
darran	1:4218cacaf696	276	if ((numdns < DNS_MAX_SERVERS) && (dns_pcb != NULL) &&
darran	1:4218cacaf696	277	(dnsserver != NULL) && !ip_addr_isany(dnsserver)) {
darran	1:4218cacaf696	278	dns_servers[numdns] = (*dnsserver);
darran	1:4218cacaf696	279	}
darran	1:4218cacaf696	280	}
darran	1:4218cacaf696	281
darran	1:4218cacaf696	282	/**
darran	1:4218cacaf696	283	* Obtain one of the currently configured DNS server.
darran	1:4218cacaf696	284	*
darran	1:4218cacaf696	285	* @param numdns the index of the DNS server
darran	1:4218cacaf696	286	* @return IP address of the indexed DNS server or "ip_addr_any" if the DNS
darran	1:4218cacaf696	287	* server has not been configured.
darran	1:4218cacaf696	288	*/
darran	1:4218cacaf696	289	ip_addr_t
darran	1:4218cacaf696	290	dns_getserver(u8_t numdns)
darran	1:4218cacaf696	291	{
darran	1:4218cacaf696	292	if (numdns < DNS_MAX_SERVERS) {
darran	1:4218cacaf696	293	return dns_servers[numdns];
darran	1:4218cacaf696	294	} else {
darran	1:4218cacaf696	295	return *IP_ADDR_ANY;
darran	1:4218cacaf696	296	}
darran	1:4218cacaf696	297	}
darran	1:4218cacaf696	298
darran	1:4218cacaf696	299	/**
darran	1:4218cacaf696	300	* The DNS resolver client timer - handle retries and timeouts and should
darran	1:4218cacaf696	301	* be called every DNS_TMR_INTERVAL milliseconds (every second by default).
darran	1:4218cacaf696	302	*/
darran	1:4218cacaf696	303	void
darran	1:4218cacaf696	304	dns_tmr(void)
darran	1:4218cacaf696	305	{
darran	1:4218cacaf696	306	if (dns_pcb != NULL) {
darran	1:4218cacaf696	307	LWIP_DEBUGF(DNS_DEBUG, ("dns_tmr: dns_check_entries\n"));
darran	1:4218cacaf696	308	dns_check_entries();
darran	1:4218cacaf696	309	}
darran	1:4218cacaf696	310	}
darran	1:4218cacaf696	311
darran	1:4218cacaf696	312	#if DNS_LOCAL_HOSTLIST
darran	1:4218cacaf696	313	static void
darran	1:4218cacaf696	314	dns_init_local()
darran	1:4218cacaf696	315	{
darran	1:4218cacaf696	316	#if DNS_LOCAL_HOSTLIST_IS_DYNAMIC && defined(DNS_LOCAL_HOSTLIST_INIT)
darran	1:4218cacaf696	317	int i;
darran	1:4218cacaf696	318	struct local_hostlist_entry *entry;
darran	1:4218cacaf696	319	/* Dynamic: copy entries from DNS_LOCAL_HOSTLIST_INIT to list */
darran	1:4218cacaf696	320	struct local_hostlist_entry local_hostlist_init[] = DNS_LOCAL_HOSTLIST_INIT;
darran	1:4218cacaf696	321	size_t namelen;
darran	1:4218cacaf696	322	for (i = 0; i < sizeof(local_hostlist_init) / sizeof(struct local_hostlist_entry); i++) {
darran	1:4218cacaf696	323	struct local_hostlist_entry *init_entry = &local_hostlist_init[i];
darran	1:4218cacaf696	324	LWIP_ASSERT("invalid host name (NULL)", init_entry->name != NULL);
darran	1:4218cacaf696	325	namelen = strlen(init_entry->name);
darran	1:4218cacaf696	326	LWIP_ASSERT("namelen <= DNS_LOCAL_HOSTLIST_MAX_NAMELEN", namelen <= DNS_LOCAL_HOSTLIST_MAX_NAMELEN);
darran	1:4218cacaf696	327	entry = (struct local_hostlist_entry *)memp_malloc(MEMP_LOCALHOSTLIST);
darran	1:4218cacaf696	328	LWIP_ASSERT("mem-error in dns_init_local", entry != NULL);
darran	1:4218cacaf696	329	if (entry != NULL) {
darran	1:4218cacaf696	330	entry->name = (char*)entry + sizeof(struct local_hostlist_entry);
darran	1:4218cacaf696	331	MEMCPY((char*)entry->name, init_entry->name, namelen);
darran	1:4218cacaf696	332	((char*)entry->name)[namelen] = 0;
darran	1:4218cacaf696	333	entry->addr = init_entry->addr;
darran	1:4218cacaf696	334	entry->next = local_hostlist_dynamic;
darran	1:4218cacaf696	335	local_hostlist_dynamic = entry;
darran	1:4218cacaf696	336	}
darran	1:4218cacaf696	337	}
darran	1:4218cacaf696	338	#endif /* DNS_LOCAL_HOSTLIST_IS_DYNAMIC && defined(DNS_LOCAL_HOSTLIST_INIT) */
darran	1:4218cacaf696	339	}
darran	1:4218cacaf696	340
darran	1:4218cacaf696	341	/**
darran	1:4218cacaf696	342	* Scans the local host-list for a hostname.
darran	1:4218cacaf696	343	*
darran	1:4218cacaf696	344	* @param hostname Hostname to look for in the local host-list
darran	1:4218cacaf696	345	* @return The first IP address for the hostname in the local host-list or
darran	1:4218cacaf696	346	* IPADDR_NONE if not found.
darran	1:4218cacaf696	347	*/
darran	1:4218cacaf696	348	static u32_t
darran	1:4218cacaf696	349	dns_lookup_local(const char *hostname)
darran	1:4218cacaf696	350	{
darran	1:4218cacaf696	351	#if DNS_LOCAL_HOSTLIST_IS_DYNAMIC
darran	1:4218cacaf696	352	struct local_hostlist_entry *entry = local_hostlist_dynamic;
darran	1:4218cacaf696	353	while(entry != NULL) {
darran	1:4218cacaf696	354	if(strcmp(entry->name, hostname) == 0) {
darran	1:4218cacaf696	355	return ip4_addr_get_u32(&entry->addr);
darran	1:4218cacaf696	356	}
darran	1:4218cacaf696	357	entry = entry->next;
darran	1:4218cacaf696	358	}
darran	1:4218cacaf696	359	#else /* DNS_LOCAL_HOSTLIST_IS_DYNAMIC */
darran	1:4218cacaf696	360	int i;
darran	1:4218cacaf696	361	for (i = 0; i < sizeof(local_hostlist_static) / sizeof(struct local_hostlist_entry); i++) {
darran	1:4218cacaf696	362	if(strcmp(local_hostlist_static[i].name, hostname) == 0) {
darran	1:4218cacaf696	363	return ip4_addr_get_u32(&local_hostlist_static[i].addr);
darran	1:4218cacaf696	364	}
darran	1:4218cacaf696	365	}
darran	1:4218cacaf696	366	#endif /* DNS_LOCAL_HOSTLIST_IS_DYNAMIC */
darran	1:4218cacaf696	367	return IPADDR_NONE;
darran	1:4218cacaf696	368	}
darran	1:4218cacaf696	369
darran	1:4218cacaf696	370	#if DNS_LOCAL_HOSTLIST_IS_DYNAMIC
darran	1:4218cacaf696	371	/** Remove all entries from the local host-list for a specific hostname
darran	1:4218cacaf696	372	* and/or IP addess
darran	1:4218cacaf696	373	*
darran	1:4218cacaf696	374	* @param hostname hostname for which entries shall be removed from the local
darran	1:4218cacaf696	375	* host-list
darran	1:4218cacaf696	376	* @param addr address for which entries shall be removed from the local host-list
darran	1:4218cacaf696	377	* @return the number of removed entries
darran	1:4218cacaf696	378	*/
darran	1:4218cacaf696	379	int
darran	1:4218cacaf696	380	dns_local_removehost(const char hostname, const ip_addr_t addr)
darran	1:4218cacaf696	381	{
darran	1:4218cacaf696	382	int removed = 0;
darran	1:4218cacaf696	383	struct local_hostlist_entry *entry = local_hostlist_dynamic;
darran	1:4218cacaf696	384	struct local_hostlist_entry *last_entry = NULL;
darran	1:4218cacaf696	385	while (entry != NULL) {
darran	1:4218cacaf696	386	if (((hostname == NULL) \|\| !strcmp(entry->name, hostname)) &&
darran	1:4218cacaf696	387	((addr == NULL) \|\| ip_addr_cmp(&entry->addr, addr))) {
darran	1:4218cacaf696	388	struct local_hostlist_entry *free_entry;
darran	1:4218cacaf696	389	if (last_entry != NULL) {
darran	1:4218cacaf696	390	last_entry->next = entry->next;
darran	1:4218cacaf696	391	} else {
darran	1:4218cacaf696	392	local_hostlist_dynamic = entry->next;
darran	1:4218cacaf696	393	}
darran	1:4218cacaf696	394	free_entry = entry;
darran	1:4218cacaf696	395	entry = entry->next;
darran	1:4218cacaf696	396	memp_free(MEMP_LOCALHOSTLIST, free_entry);
darran	1:4218cacaf696	397	removed++;
darran	1:4218cacaf696	398	} else {
darran	1:4218cacaf696	399	last_entry = entry;
darran	1:4218cacaf696	400	entry = entry->next;
darran	1:4218cacaf696	401	}
darran	1:4218cacaf696	402	}
darran	1:4218cacaf696	403	return removed;
darran	1:4218cacaf696	404	}
darran	1:4218cacaf696	405
darran	1:4218cacaf696	406	/**
darran	1:4218cacaf696	407	* Add a hostname/IP address pair to the local host-list.
darran	1:4218cacaf696	408	* Duplicates are not checked.
darran	1:4218cacaf696	409	*
darran	1:4218cacaf696	410	* @param hostname hostname of the new entry
darran	1:4218cacaf696	411	* @param addr IP address of the new entry
darran	1:4218cacaf696	412	* @return ERR_OK if succeeded or ERR_MEM on memory error
darran	1:4218cacaf696	413	*/
darran	1:4218cacaf696	414	err_t
darran	1:4218cacaf696	415	dns_local_addhost(const char hostname, const ip_addr_t addr)
darran	1:4218cacaf696	416	{
darran	1:4218cacaf696	417	struct local_hostlist_entry *entry;
darran	1:4218cacaf696	418	size_t namelen;
darran	1:4218cacaf696	419	LWIP_ASSERT("invalid host name (NULL)", hostname != NULL);
darran	1:4218cacaf696	420	namelen = strlen(hostname);
darran	1:4218cacaf696	421	LWIP_ASSERT("namelen <= DNS_LOCAL_HOSTLIST_MAX_NAMELEN", namelen <= DNS_LOCAL_HOSTLIST_MAX_NAMELEN);
darran	1:4218cacaf696	422	entry = (struct local_hostlist_entry *)memp_malloc(MEMP_LOCALHOSTLIST);
darran	1:4218cacaf696	423	if (entry == NULL) {
darran	1:4218cacaf696	424	return ERR_MEM;
darran	1:4218cacaf696	425	}
darran	1:4218cacaf696	426	entry->name = (char*)entry + sizeof(struct local_hostlist_entry);
darran	1:4218cacaf696	427	MEMCPY((char*)entry->name, hostname, namelen);
darran	1:4218cacaf696	428	((char*)entry->name)[namelen] = 0;
darran	1:4218cacaf696	429	ip_addr_copy(entry->addr, *addr);
darran	1:4218cacaf696	430	entry->next = local_hostlist_dynamic;
darran	1:4218cacaf696	431	local_hostlist_dynamic = entry;
darran	1:4218cacaf696	432	return ERR_OK;
darran	1:4218cacaf696	433	}
darran	1:4218cacaf696	434	#endif /* DNS_LOCAL_HOSTLIST_IS_DYNAMIC*/
darran	1:4218cacaf696	435	#endif /* DNS_LOCAL_HOSTLIST */
darran	1:4218cacaf696	436
darran	1:4218cacaf696	437	/**
darran	1:4218cacaf696	438	* Look up a hostname in the array of known hostnames.
darran	1:4218cacaf696	439	*
darran	1:4218cacaf696	440	* @note This function only looks in the internal array of known
darran	1:4218cacaf696	441	* hostnames, it does not send out a query for the hostname if none
darran	1:4218cacaf696	442	* was found. The function dns_enqueue() can be used to send a query
darran	1:4218cacaf696	443	* for a hostname.
darran	1:4218cacaf696	444	*
darran	1:4218cacaf696	445	* @param name the hostname to look up
darran	1:4218cacaf696	446	* @return the hostname's IP address, as u32_t (instead of ip_addr_t to
darran	1:4218cacaf696	447	* better check for failure: != IPADDR_NONE) or IPADDR_NONE if the hostname
darran	1:4218cacaf696	448	* was not found in the cached dns_table.
darran	1:4218cacaf696	449	*/
darran	1:4218cacaf696	450	static u32_t
darran	1:4218cacaf696	451	dns_lookup(const char *name)
darran	1:4218cacaf696	452	{
darran	1:4218cacaf696	453	u8_t i;
darran	1:4218cacaf696	454	#if DNS_LOCAL_HOSTLIST \|\| defined(DNS_LOOKUP_LOCAL_EXTERN)
darran	1:4218cacaf696	455	u32_t addr;
darran	1:4218cacaf696	456	#endif /* DNS_LOCAL_HOSTLIST \|\| defined(DNS_LOOKUP_LOCAL_EXTERN) */
darran	1:4218cacaf696	457	#if DNS_LOCAL_HOSTLIST
darran	1:4218cacaf696	458	if ((addr = dns_lookup_local(name)) != IPADDR_NONE) {
darran	1:4218cacaf696	459	return addr;
darran	1:4218cacaf696	460	}
darran	1:4218cacaf696	461	#endif /* DNS_LOCAL_HOSTLIST */
darran	1:4218cacaf696	462	#ifdef DNS_LOOKUP_LOCAL_EXTERN
darran	1:4218cacaf696	463	if((addr = DNS_LOOKUP_LOCAL_EXTERN(name)) != IPADDR_NONE) {
darran	1:4218cacaf696	464	return addr;
darran	1:4218cacaf696	465	}
darran	1:4218cacaf696	466	#endif /* DNS_LOOKUP_LOCAL_EXTERN */
darran	1:4218cacaf696	467
darran	1:4218cacaf696	468	/* Walk through name list, return entry if found. If not, return NULL. */
darran	1:4218cacaf696	469	for (i = 0; i < DNS_TABLE_SIZE; ++i) {
darran	1:4218cacaf696	470	if ((dns_table[i].state == DNS_STATE_DONE) &&
darran	1:4218cacaf696	471	(strcmp(name, dns_table[i].name) == 0)) {
darran	1:4218cacaf696	472	LWIP_DEBUGF(DNS_DEBUG, ("dns_lookup: \"%s\": found = ", name));
darran	1:4218cacaf696	473	ip_addr_debug_print(DNS_DEBUG, &(dns_table[i].ipaddr));
darran	1:4218cacaf696	474	LWIP_DEBUGF(DNS_DEBUG, ("\n"));
darran	1:4218cacaf696	475	return ip4_addr_get_u32(&dns_table[i].ipaddr);
darran	1:4218cacaf696	476	}
darran	1:4218cacaf696	477	}
darran	1:4218cacaf696	478
darran	1:4218cacaf696	479	return IPADDR_NONE;
darran	1:4218cacaf696	480	}
darran	1:4218cacaf696	481
darran	1:4218cacaf696	482	#if DNS_DOES_NAME_CHECK
darran	1:4218cacaf696	483	/**
darran	1:4218cacaf696	484	* Compare the "dotted" name "query" with the encoded name "response"
darran	1:4218cacaf696	485	* to make sure an answer from the DNS server matches the current dns_table
darran	1:4218cacaf696	486	* entry (otherwise, answers might arrive late for hostname not on the list
darran	1:4218cacaf696	487	* any more).
darran	1:4218cacaf696	488	*
darran	1:4218cacaf696	489	* @param query hostname (not encoded) from the dns_table
darran	1:4218cacaf696	490	* @param response encoded hostname in the DNS response
darran	1:4218cacaf696	491	* @return 0: names equal; 1: names differ
darran	1:4218cacaf696	492	*/
darran	1:4218cacaf696	493	static u8_t
darran	1:4218cacaf696	494	dns_compare_name(unsigned char query, unsigned char response)
darran	1:4218cacaf696	495	{
darran	1:4218cacaf696	496	unsigned char n;
darran	1:4218cacaf696	497
darran	1:4218cacaf696	498	do {
darran	1:4218cacaf696	499	n = *response++;
darran	1:4218cacaf696	500	/** @see RFC 1035 - 4.1.4. Message compression */
darran	1:4218cacaf696	501	if ((n & 0xc0) == 0xc0) {
darran	1:4218cacaf696	502	/* Compressed name */
darran	1:4218cacaf696	503	break;
darran	1:4218cacaf696	504	} else {
darran	1:4218cacaf696	505	/* Not compressed name */
darran	1:4218cacaf696	506	while (n > 0) {
darran	1:4218cacaf696	507	if ((query) != (response)) {
darran	1:4218cacaf696	508	return 1;
darran	1:4218cacaf696	509	}
darran	1:4218cacaf696	510	++response;
darran	1:4218cacaf696	511	++query;
darran	1:4218cacaf696	512	--n;
darran	1:4218cacaf696	513	};
darran	1:4218cacaf696	514	++query;
darran	1:4218cacaf696	515	}
darran	1:4218cacaf696	516	} while (*response != 0);
darran	1:4218cacaf696	517
darran	1:4218cacaf696	518	return 0;
darran	1:4218cacaf696	519	}
darran	1:4218cacaf696	520	#endif /* DNS_DOES_NAME_CHECK */
darran	1:4218cacaf696	521
darran	1:4218cacaf696	522	/**
darran	1:4218cacaf696	523	* Walk through a compact encoded DNS name and return the end of the name.
darran	1:4218cacaf696	524	*
darran	1:4218cacaf696	525	* @param query encoded DNS name in the DNS server response
darran	1:4218cacaf696	526	* @return end of the name
darran	1:4218cacaf696	527	*/
darran	1:4218cacaf696	528	static unsigned char *
darran	1:4218cacaf696	529	dns_parse_name(unsigned char *query)
darran	1:4218cacaf696	530	{
darran	1:4218cacaf696	531	unsigned char n;
darran	1:4218cacaf696	532
darran	1:4218cacaf696	533	do {
darran	1:4218cacaf696	534	n = *query++;
darran	1:4218cacaf696	535	/** @see RFC 1035 - 4.1.4. Message compression */
darran	1:4218cacaf696	536	if ((n & 0xc0) == 0xc0) {
darran	1:4218cacaf696	537	/* Compressed name */
darran	1:4218cacaf696	538	break;
darran	1:4218cacaf696	539	} else {
darran	1:4218cacaf696	540	/* Not compressed name */
darran	1:4218cacaf696	541	while (n > 0) {
darran	1:4218cacaf696	542	++query;
darran	1:4218cacaf696	543	--n;
darran	1:4218cacaf696	544	};
darran	1:4218cacaf696	545	}
darran	1:4218cacaf696	546	} while (*query != 0);
darran	1:4218cacaf696	547
darran	1:4218cacaf696	548	return query + 1;
darran	1:4218cacaf696	549	}
darran	1:4218cacaf696	550
darran	1:4218cacaf696	551	/**
darran	1:4218cacaf696	552	* Send a DNS query packet.
darran	1:4218cacaf696	553	*
darran	1:4218cacaf696	554	* @param numdns index of the DNS server in the dns_servers table
darran	1:4218cacaf696	555	* @param name hostname to query
darran	1:4218cacaf696	556	* @param id index of the hostname in dns_table, used as transaction ID in the
darran	1:4218cacaf696	557	* DNS query packet
darran	1:4218cacaf696	558	* @return ERR_OK if packet is sent; an err_t indicating the problem otherwise
darran	1:4218cacaf696	559	*/
darran	1:4218cacaf696	560	static err_t
darran	1:4218cacaf696	561	dns_send(u8_t numdns, const char* name, u8_t id)
darran	1:4218cacaf696	562	{
darran	1:4218cacaf696	563	err_t err;
darran	1:4218cacaf696	564	struct dns_hdr *hdr;
darran	1:4218cacaf696	565	struct dns_query qry;
darran	1:4218cacaf696	566	struct pbuf *p;
darran	1:4218cacaf696	567	char query, nptr;
darran	1:4218cacaf696	568	const char *pHostname;
darran	1:4218cacaf696	569	u8_t n;
darran	1:4218cacaf696	570
darran	1:4218cacaf696	571	LWIP_DEBUGF(DNS_DEBUG, ("dns_send: dns_servers[%"U16_F"] \"%s\": request\n",
darran	1:4218cacaf696	572	(u16_t)(numdns), name));
darran	1:4218cacaf696	573	LWIP_ASSERT("dns server out of array", numdns < DNS_MAX_SERVERS);
darran	1:4218cacaf696	574	LWIP_ASSERT("dns server has no IP address set", !ip_addr_isany(&dns_servers[numdns]));
darran	1:4218cacaf696	575
darran	1:4218cacaf696	576	/* if here, we have either a new query or a retry on a previous query to process */
darran	1:4218cacaf696	577	p = pbuf_alloc(PBUF_TRANSPORT, SIZEOF_DNS_HDR + DNS_MAX_NAME_LENGTH +
darran	1:4218cacaf696	578	SIZEOF_DNS_QUERY, PBUF_RAM);
darran	1:4218cacaf696	579	if (p != NULL) {
darran	1:4218cacaf696	580	LWIP_ASSERT("pbuf must be in one piece", p->next == NULL);
darran	1:4218cacaf696	581	/* fill dns header */
darran	1:4218cacaf696	582	hdr = (struct dns_hdr*)p->payload;
darran	1:4218cacaf696	583	memset(hdr, 0, SIZEOF_DNS_HDR);
darran	1:4218cacaf696	584	hdr->id = htons(id);
darran	1:4218cacaf696	585	hdr->flags1 = DNS_FLAG1_RD;
darran	1:4218cacaf696	586	hdr->numquestions = PP_HTONS(1);
darran	1:4218cacaf696	587	query = (char*)hdr + SIZEOF_DNS_HDR;
darran	1:4218cacaf696	588	pHostname = name;
darran	1:4218cacaf696	589	--pHostname;
darran	1:4218cacaf696	590
darran	1:4218cacaf696	591	/* convert hostname into suitable query format. */
darran	1:4218cacaf696	592	do {
darran	1:4218cacaf696	593	++pHostname;
darran	1:4218cacaf696	594	nptr = query;
darran	1:4218cacaf696	595	++query;
darran	1:4218cacaf696	596	for(n = 0; pHostname != '.' && pHostname != 0; ++pHostname) {
darran	1:4218cacaf696	597	query = pHostname;
darran	1:4218cacaf696	598	++query;
darran	1:4218cacaf696	599	++n;
darran	1:4218cacaf696	600	}
darran	1:4218cacaf696	601	*nptr = n;
darran	1:4218cacaf696	602	} while(*pHostname != 0);
darran	1:4218cacaf696	603	*query++='\0';
darran	1:4218cacaf696	604
darran	1:4218cacaf696	605	/* fill dns query */
darran	1:4218cacaf696	606	qry.type = PP_HTONS(DNS_RRTYPE_A);
darran	1:4218cacaf696	607	qry.cls = PP_HTONS(DNS_RRCLASS_IN);
darran	1:4218cacaf696	608	SMEMCPY(query, &qry, SIZEOF_DNS_QUERY);
darran	1:4218cacaf696	609
darran	1:4218cacaf696	610	/* resize pbuf to the exact dns query */
darran	1:4218cacaf696	611	pbuf_realloc(p, (u16_t)((query + SIZEOF_DNS_QUERY) - ((char*)(p->payload))));
darran	1:4218cacaf696	612
darran	1:4218cacaf696	613	/* connect to the server for faster receiving */
darran	1:4218cacaf696	614	udp_connect(dns_pcb, &dns_servers[numdns], DNS_SERVER_PORT);
darran	1:4218cacaf696	615	/* send dns packet */
darran	1:4218cacaf696	616	err = udp_sendto(dns_pcb, p, &dns_servers[numdns], DNS_SERVER_PORT);
darran	1:4218cacaf696	617
darran	1:4218cacaf696	618	/* free pbuf */
darran	1:4218cacaf696	619	pbuf_free(p);
darran	1:4218cacaf696	620	} else {
darran	1:4218cacaf696	621	err = ERR_MEM;
darran	1:4218cacaf696	622	}
darran	1:4218cacaf696	623
darran	1:4218cacaf696	624	return err;
darran	1:4218cacaf696	625	}
darran	1:4218cacaf696	626
darran	1:4218cacaf696	627	/**
darran	1:4218cacaf696	628	* dns_check_entry() - see if pEntry has not yet been queried and, if so, sends out a query.
darran	1:4218cacaf696	629	* Check an entry in the dns_table:
darran	1:4218cacaf696	630	* - send out query for new entries
darran	1:4218cacaf696	631	* - retry old pending entries on timeout (also with different servers)
darran	1:4218cacaf696	632	* - remove completed entries from the table if their TTL has expired
darran	1:4218cacaf696	633	*
darran	1:4218cacaf696	634	* @param i index of the dns_table entry to check
darran	1:4218cacaf696	635	*/
darran	1:4218cacaf696	636	static void
darran	1:4218cacaf696	637	dns_check_entry(u8_t i)
darran	1:4218cacaf696	638	{
darran	1:4218cacaf696	639	err_t err;
darran	1:4218cacaf696	640	struct dns_table_entry *pEntry = &dns_table[i];
darran	1:4218cacaf696	641
darran	1:4218cacaf696	642	LWIP_ASSERT("array index out of bounds", i < DNS_TABLE_SIZE);
darran	1:4218cacaf696	643
darran	1:4218cacaf696	644	switch(pEntry->state) {
darran	1:4218cacaf696	645
darran	1:4218cacaf696	646	case DNS_STATE_NEW: {
darran	1:4218cacaf696	647	/* initialize new entry */
darran	1:4218cacaf696	648	pEntry->state = DNS_STATE_ASKING;
darran	1:4218cacaf696	649	pEntry->numdns = 0;
darran	1:4218cacaf696	650	pEntry->tmr = 1;
darran	1:4218cacaf696	651	pEntry->retries = 0;
darran	1:4218cacaf696	652
darran	1:4218cacaf696	653	/* send DNS packet for this entry */
darran	1:4218cacaf696	654	err = dns_send(pEntry->numdns, pEntry->name, i);
darran	1:4218cacaf696	655	if (err != ERR_OK) {
darran	1:4218cacaf696	656	LWIP_DEBUGF(DNS_DEBUG \| LWIP_DBG_LEVEL_WARNING,
darran	1:4218cacaf696	657	("dns_send returned error: %d\n", err));
darran	1:4218cacaf696	658	}
darran	1:4218cacaf696	659	break;
darran	1:4218cacaf696	660	}
darran	1:4218cacaf696	661
darran	1:4218cacaf696	662	case DNS_STATE_ASKING: {
darran	1:4218cacaf696	663	if (--pEntry->tmr == 0) {
darran	1:4218cacaf696	664	if (++pEntry->retries == DNS_MAX_RETRIES) {
darran	1:4218cacaf696	665	if ((pEntry->numdns+1<DNS_MAX_SERVERS) && !ip_addr_isany(&dns_servers[pEntry->numdns+1])) {
darran	1:4218cacaf696	666	/* change of server */
darran	1:4218cacaf696	667	pEntry->numdns++;
darran	1:4218cacaf696	668	pEntry->tmr = 1;
darran	1:4218cacaf696	669	pEntry->retries = 0;
darran	1:4218cacaf696	670	break;
darran	1:4218cacaf696	671	} else {
darran	1:4218cacaf696	672	LWIP_DEBUGF(DNS_DEBUG, ("dns_check_entry: \"%s\": timeout\n", pEntry->name));
darran	1:4218cacaf696	673	/* call specified callback function if provided */
darran	1:4218cacaf696	674	if (pEntry->found)
darran	1:4218cacaf696	675	(*pEntry->found)(pEntry->name, NULL, pEntry->arg);
darran	1:4218cacaf696	676	/* flush this entry */
darran	1:4218cacaf696	677	pEntry->state = DNS_STATE_UNUSED;
darran	1:4218cacaf696	678	pEntry->found = NULL;
darran	1:4218cacaf696	679	break;
darran	1:4218cacaf696	680	}
darran	1:4218cacaf696	681	}
darran	1:4218cacaf696	682
darran	1:4218cacaf696	683	/* wait longer for the next retry */
darran	1:4218cacaf696	684	pEntry->tmr = pEntry->retries;
darran	1:4218cacaf696	685
darran	1:4218cacaf696	686	/* send DNS packet for this entry */
darran	1:4218cacaf696	687	err = dns_send(pEntry->numdns, pEntry->name, i);
darran	1:4218cacaf696	688	if (err != ERR_OK) {
darran	1:4218cacaf696	689	LWIP_DEBUGF(DNS_DEBUG \| LWIP_DBG_LEVEL_WARNING,
darran	1:4218cacaf696	690	("dns_send returned error: %d\n", err));
darran	1:4218cacaf696	691	}
darran	1:4218cacaf696	692	}
darran	1:4218cacaf696	693	break;
darran	1:4218cacaf696	694	}
darran	1:4218cacaf696	695
darran	1:4218cacaf696	696	case DNS_STATE_DONE: {
darran	1:4218cacaf696	697	/* if the time to live is nul */
darran	1:4218cacaf696	698	if (--pEntry->ttl == 0) {
darran	1:4218cacaf696	699	LWIP_DEBUGF(DNS_DEBUG, ("dns_check_entry: \"%s\": flush\n", pEntry->name));
darran	1:4218cacaf696	700	/* flush this entry */
darran	1:4218cacaf696	701	pEntry->state = DNS_STATE_UNUSED;
darran	1:4218cacaf696	702	pEntry->found = NULL;
darran	1:4218cacaf696	703	}
darran	1:4218cacaf696	704	break;
darran	1:4218cacaf696	705	}
darran	1:4218cacaf696	706	case DNS_STATE_UNUSED:
darran	1:4218cacaf696	707	/* nothing to do */
darran	1:4218cacaf696	708	break;
darran	1:4218cacaf696	709	default:
darran	1:4218cacaf696	710	LWIP_ASSERT("unknown dns_table entry state:", 0);
darran	1:4218cacaf696	711	break;
darran	1:4218cacaf696	712	}
darran	1:4218cacaf696	713	}
darran	1:4218cacaf696	714
darran	1:4218cacaf696	715	/**
darran	1:4218cacaf696	716	* Call dns_check_entry for each entry in dns_table - check all entries.
darran	1:4218cacaf696	717	*/
darran	1:4218cacaf696	718	static void
darran	1:4218cacaf696	719	dns_check_entries(void)
darran	1:4218cacaf696	720	{
darran	1:4218cacaf696	721	u8_t i;
darran	1:4218cacaf696	722
darran	1:4218cacaf696	723	for (i = 0; i < DNS_TABLE_SIZE; ++i) {
darran	1:4218cacaf696	724	dns_check_entry(i);
darran	1:4218cacaf696	725	}
darran	1:4218cacaf696	726	}
darran	1:4218cacaf696	727
darran	1:4218cacaf696	728	/**
darran	1:4218cacaf696	729	* Receive input function for DNS response packets arriving for the dns UDP pcb.
darran	1:4218cacaf696	730	*
darran	1:4218cacaf696	731	* @params see udp.h
darran	1:4218cacaf696	732	*/
darran	1:4218cacaf696	733	static void
darran	1:4218cacaf696	734	dns_recv(void arg, struct udp_pcb pcb, struct pbuf p, ip_addr_t addr, u16_t port)
darran	1:4218cacaf696	735	{
darran	1:4218cacaf696	736	u16_t i;
darran	1:4218cacaf696	737	char *pHostname;
darran	1:4218cacaf696	738	struct dns_hdr *hdr;
darran	1:4218cacaf696	739	struct dns_answer ans;
darran	1:4218cacaf696	740	struct dns_table_entry *pEntry;
darran	1:4218cacaf696	741	u16_t nquestions, nanswers;
darran	1:4218cacaf696	742
darran	1:4218cacaf696	743	LWIP_UNUSED_ARG(arg);
darran	1:4218cacaf696	744	LWIP_UNUSED_ARG(pcb);
darran	1:4218cacaf696	745	LWIP_UNUSED_ARG(addr);
darran	1:4218cacaf696	746	LWIP_UNUSED_ARG(port);
darran	1:4218cacaf696	747
darran	1:4218cacaf696	748	/* is the dns message too big ? */
darran	1:4218cacaf696	749	if (p->tot_len > DNS_MSG_SIZE) {
darran	1:4218cacaf696	750	LWIP_DEBUGF(DNS_DEBUG, ("dns_recv: pbuf too big\n"));
darran	1:4218cacaf696	751	/* free pbuf and return */
darran	1:4218cacaf696	752	goto memerr;
darran	1:4218cacaf696	753	}
darran	1:4218cacaf696	754
darran	1:4218cacaf696	755	/* is the dns message big enough ? */
darran	1:4218cacaf696	756	if (p->tot_len < (SIZEOF_DNS_HDR + SIZEOF_DNS_QUERY + SIZEOF_DNS_ANSWER)) {
darran	1:4218cacaf696	757	LWIP_DEBUGF(DNS_DEBUG, ("dns_recv: pbuf too small\n"));
darran	1:4218cacaf696	758	/* free pbuf and return */
darran	1:4218cacaf696	759	goto memerr;
darran	1:4218cacaf696	760	}
darran	1:4218cacaf696	761
darran	1:4218cacaf696	762	/* copy dns payload inside static buffer for processing */
darran	1:4218cacaf696	763	if (pbuf_copy_partial(p, dns_payload, p->tot_len, 0) == p->tot_len) {
darran	1:4218cacaf696	764	/* The ID in the DNS header should be our entry into the name table. */
darran	1:4218cacaf696	765	hdr = (struct dns_hdr*)dns_payload;
darran	1:4218cacaf696	766	i = htons(hdr->id);
darran	1:4218cacaf696	767	if (i < DNS_TABLE_SIZE) {
darran	1:4218cacaf696	768	pEntry = &dns_table[i];
darran	1:4218cacaf696	769	if(pEntry->state == DNS_STATE_ASKING) {
darran	1:4218cacaf696	770	/* This entry is now completed. */
darran	1:4218cacaf696	771	pEntry->state = DNS_STATE_DONE;
darran	1:4218cacaf696	772	pEntry->err = hdr->flags2 & DNS_FLAG2_ERR_MASK;
darran	1:4218cacaf696	773
darran	1:4218cacaf696	774	/* We only care about the question(s) and the answers. The authrr
darran	1:4218cacaf696	775	and the extrarr are simply discarded. */
darran	1:4218cacaf696	776	nquestions = htons(hdr->numquestions);
darran	1:4218cacaf696	777	nanswers = htons(hdr->numanswers);
darran	1:4218cacaf696	778
darran	1:4218cacaf696	779	/* Check for error. If so, call callback to inform. */
darran	1:4218cacaf696	780	if (((hdr->flags1 & DNS_FLAG1_RESPONSE) == 0) \|\| (pEntry->err != 0) \|\| (nquestions != 1)) {
darran	1:4218cacaf696	781	LWIP_DEBUGF(DNS_DEBUG, ("dns_recv: \"%s\": error in flags\n", pEntry->name));
darran	1:4218cacaf696	782	/* call callback to indicate error, clean up memory and return */
darran	1:4218cacaf696	783	goto responseerr;
darran	1:4218cacaf696	784	}
darran	1:4218cacaf696	785
darran	1:4218cacaf696	786	#if DNS_DOES_NAME_CHECK
darran	1:4218cacaf696	787	/* Check if the name in the "question" part match with the name in the entry. */
darran	1:4218cacaf696	788	if (dns_compare_name((unsigned char )(pEntry->name), (unsigned char )dns_payload + SIZEOF_DNS_HDR) != 0) {
darran	1:4218cacaf696	789	LWIP_DEBUGF(DNS_DEBUG, ("dns_recv: \"%s\": response not match to query\n", pEntry->name));
darran	1:4218cacaf696	790	/* call callback to indicate error, clean up memory and return */
darran	1:4218cacaf696	791	goto responseerr;
darran	1:4218cacaf696	792	}
darran	1:4218cacaf696	793	#endif /* DNS_DOES_NAME_CHECK */
darran	1:4218cacaf696	794
darran	1:4218cacaf696	795	/* Skip the name in the "question" part */
darran	1:4218cacaf696	796	pHostname = (char ) dns_parse_name((unsigned char )dns_payload + SIZEOF_DNS_HDR) + SIZEOF_DNS_QUERY;
darran	1:4218cacaf696	797
darran	1:4218cacaf696	798	while (nanswers > 0) {
darran	1:4218cacaf696	799	/* skip answer resource record's host name */
darran	1:4218cacaf696	800	pHostname = (char ) dns_parse_name((unsigned char )pHostname);
darran	1:4218cacaf696	801
darran	1:4218cacaf696	802	/* Check for IP address type and Internet class. Others are discarded. */
darran	1:4218cacaf696	803	SMEMCPY(&ans, pHostname, SIZEOF_DNS_ANSWER);
darran	1:4218cacaf696	804	if((ans.type == PP_HTONS(DNS_RRTYPE_A)) && (ans.cls == PP_HTONS(DNS_RRCLASS_IN)) &&
darran	1:4218cacaf696	805	(ans.len == PP_HTONS(sizeof(ip_addr_t))) ) {
darran	1:4218cacaf696	806	/* read the answer resource record's TTL, and maximize it if needed */
darran	1:4218cacaf696	807	pEntry->ttl = ntohl(ans.ttl);
darran	1:4218cacaf696	808	if (pEntry->ttl > DNS_MAX_TTL) {
darran	1:4218cacaf696	809	pEntry->ttl = DNS_MAX_TTL;
darran	1:4218cacaf696	810	}
darran	1:4218cacaf696	811	/* read the IP address after answer resource record's header */
darran	1:4218cacaf696	812	SMEMCPY(&(pEntry->ipaddr), (pHostname+SIZEOF_DNS_ANSWER), sizeof(ip_addr_t));
darran	1:4218cacaf696	813	LWIP_DEBUGF(DNS_DEBUG, ("dns_recv: \"%s\": response = ", pEntry->name));
darran	1:4218cacaf696	814	ip_addr_debug_print(DNS_DEBUG, (&(pEntry->ipaddr)));
darran	1:4218cacaf696	815	LWIP_DEBUGF(DNS_DEBUG, ("\n"));
darran	1:4218cacaf696	816	/* call specified callback function if provided */
darran	1:4218cacaf696	817	if (pEntry->found) {
darran	1:4218cacaf696	818	(*pEntry->found)(pEntry->name, &pEntry->ipaddr, pEntry->arg);
darran	1:4218cacaf696	819	}
darran	1:4218cacaf696	820	/* deallocate memory and return */
darran	1:4218cacaf696	821	goto memerr;
darran	1:4218cacaf696	822	} else {
darran	1:4218cacaf696	823	pHostname = pHostname + SIZEOF_DNS_ANSWER + htons(ans.len);
darran	1:4218cacaf696	824	}
darran	1:4218cacaf696	825	--nanswers;
darran	1:4218cacaf696	826	}
darran	1:4218cacaf696	827	LWIP_DEBUGF(DNS_DEBUG, ("dns_recv: \"%s\": error in response\n", pEntry->name));
darran	1:4218cacaf696	828	/* call callback to indicate error, clean up memory and return */
darran	1:4218cacaf696	829	goto responseerr;
darran	1:4218cacaf696	830	}
darran	1:4218cacaf696	831	}
darran	1:4218cacaf696	832	}
darran	1:4218cacaf696	833
darran	1:4218cacaf696	834	/* deallocate memory and return */
darran	1:4218cacaf696	835	goto memerr;
darran	1:4218cacaf696	836
darran	1:4218cacaf696	837	responseerr:
darran	1:4218cacaf696	838	/* ERROR: call specified callback function with NULL as name to indicate an error */
darran	1:4218cacaf696	839	if (pEntry->found) {
darran	1:4218cacaf696	840	(*pEntry->found)(pEntry->name, NULL, pEntry->arg);
darran	1:4218cacaf696	841	}
darran	1:4218cacaf696	842	/* flush this entry */
darran	1:4218cacaf696	843	pEntry->state = DNS_STATE_UNUSED;
darran	1:4218cacaf696	844	pEntry->found = NULL;
darran	1:4218cacaf696	845
darran	1:4218cacaf696	846	memerr:
darran	1:4218cacaf696	847	/* free pbuf */
darran	1:4218cacaf696	848	pbuf_free(p);
darran	1:4218cacaf696	849	return;
darran	1:4218cacaf696	850	}
darran	1:4218cacaf696	851
darran	1:4218cacaf696	852	/**
darran	1:4218cacaf696	853	* Queues a new hostname to resolve and sends out a DNS query for that hostname
darran	1:4218cacaf696	854	*
darran	1:4218cacaf696	855	* @param name the hostname that is to be queried
darran	1:4218cacaf696	856	* @param found a callback founction to be called on success, failure or timeout
darran	1:4218cacaf696	857	* @param callback_arg argument to pass to the callback function
darran	1:4218cacaf696	858	* @return @return a err_t return code.
darran	1:4218cacaf696	859	*/
darran	1:4218cacaf696	860	static err_t
darran	1:4218cacaf696	861	dns_enqueue(const char name, dns_found_callback found, void callback_arg)
darran	1:4218cacaf696	862	{
darran	1:4218cacaf696	863	u8_t i;
darran	1:4218cacaf696	864	u8_t lseq, lseqi;
darran	1:4218cacaf696	865	struct dns_table_entry *pEntry = NULL;
darran	1:4218cacaf696	866	size_t namelen;
darran	1:4218cacaf696	867
darran	1:4218cacaf696	868	/* search an unused entry, or the oldest one */
darran	1:4218cacaf696	869	lseq = lseqi = 0;
darran	1:4218cacaf696	870	for (i = 0; i < DNS_TABLE_SIZE; ++i) {
darran	1:4218cacaf696	871	pEntry = &dns_table[i];
darran	1:4218cacaf696	872	/* is it an unused entry ? */
darran	1:4218cacaf696	873	if (pEntry->state == DNS_STATE_UNUSED)
darran	1:4218cacaf696	874	break;
darran	1:4218cacaf696	875
darran	1:4218cacaf696	876	/* check if this is the oldest completed entry */
darran	1:4218cacaf696	877	if (pEntry->state == DNS_STATE_DONE) {
darran	1:4218cacaf696	878	if ((dns_seqno - pEntry->seqno) > lseq) {
darran	1:4218cacaf696	879	lseq = dns_seqno - pEntry->seqno;
darran	1:4218cacaf696	880	lseqi = i;
darran	1:4218cacaf696	881	}
darran	1:4218cacaf696	882	}
darran	1:4218cacaf696	883	}
darran	1:4218cacaf696	884
darran	1:4218cacaf696	885	/* if we don't have found an unused entry, use the oldest completed one */
darran	1:4218cacaf696	886	if (i == DNS_TABLE_SIZE) {
darran	1:4218cacaf696	887	if ((lseqi >= DNS_TABLE_SIZE) \|\| (dns_table[lseqi].state != DNS_STATE_DONE)) {
darran	1:4218cacaf696	888	/* no entry can't be used now, table is full */
darran	1:4218cacaf696	889	LWIP_DEBUGF(DNS_DEBUG, ("dns_enqueue: \"%s\": DNS entries table is full\n", name));
darran	1:4218cacaf696	890	return ERR_MEM;
darran	1:4218cacaf696	891	} else {
darran	1:4218cacaf696	892	/* use the oldest completed one */
darran	1:4218cacaf696	893	i = lseqi;
darran	1:4218cacaf696	894	pEntry = &dns_table[i];
darran	1:4218cacaf696	895	}
darran	1:4218cacaf696	896	}
darran	1:4218cacaf696	897
darran	1:4218cacaf696	898	/* use this entry */
darran	1:4218cacaf696	899	LWIP_DEBUGF(DNS_DEBUG, ("dns_enqueue: \"%s\": use DNS entry %"U16_F"\n", name, (u16_t)(i)));
darran	1:4218cacaf696	900
darran	1:4218cacaf696	901	/* fill the entry */
darran	1:4218cacaf696	902	pEntry->state = DNS_STATE_NEW;
darran	1:4218cacaf696	903	pEntry->seqno = dns_seqno++;
darran	1:4218cacaf696	904	pEntry->found = found;
darran	1:4218cacaf696	905	pEntry->arg = callback_arg;
darran	1:4218cacaf696	906	namelen = LWIP_MIN(strlen(name), DNS_MAX_NAME_LENGTH-1);
darran	1:4218cacaf696	907	MEMCPY(pEntry->name, name, namelen);
darran	1:4218cacaf696	908	pEntry->name[namelen] = 0;
darran	1:4218cacaf696	909
darran	1:4218cacaf696	910	/* force to send query without waiting timer */
darran	1:4218cacaf696	911	dns_check_entry(i);
darran	1:4218cacaf696	912
darran	1:4218cacaf696	913	/* dns query is enqueued */
darran	1:4218cacaf696	914	return ERR_INPROGRESS;
darran	1:4218cacaf696	915	}
darran	1:4218cacaf696	916
darran	1:4218cacaf696	917	/**
darran	1:4218cacaf696	918	* Resolve a hostname (string) into an IP address.
darran	1:4218cacaf696	919	* NON-BLOCKING callback version for use with raw API!!!
darran	1:4218cacaf696	920	*
darran	1:4218cacaf696	921	* Returns immediately with one of err_t return codes:
darran	1:4218cacaf696	922	* - ERR_OK if hostname is a valid IP address string or the host
darran	1:4218cacaf696	923	* name is already in the local names table.
darran	1:4218cacaf696	924	* - ERR_INPROGRESS enqueue a request to be sent to the DNS server
darran	1:4218cacaf696	925	* for resolution if no errors are present.
darran	1:4218cacaf696	926	*
darran	1:4218cacaf696	927	* @param hostname the hostname that is to be queried
darran	1:4218cacaf696	928	* @param addr pointer to a ip_addr_t where to store the address if it is already
darran	1:4218cacaf696	929	* cached in the dns_table (only valid if ERR_OK is returned!)
darran	1:4218cacaf696	930	* @param found a callback function to be called on success, failure or timeout (only if
darran	1:4218cacaf696	931	* ERR_INPROGRESS is returned!)
darran	1:4218cacaf696	932	* @param callback_arg argument to pass to the callback function
darran	1:4218cacaf696	933	* @return a err_t return code.
darran	1:4218cacaf696	934	*/
darran	1:4218cacaf696	935	err_t
darran	1:4218cacaf696	936	dns_gethostbyname(const char hostname, ip_addr_t addr, dns_found_callback found,
darran	1:4218cacaf696	937	void *callback_arg)
darran	1:4218cacaf696	938	{
darran	1:4218cacaf696	939	u32_t ipaddr;
darran	1:4218cacaf696	940	/* not initialized or no valid server yet, or invalid addr pointer
darran	1:4218cacaf696	941	* or invalid hostname or invalid hostname length */
darran	1:4218cacaf696	942	if ((dns_pcb == NULL) \|\| (addr == NULL) \|\|
darran	1:4218cacaf696	943	(!hostname) \|\| (!hostname[0]) \|\|
darran	1:4218cacaf696	944	(strlen(hostname) >= DNS_MAX_NAME_LENGTH)) {
darran	1:4218cacaf696	945	return ERR_VAL;
darran	1:4218cacaf696	946	}
darran	1:4218cacaf696	947
darran	1:4218cacaf696	948	#if LWIP_HAVE_LOOPIF
darran	1:4218cacaf696	949	if (strcmp(hostname, "localhost")==0) {
darran	1:4218cacaf696	950	ip_addr_set_loopback(addr);
darran	1:4218cacaf696	951	return ERR_OK;
darran	1:4218cacaf696	952	}
darran	1:4218cacaf696	953	#endif /* LWIP_HAVE_LOOPIF */
darran	1:4218cacaf696	954
darran	1:4218cacaf696	955	/* host name already in octet notation? set ip addr and return ERR_OK */
darran	1:4218cacaf696	956	ipaddr = ipaddr_addr(hostname);
darran	1:4218cacaf696	957	if (ipaddr == IPADDR_NONE) {
darran	1:4218cacaf696	958	/* already have this address cached? */
darran	1:4218cacaf696	959	ipaddr = dns_lookup(hostname);
darran	1:4218cacaf696	960	}
darran	1:4218cacaf696	961	if (ipaddr != IPADDR_NONE) {
darran	1:4218cacaf696	962	ip4_addr_set_u32(addr, ipaddr);
darran	1:4218cacaf696	963	return ERR_OK;
darran	1:4218cacaf696	964	}
darran	1:4218cacaf696	965
darran	1:4218cacaf696	966	/* queue query with specified callback */
darran	1:4218cacaf696	967	return dns_enqueue(hostname, found, callback_arg);
darran	1:4218cacaf696	968	}
darran	1:4218cacaf696	969
darran	1:4218cacaf696	970	#endif /* LWIP_DNS */

Repository toolbox

Export to desktop IDE

Build repository

Repository details

Type:	Program
Mbed OS support:	Mbed 2 deprecated
Created:	18 Jun 2010
Imports:	7
Forks:	0
Commits:	2
Dependents:	0
Dependencies:	1
Followers:	1

lwip/core/dns.c@1:4218cacaf696, 2010-06-18 (annotated)

Who changed what in which revision?

Repository toolbox

Repository details

Important Information for this Arm website

Access Warning