git://git.jankratochvil.net / reactos.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
update for HEAD-2003021201
[reactos.git] / ntoskrnl / rtl / bitmap.c
diff --git a/ntoskrnl/rtl/bitmap.c b/ntoskrnl/rtl/bitmap.c
index 3760fa6..db9b430 100644 (file)
--- a/ntoskrnl/rtl/bitmap.c
+++ b/ntoskrnl/rtl/bitmap.c
@@ -13,6 +13,8 @@
 
 #define ALIGN(x,align) (((x)+(align)-1) / (align))
 
+#define MASK(Count, Shift) ((Count) == 32 ? 0xFFFFFFFF : ~(0xFFFFFFFF << (Count)) << (Shift))
+
 
 VOID
 STDCALL
@@ -38,24 +40,24 @@ RtlAreBitsClear (
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Shift;
        ULONG Count;
-       PCHAR Ptr;
+       PULONG Ptr;
 
        if (StartingIndex >= Size ||
            !Length ||
            (StartingIndex + Length > Size))
                return FALSE;
 
-       Ptr = (PCHAR)BitMapHeader->Buffer + (StartingIndex / 8);
+       Ptr = (PULONG)BitMapHeader->Buffer + (StartingIndex / 32);
        while (Length)
        {
-               /* get bit shift in current byte */
-               Shift = StartingIndex & 7;
+               /* get bit shift in current dword */
+               Shift = StartingIndex & 0x1F;
 
-               /* get number of bits to check in current byte */
-               Count = (Length > 8 - Shift) ? 8 - Shift : Length;
+               /* get number of bits to check in current dword */
+               Count = (Length > 32 - Shift) ? 32 - Shift : Length;
 
-               /* check byte */
-               if (*Ptr++ & (~(0xFF << Count) << Shift))
+               /* check dword */
+               if (*Ptr++ & MASK(Count, Shift))
                        return FALSE;
 
                Length -= Count;
@@ -77,24 +79,24 @@ RtlAreBitsSet (
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Shift;
        ULONG Count;
-       PCHAR Ptr;
+       PULONG Ptr;
 
        if (StartingIndex >= Size ||
            !Length ||
            (StartingIndex + Length > Size))
                return FALSE;
 
-       Ptr = (PCHAR)BitMapHeader->Buffer + (StartingIndex / 8);
+       Ptr = (PULONG)BitMapHeader->Buffer + (StartingIndex / 32);
        while (Length)
        {
-               /* get bit shift in current byte */
-               Shift = StartingIndex & 7;
+               /* get bit shift in current dword */
+               Shift = StartingIndex & 0x1F;
 
-               /* get number of bits to check in current byte */
-               Count = (Length > 8 - Shift) ? 8 - Shift : Length;
+               /* get number of bits to check in current dword */
+               Count = (Length > 32 - Shift) ? 32 - Shift : Length;
 
-               /* check byte */
-               if (~*Ptr++ & (~(0xFF << Count) << Shift))
+               /* check dword */
+               if (~*Ptr++ & MASK(Count, Shift))
                        return FALSE;
 
                Length -= Count;
@@ -128,7 +130,7 @@ RtlClearBits (
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Count;
        ULONG Shift;
-       PCHAR Ptr;
+       PULONG Ptr;
 
        if (StartingIndex >= Size || NumberToClear == 0)
                return;
@@ -136,17 +138,17 @@ RtlClearBits (
        if (StartingIndex + NumberToClear > Size)
                NumberToClear = Size - StartingIndex;
 
-       Ptr = (PCHAR)BitMapHeader->Buffer + (StartingIndex / 8);
+       Ptr = (PULONG)BitMapHeader->Buffer + (StartingIndex / 32);
        while (NumberToClear)
        {
-               /* bit shift in current byte */
-               Shift = StartingIndex & 7;
+               /* bit shift in current dword */
+               Shift = StartingIndex & 0x1F;
 
-               /* number of bits to change in current byte */
-               Count = (NumberToClear > 8 - Shift ) ? 8 - Shift : NumberToClear;
+               /* number of bits to change in current dword */
+               Count = (NumberToClear > 32 - Shift ) ? 32 - Shift : NumberToClear;
 
-               /* adjust byte */
-               *Ptr++ &= ~(~(0xFF << Count) << Shift);
+               /* adjust dword */
+               *Ptr++ &= ~MASK(Count, Shift);
                NumberToClear -= Count;
                StartingIndex += Count;
        }
@@ -164,8 +166,8 @@ RtlFindClearBits (
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Index;
        ULONG Count;
-       PCHAR Ptr;
-       CHAR  Mask;
+       PULONG Ptr;
+       ULONG  Mask;
 
        if (NumberToFind > Size || NumberToFind == 0)
                return -1;
@@ -174,8 +176,8 @@ RtlFindClearBits (
                HintIndex = 0;
 
        Index = HintIndex;
-       Ptr = (PCHAR)BitMapHeader->Buffer + (Index / 8);
-       Mask  = 1 << (Index & 7);
+       Ptr = (PULONG)BitMapHeader->Buffer + (Index / 32);
+       Mask  = 1 << (Index & 0x1F);
 
        while (HintIndex < Size)
        {
@@ -241,8 +243,8 @@ RtlFindFirstRunClear (
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Index;
        ULONG Count;
-       PCHAR Ptr;
-       CHAR  Mask;
+       PULONG Ptr;
+       ULONG  Mask;
 
        if (*StartingIndex > Size)
        {
@@ -251,8 +253,8 @@ RtlFindFirstRunClear (
        }
 
        Index = *StartingIndex;
-       Ptr = (PCHAR)BitMapHeader->Buffer + (Index / 8);
-       Mask = 1 << (Index & 7);
+       Ptr = (PULONG)BitMapHeader->Buffer + (Index / 32);
+       Mask = 1 << (Index & 0x1F);
 
        /* skip set bits */
        for (; Index < Size && *Ptr & Mask; Index++)
@@ -300,8 +302,8 @@ RtlFindFirstRunSet (
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Index;
        ULONG Count;
-       PCHAR Ptr;
-       CHAR  Mask;
+       PULONG Ptr;
+       ULONG  Mask;
 
        if (*StartingIndex > Size)
        {
@@ -310,8 +312,8 @@ RtlFindFirstRunSet (
        }
 
        Index = *StartingIndex;
-       Ptr = (PCHAR)BitMapHeader->Buffer + (Index / 8);
-       Mask = 1 << (Index & 7);
+       Ptr = (PULONG)BitMapHeader->Buffer + (Index / 32);
+       Mask = 1 << (Index & 0x1F);
 
        /* skip clear bits */
        for (; Index < Size && ~*Ptr & Mask; Index++)
@@ -357,13 +359,13 @@ RtlFindLongestRunClear (
        )
 {
        ULONG Size = BitMapHeader->SizeOfBitMap;
-       PCHAR Ptr = (PCHAR)BitMapHeader->Buffer;
+       PULONG Ptr = (PULONG)BitMapHeader->Buffer;
        ULONG Index = 0;
        ULONG Count;
        ULONG Max = 0;
        ULONG Start;
        ULONG Maxstart = 0;
-       CHAR  Mask = 1;
+       ULONG  Mask = 1;
 
        while (Index < Size)
        {
@@ -414,13 +416,13 @@ RtlFindLongestRunSet (
        )
 {
        ULONG Size = BitMapHeader->SizeOfBitMap;
-       PCHAR Ptr = (PCHAR)BitMapHeader->Buffer;
+       PULONG Ptr = (PULONG)BitMapHeader->Buffer;
        ULONG Index = 0;
        ULONG Count;
        ULONG Max = 0;
        ULONG Start;
        ULONG Maxstart = 0;
-       CHAR  Mask = 1;
+       ULONG Mask = 1;
 
        while (Index < Size)
        {
@@ -474,8 +476,8 @@ RtlFindSetBits (
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Index;
        ULONG Count;
-       PCHAR Ptr;
-       CHAR  Mask;
+       PULONG Ptr;
+       ULONG Mask;
 
        if (NumberToFind > Size || NumberToFind == 0)
                return (ULONG)-1;
@@ -484,8 +486,8 @@ RtlFindSetBits (
                HintIndex = 0;
 
        Index = HintIndex;
-       Ptr = (PCHAR)BitMapHeader->Buffer + (Index / 8);
-       Mask = 1 << (Index & 7);
+       Ptr = (PULONG)BitMapHeader->Buffer + (Index / 32);
+       Mask = 1 << (Index & 0x1F);
 
        while (HintIndex < Size)
        {
@@ -547,11 +549,11 @@ RtlNumberOfClearBits (
        PRTL_BITMAP     BitMapHeader
        )
 {
-       PCHAR Ptr = (PCHAR)BitMapHeader->Buffer;
+       PULONG Ptr = (PULONG)BitMapHeader->Buffer;
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Index;
        ULONG Count;
-       CHAR Mask;
+       ULONG Mask;
 
        for (Mask = 1, Index = 0, Count = 0; Index < Size; Index++)
        {
@@ -576,11 +578,11 @@ RtlNumberOfSetBits (
        PRTL_BITMAP     BitMapHeader
        )
 {
-       PCHAR Ptr = (PCHAR)BitMapHeader->Buffer;
+       PULONG Ptr = (PULONG)BitMapHeader->Buffer;
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Index;
        ULONG Count;
-       CHAR Mask;
+       ULONG Mask;
 
        for (Mask = 1, Index = 0, Count = 0; Index < Size; Index++)
        {
@@ -622,7 +624,7 @@ RtlSetBits (
        ULONG Size = BitMapHeader->SizeOfBitMap;
        ULONG Count;
        ULONG Shift;
-       PCHAR Ptr;
+       PULONG Ptr;
 
        if (StartingIndex >= Size || NumberToSet == 0)
                return;
@@ -630,17 +632,17 @@ RtlSetBits (
        if (StartingIndex + NumberToSet > Size)
                NumberToSet = Size - StartingIndex;
 
-       Ptr = (PCHAR)BitMapHeader->Buffer + (StartingIndex / 8);
+       Ptr = (PULONG)BitMapHeader->Buffer + (StartingIndex / 32);
        while (NumberToSet)
        {
-               /* bit shift in current byte */
-               Shift = StartingIndex & 7;
+               /* bit shift in current dword */
+               Shift = StartingIndex & 0x1F;
 
-               /* number of bits to change in current byte */
-               Count = (NumberToSet > 8 - Shift) ? 8 - Shift : NumberToSet;
+               /* number of bits to change in current dword */
+               Count = (NumberToSet > 32 - Shift) ? 32 - Shift : NumberToSet;
 
-               /* adjust byte */
-               *Ptr++ |= ~(0xFF << Count) << Shift;
+               /* adjust dword */
+               *Ptr++ |= MASK(Count, Shift);
                NumberToSet -= Count;
                StartingIndex += Count;
        }
Unnamed repository; edit this file 'description' to name the repository.