NpfsHelpers.c File Reference

#include <kmt_test.h>
#include "npfs.h"

Include dependency graph for NpfsHelpers.c:

Go to the source code of this file.

Macros
#define	ok_eq_print_(value, expected, spec, FileAndLine)    KmtOk((value) == (expected), FileAndLine, #value " = " spec ", expected " spec "\n", value, expected)
#define	ok_eq_ulong_(value, expected)   ok_eq_print_(value, expected, "%lu", FileAndLine)
#define	ok_eq_ulonglong_(value, expected)   ok_eq_print_(value, expected, "%I64u", FileAndLine)
#define	ok_eq_ulongptr_(value, expected)   ok_eq_print_(value, (ULONG_PTR)(expected), "%lu", FileAndLine)
#define	ok_eq_hex_(value, expected)   ok_eq_print_(value, expected, "0x%08lx", FileAndLine)

Functions
NTSTATUS	NpCreatePipeEx (OUT PHANDLE ServerHandle, IN PCWSTR PipePath, IN ULONG ReadMode, IN ULONG CompletionMode, IN ULONG NamedPipeType, IN ULONG ShareAccess, IN ULONG MaximumInstances, IN ULONG InboundQuota, IN ULONG OutboundQuota, IN ACCESS_MASK DesiredAccess, IN ULONG Disposition, IN ULONG CreateOptions, IN PLARGE_INTEGER DefaultTimeout OPTIONAL)
NTSTATUS	NpCreatePipe (OUT PHANDLE ServerHandle, PCWSTR PipePath, ULONG ReadMode, ULONG CompletionMode, ULONG NamedPipeType, ULONG NamedPipeConfiguration, ULONG MaximumInstances, ULONG InboundQuota, ULONG OutboundQuota)
NTSTATUS	NpOpenPipeEx (OUT PHANDLE ClientHandle, IN PCWSTR PipePath, IN ACCESS_MASK DesiredAccess, IN ULONG ShareAccess, IN ULONG Disposition, IN ULONG CreateOptions)
NTSTATUS	NpOpenPipe (OUT PHANDLE ClientHandle, IN PCWSTR PipePath, IN ULONG NamedPipeConfiguration)
NTSTATUS	NpControlPipe (IN HANDLE ServerHandle, IN ULONG FsControlCode, IN PVOID InputBuffer, IN ULONG InputBufferLength)
NTSTATUS	NpWaitPipe (IN PCWSTR PipeName, IN PLARGE_INTEGER Timeout)
NTSTATUS	NpReadPipe (IN HANDLE PipeHandle, OUT PVOID Buffer, IN ULONG BufferSize, OUT PULONG_PTR BytesRead)
NTSTATUS	NpWritePipe (IN HANDLE PipeHandle, IN const VOID *Buffer, IN ULONG BufferSize, OUT PULONG_PTR BytesWritten)
static BOOLEAN	CheckBuffer (PVOID Buffer, SIZE_T Size, UCHAR Value)
VOID	NpCheckServerPipe_ (IN HANDLE ServerHandle, IN ULONG ReadMode, IN ULONG CompletionMode, IN ULONG NamedPipeType, IN ULONG NamedPipeConfiguration, IN ULONG MaximumInstances, IN ULONG CurrentInstances, IN ULONG InboundQuota, IN ULONG ReadDataAvailable, IN ULONG OutboundQuota, IN ULONG WriteQuotaAvailable, IN ULONG NamedPipeState, IN PCSTR FileAndLine)
VOID	NpCheckClientPipe_ (IN HANDLE ClientHandle, IN ULONG ReadMode, IN ULONG CompletionMode, IN ULONG NamedPipeType, IN ULONG NamedPipeConfiguration, IN ULONG MaximumInstances, IN ULONG CurrentInstances, IN ULONG InboundQuota, IN ULONG ReadDataAvailable, IN ULONG OutboundQuota, IN ULONG WriteQuotaAvailable, IN ULONG NamedPipeState, IN PCSTR FileAndLine)
VOID	NpQueryPipe_ (IN HANDLE PipeHandle, IN NTSTATUS ExpectedStatus, IN PCSTR FileAndLine)
static VOID NTAPI	PipeWorkerThread (IN PVOID ThreadContext)
VOID	StartWorkerThread (OUT PTHREAD_CONTEXT Context)
VOID	FinishWorkerThread (IN PTHREAD_CONTEXT Context)
BOOLEAN	WaitForWork (IN PTHREAD_CONTEXT Context, IN ULONG MilliSeconds)
BOOLEAN	TriggerWork (IN PTHREAD_CONTEXT Context, IN ULONG MilliSeconds)

Variables
static KSTART_ROUTINE	PipeWorkerThread

Macro Definition Documentation

ok_eq_hex_

#define ok_eq_hex_	(		value,
			expected
	)		ok_eq_print_(value, expected, "0x%08lx", FileAndLine)

Definition at line 443 of file NpfsHelpers.c.

ok_eq_print_

#define ok_eq_print_	(		value,
			expected,
			spec,
			FileAndLine
	)		KmtOk((value) == (expected), FileAndLine, #value " = " spec ", expected " spec "\n", value, expected)

Definition at line 434 of file NpfsHelpers.c.

ok_eq_ulong_

#define ok_eq_ulong_	(		value,
			expected
	)		ok_eq_print_(value, expected, "%lu", FileAndLine)

Definition at line 436 of file NpfsHelpers.c.

ok_eq_ulonglong_

#define ok_eq_ulonglong_	(		value,
			expected
	)		ok_eq_print_(value, expected, "%I64u", FileAndLine)

Definition at line 437 of file NpfsHelpers.c.

ok_eq_ulongptr_

#define ok_eq_ulongptr_	(		value,
			expected
	)		ok_eq_print_(value, (ULONG_PTR)(expected), "%lu", FileAndLine)

Definition at line 439 of file NpfsHelpers.c.

Function Documentation

CheckBuffer()

static BOOLEAN CheckBuffer ( PVOID  Buffer, SIZE_T  Size, UCHAR  Value  )

static

Definition at line 417 of file NpfsHelpers.c.

{
    PUCHAR Array = Buffer;
    SIZE_T i;
 
    for (i = 0; i < Size; i++)
        if (Array[i] != Value)
        {
            trace("Expected %x, found %x at offset %lu\n", Value, Array[i], (ULONG)i);
            return FALSE;
        }
    return TRUE;
}

Referenced by NpQueryPipe_().

FinishWorkerThread()

VOID FinishWorkerThread

(

IN PTHREAD_CONTEXT

Context

)

Definition at line 685 of file NpfsHelpers.c.

{
    KmtFinishThread(Context->Thread, &Context->ThreadDoneEvent);
}

Referenced by TestConnect(), and TestReadWrite().

NpCheckClientPipe_()

VOID NpCheckClientPipe_	(	IN HANDLE	ClientHandle,
		IN ULONG	ReadMode,
		IN ULONG	CompletionMode,
		IN ULONG	NamedPipeType,
		IN ULONG	NamedPipeConfiguration,
		IN ULONG	MaximumInstances,
		IN ULONG	CurrentInstances,
		IN ULONG	InboundQuota,
		IN ULONG	ReadDataAvailable,
		IN ULONG	OutboundQuota,
		IN ULONG	WriteQuotaAvailable,
		IN ULONG	NamedPipeState,
		IN PCSTR	FileAndLine
	)

Definition at line 520 of file NpfsHelpers.c.

{
    NTSTATUS Status;
    IO_STATUS_BLOCK IoStatusBlock;
    FILE_PIPE_INFORMATION PipeInfo;
    FILE_PIPE_LOCAL_INFORMATION PipeLocalInfo;
    FILE_PIPE_REMOTE_INFORMATION PipeRemoteInfo;
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    RtlFillMemory(&PipeInfo, sizeof(PipeInfo), 0x55);
    Status = ZwQueryInformationFile(ClientHandle,
                                    &IoStatusBlock,
                                    &PipeInfo,
                                    sizeof(PipeInfo),
                                    FilePipeInformation);
    ok_eq_hex_(Status, STATUS_SUCCESS);
    ok_eq_hex_(IoStatusBlock.Status, STATUS_SUCCESS);
    ok_eq_ulongptr_(IoStatusBlock.Information, sizeof(PipeInfo));
    ok_eq_ulong_(PipeInfo.ReadMode, ReadMode);
    ok_eq_ulong_(PipeInfo.CompletionMode, CompletionMode);
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    RtlFillMemory(&PipeLocalInfo, sizeof(PipeLocalInfo), 0x55);
    Status = ZwQueryInformationFile(ClientHandle,
                                    &IoStatusBlock,
                                    &PipeLocalInfo,
                                    sizeof(PipeLocalInfo),
                                    FilePipeLocalInformation);
    ok_eq_hex_(Status, STATUS_SUCCESS);
    ok_eq_hex_(IoStatusBlock.Status, STATUS_SUCCESS);
    ok_eq_ulongptr_(IoStatusBlock.Information, sizeof(PipeLocalInfo));
    ok_eq_ulong_(PipeLocalInfo.NamedPipeType, NamedPipeType);
    ok_eq_ulong_(PipeLocalInfo.NamedPipeConfiguration, NamedPipeConfiguration);
    ok_eq_ulong_(PipeLocalInfo.MaximumInstances, MaximumInstances);
    ok_eq_ulong_(PipeLocalInfo.CurrentInstances, CurrentInstances);
    ok_eq_ulong_(PipeLocalInfo.InboundQuota, InboundQuota);
    ok_eq_ulong_(PipeLocalInfo.ReadDataAvailable, ReadDataAvailable);
    ok_eq_ulong_(PipeLocalInfo.OutboundQuota, OutboundQuota);
    ok_eq_ulong_(PipeLocalInfo.WriteQuotaAvailable, WriteQuotaAvailable);
    ok_eq_ulong_(PipeLocalInfo.NamedPipeState, NamedPipeState);
    ok_eq_ulong_(PipeLocalInfo.NamedPipeEnd, (ULONG)FILE_PIPE_CLIENT_END);
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    RtlFillMemory(&PipeRemoteInfo, sizeof(PipeRemoteInfo), 0x55);
    Status = ZwQueryInformationFile(ClientHandle,
                                    &IoStatusBlock,
                                    &PipeRemoteInfo,
                                    sizeof(PipeRemoteInfo),
                                    FilePipeInformation);
    ok_eq_hex_(Status, STATUS_SUCCESS);
    ok_eq_hex_(IoStatusBlock.Status, STATUS_SUCCESS);
    ok_eq_ulongptr_(IoStatusBlock.Information, RTL_SIZEOF_THROUGH_FIELD(FILE_PIPE_REMOTE_INFORMATION, CollectDataTime));
    ok_eq_ulonglong_(PipeRemoteInfo.CollectDataTime.QuadPart, 0ULL);
    ok_eq_ulong_(PipeRemoteInfo.MaximumCollectionCount, 0x55555555UL);
}

NpCheckServerPipe_()

VOID NpCheckServerPipe_	(	IN HANDLE	ServerHandle,
		IN ULONG	ReadMode,
		IN ULONG	CompletionMode,
		IN ULONG	NamedPipeType,
		IN ULONG	NamedPipeConfiguration,
		IN ULONG	MaximumInstances,
		IN ULONG	CurrentInstances,
		IN ULONG	InboundQuota,
		IN ULONG	ReadDataAvailable,
		IN ULONG	OutboundQuota,
		IN ULONG	WriteQuotaAvailable,
		IN ULONG	NamedPipeState,
		IN PCSTR	FileAndLine
	)

Definition at line 446 of file NpfsHelpers.c.

{
    NTSTATUS Status;
    IO_STATUS_BLOCK IoStatusBlock;
    FILE_PIPE_INFORMATION PipeInfo;
    FILE_PIPE_LOCAL_INFORMATION PipeLocalInfo;
    FILE_PIPE_REMOTE_INFORMATION PipeRemoteInfo;
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    RtlFillMemory(&PipeInfo, sizeof(PipeInfo), 0x55);
    Status = ZwQueryInformationFile(ServerHandle,
                                    &IoStatusBlock,
                                    &PipeInfo,
                                    sizeof(PipeInfo),
                                    FilePipeInformation);
    ok_eq_hex_(Status, STATUS_SUCCESS);
    ok_eq_hex_(IoStatusBlock.Status, STATUS_SUCCESS);
    ok_eq_ulongptr_(IoStatusBlock.Information, sizeof(PipeInfo));
    ok_eq_ulong_(PipeInfo.ReadMode, ReadMode);
    ok_eq_ulong_(PipeInfo.CompletionMode, CompletionMode);
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    RtlFillMemory(&PipeLocalInfo, sizeof(PipeLocalInfo), 0x55);
    Status = ZwQueryInformationFile(ServerHandle,
                                    &IoStatusBlock,
                                    &PipeLocalInfo,
                                    sizeof(PipeLocalInfo),
                                    FilePipeLocalInformation);
    ok_eq_hex_(Status, STATUS_SUCCESS);
    ok_eq_hex_(IoStatusBlock.Status, STATUS_SUCCESS);
    ok_eq_ulongptr_(IoStatusBlock.Information, sizeof(PipeLocalInfo));
    ok_eq_ulong_(PipeLocalInfo.NamedPipeType, NamedPipeType);
    ok_eq_ulong_(PipeLocalInfo.NamedPipeConfiguration, NamedPipeConfiguration);
    ok_eq_ulong_(PipeLocalInfo.MaximumInstances, MaximumInstances);
    ok_eq_ulong_(PipeLocalInfo.CurrentInstances, CurrentInstances);
    ok_eq_ulong_(PipeLocalInfo.InboundQuota, InboundQuota);
    ok_eq_ulong_(PipeLocalInfo.ReadDataAvailable, ReadDataAvailable);
    ok_eq_ulong_(PipeLocalInfo.OutboundQuota, OutboundQuota);
    ok_eq_ulong_(PipeLocalInfo.WriteQuotaAvailable, WriteQuotaAvailable);
    ok_eq_ulong_(PipeLocalInfo.NamedPipeState, NamedPipeState);
    ok_eq_ulong_(PipeLocalInfo.NamedPipeEnd, (ULONG)FILE_PIPE_SERVER_END);
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    RtlFillMemory(&PipeRemoteInfo, sizeof(PipeRemoteInfo), 0x55);
    Status = ZwQueryInformationFile(ServerHandle,
                                    &IoStatusBlock,
                                    &PipeRemoteInfo,
                                    sizeof(PipeRemoteInfo),
                                    FilePipeInformation);
    ok_eq_hex_(Status, STATUS_SUCCESS);
    ok_eq_hex_(IoStatusBlock.Status, STATUS_SUCCESS);
    ok_eq_ulongptr_(IoStatusBlock.Information, RTL_SIZEOF_THROUGH_FIELD(FILE_PIPE_REMOTE_INFORMATION, CollectDataTime));
    ok_eq_ulonglong_(PipeRemoteInfo.CollectDataTime.QuadPart, 0ULL);
    ok_eq_ulong_(PipeRemoteInfo.MaximumCollectionCount, 0x55555555UL);
}

NpControlPipe()

NTSTATUS NpControlPipe	(	IN HANDLE	ServerHandle,
		IN ULONG	FsControlCode,
		IN PVOID	InputBuffer,
		IN ULONG	InputBufferLength
	)

Definition at line 208 of file NpfsHelpers.c.

{
    NTSTATUS Status;
    IO_STATUS_BLOCK IoStatusBlock;
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    Status = ZwFsControlFile(ServerHandle,
                             NULL,
                             NULL,
                             NULL,
                             &IoStatusBlock,
                             FsControlCode,
                             InputBuffer,
                             InputBufferLength,
                             NULL,
                             0);
    if (Status == STATUS_PENDING)
    {
        Status = ZwWaitForSingleObject(ServerHandle,
                                       FALSE,
                                       NULL);
        ok_eq_hex(Status, STATUS_SUCCESS);
        Status = IoStatusBlock.Status;
    }
    if (NT_SUCCESS(Status))
    {
        ok_eq_hex(IoStatusBlock.Status, Status);
        ok_eq_ulongptr(IoStatusBlock.Information, 0);
    }
    else
    {
        ok_eq_hex(IoStatusBlock.Status, 0x55555555UL);
        ok_eq_ulongptr(IoStatusBlock.Information, 0x5555555555555555ULL);
    }
    return Status;
}

Referenced by NpWaitPipe().

NpCreatePipe()

NTSTATUS NpCreatePipe	(	OUT PHANDLE	ServerHandle,
		PCWSTR	PipePath,
		ULONG	ReadMode,
		ULONG	CompletionMode,
		ULONG	NamedPipeType,
		ULONG	NamedPipeConfiguration,
		ULONG	MaximumInstances,
		ULONG	InboundQuota,
		ULONG	OutboundQuota
	)

Definition at line 86 of file NpfsHelpers.c.

{
    ULONG ShareAccess;
    LARGE_INTEGER DefaultTimeout;
 
    if (NamedPipeConfiguration == FILE_PIPE_INBOUND)
        ShareAccess = FILE_SHARE_WRITE;
    else if (NamedPipeConfiguration == FILE_PIPE_OUTBOUND)
        ShareAccess = FILE_SHARE_READ;
    else if (NamedPipeConfiguration == FILE_PIPE_FULL_DUPLEX)
        ShareAccess = FILE_SHARE_READ | FILE_SHARE_WRITE;
    else
    {
        ASSERTMSG("Invalid NamedPipeConfiguration parameter value!\n", FALSE);
        return STATUS_INVALID_PARAMETER_6;
    }
 
    DefaultTimeout.QuadPart = -50 * 1000 * 10;
 
    return NpCreatePipeEx(ServerHandle,
                          PipePath,
                          ReadMode,
                          CompletionMode,
                          NamedPipeType,
                          ShareAccess,
                          MaximumInstances,
                          InboundQuota,
                          OutboundQuota,
                          SYNCHRONIZE | GENERIC_READ | GENERIC_WRITE,
                          FILE_OPEN_IF,
                          FILE_SYNCHRONOUS_IO_NONALERT,
                          &DefaultTimeout);
}

NpCreatePipeEx()

NTSTATUS NpCreatePipeEx	(	OUT PHANDLE	ServerHandle,
		IN PCWSTR	PipePath,
		IN ULONG	ReadMode,
		IN ULONG	CompletionMode,
		IN ULONG	NamedPipeType,
		IN ULONG	ShareAccess,
		IN ULONG	MaximumInstances,
		IN ULONG	InboundQuota,
		IN ULONG	OutboundQuota,
		IN ACCESS_MASK	DesiredAccess,
		IN ULONG	Disposition,
		IN ULONG	CreateOptions,
		IN PLARGE_INTEGER DefaultTimeout	OPTIONAL
	)

Definition at line 12 of file NpfsHelpers.c.

{
    UNICODE_STRING ObjectName;
    OBJECT_ATTRIBUTES ObjectAttributes;
    NAMED_PIPE_CREATE_PARAMETERS Params;
    IO_STATUS_BLOCK IoStatusBlock;
    NTSTATUS Status;
 
    RtlInitUnicodeString(&ObjectName, PipePath);
    InitializeObjectAttributes(&ObjectAttributes,
                               &ObjectName,
                               OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
                               NULL,
                               NULL);
 
    Params.NamedPipeType = NamedPipeType;
    Params.ReadMode = ReadMode;
    Params.CompletionMode = CompletionMode;
    Params.MaximumInstances = MaximumInstances;
    Params.InboundQuota = InboundQuota;
    Params.OutboundQuota = OutboundQuota;
    if (DefaultTimeout)
    {
        Params.DefaultTimeout.QuadPart = DefaultTimeout->QuadPart;
        Params.TimeoutSpecified = TRUE;
    }
    else
    {
        Params.DefaultTimeout.QuadPart = 0;
        Params.TimeoutSpecified = FALSE;
    }
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    Status = IoCreateFile(ServerHandle,
                          DesiredAccess,
                          &ObjectAttributes,
                          &IoStatusBlock,
                          NULL, /* AllocationSize */
                          0, /* FileAttributes */
                          ShareAccess,
                          Disposition,
                          CreateOptions,
                          NULL, /* EaBuffer */
                          0, /* EaLength */
                          CreateFileTypeNamedPipe,
                          &Params,
                          0);
    if (NT_SUCCESS(Status))
    {
        ok_eq_hex(IoStatusBlock.Status, Status);
        ok_eq_ulongptr(IoStatusBlock.Information, FILE_CREATED);
    }
    else
    {
        ok_eq_hex(IoStatusBlock.Status, 0x55555555UL);
        ok_eq_ulongptr(IoStatusBlock.Information, 0x5555555555555555ULL);
    }
    return Status;
}

Referenced by NpCreatePipe().

NpOpenPipe()

NTSTATUS NpOpenPipe	(	OUT PHANDLE	ClientHandle,
		IN PCWSTR	PipePath,
		IN ULONG	NamedPipeConfiguration
	)

Definition at line 180 of file NpfsHelpers.c.

{
    ULONG ShareAccess;
 
    if (NamedPipeConfiguration == FILE_PIPE_INBOUND)
        ShareAccess = FILE_SHARE_WRITE;
    else if (NamedPipeConfiguration == FILE_PIPE_OUTBOUND)
        ShareAccess = FILE_SHARE_READ;
    else if (NamedPipeConfiguration == FILE_PIPE_FULL_DUPLEX)
        ShareAccess = FILE_SHARE_READ | FILE_SHARE_WRITE;
    else
    {
        ASSERTMSG("Invalid NamedPipeConfiguration parameter value!\n", FALSE);
        return STATUS_INVALID_PARAMETER_3;
    }
 
    return NpOpenPipeEx(ClientHandle,
                        PipePath,
                        SYNCHRONIZE | GENERIC_READ | GENERIC_WRITE,
                        ShareAccess,
                        FILE_OPEN,
                        FILE_SYNCHRONOUS_IO_NONALERT);
}

Referenced by ConnectPipe().

NpOpenPipeEx()

NTSTATUS NpOpenPipeEx	(	OUT PHANDLE	ClientHandle,
		IN PCWSTR	PipePath,
		IN ACCESS_MASK	DesiredAccess,
		IN ULONG	ShareAccess,
		IN ULONG	Disposition,
		IN ULONG	CreateOptions
	)

Definition at line 130 of file NpfsHelpers.c.

{
    UNICODE_STRING ObjectName;
    OBJECT_ATTRIBUTES ObjectAttributes;
    NTSTATUS Status;
    IO_STATUS_BLOCK IoStatusBlock;
 
    RtlInitUnicodeString(&ObjectName, PipePath);
    InitializeObjectAttributes(&ObjectAttributes,
                               &ObjectName,
                               OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
                               NULL,
                               NULL);
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    Status = IoCreateFile(ClientHandle,
                          DesiredAccess,
                          &ObjectAttributes,
                          &IoStatusBlock,
                          NULL, /* AllocationSize */
                          0, /* FileAttributes */
                          ShareAccess,
                          Disposition,
                          CreateOptions,
                          NULL, /* EaBuffer */
                          0, /* EaLength */
                          CreateFileTypeNone,
                          NULL,
                          0);
    if (NT_SUCCESS(Status))
    {
        ok(Status != STATUS_PENDING, "IoCreateFile returned pending\n");
        ok_eq_hex(IoStatusBlock.Status, Status);
        ok_eq_ulongptr(IoStatusBlock.Information, FILE_OPENED);
    }
    else
    {
        ok_eq_hex(IoStatusBlock.Status, 0x55555555UL);
        ok_eq_ulongptr(IoStatusBlock.Information, 0x5555555555555555ULL);
    }
    return Status;
}

Referenced by ConnectPipe(), and NpOpenPipe().

NpQueryPipe_()

VOID NpQueryPipe_	(	IN HANDLE	PipeHandle,
		IN NTSTATUS	ExpectedStatus,
		IN PCSTR	FileAndLine
	)

Definition at line 594 of file NpfsHelpers.c.

{
    NTSTATUS Status;
    IO_STATUS_BLOCK IoStatusBlock;
    FILE_PIPE_INFORMATION PipeInfo;
    FILE_PIPE_LOCAL_INFORMATION PipeLocalInfo;
    FILE_PIPE_REMOTE_INFORMATION PipeRemoteInfo;
 
    ASSERT(!NT_SUCCESS(ExpectedStatus));
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    RtlFillMemory(&PipeInfo, sizeof(PipeInfo), 0x55);
    Status = ZwQueryInformationFile(PipeHandle,
                                    &IoStatusBlock,
                                    &PipeInfo,
                                    sizeof(PipeInfo),
                                    FilePipeInformation);
    ok_eq_hex_(Status, ExpectedStatus);
    ok_bool_true(CheckBuffer(&IoStatusBlock, sizeof(IoStatusBlock), 0x55), "CheckBuffer returned");
    ok_bool_true(CheckBuffer(&PipeInfo, sizeof(PipeInfo), 0x55), "CheckBuffer returned");
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    RtlFillMemory(&PipeLocalInfo, sizeof(PipeLocalInfo), 0x55);
    Status = ZwQueryInformationFile(PipeHandle,
                                    &IoStatusBlock,
                                    &PipeLocalInfo,
                                    sizeof(PipeLocalInfo),
                                    FilePipeLocalInformation);
    ok_eq_hex_(Status, ExpectedStatus);
    ok_bool_true(CheckBuffer(&IoStatusBlock, sizeof(IoStatusBlock), 0x55), "CheckBuffer returned");
    ok_bool_true(CheckBuffer(&PipeLocalInfo, sizeof(PipeLocalInfo), 0x55), "CheckBuffer returned");
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    RtlFillMemory(&PipeRemoteInfo, sizeof(PipeRemoteInfo), 0x55);
    Status = ZwQueryInformationFile(PipeHandle,
                                    &IoStatusBlock,
                                    &PipeRemoteInfo,
                                    sizeof(PipeRemoteInfo),
                                    FilePipeInformation);
    ok_eq_hex_(Status, ExpectedStatus);
    ok_bool_true(CheckBuffer(&IoStatusBlock, sizeof(IoStatusBlock), 0x55), "CheckBuffer returned");
    ok_bool_true(CheckBuffer(&PipeRemoteInfo, sizeof(PipeRemoteInfo), 0x55), "CheckBuffer returned");
}

NpReadPipe()

NTSTATUS NpReadPipe	(	IN HANDLE	PipeHandle,
		OUT PVOID	Buffer,
		IN ULONG	BufferSize,
		OUT PULONG_PTR	BytesRead
	)

Definition at line 322 of file NpfsHelpers.c.

{
    NTSTATUS Status;
    IO_STATUS_BLOCK IoStatusBlock;
    BOOLEAN PendingReturned = FALSE;
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    Status = ZwReadFile(PipeHandle,
                        NULL,
                        NULL,
                        NULL,
                        &IoStatusBlock,
                        Buffer,
                        BufferSize,
                        NULL,
                        NULL);
    if (Status == STATUS_PENDING)
    {
        Status = ZwWaitForSingleObject(PipeHandle,
                                       FALSE,
                                       NULL);
        ok_eq_hex(Status, STATUS_SUCCESS);
        Status = IoStatusBlock.Status;
        PendingReturned = TRUE;
    }
    if (NT_SUCCESS(Status))
    {
        ok_eq_hex(IoStatusBlock.Status, Status);
        *BytesRead = IoStatusBlock.Information;
    }
    else
    {
        if (PendingReturned)
        {
            ok_eq_hex(IoStatusBlock.Status, Status);
            ok_eq_ulongptr(IoStatusBlock.Information, 0);
        }
        else
        {
            ok_eq_hex(IoStatusBlock.Status, 0x55555555UL);
            ok_eq_ulongptr(IoStatusBlock.Information, 0x5555555555555555ULL);
        }
        *BytesRead = 0;
    }
    return Status;
}

Referenced by ReadPipe().

NpWaitPipe()

NTSTATUS NpWaitPipe	(	IN PCWSTR	PipeName,
		IN PLARGE_INTEGER	Timeout
	)

Definition at line 250 of file NpfsHelpers.c.

{
    NTSTATUS Status;
    HANDLE RootHandle;
    UNICODE_STRING RootDirectoryName = RTL_CONSTANT_STRING(DEVICE_NAMED_PIPE);
    OBJECT_ATTRIBUTES ObjectAttributes;
    IO_STATUS_BLOCK IoStatusBlock;
    PFILE_PIPE_WAIT_FOR_BUFFER WaitForBuffer;
    ULONG NameLength;
    ULONG BufferSize;
 
    InitializeObjectAttributes(&ObjectAttributes,
                               &RootDirectoryName,
                               OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
                               NULL,
                               NULL);
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    Status = IoCreateFile(&RootHandle,
                          FILE_READ_ATTRIBUTES | SYNCHRONIZE,
                          &ObjectAttributes,
                          &IoStatusBlock,
                          NULL,
                          0,
                          FILE_SHARE_READ | FILE_SHARE_WRITE,
                          FILE_OPEN,
                          FILE_SYNCHRONOUS_IO_NONALERT,
                          NULL,
                          0,
                          CreateFileTypeNone,
                          NULL,
                          0);
    if (!NT_SUCCESS(Status))
    {
        ok_eq_hex(IoStatusBlock.Status, 0x55555555UL);
        ok_eq_ulongptr(IoStatusBlock.Information, 0x5555555555555555ULL);
        return Status;
    }
    ok(Status != STATUS_PENDING, "IoCreateFile returned pending\n");
    ok_eq_hex(IoStatusBlock.Status, Status);
    ok_eq_ulongptr(IoStatusBlock.Information, FILE_OPENED);
 
    NameLength = (ULONG)(wcslen(PipeName) * sizeof(WCHAR));
    BufferSize = FIELD_OFFSET(FILE_PIPE_WAIT_FOR_BUFFER,
                              Name[NameLength / sizeof(WCHAR)]);
    WaitForBuffer = ExAllocatePoolWithTag(NonPagedPool, BufferSize, 'WPmK');
    if (WaitForBuffer == NULL)
        return STATUS_INSUFFICIENT_RESOURCES;
 
    if (Timeout)
    {
        WaitForBuffer->Timeout.QuadPart = Timeout->QuadPart;
        WaitForBuffer->TimeoutSpecified = TRUE;
    }
    else
    {
        WaitForBuffer->Timeout.QuadPart = 0;
        WaitForBuffer->TimeoutSpecified = FALSE;
    }
    WaitForBuffer->NameLength = NameLength;
    RtlCopyMemory(WaitForBuffer->Name, PipeName, NameLength);
    Status = NpControlPipe(RootHandle,
                           FSCTL_PIPE_WAIT,
                           WaitForBuffer,
                           BufferSize);
    ExFreePoolWithTag(WaitForBuffer, 'WPmK');
    return Status;
}

NpWritePipe()

NTSTATUS NpWritePipe	(	IN HANDLE	PipeHandle,
		IN const VOID *	Buffer,
		IN ULONG	BufferSize,
		OUT PULONG_PTR	BytesWritten
	)

Definition at line 374 of file NpfsHelpers.c.

{
    NTSTATUS Status;
    IO_STATUS_BLOCK IoStatusBlock;
 
    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
    Status = ZwWriteFile(PipeHandle,
                         NULL,
                         NULL,
                         NULL,
                         &IoStatusBlock,
                         (PVOID)Buffer,
                         BufferSize,
                         NULL,
                         NULL);
    if (Status == STATUS_PENDING)
    {
        Status = ZwWaitForSingleObject(PipeHandle,
                                       FALSE,
                                       NULL);
        ok_eq_hex(Status, STATUS_SUCCESS);
        Status = IoStatusBlock.Status;
    }
    if (NT_SUCCESS(Status))
    {
        ok_eq_hex(IoStatusBlock.Status, Status);
        *BytesWritten = IoStatusBlock.Information;
    }
    else
    {
        ok_eq_hex(IoStatusBlock.Status, 0x55555555UL);
        ok_eq_ulongptr(IoStatusBlock.Information, 0x5555555555555555ULL);
        *BytesWritten = 0;
    }
    return Status;
}

Referenced by WritePipe().

PipeWorkerThread()

static VOID NTAPI PipeWorkerThread ( IN PVOID  ThreadContext )

static

Definition at line 645 of file NpfsHelpers.c.

{
    PTHREAD_CONTEXT Context = ThreadContext;
    PVOID WaitEvents[2] = { &Context->ThreadDoneEvent,
                            &Context->StartWorkEvent };
    NTSTATUS Status;
 
    while (TRUE)
    {
        Status = KeWaitForMultipleObjects(RTL_NUMBER_OF(WaitEvents),
                                          WaitEvents,
                                          WaitAny,
                                          Executive,
                                          KernelMode,
                                          FALSE,
                                          NULL,
                                          NULL);
        if (Status == STATUS_WAIT_0)
            break;
        ASSERT(Status == STATUS_WAIT_1);
 
        Context->Work(Context);
 
        KeSetEvent(&Context->WorkCompleteEvent, IO_NO_INCREMENT, TRUE);
    }
}

StartWorkerThread()

VOID StartWorkerThread

(

OUT PTHREAD_CONTEXT

Context

)

Definition at line 674 of file NpfsHelpers.c.

{
    KeInitializeEvent(&Context->ThreadDoneEvent, NotificationEvent, FALSE);
    KeInitializeEvent(&Context->StartWorkEvent, SynchronizationEvent, FALSE);
    KeInitializeEvent(&Context->WorkCompleteEvent, NotificationEvent, TRUE);
 
    Context->Thread = KmtStartThread(PipeWorkerThread, Context);
}

Referenced by TestConnect(), and TestReadWrite().

TriggerWork()

BOOLEAN TriggerWork	(	IN PTHREAD_CONTEXT	Context,
		IN ULONG	MilliSeconds
	)

Definition at line 710 of file NpfsHelpers.c.

{
    NTSTATUS Status;
 
    Status = KeWaitForSingleObject(&Context->WorkCompleteEvent,
                                   Executive,
                                   KernelMode,
                                   FALSE,
                                   NULL);
    ok_eq_hex(Status, STATUS_SUCCESS);
    KeClearEvent(&Context->WorkCompleteEvent);
    KeSetEvent(&Context->StartWorkEvent, IO_NO_INCREMENT, TRUE);
    return WaitForWork(Context, MilliSeconds);
}

Referenced by CheckConnectPipe(), CheckListenPipe(), CheckReadPipe(), and CheckWritePipe().

WaitForWork()

BOOLEAN WaitForWork	(	IN PTHREAD_CONTEXT	Context,
		IN ULONG	MilliSeconds
	)

Definition at line 692 of file NpfsHelpers.c.

{
    LARGE_INTEGER Timeout;
    NTSTATUS Status;
 
    Timeout.QuadPart = -10 * 1000 * (LONGLONG)MilliSeconds;
    Status = KeWaitForSingleObject(&Context->WorkCompleteEvent,
                                   Executive,
                                   KernelMode,
                                   FALSE,
                                   &Timeout);
    ok(Status == STATUS_SUCCESS || Status == STATUS_TIMEOUT, "Wait status %lx\n", Status);
    return Status != STATUS_TIMEOUT;
}

Referenced by TestConnect(), TestReadWrite(), and TriggerWork().

Variable Documentation

PipeWorkerThread

KSTART_ROUTINE PipeWorkerThread

static

Definition at line 641 of file NpfsHelpers.c.

Referenced by StartWorkerThread().