[FASTFAT]
[reactos.git] / reactos / drivers / filesystems / fastfat / rw.c
index 2ca5e70..c3834b5 100644 (file)
@@ -150,12 +150,12 @@ VfatReadFileData(
     BytesPerSector = DeviceExt->FatInfo.BytesPerSector;
     BytesPerCluster = DeviceExt->FatInfo.BytesPerCluster;
 
-    ASSERT(ReadOffset.QuadPart + Length <= ROUND_UP(Fcb->RFCB.FileSize.QuadPart, BytesPerSector));
+    ASSERT(ReadOffset.QuadPart + Length <= ROUND_UP_64(Fcb->RFCB.FileSize.QuadPart, BytesPerSector));
     ASSERT(ReadOffset.u.LowPart % BytesPerSector == 0);
     ASSERT(Length % BytesPerSector == 0);
 
     /* Is this a read of the FAT? */
-    if (Fcb->Flags & FCB_IS_FAT)
+    if (BooleanFlagOn(Fcb->Flags, FCB_IS_FAT))
     {
         ReadOffset.QuadPart += DeviceExt->FatInfo.FATStart * BytesPerSector;
         Status = VfatReadDiskPartial(IrpContext, &ReadOffset, Length, 0, TRUE);
@@ -172,7 +172,7 @@ VfatReadFileData(
     }
 
     /* Is this a read of the Volume ? */
-    if (Fcb->Flags & FCB_IS_VOLUME)
+    if (BooleanFlagOn(Fcb->Flags, FCB_IS_VOLUME))
     {
         Status = VfatReadDiskPartial(IrpContext, &ReadOffset, Length, 0, TRUE);
         if (NT_SUCCESS(Status))
@@ -367,7 +367,7 @@ VfatWriteFileData(
     ASSERT(Length % BytesPerSector == 0);
 
     /* Is this a write of the volume? */
-    if (Fcb->Flags & FCB_IS_VOLUME)
+    if (BooleanFlagOn(Fcb->Flags, FCB_IS_VOLUME))
     {
         Status = VfatWriteDiskPartial(IrpContext, &WriteOffset, Length, 0, TRUE);
         if (!NT_SUCCESS(Status))
@@ -378,7 +378,7 @@ VfatWriteFileData(
     }
 
     /* Is this a write to the FAT? */
-    if (Fcb->Flags & FCB_IS_FAT)
+    if (BooleanFlagOn(Fcb->Flags, FCB_IS_FAT))
     {
         WriteOffset.u.LowPart += DeviceExt->FatInfo.FATStart * BytesPerSector;
         IrpContext->RefCount = 1;
@@ -547,6 +547,7 @@ VfatRead(
     LARGE_INTEGER ByteOffset;
     PVOID Buffer;
     ULONG BytesPerSector;
+    BOOLEAN PagingIo, CanWait, IsVolume, NoCache;
 
     ASSERT(IrpContext);
 
@@ -554,6 +555,10 @@ VfatRead(
 
     ASSERT(IrpContext->DeviceObject);
 
+    PagingIo = BooleanFlagOn(IrpContext->Irp->Flags, IRP_PAGING_IO);
+    CanWait = BooleanFlagOn(IrpContext->Flags, IRPCONTEXT_CANWAIT);
+    NoCache = BooleanFlagOn(IrpContext->Irp->Flags, IRP_NOCACHE);
+
     // This request is not allowed on the main device object
     if (IrpContext->DeviceObject == VfatGlobalData->DeviceObject)
     {
@@ -567,7 +572,9 @@ VfatRead(
     Fcb = IrpContext->FileObject->FsContext;
     ASSERT(Fcb);
 
-    if (Fcb->Flags & FCB_IS_PAGE_FILE)
+    IsVolume = BooleanFlagOn(Fcb->Flags, FCB_IS_VOLUME);
+
+    if (BooleanFlagOn(Fcb->Flags, FCB_IS_PAGE_FILE))
     {
         PFATINFO FatInfo = &IrpContext->DeviceExt->FatInfo;
         IrpContext->Stack->Parameters.Read.ByteOffset.QuadPart += FatInfo->dataStart * FatInfo->BytesPerSector;
@@ -585,7 +592,7 @@ VfatRead(
     BytesPerSector = IrpContext->DeviceExt->FatInfo.BytesPerSector;
 
     /* fail if file is a directory and no paged read */
-    if (*Fcb->Attributes & FILE_ATTRIBUTE_DIRECTORY && !(IrpContext->Irp->Flags & IRP_PAGING_IO))
+    if (vfatFCBIsDirectory(Fcb) && !PagingIo)
     {
         Status = STATUS_INVALID_PARAMETER;
         goto ByeBye;
@@ -593,7 +600,7 @@ VfatRead(
 
     DPRINT("'%wZ', Offset: %u, Length %u\n", &Fcb->PathNameU, ByteOffset.u.LowPart, Length);
 
-    if (ByteOffset.u.HighPart && !(Fcb->Flags & FCB_IS_VOLUME))
+    if (ByteOffset.u.HighPart && !IsVolume)
     {
        Status = STATUS_INVALID_PARAMETER;
        goto ByeBye;
@@ -613,7 +620,7 @@ VfatRead(
        goto ByeBye;
     }
 
-    if (IrpContext->Irp->Flags & (IRP_PAGING_IO | IRP_NOCACHE) || (Fcb->Flags & FCB_IS_VOLUME))
+    if (NoCache || PagingIo || IsVolume)
     {
         if (ByteOffset.u.LowPart % BytesPerSector != 0 || Length % BytesPerSector != 0)
         {
@@ -624,11 +631,11 @@ VfatRead(
         }
     }
 
-    if (Fcb->Flags & FCB_IS_VOLUME)
+    if (IsVolume)
     {
         Resource = &IrpContext->DeviceExt->DirResource;
     }
-    else if (IrpContext->Irp->Flags & IRP_PAGING_IO)
+    else if (PagingIo)
     {
         Resource = &Fcb->PagingIoResource;
     }
@@ -637,15 +644,14 @@ VfatRead(
         Resource = &Fcb->MainResource;
     }
 
-    if (!ExAcquireResourceSharedLite(Resource,
-                                     IrpContext->Flags & IRPCONTEXT_CANWAIT ? TRUE : FALSE))
+    if (!ExAcquireResourceSharedLite(Resource, CanWait))
     {
         Resource = NULL;
         Status = STATUS_PENDING;
         goto ByeBye;
     }
 
-    if (!(IrpContext->Irp->Flags & IRP_PAGING_IO) &&
+    if (!PagingIo &&
         FsRtlAreThereCurrentFileLocks(&Fcb->FileLock))
     {
         if (!FsRtlCheckLockForReadAccess(&Fcb->FileLock, IrpContext->Irp))
@@ -655,15 +661,9 @@ VfatRead(
         }
     }
 
-    Buffer = VfatGetUserBuffer(IrpContext->Irp, BooleanFlagOn(IrpContext->Irp->Flags, IRP_PAGING_IO));
-    if (!Buffer && IrpContext->Irp->MdlAddress)
-    {
-        Status = STATUS_INVALID_USER_BUFFER;
-        goto ByeBye;
-    }
+    Buffer = VfatGetUserBuffer(IrpContext->Irp, PagingIo);
 
-    if (!(IrpContext->Irp->Flags & (IRP_NOCACHE|IRP_PAGING_IO)) &&
-        !(Fcb->Flags & (FCB_IS_PAGE_FILE|FCB_IS_VOLUME)))
+    if (!PagingIo && !NoCache && !IsVolume)
     {
         // cached read
         Status = STATUS_SUCCESS;
@@ -687,11 +687,11 @@ VfatRead(
             if (!CcCopyRead(IrpContext->FileObject,
                             &ByteOffset,
                             Length,
-                            (IrpContext->Flags & IRPCONTEXT_CANWAIT) != 0,
+                            CanWait,
                             Buffer,
                             &IrpContext->Irp->IoStatus))
             {
-                ASSERT((IrpContext->Flags & IRPCONTEXT_CANWAIT) == 0);
+                ASSERT(!CanWait);
                 Status = STATUS_PENDING;
                 goto ByeBye;
             }
@@ -711,17 +711,17 @@ VfatRead(
     else
     {
         // non cached read
-        if (ByteOffset.QuadPart + Length > ROUND_UP(Fcb->RFCB.FileSize.QuadPart, BytesPerSector))
-        {
-            Length = (ULONG)(ROUND_UP(Fcb->RFCB.FileSize.QuadPart, BytesPerSector) - ByteOffset.QuadPart);
-        }
-
         Status = VfatLockUserBuffer(IrpContext->Irp, Length, IoWriteAccess);
         if (!NT_SUCCESS(Status))
         {
             goto ByeBye;
         }
 
+        if (ByteOffset.QuadPart + Length > ROUND_UP_64(Fcb->RFCB.FileSize.QuadPart, BytesPerSector))
+        {
+            Length = (ULONG)(ROUND_UP_64(Fcb->RFCB.FileSize.QuadPart, BytesPerSector) - ByteOffset.QuadPart);
+        }
+
         Status = VfatReadFileData(IrpContext, Length, ByteOffset, &ReturnedLength);
         if (NT_SUCCESS(Status))
         {
@@ -746,8 +746,8 @@ ByeBye:
     else
     {
         IrpContext->Irp->IoStatus.Status = Status;
-        if (IrpContext->FileObject->Flags & FO_SYNCHRONOUS_IO &&
-            !(IrpContext->Irp->Flags & IRP_PAGING_IO) &&
+        if (BooleanFlagOn(IrpContext->FileObject->Flags, FO_SYNCHRONOUS_IO) &&
+            !PagingIo &&
             (NT_SUCCESS(Status) || Status == STATUS_END_OF_FILE))
         {
             IrpContext->FileObject->CurrentByteOffset.QuadPart =
@@ -773,6 +773,7 @@ VfatWrite(
     ULONG Length = 0;
     PVOID Buffer;
     ULONG BytesPerSector;
+    BOOLEAN PagingIo, CanWait, IsVolume, IsFAT, NoCache;
 
     ASSERT(IrpContext);
 
@@ -780,6 +781,10 @@ VfatWrite(
 
     ASSERT(IrpContext->DeviceObject);
 
+    PagingIo = BooleanFlagOn(IrpContext->Irp->Flags, IRP_PAGING_IO);
+    CanWait = BooleanFlagOn(IrpContext->Flags, IRPCONTEXT_CANWAIT);
+    NoCache = BooleanFlagOn(IrpContext->Irp->Flags, IRP_NOCACHE);
+
     // This request is not allowed on the main device object
     if (IrpContext->DeviceObject == VfatGlobalData->DeviceObject)
     {
@@ -793,7 +798,10 @@ VfatWrite(
     Fcb = IrpContext->FileObject->FsContext;
     ASSERT(Fcb);
 
-    if (Fcb->Flags & FCB_IS_PAGE_FILE)
+    IsVolume = BooleanFlagOn(Fcb->Flags, FCB_IS_VOLUME);
+    IsFAT = BooleanFlagOn(Fcb->Flags, FCB_IS_FAT);
+
+    if (BooleanFlagOn(Fcb->Flags, FCB_IS_PAGE_FILE))
     {
         PFATINFO FatInfo = &IrpContext->DeviceExt->FatInfo;
         IrpContext->Stack->Parameters.Write.ByteOffset.QuadPart += FatInfo->dataStart * FatInfo->BytesPerSector;
@@ -807,7 +815,7 @@ VfatWrite(
     DPRINT("<%wZ>\n", &Fcb->PathNameU);
 
     /* fail if file is a directory and no paged read */
-    if (*Fcb->Attributes & FILE_ATTRIBUTE_DIRECTORY && !(IrpContext->Irp->Flags & IRP_PAGING_IO))
+    if (vfatFCBIsDirectory(Fcb) && !PagingIo)
     {
         Status = STATUS_INVALID_PARAMETER;
         goto ByeBye;
@@ -822,13 +830,13 @@ VfatWrite(
     Length = IrpContext->Stack->Parameters.Write.Length;
     BytesPerSector = IrpContext->DeviceExt->FatInfo.BytesPerSector;
 
-    if (ByteOffset.u.HighPart && !(Fcb->Flags & FCB_IS_VOLUME))
+    if (ByteOffset.u.HighPart && !IsVolume)
     {
         Status = STATUS_INVALID_PARAMETER;
         goto ByeBye;
     }
 
-    if (Fcb->Flags & (FCB_IS_FAT | FCB_IS_VOLUME) ||
+    if (IsFAT || IsVolume ||
         vfatDirEntryGetFirstCluster(IrpContext->DeviceExt, &Fcb->entry) == 1)
     {
         if (ByteOffset.QuadPart + Length > Fcb->RFCB.FileSize.QuadPart)
@@ -839,7 +847,7 @@ VfatWrite(
         }
     }
 
-    if (IrpContext->Irp->Flags & (IRP_PAGING_IO|IRP_NOCACHE) || (Fcb->Flags & FCB_IS_VOLUME))
+    if (PagingIo || NoCache || IsVolume)
     {
         if (ByteOffset.u.LowPart % BytesPerSector != 0 || Length % BytesPerSector != 0)
         {
@@ -849,15 +857,17 @@ VfatWrite(
         }
     }
 
+    OldFileSize = Fcb->RFCB.FileSize;
+
     if (Length == 0)
     {
-        /* FIXME: Update last write time */
+        /* Update last write time */
         IrpContext->Irp->IoStatus.Information = 0;
         Status = STATUS_SUCCESS;
-        goto ByeBye;
+        goto Metadata;
     }
 
-    if (IrpContext->Irp->Flags & IRP_PAGING_IO)
+    if (PagingIo)
     {
         if (ByteOffset.u.LowPart + Length > Fcb->RFCB.AllocationSize.u.LowPart)
         {
@@ -871,11 +881,11 @@ VfatWrite(
         }
     }
 
-    if (Fcb->Flags & FCB_IS_VOLUME)
+    if (IsVolume)
     {
         Resource = &IrpContext->DeviceExt->DirResource;
     }
-    else if (IrpContext->Irp->Flags & IRP_PAGING_IO)
+    else if (PagingIo)
     {
         Resource = &Fcb->PagingIoResource;
     }
@@ -884,10 +894,9 @@ VfatWrite(
         Resource = &Fcb->MainResource;
     }
 
-    if (Fcb->Flags & FCB_IS_PAGE_FILE)
+    if (PagingIo)
     {
-        if (!ExAcquireResourceSharedLite(Resource,
-                                         BooleanFlagOn(IrpContext->Flags, IRPCONTEXT_CANWAIT)))
+        if (!ExAcquireResourceSharedLite(Resource, CanWait))
         {
             Resource = NULL;
             Status = STATUS_PENDING;
@@ -896,8 +905,7 @@ VfatWrite(
     }
     else
     {
-        if (!ExAcquireResourceExclusiveLite(Resource,
-                                            BooleanFlagOn(IrpContext->Flags, IRPCONTEXT_CANWAIT)))
+        if (!ExAcquireResourceExclusiveLite(Resource, CanWait))
         {
             Resource = NULL;
             Status = STATUS_PENDING;
@@ -905,7 +913,7 @@ VfatWrite(
         }
     }
 
-    if (!(IrpContext->Irp->Flags & IRP_PAGING_IO) &&
+    if (!PagingIo &&
         FsRtlAreThereCurrentFileLocks(&Fcb->FileLock))
     {
         if (!FsRtlCheckLockForWriteAccess(&Fcb->FileLock, IrpContext->Irp))
@@ -915,7 +923,7 @@ VfatWrite(
         }
     }
 
-    if (!(IrpContext->Flags & IRPCONTEXT_CANWAIT) && !(Fcb->Flags & FCB_IS_VOLUME))
+    if (!CanWait && !IsVolume)
     {
         if (ByteOffset.u.LowPart + Length > Fcb->RFCB.AllocationSize.u.LowPart)
         {
@@ -924,18 +932,9 @@ VfatWrite(
         }
     }
 
-    OldFileSize = Fcb->RFCB.FileSize;
+    Buffer = VfatGetUserBuffer(IrpContext->Irp, PagingIo);
 
-    Buffer = VfatGetUserBuffer(IrpContext->Irp, BooleanFlagOn(IrpContext->Irp->Flags, IRP_PAGING_IO));
-    if (!Buffer && IrpContext->Irp->MdlAddress)
-    {
-        Status = STATUS_INVALID_USER_BUFFER;
-        goto ByeBye;
-    }
-
-
-    if (!(Fcb->Flags & (FCB_IS_FAT|FCB_IS_VOLUME)) &&
-        !(IrpContext->Irp->Flags & IRP_PAGING_IO) &&
+    if (!IsFAT && !IsVolume && !PagingIo &&
         ByteOffset.u.LowPart + Length > Fcb->RFCB.FileSize.u.LowPart)
     {
         LARGE_INTEGER AllocationSize;
@@ -948,8 +947,7 @@ VfatWrite(
         }
     }
 
-    if (!(IrpContext->Irp->Flags & (IRP_NOCACHE|IRP_PAGING_IO)) &&
-        !(Fcb->Flags & (FCB_IS_PAGE_FILE|FCB_IS_VOLUME)))
+    if (!NoCache && !PagingIo && !IsVolume)
     {
         // cached write
 
@@ -972,7 +970,7 @@ VfatWrite(
             if (CcCopyWrite(IrpContext->FileObject,
                             &ByteOffset,
                             Length,
-                            TRUE /*(IrpContext->Flags & IRPCONTEXT_CANWAIT) != 0*/,
+                            TRUE /*CanWait*/,
                             Buffer))
             {
                 IrpContext->Irp->IoStatus.Information = Length;
@@ -980,7 +978,7 @@ VfatWrite(
             }
             else
             {
-                ASSERT(FALSE /*(IrpContext->Flags & IRPCONTEXT_CANWAIT) == 0*/);
+                ASSERT(FALSE /*!CanWait*/);
                 Status = STATUS_UNSUCCESSFUL;
             }
         }
@@ -993,18 +991,18 @@ VfatWrite(
     else
     {
         // non cached write
-
-        if (ByteOffset.QuadPart > OldFileSize.QuadPart)
-        {
-            CcZeroData(IrpContext->FileObject, &OldFileSize, &ByteOffset, TRUE);
-        }
-
         Status = VfatLockUserBuffer(IrpContext->Irp, Length, IoReadAccess);
         if (!NT_SUCCESS(Status))
         {
+            Status = STATUS_INVALID_USER_BUFFER;
             goto ByeBye;
         }
 
+        if (ByteOffset.QuadPart > OldFileSize.QuadPart)
+        {
+            CcZeroData(IrpContext->FileObject, &OldFileSize, &ByteOffset, TRUE);
+        }
+
         Status = VfatWriteFileData(IrpContext, Length, ByteOffset);
         if (NT_SUCCESS(Status))
         {
@@ -1012,17 +1010,17 @@ VfatWrite(
         }
     }
 
-    if (!(IrpContext->Irp->Flags & IRP_PAGING_IO) &&
-        !(Fcb->Flags & (FCB_IS_FAT|FCB_IS_VOLUME)))
+Metadata:
+    if (!PagingIo && !IsFAT && !IsVolume)
     {
-        if(!(*Fcb->Attributes & FILE_ATTRIBUTE_DIRECTORY))
+        if(!vfatFCBIsDirectory(Fcb))
         {
             LARGE_INTEGER SystemTime;
             ULONG Filter;
 
             // set dates and times
             KeQuerySystemTime (&SystemTime);
-            if (Fcb->Flags & FCB_IS_FATX_ENTRY)
+            if (vfatVolumeIsFatX(IrpContext->DeviceExt))
             {
                 FsdSystemTimeToDosDateTime(IrpContext->DeviceExt,
                                            &SystemTime, &Fcb->entry.FatX.UpdateDate,
@@ -1073,8 +1071,8 @@ ByeBye:
     else
     {
         IrpContext->Irp->IoStatus.Status = Status;
-        if (IrpContext->FileObject->Flags & FO_SYNCHRONOUS_IO &&
-            !(IrpContext->Irp->Flags & IRP_PAGING_IO) && NT_SUCCESS(Status))
+        if (BooleanFlagOn(IrpContext->FileObject->Flags, FO_SYNCHRONOUS_IO) &&
+            !PagingIo && NT_SUCCESS(Status))
         {
             IrpContext->FileObject->CurrentByteOffset.QuadPart =
                 ByteOffset.QuadPart + IrpContext->Irp->IoStatus.Information;