xref: /freebsd/lib/libc/inet/inet_cidr_pton.c (revision 911f0260390e18cf85f3dbf2c719b593efdc1e3c)
1 /*-
2  * SPDX-License-Identifier: ISC
3  *
4  * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
5  * Copyright (c) 1998,1999 by Internet Software Consortium.
6  *
7  * Permission to use, copy, modify, and distribute this software for any
8  * purpose with or without fee is hereby granted, provided that the above
9  * copyright notice and this permission notice appear in all copies.
10  *
11  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
12  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13  * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
14  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
17  * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18  */
19 
20 #if defined(LIBC_SCCS) && !defined(lint)
21 static const char rcsid[] = "$Id: inet_cidr_pton.c,v 1.6 2005/04/27 04:56:19 sra Exp $";
22 #endif
23 #include <sys/cdefs.h>
24 __FBSDID("$FreeBSD$");
25 
26 #include "port_before.h"
27 
28 #include <sys/types.h>
29 #include <sys/socket.h>
30 #include <netinet/in.h>
31 #include <arpa/nameser.h>
32 #include <arpa/inet.h>
33 
34 #include <assert.h>
35 #include <ctype.h>
36 #include <errno.h>
37 #include <stdio.h>
38 #include <string.h>
39 #include <stdlib.h>
40 
41 #include "port_after.h"
42 
43 #ifdef SPRINTF_CHAR
44 # define SPRINTF(x) strlen(sprintf/**/x)
45 #else
46 # define SPRINTF(x) ((size_t)sprintf x)
47 #endif
48 
49 static int	inet_cidr_pton_ipv4 (const char *src, u_char *dst, int *bits,
50 				     int ipv6);
51 static int	inet_cidr_pton_ipv6 (const char *src, u_char *dst, int *bits);
52 
53 static int	getbits(const char *, int ipv6);
54 
55 /*%
56  * int
57  * inet_cidr_pton(af, src, dst, *bits)
58  *	convert network address from presentation to network format.
59  *	accepts inet_pton()'s input for this "af" plus trailing "/CIDR".
60  *	"dst" is assumed large enough for its "af".  "bits" is set to the
61  *	/CIDR prefix length, which can have defaults (like /32 for IPv4).
62  * return:
63  *	-1 if an error occurred (inspect errno; ENOENT means bad format).
64  *	0 if successful conversion occurred.
65  * note:
66  *	192.5.5.1/28 has a nonzero host part, which means it isn't a network
67  *	as called for by inet_net_pton() but it can be a host address with
68  *	an included netmask.
69  * author:
70  *	Paul Vixie (ISC), October 1998
71  */
72 int
73 inet_cidr_pton(int af, const char *src, void *dst, int *bits) {
74 	switch (af) {
75 	case AF_INET:
76 		return (inet_cidr_pton_ipv4(src, dst, bits, 0));
77 	case AF_INET6:
78 		return (inet_cidr_pton_ipv6(src, dst, bits));
79 	default:
80 		errno = EAFNOSUPPORT;
81 		return (-1);
82 	}
83 }
84 
85 static const char digits[] = "0123456789";
86 
87 static int
88 inet_cidr_pton_ipv4(const char *src, u_char *dst, int *pbits, int ipv6) {
89 	const u_char *odst = dst;
90 	int n, ch, tmp, bits;
91 	size_t size = 4;
92 
93 	/* Get the mantissa. */
94 	while (ch = *src++, (isascii(ch) && isdigit(ch))) {
95 		tmp = 0;
96 		do {
97 			n = strchr(digits, ch) - digits;
98 			assert(n >= 0 && n <= 9);
99 			tmp *= 10;
100 			tmp += n;
101 			if (tmp > 255)
102 				goto enoent;
103 		} while ((ch = *src++) != '\0' && isascii(ch) && isdigit(ch));
104 		if (size-- == 0U)
105 			goto emsgsize;
106 		*dst++ = (u_char) tmp;
107 		if (ch == '\0' || ch == '/')
108 			break;
109 		if (ch != '.')
110 			goto enoent;
111 	}
112 
113 	/* Get the prefix length if any. */
114 	bits = -1;
115 	if (ch == '/' && dst > odst) {
116 		bits = getbits(src, ipv6);
117 		if (bits == -2)
118 			goto enoent;
119 	} else if (ch != '\0')
120 		goto enoent;
121 
122 	/* Prefix length can default to /32 only if all four octets spec'd. */
123 	if (bits == -1) {
124 		if (dst - odst == 4)
125 			bits = ipv6 ? 128 : 32;
126 		else
127 			goto enoent;
128 	}
129 
130 	/* If nothing was written to the destination, we found no address. */
131 	if (dst == odst)
132 		goto enoent;
133 
134 	/* If prefix length overspecifies mantissa, life is bad. */
135 	if (((bits - (ipv6 ? 96 : 0)) / 8) > (dst - odst))
136 		goto enoent;
137 
138 	/* Extend address to four octets. */
139 	while (size-- > 0U)
140 		*dst++ = 0;
141 
142 	*pbits = bits;
143 	return (0);
144 
145  enoent:
146 	errno = ENOENT;
147 	return (-1);
148 
149  emsgsize:
150 	errno = EMSGSIZE;
151 	return (-1);
152 }
153 
154 static int
155 inet_cidr_pton_ipv6(const char *src, u_char *dst, int *pbits) {
156 	static const char xdigits_l[] = "0123456789abcdef",
157 			  xdigits_u[] = "0123456789ABCDEF";
158 	u_char tmp[NS_IN6ADDRSZ], *tp, *endp, *colonp;
159 	const char *xdigits, *curtok;
160 	int ch, saw_xdigit;
161 	u_int val;
162 	int bits;
163 
164 	memset((tp = tmp), '\0', NS_IN6ADDRSZ);
165 	endp = tp + NS_IN6ADDRSZ;
166 	colonp = NULL;
167 	/* Leading :: requires some special handling. */
168 	if (*src == ':')
169 		if (*++src != ':')
170 			return (0);
171 	curtok = src;
172 	saw_xdigit = 0;
173 	val = 0;
174 	bits = -1;
175 	while ((ch = *src++) != '\0') {
176 		const char *pch;
177 
178 		if ((pch = strchr((xdigits = xdigits_l), ch)) == NULL)
179 			pch = strchr((xdigits = xdigits_u), ch);
180 		if (pch != NULL) {
181 			val <<= 4;
182 			val |= (pch - xdigits);
183 			if (val > 0xffff)
184 				return (0);
185 			saw_xdigit = 1;
186 			continue;
187 		}
188 		if (ch == ':') {
189 			curtok = src;
190 			if (!saw_xdigit) {
191 				if (colonp)
192 					return (0);
193 				colonp = tp;
194 				continue;
195 			} else if (*src == '\0') {
196 				return (0);
197 			}
198 			if (tp + NS_INT16SZ > endp)
199 				return (0);
200 			*tp++ = (u_char) (val >> 8) & 0xff;
201 			*tp++ = (u_char) val & 0xff;
202 			saw_xdigit = 0;
203 			val = 0;
204 			continue;
205 		}
206 		if (ch == '.' && ((tp + NS_INADDRSZ) <= endp) &&
207 		    inet_cidr_pton_ipv4(curtok, tp, &bits, 1) == 0) {
208 			tp += NS_INADDRSZ;
209 			saw_xdigit = 0;
210 			break;	/*%< '\\0' was seen by inet_pton4(). */
211 		}
212 		if (ch == '/') {
213 			bits = getbits(src, 1);
214 			if (bits == -2)
215 				goto enoent;
216 			break;
217 		}
218 		goto enoent;
219 	}
220 	if (saw_xdigit) {
221 		if (tp + NS_INT16SZ > endp)
222 			goto emsgsize;
223 		*tp++ = (u_char) (val >> 8) & 0xff;
224 		*tp++ = (u_char) val & 0xff;
225 	}
226 	if (colonp != NULL) {
227 		/*
228 		 * Since some memmove()'s erroneously fail to handle
229 		 * overlapping regions, we'll do the shift by hand.
230 		 */
231 		const int n = tp - colonp;
232 		int i;
233 
234 		if (tp == endp)
235 			goto enoent;
236 		for (i = 1; i <= n; i++) {
237 			endp[- i] = colonp[n - i];
238 			colonp[n - i] = 0;
239 		}
240 		tp = endp;
241 	}
242 
243 	memcpy(dst, tmp, NS_IN6ADDRSZ);
244 
245 	*pbits = bits;
246 	return (0);
247 
248  enoent:
249 	errno = ENOENT;
250 	return (-1);
251 
252  emsgsize:
253 	errno = EMSGSIZE;
254 	return (-1);
255 }
256 
257 static int
258 getbits(const char *src, int ipv6) {
259 	int bits = 0;
260 	char *cp, ch;
261 
262 	if (*src == '\0')			/*%< syntax */
263 		return (-2);
264 	do {
265 		ch = *src++;
266 		cp = strchr(digits, ch);
267 		if (cp == NULL)			/*%< syntax */
268 			return (-2);
269 		bits *= 10;
270 		bits += cp - digits;
271 		if (bits == 0 && *src != '\0')	/*%< no leading zeros */
272 			return (-2);
273 		if (bits > (ipv6 ? 128 : 32))	/*%< range error */
274 			return (-2);
275 	} while (*src != '\0');
276 
277 	return (bits);
278 }
279 
280 /*! \file */
281