#include "cups-private.h"
#ifdef HAVE_RESOLV_H
# include <resolv.h>
#endif
#ifdef HAVE_POLL
# include <poll.h>
#endif
#ifndef WIN32
# include <sys/fcntl.h>
#endif
http_addrlist_t *
httpAddrConnect(
http_addrlist_t *addrlist,
int *sock)
{
DEBUG_printf(("httpAddrConnect(addrlist=%p, sock=%p)", addrlist, sock));
return (httpAddrConnect2(addrlist, sock, 30000, NULL));
}
http_addrlist_t *
httpAddrConnect2(
http_addrlist_t *addrlist,
int *sock,
int msec,
int *cancel)
{
int val;
#ifdef O_NONBLOCK
socklen_t len;
http_addr_t peer;
int flags,
remaining;
# ifdef HAVE_POLL
struct pollfd pfd;
# else
fd_set input_set,
output_set;
struct timeval timeout;
# endif
int nfds;
#endif
#ifdef DEBUG
char temp[256];
#endif
DEBUG_printf(("httpAddrConnect2(addrlist=%p, sock=%p, msec=%d, cancel=%p)",
addrlist, sock, msec, cancel));
if (!sock)
{
errno = EINVAL;
_cupsSetError(IPP_INTERNAL_ERROR, strerror(errno), 0);
return (NULL);
}
if (cancel && *cancel)
return (NULL);
if (msec <= 0 || getenv("CUPS_DISABLE_ASYNC_CONNECT"))
msec = INT_MAX;
while (addrlist)
{
if (cancel && *cancel)
return (NULL);
DEBUG_printf(("2httpAddrConnect2: Trying %s:%d...",
httpAddrString(&(addrlist->addr), temp, sizeof(temp)),
_httpAddrPort(&(addrlist->addr))));
if ((*sock = (int)socket(_httpAddrFamily(&(addrlist->addr)), SOCK_STREAM,
0)) < 0)
{
addrlist = addrlist->next;
continue;
}
val = 1;
#ifdef WIN32
setsockopt(*sock, SOL_SOCKET, SO_REUSEADDR, (const char *)&val,
sizeof(val));
#else
setsockopt(*sock, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));
#endif
#ifdef SO_REUSEPORT
val = 1;
setsockopt(*sock, SOL_SOCKET, SO_REUSEPORT, &val, sizeof(val));
#endif
#ifdef SO_NOSIGPIPE
val = 1;
setsockopt(*sock, SOL_SOCKET, SO_NOSIGPIPE, &val, sizeof(val));
#endif
val = 1;
#ifdef WIN32
setsockopt(*sock, IPPROTO_TCP, TCP_NODELAY, (const char *)&val,
sizeof(val));
#else
setsockopt(*sock, IPPROTO_TCP, TCP_NODELAY, &val, sizeof(val));
#endif
#ifdef FD_CLOEXEC
fcntl(*sock, F_SETFD, FD_CLOEXEC);
#endif
#ifdef O_NONBLOCK
DEBUG_printf(("httpAddrConnect2: Setting non-blocking connect()"));
flags = fcntl(*sock, F_GETFL, 0);
if (msec != INT_MAX)
{
DEBUG_puts("httpAddrConnect2: Setting non-blocking connect()");
fcntl(*sock, F_SETFL, flags | O_NONBLOCK);
}
#endif
if (!connect(*sock, &(addrlist->addr.addr),
httpAddrLength(&(addrlist->addr))))
{
DEBUG_printf(("1httpAddrConnect2: Connected to %s:%d...",
httpAddrString(&(addrlist->addr), temp, sizeof(temp)),
_httpAddrPort(&(addrlist->addr))));
#ifdef O_NONBLOCK
fcntl(*sock, F_SETFL, flags);
#endif
return (addrlist);
}
#ifdef O_NONBLOCK
# ifdef WIN32
if (WSAGetLastError() == WSAEINPROGRESS ||
WSAGetLastError() == WSAEWOULDBLOCK)
# else
if (errno == EINPROGRESS || errno == EWOULDBLOCK)
# endif
{
DEBUG_puts("1httpAddrConnect2: Finishing async connect()");
fcntl(*sock, F_SETFL, flags);
for (remaining = msec; remaining > 0; remaining -= 250)
{
do
{
if (cancel && *cancel)
{
DEBUG_puts("1httpAddrConnect2: Canceled connect()");
# ifdef WIN32
closesocket(*sock);
# else
close(*sock);
# endif
*sock = -1;
return (NULL);
}
# ifdef HAVE_POLL
pfd.fd = *sock;
pfd.events = POLLIN | POLLOUT;
nfds = poll(&pfd, 1, remaining > 250 ? 250 : remaining);
DEBUG_printf(("1httpAddrConnect2: poll() returned %d (%d)", nfds,
errno));
# else
FD_ZERO(&input_set);
FD_SET(*sock, &input_set);
output_set = input_set;
timeout.tv_sec = 0;
timeout.tv_usec = (remaining > 250 ? 250 : remaining) * 1000;
nfds = select(*sock + 1, &input_set, &output_set, NULL, &timeout);
DEBUG_printf(("1httpAddrConnect2: select() returned %d (%d)", nfds,
errno));
# endif
}
# ifdef WIN32
while (nfds < 0 && (WSAGetLastError() == WSAEINTR ||
WSAGetLastError() == WSAEWOULDBLOCK));
# else
while (nfds < 0 && (errno == EINTR || errno == EAGAIN));
# endif
if (nfds > 0)
{
len = sizeof(peer);
if (!getpeername(*sock, (struct sockaddr *)&peer, &len))
{
DEBUG_printf(("1httpAddrConnect2: Connected to %s:%d...",
httpAddrString(&peer, temp, sizeof(temp)),
_httpAddrPort(&peer)));
return (addrlist);
}
break;
}
}
}
#endif
DEBUG_printf(("1httpAddrConnect2: Unable to connect to %s:%d: %s",
httpAddrString(&(addrlist->addr), temp, sizeof(temp)),
_httpAddrPort(&(addrlist->addr)), strerror(errno)));
#ifndef WIN32
if (errno == EINPROGRESS)
errno = ETIMEDOUT;
#endif
#ifdef WIN32
closesocket(*sock);
#else
close(*sock);
#endif
*sock = -1;
addrlist = addrlist->next;
}
if (!addrlist)
#ifdef WIN32
_cupsSetError(IPP_SERVICE_UNAVAILABLE, "Connection failed", 0);
#else
_cupsSetError(IPP_SERVICE_UNAVAILABLE, strerror(errno), 0);
#endif
return (addrlist);
}
void
httpAddrFreeList(
http_addrlist_t *addrlist)
{
http_addrlist_t *next;
while (addrlist)
{
next = addrlist->next;
free(addrlist);
addrlist = next;
}
}
http_addrlist_t *
httpAddrGetList(const char *hostname,
int family,
const char *service)
{
http_addrlist_t *first,
*addr,
*temp;
_cups_globals_t *cg = _cupsGlobals();
#ifdef DEBUG
_cups_debug_printf("httpAddrGetList(hostname=\"%s\", family=AF_%s, "
"service=\"%s\")\n",
hostname ? hostname : "(nil)",
family == AF_UNSPEC ? "UNSPEC" :
# ifdef AF_LOCAL
family == AF_LOCAL ? "LOCAL" :
# endif
# ifdef AF_INET6
family == AF_INET6 ? "INET6" :
# endif
family == AF_INET ? "INET" : "???", service);
#endif
#ifdef HAVE_RES_INIT
if (cg->need_res_init)
{
res_init();
cg->need_res_init = 0;
}
#endif
first = addr = NULL;
#ifdef AF_LOCAL
if (hostname && hostname[0] == '/')
{
if ((first = (http_addrlist_t *)calloc(1, sizeof(http_addrlist_t))) != NULL)
{
first->addr.un.sun_family = AF_LOCAL;
strlcpy(first->addr.un.sun_path, hostname, sizeof(first->addr.un.sun_path));
}
}
else
#endif
if (!hostname || _cups_strcasecmp(hostname, "localhost"))
{
#ifdef HAVE_GETADDRINFO
struct addrinfo hints,
*results,
*current;
char ipv6[64],
*ipv6zone;
int ipv6len;
int error;
memset(&hints, 0, sizeof(hints));
hints.ai_family = family;
hints.ai_flags = hostname ? 0 : AI_PASSIVE;
hints.ai_socktype = SOCK_STREAM;
if (hostname && *hostname == '[')
{
if (!strncmp(hostname, "[v1.", 4))
{
strlcpy(ipv6, hostname + 4, sizeof(ipv6));
if ((ipv6len = (int)strlen(ipv6) - 1) >= 0 && ipv6[ipv6len] == ']')
{
ipv6[ipv6len] = '\0';
hostname = ipv6;
if ((ipv6zone = strrchr(ipv6, '+')) != NULL)
*ipv6zone = '%';
}
}
else
{
strlcpy(ipv6, hostname + 1, sizeof(ipv6));
if ((ipv6len = (int)strlen(ipv6) - 1) >= 0 && ipv6[ipv6len] == ']')
{
ipv6[ipv6len] = '\0';
hostname = ipv6;
}
}
}
if ((error = getaddrinfo(hostname, service, &hints, &results)) == 0)
{
for (current = results; current; current = current->ai_next)
if (current->ai_family == AF_INET || current->ai_family == AF_INET6)
{
temp = (http_addrlist_t *)calloc(1, sizeof(http_addrlist_t));
if (!temp)
{
httpAddrFreeList(first);
_cupsSetError(IPP_INTERNAL_ERROR, strerror(errno), 0);
return (NULL);
}
if (current->ai_family == AF_INET6)
memcpy(&(temp->addr.ipv6), current->ai_addr,
sizeof(temp->addr.ipv6));
else
memcpy(&(temp->addr.ipv4), current->ai_addr,
sizeof(temp->addr.ipv4));
if (!first)
first = temp;
if (addr)
addr->next = temp;
addr = temp;
}
freeaddrinfo(results);
}
else
{
if (error == EAI_FAIL)
cg->need_res_init = 1;
_cupsSetError(IPP_INTERNAL_ERROR, gai_strerror(error), 0);
}
#else
if (hostname)
{
int i;
unsigned ip[4];
const char *ptr;
struct hostent *host;
struct servent *port;
int portnum;
if (!service)
portnum = 0;
else if (isdigit(*service & 255))
portnum = atoi(service);
else if ((port = getservbyname(service, NULL)) != NULL)
portnum = ntohs(port->s_port);
else if (!strcmp(service, "http"))
portnum = 80;
else if (!strcmp(service, "https"))
portnum = 443;
else if (!strcmp(service, "ipp") || !strcmp(service, "ipps"))
portnum = 631;
else if (!strcmp(service, "lpd"))
portnum = 515;
else if (!strcmp(service, "socket"))
portnum = 9100;
else
return (NULL);
for (ptr = hostname; isdigit(*ptr & 255) || *ptr == '.'; ptr ++);
if (!*ptr)
{
if (sscanf(hostname, "%u.%u.%u.%u", ip, ip + 1, ip + 2, ip + 3) == 4 &&
ip[0] <= 255 && ip[1] <= 255 && ip[2] <= 255 && ip[3] <= 255)
{
first = (http_addrlist_t *)calloc(1, sizeof(http_addrlist_t));
if (!first)
return (NULL);
first->addr.ipv4.sin_family = AF_INET;
first->addr.ipv4.sin_addr.s_addr = htonl(((((((ip[0] << 8) |
ip[1]) << 8) |
ip[2]) << 8) | ip[3]));
first->addr.ipv4.sin_port = htons(portnum);
}
}
else if ((host = gethostbyname(hostname)) != NULL &&
# ifdef AF_INET6
(host->h_addrtype == AF_INET || host->h_addrtype == AF_INET6))
# else
host->h_addrtype == AF_INET)
# endif
{
for (i = 0; host->h_addr_list[i]; i ++)
{
temp = (http_addrlist_t *)calloc(1, sizeof(http_addrlist_t));
if (!temp)
{
httpAddrFreeList(first);
return (NULL);
}
# ifdef AF_INET6
if (host->h_addrtype == AF_INET6)
{
temp->addr.ipv6.sin6_family = AF_INET6;
memcpy(&(temp->addr.ipv6.sin6_addr), host->h_addr_list[i],
sizeof(temp->addr.ipv6));
temp->addr.ipv6.sin6_port = htons(portnum);
}
else
# endif
{
temp->addr.ipv4.sin_family = AF_INET;
memcpy(&(temp->addr.ipv4.sin_addr), host->h_addr_list[i],
sizeof(temp->addr.ipv4));
temp->addr.ipv4.sin_port = htons(portnum);
}
if (!first)
first = temp;
if (addr)
addr->next = temp;
addr = temp;
}
}
else
{
if (h_errno == NO_RECOVERY)
cg->need_res_init = 1;
_cupsSetError(IPP_INTERNAL_ERROR, hstrerror(h_errno), 0);
}
}
#endif
}
if (!addr && (!hostname || !_cups_strcasecmp(hostname, "localhost")))
{
struct servent *port;
int portnum;
if (!service)
portnum = 0;
else if (isdigit(*service & 255))
portnum = atoi(service);
else if ((port = getservbyname(service, NULL)) != NULL)
portnum = ntohs(port->s_port);
else if (!strcmp(service, "http"))
portnum = 80;
else if (!strcmp(service, "https"))
portnum = 443;
else if (!strcmp(service, "ipp") || !strcmp(service, "ipps"))
portnum = 631;
else if (!strcmp(service, "lpd"))
portnum = 515;
else if (!strcmp(service, "socket"))
portnum = 9100;
else
{
httpAddrFreeList(first);
_cupsSetError(IPP_INTERNAL_ERROR, _("Unknown service name."), 1);
return (NULL);
}
if (hostname && !_cups_strcasecmp(hostname, "localhost"))
{
#ifdef AF_INET6
if (family != AF_INET)
{
temp = (http_addrlist_t *)calloc(1, sizeof(http_addrlist_t));
if (!temp)
{
_cupsSetError(IPP_INTERNAL_ERROR, strerror(errno), 0);
httpAddrFreeList(first);
return (NULL);
}
temp->addr.ipv6.sin6_family = AF_INET6;
temp->addr.ipv6.sin6_port = htons(portnum);
# ifdef WIN32
temp->addr.ipv6.sin6_addr.u.Byte[15] = 1;
# else
temp->addr.ipv6.sin6_addr.s6_addr32[3] = htonl(1);
# endif
if (!first)
first = temp;
addr = temp;
}
if (family != AF_INET6)
#endif
{
temp = (http_addrlist_t *)calloc(1, sizeof(http_addrlist_t));
if (!temp)
{
_cupsSetError(IPP_INTERNAL_ERROR, strerror(errno), 0);
httpAddrFreeList(first);
return (NULL);
}
temp->addr.ipv4.sin_family = AF_INET;
temp->addr.ipv4.sin_port = htons(portnum);
temp->addr.ipv4.sin_addr.s_addr = htonl(0x7f000001);
if (!first)
first = temp;
if (addr)
addr->next = temp;
}
}
else if (!hostname)
{
#ifdef AF_INET6
if (family != AF_INET)
{
temp = (http_addrlist_t *)calloc(1, sizeof(http_addrlist_t));
if (!temp)
{
_cupsSetError(IPP_INTERNAL_ERROR, strerror(errno), 0);
httpAddrFreeList(first);
return (NULL);
}
temp->addr.ipv6.sin6_family = AF_INET6;
temp->addr.ipv6.sin6_port = htons(portnum);
if (!first)
first = temp;
addr = temp;
}
if (family != AF_INET6)
#endif
{
temp = (http_addrlist_t *)calloc(1, sizeof(http_addrlist_t));
if (!temp)
{
_cupsSetError(IPP_INTERNAL_ERROR, strerror(errno), 0);
httpAddrFreeList(first);
return (NULL);
}
temp->addr.ipv4.sin_family = AF_INET;
temp->addr.ipv4.sin_port = htons(portnum);
if (!first)
first = temp;
if (addr)
addr->next = temp;
}
}
}
return (first);
}