- Silence TCPIP.

[reactos.git] / reactos / ntoskrnl / cc / pin.c

diff --git a/reactos/ntoskrnl/cc/pin.c b/reactos/ntoskrnl/cc/pin.c
index 5cbcff0..b845c6c 100644 (file)
--- a/reactos/ntoskrnl/cc/pin.c
+++ b/reactos/ntoskrnl/cc/pin.c
@@ -1,12 +1,11 @@
-/* $Id: pin.c,v 1.16 2004/08/15 16:39:00 chorns Exp $
+/* $Id$
  *
  * COPYRIGHT:       See COPYING in the top level directory
  * PROJECT:         ReactOS kernel
  * FILE:            ntoskrnl/cc/pin.c
  * PURPOSE:         Implements cache managers pinning interface
- * PROGRAMMER:      Hartmut Birr
- * UPDATE HISTORY:
- *                  Created 05.10.2001
+ *
+ * PROGRAMMERS:     Hartmut Birr
  */
 
 /* INCLUDES ******************************************************************/
@@ -17,8 +16,6 @@
 
 /* GLOBALS *******************************************************************/
 
-#define ROUND_DOWN(N, S) ((N) - ((N) % (S)))
-
 extern NPAGED_LOOKASIDE_LIST iBcbLookasideList;
 
 /* FUNCTIONS *****************************************************************/
@@ -41,19 +38,19 @@ CcMapData (IN PFILE_OBJECT FileObject,
   NTSTATUS Status;
   PINTERNAL_BCB iBcb;
   ULONG ROffset;
-  
-  DPRINT("CcMapData(FileObject %x, FileOffset %d, Length %d, Wait %d,"
-        " pBcb %x, pBuffer %x)\n", FileObject, (ULONG)FileOffset->QuadPart,
+
+  DPRINT("CcMapData(FileObject 0x%p, FileOffset %I64x, Length %d, Wait %d,"
+        " pBcb 0x%p, pBuffer 0x%p)\n", FileObject, FileOffset->QuadPart,
         Length, Wait, pBcb, pBuffer);
-  
+
   ReadOffset = (ULONG)FileOffset->QuadPart;
   Bcb = FileObject->SectionObjectPointer->SharedCacheMap;
-  assert(Bcb);
+  ASSERT(Bcb);
+
+  DPRINT("AllocationSize %I64x, FileSize %I64x\n",
+         Bcb->AllocationSize.QuadPart,
+         Bcb->FileSize.QuadPart);
 
-  DPRINT("AllocationSize %d, FileSize %d\n",
-         (ULONG)Bcb->AllocationSize.QuadPart,
-         (ULONG)Bcb->FileSize.QuadPart);
-  
   if (ReadOffset % Bcb->CacheSegmentSize + Length > Bcb->CacheSegmentSize)
     {
       return(FALSE);
@@ -81,15 +78,8 @@ CcMapData (IN PFILE_OBJECT FileObject,
          return(FALSE);
        }
     }
-#if defined(__GNUC__)
-  *pBuffer += ReadOffset % Bcb->CacheSegmentSize;
-#else
-  {
-    char* pTemp = *pBuffer;
-    pTemp += ReadOffset % Bcb->CacheSegmentSize;
-    *pBuffer = pTemp;
-  }
-#endif
+
+  *pBuffer = (PVOID)((ULONG_PTR)(*pBuffer) + (ReadOffset % Bcb->CacheSegmentSize));
   iBcb = ExAllocateFromNPagedLookasideList(&iBcbLookasideList);
   if (iBcb == NULL)
     {
@@ -97,10 +87,13 @@ CcMapData (IN PFILE_OBJECT FileObject,
       return FALSE;
     }
   memset(iBcb, 0, sizeof(INTERNAL_BCB));
+  iBcb->PFCB.NodeTypeCode = 0xDE45; /* Undocumented (CAPTIVE_PUBLIC_BCB_NODETYPECODE) */
+  iBcb->PFCB.NodeByteSize = sizeof(PUBLIC_BCB);
+  iBcb->PFCB.MappedLength = Length;
+  iBcb->PFCB.MappedFileOffset = *FileOffset;
   iBcb->CacheSegment = CacheSeg;
   iBcb->Dirty = FALSE;
-  iBcb->PFCB.MappedLength = Length;
-  iBcb->PFCB.MappedFileOffset.QuadPart = FileOffset->QuadPart;
+  iBcb->RefCount = 1;
   *pBcb = (PVOID)iBcb;
   return(TRUE);
 }
@@ -114,12 +107,12 @@ CcPinMappedData (
        IN      PFILE_OBJECT            FileObject,
        IN      PLARGE_INTEGER          FileOffset,
        IN      ULONG                   Length,
-       IN      BOOLEAN                 Wait,
+       IN      ULONG                   Flags,
        OUT     PVOID                   * Bcb
        )
 {
-       UNIMPLEMENTED;
-       return FALSE;
+  /* no-op for current implementation. */
+  return TRUE;
 }
 
 /*
@@ -131,13 +124,19 @@ CcPinRead (
        IN      PFILE_OBJECT            FileObject,
        IN      PLARGE_INTEGER          FileOffset,
        IN      ULONG                   Length,
-       IN      BOOLEAN                 Wait,
+       IN      ULONG                   Flags,
        OUT     PVOID                   * Bcb,
        OUT     PVOID                   * Buffer
        )
 {
-       UNIMPLEMENTED;
-       return FALSE;
+  if (CcMapData(FileObject, FileOffset, Length, Flags, Bcb, Buffer))
+  {
+    if (CcPinMappedData(FileObject, FileOffset, Length, Flags, Bcb))
+      return TRUE;
+    else
+      CcUnpinData(Bcb);
+  }
+  return FALSE;
 }
 
 /*
@@ -150,13 +149,20 @@ CcPreparePinWrite (
        IN      PLARGE_INTEGER          FileOffset,
        IN      ULONG                   Length,
        IN      BOOLEAN                 Zero,
-       IN      BOOLEAN                 Wait,
+       IN      ULONG                   Flags,
        OUT     PVOID                   * Bcb,
        OUT     PVOID                   * Buffer
        )
 {
-       UNIMPLEMENTED;
-       return FALSE;
+        /*
+         * FIXME: This is function is similar to CcPinRead, but doesn't
+         * read the data if they're not present. Instead it should just
+         * prepare the cache segments and zero them out if Zero == TRUE.
+         *
+         * For now calling CcPinRead is better than returning error or
+         * just having UNIMPLEMENTED here.
+         */
+        return CcPinRead(FileObject, FileOffset, Length, Flags, Bcb, Buffer);
 }
 
 /*
@@ -178,9 +184,12 @@ VOID STDCALL
 CcUnpinData (IN PVOID Bcb)
 {
   PINTERNAL_BCB iBcb = Bcb;
-  CcRosReleaseCacheSegment(iBcb->CacheSegment->Bcb, iBcb->CacheSegment, TRUE, 
-                          iBcb->Dirty, FALSE);
-  ExFreeToNPagedLookasideList(&iBcbLookasideList, iBcb);
+  CcRosReleaseCacheSegment(iBcb->CacheSegment->Bcb, iBcb->CacheSegment, TRUE,
+                           iBcb->Dirty, FALSE);
+  if (--iBcb->RefCount == 0)
+  {
+    ExFreeToNPagedLookasideList(&iBcbLookasideList, iBcb);
+  }
 }
 
 /*
@@ -195,3 +204,54 @@ CcUnpinDataForThread (
 {
        UNIMPLEMENTED;
 }
+
+/*
+ * @implemented
+ */
+VOID
+STDCALL
+CcRepinBcb (
+       IN      PVOID   Bcb
+       )
+{
+  PINTERNAL_BCB iBcb = Bcb;
+  iBcb->RefCount++;
+}
+
+/*
+ * @unimplemented
+ */
+VOID
+STDCALL
+CcUnpinRepinnedBcb (
+       IN      PVOID                   Bcb,
+       IN      BOOLEAN                 WriteThrough,
+       IN      PIO_STATUS_BLOCK        IoStatus
+       )
+{
+  PINTERNAL_BCB iBcb = Bcb;
+
+  if (--iBcb->RefCount == 0)
+    {
+      IoStatus->Information = 0;
+      if (WriteThrough)
+        {
+          ExEnterCriticalRegionAndAcquireFastMutexUnsafe(&iBcb->CacheSegment->Lock);
+          if (iBcb->CacheSegment->Dirty)
+            {
+              IoStatus->Status = CcRosFlushCacheSegment(iBcb->CacheSegment);
+            }
+          else
+            {
+              IoStatus->Status = STATUS_SUCCESS;
+            }
+          ExReleaseFastMutexUnsafeAndLeaveCriticalRegion(&iBcb->CacheSegment->Lock);
+        }
+      else
+        {
+          IoStatus->Status = STATUS_SUCCESS;
+        }
+
+      ExFreeToNPagedLookasideList(&iBcbLookasideList, iBcb);
+    }
+}