]> granicus.if.org Git - postgresql/blobdiff - src/backend/utils/adt/inet_net_pton.c
Add support for EUI-64 MAC addresses as macaddr8
[postgresql] / src / backend / utils / adt / inet_net_pton.c
index 2df36de1e9da006f4e73dcf37ab52dcaac3ebd46..be788d37cd070eadb3ceac8a1fc7cd4c532077e2 100644 (file)
@@ -1,22 +1,24 @@
 /*
- * Copyright (c) 1996 by Internet Software Consortium.
+ * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
+ * Copyright (c) 1996,1999 by Internet Software Consortium.
  *
  * Permission to use, copy, modify, and distribute this software for any
  * purpose with or without fee is hereby granted, provided that the above
  * copyright notice and this permission notice appear in all copies.
  *
- * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
- * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
- * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
- * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
- * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
- * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
- * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
- * SOFTWARE.
+ * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
+ * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ *
+ *       src/backend/utils/adt/inet_net_pton.c
  */
 
 #if defined(LIBC_SCCS) && !defined(lint)
-static const char rcsid[] = "$Id: inet_net_pton.c,v 1.14 2002/09/02 02:47:04 momjian Exp $";
+static const char rcsid[] = "Id: inet_net_pton.c,v 1.4.2.3 2004/03/17 00:40:11 marka Exp $";
 #endif
 
 #include "postgres.h"
@@ -24,24 +26,22 @@ static const char rcsid[] = "$Id: inet_net_pton.c,v 1.14 2002/09/02 02:47:04 mom
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
-
 #include <assert.h>
 #include <ctype.h>
-#include <errno.h>
 
-#include "utils/builtins.h"
+#include "utils/builtins.h" /* pgrminclude ignore */   /* needed on some
+                                                                                                                * platforms */
+#include "utils/inet.h"
 
-#ifdef SPRINTF_CHAR
-#define SPRINTF(x) strlen(sprintf/**/x)
-#else
-#define SPRINTF(x) ((size_t)sprintf x)
-#endif
 
 static int     inet_net_pton_ipv4(const char *src, u_char *dst);
 static int     inet_cidr_pton_ipv4(const char *src, u_char *dst, size_t size);
+static int     inet_net_pton_ipv6(const char *src, u_char *dst);
+static int     inet_cidr_pton_ipv6(const char *src, u_char *dst, size_t size);
+
 
 /*
- * static int
+ * int
  * inet_net_pton(af, src, dst, size)
  *     convert network number from presentation to network format.
  *     accepts hex octets, hex strings, decimal octets, and /CIDR.
@@ -63,10 +63,14 @@ inet_net_pton(int af, const char *src, void *dst, size_t size)
 {
        switch (af)
        {
-               case AF_INET:
+               case PGSQL_AF_INET:
                        return size == -1 ?
                                inet_net_pton_ipv4(src, dst) :
                                inet_cidr_pton_ipv4(src, dst, size);
+               case PGSQL_AF_INET6:
+                       return size == -1 ?
+                               inet_net_pton_ipv6(src, dst) :
+                               inet_cidr_pton_ipv6(src, dst, size);
                default:
                        errno = EAFNOSUPPORT;
                        return (-1);
@@ -85,19 +89,18 @@ inet_net_pton(int af, const char *src, void *dst, size_t size)
  *     not an IPv4 network specification.
  * note:
  *     network byte order assumed.  this means 192.5.5.240/28 has
- *     0x11110000 in its fourth octet.
+ *     0b11110000 in its fourth octet.
  * author:
  *     Paul Vixie (ISC), June 1996
  */
 static int
 inet_cidr_pton_ipv4(const char *src, u_char *dst, size_t size)
 {
-       static const char
-                               xdigits[] = "0123456789abcdef",
-                               digits[] = "0123456789";
+       static const char xdigits[] = "0123456789abcdef";
+       static const char digits[] = "0123456789";
        int                     n,
                                ch,
-                               tmp,
+                               tmp = 0,
                                dirty,
                                bits;
        const u_char *odst = dst;
@@ -107,10 +110,9 @@ inet_cidr_pton_ipv4(const char *src, u_char *dst, size_t size)
                && isxdigit((unsigned char) src[1]))
        {
                /* Hexadecimal: Eat nybble string. */
-               if (size <= 0)
+               if (size <= 0U)
                        goto emsgsize;
                dirty = 0;
-               tmp = 0;
                src++;                                  /* skip x or X. */
                while ((ch = *src++) != '\0' && isxdigit((unsigned char) ch))
                {
@@ -124,7 +126,7 @@ inet_cidr_pton_ipv4(const char *src, u_char *dst, size_t size)
                                tmp = (tmp << 4) | n;
                        if (++dirty == 2)
                        {
-                               if (size-- <= 0)
+                               if (size-- <= 0U)
                                        goto emsgsize;
                                *dst++ = (u_char) tmp;
                                dirty = 0;
@@ -132,7 +134,7 @@ inet_cidr_pton_ipv4(const char *src, u_char *dst, size_t size)
                }
                if (dirty)
                {                                               /* Odd trailing nybble? */
-                       if (size-- <= 0)
+                       if (size-- <= 0U)
                                goto emsgsize;
                        *dst++ = (u_char) (tmp << 4);
                }
@@ -153,7 +155,7 @@ inet_cidr_pton_ipv4(const char *src, u_char *dst, size_t size)
                                        goto enoent;
                        } while ((ch = *src++) != '\0' &&
                                         isdigit((unsigned char) ch));
-                       if (size-- <= 0)
+                       if (size-- <= 0U)
                                goto emsgsize;
                        *dst++ = (u_char) tmp;
                        if (ch == '\0' || ch == '/')
@@ -200,22 +202,29 @@ inet_cidr_pton_ipv4(const char *src, u_char *dst, size_t size)
                if (*odst >= 240)               /* Class E */
                        bits = 32;
                else if (*odst >= 224)  /* Class D */
-                       bits = 4;
+                       bits = 8;
                else if (*odst >= 192)  /* Class C */
                        bits = 24;
                else if (*odst >= 128)  /* Class B */
                        bits = 16;
                else
-/* Class A */
+                       /* Class A */
                        bits = 8;
                /* If imputed mask is narrower than specified octets, widen. */
-               if (bits >= 8 && bits < ((dst - odst) * 8))
+               if (bits < ((dst - odst) * 8))
                        bits = (dst - odst) * 8;
+
+               /*
+                * If there are no additional bits specified for a class D address
+                * adjust bits to 4.
+                */
+               if (bits == 8 && *odst == 224)
+                       bits = 4;
        }
        /* Extend network to cover the actual mask. */
        while (bits > ((dst - odst) * 8))
        {
-               if (size-- <= 0)
+               if (size-- <= 0U)
                        goto emsgsize;
                *dst++ = '\0';
        }
@@ -335,3 +344,221 @@ emsgsize:
        errno = EMSGSIZE;
        return (-1);
 }
+
+static int
+getbits(const char *src, int *bitsp)
+{
+       static const char digits[] = "0123456789";
+       int                     n;
+       int                     val;
+       char            ch;
+
+       val = 0;
+       n = 0;
+       while ((ch = *src++) != '\0')
+       {
+               const char *pch;
+
+               pch = strchr(digits, ch);
+               if (pch != NULL)
+               {
+                       if (n++ != 0 && val == 0)       /* no leading zeros */
+                               return (0);
+                       val *= 10;
+                       val += (pch - digits);
+                       if (val > 128)          /* range */
+                               return (0);
+                       continue;
+               }
+               return (0);
+       }
+       if (n == 0)
+               return (0);
+       *bitsp = val;
+       return (1);
+}
+
+static int
+getv4(const char *src, u_char *dst, int *bitsp)
+{
+       static const char digits[] = "0123456789";
+       u_char     *odst = dst;
+       int                     n;
+       u_int           val;
+       char            ch;
+
+       val = 0;
+       n = 0;
+       while ((ch = *src++) != '\0')
+       {
+               const char *pch;
+
+               pch = strchr(digits, ch);
+               if (pch != NULL)
+               {
+                       if (n++ != 0 && val == 0)       /* no leading zeros */
+                               return (0);
+                       val *= 10;
+                       val += (pch - digits);
+                       if (val > 255)          /* range */
+                               return (0);
+                       continue;
+               }
+               if (ch == '.' || ch == '/')
+               {
+                       if (dst - odst > 3) /* too many octets? */
+                               return (0);
+                       *dst++ = val;
+                       if (ch == '/')
+                               return (getbits(src, bitsp));
+                       val = 0;
+                       n = 0;
+                       continue;
+               }
+               return (0);
+       }
+       if (n == 0)
+               return (0);
+       if (dst - odst > 3)                     /* too many octets? */
+               return (0);
+       *dst++ = val;
+       return (1);
+}
+
+static int
+inet_net_pton_ipv6(const char *src, u_char *dst)
+{
+       return inet_cidr_pton_ipv6(src, dst, 16);
+}
+
+#define NS_IN6ADDRSZ 16
+#define NS_INT16SZ 2
+#define NS_INADDRSZ 4
+
+static int
+inet_cidr_pton_ipv6(const char *src, u_char *dst, size_t size)
+{
+       static const char xdigits_l[] = "0123456789abcdef",
+                               xdigits_u[] = "0123456789ABCDEF";
+       u_char          tmp[NS_IN6ADDRSZ],
+                          *tp,
+                          *endp,
+                          *colonp;
+       const char *xdigits,
+                          *curtok;
+       int                     ch,
+                               saw_xdigit;
+       u_int           val;
+       int                     digits;
+       int                     bits;
+
+       if (size < NS_IN6ADDRSZ)
+               goto emsgsize;
+
+       memset((tp = tmp), '\0', NS_IN6ADDRSZ);
+       endp = tp + NS_IN6ADDRSZ;
+       colonp = NULL;
+       /* Leading :: requires some special handling. */
+       if (*src == ':')
+               if (*++src != ':')
+                       goto enoent;
+       curtok = src;
+       saw_xdigit = 0;
+       val = 0;
+       digits = 0;
+       bits = -1;
+       while ((ch = *src++) != '\0')
+       {
+               const char *pch;
+
+               if ((pch = strchr((xdigits = xdigits_l), ch)) == NULL)
+                       pch = strchr((xdigits = xdigits_u), ch);
+               if (pch != NULL)
+               {
+                       val <<= 4;
+                       val |= (pch - xdigits);
+                       if (++digits > 4)
+                               goto enoent;
+                       saw_xdigit = 1;
+                       continue;
+               }
+               if (ch == ':')
+               {
+                       curtok = src;
+                       if (!saw_xdigit)
+                       {
+                               if (colonp)
+                                       goto enoent;
+                               colonp = tp;
+                               continue;
+                       }
+                       else if (*src == '\0')
+                               goto enoent;
+                       if (tp + NS_INT16SZ > endp)
+                               goto enoent;
+                       *tp++ = (u_char) (val >> 8) & 0xff;
+                       *tp++ = (u_char) val & 0xff;
+                       saw_xdigit = 0;
+                       digits = 0;
+                       val = 0;
+                       continue;
+               }
+               if (ch == '.' && ((tp + NS_INADDRSZ) <= endp) &&
+                       getv4(curtok, tp, &bits) > 0)
+               {
+                       tp += NS_INADDRSZ;
+                       saw_xdigit = 0;
+                       break;                          /* '\0' was seen by inet_pton4(). */
+               }
+               if (ch == '/' && getbits(src, &bits) > 0)
+                       break;
+               goto enoent;
+       }
+       if (saw_xdigit)
+       {
+               if (tp + NS_INT16SZ > endp)
+                       goto enoent;
+               *tp++ = (u_char) (val >> 8) & 0xff;
+               *tp++ = (u_char) val & 0xff;
+       }
+       if (bits == -1)
+               bits = 128;
+
+       endp = tmp + 16;
+
+       if (colonp != NULL)
+       {
+               /*
+                * Since some memmove()'s erroneously fail to handle overlapping
+                * regions, we'll do the shift by hand.
+                */
+               const int       n = tp - colonp;
+               int                     i;
+
+               if (tp == endp)
+                       goto enoent;
+               for (i = 1; i <= n; i++)
+               {
+                       endp[-i] = colonp[n - i];
+                       colonp[n - i] = 0;
+               }
+               tp = endp;
+       }
+       if (tp != endp)
+               goto enoent;
+
+       /*
+        * Copy out the result.
+        */
+       memcpy(dst, tmp, NS_IN6ADDRSZ);
+
+       return (bits);
+
+enoent:
+       errno = ENOENT;
+       return (-1);
+
+emsgsize:
+       errno = EMSGSIZE;
+       return (-1);
+}