source4/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-2005
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 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 #include "auth/gensec/spnego_proto.h"
  28 #include "librpc/gen_ndr/ndr_dcerpc.h"
  29 #include "lib/socket/socket.h"
  30 #include "auth/gensec/socket.h"
  31
  32 enum spnego_state_position {
  33         SPNEGO_SERVER_START,
  34         SPNEGO_CLIENT_START,
  35         SPNEGO_SERVER_TARG,
  36         SPNEGO_CLIENT_TARG,
  37         SPNEGO_FALLBACK,
  38         SPNEGO_DONE
  39 };
  40
  41 struct spnego_state {
  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         const char *neg_oid;
  48 };
  49
  50
  51 static NTSTATUS gensec_spnego_client_start(struct gensec_security *gensec_security)
  52 {
  53         struct spnego_state *spnego_state;
  54
  55         spnego_state = talloc(gensec_security, struct spnego_state);
  56         if (!spnego_state) {
  57                 return NT_STATUS_NO_MEMORY;
  58         }
  59
  60         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
  61         spnego_state->state_position = SPNEGO_CLIENT_START;
  62         spnego_state->sub_sec_security = NULL;
  63         spnego_state->no_response_expected = False;
  64
  65         gensec_security->private_data = spnego_state;
  66         return NT_STATUS_OK;
  67 }
  68
  69 static NTSTATUS gensec_spnego_server_start(struct gensec_security *gensec_security)
  70 {
  71         struct spnego_state *spnego_state;
  72
  73         spnego_state = talloc(gensec_security, struct spnego_state);
  74         if (!spnego_state) {
  75                 return NT_STATUS_NO_MEMORY;
  76         }
  77
  78         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
  79         spnego_state->state_position = SPNEGO_SERVER_START;
  80         spnego_state->sub_sec_security = NULL;
  81         spnego_state->no_response_expected = False;
  82
  83         gensec_security->private_data = spnego_state;
  84         return NT_STATUS_OK;
  85 }
  86
  87 /*
  88   wrappers for the spnego_*() functions
  89 */
  90 static NTSTATUS gensec_spnego_unseal_packet(struct gensec_security *gensec_security,
  91                                             TALLOC_CTX *mem_ctx,
  92                                             uint8_t *data, size_t length,
  93                                             const uint8_t *whole_pdu, size_t pdu_length,
  94                                             const DATA_BLOB *sig)
  95 {
  96         struct spnego_state *spnego_state = gensec_security->private_data;
  97
  98         if (spnego_state->state_position != SPNEGO_DONE
  99             && spnego_state->state_position != SPNEGO_FALLBACK) {
 100                 return NT_STATUS_INVALID_PARAMETER;
 101         }
 102
 103         return gensec_unseal_packet(spnego_state->sub_sec_security,
 104                                     mem_ctx,
 105                                     data, length,
 106                                     whole_pdu, pdu_length,
 107                                     sig);
 108 }
 109
 110 static NTSTATUS gensec_spnego_check_packet(struct gensec_security *gensec_security,
 111                                            TALLOC_CTX *mem_ctx,
 112                                            const uint8_t *data, size_t length,
 113                                            const uint8_t *whole_pdu, size_t pdu_length,
 114                                            const DATA_BLOB *sig)
 115 {
 116         struct spnego_state *spnego_state = gensec_security->private_data;
 117
 118         if (spnego_state->state_position != SPNEGO_DONE
 119             && spnego_state->state_position != SPNEGO_FALLBACK) {
 120                 return NT_STATUS_INVALID_PARAMETER;
 121         }
 122
 123         return gensec_check_packet(spnego_state->sub_sec_security,
 124                                    mem_ctx,
 125                                    data, length,
 126                                    whole_pdu, pdu_length,
 127                                    sig);
 128 }
 129
 130 static NTSTATUS gensec_spnego_seal_packet(struct gensec_security *gensec_security,
 131                                           TALLOC_CTX *mem_ctx,
 132                                           uint8_t *data, size_t length,
 133                                           const uint8_t *whole_pdu, size_t pdu_length,
 134                                           DATA_BLOB *sig)
 135 {
 136         struct spnego_state *spnego_state = gensec_security->private_data;
 137
 138         if (spnego_state->state_position != SPNEGO_DONE
 139             && spnego_state->state_position != SPNEGO_FALLBACK) {
 140                 return NT_STATUS_INVALID_PARAMETER;
 141         }
 142
 143         return gensec_seal_packet(spnego_state->sub_sec_security,
 144                                   mem_ctx,
 145                                   data, length,
 146                                   whole_pdu, pdu_length,
 147                                   sig);
 148 }
 149
 150 static NTSTATUS gensec_spnego_sign_packet(struct gensec_security *gensec_security,
 151                                           TALLOC_CTX *mem_ctx,
 152                                           const uint8_t *data, size_t length,
 153                                           const uint8_t *whole_pdu, size_t pdu_length,
 154                                           DATA_BLOB *sig)
 155 {
 156         struct spnego_state *spnego_state = gensec_security->private_data;
 157
 158         if (spnego_state->state_position != SPNEGO_DONE
 159             && spnego_state->state_position != SPNEGO_FALLBACK) {
 160                 return NT_STATUS_INVALID_PARAMETER;
 161         }
 162
 163         return gensec_sign_packet(spnego_state->sub_sec_security,
 164                                   mem_ctx,
 165                                   data, length,
 166                                   whole_pdu, pdu_length,
 167                                   sig);
 168 }
 169
 170 static NTSTATUS gensec_spnego_wrap(struct gensec_security *gensec_security,
 171                                    TALLOC_CTX *mem_ctx,
 172                                    const DATA_BLOB *in,
 173                                    DATA_BLOB *out)
 174 {
 175         struct spnego_state *spnego_state = gensec_security->private_data;
 176
 177         if (spnego_state->state_position != SPNEGO_DONE
 178             && spnego_state->state_position != SPNEGO_FALLBACK) {
 179                 DEBUG(1, ("gensec_spnego_wrap: wrong state for wrap\n"));
 180                 return NT_STATUS_INVALID_PARAMETER;
 181         }
 182
 183         return gensec_wrap(spnego_state->sub_sec_security,
 184                            mem_ctx, in, out);
 185 }
 186
 187 static NTSTATUS gensec_spnego_unwrap(struct gensec_security *gensec_security,
 188                                      TALLOC_CTX *mem_ctx,
 189                                      const DATA_BLOB *in,
 190                                      DATA_BLOB *out)
 191 {
 192         struct spnego_state *spnego_state = gensec_security->private_data;
 193
 194         if (spnego_state->state_position != SPNEGO_DONE
 195             && spnego_state->state_position != SPNEGO_FALLBACK) {
 196                 DEBUG(1, ("gensec_spnego_unwrap: wrong state for unwrap\n"));
 197                 return NT_STATUS_INVALID_PARAMETER;
 198         }
 199
 200         return gensec_unwrap(spnego_state->sub_sec_security,
 201                              mem_ctx, in, out);
 202 }
 203
 204 static NTSTATUS gensec_spnego_wrap_packets(struct gensec_security *gensec_security,
 205                                            TALLOC_CTX *mem_ctx,
 206                                            const DATA_BLOB *in,
 207                                            DATA_BLOB *out,
 208                                            size_t *len_processed)
 209 {
 210         struct spnego_state *spnego_state = gensec_security->private_data;
 211
 212         if (spnego_state->state_position != SPNEGO_DONE
 213             && spnego_state->state_position != SPNEGO_FALLBACK) {
 214                 DEBUG(1, ("gensec_spnego_wrap: wrong state for wrap\n"));
 215                 return NT_STATUS_INVALID_PARAMETER;
 216         }
 217
 218         return gensec_wrap_packets(spnego_state->sub_sec_security,
 219                                    mem_ctx, in, out,
 220                                    len_processed);
 221 }
 222
 223 static NTSTATUS gensec_spnego_packet_full_request(struct gensec_security *gensec_security,
 224                                                 DATA_BLOB blob, size_t *size)
 225 {
 226         struct spnego_state *spnego_state = gensec_security->private_data;
 227
 228         if (spnego_state->state_position != SPNEGO_DONE
 229             && spnego_state->state_position != SPNEGO_FALLBACK) {
 230                 DEBUG(1, ("gensec_spnego_unwrap: wrong state for unwrap\n"));
 231                 return NT_STATUS_INVALID_PARAMETER;
 232         }
 233
 234         return gensec_packet_full_request(spnego_state->sub_sec_security,
 235                                           blob, size);
 236 }
 237
 238 static NTSTATUS gensec_spnego_unwrap_packets(struct gensec_security *gensec_security,
 239                                              TALLOC_CTX *mem_ctx,
 240                                              const DATA_BLOB *in,
 241                                              DATA_BLOB *out,
 242                                              size_t *len_processed)
 243 {
 244         struct spnego_state *spnego_state = gensec_security->private_data;
 245
 246         if (spnego_state->state_position != SPNEGO_DONE
 247             && spnego_state->state_position != SPNEGO_FALLBACK) {
 248                 DEBUG(1, ("gensec_spnego_unwrap: wrong state for unwrap\n"));
 249                 return NT_STATUS_INVALID_PARAMETER;
 250         }
 251
 252         return gensec_unwrap_packets(spnego_state->sub_sec_security,
 253                                      mem_ctx, in, out,
 254                                      len_processed);
 255 }
 256
 257 static size_t gensec_spnego_sig_size(struct gensec_security *gensec_security, size_t data_size)
 258 {
 259         struct spnego_state *spnego_state = gensec_security->private_data;
 260
 261         if (spnego_state->state_position != SPNEGO_DONE
 262             && spnego_state->state_position != SPNEGO_FALLBACK) {
 263                 return 0;
 264         }
 265
 266         return gensec_sig_size(spnego_state->sub_sec_security, data_size);
 267 }
 268
 269 static size_t gensec_spnego_max_input_size(struct gensec_security *gensec_security)
 270 {
 271         struct spnego_state *spnego_state = gensec_security->private_data;
 272
 273         if (spnego_state->state_position != SPNEGO_DONE
 274             && spnego_state->state_position != SPNEGO_FALLBACK) {
 275                 return 0;
 276         }
 277
 278         return gensec_max_input_size(spnego_state->sub_sec_security);
 279 }
 280
 281 static size_t gensec_spnego_max_wrapped_size(struct gensec_security *gensec_security)
 282 {
 283         struct spnego_state *spnego_state = gensec_security->private_data;
 284
 285         if (spnego_state->state_position != SPNEGO_DONE
 286             && spnego_state->state_position != SPNEGO_FALLBACK) {
 287                 return 0;
 288         }
 289
 290         return gensec_max_wrapped_size(spnego_state->sub_sec_security);
 291 }
 292
 293 static NTSTATUS gensec_spnego_session_key(struct gensec_security *gensec_security,
 294                                           DATA_BLOB *session_key)
 295 {
 296         struct spnego_state *spnego_state = gensec_security->private_data;
 297         if (!spnego_state->sub_sec_security) {
 298                 return NT_STATUS_INVALID_PARAMETER;
 299         }
 300
 301         return gensec_session_key(spnego_state->sub_sec_security,
 302                                   session_key);
 303 }
 304
 305 static NTSTATUS gensec_spnego_session_info(struct gensec_security *gensec_security,
 306                                                                       struct auth_session_info **session_info)
 307 {
 308         struct spnego_state *spnego_state = gensec_security->private_data;
 309         if (!spnego_state->sub_sec_security) {
 310                 return NT_STATUS_INVALID_PARAMETER;
 311         }
 312
 313         return gensec_session_info(spnego_state->sub_sec_security,
 314                                    session_info);
 315 }
 316
 317 /** Fallback to another GENSEC mechanism, based on magic strings
 318  *
 319  * This is the 'fallback' case, where we don't get SPNEGO, and have to
 320  * try all the other options (and hope they all have a magic string
 321  * they check)
 322 */
 323
 324 static NTSTATUS gensec_spnego_server_try_fallback(struct gensec_security *gensec_security,
 325                                                   struct spnego_state *spnego_state,
 326                                                   TALLOC_CTX *out_mem_ctx,
 327                                                   const DATA_BLOB in, DATA_BLOB *out)
 328 {
 329         int i,j;
 330         struct gensec_security_ops **all_ops
 331                 = gensec_security_mechs(gensec_security, out_mem_ctx);
 332         for (i=0; all_ops[i]; i++) {
 333                 BOOL is_spnego;
 334                 NTSTATUS nt_status;
 335                 if (!all_ops[i]->oid) {
 336                         continue;
 337                 }
 338
 339                 is_spnego = False;
 340                 for (j=0; all_ops[i]->oid[j]; j++) {
 341                         if (strcasecmp(GENSEC_OID_SPNEGO,all_ops[i]->oid[j]) == 0) {
 342                                 is_spnego = True;
 343                         }
 344                 }
 345                 if (is_spnego) {
 346                         continue;
 347                 }
 348
 349                 if (!all_ops[i]->magic) {
 350                         continue;
 351                 }
 352
 353                 nt_status = all_ops[i]->magic(gensec_security, &in);
 354                 if (!NT_STATUS_IS_OK(nt_status)) {
 355                         continue;
 356                 }
 357
 358                 spnego_state->state_position = SPNEGO_FALLBACK;
 359
 360                 nt_status = gensec_subcontext_start(spnego_state,
 361                                                     gensec_security,
 362                                                     &spnego_state->sub_sec_security);
 363
 364                 if (!NT_STATUS_IS_OK(nt_status)) {
 365                         return nt_status;
 366                 }
 367                 /* select the sub context */
 368                 nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
 369                                                      all_ops[i]);
 370                 if (!NT_STATUS_IS_OK(nt_status)) {
 371                         return nt_status;
 372                 }
 373                 nt_status = gensec_update(spnego_state->sub_sec_security,
 374                                           out_mem_ctx, in, out);
 375                 return nt_status;
 376         }
 377         DEBUG(1, ("Failed to parse SPNEGO request\n"));
 378         return NT_STATUS_INVALID_PARAMETER;
 379
 380 }
 381
 382 /*
 383    Parse the netTokenInit, either from the client, to the server, or
 384    from the server to the client.
 385 */
 386
 387 static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_security,
 388                                                  struct spnego_state *spnego_state,
 389                                                  TALLOC_CTX *out_mem_ctx,
 390                                                  const char **mechType,
 391                                                  const DATA_BLOB unwrapped_in, DATA_BLOB *unwrapped_out)
 392 {
 393         int i;
 394         NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
 395         DATA_BLOB null_data_blob = data_blob(NULL,0);
 396
 397         const struct gensec_security_ops_wrapper *all_sec
 398                 = gensec_security_by_oid_list(gensec_security,
 399                                               out_mem_ctx,
 400                                               mechType,
 401                                               GENSEC_OID_SPNEGO);
 402         if (spnego_state->state_position == SPNEGO_SERVER_START) {
 403                 for (i=0; all_sec && all_sec[i].op; i++) {
 404                         /* optomisitic token */
 405                         if (strcmp(all_sec[i].oid, mechType[0]) == 0) {
 406                                 nt_status = gensec_subcontext_start(spnego_state,
 407                                                                     gensec_security,
 408                                                                     &spnego_state->sub_sec_security);
 409                                 if (!NT_STATUS_IS_OK(nt_status)) {
 410                                         return nt_status;
 411                                 }
 412                                 /* select the sub context */
 413                                 nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
 414                                                                      all_sec[i].op);
 415                                 if (!NT_STATUS_IS_OK(nt_status)) {
 416                                         talloc_free(spnego_state->sub_sec_security);
 417                                         spnego_state->sub_sec_security = NULL;
 418                                         break;
 419                                 }
 420
 421                                 nt_status = gensec_update(spnego_state->sub_sec_security,
 422                                                           out_mem_ctx,
 423                                                           unwrapped_in,
 424                                                           unwrapped_out);
 425                                 if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER) ||
 426                                     NT_STATUS_EQUAL(nt_status, NT_STATUS_CANT_ACCESS_DOMAIN_INFO)) {
 427                                         /* Pretend we never started it (lets the first run find some incompatible demand) */
 428
 429                                         DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed to parse: %s\n",
 430                                                   spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
 431                                         talloc_free(spnego_state->sub_sec_security);
 432                                         spnego_state->sub_sec_security = NULL;
 433                                         break;
 434                                 }
 435
 436                                 spnego_state->neg_oid = all_sec[i].oid;
 437                                 break;
 438                         }
 439                 }
 440         }
 441
 442         /* Having tried any optomisitc token from the client (if we
 443          * were the server), if we didn't get anywhere, walk our list
 444          * in our preference order */
 445
 446         if (!spnego_state->sub_sec_security) {
 447                 for (i=0; all_sec && all_sec[i].op; i++) {
 448                         nt_status = gensec_subcontext_start(spnego_state,
 449                                                             gensec_security,
 450                                                             &spnego_state->sub_sec_security);
 451                         if (!NT_STATUS_IS_OK(nt_status)) {
 452                                 return nt_status;
 453                         }
 454                         /* select the sub context */
 455                         nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
 456                                                              all_sec[i].op);
 457                         if (!NT_STATUS_IS_OK(nt_status)) {
 458                                 talloc_free(spnego_state->sub_sec_security);
 459                                 spnego_state->sub_sec_security = NULL;
 460                                 continue;
 461                         }
 462
 463                         spnego_state->neg_oid = all_sec[i].oid;
 464
 465                         /* only get the helping start blob for the first OID */
 466                         nt_status = gensec_update(spnego_state->sub_sec_security,
 467                                                   out_mem_ctx,
 468                                                   null_data_blob,
 469                                                   unwrapped_out);
 470
 471                         /* it is likely that a NULL input token will
 472                          * not be liked by most server mechs, but if
 473                          * we are in the client, we want the first
 474                          * update packet to be able to abort the use
 475                          * of this mech */
 476                         if (spnego_state->state_position != SPNEGO_SERVER_START) {
 477                                 if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER) ||
 478                                     NT_STATUS_EQUAL(nt_status, NT_STATUS_CANT_ACCESS_DOMAIN_INFO)) {
 479                                         /* Pretend we never started it (lets the first run find some incompatible demand) */
 480
 481                                         DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed to parse: %s\n",
 482                                                   spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
 483                                         talloc_free(spnego_state->sub_sec_security);
 484                                         spnego_state->sub_sec_security = NULL;
 485                                         continue;
 486                                 }
 487                         }
 488
 489                         break;
 490                 }
 491         }
 492
 493         if (spnego_state->sub_sec_security) {
 494                 /* it is likely that a NULL input token will
 495                  * not be liked by most server mechs, but this
 496                  * does the right thing in the CIFS client.
 497                  * just push us along the merry-go-round
 498                  * again, and hope for better luck next
 499                  * time */
 500
 501                 if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER)) {
 502                         *unwrapped_out = data_blob(NULL, 0);
 503                         nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 504                 }
 505
 506                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER)
 507                     && !NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)
 508                     && !NT_STATUS_IS_OK(nt_status)) {
 509                         DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed: %s\n",
 510                                   spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
 511                         talloc_free(spnego_state->sub_sec_security);
 512                         spnego_state->sub_sec_security = NULL;
 513
 514                         /* We started the mech correctly, and the
 515                          * input from the other side was valid.
 516                          * Return the error (say bad password, invalid
 517                          * ticket) */
 518                         return nt_status;
 519                 }
 520
 521
 522                 return nt_status; /* OK, INVALID_PARAMETER ore MORE PROCESSING */
 523         }
 524
 525         DEBUG(1, ("SPNEGO: Could not find a suitable mechtype in NEG_TOKEN_INIT\n"));
 526         /* we could re-negotiate here, but it would only work
 527          * if the client or server lied about what it could
 528          * support the first time.  Lets keep this code to
 529          * reality */
 530
 531         return NT_STATUS_INVALID_PARAMETER;
 532 }
 533
 534 /** create a negTokenInit
 535  *
 536  * This is the same packet, no matter if the client or server sends it first, but it is always the first packet
 537 */
 538 static NTSTATUS gensec_spnego_create_negTokenInit(struct gensec_security *gensec_security,
 539                                                   struct spnego_state *spnego_state,
 540                                                   TALLOC_CTX *out_mem_ctx,
 541                                                   const DATA_BLOB in, DATA_BLOB *out)
 542 {
 543         int i;
 544         NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
 545         DATA_BLOB null_data_blob = data_blob(NULL,0);
 546         const char **mechTypes = NULL;
 547         DATA_BLOB unwrapped_out = data_blob(NULL, 0);
 548         const struct gensec_security_ops_wrapper *all_sec;
 549         const char *principal = NULL;
 550
 551         mechTypes = gensec_security_oids(gensec_security,
 552                                          out_mem_ctx, GENSEC_OID_SPNEGO);
 553
 554         all_sec = gensec_security_by_oid_list(gensec_security,
 555                                               out_mem_ctx,
 556                                               mechTypes,
 557                                               GENSEC_OID_SPNEGO);
 558         for (i=0; all_sec && all_sec[i].op; i++) {
 559                 struct spnego_data spnego_out;
 560                 nt_status = gensec_subcontext_start(spnego_state,
 561                                                     gensec_security,
 562                                                     &spnego_state->sub_sec_security);
 563                 if (!NT_STATUS_IS_OK(nt_status)) {
 564                         return nt_status;
 565                 }
 566                 /* select the sub context */
 567                 nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
 568                                                      all_sec[i].op);
 569                 if (!NT_STATUS_IS_OK(nt_status)) {
 570                         talloc_free(spnego_state->sub_sec_security);
 571                         spnego_state->sub_sec_security = NULL;
 572                         continue;
 573                 }
 574
 575                 /* In the client, try and produce the first (optimistic) packet */
 576                 if (spnego_state->state_position == SPNEGO_CLIENT_START) {
 577                         nt_status = gensec_update(spnego_state->sub_sec_security,
 578                                                   out_mem_ctx,
 579                                                   null_data_blob,
 580                                                   &unwrapped_out);
 581
 582                         if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)
 583                             && !NT_STATUS_IS_OK(nt_status)) {
 584                                 DEBUG(1, ("SPNEGO(%s) creating NEG_TOKEN_INIT failed: %s\n",
 585                                           spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
 586                                 talloc_free(spnego_state->sub_sec_security);
 587                                 spnego_state->sub_sec_security = NULL;
 588                                 /* Pretend we never started it (lets the first run find some incompatible demand) */
 589
 590                                 continue;
 591                         }
 592                 }
 593
 594                 spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
 595
 596                 /* List the remaining mechs as options */
 597                 spnego_out.negTokenInit.mechTypes = gensec_security_oids_from_ops_wrapped(out_mem_ctx,
 598                                                                                           &all_sec[i]);
 599                 spnego_out.negTokenInit.reqFlags = 0;
 600
 601                 if (spnego_state->state_position == SPNEGO_SERVER_START) {
 602                         /* server credentails */
 603                         struct cli_credentials *creds = gensec_get_credentials(gensec_security);
 604                         if (creds) {
 605                                 principal = cli_credentials_get_principal(creds, out_mem_ctx);
 606                         }
 607                 }
 608                 if (principal) {
 609                         spnego_out.negTokenInit.mechListMIC
 610                                 = data_blob_string_const(principal);
 611                 } else {
 612                         spnego_out.negTokenInit.mechListMIC = null_data_blob;
 613                 }
 614
 615                 spnego_out.negTokenInit.mechToken = unwrapped_out;
 616
 617                 if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 618                         DEBUG(1, ("Failed to write NEG_TOKEN_INIT\n"));
 619                                 return NT_STATUS_INVALID_PARAMETER;
 620                 }
 621
 622                 /* set next state */
 623                 spnego_state->neg_oid = all_sec[i].oid;
 624
 625                 if (NT_STATUS_IS_OK(nt_status)) {
 626                         spnego_state->no_response_expected = True;
 627                 }
 628
 629                 return NT_STATUS_MORE_PROCESSING_REQUIRED;
 630         }
 631         talloc_free(spnego_state->sub_sec_security);
 632         spnego_state->sub_sec_security = NULL;
 633
 634         DEBUG(1, ("Failed to setup SPNEGO negTokenInit request: %s\n", nt_errstr(nt_status)));
 635         return NT_STATUS_INVALID_PARAMETER;
 636 }
 637
 638
 639 /** create a server negTokenTarg
 640  *
 641  * This is the case, where the client is the first one who sends data
 642 */
 643
 644 static NTSTATUS gensec_spnego_server_negTokenTarg(struct gensec_security *gensec_security,
 645                                                   struct spnego_state *spnego_state,
 646                                                   TALLOC_CTX *out_mem_ctx,
 647                                                   NTSTATUS nt_status,
 648                                                   const DATA_BLOB unwrapped_out, DATA_BLOB *out)
 649 {
 650         struct spnego_data spnego_out;
 651         DATA_BLOB null_data_blob = data_blob(NULL, 0);
 652
 653         /* compose reply */
 654         spnego_out.type = SPNEGO_NEG_TOKEN_TARG;
 655         spnego_out.negTokenTarg.responseToken = unwrapped_out;
 656         spnego_out.negTokenTarg.mechListMIC = null_data_blob;
 657         spnego_out.negTokenTarg.supportedMech = NULL;
 658
 659         if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 660                 spnego_out.negTokenTarg.supportedMech = spnego_state->neg_oid;
 661                 spnego_out.negTokenTarg.negResult = SPNEGO_ACCEPT_INCOMPLETE;
 662                 spnego_state->state_position = SPNEGO_SERVER_TARG;
 663         } else if (NT_STATUS_IS_OK(nt_status)) {
 664                 if (unwrapped_out.data) {
 665                         spnego_out.negTokenTarg.supportedMech = spnego_state->neg_oid;
 666                 }
 667                 spnego_out.negTokenTarg.negResult = SPNEGO_ACCEPT_COMPLETED;
 668                 spnego_state->state_position = SPNEGO_DONE;
 669         } else {
 670                 spnego_out.negTokenTarg.negResult = SPNEGO_REJECT;
 671                 DEBUG(2, ("SPNEGO login failed: %s\n", nt_errstr(nt_status)));
 672                 spnego_state->state_position = SPNEGO_DONE;
 673         }
 674
 675         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 676                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_TARG\n"));
 677                 return NT_STATUS_INVALID_PARAMETER;
 678         }
 679
 680         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 681
 682         return nt_status;
 683 }
 684
 685
 686 static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx,
 687                                      const DATA_BLOB in, DATA_BLOB *out)
 688 {
 689         struct spnego_state *spnego_state = gensec_security->private_data;
 690         DATA_BLOB null_data_blob = data_blob(NULL, 0);
 691         DATA_BLOB unwrapped_out = data_blob(NULL, 0);
 692         struct spnego_data spnego_out;
 693         struct spnego_data spnego;
 694
 695         ssize_t len;
 696
 697         *out = data_blob(NULL, 0);
 698
 699         if (!out_mem_ctx) {
 700                 out_mem_ctx = spnego_state;
 701         }
 702
 703         /* and switch into the state machine */
 704
 705         switch (spnego_state->state_position) {
 706         case SPNEGO_FALLBACK:
 707                 return gensec_update(spnego_state->sub_sec_security,
 708                                      out_mem_ctx, in, out);
 709         case SPNEGO_SERVER_START:
 710         {
 711                 NTSTATUS nt_status;
 712                 if (in.length) {
 713
 714                         len = spnego_read_data(in, &spnego);
 715                         if (len == -1) {
 716                                 return gensec_spnego_server_try_fallback(gensec_security, spnego_state,
 717                                                                          out_mem_ctx, in, out);
 718                         }
 719                         /* client sent NegTargetInit, we send NegTokenTarg */
 720
 721                         /* OK, so it's real SPNEGO, check the packet's the one we expect */
 722                         if (spnego.type != spnego_state->expected_packet) {
 723                                 DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 724                                           spnego_state->expected_packet));
 725                                 dump_data(1, in.data, in.length);
 726                                 spnego_free_data(&spnego);
 727                                 return NT_STATUS_INVALID_PARAMETER;
 728                         }
 729
 730                         nt_status = gensec_spnego_parse_negTokenInit(gensec_security,
 731                                                                      spnego_state,
 732                                                                      out_mem_ctx,
 733                                                                      spnego.negTokenInit.mechTypes,
 734                                                                      spnego.negTokenInit.mechToken,
 735                                                                      &unwrapped_out);
 736
 737                         nt_status = gensec_spnego_server_negTokenTarg(gensec_security,
 738                                                                       spnego_state,
 739                                                                       out_mem_ctx,
 740                                                                       nt_status,
 741                                                                       unwrapped_out,
 742                                                                       out);
 743
 744                         spnego_free_data(&spnego);
 745
 746                         return nt_status;
 747                 } else {
 748                         nt_status = gensec_spnego_create_negTokenInit(gensec_security, spnego_state,
 749                                                                       out_mem_ctx, in, out);
 750                         spnego_state->state_position = SPNEGO_SERVER_START;
 751                         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
 752                         return nt_status;
 753                 }
 754         }
 755
 756         case SPNEGO_CLIENT_START:
 757         {
 758                 /* The server offers a list of mechanisms */
 759
 760                 const char *my_mechs[] = {NULL, NULL};
 761                 NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
 762
 763                 if (!in.length) {
 764                         /* client to produce negTokenInit */
 765                         nt_status = gensec_spnego_create_negTokenInit(gensec_security, spnego_state,
 766                                                                  out_mem_ctx, in, out);
 767                         spnego_state->state_position = SPNEGO_CLIENT_TARG;
 768                         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 769                         return nt_status;
 770                 }
 771
 772                 len = spnego_read_data(in, &spnego);
 773
 774                 if (len == -1) {
 775                         DEBUG(1, ("Invalid SPNEGO request:\n"));
 776                         dump_data(1, in.data, in.length);
 777                         return NT_STATUS_INVALID_PARAMETER;
 778                 }
 779
 780                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
 781                 if (spnego.type != spnego_state->expected_packet) {
 782                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 783                                   spnego_state->expected_packet));
 784                         dump_data(1, in.data, in.length);
 785                         spnego_free_data(&spnego);
 786                         return NT_STATUS_INVALID_PARAMETER;
 787                 }
 788
 789                 if (spnego.negTokenInit.targetPrincipal) {
 790                         DEBUG(5, ("Server claims it's principal name is %s\n", spnego.negTokenInit.targetPrincipal));
 791                         gensec_set_target_principal(gensec_security, spnego.negTokenInit.targetPrincipal);
 792                 }
 793
 794                 nt_status = gensec_spnego_parse_negTokenInit(gensec_security,
 795                                                              spnego_state,
 796                                                              out_mem_ctx,
 797                                                              spnego.negTokenInit.mechTypes,
 798                                                              spnego.negTokenInit.mechToken,
 799                                                              &unwrapped_out);
 800
 801                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(nt_status)) {
 802                         spnego_free_data(&spnego);
 803                         return nt_status;
 804                 }
 805
 806                 my_mechs[0] = spnego_state->neg_oid;
 807                 /* compose reply */
 808                 spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
 809                 spnego_out.negTokenInit.mechTypes = my_mechs;
 810                 spnego_out.negTokenInit.reqFlags = 0;
 811                 spnego_out.negTokenInit.mechListMIC = null_data_blob;
 812                 spnego_out.negTokenInit.mechToken = unwrapped_out;
 813
 814                 if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 815                         DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n"));
 816                                 return NT_STATUS_INVALID_PARAMETER;
 817                 }
 818
 819                 /* set next state */
 820                 spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
 821                 spnego_state->state_position = SPNEGO_CLIENT_TARG;
 822
 823                 if (NT_STATUS_IS_OK(nt_status)) {
 824                         spnego_state->no_response_expected = True;
 825                 }
 826
 827                 spnego_free_data(&spnego);
 828                 return NT_STATUS_MORE_PROCESSING_REQUIRED;
 829         }
 830         case SPNEGO_SERVER_TARG:
 831         {
 832                 NTSTATUS nt_status;
 833                 if (!in.length) {
 834                         return NT_STATUS_INVALID_PARAMETER;
 835                 }
 836
 837                 len = spnego_read_data(in, &spnego);
 838
 839                 if (len == -1) {
 840                         DEBUG(1, ("Invalid SPNEGO request:\n"));
 841                         dump_data(1, in.data, in.length);
 842                         return NT_STATUS_INVALID_PARAMETER;
 843                 }
 844
 845                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
 846                 if (spnego.type != spnego_state->expected_packet) {
 847                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 848                                   spnego_state->expected_packet));
 849                         dump_data(1, in.data, in.length);
 850                         spnego_free_data(&spnego);
 851                         return NT_STATUS_INVALID_PARAMETER;
 852                 }
 853
 854                 if (!spnego_state->sub_sec_security) {
 855                         DEBUG(1, ("SPNEGO: Did not setup a mech in NEG_TOKEN_INIT\n"));
 856                         spnego_free_data(&spnego);
 857                         return NT_STATUS_INVALID_PARAMETER;
 858                 }
 859
 860                 nt_status = gensec_update(spnego_state->sub_sec_security,
 861                                           out_mem_ctx,
 862                                           spnego.negTokenTarg.responseToken,
 863                                           &unwrapped_out);
 864
 865                 nt_status = gensec_spnego_server_negTokenTarg(gensec_security,
 866                                                               spnego_state,
 867                                                               out_mem_ctx,
 868                                                               nt_status,
 869                                                               unwrapped_out,
 870                                                               out);
 871
 872                 spnego_free_data(&spnego);
 873
 874                 return nt_status;
 875         }
 876         case SPNEGO_CLIENT_TARG:
 877         {
 878                 NTSTATUS nt_status;
 879                 if (!in.length) {
 880                         return NT_STATUS_INVALID_PARAMETER;
 881                 }
 882
 883                 len = spnego_read_data(in, &spnego);
 884
 885                 if (len == -1) {
 886                         DEBUG(1, ("Invalid SPNEGO request:\n"));
 887                         dump_data(1, in.data, in.length);
 888                         return NT_STATUS_INVALID_PARAMETER;
 889                 }
 890
 891                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
 892                 if (spnego.type != spnego_state->expected_packet) {
 893                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
 894                                   spnego_state->expected_packet));
 895                         dump_data(1, in.data, in.length);
 896                         spnego_free_data(&spnego);
 897                         return NT_STATUS_INVALID_PARAMETER;
 898                 }
 899
 900                 if (spnego.negTokenTarg.negResult == SPNEGO_REJECT) {
 901                         spnego_free_data(&spnego);
 902                         return NT_STATUS_ACCESS_DENIED;
 903                 }
 904
 905                 /* Server didn't like our choice of mech, and chose something else */
 906                 if ((spnego.negTokenTarg.negResult == SPNEGO_ACCEPT_INCOMPLETE) &&
 907                     strcmp(spnego.negTokenTarg.supportedMech, spnego_state->neg_oid) != 0) {
 908                         DEBUG(3,("GENSEC SPNEGO: client preferred mech (%s) not accepted, server wants: %s\n",
 909                                  gensec_get_name_by_oid(spnego.negTokenTarg.supportedMech),
 910                                  gensec_get_name_by_oid(spnego_state->neg_oid)));
 911
 912                         talloc_free(spnego_state->sub_sec_security);
 913                         nt_status = gensec_subcontext_start(spnego_state,
 914                                                             gensec_security,
 915                                                             &spnego_state->sub_sec_security);
 916                         if (!NT_STATUS_IS_OK(nt_status)) {
 917                                 spnego_free_data(&spnego);
 918                                 return nt_status;
 919                         }
 920                         /* select the sub context */
 921                         nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
 922                                                              spnego.negTokenTarg.supportedMech);
 923                         if (!NT_STATUS_IS_OK(nt_status)) {
 924                                 spnego_free_data(&spnego);
 925                                 return nt_status;
 926                         }
 927
 928                         nt_status = gensec_update(spnego_state->sub_sec_security,
 929                                                   out_mem_ctx,
 930                                                   spnego.negTokenTarg.responseToken,
 931                                                   &unwrapped_out);
 932                         spnego_state->neg_oid = talloc_strdup(spnego_state, spnego.negTokenTarg.supportedMech);
 933                 } else if (spnego_state->no_response_expected) {
 934                         if (spnego.negTokenTarg.negResult != SPNEGO_ACCEPT_COMPLETED) {
 935                                 DEBUG(3,("GENSEC SPNEGO: client GENSEC accepted, but server rejected (bad password?)\n"));
 936                                 nt_status = NT_STATUS_INVALID_PARAMETER;
 937                         } else if (spnego.negTokenTarg.responseToken.length) {
 938                                 DEBUG(2,("GENSEC SPNEGO: client GENSEC accepted, but server continued negotiation!\n"));
 939                                 nt_status = NT_STATUS_INVALID_PARAMETER;
 940                         } else {
 941                                 nt_status = NT_STATUS_OK;
 942                         }
 943                 } else {
 944                         nt_status = gensec_update(spnego_state->sub_sec_security,
 945                                                   out_mem_ctx,
 946                                                   spnego.negTokenTarg.responseToken,
 947                                                   &unwrapped_out);
 948
 949                         if (NT_STATUS_IS_OK(nt_status)) {
 950                                 spnego_state->no_response_expected = True;
 951                         }
 952                 }
 953
 954                 spnego_free_data(&spnego);
 955
 956                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)
 957                         && !NT_STATUS_IS_OK(nt_status)) {
 958                         DEBUG(1, ("SPNEGO(%s) login failed: %s\n",
 959                                   spnego_state->sub_sec_security->ops->name,
 960                                   nt_errstr(nt_status)));
 961                         return nt_status;
 962                 }
 963
 964                 if (unwrapped_out.length) {
 965                         /* compose reply */
 966                         spnego_out.type = SPNEGO_NEG_TOKEN_TARG;
 967                         spnego_out.negTokenTarg.negResult = SPNEGO_NONE_RESULT;
 968                         spnego_out.negTokenTarg.supportedMech = NULL;
 969                         spnego_out.negTokenTarg.responseToken = unwrapped_out;
 970                         spnego_out.negTokenTarg.mechListMIC = null_data_blob;
 971
 972                         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
 973                                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_TARG\n"));
 974                                 return NT_STATUS_INVALID_PARAMETER;
 975                         }
 976
 977                         spnego_state->state_position = SPNEGO_CLIENT_TARG;
 978                         nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
 979                 } else {
 980
 981                         /* all done - server has accepted, and we agree */
 982                         *out = null_data_blob;
 983
 984                         if (spnego.negTokenTarg.negResult != SPNEGO_ACCEPT_COMPLETED) {
 985                                 /* unless of course it did not accept */
 986                                 DEBUG(1,("gensec_update ok but not accepted\n"));
 987                                 nt_status = NT_STATUS_INVALID_PARAMETER;
 988                         }
 989
 990                         spnego_state->state_position = SPNEGO_DONE;
 991                 }
 992
 993                 return nt_status;
 994         }
 995         case SPNEGO_DONE:
 996                 /* We should not be called after we are 'done' */
 997                 return NT_STATUS_INVALID_PARAMETER;
 998         }
 999         return NT_STATUS_INVALID_PARAMETER;
1000 }
1001
1002 static BOOL gensec_spnego_have_feature(struct gensec_security *gensec_security,
1003                                        uint32_t feature)
1004 {
1005         struct spnego_state *spnego_state = gensec_security->private_data;
1006         if (!spnego_state->sub_sec_security) {
1007                 return False;
1008         }
1009
1010         return gensec_have_feature(spnego_state->sub_sec_security,
1011                                    feature);
1012 }
1013
1014 static const char *gensec_spnego_oids[] = {
1015         GENSEC_OID_SPNEGO,
1016         NULL
1017 };
1018
1019 static const struct gensec_security_ops gensec_spnego_security_ops = {
1020         .name             = "spnego",
1021         .sasl_name        = "GSS-SPNEGO",
1022         .auth_type        = DCERPC_AUTH_TYPE_SPNEGO,
1023         .oid              = gensec_spnego_oids,
1024         .client_start     = gensec_spnego_client_start,
1025         .server_start     = gensec_spnego_server_start,
1026         .update           = gensec_spnego_update,
1027         .seal_packet      = gensec_spnego_seal_packet,
1028         .sign_packet      = gensec_spnego_sign_packet,
1029         .sig_size         = gensec_spnego_sig_size,
1030         .max_wrapped_size = gensec_spnego_max_wrapped_size,
1031         .max_input_size   = gensec_spnego_max_input_size,
1032         .check_packet     = gensec_spnego_check_packet,
1033         .unseal_packet    = gensec_spnego_unseal_packet,
1034         .packet_full_request = gensec_spnego_packet_full_request,
1035         .wrap             = gensec_spnego_wrap,
1036         .unwrap           = gensec_spnego_unwrap,
1037         .wrap_packets     = gensec_spnego_wrap_packets,
1038         .unwrap_packets   = gensec_spnego_unwrap_packets,
1039         .session_key      = gensec_spnego_session_key,
1040         .session_info     = gensec_spnego_session_info,
1041         .have_feature     = gensec_spnego_have_feature,
1042         .enabled          = True,
1043 };
1044
1045 NTSTATUS gensec_spnego_init(void)
1046 {
1047         NTSTATUS ret;
1048         ret = gensec_register(&gensec_spnego_security_ops);
1049         if (!NT_STATUS_IS_OK(ret)) {
1050                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
1051                         gensec_spnego_security_ops.name));
1052                 return ret;
1053         }
1054
1055         return ret;
1056 }