source4/libcli/auth/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
  10    This program is free software; you can redistribute it and/or modify
  11    it under the terms of the GNU General Public License as published by
  12    the Free Software Foundation; either version 2 of the License, or
  13    (at your option) any later version.
  14
  15    This program is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18    GNU General Public License for more details.
  19
  20
  21    You should have received a copy of the GNU General Public License
  22    along with this program; if not, write to the Free Software
  23    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  24 */
  25
  26 #include "includes.h"
  27 #include "librpc/gen_ndr/ndr_krb5pac.h"
  28
  29 #undef DBGC_CLASS
  30 #define DBGC_CLASS DBGC_AUTH
  31
  32 enum GENSEC_KRB5_STATE {
  33         GENSEC_KRB5_SERVER_START,
  34         GENSEC_KRB5_CLIENT_START,
  35         GENSEC_KRB5_CLIENT_MUTUAL_AUTH,
  36         GENSEC_KRB5_DONE
  37 };
  38
  39 struct gensec_krb5_state {
  40         DATA_BLOB session_key;
  41         struct PAC_LOGON_INFO *logon_info;
  42         enum GENSEC_KRB5_STATE state_position;
  43         krb5_context krb5_context;
  44         krb5_auth_context krb5_auth_context;
  45         krb5_ccache krb5_ccache;
  46         krb5_data ticket;
  47         krb5_keyblock krb5_keyblock;
  48         char *peer_principal;
  49 };
  50
  51 #ifdef KRB5_DO_VERIFY_PAC
  52 static NTSTATUS gensec_krb5_pac_checksum(DATA_BLOB pac_data,
  53                                             struct PAC_SIGNATURE_DATA *sig,
  54                                             struct gensec_krb5_state *gensec_krb5_state,
  55                                             uint32 keyusage)
  56 {
  57         krb5_error_code ret;
  58         krb5_crypto crypto;
  59         Checksum cksum;
  60         int i;
  61
  62         cksum.cksumtype         = (CKSUMTYPE)sig->type;
  63         cksum.checksum.length   = sizeof(sig->signature);
  64         cksum.checksum.data     = sig->signature;
  65
  66
  67         ret = krb5_crypto_init(gensec_krb5_state->krb5_context,
  68                                 &gensec_krb5_state->krb5_keyblock,
  69                                 0,
  70                                 &crypto);
  71         if (ret) {
  72                 DEBUG(0,("krb5_crypto_init() failed\n"));
  73                 return NT_STATUS_FOOBAR;
  74         }
  75         for (i=0; i < 40; i++) {
  76                 keyusage = i;
  77                 ret = krb5_verify_checksum(gensec_krb5_state->krb5_context,
  78                                            crypto,
  79                                            keyusage,
  80                                            pac_data.data,
  81                                            pac_data.length,
  82                                            &cksum);
  83                 if (!ret) {
  84                         DEBUG(0,("PAC Verified: keyusage: %d\n", keyusage));
  85                         break;
  86                 }
  87         }
  88         krb5_crypto_destroy(gensec_krb5_state->krb5_context, crypto);
  89
  90         if (ret) {
  91                 DEBUG(0,("NOT verifying PAC checksums yet!\n"));
  92                 //return NT_STATUS_LOGON_FAILURE;
  93         } else {
  94                 DEBUG(0,("PAC checksums verified!\n"));
  95         }
  96
  97         return NT_STATUS_OK;
  98 }
  99 #endif
 100
 101 static NTSTATUS gensec_krb5_decode_pac(TALLOC_CTX *mem_ctx,
 102                                 struct PAC_LOGON_INFO **logon_info_out,
 103                                 DATA_BLOB blob,
 104                                 struct gensec_krb5_state *gensec_krb5_state)
 105 {
 106         NTSTATUS status;
 107         struct PAC_SIGNATURE_DATA srv_sig;
 108         struct PAC_SIGNATURE_DATA *srv_sig_ptr;
 109         struct PAC_SIGNATURE_DATA kdc_sig;
 110         struct PAC_SIGNATURE_DATA *kdc_sig_ptr;
 111         struct PAC_LOGON_INFO *logon_info = NULL;
 112         struct PAC_DATA pac_data;
 113 #ifdef KRB5_DO_VERIFY_PAC
 114         DATA_BLOB tmp_blob = data_blob(NULL, 0);
 115 #endif
 116         int i;
 117
 118         status = ndr_pull_struct_blob(&blob, mem_ctx, &pac_data,
 119                                         (ndr_pull_flags_fn_t)ndr_pull_PAC_DATA);
 120         if (!NT_STATUS_IS_OK(status)) {
 121                 DEBUG(0,("can't parse the PAC\n"));
 122                 return status;
 123         }
 124         NDR_PRINT_DEBUG(PAC_DATA, &pac_data);
 125
 126         if (pac_data.num_buffers < 3) {
 127                 /* we need logon_ingo, service_key and kdc_key */
 128                 DEBUG(0,("less than 3 PAC buffers\n"));
 129                 return NT_STATUS_FOOBAR;
 130         }
 131
 132         for (i=0; i < pac_data.num_buffers; i++) {
 133                 switch (pac_data.buffers[i].type) {
 134                         case PAC_TYPE_LOGON_INFO:
 135                                 if (!pac_data.buffers[i].info) {
 136                                         break;
 137                                 }
 138                                 logon_info = &pac_data.buffers[i].info->logon_info;
 139                                 break;
 140                         case PAC_TYPE_SRV_CHECKSUM:
 141                                 if (!pac_data.buffers[i].info) {
 142                                         break;
 143                                 }
 144                                 srv_sig_ptr = &pac_data.buffers[i].info->srv_cksum;
 145                                 srv_sig = pac_data.buffers[i].info->srv_cksum;
 146                                 break;
 147                         case PAC_TYPE_KDC_CHECKSUM:
 148                                 if (!pac_data.buffers[i].info) {
 149                                         break;
 150                                 }
 151                                 kdc_sig_ptr = &pac_data.buffers[i].info->kdc_cksum;
 152                                 kdc_sig = pac_data.buffers[i].info->kdc_cksum;
 153                                 break;
 154                         case PAC_TYPE_UNKNOWN_10:
 155                                 break;
 156                         default:
 157                                 break;
 158                 }
 159         }
 160
 161         if (!logon_info) {
 162                 DEBUG(0,("PAC no logon_info\n"));
 163                 return NT_STATUS_FOOBAR;
 164         }
 165
 166         if (!srv_sig_ptr) {
 167                 DEBUG(0,("PAC no srv_key\n"));
 168                 return NT_STATUS_FOOBAR;
 169         }
 170
 171         if (!kdc_sig_ptr) {
 172                 DEBUG(0,("PAC no kdc_key\n"));
 173                 return NT_STATUS_FOOBAR;
 174         }
 175 #ifdef KRB5_DO_VERIFY_PAC
 176         /* clear the kdc_key */
 177 /*      memset((void *)kdc_sig_ptr , '\0', sizeof(*kdc_sig_ptr));*/
 178
 179         status = ndr_push_struct_blob(&tmp_blob, mem_ctx, &pac_data,
 180                                               (ndr_push_flags_fn_t)ndr_push_PAC_DATA);
 181         if (!NT_STATUS_IS_OK(status)) {
 182                 return status;
 183         }
 184         status = ndr_pull_struct_blob(&tmp_blob, mem_ctx, &pac_data,
 185                                         (ndr_pull_flags_fn_t)ndr_pull_PAC_DATA);
 186         if (!NT_STATUS_IS_OK(status)) {
 187                 DEBUG(0,("can't parse the PAC\n"));
 188                 return status;
 189         }
 190         /*NDR_PRINT_DEBUG(PAC_DATA, &pac_data);*/
 191
 192         /* verify by kdc_key */
 193         status = gensec_krb5_pac_checksum(tmp_blob, &kdc_sig, gensec_krb5_state, 0);
 194
 195         if (!NT_STATUS_IS_OK(status)) {
 196                 return status;
 197         }
 198
 199         /* clear the service_key */
 200 /*      memset((void *)srv_sig_ptr , '\0', sizeof(*srv_sig_ptr));*/
 201
 202         status = ndr_push_struct_blob(&tmp_blob, mem_ctx, &pac_data,
 203                                               (ndr_push_flags_fn_t)ndr_push_PAC_DATA);
 204         if (!NT_STATUS_IS_OK(status)) {
 205                 return status;
 206         }
 207         status = ndr_pull_struct_blob(&tmp_blob, mem_ctx, &pac_data,
 208                                         (ndr_pull_flags_fn_t)ndr_pull_PAC_DATA);
 209         if (!NT_STATUS_IS_OK(status)) {
 210                 DEBUG(0,("can't parse the PAC\n"));
 211                 return status;
 212         }
 213         NDR_PRINT_DEBUG(PAC_DATA, &pac_data);
 214
 215         /* verify by servie_key */
 216         status = gensec_krb5_pac_checksum(tmp_blob, &srv_sig, gensec_krb5_state, 0);
 217
 218         if (!NT_STATUS_IS_OK(status)) {
 219                 return status;
 220         }
 221 #endif
 222         DEBUG(0,("account_name: %s [%s]\n",logon_info->account_name.string, logon_info->full_name.string));
 223         *logon_info_out = logon_info;
 224
 225         return status;
 226 }
 227
 228 static void gensec_krb5_end(struct gensec_security *gensec_security)
 229 {
 230         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
 231
 232         if (gensec_krb5_state->ticket.length) {
 233         /* Hmm, early heimdal dooesn't have this - correct call would be krb5_data_free */
 234 #ifdef HAVE_KRB5_FREE_DATA_CONTENTS
 235                 krb5_free_data_contents(gensec_krb5_state->krb5_context, &gensec_krb5_state->ticket);
 236 #endif
 237         }
 238         if (gensec_krb5_state->krb5_ccache) {
 239                 /* current heimdal - 0.6.3, which we need anyway, fixes segfaults here */
 240                 krb5_cc_close(gensec_krb5_state->krb5_context, gensec_krb5_state->krb5_ccache);
 241         }
 242
 243         krb5_free_keyblock_contents(gensec_krb5_state->krb5_context,
 244                                     &gensec_krb5_state->krb5_keyblock);
 245
 246         if (gensec_krb5_state->krb5_auth_context) {
 247                 krb5_auth_con_free(gensec_krb5_state->krb5_context,
 248                                    gensec_krb5_state->krb5_auth_context);
 249         }
 250
 251         if (gensec_krb5_state->krb5_context) {
 252                 krb5_free_context(gensec_krb5_state->krb5_context);
 253         }
 254
 255         talloc_free(gensec_krb5_state);
 256         gensec_security->private_data = NULL;
 257 }
 258
 259
 260 static NTSTATUS gensec_krb5_start(struct gensec_security *gensec_security)
 261 {
 262         struct gensec_krb5_state *gensec_krb5_state;
 263         krb5_error_code ret = 0;
 264
 265         gensec_krb5_state = talloc_p(gensec_security, struct gensec_krb5_state);
 266         if (!gensec_krb5_state) {
 267                 return NT_STATUS_NO_MEMORY;
 268         }
 269
 270         gensec_security->private_data = gensec_krb5_state;
 271
 272         initialize_krb5_error_table();
 273         gensec_krb5_state->krb5_context = NULL;
 274         gensec_krb5_state->krb5_auth_context = NULL;
 275         gensec_krb5_state->krb5_ccache = NULL;
 276         ZERO_STRUCT(gensec_krb5_state->ticket);
 277         ZERO_STRUCT(gensec_krb5_state->krb5_keyblock);
 278         gensec_krb5_state->session_key = data_blob(NULL, 0);
 279
 280         ret = krb5_init_context(&gensec_krb5_state->krb5_context);
 281         if (ret) {
 282                 DEBUG(1,("gensec_krb5_start: krb5_init_context failed (%s)\n", error_message(ret)));
 283                 return NT_STATUS_INTERNAL_ERROR;
 284         }
 285
 286         if (lp_realm() && *lp_realm()) {
 287                 ret = krb5_set_default_realm(gensec_krb5_state->krb5_context, lp_realm());
 288                 if (ret) {
 289                         DEBUG(1,("gensec_krb5_start: krb5_set_default_realm failed (%s)\n", error_message(ret)));
 290                         return NT_STATUS_INTERNAL_ERROR;
 291                 }
 292         }
 293
 294         ret = krb5_auth_con_init(gensec_krb5_state->krb5_context, &gensec_krb5_state->krb5_auth_context);
 295         if (ret) {
 296                 DEBUG(1,("gensec_krb5_start: krb5_auth_con_init failed (%s)\n", error_message(ret)));
 297                 return NT_STATUS_INTERNAL_ERROR;
 298         }
 299
 300         return NT_STATUS_OK;
 301 }
 302
 303 static NTSTATUS gensec_krb5_server_start(struct gensec_security *gensec_security)
 304 {
 305         NTSTATUS nt_status;
 306         struct gensec_krb5_state *gensec_krb5_state;
 307
 308         nt_status = gensec_krb5_start(gensec_security);
 309         if (!NT_STATUS_IS_OK(nt_status)) {
 310                 return nt_status;
 311         }
 312
 313         gensec_krb5_state = gensec_security->private_data;
 314         gensec_krb5_state->state_position = GENSEC_KRB5_SERVER_START;
 315
 316         return NT_STATUS_OK;
 317 }
 318
 319 static NTSTATUS gensec_krb5_client_start(struct gensec_security *gensec_security)
 320 {
 321         struct gensec_krb5_state *gensec_krb5_state;
 322         krb5_error_code ret;
 323         NTSTATUS nt_status;
 324
 325         nt_status = gensec_krb5_start(gensec_security);
 326         if (!NT_STATUS_IS_OK(nt_status)) {
 327                 return nt_status;
 328         }
 329
 330         gensec_krb5_state = gensec_security->private_data;
 331         gensec_krb5_state->state_position = GENSEC_KRB5_CLIENT_START;
 332
 333         /* TODO: This is effecivly a static/global variable... */
 334         ret = krb5_cc_default(gensec_krb5_state->krb5_context, &gensec_krb5_state->krb5_ccache);
 335         if (ret) {
 336                 DEBUG(1,("krb5_cc_default failed (%s)\n",
 337                          error_message(ret)));
 338                 return NT_STATUS_INTERNAL_ERROR;
 339         }
 340
 341         while (1) {
 342                 if (gensec_security->target.principal) {
 343                         DEBUG(5, ("Finding ticket for target [%s]\n", gensec_security->target.principal));
 344                         ret = ads_krb5_mk_req(gensec_krb5_state->krb5_context,
 345                                               &gensec_krb5_state->krb5_auth_context,
 346                                               AP_OPTS_USE_SUBKEY | AP_OPTS_MUTUAL_REQUIRED,
 347                                               gensec_security->target.principal,
 348                                               gensec_krb5_state->krb5_ccache,
 349                                               &gensec_krb5_state->ticket);
 350                         if (ret) {
 351                                 DEBUG(1,("ads_krb5_mk_req failed (%s)\n",
 352                                          error_message(ret)));
 353                         }
 354                 } else {
 355                         krb5_data in_data;
 356                         const char *hostname = gensec_get_target_hostname(gensec_security);
 357                         if (!hostname) {
 358                                 DEBUG(1, ("Could not determine hostname for target computer, cannot use kerberos\n"));
 359                                 return NT_STATUS_ACCESS_DENIED;
 360                         }
 361
 362                         in_data.length = 0;
 363
 364                         ret = krb5_mk_req(gensec_krb5_state->krb5_context,
 365                                           &gensec_krb5_state->krb5_auth_context,
 366                                           AP_OPTS_USE_SUBKEY | AP_OPTS_MUTUAL_REQUIRED,
 367                                           gensec_get_target_service(gensec_security),
 368                                           hostname,
 369                                           &in_data, gensec_krb5_state->krb5_ccache,
 370                                           &gensec_krb5_state->ticket);
 371                         if (ret) {
 372                                 DEBUG(1,("krb5_mk_req failed (%s)\n",
 373                                          error_message(ret)));
 374                         }
 375
 376                 }
 377                 switch (ret) {
 378                 case 0:
 379                         return NT_STATUS_OK;
 380                 case KRB5KDC_ERR_S_PRINCIPAL_UNKNOWN:
 381                         DEBUG(3, ("Server is not registered with our KDC: %s\n",
 382                                   error_message(ret)));
 383                         return NT_STATUS_ACCESS_DENIED;
 384                 case KRB5KDC_ERR_PREAUTH_FAILED:
 385                 case KRB5KRB_AP_ERR_TKT_EXPIRED:
 386                 case KRB5_CC_END:
 387                 {
 388                         DEBUG(3, ("kerberos: %s\n",
 389                                   error_message(ret)));
 390                         /* fall down to remaining code */
 391                 }
 392                 /* just don't print a message for these really ordinary messages */
 393                 case KRB5_FCC_NOFILE:
 394                 case KRB5_CC_NOTFOUND:
 395                 {
 396                         char *password;
 397                         time_t kdc_time = 0;
 398                         nt_status = gensec_get_password(gensec_security,
 399                                                         gensec_security,
 400                                                         &password);
 401                         if (!NT_STATUS_IS_OK(nt_status)) {
 402                                 return nt_status;
 403                         }
 404
 405                         ret = kerberos_kinit_password_cc(gensec_krb5_state->krb5_context, gensec_krb5_state->krb5_ccache,
 406                                                          gensec_get_client_principal(gensec_security, gensec_security),
 407                                                          password, NULL, &kdc_time);
 408
 409                         /* cope with ticket being in the future due to clock skew */
 410                         if ((unsigned)kdc_time > time(NULL)) {
 411                                 time_t t = time(NULL);
 412                                 int time_offset =(unsigned)kdc_time-t;
 413                                 DEBUG(4,("Advancing clock by %d seconds to cope with clock skew\n", time_offset));
 414                                 krb5_set_real_time(gensec_krb5_state->krb5_context, t + time_offset + 1, 0);
 415                         }
 416
 417                         if (ret) {
 418                                 DEBUG(1,("kinit failed (%s)\n",
 419                                          error_message(ret)));
 420                                 return NT_STATUS_WRONG_PASSWORD;
 421                         }
 422                         break;
 423                 }
 424                 default:
 425                         DEBUG(0, ("kerberos: %s\n",
 426                                   error_message(ret)));
 427                         return NT_STATUS_UNSUCCESSFUL;
 428                 }
 429         }
 430 }
 431
 432
 433 /**
 434  * Next state function for the Krb5 GENSEC mechanism
 435  *
 436  * @param gensec_krb5_state KRB5 State
 437  * @param out_mem_ctx The TALLOC_CTX for *out to be allocated on
 438  * @param in The request, as a DATA_BLOB
 439  * @param out The reply, as an talloc()ed DATA_BLOB, on *out_mem_ctx
 440  * @return Error, MORE_PROCESSING_REQUIRED if a reply is sent,
 441  *                or NT_STATUS_OK if the user is authenticated.
 442  */
 443
 444 static NTSTATUS gensec_krb5_update(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx,
 445                                       const DATA_BLOB in, DATA_BLOB *out)
 446 {
 447         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
 448         krb5_error_code ret = 0;
 449         DATA_BLOB pac;
 450         NTSTATUS nt_status;
 451
 452         switch (gensec_krb5_state->state_position) {
 453         case GENSEC_KRB5_CLIENT_START:
 454         {
 455                 if (ret) {
 456                         DEBUG(1,("ads_krb5_mk_req (request ticket) failed (%s)\n",
 457                                  error_message(ret)));
 458                         nt_status = NT_STATUS_LOGON_FAILURE;
 459                 } else {
 460                         DATA_BLOB unwrapped_out;
 461
 462 #ifndef GENSEC_SEND_UNWRAPPED_KRB5 /* This should be a switch for the torture code to set */
 463                         unwrapped_out = data_blob_talloc(out_mem_ctx, gensec_krb5_state->ticket.data, gensec_krb5_state->ticket.length);
 464
 465                         /* wrap that up in a nice GSS-API wrapping */
 466                         *out = gensec_gssapi_gen_krb5_wrap(out_mem_ctx, &unwrapped_out, TOK_ID_KRB_AP_REQ);
 467 #else
 468                         *out = data_blob_talloc(out_mem_ctx, gensec_krb5_state->ticket.data, gensec_krb5_state->ticket.length);
 469 #endif
 470                         gensec_krb5_state->state_position = GENSEC_KRB5_CLIENT_MUTUAL_AUTH;
 471                         nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 472                 }
 473
 474                 return nt_status;
 475         }
 476
 477         case GENSEC_KRB5_CLIENT_MUTUAL_AUTH:
 478         {
 479                 krb5_data inbuf;
 480                 krb5_ap_rep_enc_part *repl = NULL;
 481                 uint8 tok_id[2];
 482                 DATA_BLOB unwrapped_in;
 483
 484                 if (!gensec_gssapi_parse_krb5_wrap(out_mem_ctx, &in, &unwrapped_in, tok_id)) {
 485                         DEBUG(1,("gensec_gssapi_parse_krb5_wrap(mutual authentication) failed to parse\n"));
 486                         dump_data_pw("Mutual authentication message:\n", in.data, in.length);
 487                         return NT_STATUS_INVALID_PARAMETER;
 488                 }
 489                 /* TODO: check the tok_id */
 490
 491                 inbuf.data = unwrapped_in.data;
 492                 inbuf.length = unwrapped_in.length;
 493                 ret = krb5_rd_rep(gensec_krb5_state->krb5_context,
 494                                   gensec_krb5_state->krb5_auth_context,
 495                                   &inbuf, &repl);
 496                 if (ret) {
 497                         DEBUG(1,("krb5_rd_rep (mutual authentication) failed (%s)\n",
 498                                  error_message(ret)));
 499                         dump_data_pw("Mutual authentication message:\n", inbuf.data, inbuf.length);
 500                         nt_status = NT_STATUS_ACCESS_DENIED;
 501                 } else {
 502                         *out = data_blob(NULL, 0);
 503                         nt_status = NT_STATUS_OK;
 504                         gensec_krb5_state->state_position = GENSEC_KRB5_DONE;
 505                 }
 506                 if (repl) {
 507                         krb5_free_ap_rep_enc_part(gensec_krb5_state->krb5_context, repl);
 508                 }
 509                 return nt_status;
 510         }
 511
 512         case GENSEC_KRB5_SERVER_START:
 513         {
 514                 char *principal;
 515                 DATA_BLOB unwrapped_in;
 516                 DATA_BLOB unwrapped_out;
 517                 uint8 tok_id[2];
 518
 519                 /* Parse the GSSAPI wrapping, if it's there... (win2k3 allows it to be omited) */
 520                 if (!gensec_gssapi_parse_krb5_wrap(out_mem_ctx, &in, &unwrapped_in, tok_id)) {
 521                         nt_status = ads_verify_ticket(out_mem_ctx,
 522                                                       gensec_krb5_state->krb5_context,
 523                                                       gensec_krb5_state->krb5_auth_context,
 524                                                       lp_realm(), &in,
 525                                                       &principal, &pac, &unwrapped_out,
 526                                                       &gensec_krb5_state->krb5_keyblock);
 527                 } else {
 528                         /* TODO: check the tok_id */
 529                         nt_status = ads_verify_ticket(out_mem_ctx,
 530                                                       gensec_krb5_state->krb5_context,
 531                                                       gensec_krb5_state->krb5_auth_context,
 532                                                       lp_realm(), &unwrapped_in,
 533                                                       &principal, &pac, &unwrapped_out,
 534                                                       &gensec_krb5_state->krb5_keyblock);
 535                 }
 536
 537                 if (!NT_STATUS_IS_OK(nt_status)) {
 538                         return nt_status;
 539                 }
 540
 541                 if (pac.data) {
 542                         /* decode and verify the pac */
 543                         nt_status = gensec_krb5_decode_pac(gensec_krb5_state, &gensec_krb5_state->logon_info, pac,
 544                                                            gensec_krb5_state);
 545                 } else {
 546                         /* NULL PAC, we might need to figure this information out the hard way */
 547                         gensec_krb5_state->logon_info = NULL;
 548                 }
 549
 550                 if (NT_STATUS_IS_OK(nt_status)) {
 551                         gensec_krb5_state->state_position = GENSEC_KRB5_DONE;
 552                         /* wrap that up in a nice GSS-API wrapping */
 553                         *out = gensec_gssapi_gen_krb5_wrap(out_mem_ctx, &unwrapped_out, TOK_ID_KRB_AP_REP);
 554
 555                         gensec_krb5_state->peer_principal = talloc_steal(gensec_krb5_state, principal);
 556                 }
 557                 return nt_status;
 558         }
 559         case GENSEC_KRB5_DONE:
 560                 return NT_STATUS_OK;
 561         }
 562
 563         return NT_STATUS_INVALID_PARAMETER;
 564 }
 565
 566 static NTSTATUS gensec_krb5_session_key(struct gensec_security *gensec_security,
 567                                            DATA_BLOB *session_key)
 568 {
 569         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
 570         krb5_context context = gensec_krb5_state->krb5_context;
 571         krb5_auth_context auth_context = gensec_krb5_state->krb5_auth_context;
 572         krb5_keyblock *skey;
 573         krb5_error_code err;
 574
 575         if (gensec_krb5_state->session_key.data) {
 576                 *session_key = gensec_krb5_state->session_key;
 577                 return NT_STATUS_OK;
 578         }
 579
 580         switch (gensec_security->gensec_role) {
 581         case GENSEC_CLIENT:
 582                 err = krb5_auth_con_getlocalsubkey(context, auth_context, &skey);
 583                 break;
 584         case GENSEC_SERVER:
 585                 err = krb5_auth_con_getremotesubkey(context, auth_context, &skey);
 586                 break;
 587         }
 588         if (err == 0 && skey != NULL) {
 589                 DEBUG(10, ("Got KRB5 session key of length %d\n",  KRB5_KEY_LENGTH(skey)));
 590                 gensec_krb5_state->session_key = data_blob_talloc(gensec_krb5_state,
 591                                                 KRB5_KEY_DATA(skey), KRB5_KEY_LENGTH(skey));
 592                 *session_key = gensec_krb5_state->session_key;
 593                 dump_data_pw("KRB5 Session Key:\n", session_key->data, session_key->length);
 594
 595                 krb5_free_keyblock(context, skey);
 596                 return NT_STATUS_OK;
 597         } else {
 598                 DEBUG(10, ("KRB5 error getting session key %d\n", err));
 599                 return NT_STATUS_NO_USER_SESSION_KEY;
 600         }
 601 }
 602
 603 static NTSTATUS gensec_krb5_session_info(struct gensec_security *gensec_security,
 604                                      struct auth_session_info **session_info_out)
 605 {
 606         NTSTATUS nt_status;
 607         struct gensec_krb5_state *gensec_krb5_state = gensec_security->private_data;
 608         struct auth_serversupplied_info *server_info = NULL;
 609         struct auth_session_info *session_info = NULL;
 610         struct PAC_LOGON_INFO *logon_info = gensec_krb5_state->logon_info;
 611         struct nt_user_token *ptoken;
 612         struct dom_sid *sid;
 613         char *p;
 614         char *principal;
 615         const char *username;
 616         const char *realm;
 617
 618         *session_info_out = NULL;
 619
 620         /* IF we have the PAC - otherwise we need to get this
 621          * data from elsewere - local ldb, or (TODO) lookup of some
 622          * kind... */
 623
 624         principal = talloc_strdup(gensec_krb5_state, gensec_krb5_state->peer_principal);
 625         p = strchr(principal, '@');
 626         if (p) {
 627                 *p = '\0';
 628         }
 629         p++;
 630         username = principal;
 631         realm = p;
 632
 633         if (logon_info) {
 634                 nt_status = make_server_info(gensec_krb5_state, &server_info, gensec_krb5_state->peer_principal);
 635                 if (!NT_STATUS_IS_OK(nt_status)) {
 636                         return nt_status;
 637                 }
 638
 639                 server_info->guest = False;
 640
 641                 server_info->account_name = talloc_strdup(server_info, principal);
 642                 server_info->domain = talloc_strdup(server_info, realm);
 643                 if (!server_info->domain) {
 644                         free_server_info(&server_info);
 645                         return NT_STATUS_NO_MEMORY;
 646                 }
 647
 648                 /* references the server_info into the session_info */
 649                 nt_status = make_session_info(gensec_krb5_state, server_info, &session_info);
 650                 if (!NT_STATUS_IS_OK(nt_status)) {
 651                         free_server_info(&server_info);
 652                         return nt_status;
 653                 }
 654
 655                 talloc_free(server_info);
 656
 657                 ptoken = talloc_p(session_info, struct nt_user_token);
 658                 if (!ptoken) {
 659                         return NT_STATUS_NO_MEMORY;
 660                 }
 661
 662                 ptoken->num_sids = 0;
 663
 664                 ptoken->user_sids = talloc_array_p(ptoken, struct dom_sid*, logon_info->groups_count + 2);
 665                 if (!ptoken->user_sids) {
 666                         return NT_STATUS_NO_MEMORY;
 667                 }
 668
 669
 670                 sid = dom_sid_dup(server_info, logon_info->dom_sid);
 671                 server_info->user_sid = dom_sid_add_rid(server_info, sid, logon_info->user_rid);
 672                 sid = dom_sid_dup(server_info, logon_info->dom_sid);
 673                 server_info->primary_group_sid = dom_sid_add_rid(server_info, sid, logon_info->group_rid);
 674
 675                 ptoken->user_sids[0] = talloc_reference(session_info, server_info->user_sid);
 676                 ptoken->num_sids++;
 677                 ptoken->user_sids[1] = talloc_reference(session_info, server_info->primary_group_sid);
 678                 ptoken->num_sids++;
 679
 680                 for (;ptoken->num_sids < (logon_info->groups_count + 2); ptoken->num_sids++) {
 681                         sid = dom_sid_dup(session_info, logon_info->dom_sid);
 682                         ptoken->user_sids[ptoken->num_sids]
 683                                 = dom_sid_add_rid(session_info, sid,
 684                                                   logon_info->groups[ptoken->num_sids - 2].rid);
 685                 }
 686
 687                 debug_nt_user_token(DBGC_AUTH, 0, ptoken);
 688
 689                 session_info->nt_user_token = ptoken;
 690         } else {
 691                 TALLOC_CTX *mem_ctx = talloc_named(gensec_krb5_state, 0, "PAC-less session info discovery for %s@%s", username, realm);
 692                 if (!mem_ctx) {
 693                         return NT_STATUS_NO_MEMORY;
 694                 }
 695                 nt_status = sam_get_server_info(username, realm, gensec_krb5_state, &server_info);
 696                 if (!NT_STATUS_IS_OK(nt_status)) {
 697                         talloc_free(mem_ctx);
 698                         return nt_status;
 699                 }
 700
 701                 /* references the server_info into the session_info */
 702                 nt_status = make_session_info(gensec_krb5_state, server_info, &session_info);
 703                 if (!NT_STATUS_IS_OK(nt_status)) {
 704                         talloc_free(mem_ctx);
 705                         return nt_status;
 706                 }
 707
 708                 talloc_free(mem_ctx);
 709         }
 710
 711         talloc_free(principal);
 712
 713         nt_status = gensec_krb5_session_key(gensec_security, &session_info->session_key);
 714
 715         session_info->workstation = NULL;
 716
 717         *session_info_out = session_info;
 718
 719         return nt_status;
 720 }
 721
 722
 723 static const struct gensec_security_ops gensec_krb5_security_ops = {
 724         .name           = "krb5",
 725         .auth_type      = DCERPC_AUTH_TYPE_KRB5,
 726         .oid            = OID_KERBEROS5,
 727         .client_start   = gensec_krb5_client_start,
 728         .server_start   = gensec_krb5_server_start,
 729         .update         = gensec_krb5_update,
 730         .session_key    = gensec_krb5_session_key,
 731         .session_info   = gensec_krb5_session_info,
 732         .end            = gensec_krb5_end
 733 };
 734
 735 static const struct gensec_security_ops gensec_ms_krb5_security_ops = {
 736         .name           = "ms_krb5",
 737         .auth_type      = DCERPC_AUTH_TYPE_KRB5,
 738         .oid            = OID_KERBEROS5_OLD,
 739         .client_start   = gensec_krb5_client_start,
 740         .server_start   = gensec_krb5_server_start,
 741         .update         = gensec_krb5_update,
 742         .session_key    = gensec_krb5_session_key,
 743         .session_info   = gensec_krb5_session_info,
 744         .end            = gensec_krb5_end
 745 };
 746
 747
 748 NTSTATUS gensec_krb5_init(void)
 749 {
 750         NTSTATUS ret;
 751
 752         ret = register_backend("gensec", &gensec_krb5_security_ops);
 753         if (!NT_STATUS_IS_OK(ret)) {
 754                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
 755                         gensec_krb5_security_ops.name));
 756                 return ret;
 757         }
 758
 759         ret = register_backend("gensec", &gensec_ms_krb5_security_ops);
 760         if (!NT_STATUS_IS_OK(ret)) {
 761                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
 762                         gensec_krb5_security_ops.name));
 763                 return ret;
 764         }
 765
 766         return ret;
 767 }