Git conversion: Make reactos the root directory, move rosapps, rostests, wallpapers...

[reactos.git] / drivers / ksfilter / ks / deviceinterface.c

diff --git a/drivers/ksfilter/ks/deviceinterface.c b/drivers/ksfilter/ks/deviceinterface.c
new file mode 100644 (file)
index 0000000..36fe943
--- /dev/null
+++ b/drivers/ksfilter/ks/deviceinterface.c
@@ -0,0 +1,138 @@
+#include "precomp.h"
+
+#define NDEBUG
+#include <debug.h>
+
+NTSTATUS
+KspSetDeviceInterfacesState(
+    IN PLIST_ENTRY ListHead,
+    IN BOOL Enable)
+{
+    NTSTATUS Status = STATUS_SUCCESS;
+    PLIST_ENTRY Entry;
+    PSYMBOLIC_LINK_ENTRY SymEntry;
+
+
+    Entry = ListHead->Flink;
+    while(Entry != ListHead)
+    {
+        /* fetch symbolic link entry */
+        SymEntry = (PSYMBOLIC_LINK_ENTRY)CONTAINING_RECORD(Entry, SYMBOLIC_LINK_ENTRY, Entry);
+        /* set device interface state */
+        Status = IoSetDeviceInterfaceState(&SymEntry->SymbolicLink, Enable);
+
+        DPRINT("KspSetDeviceInterfacesState SymbolicLink '%S' Status %lx\n", SymEntry->SymbolicLink.Buffer, Status, Enable);
+
+        /* check for success */
+        if (!NT_SUCCESS(Status))
+            return Status;
+        /* get next entry */
+        Entry = Entry->Flink;
+    }
+    /* return result */
+    return Status;
+}
+
+NTSTATUS
+KspFreeDeviceInterfaces(
+    IN PLIST_ENTRY ListHead)
+{
+    PLIST_ENTRY Entry;
+    PSYMBOLIC_LINK_ENTRY SymEntry;
+
+    while(!IsListEmpty(ListHead))
+    {
+        /* remove first entry */
+        Entry = RemoveHeadList(ListHead);
+
+        /* fetch symbolic link entry */
+        SymEntry = (PSYMBOLIC_LINK_ENTRY)CONTAINING_RECORD(Entry, SYMBOLIC_LINK_ENTRY, Entry);
+
+        /* free device interface string */
+        RtlFreeUnicodeString(&SymEntry->SymbolicLink);
+        /* free entry item */
+        FreeItem(Entry);
+    }
+
+    return STATUS_SUCCESS;
+}
+
+NTSTATUS
+KspRegisterDeviceInterfaces(
+    IN PDEVICE_OBJECT PhysicalDeviceObject,
+    IN ULONG CategoriesCount,
+    IN GUID const*Categories,
+    IN PUNICODE_STRING ReferenceString,
+    OUT PLIST_ENTRY SymbolicLinkList)
+{
+    ULONG Index;
+    NTSTATUS Status = STATUS_SUCCESS;
+    PSYMBOLIC_LINK_ENTRY SymEntry;
+
+    for(Index = 0; Index < CategoriesCount; Index++)
+    {
+        /* allocate a symbolic link entry */
+        SymEntry = AllocateItem(NonPagedPool, sizeof(SYMBOLIC_LINK_ENTRY));
+        /* check for success */
+        if (!SymEntry)
+            return STATUS_INSUFFICIENT_RESOURCES;
+
+        /* now register device interface */
+        Status = IoRegisterDeviceInterface(PhysicalDeviceObject, 
+                                           &Categories[Index],
+                                           ReferenceString,
+                                           &SymEntry->SymbolicLink);
+
+        if (!NT_SUCCESS(Status))
+        {
+            DPRINT1("Failed to register device interface %x\n", Status);
+
+            /* free entry */
+            FreeItem(SymEntry);
+
+            /* return result */
+            return Status;
+        }
+
+        /* copy device class */
+        RtlMoveMemory(&SymEntry->DeviceInterfaceClass, &Categories[Index], sizeof(CLSID));
+
+        /* insert symbolic link entry */
+        InsertTailList(SymbolicLinkList, &SymEntry->Entry);
+    }
+
+    /* return result */
+    return Status;
+}
+
+NTSTATUS
+KspSetFilterFactoriesState(
+    IN PKSIDEVICE_HEADER DeviceHeader,
+    IN BOOLEAN NewState)
+{
+    PCREATE_ITEM_ENTRY CreateEntry;
+    PLIST_ENTRY Entry;
+    NTSTATUS Status = STATUS_SUCCESS;
+
+    /* grab first device interface */
+    Entry = DeviceHeader->ItemList.Flink;
+    while(Entry != &DeviceHeader->ItemList && Status == STATUS_SUCCESS)
+    {
+        /* grab create entry */
+        CreateEntry = CONTAINING_RECORD(Entry, CREATE_ITEM_ENTRY, Entry);
+
+        /* sanity check */
+        ASSERT(CreateEntry->CreateItem);
+
+        if (CreateEntry->CreateItem->Create == IKsFilterFactory_Create)
+        {
+            /* found our own filterfactory */
+            Status = KsFilterFactorySetDeviceClassesState((PKSFILTERFACTORY)CreateEntry->CreateItem->Context, NewState);
+        }
+
+        Entry = Entry->Flink;
+    }
+
+    /* store result */
+    return Status;
+}