source/auth/gensec/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 #include "auth/auth.h"
  27
  28 #undef DBGC_CLASS
  29 #define DBGC_CLASS DBGC_AUTH
  30
  31 enum spnego_state_position {
  32         SPNEGO_SERVER_START,
  33         SPNEGO_CLIENT_START,
  34         SPNEGO_SERVER_TARG,
  35         SPNEGO_CLIENT_TARG,
  36         SPNEGO_FALLBACK,
  37         SPNEGO_DONE
  38 };
  39
  40 struct spnego_state {
  41         uint_t ref_count;
  42         enum spnego_message_type expected_packet;
  43         enum spnego_state_position state_position;
  44         struct gensec_security *sub_sec_security;
  45         BOOL no_response_expected;
  46 };
  47
  48
  49 static NTSTATUS gensec_spnego_client_start(struct gensec_security *gensec_security)
  50 {
  51         struct spnego_state *spnego_state;
  52
  53         spnego_state = talloc(gensec_security, struct spnego_state);
  54         if (!spnego_state) {
  55                 return NT_STATUS_NO_MEMORY;
  56         }
  57
  58         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
  59         spnego_state->state_position = SPNEGO_CLIENT_START;
  60         spnego_state->sub_sec_security = NULL;
  61         spnego_state->no_response_expected = False;
  62
  63         gensec_security->private_data = spnego_state;
  64         return NT_STATUS_OK;
  65 }
  66
  67 static NTSTATUS gensec_spnego_server_start(struct gensec_security *gensec_security)
  68 {
  69         struct spnego_state *spnego_state;
  70
  71         spnego_state = talloc(gensec_security, struct spnego_state);
  72         if (!spnego_state) {
  73                 return NT_STATUS_NO_MEMORY;
  74         }
  75
  76         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
  77         spnego_state->state_position = SPNEGO_SERVER_START;
  78         spnego_state->sub_sec_security = NULL;
  79         spnego_state->no_response_expected = False;
  80
  81         gensec_security->private_data = spnego_state;
  82         return NT_STATUS_OK;
  83 }
  84
  85 /*
  86   wrappers for the spnego_*() functions
  87 */
  88 static NTSTATUS gensec_spnego_unseal_packet(struct gensec_security *gensec_security,
  89                                             TALLOC_CTX *mem_ctx,
  90                                             uint8_t *data, size_t length,
  91                                             const uint8_t *whole_pdu, size_t pdu_length,
  92                                             const DATA_BLOB *sig)
  93 {
  94         struct spnego_state *spnego_state = gensec_security->private_data;
  95
  96         if (spnego_state->state_position != SPNEGO_DONE
  97             && spnego_state->state_position != SPNEGO_FALLBACK) {
  98                 return NT_STATUS_INVALID_PARAMETER;
  99         }
 100
 101         return gensec_unseal_packet(spnego_state->sub_sec_security,
 102                                     mem_ctx,
 103                                     data, length,
 104                                     whole_pdu, pdu_length,
 105                                     sig);
 106 }
 107
 108 static NTSTATUS gensec_spnego_check_packet(struct gensec_security *gensec_security,
 109                                            TALLOC_CTX *mem_ctx,
 110                                            const uint8_t *data, size_t length,
 111                                            const uint8_t *whole_pdu, size_t pdu_length,
 112                                            const DATA_BLOB *sig)
 113 {
 114         struct spnego_state *spnego_state = gensec_security->private_data;
 115
 116         if (spnego_state->state_position != SPNEGO_DONE
 117             && spnego_state->state_position != SPNEGO_FALLBACK) {
 118                 return NT_STATUS_INVALID_PARAMETER;
 119         }
 120
 121         return gensec_check_packet(spnego_state->sub_sec_security,
 122                                    mem_ctx,
 123                                    data, length,
 124                                    whole_pdu, pdu_length,
 125                                    sig);
 126 }
 127
 128 static NTSTATUS gensec_spnego_seal_packet(struct gensec_security *gensec_security,
 129                                           TALLOC_CTX *mem_ctx,
 130                                           uint8_t *data, size_t length,
 131                                           const uint8_t *whole_pdu, size_t pdu_length,
 132                                           DATA_BLOB *sig)
 133 {
 134         struct spnego_state *spnego_state = gensec_security->private_data;
 135
 136         if (spnego_state->state_position != SPNEGO_DONE
 137             && spnego_state->state_position != SPNEGO_FALLBACK) {
 138                 return NT_STATUS_INVALID_PARAMETER;
 139         }
 140
 141         return gensec_seal_packet(spnego_state->sub_sec_security,
 142                                   mem_ctx,
 143                                   data, length,
 144                                   whole_pdu, pdu_length,
 145                                   sig);
 146 }
 147
 148 static NTSTATUS gensec_spnego_sign_packet(struct gensec_security *gensec_security,
 149                                           TALLOC_CTX *mem_ctx,
 150                                           const uint8_t *data, size_t length,
 151                                           const uint8_t *whole_pdu, size_t pdu_length,
 152                                           DATA_BLOB *sig)
 153 {
 154         struct spnego_state *spnego_state = gensec_security->private_data;
 155
 156         if (spnego_state->state_position != SPNEGO_DONE
 157             && spnego_state->state_position != SPNEGO_FALLBACK) {
 158                 return NT_STATUS_INVALID_PARAMETER;
 159         }
 160
 161         return gensec_sign_packet(spnego_state->sub_sec_security,
 162                                   mem_ctx,
 163                                   data, length,
 164                                   whole_pdu, pdu_length,
 165                                   sig);
 166 }
 167
 168 static NTSTATUS gensec_spnego_wrap(struct gensec_security *gensec_security,
 169                                    TALLOC_CTX *mem_ctx,
 170                                    const DATA_BLOB *in,
 171                                    DATA_BLOB *out)
 172 {
 173         struct spnego_state *spnego_state = gensec_security->private_data;
 174
 175         if (spnego_state->state_position != SPNEGO_DONE
 176             && spnego_state->state_position != SPNEGO_FALLBACK) {
 177                 DEBUG(1, ("gensec_spnego_wrap: wrong state for wrap\n"));
 178                 return NT_STATUS_INVALID_PARAMETER;
 179         }
 180
 181         return gensec_wrap(spnego_state->sub_sec_security,
 182                            mem_ctx, in, out);
 183 }
 184
 185 static NTSTATUS gensec_spnego_unwrap(struct gensec_security *gensec_security,
 186                                      TALLOC_CTX *mem_ctx,
 187                                      const DATA_BLOB *in,
 188                                      DATA_BLOB *out)
 189 {
 190         struct spnego_state *spnego_state = gensec_security->private_data;
 191
 192         if (spnego_state->state_position != SPNEGO_DONE
 193             && spnego_state->state_position != SPNEGO_FALLBACK) {
 194                 DEBUG(1, ("gensec_spnego_unwrap: wrong state for unwrap\n"));
 195                 return NT_STATUS_INVALID_PARAMETER;
 196         }
 197
 198         return gensec_unwrap(spnego_state->sub_sec_security,
 199                              mem_ctx, in, out);
 200 }
 201
 202 static size_t gensec_spnego_sig_size(struct gensec_security *gensec_security)
 203 {
 204         struct spnego_state *spnego_state = gensec_security->private_data;
 205
 206         if (spnego_state->state_position != SPNEGO_DONE
 207             && spnego_state->state_position != SPNEGO_FALLBACK) {
 208                 return 0;
 209         }
 210
 211         return gensec_sig_size(spnego_state->sub_sec_security);
 212 }
 213
 214 static NTSTATUS gensec_spnego_session_key(struct gensec_security *gensec_security,
 215                                           DATA_BLOB *session_key)
 216 {
 217         struct spnego_state *spnego_state = gensec_security->private_data;
 218         if (!spnego_state->sub_sec_security) {
 219                 return NT_STATUS_INVALID_PARAMETER;
 220         }
 221
 222         return gensec_session_key(spnego_state->sub_sec_security,
 223                                   session_key);
 224 }
 225
 226 static NTSTATUS gensec_spnego_session_info(struct gensec_security *gensec_security,
 227                                                                       struct auth_session_info **session_info)
 228 {
 229         struct spnego_state *spnego_state = gensec_security->private_data;
 230         if (!spnego_state->sub_sec_security) {
 231                 return NT_STATUS_INVALID_PARAMETER;
 232         }
 233
 234         return gensec_session_info(spnego_state->sub_sec_security,
 235                                    session_info);
 236 }
 237
 238 /** Fallback to another GENSEC mechanism, based on magic strings
 239  *
 240  * This is the 'fallback' case, where we don't get SPNEGO, and have to
 241  * try all the other options (and hope they all have a magic string
 242  * they check)
 243 */
 244
 245 static NTSTATUS gensec_spnego_server_try_fallback(struct gensec_security *gensec_security,
 246                                                   struct spnego_state *spnego_state,
 247                                                   TALLOC_CTX *out_mem_ctx,
 248                                                   const DATA_BLOB in, DATA_BLOB *out)
 249 {
 250         int i;
 251         int num_ops;
 252         const struct gensec_security_ops **all_ops = gensec_security_all(&num_ops);
 253         for (i=0; i < num_ops; i++) {
 254                 NTSTATUS nt_status;
 255                 if (!all_ops[i]->oid) {
 256                         continue;
 257                 }
 258                 if (strcasecmp(GENSEC_OID_SPNEGO,all_ops[i]->oid) == 0) {
 259                         continue;
 260                 }
 261
 262                 nt_status = gensec_subcontext_start(spnego_state,
 263                                                     gensec_security,
 264                                                     &spnego_state->sub_sec_security);
 265                 if (!NT_STATUS_IS_OK(nt_status)) {
 266                         return nt_status;
 267                 }
 268                 /* select the sub context */
 269                 nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
 270                                                      all_ops[i]->oid);
 271                 if (!NT_STATUS_IS_OK(nt_status)) {
 272                         talloc_free(spnego_state->sub_sec_security);
 273                         spnego_state->sub_sec_security = NULL;
 274                         continue;
 275                 }
 276                 nt_status = gensec_update(spnego_state->sub_sec_security,
 277                                                           out_mem_ctx, in, out);
 278                 if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 279                         spnego_state->state_position = SPNEGO_FALLBACK;
 280                         return nt_status;
 281                 }
 282                 talloc_free(spnego_state->sub_sec_security);
 283                 spnego_state->sub_sec_security = NULL;
 284         }
 285         DEBUG(1, ("Failed to parse SPNEGO request\n"));
 286         return NT_STATUS_INVALID_PARAMETER;
 287
 288 }
 289
 290 /*
 291    Parse the netTokenInit from the client, to the server.
 292
 293
 294 */
 295
 296 static NTSTATUS gensec_spnego_server_parse_negTokenInit(struct gensec_security *gensec_security,
 297                                                         struct spnego_state *spnego_state,
 298                                                         TALLOC_CTX *out_mem_ctx,
 299                                                         const char **mechType,
 300                                                         const DATA_BLOB unwrapped_in, DATA_BLOB *unwrapped_out)
 301 {
 302         NTSTATUS nt_status;
 303
 304         if (!mechType || !mechType[0]) {
 305                 DEBUG(1, ("SPNEGO: Could not find a suitable mechtype in NEG_TOKEN_INIT\n"));
 306                 return NT_STATUS_INVALID_PARAMETER;
 307         }
 308
 309         nt_status = gensec_subcontext_start(spnego_state,
 310                                             gensec_security,
 311                                             &spnego_state->sub_sec_security);
 312         if (!NT_STATUS_IS_OK(nt_status)) {
 313                 return nt_status;
 314         }
 315         /* select the sub context */
 316         nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
 317                                              mechType[0]);
 318         if (!NT_STATUS_IS_OK(nt_status)) {
 319                 talloc_free(spnego_state->sub_sec_security);
 320                 spnego_state->sub_sec_security = NULL;
 321                 return nt_status;
 322         }
 323
 324         if (!unwrapped_in.length) {
 325                 return NT_STATUS_INVALID_PARAMETER;
 326         }
 327
 328         nt_status = gensec_update(spnego_state->sub_sec_security,
 329                                   out_mem_ctx,
 330                                   unwrapped_in,
 331                                   unwrapped_out);
 332
 333         if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(nt_status)) {
 334                 DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed: %s\n",
 335                           spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
 336                 talloc_free(spnego_state->sub_sec_security);
 337                 spnego_state->sub_sec_security = NULL;
 338         }
 339         return nt_status;
 340 }
 341
 342 static NTSTATUS gensec_spnego_client_parse_negTokenInit(struct gensec_security *gensec_security,
 343                                                         struct spnego_state *spnego_state,
 344                                                         TALLOC_CTX *out_mem_ctx,
 345                                                         const char **mechType,
 346                                                         const DATA_BLOB unwrapped_in, DATA_BLOB *unwrapped_out)
 347 {
 348         int i;
 349         NTSTATUS nt_status;
 350         DATA_BLOB null_data_blob = data_blob(NULL,0);
 351
 352         for (i=0; mechType && mechType[i]; i++) {
 353                 nt_status = gensec_subcontext_start(spnego_state,
 354                                                     gensec_security,
 355                                                     &spnego_state->sub_sec_security);
 356                 if (!NT_STATUS_IS_OK(nt_status)) {
 357                         break;
 358                 }
 359                 /* select the sub context */
 360                 nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
 361                                                      mechType[i]);
 362                 if (!NT_STATUS_IS_OK(nt_status)) {
 363                         talloc_free(spnego_state->sub_sec_security);
 364                         spnego_state->sub_sec_security = NULL;
 365                         continue;
 366                 }
 367
 368                 if (i == 0) {
 369                         nt_status = gensec_update(spnego_state->sub_sec_security,
 370                                                   out_mem_ctx,
 371                                                   unwrapped_in,
 372                                                   unwrapped_out);
 373                 } else {
 374                         /* only get the helping start blob for the first OID */
 375                         nt_status = gensec_update(spnego_state->sub_sec_security,
 376                                                   out_mem_ctx,
 377                                                   null_data_blob,
 378                                                   unwrapped_out);
 379                 }
 380                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(nt_status)) {
 381                         DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed: %s\n",
 382                                   spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
 383                         talloc_free(spnego_state->sub_sec_security);
 384                         spnego_state->sub_sec_security = NULL;
 385                         /* If the mech failed on first packet generation, pretend it never actually started */
 386                         continue;
 387                 }
 388                 return nt_status;
 389         }
 390         if (!mechType || !mechType[i]) {
 391                 DEBUG(1, ("SPNEGO: Could not find a suitable mechtype in NEG_TOKEN_INIT\n"));
 392         }
 393         return NT_STATUS_INVALID_PARAMETER;
 394 }
 395
 396 /** create a client negTokenInit
 397  *
 398  * This is the case, where the client is the first one who sends data
 399 */
 400
 401 static NTSTATUS gensec_spnego_client_negTokenInit(struct gensec_security *gensec_security,
 402                                                   struct spnego_state *spnego_state,
 403                                                   TALLOC_CTX *out_mem_ctx,
 404                                                   const DATA_BLOB in, DATA_BLOB *out)
 405 {
 406         DATA_BLOB null_data_blob = data_blob(NULL,0);
 407         NTSTATUS nt_status;
 408         const char **mechTypes = NULL;
 409         DATA_BLOB unwrapped_out = data_blob(NULL,0);
 410
 411         mechTypes = gensec_security_oids(out_mem_ctx, GENSEC_OID_SPNEGO);
 412
 413         if (!mechTypes) {
 414                 DEBUG(1, ("no GENSEC OID backends available\n"));
 415                 return NT_STATUS_INVALID_PARAMETER;
 416         }
 417
 418         nt_status = gensec_subcontext_start(spnego_state,
 419                                             gensec_security,
 420                                             &spnego_state->sub_sec_security);
 421         if (!NT_STATUS_IS_OK(nt_status)) {
 422                 return nt_status;
 423         }
 424         /* select our preferred mech */
 425         nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
 426                                              mechTypes[0]);
 427         if (!NT_STATUS_IS_OK(nt_status)) {
 428                 talloc_free(spnego_state->sub_sec_security);
 429                 spnego_state->sub_sec_security = NULL;
 430                 return nt_status;
 431         }
 432         nt_status = gensec_update(spnego_state->sub_sec_security,
 433                                   out_mem_ctx, in, &unwrapped_out);
 434         if (NT_STATUS_IS_OK(nt_status) || NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 435                 struct spnego_data spnego_out;
 436                 spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
 437                 spnego_out.negTokenInit.mechTypes = mechTypes;
 438                 spnego_out.negTokenInit.reqFlags = 0;
 439                 spnego_out.negTokenInit.mechListMIC = null_data_blob;
 440                 spnego_out.negTokenInit.mechToken = unwrapped_out;
 441
 442                 if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 443                         DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n"));
 444                                 return NT_STATUS_INVALID_PARAMETER;
 445                 }
 446
 447                 /* set next state */
 448                 spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 449                 spnego_state->state_position = SPNEGO_CLIENT_TARG;
 450
 451                 if (NT_STATUS_IS_OK(nt_status)) {
 452                         spnego_state->no_response_expected = True;
 453                 }
 454
 455                 return NT_STATUS_MORE_PROCESSING_REQUIRED;
 456         }
 457         talloc_free(spnego_state->sub_sec_security);
 458         spnego_state->sub_sec_security = NULL;
 459
 460         DEBUG(1, ("Failed to setup SPNEGO negTokenInit request: %s\n", nt_errstr(nt_status)));
 461         return NT_STATUS_INVALID_PARAMETER;
 462 }
 463
 464
 465 /** create a client negTokenTarg
 466  *
 467  * This is the case, where the client is the first one who sends data
 468 */
 469
 470 static NTSTATUS gensec_spnego_server_negTokenTarg(struct gensec_security *gensec_security,
 471                                                   struct spnego_state *spnego_state,
 472                                                   TALLOC_CTX *out_mem_ctx,
 473                                                   NTSTATUS nt_status,
 474                                                   const DATA_BLOB unwrapped_out, DATA_BLOB *out)
 475 {
 476         struct spnego_data spnego_out;
 477         DATA_BLOB null_data_blob = data_blob(NULL, 0);
 478
 479         /* compose reply */
 480         spnego_out.type = SPNEGO_NEG_TOKEN_TARG;
 481         spnego_out.negTokenTarg.responseToken = unwrapped_out;
 482         spnego_out.negTokenTarg.mechListMIC = null_data_blob;
 483         spnego_out.negTokenTarg.supportedMech = NULL;
 484
 485         if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 486                 spnego_out.negTokenTarg.supportedMech
 487                         = spnego_state->sub_sec_security->ops->oid;
 488                 spnego_out.negTokenTarg.negResult = SPNEGO_ACCEPT_INCOMPLETE;
 489                 spnego_state->state_position = SPNEGO_SERVER_TARG;
 490         } else if (NT_STATUS_IS_OK(nt_status)) {
 491                 if (unwrapped_out.data) {
 492                         spnego_out.negTokenTarg.supportedMech
 493                                 = spnego_state->sub_sec_security->ops->oid;
 494                 }
 495                 spnego_out.negTokenTarg.negResult = SPNEGO_ACCEPT_COMPLETED;
 496                 spnego_state->state_position = SPNEGO_DONE;
 497         } else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER)) {
 498                 if (spnego_state->sub_sec_security) {
 499                         /* we have a mech, but we just didn't get the input parameter */
 500                         spnego_out.negTokenTarg.supportedMech
 501                                 = spnego_state->sub_sec_security->ops->oid;
 502                 } else {
 503                         const char **mechTypes = gensec_security_oids(out_mem_ctx, GENSEC_OID_SPNEGO);
 504                         if (!mechTypes) {
 505                                 DEBUG(1, ("no GENSEC OID backends available\n"));
 506                                 return NT_STATUS_INVALID_PARAMETER;
 507                         }
 508
 509                         nt_status = gensec_subcontext_start(spnego_state,
 510                                                             gensec_security,
 511                                                             &spnego_state->sub_sec_security);
 512                         if (!NT_STATUS_IS_OK(nt_status)) {
 513                                 return nt_status;
 514                         }
 515                         /* select our preferred mech */
 516                         nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
 517                                                              mechTypes[0]);
 518                         if (!NT_STATUS_IS_OK(nt_status)) {
 519                                 talloc_free(spnego_state->sub_sec_security);
 520                                 spnego_state->sub_sec_security = NULL;
 521                                 return nt_status;
 522                         }
 523
 524                         /* we should be sending the whole list here */
 525                         spnego_out.negTokenTarg.supportedMech = mechTypes[0];
 526                 }
 527
 528                 spnego_out.negTokenTarg.negResult = SPNEGO_ACCEPT_INCOMPLETE;
 529                 spnego_state->state_position = SPNEGO_SERVER_TARG;
 530                 nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 531         } else {
 532                 spnego_out.negTokenTarg.negResult = SPNEGO_REJECT;
 533                 DEBUG(2, ("SPNEGO login failed: %s\n", nt_errstr(nt_status)));
 534                 spnego_state->state_position = SPNEGO_DONE;
 535         }
 536
 537         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 538                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_TARG\n"));
 539                 return NT_STATUS_INVALID_PARAMETER;
 540         }
 541
 542         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 543
 544         return nt_status;
 545 }
 546
 547
 548 static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx,
 549                                      const DATA_BLOB in, DATA_BLOB *out)
 550 {
 551         struct spnego_state *spnego_state = gensec_security->private_data;
 552         DATA_BLOB null_data_blob = data_blob(NULL, 0);
 553         DATA_BLOB unwrapped_out = data_blob(NULL, 0);
 554         struct spnego_data spnego_out;
 555         struct spnego_data spnego;
 556
 557         ssize_t len;
 558
 559         *out = data_blob(NULL, 0);
 560
 561         if (!out_mem_ctx) {
 562                 out_mem_ctx = spnego_state;
 563         }
 564
 565         /* and switch into the state machine */
 566
 567         switch (spnego_state->state_position) {
 568         case SPNEGO_FALLBACK:
 569                 return gensec_update(spnego_state->sub_sec_security,
 570                                      out_mem_ctx, in, out);
 571         case SPNEGO_SERVER_START:
 572         {
 573                 if (in.length) {
 574                         NTSTATUS nt_status;
 575
 576                         len = spnego_read_data(in, &spnego);
 577                         if (len == -1) {
 578                                 return gensec_spnego_server_try_fallback(gensec_security, spnego_state, out_mem_ctx, in, out);
 579                         }
 580                         /* client sent NegTargetInit, we send NegTokenTarg */
 581
 582                         /* OK, so it's real SPNEGO, check the packet's the one we expect */
 583                         if (spnego.type != spnego_state->expected_packet) {
 584                                 DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 585                                           spnego_state->expected_packet));
 586                                 dump_data(1, in.data, in.length);
 587                                 spnego_free_data(&spnego);
 588                                 return NT_STATUS_INVALID_PARAMETER;
 589                         }
 590
 591                         nt_status = gensec_spnego_server_parse_negTokenInit(gensec_security,
 592                                                                      spnego_state,
 593                                                                      out_mem_ctx,
 594                                                                      spnego.negTokenInit.mechTypes,
 595                                                                      spnego.negTokenInit.mechToken,
 596                                                                      &unwrapped_out);
 597
 598                         nt_status = gensec_spnego_server_negTokenTarg(gensec_security,
 599                                                                       spnego_state,
 600                                                                       out_mem_ctx,
 601                                                                       nt_status,
 602                                                                       unwrapped_out,
 603                                                                       out);
 604
 605                         spnego_free_data(&spnego);
 606
 607                         return nt_status;
 608                 } else {
 609                         const char **mechlist = gensec_security_oids(out_mem_ctx, GENSEC_OID_SPNEGO);
 610
 611                         spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
 612                         spnego_out.negTokenInit.mechTypes = mechlist;
 613                         spnego_out.negTokenInit.reqFlags = 0;
 614                         spnego_out.negTokenInit.mechListMIC
 615                                 = data_blob_string_const(talloc_asprintf(out_mem_ctx, "%s$@%s", lp_netbios_name(), lp_realm()));
 616                         spnego_out.negTokenInit.mechToken = unwrapped_out;
 617
 618                         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 619                                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n"));
 620                                 return NT_STATUS_INVALID_PARAMETER;
 621                         }
 622
 623                         /* set next state */
 624                         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 625                         spnego_state->state_position = SPNEGO_SERVER_TARG;
 626
 627                         return NT_STATUS_MORE_PROCESSING_REQUIRED;
 628                 }
 629         }
 630
 631         case SPNEGO_CLIENT_START:
 632         {
 633                 /* The server offers a list of mechanisms */
 634
 635                 const char *my_mechs[] = {NULL, NULL};
 636                 NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
 637
 638                 if (!in.length) {
 639                         /* client to produce negTokenInit */
 640                         return gensec_spnego_client_negTokenInit(gensec_security, spnego_state,
 641                                                                  out_mem_ctx, in, out);
 642                 }
 643
 644                 len = spnego_read_data(in, &spnego);
 645
 646                 if (len == -1) {
 647                         DEBUG(1, ("Invalid SPNEGO request:\n"));
 648                         dump_data(1, in.data, in.length);
 649                         return NT_STATUS_INVALID_PARAMETER;
 650                 }
 651
 652                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
 653                 if (spnego.type != spnego_state->expected_packet) {
 654                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 655                                   spnego_state->expected_packet));
 656                         dump_data(1, in.data, in.length);
 657                         spnego_free_data(&spnego);
 658                         return NT_STATUS_INVALID_PARAMETER;
 659                 }
 660
 661                 if (spnego.negTokenInit.targetPrincipal) {
 662                         DEBUG(5, ("Server claims it's principal name is %s (ignored)\n", spnego.negTokenInit.targetPrincipal));
 663                 }
 664
 665                 nt_status = gensec_spnego_client_parse_negTokenInit(gensec_security,
 666                                                              spnego_state,
 667                                                              out_mem_ctx,
 668                                                              spnego.negTokenInit.mechTypes,
 669                                                              spnego.negTokenInit.mechToken,
 670                                                              &unwrapped_out);
 671
 672                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(nt_status)) {
 673                         spnego_free_data(&spnego);
 674                         return nt_status;
 675                 }
 676
 677                 /* compose reply */
 678                 my_mechs[0] = spnego_state->sub_sec_security->ops->oid;
 679
 680                 spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
 681                 spnego_out.negTokenInit.mechTypes = my_mechs;
 682                 spnego_out.negTokenInit.reqFlags = 0;
 683                 spnego_out.negTokenInit.mechListMIC = null_data_blob;
 684                 spnego_out.negTokenInit.mechToken = unwrapped_out;
 685
 686                 if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 687                         DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n"));
 688                                 return NT_STATUS_INVALID_PARAMETER;
 689                 }
 690
 691                 /* set next state */
 692                 spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 693                 spnego_state->state_position = SPNEGO_CLIENT_TARG;
 694
 695                 if (NT_STATUS_IS_OK(nt_status)) {
 696                         spnego_state->no_response_expected = True;
 697                 }
 698
 699                 return NT_STATUS_MORE_PROCESSING_REQUIRED;
 700         }
 701         case SPNEGO_SERVER_TARG:
 702         {
 703                 NTSTATUS nt_status;
 704                 if (!in.length) {
 705                         return NT_STATUS_INVALID_PARAMETER;
 706                 }
 707
 708                 len = spnego_read_data(in, &spnego);
 709
 710                 if (len == -1) {
 711                         DEBUG(1, ("Invalid SPNEGO request:\n"));
 712                         dump_data(1, in.data, in.length);
 713                         return NT_STATUS_INVALID_PARAMETER;
 714                 }
 715
 716                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
 717                 if (spnego.type != spnego_state->expected_packet) {
 718                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 719                                   spnego_state->expected_packet));
 720                         dump_data(1, in.data, in.length);
 721                         spnego_free_data(&spnego);
 722                         return NT_STATUS_INVALID_PARAMETER;
 723                 }
 724
 725                 if (!spnego_state->sub_sec_security) {
 726                         DEBUG(1, ("SPNEGO: Did not setup a mech in NEG_TOKEN_INIT\n"));
 727                         return NT_STATUS_INVALID_PARAMETER;
 728                 }
 729
 730                 nt_status = gensec_update(spnego_state->sub_sec_security,
 731                                           out_mem_ctx,
 732                                           spnego.negTokenTarg.responseToken,
 733                                           &unwrapped_out);
 734
 735                 nt_status = gensec_spnego_server_negTokenTarg(gensec_security,
 736                                                               spnego_state,
 737                                                               out_mem_ctx,
 738                                                               nt_status,
 739                                                               unwrapped_out,
 740                                                               out);
 741
 742                 spnego_free_data(&spnego);
 743
 744                 return nt_status;
 745         }
 746         case SPNEGO_CLIENT_TARG:
 747         {
 748                 NTSTATUS nt_status;
 749                 if (!in.length) {
 750                         return NT_STATUS_INVALID_PARAMETER;
 751                 }
 752
 753                 len = spnego_read_data(in, &spnego);
 754
 755                 if (len == -1) {
 756                         DEBUG(1, ("Invalid SPNEGO request:\n"));
 757                         dump_data(1, in.data, in.length);
 758                         return NT_STATUS_INVALID_PARAMETER;
 759                 }
 760
 761                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
 762                 if (spnego.type != spnego_state->expected_packet) {
 763                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 764                                   spnego_state->expected_packet));
 765                         dump_data(1, in.data, in.length);
 766                         spnego_free_data(&spnego);
 767                         return NT_STATUS_INVALID_PARAMETER;
 768                 }
 769
 770                 if (spnego.negTokenTarg.negResult == SPNEGO_REJECT) {
 771                         return NT_STATUS_ACCESS_DENIED;
 772                 }
 773
 774                 if (spnego_state->no_response_expected) {
 775                         if (spnego.negTokenTarg.negResult != SPNEGO_ACCEPT_COMPLETED) {
 776                                 DEBUG(3,("GENSEC SPNEGO: client GENSEC accepted, but server rejected (bad password?)\n"));
 777                                 nt_status = NT_STATUS_INVALID_PARAMETER;
 778                         } else if (spnego.negTokenTarg.responseToken.length) {
 779                                 DEBUG(2,("GENSEC SPNEGO: client GENSEC accepted, but server continued negotiation!\n"));
 780                                 nt_status = NT_STATUS_INVALID_PARAMETER;
 781                         } else {
 782                                 nt_status = NT_STATUS_OK;
 783                         }
 784                 } else {
 785                         nt_status = gensec_update(spnego_state->sub_sec_security,
 786                                                   out_mem_ctx,
 787                                                   spnego.negTokenTarg.responseToken,
 788                                                   &unwrapped_out);
 789
 790                         if (NT_STATUS_IS_OK(nt_status)) {
 791                                 spnego_state->no_response_expected = True;
 792                         }
 793                 }
 794
 795                 spnego_free_data(&spnego);
 796
 797                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)
 798                         && !NT_STATUS_IS_OK(nt_status)) {
 799                         DEBUG(1, ("SPNEGO(%s) login failed: %s\n",
 800                                   spnego_state->sub_sec_security->ops->name,
 801                                   nt_errstr(nt_status)));
 802                         return nt_status;
 803                 }
 804
 805                 if (unwrapped_out.length) {
 806                         /* compose reply */
 807                         spnego_out.type = SPNEGO_NEG_TOKEN_TARG;
 808                         spnego_out.negTokenTarg.negResult = SPNEGO_NONE_RESULT;
 809                         spnego_out.negTokenTarg.supportedMech = NULL;
 810                         spnego_out.negTokenTarg.responseToken = unwrapped_out;
 811                         spnego_out.negTokenTarg.mechListMIC = null_data_blob;
 812
 813                         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 814                                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_TARG\n"));
 815                                 return NT_STATUS_INVALID_PARAMETER;
 816                         }
 817
 818                         spnego_state->state_position = SPNEGO_CLIENT_TARG;
 819                         nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 820                 } else {
 821
 822                         /* all done - server has accepted, and we agree */
 823                         *out = null_data_blob;
 824
 825                         if (spnego.negTokenTarg.negResult != SPNEGO_ACCEPT_COMPLETED) {
 826                                 /* unless of course it did not accept */
 827                                 DEBUG(1,("gensec_update ok but not accepted\n"));
 828                                 nt_status = NT_STATUS_INVALID_PARAMETER;
 829                         }
 830                 }
 831
 832                 spnego_state->state_position = SPNEGO_DONE;
 833
 834                 return nt_status;
 835         }
 836         case SPNEGO_DONE:
 837                 return NT_STATUS_OK;
 838         }
 839         return NT_STATUS_INVALID_PARAMETER;
 840 }
 841
 842 static BOOL gensec_spnego_have_feature(struct gensec_security *gensec_security,
 843                                        uint32_t feature)
 844 {
 845         struct spnego_state *spnego_state = gensec_security->private_data;
 846         if (!spnego_state->sub_sec_security) {
 847                 return False;
 848         }
 849
 850         return gensec_have_feature(spnego_state->sub_sec_security,
 851                                    feature);
 852 }
 853
 854 static const struct gensec_security_ops gensec_spnego_security_ops = {
 855         .name           = "spnego",
 856         .sasl_name      = "GSS-SPNEGO",
 857         .auth_type      = DCERPC_AUTH_TYPE_SPNEGO,
 858         .oid            = GENSEC_OID_SPNEGO,
 859         .client_start   = gensec_spnego_client_start,
 860         .server_start   = gensec_spnego_server_start,
 861         .update         = gensec_spnego_update,
 862         .seal_packet    = gensec_spnego_seal_packet,
 863         .sign_packet    = gensec_spnego_sign_packet,
 864         .sig_size       = gensec_spnego_sig_size,
 865         .check_packet   = gensec_spnego_check_packet,
 866         .unseal_packet  = gensec_spnego_unseal_packet,
 867         .wrap           = gensec_spnego_wrap,
 868         .unwrap         = gensec_spnego_unwrap,
 869         .session_key    = gensec_spnego_session_key,
 870         .session_info   = gensec_spnego_session_info,
 871         .have_feature   = gensec_spnego_have_feature,
 872         .enabled        = True
 873 };
 874
 875 NTSTATUS gensec_spnego_init(void)
 876 {
 877         NTSTATUS ret;
 878         ret = gensec_register(&gensec_spnego_security_ops);
 879         if (!NT_STATUS_IS_OK(ret)) {
 880                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
 881                         gensec_spnego_security_ops.name));
 882                 return ret;
 883         }
 884
 885         return ret;
 886 }