More work on SAML code
[mech_eap.git] / inquire_name.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 struct gss_eap_attribute_args {
36     enum gss_eap_attribute_type type;
37     gss_buffer_set_t attrs;
38 };
39
40 /*
41  * The purpose of this callback interface is to not expose the attribute
42  * prefixes to the attribute providers themselves.
43  */
44 static OM_uint32
45 addAttribute(OM_uint32 *minor,
46              gss_name_t name,
47              gss_buffer_t attribute,
48              void *data)
49 {
50     struct gss_eap_attribute_args *args = (struct gss_eap_attribute_args *)data;
51     OM_uint32 major, tmpMinor;
52     gss_buffer_desc qualifiedAttr;
53     gss_buffer_t prefix;
54
55     if (args->type != ATTR_TYPE_NONE)
56         prefix = gssEapAttributeTypeToPrefix(args->type);
57     else
58         prefix = GSS_C_NO_BUFFER;
59
60     if (prefix != GSS_C_NO_BUFFER && attribute != GSS_C_NO_BUFFER) {
61         major = composeAttributeName(minor, prefix, attribute, &qualifiedAttr);
62         if (GSS_ERROR(major))
63             return major;
64         major = gss_add_buffer_set_member(minor, &qualifiedAttr, &args->attrs);
65
66         gss_release_buffer(&tmpMinor, &qualifiedAttr);
67     } else {
68         assert(prefix != GSS_C_NO_BUFFER);
69         major = gss_add_buffer_set_member(minor, prefix, &args->attrs);
70     }
71
72     return major;
73 }
74
75 OM_uint32 gss_inquire_name(OM_uint32 *minor,
76                            gss_name_t name,
77                            int *name_is_MN,
78                            gss_OID *MN_mech,
79                            gss_buffer_set_t *attrs)
80 {
81     OM_uint32 major, tmpMinor;
82     krb5_context krbContext;
83     struct gss_eap_attribute_args args;
84
85     *name_is_MN = 1;
86     *MN_mech = GSS_EAP_MECHANISM;
87     *attrs = GSS_C_NO_BUFFER_SET;
88
89     if (name == GSS_C_NO_NAME) {
90         *minor = EINVAL;
91         return GSS_S_CALL_INACCESSIBLE_READ | GSS_S_BAD_NAME;
92     }
93
94     GSSEAP_KRB_INIT(&krbContext);
95     GSSEAP_MUTEX_LOCK(&name->mutex);
96
97     major = gss_create_empty_buffer_set(minor, attrs);
98     if (GSS_ERROR(major))
99         goto cleanup;
100
101     args.attrs = *attrs;
102
103     if (name->flags & NAME_FLAG_SAML_ATTRIBUTES) {
104         /* The assertion itself */
105         args.type = ATTR_TYPE_SAML_AAA_ASSERTION;
106
107         major = addAttribute(minor, name, GSS_C_NO_BUFFER, &args);
108         if (GSS_ERROR(major))
109             goto cleanup;
110
111         /* Raw SAML attributes */
112 #if 0
113         args.type = ATTR_TYPE_SAML_ATTR;
114         major = samlGetAttributeTypes(minor, args.type,
115                                       name, addAttribute, &args);
116         if (GSS_ERROR(major))
117             goto cleanup;
118 #endif
119
120         /* Cooked local attributes */
121         args.type = ATTR_TYPE_NONE;
122         major = samlGetAttributeTypes(minor, name, args.type,
123                                       addAttribute, &args);
124         if (GSS_ERROR(major))
125             goto cleanup;
126     }
127
128     if (name->flags & NAME_FLAG_RADIUS_ATTRIBUTES) {
129         /* Raw RADIUS attributes */
130         args.type = ATTR_TYPE_RADIUS_AVP;
131         major = radiusGetAttributeTypes(minor, name,
132                                         addAttribute, &args);
133         if (GSS_ERROR(major))
134             goto cleanup;
135     }
136
137 cleanup:
138     GSSEAP_MUTEX_UNLOCK(&name->mutex);
139
140     if (GSS_ERROR(major))
141         gss_release_buffer_set(&tmpMinor, attrs);
142
143     return major;
144 }