TestInitializeSecurityContext.c

 
#include <stdio.h>
#include <winpr/crt.h>
#include <winpr/sspi.h>
#include <winpr/winpr.h>
 
static const char* test_User = "User";
static const char* test_Domain = "Domain";
static const char* test_Password = "Password";
 
int TestInitializeSecurityContext(int argc, char* argv[])
{
  int rc = -1;
  UINT32 cbMaxLen = 0;
  UINT32 fContextReq = 0;
  void* output_buffer = NULL;
  CtxtHandle context;
  ULONG pfContextAttr = 0;
  SECURITY_STATUS status = 0;
  CredHandle credentials = { 0 };
  TimeStamp expiration;
  PSecPkgInfo pPackageInfo = NULL;
  SEC_WINNT_AUTH_IDENTITY identity = { 0 };
  SecurityFunctionTable* table = NULL;
  PSecBuffer p_SecBuffer = NULL;
  SecBuffer output_SecBuffer;
  SecBufferDesc output_SecBuffer_desc;
 
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
  sspi_GlobalInit();
  table = InitSecurityInterfaceEx(0);
  status = table->QuerySecurityPackageInfo(NTLM_SSP_NAME, &pPackageInfo);
 
  if (status != SEC_E_OK)
  {
    printf("QuerySecurityPackageInfo status: 0x%08" PRIX32 "\n", status);
    goto fail;
  }
 
  cbMaxLen = pPackageInfo->cbMaxToken;
  identity.User = (UINT16*)_strdup(test_User);
  identity.Domain = (UINT16*)_strdup(test_Domain);
  identity.Password = (UINT16*)_strdup(test_Password);
 
  if (!identity.User || !identity.Domain || !identity.Password)
    goto fail;
 
  identity.UserLength = strlen(test_User);
  identity.DomainLength = strlen(test_Domain);
  identity.PasswordLength = strlen(test_Password);
  identity.Flags = SEC_WINNT_AUTH_IDENTITY_ANSI;
  status = table->AcquireCredentialsHandle(NULL, NTLM_SSP_NAME, SECPKG_CRED_OUTBOUND, NULL,
                                           &identity, NULL, NULL, &credentials, &expiration);
 
  if (status != SEC_E_OK)
  {
    printf("AcquireCredentialsHandle status: 0x%08" PRIX32 "\n", status);
    goto fail;
  }
 
  fContextReq = ISC_REQ_REPLAY_DETECT | ISC_REQ_SEQUENCE_DETECT | ISC_REQ_CONFIDENTIALITY |
                ISC_REQ_DELEGATE;
  output_buffer = malloc(cbMaxLen);
 
  if (!output_buffer)
  {
    printf("Memory allocation failed\n");
    goto fail;
  }
 
  output_SecBuffer_desc.ulVersion = 0;
  output_SecBuffer_desc.cBuffers = 1;
  output_SecBuffer_desc.pBuffers = &output_SecBuffer;
  output_SecBuffer.cbBuffer = cbMaxLen;
  output_SecBuffer.BufferType = SECBUFFER_TOKEN;
  output_SecBuffer.pvBuffer = output_buffer;
  status = table->InitializeSecurityContext(&credentials, NULL, NULL, fContextReq, 0, 0, NULL, 0,
                                            &context, &output_SecBuffer_desc, &pfContextAttr,
                                            &expiration);
 
  if (status != SEC_I_CONTINUE_NEEDED)
  {
    printf("InitializeSecurityContext status: 0x%08" PRIX32 "\n", status);
    goto fail;
  }
 
  printf("cBuffers: %" PRIu32 " ulVersion: %" PRIu32 "\n", output_SecBuffer_desc.cBuffers,
         output_SecBuffer_desc.ulVersion);
  p_SecBuffer = &output_SecBuffer_desc.pBuffers[0];
  printf("BufferType: 0x%08" PRIX32 " cbBuffer: %" PRIu32 "\n", p_SecBuffer->BufferType,
         p_SecBuffer->cbBuffer);
  status = table->DeleteSecurityContext(&context);
 
  if (status != SEC_E_OK)
  {
    printf("DeleteSecurityContext status: 0x%08" PRIX32 "\n", status);
    goto fail;
  }
 
  rc = 0;
fail:
  free(identity.User);
  free(identity.Domain);
  free(identity.Password);
  free(output_buffer);
 
  if (SecIsValidHandle(&credentials))
    table->FreeCredentialsHandle(&credentials);
 
  table->FreeContextBuffer(pPackageInfo);
  sspi_GlobalFinish();
  return rc;
}