[ros-diffs] [ekohl] 50250: [NPFS] - Add NpfsDirectoryControl routine in order to enumerate pipes. - Use NpfsGetFcb and NpfsGetCcb to retrieve the FCB or CCB and use SEH to protect these routines.

Author: ekohl
Date: Sat Jan  1 12:40:24 2011
New Revision: 50250

URL: http://svn.reactos.org/svn/reactos?rev=50250&view=rev
Log:
[NPFS]
- Add NpfsDirectoryControl routine in order to enumerate pipes.
- Use NpfsGetFcb and NpfsGetCcb to retrieve the FCB or CCB and use SEH to protect these
routines.

Added:
    trunk/reactos/drivers/filesystems/npfs/dirctl.c   (with props)
Modified:
    trunk/reactos/drivers/filesystems/npfs/create.c
    trunk/reactos/drivers/filesystems/npfs/npfs.c
    trunk/reactos/drivers/filesystems/npfs/npfs.h
    trunk/reactos/drivers/filesystems/npfs/npfs.rbuild

Modified: trunk/reactos/drivers/filesystems/npfs/create.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/filesystems/npfs/c…
==============================================================================
--- trunk/reactos/drivers/filesystems/npfs/create.c [iso-8859-1] (original)
+++ trunk/reactos/drivers/filesystems/npfs/create.c [iso-8859-1] Sat Jan  1 12:40:24 2011
@@ -121,6 +121,8 @@
         return;
     }
 
+    RtlZeroMemory(Ccb, sizeof(NPFS_CCB));
+
     Ccb->Type = CCB_DEVICE;
     Ccb->Fcb = Fcb;
 
@@ -149,6 +151,8 @@
         IoStatus->Status = STATUS_NO_MEMORY;
         return;
     }
+
+    RtlZeroMemory(Ccb, sizeof(NPFS_CCB));
 
     Ccb->Type = CCB_DIRECTORY;
     Ccb->Fcb = Fcb;
@@ -881,6 +885,9 @@
     {
         DPRINT("Closing the root directory!\n");
 
+        if (Ccb->u.Directory.SearchPattern.Buffer != NULL)
+            ExFreePool(Ccb->u.Directory.SearchPattern.Buffer);
+
         ExFreePool(Ccb);
         FileObject->FsContext = NULL;
         FileObject->FsContext2 = NULL;

Added: trunk/reactos/drivers/filesystems/npfs/dirctl.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/filesystems/npfs/d…
==============================================================================
--- trunk/reactos/drivers/filesystems/npfs/dirctl.c (added)
+++ trunk/reactos/drivers/filesystems/npfs/dirctl.c [iso-8859-1] Sat Jan  1 12:40:24 2011
@@ -1,0 +1,250 @@
+/*
+ * COPYRIGHT:  See COPYING in the top level directory
+ * PROJECT:    ReactOS kernel
+ * FILE:       drivers/filesastems/npfs/dirctl.c
+ * PURPOSE:    Named pipe filesystem
+ * PROGRAMMER: Eric Kohl
+ */
+
+/* INCLUDES ******************************************************************/
+
+#include "npfs.h"
+
+#define NDEBUG
+#include <debug.h>
+
+/* FUNCTIONS *****************************************************************/
+
+static NTSTATUS
+NpfsQueryDirectory(PNPFS_CCB Ccb,
+                   PIRP Irp,
+                   PULONG Size)
+{
+    PIO_STACK_LOCATION Stack;
+    LONG BufferLength = 0;
+    PUNICODE_STRING SearchPattern = NULL;
+    FILE_INFORMATION_CLASS FileInformationClass;
+    ULONG FileIndex = 0;
+    PUCHAR Buffer = NULL;
+    BOOLEAN First = FALSE;
+    PLIST_ENTRY CurrentEntry;
+    PNPFS_VCB Vcb;
+    PNPFS_FCB PipeFcb;
+    ULONG PipeIndex;
+    BOOLEAN Found = FALSE;
+    NTSTATUS Status = STATUS_SUCCESS;
+    PFILE_NAMES_INFORMATION NamesBuffer;
+    PFILE_DIRECTORY_INFORMATION DirectoryBuffer;
+
+    Stack = IoGetCurrentIrpStackLocation(Irp);
+
+    /* Obtain the callers parameters */
+    BufferLength = Stack->Parameters.QueryDirectory.Length;
+    SearchPattern = Stack->Parameters.QueryDirectory.FileName;
+    FileInformationClass = Stack->Parameters.QueryDirectory.FileInformationClass;
+    FileIndex = Stack->Parameters.QueryDirectory.FileIndex;
+
+    DPRINT("SearchPattern: %p  '%wZ'\n", SearchPattern,
SearchPattern);
+
+    /* Determine Buffer for result */
+    if (Irp->MdlAddress)
+    {
+        Buffer = MmGetSystemAddressForMdl(Irp->MdlAddress);
+    }
+    else
+    {
+        Buffer = Irp->UserBuffer;
+    }
+
+    /* Build the search pattern string */
+    DPRINT("Ccb->u.Directory.SearchPattern.Buffer: %p\n",
Ccb->u.Directory.SearchPattern.Buffer);
+    if (Ccb->u.Directory.SearchPattern.Buffer == NULL)
+    {
+        First = TRUE;
+
+        if (SearchPattern != NULL)
+        {
+            Ccb->u.Directory.SearchPattern.Buffer =
+                ExAllocatePool(NonPagedPool, SearchPattern->Length + sizeof(WCHAR));
+            if (Ccb->u.Directory.SearchPattern.Buffer == NULL)
+            {
+                return STATUS_INSUFFICIENT_RESOURCES;
+            }
+
+            Ccb->u.Directory.SearchPattern.Length = SearchPattern->Length;
+            Ccb->u.Directory.SearchPattern.MaximumLength = SearchPattern->Length +
sizeof(WCHAR);
+            RtlCopyMemory(Ccb->u.Directory.SearchPattern.Buffer,
+                          SearchPattern->Buffer,
+                          SearchPattern->Length);
+            Ccb->u.Directory.SearchPattern.Buffer[SearchPattern->Length /
sizeof(WCHAR)] = 0;
+        }
+        else
+        {
+            Ccb->u.Directory.SearchPattern.Buffer =
+                ExAllocatePool(NonPagedPool, 2 * sizeof(WCHAR));
+            if (Ccb->u.Directory.SearchPattern.Buffer == NULL)
+            {
+                return STATUS_INSUFFICIENT_RESOURCES;
+            }
+
+            Ccb->u.Directory.SearchPattern.Length = sizeof(WCHAR);
+            Ccb->u.Directory.SearchPattern.MaximumLength = 2 * sizeof(WCHAR);
+            Ccb->u.Directory.SearchPattern.Buffer[0] = L'*';
+            Ccb->u.Directory.SearchPattern.Buffer[1] = 0;
+        }
+    }
+    DPRINT("Search pattern: '%wZ'\n",
&Ccb->u.Directory.SearchPattern);
+
+    /* Determine the file index */
+    if (First || (Stack->Flags & SL_RESTART_SCAN))
+    {
+        FileIndex = 0;
+    } else if ((Stack->Flags & SL_INDEX_SPECIFIED) == 0)
+    {
+        FileIndex = Ccb->u.Directory.FileIndex + 1;
+    }
+    DPRINT("FileIndex: %lu\n", FileIndex);
+
+    DPRINT("Buffer = %p  tofind = %wZ\n", Buffer,
&Ccb->u.Directory.SearchPattern);
+
+    PipeIndex = 0;
+
+    Vcb = Ccb->Fcb->Vcb;
+    CurrentEntry = Vcb->PipeListHead.Flink;
+    while (CurrentEntry != &Vcb->PipeListHead && Found == FALSE)
+    {
+        /* Get the FCB of the next pipe */
+        PipeFcb = CONTAINING_RECORD(CurrentEntry,
+                                    NPFS_FCB,
+                                    PipeListEntry);
+
+        /* Make sure it is a pipe FCB */
+        ASSERT(PipeFcb->Type == FCB_PIPE);
+
+        DPRINT("PipeName: %wZ\n", &PipeFcb->PipeName);
+
+        if (FsRtlIsNameInExpression(&Ccb->u.Directory.SearchPattern,
+                                    &PipeFcb->PipeName,
+                                    TRUE,
+                                    NULL))
+        {
+            DPRINT("Found pipe: %wZ\n", &PipeFcb->PipeName);
+
+            if (PipeIndex >= FileIndex)
+            {
+                switch (FileInformationClass)
+                {
+                    case FileDirectoryInformation:
+                        DirectoryBuffer = (PFILE_DIRECTORY_INFORMATION)Buffer;
+                        DirectoryBuffer->NextEntryOffset = 0;
+                        DirectoryBuffer->FileIndex = PipeIndex;
+                        DirectoryBuffer->FileAttributes = FILE_ATTRIBUTE_NORMAL;
+                        DirectoryBuffer->EndOfFile.QuadPart =
PipeFcb->CurrentInstances;
+                        DirectoryBuffer->AllocationSize.LowPart =
PipeFcb->MaximumInstances;
+                        DirectoryBuffer->FileNameLength =
PipeFcb->PipeName.Length;
+                        RtlCopyMemory(DirectoryBuffer->FileName,
+                                      PipeFcb->PipeName.Buffer,
+                                      PipeFcb->PipeName.Length);
+                        *Size = sizeof(FILE_DIRECTORY_INFORMATION) +
PipeFcb->PipeName.Length - 1;
+                        Status = STATUS_SUCCESS;
+                        break;
+
+                    case FileNamesInformation:
+                        NamesBuffer = (PFILE_NAMES_INFORMATION)Buffer;
+                        NamesBuffer->NextEntryOffset = 0;
+                        NamesBuffer->FileIndex = PipeIndex;
+                        NamesBuffer->FileNameLength = PipeFcb->PipeName.Length;
+                        RtlCopyMemory(NamesBuffer->FileName,
+                                      PipeFcb->PipeName.Buffer,
+                                      PipeFcb->PipeName.Length);
+                        *Size = sizeof(FILE_NAMES_INFORMATION) +
PipeFcb->PipeName.Length - 1;
+                        Status = STATUS_SUCCESS;
+                        break;
+
+                    default:
+                        DPRINT1("Invalid information class: %lu\n",
FileInformationClass);
+                        Status = STATUS_INVALID_INFO_CLASS;
+                        break;
+                }
+
+                Ccb->u.Directory.FileIndex = PipeIndex;
+                Found = TRUE;
+
+//                if (Stack->Flags & SL_RETURN_SINGLE_ENTRY)
+//                    return STATUS_SUCCESS;
+
+                break;
+            }
+
+            PipeIndex++;
+        }
+
+        CurrentEntry = CurrentEntry->Flink;
+    }
+
+    if (Found == FALSE)
+        Status = STATUS_NO_MORE_FILES;
+
+    return Status;
+}
+
+
+NTSTATUS NTAPI
+NpfsDirectoryControl(PDEVICE_OBJECT DeviceObject,
+                     PIRP Irp)
+{
+    PIO_STACK_LOCATION IoStack;
+    PFILE_OBJECT FileObject;
+    PNPFS_CCB Ccb;
+    PNPFS_FCB Fcb;
+    NTSTATUS Status;
+    ULONG Size = 0;
+
+    DPRINT("NpfsDirectoryControl() called\n");
+
+    IoStack = IoGetCurrentIrpStackLocation(Irp);
+
+    FileObject = IoStack->FileObject;
+
+    if (NpfsGetCcb(FileObject, &Ccb) != CCB_DIRECTORY)
+    {
+        Status = STATUS_INVALID_PARAMETER;
+
+        Irp->IoStatus.Status = Status;
+        Irp->IoStatus.Information = 0;
+
+        IoCompleteRequest(Irp, IO_NO_INCREMENT);
+
+        return Status;
+    }
+
+    Fcb = Ccb->Fcb;
+
+    switch (IoStack->MinorFunction)
+    {
+        case IRP_MN_QUERY_DIRECTORY:
+            Status = NpfsQueryDirectory(Ccb,
+                                        Irp,
+                                        &Size);
+            break;
+
+        case IRP_MN_NOTIFY_CHANGE_DIRECTORY:
+            DPRINT1("IRP_MN_NOTIFY_CHANGE_DIRECTORY\n");
+            Status = STATUS_NOT_IMPLEMENTED;
+            break;
+
+        default:
+            DPRINT1("NPFS: MinorFunction %d\n", IoStack->MinorFunction);
+            Status = STATUS_INVALID_DEVICE_REQUEST;
+            break;
+    }
+
+    Irp->IoStatus.Status = Status;
+    Irp->IoStatus.Information = Size;
+
+    IoCompleteRequest(Irp, IO_NO_INCREMENT);
+
+    return Status;
+}
+
+/* EOF */

Propchange: trunk/reactos/drivers/filesystems/npfs/dirctl.c
------------------------------------------------------------------------------
    svn:eol-style = native

Propchange: trunk/reactos/drivers/filesystems/npfs/dirctl.c
------------------------------------------------------------------------------
    svn:keywords = author date id revision

Modified: trunk/reactos/drivers/filesystems/npfs/npfs.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/filesystems/npfs/n…
==============================================================================
--- trunk/reactos/drivers/filesystems/npfs/npfs.c [iso-8859-1] (original)
+++ trunk/reactos/drivers/filesystems/npfs/npfs.c [iso-8859-1] Sat Jan  1 12:40:24 2011
@@ -44,8 +44,8 @@
         NpfsQueryVolumeInformation;
     DriverObject->MajorFunction[IRP_MJ_CLEANUP] = NpfsCleanup;
     DriverObject->MajorFunction[IRP_MJ_FLUSH_BUFFERS] = NpfsFlushBuffers;
-    //   DriverObject->MajorFunction[IRP_MJ_DIRECTORY_CONTROL] =
-    //     NpfsDirectoryControl;
+    DriverObject->MajorFunction[IRP_MJ_DIRECTORY_CONTROL] =
+        NpfsDirectoryControl;
     DriverObject->MajorFunction[IRP_MJ_FILE_SYSTEM_CONTROL] =
         NpfsFileSystemControl;
     //   DriverObject->MajorFunction[IRP_MJ_QUERY_SECURITY] =
@@ -100,4 +100,54 @@
     return STATUS_SUCCESS;
 }
 
+
+FCB_TYPE
+NpfsGetFcb(PFILE_OBJECT FileObject,
+           PNPFS_FCB *Fcb)
+{
+    PNPFS_FCB LocalFcb = NULL;
+    FCB_TYPE FcbType = FCB_INVALID;
+
+    _SEH2_TRY
+    {
+        LocalFcb = (PNPFS_FCB)FileObject->FsContext;
+        FcbType = LocalFcb->Type;
+    }
+    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
+    {
+        LocalFcb = NULL;
+        FcbType = FCB_INVALID;
+    }
+    _SEH2_END;
+
+    *Fcb = LocalFcb;
+
+    return FcbType;
+}
+
+
+CCB_TYPE
+NpfsGetCcb(PFILE_OBJECT FileObject,
+           PNPFS_CCB *Ccb)
+{
+    PNPFS_CCB LocalCcb = NULL;
+    CCB_TYPE CcbType = CCB_INVALID;
+
+    _SEH2_TRY
+    {
+        LocalCcb = (PNPFS_CCB)FileObject->FsContext2;
+        CcbType = LocalCcb->Type;
+    }
+    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
+    {
+        LocalCcb = NULL;
+        CcbType = CCB_INVALID;
+    }
+    _SEH2_END;
+
+    *Ccb = LocalCcb;
+
+    return CcbType;
+}
+
 /* EOF */

Modified: trunk/reactos/drivers/filesystems/npfs/npfs.h
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/filesystems/npfs/n…
==============================================================================
--- trunk/reactos/drivers/filesystems/npfs/npfs.h [iso-8859-1] (original)
+++ trunk/reactos/drivers/filesystems/npfs/npfs.h [iso-8859-1] Sat Jan  1 12:40:24 2011
@@ -3,9 +3,11 @@
 
 #include <ntifs.h>
 #include <ndk/iotypes.h>
+#include <pseh/pseh2.h>
 
 typedef enum _FCB_TYPE
 {
+    FCB_INVALID,
     FCB_DEVICE,
     FCB_DIRECTORY,
     FCB_PIPE
@@ -13,6 +15,7 @@
 
 typedef enum _CCB_TYPE
 {
+    CCB_INVALID,
     CCB_DEVICE,
     CCB_DIRECTORY,
     CCB_PIPE
@@ -55,6 +58,14 @@
     LARGE_INTEGER TimeOut;
 } NPFS_FCB, *PNPFS_FCB;
 
+
+typedef struct _NPFS_CCB_DIRECTORY_DATA
+{
+    UNICODE_STRING SearchPattern;
+    ULONG FileIndex;
+} NPFS_CCB_DIRECTORY_DATA, *PNPFS_CCB_DIRECTORY_DATA;
+
+
 typedef struct _NPFS_CCB
 {
     LIST_ENTRY CcbListEntry;
@@ -79,6 +90,12 @@
     ULONG MaxDataLength;
 
     FAST_MUTEX DataListLock;    /* Data queue lock */
+
+    union
+    {
+        NPFS_CCB_DIRECTORY_DATA Directory;
+    } u;
+
 } NPFS_CCB, *PNPFS_CCB;
 
 typedef struct _NPFS_CONTEXT
@@ -130,6 +147,9 @@
 DRIVER_DISPATCH NpfsClose;
 NTSTATUS NTAPI NpfsClose(PDEVICE_OBJECT DeviceObject, PIRP Irp);
 
+DRIVER_DISPATCH NpfsDirectoryControl;
+NTSTATUS NTAPI NpfsDirectoryControl(PDEVICE_OBJECT DeviceObject, PIRP Irp);
+
 DRIVER_DISPATCH NpfsRead;
 NTSTATUS NTAPI NpfsRead(PDEVICE_OBJECT DeviceObject, PIRP Irp);
 
@@ -159,5 +179,12 @@
 NpfsFindPipe(PNPFS_VCB Vcb,
              PUNICODE_STRING PipeName);
 
+FCB_TYPE
+NpfsGetFcb(PFILE_OBJECT FileObject,
+           PNPFS_FCB *Fcb);
+
+CCB_TYPE
+NpfsGetCcb(PFILE_OBJECT FileObject,
+           PNPFS_CCB *Ccb);
 
 #endif /* __DRIVERS_FS_NP_NPFS_H */

Modified: trunk/reactos/drivers/filesystems/npfs/npfs.rbuild
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/filesystems/npfs/n…
==============================================================================
--- trunk/reactos/drivers/filesystems/npfs/npfs.rbuild [iso-8859-1] (original)
+++ trunk/reactos/drivers/filesystems/npfs/npfs.rbuild [iso-8859-1] Sat Jan  1 12:40:24
2011
@@ -4,7 +4,9 @@
 	<include base="npfs">.</include>
 	<library>ntoskrnl</library>
 	<library>hal</library>
+	<library>pseh</library>
 	<file>create.c</file>
+	<file>dirctl.c</file>
 	<file>finfo.c</file>
 	<file>fsctrl.c</file>
 	<file>npfs.c</file>