/* * Copyright (c) 1999 - 2000 Kungliga Tekniska Högskolan * (Royal Institute of Technology, Stockholm, Sweden). * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * 3. Neither the name of the Institute nor the names of its contributors * may be used to endorse or promote products derived from this software * without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. */ #ifdef HAVE_CONFIG_H #include RCSID("$KTH: socket.c,v 1.5 2000/07/27 04:41:06 assar Exp $"); #endif #include #include #include /* * Set `sa' to the unitialized address of address family `af' */ void socket_set_any (struct sockaddr *sa, int af) { switch (af) { case AF_INET : { struct sockaddr_in *sin = (struct sockaddr_in *)sa; memset (sin, 0, sizeof(*sin)); sin->sin_family = AF_INET; sin->sin_port = 0; sin->sin_addr.s_addr = INADDR_ANY; break; } #ifdef HAVE_IPV6 case AF_INET6 : { struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sa; memset (sin6, 0, sizeof(*sin6)); sin6->sin6_family = AF_INET6; sin6->sin6_port = 0; sin6->sin6_addr = in6addr_any; break; } #endif default : errx (1, "unknown address family %d", sa->sa_family); break; } } /* * set `sa' to (`ptr', `port') */ void socket_set_address_and_port (struct sockaddr *sa, const void *ptr, int port) { switch (sa->sa_family) { case AF_INET : { struct sockaddr_in *sin = (struct sockaddr_in *)sa; memset (sin, 0, sizeof(*sin)); sin->sin_family = AF_INET; sin->sin_port = port; memcpy (&sin->sin_addr, ptr, sizeof(struct in_addr)); break; } #ifdef HAVE_IPV6 case AF_INET6 : { struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sa; memset (sin6, 0, sizeof(*sin6)); sin6->sin6_family = AF_INET6; sin6->sin6_port = port; memcpy (&sin6->sin6_addr, ptr, sizeof(struct in6_addr)); break; } #endif default : errx (1, "unknown address family %d", sa->sa_family); break; } } /* * Return the size of an address of the type in `sa' */ size_t socket_addr_size (const struct sockaddr *sa) { switch (sa->sa_family) { case AF_INET : return sizeof(struct in_addr); #ifdef HAVE_IPV6 case AF_INET6 : return sizeof(struct in6_addr); #endif default : errx (1, "unknown address family %d", sa->sa_family); break; } } /* * Return the size of a `struct sockaddr' in `sa'. */ size_t socket_sockaddr_size (const struct sockaddr *sa) { switch (sa->sa_family) { case AF_INET : return sizeof(struct sockaddr_in); #ifdef HAVE_IPV6 case AF_INET6 : return sizeof(struct sockaddr_in6); #endif default : errx (1, "unknown address family %d", sa->sa_family); break; } } /* * Return the binary address of `sa'. */ void * socket_get_address (struct sockaddr *sa) { switch (sa->sa_family) { case AF_INET : { struct sockaddr_in *sin = (struct sockaddr_in *)sa; return &sin->sin_addr; } #ifdef HAVE_IPV6 case AF_INET6 : { struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sa; return &sin6->sin6_addr; } #endif default : errx (1, "unknown address family %d", sa->sa_family); break; } } /* * Return the port number from `sa'. */ int socket_get_port (const struct sockaddr *sa) { switch (sa->sa_family) { case AF_INET : { const struct sockaddr_in *sin = (const struct sockaddr_in *)sa; return sin->sin_port; } #ifdef HAVE_IPV6 case AF_INET6 : { const struct sockaddr_in6 *sin6 = (const struct sockaddr_in6 *)sa; return sin6->sin6_port; } #endif default : errx (1, "unknown address family %d", sa->sa_family); break; } } /* * Set the port in `sa' to `port'. */ void socket_set_port (struct sockaddr *sa, int port) { switch (sa->sa_family) { case AF_INET : { struct sockaddr_in *sin = (struct sockaddr_in *)sa; sin->sin_port = port; break; } #ifdef HAVE_IPV6 case AF_INET6 : { struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sa; sin6->sin6_port = port; break; } #endif default : errx (1, "unknown address family %d", sa->sa_family); break; } } /* * Enable debug on `sock'. */ void socket_set_debug (int sock) { #if defined(SO_DEBUG) && defined(HAVE_SETSOCKOPT) int on = 1; if (setsockopt (sock, SOL_SOCKET, SO_DEBUG, (void *) &on, sizeof (on)) < 0) warn ("setsockopt SO_DEBUG (ignored)"); #endif } /* * Set the type-of-service of `sock' to `tos'. */ void socket_set_tos (int sock, int tos) { #if defined(IP_TOS) && defined(HAVE_SETSOCKOPT) if (setsockopt (sock, IPPROTO_IP, IP_TOS, (void *) &tos, sizeof (int)) < 0) warn ("setsockopt TOS (ignored)"); #endif } /* * set the reuse of addresses on `sock' to `val'. */ void socket_set_reuseaddr (int sock, int val) { #if defined(SO_REUSEADDR) && defined(HAVE_SETSOCKOPT) if(setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void *)&val, sizeof(val)) < 0) err (1, "setsockopt SO_REUSEADDR"); #endif }