xref: /titanic_50/usr/src/cmd/ssh/libssh/common/addrmatch.c (revision b56bf881a9655cb27b53cba1468312f7c6dfb0a2)
1 /*	$OpenBSD: addrmatch.c,v 1.4 2008/12/10 03:55:20 stevesk Exp $ */
2 
3 /*
4  * Copyright (c) 2004-2008 Damien Miller <djm@mindrot.org>
5  *
6  * Permission to use, copy, modify, and distribute this software for any
7  * purpose with or without fee is hereby granted, provided that the above
8  * copyright notice and this permission notice appear in all copies.
9  *
10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17  */
18 
19 #include "includes.h"
20 
21 #include <sys/types.h>
22 #include <sys/socket.h>
23 #include <netinet/in.h>
24 #include <arpa/inet.h>
25 
26 #include <netdb.h>
27 #include <string.h>
28 #include <stdlib.h>
29 #include <stdio.h>
30 #include <stdarg.h>
31 
32 #include "match.h"
33 #include "log.h"
34 #include "xmalloc.h"
35 
36 struct xaddr {
37 	sa_family_t	af;
38 	union {
39 		struct in_addr		v4;
40 		struct in6_addr		v6;
41 		u_int8_t		addr8[16];
42 		u_int32_t		addr32[4];
43 	} xa;		    /* 128-bit address */
44 	u_int32_t	scope_id;	/* iface scope id for v6 */
45 #define v4	xa.v4
46 #define v6	xa.v6
47 #define addr8	xa.addr8
48 #define addr32	xa.addr32
49 };
50 
51 static int
52 addr_unicast_masklen(int af)
53 {
54 	switch (af) {
55 	case AF_INET:
56 		return 32;
57 	case AF_INET6:
58 		return 128;
59 	default:
60 		return -1;
61 	}
62 }
63 
64 static inline int
65 masklen_valid(int af, u_int masklen)
66 {
67 	switch (af) {
68 	case AF_INET:
69 		return masklen <= 32 ? 0 : -1;
70 	case AF_INET6:
71 		return masklen <= 128 ? 0 : -1;
72 	default:
73 		return -1;
74 	}
75 }
76 
77 /*
78  * Convert struct sockaddr to struct xaddr
79  * Returns 0 on success, -1 on failure.
80  */
81 static int
82 addr_sa_to_xaddr(struct sockaddr *sa, socklen_t slen, struct xaddr *xa)
83 {
84 	/* LINTED E_BAD_PTR_CAST_ALIGN */
85 	struct sockaddr_in *in4 = (struct sockaddr_in *)sa;
86 	/* LINTED E_BAD_PTR_CAST_ALIGN */
87 	struct sockaddr_in6 *in6 = (struct sockaddr_in6 *)sa;
88 
89 	memset(xa, '\0', sizeof(*xa));
90 
91 	switch (sa->sa_family) {
92 	case AF_INET:
93 		if (slen < sizeof(*in4))
94 			return -1;
95 		xa->af = AF_INET;
96 		memcpy(&xa->v4, &in4->sin_addr, sizeof(xa->v4));
97 		break;
98 	case AF_INET6:
99 		if (slen < sizeof(*in6))
100 			return -1;
101 		xa->af = AF_INET6;
102 		memcpy(&xa->v6, &in6->sin6_addr, sizeof(xa->v6));
103 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
104 		xa->scope_id = in6->sin6_scope_id;
105 #endif
106 		break;
107 	default:
108 		return -1;
109 	}
110 
111 	return 0;
112 }
113 
114 /*
115  * Calculate a netmask of length 'l' for address family 'af' and
116  * store it in 'n'.
117  * Returns 0 on success, -1 on failure.
118  */
119 static int
120 addr_netmask(int af, u_int l, struct xaddr *n)
121 {
122 	int i;
123 
124 	if (masklen_valid(af, l) != 0 || n == NULL)
125 		return -1;
126 
127 	memset(n, '\0', sizeof(*n));
128 	switch (af) {
129 	case AF_INET:
130 		n->af = AF_INET;
131 		n->v4.s_addr = htonl((0xffffffff << (32 - l)) & 0xffffffff);
132 		return 0;
133 	case AF_INET6:
134 		n->af = AF_INET6;
135 		for (i = 0; i < 4 && l >= 32; i++, l -= 32)
136 			n->addr32[i] = 0xffffffffU;
137 		if (i < 4 && l != 0)
138 			n->addr32[i] = htonl((0xffffffff << (32 - l)) &
139 			    0xffffffff);
140 		return 0;
141 	default:
142 		return -1;
143 	}
144 }
145 
146 /*
147  * Perform logical AND of addresses 'a' and 'b', storing result in 'dst'.
148  * Returns 0 on success, -1 on failure.
149  */
150 static int
151 addr_and(struct xaddr *dst, const struct xaddr *a, const struct xaddr *b)
152 {
153 	int i;
154 
155 	if (dst == NULL || a == NULL || b == NULL || a->af != b->af)
156 		return -1;
157 
158 	memcpy(dst, a, sizeof(*dst));
159 	switch (a->af) {
160 	case AF_INET:
161 		dst->v4.s_addr &= b->v4.s_addr;
162 		return 0;
163 	case AF_INET6:
164 		dst->scope_id = a->scope_id;
165 		for (i = 0; i < 4; i++)
166 			dst->addr32[i] &= b->addr32[i];
167 		return 0;
168 	default:
169 		return -1;
170 	}
171 }
172 
173 /*
174  * Compare addresses 'a' and 'b'
175  * Return 0 if addresses are identical, -1 if (a < b) or 1 if (a > b)
176  */
177 static int
178 addr_cmp(const struct xaddr *a, const struct xaddr *b)
179 {
180 	int i;
181 
182 	if (a->af != b->af)
183 		return a->af == AF_INET6 ? 1 : -1;
184 
185 	switch (a->af) {
186 	case AF_INET:
187 		if (a->v4.s_addr == b->v4.s_addr)
188 			return 0;
189 		return ntohl(a->v4.s_addr) > ntohl(b->v4.s_addr) ? 1 : -1;
190 	case AF_INET6:
191 		for (i = 0; i < 16; i++)
192 			if (a->addr8[i] - b->addr8[i] != 0)
193 				return a->addr8[i] > b->addr8[i] ? 1 : -1;
194 		if (a->scope_id == b->scope_id)
195 			return 0;
196 		return a->scope_id > b->scope_id ? 1 : -1;
197 	default:
198 		return -1;
199 	}
200 }
201 
202 /*
203  * Parse string address 'p' into 'n'
204  * Returns 0 on success, -1 on failure.
205  */
206 static int
207 addr_pton(const char *p, struct xaddr *n)
208 {
209 	struct addrinfo hints, *ai;
210 
211 	memset(&hints, '\0', sizeof(hints));
212 	hints.ai_flags = AI_NUMERICHOST;
213 
214 	if (p == NULL || getaddrinfo(p, NULL, &hints, &ai) != 0)
215 		return -1;
216 
217 	if (ai == NULL || ai->ai_addr == NULL)
218 		return -1;
219 
220 	if (n != NULL &&
221 	    addr_sa_to_xaddr(ai->ai_addr, ai->ai_addrlen, n) == -1) {
222 		freeaddrinfo(ai);
223 		return -1;
224 	}
225 
226 	freeaddrinfo(ai);
227 	return 0;
228 }
229 
230 /*
231  * Perform bitwise negation of address
232  * Returns 0 on success, -1 on failure.
233  */
234 static int
235 addr_invert(struct xaddr *n)
236 {
237 	int i;
238 
239 	if (n == NULL)
240 		return (-1);
241 
242 	switch (n->af) {
243 	case AF_INET:
244 		n->v4.s_addr = ~n->v4.s_addr;
245 		return (0);
246 	case AF_INET6:
247 		for (i = 0; i < 4; i++)
248 			n->addr32[i] = ~n->addr32[i];
249 		return (0);
250 	default:
251 		return (-1);
252 	}
253 }
254 
255 /*
256  * Calculate a netmask of length 'l' for address family 'af' and
257  * store it in 'n'.
258  * Returns 0 on success, -1 on failure.
259  */
260 static int
261 addr_hostmask(int af, u_int l, struct xaddr *n)
262 {
263 	if (addr_netmask(af, l, n) == -1 || addr_invert(n) == -1)
264 		return (-1);
265 	return (0);
266 }
267 
268 /*
269  * Test whether address 'a' is all zeros (i.e. 0.0.0.0 or ::)
270  * Returns 0 on if address is all-zeros, -1 if not all zeros or on failure.
271  */
272 static int
273 addr_is_all0s(const struct xaddr *a)
274 {
275 	int i;
276 
277 	switch (a->af) {
278 	case AF_INET:
279 		return (a->v4.s_addr == 0 ? 0 : -1);
280 	case AF_INET6:;
281 		for (i = 0; i < 4; i++)
282 			if (a->addr32[i] != 0)
283 				return (-1);
284 		return (0);
285 	default:
286 		return (-1);
287 	}
288 }
289 
290 /*
291  * Test whether host portion of address 'a', as determined by 'masklen'
292  * is all zeros.
293  * Returns 0 on if host portion of address is all-zeros,
294  * -1 if not all zeros or on failure.
295  */
296 static int
297 addr_host_is_all0s(const struct xaddr *a, u_int masklen)
298 {
299 	struct xaddr tmp_addr, tmp_mask, tmp_result;
300 
301 	memcpy(&tmp_addr, a, sizeof(tmp_addr));
302 	if (addr_hostmask(a->af, masklen, &tmp_mask) == -1)
303 		return (-1);
304 	if (addr_and(&tmp_result, &tmp_addr, &tmp_mask) == -1)
305 		return (-1);
306 	return (addr_is_all0s(&tmp_result));
307 }
308 
309 /*
310  * Parse a CIDR address (x.x.x.x/y or xxxx:yyyy::/z).
311  * Return -1 on parse error, -2 on inconsistency or 0 on success.
312  */
313 static int
314 addr_pton_cidr(const char *p, struct xaddr *n, u_int *l)
315 {
316 	struct xaddr tmp;
317 	long unsigned int masklen = 999;
318 	char addrbuf[64], *mp, *cp;
319 
320 	/* Don't modify argument */
321 	if (p == NULL || strlcpy(addrbuf, p, sizeof(addrbuf)) > sizeof(addrbuf))
322 		return -1;
323 
324 	if ((mp = strchr(addrbuf, '/')) != NULL) {
325 		*mp = '\0';
326 		mp++;
327 		masklen = strtoul(mp, &cp, 10);
328 		if (*mp == '\0' || *cp != '\0' || masklen > 128)
329 			return -1;
330 	}
331 
332 	if (addr_pton(addrbuf, &tmp) == -1)
333 		return -1;
334 
335 	if (mp == NULL)
336 		masklen = addr_unicast_masklen(tmp.af);
337 	if (masklen_valid(tmp.af, masklen) == -1)
338 		return -2;
339 	if (addr_host_is_all0s(&tmp, masklen) != 0)
340 		return -2;
341 
342 	if (n != NULL)
343 		memcpy(n, &tmp, sizeof(*n));
344 	if (l != NULL)
345 		*l = masklen;
346 
347 	return 0;
348 }
349 
350 static int
351 addr_netmatch(const struct xaddr *host, const struct xaddr *net, u_int masklen)
352 {
353 	struct xaddr tmp_mask, tmp_result;
354 
355 	if (host->af != net->af)
356 		return -1;
357 
358 	if (addr_netmask(host->af, masklen, &tmp_mask) == -1)
359 		return -1;
360 	if (addr_and(&tmp_result, host, &tmp_mask) == -1)
361 		return -1;
362 	return addr_cmp(&tmp_result, net);
363 }
364 
365 /*
366  * Match "addr" against list pattern list "_list", which may contain a
367  * mix of CIDR addresses and old-school wildcards.
368  *
369  * If addr is NULL, then no matching is performed, but _list is parsed
370  * and checked for well-formedness.
371  *
372  * Returns 1 on match found (never returned when addr == NULL).
373  * Returns 0 on if no match found, or no errors found when addr == NULL.
374  * Returns -1 on negated match found (never returned when addr == NULL).
375  * Returns -2 on invalid list entry.
376  */
377 int
378 addr_match_list(const char *addr, const char *_list)
379 {
380 	char *list, *cp, *o;
381 	struct xaddr try_addr, match_addr;
382 	u_int masklen, neg;
383 	int ret = 0, r;
384 
385 	if (addr != NULL && addr_pton(addr, &try_addr) != 0) {
386 		debug2("%s: couldn't parse address %.100s", __func__, addr);
387 		return 0;
388 	}
389 	if ((o = list = strdup(_list)) == NULL)
390 		return -1;
391 	while ((cp = strsep(&list, ",")) != NULL) {
392 		neg = *cp == '!';
393 		if (neg)
394 			cp++;
395 		if (*cp == '\0') {
396 			ret = -2;
397 			break;
398 		}
399 		/* Prefer CIDR address matching */
400 		r = addr_pton_cidr(cp, &match_addr, &masklen);
401 		if (r == -2) {
402 			error("Inconsistent mask length for "
403 			    "network \"%.100s\"", cp);
404 			ret = -2;
405 			break;
406 		} else if (r == 0) {
407 			if (addr != NULL && addr_netmatch(&try_addr,
408                            &match_addr, masklen) == 0) {
409  foundit:
410 				if (neg) {
411 					ret = -1;
412 					break;
413 				}
414 				ret = 1;
415 			}
416 			continue;
417 		} else {
418 			/* If CIDR parse failed, try wildcard string match */
419 			if (addr != NULL && match_pattern(addr, cp) == 1)
420 				goto foundit;
421 		}
422 	}
423 	xfree(o);
424 
425 	return ret;
426 }
427