r11400: fix compiler warnings
[idra/samba.git] / source4 / auth / gensec / gensec_krb5.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    Kerberos backend for GENSEC
5    
6    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004
7    Copyright (C) Andrew Tridgell 2001
8    Copyright (C) Luke Howard 2002-2003
9    Copyright (C) Stefan Metzmacher 2004-2005
10
11    This program is free software; you can redistribute it and/or modify
12    it under the terms of the GNU General Public License as published by
13    the Free Software Foundation; either version 2 of the License, or
14    (at your option) any later version.
15    
16    This program is distributed in the hope that it will be useful,
17    but WITHOUT ANY WARRANTY; without even the implied warranty of
18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19    GNU General Public License for more details.
20
21    
22    You should have received a copy of the GNU General Public License
23    along with this program; if not, write to the Free Software
24    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
25 */
26
27 #include "includes.h"
28 #include "system/kerberos.h"
29 #include "system/time.h"
30 #include "system/network.h"
31 #include "auth/kerberos/kerberos.h"
32 #include "librpc/gen_ndr/ndr_krb5pac.h"
33 #include "auth/auth.h"
34
35 enum GENSEC_KRB5_STATE {
36         GENSEC_KRB5_SERVER_START,
37         GENSEC_KRB5_CLIENT_START,
38         GENSEC_KRB5_CLIENT_MUTUAL_AUTH,
39         GENSEC_KRB5_DONE
40 };
41
42 struct gensec_krb5_state {
43         DATA_BLOB session_key;
44         DATA_BLOB pac;
45         enum GENSEC_KRB5_STATE state_position;
46         struct smb_krb5_context *smb_krb5_context;
47         krb5_auth_context auth_context;
48         krb5_data enc_ticket;
49         krb5_keyblock *keyblock;
50         krb5_ticket *ticket;
51         BOOL gssapi;
52 };
53
54 static int gensec_krb5_destroy(void *ptr) 
55 {
56         struct gensec_krb5_state *gensec_krb5_state = ptr;
57
58         if (!gensec_krb5_state->smb_krb5_context) {
59                 /* We can't clean anything else up unless we started up this far */
60                 return 0;
61         }
62         if (gensec_krb5_state->enc_ticket.length) { 
63                 kerberos_free_data_contents(gensec_krb5_state->smb_krb5_context->krb5_context, 
64                                             &gensec_krb5_state->enc_ticket); 
65         }
66
67         if (gensec_krb5_state->ticket) {
68                 krb5_free_ticket(gensec_krb5_state->smb_krb5_context->krb5_context, 
69                                  gensec_krb5_state->ticket);
70         }
71
72         /* ccache freed in a child destructor */
73
74         krb5_free_keyblock(gensec_krb5_state->smb_krb5_context->krb5_context, 
75                            gensec_krb5_state->keyblock);
76                 
77         if (gensec_krb5_state->auth_context) {
78                 krb5_auth_con_free(gensec_krb5_state->smb_krb5_context->krb5_context, 
79                                    gensec_krb5_state->auth_context);
80         }
81
82         return 0;
83 }
84
85 static NTSTATUS gensec_krb5_start(struct gensec_security *gensec_security)
86 {
87         krb5_error_code ret;
88         struct gensec_krb5_state *gensec_krb5_state;
89         struct cli_credentials *creds;
90
91         creds = gensec_get_credentials(gensec_security);
92         if (!creds) {
93                 return NT_STATUS_INVALID_PARAMETER;
94         }
95
96         gensec_krb5_state = talloc(gensec_security, struct gensec_krb5_state);
97         if (!gensec_krb5_state) {
98                 return NT_STATUS_NO_MEMORY;
99         }
100
101         gensec_security->private_data = gensec_krb5_state;
102         gensec_krb5_state->smb_krb5_context = NULL;
103         gensec_krb5_state->auth_context = NULL;
104         gensec_krb5_state->ticket = NULL;
105         ZERO_STRUCT(gensec_krb5_state->enc_ticket);
106         gensec_krb5_state->keyblock = NULL;
107         gensec_krb5_state->session_key = data_blob(NULL, 0);
108         gensec_krb5_state->pac = data_blob(NULL, 0);
109         gensec_krb5_state->gssapi = False;
110
111         talloc_set_destructor(gensec_krb5_state, gensec_krb5_destroy); 
112
113         if (cli_credentials_get_krb5_context(creds, &gensec_krb5_state->smb_krb5_context)) {
114                 talloc_free(gensec_krb5_state);
115                 return NT_STATUS_INTERNAL_ERROR;
116         }
117
118         ret = krb5_auth_con_init(gensec_krb5_state->smb_krb5_context->krb5_context, &gensec_krb5_state->auth_context);
119         if (ret) {
120                 DEBUG(1,("gensec_krb5_start: krb5_auth_con_init failed (%s)\n", 
121                          smb_get_krb5_error_message(gensec_krb5_state->smb_krb5_context->krb5_context, 
122                                                     ret, gensec_krb5_state)));
123                 talloc_free(gensec_krb5_state);
124                 return NT_STATUS_INTERNAL_ERROR;
125         }
126
127         ret = krb5_auth_con_setflags(gensec_krb5_state->smb_krb5_context->krb5_context, 
128                                      gensec_krb5_state->auth_context,
129                                      KRB5_AUTH_CONTEXT_DO_SEQUENCE);
130         if (ret) {
131                 DEBUG(1,("gensec_krb5_start: krb5_auth_con_setflags failed (%s)\n", 
132                          smb_get_krb5_error_message(gensec_krb5_state->smb_krb5_context->krb5_context, 
133                                                     ret, gensec_krb5_state)));
134                 talloc_free(gensec_krb5_state);
135                 return NT_STATUS_INTERNAL_ERROR;
136         }
137
138         return NT_STATUS_OK;
139 }
140
141 static NTSTATUS gensec_krb5_server_start(struct gensec_security *gensec_security)
142 {
143         NTSTATUS nt_status;
144         struct gensec_krb5_state *gensec_krb5_state;
145
146         nt_status = gensec_krb5_start(gensec_security);
147         if (!NT_STATUS_IS_OK(nt_status)) {
148                 return nt_status;
149         }
150         
151         gensec_krb5_state = gensec_security->private_data;
152         gensec_krb5_state->state_position = GENSEC_KRB5_SERVER_START;
153
154         return NT_STATUS_OK;
155 }
156
157 static NTSTATUS gensec_fake_gssapi_krb5_server_start(struct gensec_security *gensec_security)
158 {
159         NTSTATUS nt_status = gensec_krb5_server_start(gensec_security);
160
161         if (NT_STATUS_IS_OK(nt_status)) {
162                 struct gensec_krb5_state *gensec_krb5_state;
163                 gensec_krb5_state = gensec_security->private_data;
164                 gensec_krb5_state->gssapi = True;
165         }
166         return nt_status;
167 }
168
169 static NTSTATUS gensec_krb5_client_start(struct gensec_security *gensec_security)
170 {
171         struct gensec_krb5_state *gensec_krb5_state;
172         krb5_error_code ret;
173         NTSTATUS nt_status;
174         struct ccache_container *ccache_container;
175         const char *hostname;
176         krb5_flags ap_req_options = AP_OPTS_USE_SUBKEY | AP_OPTS_MUTUAL_REQUIRED;
177
178         const char *principal;
179         krb5_data in_data;
180
181         hostname = gensec_get_target_hostname(gensec_security);
182         if (!hostname) {
183                 DEBUG(1, ("Could not determine hostname for target computer, cannot use kerberos\n"));
184                 return NT_STATUS_INVALID_PARAMETER;
185         }
186         if (is_ipaddress(hostname)) {
187                 DEBUG(2, ("Cannot do krb5 to an IP address"));
188                 return NT_STATUS_INVALID_PARAMETER;
189         }
190         if (strequal(hostname, "localhost")) {
191                 DEBUG(2, ("krb5 to 'localhost' does not make sense"));
192                 return NT_STATUS_INVALID_PARAMETER;
193         }
194                         
195         nt_status = gensec_krb5_start(gensec_security);
196         if (!NT_STATUS_IS_OK(nt_status)) {
197                 return nt_status;
198         }
199
200         gensec_krb5_state = gensec_security->private_data;
201         gensec_krb5_state->state_position = GENSEC_KRB5_CLIENT_START;
202
203         ret = cli_credentials_get_ccache(gensec_get_credentials(gensec_security), &ccache_container);
204         if (ret) {
205                 DEBUG(1,("gensec_krb5_start: cli_credentials_get_ccache failed: %s\n", 
206                          error_message(ret)));
207                 return NT_STATUS_UNSUCCESSFUL;
208         }
209
210         in_data.length = 0;
211         
212         principal = gensec_get_target_principal(gensec_security);
213         if (principal && lp_client_use_spnego_principal()) {
214                 krb5_principal target_principal;
215                 ret = krb5_parse_name(gensec_krb5_state->smb_krb5_context->krb5_context, principal,
216                                       &target_principal);
217                 if (ret == 0) {
218                         ret = krb5_mk_req_exact(gensec_krb5_state->smb_krb5_context->krb5_context, 
219                                                 &gensec_krb5_state->auth_context,
220                                                 ap_req_options, 
221                                                 target_principal,
222                                                 &in_data, ccache_container->ccache, 
223                                                 &gensec_krb5_state->enc_ticket);
224                         krb5_free_principal(gensec_krb5_state->smb_krb5_context->krb5_context, 
225                                             target_principal);
226                 }
227         } else {
228                 ret = krb5_mk_req(gensec_krb5_state->smb_krb5_context->krb5_context, 
229                                   &gensec_krb5_state->auth_context,
230                                   ap_req_options,
231                                   gensec_get_target_service(gensec_security),
232                                   hostname,
233                                   &in_data, ccache_container->ccache, 
234                                   &gensec_krb5_state->enc_ticket);
235         }
236         switch (ret) {
237         case 0:
238                 return NT_STATUS_OK;
239         case KRB5KDC_ERR_S_PRINCIPAL_UNKNOWN:
240                 DEBUG(3, ("Server [%s] is not registered with our KDC: %s\n", 
241                           hostname, smb_get_krb5_error_message(gensec_krb5_state->smb_krb5_context->krb5_context, ret, gensec_krb5_state)));
242                 return NT_STATUS_INVALID_PARAMETER; /* Make SPNEGO ignore us, we can't go any further here */
243         case KRB5_KDC_UNREACH:
244                 DEBUG(3, ("Cannot reach a KDC we require to contact host [%s]: %s\n",
245                           hostname, smb_get_krb5_error_message(gensec_krb5_state->smb_krb5_context->krb5_context, ret, gensec_krb5_state)));
246                 return NT_STATUS_INVALID_PARAMETER; /* Make SPNEGO ignore us, we can't go any further here */
247         case KRB5KDC_ERR_PREAUTH_FAILED:
248         case KRB5KRB_AP_ERR_TKT_EXPIRED:
249         case KRB5_CC_END:
250                 /* Too much clock skew - we will need to kinit to re-skew the clock */
251         case KRB5KRB_AP_ERR_SKEW:
252         case KRB5_KDCREP_SKEW:
253         {
254                 DEBUG(3, ("kerberos (mk_req) failed: %s\n", 
255                           smb_get_krb5_error_message(gensec_krb5_state->smb_krb5_context->krb5_context, ret, gensec_krb5_state)));
256                 /* fall down to remaining code */
257         }
258         
259         
260         /* just don't print a message for these really ordinary messages */
261         case KRB5_FCC_NOFILE:
262         case KRB5_CC_NOTFOUND:
263         case ENOENT:
264                 
265                 return NT_STATUS_UNSUCCESSFUL;
266                 break;
267                 
268         default:
269                 DEBUG(0, ("kerberos: %s\n", 
270                           smb_get_krb5_error_message(gensec_krb5_state->smb_krb5_context->krb5_context, ret, gensec_krb5_state)));
271                 return NT_STATUS_UNSUCCESSFUL;
272         }
273 }
274
275 static NTSTATUS gensec_fake_gssapi_krb5_client_start(struct gensec_security *gensec_security)
276 {
277         NTSTATUS nt_status = gensec_krb5_client_start(gensec_security);
278
279         if (NT_STATUS_IS_OK(nt_status)) {
280                 struct gensec_krb5_state *gensec_krb5_state;
281                 gensec_krb5_state = gensec_security->private_data;
282                 gensec_krb5_state->gssapi = True;
283         }
284         return nt_status;
285 }
286
287 /**
288  * Check if the packet is one for this mechansim
289  * 
290  * @param gensec_security GENSEC state
291  * @param in The request, as a DATA_BLOB
292  * @return Error, INVALID_PARAMETER if it's not a packet for us
293  *                or NT_STATUS_OK if the packet is ok. 
294  */
295
296 static NTSTATUS gensec_fake_gssapi_krb5_magic(struct gensec_security *gensec_security, 
297                                   const DATA_BLOB *in) 
298 {
299         if (gensec_gssapi_check_oid(in, GENSEC_OID_KERBEROS5)) {
300                 return NT_STATUS_OK;
301         } else {
302                 return NT_STATUS_INVALID_PARAMETER;
303         }
304 }
305
306
307 /**
308  * Next state function for the Krb5 GENSEC mechanism
309  * 
310  * @param gensec_krb5_state KRB5 State
311  * @param out_mem_ctx The TALLOC_CTX for *out to be allocated on
312  * @param in The request, as a DATA_BLOB
313  * @param out The reply, as an talloc()ed DATA_BLOB, on *out_mem_ctx
314  * @return Error, MORE_PROCESSING_REQUIRED if a reply is sent, 
315  *                or NT_STATUS_OK if the user is authenticated. 
316  */
317
318 static NTSTATUS gensec_krb5_update(struct gensec_security *gensec_security, 
319                                    TALLOC_CTX *out_mem_ctx, 
320                                    const DATA_BLOB in, DATA_BLOB *out) 
321 {
322         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
323         krb5_error_code ret = 0;
324         NTSTATUS nt_status;
325
326         switch (gensec_krb5_state->state_position) {
327         case GENSEC_KRB5_CLIENT_START:
328         {
329                 DATA_BLOB unwrapped_out;
330                 
331                 if (gensec_krb5_state->gssapi) {
332                         unwrapped_out = data_blob_talloc(out_mem_ctx, gensec_krb5_state->enc_ticket.data, gensec_krb5_state->enc_ticket.length);
333                         
334                         /* wrap that up in a nice GSS-API wrapping */
335                         *out = gensec_gssapi_gen_krb5_wrap(out_mem_ctx, &unwrapped_out, TOK_ID_KRB_AP_REQ);
336                 } else {
337                         *out = data_blob_talloc(out_mem_ctx, gensec_krb5_state->enc_ticket.data, gensec_krb5_state->enc_ticket.length);
338                 }
339                 gensec_krb5_state->state_position = GENSEC_KRB5_CLIENT_MUTUAL_AUTH;
340                 nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
341                 return nt_status;
342         }
343                 
344         case GENSEC_KRB5_CLIENT_MUTUAL_AUTH:
345         {
346                 DATA_BLOB unwrapped_in;
347                 krb5_data inbuf;
348                 krb5_ap_rep_enc_part *repl = NULL;
349                 uint8_t tok_id[2];
350
351                 if (gensec_krb5_state->gssapi) {
352                         if (!gensec_gssapi_parse_krb5_wrap(out_mem_ctx, &in, &unwrapped_in, tok_id)) {
353                                 DEBUG(1,("gensec_gssapi_parse_krb5_wrap(mutual authentication) failed to parse\n"));
354                                 dump_data_pw("Mutual authentication message:\n", in.data, in.length);
355                                 return NT_STATUS_INVALID_PARAMETER;
356                         }
357                 } else {
358                         unwrapped_in = in;
359                 }
360                 /* TODO: check the tok_id */
361
362                 inbuf.data = unwrapped_in.data;
363                 inbuf.length = unwrapped_in.length;
364                 ret = krb5_rd_rep(gensec_krb5_state->smb_krb5_context->krb5_context, 
365                                   gensec_krb5_state->auth_context,
366                                   &inbuf, &repl);
367                 if (ret) {
368                         DEBUG(1,("krb5_rd_rep (mutual authentication) failed (%s)\n",
369                                  smb_get_krb5_error_message(gensec_krb5_state->smb_krb5_context->krb5_context, ret, out_mem_ctx)));
370                         dump_data_pw("Mutual authentication message:\n", inbuf.data, inbuf.length);
371                         nt_status = NT_STATUS_ACCESS_DENIED;
372                 } else {
373                         *out = data_blob(NULL, 0);
374                         nt_status = NT_STATUS_OK;
375                         gensec_krb5_state->state_position = GENSEC_KRB5_DONE;
376                 }
377                 if (repl) {
378                         krb5_free_ap_rep_enc_part(gensec_krb5_state->smb_krb5_context->krb5_context, repl);
379                 }
380                 return nt_status;
381         }
382
383         case GENSEC_KRB5_SERVER_START:
384         {
385                 DATA_BLOB unwrapped_in;
386                 DATA_BLOB unwrapped_out = data_blob(NULL, 0);
387                 uint8_t tok_id[2];
388
389                 if (!in.data) {
390                         return NT_STATUS_INVALID_PARAMETER;
391                 }       
392
393                 /* Parse the GSSAPI wrapping, if it's there... (win2k3 allows it to be omited) */
394                 if (gensec_krb5_state->gssapi
395                     && gensec_gssapi_parse_krb5_wrap(out_mem_ctx, &in, &unwrapped_in, tok_id)) {
396                         nt_status = ads_verify_ticket(out_mem_ctx, 
397                                                       gensec_krb5_state->smb_krb5_context,
398                                                       &gensec_krb5_state->auth_context, 
399                                                       gensec_get_credentials(gensec_security),
400                                                       gensec_get_target_service(gensec_security), &unwrapped_in, 
401                                                       &gensec_krb5_state->ticket, &unwrapped_out,
402                                                       &gensec_krb5_state->keyblock);
403                 } else {
404                         /* TODO: check the tok_id */
405                         nt_status = ads_verify_ticket(out_mem_ctx, 
406                                                       gensec_krb5_state->smb_krb5_context,
407                                                       &gensec_krb5_state->auth_context, 
408                                                       gensec_get_credentials(gensec_security),
409                                                       gensec_get_target_service(gensec_security), 
410                                                       &in, 
411                                                       &gensec_krb5_state->ticket, &unwrapped_out,
412                                                       &gensec_krb5_state->keyblock);
413                 }
414
415                 if (!NT_STATUS_IS_OK(nt_status)) {
416                         return nt_status;
417                 }
418
419                 if (NT_STATUS_IS_OK(nt_status)) {
420                         gensec_krb5_state->state_position = GENSEC_KRB5_DONE;
421                         /* wrap that up in a nice GSS-API wrapping */
422                         if (gensec_krb5_state->gssapi) {
423                                 *out = gensec_gssapi_gen_krb5_wrap(out_mem_ctx, &unwrapped_out, TOK_ID_KRB_AP_REP);
424                         } else {
425                                 *out = unwrapped_out;
426                         }
427                 }
428                 return nt_status;
429         }
430         case GENSEC_KRB5_DONE:
431                 return NT_STATUS_OK;
432         }
433         
434         return NT_STATUS_INVALID_PARAMETER;
435 }
436
437 static NTSTATUS gensec_krb5_session_key(struct gensec_security *gensec_security, 
438                                         DATA_BLOB *session_key) 
439 {
440         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
441         krb5_context context = gensec_krb5_state->smb_krb5_context->krb5_context;
442         krb5_auth_context auth_context = gensec_krb5_state->auth_context;
443         krb5_keyblock *skey;
444         krb5_error_code err = -1;
445
446         if (gensec_krb5_state->session_key.data) {
447                 *session_key = gensec_krb5_state->session_key;
448                 return NT_STATUS_OK;
449         }
450
451         switch (gensec_security->gensec_role) {
452         case GENSEC_CLIENT:
453                 err = krb5_auth_con_getlocalsubkey(context, auth_context, &skey);
454                 break;
455         case GENSEC_SERVER:
456                 err = krb5_auth_con_getremotesubkey(context, auth_context, &skey);
457                 break;
458         }
459         if (err == 0 && skey != NULL) {
460                 DEBUG(10, ("Got KRB5 session key of length %d\n",  
461                            (int)KRB5_KEY_LENGTH(skey)));
462                 gensec_krb5_state->session_key = data_blob_talloc(gensec_krb5_state, 
463                                                 KRB5_KEY_DATA(skey), KRB5_KEY_LENGTH(skey));
464                 *session_key = gensec_krb5_state->session_key;
465                 dump_data_pw("KRB5 Session Key:\n", session_key->data, session_key->length);
466
467                 krb5_free_keyblock(context, skey);
468                 return NT_STATUS_OK;
469         } else {
470                 DEBUG(10, ("KRB5 error getting session key %d\n", err));
471                 return NT_STATUS_NO_USER_SESSION_KEY;
472         }
473 }
474
475 static NTSTATUS gensec_krb5_session_info(struct gensec_security *gensec_security,
476                                          struct auth_session_info **_session_info) 
477 {
478         NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
479         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
480         krb5_context context = gensec_krb5_state->smb_krb5_context->krb5_context;
481         struct auth_serversupplied_info *server_info = NULL;
482         struct auth_session_info *session_info = NULL;
483         struct PAC_LOGON_INFO *logon_info;
484
485         krb5_principal client_principal;
486         
487         DATA_BLOB pac;
488         krb5_data pac_data;
489
490         krb5_error_code ret;
491
492         TALLOC_CTX *mem_ctx = talloc_new(gensec_security);
493         if (!mem_ctx) {
494                 return NT_STATUS_NO_MEMORY;
495         }
496         
497         ret = krb5_ticket_get_authorization_data_type(context, gensec_krb5_state->ticket, 
498                                                       KRB5_AUTHDATA_WIN2K_PAC, 
499                                                       &pac_data);
500         
501         if (ret) {
502                 DEBUG(5, ("krb5_ticket_get_authorization_data_type failed to find PAC: %s\n", 
503                           smb_get_krb5_error_message(context, 
504                                                      ret, mem_ctx)));
505         } else {
506                 pac = data_blob_talloc(mem_ctx, pac_data.data, pac_data.length);
507                 if (!pac.data) {
508                         return NT_STATUS_NO_MEMORY;
509                 }
510
511                 ret = krb5_ticket_get_client(context, gensec_krb5_state->ticket, &client_principal);
512                 if (ret) {
513                         DEBUG(5, ("krb5_ticket_get_client failed to get cleint principal: %s\n", 
514                                   smb_get_krb5_error_message(context, 
515                                                              ret, mem_ctx)));
516                         return NT_STATUS_NO_MEMORY;
517                 }
518                 
519                 /* decode and verify the pac */
520                 nt_status = kerberos_pac_logon_info(gensec_krb5_state, &logon_info, pac,
521                                                     gensec_krb5_state->smb_krb5_context->krb5_context,
522                                                     NULL, gensec_krb5_state->keyblock,
523                                                     client_principal,
524                                                     gensec_krb5_state->ticket->ticket.authtime);
525                 krb5_free_principal(context, client_principal);
526
527                 if (NT_STATUS_IS_OK(nt_status)) {
528                         union netr_Validation validation;
529                         validation.sam3 = &logon_info->info3;
530                         nt_status = make_server_info_netlogon_validation(gensec_krb5_state, 
531                                                                          NULL,
532                                                                          3, &validation,
533                                                                          &server_info); 
534                 }
535                 talloc_free(mem_ctx);
536         }
537
538                 
539                 
540         /* IF we have the PAC - otherwise we need to get this
541          * data from elsewere - local ldb, or (TODO) lookup of some
542          * kind... 
543          */
544         if (!NT_STATUS_IS_OK(nt_status)) {
545                 /* NO pac, or can't parse or verify it */
546                 char *principal_string;
547                 ret = krb5_ticket_get_client(context, gensec_krb5_state->ticket, &client_principal);
548                 if (ret) {
549                         DEBUG(5, ("krb5_ticket_get_client failed to get cleint principal: %s\n", 
550                                   smb_get_krb5_error_message(context, 
551                                                              ret, mem_ctx)));
552                         return NT_STATUS_NO_MEMORY;
553                 }
554                 
555                 ret = krb5_unparse_name(gensec_krb5_state->smb_krb5_context->krb5_context, 
556                                         client_principal, &principal_string);
557                 krb5_free_principal(context, client_principal);
558                 if (ret) {
559                         return NT_STATUS_NO_MEMORY;
560                 }
561
562                 nt_status = sam_get_server_info_principal(mem_ctx, principal_string,
563                                                           &server_info);
564                 free(principal_string);
565                 
566                 if (!NT_STATUS_IS_OK(nt_status)) {
567                         talloc_free(mem_ctx);
568                         return nt_status;
569                 }
570         }
571
572         /* references the server_info into the session_info */
573         nt_status = auth_generate_session_info(gensec_krb5_state, server_info, &session_info);
574         talloc_free(mem_ctx);
575
576         NT_STATUS_NOT_OK_RETURN(nt_status);
577
578         nt_status = gensec_krb5_session_key(gensec_security, &session_info->session_key);
579         NT_STATUS_NOT_OK_RETURN(nt_status);
580
581         *_session_info = session_info;
582
583         return NT_STATUS_OK;
584 }
585
586 static NTSTATUS gensec_krb5_wrap(struct gensec_security *gensec_security, 
587                                    TALLOC_CTX *mem_ctx, 
588                                    const DATA_BLOB *in, 
589                                    DATA_BLOB *out)
590 {
591         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
592         krb5_context context = gensec_krb5_state->smb_krb5_context->krb5_context;
593         krb5_auth_context auth_context = gensec_krb5_state->auth_context;
594         krb5_error_code ret;
595         krb5_data input, output;
596         input.length = in->length;
597         input.data = in->data;
598         
599         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
600                 ret = krb5_mk_priv(context, auth_context, &input, &output, NULL);
601                 if (ret) {
602                         DEBUG(1, ("krb5_mk_priv failed: %s\n", 
603                                   smb_get_krb5_error_message(gensec_krb5_state->smb_krb5_context->krb5_context, 
604                                                              ret, mem_ctx)));
605                         return NT_STATUS_ACCESS_DENIED;
606                 }
607                 *out = data_blob_talloc(mem_ctx, output.data, output.length);
608                 
609                 krb5_data_free(&output);
610         } else {
611                 return NT_STATUS_ACCESS_DENIED;
612         }
613         return NT_STATUS_OK;
614 }
615
616 static NTSTATUS gensec_krb5_unwrap(struct gensec_security *gensec_security, 
617                                      TALLOC_CTX *mem_ctx, 
618                                      const DATA_BLOB *in, 
619                                      DATA_BLOB *out)
620 {
621         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
622         krb5_context context = gensec_krb5_state->smb_krb5_context->krb5_context;
623         krb5_auth_context auth_context = gensec_krb5_state->auth_context;
624         krb5_error_code ret;
625         krb5_data input, output;
626         krb5_replay_data replay;
627         input.length = in->length;
628         input.data = in->data;
629         
630         if (gensec_have_feature(gensec_security, GENSEC_FEATURE_SEAL)) {
631                 ret = krb5_rd_priv(context, auth_context, &input, &output, &replay);
632                 if (ret) {
633                         DEBUG(1, ("krb5_rd_priv failed: %s\n", 
634                                   smb_get_krb5_error_message(gensec_krb5_state->smb_krb5_context->krb5_context, 
635                                                              ret, mem_ctx)));
636                         return NT_STATUS_ACCESS_DENIED;
637                 }
638                 *out = data_blob_talloc(mem_ctx, output.data, output.length);
639                 
640                 krb5_data_free(&output);
641         } else {
642                 return NT_STATUS_ACCESS_DENIED;
643         }
644         return NT_STATUS_OK;
645 }
646
647 static BOOL gensec_krb5_have_feature(struct gensec_security *gensec_security,
648                                      uint32_t feature)
649 {
650         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
651         if (feature & GENSEC_FEATURE_SESSION_KEY) {
652                 return True;
653         } 
654         if (!gensec_krb5_state->gssapi && 
655             (feature & GENSEC_FEATURE_SEAL)) {
656                 return True;
657         } 
658         
659         return False;
660 }
661
662 static const char *gensec_krb5_oids[] = { 
663         GENSEC_OID_KERBEROS5,
664         GENSEC_OID_KERBEROS5_OLD,
665         NULL 
666 };
667
668 static const struct gensec_security_ops gensec_fake_gssapi_krb5_security_ops = {
669         .name           = "fake_gssapi_krb5",
670         .auth_type      = DCERPC_AUTH_TYPE_KRB5,
671         .oid            = gensec_krb5_oids,
672         .client_start   = gensec_fake_gssapi_krb5_client_start,
673         .server_start   = gensec_fake_gssapi_krb5_server_start,
674         .update         = gensec_krb5_update,
675         .magic          = gensec_fake_gssapi_krb5_magic,
676         .session_key    = gensec_krb5_session_key,
677         .session_info   = gensec_krb5_session_info,
678         .have_feature   = gensec_krb5_have_feature,
679         .enabled        = False
680 };
681
682 static const struct gensec_security_ops gensec_krb5_security_ops = {
683         .name           = "krb5",
684         .client_start   = gensec_krb5_client_start,
685         .server_start   = gensec_krb5_server_start,
686         .update         = gensec_krb5_update,
687         .session_key    = gensec_krb5_session_key,
688         .session_info   = gensec_krb5_session_info,
689         .have_feature   = gensec_krb5_have_feature,
690         .wrap           = gensec_krb5_wrap,
691         .unwrap         = gensec_krb5_unwrap,
692         .enabled        = True
693 };
694
695 NTSTATUS gensec_krb5_init(void)
696 {
697         NTSTATUS ret;
698
699         ret = gensec_register(&gensec_krb5_security_ops);
700         if (!NT_STATUS_IS_OK(ret)) {
701                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
702                         gensec_krb5_security_ops.name));
703                 return ret;
704         }
705
706         ret = gensec_register(&gensec_fake_gssapi_krb5_security_ops);
707         if (!NT_STATUS_IS_OK(ret)) {
708                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
709                         gensec_krb5_security_ops.name));
710                 return ret;
711         }
712
713         return ret;
714 }