Project

General

Profile

Bug #546 » socket-timeout-2.patch

socket-timeout-2.patch - Mathieu Desnoyers, 07/11/2013 01:49 PM

View differences:

src/common/sessiond-comm/Makefile.am
libsessiond_comm_la_SOURCES = sessiond-comm.c sessiond-comm.h \
unix.c unix.h inet.c inet.h inet6.c inet6.h \
relayd.h
libsessiond_comm_la_LIBADD = -lrt
src/common/sessiond-comm/inet.c
#include <sys/types.h>
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>
#include <time.h>
#include <poll.h>
#include <common/common.h>
#include "inet.h"
#define MSEC_PER_SEC 1000
#define NSEC_PER_MSEC 1000000
#define RECONNECT_DELAY 200 /* ms */
/*
* INET protocol operations.
*/
......
return ret;
}
static
int connect_no_timeout(struct lttcomm_sock *sock)
{
return connect(sock->fd, (struct sockaddr *) &sock->sockaddr.addr.sin,
sizeof(sock->sockaddr.addr.sin));
}
/*
* Return time_a - time_b in milliseconds.
*/
static
unsigned long time_diff_ms(struct timespec *time_a,
struct timespec *time_b)
{
time_t sec_diff;
long nsec_diff;
unsigned long result_ms;
sec_diff = time_a->tv_sec - time_b->tv_sec;
nsec_diff = time_a->tv_nsec - time_b->tv_nsec;
result_ms = sec_diff * MSEC_PER_SEC;
result_ms += nsec_diff / NSEC_PER_MSEC;
return result_ms;
}
static
int connect_with_timeout(struct lttcomm_sock *sock)
{
unsigned long timeout = lttcomm_get_network_timeout();
int ret, flags, connect_ret;
struct timespec orig_time, cur_time;
ret = fcntl(sock->fd, F_GETFL, 0);
if (ret == -1) {
PERROR("fcntl");
return -1;
}
flags = ret;
/* Set socket to nonblock */
ret = fcntl(sock->fd, F_SETFL, flags | O_NONBLOCK);
if (ret == -1) {
PERROR("fcntl");
return -1;
}
ret = clock_gettime(CLOCK_MONOTONIC, &orig_time);
if (ret == -1) {
PERROR("clock_gettime");
return -1;
}
do {
connect_ret = connect(sock->fd,
(struct sockaddr *) &sock->sockaddr.addr.sin,
sizeof(sock->sockaddr.addr.sin));
if (connect_ret == -1 && errno != EAGAIN
&& errno != EWOULDBLOCK
&& errno != EINPROGRESS) {
goto error;
} else if (!connect_ret) {
/* Connect succeeded */
goto success;
}
(void) poll(NULL, 0, RECONNECT_DELAY);
ret = clock_gettime(CLOCK_MONOTONIC, &cur_time);
if (ret == -1) {
PERROR("clock_gettime");
return -1;
}
} while (time_diff_ms(&cur_time, &orig_time) < timeout);
/* Timeout */
errno = ETIMEDOUT;
connect_ret = -1;
success:
/* Restore initial flags */
ret = fcntl(sock->fd, F_SETFL, flags);
if (ret == -1) {
PERROR("fcntl");
/* Continue anyway */
}
error:
return connect_ret;
}
/*
* Connect PF_INET socket.
*/
......
{
int ret, closeret;
ret = connect(sock->fd, (struct sockaddr *) &sock->sockaddr.addr.sin,
sizeof(sock->sockaddr.addr.sin));
if (lttcomm_get_network_timeout()) {
ret = connect_with_timeout(sock);
} else {
ret = connect_no_timeout(sock);
}
if (ret < 0) {
PERROR("connect");
goto error_connect;
src/common/sessiond-comm/inet6.c
#include <sys/types.h>
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>
#include <time.h>
#include <poll.h>
#include <common/common.h>
#include "inet6.h"
#define MSEC_PER_SEC 1000
#define NSEC_PER_MSEC 1000000
#define RECONNECT_DELAY 200 /* ms */
/*
* INET protocol operations.
*/
......
return ret;
}
static
int connect_no_timeout(struct lttcomm_sock *sock)
{
return connect(sock->fd, (struct sockaddr *) &sock->sockaddr.addr.sin6,
sizeof(sock->sockaddr.addr.sin6));
}
/*
* Return time_a - time_b in milliseconds.
*/
static
unsigned long time_diff_ms(struct timespec *time_a,
struct timespec *time_b)
{
time_t sec_diff;
long nsec_diff;
unsigned long result_ms;
sec_diff = time_a->tv_sec - time_b->tv_sec;
nsec_diff = time_a->tv_nsec - time_b->tv_nsec;
result_ms = sec_diff * MSEC_PER_SEC;
result_ms += nsec_diff / NSEC_PER_MSEC;
return result_ms;
}
static
int connect_with_timeout(struct lttcomm_sock *sock)
{
unsigned long timeout = lttcomm_get_network_timeout();
int ret, flags, connect_ret;
struct timespec orig_time, cur_time;
ret = fcntl(sock->fd, F_GETFL, 0);
if (ret == -1) {
PERROR("fcntl");
return -1;
}
flags = ret;
/* Set socket to nonblock */
ret = fcntl(sock->fd, F_SETFL, flags | O_NONBLOCK);
if (ret == -1) {
PERROR("fcntl");
return -1;
}
ret = clock_gettime(CLOCK_MONOTONIC, &orig_time);
if (ret == -1) {
PERROR("clock_gettime");
return -1;
}
do {
connect_ret = connect(sock->fd,
(struct sockaddr *) &sock->sockaddr.addr.sin6,
sizeof(sock->sockaddr.addr.sin6));
if (connect_ret == -1 && errno != EAGAIN
&& errno != EWOULDBLOCK
&& errno != EINPROGRESS) {
goto error;
} else if (!connect_ret) {
/* Connect succeeded */
goto success;
}
(void) poll(NULL, 0, RECONNECT_DELAY);
ret = clock_gettime(CLOCK_MONOTONIC, &cur_time);
if (ret == -1) {
PERROR("clock_gettime");
return -1;
}
} while (time_diff_ms(&cur_time, &orig_time) < timeout);
/* Timeout */
errno = ETIMEDOUT;
connect_ret = -1;
success:
/* Restore initial flags */
ret = fcntl(sock->fd, F_SETFL, flags);
if (ret == -1) {
PERROR("fcntl");
/* Continue anyway */
}
error:
return connect_ret;
}
/*
* Connect PF_INET socket.
*/
......
{
int ret, closeret;
ret = connect(sock->fd, (struct sockaddr *) &sock->sockaddr.addr.sin6,
sizeof(sock->sockaddr.addr.sin6));
if (lttcomm_get_network_timeout()) {
ret = connect_with_timeout(sock);
} else {
ret = connect_no_timeout(sock);
}
if (ret < 0) {
PERROR("connect inet6");
goto error_connect;
(14-14/19)