s4-gensec: fixed a GSSAPI SASL negotiation bug
[abartlet/samba.git/.git] / source4 / auth / gensec / gensec_gssapi.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    Kerberos backend for GENSEC
5    
6    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
7    Copyright (C) Stefan Metzmacher <metze@samba.org> 2004-2005
8
9    This program is free software; you can redistribute it and/or modify
10    it under the terms of the GNU General Public License as published by
11    the Free Software Foundation; either version 3 of the License, or
12    (at your option) any later version.
13    
14    This program is distributed in the hope that it will be useful,
15    but WITHOUT ANY WARRANTY; without even the implied warranty of
16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17    GNU General Public License for more details.
18
19    
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24 #include "includes.h"
25 #include "lib/events/events.h"
26 #include "system/kerberos.h"
27 #include "auth/kerberos/kerberos.h"
28 #include "librpc/gen_ndr/krb5pac.h"
29 #include "auth/auth.h"
30 #include "lib/ldb/include/ldb.h"
31 #include "auth/auth_sam.h"
32 #include "librpc/rpc/dcerpc.h"
33 #include "auth/credentials/credentials.h"
34 #include "auth/credentials/credentials_krb5.h"
35 #include "auth/gensec/gensec.h"
36 #include "auth/gensec/gensec_proto.h"
37 #include "param/param.h"
38 #include "auth/session_proto.h"
39 #include <gssapi/gssapi.h>
40 #include <gssapi/gssapi_krb5.h>
41 #include <gssapi/gssapi_spnego.h>
42 #include "auth/gensec/gensec_gssapi.h"
43 #include "lib/util/util_net.h"
44
45 static size_t gensec_gssapi_max_input_size(struct gensec_security *gensec_security);
46 static size_t gensec_gssapi_max_wrapped_size(struct gensec_security *gensec_security);
47
48 static char *gssapi_error_string(TALLOC_CTX *mem_ctx, 
49                                  OM_uint32 maj_stat, OM_uint32 min_stat, 
50                                  const gss_OID mech)
51 {
52         OM_uint32 disp_min_stat, disp_maj_stat;
53         gss_buffer_desc maj_error_message;
54         gss_buffer_desc min_error_message;
55         char *maj_error_string, *min_error_string;
56         OM_uint32 msg_ctx = 0;
57
58         char *ret;
59
60         maj_error_message.value = NULL;
61         min_error_message.value = NULL;
62         maj_error_message.length = 0;
63         min_error_message.length = 0;
64         
65         disp_maj_stat = gss_display_status(&disp_min_stat, maj_stat, GSS_C_GSS_CODE,
66                            mech, &msg_ctx, &maj_error_message);
67         disp_maj_stat = gss_display_status(&disp_min_stat, min_stat, GSS_C_MECH_CODE,
68                            mech, &msg_ctx, &min_error_message);
69         
70         maj_error_string = talloc_strndup(mem_ctx, (char *)maj_error_message.value, maj_error_message.length);
71
72         min_error_string = talloc_strndup(mem_ctx, (char *)min_error_message.value, min_error_message.length);
73
74         ret = talloc_asprintf(mem_ctx, "%s: %s", maj_error_string, min_error_string);
75
76         talloc_free(maj_error_string);
77         talloc_free(min_error_string);
78
79         gss_release_buffer(&disp_min_stat, &maj_error_message);
80         gss_release_buffer(&disp_min_stat, &min_error_message);
81
82         return ret;
83 }
84
85
86 static int gensec_gssapi_destructor(struct gensec_gssapi_state *gensec_gssapi_state)
87 {
88         OM_uint32 maj_stat, min_stat;
89         
90         if (gensec_gssapi_state->delegated_cred_handle != GSS_C_NO_CREDENTIAL) {
91                 maj_stat = gss_release_cred(&min_stat, 
92                                             &gensec_gssapi_state->delegated_cred_handle);
93         }
94
95         if (gensec_gssapi_state->gssapi_context != GSS_C_NO_CONTEXT) {
96                 maj_stat = gss_delete_sec_context (&min_stat,
97                                                    &gensec_gssapi_state->gssapi_context,
98                                                    GSS_C_NO_BUFFER);
99         }
100
101         if (gensec_gssapi_state->server_name != GSS_C_NO_NAME) {
102                 maj_stat = gss_release_name(&min_stat, &gensec_gssapi_state->server_name);
103         }
104         if (gensec_gssapi_state->client_name != GSS_C_NO_NAME) {
105                 maj_stat = gss_release_name(&min_stat, &gensec_gssapi_state->client_name);
106         }
107
108         if (gensec_gssapi_state->lucid) {
109                 gss_krb5_free_lucid_sec_context(&min_stat, gensec_gssapi_state->lucid);
110         }
111
112         return 0;
113 }
114
115 static NTSTATUS gensec_gssapi_init_lucid(struct gensec_gssapi_state *gensec_gssapi_state)
116 {
117         OM_uint32 maj_stat, min_stat;
118
119         if (gensec_gssapi_state->lucid) {
120                 return NT_STATUS_OK;
121         }
122
123         maj_stat = gss_krb5_export_lucid_sec_context(&min_stat,
124                                                      &gensec_gssapi_state->gssapi_context,
125                                                      1,
126                                                      (void **)&gensec_gssapi_state->lucid);
127         if (maj_stat != GSS_S_COMPLETE) {
128                 DEBUG(0,("gensec_gssapi_init_lucid: %s\n",
129                         gssapi_error_string(gensec_gssapi_state,
130                                             maj_stat, min_stat,
131                                             gensec_gssapi_state->gss_oid)));
132                 return NT_STATUS_INTERNAL_ERROR;
133         }
134
135         if (gensec_gssapi_state->lucid->version != 1) {
136                 DEBUG(0,("gensec_gssapi_init_lucid: lucid version[%d] != 1\n",
137                         gensec_gssapi_state->lucid->version));
138                 gss_krb5_free_lucid_sec_context(&min_stat, gensec_gssapi_state->lucid);
139                 gensec_gssapi_state->lucid = NULL;
140                 return NT_STATUS_INTERNAL_ERROR;
141         }
142
143         return NT_STATUS_OK;
144 }
145
146 static NTSTATUS gensec_gssapi_start(struct gensec_security *gensec_security)
147 {
148         struct gensec_gssapi_state *gensec_gssapi_state;
149         krb5_error_code ret;
150         struct gsskrb5_send_to_kdc send_to_kdc;
151         const char *realm;
152
153         gensec_gssapi_state = talloc(gensec_security, struct gensec_gssapi_state);
154         if (!gensec_gssapi_state) {
155                 return NT_STATUS_NO_MEMORY;
156         }
157
158         gensec_security->private_data = gensec_gssapi_state;
159
160         gensec_gssapi_state->gssapi_context = GSS_C_NO_CONTEXT;
161
162         /* TODO: Fill in channel bindings */
163         gensec_gssapi_state->input_chan_bindings = GSS_C_NO_CHANNEL_BINDINGS;
164
165         gensec_gssapi_state->server_name = GSS_C_NO_NAME;
166         gensec_gssapi_state->client_name = GSS_C_NO_NAME;
167         
168         gensec_gssapi_state->want_flags = 0;
169
170         if (gensec_setting_bool(gensec_security->settings, "gensec_gssapi", "delegation_by_kdc_policy", true)) {
171                 gensec_gssapi_state->want_flags |= GSS_C_DELEG_POLICY_FLAG;
172         }
173         if (gensec_setting_bool(gensec_security->settings, "gensec_gssapi", "mutual", true)) {
174                 gensec_gssapi_state->want_flags |= GSS_C_MUTUAL_FLAG;
175         }
176         if (gensec_setting_bool(gensec_security->settings, "gensec_gssapi", "delegation", true)) {
177                 gensec_gssapi_state->want_flags |= GSS_C_DELEG_FLAG;
178         }
179         if (gensec_setting_bool(gensec_security->settings, "gensec_gssapi", "replay", true)) {
180                 gensec_gssapi_state->want_flags |= GSS_C_REPLAY_FLAG;
181         }
182         if (gensec_setting_bool(gensec_security->settings, "gensec_gssapi", "sequence", true)) {
183                 gensec_gssapi_state->want_flags |= GSS_C_SEQUENCE_FLAG;
184         }
185
186         if (gensec_security->want_features & GENSEC_FEATURE_SIGN) {
187                 gensec_gssapi_state->want_flags |= GSS_C_INTEG_FLAG;
188         }
189         if (gensec_security->want_features & GENSEC_FEATURE_SEAL) {
190                 gensec_gssapi_state->want_flags |= GSS_C_CONF_FLAG;
191         }
192         if (gensec_security->want_features & GENSEC_FEATURE_DCE_STYLE) {
193                 gensec_gssapi_state->want_flags |= GSS_C_DCE_STYLE;
194         }
195
196         gensec_gssapi_state->got_flags = 0;
197
198         switch (gensec_security->ops->auth_type) {
199         case DCERPC_AUTH_TYPE_SPNEGO:
200                 gensec_gssapi_state->gss_oid = gss_mech_spnego;
201                 break;
202         case DCERPC_AUTH_TYPE_KRB5:
203         default:
204                 gensec_gssapi_state->gss_oid = gss_mech_krb5;
205                 break;
206         }
207
208         gensec_gssapi_state->session_key = data_blob(NULL, 0);
209         gensec_gssapi_state->pac = data_blob(NULL, 0);
210
211         ret = smb_krb5_init_context(gensec_gssapi_state,
212                                     gensec_security->event_ctx,
213                                     gensec_security->settings->lp_ctx,
214                                     &gensec_gssapi_state->smb_krb5_context);
215         if (ret) {
216                 DEBUG(1,("gensec_krb5_start: krb5_init_context failed (%s)\n",
217                          error_message(ret)));
218                 talloc_free(gensec_gssapi_state);
219                 return NT_STATUS_INTERNAL_ERROR;
220         }
221
222         gensec_gssapi_state->client_cred = NULL;
223         gensec_gssapi_state->server_cred = NULL;
224
225         gensec_gssapi_state->lucid = NULL;
226
227         gensec_gssapi_state->delegated_cred_handle = GSS_C_NO_CREDENTIAL;
228
229         gensec_gssapi_state->sasl = false;
230         gensec_gssapi_state->sasl_state = STAGE_GSS_NEG;
231         gensec_gssapi_state->sasl_protection = 0;
232
233         gensec_gssapi_state->max_wrap_buf_size
234                 = gensec_setting_int(gensec_security->settings, "gensec_gssapi", "max wrap buf size", 65536);
235         gensec_gssapi_state->gss_exchange_count = 0;
236         gensec_gssapi_state->sig_size = 0;
237
238         talloc_set_destructor(gensec_gssapi_state, gensec_gssapi_destructor);
239
240         send_to_kdc.func = smb_krb5_send_and_recv_func;
241         send_to_kdc.ptr = gensec_security->event_ctx;
242
243         ret = gsskrb5_set_send_to_kdc(&send_to_kdc);
244         if (ret) {
245                 DEBUG(1,("gensec_krb5_start: gsskrb5_set_send_to_kdc failed\n"));
246                 talloc_free(gensec_gssapi_state);
247                 return NT_STATUS_INTERNAL_ERROR;
248         }
249
250         realm = lpcfg_realm(gensec_security->settings->lp_ctx);
251         if (realm != NULL) {
252                 ret = gsskrb5_set_default_realm(realm);
253                 if (ret) {
254                         DEBUG(1,("gensec_krb5_start: gsskrb5_set_default_realm failed\n"));
255                         talloc_free(gensec_gssapi_state);
256                         return NT_STATUS_INTERNAL_ERROR;
257                 }
258         }
259
260         /* don't do DNS lookups of any kind, it might/will fail for a netbios name */
261         ret = gsskrb5_set_dns_canonicalize(gensec_setting_bool(gensec_security->settings, "krb5", "set_dns_canonicalize", false));
262         if (ret) {
263                 DEBUG(1,("gensec_krb5_start: gsskrb5_set_dns_canonicalize failed\n"));
264                 talloc_free(gensec_gssapi_state);
265                 return NT_STATUS_INTERNAL_ERROR;
266         }
267
268         return NT_STATUS_OK;
269 }
270
271 static NTSTATUS gensec_gssapi_server_start(struct gensec_security *gensec_security)
272 {
273         NTSTATUS nt_status;
274         int ret;
275         struct gensec_gssapi_state *gensec_gssapi_state;
276         struct cli_credentials *machine_account;
277         struct gssapi_creds_container *gcc;
278
279         nt_status = gensec_gssapi_start(gensec_security);
280         if (!NT_STATUS_IS_OK(nt_status)) {
281                 return nt_status;
282         }
283
284         gensec_gssapi_state = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
285
286         machine_account = gensec_get_credentials(gensec_security);
287         
288         if (!machine_account) {
289                 DEBUG(3, ("No machine account credentials specified\n"));
290                 return NT_STATUS_CANT_ACCESS_DOMAIN_INFO;
291         } else {
292                 ret = cli_credentials_get_server_gss_creds(machine_account, 
293                                                            gensec_security->event_ctx, 
294                                                            gensec_security->settings->lp_ctx, &gcc);
295                 if (ret) {
296                         DEBUG(1, ("Aquiring acceptor credentials failed: %s\n", 
297                                   error_message(ret)));
298                         return NT_STATUS_CANT_ACCESS_DOMAIN_INFO;
299                 }
300         }
301
302         gensec_gssapi_state->server_cred = gcc;
303         return NT_STATUS_OK;
304
305 }
306
307 static NTSTATUS gensec_gssapi_sasl_server_start(struct gensec_security *gensec_security)
308 {
309         NTSTATUS nt_status;
310         struct gensec_gssapi_state *gensec_gssapi_state;
311         nt_status = gensec_gssapi_server_start(gensec_security);
312
313         if (NT_STATUS_IS_OK(nt_status)) {
314                 gensec_gssapi_state = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
315                 gensec_gssapi_state->sasl = true;
316         }
317         return nt_status;
318 }
319
320 static NTSTATUS gensec_gssapi_client_start(struct gensec_security *gensec_security)
321 {
322         struct gensec_gssapi_state *gensec_gssapi_state;
323         struct cli_credentials *creds = gensec_get_credentials(gensec_security);
324         krb5_error_code ret;
325         NTSTATUS nt_status;
326         gss_buffer_desc name_token;
327         gss_OID name_type;
328         OM_uint32 maj_stat, min_stat;
329         const char *hostname = gensec_get_target_hostname(gensec_security);
330         const char *principal;
331         struct gssapi_creds_container *gcc;
332         const char *error_string;
333
334         if (!hostname) {
335                 DEBUG(1, ("Could not determine hostname for target computer, cannot use kerberos\n"));
336                 return NT_STATUS_INVALID_PARAMETER;
337         }
338         if (is_ipaddress(hostname)) {
339                 DEBUG(2, ("Cannot do GSSAPI to an IP address\n"));
340                 return NT_STATUS_INVALID_PARAMETER;
341         }
342         if (strcmp(hostname, "localhost") == 0) {
343                 DEBUG(2, ("GSSAPI to 'localhost' does not make sense\n"));
344                 return NT_STATUS_INVALID_PARAMETER;
345         }
346
347         nt_status = gensec_gssapi_start(gensec_security);
348         if (!NT_STATUS_IS_OK(nt_status)) {
349                 return nt_status;
350         }
351
352         gensec_gssapi_state = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
353
354         principal = gensec_get_target_principal(gensec_security);
355         if (principal && lpcfg_client_use_spnego_principal(gensec_security->settings->lp_ctx)) {
356                 name_type = GSS_C_NULL_OID;
357         } else {
358                 principal = talloc_asprintf(gensec_gssapi_state, "%s/%s@%s",
359                                             gensec_get_target_service(gensec_security), 
360                                             hostname, lpcfg_realm(gensec_security->settings->lp_ctx));
361
362                 name_type = GSS_C_NT_USER_NAME;
363         }
364         name_token.value  = discard_const_p(uint8_t, principal);
365         name_token.length = strlen(principal);
366
367
368         maj_stat = gss_import_name (&min_stat,
369                                     &name_token,
370                                     name_type,
371                                     &gensec_gssapi_state->server_name);
372         if (maj_stat) {
373                 DEBUG(2, ("GSS Import name of %s failed: %s\n",
374                           (char *)name_token.value,
375                           gssapi_error_string(gensec_gssapi_state, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
376                 return NT_STATUS_INVALID_PARAMETER;
377         }
378
379         ret = cli_credentials_get_client_gss_creds(creds, 
380                                                    gensec_security->event_ctx, 
381                                                    gensec_security->settings->lp_ctx, &gcc, &error_string);
382         switch (ret) {
383         case 0:
384                 break;
385         case KRB5KDC_ERR_PREAUTH_FAILED:
386                 return NT_STATUS_LOGON_FAILURE;
387         case KRB5_KDC_UNREACH:
388                 DEBUG(3, ("Cannot reach a KDC we require to contact %s : %s\n", principal, error_string));
389                 return NT_STATUS_INVALID_PARAMETER; /* Make SPNEGO ignore us, we can't go any further here */
390         case KRB5_CC_NOTFOUND:
391         case KRB5_CC_END:
392                 DEBUG(3, ("Error preparing credentials we require to contact %s : %s\n", principal, error_string));
393                 return NT_STATUS_INVALID_PARAMETER; /* Make SPNEGO ignore us, we can't go any further here */
394         default:
395                 DEBUG(1, ("Aquiring initiator credentials failed: %s\n", error_string));
396                 return NT_STATUS_UNSUCCESSFUL;
397         }
398
399         gensec_gssapi_state->client_cred = gcc;
400         if (!talloc_reference(gensec_gssapi_state, gcc)) {
401                 return NT_STATUS_NO_MEMORY;
402         }
403         
404         return NT_STATUS_OK;
405 }
406
407 static NTSTATUS gensec_gssapi_sasl_client_start(struct gensec_security *gensec_security)
408 {
409         NTSTATUS nt_status;
410         struct gensec_gssapi_state *gensec_gssapi_state;
411         nt_status = gensec_gssapi_client_start(gensec_security);
412
413         if (NT_STATUS_IS_OK(nt_status)) {
414                 gensec_gssapi_state = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
415                 gensec_gssapi_state->sasl = true;
416         }
417         return nt_status;
418 }
419
420
421 /**
422  * Check if the packet is one for this mechansim
423  * 
424  * @param gensec_security GENSEC state
425  * @param in The request, as a DATA_BLOB
426  * @return Error, INVALID_PARAMETER if it's not a packet for us
427  *                or NT_STATUS_OK if the packet is ok. 
428  */
429
430 static NTSTATUS gensec_gssapi_magic(struct gensec_security *gensec_security, 
431                                     const DATA_BLOB *in) 
432 {
433         if (gensec_gssapi_check_oid(in, GENSEC_OID_KERBEROS5)) {
434                 return NT_STATUS_OK;
435         } else {
436                 return NT_STATUS_INVALID_PARAMETER;
437         }
438 }
439
440
441 /**
442  * Next state function for the GSSAPI GENSEC mechanism
443  * 
444  * @param gensec_gssapi_state GSSAPI State
445  * @param out_mem_ctx The TALLOC_CTX for *out to be allocated on
446  * @param in The request, as a DATA_BLOB
447  * @param out The reply, as an talloc()ed DATA_BLOB, on *out_mem_ctx
448  * @return Error, MORE_PROCESSING_REQUIRED if a reply is sent, 
449  *                or NT_STATUS_OK if the user is authenticated. 
450  */
451
452 static NTSTATUS gensec_gssapi_update(struct gensec_security *gensec_security, 
453                                    TALLOC_CTX *out_mem_ctx, 
454                                    const DATA_BLOB in, DATA_BLOB *out) 
455 {
456         struct gensec_gssapi_state *gensec_gssapi_state
457                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
458         NTSTATUS nt_status = NT_STATUS_LOGON_FAILURE;
459         OM_uint32 maj_stat, min_stat;
460         OM_uint32 min_stat2;
461         gss_buffer_desc input_token, output_token;
462         gss_OID gss_oid_p = NULL;
463         input_token.length = in.length;
464         input_token.value = in.data;
465
466         switch (gensec_gssapi_state->sasl_state) {
467         case STAGE_GSS_NEG:
468         {
469                 switch (gensec_security->gensec_role) {
470                 case GENSEC_CLIENT:
471                 {
472                         maj_stat = gss_init_sec_context(&min_stat, 
473                                                         gensec_gssapi_state->client_cred->creds,
474                                                         &gensec_gssapi_state->gssapi_context, 
475                                                         gensec_gssapi_state->server_name, 
476                                                         gensec_gssapi_state->gss_oid,
477                                                         gensec_gssapi_state->want_flags, 
478                                                         0, 
479                                                         gensec_gssapi_state->input_chan_bindings,
480                                                         &input_token, 
481                                                         &gss_oid_p,
482                                                         &output_token, 
483                                                         &gensec_gssapi_state->got_flags, /* ret flags */
484                                                         NULL);
485                         if (gss_oid_p) {
486                                 gensec_gssapi_state->gss_oid = gss_oid_p;
487                         }
488                         break;
489                 }
490                 case GENSEC_SERVER:
491                 {
492                         maj_stat = gss_accept_sec_context(&min_stat, 
493                                                           &gensec_gssapi_state->gssapi_context, 
494                                                           gensec_gssapi_state->server_cred->creds,
495                                                           &input_token, 
496                                                           gensec_gssapi_state->input_chan_bindings,
497                                                           &gensec_gssapi_state->client_name, 
498                                                           &gss_oid_p,
499                                                           &output_token, 
500                                                           &gensec_gssapi_state->got_flags, 
501                                                           NULL, 
502                                                           &gensec_gssapi_state->delegated_cred_handle);
503                         if (gss_oid_p) {
504                                 gensec_gssapi_state->gss_oid = gss_oid_p;
505                         }
506                         break;
507                 }
508                 default:
509                         return NT_STATUS_INVALID_PARAMETER;
510                         
511                 }
512
513                 gensec_gssapi_state->gss_exchange_count++;
514
515                 if (maj_stat == GSS_S_COMPLETE) {
516                         *out = data_blob_talloc(out_mem_ctx, output_token.value, output_token.length);
517                         gss_release_buffer(&min_stat2, &output_token);
518                         
519                         if (gensec_gssapi_state->got_flags & GSS_C_DELEG_FLAG) {
520                                 DEBUG(5, ("gensec_gssapi: credentials were delegated\n"));
521                         } else {
522                                 DEBUG(5, ("gensec_gssapi: NO credentials were delegated\n"));
523                         }
524
525                         /* We may have been invoked as SASL, so there
526                          * is more work to do */
527                         if (gensec_gssapi_state->sasl) {
528                                 /* Due to a very subtle interaction
529                                  * with SASL and the LDAP libs, we
530                                  * must ensure the data pointer is 
531                                  * != NULL, but the length is 0.  
532                                  *
533                                  * This ensures we send a 'zero
534                                  * length' (rather than NULL) response 
535                                  */
536                                 
537                                 if (!out->data) {
538                                         out->data = (uint8_t *)talloc_strdup(out_mem_ctx, "\0");
539                                 }
540
541                                 gensec_gssapi_state->sasl_state = STAGE_SASL_SSF_NEG;
542                                 return NT_STATUS_MORE_PROCESSING_REQUIRED;
543                         } else {
544                                 gensec_gssapi_state->sasl_state = STAGE_DONE;
545
546                                 if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
547                                         DEBUG(5, ("GSSAPI Connection will be cryptographicly sealed\n"));
548                                 } else if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
549                                         DEBUG(5, ("GSSAPI Connection will be cryptographicly signed\n"));
550                                 } else {
551                                         DEBUG(5, ("GSSAPI Connection will have no cryptographic protection\n"));
552                                 }
553
554                                 return NT_STATUS_OK;
555                         }
556                 } else if (maj_stat == GSS_S_CONTINUE_NEEDED) {
557                         *out = data_blob_talloc(out_mem_ctx, output_token.value, output_token.length);
558                         gss_release_buffer(&min_stat2, &output_token);
559                         
560                         return NT_STATUS_MORE_PROCESSING_REQUIRED;
561                 } else if (gss_oid_equal(gensec_gssapi_state->gss_oid, gss_mech_krb5)) {
562                         switch (min_stat) {
563                         case KRB5_KDC_UNREACH:
564                                 DEBUG(3, ("Cannot reach a KDC we require: %s\n",
565                                           gssapi_error_string(gensec_gssapi_state, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
566                                 return NT_STATUS_INVALID_PARAMETER; /* Make SPNEGO ignore us, we can't go any further here */
567                         case KRB5KDC_ERR_S_PRINCIPAL_UNKNOWN:
568                                 DEBUG(3, ("Server is not registered with our KDC: %s\n", 
569                                           gssapi_error_string(gensec_gssapi_state, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
570                                 return NT_STATUS_INVALID_PARAMETER; /* Make SPNEGO ignore us, we can't go any further here */
571                         case KRB5KRB_AP_ERR_MSG_TYPE:
572                                 /* garbage input, possibly from the auto-mech detection */
573                                 return NT_STATUS_INVALID_PARAMETER;
574                         default:
575                                 DEBUG(1, ("GSS Update(krb5)(%d) Update failed: %s\n", 
576                                           gensec_gssapi_state->gss_exchange_count,
577                                           gssapi_error_string(out_mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
578                                 return nt_status;
579                         }
580                 } else {
581                         DEBUG(1, ("GSS Update(%d) failed: %s\n", 
582                                   gensec_gssapi_state->gss_exchange_count,
583                                   gssapi_error_string(out_mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
584                         return nt_status;
585                 }
586                 break;
587         }
588
589         /* These last two stages are only done if we were invoked as SASL */
590         case STAGE_SASL_SSF_NEG:
591         {
592                 switch (gensec_security->gensec_role) {
593                 case GENSEC_CLIENT:
594                 {
595                         uint8_t maxlength_proposed[4]; 
596                         uint8_t maxlength_accepted[4]; 
597                         uint8_t security_supported;
598                         int conf_state;
599                         gss_qop_t qop_state;
600                         input_token.length = in.length;
601                         input_token.value = in.data;
602
603                         /* As a client, we have just send a
604                          * zero-length blob to the server (after the
605                          * normal GSSAPI exchange), and it has replied
606                          * with it's SASL negotiation */
607                         
608                         maj_stat = gss_unwrap(&min_stat, 
609                                               gensec_gssapi_state->gssapi_context, 
610                                               &input_token,
611                                               &output_token, 
612                                               &conf_state,
613                                               &qop_state);
614                         if (GSS_ERROR(maj_stat)) {
615                                 DEBUG(1, ("gensec_gssapi_update: GSS UnWrap of SASL protection negotiation failed: %s\n", 
616                                           gssapi_error_string(out_mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
617                                 return NT_STATUS_ACCESS_DENIED;
618                         }
619                         
620                         if (output_token.length < 4) {
621                                 return NT_STATUS_INVALID_PARAMETER;
622                         }
623
624                         memcpy(maxlength_proposed, output_token.value, 4);
625                         gss_release_buffer(&min_stat, &output_token);
626
627                         /* first byte is the proposed security */
628                         security_supported = maxlength_proposed[0];
629                         maxlength_proposed[0] = '\0';
630                         
631                         /* Rest is the proposed max wrap length */
632                         gensec_gssapi_state->max_wrap_buf_size = MIN(RIVAL(maxlength_proposed, 0), 
633                                                                      gensec_gssapi_state->max_wrap_buf_size);
634                         gensec_gssapi_state->sasl_protection = 0;
635                         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
636                                 if (security_supported & NEG_SEAL) {
637                                         gensec_gssapi_state->sasl_protection |= NEG_SEAL;
638                                 }
639                         } else if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
640                                 if (security_supported & NEG_SIGN) {
641                                         gensec_gssapi_state->sasl_protection |= NEG_SIGN;
642                                 }
643                         } else if (security_supported & NEG_NONE) {
644                                 gensec_gssapi_state->sasl_protection |= NEG_NONE;
645                         } else {
646                                 DEBUG(1, ("Remote server does not support unprotected connections"));
647                                 return NT_STATUS_ACCESS_DENIED;
648                         }
649
650                         /* Send back the negotiated max length */
651
652                         RSIVAL(maxlength_accepted, 0, gensec_gssapi_state->max_wrap_buf_size);
653
654                         maxlength_accepted[0] = gensec_gssapi_state->sasl_protection;
655                         
656                         input_token.value = maxlength_accepted;
657                         input_token.length = sizeof(maxlength_accepted);
658
659                         maj_stat = gss_wrap(&min_stat, 
660                                             gensec_gssapi_state->gssapi_context, 
661                                             false,
662                                             GSS_C_QOP_DEFAULT,
663                                             &input_token,
664                                             &conf_state,
665                                             &output_token);
666                         if (GSS_ERROR(maj_stat)) {
667                                 DEBUG(1, ("GSS Update(SSF_NEG): GSS Wrap failed: %s\n", 
668                                           gssapi_error_string(out_mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
669                                 return NT_STATUS_ACCESS_DENIED;
670                         }
671                         
672                         *out = data_blob_talloc(out_mem_ctx, output_token.value, output_token.length);
673                         gss_release_buffer(&min_stat, &output_token);
674
675                         /* quirk:  This changes the value that gensec_have_feature returns, to be that after SASL negotiation */
676                         gensec_gssapi_state->sasl_state = STAGE_DONE;
677
678                         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
679                                 DEBUG(3, ("SASL/GSSAPI Connection to server will be cryptographicly sealed\n"));
680                         } else if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
681                                 DEBUG(3, ("SASL/GSSAPI Connection to server will be cryptographicly signed\n"));
682                         } else {
683                                 DEBUG(3, ("SASL/GSSAPI Connection to server will have no cryptographicly protection\n"));
684                         }
685
686                         return NT_STATUS_OK;
687                 }
688                 case GENSEC_SERVER:
689                 {
690                         uint8_t maxlength_proposed[4]; 
691                         uint8_t security_supported = 0x0;
692                         int conf_state;
693
694                         /* As a server, we have just been sent a zero-length blob (note this, but it isn't fatal) */
695                         if (in.length != 0) {
696                                 DEBUG(1, ("SASL/GSSAPI: client sent non-zero length starting SASL negotiation!\n"));
697                         }
698                         
699                         /* Give the client some idea what we will support */
700                           
701                         RSIVAL(maxlength_proposed, 0, gensec_gssapi_state->max_wrap_buf_size);
702                         /* first byte is the proposed security */
703                         maxlength_proposed[0] = '\0';
704                         
705                         gensec_gssapi_state->sasl_protection = 0;
706                         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
707                                 security_supported |= NEG_SEAL;
708                         } 
709                         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
710                                 security_supported |= NEG_SIGN;
711                         }
712                         if (security_supported == 0) {
713                                 /* If we don't support anything, this must be 0 */
714                                 RSIVAL(maxlength_proposed, 0, 0x0);
715                         }
716
717                         /* TODO:  We may not wish to support this */
718                         security_supported |= NEG_NONE;
719                         maxlength_proposed[0] = security_supported;
720                         
721                         input_token.value = maxlength_proposed;
722                         input_token.length = sizeof(maxlength_proposed);
723
724                         maj_stat = gss_wrap(&min_stat, 
725                                             gensec_gssapi_state->gssapi_context, 
726                                             false,
727                                             GSS_C_QOP_DEFAULT,
728                                             &input_token,
729                                             &conf_state,
730                                             &output_token);
731                         if (GSS_ERROR(maj_stat)) {
732                                 DEBUG(1, ("GSS Update(SSF_NEG): GSS Wrap failed: %s\n", 
733                                           gssapi_error_string(out_mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
734                                 return NT_STATUS_ACCESS_DENIED;
735                         }
736                         
737                         *out = data_blob_talloc(out_mem_ctx, output_token.value, output_token.length);
738                         gss_release_buffer(&min_stat, &output_token);
739
740                         gensec_gssapi_state->sasl_state = STAGE_SASL_SSF_ACCEPT;
741                         return NT_STATUS_MORE_PROCESSING_REQUIRED;
742                 }
743                 default:
744                         return NT_STATUS_INVALID_PARAMETER;
745                         
746                 }
747         }
748         /* This is s server-only stage */
749         case STAGE_SASL_SSF_ACCEPT:
750         {
751                 uint8_t maxlength_accepted[4]; 
752                 uint8_t security_accepted;
753                 int conf_state;
754                 gss_qop_t qop_state;
755                 input_token.length = in.length;
756                 input_token.value = in.data;
757                         
758                 maj_stat = gss_unwrap(&min_stat, 
759                                       gensec_gssapi_state->gssapi_context, 
760                                       &input_token,
761                                       &output_token, 
762                                       &conf_state,
763                                       &qop_state);
764                 if (GSS_ERROR(maj_stat)) {
765                         DEBUG(1, ("gensec_gssapi_update: GSS UnWrap of SASL protection negotiation failed: %s\n", 
766                                   gssapi_error_string(out_mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
767                         return NT_STATUS_ACCESS_DENIED;
768                 }
769                         
770                 if (output_token.length < 4) {
771                         return NT_STATUS_INVALID_PARAMETER;
772                 }
773
774                 memcpy(maxlength_accepted, output_token.value, 4);
775                 gss_release_buffer(&min_stat, &output_token);
776                 
777                 /* first byte is the proposed security */
778                 security_accepted = maxlength_accepted[0];
779                 maxlength_accepted[0] = '\0';
780
781                 /* Rest is the proposed max wrap length */
782                 gensec_gssapi_state->max_wrap_buf_size = MIN(RIVAL(maxlength_accepted, 0), 
783                                                              gensec_gssapi_state->max_wrap_buf_size);
784
785                 gensec_gssapi_state->sasl_protection = 0;
786                 if (security_accepted & NEG_SEAL) {
787                         if (!gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
788                                 DEBUG(1, ("Remote client wanted seal, but gensec refused\n"));
789                                 return NT_STATUS_ACCESS_DENIED;
790                         }
791                         gensec_gssapi_state->sasl_protection |= NEG_SEAL;
792                 }
793                 if (security_accepted & NEG_SIGN) {
794                         if (!gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
795                                 DEBUG(1, ("Remote client wanted sign, but gensec refused\n"));
796                                 return NT_STATUS_ACCESS_DENIED;
797                         }
798                         gensec_gssapi_state->sasl_protection |= NEG_SIGN;
799                 }
800                 if (security_accepted & NEG_NONE) {
801                         gensec_gssapi_state->sasl_protection |= NEG_NONE;
802                 }
803
804                 /* quirk:  This changes the value that gensec_have_feature returns, to be that after SASL negotiation */
805                 gensec_gssapi_state->sasl_state = STAGE_DONE;
806                 if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
807                         DEBUG(5, ("SASL/GSSAPI Connection from client will be cryptographicly sealed\n"));
808                 } else if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
809                         DEBUG(5, ("SASL/GSSAPI Connection from client will be cryptographicly signed\n"));
810                 } else {
811                         DEBUG(5, ("SASL/GSSAPI Connection from client will have no cryptographic protection\n"));
812                 }
813
814                 *out = data_blob(NULL, 0);
815                 return NT_STATUS_OK;    
816         }
817         default:
818                 return NT_STATUS_INVALID_PARAMETER;
819         }
820 }
821
822 static NTSTATUS gensec_gssapi_wrap(struct gensec_security *gensec_security, 
823                                    TALLOC_CTX *mem_ctx, 
824                                    const DATA_BLOB *in, 
825                                    DATA_BLOB *out)
826 {
827         struct gensec_gssapi_state *gensec_gssapi_state
828                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
829         OM_uint32 maj_stat, min_stat;
830         gss_buffer_desc input_token, output_token;
831         int conf_state;
832         input_token.length = in->length;
833         input_token.value = in->data;
834
835         maj_stat = gss_wrap(&min_stat, 
836                             gensec_gssapi_state->gssapi_context, 
837                             gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL),
838                             GSS_C_QOP_DEFAULT,
839                             &input_token,
840                             &conf_state,
841                             &output_token);
842         if (GSS_ERROR(maj_stat)) {
843                 DEBUG(1, ("gensec_gssapi_wrap: GSS Wrap failed: %s\n", 
844                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
845                 return NT_STATUS_ACCESS_DENIED;
846         }
847
848         *out = data_blob_talloc(mem_ctx, output_token.value, output_token.length);
849         gss_release_buffer(&min_stat, &output_token);
850
851         if (gensec_gssapi_state->sasl) {
852                 size_t max_wrapped_size = gensec_gssapi_max_wrapped_size(gensec_security);
853                 if (max_wrapped_size < out->length) {
854                         DEBUG(1, ("gensec_gssapi_wrap: when wrapped, INPUT data (%u) is grew to be larger than SASL negotiated maximum output size (%u > %u)\n",
855                                   (unsigned)in->length, 
856                                   (unsigned)out->length, 
857                                   (unsigned int)max_wrapped_size));
858                         return NT_STATUS_INVALID_PARAMETER;
859                 }
860         }
861         
862         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)
863             && !conf_state) {
864                 return NT_STATUS_ACCESS_DENIED;
865         }
866         return NT_STATUS_OK;
867 }
868
869 static NTSTATUS gensec_gssapi_unwrap(struct gensec_security *gensec_security, 
870                                      TALLOC_CTX *mem_ctx, 
871                                      const DATA_BLOB *in, 
872                                      DATA_BLOB *out)
873 {
874         struct gensec_gssapi_state *gensec_gssapi_state
875                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
876         OM_uint32 maj_stat, min_stat;
877         gss_buffer_desc input_token, output_token;
878         int conf_state;
879         gss_qop_t qop_state;
880         input_token.length = in->length;
881         input_token.value = in->data;
882         
883         if (gensec_gssapi_state->sasl) {
884                 size_t max_wrapped_size = gensec_gssapi_max_wrapped_size(gensec_security);
885                 if (max_wrapped_size < in->length) {
886                         DEBUG(1, ("gensec_gssapi_unwrap: WRAPPED data is larger than SASL negotiated maximum size\n"));
887                         return NT_STATUS_INVALID_PARAMETER;
888                 }
889         }
890         
891         maj_stat = gss_unwrap(&min_stat, 
892                               gensec_gssapi_state->gssapi_context, 
893                               &input_token,
894                               &output_token, 
895                               &conf_state,
896                               &qop_state);
897         if (GSS_ERROR(maj_stat)) {
898                 DEBUG(1, ("gensec_gssapi_unwrap: GSS UnWrap failed: %s\n", 
899                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
900                 return NT_STATUS_ACCESS_DENIED;
901         }
902
903         *out = data_blob_talloc(mem_ctx, output_token.value, output_token.length);
904         gss_release_buffer(&min_stat, &output_token);
905         
906         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)
907             && !conf_state) {
908                 return NT_STATUS_ACCESS_DENIED;
909         }
910         return NT_STATUS_OK;
911 }
912
913 /* Find out the maximum input size negotiated on this connection */
914
915 static size_t gensec_gssapi_max_input_size(struct gensec_security *gensec_security) 
916 {
917         struct gensec_gssapi_state *gensec_gssapi_state
918                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
919         OM_uint32 maj_stat, min_stat;
920         OM_uint32 max_input_size;
921
922         maj_stat = gss_wrap_size_limit(&min_stat, 
923                                        gensec_gssapi_state->gssapi_context,
924                                        gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL),
925                                        GSS_C_QOP_DEFAULT,
926                                        gensec_gssapi_state->max_wrap_buf_size,
927                                        &max_input_size);
928         if (GSS_ERROR(maj_stat)) {
929                 TALLOC_CTX *mem_ctx = talloc_new(NULL); 
930                 DEBUG(1, ("gensec_gssapi_max_input_size: determinaing signature size with gss_wrap_size_limit failed: %s\n", 
931                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
932                 talloc_free(mem_ctx);
933                 return 0;
934         }
935
936         return max_input_size;
937 }
938
939 /* Find out the maximum output size negotiated on this connection */
940 static size_t gensec_gssapi_max_wrapped_size(struct gensec_security *gensec_security) 
941 {
942         struct gensec_gssapi_state *gensec_gssapi_state = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);;
943         return gensec_gssapi_state->max_wrap_buf_size;
944 }
945
946 static NTSTATUS gensec_gssapi_seal_packet(struct gensec_security *gensec_security, 
947                                           TALLOC_CTX *mem_ctx, 
948                                           uint8_t *data, size_t length, 
949                                           const uint8_t *whole_pdu, size_t pdu_length, 
950                                           DATA_BLOB *sig)
951 {
952         struct gensec_gssapi_state *gensec_gssapi_state
953                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
954         OM_uint32 maj_stat, min_stat;
955         gss_buffer_desc input_token, output_token;
956         int conf_state;
957         ssize_t sig_length;
958
959         input_token.length = length;
960         input_token.value = data;
961         
962         maj_stat = gss_wrap(&min_stat, 
963                             gensec_gssapi_state->gssapi_context,
964                             gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL),
965                             GSS_C_QOP_DEFAULT,
966                             &input_token,
967                             &conf_state,
968                             &output_token);
969         if (GSS_ERROR(maj_stat)) {
970                 DEBUG(1, ("gensec_gssapi_seal_packet: GSS Wrap failed: %s\n", 
971                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
972                 return NT_STATUS_ACCESS_DENIED;
973         }
974
975         if (output_token.length < input_token.length) {
976                 DEBUG(1, ("gensec_gssapi_seal_packet: GSS Wrap length [%ld] *less* than caller length [%ld]\n", 
977                           (long)output_token.length, (long)length));
978                 return NT_STATUS_INTERNAL_ERROR;
979         }
980         sig_length = output_token.length - input_token.length;
981
982         memcpy(data, ((uint8_t *)output_token.value) + sig_length, length);
983         *sig = data_blob_talloc(mem_ctx, (uint8_t *)output_token.value, sig_length);
984
985         dump_data_pw("gensec_gssapi_seal_packet: sig\n", sig->data, sig->length);
986         dump_data_pw("gensec_gssapi_seal_packet: clear\n", data, length);
987         dump_data_pw("gensec_gssapi_seal_packet: sealed\n", ((uint8_t *)output_token.value) + sig_length, output_token.length - sig_length);
988
989         gss_release_buffer(&min_stat, &output_token);
990
991         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)
992             && !conf_state) {
993                 return NT_STATUS_ACCESS_DENIED;
994         }
995         return NT_STATUS_OK;
996 }
997
998 static NTSTATUS gensec_gssapi_unseal_packet(struct gensec_security *gensec_security, 
999                                             TALLOC_CTX *mem_ctx, 
1000                                             uint8_t *data, size_t length, 
1001                                             const uint8_t *whole_pdu, size_t pdu_length,
1002                                             const DATA_BLOB *sig)
1003 {
1004         struct gensec_gssapi_state *gensec_gssapi_state
1005                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1006         OM_uint32 maj_stat, min_stat;
1007         gss_buffer_desc input_token, output_token;
1008         int conf_state;
1009         gss_qop_t qop_state;
1010         DATA_BLOB in;
1011
1012         dump_data_pw("gensec_gssapi_unseal_packet: sig\n", sig->data, sig->length);
1013
1014         in = data_blob_talloc(mem_ctx, NULL, sig->length + length);
1015
1016         memcpy(in.data, sig->data, sig->length);
1017         memcpy(in.data + sig->length, data, length);
1018
1019         input_token.length = in.length;
1020         input_token.value = in.data;
1021         
1022         maj_stat = gss_unwrap(&min_stat, 
1023                               gensec_gssapi_state->gssapi_context, 
1024                               &input_token,
1025                               &output_token, 
1026                               &conf_state,
1027                               &qop_state);
1028         if (GSS_ERROR(maj_stat)) {
1029                 DEBUG(1, ("gensec_gssapi_unseal_packet: GSS UnWrap failed: %s\n", 
1030                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1031                 return NT_STATUS_ACCESS_DENIED;
1032         }
1033
1034         if (output_token.length != length) {
1035                 return NT_STATUS_INTERNAL_ERROR;
1036         }
1037
1038         memcpy(data, output_token.value, length);
1039
1040         gss_release_buffer(&min_stat, &output_token);
1041         
1042         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)
1043             && !conf_state) {
1044                 return NT_STATUS_ACCESS_DENIED;
1045         }
1046         return NT_STATUS_OK;
1047 }
1048
1049 static NTSTATUS gensec_gssapi_sign_packet(struct gensec_security *gensec_security, 
1050                                           TALLOC_CTX *mem_ctx, 
1051                                           const uint8_t *data, size_t length, 
1052                                           const uint8_t *whole_pdu, size_t pdu_length, 
1053                                           DATA_BLOB *sig)
1054 {
1055         struct gensec_gssapi_state *gensec_gssapi_state
1056                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1057         OM_uint32 maj_stat, min_stat;
1058         gss_buffer_desc input_token, output_token;
1059
1060         if (gensec_security->want_features & GENSEC_FEATURE_SIGN_PKT_HEADER) {
1061                 input_token.length = pdu_length;
1062                 input_token.value = discard_const_p(uint8_t *, whole_pdu);
1063         } else {
1064                 input_token.length = length;
1065                 input_token.value = discard_const_p(uint8_t *, data);
1066         }
1067
1068         maj_stat = gss_get_mic(&min_stat,
1069                             gensec_gssapi_state->gssapi_context,
1070                             GSS_C_QOP_DEFAULT,
1071                             &input_token,
1072                             &output_token);
1073         if (GSS_ERROR(maj_stat)) {
1074                 DEBUG(1, ("GSS GetMic failed: %s\n",
1075                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1076                 return NT_STATUS_ACCESS_DENIED;
1077         }
1078
1079         *sig = data_blob_talloc(mem_ctx, (uint8_t *)output_token.value, output_token.length);
1080
1081         dump_data_pw("gensec_gssapi_seal_packet: sig\n", sig->data, sig->length);
1082
1083         gss_release_buffer(&min_stat, &output_token);
1084
1085         return NT_STATUS_OK;
1086 }
1087
1088 static NTSTATUS gensec_gssapi_check_packet(struct gensec_security *gensec_security, 
1089                                            TALLOC_CTX *mem_ctx, 
1090                                            const uint8_t *data, size_t length, 
1091                                            const uint8_t *whole_pdu, size_t pdu_length, 
1092                                            const DATA_BLOB *sig)
1093 {
1094         struct gensec_gssapi_state *gensec_gssapi_state
1095                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1096         OM_uint32 maj_stat, min_stat;
1097         gss_buffer_desc input_token;
1098         gss_buffer_desc input_message;
1099         gss_qop_t qop_state;
1100
1101         dump_data_pw("gensec_gssapi_seal_packet: sig\n", sig->data, sig->length);
1102
1103         if (gensec_security->want_features & GENSEC_FEATURE_SIGN_PKT_HEADER) {
1104                 input_message.length = pdu_length;
1105                 input_message.value = discard_const(whole_pdu);
1106         } else {
1107                 input_message.length = length;
1108                 input_message.value = discard_const(data);
1109         }
1110
1111         input_token.length = sig->length;
1112         input_token.value = sig->data;
1113
1114         maj_stat = gss_verify_mic(&min_stat,
1115                               gensec_gssapi_state->gssapi_context, 
1116                               &input_message,
1117                               &input_token,
1118                               &qop_state);
1119         if (GSS_ERROR(maj_stat)) {
1120                 DEBUG(1, ("GSS VerifyMic failed: %s\n",
1121                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1122                 return NT_STATUS_ACCESS_DENIED;
1123         }
1124
1125         return NT_STATUS_OK;
1126 }
1127
1128 /* Try to figure out what features we actually got on the connection */
1129 static bool gensec_gssapi_have_feature(struct gensec_security *gensec_security, 
1130                                        uint32_t feature) 
1131 {
1132         struct gensec_gssapi_state *gensec_gssapi_state
1133                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1134         if (feature & GENSEC_FEATURE_SIGN) {
1135                 /* If we are going GSSAPI SASL, then we honour the second negotiation */
1136                 if (gensec_gssapi_state->sasl 
1137                     && gensec_gssapi_state->sasl_state == STAGE_DONE) {
1138                         return ((gensec_gssapi_state->sasl_protection & NEG_SIGN) 
1139                                 && (gensec_gssapi_state->got_flags & GSS_C_INTEG_FLAG));
1140                 }
1141                 return gensec_gssapi_state->got_flags & GSS_C_INTEG_FLAG;
1142         }
1143         if (feature & GENSEC_FEATURE_SEAL) {
1144                 /* If we are going GSSAPI SASL, then we honour the second negotiation */
1145                 if (gensec_gssapi_state->sasl 
1146                     && gensec_gssapi_state->sasl_state == STAGE_DONE) {
1147                         return ((gensec_gssapi_state->sasl_protection & NEG_SEAL) 
1148                                  && (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG));
1149                 }
1150                 return gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG;
1151         }
1152         if (feature & GENSEC_FEATURE_SESSION_KEY) {
1153                 /* Only for GSSAPI/Krb5 */
1154                 if (gss_oid_equal(gensec_gssapi_state->gss_oid, gss_mech_krb5)) {
1155                         return true;
1156                 }
1157         }
1158         if (feature & GENSEC_FEATURE_DCE_STYLE) {
1159                 return gensec_gssapi_state->got_flags & GSS_C_DCE_STYLE;
1160         }
1161         if (feature & GENSEC_FEATURE_NEW_SPNEGO) {
1162                 NTSTATUS status;
1163
1164                 if (!(gensec_gssapi_state->got_flags & GSS_C_INTEG_FLAG)) {
1165                         return false;
1166                 }
1167
1168                 if (gensec_setting_bool(gensec_security->settings, "gensec_gssapi", "force_new_spnego", false)) {
1169                         return true;
1170                 }
1171                 if (gensec_setting_bool(gensec_security->settings, "gensec_gssapi", "disable_new_spnego", false)) {
1172                         return false;
1173                 }
1174
1175                 status = gensec_gssapi_init_lucid(gensec_gssapi_state);
1176                 if (!NT_STATUS_IS_OK(status)) {
1177                         return false;
1178                 }
1179
1180                 if (gensec_gssapi_state->lucid->protocol == 1) {
1181                         return true;
1182                 }
1183
1184                 return false;
1185         }
1186         /* We can always do async (rather than strict request/reply) packets.  */
1187         if (feature & GENSEC_FEATURE_ASYNC_REPLIES) {
1188                 return true;
1189         }
1190         return false;
1191 }
1192
1193 /*
1194  * Extract the 'sesssion key' needed by SMB signing and ncacn_np 
1195  * (for encrypting some passwords).
1196  * 
1197  * This breaks all the abstractions, but what do you expect...
1198  */
1199 static NTSTATUS gensec_gssapi_session_key(struct gensec_security *gensec_security, 
1200                                           DATA_BLOB *session_key) 
1201 {
1202         struct gensec_gssapi_state *gensec_gssapi_state
1203                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1204         OM_uint32 maj_stat, min_stat;
1205         krb5_keyblock *subkey;
1206
1207         if (gensec_gssapi_state->sasl_state != STAGE_DONE) {
1208                 return NT_STATUS_NO_USER_SESSION_KEY;
1209         }
1210
1211         if (gensec_gssapi_state->session_key.data) {
1212                 *session_key = gensec_gssapi_state->session_key;
1213                 return NT_STATUS_OK;
1214         }
1215
1216         maj_stat = gsskrb5_get_subkey(&min_stat,
1217                                       gensec_gssapi_state->gssapi_context,
1218                                       &subkey);
1219         if (maj_stat != 0) {
1220                 DEBUG(1, ("NO session key for this mech\n"));
1221                 return NT_STATUS_NO_USER_SESSION_KEY;
1222         }
1223         
1224         DEBUG(10, ("Got KRB5 session key of length %d%s\n",
1225                    (int)KRB5_KEY_LENGTH(subkey),
1226                    (gensec_gssapi_state->sasl_state == STAGE_DONE)?" (done)":""));
1227         *session_key = data_blob_talloc(gensec_gssapi_state,
1228                                         KRB5_KEY_DATA(subkey), KRB5_KEY_LENGTH(subkey));
1229         krb5_free_keyblock(gensec_gssapi_state->smb_krb5_context->krb5_context, subkey);
1230         gensec_gssapi_state->session_key = *session_key;
1231         dump_data_pw("KRB5 Session Key:\n", session_key->data, session_key->length);
1232
1233         return NT_STATUS_OK;
1234 }
1235
1236 /* Get some basic (and authorization) information about the user on
1237  * this session.  This uses either the PAC (if present) or a local
1238  * database lookup */
1239 static NTSTATUS gensec_gssapi_session_info(struct gensec_security *gensec_security,
1240                                            struct auth_session_info **_session_info) 
1241 {
1242         NTSTATUS nt_status;
1243         TALLOC_CTX *mem_ctx;
1244         struct gensec_gssapi_state *gensec_gssapi_state
1245                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1246         struct auth_serversupplied_info *server_info = NULL;
1247         struct auth_session_info *session_info = NULL;
1248         OM_uint32 maj_stat, min_stat;
1249         gss_buffer_desc pac;
1250         DATA_BLOB pac_blob;
1251         
1252         if ((gensec_gssapi_state->gss_oid->length != gss_mech_krb5->length)
1253             || (memcmp(gensec_gssapi_state->gss_oid->elements, gss_mech_krb5->elements, 
1254                        gensec_gssapi_state->gss_oid->length) != 0)) {
1255                 DEBUG(1, ("NO session info available for this mech\n"));
1256                 return NT_STATUS_INVALID_PARAMETER;
1257         }
1258                 
1259         mem_ctx = talloc_named(gensec_gssapi_state, 0, "gensec_gssapi_session_info context"); 
1260         NT_STATUS_HAVE_NO_MEMORY(mem_ctx);
1261
1262         maj_stat = gsskrb5_extract_authz_data_from_sec_context(&min_stat, 
1263                                                                gensec_gssapi_state->gssapi_context, 
1264                                                                KRB5_AUTHDATA_WIN2K_PAC,
1265                                                                &pac);
1266         
1267         
1268         if (maj_stat == 0) {
1269                 pac_blob = data_blob_talloc(mem_ctx, pac.value, pac.length);
1270                 gss_release_buffer(&min_stat, &pac);
1271
1272         } else {
1273                 pac_blob = data_blob(NULL, 0);
1274         }
1275         
1276         /* IF we have the PAC - otherwise we need to get this
1277          * data from elsewere - local ldb, or (TODO) lookup of some
1278          * kind... 
1279          */
1280         if (pac_blob.length) {
1281                 nt_status = kerberos_pac_blob_to_server_info(mem_ctx, 
1282                                                              pac_blob, 
1283                                                              gensec_gssapi_state->smb_krb5_context->krb5_context,
1284                                                              &server_info);
1285                 if (!NT_STATUS_IS_OK(nt_status)) {
1286                         talloc_free(mem_ctx);
1287                         return nt_status;
1288                 }
1289         } else {
1290                 gss_buffer_desc name_token;
1291                 char *principal_string;
1292
1293                 maj_stat = gss_display_name (&min_stat,
1294                                              gensec_gssapi_state->client_name,
1295                                              &name_token,
1296                                              NULL);
1297                 if (GSS_ERROR(maj_stat)) {
1298                         DEBUG(1, ("GSS display_name failed: %s\n", 
1299                                   gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1300                         talloc_free(mem_ctx);
1301                         return NT_STATUS_FOOBAR;
1302                 }
1303                 
1304                 principal_string = talloc_strndup(mem_ctx, 
1305                                                   (const char *)name_token.value, 
1306                                                   name_token.length);
1307                 
1308                 gss_release_buffer(&min_stat, &name_token);
1309                 
1310                 if (!principal_string) {
1311                         talloc_free(mem_ctx);
1312                         return NT_STATUS_NO_MEMORY;
1313                 }
1314
1315                 if (gensec_security->auth_context && 
1316                     !gensec_setting_bool(gensec_security->settings, "gensec", "require_pac", false)) {
1317                         DEBUG(1, ("Unable to find PAC, resorting to local user lookup: %s\n",
1318                                   gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1319                         nt_status = gensec_security->auth_context->get_server_info_principal(mem_ctx, 
1320                                                                                              gensec_security->auth_context, 
1321                                                                                              principal_string,
1322                                                                                              NULL,
1323                                                                                              &server_info);
1324                         
1325                         if (!NT_STATUS_IS_OK(nt_status)) {
1326                                 talloc_free(mem_ctx);
1327                                 return nt_status;
1328                         }
1329                 } else {
1330                         DEBUG(1, ("Unable to find PAC in ticket from %s, failing to allow access: %s\n",
1331                                   principal_string,
1332                                   gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1333                         return NT_STATUS_ACCESS_DENIED;
1334                 }
1335         }
1336
1337         /* references the server_info into the session_info */
1338         nt_status = gensec_generate_session_info(mem_ctx, gensec_security,
1339                                                  server_info, &session_info);
1340         if (!NT_STATUS_IS_OK(nt_status)) {
1341                 talloc_free(mem_ctx);
1342                 return nt_status;
1343         }
1344
1345         nt_status = gensec_gssapi_session_key(gensec_security, &session_info->session_key);
1346         if (!NT_STATUS_IS_OK(nt_status)) {
1347                 talloc_free(mem_ctx);
1348                 return nt_status;
1349         }
1350
1351         if (!(gensec_gssapi_state->got_flags & GSS_C_DELEG_FLAG)) {
1352                 DEBUG(10, ("gensec_gssapi: NO delegated credentials supplied by client\n"));
1353         } else {
1354                 krb5_error_code ret;
1355                 const char *error_string;
1356
1357                 DEBUG(10, ("gensec_gssapi: delegated credentials supplied by client\n"));
1358                 session_info->credentials = cli_credentials_init(session_info);
1359                 if (!session_info->credentials) {
1360                         talloc_free(mem_ctx);
1361                         return NT_STATUS_NO_MEMORY;
1362                 }
1363
1364                 cli_credentials_set_conf(session_info->credentials, gensec_security->settings->lp_ctx);
1365                 /* Just so we don't segfault trying to get at a username */
1366                 cli_credentials_set_anonymous(session_info->credentials);
1367                 
1368                 ret = cli_credentials_set_client_gss_creds(session_info->credentials, 
1369                                                            gensec_security->event_ctx,
1370                                                            gensec_security->settings->lp_ctx,
1371                                                            gensec_gssapi_state->delegated_cred_handle,
1372                                                            CRED_SPECIFIED, &error_string);
1373                 if (ret) {
1374                         talloc_free(mem_ctx);
1375                         DEBUG(2,("Failed to get gss creds: %s\n", error_string));
1376                         return NT_STATUS_NO_MEMORY;
1377                 }
1378                 
1379                 /* This credential handle isn't useful for password authentication, so ensure nobody tries to do that */
1380                 cli_credentials_set_kerberos_state(session_info->credentials, CRED_MUST_USE_KERBEROS);
1381
1382                 /* It has been taken from this place... */
1383                 gensec_gssapi_state->delegated_cred_handle = GSS_C_NO_CREDENTIAL;
1384         }
1385         talloc_steal(gensec_gssapi_state, session_info);
1386         talloc_free(mem_ctx);
1387         *_session_info = session_info;
1388
1389         return NT_STATUS_OK;
1390 }
1391
1392 static size_t gensec_gssapi_sig_size(struct gensec_security *gensec_security, size_t data_size)
1393 {
1394         struct gensec_gssapi_state *gensec_gssapi_state
1395                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1396         NTSTATUS status;
1397
1398         if (gensec_gssapi_state->sig_size) {
1399                 return gensec_gssapi_state->sig_size;
1400         }
1401
1402         if (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG) {
1403                 gensec_gssapi_state->sig_size = 45;
1404         } else {
1405                 gensec_gssapi_state->sig_size = 37;
1406         }
1407
1408         status = gensec_gssapi_init_lucid(gensec_gssapi_state);
1409         if (!NT_STATUS_IS_OK(status)) {
1410                 return gensec_gssapi_state->sig_size;
1411         }
1412
1413         if (gensec_gssapi_state->lucid->protocol == 1) {
1414                 if (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG) {
1415                         /*
1416                          * TODO: windows uses 76 here, but we don't know
1417                          *       gss_wrap works with aes keys yet
1418                          */
1419                         gensec_gssapi_state->sig_size = 76;
1420                 } else {
1421                         gensec_gssapi_state->sig_size = 28;
1422                 }
1423         } else if (gensec_gssapi_state->lucid->protocol == 0) {
1424                 switch (gensec_gssapi_state->lucid->rfc1964_kd.ctx_key.type) {
1425                 case KEYTYPE_DES:
1426                 case KEYTYPE_ARCFOUR:
1427                 case KEYTYPE_ARCFOUR_56:
1428                         if (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG) {
1429                                 gensec_gssapi_state->sig_size = 45;
1430                         } else {
1431                                 gensec_gssapi_state->sig_size = 37;
1432                         }
1433                         break;
1434                 case KEYTYPE_DES3:
1435                         if (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG) {
1436                                 gensec_gssapi_state->sig_size = 57;
1437                         } else {
1438                                 gensec_gssapi_state->sig_size = 49;
1439                         }
1440                         break;
1441                 }
1442         }
1443
1444         return gensec_gssapi_state->sig_size;
1445 }
1446
1447 static const char *gensec_gssapi_krb5_oids[] = { 
1448         GENSEC_OID_KERBEROS5_OLD,
1449         GENSEC_OID_KERBEROS5,
1450         NULL 
1451 };
1452
1453 static const char *gensec_gssapi_spnego_oids[] = { 
1454         GENSEC_OID_SPNEGO,
1455         NULL 
1456 };
1457
1458 /* As a server, this could in theory accept any GSSAPI mech */
1459 static const struct gensec_security_ops gensec_gssapi_spnego_security_ops = {
1460         .name           = "gssapi_spnego",
1461         .sasl_name      = "GSS-SPNEGO",
1462         .auth_type      = DCERPC_AUTH_TYPE_SPNEGO,
1463         .oid            = gensec_gssapi_spnego_oids,
1464         .client_start   = gensec_gssapi_client_start,
1465         .server_start   = gensec_gssapi_server_start,
1466         .magic          = gensec_gssapi_magic,
1467         .update         = gensec_gssapi_update,
1468         .session_key    = gensec_gssapi_session_key,
1469         .session_info   = gensec_gssapi_session_info,
1470         .sign_packet    = gensec_gssapi_sign_packet,
1471         .check_packet   = gensec_gssapi_check_packet,
1472         .seal_packet    = gensec_gssapi_seal_packet,
1473         .unseal_packet  = gensec_gssapi_unseal_packet,
1474         .wrap           = gensec_gssapi_wrap,
1475         .unwrap         = gensec_gssapi_unwrap,
1476         .have_feature   = gensec_gssapi_have_feature,
1477         .enabled        = false,
1478         .kerberos       = true,
1479         .priority       = GENSEC_GSSAPI
1480 };
1481
1482 /* As a server, this could in theory accept any GSSAPI mech */
1483 static const struct gensec_security_ops gensec_gssapi_krb5_security_ops = {
1484         .name           = "gssapi_krb5",
1485         .auth_type      = DCERPC_AUTH_TYPE_KRB5,
1486         .oid            = gensec_gssapi_krb5_oids,
1487         .client_start   = gensec_gssapi_client_start,
1488         .server_start   = gensec_gssapi_server_start,
1489         .magic          = gensec_gssapi_magic,
1490         .update         = gensec_gssapi_update,
1491         .session_key    = gensec_gssapi_session_key,
1492         .session_info   = gensec_gssapi_session_info,
1493         .sig_size       = gensec_gssapi_sig_size,
1494         .sign_packet    = gensec_gssapi_sign_packet,
1495         .check_packet   = gensec_gssapi_check_packet,
1496         .seal_packet    = gensec_gssapi_seal_packet,
1497         .unseal_packet  = gensec_gssapi_unseal_packet,
1498         .wrap           = gensec_gssapi_wrap,
1499         .unwrap         = gensec_gssapi_unwrap,
1500         .have_feature   = gensec_gssapi_have_feature,
1501         .enabled        = true,
1502         .kerberos       = true,
1503         .priority       = GENSEC_GSSAPI
1504 };
1505
1506 /* As a server, this could in theory accept any GSSAPI mech */
1507 static const struct gensec_security_ops gensec_gssapi_sasl_krb5_security_ops = {
1508         .name             = "gssapi_krb5_sasl",
1509         .sasl_name        = "GSSAPI",
1510         .client_start     = gensec_gssapi_sasl_client_start,
1511         .server_start     = gensec_gssapi_sasl_server_start,
1512         .update           = gensec_gssapi_update,
1513         .session_key      = gensec_gssapi_session_key,
1514         .session_info     = gensec_gssapi_session_info,
1515         .max_input_size   = gensec_gssapi_max_input_size,
1516         .max_wrapped_size = gensec_gssapi_max_wrapped_size,
1517         .wrap             = gensec_gssapi_wrap,
1518         .unwrap           = gensec_gssapi_unwrap,
1519         .have_feature     = gensec_gssapi_have_feature,
1520         .enabled          = true,
1521         .kerberos         = true,
1522         .priority         = GENSEC_GSSAPI
1523 };
1524
1525 _PUBLIC_ NTSTATUS gensec_gssapi_init(void)
1526 {
1527         NTSTATUS ret;
1528
1529         ret = gensec_register(&gensec_gssapi_spnego_security_ops);
1530         if (!NT_STATUS_IS_OK(ret)) {
1531                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
1532                         gensec_gssapi_spnego_security_ops.name));
1533                 return ret;
1534         }
1535
1536         ret = gensec_register(&gensec_gssapi_krb5_security_ops);
1537         if (!NT_STATUS_IS_OK(ret)) {
1538                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
1539                         gensec_gssapi_krb5_security_ops.name));
1540                 return ret;
1541         }
1542
1543         ret = gensec_register(&gensec_gssapi_sasl_krb5_security_ops);
1544         if (!NT_STATUS_IS_OK(ret)) {
1545                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
1546                         gensec_gssapi_sasl_krb5_security_ops.name));
1547                 return ret;
1548         }
1549
1550         return ret;
1551 }