source4/heimdal/lib/gssapi/init.c

   1 /*
   2  * Copyright (c) 1997 - 2001, 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 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 "gssapi_locl.h"
  35
  36 RCSID("$Id: init.c,v 1.7 2003/07/22 19:50:11 lha Exp $");
  37
  38 #ifdef _SAMBA_BUILD_
  39 #include "auth/kerberos/krb5_init_context.h"
  40 #endif
  41
  42 static HEIMDAL_MUTEX gssapi_krb5_context_mutex = HEIMDAL_MUTEX_INITIALIZER;
  43 static int created_key;
  44 static HEIMDAL_thread_key gssapi_context_key;
  45
  46 static void
  47 gssapi_destroy_thread_context(void *ptr)
  48 {
  49     struct gssapi_thr_context *ctx = ptr;
  50
  51     if (ctx == NULL)
  52         return;
  53     if (ctx->error_string)
  54         free(ctx->error_string);
  55     HEIMDAL_MUTEX_destroy(&ctx->mutex);
  56     free(ctx);
  57 }
  58
  59
  60 struct gssapi_thr_context *
  61 gssapi_get_thread_context(int createp)
  62 {
  63     struct gssapi_thr_context *ctx;
  64     int ret;
  65
  66     HEIMDAL_MUTEX_lock(&gssapi_krb5_context_mutex);
  67
  68     if (!created_key)
  69         abort();
  70     ctx = HEIMDAL_getspecific(gssapi_context_key);
  71     if (ctx == NULL) {
  72         if (!createp)
  73             goto fail;
  74         ctx = malloc(sizeof(*ctx));
  75         if (ctx == NULL)
  76             goto fail;
  77         ctx->error_string = NULL;
  78         HEIMDAL_MUTEX_init(&ctx->mutex);
  79         HEIMDAL_setspecific(gssapi_context_key, ctx, ret);
  80         if (ret)
  81             goto fail;
  82     }
  83     HEIMDAL_MUTEX_unlock(&gssapi_krb5_context_mutex);
  84     return ctx;
  85  fail:
  86     HEIMDAL_MUTEX_unlock(&gssapi_krb5_context_mutex);
  87     if (ctx)
  88         free(ctx);
  89     return NULL;
  90 }
  91
  92 #ifdef _SAMBA_BUILD_
  93 /* Init krb5 with an event context.  Disgusting Samba-specific hack */
  94
  95 krb5_error_code
  96 gssapi_krb5_init_ev (void *event_context)
  97 {
  98     static struct smb_krb5_context *smb_krb5_context;
  99     krb5_error_code ret = 0;
 100
 101     HEIMDAL_MUTEX_lock(&gssapi_krb5_context_mutex);
 102
 103     if(smb_krb5_context == NULL) {
 104         ret = smb_krb5_init_context(event_context, &smb_krb5_context);
 105     }
 106     if (ret == 0 && !created_key) {
 107         HEIMDAL_key_create(&gssapi_context_key,
 108                            gssapi_destroy_thread_context,
 109                            ret);
 110         if (ret) {
 111             smb_krb5_free_context(smb_krb5_context);
 112             smb_krb5_context = NULL;
 113         } else
 114             created_key = 1;
 115     }
 116     if (ret == 0) {
 117         gssapi_krb5_context = smb_krb5_context->krb5_context;
 118     }
 119
 120     HEIMDAL_MUTEX_unlock(&gssapi_krb5_context_mutex);
 121     return ret;
 122 }
 123 #endif
 124
 125 krb5_error_code
 126 gssapi_krb5_init (void)
 127 {
 128     krb5_error_code ret = 0;
 129 #ifdef _SAMBA_BUILD_
 130     ret = gssapi_krb5_init_ev(NULL);
 131 #else
 132     HEIMDAL_MUTEX_lock(&gssapi_krb5_context_mutex);
 133
 134     if(gssapi_krb5_context == NULL) {
 135         ret = krb5_init_context (&gssapi_krb5_context);
 136     }
 137     if (ret == 0 && !created_key) {
 138         HEIMDAL_key_create(&gssapi_context_key,
 139                            gssapi_destroy_thread_context,
 140                            ret);
 141         if (ret) {
 142             krb5_free_context(gssapi_krb5_context);
 143             gssapi_krb5_context = NULL;
 144         } else
 145             created_key = 1;
 146     }
 147
 148     HEIMDAL_MUTEX_unlock(&gssapi_krb5_context_mutex);
 149 #endif
 150     return ret;
 151 }