1/*
2 * Copyright (c) 2003 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 *
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * 3. Neither the name of KTH nor the names of its contributors may be
18 *    used to endorse or promote products derived from this software without
19 *    specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY KTH AND ITS CONTRIBUTORS ``AS IS'' AND ANY
22 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
24 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL KTH OR ITS CONTRIBUTORS BE
25 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
28 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
29 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
30 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
31 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34#include "krb5_locl.h"
35#include <err.h>
36
37static int
38check_config_file(krb5_context context, char *filelist, char **res, int def)
39{
40    krb5_error_code ret;
41    char **pp;
42    int i;
43
44    pp = NULL;
45
46    if (def)
47	ret = krb5_prepend_config_files_default(filelist, &pp);
48    else
49	ret = krb5_prepend_config_files(filelist, NULL, &pp);
50
51    if (ret)
52	krb5_err(context, 1, ret, "prepend_config_files");
53
54    for (i = 0; res[i] && pp[i]; i++)
55	if (strcmp(pp[i], res[i]) != 0)
56	    krb5_errx(context, 1, "'%s' != '%s'", pp[i], res[i]);
57
58    if (res[i] != NULL)
59	krb5_errx(context, 1, "pp ended before res list");
60
61    if (def) {
62	char **deflist;
63	int j;
64
65	ret = krb5_get_default_config_files(&deflist);
66	if (ret)
67	    krb5_err(context, 1, ret, "get_default_config_files");
68
69	for (j = 0 ; pp[i] && deflist[j]; i++, j++)
70	    if (strcmp(pp[i], deflist[j]) != 0)
71		krb5_errx(context, 1, "'%s' != '%s'", pp[i], deflist[j]);
72
73	if (deflist[j] != NULL)
74	    krb5_errx(context, 1, "pp ended before def list");
75	krb5_free_config_files(deflist);
76    }
77
78    if (pp[i] != NULL)
79	krb5_errx(context, 1, "pp ended after res (and def) list");
80
81    krb5_free_config_files(pp);
82
83    return 0;
84}
85
86char *list0[] =  { "/tmp/foo", NULL };
87char *list1[] =  { "/tmp/foo", "/tmp/foo/bar", NULL };
88char *list2[] =  { "", NULL };
89
90struct {
91    char *fl;
92    char **res;
93} test[] = {
94    { "/tmp/foo", NULL },
95    { "/tmp/foo" PATH_SEP "/tmp/foo/bar", NULL },
96    { "", NULL }
97};
98
99static void
100check_config_files(void)
101{
102    krb5_context context;
103    krb5_error_code ret;
104    size_t i;
105
106    ret = krb5_init_context(&context);
107    if (ret)
108	errx(1, "krb5_init_context %d", ret);
109
110    test[0].res = list0;
111    test[1].res = list1;
112    test[2].res = list2;
113
114    for (i = 0; i < sizeof(test)/sizeof(*test); i++) {
115	check_config_file(context, test[i].fl, test[i].res, 0);
116	check_config_file(context, test[i].fl, test[i].res, 1);
117    }
118
119    krb5_free_context(context);
120}
121
122static void
123check_notify(void)
124{
125    krb5_context context, context2;
126    krb5_error_code ret;
127    krb5_boolean reread = 0, reread2 = 0;
128
129    /* check notify */
130
131    ret = krb5_init_context(&context);
132    if (ret)
133	errx(1, "krb5_init_context %d", ret);
134
135    ret = krb5_init_context(&context2);
136    if (ret)
137	errx(1, "krb5_init_context %d", ret);
138
139
140#ifdef HAVE_NOTIFY_H
141    notify_post(KRB5_CONFIGURATION_CHANGE_NOTIFY_NAME);
142#endif
143
144    ret = krb5_reload_config(context, 0, &reread);
145    if (ret)
146	errx(1, "krb5_reload_config failed");
147
148    ret = krb5_reload_config(context2, 0, &reread2);
149    if (ret)
150	errx(1, "krb5_reload_config failed");
151
152#ifdef HAVE_NOTIFY_H
153    if (!reread)
154	errx(1, "configuration not reread!");
155#endif
156#ifdef HAVE_NOTIFY_H
157    if (!reread2)
158	errx(1, "configuration not reread!");
159#endif
160
161    krb5_free_context(context);
162    krb5_free_context(context2);
163}
164
165static void
166check_alloc_loop(void)
167{
168    krb5_context context;
169    krb5_error_code ret;
170    unsigned int i;
171
172    /*
173     * Check that looping a couple of times doesn't corrupt heap
174     */
175    for (i = 0; i < 10000; i++) {
176	ret = krb5_init_context(&context);
177	if (ret)
178	    errx(1, "krb5_init_context %d", ret);
179	krb5_free_context(context);
180    }
181}
182
183
184const char *config_string_result0[] = {
185    "A", "B", "C", "D", NULL
186};
187
188const char *config_string_result1[] = {
189    "A", "B", "C D", NULL
190};
191
192const char *config_string_result2[] = {
193    "A", "B", "", NULL
194};
195
196const char *config_string_result3[] = {
197    "A B;C: D", NULL
198};
199
200const char *config_string_result4[] = {
201    "\"\"", "", "\"\"", NULL
202};
203
204const char *config_string_result5[] = {
205    "A\"BQd", NULL
206};
207
208const char *config_string_result6[] = {
209    "efgh\"", "ABC", NULL
210};
211
212const char *config_string_result7[] = {
213    "SnapeKills\\", "Dumbledore", NULL
214};
215
216const char *config_string_result8[] = {
217    "\"TownOf Sandwich: Massachusetts\"Oldest", "Town", "In", "Cape Cod", NULL
218};
219
220const char *config_string_result9[] = {
221    "\"Begins and\"ends", "In", "One", "String", NULL
222};
223
224const char *config_string_result10[] = {
225    "Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:",
226    "1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.",
227    "2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.",
228    "3. Neither the name of the Institute nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.",
229    "THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.",
230    "Why do we test with such long strings? Because some people have config files",
231    "That", "look", "Like this.", NULL
232};
233
234const struct {
235    const char * name;
236    const char ** expected;
237} config_strings_tests[] = {
238    { "foo", config_string_result0 },
239    { "bar", config_string_result1 },
240    { "baz", config_string_result2 },
241    { "quux", config_string_result3 },
242    { "questionable", config_string_result4 },
243    { "mismatch1", config_string_result5 },
244    { "mismatch2", config_string_result6 },
245    { "internal1", config_string_result7 },
246    { "internal2", config_string_result8 },
247    { "internal3", config_string_result9 },
248    { "longer_strings", config_string_result10 }
249};
250
251static void
252check_escaped_strings(void)
253{
254    krb5_context context;
255    krb5_config_section *c = NULL;
256    krb5_error_code ret;
257    size_t i;
258
259    ret = krb5_init_context(&context);
260    if (ret)
261        errx(1, "krb5_init_context %d", ret);
262
263    ret = krb5_config_parse_file(context, "test_config_strings.out", &c);
264    if (ret)
265        krb5_errx(context, 1, "krb5_config_parse_file()");
266
267    for (i=0; i < sizeof(config_strings_tests)/sizeof(config_strings_tests[0]); i++) {
268        char **ps;
269        const char **s;
270        const char **e;
271
272        ps = krb5_config_get_strings(context, c, "escapes", config_strings_tests[i].name,
273                                     NULL);
274        if (ps == NULL)
275            errx(1, "Failed to read string value %s", config_strings_tests[i].name);
276
277        e = config_strings_tests[i].expected;
278
279        for (s = (const char **)ps; *s && *e; s++, e++) {
280            if (strcmp(*s, *e))
281                errx(1,
282                     "Unexpected configuration string at value [%s].\n"
283                     "Actual=[%s]\n"
284                     "Expected=[%s]\n",
285                     config_strings_tests[i].name, *s, *e);
286        }
287
288        if (*s || *e)
289            errx(1, "Configuation string list for value [%s] has incorrect length.",
290		 config_strings_tests[i].name);
291
292        krb5_config_free_strings(ps);
293    }
294
295    ret = krb5_config_file_free(context, c);
296    if (ret)
297        krb5_errx(context, 1, "krb5_config_file_free()");
298
299    krb5_free_context(context);
300}
301
302int
303main(int argc, char **argv)
304{
305    check_config_files();
306    check_notify();
307    check_alloc_loop();
308    check_escaped_strings();
309
310    return 0;
311}
312