1/*
2 * Copyright (c) 2001-2002 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 the Institute nor the names of its contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#include "krb5_locl.h"
35
36RCSID("$Id: keytab_any.c 17035 2006-04-10 09:20:13Z lha $");
37
38struct any_data {
39    krb5_keytab kt;
40    char *name;
41    struct any_data *next;
42};
43
44static void
45free_list (krb5_context context, struct any_data *a)
46{
47    struct any_data *next;
48
49    for (; a != NULL; a = next) {
50	next = a->next;
51	free (a->name);
52	if(a->kt)
53	    krb5_kt_close(context, a->kt);
54	free (a);
55    }
56}
57
58static krb5_error_code
59any_resolve(krb5_context context, const char *name, krb5_keytab id)
60{
61    struct any_data *a, *a0 = NULL, *prev = NULL;
62    krb5_error_code ret;
63    char buf[256];
64
65    while (strsep_copy(&name, ",", buf, sizeof(buf)) != -1) {
66	a = malloc(sizeof(*a));
67	if (a == NULL) {
68	    ret = ENOMEM;
69	    goto fail;
70	}
71	if (a0 == NULL) {
72	    a0 = a;
73	    a->name = strdup(buf);
74	    if (a->name == NULL) {
75		krb5_set_error_string(context, "malloc: out of memory");
76		ret = ENOMEM;
77		goto fail;
78	    }
79	} else
80	    a->name = NULL;
81	if (prev != NULL)
82	    prev->next = a;
83	a->next = NULL;
84	ret = krb5_kt_resolve (context, buf, &a->kt);
85	if (ret)
86	    goto fail;
87	prev = a;
88    }
89    if (a0 == NULL) {
90	krb5_set_error_string(context, "empty ANY: keytab");
91	return ENOENT;
92    }
93    id->data = a0;
94    return 0;
95 fail:
96    free_list (context, a0);
97    return ret;
98}
99
100static krb5_error_code
101any_get_name (krb5_context context,
102	      krb5_keytab id,
103	      char *name,
104	      size_t namesize)
105{
106    struct any_data *a = id->data;
107    strlcpy(name, a->name, namesize);
108    return 0;
109}
110
111static krb5_error_code
112any_close (krb5_context context,
113	   krb5_keytab id)
114{
115    struct any_data *a = id->data;
116
117    free_list (context, a);
118    return 0;
119}
120
121struct any_cursor_extra_data {
122    struct any_data *a;
123    krb5_kt_cursor cursor;
124};
125
126static krb5_error_code
127any_start_seq_get(krb5_context context,
128		  krb5_keytab id,
129		  krb5_kt_cursor *c)
130{
131    struct any_data *a = id->data;
132    struct any_cursor_extra_data *ed;
133    krb5_error_code ret;
134
135    c->data = malloc (sizeof(struct any_cursor_extra_data));
136    if(c->data == NULL){
137	krb5_set_error_string (context, "malloc: out of memory");
138	return ENOMEM;
139    }
140    ed = (struct any_cursor_extra_data *)c->data;
141    ed->a = a;
142    ret = krb5_kt_start_seq_get(context, ed->a->kt, &ed->cursor);
143    if (ret) {
144	free (c->data);
145	c->data = NULL;
146	return ret;
147    }
148    return 0;
149}
150
151static krb5_error_code
152any_next_entry (krb5_context context,
153		krb5_keytab id,
154		krb5_keytab_entry *entry,
155		krb5_kt_cursor *cursor)
156{
157    krb5_error_code ret, ret2;
158    struct any_cursor_extra_data *ed;
159
160    ed = (struct any_cursor_extra_data *)cursor->data;
161    do {
162	ret = krb5_kt_next_entry(context, ed->a->kt, entry, &ed->cursor);
163	if (ret == 0)
164	    return 0;
165	else if (ret != KRB5_KT_END)
166	    return ret;
167
168	ret2 = krb5_kt_end_seq_get (context, ed->a->kt, &ed->cursor);
169	if (ret2)
170	    return ret2;
171	while ((ed->a = ed->a->next) != NULL) {
172	    ret2 = krb5_kt_start_seq_get(context, ed->a->kt, &ed->cursor);
173	    if (ret2 == 0)
174		break;
175	}
176	if (ed->a == NULL) {
177	    krb5_clear_error_string (context);
178	    return KRB5_KT_END;
179	}
180    } while (1);
181}
182
183static krb5_error_code
184any_end_seq_get(krb5_context context,
185		krb5_keytab id,
186		krb5_kt_cursor *cursor)
187{
188    krb5_error_code ret = 0;
189    struct any_cursor_extra_data *ed;
190
191    ed = (struct any_cursor_extra_data *)cursor->data;
192    if (ed->a != NULL)
193	ret = krb5_kt_end_seq_get(context, ed->a->kt, &ed->cursor);
194    free (ed);
195    cursor->data = NULL;
196    return ret;
197}
198
199static krb5_error_code
200any_add_entry(krb5_context context,
201	      krb5_keytab id,
202	      krb5_keytab_entry *entry)
203{
204    struct any_data *a = id->data;
205    krb5_error_code ret;
206    while(a != NULL) {
207	ret = krb5_kt_add_entry(context, a->kt, entry);
208	if(ret != 0 && ret != KRB5_KT_NOWRITE) {
209	    krb5_set_error_string(context, "failed to add entry to %s",
210				  a->name);
211	    return ret;
212	}
213	a = a->next;
214    }
215    return 0;
216}
217
218static krb5_error_code
219any_remove_entry(krb5_context context,
220		 krb5_keytab id,
221		 krb5_keytab_entry *entry)
222{
223    struct any_data *a = id->data;
224    krb5_error_code ret;
225    int found = 0;
226    while(a != NULL) {
227	ret = krb5_kt_remove_entry(context, a->kt, entry);
228	if(ret == 0)
229	    found++;
230	else {
231	    if(ret != KRB5_KT_NOWRITE && ret != KRB5_KT_NOTFOUND) {
232		krb5_set_error_string(context, "failed to remove entry from %s",
233				      a->name);
234		return ret;
235	    }
236	}
237	a = a->next;
238    }
239    if(!found)
240	return KRB5_KT_NOTFOUND;
241    return 0;
242}
243
244const krb5_kt_ops krb5_any_ops = {
245    "ANY",
246    any_resolve,
247    any_get_name,
248    any_close,
249    NULL, /* get */
250    any_start_seq_get,
251    any_next_entry,
252    any_end_seq_get,
253    any_add_entry,
254    any_remove_entry
255};
256