/******************************************************************************
* *
* Function: zbx_tcp_close *
* *
* Purpose: close open socket *
* *
* Author: Alexei Vladishev *
* *
******************************************************************************/
void zbx_tcp_close(zbx_sock_t *s)
{
zbx_tcp_unaccept(s);
zbx_tcp_free(s);
zbx_tcp_timeout_cleanup(s);
zbx_sock_close(s->socket);
}
/******************************************************************************
* *
* Function: zbx_tcp_unaccept *
* *
* Purpose: close accepted connection *
* *
* Author: Eugene Grigorjev *
* *
******************************************************************************/
void zbx_tcp_unaccept(zbx_sock_t *s)
{
if (!s->accepted) return;
shutdown(s->socket, 2);
zbx_sock_close(s->socket);
s->socket = s->socket_orig; /* restore main socket */
s->socket_orig = ZBX_SOCK_ERROR;
s->accepted = 0;
}
/******************************************************************************
* *
* Function: zbx_tcp_close *
* *
* Purpose: close open socket *
* *
* Parameters: *
* *
* Return value: *
* *
* Author: Alexei Vladishev *
* *
* Comments: *
* *
******************************************************************************/
void zbx_tcp_close(zbx_sock_t *s)
{
zbx_tcp_unaccept(s);
zbx_tcp_free(s);
#if !defined(_WINDOWS)
if (0 != s->timeout)
alarm(0);
#endif
zbx_sock_close(s->socket);
}
int zbx_tcp_listen(zbx_sock_t *s, const char *listen_ip, unsigned short listen_port)
{
ZBX_SOCKADDR serv_addr;
char *ip, *ips, *delim;
int i, on, ret = FAIL;
ZBX_TCP_START();
zbx_tcp_clean(s);
ip = ips = (NULL == listen_ip ? NULL : strdup(listen_ip));
while (1)
{
delim = (NULL == ip ? NULL : strchr(ip, ','));
if (NULL != delim)
*delim = '\0';
if (NULL != ip && FAIL == is_ip4(ip))
{
zbx_set_tcp_strerror("incorrect IPv4 address [%s]", ip);
goto out;
}
if (ZBX_SOCKET_COUNT == s->num_socks)
{
zbx_set_tcp_strerror("not enough space for socket [[%s]:%hu]",
ip ? ip : "-", listen_port);
goto out;
}
if (ZBX_SOCK_ERROR == (s->sockets[s->num_socks] = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, 0)))
{
zbx_set_tcp_strerror("socket() for [[%s]:%hu] failed: %s",
ip ? ip : "-", listen_port, strerror_from_system(zbx_sock_last_error()));
goto out;
}
#if !defined(_WINDOWS) && !SOCK_CLOEXEC
fcntl(s->sockets[s->num_socks], F_SETFD, FD_CLOEXEC);
#endif
/* Enable address reuse */
/* This is to immediately use the address even if it is in TIME_WAIT state */
/* http://www-128.ibm.com/developerworks/linux/library/l-sockpit/index.html */
on = 1;
if (ZBX_TCP_ERROR == setsockopt(s->sockets[s->num_socks], SOL_SOCKET, SO_REUSEADDR, (void *)&on, sizeof(on)))
{
zbx_set_tcp_strerror("setsockopt() for [[%s]:%hu] failed: %s",
ip ? ip : "-", listen_port, strerror_from_system(zbx_sock_last_error()));
}
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = NULL != ip ? inet_addr(ip) : htonl(INADDR_ANY);
serv_addr.sin_port = htons((unsigned short)listen_port);
if (ZBX_TCP_ERROR == bind(s->sockets[s->num_socks], (struct sockaddr *)&serv_addr, sizeof(serv_addr)))
{
zbx_set_tcp_strerror("bind() for [[%s]:%hu] failed: %s",
ip ? ip : "-", listen_port, strerror_from_system(zbx_sock_last_error()));
zbx_sock_close(s->sockets[s->num_socks]);
goto out;
}
if (ZBX_TCP_ERROR == listen(s->sockets[s->num_socks], SOMAXCONN))
{
zbx_set_tcp_strerror("listen() for [[%s]:%hu] failed: %s",
ip ? ip : "-", listen_port, strerror_from_system(zbx_sock_last_error()));
zbx_sock_close(s->sockets[s->num_socks]);
goto out;
}
s->num_socks++;
if (NULL == ip || NULL == delim)
break;
*delim = ',';
ip = delim + 1;
}
if (0 == s->num_socks)
{
zbx_set_tcp_strerror("zbx_tcp_listen() fatal error: unable to serve on any address [[%s]:%hu]",
listen_ip ? listen_ip : "-", listen_port);
goto out;
}
ret = SUCCEED;
out:
if (NULL != ips)
zbx_free(ips);
if (SUCCEED != ret)
{
for (i = 0; i < s->num_socks; i++)
zbx_sock_close(s->sockets[i]);
}
return ret;
}
int zbx_tcp_listen(zbx_sock_t *s, const char *listen_ip, unsigned short listen_port)
{
struct addrinfo hints, *ai = NULL, *current_ai;
char port[8], *ip, *ips, *delim;
int i, err, on, ret = FAIL;
ZBX_TCP_START();
zbx_tcp_clean(s);
memset(&hints, 0, sizeof(hints));
hints.ai_family = PF_UNSPEC;
hints.ai_flags = AI_NUMERICHOST | AI_PASSIVE;
hints.ai_socktype = SOCK_STREAM;
zbx_snprintf(port, sizeof(port), "%hu", listen_port);
ip = ips = (NULL == listen_ip ? NULL : strdup(listen_ip));
while (1)
{
delim = (NULL == ip ? NULL : strchr(ip, ','));
if (NULL != delim)
*delim = '\0';
if (0 != (err = getaddrinfo(ip, port, &hints, &ai)))
{
zbx_set_tcp_strerror("cannot resolve address [[%s]:%s]: [%d] %s",
ip ? ip : "-", port, err, gai_strerror(err));
goto out;
}
for (current_ai = ai; NULL != current_ai; current_ai = current_ai->ai_next)
{
if (ZBX_SOCKET_COUNT == s->num_socks)
{
zbx_set_tcp_strerror("not enough space for socket [[%s]:%s]",
ip ? ip : "-", port);
goto out;
}
if (PF_INET != current_ai->ai_family && PF_INET6 != current_ai->ai_family)
continue;
if (ZBX_SOCK_ERROR == (s->sockets[s->num_socks] =
socket(current_ai->ai_family, current_ai->ai_socktype | SOCK_CLOEXEC, current_ai->ai_protocol)))
{
zbx_set_tcp_strerror("socket() for [[%s]:%s] failed: %s",
ip ? ip : "-", port, strerror_from_system(zbx_sock_last_error()));
#ifdef _WINDOWS
if (WSAEAFNOSUPPORT == zbx_sock_last_error())
#else
if (EAFNOSUPPORT == zbx_sock_last_error())
#endif
continue;
else
goto out;
}
#if !defined(_WINDOWS) && !SOCK_CLOEXEC
fcntl(s->sockets[s->num_socks], F_SETFD, FD_CLOEXEC);
#endif
/* enable address reuse */
/* this is to immediately use the address even if it is in TIME_WAIT state */
/* http://www-128.ibm.com/developerworks/linux/library/l-sockpit/index.html */
on = 1;
if (ZBX_TCP_ERROR == setsockopt(s->sockets[s->num_socks], SOL_SOCKET, SO_REUSEADDR, (void *)&on, sizeof(on)))
{
zbx_set_tcp_strerror("setsockopt() with SO_REUSEADDR for [[%s]:%s] failed: %s",
ip ? ip : "-", port, strerror_from_system(zbx_sock_last_error()));
}
#if defined(IPPROTO_IPV6) && defined(IPV6_V6ONLY)
if (PF_INET6 == current_ai->ai_family &&
ZBX_TCP_ERROR == setsockopt(s->sockets[s->num_socks], IPPROTO_IPV6, IPV6_V6ONLY, (void *)&on, sizeof(on)))
{
zbx_set_tcp_strerror("setsockopt() with IPV6_V6ONLY for [[%s]:%s] failed: %s",
ip ? ip : "-", port, strerror_from_system(zbx_sock_last_error()));
}
#endif
if (ZBX_TCP_ERROR == bind(s->sockets[s->num_socks], current_ai->ai_addr, current_ai->ai_addrlen))
{
zbx_set_tcp_strerror("bind() for [[%s]:%s] failed: %s",
ip ? ip : "-", port, strerror_from_system(zbx_sock_last_error()));
zbx_sock_close(s->sockets[s->num_socks]);
#ifdef _WINDOWS
if (WSAEADDRINUSE == zbx_sock_last_error())
#else
if (EADDRINUSE == zbx_sock_last_error())
#endif
continue;
else
goto out;
}
if (ZBX_TCP_ERROR == listen(s->sockets[s->num_socks], SOMAXCONN))
{
zbx_set_tcp_strerror("listen() for [[%s]:%s] failed: %s",
ip ? ip : "-", port, strerror_from_system(zbx_sock_last_error()));
zbx_sock_close(s->sockets[s->num_socks]);
goto out;
}
s->num_socks++;
}
if (NULL != ai)
{
freeaddrinfo(ai);
ai = NULL;
}
if (NULL == ip || NULL == delim)
break;
*delim = ',';
ip = delim + 1;
}
if (0 == s->num_socks)
{
zbx_set_tcp_strerror("zbx_tcp_listen() fatal error: unable to serve on any address [[%s]:%hu]",
listen_ip ? listen_ip : "-", listen_port);
goto out;
}
ret = SUCCEED;
out:
if (NULL != ips)
zbx_free(ips);
if (NULL != ai)
freeaddrinfo(ai);
if (SUCCEED != ret)
{
for (i = 0; i < s->num_socks; i++)
zbx_sock_close(s->sockets[i]);
}
return ret;
}
int zbx_tcp_listen(
zbx_sock_t *s,
const char *listen_ip,
unsigned short listen_port
)
{
struct addrinfo hints, *ai = NULL, *current_ai;
char port[MAX_STRING_LEN];
int e, on, ret = FAIL;
ZBX_TCP_START();
zbx_tcp_clean(s);
memset(&hints, 0, sizeof(hints));
hints.ai_family = PF_UNSPEC;
hints.ai_flags = AI_NUMERICHOST | AI_PASSIVE;
hints.ai_socktype = SOCK_STREAM;
zbx_snprintf(port, sizeof(port), "%d", listen_port);
if(0 != (e = getaddrinfo(listen_ip, port, &hints, &ai)))
{
zbx_set_tcp_strerror("Cannot resolve address [[%s]:%u], error %d: %s", listen_ip, listen_port, e, gai_strerror(e));
goto out;
}
for(s->num_socks = 0, current_ai = ai; current_ai != NULL; current_ai = current_ai->ai_next)
{
if(s->num_socks == FD_SETSIZE) {
break;
}
/* This example only supports PF_INET and PF_INET6. */
if((current_ai->ai_family != PF_INET) && (current_ai->ai_family != PF_INET6))
continue;
if((s->sockets[s->num_socks] = socket(current_ai->ai_family, current_ai->ai_socktype, current_ai->ai_protocol)) == ZBX_SOCK_ERROR) {
zbx_set_tcp_strerror("socket() failed with error %d: %s", zbx_sock_last_error(), strerror_from_system(zbx_sock_last_error()));
continue;
}
/* Enable address reuse */
/* This is to immediately use the address even if it is in TIME_WAIT state */
/* http://www-128.ibm.com/developerworks/linux/library/l-sockpit/index.html */
on = 1;
if(setsockopt(s->sockets[s->num_socks], SOL_SOCKET, SO_REUSEADDR, (void *)&on, sizeof(on)) == ZBX_TCP_ERROR)
{
zbx_set_tcp_strerror("setsockopt() failed with error %d: %s", zbx_sock_last_error(), strerror_from_system(zbx_sock_last_error()));
}
/* Create socket Fill in local address structure */
if(bind(s->sockets[s->num_socks], current_ai->ai_addr, current_ai->ai_addrlen) == ZBX_TCP_ERROR) {
zbx_set_tcp_strerror("bind() failed with error %d: %s", zbx_sock_last_error(), strerror_from_system(zbx_sock_last_error()));
zbx_sock_close(s->sockets[s->num_socks]);
continue;
}
if(ZBX_SOCK_ERROR == listen(s->sockets[s->num_socks], SOMAXCONN) )
{
zbx_set_tcp_strerror("listen() failed with error %d: %s", zbx_sock_last_error(), strerror_from_system(zbx_sock_last_error()));
zbx_sock_close(s->sockets[s->num_socks]);
continue;
}
s->num_socks++;
}
if(s->num_socks == 0) {
zbx_set_tcp_strerror("zbx_tcp_listen() Fatal error: unable to serve on any address. [[%s]:%u]", listen_ip, listen_port);
goto out;
}
ret = SUCCEED;
out:
if (NULL != ai)
freeaddrinfo(ai);
return ret;
}
