[03/10] secur32: Implement AcceptSecurityContext for Kerberos.

[Hans Leidekker]

Signed-off-by: Hans Leidekker <hans at codeweavers.com>
---
 dlls/secur32/kerberos.c | 77 ++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 73 insertions(+), 4 deletions(-)

diff --git a/dlls/secur32/kerberos.c b/dlls/secur32/kerberos.c
index 0e486d7c51..474943e10c 100644
--- a/dlls/secur32/kerberos.c
+++ b/dlls/secur32/kerberos.c
@@ -46,6 +46,7 @@ WINE_DECLARE_DEBUG_CHANNEL(winediag);
 static void *libgssapi_krb5_handle;
 
 #define MAKE_FUNCPTR(f) static typeof(f) * p##f
+MAKE_FUNCPTR(gss_accept_sec_context);
 MAKE_FUNCPTR(gss_acquire_cred);
 MAKE_FUNCPTR(gss_delete_sec_context);
 MAKE_FUNCPTR(gss_import_name);
@@ -69,6 +70,7 @@ static BOOL load_gssapi_krb5(void)
         goto fail; \
     }
 
+    LOAD_FUNCPTR(gss_accept_sec_context)
     LOAD_FUNCPTR(gss_acquire_cred)
     LOAD_FUNCPTR(gss_delete_sec_context)
     LOAD_FUNCPTR(gss_import_name)
@@ -202,6 +204,19 @@ static ULONG flags_gss_to_isc_ret( ULONG flags )
     return ret;
 }
 
+static ULONG flags_gss_to_asc_ret(ULONG flags)
+{
+    ULONG ret = 0;
+    if (flags & GSS_C_DELEG_FLAG)    ret |= ASC_RET_DELEGATE;
+    if (flags & GSS_C_MUTUAL_FLAG)   ret |= ASC_RET_MUTUAL_AUTH;
+    if (flags & GSS_C_REPLAY_FLAG)   ret |= ASC_RET_REPLAY_DETECT;
+    if (flags & GSS_C_SEQUENCE_FLAG) ret |= ASC_RET_SEQUENCE_DETECT;
+    if (flags & GSS_C_CONF_FLAG)     ret |= ASC_RET_CONFIDENTIALITY;
+    if (flags & GSS_C_INTEG_FLAG)    ret |= ASC_RET_INTEGRITY;
+    if (flags & GSS_C_ANON_FLAG)     ret |= ASC_RET_NULL_SESSION;
+    return ret;
+}
+
 static int get_buffer_index( SecBufferDesc *desc, DWORD type )
 {
     UINT i;
@@ -470,12 +485,66 @@ static SECURITY_STATUS SEC_ENTRY kerberos_InitializeSecurityContextA(CredHandle
 /***********************************************************************
  *              AcceptSecurityContext
  */
-static SECURITY_STATUS SEC_ENTRY kerberos_AcceptSecurityContext(CredHandle *phCredential, CtxtHandle *phContext, SecBufferDesc *pInput,
-        ULONG fContextReq, ULONG TargetDataRep, CtxtHandle *phNewContext, SecBufferDesc *pOutput, ULONG *pfContextAttr, TimeStamp *ptsExpiry)
+static SECURITY_STATUS SEC_ENTRY kerberos_AcceptSecurityContext( CredHandle *phCredential, CtxtHandle *phContext,
+    SecBufferDesc *pInput, ULONG fContextReq, ULONG TargetDataRep, CtxtHandle *phNewContext, SecBufferDesc *pOutput,
+    ULONG *pfContextAttr, TimeStamp *ptsExpiry )
 {
-    FIXME("(%p %p %p %d %d %p %p %p %p)\n", phCredential, phContext, pInput, fContextReq, TargetDataRep, phNewContext, pOutput,
-          pfContextAttr, ptsExpiry);
+#ifdef SONAME_LIBGSSAPI_KRB5
+    OM_uint32 ret, minor_status, ret_flags, expiry_time;
+    gss_cred_id_t cred_handle;
+    gss_ctx_id_t ctxt_handle;
+    gss_buffer_desc input_token, output_token;
+    gss_name_t target = GSS_C_NO_NAME;
+    int idx;
+
+    TRACE( "(%p %p %p %u %u %p %p %p %p)\n", phCredential, phContext, pInput, fContextReq, TargetDataRep, phNewContext,
+           pOutput, pfContextAttr, ptsExpiry );
+    if (fContextReq) FIXME( "ignoring flags 0x%08x\n", fContextReq );
+
+    if (!phContext && !pInput && !phCredential) return SEC_E_INVALID_HANDLE;
+    cred_handle = credhandle_sspi_to_gss( phCredential, &target );
+    ctxt_handle = ctxthandle_sspi_to_gss( phContext );
+
+    if (!pInput) input_token.length = 0;
+    else
+    {
+        if ((idx = get_buffer_index( pInput, SECBUFFER_TOKEN )) == -1) return SEC_E_INVALID_TOKEN;
+        input_token.length = pInput->pBuffers[idx].cbBuffer;
+        input_token.value  = pInput->pBuffers[idx].pvBuffer;
+    }
+
+    if ((idx = get_buffer_index( pOutput, SECBUFFER_TOKEN )) == -1) return SEC_E_INVALID_TOKEN;
+    output_token.length = 0;
+    output_token.value  = NULL;
+
+    ret = pgss_accept_sec_context( &minor_status, &ctxt_handle, cred_handle, &input_token, GSS_C_NO_CHANNEL_BINDINGS,
+                                   &target, NULL, &output_token, &ret_flags, &expiry_time, NULL );
+    TRACE( "gss_accept_sec_context returned %08x minor status %08x ctxt_handle %p\n", ret, minor_status, ctxt_handle );
+    if (ret == GSS_S_COMPLETE || ret == GSS_S_CONTINUE_NEEDED)
+    {
+        if (output_token.length > pOutput->pBuffers[idx].cbBuffer) /* FIXME: check if larger buffer exists */
+        {
+            TRACE( "buffer too small %u > %u\n", output_token.length, pOutput->pBuffers[idx].cbBuffer );
+            pgss_release_buffer( &minor_status, &output_token );
+            pgss_delete_sec_context( &minor_status, &ctxt_handle, GSS_C_NO_BUFFER );
+            if (phContext) phContext->dwLower = 0;
+            return SEC_E_BUFFER_TOO_SMALL;
+        }
+        pOutput->pBuffers[idx].cbBuffer = output_token.length;
+        memcpy( pOutput->pBuffers[idx].pvBuffer, output_token.value, output_token.length );
+        pgss_release_buffer( &minor_status, &output_token );
+
+        ctxthandle_gss_to_sspi( ctxt_handle, phNewContext );
+        if (pfContextAttr) *pfContextAttr = flags_gss_to_asc_ret( ret_flags );
+        expirytime_gss_to_sspi( expiry_time, ptsExpiry );
+    }
+
+    return status_gss_to_sspi( ret );
+#else
+    FIXME( "(%p %p %p %u %u %p %p %p %p)\n", phCredential, phContext, pInput, fContextReq, TargetDataRep, phNewContext,
+           pOutput, pfContextAttr, ptsExpiry );
     return SEC_E_UNSUPPORTED_FUNCTION;
+#endif
 }
 
 /***********************************************************************
-- 
2.11.0