source4/libcli/auth/spnego.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    RFC2478 Compliant SPNEGO implementation
   5
   6    Copyright (C) Jim McDonough <jmcd@us.ibm.com>      2003
   7    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004
   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 2 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, write to the Free Software
  22    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  23 */
  24
  25 #include "includes.h"
  26
  27 #undef DBGC_CLASS
  28 #define DBGC_CLASS DBGC_AUTH
  29
  30 enum spnego_state_position {
  31         SPNEGO_SERVER_START,
  32         SPNEGO_CLIENT_START,
  33         SPNEGO_SERVER_TARG,
  34         SPNEGO_CLIENT_TARG,
  35         SPNEGO_FALLBACK,
  36         SPNEGO_DONE
  37 };
  38
  39 struct spnego_state {
  40         uint_t ref_count;
  41         enum spnego_message_type expected_packet;
  42         enum spnego_state_position state_position;
  43         struct gensec_security *sub_sec_security;
  44 };
  45
  46 static NTSTATUS gensec_spnego_client_start(struct gensec_security *gensec_security)
  47 {
  48         struct spnego_state *spnego_state;
  49
  50         spnego_state = talloc_p(gensec_security, struct spnego_state);
  51         if (!spnego_state) {
  52                 return NT_STATUS_NO_MEMORY;
  53         }
  54
  55         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
  56         spnego_state->state_position = SPNEGO_CLIENT_START;
  57         spnego_state->sub_sec_security = NULL;
  58
  59         gensec_security->private_data = spnego_state;
  60         return NT_STATUS_OK;
  61 }
  62
  63 static NTSTATUS gensec_spnego_server_start(struct gensec_security *gensec_security)
  64 {
  65         struct spnego_state *spnego_state;
  66
  67         spnego_state = talloc_p(gensec_security, struct spnego_state);
  68         if (!spnego_state) {
  69                 return NT_STATUS_NO_MEMORY;
  70         }
  71
  72         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
  73         spnego_state->state_position = SPNEGO_SERVER_START;
  74         spnego_state->sub_sec_security = NULL;
  75
  76         gensec_security->private_data = spnego_state;
  77         return NT_STATUS_OK;
  78 }
  79
  80 /*
  81   wrappers for the spnego_*() functions
  82 */
  83 static NTSTATUS gensec_spnego_unseal_packet(struct gensec_security *gensec_security,
  84                                             TALLOC_CTX *mem_ctx,
  85                                             uint8_t *data, size_t length,
  86                                             const uint8_t *whole_pdu, size_t pdu_length,
  87                                             DATA_BLOB *sig)
  88 {
  89         struct spnego_state *spnego_state = gensec_security->private_data;
  90
  91         if (spnego_state->state_position != SPNEGO_DONE
  92             && spnego_state->state_position != SPNEGO_FALLBACK) {
  93                 return NT_STATUS_INVALID_PARAMETER;
  94         }
  95
  96         return gensec_unseal_packet(spnego_state->sub_sec_security,
  97                                     mem_ctx,
  98                                     data, length,
  99                                     whole_pdu, pdu_length,
 100                                     sig);
 101 }
 102
 103 static NTSTATUS gensec_spnego_check_packet(struct gensec_security *gensec_security,
 104                                            TALLOC_CTX *mem_ctx,
 105                                            const uint8_t *data, size_t length,
 106                                            const uint8_t *whole_pdu, size_t pdu_length,
 107                                            const DATA_BLOB *sig)
 108 {
 109         struct spnego_state *spnego_state = gensec_security->private_data;
 110
 111         if (spnego_state->state_position != SPNEGO_DONE
 112             && spnego_state->state_position != SPNEGO_FALLBACK) {
 113                 return NT_STATUS_INVALID_PARAMETER;
 114         }
 115
 116         return gensec_check_packet(spnego_state->sub_sec_security,
 117                                    mem_ctx,
 118                                    data, length,
 119                                    whole_pdu, pdu_length,
 120                                    sig);
 121 }
 122
 123 static NTSTATUS gensec_spnego_seal_packet(struct gensec_security *gensec_security,
 124                                           TALLOC_CTX *mem_ctx,
 125                                           uint8_t *data, size_t length,
 126                                           const uint8_t *whole_pdu, size_t pdu_length,
 127                                           DATA_BLOB *sig)
 128 {
 129         struct spnego_state *spnego_state = gensec_security->private_data;
 130
 131         if (spnego_state->state_position != SPNEGO_DONE
 132             && spnego_state->state_position != SPNEGO_FALLBACK) {
 133                 return NT_STATUS_INVALID_PARAMETER;
 134         }
 135
 136         return gensec_seal_packet(spnego_state->sub_sec_security,
 137                                   mem_ctx,
 138                                   data, length,
 139                                   whole_pdu, pdu_length,
 140                                   sig);
 141 }
 142
 143 static NTSTATUS gensec_spnego_sign_packet(struct gensec_security *gensec_security,
 144                                           TALLOC_CTX *mem_ctx,
 145                                           const uint8_t *data, size_t length,
 146                                           const uint8_t *whole_pdu, size_t pdu_length,
 147                                           DATA_BLOB *sig)
 148 {
 149         struct spnego_state *spnego_state = gensec_security->private_data;
 150
 151         if (spnego_state->state_position != SPNEGO_DONE
 152             && spnego_state->state_position != SPNEGO_FALLBACK) {
 153                 return NT_STATUS_INVALID_PARAMETER;
 154         }
 155
 156         return gensec_sign_packet(spnego_state->sub_sec_security,
 157                                   mem_ctx,
 158                                   data, length,
 159                                   whole_pdu, pdu_length,
 160                                   sig);
 161 }
 162
 163 static size_t gensec_spnego_sig_size(struct gensec_security *gensec_security)
 164 {
 165         struct spnego_state *spnego_state = gensec_security->private_data;
 166
 167         if (spnego_state->state_position != SPNEGO_DONE
 168             && spnego_state->state_position != SPNEGO_FALLBACK) {
 169                 return 0;
 170         }
 171
 172         return gensec_sig_size(spnego_state->sub_sec_security);
 173 }
 174
 175 static NTSTATUS gensec_spnego_session_key(struct gensec_security *gensec_security,
 176                                           DATA_BLOB *session_key)
 177 {
 178         struct spnego_state *spnego_state = gensec_security->private_data;
 179         if (!spnego_state->sub_sec_security) {
 180                 return NT_STATUS_INVALID_PARAMETER;
 181         }
 182
 183         return gensec_session_key(spnego_state->sub_sec_security,
 184                                   session_key);
 185 }
 186
 187 static NTSTATUS gensec_spnego_session_info(struct gensec_security *gensec_security,
 188                                                                       struct auth_session_info **session_info)
 189 {
 190         struct spnego_state *spnego_state = gensec_security->private_data;
 191         if (!spnego_state->sub_sec_security) {
 192                 return NT_STATUS_INVALID_PARAMETER;
 193         }
 194
 195         return gensec_session_info(spnego_state->sub_sec_security,
 196                                    session_info);
 197 }
 198
 199 /** Fallback to another GENSEC mechanism, based on magic strings
 200  *
 201  * This is the 'fallback' case, where we don't get SPNEGO, and have to
 202  * try all the other options (and hope they all have a magic string
 203  * they check)
 204 */
 205
 206 static NTSTATUS gensec_spnego_server_try_fallback(struct gensec_security *gensec_security,
 207                                                   struct spnego_state *spnego_state,
 208                                                   TALLOC_CTX *out_mem_ctx,
 209                                                   const DATA_BLOB in, DATA_BLOB *out)
 210 {
 211         int i;
 212         int num_ops;
 213         const struct gensec_security_ops **all_ops = gensec_security_all(&num_ops);
 214         for (i=0; i < num_ops; i++) {
 215                 NTSTATUS nt_status;
 216                 if (!all_ops[i]->oid) {
 217                         continue;
 218                 }
 219                 nt_status = gensec_subcontext_start(gensec_security,
 220                                                     &spnego_state->sub_sec_security);
 221                 if (!NT_STATUS_IS_OK(nt_status)) {
 222                         return nt_status;
 223                 }
 224                 /* select the sub context */
 225                 nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
 226                                                      all_ops[i]->oid);
 227                 if (!NT_STATUS_IS_OK(nt_status)) {
 228                         gensec_end(&spnego_state->sub_sec_security);
 229                                         continue;
 230                 }
 231                 nt_status = gensec_update(spnego_state->sub_sec_security,
 232                                                           out_mem_ctx, in, out);
 233                 if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 234                         spnego_state->state_position = SPNEGO_FALLBACK;
 235                         return nt_status;
 236                 }
 237                 gensec_end(&spnego_state->sub_sec_security);
 238         }
 239         DEBUG(1, ("Failed to parse SPNEGO request\n"));
 240         return NT_STATUS_INVALID_PARAMETER;
 241
 242 }
 243
 244 static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_security,
 245                                                  struct spnego_state *spnego_state,
 246                                                  TALLOC_CTX *out_mem_ctx,
 247                                                  const char **mechType,
 248                                                  const DATA_BLOB unwrapped_in, DATA_BLOB *unwrapped_out)
 249 {
 250         int i;
 251         NTSTATUS nt_status;
 252         DATA_BLOB null_data_blob = data_blob(NULL,0);
 253
 254         for (i=0; mechType && mechType[i]; i++) {
 255                 nt_status = gensec_subcontext_start(gensec_security,
 256                                                     &spnego_state->sub_sec_security);
 257                 if (!NT_STATUS_IS_OK(nt_status)) {
 258                         break;
 259                 }
 260                 /* select the sub context */
 261                 nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
 262                                                      mechType[i]);
 263                 if (!NT_STATUS_IS_OK(nt_status)) {
 264                         gensec_end(&spnego_state->sub_sec_security);
 265                         continue;
 266                 }
 267
 268                 if (i == 0) {
 269                         nt_status = gensec_update(spnego_state->sub_sec_security,
 270                                                   out_mem_ctx,
 271                                                   unwrapped_in,
 272                                                   unwrapped_out);
 273                 } else {
 274                         /* only get the helping start blob for the first OID */
 275                         nt_status = gensec_update(spnego_state->sub_sec_security,
 276                                                   out_mem_ctx,
 277                                                   null_data_blob,
 278                                                   unwrapped_out);
 279                 }
 280                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(nt_status)) {
 281                         DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed: %s\n",
 282                                   spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
 283                         gensec_end(&spnego_state->sub_sec_security);
 284                 }
 285                 return nt_status;
 286         }
 287         if (!mechType || !mechType[i]) {
 288                 DEBUG(1, ("SPNEGO: Could not find a suitable mechtype in NEG_TOKEN_INIT\n"));
 289         }
 290         return NT_STATUS_INVALID_PARAMETER;
 291 }
 292
 293 /** create a client negTokenInit
 294  *
 295  * This is the case, where the client is the first one who sends data
 296 */
 297
 298 static NTSTATUS gensec_spnego_client_negTokenInit(struct gensec_security *gensec_security,
 299                                                   struct spnego_state *spnego_state,
 300                                                   TALLOC_CTX *out_mem_ctx,
 301                                                   const DATA_BLOB in, DATA_BLOB *out)
 302 {
 303         DATA_BLOB null_data_blob = data_blob(NULL,0);
 304         NTSTATUS nt_status;
 305         const char **mechTypes = NULL;
 306         DATA_BLOB unwrapped_out = data_blob(NULL,0);
 307
 308         mechTypes = gensec_security_oids(out_mem_ctx, OID_SPNEGO);
 309
 310         if (!mechTypes) {
 311                 DEBUG(1, ("no GENSEC OID backends available\n"));
 312                 return NT_STATUS_INVALID_PARAMETER;
 313         }
 314
 315         nt_status = gensec_subcontext_start(gensec_security,
 316                                             &spnego_state->sub_sec_security);
 317         if (!NT_STATUS_IS_OK(nt_status)) {
 318                 return nt_status;
 319         }
 320         /* select our preferred mech */
 321         nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
 322                                              mechTypes[0]);
 323         if (!NT_STATUS_IS_OK(nt_status)) {
 324                 gensec_end(&spnego_state->sub_sec_security);
 325                         return nt_status;
 326         }
 327         nt_status = gensec_update(spnego_state->sub_sec_security,
 328                                   out_mem_ctx, in, &unwrapped_out);
 329         if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 330                 struct spnego_data spnego_out;
 331                 spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
 332                 spnego_out.negTokenInit.mechTypes = mechTypes;
 333                 spnego_out.negTokenInit.reqFlags = 0;
 334                 spnego_out.negTokenInit.mechListMIC = null_data_blob;
 335                 spnego_out.negTokenInit.mechToken = unwrapped_out;
 336
 337                 if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 338                         DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n"));
 339                                 return NT_STATUS_INVALID_PARAMETER;
 340                 }
 341
 342                 /* set next state */
 343                 spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 344                 spnego_state->state_position = SPNEGO_CLIENT_TARG;
 345                 return nt_status;
 346         }
 347         gensec_end(&spnego_state->sub_sec_security);
 348
 349         DEBUG(1, ("Failed to setup SPNEGO netTokenInit request\n"));
 350         return NT_STATUS_INVALID_PARAMETER;
 351 }
 352
 353
 354 /** create a client negTokenTarg
 355  *
 356  * This is the case, where the client is the first one who sends data
 357 */
 358
 359 static NTSTATUS gensec_spnego_server_negTokenTarg(struct gensec_security *gensec_security,
 360                                                   struct spnego_state *spnego_state,
 361                                                   TALLOC_CTX *out_mem_ctx,
 362                                                   NTSTATUS nt_status,
 363                                                   const DATA_BLOB unwrapped_out, DATA_BLOB *out)
 364 {
 365         struct spnego_data spnego_out;
 366         DATA_BLOB null_data_blob = data_blob(NULL, 0);
 367
 368         /* compose reply */
 369         spnego_out.type = SPNEGO_NEG_TOKEN_TARG;
 370         spnego_out.negTokenTarg.responseToken = unwrapped_out;
 371         spnego_out.negTokenTarg.mechListMIC = null_data_blob;
 372
 373         if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 374                 spnego_out.negTokenTarg.supportedMech
 375                         = spnego_state->sub_sec_security->ops->oid;
 376                 spnego_out.negTokenTarg.negResult = SPNEGO_ACCEPT_INCOMPLETE;
 377                 spnego_state->state_position = SPNEGO_SERVER_TARG;
 378         } else if (NT_STATUS_IS_OK(nt_status)) {
 379                 spnego_out.negTokenTarg.supportedMech = NULL;
 380                 spnego_out.negTokenTarg.negResult = SPNEGO_ACCEPT_COMPLETED;
 381                 spnego_state->state_position = SPNEGO_DONE;
 382         } else {
 383                 spnego_out.negTokenTarg.supportedMech = NULL;
 384                 spnego_out.negTokenTarg.negResult = SPNEGO_REJECT;
 385                 DEBUG(1, ("SPNEGO login failed: %s\n", nt_errstr(nt_status)));
 386                 spnego_state->state_position = SPNEGO_DONE;
 387         }
 388
 389         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 390                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_TARG\n"));
 391                 return NT_STATUS_INVALID_PARAMETER;
 392         }
 393
 394         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 395
 396         return nt_status;
 397 }
 398
 399
 400 static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx,
 401                                      const DATA_BLOB in, DATA_BLOB *out)
 402 {
 403         struct spnego_state *spnego_state = gensec_security->private_data;
 404         DATA_BLOB null_data_blob = data_blob(NULL, 0);
 405         DATA_BLOB unwrapped_out = data_blob(NULL, 0);
 406         struct spnego_data spnego_out;
 407         struct spnego_data spnego;
 408
 409         ssize_t len;
 410
 411         *out = data_blob(NULL, 0);
 412
 413         if (!out_mem_ctx) {
 414                 out_mem_ctx = spnego_state;
 415         }
 416
 417         /* and switch into the state machine */
 418
 419         switch (spnego_state->state_position) {
 420         case SPNEGO_FALLBACK:
 421                 return gensec_update(spnego_state->sub_sec_security,
 422                                      out_mem_ctx, in, out);
 423         case SPNEGO_SERVER_START:
 424         {
 425                 if (in.length) {
 426                         NTSTATUS nt_status;
 427
 428                         len = spnego_read_data(in, &spnego);
 429                         if (len == -1) {
 430                                 return gensec_spnego_server_try_fallback(gensec_security, spnego_state, out_mem_ctx, in, out);
 431                         }
 432                         /* client sent NegTargetInit, we send NegTokenTarg */
 433
 434                         /* OK, so it's real SPNEGO, check the packet's the one we expect */
 435                         if (spnego.type != spnego_state->expected_packet) {
 436                                 DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 437                                           spnego_state->expected_packet));
 438                                 dump_data(1, (const char *)in.data, in.length);
 439                                 spnego_free_data(&spnego);
 440                                 return NT_STATUS_INVALID_PARAMETER;
 441                         }
 442
 443                         nt_status = gensec_spnego_parse_negTokenInit(gensec_security,
 444                                                                      spnego_state,
 445                                                                      out_mem_ctx,
 446                                                                      spnego.negTokenInit.mechTypes,
 447                                                                      spnego.negTokenInit.mechToken,
 448                                                                      &unwrapped_out);
 449
 450                         nt_status = gensec_spnego_server_negTokenTarg(gensec_security,
 451                                                                       spnego_state,
 452                                                                       out_mem_ctx,
 453                                                                       nt_status,
 454                                                                       unwrapped_out,
 455                                                                       out);
 456
 457                         spnego_free_data(&spnego);
 458
 459                         return nt_status;
 460                 } else {
 461                         const char **mechlist = gensec_security_oids(out_mem_ctx, OID_SPNEGO);
 462
 463                         spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
 464                         spnego_out.negTokenInit.mechTypes = mechlist;
 465                         spnego_out.negTokenInit.reqFlags = 0;
 466                         spnego_out.negTokenInit.mechListMIC = null_data_blob;
 467                         spnego_out.negTokenInit.mechToken = unwrapped_out;
 468
 469                         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 470                                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n"));
 471                                 return NT_STATUS_INVALID_PARAMETER;
 472                         }
 473
 474                         /* set next state */
 475                         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 476                         spnego_state->state_position = SPNEGO_SERVER_TARG;
 477
 478                         return NT_STATUS_MORE_PROCESSING_REQUIRED;
 479                 }
 480         }
 481
 482         case SPNEGO_CLIENT_START:
 483         {
 484                 /* The server offers a list of mechanisms */
 485
 486                 char *my_mechs[] = {NULL, NULL};
 487                 NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
 488
 489                 if (!in.length) {
 490                         /* client to produce negTokenInit */
 491                         return gensec_spnego_client_negTokenInit(gensec_security, spnego_state, out_mem_ctx, in, out);
 492                 }
 493
 494                 len = spnego_read_data(in, &spnego);
 495
 496                 if (len == -1) {
 497                         DEBUG(1, ("Invalid SPNEGO request:\n"));
 498                         dump_data(1, (const char *)in.data, in.length);
 499                         return NT_STATUS_INVALID_PARAMETER;
 500                 }
 501
 502                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
 503                 if (spnego.type != spnego_state->expected_packet) {
 504                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 505                                   spnego_state->expected_packet));
 506                         dump_data(1, (const char *)in.data, in.length);
 507                         spnego_free_data(&spnego);
 508                         return NT_STATUS_INVALID_PARAMETER;
 509                 }
 510
 511                 if (spnego.negTokenInit.targetPrincipal) {
 512                         DEBUG(5, ("Server claims it's principal name is %s\n", spnego.negTokenInit.targetPrincipal));
 513                         nt_status = gensec_set_target_principal(gensec_security,
 514                                                                 spnego.negTokenInit.targetPrincipal);
 515                         if (!NT_STATUS_IS_OK(nt_status)) {
 516                                 spnego_free_data(&spnego);
 517                                 return nt_status;
 518                         }
 519                 }
 520
 521                 nt_status = gensec_spnego_parse_negTokenInit(gensec_security,
 522                                                              spnego_state,
 523                                                              out_mem_ctx,
 524                                                              spnego.negTokenInit.mechTypes,
 525                                                              spnego.negTokenInit.mechToken,
 526                                                              &unwrapped_out);
 527
 528                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(nt_status)) {
 529                         spnego_free_data(&spnego);
 530                         return nt_status;
 531                 }
 532
 533                 /* compose reply */
 534                 my_mechs[0] = spnego_state->sub_sec_security->ops->oid;
 535
 536                 spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
 537                 spnego_out.negTokenInit.mechTypes = my_mechs;
 538                 spnego_out.negTokenInit.reqFlags = 0;
 539                 spnego_out.negTokenInit.mechListMIC = null_data_blob;
 540                 spnego_out.negTokenInit.mechToken = unwrapped_out;
 541
 542                 if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 543                         DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n"));
 544                                 return NT_STATUS_INVALID_PARAMETER;
 545                 }
 546
 547                 /* set next state */
 548                 spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 549                 spnego_state->state_position = SPNEGO_CLIENT_TARG;
 550
 551                 return NT_STATUS_MORE_PROCESSING_REQUIRED;
 552         }
 553         case SPNEGO_SERVER_TARG:
 554         {
 555                 NTSTATUS nt_status;
 556                 if (!in.length) {
 557                         return NT_STATUS_INVALID_PARAMETER;
 558                 }
 559
 560                 len = spnego_read_data(in, &spnego);
 561
 562                 if (len == -1) {
 563                         DEBUG(1, ("Invalid SPNEGO request:\n"));
 564                         dump_data(1, (const char *)in.data, in.length);
 565                         return NT_STATUS_INVALID_PARAMETER;
 566                 }
 567
 568                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
 569                 if (spnego.type != spnego_state->expected_packet) {
 570                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 571                                   spnego_state->expected_packet));
 572                         dump_data(1, (const char *)in.data, in.length);
 573                         spnego_free_data(&spnego);
 574                         return NT_STATUS_INVALID_PARAMETER;
 575                 }
 576
 577                 nt_status = gensec_update(spnego_state->sub_sec_security,
 578                                           out_mem_ctx,
 579                                           spnego.negTokenTarg.responseToken,
 580                                           &unwrapped_out);
 581
 582                 nt_status = gensec_spnego_server_negTokenTarg(gensec_security,
 583                                                               spnego_state,
 584                                                               out_mem_ctx,
 585                                                               nt_status,
 586                                                               unwrapped_out,
 587                                                               out);
 588
 589                 spnego_free_data(&spnego);
 590
 591                 return nt_status;
 592         }
 593         case SPNEGO_CLIENT_TARG:
 594         {
 595                 NTSTATUS nt_status;
 596                 if (!in.length) {
 597                         return NT_STATUS_INVALID_PARAMETER;
 598                 }
 599
 600                 len = spnego_read_data(in, &spnego);
 601
 602                 if (len == -1) {
 603                         DEBUG(1, ("Invalid SPNEGO request:\n"));
 604                         dump_data(1, (const char *)in.data, in.length);
 605                         return NT_STATUS_INVALID_PARAMETER;
 606                 }
 607
 608                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
 609                 if (spnego.type != spnego_state->expected_packet) {
 610                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 611                                   spnego_state->expected_packet));
 612                         dump_data(1, (const char *)in.data, in.length);
 613                         spnego_free_data(&spnego);
 614                         return NT_STATUS_INVALID_PARAMETER;
 615                 }
 616
 617                 if (spnego.negTokenTarg.negResult == SPNEGO_REJECT) {
 618                         return NT_STATUS_ACCESS_DENIED;
 619                 }
 620
 621                 nt_status = gensec_update(spnego_state->sub_sec_security,
 622                                           out_mem_ctx,
 623                                           spnego.negTokenTarg.responseToken,
 624                                           &unwrapped_out);
 625
 626
 627                 if (NT_STATUS_IS_OK(nt_status)
 628                     && (spnego.negTokenTarg.negResult != SPNEGO_ACCEPT_COMPLETED)) {
 629                         DEBUG(1,("gensec_update ok but not accepted\n"));
 630                         nt_status = NT_STATUS_INVALID_PARAMETER;
 631                 }
 632
 633                 spnego_free_data(&spnego);
 634
 635                 if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 636                         /* compose reply */
 637                         spnego_out.type = SPNEGO_NEG_TOKEN_TARG;
 638                         spnego_out.negTokenTarg.negResult = SPNEGO_NONE_RESULT;
 639                         spnego_out.negTokenTarg.supportedMech = NULL;
 640                         spnego_out.negTokenTarg.responseToken = unwrapped_out;
 641                         spnego_out.negTokenTarg.mechListMIC = null_data_blob;
 642
 643                         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 644                                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_TARG\n"));
 645                                 return NT_STATUS_INVALID_PARAMETER;
 646                         }
 647
 648                         spnego_state->state_position = SPNEGO_CLIENT_TARG;
 649                 } else if (NT_STATUS_IS_OK(nt_status)) {
 650                         /* all done - server has accepted, and we agree */
 651
 652                         if (unwrapped_out.length) {
 653                                 spnego_out.type = SPNEGO_NEG_TOKEN_TARG;
 654                                 spnego_out.negTokenTarg.negResult = SPNEGO_NONE_RESULT;
 655                                 spnego_out.negTokenTarg.supportedMech = NULL;
 656                                 spnego_out.negTokenTarg.responseToken = unwrapped_out;
 657                                 spnego_out.negTokenTarg.mechListMIC = null_data_blob;
 658
 659                                 if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 660                                         DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_TARG\n"));
 661                                         return NT_STATUS_INVALID_PARAMETER;
 662                                 }
 663                         } else {
 664                                 *out = null_data_blob;
 665                         }
 666
 667                         spnego_state->state_position = SPNEGO_DONE;
 668                 } else {
 669                         DEBUG(1, ("SPNEGO(%s) login failed: %s\n",
 670                                   spnego_state->sub_sec_security->ops->name,
 671                                   nt_errstr(nt_status)));
 672                 }
 673                 return nt_status;
 674         }
 675         case SPNEGO_DONE:
 676                 return NT_STATUS_OK;
 677         }
 678         return NT_STATUS_INVALID_PARAMETER;
 679 }
 680
 681 static void gensec_spnego_end(struct gensec_security *gensec_security)
 682 {
 683         struct spnego_state *spnego_state = gensec_security->private_data;
 684
 685         if (spnego_state->sub_sec_security) {
 686                 gensec_end(&spnego_state->sub_sec_security);
 687         }
 688
 689         talloc_free(spnego_state);
 690
 691         gensec_security->private_data = NULL;
 692 }
 693
 694 static const struct gensec_security_ops gensec_spnego_security_ops = {
 695         .name           = "spnego",
 696         .sasl_name      = "GSS-SPNEGO",
 697         .auth_type      = DCERPC_AUTH_TYPE_SPNEGO,
 698         .oid            = OID_SPNEGO,
 699         .client_start   = gensec_spnego_client_start,
 700         .server_start   = gensec_spnego_server_start,
 701         .update         = gensec_spnego_update,
 702         .seal_packet    = gensec_spnego_seal_packet,
 703         .sign_packet    = gensec_spnego_sign_packet,
 704         .sig_size       = gensec_spnego_sig_size,
 705         .check_packet   = gensec_spnego_check_packet,
 706         .unseal_packet  = gensec_spnego_unseal_packet,
 707         .session_key    = gensec_spnego_session_key,
 708         .session_info   = gensec_spnego_session_info,
 709         .end            = gensec_spnego_end
 710 };
 711
 712 NTSTATUS gensec_spnego_init(void)
 713 {
 714         NTSTATUS ret;
 715         ret = register_backend("gensec", &gensec_spnego_security_ops);
 716         if (!NT_STATUS_IS_OK(ret)) {
 717                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
 718                         gensec_spnego_security_ops.name));
 719                 return ret;
 720         }
 721
 722         return ret;
 723 }