More work on SAML code
[mech_eap.git] / util.h
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  * Portions Copyright 2003-2010 Massachusetts Institute of Technology.
34  * All Rights Reserved.
35  *
36  * Export of this software from the United States of America may
37  *   require a specific license from the United States Government.
38  *   It is the responsibility of any person or organization contemplating
39  *   export to obtain such a license before exporting.
40  *
41  * WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
42  * distribute this software and its documentation for any purpose and
43  * without fee is hereby granted, provided that the above copyright
44  * notice appear in all copies and that both that copyright notice and
45  * this permission notice appear in supporting documentation, and that
46  * the name of M.I.T. not be used in advertising or publicity pertaining
47  * to distribution of the software without specific, written prior
48  * permission.  Furthermore if you modify this software you must label
49  * your software as modified software and not distribute it in such a
50  * fashion that it might be confused with the original M.I.T. software.
51  * M.I.T. makes no representations about the suitability of
52  * this software for any purpose.  It is provided "as is" without express
53  * or implied warranty.
54  *
55  */
56
57 #ifndef _UTIL_H_
58 #define _UTIL_H_ 1
59
60 #include <string.h>
61 #include <errno.h>
62
63 #include <krb5.h>
64
65 #ifdef __cplusplus
66 extern "C" {
67 #endif
68
69 #ifndef MIN             /* Usually found in <sys/param.h>. */
70 #define MIN(_a,_b)  ((_a)<(_b)?(_a):(_b))
71 #endif
72
73 #define KRB_KEY_TYPE(key)       ((key)->enctype)
74 #define KRB_KEY_DATA(key)       ((key)->contents)
75 #define KRB_KEY_LENGTH(key)     ((key)->length)
76 #define KRB_KEY_INIT(key)       do {        \
77         KRB_KEY_TYPE(key) = ENCTYPE_NULL;   \
78         KRB_KEY_DATA(key) = NULL;           \
79         KRB_KEY_LENGTH(key) = 0;            \
80     } while (0)
81
82 enum gss_eap_token_type {
83     TOK_TYPE_NONE                    = 0x0000,  /* no token */
84     TOK_TYPE_MIC                     = 0x0404,  /* RFC 4121 MIC token */
85     TOK_TYPE_WRAP                    = 0x0504,  /* RFC 4121 wrap token */
86     TOK_TYPE_EXPORT_NAME             = 0x0401,  /* RFC 2743 exported name */
87     TOK_TYPE_EXPORT_NAME_COMPOSITE   = 0x0402,  /* draft-ietf-kitten-gss-naming */
88     TOK_TYPE_DELETE_CONTEXT          = 0x0405,  /* RFC 2743 delete context */
89     TOK_TYPE_EAP_RESP                = 0x0601,  /* draft-howlett-eap-gss */
90     TOK_TYPE_EAP_REQ                 = 0x0602,  /* draft-howlett-eap-gss */
91     TOK_TYPE_GSS_CB                  = 0x0603,  /* draft-howlett-eap-gss */
92 };
93
94 #define EAP_EXPORT_CONTEXT_V1           1
95
96 /* util_buffer.c */
97 OM_uint32
98 makeStringBuffer(OM_uint32 *minor,
99                  const char *string,
100                  gss_buffer_t buffer);
101
102 OM_uint32
103 bufferToString(OM_uint32 *minor,
104                const gss_buffer_t buffer,
105                char **pString);
106
107 OM_uint32
108 duplicateBuffer(OM_uint32 *minor,
109                 const gss_buffer_t src,
110                 gss_buffer_t dst);
111
112 static inline int
113 bufferEqual(const gss_buffer_t b1, const gss_buffer_t b2)
114 {
115     return (b1->length == b2->length &&
116             memcmp(b1->value, b2->value, b2->length) == 0);
117 }
118
119 static inline int
120 bufferEqualString(const gss_buffer_t b1, const char *s)
121 {
122     gss_buffer_desc b2;
123
124     b2.length = strlen(s);
125     b2.value = (char *)s;
126
127     return bufferEqual(b1, &b2);
128 }
129
130 /* util_cksum.c */
131 int
132 gssEapSign(krb5_context context,
133            krb5_cksumtype type,
134            size_t rrc,
135            krb5_keyblock *key,
136            krb5_keyusage sign_usage,
137            gss_iov_buffer_desc *iov,
138            int iov_count);
139
140 int
141 gssEapVerify(krb5_context context,
142              krb5_cksumtype type,
143              size_t rrc,  
144              krb5_keyblock *key,
145              krb5_keyusage sign_usage,
146              gss_iov_buffer_desc *iov,
147              int iov_count,
148              int *valid);
149
150 #if 0
151 OM_uint32
152 gssEapEncodeGssChannelBindings(OM_uint32 *minor,
153                                gss_channel_bindings_t chanBindings,
154                                gss_buffer_t encodedBindings);
155 #endif
156
157 /* util_context.c */
158 OM_uint32 gssEapAllocContext(OM_uint32 *minor, gss_ctx_id_t *pCtx);
159 OM_uint32 gssEapReleaseContext(OM_uint32 *minor, gss_ctx_id_t *pCtx);
160
161 OM_uint32
162 gssEapMakeToken(OM_uint32 *minor,
163                 gss_ctx_id_t ctx,
164                 const gss_buffer_t innerToken,
165                 enum gss_eap_token_type tokenType,
166                 gss_buffer_t outputToken);
167
168 OM_uint32
169 gssEapVerifyToken(OM_uint32 *minor,
170                   gss_ctx_id_t ctx,
171                   const gss_buffer_t inputToken,
172                   enum gss_eap_token_type tokenType,
173                   gss_buffer_t innerInputToken);
174
175 /* util_cred.c */
176 OM_uint32 gssEapAllocCred(OM_uint32 *minor, gss_cred_id_t *pCred);
177 OM_uint32 gssEapReleaseCred(OM_uint32 *minor, gss_cred_id_t *pCred);
178
179 OM_uint32
180 gssEapAcquireCred(OM_uint32 *minor,
181                   const gss_name_t desiredName,
182                   const gss_buffer_t password,
183                   OM_uint32 timeReq,
184                   const gss_OID_set desiredMechs,
185                   int cred_usage,
186                   gss_cred_id_t *pCred,
187                   gss_OID_set *pActualMechs,
188                   OM_uint32 *timeRec);
189
190 int gssEapCredAvailable(gss_cred_id_t cred, gss_OID mech);
191
192 /* util_crypt.c */
193 int
194 gssEapEncrypt(krb5_context context, int dce_style, size_t ec,
195               size_t rrc, krb5_keyblock *key, int usage, krb5_pointer iv,
196               gss_iov_buffer_desc *iov, int iov_count);
197
198 int
199 gssEapDecrypt(krb5_context context, int dce_style, size_t ec,
200               size_t rrc, krb5_keyblock *key, int usage, krb5_pointer iv,
201               gss_iov_buffer_desc *iov, int iov_count);
202
203 krb5_cryptotype
204 gssEapMapCryptoFlag(OM_uint32 type);
205
206 gss_iov_buffer_t
207 gssEapLocateIov(gss_iov_buffer_desc *iov,
208                 int iov_count,
209                 OM_uint32 type);
210
211 void
212 gssEapIovMessageLength(gss_iov_buffer_desc *iov,
213                        int iov_count,
214                        size_t *data_length,
215                        size_t *assoc_data_length);
216
217 void
218 gssEapReleaseIov(gss_iov_buffer_desc *iov, int iov_count);
219
220 int
221 gssEapIsIntegrityOnly(gss_iov_buffer_desc *iov, int iov_count);
222
223 int
224 gssEapAllocIov(gss_iov_buffer_t iov, size_t size);
225
226 OM_uint32
227 gssEapDeriveRfc3961Key(OM_uint32 *minor,
228                        const unsigned char *key,
229                        size_t keyLength,
230                        krb5_enctype enctype,
231                        krb5_keyblock *pKey);
232
233 /* util_krb.c */
234 OM_uint32
235 gssEapKerberosInit(OM_uint32 *minor, krb5_context *context);
236
237 OM_uint32
238 rfc3961ChecksumTypeForKey(OM_uint32 *minor,
239                           krb5_keyblock *key,
240                           krb5_cksumtype *cksumtype);
241
242 #define GSSEAP_KRB_INIT(ctx) do {                   \
243         OM_uint32 tmpMajor;                         \
244         tmpMajor  = gssEapKerberosInit(minor, ctx); \
245         if (GSS_ERROR(tmpMajor)) {                  \
246             return tmpMajor;                        \
247         }                                           \
248     } while (0)
249
250 /* util_mech.c */
251 int
252 gssEapInternalizeOid(const gss_OID oid,
253                      gss_OID *const pInternalizedOid);
254
255 OM_uint32
256 gssEapDefaultMech(OM_uint32 *minor,
257                   gss_OID *oid);
258
259 OM_uint32
260 gssEapIndicateMechs(OM_uint32 *minor,
261                     gss_OID_set *mechs);
262
263 OM_uint32
264 gssEapEnctypeToOid(OM_uint32 *minor,
265                    krb5_enctype enctype,
266                    gss_OID *pOid);
267
268 OM_uint32
269 gssEapOidToEnctype(OM_uint32 *minor,
270                    const gss_OID oid,
271                    krb5_enctype *enctype);
272
273 int
274 gssEapIsMechanismOid(const gss_OID oid);
275
276 int
277 gssEapIsConcreteMechanismOid(const gss_OID oid);
278
279 OM_uint32
280 gssEapValidateMechs(OM_uint32 *minor,
281                    const gss_OID_set mechs);
282
283 /* util_name.c */
284 enum gss_eap_attribute_type {
285     ATTR_TYPE_NONE                  = 0,
286     ATTR_TYPE_SAML_AAA_ASSERTION    = 1,
287     ATTR_TYPE_SAML_ATTR             = 2,
288     ATTR_TYPE_RADIUS_AVP            = 3
289 };
290
291 typedef OM_uint32 (*gss_eap_add_attr_cb)(OM_uint32 *minor,
292                                          gss_name_t name,
293                                          gss_buffer_t attribute,
294                                          void *data);
295
296 OM_uint32 gssEapAllocName(OM_uint32 *minor, gss_name_t *pName);
297 OM_uint32 gssEapReleaseName(OM_uint32 *minor, gss_name_t *pName);
298 OM_uint32 gssEapExportName(OM_uint32 *minor,
299                            const gss_name_t name,
300                            gss_buffer_t exportedName,
301                            int composite);
302 OM_uint32 gssEapImportName(OM_uint32 *minor,
303                            const gss_buffer_t input_name_buffer,
304                            gss_OID input_name_type,
305                            gss_name_t *output_name);
306
307 enum gss_eap_attribute_type
308 gssEapAttributePrefixToType(const gss_buffer_t prefix);
309 gss_buffer_t
310 gssEapAttributeTypeToPrefix(enum gss_eap_attribute_type type);
311 OM_uint32
312 decomposeAttributeName(OM_uint32 *minor,
313                        const gss_buffer_t attribute,
314                        gss_buffer_t prefix,
315                        gss_buffer_t suffix);
316 OM_uint32
317 composeAttributeName(OM_uint32 *minor,
318                      const gss_buffer_t prefix,
319                      const gss_buffer_t suffix,
320                      gss_buffer_t attribute);
321
322 /* util_oid.c */
323 OM_uint32
324 composeOid(OM_uint32 *minor_status,
325            const char *prefix,
326            size_t prefix_len,
327            int suffix,  
328            gss_OID_desc *oid);
329
330 OM_uint32
331 decomposeOid(OM_uint32 *minor_status,
332              const char *prefix,
333              size_t prefix_len,
334              gss_OID_desc *oid,
335              int *suffix) ;
336
337 OM_uint32
338 duplicateOid(OM_uint32 *minor_status,
339              const gss_OID_desc * const oid,
340              gss_OID *new_oid);
341
342 OM_uint32
343 duplicateOidSet(OM_uint32 *minor,
344                 const gss_OID_set src,
345                 gss_OID_set *dst);
346
347 static inline int
348 oidEqual(const gss_OID_desc *o1, const gss_OID_desc *o2)
349 {
350     if (o1 == GSS_C_NO_OID)
351         return (o2 == GSS_C_NO_OID);
352     else if (o2 == GSS_C_NO_OID)
353         return (o1 == GSS_C_NO_OID);
354     else
355         return (o1->length == o2->length &&
356                 memcmp(o1->elements, o2->elements, o1->length) == 0);
357 }
358
359 /* util_ordering.c */
360 OM_uint32
361 sequenceInternalize(OM_uint32 *minor,
362                     void **vqueue,
363                     unsigned char **buf,
364                     size_t *lenremain);
365
366 OM_uint32
367 sequenceExternalize(OM_uint32 *minor,
368                     void *vqueue,
369                     unsigned char **buf,
370                     size_t *lenremain);
371
372 size_t
373 sequenceSize(void *vqueue);
374
375 OM_uint32
376 sequenceFree(OM_uint32 *minor, void **vqueue);
377
378 OM_uint32
379 sequenceCheck(OM_uint32 *minor, void **vqueue, uint64_t seqnum);
380
381 OM_uint32
382 sequenceInit(OM_uint32 *minor, void **vqueue, uint64_t seqnum,
383              int do_replay, int do_sequence, int wide_nums);
384
385 /* util_token.c */
386 size_t
387 tokenSize(const gss_OID_desc *mech, size_t body_size);
388
389 void
390 makeTokenHeader(const gss_OID_desc *mech,
391                 size_t body_size,
392                 unsigned char **buf,
393                 enum gss_eap_token_type tok_type);
394
395 OM_uint32
396 verifyTokenHeader(OM_uint32 *minor,
397                   gss_OID mech,
398                   size_t *body_size,
399                   unsigned char **buf_in,
400                   size_t toksize_in,
401                   enum gss_eap_token_type tok_type);
402
403 /* Helper macros */
404 #define GSSEAP_CALLOC(count, size)      (calloc((count), (size)))
405 #define GSSEAP_FREE(ptr)                (free((ptr)))
406 #define GSSEAP_MALLOC(size)             (malloc((size)))
407 #define GSSEAP_REALLOC(ptr, size)       (realloc((ptr), (size)))
408
409 #define GSSEAP_NOT_IMPLEMENTED          do {            \
410         assert(0 && "not implemented");                 \
411         *minor = ENOSYS;                                \
412         return GSS_S_FAILURE;                           \
413     } while (0)
414
415 #include <pthread.h>
416
417 #define GSSEAP_MUTEX                    pthread_mutex_t
418 #define GSSEAP_MUTEX_INITIALIZER        PTHREAD_MUTEX_INITIALIZER
419
420 #define GSSEAP_MUTEX_INIT(m)            pthread_mutex_init((m), NULL)
421 #define GSSEAP_MUTEX_DESTROY(m)         pthread_mutex_destroy((m))
422 #define GSSEAP_MUTEX_LOCK(m)            pthread_mutex_lock((m))
423 #define GSSEAP_MUTEX_UNLOCK(m)          pthread_mutex_unlock((m))
424
425 #define GSSEAP_THREAD_KEY               pthread_key_t
426 #define GSSEAP_KEY_CREATE(k, d)         pthread_key_create((k), (d))
427 #define GSSEAP_GETSPECIFIC(k)           pthread_getspecific((k))
428 #define GSSEAP_SETSPECIFIC(k, d)        pthread_setspecific((k), (d))
429
430 #define GSSEAP_THREAD_ONCE              pthread_once_t
431 #define GSSEAP_ONCE(o, i)               pthread_once((o), (i))
432 #define GSSEAP_ONCE_INITIALIZER         PTHREAD_ONCE_INIT
433
434 #include "util_radius.h"
435 #include "util_saml.h"
436
437 /* Helper functions */
438 static inline void
439 store_uint16_be(uint16_t val, void *vp)
440 {
441     unsigned char *p = (unsigned char *)vp;
442
443     p[0] = (val >>  8) & 0xff;
444     p[1] = (val      ) & 0xff;
445 }
446
447 static inline uint16_t
448 load_uint16_be(const void *cvp)
449 {
450     const unsigned char *p = (const unsigned char *)cvp;
451
452     return (p[1] | (p[0] << 8));
453 }
454
455 static inline void
456 store_uint32_be(uint32_t val, void *vp)
457 {
458     unsigned char *p = (unsigned char *)vp;
459
460     p[0] = (val >> 24) & 0xff;
461     p[1] = (val >> 16) & 0xff;
462     p[2] = (val >>  8) & 0xff;
463     p[3] = (val      ) & 0xff;
464 }
465
466 static inline uint32_t
467 load_uint32_be(const void *cvp)
468 {
469     const unsigned char *p = (const unsigned char *)cvp;
470
471     return (p[3] | (p[2] << 8)
472             | ((uint32_t) p[1] << 16)
473             | ((uint32_t) p[0] << 24));
474 }
475
476 static inline void
477 store_uint64_be(uint64_t val, void *vp)
478 {
479     unsigned char *p = (unsigned char *)vp;
480
481     p[0] = (unsigned char)((val >> 56) & 0xff);
482     p[1] = (unsigned char)((val >> 48) & 0xff);
483     p[2] = (unsigned char)((val >> 40) & 0xff);
484     p[3] = (unsigned char)((val >> 32) & 0xff);
485     p[4] = (unsigned char)((val >> 24) & 0xff);
486     p[5] = (unsigned char)((val >> 16) & 0xff);
487     p[6] = (unsigned char)((val >>  8) & 0xff);
488     p[7] = (unsigned char)((val      ) & 0xff);
489 }
490
491 static inline uint64_t
492 load_uint64_be(const void *cvp)
493 {
494     const unsigned char *p = (const unsigned char *)cvp;
495
496     return ((uint64_t)load_uint32_be(p) << 32) | load_uint32_be(p + 4);
497 }
498
499 static inline unsigned char *
500 store_buffer(gss_buffer_t buffer, void *vp, int wide_nums)
501 {
502     unsigned char *p = (unsigned char *)vp;
503
504     if (wide_nums) {
505         store_uint64_be(buffer->length, p);
506         p += 8;
507     } else {
508         store_uint32_be(buffer->length, p);
509         p += 4;
510     }
511
512     if (buffer->value != NULL) {
513         memcpy(p, buffer->value, buffer->length);
514         p += buffer->length;
515     }
516
517     return p;
518 }
519
520 static inline unsigned char *
521 load_buffer(const void *cvp, size_t length, gss_buffer_t buffer)
522 {
523     buffer->length = 0;
524     buffer->value = GSSEAP_MALLOC(length);
525     if (buffer->value == NULL)
526         return NULL;
527     buffer->length = length;
528     memcpy(buffer->value, cvp, length);
529     return (unsigned char *)cvp + length;
530 }
531
532 static inline unsigned char *
533 store_oid(gss_OID oid, void *vp)
534 {
535     gss_buffer_desc buf;
536
537     if (oid != GSS_C_NO_OID) {
538         buf.length = oid->length;
539         buf.value = oid->elements;
540     } else {
541         buf.length = 0;
542         buf.value = NULL;
543     }
544
545     return store_buffer(&buf, vp, FALSE);
546 }
547
548 #ifdef __cplusplus
549 }
550 #endif
551
552 #endif /* _UTIL_H_ */