2 * usock - socket helper functions
4 * Copyright (C) 2010 Steven Barth <steven@midlink.org>
5 * Copyright (C) 2011-2012 Felix Fietkau <nbd@openwrt.org>
7 * Permission to use, copy, modify, and/or distribute this software for any
8 * purpose with or without fee is hereby granted, provided that the above
9 * copyright notice and this permission notice appear in all copies.
11 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
19 #include <sys/types.h>
20 #include <sys/socket.h>
36 static void usock_set_flags(int sock
, unsigned int type
)
38 if (!(type
& USOCK_NOCLOEXEC
))
39 fcntl(sock
, F_SETFD
, fcntl(sock
, F_GETFD
) | FD_CLOEXEC
);
41 if (type
& USOCK_NONBLOCK
)
42 fcntl(sock
, F_SETFL
, fcntl(sock
, F_GETFL
) | O_NONBLOCK
);
45 static int usock_connect(int type
, struct sockaddr
*sa
, int sa_len
, int family
, int socktype
, bool server
)
49 sock
= socket(family
, socktype
, 0);
53 usock_set_flags(sock
, type
);
57 setsockopt(sock
, SOL_SOCKET
, SO_REUSEADDR
, &one
, sizeof(one
));
59 if (!bind(sock
, sa
, sa_len
) &&
60 (socktype
!= SOCK_STREAM
|| !listen(sock
, SOMAXCONN
)))
63 if (!connect(sock
, sa
, sa_len
) || errno
== EINPROGRESS
)
71 static int usock_unix(int type
, const char *host
)
73 struct sockaddr_un sun
= {.sun_family
= AF_UNIX
};
74 bool server
= !!(type
& USOCK_SERVER
);
75 int socktype
= ((type
& 0xff) == USOCK_TCP
) ? SOCK_STREAM
: SOCK_DGRAM
;
77 if (strlen(host
) >= sizeof(sun
.sun_path
)) {
81 strcpy(sun
.sun_path
, host
);
83 return usock_connect(type
, (struct sockaddr
*)&sun
, sizeof(sun
), AF_UNIX
, socktype
, server
);
87 usock_inet_notimeout(int type
, struct addrinfo
*result
, void *addr
)
90 int socktype
= ((type
& 0xff) == USOCK_TCP
) ? SOCK_STREAM
: SOCK_DGRAM
;
91 bool server
= !!(type
& USOCK_SERVER
);
94 for (rp
= result
; rp
!= NULL
; rp
= rp
->ai_next
) {
95 sock
= usock_connect(type
, rp
->ai_addr
, rp
->ai_addrlen
, rp
->ai_family
, socktype
, server
);
98 memcpy(addr
, rp
->ai_addr
, rp
->ai_addrlen
);
106 static int poll_restart(struct pollfd
*fds
, int nfds
, int timeout
)
108 struct timespec ts
, cur
;
109 int msec
= timeout
% 1000;
112 clock_gettime(CLOCK_MONOTONIC
, &ts
);
114 ts
.tv_nsec
+= msec
* 1000000;
115 if (ts
.tv_nsec
> 1000000000) {
117 ts
.tv_nsec
-= 1000000000;
119 ts
.tv_sec
+= timeout
/ 1000;
122 ret
= poll(fds
, nfds
, timeout
);
129 clock_gettime(CLOCK_MONOTONIC
, &cur
);
130 timeout
= (ts
.tv_sec
- cur
.tv_sec
) * 1000;
131 timeout
+= (ts
.tv_nsec
- cur
.tv_nsec
) / 1000000;
137 int usock_inet_timeout(int type
, const char *host
, const char *service
,
138 void *addr
, int timeout
)
140 int socktype
= ((type
& 0xff) == USOCK_TCP
) ? SOCK_STREAM
: SOCK_DGRAM
;
141 bool server
= !!(type
& USOCK_SERVER
);
142 struct addrinfo
*result
, *rp
;
143 struct addrinfo hints
= {
144 .ai_family
= (type
& USOCK_IPV6ONLY
) ? AF_INET6
:
145 (type
& USOCK_IPV4ONLY
) ? AF_INET
: AF_UNSPEC
,
146 .ai_socktype
= socktype
,
147 .ai_flags
= AI_ADDRCONFIG
148 | ((type
& USOCK_SERVER
) ? AI_PASSIVE
: 0)
149 | ((type
& USOCK_NUMERIC
) ? AI_NUMERICHOST
: 0),
151 struct addrinfo
*rp_v6
= NULL
;
152 struct addrinfo
*rp_v4
= NULL
;
153 struct pollfd pfds
[2] = {
154 { .fd
= -1, .events
= POLLOUT
},
155 { .fd
= -1, .events
= POLLOUT
},
160 if (getaddrinfo(host
, service
, &hints
, &result
))
163 if (timeout
<= 0 || server
) {
164 sock
= usock_inet_notimeout(type
, result
, addr
);
168 for (rp
= result
; rp
!= NULL
; rp
= rp
->ai_next
) {
169 if (rp
->ai_family
== AF_INET6
&& !rp_v6
)
171 if (rp
->ai_family
== AF_INET
&& !rp_v4
)
175 if (!rp_v6
&& !rp_v4
)
180 pfds
[0].fd
= usock_connect(type
| USOCK_NONBLOCK
, rp
->ai_addr
,
181 rp
->ai_addrlen
, rp
->ai_family
,
183 if (pfds
[0].fd
< 0) {
189 if (poll_restart(pfds
, 1, 300) == 1) {
201 pfds
[1].fd
= usock_connect(type
| USOCK_NONBLOCK
, rp
->ai_addr
,
202 rp
->ai_addrlen
, rp
->ai_family
,
204 if (pfds
[1].fd
< 0) {
213 poll_restart(pfds
+ !rp_v6
, !!rp_v6
+ !!rp_v4
, timeout
);
214 if (pfds
[0].revents
& POLLOUT
) {
220 if (pfds
[1].revents
& POLLOUT
) {
227 for (i
= 0; i
< 2; i
++) {
229 if (fd
>= 0 && fd
!= sock
)
233 if (!(type
& USOCK_NONBLOCK
))
234 fcntl(sock
, F_SETFL
, fcntl(sock
, F_GETFL
) & ~O_NONBLOCK
);
236 if (addr
&& sock
>= 0)
237 memcpy(addr
, rp
->ai_addr
, rp
->ai_addrlen
);
239 freeaddrinfo(result
);
243 const char *usock_port(int port
)
245 static char buffer
[sizeof("65535\0")];
247 if (port
< 0 || port
> 65535)
250 snprintf(buffer
, sizeof(buffer
), "%u", port
);
255 int usock(int type
, const char *host
, const char *service
) {
258 if (type
& USOCK_UNIX
)
259 sock
= usock_unix(type
, host
);
261 sock
= usock_inet(type
, host
, service
, NULL
);
269 int usock_wait_ready(int fd
, int msecs
) {
270 struct pollfd fds
[1];
274 fds
[0].events
= POLLOUT
;
276 res
= poll(fds
, 1, msecs
);
279 } else if (res
== 0) {
283 socklen_t optlen
= sizeof(err
);
285 res
= getsockopt(fd
, SOL_SOCKET
, SO_ERROR
, &err
, &optlen
);