From 3f2de3d1019140571a59c04249de085a318f9266 Mon Sep 17 00:00:00 2001 From: Yang Tse Date: Tue, 21 Oct 2008 01:58:23 +0000 Subject: Charles Hardin patch: - handles the EINPROGRESS for UDP connects - uses closesocket instead of close on some paths that were noticed --- ares/ares_process.c | 29 ++++++++++++++++++----------- 1 file changed, 18 insertions(+), 11 deletions(-) (limited to 'ares/ares_process.c') diff --git a/ares/ares_process.c b/ares/ares_process.c index 798df7a21..675af489d 100644 --- a/ares/ares_process.c +++ b/ares/ares_process.c @@ -906,7 +906,7 @@ static int open_tcp_socket(ares_channel channel, struct server_state *server) /* Configure it. */ if (configure_socket(s, channel) < 0) { - close(s); + closesocket(s); return -1; } @@ -920,7 +920,7 @@ static int open_tcp_socket(ares_channel channel, struct server_state *server) if (setsockopt(s, IPPROTO_TCP, TCP_NODELAY, (void *)&opt, sizeof(opt)) == -1) { - close(s); + closesocket(s); return -1; } @@ -929,14 +929,16 @@ static int open_tcp_socket(ares_channel channel, struct server_state *server) sockin.sin_family = AF_INET; sockin.sin_addr = server->addr; sockin.sin_port = (unsigned short)(channel->tcp_port & 0xffff); - if (connect(s, (struct sockaddr *) &sockin, sizeof(sockin)) == -1) { - int err = SOCKERRNO; + if (connect(s, (struct sockaddr *) &sockin, sizeof(sockin)) == -1) + { + int err = SOCKERRNO; - if (err != EINPROGRESS && err != EWOULDBLOCK) { - closesocket(s); - return -1; + if (err != EINPROGRESS && err != EWOULDBLOCK) + { + closesocket(s); + return -1; + } } - } SOCK_STATE_CALLBACK(channel, s, 1, 0); server->tcp_buffer_pos = 0; @@ -958,7 +960,7 @@ static int open_udp_socket(ares_channel channel, struct server_state *server) /* Set the socket non-blocking. */ if (configure_socket(s, channel) < 0) { - close(s); + closesocket(s); return -1; } @@ -969,8 +971,13 @@ static int open_udp_socket(ares_channel channel, struct server_state *server) sockin.sin_port = (unsigned short)(channel->udp_port & 0xffff); if (connect(s, (struct sockaddr *) &sockin, sizeof(sockin)) == -1) { - closesocket(s); - return -1; + int err = SOCKERRNO; + + if (err != EINPROGRESS && err != EWOULDBLOCK) + { + closesocket(s); + return -1; + } } SOCK_STATE_CALLBACK(channel, s, 1, 0); -- cgit v1.2.3