ordering.c File Reference

#include <ntifs.h>
#include <ndk/rtlfuncs.h>
#include "arbiter.h"
#include <debug.h>

Include dependency graph for ordering.c:

Go to the source code of this file.

Macros
#define	NDEBUG
#define	ARBITER_ORDERING_GRANULARITY   16
#define	ARBITER_ORDERING_LIMIT   1024
#define	ARBP_POLICY_ROOT   L"\\Registry\\Machine\\System\\CurrentControlSet\\Control\\Arbiters\\"
#define	ARBP_LONGEST_SUBKEY   L"ReservedResources"

Typedefs
typedef VOID(NTAPI *	PARB_RANGE_CALLBACK) (_In_ PARBITER_INSTANCE Arbiter, _In_opt_ PVOID Context, _In_ ULONGLONG Start, _In_ ULONGLONG End)

Functions
static NTSTATUS	ArbpCreateOrderingList (_Out_ PARBITER_ORDERING_LIST OrderingList)
VOID NTAPI	ArbiterLibFreeOrderingList (_Inout_ PARBITER_ORDERING_LIST OrderingList)
NTSTATUS NTAPI	ArbiterLibAddOrdering (_Inout_ PARBITER_ORDERING_LIST OrderingList, _In_ UINT64 Start, _In_ UINT64 End)
static NTSTATUS	ArbpOpenPolicySubkey (_In_ PCWSTR SubkeyName, _Out_ PHANDLE KeyHandle)
static NTSTATUS	ArbpReadPolicyValue (_In_ HANDLE KeyHandle, _In_ PCWSTR ValueName, _Out_ PKEY_VALUE_FULL_INFORMATION *Value)
static BOOLEAN	ArbpDescriptorMatchesArbiter (_In_ PARBITER_INSTANCE Arbiter, _In_ UCHAR DescriptorType)
static NTSTATUS	ArbpForEachRegistryRange (_In_ PARBITER_INSTANCE Arbiter, _In_ PCWSTR Subkey, _In_ PCWSTR ValueName, _In_ PARB_RANGE_CALLBACK Callback, _In_opt_ PVOID Context)
static VOID NTAPI	ArbpAddOrderingCallback (_In_ PARBITER_INSTANCE Arbiter, _In_opt_ PVOID Context, _In_ ULONGLONG Start, _In_ ULONGLONG End)
NTSTATUS NTAPI	ArbiterLibDefaultAssignmentOrdering (_Inout_ PARBITER_INSTANCE Arbiter, _In_ PCWSTR AllocationOrderName, _In_ PCWSTR ReservedResourcesName, _In_opt_ PARB_TRANSLATE_ORDERING TranslateOrderingFunction)

Macro Definition Documentation

ARBITER_ORDERING_GRANULARITY

#define ARBITER_ORDERING_GRANULARITY   16

Definition at line 17 of file ordering.c.

ARBITER_ORDERING_LIMIT

#define ARBITER_ORDERING_LIMIT   1024

Definition at line 18 of file ordering.c.

ARBP_LONGEST_SUBKEY

#define ARBP_LONGEST_SUBKEY   L"ReservedResources"

Definition at line 21 of file ordering.c.

ARBP_POLICY_ROOT

#define ARBP_POLICY_ROOT   L"\\Registry\\Machine\\System\\CurrentControlSet\\Control\\Arbiters\\"

Definition at line 20 of file ordering.c.

NDEBUG

#define NDEBUG

Definition at line 14 of file ordering.c.

Typedef Documentation

PARB_RANGE_CALLBACK

typedef VOID(NTAPI * PARB_RANGE_CALLBACK) (_In_ PARBITER_INSTANCE Arbiter, _In_opt_ PVOID Context, _In_ ULONGLONG Start, _In_ ULONGLONG End)

Definition at line 123 of file ordering.c.

Function Documentation

ArbiterLibAddOrdering()

NTSTATUS NTAPI ArbiterLibAddOrdering	(	_Inout_ PARBITER_ORDERING_LIST	OrderingList,
		_In_ UINT64	Start,
		_In_ UINT64	End
	)

Definition at line 70 of file ordering.c.

{
    PAGED_CODE();
 
    if (End < Start)
        return STATUS_INVALID_PARAMETER;
 
    /* Double the array capacity when it fills up, up to the sanity cap. */
    if (OrderingList->Count == OrderingList->Maximum)
    {
        ULONG NewMaximum;
        PARBITER_ORDERING NewArray;
 
        NewMaximum = OrderingList->Maximum ? (ULONG)OrderingList->Maximum * 2
                                           : ARBITER_ORDERING_GRANULARITY;
        if (NewMaximum > ARBITER_ORDERING_LIMIT)
            return STATUS_INSUFFICIENT_RESOURCES;
 
        NewArray = ExAllocatePoolWithTag(PagedPool,
                                         NewMaximum * sizeof(ARBITER_ORDERING),
                                         TAG_ARBITER);
        if (NewArray == NULL)
            return STATUS_INSUFFICIENT_RESOURCES;
 
        if (OrderingList->Orderings != NULL)
        {
            RtlCopyMemory(NewArray, OrderingList->Orderings,
                          OrderingList->Count * sizeof(ARBITER_ORDERING));
            ExFreePoolWithTag(OrderingList->Orderings, TAG_ARBITER);
        }
 
        OrderingList->Maximum = (UINT16)NewMaximum;
        OrderingList->Orderings = NewArray;
    }
 
    OrderingList->Orderings[OrderingList->Count].Start = Start;
    OrderingList->Orderings[OrderingList->Count].End = End;
    OrderingList->Count++;
    return STATUS_SUCCESS;
}

Referenced by ArbiterLibDefaultAssignmentOrdering(), and ArbpAddOrderingCallback().

ArbiterLibDefaultAssignmentOrdering()

NTSTATUS NTAPI ArbiterLibDefaultAssignmentOrdering	(	_Inout_ PARBITER_INSTANCE	Arbiter,
		_In_ PCWSTR	AllocationOrderName,
		_In_ PCWSTR	ReservedResourcesName,
		_In_opt_ PARB_TRANSLATE_ORDERING	TranslateOrderingFunction
	)

Definition at line 328 of file ordering.c.

{
    NTSTATUS Status;
 
    PAGED_CODE();
 
    UNREFERENCED_PARAMETER(ReservedResourcesName);
    UNREFERENCED_PARAMETER(TranslateOrderingFunction);
 
    ArbiterLibFreeOrderingList(&Arbiter->OrderingList);
    ArbiterLibFreeOrderingList(&Arbiter->ReservedList);
 
    Status = ArbpCreateOrderingList(&Arbiter->OrderingList);
    if (!NT_SUCCESS(Status))
        return Status;
    Status = ArbpCreateOrderingList(&Arbiter->ReservedList);
    if (!NT_SUCCESS(Status))
    {
        ArbiterLibFreeOrderingList(&Arbiter->OrderingList);
        return Status;
    }
 
    ArbpForEachRegistryRange(Arbiter, L"AllocationOrder", AllocationOrderName,
                             ArbpAddOrderingCallback, NULL);
 
    return ArbiterLibAddOrdering(&Arbiter->OrderingList, 0, 0xFFFFFFFFFFFFFFFFULL);
}

Referenced by ArbiterLibInitializeInstance(), and ario_ApplyBrokenVideoHack().

ArbiterLibFreeOrderingList()

VOID NTAPI ArbiterLibFreeOrderingList

(

_Inout_ PARBITER_ORDERING_LIST

OrderingList

)

Definition at line 52 of file ordering.c.

{
    PAGED_CODE();
 
    if (OrderingList->Orderings != NULL)
        ExFreePoolWithTag(OrderingList->Orderings, TAG_ARBITER);
 
    OrderingList->Orderings = NULL;
    OrderingList->Count = 0;
    OrderingList->Maximum = 0;
}

Referenced by ArbiterLibDefaultAssignmentOrdering(), ArbiterLibDeleteInstance(), and ario_ApplyBrokenVideoHack().

ArbpAddOrderingCallback()

static VOID NTAPI ArbpAddOrderingCallback ( _In_ PARBITER_INSTANCE  Arbiter, _In_opt_ PVOID  Context, _In_ ULONGLONG  Start, _In_ ULONGLONG  End  )

static

Definition at line 305 of file ordering.c.

{
    PAGED_CODE();
    UNREFERENCED_PARAMETER(Context);
    ArbiterLibAddOrdering(&Arbiter->OrderingList, Start, End);
}

Referenced by ArbiterLibDefaultAssignmentOrdering().

ArbpCreateOrderingList()

static NTSTATUS ArbpCreateOrderingList ( _Out_ PARBITER_ORDERING_LIST  OrderingList )

static

Definition at line 29 of file ordering.c.

{
    PAGED_CODE();
 
    OrderingList->Orderings = ExAllocatePoolWithTag(PagedPool,
                                                    ARBITER_ORDERING_GRANULARITY * sizeof(ARBITER_ORDERING),
                                                    TAG_ARBITER);
    if (OrderingList->Orderings == NULL)
    {
        OrderingList->Count = 0;
        OrderingList->Maximum = 0;
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    OrderingList->Count = 0;
    OrderingList->Maximum = ARBITER_ORDERING_GRANULARITY;
    return STATUS_SUCCESS;
}

Referenced by ArbiterLibDefaultAssignmentOrdering().

ArbpDescriptorMatchesArbiter()

static BOOLEAN ArbpDescriptorMatchesArbiter ( _In_ PARBITER_INSTANCE  Arbiter, _In_ UCHAR  DescriptorType  )

static

Definition at line 209 of file ordering.c.

{
    PAGED_CODE();
 
    if (DescriptorType == (UCHAR)Arbiter->ResourceType)
        return TRUE;
 
    /* Large-memory descriptors also satisfy the plain memory arbiter. */
    return (DescriptorType == CmResourceTypeMemoryLarge &&
            Arbiter->ResourceType == CmResourceTypeMemory);
}

Referenced by ArbpForEachRegistryRange().

ArbpForEachRegistryRange()

static NTSTATUS ArbpForEachRegistryRange ( _In_ PARBITER_INSTANCE  Arbiter, _In_ PCWSTR  Subkey, _In_ PCWSTR  ValueName, _In_ PARB_RANGE_CALLBACK  Callback, _In_opt_ PVOID  Context  )

static

Definition at line 230 of file ordering.c.

{
    HANDLE KeyHandle;
    PKEY_VALUE_FULL_INFORMATION Value = NULL;
    NTSTATUS Status;
 
    PAGED_CODE();
 
    if (!NT_SUCCESS(ArbpOpenPolicySubkey(Subkey, &KeyHandle)))
        return STATUS_SUCCESS;
 
    Status = ArbpReadPolicyValue(KeyHandle, ValueName, &Value);
 
    /* Follow a REG_SZ indirection to the named sibling value. */
    if (NT_SUCCESS(Status) && Value->Type == REG_SZ)
    {
        PWSTR TargetName = (PWSTR)((PUCHAR)Value + Value->DataOffset);
        ULONG TargetChars = Value->DataLength / sizeof(WCHAR);
        PKEY_VALUE_FULL_INFORMATION Target = NULL;
 
        if (TargetChars != 0 &&
            TargetName[TargetChars - 1] == UNICODE_NULL &&
            NT_SUCCESS(ArbpReadPolicyValue(KeyHandle, TargetName, &Target)))
        {
            ExFreePoolWithTag(Value, TAG_ARBITER);
            Value = Target;
        }
    }
 
    if (NT_SUCCESS(Status) &&
        Value->Type == REG_RESOURCE_REQUIREMENTS_LIST &&
        Value->DataLength >= sizeof(IO_RESOURCE_REQUIREMENTS_LIST))
    {
        PIO_RESOURCE_REQUIREMENTS_LIST Requirements;
        PIO_RESOURCE_LIST Alternative;
        PUCHAR DataEnd;
        ULONG Index;
 
        Requirements = (PIO_RESOURCE_REQUIREMENTS_LIST)((PUCHAR)Value + Value->DataOffset);
        DataEnd = (PUCHAR)Requirements + Value->DataLength;
        Alternative = &Requirements->List[0];
 
        for (Index = 0; Index < Alternative->Count; ++Index)
        {
            PIO_RESOURCE_DESCRIPTOR Descriptor = &Alternative->Descriptors[Index];
 
            /* Never trust the stored Count past the actual value data. */
            if ((PUCHAR)(Descriptor + 1) > DataEnd)
                break;
 
            if (ArbpDescriptorMatchesArbiter(Arbiter, Descriptor->Type))
            {
                Callback(Arbiter, Context,
                         (ULONGLONG)Descriptor->u.Generic.MinimumAddress.QuadPart,
                         (ULONGLONG)Descriptor->u.Generic.MaximumAddress.QuadPart);
            }
        }
    }
 
    if (Value != NULL)
        ExFreePoolWithTag(Value, TAG_ARBITER);
    ZwClose(KeyHandle);
    return STATUS_SUCCESS;
}

Referenced by ArbiterLibDefaultAssignmentOrdering().

ArbpOpenPolicySubkey()

static NTSTATUS ArbpOpenPolicySubkey ( _In_ PCWSTR  SubkeyName, _Out_ PHANDLE  KeyHandle  )

static

Definition at line 134 of file ordering.c.

{
    WCHAR PathBuffer[sizeof(ARBP_POLICY_ROOT ARBP_LONGEST_SUBKEY) / sizeof(WCHAR)];
    UNICODE_STRING KeyPath;
    OBJECT_ATTRIBUTES ObjectAttributes;
    NTSTATUS Status;
 
    PAGED_CODE();
 
    RtlInitEmptyUnicodeString(&KeyPath, PathBuffer, sizeof(PathBuffer));
    Status = RtlAppendUnicodeToString(&KeyPath, ARBP_POLICY_ROOT);
    if (NT_SUCCESS(Status))
        Status = RtlAppendUnicodeToString(&KeyPath, SubkeyName);
    if (!NT_SUCCESS(Status))
        return Status;
 
    InitializeObjectAttributes(&ObjectAttributes, &KeyPath,
                               OBJ_KERNEL_HANDLE | OBJ_CASE_INSENSITIVE, NULL, NULL);
    return ZwOpenKey(KeyHandle, KEY_READ, &ObjectAttributes);
}

Referenced by ArbpForEachRegistryRange().

ArbpReadPolicyValue()

static NTSTATUS ArbpReadPolicyValue ( _In_ HANDLE  KeyHandle, _In_ PCWSTR  ValueName, _Out_ PKEY_VALUE_FULL_INFORMATION *  Value  )

static

Definition at line 161 of file ordering.c.

{
    UNICODE_STRING NameString;
    PKEY_VALUE_FULL_INFORMATION Buffer;
    ULONG Size = 0;
    ULONG ResultLength = 0;
    NTSTATUS Status;
 
    PAGED_CODE();
 
    RtlInitUnicodeString(&NameString, ValueName);
 
    /* Size probe: only a buffer-size result means the value is readable. */
    Status = ZwQueryValueKey(KeyHandle, &NameString, KeyValueFullInformationAlign64,
                             NULL, 0, &Size);
    if (Status != STATUS_BUFFER_TOO_SMALL && Status != STATUS_BUFFER_OVERFLOW)
        return NT_SUCCESS(Status) ? STATUS_UNSUCCESSFUL : Status;
 
    Buffer = ExAllocatePoolWithTag(PagedPool, Size, TAG_ARBITER);
    if (Buffer == NULL)
        return STATUS_INSUFFICIENT_RESOURCES;
 
    Status = ZwQueryValueKey(KeyHandle, &NameString, KeyValueFullInformationAlign64,
                             Buffer, Size, &ResultLength);
 
    if (NT_SUCCESS(Status) &&
        ((ULONGLONG)Buffer->DataOffset + Buffer->DataLength > ResultLength))
    {
        Status = STATUS_UNSUCCESSFUL;
    }
 
    if (!NT_SUCCESS(Status))
    {
        ExFreePoolWithTag(Buffer, TAG_ARBITER);
        return Status;
    }
 
    *Value = Buffer;
    return STATUS_SUCCESS;
}

Referenced by ArbpForEachRegistryRange().