Merge pull request #11 from asankah/master
[heimdal.git] / kcm / cache.c
blob1bd220c8a70aa443f97d8a12250180d534fa48b3
1 /*
2 * Copyright (c) 2005, PADL Software Pty Ltd.
3 * All rights reserved.
5 * Portions Copyright (c) 2009 Apple Inc. All rights reserved.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
18 * 3. Neither the name of PADL Software nor the names of its contributors
19 * may be used to endorse or promote products derived from this software
20 * without specific prior written permission.
22 * THIS SOFTWARE IS PROVIDED BY PADL SOFTWARE AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED. IN NO EVENT SHALL PADL SOFTWARE OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
35 #include "kcm_locl.h"
37 HEIMDAL_MUTEX ccache_mutex = HEIMDAL_MUTEX_INITIALIZER;
38 kcm_ccache_data *ccache_head = NULL;
39 static unsigned int ccache_nextid = 0;
41 char *kcm_ccache_nextid(pid_t pid, uid_t uid, gid_t gid)
43 unsigned n;
44 char *name;
46 HEIMDAL_MUTEX_lock(&ccache_mutex);
47 n = ++ccache_nextid;
48 HEIMDAL_MUTEX_unlock(&ccache_mutex);
50 asprintf(&name, "%ld:%u", (long)uid, n);
52 return name;
55 krb5_error_code
56 kcm_ccache_resolve(krb5_context context,
57 const char *name,
58 kcm_ccache *ccache)
60 kcm_ccache p;
61 krb5_error_code ret;
63 *ccache = NULL;
65 ret = KRB5_FCC_NOFILE;
67 HEIMDAL_MUTEX_lock(&ccache_mutex);
69 for (p = ccache_head; p != NULL; p = p->next) {
70 if ((p->flags & KCM_FLAGS_VALID) == 0)
71 continue;
72 if (strcmp(p->name, name) == 0) {
73 ret = 0;
74 break;
78 if (ret == 0) {
79 kcm_retain_ccache(context, p);
80 *ccache = p;
83 HEIMDAL_MUTEX_unlock(&ccache_mutex);
85 return ret;
88 krb5_error_code
89 kcm_ccache_resolve_by_uuid(krb5_context context,
90 kcmuuid_t uuid,
91 kcm_ccache *ccache)
93 kcm_ccache p;
94 krb5_error_code ret;
96 *ccache = NULL;
98 ret = KRB5_FCC_NOFILE;
100 HEIMDAL_MUTEX_lock(&ccache_mutex);
102 for (p = ccache_head; p != NULL; p = p->next) {
103 if ((p->flags & KCM_FLAGS_VALID) == 0)
104 continue;
105 if (memcmp(p->uuid, uuid, sizeof(uuid)) == 0) {
106 ret = 0;
107 break;
111 if (ret == 0) {
112 kcm_retain_ccache(context, p);
113 *ccache = p;
116 HEIMDAL_MUTEX_unlock(&ccache_mutex);
118 return ret;
121 krb5_error_code
122 kcm_ccache_get_uuids(krb5_context context, kcm_client *client, kcm_operation opcode, krb5_storage *sp)
124 krb5_error_code ret;
125 kcm_ccache p;
127 ret = KRB5_FCC_NOFILE;
129 HEIMDAL_MUTEX_lock(&ccache_mutex);
131 for (p = ccache_head; p != NULL; p = p->next) {
132 if ((p->flags & KCM_FLAGS_VALID) == 0)
133 continue;
134 ret = kcm_access(context, client, opcode, p);
135 if (ret) {
136 ret = 0;
137 continue;
139 krb5_storage_write(sp, p->uuid, sizeof(p->uuid));
142 HEIMDAL_MUTEX_unlock(&ccache_mutex);
144 return ret;
148 krb5_error_code kcm_debug_ccache(krb5_context context)
150 kcm_ccache p;
152 for (p = ccache_head; p != NULL; p = p->next) {
153 char *cpn = NULL, *spn = NULL;
154 int ncreds = 0;
155 struct kcm_creds *k;
157 if ((p->flags & KCM_FLAGS_VALID) == 0) {
158 kcm_log(7, "cache %08x: empty slot");
159 continue;
162 KCM_ASSERT_VALID(p);
164 for (k = p->creds; k != NULL; k = k->next)
165 ncreds++;
167 if (p->client != NULL)
168 krb5_unparse_name(context, p->client, &cpn);
169 if (p->server != NULL)
170 krb5_unparse_name(context, p->server, &spn);
172 kcm_log(7, "cache %08x: name %s refcnt %d flags %04x mode %04o "
173 "uid %d gid %d client %s server %s ncreds %d",
174 p, p->name, p->refcnt, p->flags, p->mode, p->uid, p->gid,
175 (cpn == NULL) ? "<none>" : cpn,
176 (spn == NULL) ? "<none>" : spn,
177 ncreds);
179 if (cpn != NULL)
180 free(cpn);
181 if (spn != NULL)
182 free(spn);
185 return 0;
188 static void
189 kcm_free_ccache_data_internal(krb5_context context,
190 kcm_ccache_data *cache)
192 KCM_ASSERT_VALID(cache);
194 if (cache->name != NULL) {
195 free(cache->name);
196 cache->name = NULL;
199 if (cache->flags & KCM_FLAGS_USE_KEYTAB) {
200 krb5_kt_close(context, cache->key.keytab);
201 cache->key.keytab = NULL;
202 } else if (cache->flags & KCM_FLAGS_USE_CACHED_KEY) {
203 krb5_free_keyblock_contents(context, &cache->key.keyblock);
204 krb5_keyblock_zero(&cache->key.keyblock);
207 cache->flags = 0;
208 cache->mode = 0;
209 cache->uid = -1;
210 cache->gid = -1;
211 cache->session = -1;
213 kcm_zero_ccache_data_internal(context, cache);
215 cache->tkt_life = 0;
216 cache->renew_life = 0;
218 cache->next = NULL;
219 cache->refcnt = 0;
221 HEIMDAL_MUTEX_unlock(&cache->mutex);
222 HEIMDAL_MUTEX_destroy(&cache->mutex);
226 krb5_error_code
227 kcm_ccache_destroy(krb5_context context, const char *name)
229 kcm_ccache *p, ccache;
230 krb5_error_code ret;
232 ret = KRB5_FCC_NOFILE;
234 HEIMDAL_MUTEX_lock(&ccache_mutex);
235 for (p = &ccache_head; *p != NULL; p = &(*p)->next) {
236 if (((*p)->flags & KCM_FLAGS_VALID) == 0)
237 continue;
238 if (strcmp((*p)->name, name) == 0) {
239 ret = 0;
240 break;
243 if (ret)
244 goto out;
246 if ((*p)->refcnt != 1) {
247 ret = EAGAIN;
248 goto out;
251 ccache = *p;
252 *p = (*p)->next;
253 kcm_free_ccache_data_internal(context, ccache);
254 free(ccache);
256 out:
257 HEIMDAL_MUTEX_unlock(&ccache_mutex);
259 return ret;
262 static krb5_error_code
263 kcm_ccache_alloc(krb5_context context,
264 const char *name,
265 kcm_ccache *ccache)
267 kcm_ccache slot = NULL, p;
268 krb5_error_code ret;
269 int new_slot = 0;
271 *ccache = NULL;
273 /* First, check for duplicates */
274 HEIMDAL_MUTEX_lock(&ccache_mutex);
275 ret = 0;
276 for (p = ccache_head; p != NULL; p = p->next) {
277 if (p->flags & KCM_FLAGS_VALID) {
278 if (strcmp(p->name, name) == 0) {
279 ret = KRB5_CC_WRITE;
280 break;
282 } else if (slot == NULL)
283 slot = p;
286 if (ret)
287 goto out;
290 * Create an enpty slot for us.
292 if (slot == NULL) {
293 slot = (kcm_ccache_data *)malloc(sizeof(*slot));
294 if (slot == NULL) {
295 ret = KRB5_CC_NOMEM;
296 goto out;
298 slot->next = ccache_head;
299 HEIMDAL_MUTEX_init(&slot->mutex);
300 new_slot = 1;
303 RAND_bytes(slot->uuid, sizeof(slot->uuid));
305 slot->name = strdup(name);
306 if (slot->name == NULL) {
307 ret = KRB5_CC_NOMEM;
308 goto out;
311 slot->refcnt = 1;
312 slot->flags = KCM_FLAGS_VALID;
313 slot->mode = S_IRUSR | S_IWUSR;
314 slot->uid = -1;
315 slot->gid = -1;
316 slot->client = NULL;
317 slot->server = NULL;
318 slot->creds = NULL;
319 slot->key.keytab = NULL;
320 slot->tkt_life = 0;
321 slot->renew_life = 0;
323 if (new_slot)
324 ccache_head = slot;
326 *ccache = slot;
328 HEIMDAL_MUTEX_unlock(&ccache_mutex);
329 return 0;
331 out:
332 HEIMDAL_MUTEX_unlock(&ccache_mutex);
333 if (new_slot && slot != NULL) {
334 HEIMDAL_MUTEX_destroy(&slot->mutex);
335 free(slot);
337 return ret;
340 krb5_error_code
341 kcm_ccache_remove_creds_internal(krb5_context context,
342 kcm_ccache ccache)
344 struct kcm_creds *k;
346 k = ccache->creds;
347 while (k != NULL) {
348 struct kcm_creds *old;
350 krb5_free_cred_contents(context, &k->cred);
351 old = k;
352 k = k->next;
353 free(old);
355 ccache->creds = NULL;
357 return 0;
360 krb5_error_code
361 kcm_ccache_remove_creds(krb5_context context,
362 kcm_ccache ccache)
364 krb5_error_code ret;
366 KCM_ASSERT_VALID(ccache);
368 HEIMDAL_MUTEX_lock(&ccache->mutex);
369 ret = kcm_ccache_remove_creds_internal(context, ccache);
370 HEIMDAL_MUTEX_unlock(&ccache->mutex);
372 return ret;
375 krb5_error_code
376 kcm_zero_ccache_data_internal(krb5_context context,
377 kcm_ccache_data *cache)
379 if (cache->client != NULL) {
380 krb5_free_principal(context, cache->client);
381 cache->client = NULL;
384 if (cache->server != NULL) {
385 krb5_free_principal(context, cache->server);
386 cache->server = NULL;
389 kcm_ccache_remove_creds_internal(context, cache);
391 return 0;
394 krb5_error_code
395 kcm_zero_ccache_data(krb5_context context,
396 kcm_ccache cache)
398 krb5_error_code ret;
400 KCM_ASSERT_VALID(cache);
402 HEIMDAL_MUTEX_lock(&cache->mutex);
403 ret = kcm_zero_ccache_data_internal(context, cache);
404 HEIMDAL_MUTEX_unlock(&cache->mutex);
406 return ret;
409 krb5_error_code
410 kcm_retain_ccache(krb5_context context,
411 kcm_ccache ccache)
413 KCM_ASSERT_VALID(ccache);
415 HEIMDAL_MUTEX_lock(&ccache->mutex);
416 ccache->refcnt++;
417 HEIMDAL_MUTEX_unlock(&ccache->mutex);
419 return 0;
422 krb5_error_code
423 kcm_release_ccache(krb5_context context, kcm_ccache c)
425 krb5_error_code ret = 0;
427 KCM_ASSERT_VALID(c);
429 HEIMDAL_MUTEX_lock(&c->mutex);
430 if (c->refcnt == 1) {
431 kcm_free_ccache_data_internal(context, c);
432 free(c);
433 } else {
434 c->refcnt--;
435 HEIMDAL_MUTEX_unlock(&c->mutex);
438 return ret;
441 krb5_error_code
442 kcm_ccache_gen_new(krb5_context context,
443 pid_t pid,
444 uid_t uid,
445 gid_t gid,
446 kcm_ccache *ccache)
448 krb5_error_code ret;
449 char *name;
451 name = kcm_ccache_nextid(pid, uid, gid);
452 if (name == NULL) {
453 return KRB5_CC_NOMEM;
456 ret = kcm_ccache_new(context, name, ccache);
458 free(name);
459 return ret;
462 krb5_error_code
463 kcm_ccache_new(krb5_context context,
464 const char *name,
465 kcm_ccache *ccache)
467 krb5_error_code ret;
469 ret = kcm_ccache_alloc(context, name, ccache);
470 if (ret == 0) {
472 * one reference is held by the linked list,
473 * one by the caller
475 kcm_retain_ccache(context, *ccache);
478 return ret;
481 krb5_error_code
482 kcm_ccache_destroy_if_empty(krb5_context context,
483 kcm_ccache ccache)
485 krb5_error_code ret;
487 KCM_ASSERT_VALID(ccache);
489 if (ccache->creds == NULL) {
490 ret = kcm_ccache_destroy(context, ccache->name);
491 } else
492 ret = 0;
494 return ret;
497 krb5_error_code
498 kcm_ccache_store_cred(krb5_context context,
499 kcm_ccache ccache,
500 krb5_creds *creds,
501 int copy)
503 krb5_error_code ret;
504 krb5_creds *tmp;
506 KCM_ASSERT_VALID(ccache);
508 HEIMDAL_MUTEX_lock(&ccache->mutex);
509 ret = kcm_ccache_store_cred_internal(context, ccache, creds, copy, &tmp);
510 HEIMDAL_MUTEX_unlock(&ccache->mutex);
512 return ret;
515 struct kcm_creds *
516 kcm_ccache_find_cred_uuid(krb5_context context,
517 kcm_ccache ccache,
518 kcmuuid_t uuid)
520 struct kcm_creds *c;
522 for (c = ccache->creds; c != NULL; c = c->next)
523 if (memcmp(c->uuid, uuid, sizeof(c->uuid)) == 0)
524 return c;
526 return NULL;
531 krb5_error_code
532 kcm_ccache_store_cred_internal(krb5_context context,
533 kcm_ccache ccache,
534 krb5_creds *creds,
535 int copy,
536 krb5_creds **credp)
538 struct kcm_creds **c;
539 krb5_error_code ret;
541 for (c = &ccache->creds; *c != NULL; c = &(*c)->next)
544 *c = (struct kcm_creds *)calloc(1, sizeof(**c));
545 if (*c == NULL)
546 return KRB5_CC_NOMEM;
548 RAND_bytes((*c)->uuid, sizeof((*c)->uuid));
550 *credp = &(*c)->cred;
552 if (copy) {
553 ret = krb5_copy_creds_contents(context, creds, *credp);
554 if (ret) {
555 free(*c);
556 *c = NULL;
558 } else {
559 **credp = *creds;
560 ret = 0;
563 return ret;
566 krb5_error_code
567 kcm_ccache_remove_cred_internal(krb5_context context,
568 kcm_ccache ccache,
569 krb5_flags whichfields,
570 const krb5_creds *mcreds)
572 krb5_error_code ret;
573 struct kcm_creds **c;
575 ret = KRB5_CC_NOTFOUND;
577 for (c = &ccache->creds; *c != NULL; c = &(*c)->next) {
578 if (krb5_compare_creds(context, whichfields, mcreds, &(*c)->cred)) {
579 struct kcm_creds *cred = *c;
581 *c = cred->next;
582 krb5_free_cred_contents(context, &cred->cred);
583 free(cred);
584 ret = 0;
585 if (*c == NULL)
586 break;
590 return ret;
593 krb5_error_code
594 kcm_ccache_remove_cred(krb5_context context,
595 kcm_ccache ccache,
596 krb5_flags whichfields,
597 const krb5_creds *mcreds)
599 krb5_error_code ret;
601 KCM_ASSERT_VALID(ccache);
603 HEIMDAL_MUTEX_lock(&ccache->mutex);
604 ret = kcm_ccache_remove_cred_internal(context, ccache, whichfields, mcreds);
605 HEIMDAL_MUTEX_unlock(&ccache->mutex);
607 return ret;
610 krb5_error_code
611 kcm_ccache_retrieve_cred_internal(krb5_context context,
612 kcm_ccache ccache,
613 krb5_flags whichfields,
614 const krb5_creds *mcreds,
615 krb5_creds **creds)
617 krb5_boolean match;
618 struct kcm_creds *c;
619 krb5_error_code ret;
621 memset(creds, 0, sizeof(*creds));
623 ret = KRB5_CC_END;
625 match = FALSE;
626 for (c = ccache->creds; c != NULL; c = c->next) {
627 match = krb5_compare_creds(context, whichfields, mcreds, &c->cred);
628 if (match)
629 break;
632 if (match) {
633 ret = 0;
634 *creds = &c->cred;
637 return ret;
640 krb5_error_code
641 kcm_ccache_retrieve_cred(krb5_context context,
642 kcm_ccache ccache,
643 krb5_flags whichfields,
644 const krb5_creds *mcreds,
645 krb5_creds **credp)
647 krb5_error_code ret;
649 KCM_ASSERT_VALID(ccache);
651 HEIMDAL_MUTEX_lock(&ccache->mutex);
652 ret = kcm_ccache_retrieve_cred_internal(context, ccache,
653 whichfields, mcreds, credp);
654 HEIMDAL_MUTEX_unlock(&ccache->mutex);
656 return ret;
659 char *
660 kcm_ccache_first_name(kcm_client *client)
662 kcm_ccache p;
663 char *name = NULL;
665 HEIMDAL_MUTEX_lock(&ccache_mutex);
667 for (p = ccache_head; p != NULL; p = p->next) {
668 if (kcm_is_same_session(client, p->uid, p->session))
669 break;
671 if (p)
672 name = strdup(p->name);
673 HEIMDAL_MUTEX_unlock(&ccache_mutex);
674 return name;