lighttpd 1.4.x
https://www.lighttpd.net/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
310 lines
9.4 KiB
310 lines
9.4 KiB
#include "first.h" |
|
|
|
#include "inet_ntop_cache.h" |
|
#include "base.h" |
|
#include "log.h" |
|
|
|
#include "sys-socket.h" |
|
#include <sys/types.h> |
|
#include <errno.h> |
|
#include <string.h> |
|
#ifndef _WIN32 |
|
#include <netdb.h> |
|
#endif |
|
|
|
|
|
unsigned short sock_addr_get_port (const sock_addr *addr) |
|
{ |
|
switch (addr->plain.sa_family) { |
|
case AF_INET: |
|
return ntohs(addr->ipv4.sin_port); |
|
#ifdef HAVE_IPV6 |
|
case AF_INET6: |
|
return ntohs(addr->ipv6.sin6_port); |
|
#endif |
|
default: /* case AF_UNIX: */ |
|
return 0; |
|
} |
|
} |
|
|
|
|
|
int sock_addr_inet_pton(sock_addr *addr, const char *str, |
|
int family, unsigned short port) |
|
{ |
|
if (AF_INET == family) { |
|
memset(&addr->ipv4, 0, sizeof(struct sockaddr_in)); |
|
addr->ipv4.sin_family = AF_INET; |
|
addr->ipv4.sin_port = htons(port); |
|
#if defined(HAVE_INET_ATON) /*(Windows does not provide inet_aton())*/ |
|
return (0 != inet_aton(str, &addr->ipv4.sin_addr)); |
|
#else |
|
return ((addr->ipv4.sin_addr.s_addr = inet_addr(str)) != INADDR_NONE); |
|
#endif |
|
} |
|
#ifdef HAVE_IPV6 |
|
else if (AF_INET6 == family) { |
|
memset(&addr->ipv6, 0, sizeof(struct sockaddr_in6)); |
|
addr->ipv6.sin6_family = AF_INET6; |
|
addr->ipv6.sin6_port = htons(port); |
|
return inet_pton(AF_INET6, str, &addr->ipv6.sin6_addr); |
|
} |
|
#endif |
|
else { |
|
errno = EAFNOSUPPORT; |
|
return -1; |
|
} |
|
} |
|
|
|
|
|
const char * sock_addr_inet_ntop(const sock_addr *addr, char *buf, socklen_t sz) |
|
{ |
|
if (addr->plain.sa_family == AF_INET) { |
|
#if defined(HAVE_INET_PTON) /*(expect inet_ntop if inet_pton)*/ |
|
return inet_ntop(AF_INET,(const void *)&addr->ipv4.sin_addr,buf,sz); |
|
#else /*(inet_ntoa() not thread-safe)*/ |
|
return inet_ntoa(addr->ipv4.sin_addr); |
|
#endif |
|
} |
|
#ifdef HAVE_IPV6 |
|
else if (addr->plain.sa_family == AF_INET6) { |
|
return inet_ntop(AF_INET6,(const void *)&addr->ipv6.sin6_addr,buf,sz); |
|
} |
|
#endif |
|
#ifdef HAVE_SYS_UN_H |
|
else if (addr->plain.sa_family == AF_UNIX) { |
|
return addr->un.sun_path; |
|
} |
|
#endif |
|
else { |
|
errno = EAFNOSUPPORT; |
|
return NULL; |
|
} |
|
} |
|
|
|
|
|
int sock_addr_inet_ntop_copy_buffer(buffer *b, const sock_addr *addr) |
|
{ |
|
/*(incur cost of extra copy to avoid potential extra memory allocation)*/ |
|
char buf[UNIX_PATH_MAX]; |
|
const char *s = sock_addr_inet_ntop(addr, buf, sizeof(buf)); |
|
if (NULL == s) return -1; /*(buffer not modified if any error occurs)*/ |
|
buffer_copy_string(b, s); |
|
return 0; |
|
} |
|
|
|
|
|
int sock_addr_inet_ntop_append_buffer(buffer *b, const sock_addr *addr) |
|
{ |
|
/*(incur cost of extra copy to avoid potential extra memory allocation)*/ |
|
char buf[UNIX_PATH_MAX]; |
|
const char *s = sock_addr_inet_ntop(addr, buf, sizeof(buf)); |
|
if (NULL == s) return -1; /*(buffer not modified if any error occurs)*/ |
|
buffer_append_string(b, s); |
|
return 0; |
|
} |
|
|
|
|
|
int sock_addr_from_str_hints(server *srv, sock_addr *addr, socklen_t *len, const char *str, int family, unsigned short port) |
|
{ |
|
/*(note: name resolution here is *blocking*)*/ |
|
switch(family) { |
|
#ifdef HAVE_IPV6 |
|
case AF_INET6: |
|
memset(addr, 0, sizeof(struct sockaddr_in6)); |
|
addr->ipv6.sin6_family = AF_INET6; |
|
if (0 == strcmp(str, "::")) { |
|
addr->ipv6.sin6_addr = in6addr_any; |
|
} |
|
else if (0 == strcmp(str, "::1")) { |
|
addr->ipv6.sin6_addr = in6addr_loopback; |
|
} |
|
else { |
|
struct addrinfo hints, *res; |
|
int r; |
|
|
|
memset(&hints, 0, sizeof(hints)); |
|
|
|
hints.ai_family = AF_INET6; |
|
hints.ai_socktype = SOCK_STREAM; |
|
hints.ai_protocol = IPPROTO_TCP; |
|
|
|
if (0 != (r = getaddrinfo(str, NULL, &hints, &res))) { |
|
hints.ai_family = AF_INET; |
|
if ( |
|
#ifdef EAI_ADDRFAMILY |
|
EAI_ADDRFAMILY == r && |
|
#endif |
|
0 == getaddrinfo(str, NULL, &hints, &res)) { |
|
memcpy(addr, res->ai_addr, res->ai_addrlen); |
|
addr->ipv4.sin_family = AF_INET; |
|
addr->ipv4.sin_port = htons(port); |
|
*len = sizeof(struct sockaddr_in); |
|
/*assert(*len == res->ai_addrlen);*/ |
|
freeaddrinfo(res); |
|
return 1; |
|
} |
|
|
|
log_error_write(srv, __FILE__, __LINE__, |
|
"sssss", "getaddrinfo failed: ", |
|
gai_strerror(r), "'", str, "'"); |
|
|
|
return 0; |
|
} |
|
|
|
memcpy(addr, res->ai_addr, res->ai_addrlen); |
|
freeaddrinfo(res); |
|
} |
|
addr->ipv6.sin6_port = htons(port); |
|
*len = sizeof(struct sockaddr_in6); |
|
return 1; |
|
#endif |
|
case AF_INET: |
|
memset(addr, 0, sizeof(struct sockaddr_in)); |
|
addr->ipv4.sin_family = AF_INET; |
|
if (0 == strcmp(str, "0.0.0.0")) { |
|
addr->ipv4.sin_addr.s_addr = htonl(INADDR_ANY); |
|
} |
|
else if (0 == strcmp(str, "127.0.0.1")) { |
|
addr->ipv4.sin_addr.s_addr = htonl(INADDR_LOOPBACK); |
|
} |
|
else { |
|
#ifdef HAVE_INET_PTON |
|
/*(reuse HAVE_INET_PTON for presence of getaddrinfo())*/ |
|
struct addrinfo hints, *res; |
|
int r; |
|
memset(&hints, 0, sizeof(hints)); |
|
hints.ai_family = AF_INET; |
|
hints.ai_socktype = SOCK_STREAM; |
|
hints.ai_protocol = IPPROTO_TCP; |
|
|
|
if (0 != (r = getaddrinfo(str, NULL, &hints, &res))) { |
|
log_error_write(srv, __FILE__, __LINE__, |
|
"sssss", "getaddrinfo failed: ", |
|
gai_strerror(r), "'", str, "'"); |
|
return 0; |
|
} |
|
|
|
memcpy(addr, res->ai_addr, res->ai_addrlen); |
|
freeaddrinfo(res); |
|
#else |
|
struct hostent *he = gethostbyname(str); |
|
if (NULL == he) { |
|
log_error_write(srv, __FILE__, __LINE__, "sds", |
|
"gethostbyname failed:", h_errno, str); |
|
return 0; |
|
} |
|
|
|
if (he->h_addrtype != AF_INET) { |
|
log_error_write(srv, __FILE__, __LINE__, "sd", |
|
"addr-type != AF_INET:", he->h_addrtype); |
|
return 0; |
|
} |
|
|
|
if (he->h_length != sizeof(struct in_addr)) { |
|
log_error_write(srv, __FILE__, __LINE__, "sd", |
|
"addr-length != sizeof(in_addr):",he->h_length); |
|
return 0; |
|
} |
|
|
|
memcpy(&addr->ipv4.sin_addr.s_addr,he->h_addr_list[0],he->h_length); |
|
#endif |
|
} |
|
addr->ipv4.sin_port = htons(port); |
|
*len = sizeof(struct sockaddr_in); |
|
return 1; |
|
#ifdef HAVE_SYS_UN_H |
|
case AF_UNIX: |
|
memset(addr, 0, sizeof(struct sockaddr_un)); |
|
addr->un.sun_family = AF_UNIX; |
|
{ |
|
size_t hostlen = strlen(str) + 1; |
|
if (hostlen > sizeof(addr->un.sun_path)) { |
|
log_error_write(srv, __FILE__, __LINE__, "sS", |
|
"unix socket filename too long:", str); |
|
/*errno = ENAMETOOLONG;*/ |
|
return 0; |
|
} |
|
memcpy(addr->un.sun_path, str, hostlen); |
|
#if defined(SUN_LEN) |
|
*len = SUN_LEN(&addr->un); |
|
#else |
|
/* stevens says: */ |
|
*len = hostlen + sizeof(addr->un.sun_family); |
|
#endif |
|
} |
|
return 1; |
|
#else |
|
case AF_UNIX: |
|
log_error_write(srv, __FILE__, __LINE__, "s", |
|
"unix domain sockets are not supported."); |
|
return 0; |
|
#endif |
|
default: |
|
log_error_write(srv, __FILE__, __LINE__, "sd", |
|
"address family unsupported:", family); |
|
/*errno = EAFNOSUPPORT;*/ |
|
return 0; |
|
} |
|
} |
|
|
|
|
|
const char * inet_ntop_cache_get_ip(server *srv, sock_addr *addr) { |
|
#ifdef HAVE_IPV6 |
|
typedef struct { |
|
int family; |
|
union { |
|
struct in6_addr ipv6; |
|
struct in_addr ipv4; |
|
} addr; |
|
char b2[INET6_ADDRSTRLEN + 1]; |
|
} inet_ntop_cache_type; |
|
#define INET_NTOP_CACHE_MAX 4 |
|
static inet_ntop_cache_type inet_ntop_cache[INET_NTOP_CACHE_MAX]; |
|
static int ndx; |
|
|
|
int i; |
|
UNUSED(srv); |
|
for (i = 0; i < INET_NTOP_CACHE_MAX; i++) { |
|
if (inet_ntop_cache[i].family == addr->plain.sa_family) { |
|
if (inet_ntop_cache[i].family == AF_INET6 && |
|
0 == memcmp(inet_ntop_cache[i].addr.ipv6.s6_addr, addr->ipv6.sin6_addr.s6_addr, 16)) { |
|
/* IPv6 found in cache */ |
|
break; |
|
} else if (inet_ntop_cache[i].family == AF_INET && |
|
inet_ntop_cache[i].addr.ipv4.s_addr == addr->ipv4.sin_addr.s_addr) { |
|
/* IPv4 found in cache */ |
|
break; |
|
|
|
} |
|
} |
|
} |
|
|
|
if (i == INET_NTOP_CACHE_MAX) { |
|
/* not found in cache */ |
|
const char *s; |
|
|
|
i = ndx; |
|
if (++ndx >= INET_NTOP_CACHE_MAX) ndx = 0; |
|
s = |
|
inet_ntop(addr->plain.sa_family, |
|
addr->plain.sa_family == AF_INET6 ? |
|
(const void *) &(addr->ipv6.sin6_addr) : |
|
(const void *) &(addr->ipv4.sin_addr), |
|
inet_ntop_cache[i].b2, INET6_ADDRSTRLEN); |
|
if (NULL == s) return ""; |
|
|
|
inet_ntop_cache[i].family = addr->plain.sa_family; |
|
|
|
if (inet_ntop_cache[i].family == AF_INET) { |
|
inet_ntop_cache[i].addr.ipv4.s_addr = addr->ipv4.sin_addr.s_addr; |
|
} else if (inet_ntop_cache[i].family == AF_INET6) { |
|
memcpy(inet_ntop_cache[i].addr.ipv6.s6_addr, addr->ipv6.sin6_addr.s6_addr, 16); |
|
} |
|
} |
|
|
|
return inet_ntop_cache[i].b2; |
|
#else |
|
UNUSED(srv); |
|
return inet_ntoa(addr->ipv4.sin_addr); |
|
#endif |
|
}
|
|
|