dirctl.c File Reference

#include "ntfs.h"
#include <debug.h>

Include dependency graph for dirctl.c:

Go to the source code of this file.

Macros
#define	NDEBUG
#define	ULONG_ROUND_UP(x)   ROUND_UP((x), (sizeof(ULONG)))

Functions
ULONGLONG	NtfsGetFileSize (PDEVICE_EXTENSION DeviceExt, PFILE_RECORD_HEADER FileRecord, PCWSTR Stream, ULONG StreamLength, PULONGLONG AllocatedSize)
static NTSTATUS	NtfsGetNamesInformation (PDEVICE_EXTENSION DeviceExt, PFILE_RECORD_HEADER FileRecord, ULONGLONG MFTIndex, PFILE_NAMES_INFORMATION Info, ULONG BufferLength, PULONG Written, BOOLEAN First, PCUNICODE_STRING OverrideName)
static NTSTATUS	NtfsGetDirectoryInformation (PDEVICE_EXTENSION DeviceExt, PFILE_RECORD_HEADER FileRecord, ULONGLONG MFTIndex, PFILE_DIRECTORY_INFORMATION Info, ULONG BufferLength, PULONG Written, BOOLEAN First, PCUNICODE_STRING OverrideName)
static NTSTATUS	NtfsGetFullDirectoryInformation (PDEVICE_EXTENSION DeviceExt, PFILE_RECORD_HEADER FileRecord, ULONGLONG MFTIndex, PFILE_FULL_DIRECTORY_INFORMATION Info, ULONG BufferLength, PULONG Written, BOOLEAN First, PCUNICODE_STRING OverrideName)
static NTSTATUS	NtfsGetBothDirectoryInformation (PDEVICE_EXTENSION DeviceExt, PFILE_RECORD_HEADER FileRecord, ULONGLONG MFTIndex, PFILE_BOTH_DIR_INFORMATION Info, ULONG BufferLength, PULONG Written, BOOLEAN First, PCUNICODE_STRING OverrideName)
static BOOLEAN	NtfsDotNameMatches (PUNICODE_STRING Pattern, PUNICODE_STRING Name, BOOLEAN CaseSensitive)
static NTSTATUS	NtfsGetDirEntryInformation (PDEVICE_EXTENSION DeviceExt, FILE_INFORMATION_CLASS FileInformationClass, PFILE_RECORD_HEADER FileRecord, ULONGLONG MFTIndex, PCUNICODE_STRING OverrideName, PVOID Buffer, ULONG BufferLength, PULONG Written, BOOLEAN First)
static NTSTATUS	NtfsGetDotEntry (PDEVICE_EXTENSION DeviceExt, ULONGLONG MFTIndex, PCUNICODE_STRING Name, FILE_INFORMATION_CLASS FileInformationClass, PVOID Buffer, ULONG BufferLength, PULONG Written, BOOLEAN First)
static ULONGLONG	NtfsGetParentMftIndex (PDEVICE_EXTENSION DeviceExt, ULONGLONG MFTIndex)
NTSTATUS	NtfsQueryDirectory (PNTFS_IRP_CONTEXT IrpContext)
NTSTATUS	NtfsDirectoryControl (PNTFS_IRP_CONTEXT IrpContext)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 32 of file dirctl.c.

ULONG_ROUND_UP

#define ULONG_ROUND_UP

(

x

)

ROUND_UP((x), (sizeof(ULONG)))

Definition at line 63 of file dirctl.c.

Function Documentation

NtfsDirectoryControl()

NTSTATUS NtfsDirectoryControl

(

PNTFS_IRP_CONTEXT

IrpContext

)

Definition at line 793 of file dirctl.c.

{
    NTSTATUS Status = STATUS_UNSUCCESSFUL;
 
    DPRINT("NtfsDirectoryControl() called\n");
 
    /* Handlers that fill the caller's buffer overwrite this with the byte count */
    IrpContext->Irp->IoStatus.Information = 0;
 
    switch (IrpContext->MinorFunction)
    {
        case IRP_MN_QUERY_DIRECTORY:
            Status = NtfsQueryDirectory(IrpContext);
            break;
 
        case IRP_MN_NOTIFY_CHANGE_DIRECTORY:
            DPRINT1("IRP_MN_NOTIFY_CHANGE_DIRECTORY\n");
            Status = STATUS_NOT_IMPLEMENTED;
            break;
 
        default:
            Status = STATUS_INVALID_DEVICE_REQUEST;
            break;
    }
 
    if (Status == STATUS_PENDING && IrpContext->Flags & IRPCONTEXT_COMPLETE)
    {
        return NtfsMarkIrpContextForQueue(IrpContext);
    }
 
    return Status;
}

Referenced by NtfsDispatch().

NtfsDotNameMatches()

static BOOLEAN NtfsDotNameMatches ( PUNICODE_STRING  Pattern, PUNICODE_STRING  Name, BOOLEAN  CaseSensitive  )

static

Definition at line 413 of file dirctl.c.

{
    UNICODE_STRING Upcased;
    BOOLEAN Ret;
 
    if (!FsRtlDoesNameContainWildCards(Pattern))
    {
        return (RtlCompareUnicodeString(Pattern, Name, !CaseSensitive) == 0);
    }
 
    /* FsRtlIsNameInExpression() wants the expression upcased when matching
     * case-insensitively. "." and ".." need no upcasing themselves. */
    if (CaseSensitive)
    {
        return FsRtlIsNameInExpression(Pattern, Name, FALSE, NULL);
    }
 
    if (!NT_SUCCESS(RtlUpcaseUnicodeString(&Upcased, Pattern, TRUE)))
    {
        return FALSE;
    }
 
    Ret = FsRtlIsNameInExpression(&Upcased, Name, TRUE, NULL);
    RtlFreeUnicodeString(&Upcased);
 
    return Ret;
}

Referenced by NtfsQueryDirectory().

NtfsGetBothDirectoryInformation()

static NTSTATUS NtfsGetBothDirectoryInformation ( PDEVICE_EXTENSION  DeviceExt, PFILE_RECORD_HEADER  FileRecord, ULONGLONG  MFTIndex, PFILE_BOTH_DIR_INFORMATION  Info, ULONG  BufferLength, PULONG  Written, BOOLEAN  First, PCUNICODE_STRING  OverrideName  )

static

Definition at line 307 of file dirctl.c.

{
    ULONG Length;
    NTSTATUS Status;
    ULONG BytesToCopy = 0;
    PFILENAME_ATTRIBUTE FileName, ShortFileName;
    PCWSTR NameBuffer;
    PSTANDARD_INFORMATION StdInfo;
 
    DPRINT("NtfsGetBothDirectoryInformation() called\n");
 
    *Written = 0;
    Status = STATUS_BUFFER_OVERFLOW;
    if (FIELD_OFFSET(FILE_BOTH_DIR_INFORMATION, FileName) > BufferLength)
    {
        return Status;
    }
 
    FileName = GetBestFileNameFromRecord(DeviceExt, FileRecord);
    if (FileName == NULL)
    {
        DPRINT1("No name information for file ID: %#I64x\n", MFTIndex);
        NtfsDumpFileAttributes(DeviceExt, FileRecord);
        return STATUS_OBJECT_NAME_NOT_FOUND;
    }
    ShortFileName = GetFileNameFromRecord(DeviceExt, FileRecord, NTFS_FILE_NAME_DOS);
 
    StdInfo = GetStandardInformationFromRecord(DeviceExt, FileRecord);
    ASSERT(StdInfo != NULL);
 
    if (OverrideName != NULL)
    {
        NameBuffer = OverrideName->Buffer;
        Length = OverrideName->Length;
    }
    else
    {
        NameBuffer = FileName->Name;
        Length = FileName->NameLength * sizeof (WCHAR);
    }
 
    if (First || (BufferLength >= FIELD_OFFSET(FILE_BOTH_DIR_INFORMATION, FileName) + Length))
    {
        Info->FileNameLength = Length;
 
        *Written = FIELD_OFFSET(FILE_BOTH_DIR_INFORMATION, FileName);
        Info->NextEntryOffset = 0;
        if (BufferLength > FIELD_OFFSET(FILE_BOTH_DIR_INFORMATION, FileName))
        {
            BytesToCopy = min(Length, BufferLength - FIELD_OFFSET(FILE_BOTH_DIR_INFORMATION, FileName));
            RtlCopyMemory(Info->FileName, NameBuffer, BytesToCopy);
            *Written += BytesToCopy;
 
            if (BytesToCopy == Length)
            {
                Info->NextEntryOffset = ULONG_ROUND_UP(sizeof(FILE_BOTH_DIR_INFORMATION) +
                                                       BytesToCopy);
                Status = STATUS_SUCCESS;
            }
        }
 
        if (ShortFileName && OverrideName == NULL)
        {
            /* Should we upcase the filename? */
            ASSERT(ShortFileName->NameLength <= ARRAYSIZE(Info->ShortName));
            Info->ShortNameLength = ShortFileName->NameLength * sizeof(WCHAR);
            RtlCopyMemory(Info->ShortName, ShortFileName->Name, Info->ShortNameLength);
        }
        else
        {
            Info->ShortName[0] = 0;
            Info->ShortNameLength = 0;
        }
 
        Info->CreationTime.QuadPart = FileName->CreationTime;
        Info->LastAccessTime.QuadPart = FileName->LastAccessTime;
        Info->LastWriteTime.QuadPart = FileName->LastWriteTime;
        Info->ChangeTime.QuadPart = FileName->ChangeTime;
 
        /* Convert file flags */
        NtfsFileFlagsToAttributes(FileName->FileAttributes | StdInfo->FileAttribute, &Info->FileAttributes);
 
        Info->EndOfFile.QuadPart = NtfsGetFileSize(DeviceExt, FileRecord, L"", 0, (PULONGLONG)&Info->AllocationSize.QuadPart);
 
        Info->FileIndex = MFTIndex;
        Info->EaSize = 0;
    }
 
    return Status;
}

Referenced by NtfsGetDirEntryInformation().

NtfsGetDirectoryInformation()

static NTSTATUS NtfsGetDirectoryInformation ( PDEVICE_EXTENSION  DeviceExt, PFILE_RECORD_HEADER  FileRecord, ULONGLONG  MFTIndex, PFILE_DIRECTORY_INFORMATION  Info, ULONG  BufferLength, PULONG  Written, BOOLEAN  First, PCUNICODE_STRING  OverrideName  )

static

Definition at line 136 of file dirctl.c.

{
    ULONG Length;
    NTSTATUS Status;
    ULONG BytesToCopy = 0;
    PFILENAME_ATTRIBUTE FileName;
    PCWSTR NameBuffer;
    PSTANDARD_INFORMATION StdInfo;
 
    DPRINT("NtfsGetDirectoryInformation() called\n");
 
    *Written = 0;
    Status = STATUS_BUFFER_OVERFLOW;
    if (FIELD_OFFSET(FILE_DIRECTORY_INFORMATION, FileName) > BufferLength)
    {
        return Status;
    }
 
    FileName = GetBestFileNameFromRecord(DeviceExt, FileRecord);
    if (FileName == NULL)
    {
        DPRINT1("No name information for file ID: %#I64x\n", MFTIndex);
        NtfsDumpFileAttributes(DeviceExt, FileRecord);
        return STATUS_OBJECT_NAME_NOT_FOUND;
    }
 
    StdInfo = GetStandardInformationFromRecord(DeviceExt, FileRecord);
    ASSERT(StdInfo != NULL);
 
    if (OverrideName != NULL)
    {
        NameBuffer = OverrideName->Buffer;
        Length = OverrideName->Length;
    }
    else
    {
        NameBuffer = FileName->Name;
        Length = FileName->NameLength * sizeof (WCHAR);
    }
 
    if (First || (BufferLength >= FIELD_OFFSET(FILE_DIRECTORY_INFORMATION, FileName) + Length))
    {
        Info->FileNameLength = Length;
 
        *Written = FIELD_OFFSET(FILE_DIRECTORY_INFORMATION, FileName);
        Info->NextEntryOffset = 0;
        if (BufferLength > FIELD_OFFSET(FILE_DIRECTORY_INFORMATION, FileName))
        {
            BytesToCopy = min(Length, BufferLength - FIELD_OFFSET(FILE_DIRECTORY_INFORMATION, FileName));
            RtlCopyMemory(Info->FileName, NameBuffer, BytesToCopy);
            *Written += BytesToCopy;
 
            if (BytesToCopy == Length)
            {
                Info->NextEntryOffset = ULONG_ROUND_UP(sizeof(FILE_DIRECTORY_INFORMATION) +
                                                       BytesToCopy);
                Status = STATUS_SUCCESS;
            }
        }
 
        Info->CreationTime.QuadPart = FileName->CreationTime;
        Info->LastAccessTime.QuadPart = FileName->LastAccessTime;
        Info->LastWriteTime.QuadPart = FileName->LastWriteTime;
        Info->ChangeTime.QuadPart = FileName->ChangeTime;
 
        /* Convert file flags */
        NtfsFileFlagsToAttributes(FileName->FileAttributes | StdInfo->FileAttribute, &Info->FileAttributes);
 
        Info->EndOfFile.QuadPart = NtfsGetFileSize(DeviceExt, FileRecord, L"", 0, (PULONGLONG)&Info->AllocationSize.QuadPart);
 
        Info->FileIndex = MFTIndex;
    }
 
    return Status;
}

Referenced by NtfsGetDirEntryInformation().

NtfsGetDirEntryInformation()

static NTSTATUS NtfsGetDirEntryInformation ( PDEVICE_EXTENSION  DeviceExt, FILE_INFORMATION_CLASS  FileInformationClass, PFILE_RECORD_HEADER  FileRecord, ULONGLONG  MFTIndex, PCUNICODE_STRING  OverrideName, PVOID  Buffer, ULONG  BufferLength, PULONG  Written, BOOLEAN  First  )

static

Definition at line 444 of file dirctl.c.

{
    switch (FileInformationClass)
    {
        case FileNamesInformation:
            return NtfsGetNamesInformation(DeviceExt, FileRecord, MFTIndex,
                                           (PFILE_NAMES_INFORMATION)Buffer,
                                           BufferLength, Written, First, OverrideName);
 
        case FileDirectoryInformation:
            return NtfsGetDirectoryInformation(DeviceExt, FileRecord, MFTIndex,
                                               (PFILE_DIRECTORY_INFORMATION)Buffer,
                                               BufferLength, Written, First, OverrideName);
 
        case FileFullDirectoryInformation:
            return NtfsGetFullDirectoryInformation(DeviceExt, FileRecord, MFTIndex,
                                                   (PFILE_FULL_DIRECTORY_INFORMATION)Buffer,
                                                   BufferLength, Written, First, OverrideName);
 
        case FileBothDirectoryInformation:
            return NtfsGetBothDirectoryInformation(DeviceExt, FileRecord, MFTIndex,
                                                   (PFILE_BOTH_DIR_INFORMATION)Buffer,
                                                   BufferLength, Written, First, OverrideName);
 
        default:
            return STATUS_INVALID_INFO_CLASS;
    }
}

Referenced by NtfsGetDotEntry(), and NtfsQueryDirectory().

NtfsGetDotEntry()

static NTSTATUS NtfsGetDotEntry ( PDEVICE_EXTENSION  DeviceExt, ULONGLONG  MFTIndex, PCUNICODE_STRING  Name, FILE_INFORMATION_CLASS  FileInformationClass, PVOID  Buffer, ULONG  BufferLength, PULONG  Written, BOOLEAN  First  )

static

Definition at line 484 of file dirctl.c.

{
    PFILE_RECORD_HEADER FileRecord;
    NTSTATUS Status;
 
    FileRecord = ExAllocateFromNPagedLookasideList(&DeviceExt->FileRecLookasideList);
    if (FileRecord == NULL)
    {
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    Status = ReadFileRecord(DeviceExt, MFTIndex, FileRecord);
    if (NT_SUCCESS(Status))
    {
        Status = NtfsGetDirEntryInformation(DeviceExt, FileInformationClass, FileRecord,
                                            MFTIndex, Name, Buffer, BufferLength,
                                            Written, First);
    }
 
    ExFreeToNPagedLookasideList(&DeviceExt->FileRecLookasideList, FileRecord);
 
    return Status;
}

Referenced by NtfsQueryDirectory().

NtfsGetFileSize()

ULONGLONG NtfsGetFileSize	(	PDEVICE_EXTENSION	DeviceExt,
		PFILE_RECORD_HEADER	FileRecord,
		PCWSTR	Stream,
		ULONG	StreamLength,
		PULONGLONG	AllocatedSize
	)

Definition at line 38 of file dirctl.c.

{
    ULONGLONG Size = 0ULL;
    ULONGLONG Allocated = 0ULL;
    NTSTATUS Status;
    PNTFS_ATTR_CONTEXT DataContext;
 
    Status = FindAttribute(DeviceExt, FileRecord, AttributeData, Stream, StreamLength, &DataContext, NULL);
    if (NT_SUCCESS(Status))
    {
        Size = AttributeDataLength(DataContext->pRecord);
        Allocated = AttributeAllocatedLength(DataContext->pRecord);
        ReleaseAttributeContext(DataContext);
    }
 
    if (AllocatedSize != NULL) *AllocatedSize = Allocated;
 
    return Size;
}

Referenced by NtfsGetBothDirectoryInformation(), NtfsGetDirectoryInformation(), NtfsGetFullDirectoryInformation(), NtfsMakeFCBFromDirEntry(), NtfsSetEndOfFile(), and NtfsUpdateDuplicatedInformation().

NtfsGetFullDirectoryInformation()

static NTSTATUS NtfsGetFullDirectoryInformation ( PDEVICE_EXTENSION  DeviceExt, PFILE_RECORD_HEADER  FileRecord, ULONGLONG  MFTIndex, PFILE_FULL_DIRECTORY_INFORMATION  Info, ULONG  BufferLength, PULONG  Written, BOOLEAN  First, PCUNICODE_STRING  OverrideName  )

static

Definition at line 221 of file dirctl.c.

{
    ULONG Length;
    NTSTATUS Status;
    ULONG BytesToCopy = 0;
    PFILENAME_ATTRIBUTE FileName;
    PCWSTR NameBuffer;
    PSTANDARD_INFORMATION StdInfo;
 
    DPRINT("NtfsGetFullDirectoryInformation() called\n");
 
    *Written = 0;
    Status = STATUS_BUFFER_OVERFLOW;
    if (FIELD_OFFSET(FILE_FULL_DIR_INFORMATION, FileName) > BufferLength)
    {
        return Status;
    }
 
    FileName = GetBestFileNameFromRecord(DeviceExt, FileRecord);
    if (FileName == NULL)
    {
        DPRINT1("No name information for file ID: %#I64x\n", MFTIndex);
        NtfsDumpFileAttributes(DeviceExt, FileRecord);
        return STATUS_OBJECT_NAME_NOT_FOUND;
    }
 
    StdInfo = GetStandardInformationFromRecord(DeviceExt, FileRecord);
    ASSERT(StdInfo != NULL);
 
    if (OverrideName != NULL)
    {
        NameBuffer = OverrideName->Buffer;
        Length = OverrideName->Length;
    }
    else
    {
        NameBuffer = FileName->Name;
        Length = FileName->NameLength * sizeof (WCHAR);
    }
 
    if (First || (BufferLength >= FIELD_OFFSET(FILE_FULL_DIR_INFORMATION, FileName) + Length))
    {
        Info->FileNameLength = Length;
 
        *Written = FIELD_OFFSET(FILE_FULL_DIR_INFORMATION, FileName);
        Info->NextEntryOffset = 0;
        if (BufferLength > FIELD_OFFSET(FILE_FULL_DIR_INFORMATION, FileName))
        {
            BytesToCopy = min(Length, BufferLength - FIELD_OFFSET(FILE_FULL_DIR_INFORMATION, FileName));
            RtlCopyMemory(Info->FileName, NameBuffer, BytesToCopy);
            *Written += BytesToCopy;
 
            if (BytesToCopy == Length)
            {
                Info->NextEntryOffset = ULONG_ROUND_UP(sizeof(FILE_FULL_DIR_INFORMATION) +
                                                       BytesToCopy);
                Status = STATUS_SUCCESS;
            }
        }
 
        Info->CreationTime.QuadPart = FileName->CreationTime;
        Info->LastAccessTime.QuadPart = FileName->LastAccessTime;
        Info->LastWriteTime.QuadPart = FileName->LastWriteTime;
        Info->ChangeTime.QuadPart = FileName->ChangeTime;
 
        /* Convert file flags */
        NtfsFileFlagsToAttributes(FileName->FileAttributes | StdInfo->FileAttribute, &Info->FileAttributes);
 
        Info->EndOfFile.QuadPart = NtfsGetFileSize(DeviceExt, FileRecord, L"", 0, (PULONGLONG)&Info->AllocationSize.QuadPart);
 
        Info->FileIndex = MFTIndex;
        Info->EaSize = 0;
    }
 
    return Status;
}

Referenced by NtfsGetDirEntryInformation().

NtfsGetNamesInformation()

static NTSTATUS NtfsGetNamesInformation ( PDEVICE_EXTENSION  DeviceExt, PFILE_RECORD_HEADER  FileRecord, ULONGLONG  MFTIndex, PFILE_NAMES_INFORMATION  Info, ULONG  BufferLength, PULONG  Written, BOOLEAN  First, PCUNICODE_STRING  OverrideName  )

static

Definition at line 67 of file dirctl.c.

{
    ULONG Length;
    NTSTATUS Status;
    ULONG BytesToCopy = 0;
    PFILENAME_ATTRIBUTE FileName;
    PCWSTR NameBuffer;
 
    DPRINT("NtfsGetNamesInformation() called\n");
 
    *Written = 0;
    Status = STATUS_BUFFER_OVERFLOW;
    if (FIELD_OFFSET(FILE_NAMES_INFORMATION, FileName) > BufferLength)
    {
        return Status;
    }
 
    FileName = GetBestFileNameFromRecord(DeviceExt, FileRecord);
    if (FileName == NULL)
    {
        DPRINT1("No name information for file ID: %#I64x\n", MFTIndex);
        NtfsDumpFileAttributes(DeviceExt, FileRecord);
        return STATUS_OBJECT_NAME_NOT_FOUND;
    }
 
    if (OverrideName != NULL)
    {
        NameBuffer = OverrideName->Buffer;
        Length = OverrideName->Length;
    }
    else
    {
        NameBuffer = FileName->Name;
        Length = FileName->NameLength * sizeof (WCHAR);
    }
 
    if (First || (BufferLength >= FIELD_OFFSET(FILE_NAMES_INFORMATION, FileName) + Length))
    {
        Info->FileNameLength = Length;
 
        *Written = FIELD_OFFSET(FILE_NAMES_INFORMATION, FileName);
        Info->NextEntryOffset = 0;
        if (BufferLength > FIELD_OFFSET(FILE_NAMES_INFORMATION, FileName))
        {
            BytesToCopy = min(Length, BufferLength - FIELD_OFFSET(FILE_NAMES_INFORMATION, FileName));
            RtlCopyMemory(Info->FileName, NameBuffer, BytesToCopy);
            *Written += BytesToCopy;
 
            if (BytesToCopy == Length)
            {
                Info->NextEntryOffset = ULONG_ROUND_UP(sizeof(FILE_NAMES_INFORMATION) +
                                                       BytesToCopy);
                Status = STATUS_SUCCESS;
            }
        }
    }
 
    return Status;
}

Referenced by NtfsGetDirEntryInformation().

NtfsGetParentMftIndex()

static ULONGLONG NtfsGetParentMftIndex ( PDEVICE_EXTENSION  DeviceExt, ULONGLONG  MFTIndex  )

static

Definition at line 516 of file dirctl.c.

{
    PFILE_RECORD_HEADER FileRecord;
    PFILENAME_ATTRIBUTE FileName;
    ULONGLONG Parent = MFTIndex;
 
    FileRecord = ExAllocateFromNPagedLookasideList(&DeviceExt->FileRecLookasideList);
    if (FileRecord == NULL)
    {
        return Parent;
    }
 
    if (NT_SUCCESS(ReadFileRecord(DeviceExt, MFTIndex, FileRecord)))
    {
        FileName = GetBestFileNameFromRecord(DeviceExt, FileRecord);
        if (FileName != NULL)
        {
            Parent = FileName->DirectoryFileReferenceNumber & NTFS_MFT_MASK;
        }
    }
 
    ExFreeToNPagedLookasideList(&DeviceExt->FileRecLookasideList, FileRecord);
 
    return Parent;
}

Referenced by NtfsQueryDirectory().

NtfsQueryDirectory()

NTSTATUS NtfsQueryDirectory

(

PNTFS_IRP_CONTEXT

IrpContext

)

Definition at line 544 of file dirctl.c.

{
    PIRP Irp;
    PDEVICE_OBJECT DeviceObject;
    PDEVICE_EXTENSION DeviceExtension;
    LONG BufferLength = 0;
    PUNICODE_STRING SearchPattern = NULL;
    FILE_INFORMATION_CLASS FileInformationClass;
    ULONG FileIndex = 0;
    PUCHAR Buffer = NULL;
    PFILE_NAMES_INFORMATION Buffer0 = NULL;
    PNTFS_FCB Fcb;
    PNTFS_CCB Ccb;
    BOOLEAN First = FALSE;
    PIO_STACK_LOCATION Stack;
    PFILE_OBJECT FileObject;
    NTSTATUS Status = STATUS_SUCCESS;
    PFILE_RECORD_HEADER FileRecord;
    ULONGLONG MFTRecord, OldMFTRecord = 0;
    UNICODE_STRING Pattern;
    ULONG Written;
 
    DPRINT("NtfsQueryDirectory() called\n");
 
    ASSERT(IrpContext);
    Irp = IrpContext->Irp;
    DeviceObject = IrpContext->DeviceObject;
 
    DeviceExtension = DeviceObject->DeviceExtension;
    Stack = IoGetCurrentIrpStackLocation(Irp);
    FileObject = Stack->FileObject;
 
    Ccb = (PNTFS_CCB)FileObject->FsContext2;
    Fcb = (PNTFS_FCB)FileObject->FsContext;
 
    /* Obtain the callers parameters */
    BufferLength = Stack->Parameters.QueryDirectory.Length;
    SearchPattern = Stack->Parameters.QueryDirectory.FileName;
    FileInformationClass = Stack->Parameters.QueryDirectory.FileInformationClass;
    FileIndex = Stack->Parameters.QueryDirectory.FileIndex;
 
    if (NtfsFCBIsCompressed(Fcb))
    {
        DPRINT1("Compressed directory!\n");
        UNIMPLEMENTED;
        return STATUS_NOT_IMPLEMENTED;
    }
 
    if (!ExAcquireResourceSharedLite(&Fcb->MainResource,
                                     BooleanFlagOn(IrpContext->Flags, IRPCONTEXT_CANWAIT)))
    {
        return STATUS_PENDING;
    }
 
    if (SearchPattern != NULL)
    {
        if (!Ccb->DirectorySearchPattern)
        {
            First = TRUE;
            Pattern.Length = 0;
            Pattern.MaximumLength = SearchPattern->Length + sizeof(WCHAR);
            Ccb->DirectorySearchPattern = Pattern.Buffer =
                ExAllocatePoolWithTag(NonPagedPool, Pattern.MaximumLength, TAG_NTFS);
            if (!Ccb->DirectorySearchPattern)
            {
                ExReleaseResourceLite(&Fcb->MainResource);
                return STATUS_INSUFFICIENT_RESOURCES;
            }
 
            memcpy(Ccb->DirectorySearchPattern, SearchPattern->Buffer, SearchPattern->Length);
            Ccb->DirectorySearchPattern[SearchPattern->Length / sizeof(WCHAR)] = 0;
        }
    }
    else if (!Ccb->DirectorySearchPattern)
    {
        First = TRUE;
        Ccb->DirectorySearchPattern = ExAllocatePoolWithTag(NonPagedPool, 2 * sizeof(WCHAR), TAG_NTFS);
        if (!Ccb->DirectorySearchPattern)
        {
            ExReleaseResourceLite(&Fcb->MainResource);
            return STATUS_INSUFFICIENT_RESOURCES;
        }
 
        Ccb->DirectorySearchPattern[0] = L'*';
        Ccb->DirectorySearchPattern[1] = 0;
    }
 
    RtlInitUnicodeString(&Pattern, Ccb->DirectorySearchPattern);
    DPRINT("Search pattern '%S'\n", Ccb->DirectorySearchPattern);
    DPRINT("In: '%S'\n", Fcb->PathName);
 
    /* Determine directory index */
    if (Stack->Flags & SL_INDEX_SPECIFIED)
    {
        Ccb->Entry = FileIndex;
        SetFlag(Ccb->Flags, CCB_RETURNED_DOT | CCB_RETURNED_DOTDOT);
    }
    else if (First || (Stack->Flags & SL_RESTART_SCAN))
    {
        Ccb->Entry = 0;
        ClearFlag(Ccb->Flags, CCB_RETURNED_DOT | CCB_RETURNED_DOTDOT);
    }
 
    /* Get Buffer for result */
    Buffer = NtfsGetUserBuffer(Irp, FALSE);
 
    DPRINT("Buffer=%p tofind=%S\n", Buffer, Ccb->DirectorySearchPattern);
 
    if (!ExAcquireResourceExclusiveLite(&DeviceExtension->DirResource,
                                        BooleanFlagOn(IrpContext->Flags, IRPCONTEXT_CANWAIT)))
    {
        ExReleaseResourceLite(&Fcb->MainResource);
        return STATUS_PENDING;
    }
 
    Written = 0;
    while (Status == STATUS_SUCCESS && BufferLength > 0)
    {
        if (!BooleanFlagOn(Ccb->Flags, CCB_RETURNED_DOT) ||
            !BooleanFlagOn(Ccb->Flags, CCB_RETURNED_DOTDOT))
        {
            BOOLEAN DotDot = BooleanFlagOn(Ccb->Flags, CCB_RETURNED_DOT);
            UNICODE_STRING DotName;
            ULONGLONG DotIndex;
 
            RtlInitUnicodeString(&DotName, DotDot ? L".." : L".");
 
            if (!NtfsDotNameMatches(&Pattern, &DotName,
                                    BooleanFlagOn(Stack->Flags, SL_CASE_SENSITIVE)))
            {
                SetFlag(Ccb->Flags, DotDot ? CCB_RETURNED_DOTDOT : CCB_RETURNED_DOT);
                continue;
            }
 
            DotIndex = DotDot ? NtfsGetParentMftIndex(DeviceExtension, Fcb->MFTIndex)
                              : Fcb->MFTIndex;
 
            Status = NtfsGetDotEntry(DeviceExtension, DotIndex, &DotName,
                                     FileInformationClass, Buffer, BufferLength,
                                     &Written, Buffer0 == NULL);
 
            /* The entry did not fit. Leave the flag alone so the next call,
             * with a bigger buffer, hands it out instead of skipping it. */
            if (Status == STATUS_BUFFER_OVERFLOW || Status == STATUS_INVALID_INFO_CLASS)
            {
                break;
            }
 
            SetFlag(Ccb->Flags, DotDot ? CCB_RETURNED_DOTDOT : CCB_RETURNED_DOT);
 
            if (!NT_SUCCESS(Status))
            {
                /* Losing "." is not worth failing the whole enumeration over */
                Status = STATUS_SUCCESS;
                continue;
            }
 
            Buffer0 = (PFILE_NAMES_INFORMATION)Buffer;
            Buffer0->FileIndex = FileIndex++;
            BufferLength -= Buffer0->NextEntryOffset;
 
            if (Stack->Flags & SL_RETURN_SINGLE_ENTRY)
            {
                break;
            }
 
            Buffer += Buffer0->NextEntryOffset;
            continue;
        }
 
        Status = NtfsFindFileAt(DeviceExtension,
                                &Pattern,
                                &Ccb->Entry,
                                &FileRecord,
                                &MFTRecord,
                                Fcb->MFTIndex,
                                BooleanFlagOn(Stack->Flags, SL_CASE_SENSITIVE));
 
        if (NT_SUCCESS(Status))
        {
            /* HACK: files with both a short name and a long name are present twice in the index.
             * Ignore the second entry, if it is immediately following the first one.
             */
            if (MFTRecord == OldMFTRecord)
            {
                DPRINT1("Ignoring duplicate MFT entry 0x%x\n", MFTRecord);
                Ccb->Entry++;
                ExFreeToNPagedLookasideList(&DeviceExtension->FileRecLookasideList, FileRecord);
                continue;
            }
            OldMFTRecord = MFTRecord;
 
            Status = NtfsGetDirEntryInformation(DeviceExtension,
                                                FileInformationClass,
                                                FileRecord,
                                                MFTRecord,
                                                NULL,
                                                Buffer,
                                                BufferLength,
                                                &Written,
                                                Buffer0 == NULL);
 
            if (Status == STATUS_BUFFER_OVERFLOW || Status == STATUS_INVALID_INFO_CLASS)
            {
                break;
            }
        }
        else
        {
            Status = (First ? STATUS_NO_SUCH_FILE : STATUS_NO_MORE_FILES);
            break;
        }
 
        Buffer0 = (PFILE_NAMES_INFORMATION)Buffer;
        Buffer0->FileIndex = FileIndex++;
        Ccb->Entry++;
        BufferLength -= Buffer0->NextEntryOffset;
 
        ExFreeToNPagedLookasideList(&DeviceExtension->FileRecLookasideList, FileRecord);
 
        if (Stack->Flags & SL_RETURN_SINGLE_ENTRY)
        {
            break;
        }
 
        Buffer += Buffer0->NextEntryOffset;
    }
 
    if (Buffer0)
    {
        Buffer0->NextEntryOffset = 0;
        Status = STATUS_SUCCESS;
        IrpContext->Irp->IoStatus.Information = Stack->Parameters.QueryDirectory.Length - BufferLength;
    }
    else
    {
        ASSERT(Status != STATUS_SUCCESS || BufferLength == 0);
        ASSERT(Written <= Stack->Parameters.QueryDirectory.Length);
        IrpContext->Irp->IoStatus.Information = Written;
    }
 
    ExReleaseResourceLite(&DeviceExtension->DirResource);
    ExReleaseResourceLite(&Fcb->MainResource);
 
    return Status;
}

Referenced by NtfsDirectoryControl().