1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/*
** Copyright 2006, The Android Open Source Project
**
** Licensed under the Apache License, Version 2.0 (the "License");
** you may not use this file except in compliance with the License.
** You may obtain a copy of the License at
**
** http://www.apache.org/licenses/LICENSE-2.0
**
** Unless required by applicable law or agreed to in writing, software
** distributed under the License is distributed on an "AS IS" BASIS,
** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
** See the License for the specific language governing permissions and
** limitations under the License.
*/
#include <errno.h>
#include <fcntl.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <netdb.h>
#include <cutils/sockets.h>
/* Connect to port on the IP interface. type is
* SOCK_STREAM or SOCK_DGRAM.
* return is a file descriptor or -1 on error
*/
int socket_network_client(const char *host, int port, int type)
{
return socket_network_client_timeout(host, port, type, 0);
}
/* Connect to port on the IP interface. type is SOCK_STREAM or SOCK_DGRAM.
* timeout in seconds return is a file descriptor or -1 on error
*/
int socket_network_client_timeout(const char *host, int port, int type, int timeout)
{
struct hostent *hp;
struct sockaddr_in addr;
socklen_t alen;
int s;
int flags = 0, error = 0, ret = 0;
fd_set rset, wset;
socklen_t len = sizeof(error);
struct timeval ts;
ts.tv_sec = timeout;
ts.tv_usec = 0;
hp = gethostbyname(host);
if (hp == 0) return -1;
memset(&addr, 0, sizeof(addr));
addr.sin_family = hp->h_addrtype;
addr.sin_port = htons(port);
memcpy(&addr.sin_addr, hp->h_addr, hp->h_length);
s = socket(hp->h_addrtype, type, 0);
if (s < 0) return -1;
if ((flags = fcntl(s, F_GETFL, 0)) < 0) {
close(s);
return -1;
}
if (fcntl(s, F_SETFL, flags | O_NONBLOCK) < 0) {
close(s);
return -1;
}
if ((ret = connect(s, (struct sockaddr *) &addr, sizeof(addr))) < 0) {
if (errno != EINPROGRESS) {
close(s);
return -1;
}
}
if (ret == 0)
goto done;
FD_ZERO(&rset);
FD_SET(s, &rset);
wset = rset;
if ((ret = select(s + 1, &rset, &wset, NULL, (timeout) ? &ts : NULL)) < 0) {
close(s);
return -1;
}
if (ret == 0) { // we had a timeout
errno = ETIMEDOUT;
close(s);
return -1;
}
if (FD_ISSET(s, &rset) || FD_ISSET(s, &wset)) {
if (getsockopt(s, SOL_SOCKET, SO_ERROR, &error, &len) < 0) {
close(s);
return -1;
}
} else {
close(s);
return -1;
}
if (error) { // check if we had a socket error
errno = error;
close(s);
return -1;
}
done:
if (fcntl(s, F_SETFL, flags) < 0) {
close(s);
return -1;
}
return s;
}
|