s4-loadparm: 2nd half of lp_ to lpcfg_ conversion
[sfrench/samba-autobuild/.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 (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
787                         if (security_accepted & NEG_SEAL) {
788                                 gensec_gssapi_state->sasl_protection |= NEG_SEAL;
789                         }
790                 } else if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
791                         if (security_accepted & NEG_SIGN) {
792                                 gensec_gssapi_state->sasl_protection |= NEG_SIGN;
793                         }
794                 } else if (security_accepted & NEG_NONE) {
795                         gensec_gssapi_state->sasl_protection |= NEG_NONE;
796                 } else {
797                         DEBUG(1, ("Remote client does not support unprotected connections, but we failed to negotiate anything better"));
798                         return NT_STATUS_ACCESS_DENIED;
799                 }
800
801                 /* quirk:  This changes the value that gensec_have_feature returns, to be that after SASL negotiation */
802                 gensec_gssapi_state->sasl_state = STAGE_DONE;
803                 if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
804                         DEBUG(5, ("SASL/GSSAPI Connection from client will be cryptographicly sealed\n"));
805                 } else if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SIGN)) {
806                         DEBUG(5, ("SASL/GSSAPI Connection from client will be cryptographicly signed\n"));
807                 } else {
808                         DEBUG(5, ("SASL/GSSAPI Connection from client will have no cryptographic protection\n"));
809                 }
810
811                 *out = data_blob(NULL, 0);
812                 return NT_STATUS_OK;    
813         }
814         default:
815                 return NT_STATUS_INVALID_PARAMETER;
816         }
817 }
818
819 static NTSTATUS gensec_gssapi_wrap(struct gensec_security *gensec_security, 
820                                    TALLOC_CTX *mem_ctx, 
821                                    const DATA_BLOB *in, 
822                                    DATA_BLOB *out)
823 {
824         struct gensec_gssapi_state *gensec_gssapi_state
825                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
826         OM_uint32 maj_stat, min_stat;
827         gss_buffer_desc input_token, output_token;
828         int conf_state;
829         input_token.length = in->length;
830         input_token.value = in->data;
831
832         maj_stat = gss_wrap(&min_stat, 
833                             gensec_gssapi_state->gssapi_context, 
834                             gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL),
835                             GSS_C_QOP_DEFAULT,
836                             &input_token,
837                             &conf_state,
838                             &output_token);
839         if (GSS_ERROR(maj_stat)) {
840                 DEBUG(1, ("gensec_gssapi_wrap: GSS Wrap failed: %s\n", 
841                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
842                 return NT_STATUS_ACCESS_DENIED;
843         }
844
845         *out = data_blob_talloc(mem_ctx, output_token.value, output_token.length);
846         gss_release_buffer(&min_stat, &output_token);
847
848         if (gensec_gssapi_state->sasl) {
849                 size_t max_wrapped_size = gensec_gssapi_max_wrapped_size(gensec_security);
850                 if (max_wrapped_size < out->length) {
851                         DEBUG(1, ("gensec_gssapi_wrap: when wrapped, INPUT data (%u) is grew to be larger than SASL negotiated maximum output size (%u > %u)\n",
852                                   (unsigned)in->length, 
853                                   (unsigned)out->length, 
854                                   (unsigned int)max_wrapped_size));
855                         return NT_STATUS_INVALID_PARAMETER;
856                 }
857         }
858         
859         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)
860             && !conf_state) {
861                 return NT_STATUS_ACCESS_DENIED;
862         }
863         return NT_STATUS_OK;
864 }
865
866 static NTSTATUS gensec_gssapi_unwrap(struct gensec_security *gensec_security, 
867                                      TALLOC_CTX *mem_ctx, 
868                                      const DATA_BLOB *in, 
869                                      DATA_BLOB *out)
870 {
871         struct gensec_gssapi_state *gensec_gssapi_state
872                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
873         OM_uint32 maj_stat, min_stat;
874         gss_buffer_desc input_token, output_token;
875         int conf_state;
876         gss_qop_t qop_state;
877         input_token.length = in->length;
878         input_token.value = in->data;
879         
880         if (gensec_gssapi_state->sasl) {
881                 size_t max_wrapped_size = gensec_gssapi_max_wrapped_size(gensec_security);
882                 if (max_wrapped_size < in->length) {
883                         DEBUG(1, ("gensec_gssapi_unwrap: WRAPPED data is larger than SASL negotiated maximum size\n"));
884                         return NT_STATUS_INVALID_PARAMETER;
885                 }
886         }
887         
888         maj_stat = gss_unwrap(&min_stat, 
889                               gensec_gssapi_state->gssapi_context, 
890                               &input_token,
891                               &output_token, 
892                               &conf_state,
893                               &qop_state);
894         if (GSS_ERROR(maj_stat)) {
895                 DEBUG(1, ("gensec_gssapi_unwrap: GSS UnWrap failed: %s\n", 
896                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
897                 return NT_STATUS_ACCESS_DENIED;
898         }
899
900         *out = data_blob_talloc(mem_ctx, output_token.value, output_token.length);
901         gss_release_buffer(&min_stat, &output_token);
902         
903         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)
904             && !conf_state) {
905                 return NT_STATUS_ACCESS_DENIED;
906         }
907         return NT_STATUS_OK;
908 }
909
910 /* Find out the maximum input size negotiated on this connection */
911
912 static size_t gensec_gssapi_max_input_size(struct gensec_security *gensec_security) 
913 {
914         struct gensec_gssapi_state *gensec_gssapi_state
915                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
916         OM_uint32 maj_stat, min_stat;
917         OM_uint32 max_input_size;
918
919         maj_stat = gss_wrap_size_limit(&min_stat, 
920                                        gensec_gssapi_state->gssapi_context,
921                                        gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL),
922                                        GSS_C_QOP_DEFAULT,
923                                        gensec_gssapi_state->max_wrap_buf_size,
924                                        &max_input_size);
925         if (GSS_ERROR(maj_stat)) {
926                 TALLOC_CTX *mem_ctx = talloc_new(NULL); 
927                 DEBUG(1, ("gensec_gssapi_max_input_size: determinaing signature size with gss_wrap_size_limit failed: %s\n", 
928                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
929                 talloc_free(mem_ctx);
930                 return 0;
931         }
932
933         return max_input_size;
934 }
935
936 /* Find out the maximum output size negotiated on this connection */
937 static size_t gensec_gssapi_max_wrapped_size(struct gensec_security *gensec_security) 
938 {
939         struct gensec_gssapi_state *gensec_gssapi_state = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);;
940         return gensec_gssapi_state->max_wrap_buf_size;
941 }
942
943 static NTSTATUS gensec_gssapi_seal_packet(struct gensec_security *gensec_security, 
944                                           TALLOC_CTX *mem_ctx, 
945                                           uint8_t *data, size_t length, 
946                                           const uint8_t *whole_pdu, size_t pdu_length, 
947                                           DATA_BLOB *sig)
948 {
949         struct gensec_gssapi_state *gensec_gssapi_state
950                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
951         OM_uint32 maj_stat, min_stat;
952         gss_buffer_desc input_token, output_token;
953         int conf_state;
954         ssize_t sig_length;
955
956         input_token.length = length;
957         input_token.value = data;
958         
959         maj_stat = gss_wrap(&min_stat, 
960                             gensec_gssapi_state->gssapi_context,
961                             gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL),
962                             GSS_C_QOP_DEFAULT,
963                             &input_token,
964                             &conf_state,
965                             &output_token);
966         if (GSS_ERROR(maj_stat)) {
967                 DEBUG(1, ("gensec_gssapi_seal_packet: GSS Wrap failed: %s\n", 
968                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
969                 return NT_STATUS_ACCESS_DENIED;
970         }
971
972         if (output_token.length < input_token.length) {
973                 DEBUG(1, ("gensec_gssapi_seal_packet: GSS Wrap length [%ld] *less* than caller length [%ld]\n", 
974                           (long)output_token.length, (long)length));
975                 return NT_STATUS_INTERNAL_ERROR;
976         }
977         sig_length = output_token.length - input_token.length;
978
979         memcpy(data, ((uint8_t *)output_token.value) + sig_length, length);
980         *sig = data_blob_talloc(mem_ctx, (uint8_t *)output_token.value, sig_length);
981
982         dump_data_pw("gensec_gssapi_seal_packet: sig\n", sig->data, sig->length);
983         dump_data_pw("gensec_gssapi_seal_packet: clear\n", data, length);
984         dump_data_pw("gensec_gssapi_seal_packet: sealed\n", ((uint8_t *)output_token.value) + sig_length, output_token.length - sig_length);
985
986         gss_release_buffer(&min_stat, &output_token);
987
988         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)
989             && !conf_state) {
990                 return NT_STATUS_ACCESS_DENIED;
991         }
992         return NT_STATUS_OK;
993 }
994
995 static NTSTATUS gensec_gssapi_unseal_packet(struct gensec_security *gensec_security, 
996                                             TALLOC_CTX *mem_ctx, 
997                                             uint8_t *data, size_t length, 
998                                             const uint8_t *whole_pdu, size_t pdu_length,
999                                             const DATA_BLOB *sig)
1000 {
1001         struct gensec_gssapi_state *gensec_gssapi_state
1002                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1003         OM_uint32 maj_stat, min_stat;
1004         gss_buffer_desc input_token, output_token;
1005         int conf_state;
1006         gss_qop_t qop_state;
1007         DATA_BLOB in;
1008
1009         dump_data_pw("gensec_gssapi_unseal_packet: sig\n", sig->data, sig->length);
1010
1011         in = data_blob_talloc(mem_ctx, NULL, sig->length + length);
1012
1013         memcpy(in.data, sig->data, sig->length);
1014         memcpy(in.data + sig->length, data, length);
1015
1016         input_token.length = in.length;
1017         input_token.value = in.data;
1018         
1019         maj_stat = gss_unwrap(&min_stat, 
1020                               gensec_gssapi_state->gssapi_context, 
1021                               &input_token,
1022                               &output_token, 
1023                               &conf_state,
1024                               &qop_state);
1025         if (GSS_ERROR(maj_stat)) {
1026                 DEBUG(1, ("gensec_gssapi_unseal_packet: GSS UnWrap failed: %s\n", 
1027                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1028                 return NT_STATUS_ACCESS_DENIED;
1029         }
1030
1031         if (output_token.length != length) {
1032                 return NT_STATUS_INTERNAL_ERROR;
1033         }
1034
1035         memcpy(data, output_token.value, length);
1036
1037         gss_release_buffer(&min_stat, &output_token);
1038         
1039         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)
1040             && !conf_state) {
1041                 return NT_STATUS_ACCESS_DENIED;
1042         }
1043         return NT_STATUS_OK;
1044 }
1045
1046 static NTSTATUS gensec_gssapi_sign_packet(struct gensec_security *gensec_security, 
1047                                           TALLOC_CTX *mem_ctx, 
1048                                           const uint8_t *data, size_t length, 
1049                                           const uint8_t *whole_pdu, size_t pdu_length, 
1050                                           DATA_BLOB *sig)
1051 {
1052         struct gensec_gssapi_state *gensec_gssapi_state
1053                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1054         OM_uint32 maj_stat, min_stat;
1055         gss_buffer_desc input_token, output_token;
1056
1057         if (gensec_security->want_features & GENSEC_FEATURE_SIGN_PKT_HEADER) {
1058                 input_token.length = pdu_length;
1059                 input_token.value = discard_const_p(uint8_t *, whole_pdu);
1060         } else {
1061                 input_token.length = length;
1062                 input_token.value = discard_const_p(uint8_t *, data);
1063         }
1064
1065         maj_stat = gss_get_mic(&min_stat,
1066                             gensec_gssapi_state->gssapi_context,
1067                             GSS_C_QOP_DEFAULT,
1068                             &input_token,
1069                             &output_token);
1070         if (GSS_ERROR(maj_stat)) {
1071                 DEBUG(1, ("GSS GetMic failed: %s\n",
1072                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1073                 return NT_STATUS_ACCESS_DENIED;
1074         }
1075
1076         *sig = data_blob_talloc(mem_ctx, (uint8_t *)output_token.value, output_token.length);
1077
1078         dump_data_pw("gensec_gssapi_seal_packet: sig\n", sig->data, sig->length);
1079
1080         gss_release_buffer(&min_stat, &output_token);
1081
1082         return NT_STATUS_OK;
1083 }
1084
1085 static NTSTATUS gensec_gssapi_check_packet(struct gensec_security *gensec_security, 
1086                                            TALLOC_CTX *mem_ctx, 
1087                                            const uint8_t *data, size_t length, 
1088                                            const uint8_t *whole_pdu, size_t pdu_length, 
1089                                            const DATA_BLOB *sig)
1090 {
1091         struct gensec_gssapi_state *gensec_gssapi_state
1092                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1093         OM_uint32 maj_stat, min_stat;
1094         gss_buffer_desc input_token;
1095         gss_buffer_desc input_message;
1096         gss_qop_t qop_state;
1097
1098         dump_data_pw("gensec_gssapi_seal_packet: sig\n", sig->data, sig->length);
1099
1100         if (gensec_security->want_features & GENSEC_FEATURE_SIGN_PKT_HEADER) {
1101                 input_message.length = pdu_length;
1102                 input_message.value = discard_const(whole_pdu);
1103         } else {
1104                 input_message.length = length;
1105                 input_message.value = discard_const(data);
1106         }
1107
1108         input_token.length = sig->length;
1109         input_token.value = sig->data;
1110
1111         maj_stat = gss_verify_mic(&min_stat,
1112                               gensec_gssapi_state->gssapi_context, 
1113                               &input_message,
1114                               &input_token,
1115                               &qop_state);
1116         if (GSS_ERROR(maj_stat)) {
1117                 DEBUG(1, ("GSS VerifyMic failed: %s\n",
1118                           gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1119                 return NT_STATUS_ACCESS_DENIED;
1120         }
1121
1122         return NT_STATUS_OK;
1123 }
1124
1125 /* Try to figure out what features we actually got on the connection */
1126 static bool gensec_gssapi_have_feature(struct gensec_security *gensec_security, 
1127                                        uint32_t feature) 
1128 {
1129         struct gensec_gssapi_state *gensec_gssapi_state
1130                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1131         if (feature & GENSEC_FEATURE_SIGN) {
1132                 /* If we are going GSSAPI SASL, then we honour the second negotiation */
1133                 if (gensec_gssapi_state->sasl 
1134                     && gensec_gssapi_state->sasl_state == STAGE_DONE) {
1135                         return ((gensec_gssapi_state->sasl_protection & NEG_SIGN) 
1136                                 && (gensec_gssapi_state->got_flags & GSS_C_INTEG_FLAG));
1137                 }
1138                 return gensec_gssapi_state->got_flags & GSS_C_INTEG_FLAG;
1139         }
1140         if (feature & GENSEC_FEATURE_SEAL) {
1141                 /* If we are going GSSAPI SASL, then we honour the second negotiation */
1142                 if (gensec_gssapi_state->sasl 
1143                     && gensec_gssapi_state->sasl_state == STAGE_DONE) {
1144                         return ((gensec_gssapi_state->sasl_protection & NEG_SEAL) 
1145                                  && (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG));
1146                 }
1147                 return gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG;
1148         }
1149         if (feature & GENSEC_FEATURE_SESSION_KEY) {
1150                 /* Only for GSSAPI/Krb5 */
1151                 if (gss_oid_equal(gensec_gssapi_state->gss_oid, gss_mech_krb5)) {
1152                         return true;
1153                 }
1154         }
1155         if (feature & GENSEC_FEATURE_DCE_STYLE) {
1156                 return gensec_gssapi_state->got_flags & GSS_C_DCE_STYLE;
1157         }
1158         if (feature & GENSEC_FEATURE_NEW_SPNEGO) {
1159                 NTSTATUS status;
1160
1161                 if (!(gensec_gssapi_state->got_flags & GSS_C_INTEG_FLAG)) {
1162                         return false;
1163                 }
1164
1165                 if (gensec_setting_bool(gensec_security->settings, "gensec_gssapi", "force_new_spnego", false)) {
1166                         return true;
1167                 }
1168                 if (gensec_setting_bool(gensec_security->settings, "gensec_gssapi", "disable_new_spnego", false)) {
1169                         return false;
1170                 }
1171
1172                 status = gensec_gssapi_init_lucid(gensec_gssapi_state);
1173                 if (!NT_STATUS_IS_OK(status)) {
1174                         return false;
1175                 }
1176
1177                 if (gensec_gssapi_state->lucid->protocol == 1) {
1178                         return true;
1179                 }
1180
1181                 return false;
1182         }
1183         /* We can always do async (rather than strict request/reply) packets.  */
1184         if (feature & GENSEC_FEATURE_ASYNC_REPLIES) {
1185                 return true;
1186         }
1187         return false;
1188 }
1189
1190 /*
1191  * Extract the 'sesssion key' needed by SMB signing and ncacn_np 
1192  * (for encrypting some passwords).
1193  * 
1194  * This breaks all the abstractions, but what do you expect...
1195  */
1196 static NTSTATUS gensec_gssapi_session_key(struct gensec_security *gensec_security, 
1197                                           DATA_BLOB *session_key) 
1198 {
1199         struct gensec_gssapi_state *gensec_gssapi_state
1200                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1201         OM_uint32 maj_stat, min_stat;
1202         krb5_keyblock *subkey;
1203
1204         if (gensec_gssapi_state->sasl_state != STAGE_DONE) {
1205                 return NT_STATUS_NO_USER_SESSION_KEY;
1206         }
1207
1208         if (gensec_gssapi_state->session_key.data) {
1209                 *session_key = gensec_gssapi_state->session_key;
1210                 return NT_STATUS_OK;
1211         }
1212
1213         maj_stat = gsskrb5_get_subkey(&min_stat,
1214                                       gensec_gssapi_state->gssapi_context,
1215                                       &subkey);
1216         if (maj_stat != 0) {
1217                 DEBUG(1, ("NO session key for this mech\n"));
1218                 return NT_STATUS_NO_USER_SESSION_KEY;
1219         }
1220         
1221         DEBUG(10, ("Got KRB5 session key of length %d%s\n",
1222                    (int)KRB5_KEY_LENGTH(subkey),
1223                    (gensec_gssapi_state->sasl_state == STAGE_DONE)?" (done)":""));
1224         *session_key = data_blob_talloc(gensec_gssapi_state,
1225                                         KRB5_KEY_DATA(subkey), KRB5_KEY_LENGTH(subkey));
1226         krb5_free_keyblock(gensec_gssapi_state->smb_krb5_context->krb5_context, subkey);
1227         gensec_gssapi_state->session_key = *session_key;
1228         dump_data_pw("KRB5 Session Key:\n", session_key->data, session_key->length);
1229
1230         return NT_STATUS_OK;
1231 }
1232
1233 /* Get some basic (and authorization) information about the user on
1234  * this session.  This uses either the PAC (if present) or a local
1235  * database lookup */
1236 static NTSTATUS gensec_gssapi_session_info(struct gensec_security *gensec_security,
1237                                            struct auth_session_info **_session_info) 
1238 {
1239         NTSTATUS nt_status;
1240         TALLOC_CTX *mem_ctx;
1241         struct gensec_gssapi_state *gensec_gssapi_state
1242                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1243         struct auth_serversupplied_info *server_info = NULL;
1244         struct auth_session_info *session_info = NULL;
1245         OM_uint32 maj_stat, min_stat;
1246         gss_buffer_desc pac;
1247         DATA_BLOB pac_blob;
1248         
1249         if ((gensec_gssapi_state->gss_oid->length != gss_mech_krb5->length)
1250             || (memcmp(gensec_gssapi_state->gss_oid->elements, gss_mech_krb5->elements, 
1251                        gensec_gssapi_state->gss_oid->length) != 0)) {
1252                 DEBUG(1, ("NO session info available for this mech\n"));
1253                 return NT_STATUS_INVALID_PARAMETER;
1254         }
1255                 
1256         mem_ctx = talloc_named(gensec_gssapi_state, 0, "gensec_gssapi_session_info context"); 
1257         NT_STATUS_HAVE_NO_MEMORY(mem_ctx);
1258
1259         maj_stat = gsskrb5_extract_authz_data_from_sec_context(&min_stat, 
1260                                                                gensec_gssapi_state->gssapi_context, 
1261                                                                KRB5_AUTHDATA_WIN2K_PAC,
1262                                                                &pac);
1263         
1264         
1265         if (maj_stat == 0) {
1266                 pac_blob = data_blob_talloc(mem_ctx, pac.value, pac.length);
1267                 gss_release_buffer(&min_stat, &pac);
1268
1269         } else {
1270                 pac_blob = data_blob(NULL, 0);
1271         }
1272         
1273         /* IF we have the PAC - otherwise we need to get this
1274          * data from elsewere - local ldb, or (TODO) lookup of some
1275          * kind... 
1276          */
1277         if (pac_blob.length) {
1278                 nt_status = kerberos_pac_blob_to_server_info(mem_ctx, 
1279                                                              pac_blob, 
1280                                                              gensec_gssapi_state->smb_krb5_context->krb5_context,
1281                                                              &server_info);
1282                 if (!NT_STATUS_IS_OK(nt_status)) {
1283                         talloc_free(mem_ctx);
1284                         return nt_status;
1285                 }
1286         } else {
1287                 gss_buffer_desc name_token;
1288                 char *principal_string;
1289
1290                 maj_stat = gss_display_name (&min_stat,
1291                                              gensec_gssapi_state->client_name,
1292                                              &name_token,
1293                                              NULL);
1294                 if (GSS_ERROR(maj_stat)) {
1295                         DEBUG(1, ("GSS display_name failed: %s\n", 
1296                                   gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1297                         talloc_free(mem_ctx);
1298                         return NT_STATUS_FOOBAR;
1299                 }
1300                 
1301                 principal_string = talloc_strndup(mem_ctx, 
1302                                                   (const char *)name_token.value, 
1303                                                   name_token.length);
1304                 
1305                 gss_release_buffer(&min_stat, &name_token);
1306                 
1307                 if (!principal_string) {
1308                         talloc_free(mem_ctx);
1309                         return NT_STATUS_NO_MEMORY;
1310                 }
1311
1312                 if (gensec_security->auth_context && 
1313                     !gensec_setting_bool(gensec_security->settings, "gensec", "require_pac", false)) {
1314                         DEBUG(1, ("Unable to find PAC, resorting to local user lookup: %s\n",
1315                                   gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1316                         nt_status = gensec_security->auth_context->get_server_info_principal(mem_ctx, 
1317                                                                                              gensec_security->auth_context, 
1318                                                                                              principal_string,
1319                                                                                              NULL,
1320                                                                                              &server_info);
1321                         
1322                         if (!NT_STATUS_IS_OK(nt_status)) {
1323                                 talloc_free(mem_ctx);
1324                                 return nt_status;
1325                         }
1326                 } else {
1327                         DEBUG(1, ("Unable to find PAC in ticket from %s, failing to allow access: %s\n",
1328                                   principal_string,
1329                                   gssapi_error_string(mem_ctx, maj_stat, min_stat, gensec_gssapi_state->gss_oid)));
1330                         return NT_STATUS_ACCESS_DENIED;
1331                 }
1332         }
1333
1334         /* references the server_info into the session_info */
1335         nt_status = gensec_generate_session_info(mem_ctx, gensec_security,
1336                                                  server_info, &session_info);
1337         if (!NT_STATUS_IS_OK(nt_status)) {
1338                 talloc_free(mem_ctx);
1339                 return nt_status;
1340         }
1341
1342         nt_status = gensec_gssapi_session_key(gensec_security, &session_info->session_key);
1343         if (!NT_STATUS_IS_OK(nt_status)) {
1344                 talloc_free(mem_ctx);
1345                 return nt_status;
1346         }
1347
1348         if (!(gensec_gssapi_state->got_flags & GSS_C_DELEG_FLAG)) {
1349                 DEBUG(10, ("gensec_gssapi: NO delegated credentials supplied by client\n"));
1350         } else {
1351                 krb5_error_code ret;
1352                 const char *error_string;
1353
1354                 DEBUG(10, ("gensec_gssapi: delegated credentials supplied by client\n"));
1355                 session_info->credentials = cli_credentials_init(session_info);
1356                 if (!session_info->credentials) {
1357                         talloc_free(mem_ctx);
1358                         return NT_STATUS_NO_MEMORY;
1359                 }
1360
1361                 cli_credentials_set_conf(session_info->credentials, gensec_security->settings->lp_ctx);
1362                 /* Just so we don't segfault trying to get at a username */
1363                 cli_credentials_set_anonymous(session_info->credentials);
1364                 
1365                 ret = cli_credentials_set_client_gss_creds(session_info->credentials, 
1366                                                            gensec_security->event_ctx,
1367                                                            gensec_security->settings->lp_ctx,
1368                                                            gensec_gssapi_state->delegated_cred_handle,
1369                                                            CRED_SPECIFIED, &error_string);
1370                 if (ret) {
1371                         talloc_free(mem_ctx);
1372                         DEBUG(2,("Failed to get gss creds: %s\n", error_string));
1373                         return NT_STATUS_NO_MEMORY;
1374                 }
1375                 
1376                 /* This credential handle isn't useful for password authentication, so ensure nobody tries to do that */
1377                 cli_credentials_set_kerberos_state(session_info->credentials, CRED_MUST_USE_KERBEROS);
1378
1379                 /* It has been taken from this place... */
1380                 gensec_gssapi_state->delegated_cred_handle = GSS_C_NO_CREDENTIAL;
1381         }
1382         talloc_steal(gensec_gssapi_state, session_info);
1383         talloc_free(mem_ctx);
1384         *_session_info = session_info;
1385
1386         return NT_STATUS_OK;
1387 }
1388
1389 static size_t gensec_gssapi_sig_size(struct gensec_security *gensec_security, size_t data_size)
1390 {
1391         struct gensec_gssapi_state *gensec_gssapi_state
1392                 = talloc_get_type(gensec_security->private_data, struct gensec_gssapi_state);
1393         NTSTATUS status;
1394
1395         if (gensec_gssapi_state->sig_size) {
1396                 return gensec_gssapi_state->sig_size;
1397         }
1398
1399         if (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG) {
1400                 gensec_gssapi_state->sig_size = 45;
1401         } else {
1402                 gensec_gssapi_state->sig_size = 37;
1403         }
1404
1405         status = gensec_gssapi_init_lucid(gensec_gssapi_state);
1406         if (!NT_STATUS_IS_OK(status)) {
1407                 return gensec_gssapi_state->sig_size;
1408         }
1409
1410         if (gensec_gssapi_state->lucid->protocol == 1) {
1411                 if (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG) {
1412                         /*
1413                          * TODO: windows uses 76 here, but we don't know
1414                          *       gss_wrap works with aes keys yet
1415                          */
1416                         gensec_gssapi_state->sig_size = 76;
1417                 } else {
1418                         gensec_gssapi_state->sig_size = 28;
1419                 }
1420         } else if (gensec_gssapi_state->lucid->protocol == 0) {
1421                 switch (gensec_gssapi_state->lucid->rfc1964_kd.ctx_key.type) {
1422                 case KEYTYPE_DES:
1423                 case KEYTYPE_ARCFOUR:
1424                 case KEYTYPE_ARCFOUR_56:
1425                         if (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG) {
1426                                 gensec_gssapi_state->sig_size = 45;
1427                         } else {
1428                                 gensec_gssapi_state->sig_size = 37;
1429                         }
1430                         break;
1431                 case KEYTYPE_DES3:
1432                         if (gensec_gssapi_state->got_flags & GSS_C_CONF_FLAG) {
1433                                 gensec_gssapi_state->sig_size = 57;
1434                         } else {
1435                                 gensec_gssapi_state->sig_size = 49;
1436                         }
1437                         break;
1438                 }
1439         }
1440
1441         return gensec_gssapi_state->sig_size;
1442 }
1443
1444 static const char *gensec_gssapi_krb5_oids[] = { 
1445         GENSEC_OID_KERBEROS5_OLD,
1446         GENSEC_OID_KERBEROS5,
1447         NULL 
1448 };
1449
1450 static const char *gensec_gssapi_spnego_oids[] = { 
1451         GENSEC_OID_SPNEGO,
1452         NULL 
1453 };
1454
1455 /* As a server, this could in theory accept any GSSAPI mech */
1456 static const struct gensec_security_ops gensec_gssapi_spnego_security_ops = {
1457         .name           = "gssapi_spnego",
1458         .sasl_name      = "GSS-SPNEGO",
1459         .auth_type      = DCERPC_AUTH_TYPE_SPNEGO,
1460         .oid            = gensec_gssapi_spnego_oids,
1461         .client_start   = gensec_gssapi_client_start,
1462         .server_start   = gensec_gssapi_server_start,
1463         .magic          = gensec_gssapi_magic,
1464         .update         = gensec_gssapi_update,
1465         .session_key    = gensec_gssapi_session_key,
1466         .session_info   = gensec_gssapi_session_info,
1467         .sign_packet    = gensec_gssapi_sign_packet,
1468         .check_packet   = gensec_gssapi_check_packet,
1469         .seal_packet    = gensec_gssapi_seal_packet,
1470         .unseal_packet  = gensec_gssapi_unseal_packet,
1471         .wrap           = gensec_gssapi_wrap,
1472         .unwrap         = gensec_gssapi_unwrap,
1473         .have_feature   = gensec_gssapi_have_feature,
1474         .enabled        = false,
1475         .kerberos       = true,
1476         .priority       = GENSEC_GSSAPI
1477 };
1478
1479 /* As a server, this could in theory accept any GSSAPI mech */
1480 static const struct gensec_security_ops gensec_gssapi_krb5_security_ops = {
1481         .name           = "gssapi_krb5",
1482         .auth_type      = DCERPC_AUTH_TYPE_KRB5,
1483         .oid            = gensec_gssapi_krb5_oids,
1484         .client_start   = gensec_gssapi_client_start,
1485         .server_start   = gensec_gssapi_server_start,
1486         .magic          = gensec_gssapi_magic,
1487         .update         = gensec_gssapi_update,
1488         .session_key    = gensec_gssapi_session_key,
1489         .session_info   = gensec_gssapi_session_info,
1490         .sig_size       = gensec_gssapi_sig_size,
1491         .sign_packet    = gensec_gssapi_sign_packet,
1492         .check_packet   = gensec_gssapi_check_packet,
1493         .seal_packet    = gensec_gssapi_seal_packet,
1494         .unseal_packet  = gensec_gssapi_unseal_packet,
1495         .wrap           = gensec_gssapi_wrap,
1496         .unwrap         = gensec_gssapi_unwrap,
1497         .have_feature   = gensec_gssapi_have_feature,
1498         .enabled        = true,
1499         .kerberos       = true,
1500         .priority       = GENSEC_GSSAPI
1501 };
1502
1503 /* As a server, this could in theory accept any GSSAPI mech */
1504 static const struct gensec_security_ops gensec_gssapi_sasl_krb5_security_ops = {
1505         .name             = "gssapi_krb5_sasl",
1506         .sasl_name        = "GSSAPI",
1507         .client_start     = gensec_gssapi_sasl_client_start,
1508         .server_start     = gensec_gssapi_sasl_server_start,
1509         .update           = gensec_gssapi_update,
1510         .session_key      = gensec_gssapi_session_key,
1511         .session_info     = gensec_gssapi_session_info,
1512         .max_input_size   = gensec_gssapi_max_input_size,
1513         .max_wrapped_size = gensec_gssapi_max_wrapped_size,
1514         .wrap             = gensec_gssapi_wrap,
1515         .unwrap           = gensec_gssapi_unwrap,
1516         .have_feature     = gensec_gssapi_have_feature,
1517         .enabled          = true,
1518         .kerberos         = true,
1519         .priority         = GENSEC_GSSAPI
1520 };
1521
1522 _PUBLIC_ NTSTATUS gensec_gssapi_init(void)
1523 {
1524         NTSTATUS ret;
1525
1526         ret = gensec_register(&gensec_gssapi_spnego_security_ops);
1527         if (!NT_STATUS_IS_OK(ret)) {
1528                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
1529                         gensec_gssapi_spnego_security_ops.name));
1530                 return ret;
1531         }
1532
1533         ret = gensec_register(&gensec_gssapi_krb5_security_ops);
1534         if (!NT_STATUS_IS_OK(ret)) {
1535                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
1536                         gensec_gssapi_krb5_security_ops.name));
1537                 return ret;
1538         }
1539
1540         ret = gensec_register(&gensec_gssapi_sasl_krb5_security_ops);
1541         if (!NT_STATUS_IS_OK(ret)) {
1542                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
1543                         gensec_gssapi_sasl_krb5_security_ops.name));
1544                 return ret;
1545         }
1546
1547         return ret;
1548 }