finfo.c

来自「ReactOS是一些高手根据Windows XP的内核编写出的类XP。内核实现机」· C语言 代码 · 共 286 行

/*
* COPYRIGHT:  See COPYING in the top level directory
* PROJECT:    ReactOS kernel
* FILE:       drivers/fs/np/finfo.c
* PURPOSE:    Named pipe filesystem
* PROGRAMMER: Eric Kohl
*/

/* INCLUDES ******************************************************************/

#include "npfs.h"

#define NDEBUG
#include <debug.h>

/* FUNCTIONS *****************************************************************/

static
NTSTATUS
NpfsSetPipeInformation(PDEVICE_OBJECT DeviceObject,
					   PNPFS_CCB Ccb,
					   PFILE_PIPE_INFORMATION Info,
					   PULONG BufferLength)
{
	PNPFS_FCB Fcb;
	PFILE_PIPE_INFORMATION Request;
	DPRINT("NpfsSetPipeInformation()\n");

	/* Get the Pipe and data */
	Fcb = Ccb->Fcb;
	Request = (PFILE_PIPE_INFORMATION)Info;

	/* Set Pipe Data */
	Fcb->ReadMode = Request->ReadMode;
	Fcb->CompletionMode =  Request->CompletionMode;

	/* Return Success */
	return STATUS_SUCCESS;
}

static
NTSTATUS
NpfsSetPipeRemoteInformation(PDEVICE_OBJECT DeviceObject,
							 PNPFS_CCB Ccb,
							 PFILE_PIPE_INFORMATION Info,
							 PULONG BufferLength)
{
	PNPFS_FCB Fcb;
	PFILE_PIPE_REMOTE_INFORMATION Request;
	DPRINT("NpfsSetPipeRemoteInformation()\n");

	/* Get the Pipe and data */
	Fcb = Ccb->Fcb;
	Request = (PFILE_PIPE_REMOTE_INFORMATION)Info;

	/* Set the Settings */
	Fcb->TimeOut = Request->CollectDataTime;
	Fcb->InboundQuota = Request->MaximumCollectionCount;

	/* Return Success */
	return STATUS_SUCCESS;
}

static
NTSTATUS
NpfsQueryPipeInformation(PDEVICE_OBJECT DeviceObject,
						 PNPFS_CCB Ccb,
						 PFILE_PIPE_INFORMATION Info,
						 PULONG BufferLength)
{
	PNPFS_FCB Fcb;
	DPRINT("NpfsQueryPipeInformation()\n");

	/* Get the Pipe */
	Fcb = Ccb->Fcb;

	/* Clear Info */
	RtlZeroMemory(Info, sizeof(FILE_PIPE_INFORMATION));

	/* Return Info */
	Info->CompletionMode = Fcb->CompletionMode;
	Info->ReadMode = Fcb->ReadMode;

	/* Return success */
	*BufferLength -= sizeof(FILE_PIPE_INFORMATION);
	return STATUS_SUCCESS;
}

static
NTSTATUS
NpfsQueryPipeRemoteInformation(PDEVICE_OBJECT DeviceObject,
							   PNPFS_CCB Ccb,
							   PFILE_PIPE_REMOTE_INFORMATION Info,
							   PULONG BufferLength)
{
	PNPFS_FCB Fcb;
	DPRINT("NpfsQueryPipeRemoteInformation()\n");

	/* Get the Pipe */
	Fcb = Ccb->Fcb;

	/* Clear Info */
	RtlZeroMemory(Info, sizeof(FILE_PIPE_REMOTE_INFORMATION));

	/* Return Info */
	Info->MaximumCollectionCount = Fcb->InboundQuota;
	Info->CollectDataTime = Fcb->TimeOut;

	/* Return success */
	*BufferLength -= sizeof(FILE_PIPE_REMOTE_INFORMATION);
	return STATUS_SUCCESS;
}


static NTSTATUS
NpfsQueryLocalPipeInformation(PDEVICE_OBJECT DeviceObject,
							  PNPFS_CCB Ccb,
							  PFILE_PIPE_LOCAL_INFORMATION Info,
							  PULONG BufferLength)
{
	PNPFS_FCB Fcb;

	DPRINT("NpfsQueryLocalPipeInformation()\n");

	Fcb = Ccb->Fcb;

	RtlZeroMemory(Info,
		sizeof(FILE_PIPE_LOCAL_INFORMATION));

	Info->NamedPipeType = Fcb->PipeType;
	Info->NamedPipeConfiguration = Fcb->PipeConfiguration;
	Info->MaximumInstances = Fcb->MaximumInstances;
	Info->CurrentInstances = Fcb->CurrentInstances;
	Info->InboundQuota = Fcb->InboundQuota;
	Info->OutboundQuota = Fcb->OutboundQuota;
	Info->NamedPipeState = Ccb->PipeState;
	Info->NamedPipeEnd = Ccb->PipeEnd;

	if (Ccb->PipeEnd == FILE_PIPE_SERVER_END)
	{
		Info->ReadDataAvailable = Ccb->ReadDataAvailable;
		Info->WriteQuotaAvailable = Ccb->WriteQuotaAvailable;
	}
	else if (Ccb->OtherSide != NULL)
	{
		Info->ReadDataAvailable = Ccb->OtherSide->ReadDataAvailable;
		Info->WriteQuotaAvailable = Ccb->OtherSide->WriteQuotaAvailable;
	}

	*BufferLength -= sizeof(FILE_PIPE_LOCAL_INFORMATION);
	return STATUS_SUCCESS;
}


NTSTATUS STDCALL
NpfsQueryInformation(PDEVICE_OBJECT DeviceObject,
					 PIRP Irp)
{
	PIO_STACK_LOCATION IoStack;
	FILE_INFORMATION_CLASS FileInformationClass;
	PFILE_OBJECT FileObject;
	PNPFS_DEVICE_EXTENSION DeviceExtension;
	PNPFS_FCB Fcb;
	PNPFS_CCB Ccb;
	PVOID SystemBuffer;
	ULONG BufferLength;
	NTSTATUS Status;

	DPRINT("NpfsQueryInformation(DeviceObject %p Irp %p)\n", DeviceObject, Irp);

	IoStack = IoGetCurrentIrpStackLocation (Irp);
	FileInformationClass = IoStack->Parameters.QueryFile.FileInformationClass;
	DeviceExtension = DeviceObject->DeviceExtension;
	FileObject = IoStack->FileObject;
	Ccb = (PNPFS_CCB)FileObject->FsContext2;
	Fcb = Ccb->Fcb;

	SystemBuffer = Irp->AssociatedIrp.SystemBuffer;
	BufferLength = IoStack->Parameters.QueryFile.Length;

	DPRINT("Pipe name: %wZ\n", &Fcb->PipeName);
	DPRINT("FileInformationClass %d\n", FileInformationClass);
	DPRINT("SystemBuffer %p\n", SystemBuffer);
	DPRINT("BufferLength %lu\n", BufferLength);

	switch (FileInformationClass)
	{
	case FilePipeInformation:
		Status = NpfsQueryPipeInformation(DeviceObject,
			Ccb,
			SystemBuffer,
			&BufferLength);
		break;

	case FilePipeLocalInformation:
		Status = NpfsQueryLocalPipeInformation(DeviceObject,
			Ccb,
			SystemBuffer,
			&BufferLength);
		break;

	case FilePipeRemoteInformation:
		Status = NpfsQueryPipeRemoteInformation(DeviceObject,
			Ccb,
			SystemBuffer,
			&BufferLength);
		break;

	default:
		Status = STATUS_NOT_SUPPORTED;
	}

	Irp->IoStatus.Status = Status;
	if (NT_SUCCESS(Status))
		Irp->IoStatus.Information =
		IoStack->Parameters.QueryFile.Length - BufferLength;
	else
		Irp->IoStatus.Information = 0;
	IoCompleteRequest (Irp, IO_NO_INCREMENT);

	return Status;
}


NTSTATUS STDCALL
NpfsSetInformation(PDEVICE_OBJECT DeviceObject,
				   PIRP Irp)
{
	PIO_STACK_LOCATION IoStack;
	FILE_INFORMATION_CLASS FileInformationClass;
	PFILE_OBJECT FileObject;
	PNPFS_FCB Fcb;
	PNPFS_CCB Ccb;
	PVOID SystemBuffer;
	ULONG BufferLength;
	NTSTATUS Status;

	DPRINT("NpfsSetInformation(DeviceObject %p Irp %p)\n", DeviceObject, Irp);

	IoStack = IoGetCurrentIrpStackLocation (Irp);
	FileInformationClass = IoStack->Parameters.QueryFile.FileInformationClass;
	FileObject = IoStack->FileObject;
	Ccb = (PNPFS_CCB)FileObject->FsContext2;
	Fcb = Ccb->Fcb;

	SystemBuffer = Irp->AssociatedIrp.SystemBuffer;
	BufferLength = IoStack->Parameters.QueryFile.Length;

	DPRINT("Pipe name: %wZ\n", &Fcb->PipeName);
	DPRINT("FileInformationClass %d\n", FileInformationClass);
	DPRINT("SystemBuffer %p\n", SystemBuffer);
	DPRINT("BufferLength %lu\n", BufferLength);

	switch (FileInformationClass)
	{
	case FilePipeInformation:
		/* Call the handler */
		Status = NpfsSetPipeInformation(DeviceObject,
			Ccb,
			SystemBuffer,
			&BufferLength);
		break;

	case FilePipeLocalInformation:
		Status = STATUS_NOT_IMPLEMENTED;
		break;

	case FilePipeRemoteInformation:
		/* Call the handler */
		Status = NpfsSetPipeRemoteInformation(DeviceObject,
			Ccb,
			SystemBuffer,
			&BufferLength);
		break;
	default:
		Status = STATUS_NOT_SUPPORTED;
	}

	Irp->IoStatus.Status = Status;
	Irp->IoStatus.Information = 0;
	IoCompleteRequest(Irp, IO_NO_INCREMENT);
	return Status;
}

/* EOF */