[mjmartin] 38899: - Fix calculation of WriteQuotaAvailable and WritePtr when moving memory. - Due to the new implementation, Quota is not charged for partial message retrievals. Therefore reset Quota Available to MaxDataLength when last message is read from buffer and check that Quota is greater than 0 before waking the write thread.

Author: mjmartin
Date: Sun Jan 18 10:31:45 2009
New Revision: 38899
URL: http://svn.reactos.org/svn/reactos?rev=38899&view=rev
Log:
- Fix calculation of WriteQuotaAvailable and WritePtr when moving memory.
- Due to the new implementation, Quota is not charged for partial message retrievals.
Therefore reset Quota Available to MaxDataLength when last message is read from buffer and
check that Quota is greater than 0 before waking the write thread.
Modified:
    trunk/reactos/drivers/filesystems/npfs/rw.c
Modified: trunk/reactos/drivers/filesystems/npfs/rw.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/filesystems/npfs/r…
==============================================================================
--- trunk/reactos/drivers/filesystems/npfs/rw.c [iso-8859-1] (original)
+++ trunk/reactos/drivers/filesystems/npfs/rw.c [iso-8859-1] Sun Jan 18 10:31:45 2009
@@ -537,10 +537,10 @@
                                /* Check if buffer is full and the read pointer is not at
the start of the buffer */
                                if ((Ccb->WriteQuotaAvailable == 0) &&
(Ccb->ReadPtr > Ccb->Data))
                                {
+                                       Ccb->WriteQuotaAvailable +=
(ULONG_PTR)Ccb->ReadPtr - (ULONG_PTR)Ccb->Data;
                                        memcpy(Ccb->Data, Ccb->ReadPtr,
(ULONG_PTR)Ccb->WritePtr - (ULONG_PTR)Ccb->ReadPtr);
-                                       Ccb->WritePtr =
(PVOID)((ULONG_PTR)Ccb->WritePtr - ((ULONG_PTR)Ccb->WritePtr -
(ULONG_PTR)Ccb->ReadPtr));
+                                       Ccb->WritePtr =
(PVOID)((ULONG_PTR)Ccb->WritePtr - ((ULONG_PTR)Ccb->ReadPtr -
(ULONG_PTR)Ccb->Data));
                                        Ccb->ReadPtr = Ccb->Data;
-                                       Ccb->WriteQuotaAvailable +=
(ULONG_PTR)Ccb->WritePtr - (ULONG_PTR)Ccb->ReadPtr;
                                }
                                /* For Message mode, the Message length is stored in the
buffer preceeding the Message. */
@@ -560,7 +560,8 @@
                                        /* Use the smaller value */
                                        CopyLength = min(NextMessageLength, Length);
-
+                                       ASSERT(CopyLength > 0);
+                                       ASSERT(CopyLength <=
Ccb->ReadDataAvailable);
                                        /* retrieve the message from the buffer */
                                        memcpy(Buffer, (PVOID)((ULONG_PTR)Ccb->ReadPtr
+ sizeof(NextMessageLength)), CopyLength);
@@ -571,6 +572,7 @@
                                                {
                                                        /* Calculate the remaining message
new size */
                                                        ULONG NewMessageSize =
NextMessageLength-CopyLength;
+
                                                        /* Update ReadPtr to point to new
Message size location */
                                                        Ccb->ReadPtr =
(PVOID)((ULONG_PTR)Ccb->ReadPtr + CopyLength);
@@ -586,16 +588,16 @@
                                        }
                                        else
                                        {
-                                               /* This was the last Message, so just zero
this messages for safety sake */
+                                               /* This was the last Message, so just zero
start of buffer for safety sake */
                                                memset(Ccb->Data, 0, NextMessageLength
+ sizeof(NextMessageLength));
+
+                                               /* Reset to MaxDataLength as partial
message retrievals dont
+                                                  give the length back to Quota */
+                                               Ccb->WriteQuotaAvailable =
Ccb->MaxDataLength;
                                                /* reset read and write pointer to
beginning of buffer */
                                                Ccb->WritePtr = Ccb->Data;
                                                Ccb->ReadPtr = Ccb->Data;
-
-                                               /* Add both the message length and the
header to the WriteQuotaAvailable
-                                                  as they both were removed */
-                                               Ccb->WriteQuotaAvailable += (CopyLength
+ sizeof(CopyLength));
                                        }
 #ifndef NDEBUG
                                        DPRINT("Length %d Buffer
%x\n",CopyLength,Buffer);
@@ -619,7 +621,8 @@
                                        else
                                        {
                                                KeResetEvent(&Ccb->ReadEvent);
-                                               if (Ccb->PipeState ==
FILE_PIPE_CONNECTED_STATE)
+
+                                               if ((Ccb->PipeState ==
FILE_PIPE_CONNECTED_STATE) && (Ccb->WriteQuotaAvailable > 0))
                                                {
KeSetEvent(&Ccb->OtherSide->WriteEvent, IO_NO_INCREMENT, FALSE);
                                                }
@@ -870,7 +873,7 @@
                                if (CopyLength > ReaderCcb->WriteQuotaAvailable)
                                {
                                        DPRINT1("Writing %lu byte to pipe would
overflow as only %lu bytes are available\n",
-                                               CopyLength,
ReaderCcb->ReadDataAvailable);
+                                               CopyLength,
ReaderCcb->WriteQuotaAvailable);
                                        ASSERT(FALSE);
                                }