From 86fb7c58cfb5a78db2bcb991b283edbc97124722 Mon Sep 17 00:00:00 2001
From: Luke Howard <lukeh@padl.com>
Date: Thu, 9 Sep 2010 14:28:37 +0200
Subject: [PATCH] more work on state machines

---
 mech_eap/accept_sec_context.c | 18 ++++++++++--------
 mech_eap/gssapiP_eap.h        |  1 +
 mech_eap/init_sec_context.c   | 38 +++++++++++++++++++++++++-------------
 3 files changed, 36 insertions(+), 21 deletions(-)

diff --git a/mech_eap/accept_sec_context.c b/mech_eap/accept_sec_context.c
index 25f35ce..80b2697 100644
--- a/mech_eap/accept_sec_context.c
+++ b/mech_eap/accept_sec_context.c
@@ -111,14 +111,16 @@ gss_accept_sec_context(OM_uint32 *minor,
     if (GSS_ERROR(major))
         goto cleanup;
 
-    major = (sm->processToken)(minor,
-                               ctx,
-                               cred,
-                               &innerInputToken,
-                               input_chan_bindings,
-                               &innerOutputToken);
-    if (GSS_ERROR(major))
-        goto cleanup;
+    do {
+        major = (sm->processToken)(minor,
+                                   ctx,
+                                   cred,
+                                   &innerInputToken,
+                                   input_chan_bindings,
+                                   &innerOutputToken);
+        if (GSS_ERROR(major))
+            goto cleanup;
+    } while (major == GSS_S_CONTINUE_NEEDED && innerOutputToken.length == 0);
 
     if (src_name != NULL && ctx->initiatorName != GSS_C_NO_NAME) {
         major = gss_duplicate_name(&minor, ctx->initiatorName, src_name);
diff --git a/mech_eap/gssapiP_eap.h b/mech_eap/gssapiP_eap.h
index 28e65b3..6013c1d 100644
--- a/mech_eap/gssapiP_eap.h
+++ b/mech_eap/gssapiP_eap.h
@@ -118,6 +118,7 @@ struct eap_gss_initiator_ctx {
     struct eap_peer_config eapPeerConfig;
     struct eap_config eapConfig;
     struct eap_sm *eap;
+    struct wpabuf reqData;
 };
 
 /* Acceptor context flags */
diff --git a/mech_eap/init_sec_context.c b/mech_eap/init_sec_context.c
index a25e123..15bdff3 100644
--- a/mech_eap/init_sec_context.c
+++ b/mech_eap/init_sec_context.c
@@ -150,7 +150,9 @@ peerSetInt(void *data, enum eapol_int_var variable,
 static struct wpabuf *
 peerGetEapReqData(void *ctx)
 {
-    return NULL;
+    gss_ctx_id_t gssCtx = (gss_ctx_id_t)ctx;
+
+    return &gssCtx->initiatorCtx.reqData;
 }
 
 static void
@@ -318,6 +320,9 @@ eapGssSmInitAuthenticate(OM_uint32 *minor,
             goto cleanup;
     }
 
+    wpabuf_set(&ctx->initiatorCtx.reqData,
+               inputToken->value, inputToken->length);
+
     code = eap_peer_sm_step(ctx->initiatorCtx.eap);
 
     if (ctx->flags & CTX_FLAG_EAP_RESP) {
@@ -351,6 +356,7 @@ eapGssSmInitAuthenticate(OM_uint32 *minor,
     }
 
 cleanup:
+    wpabuf_set(&ctx->initiatorCtx.reqData, NULL, 0);
     peerConfigFree(&tmpMinor, ctx);
 
     return major;
@@ -499,18 +505,24 @@ gss_init_sec_context(OM_uint32 *minor,
         innerInputToken.value = NULL;
     }
 
-    major = (sm->processToken)(minor,
-                               cred,
-                               ctx,
-                               target_name,
-                               mech_type,
-                               req_flags,
-                               time_req,
-                               input_chan_bindings,
-                               &innerInputToken,
-                               &innerOutputToken);
-    if (GSS_ERROR(major))
-        goto cleanup;
+    /*
+     * Advance through state machine whilst empty tokens are emitted and
+     * the status is not GSS_S_COMPLETE or an error status.
+     */
+    do {
+        major = (sm->processToken)(minor,
+                                   cred,
+                                   ctx,
+                                   target_name,
+                                   mech_type,
+                                   req_flags,
+                                   time_req,
+                                   input_chan_bindings,
+                                   &innerInputToken,
+                                   &innerOutputToken);
+        if (GSS_ERROR(major))
+            goto cleanup;
+    } while (major == GSS_S_CONTINUE_NEEDED && innerOutputToken.length == 0);
 
     if (actual_mech_type != NULL) {
         if (!gssEapInternalizeOid(ctx->mechanismUsed, actual_mech_type))
-- 
2.1.4