util.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	_NTLM_CLIENT_BUFFER

Typedefs
typedef struct _NTLM_CLIENT_BUFFER	NTLM_CLIENT_BUFFER
typedef struct _NTLM_CLIENT_BUFFER *	PNTLM_CLIENT_BUFFER

Functions
bool	NtlmUStrAlloc (_Out_ PUNICODE_STRING Dst, _In_ UINT16 SizeInBytes, _In_ UINT16 InitLength)
VOID	NtlmUStrFree (_In_ PUNICODE_STRING String)
bool	NtlmUStrWriteToStruct (_In_ PVOID DataStart, _In_ ULONG DataSize, _Out_ PUNICODE_STRING DstData, _In_ const PUNICODE_STRING SrcData, _In_ OUT PBYTE *AbsoluteOffsetPtr, _In_ bool TerminateWith0)
bool	NtlmFixupAndValidateUStr (_Inout_ PUNICODE_STRING String, _In_ ULONG_PTR FixupOffset)
bool	NtlmFixupAStr (_Inout_ PSTRING String, _In_ ULONG_PTR FixupOffset)
NTSTATUS	NtlmAllocateClientBuffer (_In_ PLSA_CLIENT_REQUEST ClientRequest, _In_ ULONG BufferLength, _Inout_ PNTLM_CLIENT_BUFFER Buffer)
NTSTATUS	NtlmCopyToClientBuffer (_In_ PLSA_CLIENT_REQUEST ClientRequest, _In_ ULONG BufferLength, _In_ OUT PNTLM_CLIENT_BUFFER Buffer)
VOID	NtlmFreeClientBuffer (_In_ PLSA_CLIENT_REQUEST ClientRequest, _In_ bool FreeClientBuffer, _Inout_ PNTLM_CLIENT_BUFFER Buffer)

Typedef Documentation

NTLM_CLIENT_BUFFER

typedef struct _NTLM_CLIENT_BUFFER NTLM_CLIENT_BUFFER

PNTLM_CLIENT_BUFFER

typedef struct _NTLM_CLIENT_BUFFER * PNTLM_CLIENT_BUFFER

Function Documentation

NtlmAllocateClientBuffer()

NTSTATUS NtlmAllocateClientBuffer	(	_In_ PLSA_CLIENT_REQUEST	ClientRequest,
		_In_ ULONG	BufferLength,
		_Inout_ PNTLM_CLIENT_BUFFER	Buffer
	)

Definition at line 264 of file util.c.

{
    NTSTATUS Status = STATUS_SUCCESS;

    if (!Buffer)
        return STATUS_NO_MEMORY;

    Buffer->LocalBuffer = NtlmAllocate(BufferLength, false);
    if (!Buffer->LocalBuffer)
        return STATUS_NO_MEMORY;

    if ((HANDLE)ClientRequest == INVALID_HANDLE_VALUE)
    {
        Buffer->ClientBaseAddress = Buffer->LocalBuffer;
        //if (!ClientBaseAddress)
        //    return STATUS_INSUFFICIENT_RESOURCES;
    }
    else
    {
        Status = DispatchTable.AllocateClientBuffer(ClientRequest,
                                                    BufferLength,
                                                    &Buffer->ClientBaseAddress);
        if (!NT_SUCCESS(Status))
        {
            NtlmFree(Buffer->LocalBuffer, false);
            Buffer->LocalBuffer = NULL;
        }
        //FIXME: Maybe we have to free ClientBaseAddress if something
        //       goes wrong ...? I'm not sure about that ...
    }
    return Status;
}

Referenced by BuildLm20LogonProfileBuffer().

NtlmCopyToClientBuffer()

NTSTATUS NtlmCopyToClientBuffer	(	_In_ PLSA_CLIENT_REQUEST	ClientRequest,
		_In_ ULONG	BufferLength,
		_In_ OUT PNTLM_CLIENT_BUFFER	Buffer
	)

NtlmFixupAndValidateUStr()

bool NtlmFixupAndValidateUStr	(	_Inout_ PUNICODE_STRING	String,
		_In_ ULONG_PTR	FixupOffset
	)

Definition at line 223 of file util.c.

{
    NTSTATUS Status;

    if (String->Length)
    {
        String->Buffer = FIXUP_POINTER(String->Buffer, FixupOffset);
        String->MaximumLength = String->Length;
    }
    else
    {
        String->Buffer = NULL;
        String->MaximumLength = 0;
    }

    Status = RtlValidateUnicodeString(0, String);
    return NT_SUCCESS(Status);
}

Referenced by LsaApLogonUserEx2_Network().

NtlmFixupAStr()

bool NtlmFixupAStr	(	_Inout_ PSTRING	String,
		_In_ ULONG_PTR	FixupOffset
	)

Definition at line 245 of file util.c.

{
    if (String->Length)
    {
        String->Buffer = (PCHAR)FIXUP_POINTER(String->Buffer, FixupOffset);
        String->MaximumLength = String->Length;
    }
    else
    {
        String->Buffer = NULL;
        String->MaximumLength = 0;
    }

    return true;
}

Referenced by LsaApLogonUserEx2_Network().

NtlmFreeClientBuffer()

VOID NtlmFreeClientBuffer	(	_In_ PLSA_CLIENT_REQUEST	ClientRequest,
		_In_ bool	FreeClientBuffer,
		_Inout_ PNTLM_CLIENT_BUFFER	Buffer
	)

Definition at line 335 of file util.c.

{
    if (!Buffer->ClientBaseAddress)
        return;

    if ((HANDLE)ClientRequest == INVALID_HANDLE_VALUE)
    {
        if (Buffer->ClientBaseAddress != Buffer->LocalBuffer)
        {
            ERR("Buffer->ClientBaseAddress != Buffer->LocalBuffer (something must be wrong!)\n");
            return;
        }
        // LocalBuffer and ClientBaseAddress is the same
        // so we have only to free it if FreeClientBuffer is TRUE.
        Buffer->LocalBuffer = NULL;
        if (FreeClientBuffer)
        {
            NtlmFree(Buffer->ClientBaseAddress, false);
            Buffer->ClientBaseAddress = NULL;
        }
    }
    else
    {
        NtlmFree(Buffer->LocalBuffer, false);
        Buffer->LocalBuffer = NULL;
        if (FreeClientBuffer)
            DispatchTable.FreeClientBuffer(ClientRequest, Buffer->ClientBaseAddress);
        Buffer->ClientBaseAddress = NULL;
    }
}

Referenced by BuildLm20LogonProfileBuffer().

NtlmUStrAlloc()

bool NtlmUStrAlloc	(	_Out_ PUNICODE_STRING	Dst,
		_In_ UINT16	SizeInBytes,
		_In_ UINT16	InitLength
	)

NtlmUStrFree()

VOID NtlmUStrFree

(

_In_ PUNICODE_STRING

String

)

Definition at line 115 of file util.c.

{
    if (String == NULL || String->Buffer == NULL || String->MaximumLength == 0)
        return;

    NtlmFree(String->Buffer, false);
    String->Buffer = NULL;
    String->MaximumLength = 0;
}

Referenced by BuildLm20LogonProfileBuffer().

NtlmUStrWriteToStruct()

bool NtlmUStrWriteToStruct	(	_In_ PVOID	DataStart,
		_In_ ULONG	DataSize,
		_Out_ PUNICODE_STRING	DstData,
		_In_ const PUNICODE_STRING	SrcData,
		_In_ OUT PBYTE *	AbsoluteOffsetPtr,
		_In_ bool	TerminateWith0
	)