lib/libutil/sockaddr_snprintf.c

   1 /*      $NetBSD: sockaddr_snprintf.c,v 1.8 2007/07/24 08:45:45 dyoung Exp $     */
   2
   3 /*-
   4  * Copyright (c) 2004 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Christos Zoulas.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31 #include <sys/cdefs.h>
  32 #if defined(LIBC_SCCS) && !defined(lint)
  33 __RCSID("$NetBSD: sockaddr_snprintf.c,v 1.8 2007/07/24 08:45:45 dyoung Exp $");
  34 #endif /* LIBC_SCCS and not lint */
  35
  36 #include <sys/types.h>
  37 #include <sys/socket.h>
  38 #include <sys/un.h>
  39
  40 #include <netinet/in.h>
  41 #include <netatalk/at.h>
  42 #include <net/if_dl.h>
  43
  44 #include <stdio.h>
  45 #include <string.h>
  46 #include <errno.h>
  47 #include <util.h>
  48 #include <netdb.h>
  49
  50 int
  51 sockaddr_snprintf(char * const sbuf, const size_t len, const char * const fmt,
  52     const struct sockaddr * const sa)
  53 {
  54         const void *a = NULL;
  55         char abuf[1024], nbuf[1024], *addr = NULL, *w = NULL;
  56         char Abuf[1024], pbuf[32], *name = NULL, *port = NULL;
  57         char *ebuf = &sbuf[len - 1], *buf = sbuf;
  58         const char *ptr, *s;
  59         int p = -1;
  60         const struct sockaddr_at *sat = NULL;
  61         const struct sockaddr_in *sin4 = NULL;
  62         const struct sockaddr_in6 *sin6 = NULL;
  63         const struct sockaddr_un *sun = NULL;
  64         const struct sockaddr_dl *sdl = NULL;
  65         int na = 1;
  66
  67 #define ADDC(c) do { if (buf < ebuf) *buf++ = c; else buf++; } \
  68         while (/*CONSTCOND*/0)
  69 #define ADDS(p) do { for (s = p; *s; s++) ADDC(*s); } \
  70         while (/*CONSTCOND*/0)
  71 #define ADDNA() do { if (na) ADDS("N/A"); } \
  72         while (/*CONSTCOND*/0)
  73
  74         switch (sa->sa_family) {
  75         case AF_UNSPEC:
  76                 goto done;
  77         case AF_APPLETALK:
  78                 sat = ((const struct sockaddr_at *)(const void *)sa);
  79                 p = ntohs(sat->sat_port);
  80                 (void)snprintf(addr = abuf, sizeof(abuf), "%u.%u",
  81                         ntohs(sat->sat_addr.s_net), sat->sat_addr.s_node);
  82                 (void)snprintf(port = pbuf, sizeof(pbuf), "%d", p);
  83                 break;
  84         case AF_LOCAL:
  85                 sun = ((const struct sockaddr_un *)(const void *)sa);
  86                 (void)strlcpy(addr = abuf, sun->sun_path, SUN_LEN(sun));
  87                 break;
  88         case AF_INET:
  89                 sin4 = ((const struct sockaddr_in *)(const void *)sa);
  90                 p = ntohs(sin4->sin_port);
  91                 a = &sin4->sin_addr;
  92                 break;
  93         case AF_INET6:
  94                 sin6 = ((const struct sockaddr_in6 *)(const void *)sa);
  95                 p = ntohs(sin6->sin6_port);
  96                 a = &sin6->sin6_addr;
  97                 break;
  98         case AF_LINK:
  99                 sdl = ((const struct sockaddr_dl *)(const void *)sa);
 100                 (void)strlcpy(addr = abuf, link_ntoa(sdl), sizeof(abuf));
 101                 if ((w = strchr(addr, ':')) != 0) {
 102                         *w++ = '\0';
 103                         addr = w;
 104                 }
 105                 break;
 106         default:
 107                 errno = EAFNOSUPPORT;
 108                 return -1;
 109         }
 110
 111         if (addr == abuf)
 112                 name = addr;
 113
 114         if (a && getnameinfo(sa, (socklen_t)sa->sa_len, addr = abuf,
 115             (unsigned int)sizeof(abuf), NULL, 0,
 116             NI_NUMERICHOST|NI_NUMERICSERV) != 0)
 117                 return -1;
 118
 119         for (ptr = fmt; *ptr; ptr++) {
 120                 if (*ptr != '%') {
 121                         ADDC(*ptr);
 122                         continue;
 123                 }
 124           next_char:
 125                 switch (*++ptr) {
 126                 case '?':
 127                         na = 0;
 128                         goto next_char;
 129                 case 'a':
 130                         ADDS(addr);
 131                         break;
 132                 case 'p':
 133                         if (p != -1) {
 134                                 (void)snprintf(nbuf, sizeof(nbuf), "%d", p);
 135                                 ADDS(nbuf);
 136                         } else
 137                                 ADDNA();
 138                         break;
 139                 case 'f':
 140                         (void)snprintf(nbuf, sizeof(nbuf), "%d", sa->sa_family);
 141                         ADDS(nbuf);
 142                         break;
 143                 case 'l':
 144                         (void)snprintf(nbuf, sizeof(nbuf), "%d", sa->sa_len);
 145                         ADDS(nbuf);
 146                         break;
 147                 case 'A':
 148                         if (name)
 149                                 ADDS(name);
 150                         else if (!a)
 151                                 ADDNA();
 152                         else {
 153                                 getnameinfo(sa, (socklen_t)sa->sa_len,
 154                                         name = Abuf,
 155                                         (unsigned int)sizeof(nbuf), NULL, 0, 0);
 156                                 ADDS(name);
 157                         }
 158                         break;
 159                 case 'P':
 160                         if (port)
 161                                 ADDS(port);
 162                         else if (p == -1)
 163                                 ADDNA();
 164                         else {
 165                                 getnameinfo(sa, (socklen_t)sa->sa_len, NULL, 0,
 166                                         port = pbuf,
 167                                         (unsigned int)sizeof(pbuf), 0);
 168                                 ADDS(port);
 169                         }
 170                         break;
 171                 case 'I':
 172                         if (sdl && addr != abuf) {
 173                                 ADDS(abuf);
 174                         } else {
 175                                 ADDNA();
 176                         }
 177                         break;
 178                 case 'F':
 179                         if (sin6) {
 180                                 (void)snprintf(nbuf, sizeof(nbuf), "%d",
 181                                     sin6->sin6_flowinfo);
 182                                 ADDS(nbuf);
 183                                 break;
 184                         } else {
 185                                 ADDNA();
 186                         }
 187                         break;
 188                 case 'S':
 189                         if (sin6) {
 190                                 (void)snprintf(nbuf, sizeof(nbuf), "%d",
 191                                     sin6->sin6_scope_id);
 192                                 ADDS(nbuf);
 193                                 break;
 194                         } else {
 195                                 ADDNA();
 196                         }
 197                         break;
 198                 case 'R':
 199                         if (sat) {
 200                                 const struct netrange *n =
 201                                     &sat->sat_range.r_netrange;
 202                                 (void)snprintf(nbuf, sizeof(nbuf),
 203                                     "%d:[%d,%d]", n->nr_phase , n->nr_firstnet,
 204                                     n->nr_lastnet);
 205                                 ADDS(nbuf);
 206                         } else {
 207                                 ADDNA();
 208                         }
 209                         break;
 210                 default:
 211                         ADDC('%');
 212                         if (na == 0)
 213                                 ADDC('?');
 214                         if (*ptr == '\0')
 215                                 goto done;
 216                         /*FALLTHROUGH*/
 217                 case '%':
 218                         ADDC(*ptr);
 219                         break;
 220                 }
 221                 na = 1;
 222         }
 223 done:
 224         if (buf < ebuf)
 225                 *buf = '\0';
 226         else if (len != 0)
 227                 sbuf[len - 1] = '\0';
 228         return (int)(buf - sbuf);
 229 }