1/*	$NetBSD: context.c,v 1.2 2017/01/28 21:31:46 christos Exp $	*/
2
3#include "mech_locl.h"
4#include "heim_threads.h"
5
6struct mg_thread_ctx {
7    gss_OID mech;
8    OM_uint32 maj_stat;
9    OM_uint32 min_stat;
10    gss_buffer_desc maj_error;
11    gss_buffer_desc min_error;
12};
13
14static HEIMDAL_MUTEX context_mutex = HEIMDAL_MUTEX_INITIALIZER;
15static int created_key;
16static HEIMDAL_thread_key context_key;
17
18
19static void
20destroy_context(void *ptr)
21{
22    struct mg_thread_ctx *mg = ptr;
23    OM_uint32 junk;
24
25    if (mg == NULL)
26	return;
27
28    gss_release_buffer(&junk, &mg->maj_error);
29    gss_release_buffer(&junk, &mg->min_error);
30    free(mg);
31}
32
33
34static struct mg_thread_ctx *
35_gss_mechglue_thread(void)
36{
37    struct mg_thread_ctx *ctx;
38    int ret = 0;
39
40    HEIMDAL_MUTEX_lock(&context_mutex);
41
42    if (!created_key) {
43	HEIMDAL_key_create(&context_key, destroy_context, ret);
44	if (ret) {
45	    HEIMDAL_MUTEX_unlock(&context_mutex);
46	    return NULL;
47	}
48	created_key = 1;
49    }
50    HEIMDAL_MUTEX_unlock(&context_mutex);
51
52    ctx = HEIMDAL_getspecific(context_key);
53    if (ctx == NULL) {
54
55	ctx = calloc(1, sizeof(*ctx));
56	if (ctx == NULL)
57	    return NULL;
58	HEIMDAL_setspecific(context_key, ctx, ret);
59	if (ret) {
60	    free(ctx);
61	    return NULL;
62	}
63    }
64    return ctx;
65}
66
67OM_uint32
68_gss_mg_get_error(const gss_OID mech, OM_uint32 type,
69		  OM_uint32 value, gss_buffer_t string)
70{
71    struct mg_thread_ctx *mg;
72
73    mg = _gss_mechglue_thread();
74    if (mg == NULL)
75	return GSS_S_BAD_STATUS;
76
77#if 0
78    /*
79     * We cant check the mech here since a pseudo-mech might have
80     * called an lower layer and then the mech info is all broken
81     */
82    if (mech != NULL && gss_oid_equal(mg->mech, mech) == 0)
83	return GSS_S_BAD_STATUS;
84#endif
85
86    switch (type) {
87    case GSS_C_GSS_CODE: {
88	if (value != mg->maj_stat || mg->maj_error.length == 0)
89	    break;
90	string->value = malloc(mg->maj_error.length + 1);
91	string->length = mg->maj_error.length;
92	memcpy(string->value, mg->maj_error.value, mg->maj_error.length);
93        ((char *) string->value)[string->length] = '\0';
94	return GSS_S_COMPLETE;
95    }
96    case GSS_C_MECH_CODE: {
97	if (value != mg->min_stat || mg->min_error.length == 0)
98	    break;
99	string->value = malloc(mg->min_error.length + 1);
100	string->length = mg->min_error.length;
101	memcpy(string->value, mg->min_error.value, mg->min_error.length);
102        ((char *) string->value)[string->length] = '\0';
103	return GSS_S_COMPLETE;
104    }
105    }
106    string->value = NULL;
107    string->length = 0;
108    return GSS_S_BAD_STATUS;
109}
110
111void
112_gss_mg_error(gssapi_mech_interface m, OM_uint32 maj, OM_uint32 min)
113{
114    OM_uint32 major_status, minor_status;
115    OM_uint32 message_content;
116    struct mg_thread_ctx *mg;
117
118    /*
119     * Mechs without gss_display_status() does
120     * gss_mg_collect_error() by themself.
121     */
122    if (m->gm_display_status == NULL)
123	return ;
124
125    mg = _gss_mechglue_thread();
126    if (mg == NULL)
127	return;
128
129    gss_release_buffer(&minor_status, &mg->maj_error);
130    gss_release_buffer(&minor_status, &mg->min_error);
131
132    mg->mech = &m->gm_mech_oid;
133    mg->maj_stat = maj;
134    mg->min_stat = min;
135
136    major_status = m->gm_display_status(&minor_status,
137					maj,
138					GSS_C_GSS_CODE,
139					&m->gm_mech_oid,
140					&message_content,
141					&mg->maj_error);
142    if (GSS_ERROR(major_status)) {
143	mg->maj_error.value = NULL;
144	mg->maj_error.length = 0;
145    }
146    major_status = m->gm_display_status(&minor_status,
147					min,
148					GSS_C_MECH_CODE,
149					&m->gm_mech_oid,
150					&message_content,
151					&mg->min_error);
152    if (GSS_ERROR(major_status)) {
153	mg->min_error.value = NULL;
154	mg->min_error.length = 0;
155    }
156}
157
158void
159gss_mg_collect_error(gss_OID mech, OM_uint32 maj, OM_uint32 min)
160{
161    gssapi_mech_interface m = __gss_get_mechanism(mech);
162    if (m == NULL)
163	return;
164    _gss_mg_error(m, maj, min);
165}
166