#ifdef HAVE_CONFIG_H
# include <config.h>
#endif
#include "getaddrinfo.h"
#include <stdlib.h>
#include <string.h>
#include <stdbool.h>
#include "gettext.h"
#define _(String) gettext (String)
#define N_(String) String
#include "strdup.h"
static inline bool
validate_family (int family)
{
#if HAVE_IPV4
if (family == PF_INET)
return true;
#endif
#if HAVE_IPV6
if (family == PF_INET6)
return true;
#endif
if (family == PF_UNSPEC)
return true;
return false;
}
int
getaddrinfo (const char *restrict nodename,
const char *restrict servname,
const struct addrinfo *restrict hints,
struct addrinfo **restrict res)
{
struct addrinfo *tmp;
struct servent *se;
struct hostent *he;
size_t sinlen;
if (hints && (hints->ai_flags & ~AI_CANONNAME))
return EAI_BADFLAGS;
if (hints && !validate_family (hints->ai_family))
return EAI_FAMILY;
if (hints &&
hints->ai_socktype != SOCK_STREAM && hints->ai_socktype != SOCK_DGRAM)
return EAI_SOCKTYPE;
if (!nodename)
return EAI_NONAME;
if (servname)
{
const char *proto =
(hints && hints->ai_socktype == SOCK_DGRAM) ? "udp" : "tcp";
se = getservbyname (servname, proto);
if (!se)
return EAI_SERVICE;
}
he = gethostbyname (nodename);
if (!he || he->h_addr_list[0] == NULL)
return EAI_NONAME;
switch (he->h_addrtype)
{
#if HAVE_IPV6
case PF_INET6:
sinlen = sizeof (struct sockaddr_in6);
break;
#endif
#if HAVE_IPV4
case PF_INET:
sinlen = sizeof (struct sockaddr_in);
break;
#endif
default:
return EAI_NODATA;
}
tmp = calloc (1, sizeof (*tmp) + sinlen);
if (!tmp)
return EAI_MEMORY;
switch (he->h_addrtype)
{
#if HAVE_IPV6
case PF_INET6:
{
struct sockaddr_in6 *sinp = (char *) tmp + sizeof (*tmp);
if (se)
sinp->sin6_port = se->s_port;
if (he->h_length != sizeof (sinp->sin6_addr))
return EAI_SYSTEM;
memcpy (&sinp->sin6_addr, he->h_addr_list[0], he->h_length);
tmp->ai_addr = (struct sockaddr *) sinp;
tmp->ai_addrlen = sinlen;
}
break;
#endif
#if HAVE_IPV4
case PF_INET:
{
struct sockaddr_in *sinp = (char *) tmp + sizeof (*tmp);
if (se)
sinp->sin_port = se->s_port;
if (he->h_length != sizeof (sinp->sin_addr))
return EAI_SYSTEM;
memcpy (&sinp->sin_addr, he->h_addr_list[0], he->h_length);
tmp->ai_addr = (struct sockaddr *) sinp;
tmp->ai_addrlen = sinlen;
}
break;
#endif
default:
free (tmp);
return EAI_NODATA;
}
if (hints && hints->ai_flags & AI_CANONNAME)
{
const char *cn;
if (he->h_name)
cn = he->h_name;
else
cn = nodename;
tmp->ai_canonname = strdup (cn);
if (!tmp->ai_canonname)
{
free (tmp);
return EAI_MEMORY;
}
}
tmp->ai_protocol = (hints) ? hints->ai_protocol : 0;
tmp->ai_socktype = (hints) ? hints->ai_socktype : 0;
tmp->ai_addr->sa_family = he->h_addrtype;
*res = tmp;
return 0;
}
void
freeaddrinfo (struct addrinfo *ai)
{
while (ai)
{
struct addrinfo *cur;
cur = ai;
ai = ai->ai_next;
if (cur->ai_canonname) free (cur->ai_canonname);
free (cur);
}
}