#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;
	UINT32 fContextReq;
	void* output_buffer = NULL;
	CtxtHandle context;
	ULONG pfContextAttr;
	SECURITY_STATUS status;
	CredHandle credentials = { 0 };
	TimeStamp expiration;
	PSecPkgInfo pPackageInfo;
	SEC_WINNT_AUTH_IDENTITY identity = { 0 };
	SecurityFunctionTable* table;
	PSecBuffer p_SecBuffer;
	SecBuffer output_SecBuffer;
	SecBufferDesc output_SecBuffer_desc;
	sspi_GlobalInit();
	table = InitSecurityInterface();
	status = 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);

	FreeContextBuffer(pPackageInfo);
	sspi_GlobalFinish();
	return rc;
}