/* Copyright 2000-2004 The Apache Software Foundation * * 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 "apr_arch_networkio.h" #include "apr_strings.h" static apr_status_t soblock(int sd) ###### { /* BeOS uses setsockopt at present for non blocking... */ #ifndef BEOS ###### int fd_flags; ###### fd_flags = fcntl(sd, F_GETFL, 0); #if defined(O_NONBLOCK) ###### fd_flags &= ~O_NONBLOCK; #elif defined(O_NDELAY) fd_flags &= ~O_NDELAY; #elif defined(FNDELAY) fd_flags &= ~FNDELAY; #else #error Please teach APR how to make sockets blocking on your platform. #endif ###### if (fcntl(sd, F_SETFL, fd_flags) == -1) { ###### return errno; } #else int on = 0; if (setsockopt(sd, SOL_SOCKET, SO_NONBLOCK, &on, sizeof(int)) < 0) return errno; #endif /* BEOS */ ###### return APR_SUCCESS; } static apr_status_t sononblock(int sd) 3 { #ifndef BEOS 3 int fd_flags; 3 fd_flags = fcntl(sd, F_GETFL, 0); #if defined(O_NONBLOCK) 3 fd_flags |= O_NONBLOCK; #elif defined(O_NDELAY) fd_flags |= O_NDELAY; #elif defined(FNDELAY) fd_flags |= FNDELAY; #else #error Please teach APR how to make sockets non-blocking on your platform. #endif 3 if (fcntl(sd, F_SETFL, fd_flags) == -1) { ###### return errno; } #else int on = 1; if (setsockopt(sd, SOL_SOCKET, SO_NONBLOCK, &on, sizeof(int)) < 0) return errno; #endif /* BEOS */ 3 return APR_SUCCESS; } apr_status_t apr_socket_timeout_set(apr_socket_t *sock, apr_interval_time_t t) 3 { 3 apr_status_t stat; /* If our timeout is positive or zero and our last timeout was * negative, then we need to ensure that we are non-blocking. * Conversely, if our timeout is negative and we had a positive * or zero timeout, we must make sure our socket is blocking. * We want to avoid calling fcntl more than necessary on the socket, */ 3 if (t >= 0 && sock->timeout < 0) { 3 if (apr_is_option_set(sock, APR_SO_NONBLOCK) != 1) { 3 if ((stat = sononblock(sock->socketdes)) != APR_SUCCESS) { ###### return stat; } 3 apr_set_option(sock, APR_SO_NONBLOCK, 1); } } ###### else if (t < 0 && sock->timeout >= 0) { ###### if (apr_is_option_set(sock, APR_SO_NONBLOCK) != 0) { ###### if ((stat = soblock(sock->socketdes)) != APR_SUCCESS) { ###### return stat; } ###### apr_set_option(sock, APR_SO_NONBLOCK, 0); } } /* must disable the incomplete read support if we disable * a timeout */ 3 if (t <= 0) { ###### sock->options &= ~APR_INCOMPLETE_READ; } 3 sock->timeout = t; 3 return APR_SUCCESS; } apr_status_t apr_socket_opt_set(apr_socket_t *sock, apr_int32_t opt, apr_int32_t on) 6 { 6 int one; 6 apr_status_t rv; 6 if (on) 4 one = 1; else 2 one = 0; 6 switch(opt) { case APR_SO_KEEPALIVE: #ifdef SO_KEEPALIVE 2 if (on != apr_is_option_set(sock, APR_SO_KEEPALIVE)) { 2 if (setsockopt(sock->socketdes, SOL_SOCKET, SO_KEEPALIVE, (void *)&one, sizeof(int)) == -1) { ###### return errno; } 2 apr_set_option(sock, APR_SO_KEEPALIVE, on); } #else return APR_ENOTIMPL; #endif 1 break; case APR_SO_DEBUG: 1 if (on != apr_is_option_set(sock, APR_SO_DEBUG)) { 1 if (setsockopt(sock->socketdes, SOL_SOCKET, SO_DEBUG, (void *)&one, sizeof(int)) == -1) { 1 return errno; } ###### apr_set_option(sock, APR_SO_DEBUG, on); } ###### break; case APR_SO_REUSEADDR: ###### if (on != apr_is_option_set(sock, APR_SO_REUSEADDR)) { ###### if (setsockopt(sock->socketdes, SOL_SOCKET, SO_REUSEADDR, (void *)&one, sizeof(int)) == -1) { ###### return errno; } ###### apr_set_option(sock, APR_SO_REUSEADDR, on); } ###### break; case APR_SO_SNDBUF: #ifdef SO_SNDBUF ###### if (apr_is_option_set(sock, APR_SO_SNDBUF) != on) { ###### if (setsockopt(sock->socketdes, SOL_SOCKET, SO_SNDBUF, (void *)&on, sizeof(int)) == -1) { ###### return errno; } ###### apr_set_option(sock, APR_SO_SNDBUF, on); } #else return APR_ENOTIMPL; #endif ###### break; case APR_SO_RCVBUF: #ifdef SO_RCVBUF ###### if (apr_is_option_set(sock, APR_SO_RCVBUF) != on) { ###### if (setsockopt(sock->socketdes, SOL_SOCKET, SO_RCVBUF, (void *)&on, sizeof(int)) == -1) { ###### return errno; } ###### apr_set_option(sock, APR_SO_RCVBUF, on); } #else return APR_ENOTIMPL; #endif ###### break; case APR_SO_NONBLOCK: ###### if (apr_is_option_set(sock, APR_SO_NONBLOCK) != on) { ###### if (on) { ###### if ((rv = sononblock(sock->socketdes)) != APR_SUCCESS) ###### return rv; } else { ###### if ((rv = soblock(sock->socketdes)) != APR_SUCCESS) ###### return rv; } ###### apr_set_option(sock, APR_SO_NONBLOCK, on); } ###### break; case APR_SO_LINGER: #ifdef SO_LINGER ###### if (apr_is_option_set(sock, APR_SO_LINGER) != on) { ###### struct linger li; ###### li.l_onoff = on; ###### li.l_linger = APR_MAX_SECS_TO_LINGER; ###### if (setsockopt(sock->socketdes, SOL_SOCKET, SO_LINGER, (char *) &li, sizeof(struct linger)) == -1) { ###### return errno; } ###### apr_set_option(sock, APR_SO_LINGER, on); } #else return APR_ENOTIMPL; #endif ###### break; case APR_TCP_NODELAY: #if defined(TCP_NODELAY) 1 if (apr_is_option_set(sock, APR_TCP_NODELAY) != on) { 1 int optlevel = IPPROTO_TCP; 1 int optname = TCP_NODELAY; #if APR_HAVE_SCTP if (sock->protocol == IPPROTO_SCTP) { optlevel = IPPROTO_SCTP; optname = SCTP_NODELAY; } #endif 1 if (setsockopt(sock->socketdes, optlevel, optname, (void *)&on, sizeof(int)) == -1) { ###### return errno; } 1 apr_set_option(sock, APR_TCP_NODELAY, on); } #else /* BeOS pre-BONE has TCP_NODELAY set by default. * As it can't be turned off we might as well check if they're asking * for it to be turned on! */ #ifdef BEOS if (on == 1) return APR_SUCCESS; else #endif return APR_ENOTIMPL; #endif ###### break; case APR_TCP_NOPUSH: #if APR_TCP_NOPUSH_FLAG 2 if (apr_is_option_set(sock, APR_TCP_NOPUSH) != on) { 2 int optlevel = IPPROTO_TCP; 2 int optname = TCP_NODELAY; #if APR_HAVE_SCTP if (sock->protocol == IPPROTO_SCTP) { optlevel = IPPROTO_SCTP; optname = SCTP_NODELAY; } #endif /* OK we're going to change some settings here... */ /* TCP_NODELAY is mutually exclusive, so do we have it set? */ 2 if (apr_is_option_set(sock, APR_TCP_NODELAY) == 1 && on) { /* If we want to set NOPUSH then if we have the TCP_NODELAY * flag set we need to switch it off... */ 1 int tmpflag = 0; 1 if (setsockopt(sock->socketdes, optlevel, optname, (void*)&tmpflag, sizeof(int)) == -1) { ###### return errno; } 1 apr_set_option(sock, APR_RESET_NODELAY, 1); 1 apr_set_option(sock, APR_TCP_NODELAY, 0); 1 } else if (on) { ###### apr_set_option(sock, APR_RESET_NODELAY, 0); } /* OK, now we can just set the TCP_NOPUSH flag accordingly...*/ 2 if (setsockopt(sock->socketdes, IPPROTO_TCP, APR_TCP_NOPUSH_FLAG, (void*)&on, sizeof(int)) == -1) { ###### return errno; } 2 apr_set_option(sock, APR_TCP_NOPUSH, on); 2 if (!on && apr_is_option_set(sock, APR_RESET_NODELAY)) { 1 int tmpflag = 1; 1 if (setsockopt(sock->socketdes, optlevel, optname, (void*)&tmpflag, sizeof(int)) == -1) { ###### return errno; } 1 apr_set_option(sock, APR_RESET_NODELAY,0); 1 apr_set_option(sock, APR_TCP_NODELAY, 1); } } #else return APR_ENOTIMPL; #endif 1 break; case APR_INCOMPLETE_READ: ###### apr_set_option(sock, APR_INCOMPLETE_READ, on); ###### break; case APR_IPV6_V6ONLY: #if APR_HAVE_IPV6 && defined(IPV6_V6ONLY) /* we don't know the initial setting of this option, * so don't check/set sock->options since that optimization * won't work */ if (setsockopt(sock->socketdes, IPPROTO_IPV6, IPV6_V6ONLY, (void *)&on, sizeof(int)) == -1) { return errno; } #else ###### return APR_ENOTIMPL; #endif ###### break; default: ###### return APR_EINVAL; } 5 return APR_SUCCESS; } apr_status_t apr_socket_timeout_get(apr_socket_t *sock, apr_interval_time_t *t) ###### { ###### *t = sock->timeout; ###### return APR_SUCCESS; } apr_status_t apr_socket_opt_get(apr_socket_t *sock, apr_int32_t opt, apr_int32_t *on) 8 { 8 switch(opt) { default: 8 *on = apr_is_option_set(sock, opt); } 8 return APR_SUCCESS; } apr_status_t apr_socket_atmark(apr_socket_t *sock, int *atmark) ###### { ###### int oobmark; ###### if (ioctl(sock->socketdes, SIOCATMARK, (void*) &oobmark) < 0) ###### return apr_get_netos_error(); ###### *atmark = (oobmark != 0); ###### return APR_SUCCESS; } apr_status_t apr_gethostname(char *buf, apr_int32_t len, apr_pool_t *cont) ###### { ###### if (gethostname(buf, len) == -1) { ###### buf[0] = '\0'; ###### return errno; } ###### else if (!memchr(buf, '\0', len)) { /* buffer too small */ /* note... most platforms just truncate in this condition * linux+glibc return an error */ ###### buf[0] = '\0'; ###### return APR_ENAMETOOLONG; } ###### return APR_SUCCESS; } #if APR_HAS_SO_ACCEPTFILTER apr_status_t apr_socket_accept_filter(apr_socket_t *sock, char *name, char *args) { struct accept_filter_arg af; strncpy(af.af_name, name, 16); strncpy(af.af_arg, args, 256 - 16); if ((setsockopt(sock->socketdes, SOL_SOCKET, SO_ACCEPTFILTER, &af, sizeof(af))) < 0) { return errno; } return APR_SUCCESS; } #endif