#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;
}
