source: trunk/third-party/miniupnp/connecthostport.c @ 11577

Last change on this file since 11577 was 11577, checked in by livings124, 11 years ago

update miniupnp to 1.4.20101221

File size: 5.4 KB
Line 
1/* $Id: connecthostport.c,v 1.3 2010/12/21 16:13:14 nanard Exp $ */
2/* Project : miniupnp
3 * Author : Thomas Bernard
4 * Copyright (c) 2010 Thomas Bernard
5 * This software is subject to the conditions detailed in the
6 * LICENCE file provided in this distribution. */
7
8/* use getaddrinfo() or gethostbyname()
9 * uncomment the following line in order to use gethostbyname() */
10#ifdef NO_GETADDRINFO
11#define USE_GETHOSTBYNAME
12#endif
13
14#include <string.h>
15#include <stdio.h>
16#ifdef WIN32
17#include <winsock2.h>
18#include <ws2tcpip.h>
19#include <io.h>
20#define snprintf _snprintf
21#define herror
22#define socklen_t int
23#else /* #ifdef WIN32 */
24#include <unistd.h>
25#include <errno.h>
26#define closesocket close
27#include <netdb.h>
28/* defining MINIUPNPC_IGNORE_EINTR enable the ignore of interruptions
29 * during the connect() call */
30#define MINIUPNPC_IGNORE_EINTR
31#ifndef USE_GETHOSTBYNAME
32#include <sys/types.h>
33#include <sys/socket.h>
34#endif /* #ifndef USE_GETHOSTBYNAME */
35#endif /* #else WIN32 */
36
37/* definition of PRINT_SOCKET_ERROR */
38#ifdef WIN32
39#define PRINT_SOCKET_ERROR(x)    printf("Socket error: %s, %d\n", x, WSAGetLastError());
40#else
41#define PRINT_SOCKET_ERROR(x) perror(x)
42#endif
43
44#if defined(__amigaos__) || defined(__amigaos4__)
45#define herror(A) printf("%s\n", A)
46#endif
47
48#include "connecthostport.h"
49
50/* connecthostport()
51 * return a socket connected (TCP) to the host and port
52 * or -1 in case of error */
53int connecthostport(const char * host, unsigned short port)
54{
55        int s, n;
56#ifdef USE_GETHOSTBYNAME
57        struct sockaddr_in dest;
58        struct hostent *hp;
59#else /* #ifdef USE_GETHOSTBYNAME */
60        char port_str[8];
61        struct addrinfo *ai, *p;
62        struct addrinfo hints;
63#endif /* #ifdef USE_GETHOSTBYNAME */
64#ifdef MINIUPNPC_SET_SOCKET_TIMEOUT
65        struct timeval timeout;
66#endif /* #ifdef MINIUPNPC_SET_SOCKET_TIMEOUT */
67       
68#ifdef USE_GETHOSTBYNAME
69        hp = gethostbyname(host);
70        if(hp == NULL)
71        {
72                herror(host);
73                return -1;
74        }
75        memcpy(&dest.sin_addr, hp->h_addr, sizeof(dest.sin_addr));
76        memset(dest.sin_zero, 0, sizeof(dest.sin_zero));
77        s = socket(PF_INET, SOCK_STREAM, 0);
78        if(s < 0)
79        {
80                PRINT_SOCKET_ERROR("socket");
81                return -1;
82        }
83#ifdef MINIUPNPC_SET_SOCKET_TIMEOUT
84        /* setting a 3 seconds timeout for the connect() call */
85        timeout.tv_sec = 3;
86        timeout.tv_usec = 0;
87        if(setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(struct timeval)) < 0)
88        {
89                PRINT_SOCKET_ERROR("setsockopt");
90        }
91        timeout.tv_sec = 3;
92        timeout.tv_usec = 0;
93        if(setsockopt(s, SOL_SOCKET, SO_SNDTIMEO, &timeout, sizeof(struct timeval)) < 0)
94        {
95                PRINT_SOCKET_ERROR("setsockopt");
96        }
97#endif /* #ifdef MINIUPNPC_SET_SOCKET_TIMEOUT */
98        dest.sin_family = AF_INET;
99        dest.sin_port = htons(port);
100        n = connect(s, (struct sockaddr *)&dest, sizeof(struct sockaddr_in));
101#ifdef MINIUPNPC_IGNORE_EINTR
102        while(n < 0 && errno == EINTR)
103        {
104                socklen_t len;
105                fd_set wset;
106                int err;
107                FD_ZERO(&wset);
108                FD_SET(s, &wset);
109                if((n = select(s + 1, NULL, &wset, NULL, NULL)) == -1 && errno == EINTR)
110                        continue;
111                /*len = 0;*/
112                /*n = getpeername(s, NULL, &len);*/
113                len = sizeof(err);
114                if(getsockopt(s, SOL_SOCKET, SO_ERROR, &err, &len) < 0) {
115                        PRINT_SOCKET_ERROR("getsockopt");
116                        closesocket(s);
117                        return -1;
118                }
119                if(err != 0) {
120                        errno = err;
121                        n = -1;
122                }
123        }
124#endif /* #ifdef MINIUPNPC_IGNORE_EINTR */
125        if(n<0)
126        {
127                PRINT_SOCKET_ERROR("connect");
128                closesocket(s);
129                return -1;
130        }
131#else /* #ifdef USE_GETHOSTBYNAME */
132        /* use getaddrinfo() instead of gethostbyname() */
133        memset(&hints, 0, sizeof(hints));
134        /* hints.ai_flags = AI_ADDRCONFIG; */
135#ifdef AI_NUMERICSERV
136        hints.ai_flags = AI_NUMERICSERV;
137#endif
138        hints.ai_socktype = SOCK_STREAM;
139        hints.ai_family = AF_UNSPEC; /* AF_INET, AF_INET6 or AF_UNSPEC */
140        /* hints.ai_protocol = IPPROTO_TCP; */
141        snprintf(port_str, sizeof(port_str), "%hu", port);
142        n = getaddrinfo(host, port_str, &hints, &ai);
143        if(n != 0)
144        {
145#ifdef WIN32
146                fprintf(stderr, "getaddrinfo() error : %d\n", n);
147#else
148                fprintf(stderr, "getaddrinfo() error : %s\n", gai_strerror(n));
149#endif
150                return -1;
151        }
152        s = -1;
153        for(p = ai; p; p = p->ai_next)
154        {
155                s = socket(p->ai_family, p->ai_socktype, p->ai_protocol);
156                if(s < 0)
157                        continue;
158#ifdef MINIUPNPC_SET_SOCKET_TIMEOUT
159                /* setting a 3 seconds timeout for the connect() call */
160                timeout.tv_sec = 3;
161                timeout.tv_usec = 0;
162                if(setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(struct timeval)) < 0)
163                {
164                        PRINT_SOCKET_ERROR("setsockopt");
165                }
166                timeout.tv_sec = 3;
167                timeout.tv_usec = 0;
168                if(setsockopt(s, SOL_SOCKET, SO_SNDTIMEO, &timeout, sizeof(struct timeval)) < 0)
169                {
170                        PRINT_SOCKET_ERROR("setsockopt");
171                }
172#endif /* #ifdef MINIUPNPC_SET_SOCKET_TIMEOUT */
173                n = connect(s, p->ai_addr, p->ai_addrlen);
174#ifdef MINIUPNPC_IGNORE_EINTR
175                while(n < 0 && errno == EINTR)
176                {
177                        socklen_t len;
178                        fd_set wset;
179                        int err;
180                        FD_ZERO(&wset);
181                        FD_SET(s, &wset);
182                        if((n = select(s + 1, NULL, &wset, NULL, NULL)) == -1 && errno == EINTR)
183                                continue;
184                        /*len = 0;*/
185                        /*n = getpeername(s, NULL, &len);*/
186                        len = sizeof(err);
187                        if(getsockopt(s, SOL_SOCKET, SO_ERROR, &err, &len) < 0) {
188                                PRINT_SOCKET_ERROR("getsockopt");
189                                closesocket(s);
190                                freeaddrinfo(ai);
191                                return -1;
192                        }
193                        if(err != 0) {
194                                errno = err;
195                                n = -1;
196                        }
197                }
198#endif /* #ifdef MINIUPNPC_IGNORE_EINTR */
199                if(n < 0)
200                {
201                        closesocket(s);
202                        continue;
203                }
204                else
205                {
206                        break;
207                }
208        }
209        freeaddrinfo(ai);
210        if(s < 0)
211        {
212                PRINT_SOCKET_ERROR("socket");
213                return -1;
214        }
215        if(n < 0)
216        {
217                PRINT_SOCKET_ERROR("connect");
218                return -1;
219        }
220#endif /* #ifdef USE_GETHOSTBYNAME */
221        return s;
222}
223
Note: See TracBrowser for help on using the repository browser.