1 /* Licensed to the Apache Software Foundation (ASF) under one or more
2 * contributor license agreements. See the NOTICE file distributed with
3 * this work for additional information regarding copyright ownership.
4 * The ASF licenses this file to You under the Apache License, Version 2.0
5 * (the "License"); you may not use this file except in compliance with
6 * the License. You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #include "apr_arch_networkio.h"
18 #include "apr_strings.h"
21 static apr_status_t soblock(int sd)
23 /* BeOS uses setsockopt at present for non blocking... */
27 fd_flags = fcntl(sd, F_GETFL, 0);
28 #if defined(O_NONBLOCK)
29 fd_flags &= ~O_NONBLOCK;
30 #elif defined(O_NDELAY)
31 fd_flags &= ~O_NDELAY;
32 #elif defined(FNDELAY)
35 #error Please teach APR how to make sockets blocking on your platform.
37 if (fcntl(sd, F_SETFL, fd_flags) == -1) {
42 if (setsockopt(sd, SOL_SOCKET, SO_NONBLOCK, &on, sizeof(int)) < 0)
48 static apr_status_t sononblock(int sd)
53 fd_flags = fcntl(sd, F_GETFL, 0);
54 #if defined(O_NONBLOCK)
55 fd_flags |= O_NONBLOCK;
56 #elif defined(O_NDELAY)
58 #elif defined(FNDELAY)
61 #error Please teach APR how to make sockets non-blocking on your platform.
63 if (fcntl(sd, F_SETFL, fd_flags) == -1) {
68 if (setsockopt(sd, SOL_SOCKET, SO_NONBLOCK, &on, sizeof(int)) < 0)
75 apr_status_t apr_socket_timeout_set(apr_socket_t *sock, apr_interval_time_t t)
79 /* If our new timeout is non-negative and our old timeout was
80 * negative, then we need to ensure that we are non-blocking.
81 * Conversely, if our new timeout is negative and we had
82 * non-negative timeout, we must make sure our socket is blocking.
83 * We want to avoid calling fcntl more than necessary on the
86 if (t >= 0 && sock->timeout < 0) {
87 if (apr_is_option_set(sock, APR_SO_NONBLOCK) != 1) {
88 if ((stat = sononblock(sock->socketdes)) != APR_SUCCESS) {
91 apr_set_option(sock, APR_SO_NONBLOCK, 1);
94 else if (t < 0 && sock->timeout >= 0) {
95 if (apr_is_option_set(sock, APR_SO_NONBLOCK) != 0) {
96 if ((stat = soblock(sock->socketdes)) != APR_SUCCESS) {
99 apr_set_option(sock, APR_SO_NONBLOCK, 0);
102 /* must disable the incomplete read support if we disable
106 sock->options &= ~APR_INCOMPLETE_READ;
113 apr_status_t apr_socket_opt_set(apr_socket_t *sock,
114 apr_int32_t opt, apr_int32_t on)
124 case APR_SO_KEEPALIVE:
126 if (on != apr_is_option_set(sock, APR_SO_KEEPALIVE)) {
127 if (setsockopt(sock->socketdes, SOL_SOCKET, SO_KEEPALIVE, (void *)&one, sizeof(int)) == -1) {
130 apr_set_option(sock, APR_SO_KEEPALIVE, on);
137 if (on != apr_is_option_set(sock, APR_SO_DEBUG)) {
138 if (setsockopt(sock->socketdes, SOL_SOCKET, SO_DEBUG, (void *)&one, sizeof(int)) == -1) {
141 apr_set_option(sock, APR_SO_DEBUG, on);
144 case APR_SO_REUSEADDR:
145 if (on != apr_is_option_set(sock, APR_SO_REUSEADDR)) {
146 if (setsockopt(sock->socketdes, SOL_SOCKET, SO_REUSEADDR, (void *)&one, sizeof(int)) == -1) {
149 apr_set_option(sock, APR_SO_REUSEADDR, on);
154 if (setsockopt(sock->socketdes, SOL_SOCKET, SO_SNDBUF, (void *)&on, sizeof(int)) == -1) {
163 if (setsockopt(sock->socketdes, SOL_SOCKET, SO_RCVBUF, (void *)&on, sizeof(int)) == -1) {
170 case APR_SO_NONBLOCK:
171 if (apr_is_option_set(sock, APR_SO_NONBLOCK) != on) {
173 if ((rv = sononblock(sock->socketdes)) != APR_SUCCESS)
177 if ((rv = soblock(sock->socketdes)) != APR_SUCCESS)
180 apr_set_option(sock, APR_SO_NONBLOCK, on);
185 if (apr_is_option_set(sock, APR_SO_LINGER) != on) {
188 li.l_linger = APR_MAX_SECS_TO_LINGER;
189 if (setsockopt(sock->socketdes, SOL_SOCKET, SO_LINGER, (char *) &li, sizeof(struct linger)) == -1) {
192 apr_set_option(sock, APR_SO_LINGER, on);
198 case APR_TCP_DEFER_ACCEPT:
199 #if defined(TCP_DEFER_ACCEPT)
200 if (apr_is_option_set(sock, APR_TCP_DEFER_ACCEPT) != on) {
201 int optlevel = IPPROTO_TCP;
202 int optname = TCP_DEFER_ACCEPT;
204 if (setsockopt(sock->socketdes, optlevel, optname,
205 (void *)&on, sizeof(int)) == -1) {
208 apr_set_option(sock, APR_TCP_DEFER_ACCEPT, on);
214 case APR_TCP_NODELAY:
215 #if defined(TCP_NODELAY)
216 if (apr_is_option_set(sock, APR_TCP_NODELAY) != on) {
217 int optlevel = IPPROTO_TCP;
218 int optname = TCP_NODELAY;
221 if (sock->protocol == IPPROTO_SCTP) {
222 optlevel = IPPROTO_SCTP;
223 optname = SCTP_NODELAY;
226 if (setsockopt(sock->socketdes, optlevel, optname, (void *)&on, sizeof(int)) == -1) {
229 apr_set_option(sock, APR_TCP_NODELAY, on);
232 /* BeOS pre-BONE has TCP_NODELAY set by default.
233 * As it can't be turned off we might as well check if they're asking
234 * for it to be turned on!
245 #if APR_TCP_NOPUSH_FLAG
246 /* TCP_NODELAY and TCP_CORK are mutually exclusive on Linux
247 * kernels < 2.6; on newer kernels they can be used together
248 * and TCP_CORK takes preference, which is the desired
249 * behaviour. On older kernels, TCP_NODELAY must be toggled
250 * to "off" whilst TCP_CORK is in effect. */
251 if (apr_is_option_set(sock, APR_TCP_NOPUSH) != on) {
252 #ifndef HAVE_TCP_NODELAY_WITH_CORK
253 int optlevel = IPPROTO_TCP;
254 int optname = TCP_NODELAY;
257 if (sock->protocol == IPPROTO_SCTP) {
258 optlevel = IPPROTO_SCTP;
259 optname = SCTP_NODELAY;
262 /* OK we're going to change some settings here... */
263 if (apr_is_option_set(sock, APR_TCP_NODELAY) == 1 && on) {
264 /* Now toggle TCP_NODELAY to off, if TCP_CORK is being
267 if (setsockopt(sock->socketdes, optlevel, optname,
268 (void*)&tmpflag, sizeof(int)) == -1) {
271 apr_set_option(sock, APR_RESET_NODELAY, 1);
272 apr_set_option(sock, APR_TCP_NODELAY, 0);
274 apr_set_option(sock, APR_RESET_NODELAY, 0);
276 #endif /* HAVE_TCP_NODELAY_WITH_CORK */
278 /* OK, now we can just set the TCP_NOPUSH flag accordingly...*/
279 if (setsockopt(sock->socketdes, IPPROTO_TCP, APR_TCP_NOPUSH_FLAG,
280 (void*)&on, sizeof(int)) == -1) {
283 apr_set_option(sock, APR_TCP_NOPUSH, on);
284 #ifndef HAVE_TCP_NODELAY_WITH_CORK
285 if (!on && apr_is_option_set(sock, APR_RESET_NODELAY)) {
286 /* Now, if TCP_CORK was just turned off, turn
287 * TCP_NODELAY back on again if it was earlier toggled
290 if (setsockopt(sock->socketdes, optlevel, optname,
291 (void*)&tmpflag, sizeof(int)) == -1) {
294 apr_set_option(sock, APR_RESET_NODELAY,0);
295 apr_set_option(sock, APR_TCP_NODELAY, 1);
297 #endif /* HAVE_TCP_NODELAY_WITH_CORK */
303 case APR_INCOMPLETE_READ:
304 apr_set_option(sock, APR_INCOMPLETE_READ, on);
306 case APR_IPV6_V6ONLY:
307 #if APR_HAVE_IPV6 && defined(IPV6_V6ONLY)
308 /* we don't know the initial setting of this option,
309 * so don't check sock->options since that optimization
312 if (setsockopt(sock->socketdes, IPPROTO_IPV6, IPV6_V6ONLY,
313 (void *)&on, sizeof(int)) == -1) {
316 apr_set_option(sock, APR_IPV6_V6ONLY, on);
329 apr_status_t apr_socket_timeout_get(apr_socket_t *sock, apr_interval_time_t *t)
336 apr_status_t apr_socket_opt_get(apr_socket_t *sock,
337 apr_int32_t opt, apr_int32_t *on)
341 *on = apr_is_option_set(sock, opt);
347 apr_status_t apr_socket_atmark(apr_socket_t *sock, int *atmark)
352 if (ioctl(sock->socketdes, SIOCATMARK, (void*) &oobmark) < 0)
353 return apr_get_netos_error();
355 *atmark = (oobmark != 0);
363 apr_status_t apr_gethostname(char *buf, apr_int32_t len, apr_pool_t *cont)
366 if (gethostname(buf, len) == 0) {
368 if (gethostname(buf, len) != 0) {
373 else if (!memchr(buf, '\0', len)) { /* buffer too small */
374 /* note... most platforms just truncate in this condition
375 * linux+glibc return an error
378 return APR_ENAMETOOLONG;
383 #if APR_HAS_SO_ACCEPTFILTER
384 apr_status_t apr_socket_accept_filter(apr_socket_t *sock, char *nonconst_name,
387 /* these should have been const; act like they are */
388 const char *name = nonconst_name;
389 const char *args = nonconst_args;
391 struct accept_filter_arg af;
392 socklen_t optlen = sizeof(af);
394 /* FreeBSD returns an error if the filter is already set; ignore
395 * this call if we previously set it to the same value.
397 if ((getsockopt(sock->socketdes, SOL_SOCKET, SO_ACCEPTFILTER,
398 &af, &optlen)) == 0) {
399 if (!strcmp(name, af.af_name) && !strcmp(args, af.af_arg)) {
404 /* Uhh, at least in FreeBSD 9 the fields are declared as arrays of
405 * these lengths; did sizeof not work in some ancient release?
407 * FreeBSD kernel sets the last byte to a '\0'.
409 apr_cpystrn(af.af_name, name, 16);
410 apr_cpystrn(af.af_arg, args, 256 - 16);
412 if ((setsockopt(sock->socketdes, SOL_SOCKET, SO_ACCEPTFILTER,
413 &af, sizeof(af))) < 0) {