793 lines
26 KiB
C
793 lines
26 KiB
C
|
|
#include <winpr/crt.h>
|
|
#include <winpr/assert.h>
|
|
#include <winpr/sspi.h>
|
|
#include <winpr/print.h>
|
|
#include <winpr/wlog.h>
|
|
|
|
struct test_input_t
|
|
{
|
|
const char* user;
|
|
const char* domain;
|
|
const char* pwd;
|
|
const BYTE* ntlm;
|
|
const BYTE* ntlmv2;
|
|
BOOL dynamic;
|
|
BOOL expected;
|
|
};
|
|
|
|
typedef struct
|
|
{
|
|
CtxtHandle context;
|
|
ULONG cbMaxToken;
|
|
ULONG fContextReq;
|
|
ULONG pfContextAttr;
|
|
TimeStamp expiration;
|
|
PSecBuffer pBuffer;
|
|
SecBuffer inputBuffer[2];
|
|
SecBuffer outputBuffer[2];
|
|
BOOL haveContext;
|
|
BOOL haveInputBuffer;
|
|
BOOL UseNtlmV2Hash;
|
|
LPTSTR ServicePrincipalName;
|
|
SecBufferDesc inputBufferDesc;
|
|
SecBufferDesc outputBufferDesc;
|
|
CredHandle credentials;
|
|
BOOL confidentiality;
|
|
SecPkgInfo* pPackageInfo;
|
|
SecurityFunctionTable* table;
|
|
SEC_WINNT_AUTH_IDENTITY identity;
|
|
} TEST_NTLM_SERVER;
|
|
|
|
static BYTE TEST_NTLM_TIMESTAMP[8] = { 0x33, 0x57, 0xbd, 0xb1, 0x07, 0x8b, 0xcf, 0x01 };
|
|
|
|
static BYTE TEST_NTLM_CLIENT_CHALLENGE[8] = { 0x20, 0xc0, 0x2b, 0x3d, 0xc0, 0x61, 0xa7, 0x73 };
|
|
|
|
static BYTE TEST_NTLM_SERVER_CHALLENGE[8] = { 0xa4, 0xf1, 0xba, 0xa6, 0x7c, 0xdc, 0x1a, 0x12 };
|
|
|
|
static BYTE TEST_NTLM_NEGOTIATE[] =
|
|
"\x4e\x54\x4c\x4d\x53\x53\x50\x00\x01\x00\x00\x00\x07\x82\x08\xa2"
|
|
"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
|
|
"\x06\x03\x80\x25\x00\x00\x00\x0f";
|
|
|
|
static BYTE TEST_NTLM_CHALLENGE[] =
|
|
"\x4e\x54\x4c\x4d\x53\x53\x50\x00\x02\x00\x00\x00\x00\x00\x00\x00"
|
|
"\x38\x00\x00\x00\x07\x82\x88\xa2\xa4\xf1\xba\xa6\x7c\xdc\x1a\x12"
|
|
"\x00\x00\x00\x00\x00\x00\x00\x00\x66\x00\x66\x00\x38\x00\x00\x00"
|
|
"\x06\x03\x80\x25\x00\x00\x00\x0f\x02\x00\x0e\x00\x4e\x00\x45\x00"
|
|
"\x57\x00\x59\x00\x45\x00\x41\x00\x52\x00\x01\x00\x0e\x00\x4e\x00"
|
|
"\x45\x00\x57\x00\x59\x00\x45\x00\x41\x00\x52\x00\x04\x00\x1c\x00"
|
|
"\x6c\x00\x61\x00\x62\x00\x2e\x00\x77\x00\x61\x00\x79\x00\x6b\x00"
|
|
"\x2e\x00\x6c\x00\x6f\x00\x63\x00\x61\x00\x6c\x00\x03\x00\x0e\x00"
|
|
"\x6e\x00\x65\x00\x77\x00\x79\x00\x65\x00\x61\x00\x72\x00\x07\x00"
|
|
"\x08\x00\x33\x57\xbd\xb1\x07\x8b\xcf\x01\x00\x00\x00\x00";
|
|
|
|
static BYTE TEST_NTLM_AUTHENTICATE[] =
|
|
"\x4e\x54\x4c\x4d\x53\x53\x50\x00\x03\x00\x00\x00\x18\x00\x18\x00"
|
|
"\x82\x00\x00\x00\x08\x01\x08\x01\x9a\x00\x00\x00\x0c\x00\x0c\x00"
|
|
"\x58\x00\x00\x00\x10\x00\x10\x00\x64\x00\x00\x00\x0e\x00\x0e\x00"
|
|
"\x74\x00\x00\x00\x00\x00\x00\x00\xa2\x01\x00\x00\x05\x82\x88\xa2"
|
|
"\x06\x03\x80\x25\x00\x00\x00\x0f\x12\xe5\x5a\xf5\x80\xee\x3f\x29"
|
|
"\xe1\xde\x90\x4d\x73\x77\x06\x25\x44\x00\x6f\x00\x6d\x00\x61\x00"
|
|
"\x69\x00\x6e\x00\x55\x00\x73\x00\x65\x00\x72\x00\x6e\x00\x61\x00"
|
|
"\x6d\x00\x65\x00\x4e\x00\x45\x00\x57\x00\x59\x00\x45\x00\x41\x00"
|
|
"\x52\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
|
|
"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x62\x14\x68\xc8\x98\x12"
|
|
"\xe7\x39\xd8\x76\x1b\xe9\xf7\x54\xb5\xe3\x01\x01\x00\x00\x00\x00"
|
|
"\x00\x00\x33\x57\xbd\xb1\x07\x8b\xcf\x01\x20\xc0\x2b\x3d\xc0\x61"
|
|
"\xa7\x73\x00\x00\x00\x00\x02\x00\x0e\x00\x4e\x00\x45\x00\x57\x00"
|
|
"\x59\x00\x45\x00\x41\x00\x52\x00\x01\x00\x0e\x00\x4e\x00\x45\x00"
|
|
"\x57\x00\x59\x00\x45\x00\x41\x00\x52\x00\x04\x00\x1c\x00\x6c\x00"
|
|
"\x61\x00\x62\x00\x2e\x00\x77\x00\x61\x00\x79\x00\x6b\x00\x2e\x00"
|
|
"\x6c\x00\x6f\x00\x63\x00\x61\x00\x6c\x00\x03\x00\x0e\x00\x6e\x00"
|
|
"\x65\x00\x77\x00\x79\x00\x65\x00\x61\x00\x72\x00\x07\x00\x08\x00"
|
|
"\x33\x57\xbd\xb1\x07\x8b\xcf\x01\x06\x00\x04\x00\x02\x00\x00\x00"
|
|
"\x08\x00\x30\x00\x30\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00"
|
|
"\x00\x20\x00\x00\x1e\x10\xf5\x2c\x54\x2f\x2e\x77\x1c\x13\xbf\xc3"
|
|
"\x3f\xe1\x7b\x28\x7e\x0b\x93\x5a\x39\xd2\xce\x12\xd7\xbd\x8c\x4e"
|
|
"\x2b\xb5\x0b\xf5\x0a\x00\x10\x00\x00\x00\x00\x00\x00\x00\x00\x00"
|
|
"\x00\x00\x00\x00\x00\x00\x00\x00\x09\x00\x1a\x00\x48\x00\x54\x00"
|
|
"\x54\x00\x50\x00\x2f\x00\x72\x00\x77\x00\x2e\x00\x6c\x00\x6f\x00"
|
|
"\x63\x00\x61\x00\x6c\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
|
|
"\x00\x00";
|
|
|
|
#define TEST_SSPI_INTERFACE SSPI_INTERFACE_WINPR
|
|
|
|
static const char* TEST_NTLM_USER = "Username";
|
|
static const char* TEST_NTLM_DOMAIN = "Domain";
|
|
static const char* TEST_NTLM_PASSWORD = "P4ss123!";
|
|
|
|
// static const char* TEST_NTLM_HASH_STRING = "d5922a65c4d5c082ca444af1be0001db";
|
|
|
|
static const BYTE TEST_NTLM_HASH[16] = { 0xd5, 0x92, 0x2a, 0x65, 0xc4, 0xd5, 0xc0, 0x82,
|
|
0xca, 0x44, 0x4a, 0xf1, 0xbe, 0x00, 0x01, 0xdb };
|
|
|
|
// static const char* TEST_NTLM_HASH_V2_STRING = "4c7f706f7dde05a9d1a0f4e7ffe3bfb8";
|
|
|
|
static const BYTE TEST_NTLM_V2_HASH[16] = { 0x4c, 0x7f, 0x70, 0x6f, 0x7d, 0xde, 0x05, 0xa9,
|
|
0xd1, 0xa0, 0xf4, 0xe7, 0xff, 0xe3, 0xbf, 0xb8 };
|
|
|
|
static const BYTE TEST_EMPTY_PWD_NTLM_HASH[] = { 0x31, 0xd6, 0xcf, 0xe0, 0xd1, 0x6a, 0xe9, 0x31,
|
|
0xb7, 0x3c, 0x59, 0xd7, 0xe0, 0xc0, 0x89, 0xc0 };
|
|
|
|
static const BYTE TEST_EMPTY_PWD_NTLM_V2_HASH[] = {
|
|
0x0b, 0xce, 0x54, 0x87, 0x4e, 0x94, 0x20, 0x9e, 0x34, 0x48, 0x97, 0xc1, 0x60, 0x03, 0x6e, 0x3b
|
|
};
|
|
|
|
#define NTLM_PACKAGE_NAME NTLM_SSP_NAME
|
|
|
|
typedef struct
|
|
{
|
|
CtxtHandle context;
|
|
ULONG cbMaxToken;
|
|
ULONG fContextReq;
|
|
ULONG pfContextAttr;
|
|
TimeStamp expiration;
|
|
PSecBuffer pBuffer;
|
|
SecBuffer inputBuffer[2];
|
|
SecBuffer outputBuffer[2];
|
|
BOOL haveContext;
|
|
BOOL haveInputBuffer;
|
|
LPTSTR ServicePrincipalName;
|
|
SecBufferDesc inputBufferDesc;
|
|
SecBufferDesc outputBufferDesc;
|
|
CredHandle credentials;
|
|
BOOL confidentiality;
|
|
SecPkgInfo* pPackageInfo;
|
|
SecurityFunctionTable* table;
|
|
SEC_WINNT_AUTH_IDENTITY identity;
|
|
} TEST_NTLM_CLIENT;
|
|
|
|
static int test_ntlm_client_init(TEST_NTLM_CLIENT* ntlm, const char* user, const char* domain,
|
|
const char* password)
|
|
{
|
|
SECURITY_STATUS status = SEC_E_INTERNAL_ERROR;
|
|
|
|
WINPR_ASSERT(ntlm);
|
|
|
|
SecInvalidateHandle(&(ntlm->context));
|
|
ntlm->table = InitSecurityInterfaceEx(TEST_SSPI_INTERFACE);
|
|
if (!ntlm->table)
|
|
return -1;
|
|
const int rc = sspi_SetAuthIdentity(&(ntlm->identity), user, domain, password);
|
|
if (rc < 0)
|
|
return rc;
|
|
|
|
status = ntlm->table->QuerySecurityPackageInfo(NTLM_PACKAGE_NAME, &ntlm->pPackageInfo);
|
|
|
|
if (status != SEC_E_OK)
|
|
{
|
|
(void)fprintf(stderr, "QuerySecurityPackageInfo status: %s (0x%08" PRIX32 ")\n",
|
|
GetSecurityStatusString(status), status);
|
|
return -1;
|
|
}
|
|
|
|
ntlm->cbMaxToken = ntlm->pPackageInfo->cbMaxToken;
|
|
status = ntlm->table->AcquireCredentialsHandle(nullptr, NTLM_PACKAGE_NAME, SECPKG_CRED_OUTBOUND,
|
|
nullptr, &ntlm->identity, nullptr, nullptr,
|
|
&ntlm->credentials, &ntlm->expiration);
|
|
|
|
if (status != SEC_E_OK)
|
|
{
|
|
(void)fprintf(stderr, "AcquireCredentialsHandle status: %s (0x%08" PRIX32 ")\n",
|
|
GetSecurityStatusString(status), status);
|
|
return -1;
|
|
}
|
|
|
|
ntlm->haveContext = FALSE;
|
|
ntlm->haveInputBuffer = FALSE;
|
|
ZeroMemory(&ntlm->inputBuffer, sizeof(SecBuffer));
|
|
ZeroMemory(&ntlm->outputBuffer, sizeof(SecBuffer));
|
|
ntlm->fContextReq = 0;
|
|
|
|
/* NLA authentication flags */
|
|
ntlm->fContextReq |= ISC_REQ_MUTUAL_AUTH;
|
|
ntlm->fContextReq |= ISC_REQ_CONFIDENTIALITY;
|
|
ntlm->fContextReq |= ISC_REQ_USE_SESSION_KEY;
|
|
return 1;
|
|
}
|
|
|
|
static void test_ntlm_client_uninit(TEST_NTLM_CLIENT* ntlm)
|
|
{
|
|
if (!ntlm)
|
|
return;
|
|
|
|
if (ntlm->outputBuffer[0].pvBuffer)
|
|
{
|
|
free(ntlm->outputBuffer[0].pvBuffer);
|
|
ntlm->outputBuffer[0].pvBuffer = nullptr;
|
|
}
|
|
|
|
free(ntlm->identity.User);
|
|
free(ntlm->identity.Domain);
|
|
free(ntlm->identity.Password);
|
|
free(ntlm->ServicePrincipalName);
|
|
|
|
if (ntlm->table)
|
|
{
|
|
SECURITY_STATUS status = ntlm->table->FreeCredentialsHandle(&ntlm->credentials);
|
|
WINPR_ASSERT((status == SEC_E_OK) || (status == SEC_E_SECPKG_NOT_FOUND) ||
|
|
(status == SEC_E_UNSUPPORTED_FUNCTION));
|
|
|
|
status = ntlm->table->FreeContextBuffer(ntlm->pPackageInfo);
|
|
WINPR_ASSERT((status == SEC_E_OK) || (status = SEC_E_INVALID_HANDLE));
|
|
|
|
status = ntlm->table->DeleteSecurityContext(&ntlm->context);
|
|
WINPR_ASSERT((status == SEC_E_OK) || (status == SEC_E_SECPKG_NOT_FOUND) ||
|
|
(status == SEC_E_UNSUPPORTED_FUNCTION));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* SSPI Client Ceremony
|
|
*
|
|
* --------------
|
|
* ( Client Begin )
|
|
* --------------
|
|
* |
|
|
* |
|
|
* \|/
|
|
* -----------+--------------
|
|
* | AcquireCredentialsHandle |
|
|
* --------------------------
|
|
* |
|
|
* |
|
|
* \|/
|
|
* -------------+--------------
|
|
* +---------------> / InitializeSecurityContext /
|
|
* | ----------------------------
|
|
* | |
|
|
* | |
|
|
* | \|/
|
|
* --------------------------- ---------+------------- ----------------------
|
|
* / Receive blob from server / < Received security blob? > --Yes-> / Send blob to server /
|
|
* -------------+------------- ----------------------- ----------------------
|
|
* /|\ | |
|
|
* | No |
|
|
* Yes \|/ |
|
|
* | ------------+----------- |
|
|
* +---------------- < Received Continue Needed > <-----------------+
|
|
* ------------------------
|
|
* |
|
|
* No
|
|
* \|/
|
|
* ------+-------
|
|
* ( Client End )
|
|
* --------------
|
|
*/
|
|
static BOOL IsSecurityStatusError(SECURITY_STATUS status)
|
|
{
|
|
BOOL error = TRUE;
|
|
|
|
switch (status)
|
|
{
|
|
case SEC_E_OK:
|
|
case SEC_I_CONTINUE_NEEDED:
|
|
case SEC_I_COMPLETE_NEEDED:
|
|
case SEC_I_COMPLETE_AND_CONTINUE:
|
|
case SEC_I_LOCAL_LOGON:
|
|
case SEC_I_CONTEXT_EXPIRED:
|
|
case SEC_I_INCOMPLETE_CREDENTIALS:
|
|
case SEC_I_RENEGOTIATE:
|
|
case SEC_I_NO_LSA_CONTEXT:
|
|
case SEC_I_SIGNATURE_NEEDED:
|
|
case SEC_I_NO_RENEGOTIATION:
|
|
error = FALSE;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return error;
|
|
}
|
|
|
|
static int test_ntlm_client_authenticate(TEST_NTLM_CLIENT* ntlm)
|
|
{
|
|
SECURITY_STATUS status = SEC_E_INTERNAL_ERROR;
|
|
|
|
WINPR_ASSERT(ntlm);
|
|
if (ntlm->outputBuffer[0].pvBuffer)
|
|
{
|
|
free(ntlm->outputBuffer[0].pvBuffer);
|
|
ntlm->outputBuffer[0].pvBuffer = nullptr;
|
|
}
|
|
|
|
ntlm->outputBufferDesc.ulVersion = SECBUFFER_VERSION;
|
|
ntlm->outputBufferDesc.cBuffers = 1;
|
|
ntlm->outputBufferDesc.pBuffers = ntlm->outputBuffer;
|
|
ntlm->outputBuffer[0].BufferType = SECBUFFER_TOKEN;
|
|
ntlm->outputBuffer[0].cbBuffer = ntlm->cbMaxToken;
|
|
ntlm->outputBuffer[0].pvBuffer = malloc(ntlm->outputBuffer[0].cbBuffer);
|
|
|
|
if (!ntlm->outputBuffer[0].pvBuffer)
|
|
return -1;
|
|
|
|
if (ntlm->haveInputBuffer)
|
|
{
|
|
ntlm->inputBufferDesc.ulVersion = SECBUFFER_VERSION;
|
|
ntlm->inputBufferDesc.cBuffers = 1;
|
|
ntlm->inputBufferDesc.pBuffers = ntlm->inputBuffer;
|
|
ntlm->inputBuffer[0].BufferType = SECBUFFER_TOKEN;
|
|
}
|
|
|
|
if ((!ntlm) || (!ntlm->table))
|
|
{
|
|
(void)fprintf(stderr, "ntlm_authenticate: invalid ntlm context\n");
|
|
return -1;
|
|
}
|
|
|
|
status = ntlm->table->InitializeSecurityContext(
|
|
&ntlm->credentials, (ntlm->haveContext) ? &ntlm->context : nullptr,
|
|
(ntlm->ServicePrincipalName) ? ntlm->ServicePrincipalName : nullptr, ntlm->fContextReq, 0,
|
|
SECURITY_NATIVE_DREP, (ntlm->haveInputBuffer) ? &ntlm->inputBufferDesc : nullptr, 0,
|
|
&ntlm->context, &ntlm->outputBufferDesc, &ntlm->pfContextAttr, &ntlm->expiration);
|
|
|
|
if (IsSecurityStatusError(status))
|
|
return -1;
|
|
|
|
if ((status == SEC_I_COMPLETE_AND_CONTINUE) || (status == SEC_I_COMPLETE_NEEDED))
|
|
{
|
|
if (ntlm->table->CompleteAuthToken)
|
|
{
|
|
SECURITY_STATUS rc =
|
|
ntlm->table->CompleteAuthToken(&ntlm->context, &ntlm->outputBufferDesc);
|
|
if (rc != SEC_E_OK)
|
|
return -1;
|
|
}
|
|
|
|
if (status == SEC_I_COMPLETE_NEEDED)
|
|
status = SEC_E_OK;
|
|
else if (status == SEC_I_COMPLETE_AND_CONTINUE)
|
|
status = SEC_I_CONTINUE_NEEDED;
|
|
}
|
|
|
|
if (ntlm->haveInputBuffer)
|
|
{
|
|
free(ntlm->inputBuffer[0].pvBuffer);
|
|
}
|
|
|
|
ntlm->haveInputBuffer = TRUE;
|
|
ntlm->haveContext = TRUE;
|
|
return (status == SEC_I_CONTINUE_NEEDED) ? 1 : 0;
|
|
}
|
|
|
|
static TEST_NTLM_CLIENT* test_ntlm_client_new(void)
|
|
{
|
|
TEST_NTLM_CLIENT* ntlm = (TEST_NTLM_CLIENT*)calloc(1, sizeof(TEST_NTLM_CLIENT));
|
|
|
|
if (!ntlm)
|
|
return nullptr;
|
|
|
|
return ntlm;
|
|
}
|
|
|
|
static void test_ntlm_client_free(TEST_NTLM_CLIENT* ntlm)
|
|
{
|
|
if (!ntlm)
|
|
return;
|
|
|
|
test_ntlm_client_uninit(ntlm);
|
|
free(ntlm);
|
|
}
|
|
|
|
static int test_ntlm_server_init(TEST_NTLM_SERVER* ntlm)
|
|
{
|
|
SECURITY_STATUS status = SEC_E_INTERNAL_ERROR;
|
|
|
|
WINPR_ASSERT(ntlm);
|
|
|
|
ntlm->UseNtlmV2Hash = TRUE;
|
|
SecInvalidateHandle(&(ntlm->context));
|
|
ntlm->table = InitSecurityInterfaceEx(TEST_SSPI_INTERFACE);
|
|
if (!ntlm->table)
|
|
return SEC_E_INTERNAL_ERROR;
|
|
|
|
status = ntlm->table->QuerySecurityPackageInfo(NTLM_PACKAGE_NAME, &ntlm->pPackageInfo);
|
|
|
|
if (status != SEC_E_OK)
|
|
{
|
|
(void)fprintf(stderr, "QuerySecurityPackageInfo status: %s (0x%08" PRIX32 ")\n",
|
|
GetSecurityStatusString(status), status);
|
|
return -1;
|
|
}
|
|
|
|
ntlm->cbMaxToken = ntlm->pPackageInfo->cbMaxToken;
|
|
status = ntlm->table->AcquireCredentialsHandle(nullptr, NTLM_PACKAGE_NAME, SECPKG_CRED_INBOUND,
|
|
nullptr, nullptr, nullptr, nullptr,
|
|
&ntlm->credentials, &ntlm->expiration);
|
|
|
|
if (status != SEC_E_OK)
|
|
{
|
|
(void)fprintf(stderr, "AcquireCredentialsHandle status: %s (0x%08" PRIX32 ")\n",
|
|
GetSecurityStatusString(status), status);
|
|
return -1;
|
|
}
|
|
|
|
ntlm->haveContext = FALSE;
|
|
ntlm->haveInputBuffer = FALSE;
|
|
ZeroMemory(&ntlm->inputBuffer, sizeof(SecBuffer));
|
|
ZeroMemory(&ntlm->outputBuffer, sizeof(SecBuffer));
|
|
ntlm->fContextReq = 0;
|
|
/* NLA authentication flags */
|
|
ntlm->fContextReq |= ASC_REQ_MUTUAL_AUTH;
|
|
ntlm->fContextReq |= ASC_REQ_CONFIDENTIALITY;
|
|
ntlm->fContextReq |= ASC_REQ_CONNECTION;
|
|
ntlm->fContextReq |= ASC_REQ_USE_SESSION_KEY;
|
|
ntlm->fContextReq |= ASC_REQ_REPLAY_DETECT;
|
|
ntlm->fContextReq |= ASC_REQ_SEQUENCE_DETECT;
|
|
ntlm->fContextReq |= ASC_REQ_EXTENDED_ERROR;
|
|
return 1;
|
|
}
|
|
|
|
static void test_ntlm_server_uninit(TEST_NTLM_SERVER* ntlm)
|
|
{
|
|
if (!ntlm)
|
|
return;
|
|
|
|
if (ntlm->outputBuffer[0].pvBuffer)
|
|
{
|
|
free(ntlm->outputBuffer[0].pvBuffer);
|
|
ntlm->outputBuffer[0].pvBuffer = nullptr;
|
|
}
|
|
|
|
free(ntlm->identity.User);
|
|
free(ntlm->identity.Domain);
|
|
free(ntlm->identity.Password);
|
|
free(ntlm->ServicePrincipalName);
|
|
|
|
if (ntlm->table)
|
|
{
|
|
SECURITY_STATUS status = ntlm->table->FreeCredentialsHandle(&ntlm->credentials);
|
|
WINPR_ASSERT(status == SEC_E_OK);
|
|
status = ntlm->table->FreeContextBuffer(ntlm->pPackageInfo);
|
|
WINPR_ASSERT(status == SEC_E_OK);
|
|
status = ntlm->table->DeleteSecurityContext(&ntlm->context);
|
|
WINPR_ASSERT(status == SEC_E_OK);
|
|
}
|
|
}
|
|
|
|
static int test_ntlm_server_authenticate(const struct test_input_t* targ, TEST_NTLM_SERVER* ntlm)
|
|
{
|
|
SECURITY_STATUS status = SEC_E_INTERNAL_ERROR;
|
|
|
|
WINPR_ASSERT(ntlm);
|
|
WINPR_ASSERT(targ);
|
|
|
|
ntlm->inputBufferDesc.ulVersion = SECBUFFER_VERSION;
|
|
ntlm->inputBufferDesc.cBuffers = 1;
|
|
ntlm->inputBufferDesc.pBuffers = ntlm->inputBuffer;
|
|
ntlm->inputBuffer[0].BufferType = SECBUFFER_TOKEN;
|
|
ntlm->outputBufferDesc.ulVersion = SECBUFFER_VERSION;
|
|
ntlm->outputBufferDesc.cBuffers = 1;
|
|
ntlm->outputBufferDesc.pBuffers = &ntlm->outputBuffer[0];
|
|
ntlm->outputBuffer[0].BufferType = SECBUFFER_TOKEN;
|
|
ntlm->outputBuffer[0].cbBuffer = ntlm->cbMaxToken;
|
|
ntlm->outputBuffer[0].pvBuffer = malloc(ntlm->outputBuffer[0].cbBuffer);
|
|
|
|
if (!ntlm->outputBuffer[0].pvBuffer)
|
|
return -1;
|
|
|
|
status = ntlm->table->AcceptSecurityContext(
|
|
&ntlm->credentials, ntlm->haveContext ? &ntlm->context : nullptr, &ntlm->inputBufferDesc,
|
|
ntlm->fContextReq, SECURITY_NATIVE_DREP, &ntlm->context, &ntlm->outputBufferDesc,
|
|
&ntlm->pfContextAttr, &ntlm->expiration);
|
|
|
|
if (status == SEC_I_CONTINUE_NEEDED)
|
|
{
|
|
SecPkgContext_AuthNtlmHash AuthNtlmHash = WINPR_C_ARRAY_INIT;
|
|
|
|
if (ntlm->UseNtlmV2Hash)
|
|
{
|
|
AuthNtlmHash.Version = 2;
|
|
CopyMemory(AuthNtlmHash.NtlmHash, targ->ntlmv2, 16);
|
|
}
|
|
else
|
|
{
|
|
AuthNtlmHash.Version = 1;
|
|
CopyMemory(AuthNtlmHash.NtlmHash, targ->ntlm, 16);
|
|
}
|
|
|
|
status =
|
|
ntlm->table->SetContextAttributes(&ntlm->context, SECPKG_ATTR_AUTH_NTLM_HASH,
|
|
&AuthNtlmHash, sizeof(SecPkgContext_AuthNtlmHash));
|
|
}
|
|
|
|
if ((status != SEC_E_OK) && (status != SEC_I_CONTINUE_NEEDED))
|
|
{
|
|
(void)fprintf(stderr, "AcceptSecurityContext status: %s (0x%08" PRIX32 ")\n",
|
|
GetSecurityStatusString(status), status);
|
|
return -1; /* Access Denied */
|
|
}
|
|
|
|
ntlm->haveContext = TRUE;
|
|
return (status == SEC_I_CONTINUE_NEEDED) ? 1 : 0;
|
|
}
|
|
|
|
static TEST_NTLM_SERVER* test_ntlm_server_new(void)
|
|
{
|
|
TEST_NTLM_SERVER* ntlm = (TEST_NTLM_SERVER*)calloc(1, sizeof(TEST_NTLM_SERVER));
|
|
|
|
if (!ntlm)
|
|
return nullptr;
|
|
|
|
return ntlm;
|
|
}
|
|
|
|
static void test_ntlm_server_free(TEST_NTLM_SERVER* ntlm)
|
|
{
|
|
if (!ntlm)
|
|
return;
|
|
|
|
test_ntlm_server_uninit(ntlm);
|
|
free(ntlm);
|
|
}
|
|
|
|
static BOOL test_default(const struct test_input_t* arg)
|
|
{
|
|
BOOL rc = FALSE;
|
|
PSecBuffer pSecBuffer = nullptr;
|
|
|
|
WINPR_ASSERT(arg);
|
|
|
|
printf("testcase {user=%s, domain=%s, password=%s, dynamic=%s}\n", arg->user, arg->domain,
|
|
arg->pwd, arg->dynamic ? "TRUE" : "FALSE");
|
|
|
|
/**
|
|
* Client Initialization
|
|
*/
|
|
TEST_NTLM_CLIENT* client = test_ntlm_client_new();
|
|
TEST_NTLM_SERVER* server = test_ntlm_server_new();
|
|
|
|
if (!client || !server)
|
|
{
|
|
printf("Memory allocation failed\n");
|
|
goto fail;
|
|
}
|
|
|
|
int status = test_ntlm_client_init(client, arg->user, arg->domain, arg->pwd);
|
|
|
|
if (status < 0)
|
|
{
|
|
printf("test_ntlm_client_init failure\n");
|
|
goto fail;
|
|
}
|
|
|
|
/**
|
|
* Server Initialization
|
|
*/
|
|
|
|
status = test_ntlm_server_init(server);
|
|
|
|
if (status < 0)
|
|
{
|
|
printf("test_ntlm_server_init failure\n");
|
|
goto fail;
|
|
}
|
|
|
|
/**
|
|
* Client -> Negotiate Message
|
|
*/
|
|
status = test_ntlm_client_authenticate(client);
|
|
|
|
if (status < 0)
|
|
{
|
|
printf("test_ntlm_client_authenticate failure\n");
|
|
goto fail;
|
|
}
|
|
|
|
if (!arg->dynamic)
|
|
{
|
|
SecPkgContext_AuthNtlmTimestamp AuthNtlmTimestamp = WINPR_C_ARRAY_INIT;
|
|
SecPkgContext_AuthNtlmClientChallenge AuthNtlmClientChallenge = WINPR_C_ARRAY_INIT;
|
|
SecPkgContext_AuthNtlmServerChallenge AuthNtlmServerChallenge = WINPR_C_ARRAY_INIT;
|
|
CopyMemory(AuthNtlmTimestamp.Timestamp, TEST_NTLM_TIMESTAMP, 8);
|
|
AuthNtlmTimestamp.ChallengeOrResponse = TRUE;
|
|
SECURITY_STATUS rc = client->table->SetContextAttributes(
|
|
&client->context, SECPKG_ATTR_AUTH_NTLM_TIMESTAMP, &AuthNtlmTimestamp,
|
|
sizeof(SecPkgContext_AuthNtlmTimestamp));
|
|
WINPR_ASSERT((rc == SEC_E_OK) || (rc == SEC_E_SECPKG_NOT_FOUND));
|
|
|
|
AuthNtlmTimestamp.ChallengeOrResponse = FALSE;
|
|
rc = client->table->SetContextAttributes(&client->context, SECPKG_ATTR_AUTH_NTLM_TIMESTAMP,
|
|
&AuthNtlmTimestamp,
|
|
sizeof(SecPkgContext_AuthNtlmTimestamp));
|
|
WINPR_ASSERT((rc == SEC_E_OK) || (rc == SEC_E_SECPKG_NOT_FOUND));
|
|
|
|
CopyMemory(AuthNtlmClientChallenge.ClientChallenge, TEST_NTLM_CLIENT_CHALLENGE, 8);
|
|
CopyMemory(AuthNtlmServerChallenge.ServerChallenge, TEST_NTLM_SERVER_CHALLENGE, 8);
|
|
rc = client->table->SetContextAttributes(
|
|
&client->context, SECPKG_ATTR_AUTH_NTLM_CLIENT_CHALLENGE, &AuthNtlmClientChallenge,
|
|
sizeof(SecPkgContext_AuthNtlmClientChallenge));
|
|
WINPR_ASSERT((rc == SEC_E_OK) || (rc == SEC_E_SECPKG_NOT_FOUND));
|
|
|
|
rc = client->table->SetContextAttributes(
|
|
&client->context, SECPKG_ATTR_AUTH_NTLM_SERVER_CHALLENGE, &AuthNtlmServerChallenge,
|
|
sizeof(SecPkgContext_AuthNtlmServerChallenge));
|
|
WINPR_ASSERT((rc == SEC_E_OK) || (rc == SEC_E_SECPKG_NOT_FOUND));
|
|
}
|
|
|
|
pSecBuffer = &(client->outputBuffer[0]);
|
|
|
|
if (!arg->dynamic)
|
|
{
|
|
pSecBuffer->cbBuffer = sizeof(TEST_NTLM_NEGOTIATE) - 1;
|
|
free(pSecBuffer->pvBuffer);
|
|
pSecBuffer->pvBuffer = malloc(pSecBuffer->cbBuffer);
|
|
|
|
if (!pSecBuffer->pvBuffer)
|
|
{
|
|
printf("Memory allocation failed\n");
|
|
goto fail;
|
|
}
|
|
|
|
CopyMemory(pSecBuffer->pvBuffer, TEST_NTLM_NEGOTIATE, pSecBuffer->cbBuffer);
|
|
}
|
|
|
|
(void)fprintf(stderr, "NTLM_NEGOTIATE (length = %" PRIu32 "):\n", pSecBuffer->cbBuffer);
|
|
winpr_HexDump("sspi.test", WLOG_DEBUG, (BYTE*)pSecBuffer->pvBuffer, pSecBuffer->cbBuffer);
|
|
/**
|
|
* Server <- Negotiate Message
|
|
* Server -> Challenge Message
|
|
*/
|
|
server->haveInputBuffer = TRUE;
|
|
server->inputBuffer[0].BufferType = SECBUFFER_TOKEN;
|
|
server->inputBuffer[0].pvBuffer = pSecBuffer->pvBuffer;
|
|
server->inputBuffer[0].cbBuffer = pSecBuffer->cbBuffer;
|
|
status = test_ntlm_server_authenticate(arg, server);
|
|
|
|
if (status < 0)
|
|
{
|
|
printf("test_ntlm_server_authenticate failure\n");
|
|
goto fail;
|
|
}
|
|
|
|
if (!arg->dynamic)
|
|
{
|
|
SecPkgContext_AuthNtlmTimestamp AuthNtlmTimestamp = WINPR_C_ARRAY_INIT;
|
|
SecPkgContext_AuthNtlmClientChallenge AuthNtlmClientChallenge = WINPR_C_ARRAY_INIT;
|
|
SecPkgContext_AuthNtlmServerChallenge AuthNtlmServerChallenge = WINPR_C_ARRAY_INIT;
|
|
CopyMemory(AuthNtlmTimestamp.Timestamp, TEST_NTLM_TIMESTAMP, 8);
|
|
AuthNtlmTimestamp.ChallengeOrResponse = TRUE;
|
|
SECURITY_STATUS rc = client->table->SetContextAttributes(
|
|
&server->context, SECPKG_ATTR_AUTH_NTLM_TIMESTAMP, &AuthNtlmTimestamp,
|
|
sizeof(SecPkgContext_AuthNtlmTimestamp));
|
|
WINPR_ASSERT(rc == SEC_E_OK);
|
|
|
|
AuthNtlmTimestamp.ChallengeOrResponse = FALSE;
|
|
rc = client->table->SetContextAttributes(&server->context, SECPKG_ATTR_AUTH_NTLM_TIMESTAMP,
|
|
&AuthNtlmTimestamp,
|
|
sizeof(SecPkgContext_AuthNtlmTimestamp));
|
|
WINPR_ASSERT(rc == SEC_E_OK);
|
|
|
|
CopyMemory(AuthNtlmClientChallenge.ClientChallenge, TEST_NTLM_CLIENT_CHALLENGE, 8);
|
|
CopyMemory(AuthNtlmServerChallenge.ServerChallenge, TEST_NTLM_SERVER_CHALLENGE, 8);
|
|
rc = server->table->SetContextAttributes(
|
|
&server->context, SECPKG_ATTR_AUTH_NTLM_CLIENT_CHALLENGE, &AuthNtlmClientChallenge,
|
|
sizeof(SecPkgContext_AuthNtlmClientChallenge));
|
|
WINPR_ASSERT(rc == SEC_E_OK);
|
|
|
|
rc = server->table->SetContextAttributes(
|
|
&server->context, SECPKG_ATTR_AUTH_NTLM_SERVER_CHALLENGE, &AuthNtlmServerChallenge,
|
|
sizeof(SecPkgContext_AuthNtlmServerChallenge));
|
|
WINPR_ASSERT(rc == SEC_E_OK);
|
|
}
|
|
|
|
pSecBuffer = &(server->outputBuffer[0]);
|
|
|
|
if (!arg->dynamic)
|
|
{
|
|
SecPkgContext_AuthNtlmMessage AuthNtlmMessage = WINPR_C_ARRAY_INIT;
|
|
pSecBuffer->cbBuffer = sizeof(TEST_NTLM_CHALLENGE) - 1;
|
|
free(pSecBuffer->pvBuffer);
|
|
pSecBuffer->pvBuffer = malloc(pSecBuffer->cbBuffer);
|
|
|
|
if (!pSecBuffer->pvBuffer)
|
|
{
|
|
printf("Memory allocation failed\n");
|
|
goto fail;
|
|
}
|
|
|
|
CopyMemory(pSecBuffer->pvBuffer, TEST_NTLM_CHALLENGE, pSecBuffer->cbBuffer);
|
|
AuthNtlmMessage.type = 2;
|
|
AuthNtlmMessage.length = pSecBuffer->cbBuffer;
|
|
AuthNtlmMessage.buffer = (BYTE*)pSecBuffer->pvBuffer;
|
|
SECURITY_STATUS rc = server->table->SetContextAttributes(
|
|
&server->context, SECPKG_ATTR_AUTH_NTLM_MESSAGE, &AuthNtlmMessage,
|
|
sizeof(SecPkgContext_AuthNtlmMessage));
|
|
if (rc != SEC_E_OK)
|
|
goto fail;
|
|
}
|
|
|
|
(void)fprintf(stderr, "NTLM_CHALLENGE (length = %" PRIu32 "):\n", pSecBuffer->cbBuffer);
|
|
winpr_HexDump("sspi.test", WLOG_DEBUG, (BYTE*)pSecBuffer->pvBuffer, pSecBuffer->cbBuffer);
|
|
/**
|
|
* Client <- Challenge Message
|
|
* Client -> Authenticate Message
|
|
*/
|
|
client->haveInputBuffer = TRUE;
|
|
client->inputBuffer[0].BufferType = SECBUFFER_TOKEN;
|
|
client->inputBuffer[0].pvBuffer = pSecBuffer->pvBuffer;
|
|
client->inputBuffer[0].cbBuffer = pSecBuffer->cbBuffer;
|
|
status = test_ntlm_client_authenticate(client);
|
|
|
|
if (status < 0)
|
|
{
|
|
printf("test_ntlm_client_authenticate failure\n");
|
|
goto fail;
|
|
}
|
|
|
|
pSecBuffer = &(client->outputBuffer[0]);
|
|
|
|
if (!arg->dynamic)
|
|
{
|
|
pSecBuffer->cbBuffer = sizeof(TEST_NTLM_AUTHENTICATE) - 1;
|
|
free(pSecBuffer->pvBuffer);
|
|
pSecBuffer->pvBuffer = malloc(pSecBuffer->cbBuffer);
|
|
|
|
if (!pSecBuffer->pvBuffer)
|
|
{
|
|
printf("Memory allocation failed\n");
|
|
goto fail;
|
|
}
|
|
|
|
CopyMemory(pSecBuffer->pvBuffer, TEST_NTLM_AUTHENTICATE, pSecBuffer->cbBuffer);
|
|
}
|
|
|
|
(void)fprintf(stderr, "NTLM_AUTHENTICATE (length = %" PRIu32 "):\n", pSecBuffer->cbBuffer);
|
|
winpr_HexDump("sspi.test", WLOG_DEBUG, (BYTE*)pSecBuffer->pvBuffer, pSecBuffer->cbBuffer);
|
|
/**
|
|
* Server <- Authenticate Message
|
|
*/
|
|
server->haveInputBuffer = TRUE;
|
|
server->inputBuffer[0].BufferType = SECBUFFER_TOKEN;
|
|
server->inputBuffer[0].pvBuffer = pSecBuffer->pvBuffer;
|
|
server->inputBuffer[0].cbBuffer = pSecBuffer->cbBuffer;
|
|
status = test_ntlm_server_authenticate(arg, server);
|
|
|
|
if (status < 0)
|
|
{
|
|
printf("test_ntlm_server_authenticate failure\n");
|
|
goto fail;
|
|
}
|
|
|
|
rc = TRUE;
|
|
|
|
fail:
|
|
/**
|
|
* Cleanup & Termination
|
|
*/
|
|
test_ntlm_client_free(client);
|
|
test_ntlm_server_free(server);
|
|
|
|
printf("testcase {user=%s, domain=%s, password=%s, dynamic=%s} returns %d\n", arg->user,
|
|
arg->domain, arg->pwd, arg->dynamic ? "TRUE" : "FALSE", rc);
|
|
return rc;
|
|
}
|
|
|
|
int TestNTLM(int argc, char* argv[])
|
|
{
|
|
WINPR_UNUSED(argc);
|
|
WINPR_UNUSED(argv);
|
|
|
|
const struct test_input_t inputs[] = {
|
|
{ TEST_NTLM_USER, TEST_NTLM_DOMAIN, TEST_NTLM_PASSWORD, TEST_NTLM_HASH, TEST_NTLM_V2_HASH,
|
|
TRUE, TRUE },
|
|
{ TEST_NTLM_USER, TEST_NTLM_DOMAIN, TEST_NTLM_PASSWORD, TEST_NTLM_HASH, TEST_NTLM_V2_HASH,
|
|
FALSE, TRUE },
|
|
{ TEST_NTLM_USER, TEST_NTLM_DOMAIN, "", TEST_EMPTY_PWD_NTLM_HASH,
|
|
TEST_EMPTY_PWD_NTLM_V2_HASH, TRUE, TRUE },
|
|
{ TEST_NTLM_USER, TEST_NTLM_DOMAIN, nullptr, TEST_EMPTY_PWD_NTLM_HASH,
|
|
TEST_EMPTY_PWD_NTLM_V2_HASH, TRUE, FALSE }
|
|
};
|
|
|
|
int rc = 0;
|
|
for (size_t x = 0; x < ARRAYSIZE(inputs); x++)
|
|
{
|
|
const struct test_input_t* cur = &inputs[x];
|
|
const BOOL res = test_default(cur);
|
|
if (res != cur->expected)
|
|
rc = -1;
|
|
}
|
|
return rc;
|
|
}
|