support multi-homed servers

This commit is contained in:
Oswald Buddenhagen 2013-09-01 16:32:45 +02:00
parent 0b59ee0df3
commit 4a39cae8c4
2 changed files with 60 additions and 32 deletions

View File

@ -85,6 +85,7 @@ typedef struct {
int fd; int fd;
int state; int state;
const server_conf_t *conf; /* needed during connect */ const server_conf_t *conf; /* needed during connect */
char **curr_addr; /* needed during connect */
char *name; char *name;
#ifdef HAVE_LIBSSL #ifdef HAVE_LIBSSL
SSL *ssl; SSL *ssl;

View File

@ -296,6 +296,7 @@ static void start_tls_p3( conn_t *conn, int ok )
static void socket_fd_cb( int, void * ); static void socket_fd_cb( int, void * );
static void socket_connect_one( conn_t * );
static void socket_connect_failed( conn_t * ); static void socket_connect_failed( conn_t * );
static void socket_connected( conn_t * ); static void socket_connected( conn_t * );
static void socket_connect_bail( conn_t * ); static void socket_connect_bail( conn_t * );
@ -312,14 +313,13 @@ void
socket_connect( conn_t *sock, void (*cb)( int ok, void *aux ) ) socket_connect( conn_t *sock, void (*cb)( int ok, void *aux ) )
{ {
const server_conf_t *conf = sock->conf; const server_conf_t *conf = sock->conf;
struct hostent *he;
struct sockaddr_in addr;
int s, a[2];
sock->callbacks.connect = cb; sock->callbacks.connect = cb;
/* open connection to IMAP server */ /* open connection to IMAP server */
if (conf->tunnel) { if (conf->tunnel) {
int a[2];
nfasprintf( &sock->name, "tunnel '%s'", conf->tunnel ); nfasprintf( &sock->name, "tunnel '%s'", conf->tunnel );
infon( "Starting %s... ", sock->name ); infon( "Starting %s... ", sock->name );
@ -343,14 +343,10 @@ socket_connect( conn_t *sock, void (*cb)( int ok, void *aux ) )
fcntl( a[1], F_SETFL, O_NONBLOCK ); fcntl( a[1], F_SETFL, O_NONBLOCK );
add_fd( a[1], socket_fd_cb, sock ); add_fd( a[1], socket_fd_cb, sock );
info( "\vok\n" );
socket_connected( sock );
} else { } else {
memset( &addr, 0, sizeof(addr) ); struct hostent *he;
addr.sin_port = conf->port ? htons( conf->port ) :
#ifdef HAVE_LIBSSL
conf->use_imaps ? htons( 993 ) :
#endif
htons( 143 );
addr.sin_family = AF_INET;
infon( "Resolving %s... ", conf->host ); infon( "Resolving %s... ", conf->host );
he = gethostbyname( conf->host ); he = gethostbyname( conf->host );
@ -361,7 +357,40 @@ socket_connect( conn_t *sock, void (*cb)( int ok, void *aux ) )
} }
info( "\vok\n" ); info( "\vok\n" );
addr.sin_addr.s_addr = *((int *)he->h_addr_list[0]); sock->curr_addr = he->h_addr_list;
socket_connect_one( sock );
}
}
static void
socket_connect_one( conn_t *sock )
{
int s;
ushort port;
struct {
struct sockaddr_in ai_addr[1];
} ai[1];
if (!*sock->curr_addr) {
error( "No working address found for %s\n", sock->conf->host );
socket_connect_bail( sock );
return;
}
port = sock->conf->port ? sock->conf->port :
#ifdef HAVE_LIBSSL
sock->conf->use_imaps ? 993 :
#endif
143;
{
struct sockaddr_in *in = ((struct sockaddr_in *)ai->ai_addr);
memset( in, 0, sizeof(*in) );
in->sin_family = AF_INET;
in->sin_addr.s_addr = *((int *)*sock->curr_addr);
in->sin_port = htons( port );
nfasprintf( &sock->name, "%s (%s:%hu)",
sock->conf->host, inet_ntoa( in->sin_addr ), port );
}
s = socket( PF_INET, SOCK_STREAM, 0 ); s = socket( PF_INET, SOCK_STREAM, 0 );
if (s < 0) { if (s < 0) {
@ -372,10 +401,8 @@ socket_connect( conn_t *sock, void (*cb)( int ok, void *aux ) )
fcntl( s, F_SETFL, O_NONBLOCK ); fcntl( s, F_SETFL, O_NONBLOCK );
add_fd( s, socket_fd_cb, sock ); add_fd( s, socket_fd_cb, sock );
nfasprintf( &sock->name, "%s (%s:%hu)",
conf->host, inet_ntoa( addr.sin_addr ), ntohs( addr.sin_port ) );
infon( "Connecting to %s... ", sock->name ); infon( "Connecting to %s... ", sock->name );
if (connect( s, (struct sockaddr *)&addr, sizeof(addr) )) { if (connect( s, ai->ai_addr, sizeof(*ai->ai_addr) )) {
if (errno != EINPROGRESS) { if (errno != EINPROGRESS) {
socket_connect_failed( sock ); socket_connect_failed( sock );
return; return;
@ -385,11 +412,8 @@ socket_connect( conn_t *sock, void (*cb)( int ok, void *aux ) )
info( "\v\n" ); info( "\v\n" );
return; return;
} }
}
info( "\vok\n" ); info( "\vok\n" );
socket_connected( sock ); socket_connected( sock );
return;
} }
static void static void
@ -397,7 +421,10 @@ socket_connect_failed( conn_t *conn )
{ {
sys_error( "Cannot connect to %s", conn->name ); sys_error( "Cannot connect to %s", conn->name );
socket_close_internal( conn ); socket_close_internal( conn );
socket_connect_bail( conn ); free( conn->name );
conn->name = 0;
conn->curr_addr++;
socket_connect_one( conn );
} }
static void static void