mirror of
https://github.com/reactos/reactos.git
synced 2024-09-12 22:02:18 +00:00
dfb776380d
[FLTMGR][KMTEST] Squash and push my local branch across to master as the patch is getting a bit large. This is still WIP and none of this code is run in ros yet, so don't fret if you see ugly/unfinished code or int3's dotted around. [FLTMGR] Improve loading/reg of filters and start to implement client connections - Implement handling of connections from clients - Implement closing of client ports - Add a basic message waiter queue using CSQ's (untested) - Hand off messages for the comms object to be handled by the comms file - Initialize the connection list - Add a registry file which will contain lib functions for accessing filter service entries - [KMTEST] Initial usermode support for testing FS mini-filters - Add base routines to wrap the win32 'Filter' APis - Add support routines to be used when testing FS filter drivers - Move KmtCreateService to a private routine so it can be shared with KmtFltCreateService - Completely untested at the mo, so likely contains bugs at this point - Add support for adding altitude and flags registry entries for minifilters - Allow minifilters to setup without requiring instance attach/detach callbacks - Add tests for FltRegisterFilter and FltUnregisterFilter and start to add associated tests
143 lines
4 KiB
C
143 lines
4 KiB
C
/*
|
|
* PROJECT: Filesystem Filter Manager
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: drivers/filters/fltmgr/Misc.c
|
|
* PURPOSE: Uncataloged functions
|
|
* PROGRAMMERS: Ged Murphy (gedmurphy@reactos.org)
|
|
*/
|
|
|
|
/* INCLUDES ******************************************************************/
|
|
|
|
#include "fltmgr.h"
|
|
#include "fltmgrint.h"
|
|
|
|
#define NDEBUG
|
|
#include <debug.h>
|
|
|
|
|
|
/* DATA *********************************************************************/
|
|
|
|
#define REG_SERVICES_KEY L"\\Registry\\Machine\\System\\CurrentControlSet\\Services\\"
|
|
#define REG_PATH_LENGTH 512
|
|
|
|
|
|
/* INTERNAL FUNCTIONS ******************************************************/
|
|
|
|
|
|
NTSTATUS
|
|
FltpOpenFilterServicesKey(
|
|
_In_ PFLT_FILTER Filter,
|
|
_In_ ACCESS_MASK DesiredAccess,
|
|
_In_opt_ PUNICODE_STRING SubKey,
|
|
_Out_ PHANDLE Handle)
|
|
{
|
|
OBJECT_ATTRIBUTES ObjectAttributes;
|
|
UNICODE_STRING ServicesKey;
|
|
UNICODE_STRING Path;
|
|
WCHAR Buffer[REG_PATH_LENGTH];
|
|
|
|
/* Setup a local buffer to hold the services key path */
|
|
Path.Length = 0;
|
|
Path.MaximumLength = REG_PATH_LENGTH;
|
|
Path.Buffer = Buffer;
|
|
|
|
/* Build up the serices key name */
|
|
RtlInitUnicodeString(&ServicesKey, REG_SERVICES_KEY);
|
|
RtlCopyUnicodeString(&Path, &ServicesKey);
|
|
RtlAppendUnicodeStringToString(&Path, &Filter->Name);
|
|
|
|
if (SubKey)
|
|
{
|
|
/* Tag on any child key */
|
|
RtlAppendUnicodeToString(&Path, L"\\");
|
|
RtlAppendUnicodeStringToString(&Path, SubKey);
|
|
}
|
|
|
|
InitializeObjectAttributes(&ObjectAttributes,
|
|
&Path,
|
|
OBJ_KERNEL_HANDLE | OBJ_CASE_INSENSITIVE,
|
|
NULL,
|
|
NULL);
|
|
|
|
/* Open and return the key handle param*/
|
|
return ZwOpenKey(Handle, DesiredAccess, &ObjectAttributes);
|
|
}
|
|
|
|
NTSTATUS
|
|
FltpReadRegistryValue(_In_ HANDLE KeyHandle,
|
|
_In_ PUNICODE_STRING ValueName,
|
|
_In_opt_ ULONG Type,
|
|
_Out_writes_bytes_(BufferSize) PVOID Buffer,
|
|
_In_ ULONG BufferSize,
|
|
_Out_opt_ PULONG BytesRequired)
|
|
{
|
|
PKEY_VALUE_PARTIAL_INFORMATION Value = NULL;
|
|
ULONG ValueLength = 0;
|
|
NTSTATUS Status;
|
|
|
|
PAGED_CODE();
|
|
|
|
/* Get the size of the buffer required to hold the string */
|
|
Status = ZwQueryValueKey(KeyHandle,
|
|
ValueName,
|
|
KeyValuePartialInformation,
|
|
NULL,
|
|
0,
|
|
&ValueLength);
|
|
if (Status != STATUS_BUFFER_TOO_SMALL && Status != STATUS_BUFFER_OVERFLOW)
|
|
{
|
|
return Status;
|
|
}
|
|
|
|
/* Allocate the buffer */
|
|
Value = (PKEY_VALUE_PARTIAL_INFORMATION)ExAllocatePoolWithTag(PagedPool,
|
|
ValueLength,
|
|
FM_TAG_TEMP_REGISTRY);
|
|
if (Value == NULL)
|
|
{
|
|
Status = STATUS_INSUFFICIENT_RESOURCES;
|
|
goto Quit;
|
|
}
|
|
|
|
/* Now read in the value */
|
|
Status = ZwQueryValueKey(KeyHandle,
|
|
ValueName,
|
|
KeyValuePartialInformation,
|
|
Value,
|
|
ValueLength,
|
|
&ValueLength);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
goto Quit;
|
|
}
|
|
|
|
/* Make sure we got the type expected */
|
|
if (Value->Type != Type)
|
|
{
|
|
Status = STATUS_INVALID_PARAMETER;
|
|
goto Quit;
|
|
}
|
|
|
|
if (BytesRequired)
|
|
{
|
|
*BytesRequired = Value->DataLength;
|
|
}
|
|
|
|
/* Make sure the caller buffer is big enough to hold the data */
|
|
if (!BufferSize || BufferSize < Value->DataLength)
|
|
{
|
|
Status = STATUS_BUFFER_TOO_SMALL;
|
|
goto Quit;
|
|
}
|
|
|
|
/* Copy the data into the caller buffer */
|
|
RtlCopyMemory(Buffer, Value->Data, Value->DataLength);
|
|
|
|
Quit:
|
|
|
|
if (Value)
|
|
ExFreePoolWithTag(Value, FM_TAG_TEMP_REGISTRY);
|
|
|
|
return Status;
|
|
}
|