LCOV - code coverage report
Current view: top level - src/backend/utils/adt - inet_cidr_ntop.c (source / functions) Hit Total Coverage
Test: PostgreSQL 18devel Lines: 81 111 73.0 %
Date: 2024-12-12 17:14:55 Functions: 3 3 100.0 %
Legend: Lines: hit not hit

          Line data    Source code
       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             :  *    src/backend/utils/adt/inet_cidr_ntop.c
      18             :  */
      19             : 
      20             : #if defined(LIBC_SCCS) && !defined(lint)
      21             : static const char rcsid[] = "Id: inet_net_ntop.c,v 1.1.2.2 2004/03/09 09:17:27 marka Exp $";
      22             : #endif
      23             : 
      24             : #include "postgres.h"
      25             : 
      26             : #include <sys/socket.h>
      27             : #include <netinet/in.h>
      28             : #include <arpa/inet.h>
      29             : 
      30             : #include "utils/builtins.h"
      31             : #include "utils/inet.h"
      32             : 
      33             : 
      34             : #ifdef SPRINTF_CHAR
      35             : #define SPRINTF(x) strlen(sprintf/**/x)
      36             : #else
      37             : #define SPRINTF(x) ((size_t)sprintf x)
      38             : #endif
      39             : 
      40             : static char *inet_cidr_ntop_ipv4(const u_char *src, int bits,
      41             :                                  char *dst, size_t size);
      42             : static char *inet_cidr_ntop_ipv6(const u_char *src, int bits,
      43             :                                  char *dst, size_t size);
      44             : 
      45             : /*
      46             :  * char *
      47             :  * pg_inet_cidr_ntop(af, src, bits, dst, size)
      48             :  *  convert network number from network to presentation format.
      49             :  *  generates CIDR style result always.
      50             :  * return:
      51             :  *  pointer to dst, or NULL if an error occurred (check errno).
      52             :  * author:
      53             :  *  Paul Vixie (ISC), July 1996
      54             :  */
      55             : char *
      56         102 : pg_inet_cidr_ntop(int af, const void *src, int bits, char *dst, size_t size)
      57             : {
      58         102 :     switch (af)
      59             :     {
      60          84 :         case PGSQL_AF_INET:
      61          84 :             return inet_cidr_ntop_ipv4(src, bits, dst, size);
      62          18 :         case PGSQL_AF_INET6:
      63          18 :             return inet_cidr_ntop_ipv6(src, bits, dst, size);
      64           0 :         default:
      65           0 :             errno = EAFNOSUPPORT;
      66           0 :             return NULL;
      67             :     }
      68             : }
      69             : 
      70             : 
      71             : /*
      72             :  * static char *
      73             :  * inet_cidr_ntop_ipv4(src, bits, dst, size)
      74             :  *  convert IPv4 network number from network to presentation format.
      75             :  *  generates CIDR style result always.
      76             :  * return:
      77             :  *  pointer to dst, or NULL if an error occurred (check errno).
      78             :  * note:
      79             :  *  network byte order assumed.  this means 192.5.5.240/28 has
      80             :  *  0b11110000 in its fourth octet.
      81             :  * author:
      82             :  *  Paul Vixie (ISC), July 1996
      83             :  */
      84             : static char *
      85          84 : inet_cidr_ntop_ipv4(const u_char *src, int bits, char *dst, size_t size)
      86             : {
      87          84 :     char       *odst = dst;
      88             :     char       *t;
      89             :     u_int       m;
      90             :     int         b;
      91             : 
      92          84 :     if (bits < 0 || bits > 32)
      93             :     {
      94           0 :         errno = EINVAL;
      95           0 :         return NULL;
      96             :     }
      97             : 
      98          84 :     if (bits == 0)
      99             :     {
     100           0 :         if (size < sizeof "0")
     101           0 :             goto emsgsize;
     102           0 :         *dst++ = '0';
     103           0 :         size--;
     104           0 :         *dst = '\0';
     105             :     }
     106             : 
     107             :     /* Format whole octets. */
     108         294 :     for (b = bits / 8; b > 0; b--)
     109             :     {
     110         210 :         if (size <= sizeof "255.")
     111           0 :             goto emsgsize;
     112         210 :         t = dst;
     113         210 :         dst += SPRINTF((dst, "%u", *src++));
     114         210 :         if (b > 1)
     115             :         {
     116         126 :             *dst++ = '.';
     117         126 :             *dst = '\0';
     118             :         }
     119         210 :         size -= (size_t) (dst - t);
     120             :     }
     121             : 
     122             :     /* Format partial octet. */
     123          84 :     b = bits % 8;
     124          84 :     if (b > 0)
     125             :     {
     126           6 :         if (size <= sizeof ".255")
     127           0 :             goto emsgsize;
     128           6 :         t = dst;
     129           6 :         if (dst != odst)
     130           6 :             *dst++ = '.';
     131           6 :         m = ((1 << b) - 1) << (8 - b);
     132           6 :         dst += SPRINTF((dst, "%u", *src & m));
     133           6 :         size -= (size_t) (dst - t);
     134             :     }
     135             : 
     136             :     /* Format CIDR /width. */
     137          84 :     if (size <= sizeof "/32")
     138           0 :         goto emsgsize;
     139          84 :     dst += SPRINTF((dst, "/%u", bits));
     140          84 :     return odst;
     141             : 
     142           0 : emsgsize:
     143           0 :     errno = EMSGSIZE;
     144           0 :     return NULL;
     145             : }
     146             : 
     147             : /*
     148             :  * static char *
     149             :  * inet_cidr_ntop_ipv6(src, bits, dst, size)
     150             :  *  convert IPv6 network number from network to presentation format.
     151             :  *  generates CIDR style result always. Picks the shortest representation
     152             :  *  unless the IP is really IPv4.
     153             :  *  always prints specified number of bits (bits).
     154             :  * return:
     155             :  *  pointer to dst, or NULL if an error occurred (check errno).
     156             :  * note:
     157             :  *  network byte order assumed.  this means 192.5.5.240/28 has
     158             :  *  0x11110000 in its fourth octet.
     159             :  * author:
     160             :  *  Vadim Kogan (UCB), June 2001
     161             :  *  Original version (IPv4) by Paul Vixie (ISC), July 1996
     162             :  */
     163             : 
     164             : static char *
     165          18 : inet_cidr_ntop_ipv6(const u_char *src, int bits, char *dst, size_t size)
     166             : {
     167             :     u_int       m;
     168             :     int         b;
     169             :     int         p;
     170             :     int         zero_s,
     171             :                 zero_l,
     172             :                 tmp_zero_s,
     173             :                 tmp_zero_l;
     174             :     int         i;
     175          18 :     int         is_ipv4 = 0;
     176             :     unsigned char inbuf[16];
     177             :     char        outbuf[sizeof("xxxx:xxxx:xxxx:xxxx:xxxx:xxxx:255.255.255.255/128")];
     178             :     char       *cp;
     179             :     int         words;
     180             :     u_char     *s;
     181             : 
     182          18 :     if (bits < 0 || bits > 128)
     183             :     {
     184           0 :         errno = EINVAL;
     185           0 :         return NULL;
     186             :     }
     187             : 
     188          18 :     cp = outbuf;
     189             : 
     190          18 :     if (bits == 0)
     191             :     {
     192           0 :         *cp++ = ':';
     193           0 :         *cp++ = ':';
     194           0 :         *cp = '\0';
     195             :     }
     196             :     else
     197             :     {
     198             :         /* Copy src to private buffer.  Zero host part. */
     199          18 :         p = (bits + 7) / 8;
     200          18 :         memcpy(inbuf, src, p);
     201          18 :         memset(inbuf + p, 0, 16 - p);
     202          18 :         b = bits % 8;
     203          18 :         if (b != 0)
     204             :         {
     205           6 :             m = ((u_int) ~0) << (8 - b);
     206           6 :             inbuf[p - 1] &= m;
     207             :         }
     208             : 
     209          18 :         s = inbuf;
     210             : 
     211             :         /* how many words need to be displayed in output */
     212          18 :         words = (bits + 15) / 16;
     213          18 :         if (words == 1)
     214           0 :             words = 2;
     215             : 
     216             :         /* Find the longest substring of zero's */
     217          18 :         zero_s = zero_l = tmp_zero_s = tmp_zero_l = 0;
     218         162 :         for (i = 0; i < (words * 2); i += 2)
     219             :         {
     220         144 :             if ((s[i] | s[i + 1]) == 0)
     221             :             {
     222          90 :                 if (tmp_zero_l == 0)
     223          18 :                     tmp_zero_s = i / 2;
     224          90 :                 tmp_zero_l++;
     225             :             }
     226             :             else
     227             :             {
     228          54 :                 if (tmp_zero_l && zero_l < tmp_zero_l)
     229             :                 {
     230          18 :                     zero_s = tmp_zero_s;
     231          18 :                     zero_l = tmp_zero_l;
     232          18 :                     tmp_zero_l = 0;
     233             :                 }
     234             :             }
     235             :         }
     236             : 
     237          18 :         if (tmp_zero_l && zero_l < tmp_zero_l)
     238             :         {
     239           0 :             zero_s = tmp_zero_s;
     240           0 :             zero_l = tmp_zero_l;
     241             :         }
     242             : 
     243          18 :         if (zero_l != words && zero_s == 0 && ((zero_l == 6) ||
     244           6 :                                                ((zero_l == 5 && s[10] == 0xff && s[11] == 0xff) ||
     245           0 :                                                 ((zero_l == 7 && s[14] != 0 && s[15] != 1)))))
     246           6 :             is_ipv4 = 1;
     247             : 
     248             :         /* Format whole words. */
     249         162 :         for (p = 0; p < words; p++)
     250             :         {
     251         144 :             if (zero_l != 0 && p >= zero_s && p < zero_s + zero_l)
     252             :             {
     253             :                 /* Time to skip some zeros */
     254          90 :                 if (p == zero_s)
     255          18 :                     *cp++ = ':';
     256          90 :                 if (p == words - 1)
     257           0 :                     *cp++ = ':';
     258          90 :                 s++;
     259          90 :                 s++;
     260          90 :                 continue;
     261             :             }
     262             : 
     263          54 :             if (is_ipv4 && p > 5)
     264             :             {
     265          12 :                 *cp++ = (p == 6) ? ':' : '.';
     266          12 :                 cp += SPRINTF((cp, "%u", *s++));
     267             :                 /* we can potentially drop the last octet */
     268          12 :                 if (p != 7 || bits > 120)
     269             :                 {
     270          12 :                     *cp++ = '.';
     271          12 :                     cp += SPRINTF((cp, "%u", *s++));
     272             :                 }
     273             :             }
     274             :             else
     275             :             {
     276          42 :                 if (cp != outbuf)
     277          30 :                     *cp++ = ':';
     278          42 :                 cp += SPRINTF((cp, "%x", *s * 256 + s[1]));
     279          42 :                 s += 2;
     280             :             }
     281             :         }
     282             :     }
     283             :     /* Format CIDR /width. */
     284          18 :     (void) SPRINTF((cp, "/%u", bits));
     285          18 :     if (strlen(outbuf) + 1 > size)
     286           0 :         goto emsgsize;
     287          18 :     strcpy(dst, outbuf);
     288             : 
     289          18 :     return dst;
     290             : 
     291           0 : emsgsize:
     292           0 :     errno = EMSGSIZE;
     293           0 :     return NULL;
     294             : }

Generated by: LCOV version 1.14