Various cleanups
[mech_eap.git] / util_krb.c
1 /*
2  * Copyright (c) 2010, JANET(UK)
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  *
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * 3. Neither the name of JANET(UK) nor the names of its contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
21  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  */
32
33 #include "gssapiP_eap.h"
34
35 static GSSEAP_THREAD_ONCE krbContextKeyOnce = GSSEAP_ONCE_INITIALIZER;
36 static GSSEAP_THREAD_KEY krbContextKey;
37
38 static void
39 destroyKrbContext(void *arg)
40 {
41     krb5_context context = (krb5_context)arg;
42
43     if (context != NULL)
44         krb5_free_context(context);
45 }
46
47 static void
48 createKrbContextKey(void)
49 {
50     GSSEAP_KEY_CREATE(&krbContextKey, destroyKrbContext);
51 }
52
53 OM_uint32
54 gssEapKerberosInit(OM_uint32 *minor, krb5_context *context)
55 {
56     *minor = 0;
57
58     GSSEAP_ONCE(&krbContextKeyOnce, createKrbContextKey);
59
60     *context = GSSEAP_GETSPECIFIC(krbContextKey);
61     if (*context == NULL) {
62         *minor = krb5_init_context(context);
63         if (*minor == 0) {
64             if (GSSEAP_SETSPECIFIC(krbContextKey, *context) != 0) {
65                 *minor = errno;
66                 krb5_free_context(*context);
67                 *context = NULL;
68             }
69         }
70     }
71
72     return *minor == 0 ? GSS_S_COMPLETE : GSS_S_FAILURE;
73 }
74
75 /*
76  * Derive a key for RFC 4121 use by using the following
77  * derivation function:
78  *
79  *    random-to-key(prf(random-to-key([e]msk), "rfc4121-gss-eap"))
80  *
81  * where random-to-key and prf are defined in RFC 3961.
82  */
83 OM_uint32
84 gssEapDeriveRFC3961Key(OM_uint32 *minor,
85                        gss_buffer_t msk,
86                        krb5_enctype enctype,
87                        krb5_keyblock *pKey)
88 {
89     krb5_context context;
90     krb5_data data, prf;
91     krb5_keyblock kd;
92     krb5_error_code code;
93     size_t keybytes, keylength, prflength;
94
95     memset(pKey, 0, sizeof(*pKey));
96
97     GSSEAP_KRB_INIT(&context);
98
99     kd.contents = NULL;
100     kd.length = 0;
101     KRB_KEYTYPE(&kd) = enctype;
102
103     prf.data = NULL;
104     prf.length = 0;
105
106     code = krb5_c_keylengths(context, enctype, &keybytes, &keylength);
107     if (code != 0)
108         goto cleanup;
109
110     data.length = msk->length;
111     data.data = (char *)msk->value;
112
113     kd.contents = GSSEAP_MALLOC(keylength);
114     if (kd.contents == NULL) {
115         code = ENOMEM;
116         goto cleanup;
117     }
118     kd.length = keylength;
119
120     /* Convert MSK into a Kerberos key */
121     code = krb5_c_random_to_key(context, enctype, &data, &kd);
122     if (code != 0)
123         goto cleanup;
124
125     data.length = sizeof("rfc4121-gss-eap") - 1;
126     data.data = "rfc4121-gss-eap";
127
128     /* Plug derivation constant and key into PRF */
129     code = krb5_c_prf_length(context, enctype, &prflength);
130     if (code != 0)
131         goto cleanup;
132
133     prf.length = prflength;
134     prf.data = GSSEAP_MALLOC(prflength);
135     if (data.data == NULL) {
136         code = ENOMEM;
137         goto cleanup;
138     }
139
140     code = krb5_c_prf(context, &kd, &data, &prf);
141     if (code != 0)
142         goto cleanup;
143
144     /* Finally, convert PRF output into a new key which we will return */
145     code = krb5_c_random_to_key(context, enctype, &prf, &kd);
146     if (code != 0)
147         goto cleanup;
148
149     *pKey = kd;
150     kd.contents = NULL;
151
152 cleanup:
153     if (kd.contents != NULL) {
154         memset(kd.contents, 0, kd.length);
155         GSSEAP_FREE(kd.contents);
156     }
157     if (prf.data != NULL) {
158         memset(prf.data, 0, prf.length);
159         GSSEAP_FREE(prf.data);
160     }
161
162     *minor = code;
163     return (code == 0) ? GSS_S_COMPLETE : GSS_S_FAILURE;
164 }