lib/socket/interfaces.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    return a list of network interfaces
   4    Copyright (C) Andrew Tridgell 1998
   5    Copyright (C) Jeremy Allison 2007
   6    Copyright (C) Jelmer Vernooij 2007
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 3 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22
  23 #include "includes.h"
  24 #include "system/network.h"
  25 #include "interfaces.h"
  26 #include "lib/util/tsort.h"
  27
  28 /****************************************************************************
  29  Create a struct sockaddr_storage with the netmask bits set to 1.
  30 ****************************************************************************/
  31
  32 bool make_netmask(struct sockaddr_storage *pss_out,
  33                         const struct sockaddr_storage *pss_in,
  34                         unsigned long masklen)
  35 {
  36         *pss_out = *pss_in;
  37         /* Now apply masklen bits of mask. */
  38 #if defined(HAVE_IPV6)
  39         if (pss_in->ss_family == AF_INET6) {
  40                 char *p = (char *)&((struct sockaddr_in6 *)pss_out)->sin6_addr;
  41                 unsigned int i;
  42
  43                 if (masklen > 128) {
  44                         return false;
  45                 }
  46                 for (i = 0; masklen >= 8; masklen -= 8, i++) {
  47                         *p++ = 0xff;
  48                 }
  49                 /* Deal with the partial byte. */
  50                 *p++ &= (0xff & ~(0xff>>masklen));
  51                 i++;
  52                 for (;i < sizeof(struct in6_addr); i++) {
  53                         *p++ = '\0';
  54                 }
  55                 return true;
  56         }
  57 #endif
  58         if (pss_in->ss_family == AF_INET) {
  59                 if (masklen > 32) {
  60                         return false;
  61                 }
  62                 ((struct sockaddr_in *)pss_out)->sin_addr.s_addr =
  63                         htonl(((0xFFFFFFFFL >> masklen) ^ 0xFFFFFFFFL));
  64                 return true;
  65         }
  66         return false;
  67 }
  68
  69 /****************************************************************************
  70  Create a struct sockaddr_storage set to the broadcast or network adress from
  71  an incoming sockaddr_storage.
  72 ****************************************************************************/
  73
  74 static void make_bcast_or_net(struct sockaddr_storage *pss_out,
  75                         const struct sockaddr_storage *pss_in,
  76                         const struct sockaddr_storage *nmask,
  77                         bool make_bcast_p)
  78 {
  79         unsigned int i = 0, len = 0;
  80         const char *pmask = NULL;
  81         char *p = NULL;
  82         *pss_out = *pss_in;
  83
  84         /* Set all zero netmask bits to 1. */
  85 #if defined(HAVE_IPV6)
  86         if (pss_in->ss_family == AF_INET6) {
  87                 p = (char *)&((struct sockaddr_in6 *)pss_out)->sin6_addr;
  88                 pmask = (const char *)&((const struct sockaddr_in6 *)nmask)->sin6_addr;
  89                 len = 16;
  90         }
  91 #endif
  92         if (pss_in->ss_family == AF_INET) {
  93                 p = (char *)&((struct sockaddr_in *)pss_out)->sin_addr;
  94                 pmask = (const char *)&((const struct sockaddr_in *)nmask)->sin_addr;
  95                 len = 4;
  96         }
  97
  98         for (i = 0; i < len; i++, p++, pmask++) {
  99                 if (make_bcast_p) {
 100                         *p = (*p & *pmask) | (*pmask ^ 0xff);
 101                 } else {
 102                         /* make_net */
 103                         *p = (*p & *pmask);
 104                 }
 105         }
 106 }
 107
 108 void make_bcast(struct sockaddr_storage *pss_out,
 109                         const struct sockaddr_storage *pss_in,
 110                         const struct sockaddr_storage *nmask)
 111 {
 112         make_bcast_or_net(pss_out, pss_in, nmask, true);
 113 }
 114
 115 void make_net(struct sockaddr_storage *pss_out,
 116                         const struct sockaddr_storage *pss_in,
 117                         const struct sockaddr_storage *nmask)
 118 {
 119         make_bcast_or_net(pss_out, pss_in, nmask, false);
 120 }
 121
 122
 123 /****************************************************************************
 124  Try the "standard" getifaddrs/freeifaddrs interfaces.
 125  Also gets IPv6 interfaces.
 126 ****************************************************************************/
 127
 128 /****************************************************************************
 129  Get the netmask address for a local interface.
 130 ****************************************************************************/
 131
 132 static int _get_interfaces(TALLOC_CTX *mem_ctx, struct iface_struct **pifaces)
 133 {
 134         struct iface_struct *ifaces;
 135         struct ifaddrs *iflist = NULL;
 136         struct ifaddrs *ifptr = NULL;
 137         int count;
 138         int total = 0;
 139         size_t copy_size;
 140
 141         if (getifaddrs(&iflist) < 0) {
 142                 return -1;
 143         }
 144
 145         count = 0;
 146         for (ifptr = iflist; ifptr != NULL; ifptr = ifptr->ifa_next) {
 147                 if (!ifptr->ifa_addr || !ifptr->ifa_netmask) {
 148                         continue;
 149                 }
 150                 if (!(ifptr->ifa_flags & IFF_UP)) {
 151                         continue;
 152                 }
 153                 count += 1;
 154         }
 155
 156         ifaces = talloc_array(mem_ctx, struct iface_struct, count);
 157         if (ifaces == NULL) {
 158                 errno = ENOMEM;
 159                 return -1;
 160         }
 161
 162         /* Loop through interfaces, looking for given IP address */
 163         for (ifptr = iflist; ifptr != NULL; ifptr = ifptr->ifa_next) {
 164
 165                 if (!ifptr->ifa_addr || !ifptr->ifa_netmask) {
 166                         continue;
 167                 }
 168
 169                 /* Check the interface is up. */
 170                 if (!(ifptr->ifa_flags & IFF_UP)) {
 171                         continue;
 172                 }
 173
 174                 memset(&ifaces[total], '\0', sizeof(ifaces[total]));
 175
 176                 copy_size = sizeof(struct sockaddr_in);
 177
 178                 ifaces[total].flags = ifptr->ifa_flags;
 179
 180 #if defined(HAVE_IPV6)
 181                 if (ifptr->ifa_addr->sa_family == AF_INET6) {
 182                         copy_size = sizeof(struct sockaddr_in6);
 183                 }
 184 #endif
 185
 186                 memcpy(&ifaces[total].ip, ifptr->ifa_addr, copy_size);
 187                 memcpy(&ifaces[total].netmask, ifptr->ifa_netmask, copy_size);
 188
 189                 /* calculate broadcast address */
 190 #if defined(HAVE_IPV6)
 191                 if (ifptr->ifa_addr->sa_family == AF_INET6) {
 192                         struct sockaddr_in6 *sin6 =
 193                                 (struct sockaddr_in6 *)ifptr->ifa_addr;
 194                         struct in6_addr *in6 =
 195                                 (struct in6_addr *)&sin6->sin6_addr;
 196
 197                         if (IN6_IS_ADDR_LINKLOCAL(in6) || IN6_IS_ADDR_V4COMPAT(in6)) {
 198                                 continue;
 199                         }
 200                         /* IPv6 does not have broadcast it uses multicast. */
 201                         memset(&ifaces[total].bcast, '\0', copy_size);
 202                 } else
 203 #endif
 204                 if (ifaces[total].flags & (IFF_BROADCAST|IFF_LOOPBACK)) {
 205                         make_bcast(&ifaces[total].bcast,
 206                                 &ifaces[total].ip,
 207                                 &ifaces[total].netmask);
 208                 } else if ((ifaces[total].flags & IFF_POINTOPOINT) &&
 209                                ifptr->ifa_dstaddr ) {
 210                         memcpy(&ifaces[total].bcast,
 211                                 ifptr->ifa_dstaddr,
 212                                 copy_size);
 213                 } else {
 214                         continue;
 215                 }
 216
 217                 if (strlcpy(ifaces[total].name, ifptr->ifa_name,
 218                         sizeof(ifaces[total].name)) >=
 219                                 sizeof(ifaces[total].name)) {
 220                         /* Truncation ! Ignore. */
 221                         continue;
 222                 }
 223                 total++;
 224         }
 225
 226         freeifaddrs(iflist);
 227
 228         *pifaces = ifaces;
 229         return total;
 230 }
 231
 232 static int iface_comp(struct iface_struct *i1, struct iface_struct *i2)
 233 {
 234         int r;
 235
 236 #if defined(HAVE_IPV6)
 237         /*
 238          * If we have IPv6 - sort these interfaces lower
 239          * than any IPv4 ones.
 240          */
 241         if (i1->ip.ss_family == AF_INET6 &&
 242                         i2->ip.ss_family == AF_INET) {
 243                 return -1;
 244         } else if (i1->ip.ss_family == AF_INET &&
 245                         i2->ip.ss_family == AF_INET6) {
 246                 return 1;
 247         }
 248
 249         if (i1->ip.ss_family == AF_INET6) {
 250                 struct sockaddr_in6 *s1 = (struct sockaddr_in6 *)&i1->ip;
 251                 struct sockaddr_in6 *s2 = (struct sockaddr_in6 *)&i2->ip;
 252
 253                 r = memcmp(&s1->sin6_addr,
 254                                 &s2->sin6_addr,
 255                                 sizeof(struct in6_addr));
 256                 if (r) {
 257                         return r;
 258                 }
 259
 260                 s1 = (struct sockaddr_in6 *)&i1->netmask;
 261                 s2 = (struct sockaddr_in6 *)&i2->netmask;
 262
 263                 r = memcmp(&s1->sin6_addr,
 264                                 &s2->sin6_addr,
 265                                 sizeof(struct in6_addr));
 266                 if (r) {
 267                         return r;
 268                 }
 269         }
 270 #endif
 271
 272         /* AIX uses __ss_family instead of ss_family inside of
 273            sockaddr_storage. Instead of trying to figure out which field to
 274            use, we can just cast it to a sockaddr.
 275          */
 276
 277         if (((struct sockaddr *)&i1->ip)->sa_family == AF_INET) {
 278                 struct sockaddr_in *s1 = (struct sockaddr_in *)&i1->ip;
 279                 struct sockaddr_in *s2 = (struct sockaddr_in *)&i2->ip;
 280
 281                 r = ntohl(s1->sin_addr.s_addr) -
 282                         ntohl(s2->sin_addr.s_addr);
 283                 if (r) {
 284                         return r;
 285                 }
 286
 287                 s1 = (struct sockaddr_in *)&i1->netmask;
 288                 s2 = (struct sockaddr_in *)&i2->netmask;
 289
 290                 return ntohl(s1->sin_addr.s_addr) -
 291                         ntohl(s2->sin_addr.s_addr);
 292         }
 293         return 0;
 294 }
 295
 296 /* this wrapper is used to remove duplicates from the interface list generated
 297    above */
 298 int get_interfaces(TALLOC_CTX *mem_ctx, struct iface_struct **pifaces)
 299 {
 300         struct iface_struct *ifaces;
 301         int total, i, j;
 302
 303         total = _get_interfaces(mem_ctx, &ifaces);
 304         if (total <= 0) return total;
 305
 306         /* now we need to remove duplicates */
 307         TYPESAFE_QSORT(ifaces, total, iface_comp);
 308
 309         for (i=1;i<total;) {
 310                 if (iface_comp(&ifaces[i-1], &ifaces[i]) == 0) {
 311                         for (j=i-1;j<total-1;j++) {
 312                                 ifaces[j] = ifaces[j+1];
 313                         }
 314                         total--;
 315                 } else {
 316                         i++;
 317                 }
 318         }
 319
 320         *pifaces = ifaces;
 321         return total;
 322 }