mirror of
https://github.com/transmission/transmission
synced 2024-12-25 09:13:06 +00:00
223 lines
5.4 KiB
C
223 lines
5.4 KiB
C
/* $Id: connecthostport.c,v 1.3 2010/12/21 16:13:14 nanard Exp $ */
|
|
/* Project : miniupnp
|
|
* Author : Thomas Bernard
|
|
* Copyright (c) 2010 Thomas Bernard
|
|
* This software is subject to the conditions detailed in the
|
|
* LICENCE file provided in this distribution. */
|
|
|
|
/* use getaddrinfo() or gethostbyname()
|
|
* uncomment the following line in order to use gethostbyname() */
|
|
#ifdef NO_GETADDRINFO
|
|
#define USE_GETHOSTBYNAME
|
|
#endif
|
|
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#ifdef WIN32
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
#include <io.h>
|
|
#define snprintf _snprintf
|
|
#define herror
|
|
#define socklen_t int
|
|
#else /* #ifdef WIN32 */
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#define closesocket close
|
|
#include <netdb.h>
|
|
/* defining MINIUPNPC_IGNORE_EINTR enable the ignore of interruptions
|
|
* during the connect() call */
|
|
#define MINIUPNPC_IGNORE_EINTR
|
|
#ifndef USE_GETHOSTBYNAME
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#endif /* #ifndef USE_GETHOSTBYNAME */
|
|
#endif /* #else WIN32 */
|
|
|
|
/* definition of PRINT_SOCKET_ERROR */
|
|
#ifdef WIN32
|
|
#define PRINT_SOCKET_ERROR(x) printf("Socket error: %s, %d\n", x, WSAGetLastError());
|
|
#else
|
|
#define PRINT_SOCKET_ERROR(x) perror(x)
|
|
#endif
|
|
|
|
#if defined(__amigaos__) || defined(__amigaos4__)
|
|
#define herror(A) printf("%s\n", A)
|
|
#endif
|
|
|
|
#include "connecthostport.h"
|
|
|
|
/* connecthostport()
|
|
* return a socket connected (TCP) to the host and port
|
|
* or -1 in case of error */
|
|
int connecthostport(const char * host, unsigned short port)
|
|
{
|
|
int s, n;
|
|
#ifdef USE_GETHOSTBYNAME
|
|
struct sockaddr_in dest;
|
|
struct hostent *hp;
|
|
#else /* #ifdef USE_GETHOSTBYNAME */
|
|
char port_str[8];
|
|
struct addrinfo *ai, *p;
|
|
struct addrinfo hints;
|
|
#endif /* #ifdef USE_GETHOSTBYNAME */
|
|
#ifdef MINIUPNPC_SET_SOCKET_TIMEOUT
|
|
struct timeval timeout;
|
|
#endif /* #ifdef MINIUPNPC_SET_SOCKET_TIMEOUT */
|
|
|
|
#ifdef USE_GETHOSTBYNAME
|
|
hp = gethostbyname(host);
|
|
if(hp == NULL)
|
|
{
|
|
herror(host);
|
|
return -1;
|
|
}
|
|
memcpy(&dest.sin_addr, hp->h_addr, sizeof(dest.sin_addr));
|
|
memset(dest.sin_zero, 0, sizeof(dest.sin_zero));
|
|
s = socket(PF_INET, SOCK_STREAM, 0);
|
|
if(s < 0)
|
|
{
|
|
PRINT_SOCKET_ERROR("socket");
|
|
return -1;
|
|
}
|
|
#ifdef MINIUPNPC_SET_SOCKET_TIMEOUT
|
|
/* setting a 3 seconds timeout for the connect() call */
|
|
timeout.tv_sec = 3;
|
|
timeout.tv_usec = 0;
|
|
if(setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(struct timeval)) < 0)
|
|
{
|
|
PRINT_SOCKET_ERROR("setsockopt");
|
|
}
|
|
timeout.tv_sec = 3;
|
|
timeout.tv_usec = 0;
|
|
if(setsockopt(s, SOL_SOCKET, SO_SNDTIMEO, &timeout, sizeof(struct timeval)) < 0)
|
|
{
|
|
PRINT_SOCKET_ERROR("setsockopt");
|
|
}
|
|
#endif /* #ifdef MINIUPNPC_SET_SOCKET_TIMEOUT */
|
|
dest.sin_family = AF_INET;
|
|
dest.sin_port = htons(port);
|
|
n = connect(s, (struct sockaddr *)&dest, sizeof(struct sockaddr_in));
|
|
#ifdef MINIUPNPC_IGNORE_EINTR
|
|
while(n < 0 && errno == EINTR)
|
|
{
|
|
socklen_t len;
|
|
fd_set wset;
|
|
int err;
|
|
FD_ZERO(&wset);
|
|
FD_SET(s, &wset);
|
|
if((n = select(s + 1, NULL, &wset, NULL, NULL)) == -1 && errno == EINTR)
|
|
continue;
|
|
/*len = 0;*/
|
|
/*n = getpeername(s, NULL, &len);*/
|
|
len = sizeof(err);
|
|
if(getsockopt(s, SOL_SOCKET, SO_ERROR, &err, &len) < 0) {
|
|
PRINT_SOCKET_ERROR("getsockopt");
|
|
closesocket(s);
|
|
return -1;
|
|
}
|
|
if(err != 0) {
|
|
errno = err;
|
|
n = -1;
|
|
}
|
|
}
|
|
#endif /* #ifdef MINIUPNPC_IGNORE_EINTR */
|
|
if(n<0)
|
|
{
|
|
PRINT_SOCKET_ERROR("connect");
|
|
closesocket(s);
|
|
return -1;
|
|
}
|
|
#else /* #ifdef USE_GETHOSTBYNAME */
|
|
/* use getaddrinfo() instead of gethostbyname() */
|
|
memset(&hints, 0, sizeof(hints));
|
|
/* hints.ai_flags = AI_ADDRCONFIG; */
|
|
#ifdef AI_NUMERICSERV
|
|
hints.ai_flags = AI_NUMERICSERV;
|
|
#endif
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
hints.ai_family = AF_UNSPEC; /* AF_INET, AF_INET6 or AF_UNSPEC */
|
|
/* hints.ai_protocol = IPPROTO_TCP; */
|
|
snprintf(port_str, sizeof(port_str), "%hu", port);
|
|
n = getaddrinfo(host, port_str, &hints, &ai);
|
|
if(n != 0)
|
|
{
|
|
#ifdef WIN32
|
|
fprintf(stderr, "getaddrinfo() error : %d\n", n);
|
|
#else
|
|
fprintf(stderr, "getaddrinfo() error : %s\n", gai_strerror(n));
|
|
#endif
|
|
return -1;
|
|
}
|
|
s = -1;
|
|
for(p = ai; p; p = p->ai_next)
|
|
{
|
|
s = socket(p->ai_family, p->ai_socktype, p->ai_protocol);
|
|
if(s < 0)
|
|
continue;
|
|
#ifdef MINIUPNPC_SET_SOCKET_TIMEOUT
|
|
/* setting a 3 seconds timeout for the connect() call */
|
|
timeout.tv_sec = 3;
|
|
timeout.tv_usec = 0;
|
|
if(setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(struct timeval)) < 0)
|
|
{
|
|
PRINT_SOCKET_ERROR("setsockopt");
|
|
}
|
|
timeout.tv_sec = 3;
|
|
timeout.tv_usec = 0;
|
|
if(setsockopt(s, SOL_SOCKET, SO_SNDTIMEO, &timeout, sizeof(struct timeval)) < 0)
|
|
{
|
|
PRINT_SOCKET_ERROR("setsockopt");
|
|
}
|
|
#endif /* #ifdef MINIUPNPC_SET_SOCKET_TIMEOUT */
|
|
n = connect(s, p->ai_addr, p->ai_addrlen);
|
|
#ifdef MINIUPNPC_IGNORE_EINTR
|
|
while(n < 0 && errno == EINTR)
|
|
{
|
|
socklen_t len;
|
|
fd_set wset;
|
|
int err;
|
|
FD_ZERO(&wset);
|
|
FD_SET(s, &wset);
|
|
if((n = select(s + 1, NULL, &wset, NULL, NULL)) == -1 && errno == EINTR)
|
|
continue;
|
|
/*len = 0;*/
|
|
/*n = getpeername(s, NULL, &len);*/
|
|
len = sizeof(err);
|
|
if(getsockopt(s, SOL_SOCKET, SO_ERROR, &err, &len) < 0) {
|
|
PRINT_SOCKET_ERROR("getsockopt");
|
|
closesocket(s);
|
|
freeaddrinfo(ai);
|
|
return -1;
|
|
}
|
|
if(err != 0) {
|
|
errno = err;
|
|
n = -1;
|
|
}
|
|
}
|
|
#endif /* #ifdef MINIUPNPC_IGNORE_EINTR */
|
|
if(n < 0)
|
|
{
|
|
closesocket(s);
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
freeaddrinfo(ai);
|
|
if(s < 0)
|
|
{
|
|
PRINT_SOCKET_ERROR("socket");
|
|
return -1;
|
|
}
|
|
if(n < 0)
|
|
{
|
|
PRINT_SOCKET_ERROR("connect");
|
|
return -1;
|
|
}
|
|
#endif /* #ifdef USE_GETHOSTBYNAME */
|
|
return s;
|
|
}
|
|
|