resolv/inet_pton.c

   1 /* Copyright (C) 1996-2017 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3
   4    The GNU C Library is free software; you can redistribute it and/or
   5    modify it under the terms of the GNU Lesser General Public
   6    License as published by the Free Software Foundation; either
   7    version 2.1 of the License, or (at your option) any later version.
   8
   9    The GNU C Library is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12    Lesser General Public License for more details.
  13
  14    You should have received a copy of the GNU Lesser General Public
  15    License along with the GNU C Library; if not, see
  16    <http://www.gnu.org/licenses/>.  */
  17
  18 /*
  19  * Copyright (c) 1996,1999 by Internet Software Consortium.
  20  *
  21  * Permission to use, copy, modify, and distribute this software for any
  22  * purpose with or without fee is hereby granted, provided that the above
  23  * copyright notice and this permission notice appear in all copies.
  24  *
  25  * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
  26  * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
  27  * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
  28  * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
  29  * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
  30  * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
  31  * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
  32  * SOFTWARE.
  33  */
  34
  35 #include <arpa/inet.h>
  36 #include <arpa/nameser.h>
  37 #include <ctype.h>
  38 #include <errno.h>
  39 #include <netinet/in.h>
  40 #include <resolv/resolv-internal.h>
  41 #include <string.h>
  42 #include <sys/socket.h>
  43 #include <sys/types.h>
  44
  45 static int inet_pton4 (const char *src, const char *src_end, u_char *dst);
  46 static int inet_pton6 (const char *src, const char *src_end, u_char *dst);
  47
  48 int
  49 __inet_pton_length (int af, const char *src, size_t srclen, void *dst)
  50 {
  51   switch (af)
  52     {
  53     case AF_INET:
  54       return inet_pton4 (src, src + srclen, dst);
  55     case AF_INET6:
  56       return inet_pton6 (src, src + srclen, dst);
  57     default:
  58       __set_errno (EAFNOSUPPORT);
  59       return -1;
  60     }
  61 }
  62 libc_hidden_def (__inet_pton_length)
  63
  64 /* Like __inet_pton_length, but use strlen (SRC) as the length of
  65    SRC.  */
  66 int
  67 __inet_pton (int af, const char *src, void *dst)
  68 {
  69   return __inet_pton_length (af, src, strlen (src), dst);
  70 }
  71 libc_hidden_def (__inet_pton)
  72 weak_alias (__inet_pton, inet_pton)
  73 libc_hidden_weak (inet_pton)
  74
  75 /* Like inet_aton but without all the hexadecimal, octal and shorthand
  76    (and trailing garbage is not ignored).  Return 1 if SRC is a valid
  77    dotted quad, else 0.  This function does not touch DST unless it's
  78    returning 1.
  79    Author: Paul Vixie, 1996.  */
  80 static int
  81 inet_pton4 (const char *src, const char *end, unsigned char *dst)
  82 {
  83   int saw_digit, octets, ch;
  84   unsigned char tmp[NS_INADDRSZ], *tp;
  85
  86   saw_digit = 0;
  87   octets = 0;
  88   *(tp = tmp) = 0;
  89   while (src < end)
  90     {
  91       ch = *src++;
  92       if (ch >= '0' && ch <= '9')
  93         {
  94           unsigned int new = *tp * 10 + (ch - '0');
  95
  96           if (saw_digit && *tp == 0)
  97             return 0;
  98           if (new > 255)
  99             return 0;
 100           *tp = new;
 101           if (! saw_digit)
 102             {
 103               if (++octets > 4)
 104                 return 0;
 105               saw_digit = 1;
 106             }
 107         }
 108       else if (ch == '.' && saw_digit)
 109         {
 110           if (octets == 4)
 111             return 0;
 112           *++tp = 0;
 113           saw_digit = 0;
 114         }
 115       else
 116         return 0;
 117     }
 118   if (octets < 4)
 119     return 0;
 120   memcpy (dst, tmp, NS_INADDRSZ);
 121   return 1;
 122 }
 123
 124 /* Return the value of CH as a hexademical digit, or -1 if it is a
 125    different type of character.  */
 126 static int
 127 hex_digit_value (char ch)
 128 {
 129   if ('0' <= ch && ch <= '9')
 130     return ch - '0';
 131   if ('a' <= ch && ch <= 'f')
 132     return ch - 'a' + 10;
 133   if ('A' <= ch && ch <= 'F')
 134     return ch - 'A' + 10;
 135   return -1;
 136 }
 137
 138 /* Convert presentation-level IPv6 address to network order binary
 139    form.  Return 1 if SRC is a valid [RFC1884 2.2] address, else 0.
 140    This function does not touch DST unless it's returning 1.
 141    Author: Paul Vixie, 1996.  Inspired by Mark Andrews.  */
 142 static int
 143 inet_pton6 (const char *src, const char *src_endp, unsigned char *dst)
 144 {
 145   unsigned char tmp[NS_IN6ADDRSZ], *tp, *endp, *colonp;
 146   const char *curtok;
 147   int ch, saw_xdigit;
 148   unsigned int val;
 149
 150   tp = memset (tmp, '\0', NS_IN6ADDRSZ);
 151   endp = tp + NS_IN6ADDRSZ;
 152   colonp = NULL;
 153
 154   /* Leading :: requires some special handling.  */
 155   if (src == src_endp)
 156     return 0;
 157   if (*src == ':')
 158     {
 159       ++src;
 160       if (src == src_endp || *src != ':')
 161         return 0;
 162     }
 163
 164   curtok = src;
 165   saw_xdigit = 0;
 166   val = 0;
 167   while (src < src_endp)
 168     {
 169       ch = *src++;
 170       int digit = hex_digit_value (ch);
 171       if (digit >= 0)
 172         {
 173           val <<= 4;
 174           val |= digit;
 175           if (val > 0xffff)
 176             return 0;
 177           saw_xdigit = 1;
 178           continue;
 179         }
 180       if (ch == ':')
 181         {
 182           curtok = src;
 183           if (!saw_xdigit)
 184             {
 185               if (colonp)
 186                 return 0;
 187               colonp = tp;
 188               continue;
 189             }
 190           else if (src == src_endp)
 191             return 0;
 192           if (tp + NS_INT16SZ > endp)
 193             return 0;
 194           *tp++ = (unsigned char) (val >> 8) & 0xff;
 195           *tp++ = (unsigned char) val & 0xff;
 196           saw_xdigit = 0;
 197           val = 0;
 198           continue;
 199         }
 200       if (ch == '.' && ((tp + NS_INADDRSZ) <= endp)
 201           && inet_pton4 (curtok, src_endp, tp) > 0)
 202         {
 203           tp += NS_INADDRSZ;
 204           saw_xdigit = 0;
 205           break;  /* '\0' was seen by inet_pton4.  */
 206         }
 207       return 0;
 208     }
 209   if (saw_xdigit)
 210     {
 211       if (tp + NS_INT16SZ > endp)
 212         return 0;
 213       *tp++ = (unsigned char) (val >> 8) & 0xff;
 214       *tp++ = (unsigned char) val & 0xff;
 215     }
 216   if (colonp != NULL)
 217     {
 218       /* Replace :: with zeros.  */
 219       if (tp == endp)
 220         /* :: would expand to a zero-width field.  */
 221         return 0;
 222       size_t n = tp - colonp;
 223       memmove (endp - n, colonp, n);
 224       memset (colonp, 0, endp - n - colonp);
 225       tp = endp;
 226     }
 227   if (tp != endp)
 228     return 0;
 229   memcpy (dst, tmp, NS_IN6ADDRSZ);
 230   return 1;
 231 }