secur32: Fix the wrapper InitializeSecurityContextA/W functions to handle phContext and phNewContext parameters being optional for some SSPs.
This commit is contained in:
parent
e24667cc6e
commit
ee47c5a23a
|
@ -247,14 +247,24 @@ SECURITY_STATUS WINAPI InitializeSecurityContextA(
|
|||
ULONG *pfContextAttr, PTimeStamp ptsExpiry)
|
||||
{
|
||||
SECURITY_STATUS ret;
|
||||
SecurePackage *package = NULL;
|
||||
PCredHandle cred = NULL;
|
||||
PCredHandle ctxt = NULL;
|
||||
|
||||
TRACE("%p %p %s %d %d %d %p %d %p %p %p %p\n", phCredential, phContext,
|
||||
debugstr_a(pszTargetName), fContextReq, Reserved1, TargetDataRep, pInput,
|
||||
Reserved1, phNewContext, pOutput, pfContextAttr, ptsExpiry);
|
||||
|
||||
if (phContext)
|
||||
{
|
||||
package = (SecurePackage *)phContext->dwUpper;
|
||||
ctxt = (PCtxtHandle)phContext->dwLower;
|
||||
}
|
||||
if (phCredential)
|
||||
{
|
||||
SecurePackage *package = (SecurePackage *)phCredential->dwUpper;
|
||||
PCredHandle cred = (PCredHandle)phCredential->dwLower;
|
||||
package = (SecurePackage *)phCredential->dwUpper;
|
||||
cred = (PCredHandle)phCredential->dwLower;
|
||||
}
|
||||
|
||||
if (package && package->provider)
|
||||
{
|
||||
|
@ -262,7 +272,7 @@ SECURITY_STATUS WINAPI InitializeSecurityContextA(
|
|||
{
|
||||
CtxtHandle myCtxt;
|
||||
|
||||
if(phContext)
|
||||
if (phContext)
|
||||
{
|
||||
PCtxtHandle realCtxt = (PCtxtHandle)phContext->dwLower;
|
||||
myCtxt.dwUpper = realCtxt->dwUpper;
|
||||
|
@ -270,16 +280,15 @@ SECURITY_STATUS WINAPI InitializeSecurityContextA(
|
|||
}
|
||||
|
||||
ret = package->provider->fnTableA.InitializeSecurityContextA(
|
||||
cred, phContext ? &myCtxt : NULL, pszTargetName, fContextReq,
|
||||
Reserved1, TargetDataRep, pInput, Reserved2, &myCtxt,
|
||||
cred, ctxt, pszTargetName, fContextReq,
|
||||
Reserved1, TargetDataRep, pInput, Reserved2, phNewContext ? &myCtxt : NULL,
|
||||
pOutput, pfContextAttr, ptsExpiry);
|
||||
if (ret == SEC_E_OK || ret == SEC_I_CONTINUE_NEEDED)
|
||||
if ((ret == SEC_E_OK || ret == SEC_I_CONTINUE_NEEDED) && phNewContext)
|
||||
{
|
||||
SECURITY_STATUS ret2;
|
||||
ret2 = SECUR32_makeSecHandle(phNewContext, package, &myCtxt);
|
||||
if (ret2 != SEC_E_OK)
|
||||
package->provider->fnTableW.DeleteSecurityContext(
|
||||
&myCtxt);
|
||||
package->provider->fnTableA.DeleteSecurityContext(&myCtxt);
|
||||
}
|
||||
}
|
||||
else
|
||||
|
@ -287,9 +296,6 @@ SECURITY_STATUS WINAPI InitializeSecurityContextA(
|
|||
}
|
||||
else
|
||||
ret = SEC_E_INVALID_HANDLE;
|
||||
}
|
||||
else
|
||||
ret = SEC_E_INVALID_HANDLE;
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
@ -304,22 +310,32 @@ SECURITY_STATUS WINAPI InitializeSecurityContextW(
|
|||
ULONG *pfContextAttr, PTimeStamp ptsExpiry)
|
||||
{
|
||||
SECURITY_STATUS ret;
|
||||
SecurePackage *package = NULL;
|
||||
PCredHandle cred = NULL;
|
||||
PCredHandle ctxt = NULL;
|
||||
|
||||
TRACE("%p %p %s %d %d %d %p %d %p %p %p %p\n", phCredential, phContext,
|
||||
debugstr_w(pszTargetName), fContextReq, Reserved1, TargetDataRep, pInput,
|
||||
Reserved1, phNewContext, pOutput, pfContextAttr, ptsExpiry);
|
||||
|
||||
if (phContext)
|
||||
{
|
||||
package = (SecurePackage *)phContext->dwUpper;
|
||||
ctxt = (PCtxtHandle)phContext->dwLower;
|
||||
}
|
||||
if (phCredential)
|
||||
{
|
||||
SecurePackage *package = (SecurePackage *)phCredential->dwUpper;
|
||||
PCredHandle cred = (PCredHandle)phCredential->dwLower;
|
||||
package = (SecurePackage *)phCredential->dwUpper;
|
||||
cred = (PCredHandle)phCredential->dwLower;
|
||||
}
|
||||
|
||||
if (package && package->provider)
|
||||
{
|
||||
if (package->provider->fnTableW.QueryCredentialsAttributesW)
|
||||
if (package->provider->fnTableW.InitializeSecurityContextW)
|
||||
{
|
||||
CtxtHandle myCtxt;
|
||||
|
||||
if(phContext)
|
||||
if (phContext)
|
||||
{
|
||||
PCtxtHandle realCtxt = (PCtxtHandle)phContext->dwLower;
|
||||
myCtxt.dwUpper = realCtxt->dwUpper;
|
||||
|
@ -327,16 +343,15 @@ SECURITY_STATUS WINAPI InitializeSecurityContextW(
|
|||
}
|
||||
|
||||
ret = package->provider->fnTableW.InitializeSecurityContextW(
|
||||
cred, phContext ? &myCtxt : NULL, pszTargetName, fContextReq,
|
||||
Reserved1, TargetDataRep, pInput, Reserved2, &myCtxt,
|
||||
cred, ctxt, pszTargetName, fContextReq,
|
||||
Reserved1, TargetDataRep, pInput, Reserved2, phNewContext ? &myCtxt : NULL,
|
||||
pOutput, pfContextAttr, ptsExpiry);
|
||||
if (ret == SEC_E_OK || ret == SEC_I_CONTINUE_NEEDED)
|
||||
if ((ret == SEC_E_OK || ret == SEC_I_CONTINUE_NEEDED) && phNewContext)
|
||||
{
|
||||
SECURITY_STATUS ret2;
|
||||
ret2 = SECUR32_makeSecHandle(phNewContext, package, &myCtxt);
|
||||
if (ret2 != SEC_E_OK)
|
||||
package->provider->fnTableW.DeleteSecurityContext(
|
||||
&myCtxt);
|
||||
package->provider->fnTableW.DeleteSecurityContext(&myCtxt);
|
||||
}
|
||||
}
|
||||
else
|
||||
|
@ -344,9 +359,6 @@ SECURITY_STATUS WINAPI InitializeSecurityContextW(
|
|||
}
|
||||
else
|
||||
ret = SEC_E_INVALID_HANDLE;
|
||||
}
|
||||
else
|
||||
ret = SEC_E_INVALID_HANDLE;
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue