/**************************************************************************** * libs/libc/netdb/lib_getaddrinfo.c * * Copyright (C) 2018-2019 Gregory Nutt. All rights reserved. * Author: Juha Niskanen * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in * the documentation and/or other materials provided with the * distribution. * 3. Neither the name NuttX nor the names of its contributors may be * used to endorse or promote products derived from this software * without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * POSSIBILITY OF SUCH DAMAGE. * ****************************************************************************/ /**************************************************************************** * Included Files ****************************************************************************/ #include #include #include #include #include #include "libc.h" #include "lib_netdb.h" /**************************************************************************** * Private Data Types ****************************************************************************/ struct ai_s { struct addrinfo ai; union { struct sockaddr_in sin; struct sockaddr_in6 sin6; } sa; }; /**************************************************************************** * Private Functions ****************************************************************************/ FAR static struct ai_s *alloc_ai(int family, int socktype, int protocol, int port, FAR void *addr) { FAR struct ai_s *ai; socklen_t addrlen; addrlen = (family == AF_INET) ? sizeof(struct sockaddr_in) : sizeof(struct sockaddr_in6); ai = lib_zalloc(sizeof(struct ai_s)); if (ai == NULL) { return ai; } ai->ai.ai_addr = (FAR struct sockaddr *)&ai->sa; ai->ai.ai_addrlen = addrlen; ai->ai.ai_family = family; ai->ai.ai_socktype = socktype; ai->ai.ai_protocol = protocol; switch (family) { #ifdef CONFIG_NET_IPv4 case AF_INET: ai->sa.sin.sin_family = AF_INET; ai->sa.sin.sin_port = port; /* Already network order */ memcpy(&ai->sa.sin.sin_addr, addr, sizeof(ai->sa.sin.sin_addr)); break; #endif #ifdef CONFIG_NET_IPv6 case AF_INET6: ai->sa.sin6.sin6_family = AF_INET6; ai->sa.sin6.sin6_port = port; /* Already network order */ memcpy(&ai->sa.sin6.sin6_addr, addr, sizeof(ai->sa.sin6.sin6_addr)); break; #endif } return ai; } /**************************************************************************** * Public Functions ****************************************************************************/ /**************************************************************************** * Name: getaddrinfo ****************************************************************************/ int getaddrinfo(FAR const char *hostname, FAR const char *servname, FAR const struct addrinfo *hint, FAR struct addrinfo **res) { int family = AF_UNSPEC; int port = 0; int flags = 0; int proto = 0; int socktype = 0; char hostbuffer[CONFIG_NETDB_BUFSIZE]; FAR struct hostent_s host; FAR struct ai_s *ai; FAR struct ai_s *prev_ai = NULL; const int valid_flags = AI_PASSIVE | AI_CANONNAME | AI_NUMERICHOST | AI_NUMERICSERV | AI_V4MAPPED | AI_ALL | AI_ADDRCONFIG; int ret = OK; int i; if (hostname == NULL && servname == NULL) { return EAI_NONAME; } if (hint) { family = hint->ai_family; flags = hint->ai_flags; proto = hint->ai_protocol; socktype = hint->ai_socktype; if ((flags & valid_flags) != flags) { return EAI_BADFLAGS; } if (family != AF_INET && family != AF_INET6 && family != AF_UNSPEC) { return EAI_FAMILY; } } if (servname != NULL) { struct servent ent; FAR struct servent *sp; FAR char *endp; port = strtol(servname, &endp, 10); if (port > 0 && port <= 65535 && *endp == '\0') { /* Force network byte order */ port = htons(port); } else if ((flags & AI_NUMERICSERV) != 0) { return EAI_NONAME; } else if (getservbyname_r(servname, NULL, &ent, NULL, 0, &sp) == OK) { /* The s_port field of struct servent is required to * be in network byte order (per OpenGroup.org) */ port = sp->s_port; } else { return EAI_SERVICE; } } *res = NULL; /* If hostname is not NULL, then the AI_PASSIVE flag is ignored. */ if ((flags & AI_PASSIVE) != 0 && hostname == NULL) { struct in6_addr addr; memset(&addr, 0, sizeof(struct in6_addr)); #ifdef CONFIG_NET_IPv4 if (family == AF_INET || family == AF_UNSPEC) { ai = alloc_ai(AF_INET, socktype, proto, port, (FAR void *)&addr); if (ai != NULL) { *res = (FAR struct addrinfo *)ai; } } #endif #ifdef CONFIG_NET_IPv6 if (family == AF_INET6 || family == AF_UNSPEC) { ai = alloc_ai(AF_INET6, socktype, proto, port, (FAR void *)&addr); if (ai != NULL) { /* Can return both IPv4 and IPv6 loopback. */ if (*res != NULL) { (*res)->ai_next = (FAR struct addrinfo *)ai; } else { *res = (FAR struct addrinfo *)ai; } } } #endif return (*res != NULL) ? OK : EAI_MEMORY; } if (hostname == NULL) { #ifdef CONFIG_NET_LOOPBACK /* Local service. */ #ifdef CONFIG_NET_IPv4 if (family == AF_INET || family == AF_UNSPEC) { ai = alloc_ai(AF_INET, socktype, proto, port, (FAR void *)&g_lo_ipv4addr); if (ai != NULL) { *res = (FAR struct addrinfo *)ai; } } #endif #ifdef CONFIG_NET_IPv6 if (family == AF_INET6 || family == AF_UNSPEC) { ai = alloc_ai(AF_INET6, socktype, proto, port, (FAR void *)&g_lo_ipv6addr); if (ai != NULL) { /* Can return both IPv4 and IPv6 loopback. */ if (*res != NULL) { (*res)->ai_next = (FAR struct addrinfo *)ai; } else { *res = (FAR struct addrinfo *)ai; } } } #endif return (*res != NULL) ? OK : EAI_MEMORY; #else /* Local service, but no loopback so cannot succeed. */ return EAI_FAIL; #endif /* CONFIG_NET_LOOPBACK */ } /* REVISIT: no check for AI_NUMERICHOST flag. */ gethostentbyname_r(hostname, &host, hostbuffer, sizeof(hostbuffer), &ret); if (ret != OK) { return ret; } for (i = 0; host.h_addr_list[i]; i++) { if (family != AF_UNSPEC && host.h_addrtypes[i] != family) { /* Filter by protocol family. */ continue; } /* REVISIT: filter by socktype and protocol not implemented. */ ai = alloc_ai(host.h_addrtypes[i], socktype, proto, port, host.h_addr_list[i]); if (ai == NULL) { if (*res) { freeaddrinfo(*res); } return EAI_MEMORY; } /* REVISIT: grok canonical name. * * OpenGroup: "if the canonical name is not available, then * ai_canonname shall refer to the hostname argument or a string * with the same contents." */ ai->ai.ai_canonname = (FAR char *)hostname; /* Add result to linked list. * TODO: RFC 3484/6724 destination address sort not implemented. */ if (prev_ai != NULL) { prev_ai->ai.ai_next = (FAR struct addrinfo *)ai; } else { *res = (FAR struct addrinfo *)ai; } prev_ai = ai; } return (*res != NULL) ? OK : EAI_FAMILY; }