auth/spnego: split out a gensec_spnego_server_negTokenInit() function.
[samba.git] / auth / gensec / spnego.c
index e3f9def8b82947104c9c8517386754adf66ee30b..53fd81c591dcda8d52bc83c59a149f45c983dc4b 100644 (file)
@@ -77,6 +77,11 @@ struct spnego_state {
        NTSTATUS out_status;
 };
 
+static void gensec_spnego_update_sub_abort(struct spnego_state *spnego_state)
+{
+       spnego_state->sub_sec_ready = false;
+       TALLOC_FREE(spnego_state->sub_sec_security);
+}
 
 static NTSTATUS gensec_spnego_client_start(struct gensec_security *gensec_security)
 {
@@ -135,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;
@@ -190,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;
@@ -207,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);
@@ -246,8 +260,10 @@ static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_
                                nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
                                                                     all_sec[i].op);
                                if (!NT_STATUS_IS_OK(nt_status)) {
-                                       talloc_free(spnego_state->sub_sec_security);
-                                       spnego_state->sub_sec_security = NULL;
+                                       /*
+                                        * Pretend we never started it
+                                        */
+                                       gensec_spnego_update_sub_abort(spnego_state);
                                        break;
                                }
 
@@ -275,12 +291,14 @@ static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_
                                }
                                if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER) || 
                                    NT_STATUS_EQUAL(nt_status, NT_STATUS_CANT_ACCESS_DOMAIN_INFO)) {
-                                       /* Pretend we never started it (lets the first run find some incompatible demand) */
 
                                        DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed to parse contents: %s\n", 
                                                  spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
-                                       talloc_free(spnego_state->sub_sec_security);
-                                       spnego_state->sub_sec_security = NULL;
+
+                                       /*
+                                        * Pretend we never started it
+                                        */
+                                       gensec_spnego_update_sub_abort(spnego_state);
                                        break;
                                }
 
@@ -301,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++) {
@@ -314,8 +333,10 @@ static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_
                        nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
                                                             all_sec[i].op);
                        if (!NT_STATUS_IS_OK(nt_status)) {
-                               talloc_free(spnego_state->sub_sec_security);
-                               spnego_state->sub_sec_security = NULL;
+                               /*
+                                * Pretend we never started it.
+                                */
+                               gensec_spnego_update_sub_abort(spnego_state);
                                continue;
                        }
 
@@ -325,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;
@@ -368,9 +389,10 @@ static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_
                                                          principal,
                                                          next, nt_errstr(nt_status)));
 
-                                       /* Pretend we never started it (lets the first run find some incompatible demand) */
-                                       talloc_free(spnego_state->sub_sec_security);
-                                       spnego_state->sub_sec_security = NULL;
+                                       /*
+                                        * Pretend we never started it.
+                                        */
+                                       gensec_spnego_update_sub_abort(spnego_state);
                                        continue;
                                }
                        }
@@ -392,22 +414,19 @@ 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)));
-                       talloc_free(spnego_state->sub_sec_security);
-                       spnego_state->sub_sec_security = NULL;
 
                        /* We started the mech correctly, and the
                         * input from the other side was valid.
                         * Return the error (say bad password, invalid
                         * ticket) */
+                       gensec_spnego_update_sub_abort(spnego_state);
                        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"));
@@ -427,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;
@@ -457,8 +476,7 @@ static NTSTATUS gensec_spnego_create_negTokenInit(struct gensec_security *gensec
                nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
                                                     all_sec[i].op);
                if (!NT_STATUS_IS_OK(nt_status)) {
-                       talloc_free(spnego_state->sub_sec_security);
-                       spnego_state->sub_sec_security = NULL;
+                       gensec_spnego_update_sub_abort(spnego_state);
                        continue;
                }
 
@@ -473,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;
@@ -501,10 +518,11 @@ static NTSTATUS gensec_spnego_create_negTokenInit(struct gensec_security *gensec
                                          spnego_state->sub_sec_security->ops->name,
                                          principal,
                                          next, nt_errstr(nt_status)));
-                               talloc_free(spnego_state->sub_sec_security);
-                               spnego_state->sub_sec_security = NULL;
-                               /* Pretend we never started it (lets the first run find some incompatible demand) */
 
+                               /*
+                                * Pretend we never started it
+                                */
+                               gensec_spnego_update_sub_abort(spnego_state);
                                continue;
                        }
                }
@@ -544,27 +562,98 @@ 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;
-       } 
-       talloc_free(spnego_state->sub_sec_security);
-       spnego_state->sub_sec_security = NULL;
+       }
+       gensec_spnego_update_sub_abort(spnego_state);
 
        DEBUG(10, ("Failed to setup SPNEGO negTokenInit request: %s\n", nt_errstr(nt_status)));
        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;
 
@@ -607,457 +696,188 @@ 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(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx, 
-                                    struct tevent_context *ev,
-                                    const DATA_BLOB in, DATA_BLOB *out) 
+static NTSTATUS gensec_spnego_update_client(struct gensec_security *gensec_security,
+                                           TALLOC_CTX *out_mem_ctx,
+                                           struct tevent_context *ev,
+                                           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;
 
-       if (!out_mem_ctx) {
-               out_mem_ctx = spnego_state;
-       }
-
        /* and switch into the state machine */
 
        switch (spnego_state->state_position) {
-       case SPNEGO_FALLBACK:
-               return gensec_update_ev(spnego_state->sub_sec_security,
-                                       out_mem_ctx, ev, in, out);
-       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;
-               }
-       }
-
        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;
+               return gensec_spnego_client_negTokenInit(gensec_security,
+                                                        spnego_state,
+                                                        ev, spnego_in,
+                                                        out_mem_ctx, out);
 
-               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;
-               }
+       case SPNEGO_CLIENT_TARG:
+       {
+               NTSTATUS nt_status = NT_STATUS_INTERNAL_ERROR;
+               const struct spnego_negTokenTarg *ta =
+                       &spnego_in->negTokenTarg;
 
-               len = spnego_read_data(gensec_security, in, &spnego);
+               spnego_state->num_targs++;
 
-               if (len == -1) {
-                       DEBUG(1, ("Invalid SPNEGO request:\n"));
-                       dump_data(1, in.data, in.length);
-                       return NT_STATUS_INVALID_PARAMETER;
+               if (ta->negResult == SPNEGO_REJECT) {
+                       return NT_STATUS_LOGON_FAILURE;
                }
 
-               /* 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_in->negTokenTarg.negResult == SPNEGO_REQUEST_MIC) {
+                       spnego_state->mic_requested = true;
                }
 
-               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);
+               /* Server didn't like our choice of mech, and chose something else */
+               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, 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)) {
+                               return nt_status;
+                       }
+                       /* select the sub context */
+                       nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
+                                                            ta->supportedMech);
+                       if (!NT_STATUS_IS_OK(nt_status)) {
+                               return nt_status;
                        }
-               }
-
-               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;
-       }
-       case SPNEGO_SERVER_TARG:
-       {
-               NTSTATUS nt_status;
-               bool have_sign = true;
-               bool new_spnego = false;
 
-               if (!in.length) {
-                       return NT_STATUS_INVALID_PARAMETER;
+                       spnego_state->neg_oid = talloc_strdup(spnego_state,
+                                               ta->supportedMech);
+                       if (spnego_state->neg_oid == NULL) {
+                               return NT_STATUS_NO_MEMORY;
+                       };
                }
 
-               len = spnego_read_data(gensec_security, in, &spnego);
+               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
+                       DATA_BLOB *m = &spnego_in->negTokenTarg.mechListMIC;
+                       const DATA_BLOB *r = &spnego_in->negTokenTarg.responseToken;
 
-               if (len == -1) {
-                       DEBUG(1, ("Invalid SPNEGO request:\n"));
-                       dump_data(1, in.data, in.length);
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
+                       /*
+                        * Windows 2000 has a bug, it repeats the
+                        * responseToken in the mechListMIC field.
+                        */
+                       if (m->length == r->length) {
+                               int cmp;
 
-               /* 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;
+                               cmp = memcmp(m->data, r->data, m->length);
+                               if (cmp == 0) {
+                                       data_blob_free(m);
+                               }
+                       }
                }
 
-               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_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_in->negTokenTarg.mechListMIC.length == 0
+                           && spnego_state->may_skip_mic_check) {
+                               /*
+                                * In this case we don't require
+                                * a mechListMIC from the server.
+                                *
+                                * This works around bugs in the Azure
+                                * and Apple spnego implementations.
+                                *
+                                * See
+                                * https://bugzilla.samba.org/show_bug.cgi?id=11994
+                                */
+                               spnego_state->needs_mic_check = false;
+                               nt_status = NT_STATUS_OK;
+                               goto client_response;
+                       }
+
                        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);
-                       if (NT_STATUS_IS_OK(nt_status)) {
-                               spnego_state->needs_mic_check = false;
-                               spnego_state->done_mic_check = true;
-                       } else {
+                                                       &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)));
+                               return nt_status;
                        }
-                       goto server_response;
+                       spnego_state->needs_mic_check = false;
+                       spnego_state->done_mic_check = true;
+                       goto client_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;
-               }
-
-               have_sign = gensec_have_feature(spnego_state->sub_sec_security,
-                                               GENSEC_FEATURE_SIGN);
-               if (spnego_state->simulate_w2k) {
-                       have_sign = false;
-               }
-               new_spnego = gensec_have_feature(spnego_state->sub_sec_security,
-                                                GENSEC_FEATURE_NEW_SPNEGO);
-               if (spnego.negTokenTarg.mechListMIC.length > 0) {
-                       new_spnego = true;
-               }
-
-               if (have_sign && new_spnego) {
-                       spnego_state->needs_mic_check = true;
-                       spnego_state->needs_mic_sign = true;
-               }
-
-               if (have_sign && spnego.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);
-                       if (!NT_STATUS_IS_OK(nt_status)) {
-                               DEBUG(2,("GENSEC SPNEGO: failed to verify mechListMIC: %s\n",
-                                       nt_errstr(nt_status)));
-                               goto server_response;
-                       }
-
-                       spnego_state->needs_mic_check = false;
-                       spnego_state->done_mic_check = true;
-               }
-
-               if (spnego_state->needs_mic_sign) {
-                       nt_status = gensec_sign_packet(spnego_state->sub_sec_security,
-                                                      out_mem_ctx,
-                                                      spnego_state->mech_types.data,
-                                                      spnego_state->mech_types.length,
-                                                      spnego_state->mech_types.data,
-                                                      spnego_state->mech_types.length,
-                                                      &mech_list_mic);
-                       if (!NT_STATUS_IS_OK(nt_status)) {
-                               DEBUG(2,("GENSEC SPNEGO: failed to sign mechListMIC: %s\n",
-                                       nt_errstr(nt_status)));
-                               goto server_response;
-                       }
-                       spnego_state->needs_mic_sign = false;
-               }
-
-               if (spnego_state->needs_mic_check) {
-                       nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
-               }
-
- 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);
-
-               return nt_status;
-       }
-       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;
-               }
-
-               spnego_state->num_targs++;
-
-               if (spnego.negTokenTarg.negResult == SPNEGO_REJECT) {
-                       spnego_free_data(&spnego);
-                       return NT_STATUS_LOGON_FAILURE;
-               }
-
-               if (spnego.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) {
-                       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)));
-                       spnego_state->downgraded = true;
-                       spnego_state->sub_sec_ready = false;
-                       talloc_free(spnego_state->sub_sec_security);
-                       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);
-                       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);
-                       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;
-
-                       /*
-                        * Windows 2000 has a bug, it repeats the
-                        * responseToken in the mechListMIC field.
-                        */
-                       if (m->length == r->length) {
-                               int cmp;
-
-                               cmp = memcmp(m->data, r->data, m->length);
-                               if (cmp == 0) {
-                                       data_blob_free(m);
-                               }
-                       }
-               }
-
-               if (spnego.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) {
-                               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
-                           && spnego_state->may_skip_mic_check) {
-                               /*
-                                * In this case we don't require
-                                * a mechListMIC from the server.
-                                *
-                                * This works around bugs in the Azure
-                                * and Apple spnego implementations.
-                                *
-                                * See
-                                * https://bugzilla.samba.org/show_bug.cgi?id=11994
-                                */
-                               spnego_state->needs_mic_check = false;
-                               nt_status = NT_STATUS_OK;
-                               goto client_response;
-                       }
-
-                       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);
-                       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;
-                       spnego_state->done_mic_check = true;
-                       goto client_response;
-               }
-
-               if (!spnego_state->sub_sec_ready) {
-                       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 client_response;
-                       }
-               } else {
-                       nt_status = NT_STATUS_OK;
+               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 client_response;
+                       }
+               } else {
+                       nt_status = NT_STATUS_OK;
                }
 
                if (!spnego_state->done_mic_check) {
@@ -1072,7 +892,7 @@ static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TA
                        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) {
@@ -1086,7 +906,7 @@ static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TA
                                break;
 
                        case SPNEGO_ACCEPT_INCOMPLETE:
-                               if (spnego.negTokenTarg.mechListMIC.length > 0) {
+                               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
                                        new_spnego = true;
                                        break;
                                }
@@ -1133,7 +953,7 @@ static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TA
                                break;
 
                        case SPNEGO_REQUEST_MIC:
-                               if (spnego.negTokenTarg.mechListMIC.length > 0) {
+                               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
                                        new_spnego = true;
                                }
                                break;
@@ -1153,17 +973,16 @@ static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TA
                        }
                }
 
-               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;
@@ -1181,7 +1000,6 @@ static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TA
                        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;
@@ -1192,10 +1010,7 @@ static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TA
                }
 
  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)));
@@ -1223,7 +1038,7 @@ static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TA
                        /* 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;
@@ -1234,17 +1049,162 @@ static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TA
 
                return nt_status;
        }
-       case SPNEGO_DONE:
-               /* We should not be called after we are 'done' */
-               return NT_STATUS_INVALID_PARAMETER;
+
+       default:
+               break;
        }
-       return NT_STATUS_INVALID_PARAMETER;
+
+       smb_panic(__location__);
+       return NT_STATUS_INTERNAL_ERROR;
+}
+
+static NTSTATUS gensec_spnego_update_server(struct gensec_security *gensec_security,
+                                           TALLOC_CTX *out_mem_ctx,
+                                           struct tevent_context *ev,
+                                           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;
+
+       /* and switch into the state machine */
+
+       switch (spnego_state->state_position) {
+       case SPNEGO_SERVER_START:
+               return gensec_spnego_server_negTokenInit(gensec_security,
+                                                        spnego_state,
+                                                        ev, spnego_in,
+                                                        out_mem_ctx, out);
+
+       case SPNEGO_SERVER_TARG:
+       {
+               NTSTATUS nt_status;
+               bool have_sign = true;
+               bool new_spnego = false;
+
+               spnego_state->num_targs++;
+
+               if (!spnego_state->sub_sec_security) {
+                       DEBUG(1, ("SPNEGO: Did not setup a mech in NEG_TOKEN_INIT\n"));
+                       return NT_STATUS_INVALID_PARAMETER;
+               }
+
+               if (spnego_state->needs_mic_check) {
+                       if (spnego_in->negTokenTarg.responseToken.length != 0) {
+                               DEBUG(1, ("SPNEGO: Did not setup a mech in NEG_TOKEN_INIT\n"));
+                               return NT_STATUS_INVALID_PARAMETER;
+                       }
+
+                       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_in->negTokenTarg.mechListMIC);
+                       if (NT_STATUS_IS_OK(nt_status)) {
+                               spnego_state->needs_mic_check = false;
+                               spnego_state->done_mic_check = true;
+                       } else {
+                               DEBUG(2,("GENSEC SPNEGO: failed to verify mechListMIC: %s\n",
+                                       nt_errstr(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,
+                                               GENSEC_FEATURE_SIGN);
+               if (spnego_state->simulate_w2k) {
+                       have_sign = false;
+               }
+               new_spnego = gensec_have_feature(spnego_state->sub_sec_security,
+                                                GENSEC_FEATURE_NEW_SPNEGO);
+               if (spnego_in->negTokenTarg.mechListMIC.length > 0) {
+                       new_spnego = true;
+               }
+
+               if (have_sign && new_spnego) {
+                       spnego_state->needs_mic_check = true;
+                       spnego_state->needs_mic_sign = true;
+               }
+
+               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_in->negTokenTarg.mechListMIC);
+                       if (!NT_STATUS_IS_OK(nt_status)) {
+                               DEBUG(2,("GENSEC SPNEGO: failed to verify mechListMIC: %s\n",
+                                       nt_errstr(nt_status)));
+                               goto server_response;
+                       }
+
+                       spnego_state->needs_mic_check = false;
+                       spnego_state->done_mic_check = true;
+               }
+
+               if (spnego_state->needs_mic_sign) {
+                       nt_status = gensec_sign_packet(spnego_state->sub_sec_security,
+                                                      out_mem_ctx,
+                                                      spnego_state->mech_types.data,
+                                                      spnego_state->mech_types.length,
+                                                      spnego_state->mech_types.data,
+                                                      spnego_state->mech_types.length,
+                                                      &mech_list_mic);
+                       if (!NT_STATUS_IS_OK(nt_status)) {
+                               DEBUG(2,("GENSEC SPNEGO: failed to sign mechListMIC: %s\n",
+                                       nt_errstr(nt_status)));
+                               goto server_response;
+                       }
+                       spnego_state->needs_mic_sign = false;
+               }
+
+               if (spnego_state->needs_mic_check) {
+                       nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
+               }
+
+ server_response:
+               nt_status = gensec_spnego_server_response(spnego_state,
+                                                         out_mem_ctx,
+                                                         nt_status,
+                                                         unwrapped_out,
+                                                         mech_list_mic,
+                                                         out);
+
+               return nt_status;
+       }
+
+       default:
+               break;
+       }
+
+       smb_panic(__location__);
+       return NT_STATUS_INTERNAL_ERROR;
 }
 
 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;
 };
@@ -1288,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);
@@ -1306,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);
        }
@@ -1330,10 +1288,132 @@ static struct tevent_req *gensec_spnego_update_send(TALLOC_CTX *mem_ctx,
                return tevent_req_post(req, ev);
        }
 
-       status = gensec_spnego_update(gensec_security,
-                                     state, ev,
-                                     state->full_in,
-                                     &spnego_state->out_frag);
+       /* 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) {
+       case SPNEGO_FALLBACK:
+               status = gensec_update_ev(spnego_state->sub_sec_security,
+                                         state, ev,
+                                         state->full_in,
+                                         &spnego_state->out_frag);
+               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->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->spnego_in,
+                                                    &spnego_state->out_frag);
+               break;
+
+       default:
+               smb_panic(__location__);
+               return NULL;
+       }
+
+       if (GENSEC_UPDATE_IS_NTERROR(status)) {
+               tevent_req_nterror(req, status);
+               return tevent_req_post(req, ev);
+       }
+
        if (NT_STATUS_IS_OK(status)) {
                bool reset_full = true;
 
@@ -1341,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);
 }
@@ -1375,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;