sys/rpc/inet_pton.c

   1 /*
   2  * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
   3  * Copyright (c) 1996,1999 by Internet Software Consortium.
   4  *
   5  * Permission to use, copy, modify, and distribute this software for any
   6  * purpose with or without fee is hereby granted, provided that the above
   7  * copyright notice and this permission notice appear in all copies.
   8  *
   9  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
  10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11  * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
  12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
  15  * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16  */
  17
  18 #if defined(LIBC_SCCS) && !defined(lint)
  19 static const char rcsid[] = "$Id: inet_pton.c,v 1.3.18.2 2005/07/28 07:38:07 marka Exp $";
  20 #endif /* LIBC_SCCS and not lint */
  21 #include <sys/cdefs.h>
  22 __FBSDID("$FreeBSD$");
  23
  24 #include <sys/param.h>
  25 #include <sys/types.h>
  26 #include <sys/socket.h>
  27 #include <sys/systm.h>
  28
  29 #include <rpc/types.h>
  30 #include <rpc/rpc_com.h>
  31
  32 #if __FreeBSD_version < 700000
  33 #define strchr index
  34 #endif
  35
  36 /*%
  37  * WARNING: Don't even consider trying to compile this on a system where
  38  * sizeof(int) < 4.  sizeof(int) > 4 is fine; all the world's not a VAX.
  39  */
  40
  41 static int      inet_pton4(const char *src, u_char *dst);
  42 static int      inet_pton6(const char *src, u_char *dst);
  43
  44 /* int
  45  * inet_pton(af, src, dst)
  46  *      convert from presentation format (which usually means ASCII printable)
  47  *      to network format (which is usually some kind of binary format).
  48  * return:
  49  *      1 if the address was valid for the specified address family
  50  *      0 if the address wasn't valid (`dst' is untouched in this case)
  51  *      -1 if some other error occurred (`dst' is untouched in this case, too)
  52  * author:
  53  *      Paul Vixie, 1996.
  54  */
  55 int
  56 __rpc_inet_pton(int af, const char * __restrict src, void * __restrict dst)
  57 {
  58         switch (af) {
  59         case AF_INET:
  60                 return (inet_pton4(src, dst));
  61         case AF_INET6:
  62                 return (inet_pton6(src, dst));
  63         default:
  64                 return (-1);
  65         }
  66         /* NOTREACHED */
  67 }
  68
  69 /* int
  70  * inet_pton4(src, dst)
  71  *      like inet_aton() but without all the hexadecimal and shorthand.
  72  * return:
  73  *      1 if `src' is a valid dotted quad, else 0.
  74  * notice:
  75  *      does not touch `dst' unless it's returning 1.
  76  * author:
  77  *      Paul Vixie, 1996.
  78  */
  79 static int
  80 inet_pton4(const char *src, u_char *dst)
  81 {
  82         static const char digits[] = "0123456789";
  83         int saw_digit, octets, ch;
  84 #define NS_INADDRSZ     4
  85         u_char tmp[NS_INADDRSZ], *tp;
  86
  87         saw_digit = 0;
  88         octets = 0;
  89         *(tp = tmp) = 0;
  90         while ((ch = *src++) != '\0') {
  91                 const char *pch;
  92
  93                 if ((pch = strchr(digits, ch)) != NULL) {
  94                         u_int new = *tp * 10 + (pch - digits);
  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                                 if (++octets > 4)
 103                                         return (0);
 104                                 saw_digit = 1;
 105                         }
 106                 } else if (ch == '.' && saw_digit) {
 107                         if (octets == 4)
 108                                 return (0);
 109                         *++tp = 0;
 110                         saw_digit = 0;
 111                 } else
 112                         return (0);
 113         }
 114         if (octets < 4)
 115                 return (0);
 116         memcpy(dst, tmp, NS_INADDRSZ);
 117         return (1);
 118 }
 119
 120 /* int
 121  * inet_pton6(src, dst)
 122  *      convert presentation level address to network order binary form.
 123  * return:
 124  *      1 if `src' is a valid [RFC1884 2.2] address, else 0.
 125  * notice:
 126  *      (1) does not touch `dst' unless it's returning 1.
 127  *      (2) :: in a full address is silently ignored.
 128  * credit:
 129  *      inspired by Mark Andrews.
 130  * author:
 131  *      Paul Vixie, 1996.
 132  */
 133 static int
 134 inet_pton6(const char *src, u_char *dst)
 135 {
 136         static const char xdigits_l[] = "0123456789abcdef",
 137                           xdigits_u[] = "0123456789ABCDEF";
 138 #define NS_IN6ADDRSZ    16
 139 #define NS_INT16SZ      2
 140         u_char tmp[NS_IN6ADDRSZ], *tp, *endp, *colonp;
 141         const char *xdigits, *curtok;
 142         int ch, seen_xdigits;
 143         u_int val;
 144
 145         memset((tp = tmp), '\0', NS_IN6ADDRSZ);
 146         endp = tp + NS_IN6ADDRSZ;
 147         colonp = NULL;
 148         /* Leading :: requires some special handling. */
 149         if (*src == ':')
 150                 if (*++src != ':')
 151                         return (0);
 152         curtok = src;
 153         seen_xdigits = 0;
 154         val = 0;
 155         while ((ch = *src++) != '\0') {
 156                 const char *pch;
 157
 158                 if ((pch = strchr((xdigits = xdigits_l), ch)) == NULL)
 159                         pch = strchr((xdigits = xdigits_u), ch);
 160                 if (pch != NULL) {
 161                         val <<= 4;
 162                         val |= (pch - xdigits);
 163                         if (++seen_xdigits > 4)
 164                                 return (0);
 165                         continue;
 166                 }
 167                 if (ch == ':') {
 168                         curtok = src;
 169                         if (!seen_xdigits) {
 170                                 if (colonp)
 171                                         return (0);
 172                                 colonp = tp;
 173                                 continue;
 174                         } else if (*src == '\0') {
 175                                 return (0);
 176                         }
 177                         if (tp + NS_INT16SZ > endp)
 178                                 return (0);
 179                         *tp++ = (u_char) (val >> 8) & 0xff;
 180                         *tp++ = (u_char) val & 0xff;
 181                         seen_xdigits = 0;
 182                         val = 0;
 183                         continue;
 184                 }
 185                 if (ch == '.' && ((tp + NS_INADDRSZ) <= endp) &&
 186                     inet_pton4(curtok, tp) > 0) {
 187                         tp += NS_INADDRSZ;
 188                         seen_xdigits = 0;
 189                         break;  /*%< '\\0' was seen by inet_pton4(). */
 190                 }
 191                 return (0);
 192         }
 193         if (seen_xdigits) {
 194                 if (tp + NS_INT16SZ > endp)
 195                         return (0);
 196                 *tp++ = (u_char) (val >> 8) & 0xff;
 197                 *tp++ = (u_char) val & 0xff;
 198         }
 199         if (colonp != NULL) {
 200                 /*
 201                  * Since some memmove()'s erroneously fail to handle
 202                  * overlapping regions, we'll do the shift by hand.
 203                  */
 204                 const int n = tp - colonp;
 205                 int i;
 206
 207                 if (tp == endp)
 208                         return (0);
 209                 for (i = 1; i <= n; i++) {
 210                         endp[- i] = colonp[n - i];
 211                         colonp[n - i] = 0;
 212                 }
 213                 tp = endp;
 214         }
 215         if (tp != endp)
 216                 return (0);
 217         memcpy(dst, tmp, NS_IN6ADDRSZ);
 218         return (1);
 219 }
 220
 221 /*
 222  * Weak aliases for applications that use certain private entry points,
 223  * and fail to include <arpa/inet.h>.
 224  */
 225 #undef inet_pton
 226 __weak_reference(__inet_pton, inet_pton);
 227
 228 /*! \file */