sjero.net Git - iperf/blob - src/SocketAddr.c

   1 /*---------------------------------------------------------------
   2  * Copyright (c) 1999,2000,2001,2002,2003
   3  * The Board of Trustees of the University of Illinois
   4  * All Rights Reserved.
   5  *---------------------------------------------------------------
   6  * Permission is hereby granted, free of charge, to any person
   7  * obtaining a copy of this software (Iperf) and associated
   8  * documentation files (the "Software"), to deal in the Software
   9  * without restriction, including without limitation the
  10  * rights to use, copy, modify, merge, publish, distribute,
  11  * sublicense, and/or sell copies of the Software, and to permit
  12  * persons to whom the Software is furnished to do
  13  * so, subject to the following conditions:
  14  *
  15  *
  16  * Redistributions of source code must retain the above
  17  * copyright notice, this list of conditions and
  18  * the following disclaimers.
  19  *
  20  *
  21  * Redistributions in binary form must reproduce the above
  22  * copyright notice, this list of conditions and the following
  23  * disclaimers in the documentation and/or other materials
  24  * provided with the distribution.
  25  *
  26  *
  27  * Neither the names of the University of Illinois, NCSA,
  28  * nor the names of its contributors may be used to endorse
  29  * or promote products derived from this Software without
  30  * specific prior written permission.
  31  *
  32  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  33  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
  34  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  35  * NONINFRINGEMENT. IN NO EVENT SHALL THE CONTIBUTORS OR COPYRIGHT
  36  * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  37  * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  38  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  39  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  40  * ________________________________________________________________
  41  * National Laboratory for Applied Network Research
  42  * National Center for Supercomputing Applications
  43  * University of Illinois at Urbana-Champaign
  44  * http://www.ncsa.uiuc.edu
  45  * ________________________________________________________________
  46  *
  47  * Socket.cpp
  48  * by       Ajay Tirumala <tirumala@ncsa.uiuc.edu>
  49  * and      Mark Gates <mgates@nlanr.net>
  50  * ------------------------------------------------------------------- */
  51
  52 #define HEADERS()
  53
  54 #include "headers.h"
  55
  56 #include "SocketAddr.h"
  57
  58
  59 #ifdef __cplusplus
  60 extern "C" {
  61 #endif
  62
  63 /* return the port of @ss, -1 if undefined */
  64 int SockAddr_port(struct sockaddr_storage *ss)
  65 {
  66         if (ss->ss_family == AF_INET)
  67                 return ntohs(((struct sockaddr_in *)ss)->sin_port);
  68         else if (ss->ss_family == AF_INET6)
  69                 return ntohs(((struct sockaddr_in6 *)ss)->sin6_port);
  70         return -1;
  71 }
  72
  73 /* -------------------------------------------------------------------
  74  * Print socket name into output (must be INET6_ADDRSTRLEN long).
  75  * This one works around the quirk of inet_ntop/getaddrinfo to represent
  76  * print IPv4-mapped-IPv6 addresses as sockaddr_in6 (::FFFF:x.x.x.x).
  77  * Make the #ifndef an `#ifdef' to disable this conversion.
  78  * ------------------------------------------------------------------- */
  79 const char *SockAddr_name(struct sockaddr_storage *ss, char *hbuf, size_t hbuflen)
  80 {
  81         int af = ss->ss_family;
  82         const void *ptr;
  83
  84         if (ss->ss_family == AF_INET)
  85                 ptr = &((struct sockaddr_in *)ss)->sin_addr;
  86         else if (ss->ss_family == AF_INET6) {
  87                 const struct in6_addr *v6_addr;
  88
  89                 ptr = v6_addr = &((struct sockaddr_in6 *)ss)->sin6_addr;
  90 #ifndef I_WANT_V4_MAPPED_IN_MY_OUTPUT
  91                 if (IN6_IS_ADDR_V4MAPPED(v6_addr)) {
  92                         struct in_addr ia;
  93
  94                         // extract the IPv4 address from the last 4 bytes
  95                         memcpy(&ia.s_addr, &v6_addr->s6_addr[12], 4);
  96                         ptr = &ia;
  97                         af  = AF_INET;
  98                 }
  99 #endif
 100         } else {
 101                 return NULL;    // undetermined family (e.g. AF_UNSPEC)
 102         }
 103         return inet_ntop(af, ptr, hbuf, hbuflen);
 104 }
 105
 106 static int SockAddr_AF_are_Equal(struct sockaddr_storage *first, struct sockaddr_storage *second)
 107 {
 108         return first->ss_family == second->ss_family;
 109 }
 110
 111 /* -------------------------------------------------------------------
 112  * Compare two sockaddrs and return true if the hosts are equal
 113  * ------------------------------------------------------------------- */
 114 int SockAddr_Hostare_Equal(struct sockaddr_storage *first, struct sockaddr_storage *second)
 115 {
 116         if (!SockAddr_AF_are_Equal(first, second))
 117                 return 0;
 118         if (first->ss_family == AF_INET)
 119                 return !memcmp(&((struct sockaddr_in *)first)->sin_addr,
 120                                &((struct sockaddr_in *)second)->sin_addr, sizeof(struct in_addr));
 121         if (first->ss_family == AF_INET6)
 122                 return !memcmp(&((struct sockaddr_in6 *)first)->sin6_addr,
 123                                &((struct sockaddr_in6 *)second)->sin6_addr, sizeof(struct in6_addr));
 124         return 0;
 125 }
 126
 127 /* -------------------------------------------------------------------
 128  * Compare two socket structs and return true if they are equal
 129  * ------------------------------------------------------------------- */
 130 int SockAddr_are_Equal(struct sockaddr_storage *first, struct sockaddr_storage *second)
 131 {
 132         if (!SockAddr_Hostare_Equal(first, second))
 133                 return 0;
 134         if (first->ss_family == AF_INET)
 135                 return ((struct sockaddr_in*)first)->sin_port == ((struct sockaddr_in*)second)->sin_port;
 136         if (first->ss_family == AF_INET6)
 137                 return (((struct sockaddr_in6*)first)->sin6_port == ((struct sockaddr_in6*)second)->sin6_port);
 138         return 0;
 139 }
 140
 141 #ifdef __cplusplus
 142 } /* end extern "C" */
 143 #endif