1266 lines
38 KiB
C
1266 lines
38 KiB
C
/**
|
|
* WinPR: Windows Portable Runtime
|
|
* NTLM Security Package
|
|
*
|
|
* Copyright 2011-2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <winpr/crt.h>
|
|
#include <winpr/assert.h>
|
|
#include <winpr/sspi.h>
|
|
#include <winpr/print.h>
|
|
#include <winpr/tchar.h>
|
|
#include <winpr/sysinfo.h>
|
|
#include <winpr/registry.h>
|
|
#include <winpr/endian.h>
|
|
#include <winpr/build-config.h>
|
|
|
|
#include "ntlm.h"
|
|
#include "ntlm_export.h"
|
|
#include "../sspi.h"
|
|
|
|
#include "ntlm_message.h"
|
|
|
|
#include "../../log.h"
|
|
#define TAG WINPR_TAG("sspi.NTLM")
|
|
|
|
#define WINPR_KEY "Software\\" WINPR_VENDOR_STRING "\\" WINPR_PRODUCT_STRING "\\WinPR\\NTLM"
|
|
|
|
static char* NTLM_PACKAGE_NAME = "NTLM";
|
|
|
|
static int ntlm_SetContextWorkstation(NTLM_CONTEXT* context, char* Workstation)
|
|
{
|
|
int status;
|
|
char* ws = Workstation;
|
|
DWORD nSize = 0;
|
|
CHAR* computerName;
|
|
|
|
if (!Workstation)
|
|
{
|
|
if (GetComputerNameExA(ComputerNameNetBIOS, NULL, &nSize) ||
|
|
GetLastError() != ERROR_MORE_DATA)
|
|
return -1;
|
|
|
|
computerName = calloc(nSize, sizeof(CHAR));
|
|
|
|
if (!computerName)
|
|
return -1;
|
|
|
|
if (!GetComputerNameExA(ComputerNameNetBIOS, computerName, &nSize))
|
|
{
|
|
free(computerName);
|
|
return -1;
|
|
}
|
|
|
|
if (nSize > MAX_COMPUTERNAME_LENGTH)
|
|
computerName[MAX_COMPUTERNAME_LENGTH] = '\0';
|
|
|
|
ws = computerName;
|
|
|
|
if (!ws)
|
|
return -1;
|
|
}
|
|
|
|
context->Workstation.Buffer = NULL;
|
|
status = ConvertToUnicode(CP_UTF8, 0, ws, -1, &context->Workstation.Buffer, 0);
|
|
|
|
if (!Workstation)
|
|
free(ws);
|
|
|
|
if (status <= 0)
|
|
return -1;
|
|
|
|
context->Workstation.Length = (USHORT)(status - 1);
|
|
context->Workstation.Length *= 2;
|
|
return 1;
|
|
}
|
|
|
|
static int ntlm_SetContextServicePrincipalNameW(NTLM_CONTEXT* context, LPWSTR ServicePrincipalName)
|
|
{
|
|
if (!ServicePrincipalName)
|
|
{
|
|
context->ServicePrincipalName.Buffer = NULL;
|
|
context->ServicePrincipalName.Length = 0;
|
|
return 1;
|
|
}
|
|
|
|
context->ServicePrincipalName.Length = (USHORT)(_wcslen(ServicePrincipalName) * 2);
|
|
context->ServicePrincipalName.Buffer = (PWSTR)malloc(context->ServicePrincipalName.Length + 2);
|
|
|
|
if (!context->ServicePrincipalName.Buffer)
|
|
return -1;
|
|
|
|
CopyMemory(context->ServicePrincipalName.Buffer, ServicePrincipalName,
|
|
context->ServicePrincipalName.Length + 2);
|
|
return 1;
|
|
}
|
|
|
|
static int ntlm_SetContextTargetName(NTLM_CONTEXT* context, char* TargetName)
|
|
{
|
|
int status;
|
|
char* name = TargetName;
|
|
DWORD nSize = 0;
|
|
CHAR* computerName = NULL;
|
|
|
|
if (!name)
|
|
{
|
|
if (GetComputerNameExA(ComputerNameNetBIOS, NULL, &nSize) ||
|
|
GetLastError() != ERROR_MORE_DATA)
|
|
return -1;
|
|
|
|
computerName = calloc(nSize, sizeof(CHAR));
|
|
|
|
if (!computerName)
|
|
return -1;
|
|
|
|
if (!GetComputerNameExA(ComputerNameNetBIOS, computerName, &nSize))
|
|
{
|
|
free(computerName);
|
|
return -1;
|
|
}
|
|
|
|
if (nSize > MAX_COMPUTERNAME_LENGTH)
|
|
computerName[MAX_COMPUTERNAME_LENGTH] = '\0';
|
|
|
|
name = computerName;
|
|
|
|
if (!name)
|
|
return -1;
|
|
|
|
CharUpperA(name);
|
|
}
|
|
|
|
context->TargetName.pvBuffer = NULL;
|
|
status = ConvertToUnicode(CP_UTF8, 0, name, -1, (LPWSTR*)&context->TargetName.pvBuffer, 0);
|
|
|
|
if (status <= 0)
|
|
{
|
|
if (!TargetName)
|
|
free(name);
|
|
|
|
return -1;
|
|
}
|
|
|
|
context->TargetName.cbBuffer = (USHORT)((status - 1) * 2);
|
|
|
|
if (!TargetName)
|
|
free(name);
|
|
|
|
return 1;
|
|
}
|
|
|
|
static NTLM_CONTEXT* ntlm_ContextNew(void)
|
|
{
|
|
HKEY hKey;
|
|
LONG status;
|
|
DWORD dwType;
|
|
DWORD dwSize;
|
|
DWORD dwValue;
|
|
NTLM_CONTEXT* context;
|
|
context = (NTLM_CONTEXT*)calloc(1, sizeof(NTLM_CONTEXT));
|
|
|
|
if (!context)
|
|
return NULL;
|
|
|
|
context->NTLMv2 = TRUE;
|
|
context->UseMIC = FALSE;
|
|
context->SendVersionInfo = TRUE;
|
|
context->SendSingleHostData = FALSE;
|
|
context->SendWorkstationName = TRUE;
|
|
context->NegotiateKeyExchange = TRUE;
|
|
context->UseSamFileDatabase = TRUE;
|
|
status = RegOpenKeyExA(HKEY_LOCAL_MACHINE, WINPR_KEY, 0, KEY_READ | KEY_WOW64_64KEY, &hKey);
|
|
|
|
if (status == ERROR_SUCCESS)
|
|
{
|
|
if (RegQueryValueEx(hKey, _T("NTLMv2"), NULL, &dwType, (BYTE*)&dwValue, &dwSize) ==
|
|
ERROR_SUCCESS)
|
|
context->NTLMv2 = dwValue ? 1 : 0;
|
|
|
|
if (RegQueryValueEx(hKey, _T("UseMIC"), NULL, &dwType, (BYTE*)&dwValue, &dwSize) ==
|
|
ERROR_SUCCESS)
|
|
context->UseMIC = dwValue ? 1 : 0;
|
|
|
|
if (RegQueryValueEx(hKey, _T("SendVersionInfo"), NULL, &dwType, (BYTE*)&dwValue, &dwSize) ==
|
|
ERROR_SUCCESS)
|
|
context->SendVersionInfo = dwValue ? 1 : 0;
|
|
|
|
if (RegQueryValueEx(hKey, _T("SendSingleHostData"), NULL, &dwType, (BYTE*)&dwValue,
|
|
&dwSize) == ERROR_SUCCESS)
|
|
context->SendSingleHostData = dwValue ? 1 : 0;
|
|
|
|
if (RegQueryValueEx(hKey, _T("SendWorkstationName"), NULL, &dwType, (BYTE*)&dwValue,
|
|
&dwSize) == ERROR_SUCCESS)
|
|
context->SendWorkstationName = dwValue ? 1 : 0;
|
|
|
|
if (RegQueryValueEx(hKey, _T("WorkstationName"), NULL, &dwType, NULL, &dwSize) ==
|
|
ERROR_SUCCESS)
|
|
{
|
|
char* workstation = (char*)malloc(dwSize + 1);
|
|
|
|
if (!workstation)
|
|
{
|
|
free(context);
|
|
return NULL;
|
|
}
|
|
|
|
status = RegQueryValueExA(hKey, "WorkstationName", NULL, &dwType, (BYTE*)workstation,
|
|
&dwSize);
|
|
workstation[dwSize] = '\0';
|
|
|
|
if (ntlm_SetContextWorkstation(context, workstation) < 0)
|
|
{
|
|
free(workstation);
|
|
free(context);
|
|
return NULL;
|
|
}
|
|
|
|
free(workstation);
|
|
}
|
|
|
|
RegCloseKey(hKey);
|
|
}
|
|
|
|
/*
|
|
* Extended Protection is enabled by default in Windows 7,
|
|
* but enabling it in WinPR breaks TS Gateway at this point
|
|
*/
|
|
context->SuppressExtendedProtection = FALSE;
|
|
status = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("System\\CurrentControlSet\\Control\\LSA"), 0,
|
|
KEY_READ | KEY_WOW64_64KEY, &hKey);
|
|
|
|
if (status == ERROR_SUCCESS)
|
|
{
|
|
if (RegQueryValueEx(hKey, _T("SuppressExtendedProtection"), NULL, &dwType, (BYTE*)&dwValue,
|
|
&dwSize) == ERROR_SUCCESS)
|
|
context->SuppressExtendedProtection = dwValue ? 1 : 0;
|
|
|
|
RegCloseKey(hKey);
|
|
}
|
|
|
|
context->NegotiateFlags = 0;
|
|
context->LmCompatibilityLevel = 3;
|
|
context->state = NTLM_STATE_INITIAL;
|
|
FillMemory(context->MachineID, sizeof(context->MachineID), 0xAA);
|
|
|
|
if (context->NTLMv2)
|
|
context->UseMIC = TRUE;
|
|
|
|
return context;
|
|
}
|
|
|
|
static void ntlm_ContextFree(NTLM_CONTEXT* context)
|
|
{
|
|
if (!context)
|
|
return;
|
|
|
|
winpr_RC4_Free(context->SendRc4Seal);
|
|
winpr_RC4_Free(context->RecvRc4Seal);
|
|
sspi_SecBufferFree(&context->NegotiateMessage);
|
|
sspi_SecBufferFree(&context->ChallengeMessage);
|
|
sspi_SecBufferFree(&context->AuthenticateMessage);
|
|
sspi_SecBufferFree(&context->ChallengeTargetInfo);
|
|
sspi_SecBufferFree(&context->TargetName);
|
|
sspi_SecBufferFree(&context->NtChallengeResponse);
|
|
sspi_SecBufferFree(&context->LmChallengeResponse);
|
|
free(context->ServicePrincipalName.Buffer);
|
|
free(context->Workstation.Buffer);
|
|
free(context);
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_AcquireCredentialsHandleW(
|
|
SEC_WCHAR* pszPrincipal, SEC_WCHAR* pszPackage, ULONG fCredentialUse, void* pvLogonID,
|
|
void* pAuthData, SEC_GET_KEY_FN pGetKeyFn, void* pvGetKeyArgument, PCredHandle phCredential,
|
|
PTimeStamp ptsExpiry)
|
|
{
|
|
SSPI_CREDENTIALS* credentials;
|
|
SEC_WINNT_AUTH_IDENTITY* identity;
|
|
|
|
if ((fCredentialUse != SECPKG_CRED_OUTBOUND) && (fCredentialUse != SECPKG_CRED_INBOUND) &&
|
|
(fCredentialUse != SECPKG_CRED_BOTH))
|
|
{
|
|
return SEC_E_INVALID_PARAMETER;
|
|
}
|
|
|
|
credentials = sspi_CredentialsNew();
|
|
|
|
if (!credentials)
|
|
return SEC_E_INTERNAL_ERROR;
|
|
|
|
credentials->fCredentialUse = fCredentialUse;
|
|
credentials->pGetKeyFn = pGetKeyFn;
|
|
credentials->pvGetKeyArgument = pvGetKeyArgument;
|
|
identity = (SEC_WINNT_AUTH_IDENTITY*)pAuthData;
|
|
|
|
if (identity)
|
|
sspi_CopyAuthIdentity(&(credentials->identity), identity);
|
|
|
|
sspi_SecureHandleSetLowerPointer(phCredential, (void*)credentials);
|
|
sspi_SecureHandleSetUpperPointer(phCredential, (void*)NTLM_PACKAGE_NAME);
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_AcquireCredentialsHandleA(
|
|
SEC_CHAR* pszPrincipal, SEC_CHAR* pszPackage, ULONG fCredentialUse, void* pvLogonID,
|
|
void* pAuthData, SEC_GET_KEY_FN pGetKeyFn, void* pvGetKeyArgument, PCredHandle phCredential,
|
|
PTimeStamp ptsExpiry)
|
|
{
|
|
SSPI_CREDENTIALS* credentials;
|
|
SEC_WINNT_AUTH_IDENTITY* identity;
|
|
|
|
if ((fCredentialUse != SECPKG_CRED_OUTBOUND) && (fCredentialUse != SECPKG_CRED_INBOUND) &&
|
|
(fCredentialUse != SECPKG_CRED_BOTH))
|
|
{
|
|
return SEC_E_INVALID_PARAMETER;
|
|
}
|
|
|
|
credentials = sspi_CredentialsNew();
|
|
|
|
if (!credentials)
|
|
return SEC_E_INTERNAL_ERROR;
|
|
|
|
credentials->fCredentialUse = fCredentialUse;
|
|
credentials->pGetKeyFn = pGetKeyFn;
|
|
credentials->pvGetKeyArgument = pvGetKeyArgument;
|
|
identity = (SEC_WINNT_AUTH_IDENTITY*)pAuthData;
|
|
|
|
if (identity)
|
|
sspi_CopyAuthIdentity(&(credentials->identity), identity);
|
|
|
|
sspi_SecureHandleSetLowerPointer(phCredential, (void*)credentials);
|
|
sspi_SecureHandleSetUpperPointer(phCredential, (void*)NTLM_PACKAGE_NAME);
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_FreeCredentialsHandle(PCredHandle phCredential)
|
|
{
|
|
SSPI_CREDENTIALS* credentials;
|
|
|
|
if (!phCredential)
|
|
return SEC_E_INVALID_HANDLE;
|
|
|
|
credentials = (SSPI_CREDENTIALS*)sspi_SecureHandleGetLowerPointer(phCredential);
|
|
|
|
if (!credentials)
|
|
return SEC_E_INVALID_HANDLE;
|
|
|
|
sspi_CredentialsFree(credentials);
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_QueryCredentialsAttributesW(PCredHandle phCredential,
|
|
ULONG ulAttribute, void* pBuffer)
|
|
{
|
|
if (ulAttribute == SECPKG_CRED_ATTR_NAMES)
|
|
{
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
WLog_ERR(TAG, "[%s]: TODO: Implement", __FUNCTION__);
|
|
return SEC_E_UNSUPPORTED_FUNCTION;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_QueryCredentialsAttributesA(PCredHandle phCredential,
|
|
ULONG ulAttribute, void* pBuffer)
|
|
{
|
|
return ntlm_QueryCredentialsAttributesW(phCredential, ulAttribute, pBuffer);
|
|
}
|
|
|
|
/**
|
|
* @see http://msdn.microsoft.com/en-us/library/windows/desktop/aa374707
|
|
*/
|
|
static SECURITY_STATUS SEC_ENTRY
|
|
ntlm_AcceptSecurityContext(PCredHandle phCredential, PCtxtHandle phContext, PSecBufferDesc pInput,
|
|
ULONG fContextReq, ULONG TargetDataRep, PCtxtHandle phNewContext,
|
|
PSecBufferDesc pOutput, PULONG pfContextAttr, PTimeStamp ptsTimeStamp)
|
|
{
|
|
NTLM_CONTEXT* context;
|
|
SECURITY_STATUS status;
|
|
SSPI_CREDENTIALS* credentials;
|
|
PSecBuffer input_buffer;
|
|
PSecBuffer output_buffer;
|
|
context = (NTLM_CONTEXT*)sspi_SecureHandleGetLowerPointer(phContext);
|
|
|
|
if (!context)
|
|
{
|
|
context = ntlm_ContextNew();
|
|
|
|
if (!context)
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
context->server = TRUE;
|
|
|
|
if (fContextReq & ASC_REQ_CONFIDENTIALITY)
|
|
context->confidentiality = TRUE;
|
|
|
|
credentials = (SSPI_CREDENTIALS*)sspi_SecureHandleGetLowerPointer(phCredential);
|
|
context->credentials = credentials;
|
|
ntlm_SetContextTargetName(context, NULL);
|
|
sspi_SecureHandleSetLowerPointer(phNewContext, context);
|
|
sspi_SecureHandleSetUpperPointer(phNewContext, (void*)NTLM_PACKAGE_NAME);
|
|
}
|
|
|
|
if (context->state == NTLM_STATE_INITIAL)
|
|
{
|
|
context->state = NTLM_STATE_NEGOTIATE;
|
|
|
|
if (!pInput)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (pInput->cBuffers < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
input_buffer = sspi_FindSecBuffer(pInput, SECBUFFER_TOKEN);
|
|
|
|
if (!input_buffer)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (input_buffer->cbBuffer < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
status = ntlm_read_NegotiateMessage(context, input_buffer);
|
|
|
|
if (context->state == NTLM_STATE_CHALLENGE)
|
|
{
|
|
if (!pOutput)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (pOutput->cBuffers < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
output_buffer = sspi_FindSecBuffer(pOutput, SECBUFFER_TOKEN);
|
|
|
|
if (!output_buffer->BufferType)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (output_buffer->cbBuffer < 1)
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
return ntlm_write_ChallengeMessage(context, output_buffer);
|
|
}
|
|
|
|
return SEC_E_OUT_OF_SEQUENCE;
|
|
}
|
|
else if (context->state == NTLM_STATE_AUTHENTICATE)
|
|
{
|
|
if (!pInput)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (pInput->cBuffers < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
input_buffer = sspi_FindSecBuffer(pInput, SECBUFFER_TOKEN);
|
|
|
|
if (!input_buffer)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (input_buffer->cbBuffer < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
status = ntlm_read_AuthenticateMessage(context, input_buffer);
|
|
|
|
if (pOutput)
|
|
{
|
|
ULONG i;
|
|
|
|
for (i = 0; i < pOutput->cBuffers; i++)
|
|
{
|
|
pOutput->pBuffers[i].cbBuffer = 0;
|
|
pOutput->pBuffers[i].BufferType = SECBUFFER_TOKEN;
|
|
}
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
return SEC_E_OUT_OF_SEQUENCE;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_ImpersonateSecurityContext(PCtxtHandle phContext)
|
|
{
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_InitializeSecurityContextW(
|
|
PCredHandle phCredential, PCtxtHandle phContext, SEC_WCHAR* pszTargetName, ULONG fContextReq,
|
|
ULONG Reserved1, ULONG TargetDataRep, PSecBufferDesc pInput, ULONG Reserved2,
|
|
PCtxtHandle phNewContext, PSecBufferDesc pOutput, PULONG pfContextAttr, PTimeStamp ptsExpiry)
|
|
{
|
|
NTLM_CONTEXT* context;
|
|
SECURITY_STATUS status;
|
|
SSPI_CREDENTIALS* credentials;
|
|
PSecBuffer input_buffer = NULL;
|
|
PSecBuffer output_buffer = NULL;
|
|
PSecBuffer channel_bindings = NULL;
|
|
context = (NTLM_CONTEXT*)sspi_SecureHandleGetLowerPointer(phContext);
|
|
|
|
if (!context)
|
|
{
|
|
context = ntlm_ContextNew();
|
|
|
|
if (!context)
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
if (fContextReq & ISC_REQ_CONFIDENTIALITY)
|
|
context->confidentiality = TRUE;
|
|
|
|
credentials = (SSPI_CREDENTIALS*)sspi_SecureHandleGetLowerPointer(phCredential);
|
|
context->credentials = credentials;
|
|
|
|
if (context->Workstation.Length < 1)
|
|
{
|
|
if (ntlm_SetContextWorkstation(context, NULL) < 0)
|
|
{
|
|
ntlm_ContextFree(context);
|
|
return SEC_E_INTERNAL_ERROR;
|
|
}
|
|
}
|
|
|
|
if (ntlm_SetContextServicePrincipalNameW(context, pszTargetName) < 0)
|
|
{
|
|
ntlm_ContextFree(context);
|
|
return SEC_E_INTERNAL_ERROR;
|
|
}
|
|
|
|
sspi_SecureHandleSetLowerPointer(phNewContext, context);
|
|
sspi_SecureHandleSetUpperPointer(phNewContext, (void*)NTLM_PACKAGE_NAME);
|
|
}
|
|
|
|
if ((!pInput) || (context->state == NTLM_STATE_AUTHENTICATE))
|
|
{
|
|
if (!pOutput)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (pOutput->cBuffers < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
output_buffer = sspi_FindSecBuffer(pOutput, SECBUFFER_TOKEN);
|
|
|
|
if (!output_buffer)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (output_buffer->cbBuffer < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (context->state == NTLM_STATE_INITIAL)
|
|
context->state = NTLM_STATE_NEGOTIATE;
|
|
|
|
if (context->state == NTLM_STATE_NEGOTIATE)
|
|
return ntlm_write_NegotiateMessage(context, output_buffer);
|
|
|
|
return SEC_E_OUT_OF_SEQUENCE;
|
|
}
|
|
else
|
|
{
|
|
if (pInput->cBuffers < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
input_buffer = sspi_FindSecBuffer(pInput, SECBUFFER_TOKEN);
|
|
|
|
if (!input_buffer)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (input_buffer->cbBuffer < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
channel_bindings = sspi_FindSecBuffer(pInput, SECBUFFER_CHANNEL_BINDINGS);
|
|
|
|
if (channel_bindings)
|
|
{
|
|
context->Bindings.BindingsLength = channel_bindings->cbBuffer;
|
|
context->Bindings.Bindings = (SEC_CHANNEL_BINDINGS*)channel_bindings->pvBuffer;
|
|
}
|
|
|
|
if (context->state == NTLM_STATE_CHALLENGE)
|
|
{
|
|
status = ntlm_read_ChallengeMessage(context, input_buffer);
|
|
|
|
if (!pOutput)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (pOutput->cBuffers < 1)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
output_buffer = sspi_FindSecBuffer(pOutput, SECBUFFER_TOKEN);
|
|
|
|
if (!output_buffer)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (output_buffer->cbBuffer < 1)
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
if (context->state == NTLM_STATE_AUTHENTICATE)
|
|
return ntlm_write_AuthenticateMessage(context, output_buffer);
|
|
}
|
|
|
|
return SEC_E_OUT_OF_SEQUENCE;
|
|
}
|
|
|
|
return SEC_E_OUT_OF_SEQUENCE;
|
|
}
|
|
|
|
/**
|
|
* @see http://msdn.microsoft.com/en-us/library/windows/desktop/aa375512%28v=vs.85%29.aspx
|
|
*/
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_InitializeSecurityContextA(
|
|
PCredHandle phCredential, PCtxtHandle phContext, SEC_CHAR* pszTargetName, ULONG fContextReq,
|
|
ULONG Reserved1, ULONG TargetDataRep, PSecBufferDesc pInput, ULONG Reserved2,
|
|
PCtxtHandle phNewContext, PSecBufferDesc pOutput, PULONG pfContextAttr, PTimeStamp ptsExpiry)
|
|
{
|
|
SECURITY_STATUS status;
|
|
SEC_WCHAR* pszTargetNameW = NULL;
|
|
|
|
if (pszTargetName)
|
|
{
|
|
if (ConvertToUnicode(CP_UTF8, 0, pszTargetName, -1, &pszTargetNameW, 0) <= 0)
|
|
return SEC_E_INTERNAL_ERROR;
|
|
}
|
|
|
|
status = ntlm_InitializeSecurityContextW(phCredential, phContext, pszTargetNameW, fContextReq,
|
|
Reserved1, TargetDataRep, pInput, Reserved2,
|
|
phNewContext, pOutput, pfContextAttr, ptsExpiry);
|
|
free(pszTargetNameW);
|
|
return status;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_CompleteAuthToken(PCtxtHandle phContext,
|
|
PSecBufferDesc pToken)
|
|
{
|
|
NTLM_CONTEXT* context;
|
|
SECURITY_STATUS status = SEC_E_OK;
|
|
context = (NTLM_CONTEXT*)sspi_SecureHandleGetLowerPointer(phContext);
|
|
|
|
if (!context)
|
|
return SEC_E_INVALID_HANDLE;
|
|
|
|
if (context->server)
|
|
{
|
|
status = ntlm_server_AuthenticateComplete(context);
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/windows/desktop/aa375354 */
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_DeleteSecurityContext(PCtxtHandle phContext)
|
|
{
|
|
NTLM_CONTEXT* context;
|
|
context = (NTLM_CONTEXT*)sspi_SecureHandleGetLowerPointer(phContext);
|
|
|
|
if (!context)
|
|
return SEC_E_INVALID_HANDLE;
|
|
|
|
ntlm_ContextFree(context);
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
SECURITY_STATUS ntlm_computeProofValue(NTLM_CONTEXT* ntlm, SecBuffer* ntproof)
|
|
{
|
|
BYTE* blob;
|
|
SecBuffer* target = &ntlm->ChallengeTargetInfo;
|
|
|
|
if (!sspi_SecBufferAlloc(ntproof, 36 + target->cbBuffer))
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
blob = (BYTE*)ntproof->pvBuffer;
|
|
CopyMemory(blob, ntlm->ServerChallenge, 8); /* Server challenge. */
|
|
blob[8] = 1; /* Response version. */
|
|
blob[9] = 1; /* Highest response version understood by the client. */
|
|
/* Reserved 6B. */
|
|
CopyMemory(&blob[16], ntlm->Timestamp, 8); /* Time. */
|
|
CopyMemory(&blob[24], ntlm->ClientChallenge, 8); /* Client challenge. */
|
|
/* Reserved 4B. */
|
|
/* Server name. */
|
|
CopyMemory(&blob[36], target->pvBuffer, target->cbBuffer);
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
SECURITY_STATUS ntlm_computeMicValue(NTLM_CONTEXT* ntlm, SecBuffer* micvalue)
|
|
{
|
|
BYTE* blob;
|
|
ULONG msgSize = ntlm->NegotiateMessage.cbBuffer + ntlm->ChallengeMessage.cbBuffer +
|
|
ntlm->AuthenticateMessage.cbBuffer;
|
|
|
|
if (!sspi_SecBufferAlloc(micvalue, msgSize))
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
blob = (BYTE*)micvalue->pvBuffer;
|
|
CopyMemory(blob, ntlm->NegotiateMessage.pvBuffer, ntlm->NegotiateMessage.cbBuffer);
|
|
blob += ntlm->NegotiateMessage.cbBuffer;
|
|
CopyMemory(blob, ntlm->ChallengeMessage.pvBuffer, ntlm->ChallengeMessage.cbBuffer);
|
|
blob += ntlm->ChallengeMessage.cbBuffer;
|
|
CopyMemory(blob, ntlm->AuthenticateMessage.pvBuffer, ntlm->AuthenticateMessage.cbBuffer);
|
|
blob += ntlm->MessageIntegrityCheckOffset;
|
|
ZeroMemory(blob, 16);
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/windows/desktop/aa379337/ */
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_QueryContextAttributesW(PCtxtHandle phContext,
|
|
ULONG ulAttribute, void* pBuffer)
|
|
{
|
|
NTLM_CONTEXT* context;
|
|
|
|
if (!phContext)
|
|
return SEC_E_INVALID_HANDLE;
|
|
|
|
if (!pBuffer)
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
context = (NTLM_CONTEXT*)sspi_SecureHandleGetLowerPointer(phContext);
|
|
|
|
if (ulAttribute == SECPKG_ATTR_SIZES)
|
|
{
|
|
SecPkgContext_Sizes* ContextSizes = (SecPkgContext_Sizes*)pBuffer;
|
|
ContextSizes->cbMaxToken = 2010;
|
|
ContextSizes->cbMaxSignature = 16; /* the size of expected signature is 16 bytes */
|
|
ContextSizes->cbBlockSize = 0; /* no padding */
|
|
ContextSizes->cbSecurityTrailer = 16; /* no security trailer appended in NTLM
|
|
contrary to Kerberos */
|
|
return SEC_E_OK;
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_IDENTITY)
|
|
{
|
|
int status;
|
|
char* UserA = NULL;
|
|
char* DomainA = NULL;
|
|
SSPI_CREDENTIALS* credentials;
|
|
SecPkgContext_AuthIdentity* AuthIdentity = (SecPkgContext_AuthIdentity*)pBuffer;
|
|
context->UseSamFileDatabase = FALSE;
|
|
credentials = context->credentials;
|
|
ZeroMemory(AuthIdentity, sizeof(SecPkgContext_AuthIdentity));
|
|
UserA = AuthIdentity->User;
|
|
|
|
if (credentials->identity.UserLength > 0)
|
|
{
|
|
WINPR_ASSERT(credentials->identity.UserLength <= INT_MAX);
|
|
status =
|
|
ConvertFromUnicode(CP_UTF8, 0, (WCHAR*)credentials->identity.User,
|
|
(int)credentials->identity.UserLength, &UserA, 256, NULL, NULL);
|
|
|
|
if (status <= 0)
|
|
return SEC_E_INTERNAL_ERROR;
|
|
}
|
|
|
|
DomainA = AuthIdentity->Domain;
|
|
|
|
if (credentials->identity.DomainLength > 0)
|
|
{
|
|
WINPR_ASSERT(credentials->identity.DomainLength <= INT_MAX);
|
|
status = ConvertFromUnicode(CP_UTF8, 0, (WCHAR*)credentials->identity.Domain,
|
|
(int)credentials->identity.DomainLength, &DomainA, 256,
|
|
NULL, NULL);
|
|
|
|
if (status <= 0)
|
|
return SEC_E_INTERNAL_ERROR;
|
|
}
|
|
|
|
return SEC_E_OK;
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_NTPROOF_VALUE)
|
|
{
|
|
return ntlm_computeProofValue(context, (SecBuffer*)pBuffer);
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_RANDKEY)
|
|
{
|
|
SecBuffer* randkey;
|
|
randkey = (SecBuffer*)pBuffer;
|
|
|
|
if (!sspi_SecBufferAlloc(randkey, 16))
|
|
return (SEC_E_INSUFFICIENT_MEMORY);
|
|
|
|
CopyMemory(randkey->pvBuffer, context->EncryptedRandomSessionKey, 16);
|
|
return (SEC_E_OK);
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_MIC)
|
|
{
|
|
SecBuffer* mic;
|
|
NTLM_AUTHENTICATE_MESSAGE* message;
|
|
mic = (SecBuffer*)pBuffer;
|
|
message = &context->AUTHENTICATE_MESSAGE;
|
|
|
|
if (!sspi_SecBufferAlloc(mic, 16))
|
|
return (SEC_E_INSUFFICIENT_MEMORY);
|
|
|
|
CopyMemory(mic->pvBuffer, message->MessageIntegrityCheck, 16);
|
|
return (SEC_E_OK);
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_MIC_VALUE)
|
|
{
|
|
return ntlm_computeMicValue(context, (SecBuffer*)pBuffer);
|
|
}
|
|
|
|
WLog_ERR(TAG, "[%s]: TODO: Implement ulAttribute%08" PRIx32, __FUNCTION__, ulAttribute);
|
|
return SEC_E_UNSUPPORTED_FUNCTION;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_QueryContextAttributesA(PCtxtHandle phContext,
|
|
ULONG ulAttribute, void* pBuffer)
|
|
{
|
|
return ntlm_QueryContextAttributesW(phContext, ulAttribute, pBuffer);
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_SetContextAttributesW(PCtxtHandle phContext,
|
|
ULONG ulAttribute, void* pBuffer,
|
|
ULONG cbBuffer)
|
|
{
|
|
NTLM_CONTEXT* context;
|
|
|
|
if (!phContext)
|
|
return SEC_E_INVALID_HANDLE;
|
|
|
|
if (!pBuffer)
|
|
return SEC_E_INVALID_PARAMETER;
|
|
|
|
context = (NTLM_CONTEXT*)sspi_SecureHandleGetLowerPointer(phContext);
|
|
|
|
if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_HASH)
|
|
{
|
|
SecPkgContext_AuthNtlmHash* AuthNtlmHash = (SecPkgContext_AuthNtlmHash*)pBuffer;
|
|
|
|
if (cbBuffer < sizeof(SecPkgContext_AuthNtlmHash))
|
|
return SEC_E_INVALID_PARAMETER;
|
|
|
|
if (AuthNtlmHash->Version == 1)
|
|
CopyMemory(context->NtlmHash, AuthNtlmHash->NtlmHash, 16);
|
|
else if (AuthNtlmHash->Version == 2)
|
|
CopyMemory(context->NtlmV2Hash, AuthNtlmHash->NtlmHash, 16);
|
|
|
|
return SEC_E_OK;
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_SAM_FILE)
|
|
{
|
|
const char* filename = (char*)pBuffer;
|
|
free(context->SamFile);
|
|
context->SamFile = NULL;
|
|
|
|
if (filename)
|
|
{
|
|
context->SamFile = _strdup(filename);
|
|
|
|
if (!context->SamFile)
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
}
|
|
|
|
return SEC_E_OK;
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_MESSAGE)
|
|
{
|
|
SecPkgContext_AuthNtlmMessage* AuthNtlmMessage = (SecPkgContext_AuthNtlmMessage*)pBuffer;
|
|
|
|
if (cbBuffer < sizeof(SecPkgContext_AuthNtlmMessage))
|
|
return SEC_E_INVALID_PARAMETER;
|
|
|
|
if (AuthNtlmMessage->type == 1)
|
|
{
|
|
sspi_SecBufferFree(&context->NegotiateMessage);
|
|
|
|
if (!sspi_SecBufferAlloc(&context->NegotiateMessage, AuthNtlmMessage->length))
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
CopyMemory(context->NegotiateMessage.pvBuffer, AuthNtlmMessage->buffer,
|
|
AuthNtlmMessage->length);
|
|
}
|
|
else if (AuthNtlmMessage->type == 2)
|
|
{
|
|
sspi_SecBufferFree(&context->ChallengeMessage);
|
|
|
|
if (!sspi_SecBufferAlloc(&context->ChallengeMessage, AuthNtlmMessage->length))
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
CopyMemory(context->ChallengeMessage.pvBuffer, AuthNtlmMessage->buffer,
|
|
AuthNtlmMessage->length);
|
|
}
|
|
else if (AuthNtlmMessage->type == 3)
|
|
{
|
|
sspi_SecBufferFree(&context->AuthenticateMessage);
|
|
|
|
if (!sspi_SecBufferAlloc(&context->AuthenticateMessage, AuthNtlmMessage->length))
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
CopyMemory(context->AuthenticateMessage.pvBuffer, AuthNtlmMessage->buffer,
|
|
AuthNtlmMessage->length);
|
|
}
|
|
|
|
return SEC_E_OK;
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_TIMESTAMP)
|
|
{
|
|
SecPkgContext_AuthNtlmTimestamp* AuthNtlmTimestamp =
|
|
(SecPkgContext_AuthNtlmTimestamp*)pBuffer;
|
|
|
|
if (cbBuffer < sizeof(SecPkgContext_AuthNtlmTimestamp))
|
|
return SEC_E_INVALID_PARAMETER;
|
|
|
|
if (AuthNtlmTimestamp->ChallengeOrResponse)
|
|
CopyMemory(context->ChallengeTimestamp, AuthNtlmTimestamp->Timestamp, 8);
|
|
else
|
|
CopyMemory(context->Timestamp, AuthNtlmTimestamp->Timestamp, 8);
|
|
|
|
return SEC_E_OK;
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_CLIENT_CHALLENGE)
|
|
{
|
|
SecPkgContext_AuthNtlmClientChallenge* AuthNtlmClientChallenge =
|
|
(SecPkgContext_AuthNtlmClientChallenge*)pBuffer;
|
|
|
|
if (cbBuffer < sizeof(SecPkgContext_AuthNtlmClientChallenge))
|
|
return SEC_E_INVALID_PARAMETER;
|
|
|
|
CopyMemory(context->ClientChallenge, AuthNtlmClientChallenge->ClientChallenge, 8);
|
|
return SEC_E_OK;
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_SERVER_CHALLENGE)
|
|
{
|
|
SecPkgContext_AuthNtlmServerChallenge* AuthNtlmServerChallenge =
|
|
(SecPkgContext_AuthNtlmServerChallenge*)pBuffer;
|
|
|
|
if (cbBuffer < sizeof(SecPkgContext_AuthNtlmServerChallenge))
|
|
return SEC_E_INVALID_PARAMETER;
|
|
|
|
CopyMemory(context->ServerChallenge, AuthNtlmServerChallenge->ServerChallenge, 8);
|
|
return SEC_E_OK;
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_HASH_CB)
|
|
{
|
|
context->HashCallback = (psPeerComputeNtlmHash)pBuffer;
|
|
return SEC_E_OK;
|
|
}
|
|
else if (ulAttribute == SECPKG_ATTR_AUTH_NTLM_HASH_CB_DATA)
|
|
{
|
|
context->HashCallbackArg = pBuffer;
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
WLog_ERR(TAG, "[%s]: TODO: Implement ulAttribute=%08" PRIx32, __FUNCTION__, ulAttribute);
|
|
return SEC_E_UNSUPPORTED_FUNCTION;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_SetContextAttributesA(PCtxtHandle phContext,
|
|
ULONG ulAttribute, void* pBuffer,
|
|
ULONG cbBuffer)
|
|
{
|
|
return ntlm_SetContextAttributesW(phContext, ulAttribute, pBuffer, cbBuffer);
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_RevertSecurityContext(PCtxtHandle phContext)
|
|
{
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_EncryptMessage(PCtxtHandle phContext, ULONG fQOP,
|
|
PSecBufferDesc pMessage, ULONG MessageSeqNo)
|
|
{
|
|
ULONG index;
|
|
size_t length;
|
|
void* data;
|
|
UINT32 SeqNo;
|
|
UINT32 value;
|
|
BYTE digest[WINPR_MD5_DIGEST_LENGTH];
|
|
BYTE checksum[8];
|
|
BYTE* signature;
|
|
ULONG version = 1;
|
|
WINPR_HMAC_CTX* hmac;
|
|
NTLM_CONTEXT* context;
|
|
PSecBuffer data_buffer = NULL;
|
|
PSecBuffer signature_buffer = NULL;
|
|
SeqNo = MessageSeqNo;
|
|
context = (NTLM_CONTEXT*)sspi_SecureHandleGetLowerPointer(phContext);
|
|
|
|
for (index = 0; index < pMessage->cBuffers; index++)
|
|
{
|
|
SecBuffer* cur = &pMessage->pBuffers[index];
|
|
|
|
if (cur->BufferType & SECBUFFER_DATA)
|
|
data_buffer = cur;
|
|
else if (cur->BufferType & SECBUFFER_TOKEN)
|
|
signature_buffer = cur;
|
|
}
|
|
|
|
if (!data_buffer)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (!signature_buffer)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
/* Copy original data buffer */
|
|
length = data_buffer->cbBuffer;
|
|
data = malloc(length);
|
|
|
|
if (!data)
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
CopyMemory(data, data_buffer->pvBuffer, length);
|
|
/* Compute the HMAC-MD5 hash of ConcatenationOf(seq_num,data) using the client signing key */
|
|
hmac = winpr_HMAC_New();
|
|
|
|
if (hmac &&
|
|
winpr_HMAC_Init(hmac, WINPR_MD_MD5, context->SendSigningKey, WINPR_MD5_DIGEST_LENGTH))
|
|
{
|
|
Data_Write_UINT32(&value, SeqNo);
|
|
winpr_HMAC_Update(hmac, (void*)&value, 4);
|
|
winpr_HMAC_Update(hmac, (void*)data, length);
|
|
winpr_HMAC_Final(hmac, digest, WINPR_MD5_DIGEST_LENGTH);
|
|
winpr_HMAC_Free(hmac);
|
|
}
|
|
else
|
|
{
|
|
winpr_HMAC_Free(hmac);
|
|
free(data);
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
}
|
|
|
|
/* Encrypt message using with RC4, result overwrites original buffer */
|
|
if ((data_buffer->BufferType & SECBUFFER_READONLY) == 0)
|
|
{
|
|
if (context->confidentiality)
|
|
winpr_RC4_Update(context->SendRc4Seal, length, (BYTE*)data,
|
|
(BYTE*)data_buffer->pvBuffer);
|
|
else
|
|
CopyMemory(data_buffer->pvBuffer, data, length);
|
|
}
|
|
|
|
#ifdef WITH_DEBUG_NTLM
|
|
WLog_DBG(TAG, "Data Buffer (length = %d)", length);
|
|
winpr_HexDump(TAG, WLOG_DEBUG, data, length);
|
|
WLog_DBG(TAG, "Encrypted Data Buffer (length = %" PRIu32 ")", data_buffer->cbBuffer);
|
|
winpr_HexDump(TAG, WLOG_DEBUG, data_buffer->pvBuffer, data_buffer->cbBuffer);
|
|
#endif
|
|
free(data);
|
|
/* RC4-encrypt first 8 bytes of digest */
|
|
winpr_RC4_Update(context->SendRc4Seal, 8, digest, checksum);
|
|
if ((signature_buffer->BufferType & SECBUFFER_READONLY) == 0)
|
|
{
|
|
signature = (BYTE*)signature_buffer->pvBuffer;
|
|
/* Concatenate version, ciphertext and sequence number to build signature */
|
|
Data_Write_UINT32(signature, version);
|
|
CopyMemory(&signature[4], (void*)checksum, 8);
|
|
Data_Write_UINT32(&signature[12], SeqNo);
|
|
}
|
|
context->SendSeqNum++;
|
|
#ifdef WITH_DEBUG_NTLM
|
|
WLog_DBG(TAG, "Signature (length = %" PRIu32 ")", signature_buffer->cbBuffer);
|
|
winpr_HexDump(TAG, WLOG_DEBUG, signature_buffer->pvBuffer, signature_buffer->cbBuffer);
|
|
#endif
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_DecryptMessage(PCtxtHandle phContext, PSecBufferDesc pMessage,
|
|
ULONG MessageSeqNo, PULONG pfQOP)
|
|
{
|
|
ULONG index;
|
|
size_t length;
|
|
void* data;
|
|
UINT32 SeqNo;
|
|
UINT32 value;
|
|
BYTE digest[WINPR_MD5_DIGEST_LENGTH];
|
|
BYTE checksum[8];
|
|
UINT32 version = 1;
|
|
WINPR_HMAC_CTX* hmac;
|
|
NTLM_CONTEXT* context;
|
|
BYTE expected_signature[WINPR_MD5_DIGEST_LENGTH];
|
|
PSecBuffer data_buffer = NULL;
|
|
PSecBuffer signature_buffer = NULL;
|
|
SeqNo = (UINT32)MessageSeqNo;
|
|
context = (NTLM_CONTEXT*)sspi_SecureHandleGetLowerPointer(phContext);
|
|
|
|
for (index = 0; index < pMessage->cBuffers; index++)
|
|
{
|
|
if (pMessage->pBuffers[index].BufferType == SECBUFFER_DATA)
|
|
data_buffer = &pMessage->pBuffers[index];
|
|
else if (pMessage->pBuffers[index].BufferType == SECBUFFER_TOKEN)
|
|
signature_buffer = &pMessage->pBuffers[index];
|
|
}
|
|
|
|
if (!data_buffer)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
if (!signature_buffer)
|
|
return SEC_E_INVALID_TOKEN;
|
|
|
|
/* Copy original data buffer */
|
|
length = data_buffer->cbBuffer;
|
|
data = malloc(length);
|
|
|
|
if (!data)
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
|
|
CopyMemory(data, data_buffer->pvBuffer, length);
|
|
|
|
/* Decrypt message using with RC4, result overwrites original buffer */
|
|
|
|
if (context->confidentiality)
|
|
winpr_RC4_Update(context->RecvRc4Seal, length, (BYTE*)data, (BYTE*)data_buffer->pvBuffer);
|
|
else
|
|
CopyMemory(data_buffer->pvBuffer, data, length);
|
|
|
|
/* Compute the HMAC-MD5 hash of ConcatenationOf(seq_num,data) using the client signing key */
|
|
hmac = winpr_HMAC_New();
|
|
|
|
if (hmac &&
|
|
winpr_HMAC_Init(hmac, WINPR_MD_MD5, context->RecvSigningKey, WINPR_MD5_DIGEST_LENGTH))
|
|
{
|
|
Data_Write_UINT32(&value, SeqNo);
|
|
winpr_HMAC_Update(hmac, (void*)&value, 4);
|
|
winpr_HMAC_Update(hmac, (void*)data_buffer->pvBuffer, data_buffer->cbBuffer);
|
|
winpr_HMAC_Final(hmac, digest, WINPR_MD5_DIGEST_LENGTH);
|
|
winpr_HMAC_Free(hmac);
|
|
}
|
|
else
|
|
{
|
|
winpr_HMAC_Free(hmac);
|
|
free(data);
|
|
return SEC_E_INSUFFICIENT_MEMORY;
|
|
}
|
|
|
|
#ifdef WITH_DEBUG_NTLM
|
|
WLog_DBG(TAG, "Encrypted Data Buffer (length = %d)", length);
|
|
winpr_HexDump(TAG, WLOG_DEBUG, data, length);
|
|
WLog_DBG(TAG, "Data Buffer (length = %" PRIu32 ")", data_buffer->cbBuffer);
|
|
winpr_HexDump(TAG, WLOG_DEBUG, data_buffer->pvBuffer, data_buffer->cbBuffer);
|
|
#endif
|
|
free(data);
|
|
/* RC4-encrypt first 8 bytes of digest */
|
|
winpr_RC4_Update(context->RecvRc4Seal, 8, digest, checksum);
|
|
/* Concatenate version, ciphertext and sequence number to build signature */
|
|
Data_Write_UINT32(expected_signature, version);
|
|
CopyMemory(&expected_signature[4], (void*)checksum, 8);
|
|
Data_Write_UINT32(&expected_signature[12], SeqNo);
|
|
context->RecvSeqNum++;
|
|
|
|
if (memcmp(signature_buffer->pvBuffer, expected_signature, 16) != 0)
|
|
{
|
|
/* signature verification failed! */
|
|
WLog_ERR(TAG, "signature verification failed, something nasty is going on!");
|
|
#ifdef WITH_DEBUG_NTLM
|
|
WLog_ERR(TAG, "Expected Signature:");
|
|
winpr_HexDump(TAG, WLOG_ERROR, expected_signature, 16);
|
|
WLog_ERR(TAG, "Actual Signature:");
|
|
winpr_HexDump(TAG, WLOG_ERROR, (BYTE*)signature_buffer->pvBuffer, 16);
|
|
#endif
|
|
return SEC_E_MESSAGE_ALTERED;
|
|
}
|
|
|
|
return SEC_E_OK;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_MakeSignature(PCtxtHandle phContext, ULONG fQOP,
|
|
PSecBufferDesc pMessage, ULONG MessageSeqNo)
|
|
{
|
|
WLog_ERR(TAG, "[%s]: TODO: Implement", __FUNCTION__);
|
|
return SEC_E_UNSUPPORTED_FUNCTION;
|
|
}
|
|
|
|
static SECURITY_STATUS SEC_ENTRY ntlm_VerifySignature(PCtxtHandle phContext,
|
|
PSecBufferDesc pMessage, ULONG MessageSeqNo,
|
|
PULONG pfQOP)
|
|
{
|
|
WLog_ERR(TAG, "[%s]: TODO: Implement", __FUNCTION__);
|
|
return SEC_E_UNSUPPORTED_FUNCTION;
|
|
}
|
|
|
|
const SecurityFunctionTableA NTLM_SecurityFunctionTableA = {
|
|
1, /* dwVersion */
|
|
NULL, /* EnumerateSecurityPackages */
|
|
ntlm_QueryCredentialsAttributesA, /* QueryCredentialsAttributes */
|
|
ntlm_AcquireCredentialsHandleA, /* AcquireCredentialsHandle */
|
|
ntlm_FreeCredentialsHandle, /* FreeCredentialsHandle */
|
|
NULL, /* Reserved2 */
|
|
ntlm_InitializeSecurityContextA, /* InitializeSecurityContext */
|
|
ntlm_AcceptSecurityContext, /* AcceptSecurityContext */
|
|
ntlm_CompleteAuthToken, /* CompleteAuthToken */
|
|
ntlm_DeleteSecurityContext, /* DeleteSecurityContext */
|
|
NULL, /* ApplyControlToken */
|
|
ntlm_QueryContextAttributesA, /* QueryContextAttributes */
|
|
ntlm_ImpersonateSecurityContext, /* ImpersonateSecurityContext */
|
|
ntlm_RevertSecurityContext, /* RevertSecurityContext */
|
|
ntlm_MakeSignature, /* MakeSignature */
|
|
ntlm_VerifySignature, /* VerifySignature */
|
|
NULL, /* FreeContextBuffer */
|
|
NULL, /* QuerySecurityPackageInfo */
|
|
NULL, /* Reserved3 */
|
|
NULL, /* Reserved4 */
|
|
NULL, /* ExportSecurityContext */
|
|
NULL, /* ImportSecurityContext */
|
|
NULL, /* AddCredentials */
|
|
NULL, /* Reserved8 */
|
|
NULL, /* QuerySecurityContextToken */
|
|
ntlm_EncryptMessage, /* EncryptMessage */
|
|
ntlm_DecryptMessage, /* DecryptMessage */
|
|
ntlm_SetContextAttributesA, /* SetContextAttributes */
|
|
};
|
|
|
|
const SecurityFunctionTableW NTLM_SecurityFunctionTableW = {
|
|
1, /* dwVersion */
|
|
NULL, /* EnumerateSecurityPackages */
|
|
ntlm_QueryCredentialsAttributesW, /* QueryCredentialsAttributes */
|
|
ntlm_AcquireCredentialsHandleW, /* AcquireCredentialsHandle */
|
|
ntlm_FreeCredentialsHandle, /* FreeCredentialsHandle */
|
|
NULL, /* Reserved2 */
|
|
ntlm_InitializeSecurityContextW, /* InitializeSecurityContext */
|
|
ntlm_AcceptSecurityContext, /* AcceptSecurityContext */
|
|
ntlm_CompleteAuthToken, /* CompleteAuthToken */
|
|
ntlm_DeleteSecurityContext, /* DeleteSecurityContext */
|
|
NULL, /* ApplyControlToken */
|
|
ntlm_QueryContextAttributesW, /* QueryContextAttributes */
|
|
ntlm_ImpersonateSecurityContext, /* ImpersonateSecurityContext */
|
|
ntlm_RevertSecurityContext, /* RevertSecurityContext */
|
|
ntlm_MakeSignature, /* MakeSignature */
|
|
ntlm_VerifySignature, /* VerifySignature */
|
|
NULL, /* FreeContextBuffer */
|
|
NULL, /* QuerySecurityPackageInfo */
|
|
NULL, /* Reserved3 */
|
|
NULL, /* Reserved4 */
|
|
NULL, /* ExportSecurityContext */
|
|
NULL, /* ImportSecurityContext */
|
|
NULL, /* AddCredentials */
|
|
NULL, /* Reserved8 */
|
|
NULL, /* QuerySecurityContextToken */
|
|
ntlm_EncryptMessage, /* EncryptMessage */
|
|
ntlm_DecryptMessage, /* DecryptMessage */
|
|
ntlm_SetContextAttributesA, /* SetContextAttributes */
|
|
};
|
|
|
|
const SecPkgInfoA NTLM_SecPkgInfoA = {
|
|
0x00082B37, /* fCapabilities */
|
|
1, /* wVersion */
|
|
0x000A, /* wRPCID */
|
|
0x00000B48, /* cbMaxToken */
|
|
"NTLM", /* Name */
|
|
"NTLM Security Package" /* Comment */
|
|
};
|
|
|
|
static WCHAR NTLM_SecPkgInfoW_Name[] = { 'N', 'T', 'L', 'M', '\0' };
|
|
|
|
static WCHAR NTLM_SecPkgInfoW_Comment[] = {
|
|
'N', 'T', 'L', 'M', ' ', 'S', 'e', 'c', 'u', 'r', 'i',
|
|
't', 'y', ' ', 'P', 'a', 'c', 'k', 'a', 'g', 'e', '\0'
|
|
};
|
|
|
|
const SecPkgInfoW NTLM_SecPkgInfoW = {
|
|
0x00082B37, /* fCapabilities */
|
|
1, /* wVersion */
|
|
0x000A, /* wRPCID */
|
|
0x00000B48, /* cbMaxToken */
|
|
NTLM_SecPkgInfoW_Name, /* Name */
|
|
NTLM_SecPkgInfoW_Comment /* Comment */
|
|
};
|