auth/spnego: split out a gensec_spnego_server_negTokenInit() function.
[samba.git] / auth / gensec / spnego.c
index bb2aa70a4d8c290590204bb42dbc487aab07cfca..53fd81c591dcda8d52bc83c59a149f45c983dc4b 100644 (file)
@@ -140,22 +140,23 @@ static NTSTATUS gensec_spnego_server_start(struct gensec_security *gensec_securi
 
 static NTSTATUS gensec_spnego_server_try_fallback(struct gensec_security *gensec_security, 
                                                  struct spnego_state *spnego_state,
-                                                 struct tevent_context *ev,
-                                                 TALLOC_CTX *out_mem_ctx, 
-                                                 const DATA_BLOB in, DATA_BLOB *out) 
+                                                 TALLOC_CTX *mem_ctx,
+                                                 const DATA_BLOB in)
 {
        int i,j;
        const struct gensec_security_ops **all_ops;
 
-       all_ops = gensec_security_mechs(gensec_security, out_mem_ctx);
+       all_ops = gensec_security_mechs(gensec_security, mem_ctx);
 
        for (i=0; all_ops && all_ops[i]; i++) {
                bool is_spnego;
                NTSTATUS nt_status;
 
-               if (gensec_security != NULL && 
-                               !gensec_security_ops_enabled(all_ops[i], gensec_security))
-                   continue;
+               if (gensec_security != NULL &&
+                   !gensec_security_ops_enabled(all_ops[i], gensec_security))
+               {
+                       continue;
+               }
 
                if (!all_ops[i]->oid) {
                        continue;
@@ -195,9 +196,8 @@ static NTSTATUS gensec_spnego_server_try_fallback(struct gensec_security *gensec
                if (!NT_STATUS_IS_OK(nt_status)) {
                        return nt_status;
                }
-               nt_status = gensec_update_ev(spnego_state->sub_sec_security,
-                                            out_mem_ctx, ev, in, out);
-               return nt_status;
+
+               return NT_STATUS_OK;
        }
        DEBUG(1, ("Failed to parse SPNEGO request\n"));
        return NT_STATUS_INVALID_PARAMETER;
@@ -212,15 +212,24 @@ static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_
                                                 struct spnego_state *spnego_state, 
                                                 TALLOC_CTX *out_mem_ctx, 
                                                 struct tevent_context *ev,
-                                                const char * const *mechType,
-                                                const DATA_BLOB unwrapped_in, DATA_BLOB *unwrapped_out) 
+                                                struct spnego_data *spnego_in,
+                                                DATA_BLOB *unwrapped_out)
 {
        int i;
        NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
+       const char * const *mechType = NULL;
+       DATA_BLOB unwrapped_in = data_blob_null;
        bool ok;
+       const struct gensec_security_ops_wrapper *all_sec = NULL;
+
+       if (spnego_in->type != SPNEGO_NEG_TOKEN_INIT) {
+               return NT_STATUS_INTERNAL_ERROR;
+       }
 
-       const struct gensec_security_ops_wrapper *all_sec
-               = gensec_security_by_oid_list(gensec_security, 
+       mechType = spnego_in->negTokenInit.mechTypes;
+       unwrapped_in = spnego_in->negTokenInit.mechToken;
+
+       all_sec = gensec_security_by_oid_list(gensec_security,
                                              out_mem_ctx, 
                                              mechType,
                                              GENSEC_OID_SPNEGO);
@@ -310,6 +319,7 @@ static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_
        /* Having tried any optimistic token from the client (if we
         * were the server), if we didn't get anywhere, walk our list
         * in our preference order */
+       unwrapped_in = data_blob_null;
 
        if (!spnego_state->sub_sec_security) {
                for (i=0; all_sec && all_sec[i].op; i++) {
@@ -336,7 +346,7 @@ static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_
                        nt_status = gensec_update_ev(spnego_state->sub_sec_security,
                                                  out_mem_ctx, 
                                                  ev,
-                                                 data_blob_null,
+                                                 unwrapped_in,
                                                  unwrapped_out);
                        if (NT_STATUS_IS_OK(nt_status)) {
                                spnego_state->sub_sec_ready = true;
@@ -404,9 +414,7 @@ static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_
                        nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
                }
 
-               if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER) 
-                   && !NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) 
-                   && !NT_STATUS_IS_OK(nt_status)) {
+               if (GENSEC_UPDATE_IS_NTERROR(nt_status)) {
                        DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed: %s\n", 
                                  spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
 
@@ -418,7 +426,7 @@ static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_
                        return nt_status;
                }
 
-               return nt_status; /* OK, INVALID_PARAMETER ore MORE PROCESSING */
+               return nt_status; /* OK or MORE PROCESSING */
        }
 
        DEBUG(1, ("SPNEGO: Could not find a suitable mechtype in NEG_TOKEN_INIT\n"));
@@ -438,7 +446,7 @@ static NTSTATUS gensec_spnego_create_negTokenInit(struct gensec_security *gensec
                                                  struct spnego_state *spnego_state,
                                                  TALLOC_CTX *out_mem_ctx, 
                                                  struct tevent_context *ev,
-                                                 const DATA_BLOB in, DATA_BLOB *out) 
+                                                 DATA_BLOB *out)
 {
        int i;
        NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
@@ -483,8 +491,7 @@ static NTSTATUS gensec_spnego_create_negTokenInit(struct gensec_security *gensec
                                spnego_state->sub_sec_ready = true;
                        }
 
-                       if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) 
-                           && !NT_STATUS_IS_OK(nt_status)) {
+                       if (GENSEC_UPDATE_IS_NTERROR(nt_status)) {
                                const char *next = NULL;
                                const char *principal = NULL;
                                int dbg_level = DBGLVL_WARNING;
@@ -555,6 +562,14 @@ static NTSTATUS gensec_spnego_create_negTokenInit(struct gensec_security *gensec
                /* set next state */
                spnego_state->neg_oid = all_sec[i].oid;
 
+               if (spnego_state->state_position == SPNEGO_SERVER_START) {
+                       spnego_state->state_position = SPNEGO_SERVER_START;
+                       spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
+               } else {
+                       spnego_state->state_position = SPNEGO_CLIENT_TARG;
+                       spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
+               }
+
                return NT_STATUS_MORE_PROCESSING_REQUIRED;
        }
        gensec_spnego_update_sub_abort(spnego_state);
@@ -563,18 +578,82 @@ static NTSTATUS gensec_spnego_create_negTokenInit(struct gensec_security *gensec
        return nt_status;
 }
 
+static NTSTATUS gensec_spnego_client_negTokenInit(struct gensec_security *gensec_security,
+                                                 struct spnego_state *spnego_state,
+                                                 struct tevent_context *ev,
+                                                 struct spnego_data *spnego_in,
+                                                 TALLOC_CTX *out_mem_ctx,
+                                                 DATA_BLOB *out)
+{
+       DATA_BLOB sub_out = data_blob_null;
+       const char *tp = NULL;
+       struct spnego_data spnego_out;
+       const char *my_mechs[] = {NULL, NULL};
+       NTSTATUS status;
+       bool ok;
+
+       *out = data_blob_null;
+
+       /* The server offers a list of mechanisms */
+
+       tp = spnego_in->negTokenInit.targetPrincipal;
+       if (tp != NULL && strcmp(tp, ADS_IGNORE_PRINCIPAL) != 0) {
+               DBG_INFO("Server claims it's principal name is %s\n", tp);
+               if (lpcfg_client_use_spnego_principal(gensec_security->settings->lp_ctx)) {
+                       gensec_set_target_principal(gensec_security, tp);
+               }
+       }
+
+       status = gensec_spnego_parse_negTokenInit(gensec_security,
+                                                 spnego_state,
+                                                 out_mem_ctx,
+                                                 ev,
+                                                 spnego_in,
+                                                 &sub_out);
+       if (GENSEC_UPDATE_IS_NTERROR(status)) {
+               return status;
+       }
+
+       my_mechs[0] = spnego_state->neg_oid;
+       /* compose reply */
+       spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
+       spnego_out.negTokenInit.mechTypes = my_mechs;
+       spnego_out.negTokenInit.reqFlags = data_blob_null;
+       spnego_out.negTokenInit.reqFlagsPadding = 0;
+       spnego_out.negTokenInit.mechListMIC = data_blob_null;
+       spnego_out.negTokenInit.mechToken = sub_out;
+
+       if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
+               DBG_ERR("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n");
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+
+       ok = spnego_write_mech_types(spnego_state,
+                                    my_mechs,
+                                    &spnego_state->mech_types);
+       if (!ok) {
+               DBG_ERR("failed to write mechTypes\n");
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       /* set next state */
+       spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
+       spnego_state->state_position = SPNEGO_CLIENT_TARG;
+
+       return NT_STATUS_MORE_PROCESSING_REQUIRED;
+}
 
 /** create a server negTokenTarg 
  *
  * This is the case, where the client is the first one who sends data
 */
 
-static NTSTATUS gensec_spnego_server_negTokenTarg(struct spnego_state *spnego_state,
-                                                 TALLOC_CTX *out_mem_ctx, 
-                                                 NTSTATUS nt_status,
-                                                 const DATA_BLOB unwrapped_out,
-                                                 DATA_BLOB mech_list_mic,
-                                                 DATA_BLOB *out)
+static NTSTATUS gensec_spnego_server_response(struct spnego_state *spnego_state,
+                                             TALLOC_CTX *out_mem_ctx,
+                                             NTSTATUS nt_status,
+                                             const DATA_BLOB unwrapped_out,
+                                             DATA_BLOB mech_list_mic,
+                                             DATA_BLOB *out)
 {
        struct spnego_data spnego_out;
 
@@ -617,17 +696,54 @@ static NTSTATUS gensec_spnego_server_negTokenTarg(struct spnego_state *spnego_st
        return nt_status;
 }
 
+static NTSTATUS gensec_spnego_server_negTokenInit(struct gensec_security *gensec_security,
+                                                 struct spnego_state *spnego_state,
+                                                 struct tevent_context *ev,
+                                                 struct spnego_data *spnego_in,
+                                                 TALLOC_CTX *out_mem_ctx,
+                                                 DATA_BLOB *out)
+{
+       DATA_BLOB sub_out = data_blob_null;
+       DATA_BLOB mech_list_mic = data_blob_null;
+       NTSTATUS status;
+
+       status = gensec_spnego_parse_negTokenInit(gensec_security,
+                                                 spnego_state,
+                                                 out_mem_ctx,
+                                                 ev,
+                                                 spnego_in,
+                                                 &sub_out);
+
+       if (spnego_state->simulate_w2k) {
+               /*
+                * Windows 2000 returns the unwrapped token
+                * also in the mech_list_mic field.
+                *
+                * In order to verify our client code,
+                * we need a way to have a server with this
+                * broken behaviour
+                */
+               mech_list_mic = sub_out;
+       }
+
+       return gensec_spnego_server_response(spnego_state,
+                                            out_mem_ctx,
+                                            status,
+                                            sub_out,
+                                            mech_list_mic,
+                                            out);
+}
+
 static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_security,
                                            TALLOC_CTX *out_mem_ctx,
                                            struct tevent_context *ev,
-                                           const DATA_BLOB in, DATA_BLOB *out)
+                                           struct spnego_data *spnego_in,
+                                           DATA_BLOB *out)
 {
        struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
        DATA_BLOB mech_list_mic = data_blob_null;
        DATA_BLOB unwrapped_out = data_blob_null;
        struct spnego_data spnego_out;
-       struct spnego_data spnego;
-       ssize_t len;
 
        *out = data_blob_null;
 
@@ -635,162 +751,60 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
 
        switch (spnego_state->state_position) {
        case SPNEGO_CLIENT_START:
-       {
-               /* The server offers a list of mechanisms */
-
-               const char *my_mechs[] = {NULL, NULL};
-               NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
-               bool ok;
-
-               if (!in.length) {
-                       /* client to produce negTokenInit */
-                       nt_status = gensec_spnego_create_negTokenInit(gensec_security, spnego_state, 
-                                                                     out_mem_ctx, ev, in, out);
-                       spnego_state->state_position = SPNEGO_CLIENT_TARG;
-                       spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
-                       return nt_status;
-               }
-
-               len = spnego_read_data(gensec_security, in, &spnego);
-
-               if (len == -1) {
-                       DEBUG(1, ("Invalid SPNEGO request:\n"));
-                       dump_data(1, in.data, in.length);
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
-
-               /* OK, so it's real SPNEGO, check the packet's the one we expect */
-               if (spnego.type != spnego_state->expected_packet) {
-                       DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type, 
-                                 spnego_state->expected_packet));
-                       dump_data(1, in.data, in.length);
-                       spnego_free_data(&spnego);
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
-
-               if (spnego.negTokenInit.targetPrincipal
-                   && strcmp(spnego.negTokenInit.targetPrincipal, ADS_IGNORE_PRINCIPAL) != 0) {
-                       DEBUG(5, ("Server claims it's principal name is %s\n", spnego.negTokenInit.targetPrincipal));
-                       if (lpcfg_client_use_spnego_principal(gensec_security->settings->lp_ctx)) {
-                               gensec_set_target_principal(gensec_security, spnego.negTokenInit.targetPrincipal);
-                       }
-               }
-
-               nt_status = gensec_spnego_parse_negTokenInit(gensec_security,
-                                                            spnego_state,
-                                                            out_mem_ctx, 
-                                                            ev,
-                                                            spnego.negTokenInit.mechTypes,
-                                                            spnego.negTokenInit.mechToken, 
-                                                            &unwrapped_out);
-
-               if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(nt_status)) {
-                       spnego_free_data(&spnego);
-                       return nt_status;
-               }
-
-               my_mechs[0] = spnego_state->neg_oid;
-               /* compose reply */
-               spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
-               spnego_out.negTokenInit.mechTypes = my_mechs;
-               spnego_out.negTokenInit.reqFlags = data_blob_null;
-               spnego_out.negTokenInit.reqFlagsPadding = 0;
-               spnego_out.negTokenInit.mechListMIC = data_blob_null;
-               spnego_out.negTokenInit.mechToken = unwrapped_out;
-
-               if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
-                       DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n"));
-                               return NT_STATUS_INVALID_PARAMETER;
-               }
-
-               ok = spnego_write_mech_types(spnego_state,
-                                            my_mechs,
-                                            &spnego_state->mech_types);
-               if (!ok) {
-                       DEBUG(1, ("SPNEGO: Failed to write mechTypes\n"));
-                       return NT_STATUS_NO_MEMORY;
-               }
-
-               /* set next state */
-               spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
-               spnego_state->state_position = SPNEGO_CLIENT_TARG;
-
-               spnego_free_data(&spnego);
-               return NT_STATUS_MORE_PROCESSING_REQUIRED;
-       }
+               return gensec_spnego_client_negTokenInit(gensec_security,
+                                                        spnego_state,
+                                                        ev, spnego_in,
+                                                        out_mem_ctx, out);
 
        case SPNEGO_CLIENT_TARG:
        {
                NTSTATUS nt_status = NT_STATUS_INTERNAL_ERROR;
-
-               if (!in.length) {
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
-
-               len = spnego_read_data(gensec_security, in, &spnego);
-
-               if (len == -1) {
-                       DEBUG(1, ("Invalid SPNEGO request:\n"));
-                       dump_data(1, in.data, in.length);
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
-
-               /* OK, so it's real SPNEGO, check the packet's the one we expect */
-               if (spnego.type != spnego_state->expected_packet) {
-                       DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type, 
-                                 spnego_state->expected_packet));
-                       dump_data(1, in.data, in.length);
-                       spnego_free_data(&spnego);
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
+               const struct spnego_negTokenTarg *ta =
+                       &spnego_in->negTokenTarg;
 
                spnego_state->num_targs++;
 
-               if (spnego.negTokenTarg.negResult == SPNEGO_REJECT) {
-                       spnego_free_data(&spnego);
+               if (ta->negResult == SPNEGO_REJECT) {
                        return NT_STATUS_LOGON_FAILURE;
                }
 
-               if (spnego.negTokenTarg.negResult == SPNEGO_REQUEST_MIC) {
+               if (spnego_in->negTokenTarg.negResult == SPNEGO_REQUEST_MIC) {
                        spnego_state->mic_requested = true;
                }
 
                /* Server didn't like our choice of mech, and chose something else */
-               if (((spnego.negTokenTarg.negResult == SPNEGO_ACCEPT_INCOMPLETE) ||
-                    (spnego.negTokenTarg.negResult == SPNEGO_REQUEST_MIC)) &&
-                   spnego.negTokenTarg.supportedMech &&
-                   strcmp(spnego.negTokenTarg.supportedMech, spnego_state->neg_oid) != 0) {
+               if (((ta->negResult == SPNEGO_ACCEPT_INCOMPLETE) ||
+                    (ta->negResult == SPNEGO_REQUEST_MIC)) &&
+                   ta->supportedMech != NULL&&
+                   strcmp(ta->supportedMech, spnego_state->neg_oid) != 0) {
                        DEBUG(3,("GENSEC SPNEGO: client preferred mech (%s) not accepted, server wants: %s\n",
                                 gensec_get_name_by_oid(gensec_security, spnego_state->neg_oid),
-                                gensec_get_name_by_oid(gensec_security, spnego.negTokenTarg.supportedMech)));
+                                gensec_get_name_by_oid(gensec_security, ta->supportedMech)));
                        spnego_state->downgraded = true;
                        gensec_spnego_update_sub_abort(spnego_state);
                        nt_status = gensec_subcontext_start(spnego_state,
                                                            gensec_security,
                                                            &spnego_state->sub_sec_security);
                        if (!NT_STATUS_IS_OK(nt_status)) {
-                               spnego_free_data(&spnego);
                                return nt_status;
                        }
                        /* select the sub context */
                        nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
-                                                            spnego.negTokenTarg.supportedMech);
+                                                            ta->supportedMech);
                        if (!NT_STATUS_IS_OK(nt_status)) {
-                               spnego_free_data(&spnego);
                                return nt_status;
                        }
 
                        spnego_state->neg_oid = talloc_strdup(spnego_state,
-                                               spnego.negTokenTarg.supportedMech);
+                                               ta->supportedMech);
                        if (spnego_state->neg_oid == NULL) {
-                               spnego_free_data(&spnego);
                                return NT_STATUS_NO_MEMORY;
                        };
                }
 
-               if (spnego.negTokenTarg.mechListMIC.length > 0) {
-                       DATA_BLOB *m = &spnego.negTokenTarg.mechListMIC;
-                       const DATA_BLOB *r = &spnego.negTokenTarg.responseToken;
+               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
+                       DATA_BLOB *m = &spnego_in->negTokenTarg.mechListMIC;
+                       const DATA_BLOB *r = &spnego_in->negTokenTarg.responseToken;
 
                        /*
                         * Windows 2000 has a bug, it repeats the
@@ -806,20 +820,19 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                        }
                }
 
-               if (spnego.negTokenTarg.mechListMIC.length > 0) {
+               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
                        if (spnego_state->sub_sec_ready) {
                                spnego_state->needs_mic_check = true;
                        }
                }
 
                if (spnego_state->needs_mic_check) {
-                       if (spnego.negTokenTarg.responseToken.length != 0) {
+                       if (spnego_in->negTokenTarg.responseToken.length != 0) {
                                DEBUG(1, ("SPNEGO: Did not setup a mech in NEG_TOKEN_INIT\n"));
-                               spnego_free_data(&spnego);
                                return NT_STATUS_INVALID_PARAMETER;
                        }
 
-                       if (spnego.negTokenTarg.mechListMIC.length == 0
+                       if (spnego_in->negTokenTarg.mechListMIC.length == 0
                            && spnego_state->may_skip_mic_check) {
                                /*
                                 * In this case we don't require
@@ -841,11 +854,10 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                                                        spnego_state->mech_types.length,
                                                        spnego_state->mech_types.data,
                                                        spnego_state->mech_types.length,
-                                                       &spnego.negTokenTarg.mechListMIC);
+                                                       &spnego_in->negTokenTarg.mechListMIC);
                        if (!NT_STATUS_IS_OK(nt_status)) {
                                DEBUG(2,("GENSEC SPNEGO: failed to verify mechListMIC: %s\n",
                                        nt_errstr(nt_status)));
-                               spnego_free_data(&spnego);
                                return nt_status;
                        }
                        spnego_state->needs_mic_check = false;
@@ -856,7 +868,7 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                if (!spnego_state->sub_sec_ready) {
                        nt_status = gensec_update_ev(spnego_state->sub_sec_security,
                                                  out_mem_ctx, ev,
-                                                 spnego.negTokenTarg.responseToken, 
+                                                 spnego_in->negTokenTarg.responseToken,
                                                  &unwrapped_out);
                        if (NT_STATUS_IS_OK(nt_status)) {
                                spnego_state->sub_sec_ready = true;
@@ -880,7 +892,7 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                        new_spnego = gensec_have_feature(spnego_state->sub_sec_security,
                                                         GENSEC_FEATURE_NEW_SPNEGO);
 
-                       switch (spnego.negTokenTarg.negResult) {
+                       switch (spnego_in->negTokenTarg.negResult) {
                        case SPNEGO_ACCEPT_COMPLETED:
                        case SPNEGO_NONE_RESULT:
                                if (spnego_state->num_targs == 1) {
@@ -894,7 +906,7 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                                break;
 
                        case SPNEGO_ACCEPT_INCOMPLETE:
-                               if (spnego.negTokenTarg.mechListMIC.length > 0) {
+                               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
                                        new_spnego = true;
                                        break;
                                }
@@ -941,7 +953,7 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                                break;
 
                        case SPNEGO_REQUEST_MIC:
-                               if (spnego.negTokenTarg.mechListMIC.length > 0) {
+                               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
                                        new_spnego = true;
                                }
                                break;
@@ -961,17 +973,16 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                        }
                }
 
-               if (spnego.negTokenTarg.mechListMIC.length > 0) {
+               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
                        nt_status = gensec_check_packet(spnego_state->sub_sec_security,
                                                        spnego_state->mech_types.data,
                                                        spnego_state->mech_types.length,
                                                        spnego_state->mech_types.data,
                                                        spnego_state->mech_types.length,
-                                                       &spnego.negTokenTarg.mechListMIC);
+                                                       &spnego_in->negTokenTarg.mechListMIC);
                        if (!NT_STATUS_IS_OK(nt_status)) {
                                DEBUG(2,("GENSEC SPNEGO: failed to verify mechListMIC: %s\n",
                                        nt_errstr(nt_status)));
-                               spnego_free_data(&spnego);
                                return nt_status;
                        }
                        spnego_state->needs_mic_check = false;
@@ -989,7 +1000,6 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                        if (!NT_STATUS_IS_OK(nt_status)) {
                                DEBUG(2,("GENSEC SPNEGO: failed to sign mechListMIC: %s\n",
                                        nt_errstr(nt_status)));
-                               spnego_free_data(&spnego);
                                return nt_status;
                        }
                        spnego_state->needs_mic_sign = false;
@@ -1000,10 +1010,7 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                }
 
  client_response:
-               spnego_free_data(&spnego);
-
-               if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)
-                       && !NT_STATUS_IS_OK(nt_status)) {
+               if (GENSEC_UPDATE_IS_NTERROR(nt_status)) {
                        DEBUG(1, ("SPNEGO(%s) login failed: %s\n", 
                                  spnego_state->sub_sec_security->ops->name, 
                                  nt_errstr(nt_status)));
@@ -1031,7 +1038,7 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
                        /* all done - server has accepted, and we agree */
                        *out = data_blob_null;
 
-                       if (spnego.negTokenTarg.negResult != SPNEGO_ACCEPT_COMPLETED) {
+                       if (ta->negResult != SPNEGO_ACCEPT_COMPLETED) {
                                /* unless of course it did not accept */
                                DEBUG(1,("gensec_update ok but not accepted\n"));
                                nt_status = NT_STATUS_INVALID_PARAMETER;
@@ -1054,76 +1061,21 @@ static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_secur
 static NTSTATUS gensec_spnego_update_server(struct gensec_security *gensec_security,
                                            TALLOC_CTX *out_mem_ctx,
                                            struct tevent_context *ev,
-                                           const DATA_BLOB in, DATA_BLOB *out)
+                                           struct spnego_data *spnego_in,
+                                           DATA_BLOB *out)
 {
        struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
        DATA_BLOB mech_list_mic = data_blob_null;
        DATA_BLOB unwrapped_out = data_blob_null;
-       struct spnego_data spnego;
-       ssize_t len;
 
        /* and switch into the state machine */
 
        switch (spnego_state->state_position) {
        case SPNEGO_SERVER_START:
-       {
-               NTSTATUS nt_status;
-               if (in.length) {
-
-                       len = spnego_read_data(gensec_security, in, &spnego);
-                       if (len == -1) {
-                               return gensec_spnego_server_try_fallback(gensec_security, spnego_state,
-                                                                        ev, out_mem_ctx, in, out);
-                       }
-                       /* client sent NegTargetInit, we send NegTokenTarg */
-
-                       /* OK, so it's real SPNEGO, check the packet's the one we expect */
-                       if (spnego.type != spnego_state->expected_packet) {
-                               DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
-                                         spnego_state->expected_packet));
-                               dump_data(1, in.data, in.length);
-                               spnego_free_data(&spnego);
-                               return NT_STATUS_INVALID_PARAMETER;
-                       }
-
-                       nt_status = gensec_spnego_parse_negTokenInit(gensec_security,
-                                                                    spnego_state,
-                                                                    out_mem_ctx,
-                                                                    ev,
-                                                                    spnego.negTokenInit.mechTypes,
-                                                                    spnego.negTokenInit.mechToken,
-                                                                    &unwrapped_out);
-
-                       if (spnego_state->simulate_w2k) {
-                               /*
-                                * Windows 2000 returns the unwrapped token
-                                * also in the mech_list_mic field.
-                                *
-                                * In order to verify our client code,
-                                * we need a way to have a server with this
-                                * broken behaviour
-                                */
-                               mech_list_mic = unwrapped_out;
-                       }
-
-                       nt_status = gensec_spnego_server_negTokenTarg(spnego_state,
-                                                                     out_mem_ctx,
-                                                                     nt_status,
-                                                                     unwrapped_out,
-                                                                     mech_list_mic,
-                                                                     out);
-
-                       spnego_free_data(&spnego);
-
-                       return nt_status;
-               } else {
-                       nt_status = gensec_spnego_create_negTokenInit(gensec_security, spnego_state,
-                                                                     out_mem_ctx, ev, in, out);
-                       spnego_state->state_position = SPNEGO_SERVER_START;
-                       spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
-                       return nt_status;
-               }
-       }
+               return gensec_spnego_server_negTokenInit(gensec_security,
+                                                        spnego_state,
+                                                        ev, spnego_in,
+                                                        out_mem_ctx, out);
 
        case SPNEGO_SERVER_TARG:
        {
@@ -1131,39 +1083,16 @@ static NTSTATUS gensec_spnego_update_server(struct gensec_security *gensec_secur
                bool have_sign = true;
                bool new_spnego = false;
 
-               if (!in.length) {
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
-
-               len = spnego_read_data(gensec_security, in, &spnego);
-
-               if (len == -1) {
-                       DEBUG(1, ("Invalid SPNEGO request:\n"));
-                       dump_data(1, in.data, in.length);
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
-
-               /* OK, so it's real SPNEGO, check the packet's the one we expect */
-               if (spnego.type != spnego_state->expected_packet) {
-                       DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type,
-                                 spnego_state->expected_packet));
-                       dump_data(1, in.data, in.length);
-                       spnego_free_data(&spnego);
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
-
                spnego_state->num_targs++;
 
                if (!spnego_state->sub_sec_security) {
                        DEBUG(1, ("SPNEGO: Did not setup a mech in NEG_TOKEN_INIT\n"));
-                       spnego_free_data(&spnego);
                        return NT_STATUS_INVALID_PARAMETER;
                }
 
                if (spnego_state->needs_mic_check) {
-                       if (spnego.negTokenTarg.responseToken.length != 0) {
+                       if (spnego_in->negTokenTarg.responseToken.length != 0) {
                                DEBUG(1, ("SPNEGO: Did not setup a mech in NEG_TOKEN_INIT\n"));
-                               spnego_free_data(&spnego);
                                return NT_STATUS_INVALID_PARAMETER;
                        }
 
@@ -1172,7 +1101,7 @@ static NTSTATUS gensec_spnego_update_server(struct gensec_security *gensec_secur
                                                        spnego_state->mech_types.length,
                                                        spnego_state->mech_types.data,
                                                        spnego_state->mech_types.length,
-                                                       &spnego.negTokenTarg.mechListMIC);
+                                                       &spnego_in->negTokenTarg.mechListMIC);
                        if (NT_STATUS_IS_OK(nt_status)) {
                                spnego_state->needs_mic_check = false;
                                spnego_state->done_mic_check = true;
@@ -1183,15 +1112,19 @@ static NTSTATUS gensec_spnego_update_server(struct gensec_security *gensec_secur
                        goto server_response;
                }
 
-               nt_status = gensec_update_ev(spnego_state->sub_sec_security,
-                                            out_mem_ctx, ev,
-                                            spnego.negTokenTarg.responseToken,
-                                            &unwrapped_out);
-               if (NT_STATUS_IS_OK(nt_status)) {
-                       spnego_state->sub_sec_ready = true;
-               }
-               if (!NT_STATUS_IS_OK(nt_status)) {
-                       goto server_response;
+               if (!spnego_state->sub_sec_ready) {
+                       nt_status = gensec_update_ev(spnego_state->sub_sec_security,
+                                                    out_mem_ctx, ev,
+                                                    spnego_in->negTokenTarg.responseToken,
+                                                    &unwrapped_out);
+                       if (NT_STATUS_IS_OK(nt_status)) {
+                               spnego_state->sub_sec_ready = true;
+                       }
+                       if (!NT_STATUS_IS_OK(nt_status)) {
+                               goto server_response;
+                       }
+               } else {
+                       nt_status = NT_STATUS_OK;
                }
 
                have_sign = gensec_have_feature(spnego_state->sub_sec_security,
@@ -1201,7 +1134,7 @@ static NTSTATUS gensec_spnego_update_server(struct gensec_security *gensec_secur
                }
                new_spnego = gensec_have_feature(spnego_state->sub_sec_security,
                                                 GENSEC_FEATURE_NEW_SPNEGO);
-               if (spnego.negTokenTarg.mechListMIC.length > 0) {
+               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
                        new_spnego = true;
                }
 
@@ -1210,13 +1143,13 @@ static NTSTATUS gensec_spnego_update_server(struct gensec_security *gensec_secur
                        spnego_state->needs_mic_sign = true;
                }
 
-               if (have_sign && spnego.negTokenTarg.mechListMIC.length > 0) {
+               if (have_sign && spnego_in->negTokenTarg.mechListMIC.length > 0) {
                        nt_status = gensec_check_packet(spnego_state->sub_sec_security,
                                                        spnego_state->mech_types.data,
                                                        spnego_state->mech_types.length,
                                                        spnego_state->mech_types.data,
                                                        spnego_state->mech_types.length,
-                                                       &spnego.negTokenTarg.mechListMIC);
+                                                       &spnego_in->negTokenTarg.mechListMIC);
                        if (!NT_STATUS_IS_OK(nt_status)) {
                                DEBUG(2,("GENSEC SPNEGO: failed to verify mechListMIC: %s\n",
                                        nt_errstr(nt_status)));
@@ -1248,14 +1181,12 @@ static NTSTATUS gensec_spnego_update_server(struct gensec_security *gensec_secur
                }
 
  server_response:
-               nt_status = gensec_spnego_server_negTokenTarg(spnego_state,
-                                                             out_mem_ctx,
-                                                             nt_status,
-                                                             unwrapped_out,
-                                                             mech_list_mic,
-                                                             out);
-
-               spnego_free_data(&spnego);
+               nt_status = gensec_spnego_server_response(spnego_state,
+                                                         out_mem_ctx,
+                                                         nt_status,
+                                                         unwrapped_out,
+                                                         mech_list_mic,
+                                                         out);
 
                return nt_status;
        }
@@ -1272,6 +1203,8 @@ struct gensec_spnego_update_state {
        struct gensec_security *gensec;
        struct spnego_state *spnego;
        DATA_BLOB full_in;
+       struct spnego_data _spnego_in;
+       struct spnego_data *spnego_in;
        NTSTATUS status;
        DATA_BLOB out;
 };
@@ -1315,6 +1248,7 @@ static struct tevent_req *gensec_spnego_update_send(TALLOC_CTX *mem_ctx,
        struct tevent_req *req = NULL;
        struct gensec_spnego_update_state *state = NULL;
        NTSTATUS status;
+       ssize_t len;
 
        req = tevent_req_create(mem_ctx, &state,
                                struct gensec_spnego_update_state);
@@ -1333,15 +1267,12 @@ static struct tevent_req *gensec_spnego_update_send(TALLOC_CTX *mem_ctx,
 
                status = gensec_spnego_update_out(gensec_security,
                                                  state, &state->out);
-               state->status = status;
-               if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-                       tevent_req_done(req);
-                       return tevent_req_post(req, ev);
-               }
-               if (tevent_req_nterror(req, status)) {
+               if (GENSEC_UPDATE_IS_NTERROR(status)) {
+                       tevent_req_nterror(req, status);
                        return tevent_req_post(req, ev);
                }
 
+               state->status = status;
                tevent_req_done(req);
                return tevent_req_post(req, ev);
        }
@@ -1357,6 +1288,78 @@ static struct tevent_req *gensec_spnego_update_send(TALLOC_CTX *mem_ctx,
                return tevent_req_post(req, ev);
        }
 
+       /* Check if we got a valid SPNEGO blob... */
+
+       switch (spnego_state->state_position) {
+       case SPNEGO_FALLBACK:
+               break;
+
+       case SPNEGO_CLIENT_TARG:
+       case SPNEGO_SERVER_TARG:
+               if (state->full_in.length == 0) {
+                       tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
+                       return tevent_req_post(req, ev);
+               }
+
+               /* fall through */
+       case SPNEGO_CLIENT_START:
+       case SPNEGO_SERVER_START:
+
+               if (state->full_in.length == 0) {
+                       /* create_negTokenInit later */
+                       break;
+               }
+
+               len = spnego_read_data(state,
+                                      state->full_in,
+                                      &state->_spnego_in);
+               if (len == -1) {
+                       if (spnego_state->state_position != SPNEGO_SERVER_START) {
+                               DEBUG(1, ("Invalid SPNEGO request:\n"));
+                               dump_data(1, state->full_in.data,
+                                         state->full_in.length);
+                               tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
+                               return tevent_req_post(req, ev);
+                       }
+
+                       /*
+                        * This is the 'fallback' case, where we don't get
+                        * SPNEGO, and have to try all the other options (and
+                        * hope they all have a magic string they check)
+                        */
+                       status = gensec_spnego_server_try_fallback(gensec_security,
+                                                                  spnego_state,
+                                                                  state,
+                                                                  state->full_in);
+                       if (tevent_req_nterror(req, status)) {
+                               return tevent_req_post(req, ev);
+                       }
+
+                       /*
+                        * We'll continue with SPNEGO_FALLBACK below...
+                        */
+                       break;
+               }
+               state->spnego_in = &state->_spnego_in;
+
+               /* OK, so it's real SPNEGO, check the packet's the one we expect */
+               if (state->spnego_in->type != spnego_state->expected_packet) {
+                       DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n",
+                                 state->spnego_in->type,
+                                 spnego_state->expected_packet));
+                       dump_data(1, state->full_in.data,
+                                 state->full_in.length);
+                       tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
+                       return tevent_req_post(req, ev);
+               }
+
+               break;
+
+       default:
+               smb_panic(__location__);
+               return NULL;
+       }
+
        /* and switch into the state machine */
 
        switch (spnego_state->state_position) {
@@ -1368,28 +1371,46 @@ static struct tevent_req *gensec_spnego_update_send(TALLOC_CTX *mem_ctx,
                break;
 
        case SPNEGO_CLIENT_START:
+               if (state->spnego_in == NULL) {
+                       /* client to produce negTokenInit */
+                       status = gensec_spnego_create_negTokenInit(gensec_security,
+                                                       spnego_state, state, ev,
+                                                       &spnego_state->out_frag);
+                       break;
+               }
+
+               /* fall through */
        case SPNEGO_CLIENT_TARG:
                status = gensec_spnego_update_client(gensec_security,
                                                     state, ev,
-                                                    state->full_in,
+                                                    state->spnego_in,
                                                     &spnego_state->out_frag);
                break;
 
        case SPNEGO_SERVER_START:
+               if (state->spnego_in == NULL) {
+                       /* server to produce negTokenInit */
+                       status = gensec_spnego_create_negTokenInit(gensec_security,
+                                                       spnego_state, state, ev,
+                                                       &spnego_state->out_frag);
+                       break;
+               }
+
+               /* fall through */
        case SPNEGO_SERVER_TARG:
                status = gensec_spnego_update_server(gensec_security,
                                                     state, ev,
-                                                    state->full_in,
+                                                    state->spnego_in,
                                                     &spnego_state->out_frag);
                break;
 
-       case SPNEGO_DONE:
-               /* We should not be called after we are 'done' */
-               tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
-               return tevent_req_post(req, ev);
-
        default:
-               tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
+               smb_panic(__location__);
+               return NULL;
+       }
+
+       if (GENSEC_UPDATE_IS_NTERROR(status)) {
+               tevent_req_nterror(req, status);
                return tevent_req_post(req, ev);
        }
 
@@ -1400,26 +1421,21 @@ static struct tevent_req *gensec_spnego_update_send(TALLOC_CTX *mem_ctx,
 
                status = gensec_may_reset_crypto(spnego_state->sub_sec_security,
                                                 reset_full);
-       }
-       if (!NT_STATUS_IS_OK(status) &&
-           !NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-               tevent_req_nterror(req, status);
-               return tevent_req_post(req, ev);
+               if (tevent_req_nterror(req, status)) {
+                       return tevent_req_post(req, ev);
+               }
        }
 
        spnego_state->out_status = status;
 
        status = gensec_spnego_update_out(gensec_security,
                                          state, &state->out);
-       state->status = status;
-       if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-               tevent_req_done(req);
-               return tevent_req_post(req, ev);
-       }
-       if (tevent_req_nterror(req, status)) {
+       if (GENSEC_UPDATE_IS_NTERROR(status)) {
+               tevent_req_nterror(req, status);
                return tevent_req_post(req, ev);
        }
 
+       state->status = status;
        tevent_req_done(req);
        return tevent_req_post(req, ev);
 }
@@ -1434,6 +1450,23 @@ static NTSTATUS gensec_spnego_update_in(struct gensec_security *gensec_security,
 
        *full_in = data_blob_null;
 
+       switch (spnego_state->state_position) {
+       case SPNEGO_FALLBACK:
+               *full_in = in;
+               spnego_state->in_needed = 0;
+               return NT_STATUS_OK;
+
+       case SPNEGO_CLIENT_START:
+       case SPNEGO_CLIENT_TARG:
+       case SPNEGO_SERVER_START:
+       case SPNEGO_SERVER_TARG:
+               break;
+
+       case SPNEGO_DONE:
+       default:
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+
        if (spnego_state->in_needed == 0) {
                size_t size = 0;
                int ret;