release/src/router/minidlna/getifaddr.c

   1 /* $Id: getifaddr.c,v 1.14 2011/05/02 23:50:52 jmaggard Exp $ */
   2 /* MiniUPnP project
   3  * http://miniupnp.free.fr/ or http://miniupnp.tuxfamily.org/
   4  *
   5  * Copyright (c) 2006, Thomas Bernard
   6  * All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions are met:
  10  *     * Redistributions of source code must retain the above copyright
  11  *       notice, this list of conditions and the following disclaimer.
  12  *     * Redistributions in binary form must reproduce the above copyright
  13  *       notice, this list of conditions and the following disclaimer in the
  14  *       documentation and/or other materials provided with the distribution.
  15  *     * The name of the author may not be used to endorse or promote products
  16  *       derived from this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  19  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  22  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  23  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  24  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  25  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  26  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  27  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28  * POSSIBILITY OF SUCH DAMAGE.
  29  */
  30 #include <stdio.h>
  31 #include <stdlib.h>
  32 #include <string.h>
  33 #include <unistd.h>
  34 #include <sys/ioctl.h>
  35 #include <sys/types.h>
  36 #include <sys/socket.h>
  37 #include <net/if.h>
  38 #include <arpa/inet.h>
  39 #include <netinet/in.h>
  40 #include <netdb.h>
  41 #include <errno.h>
  42 #if defined(sun)
  43 #include <sys/sockio.h>
  44 #endif
  45
  46 #include "getifaddr.h"
  47 #include "log.h"
  48
  49 int
  50 getifaddr(const char * ifname, char * buf, int len)
  51 {
  52         /* SIOCGIFADDR struct ifreq *  */
  53         int s;
  54         struct ifreq ifr;
  55         int ifrlen;
  56         struct sockaddr_in * addr;
  57         uint32_t mask;
  58         int i;
  59
  60         ifrlen = sizeof(ifr);
  61         s = socket(PF_INET, SOCK_DGRAM, 0);
  62         if(s < 0)
  63         {
  64                 DPRINTF(E_ERROR, L_GENERAL, "socket(PF_INET, SOCK_DGRAM): %s\n", strerror(errno));
  65                 return -1;
  66         }
  67         strncpy(ifr.ifr_name, ifname, IFNAMSIZ);
  68         if(ioctl(s, SIOCGIFADDR, &ifr, &ifrlen) < 0)
  69         {
  70                 DPRINTF(E_ERROR, L_GENERAL, "ioctl(s, SIOCGIFADDR, ...): %s\n", strerror(errno));
  71                 close(s);
  72                 return -1;
  73         }
  74         addr = (struct sockaddr_in *)&ifr.ifr_addr;
  75         if(!inet_ntop(AF_INET, &addr->sin_addr, buf, len))
  76         {
  77                 DPRINTF(E_ERROR, L_GENERAL, "inet_ntop(): %s\n", strerror(errno));
  78                 close(s);
  79                 return -1;
  80         }
  81         if(ioctl(s, SIOCGIFNETMASK, &ifr, &ifrlen) == 0)
  82         {
  83                 addr = (struct sockaddr_in *)&ifr.ifr_netmask;
  84                 mask = ntohl(addr->sin_addr.s_addr);
  85                 for (i = 0; i < 32; i++)
  86                 {
  87                         if ((mask >> i) & 1)
  88                                 break;
  89                 }
  90                 mask = 32 - i;
  91                 if (mask)
  92                 {
  93                         i = strlen(buf);
  94                         snprintf(buf+i, len-i, "/%u", mask);
  95                 }
  96         }
  97         else
  98                 DPRINTF(E_ERROR, L_GENERAL, "ioctl(s, SIOCGIFNETMASK, ...): %s\n", strerror(errno));
  99         close(s);
 100         return 0;
 101 }
 102
 103 int
 104 getsysaddr(char * buf, int len)
 105 {
 106         int i;
 107         int s = socket(PF_INET, SOCK_STREAM, 0);
 108         struct sockaddr_in addr;
 109         struct ifreq ifr;
 110         uint32_t mask;
 111         int ret = -1;
 112
 113         for (i=1; i > 0; i++)
 114         {
 115                 ifr.ifr_ifindex = i;
 116                 if( ioctl(s, SIOCGIFNAME, &ifr) < 0 )
 117                         break;
 118                 if(ioctl(s, SIOCGIFADDR, &ifr, sizeof(struct ifreq)) < 0)
 119                         continue;
 120                 memcpy(&addr, &ifr.ifr_addr, sizeof(addr));
 121                 if(strncmp(inet_ntoa(addr.sin_addr), "127.", 4) == 0)
 122                         continue;
 123                 if(ioctl(s, SIOCGIFNETMASK, &ifr, sizeof(struct ifreq)) < 0)
 124                         continue;
 125                 if(!inet_ntop(AF_INET, &addr.sin_addr, buf, len))
 126                 {
 127                         DPRINTF(E_ERROR, L_GENERAL, "inet_ntop(): %s\n", strerror(errno));
 128                         close(s);
 129                         break;
 130                 }
 131                 ret = 0;
 132
 133                 memcpy(&addr, &ifr.ifr_netmask, sizeof(addr));
 134                 mask = ntohl(addr.sin_addr.s_addr);
 135                 for (i = 0; i < 32; i++)
 136                 {
 137                         if ((mask >> i) & 1)
 138                                 break;
 139                 }
 140                 mask = 32 - i;
 141                 if (mask)
 142                 {
 143                         i = strlen(buf);
 144                         snprintf(buf+i, len-i, "/%u", mask);
 145                 }
 146                 break;
 147         }
 148         close(s);
 149
 150         return(ret);
 151 }
 152
 153 int
 154 getsyshwaddr(char * buf, int len)
 155 {
 156         struct if_nameindex *ifaces, *if_idx;
 157         unsigned char mac[6];
 158         struct ifreq ifr;
 159         int fd;
 160         int ret = -1;
 161
 162         memset(&mac, '\0', sizeof(mac));
 163         /* Get the spatially unique node identifier */
 164         fd = socket(AF_INET, SOCK_DGRAM, 0);
 165         if( fd < 0 )
 166                 return(ret);
 167
 168         ifaces = if_nameindex();
 169         if(!ifaces)
 170                 return(ret);
 171
 172         for(if_idx = ifaces; if_idx->if_index; if_idx++)
 173         {
 174                 strncpy(ifr.ifr_name, if_idx->if_name, IFNAMSIZ);
 175                 if(ioctl(fd, SIOCGIFFLAGS, &ifr) < 0)
 176                         continue;
 177                 if(ifr.ifr_ifru.ifru_flags & IFF_LOOPBACK)
 178                         continue;
 179                 if( ioctl(fd, SIOCGIFHWADDR, &ifr) < 0 )
 180                         continue;
 181                 if( MACADDR_IS_ZERO(ifr.ifr_hwaddr.sa_data) )
 182                         continue;
 183                 ret = 0;
 184                 break;
 185         }
 186         if_freenameindex(ifaces);
 187         close(fd);
 188
 189         if(ret == 0)
 190         {
 191                 if(len > 12)
 192                 {
 193                         memmove(mac, ifr.ifr_hwaddr.sa_data, 6);
 194                         sprintf(buf, "%02x%02x%02x%02x%02x%02x",
 195                                 mac[0]&0xFF, mac[1]&0xFF, mac[2]&0xFF,
 196                                 mac[3]&0xFF, mac[4]&0xFF, mac[5]&0xFF);
 197                 }
 198                 else if(len == 6)
 199                 {
 200                         memmove(buf, ifr.ifr_hwaddr.sa_data, 6);
 201                 }
 202         }
 203         return ret;
 204 }
 205
 206 int
 207 get_remote_mac(struct in_addr ip_addr, unsigned char * mac)
 208 {
 209         struct in_addr arp_ent;
 210         FILE * arp;
 211         char remote_ip[16];
 212         int matches, hwtype, flags;
 213         memset(mac, 0xFF, 6);
 214
 215         arp = fopen("/proc/net/arp", "r");
 216         if( !arp )
 217                 return 1;
 218         while( !feof(arp) )
 219         {
 220                 matches = fscanf(arp, "%15s 0x%8X 0x%8X %2hhx:%2hhx:%2hhx:%2hhx:%2hhx:%2hhx",
 221                                       remote_ip, &hwtype, &flags,
 222                                       &mac[0], &mac[1], &mac[2], &mac[3], &mac[4], &mac[5]);
 223                 if( matches != 9 )
 224                         continue;
 225                 inet_pton(AF_INET, remote_ip, &arp_ent);
 226                 if( ip_addr.s_addr == arp_ent.s_addr )
 227                         break;
 228                 mac[0] = 0xFF;
 229         }
 230         fclose(arp);
 231
 232         if( mac[0] == 0xFF )
 233         {
 234                 memset(mac, 0xFF, 6);
 235                 return 1;
 236         }
 237
 238         return 0;
 239 }