cleanup key derivation on acceptor
[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                        const unsigned char *inputKey,
86                        size_t inputKeyLength,
87                        krb5_enctype encryptionType,
88                        krb5_keyblock *pKey)
89 {
90     krb5_context krbContext;
91     krb5_data data, ns, t, prfOut;
92     krb5_keyblock kd;
93     krb5_error_code code;
94     size_t randomLength, keyLength, prfLength;
95     unsigned char constant[4 + sizeof("rfc4121-gss-eap") - 1], *p;
96     ssize_t i, remain;
97
98     assert(encryptionType != ENCTYPE_NULL);
99
100     memset(pKey, 0, sizeof(*pKey));
101
102     GSSEAP_KRB_INIT(&krbContext);
103
104     KRB_KEY_INIT(&kd);
105     KRB_KEY_TYPE(&kd) = encryptionType;
106
107     t.data = NULL;
108     t.length = 0;
109
110     prfOut.data = NULL;
111     prfOut.length = 0;
112
113     code = krb5_c_keylengths(krbContext, encryptionType,
114                              &randomLength, &keyLength);
115     if (code != 0)
116         goto cleanup;
117
118     data.length = MIN(inputKeyLength, randomLength);
119     data.data = (char *)inputKey;
120
121     KRB_KEY_DATA(&kd) = GSSEAP_MALLOC(keyLength);
122     if (KRB_KEY_DATA(&kd) == NULL) {
123         code = ENOMEM;
124         goto cleanup;
125     }
126     KRB_KEY_LENGTH(&kd) = keyLength;
127
128     /* Convert MSK into a Kerberos key */
129     code = krb5_c_random_to_key(krbContext, encryptionType, &data, &kd);
130     if (code != 0)
131         goto cleanup;
132
133     memset(&constant[0], 0, 4);
134     memcpy(&constant[4], "rfc4121-gss-eap", sizeof("rfc4121-gss-eap") - 1);
135
136     ns.length = sizeof(constant);
137     ns.data = (char *)constant;
138
139     /* Plug derivation constant and key into PRF */
140     code = krb5_c_prf_length(krbContext, encryptionType, &prfLength);
141     if (code != 0)
142         goto cleanup;
143
144     t.length = prfLength;
145     t.data = GSSEAP_MALLOC(t.length);
146     if (t.data == NULL) {
147         code = ENOMEM;
148         goto cleanup;
149     }
150
151     prfOut.length = randomLength;
152     prfOut.data = GSSEAP_MALLOC(prfOut.length);
153     if (prfOut.data == NULL) {
154         code = ENOMEM;
155         goto cleanup;
156     }
157
158     for (i = 0, p = (unsigned char *)prfOut.data, remain = randomLength;
159          remain > 0;
160          p += t.length, remain -= t.length, i++)
161     {
162         store_uint32_be(i, ns.data);
163
164         code = krb5_c_prf(krbContext, &kd, &ns, &t);
165         if (code != 0)
166             goto cleanup;
167
168         memcpy(p, t.data, MIN(t.length, remain));
169      }
170
171     /* Finally, convert PRF output into a new key which we will return */
172     code = krb5_c_random_to_key(krbContext, encryptionType, &prfOut, &kd);
173     if (code != 0)
174         goto cleanup;
175
176     *pKey = kd;
177     KRB_KEY_DATA(&kd) = NULL;
178
179 cleanup:
180     if (KRB_KEY_DATA(&kd) != NULL) {
181         memset(KRB_KEY_DATA(&kd), 0, KRB_KEY_LENGTH(&kd));
182         GSSEAP_FREE(KRB_KEY_DATA(&kd));
183     }
184     if (t.data != NULL) {
185         memset(t.data, 0, t.length);
186         GSSEAP_FREE(t.data);
187     }
188     if (prfOut.data != NULL) {
189         memset(prfOut.data, 0, prfOut.length);
190         GSSEAP_FREE(prfOut.data);
191     }
192     *minor = code;
193     return (code == 0) ? GSS_S_COMPLETE : GSS_S_FAILURE;
194 }
195
196 #ifdef HAVE_KRB5INT_C_MANDATORY_CKSUMTYPE
197 extern krb5_error_code
198 krb5int_c_mandatory_cksumtype(krb5_context, krb5_enctype, krb5_cksumtype *);
199 #endif
200
201 OM_uint32
202 rfc3961ChecksumTypeForKey(OM_uint32 *minor,
203                           krb5_keyblock *key,
204                           krb5_cksumtype *cksumtype)
205 {
206     krb5_context krbContext;
207 #ifndef HAVE_KRB5INT_C_MANDATORY_CKSUMTYPE
208     krb5_data data;
209     krb5_checksum cksum;
210 #endif
211
212     GSSEAP_KRB_INIT(&krbContext);
213
214 #ifdef HAVE_KRB5INT_C_MANDATORY_CKSUMTYPE
215     *minor = krb5int_c_mandatory_cksumtype(krbContext, KRB_KEY_TYPE(key),
216                                            cksumtype);
217     if (*minor != 0)
218         return GSS_S_FAILURE;
219 #else
220     data.length = 0;
221     data.data = NULL;
222
223     memset(&cksum, 0, sizeof(cksum));
224
225     /*
226      * This is a complete hack but it's the only way to work with
227      * MIT Kerberos pre-1.9 without using private API, as it does
228      * not support passing in zero as the checksum type.
229      */
230     *minor = krb5_c_make_checksum(krbContext, 0, key, 0, &data, &cksum);
231     if (*minor != 0)
232         return GSS_S_FAILURE;
233
234     *cksumtype = cksum.checksum_type;
235
236     krb5_free_checksum_contents(krbContext, &cksum);
237 #endif /* HAVE_KRB5INT_C_MANDATORY_CKSUMTYPE */
238
239     return GSS_S_COMPLETE;
240 }