1 /* $NetBSD: test_addr.c,v 1.1.1.1 2011/04/13 18:15:38 elric Exp $ */
2
3 /*
4 * Copyright (c) 2005 Kungliga Tekniska Högskolan
5 * (Royal Institute of Technology, Stockholm, Sweden).
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 *
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 *
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 * 3. Neither the name of KTH nor the names of its contributors may be
20 * used to endorse or promote products derived from this software without
21 * specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY KTH AND ITS CONTRIBUTORS ``AS IS'' AND ANY
24 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
26 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL KTH OR ITS CONTRIBUTORS BE
27 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
28 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
29 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
30 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
31 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
32 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
33 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */
34
35 #include "krb5_locl.h"
36 #include <err.h>
37
38 static void
print_addr(krb5_context context,const char * addr)39 print_addr(krb5_context context, const char *addr)
40 {
41 krb5_addresses addresses;
42 krb5_error_code ret;
43 char buf[38];
44 char buf2[1000];
45 size_t len;
46 int i;
47
48 ret = krb5_parse_address(context, addr, &addresses);
49 if (ret)
50 krb5_err(context, 1, ret, "krb5_parse_address");
51
52 if (addresses.len < 1)
53 krb5_err(context, 1, ret, "too few addresses");
54
55 for (i = 0; i < addresses.len; i++) {
56 krb5_print_address(&addresses.val[i], buf, sizeof(buf), &len);
57 #if 0
58 printf("addr %d: %s (%d/%d)\n", i, buf, (int)len, (int)strlen(buf));
59 #endif
60 if (strlen(buf) > sizeof(buf))
61 krb5_err(context, 1, ret, "len %d larger then buf %d",
62 (int)strlen(buf), (int)sizeof(buf));
63 krb5_print_address(&addresses.val[i], buf2, sizeof(buf2), &len);
64 #if 0
65 printf("addr %d: %s (%d/%d)\n", i, buf2, (int)len, (int)strlen(buf2));
66 #endif
67 if (strlen(buf2) > sizeof(buf2))
68 krb5_err(context, 1, ret, "len %d larger then buf %d",
69 (int)strlen(buf2), (int)sizeof(buf2));
70
71 }
72 krb5_free_addresses(context, &addresses);
73
74 }
75
76 static void
truncated_addr(krb5_context context,const char * addr,size_t truncate_len,size_t outlen)77 truncated_addr(krb5_context context, const char *addr,
78 size_t truncate_len, size_t outlen)
79 {
80 krb5_addresses addresses;
81 krb5_error_code ret;
82 char *buf;
83 size_t len;
84
85 buf = ecalloc(1, outlen + 1);
86
87 ret = krb5_parse_address(context, addr, &addresses);
88 if (ret)
89 krb5_err(context, 1, ret, "krb5_parse_address");
90
91 if (addresses.len != 1)
92 krb5_err(context, 1, ret, "addresses should be one");
93
94 krb5_print_address(&addresses.val[0], buf, truncate_len, &len);
95
96 #if 0
97 printf("addr %s (%d/%d) should be %d\n", buf, (int)len, (int)strlen(buf), (int)outlen);
98 #endif
99
100 if (truncate_len > strlen(buf) + 1)
101 krb5_err(context, 1, ret, "%s truncate_len %d larger then strlen %d source %s",
102 buf, (int)truncate_len, (int)strlen(buf), addr);
103
104 if (outlen != len)
105 krb5_err(context, 1, ret, "%s: outlen %d != len %d",
106 buf, (int)outlen, (int)strlen(buf));
107
108 krb5_print_address(&addresses.val[0], buf, outlen + 1, &len);
109
110 #if 0
111 printf("addr %s (%d/%d)\n", buf, (int)len, (int)strlen(buf));
112 #endif
113
114 if (len != outlen)
115 abort();
116 if (strlen(buf) != len)
117 abort();
118
119 krb5_free_addresses(context, &addresses);
120 free(buf);
121 }
122
123 static void
check_truncation(krb5_context context,const char * addr)124 check_truncation(krb5_context context, const char *addr)
125 {
126 int i, len = strlen(addr);
127
128 truncated_addr(context, addr, len, len);
129
130 for (i = 0; i < len; i++)
131 truncated_addr(context, addr, i, len);
132 }
133
134 static void
match_addr(krb5_context context,const char * range_addr,const char * one_addr,int match)135 match_addr(krb5_context context, const char *range_addr,
136 const char *one_addr, int match)
137 {
138 krb5_addresses range, one;
139 krb5_error_code ret;
140
141 ret = krb5_parse_address(context, range_addr, &range);
142 if (ret)
143 krb5_err(context, 1, ret, "krb5_parse_address");
144
145 if (range.len != 1)
146 krb5_err(context, 1, ret, "wrong num of addresses");
147
148 ret = krb5_parse_address(context, one_addr, &one);
149 if (ret)
150 krb5_err(context, 1, ret, "krb5_parse_address");
151
152 if (one.len != 1)
153 krb5_err(context, 1, ret, "wrong num of addresses");
154
155 if (krb5_address_order(context, &range.val[0], &one.val[0]) == 0) {
156 if (!match)
157 krb5_errx(context, 1, "match when one shouldn't be");
158 } else {
159 if (match)
160 krb5_errx(context, 1, "no match when one should be");
161 }
162
163 krb5_free_addresses(context, &range);
164 krb5_free_addresses(context, &one);
165 }
166
167 #ifdef _MSC_VER
168
169 /* For the truncation tests, calling strcpy_s() or strcat_s() with a
170 size of 0 results in the invalid parameter handler being invoked.
171 For the debug version, the runtime also throws an assert. */
172
173 static void
inv_param_handler(const wchar_t * expression,const wchar_t * function,const wchar_t * file,unsigned int line,uintptr_t pReserved)174 inv_param_handler(const wchar_t* expression,
175 const wchar_t* function,
176 const wchar_t* file,
177 unsigned int line,
178 uintptr_t pReserved)
179 {
180 printf("Invalid parameter handler invoked for: %S in %S(%d) [%S]\n",
181 function, file, line, expression);
182 }
183
184 static _invalid_parameter_handler _inv_old = NULL;
185
186 #define SET_INVALID_PARAM_HANDLER _inv_old = _set_invalid_parameter_handler(inv_param_handler)
187
188 #else
189
190 #define SET_INVALID_PARAM_HANDLER ((void) 0)
191
192 #endif
193
194 int
main(int argc,char ** argv)195 main(int argc, char **argv)
196 {
197 krb5_context context;
198 krb5_error_code ret;
199
200 SET_INVALID_PARAM_HANDLER;
201
202 setprogname(argv[0]);
203
204 ret = krb5_init_context(&context);
205 if (ret)
206 errx (1, "krb5_init_context failed: %d", ret);
207
208 print_addr(context, "RANGE:127.0.0.0/8");
209 print_addr(context, "RANGE:127.0.0.0/24");
210 print_addr(context, "RANGE:IPv4:127.0.0.0-IPv4:127.0.0.255");
211 print_addr(context, "RANGE:130.237.237.4/29");
212 #ifdef HAVE_IPV6
213 print_addr(context, "RANGE:2001:db8:1:2:3:4:1428:7ab/64");
214 print_addr(context, "RANGE:IPv6:fe80::209:6bff:fea0:e522/64");
215 print_addr(context, "RANGE:IPv6:fe80::-IPv6:fe80::ffff:ffff:ffff:ffff");
216 print_addr(context, "RANGE:fe80::-fe80::ffff:ffff:ffff:ffff");
217 #endif
218
219 check_truncation(context, "IPv4:127.0.0.0");
220 check_truncation(context, "RANGE:IPv4:127.0.0.0-IPv4:127.0.0.255");
221 #ifdef HAVE_IPV6
222 check_truncation(context, "IPv6:::");
223 check_truncation(context, "IPv6:::1");
224 check_truncation(context, "IPv6:2001:db8:1:2:3:4:1428:7ab");
225 check_truncation(context, "IPv6:fe80::209:0:0:0");
226 check_truncation(context, "IPv6:fe80::ffff:ffff:ffff:ffff");
227 #endif
228
229 match_addr(context, "RANGE:127.0.0.0/8", "inet:127.0.0.0", 1);
230 match_addr(context, "RANGE:127.0.0.0/8", "inet:127.255.255.255", 1);
231 match_addr(context, "RANGE:127.0.0.0/8", "inet:128.0.0.0", 0);
232
233 match_addr(context, "RANGE:130.237.237.8/29", "inet:130.237.237.7", 0);
234 match_addr(context, "RANGE:130.237.237.8/29", "inet:130.237.237.8", 1);
235 match_addr(context, "RANGE:130.237.237.8/29", "inet:130.237.237.15", 1);
236 match_addr(context, "RANGE:130.237.237.8/29", "inet:130.237.237.16", 0);
237
238 krb5_free_context(context);
239
240 return 0;
241 }
242