rpcrt4: Call CompleteAuthToken if InitializeSecurityContext returned a code that…

rpcrt4: Call CompleteAuthToken if InitializeSecurityContext returned a code that says it should be called. Currently, the only security provider that requires this is the Digest one, which we don't really care about, but is supported for completeness.

rpcrt4: Call CompleteAuthToken if InitializeSecurityContext returned a code that…
88295b9a · Rob Shearman · Alexandre Julliard · 88e9b3fa · 88295b9a
Commit 88295b9a authored Mar 26, 2007 by Rob Shearman Committed by Alexandre Julliard Mar 27, 2007
Hide whitespace changes
Inline Side-by-side

Showing with 16 additions and 1 deletion

rpc_message.c dlls/rpcrt4/rpc_message.c +16 -1

No files found.
--- a/dlls/rpcrt4/rpc_message.c
+++ b/dlls/rpcrt4/rpc_message.c
@@ -501,7 +501,22 @@ static RPC_STATUS RPCRT4_ClientAuthorize(RpcConnection *conn, SecBuffer *in,
      return ERROR_ACCESS_DENIED; /* FIXME: is this correct? */
  }

-  TRACE("r = 0x%08x, cbBuffer = %ld, attr = 0x%08x\n", r, out->cbBuffer, conn->attr);
+  TRACE("r = 0x%08x, attr = 0x%08x\n", r, conn->attr);
+
+  if ((r == SEC_I_COMPLETE_NEEDED) || (r == SEC_I_COMPLETE_AND_CONTINUE))
+  {
+      TRACE("complete needed\n");
+      r = CompleteAuthToken(&conn->ctx, &out_desc);
+      if (FAILED(r))
+      {
+          HeapFree(GetProcessHeap(), 0, out->pvBuffer);
+          out->pvBuffer = NULL;
+          WARN("CompleteAuthToken failed with error 0x%08x\n", r);
+          return ERROR_ACCESS_DENIED; /* FIXME: is this correct? */
+      }
+  }
+
+  TRACE("cbBuffer = %ld\n", out->cbBuffer);

  return RPC_S_OK;
 }