Merge branch 'vendor/OPENSSH'
[dragonfly.git] / crypto / openssh / addrmatch.c
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         struct sockaddr_in *in4 = (struct sockaddr_in *)sa;
85         struct sockaddr_in6 *in6 = (struct sockaddr_in6 *)sa;
86
87         memset(xa, '\0', sizeof(*xa));
88
89         switch (sa->sa_family) {
90         case AF_INET:
91                 if (slen < sizeof(*in4))
92                         return -1;
93                 xa->af = AF_INET;
94                 memcpy(&xa->v4, &in4->sin_addr, sizeof(xa->v4));
95                 break;
96         case AF_INET6:
97                 if (slen < sizeof(*in6))
98                         return -1;
99                 xa->af = AF_INET6;
100                 memcpy(&xa->v6, &in6->sin6_addr, sizeof(xa->v6));
101 #ifdef HAVE_STRUCT_SOCKADDR_IN6_SIN6_SCOPE_ID
102                 xa->scope_id = in6->sin6_scope_id;
103 #endif
104                 break;
105         default:
106                 return -1;
107         }
108
109         return 0;
110 }
111
112 /*
113  * Calculate a netmask of length 'l' for address family 'af' and
114  * store it in 'n'.
115  * Returns 0 on success, -1 on failure.
116  */
117 static int
118 addr_netmask(int af, u_int l, struct xaddr *n)
119 {
120         int i;
121
122         if (masklen_valid(af, l) != 0 || n == NULL)
123                 return -1;
124
125         memset(n, '\0', sizeof(*n));
126         switch (af) {
127         case AF_INET:
128                 n->af = AF_INET;
129                 n->v4.s_addr = htonl((0xffffffff << (32 - l)) & 0xffffffff);
130                 return 0;
131         case AF_INET6:
132                 n->af = AF_INET6;
133                 for (i = 0; i < 4 && l >= 32; i++, l -= 32)
134                         n->addr32[i] = 0xffffffffU;
135                 if (i < 4 && l != 0)
136                         n->addr32[i] = htonl((0xffffffff << (32 - l)) &
137                             0xffffffff);
138                 return 0;
139         default:
140                 return -1;
141         }
142 }
143
144 /*
145  * Perform logical AND of addresses 'a' and 'b', storing result in 'dst'.
146  * Returns 0 on success, -1 on failure.
147  */
148 static int
149 addr_and(struct xaddr *dst, const struct xaddr *a, const struct xaddr *b)
150 {
151         int i;
152
153         if (dst == NULL || a == NULL || b == NULL || a->af != b->af)
154                 return -1;
155
156         memcpy(dst, a, sizeof(*dst));
157         switch (a->af) {
158         case AF_INET:
159                 dst->v4.s_addr &= b->v4.s_addr;
160                 return 0;
161         case AF_INET6:
162                 dst->scope_id = a->scope_id;
163                 for (i = 0; i < 4; i++)
164                         dst->addr32[i] &= b->addr32[i];
165                 return 0;
166         default:
167                 return -1;
168         }
169 }
170
171 /*
172  * Compare addresses 'a' and 'b'
173  * Return 0 if addresses are identical, -1 if (a < b) or 1 if (a > b)
174  */
175 static int
176 addr_cmp(const struct xaddr *a, const struct xaddr *b)
177 {
178         int i;
179
180         if (a->af != b->af)
181                 return a->af == AF_INET6 ? 1 : -1;
182
183         switch (a->af) {
184         case AF_INET:
185                 if (a->v4.s_addr == b->v4.s_addr)
186                         return 0;
187                 return ntohl(a->v4.s_addr) > ntohl(b->v4.s_addr) ? 1 : -1;
188         case AF_INET6:
189                 for (i = 0; i < 16; i++)
190                         if (a->addr8[i] - b->addr8[i] != 0)
191                                 return a->addr8[i] > b->addr8[i] ? 1 : -1;
192                 if (a->scope_id == b->scope_id)
193                         return 0;
194                 return a->scope_id > b->scope_id ? 1 : -1;
195         default:
196                 return -1;
197         }
198 }
199
200 /*
201  * Parse string address 'p' into 'n'
202  * Returns 0 on success, -1 on failure.
203  */
204 static int
205 addr_pton(const char *p, struct xaddr *n)
206 {
207         struct addrinfo hints, *ai;
208
209         memset(&hints, '\0', sizeof(hints));
210         hints.ai_flags = AI_NUMERICHOST;
211
212         if (p == NULL || getaddrinfo(p, NULL, &hints, &ai) != 0)
213                 return -1;
214
215         if (ai == NULL || ai->ai_addr == NULL)
216                 return -1;
217
218         if (n != NULL &&
219             addr_sa_to_xaddr(ai->ai_addr, ai->ai_addrlen, n) == -1) {
220                 freeaddrinfo(ai);
221                 return -1;
222         }
223
224         freeaddrinfo(ai);
225         return 0;
226 }
227
228 /*
229  * Perform bitwise negation of address
230  * Returns 0 on success, -1 on failure.
231  */
232 static int
233 addr_invert(struct xaddr *n)
234 {
235         int i;
236
237         if (n == NULL)
238                 return (-1);
239
240         switch (n->af) {
241         case AF_INET:
242                 n->v4.s_addr = ~n->v4.s_addr;
243                 return (0);
244         case AF_INET6:
245                 for (i = 0; i < 4; i++)
246                         n->addr32[i] = ~n->addr32[i];
247                 return (0);
248         default:
249                 return (-1);
250         }
251 }
252
253 /*
254  * Calculate a netmask of length 'l' for address family 'af' and
255  * store it in 'n'.
256  * Returns 0 on success, -1 on failure.
257  */
258 static int
259 addr_hostmask(int af, u_int l, struct xaddr *n)
260 {
261         if (addr_netmask(af, l, n) == -1 || addr_invert(n) == -1)
262                 return (-1);
263         return (0);
264 }
265
266 /*
267  * Test whether address 'a' is all zeros (i.e. 0.0.0.0 or ::)
268  * Returns 0 on if address is all-zeros, -1 if not all zeros or on failure.
269  */
270 static int
271 addr_is_all0s(const struct xaddr *a)
272 {
273         int i;
274
275         switch (a->af) {
276         case AF_INET:
277                 return (a->v4.s_addr == 0 ? 0 : -1);
278         case AF_INET6:;
279                 for (i = 0; i < 4; i++)
280                         if (a->addr32[i] != 0)
281                                 return (-1);
282                 return (0);
283         default:
284                 return (-1);
285         }
286 }
287
288 /*
289  * Test whether host portion of address 'a', as determined by 'masklen'
290  * is all zeros.
291  * Returns 0 on if host portion of address is all-zeros,
292  * -1 if not all zeros or on failure.
293  */
294 static int
295 addr_host_is_all0s(const struct xaddr *a, u_int masklen)
296 {
297         struct xaddr tmp_addr, tmp_mask, tmp_result;
298
299         memcpy(&tmp_addr, a, sizeof(tmp_addr));
300         if (addr_hostmask(a->af, masklen, &tmp_mask) == -1)
301                 return (-1);
302         if (addr_and(&tmp_result, &tmp_addr, &tmp_mask) == -1)
303                 return (-1);
304         return (addr_is_all0s(&tmp_result));
305 }
306
307 /*
308  * Parse a CIDR address (x.x.x.x/y or xxxx:yyyy::/z).
309  * Return -1 on parse error, -2 on inconsistency or 0 on success.
310  */
311 static int
312 addr_pton_cidr(const char *p, struct xaddr *n, u_int *l)
313 {
314         struct xaddr tmp;
315         long unsigned int masklen = 999;
316         char addrbuf[64], *mp, *cp;
317
318         /* Don't modify argument */
319         if (p == NULL || strlcpy(addrbuf, p, sizeof(addrbuf)) > sizeof(addrbuf))
320                 return -1;
321
322         if ((mp = strchr(addrbuf, '/')) != NULL) {
323                 *mp = '\0';
324                 mp++;
325                 masklen = strtoul(mp, &cp, 10);
326                 if (*mp == '\0' || *cp != '\0' || masklen > 128)
327                         return -1;
328         }
329
330         if (addr_pton(addrbuf, &tmp) == -1)
331                 return -1;
332
333         if (mp == NULL)
334                 masklen = addr_unicast_masklen(tmp.af);
335         if (masklen_valid(tmp.af, masklen) == -1)
336                 return -2;
337         if (addr_host_is_all0s(&tmp, masklen) != 0)
338                 return -2;
339
340         if (n != NULL)
341                 memcpy(n, &tmp, sizeof(*n));
342         if (l != NULL)
343                 *l = masklen;
344
345         return 0;
346 }
347
348 static int
349 addr_netmatch(const struct xaddr *host, const struct xaddr *net, u_int masklen)
350 {
351         struct xaddr tmp_mask, tmp_result;
352
353         if (host->af != net->af)
354                 return -1;
355
356         if (addr_netmask(host->af, masklen, &tmp_mask) == -1)
357                 return -1;
358         if (addr_and(&tmp_result, host, &tmp_mask) == -1)
359                 return -1;
360         return addr_cmp(&tmp_result, net);
361 }
362
363 /*
364  * Match "addr" against list pattern list "_list", which may contain a
365  * mix of CIDR addresses and old-school wildcards.
366  *
367  * If addr is NULL, then no matching is performed, but _list is parsed
368  * and checked for well-formedness.
369  *
370  * Returns 1 on match found (never returned when addr == NULL).
371  * Returns 0 on if no match found, or no errors found when addr == NULL.
372  * Returns -1 on negated match found (never returned when addr == NULL).
373  * Returns -2 on invalid list entry.
374  */
375 int
376 addr_match_list(const char *addr, const char *_list)
377 {
378         char *list, *cp, *o;
379         struct xaddr try_addr, match_addr;
380         u_int masklen, neg;
381         int ret = 0, r;
382
383         if (addr != NULL && addr_pton(addr, &try_addr) != 0) {
384                 debug2("%s: couldn't parse address %.100s", __func__, addr);
385                 return 0;
386         }
387         if ((o = list = strdup(_list)) == NULL)
388                 return -1;
389         while ((cp = strsep(&list, ",")) != NULL) {
390                 neg = *cp == '!';
391                 if (neg)
392                         cp++;
393                 if (*cp == '\0') {
394                         ret = -2;
395                         break;
396                 }
397                 /* Prefer CIDR address matching */
398                 r = addr_pton_cidr(cp, &match_addr, &masklen);
399                 if (r == -2) {
400                         error("Inconsistent mask length for "
401                             "network \"%.100s\"", cp);
402                         ret = -2;
403                         break;
404                 } else if (r == 0) {
405                         if (addr != NULL && addr_netmatch(&try_addr,
406                            &match_addr, masklen) == 0) {
407  foundit:
408                                 if (neg) {
409                                         ret = -1;
410                                         break;
411                                 }
412                                 ret = 1;
413                         }
414                         continue;
415                 } else {
416                         /* If CIDR parse failed, try wildcard string match */
417                         if (addr != NULL && match_pattern(addr, cp) == 1)
418                                 goto foundit;
419                 }
420         }
421         xfree(o);
422
423         return ret;
424 }