1 /** 2 * @file 3 * API functions for name resolving 4 * 5 */ 6 7 /* 8 * Redistribution and use in source and binary forms, with or without modification, 9 * are permitted provided that the following conditions are met: 10 * 11 * 1. Redistributions of source code must retain the above copyright notice, 12 * this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright notice, 14 * this list of conditions and the following disclaimer in the documentation 15 * and/or other materials provided with the distribution. 16 * 3. The name of the author may not be used to endorse or promote products 17 * derived from this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 20 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 21 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT 22 * SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 23 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT 24 * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 25 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 26 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING 27 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY 28 * OF SUCH DAMAGE. 29 * 30 * This file is part of the lwIP TCP/IP stack. 31 * 32 * Author: Simon Goldschmidt 33 * 34 */ 35 36 #include "lwip/netdb.h" 37 38 #if LWIP_DNS && LWIP_SOCKET 39 40 #include "lwip/err.h" 41 #include "lwip/mem.h" 42 #include "lwip/memp.h" 43 #include "lwip/ip_addr.h" 44 #include "lwip/api.h" 45 #include "lwip/dns.h" 46 47 #include <string.h> 48 #include <stdlib.h> 49 50 /** helper struct for gethostbyname_r to access the char* buffer */ 51 struct gethostbyname_r_helper { 52 ip_addr_t *addrs; 53 ip_addr_t addr; 54 char *aliases; 55 }; 56 57 /** h_errno is exported in netdb.h for access by applications. */ 58 #if LWIP_DNS_API_DECLARE_H_ERRNO 59 int h_errno; 60 #endif /* LWIP_DNS_API_DECLARE_H_ERRNO */ 61 62 /** define "hostent" variables storage: 0 if we use a static (but unprotected) 63 * set of variables for lwip_gethostbyname, 1 if we use a local storage */ 64 #ifndef LWIP_DNS_API_HOSTENT_STORAGE 65 #define LWIP_DNS_API_HOSTENT_STORAGE 0 66 #endif 67 68 /** define "hostent" variables storage */ 69 #if LWIP_DNS_API_HOSTENT_STORAGE 70 #define HOSTENT_STORAGE 71 #else 72 #define HOSTENT_STORAGE static 73 #endif /* LWIP_DNS_API_STATIC_HOSTENT */ 74 75 /** 76 * Returns an entry containing addresses of address family AF_INET 77 * for the host with name name. 78 * Due to dns_gethostbyname limitations, only one address is returned. 79 * 80 * @param name the hostname to resolve 81 * @return an entry containing addresses of address family AF_INET 82 * for the host with name name 83 */ 84 struct hostent* 85 lwip_gethostbyname(const char *name) 86 { 87 err_t err; 88 ip_addr_t addr; 89 90 /* buffer variables for lwip_gethostbyname() */ 91 HOSTENT_STORAGE struct hostent s_hostent; 92 HOSTENT_STORAGE char *s_aliases; 93 HOSTENT_STORAGE ip_addr_t s_hostent_addr; 94 HOSTENT_STORAGE ip_addr_t *s_phostent_addr[2]; 95 96 /* query host IP address */ 97 err = netconn_gethostbyname(name, &addr); 98 if (err != ERR_OK) { 99 LWIP_DEBUGF(DNS_DEBUG, ("lwip_gethostbyname(%s) failed, err=%d\n", name, err)); 100 h_errno = HOST_NOT_FOUND; 101 return NULL; 102 } 103 104 /* fill hostent */ 105 s_hostent_addr = addr; 106 s_phostent_addr[0] = &s_hostent_addr; 107 s_phostent_addr[1] = NULL; 108 s_hostent.h_name = (char*)name; 109 s_hostent.h_aliases = &s_aliases; 110 s_hostent.h_addrtype = AF_INET; 111 s_hostent.h_length = sizeof(ip_addr_t); 112 s_hostent.h_addr_list = (char**)&s_phostent_addr; 113 114 #if DNS_DEBUG 115 /* dump hostent */ 116 LWIP_DEBUGF(DNS_DEBUG, ("hostent.h_name == %s\n", s_hostent.h_name)); 117 LWIP_DEBUGF(DNS_DEBUG, ("hostent.h_aliases == %p\n", s_hostent.h_aliases)); 118 if (s_hostent.h_aliases != NULL) { 119 u8_t idx; 120 for ( idx=0; s_hostent.h_aliases[idx]; idx++) { 121 LWIP_DEBUGF(DNS_DEBUG, ("hostent.h_aliases[%i]-> == %p\n", idx, s_hostent.h_aliases[idx])); 122 LWIP_DEBUGF(DNS_DEBUG, ("hostent.h_aliases[%i]-> == %s\n", idx, s_hostent.h_aliases[idx])); 123 } 124 } 125 LWIP_DEBUGF(DNS_DEBUG, ("hostent.h_addrtype == %d\n", s_hostent.h_addrtype)); 126 LWIP_DEBUGF(DNS_DEBUG, ("hostent.h_length == %d\n", s_hostent.h_length)); 127 LWIP_DEBUGF(DNS_DEBUG, ("hostent.h_addr_list == %p\n", s_hostent.h_addr_list)); 128 if (s_hostent.h_addr_list != NULL) { 129 u8_t idx; 130 for ( idx=0; s_hostent.h_addr_list[idx]; idx++) { 131 LWIP_DEBUGF(DNS_DEBUG, ("hostent.h_addr_list[%i] == %p\n", idx, s_hostent.h_addr_list[idx])); 132 LWIP_DEBUGF(DNS_DEBUG, ("hostent.h_addr_list[%i]-> == %s\n", idx, ip_ntoa((ip_addr_t*)s_hostent.h_addr_list[idx]))); 133 } 134 } 135 #endif /* DNS_DEBUG */ 136 137 #if LWIP_DNS_API_HOSTENT_STORAGE 138 /* this function should return the "per-thread" hostent after copy from s_hostent */ 139 return sys_thread_hostent(&s_hostent); 140 #else 141 return &s_hostent; 142 #endif /* LWIP_DNS_API_HOSTENT_STORAGE */ 143 } 144 145 /** 146 * Thread-safe variant of lwip_gethostbyname: instead of using a static 147 * buffer, this function takes buffer and errno pointers as arguments 148 * and uses these for the result. 149 * 150 * @param name the hostname to resolve 151 * @param ret pre-allocated struct where to store the result 152 * @param buf pre-allocated buffer where to store additional data 153 * @param buflen the size of buf 154 * @param result pointer to a hostent pointer that is set to ret on success 155 * and set to zero on error 156 * @param h_errnop pointer to an int where to store errors (instead of modifying 157 * the global h_errno) 158 * @return 0 on success, non-zero on error, additional error information 159 * is stored in *h_errnop instead of h_errno to be thread-safe 160 */ 161 int 162 lwip_gethostbyname_r(const char *name, struct hostent *ret, char *buf, 163 size_t buflen, struct hostent **result, int *h_errnop) 164 { 165 err_t err; 166 struct gethostbyname_r_helper *h; 167 char *hostname; 168 size_t namelen; 169 int lh_errno; 170 171 if (h_errnop == NULL) { 172 /* ensure h_errnop is never NULL */ 173 h_errnop = &lh_errno; 174 } 175 176 if (result == NULL) { 177 /* not all arguments given */ 178 *h_errnop = EINVAL; 179 return -1; 180 } 181 /* first thing to do: set *result to nothing */ 182 *result = NULL; 183 if ((name == NULL) || (ret == NULL) || (buf == 0)) { 184 /* not all arguments given */ 185 *h_errnop = EINVAL; 186 return -1; 187 } 188 189 namelen = strlen(name); 190 if (buflen < (sizeof(struct gethostbyname_r_helper) + namelen + 1 + (MEM_ALIGNMENT - 1))) { 191 /* buf can't hold the data needed + a copy of name */ 192 *h_errnop = ERANGE; 193 return -1; 194 } 195 196 h = (struct gethostbyname_r_helper*)LWIP_MEM_ALIGN(buf); 197 hostname = ((char*)h) + sizeof(struct gethostbyname_r_helper); 198 199 /* query host IP address */ 200 err = netconn_gethostbyname(name, &(h->addr)); 201 if (err != ERR_OK) { 202 LWIP_DEBUGF(DNS_DEBUG, ("lwip_gethostbyname(%s) failed, err=%d\n", name, err)); 203 *h_errnop = ENSRNOTFOUND; 204 return -1; 205 } 206 207 /* copy the hostname into buf */ 208 MEMCPY(hostname, name, namelen); 209 hostname[namelen] = 0; 210 211 /* fill hostent */ 212 h->addrs = &(h->addr); 213 h->aliases = NULL; 214 ret->h_name = (char*)hostname; 215 ret->h_aliases = &(h->aliases); 216 ret->h_addrtype = AF_INET; 217 ret->h_length = sizeof(ip_addr_t); 218 ret->h_addr_list = (char**)&(h->addrs); 219 220 /* set result != NULL */ 221 *result = ret; 222 223 /* return success */ 224 return 0; 225 } 226 227 /** 228 * Frees one or more addrinfo structures returned by getaddrinfo(), along with 229 * any additional storage associated with those structures. If the ai_next field 230 * of the structure is not null, the entire list of structures is freed. 231 * 232 * @param ai struct addrinfo to free 233 */ 234 void 235 lwip_freeaddrinfo(struct addrinfo *ai) 236 { 237 struct addrinfo *next; 238 239 while (ai != NULL) { 240 next = ai->ai_next; 241 memp_free(MEMP_NETDB, ai); 242 ai = next; 243 } 244 } 245 246 /** 247 * Translates the name of a service location (for example, a host name) and/or 248 * a service name and returns a set of socket addresses and associated 249 * information to be used in creating a socket with which to address the 250 * specified service. 251 * Memory for the result is allocated internally and must be freed by calling 252 * lwip_freeaddrinfo()! 253 * 254 * Due to a limitation in dns_gethostbyname, only the first address of a 255 * host is returned. 256 * Also, service names are not supported (only port numbers)! 257 * 258 * @param nodename descriptive name or address string of the host 259 * (may be NULL -> local address) 260 * @param servname port number as string of NULL 261 * @param hints structure containing input values that set socktype and protocol 262 * @param res pointer to a pointer where to store the result (set to NULL on failure) 263 * @return 0 on success, non-zero on failure 264 */ 265 int 266 lwip_getaddrinfo(const char *nodename, const char *servname, 267 const struct addrinfo *hints, struct addrinfo **res) 268 { 269 err_t err; 270 ip_addr_t addr; 271 struct addrinfo *ai; 272 struct sockaddr_in *sa = NULL; 273 int port_nr = 0; 274 size_t total_size; 275 size_t namelen = 0; 276 277 if (res == NULL) { 278 return EAI_FAIL; 279 } 280 *res = NULL; 281 if ((nodename == NULL) && (servname == NULL)) { 282 return EAI_NONAME; 283 } 284 285 if (servname != NULL) { 286 /* service name specified: convert to port number 287 * @todo?: currently, only ASCII integers (port numbers) are supported! */ 288 port_nr = atoi(servname); 289 if ((port_nr <= 0) || (port_nr > 0xffff)) { 290 return EAI_SERVICE; 291 } 292 } 293 294 if (nodename != NULL) { 295 /* service location specified, try to resolve */ 296 err = netconn_gethostbyname(nodename, &addr); 297 if (err != ERR_OK) { 298 return EAI_FAIL; 299 } 300 } else { 301 /* service location specified, use loopback address */ 302 ip_addr_set_loopback(&addr); 303 } 304 305 total_size = sizeof(struct addrinfo) + sizeof(struct sockaddr_in); 306 if (nodename != NULL) { 307 namelen = strlen(nodename); 308 LWIP_ASSERT("namelen is too long", (namelen + 1) <= (mem_size_t)-1); 309 total_size += namelen + 1; 310 } 311 /* If this fails, please report to lwip-devel! :-) */ 312 LWIP_ASSERT("total_size <= NETDB_ELEM_SIZE: please report this!", 313 total_size <= NETDB_ELEM_SIZE); 314 ai = (struct addrinfo *)memp_malloc(MEMP_NETDB); 315 if (ai == NULL) { 316 goto memerr; 317 } 318 memset(ai, 0, total_size); 319 sa = (struct sockaddr_in*)((u8_t*)ai + sizeof(struct addrinfo)); 320 /* set up sockaddr */ 321 inet_addr_from_ipaddr(&sa->sin_addr, &addr); 322 sa->sin_family = AF_INET; 323 sa->sin_len = sizeof(struct sockaddr_in); 324 sa->sin_port = htons((u16_t)port_nr); 325 326 /* set up addrinfo */ 327 ai->ai_family = AF_INET; 328 if (hints != NULL) { 329 /* copy socktype & protocol from hints if specified */ 330 ai->ai_socktype = hints->ai_socktype; 331 ai->ai_protocol = hints->ai_protocol; 332 } 333 if (nodename != NULL) { 334 /* copy nodename to canonname if specified */ 335 ai->ai_canonname = ((char*)ai + sizeof(struct addrinfo) + sizeof(struct sockaddr_in)); 336 MEMCPY(ai->ai_canonname, nodename, namelen); 337 ai->ai_canonname[namelen] = 0; 338 } 339 ai->ai_addrlen = sizeof(struct sockaddr_in); 340 ai->ai_addr = (struct sockaddr*)sa; 341 342 *res = ai; 343 344 return 0; 345 memerr: 346 if (ai != NULL) { 347 memp_free(MEMP_NETDB, ai); 348 } 349 return EAI_MEMORY; 350 } 351 352 #endif /* LWIP_DNS && LWIP_SOCKET */ 353