[NTOSKRNL]

- Implemented FsRtlLookupPerStreamContextInternal(), FsRtlInsertPerStreamContext(), FsRtlRemovePerStreamContext(), FsRtlTeardownPerStreamContexts()
Based on my previous work on pierre-fsd branch.

svn path=/trunk/; revision=49716

diff --git a/reactos/ntoskrnl/fsrtl/filtrctx.c b/reactos/ntoskrnl/fsrtl/filtrctx.c
index 9d5bdf6..458d907 100644 (file)
--- a/reactos/ntoskrnl/fsrtl/filtrctx.c
+++ b/reactos/ntoskrnl/fsrtl/filtrctx.c
@@ -3,7 +3,7 @@
  * LICENSE:         GPL - See COPYING in the top level directory
  * FILE:            ntoskrnl/fsrtl/filtrctx.c
  * PURPOSE:         File Stream Filter Context support for File System Drivers
- * PROGRAMMERS:     None.
+ * PROGRAMMERS:     Pierre Schweitzer (pierre.schweitzer@reactos.org)
  */
 
 /* INCLUDES ******************************************************************/
@@ -36,16 +36,51 @@ FsRtlIsPagingFile(IN PFILE_OBJECT FileObject)
 }
 
 /*
- * @unimplemented
+ * @implemented
  */
 PFSRTL_PER_STREAM_CONTEXT
 NTAPI
-FsRtlLookupPerStreamContextInternal(IN PFSRTL_ADVANCED_FCB_HEADER StreamContext,
+FsRtlLookupPerStreamContextInternal(IN PFSRTL_ADVANCED_FCB_HEADER AdvFcbHeader,
                                     IN PVOID OwnerId OPTIONAL,
                                     IN PVOID InstanceId OPTIONAL)
 {
-    KeBugCheck(FILE_SYSTEM);
-    return FALSE;
+    PLIST_ENTRY NextEntry;
+    PFSRTL_PER_STREAM_CONTEXT TmpPerStreamContext, PerStreamContext = NULL;
+
+    ASSERT(AdvFcbHeader);
+    ASSERT(FlagOn(AdvFcbHeader->Flags2, FSRTL_FLAG2_SUPPORTS_FILTER_CONTEXTS));
+
+    ExAcquireFastMutex(AdvFcbHeader->FastMutex);
+
+    /* If list is empty, no need to browse it */
+    if (!IsListEmpty(&(AdvFcbHeader->FilterContexts)))
+    {
+        for (NextEntry = AdvFcbHeader->FilterContexts.Flink;
+             NextEntry != &(AdvFcbHeader->FilterContexts);
+             NextEntry = NextEntry->Flink)
+        {
+            /* If we don't have any criteria for search, first entry will be enough */
+            if (!OwnerId && !InstanceId)
+            {
+                PerStreamContext = (PFSRTL_PER_STREAM_CONTEXT)NextEntry;
+                break;
+            }
+            /* Else, we've to find something that matches with the parameters. */
+            else
+            {
+                TmpPerStreamContext = CONTAINING_RECORD(NextEntry, FSRTL_PER_STREAM_CONTEXT, Links);
+                if ((InstanceId && TmpPerStreamContext->InstanceId == InstanceId && TmpPerStreamContext->OwnerId == OwnerId) ||
+                    (OwnerId && TmpPerStreamContext->OwnerId == OwnerId))
+                {
+                    PerStreamContext = TmpPerStreamContext;
+                    break;
+                }
+            }
+        }
+    }
+    ExReleaseFastMutex(AdvFcbHeader->FastMutex);
+
+    return PerStreamContext;
 }
 
 /*
@@ -62,28 +97,77 @@ FsRtlLookupPerFileObjectContext(IN PFILE_OBJECT FileObject,
 }
 
 /*
- * @unimplemented
+ * @implemented
  */
 NTSTATUS
 NTAPI
-FsRtlInsertPerStreamContext(IN PFSRTL_ADVANCED_FCB_HEADER PerStreamContext,
-                            IN PFSRTL_PER_STREAM_CONTEXT Ptr)
+FsRtlInsertPerStreamContext(IN PFSRTL_ADVANCED_FCB_HEADER AdvFcbHeader,
+                            IN PFSRTL_PER_STREAM_CONTEXT PerStreamContext)
 {
-    KeBugCheck(FILE_SYSTEM);
-    return STATUS_NOT_IMPLEMENTED;
+    if (!(AdvFcbHeader) || !(AdvFcbHeader->Flags2 & FSRTL_FLAG2_SUPPORTS_FILTER_CONTEXTS))
+    {
+        return STATUS_INVALID_DEVICE_REQUEST;
+    }
+
+    ExAcquireFastMutex(AdvFcbHeader->FastMutex);
+    InsertHeadList(&(AdvFcbHeader->FilterContexts), &(PerStreamContext->Links));
+    ExReleaseFastMutex(AdvFcbHeader->FastMutex);
+    return STATUS_SUCCESS;
 }
 
 /*
- * @unimplemented
+ * @implemented
  */
 PFSRTL_PER_STREAM_CONTEXT
 NTAPI
-FsRtlRemovePerStreamContext(IN PFSRTL_ADVANCED_FCB_HEADER StreamContext,
+FsRtlRemovePerStreamContext(IN PFSRTL_ADVANCED_FCB_HEADER AdvFcbHeader,
                             IN PVOID OwnerId OPTIONAL,
                             IN PVOID InstanceId OPTIONAL)
 {
-    KeBugCheck(FILE_SYSTEM);
-    return NULL;
+    PLIST_ENTRY NextEntry;
+    PFSRTL_PER_STREAM_CONTEXT TmpPerStreamContext, PerStreamContext = NULL;
+
+    if (!(AdvFcbHeader) || !(AdvFcbHeader->Flags2 & FSRTL_FLAG2_SUPPORTS_FILTER_CONTEXTS))
+    {
+        return NULL;
+    }
+
+    ExAcquireFastMutex(AdvFcbHeader->FastMutex);
+    /* If list is empty, no need to browse it */
+    if (!IsListEmpty(&(AdvFcbHeader->FilterContexts)))
+    {
+        for (NextEntry = AdvFcbHeader->FilterContexts.Flink;
+             NextEntry != &(AdvFcbHeader->FilterContexts);
+             NextEntry = NextEntry->Flink)
+        {
+            /* If we don't have any criteria for search, first entry will be enough */
+            if (!OwnerId && !InstanceId)
+            {
+                PerStreamContext = (PFSRTL_PER_STREAM_CONTEXT)NextEntry;
+                break;
+            }
+            /* Else, we've to find something that matches with the parameters. */
+            else
+            {
+                TmpPerStreamContext = CONTAINING_RECORD(NextEntry, FSRTL_PER_STREAM_CONTEXT, Links);
+                if ((InstanceId && TmpPerStreamContext->InstanceId == InstanceId && TmpPerStreamContext->OwnerId == OwnerId) ||
+                    (OwnerId && TmpPerStreamContext->OwnerId == OwnerId))
+                {
+                    PerStreamContext = TmpPerStreamContext;
+                    break;
+                }
+            }
+        }
+        /* Finally remove entry from list */
+        if (PerStreamContext)
+        {
+            RemoveEntryList(&(PerStreamContext->Links));
+        }
+    }
+    ExReleaseFastMutex(AdvFcbHeader->FastMutex);
+
+    return PerStreamContext;
+
 }
 
 /*
@@ -112,12 +196,49 @@ FsRtlRemovePerFileObjectContext(IN PFILE_OBJECT PerFileObjectContext,
 }
 
 /*
- * @unimplemented
+ * @implemented
  */
 VOID
 NTAPI
-FsRtlTeardownPerStreamContexts(IN PFSRTL_ADVANCED_FCB_HEADER AdvancedHeader)
+FsRtlTeardownPerStreamContexts(IN PFSRTL_ADVANCED_FCB_HEADER AdvFcbHeader)
 {
-    KeBugCheck(FILE_SYSTEM);
-}
+    PLIST_ENTRY NextEntry;
+    BOOLEAN IsMutexLocked = FALSE;
+    PFSRTL_PER_STREAM_CONTEXT PerStreamContext;
+
+    _SEH2_TRY
+    {
+        /* Acquire mutex to deal with the list */
+        ExAcquireFastMutex(AdvFcbHeader->FastMutex);
+        IsMutexLocked = TRUE;
+
+        /* While there are items... */
+        while (!IsListEmpty(&(AdvFcbHeader->FilterContexts)))
+        {
+            /* ...remove one */
+            NextEntry = RemoveHeadList(&(AdvFcbHeader->FilterContexts));
+            PerStreamContext = CONTAINING_RECORD(NextEntry, FSRTL_PER_STREAM_CONTEXT, Links);
 
+            /* Release mutex before calling callback */
+            ExReleaseFastMutex(AdvFcbHeader->FastMutex);
+            IsMutexLocked = FALSE;
+
+            /* Call the callback */
+            ASSERT(PerStreamContext->FreeCallback);
+            (*PerStreamContext->FreeCallback)(PerStreamContext);
+
+            /* Relock the list to continue */
+            ExAcquireFastMutex(AdvFcbHeader->FastMutex);
+            IsMutexLocked = TRUE;
+        }
+    }
+    _SEH2_FINALLY
+    {
+        /* If mutex was locked, release */
+        if (IsMutexLocked)
+        {
+            ExReleaseFastMutex(AdvFcbHeader->FastMutex);
+        }
+    }
+    _SEH2_END;
+}