Unborking some threading code
[tpg/acess2.git] / Kernel / arch / x86_64 / mm_phys.c
index 77ca92e..c67b905 100644 (file)
@@ -3,7 +3,7 @@
  * 
  * Physical Memory Manager
  */
-#define DEBUG  1
+#define DEBUG  0
 #include <acess.h>
 #include <mboot.h>
 #include <mm_virt.h>
@@ -19,8 +19,8 @@ enum eMMPhys_Ranges
 };
 
 // === IMPORTS ===
-extern void    gKernelBase;
-extern void    gKernelEnd;
+extern char    gKernelBase[];
+extern char    gKernelEnd[];
 
 // === PROTOTYPES ===
 void   MM_InitPhys_Multiboot(tMBoot_Info *MBoot);
@@ -31,10 +31,10 @@ void        MM_DerefPhys(tPAddr PAddr);
  int   MM_int_GetRangeID( tPAddr Addr );
 
 // === GLOBALS ===
-tSpinlock      glPhysicalPages;
-Uint64 *gaSuperBitmap; // 1 bit = 64 Pages, 16 MiB Per Word
-Uint64 *gaMainBitmap;  // 1 bit = 1 Page, 256 KiB per Word
-Uint64 *gaMultiBitmap; // Each bit means that the page is being used multiple times
+tMutex glPhysicalPages;
+Uint64 *gaSuperBitmap = (void*)MM_PAGE_SUPBMP; // 1 bit = 64 Pages, 16 MiB Per Word
+Uint64 *gaMainBitmap = (void*)MM_PAGE_BITMAP;  // 1 bit = 1 Page, 256 KiB per Word
+Uint64 *gaMultiBitmap = (void*)MM_PAGE_DBLBMP; // Each bit means that the page is being used multiple times
 Uint32 *gaiPageReferences = (void*)MM_PAGE_COUNTS;     // Reference Counts
 tPAddr giFirstFreePage;        // First possibly free page
 Uint64 giPhysRangeFree[NUM_MM_PHYS_RANGES];    // Number of free pages in each range
@@ -208,9 +208,10 @@ void MM_InitPhys_Multiboot(tMBoot_Info *MBoot)
        
        LOG("Clearing multi bitmap");
        // Fill the bitmaps
-       memset(gaMultiBitmap, 0, numPages<<12);
+       memset(gaMultiBitmap, 0, (numPages<<12)/8);
        // - initialise to one, then clear the avaliable areas
-       memset(gaMainBitmap, -1, numPages<<12);
+       memset(gaMainBitmap, -1, (numPages<<12)/8);
+       memset(gaSuperBitmap, -1, (numPages<<12)/(8*64));
        LOG("Setting main bitmap");
        // - Clear all Type=1 areas
        LOG("Clearing valid regions");
@@ -228,14 +229,14 @@ void MM_InitPhys_Multiboot(tMBoot_Info *MBoot)
                size = ent->Size >> 12;
                
                if(base & 63) {
-                       Uint64  val = -1 << (base & 63);
+                       Uint64  val = -1LL << (base & 63);
                        gaMainBitmap[base / 64] &= ~val;
                        size -= (base & 63);
                        base += 64 - (base & 63);
                }
                memset( &gaMainBitmap[base / 64], 0, size/8 );
                if( size & 7 ) {
-                       Uint64  val = -1 << (size & 7);
+                       Uint64  val = -1LL << (size & 7);
                        val <<= (size/8)&7;
                        gaMainBitmap[base / 64] &= ~val;
                }
@@ -246,7 +247,7 @@ void MM_InitPhys_Multiboot(tMBoot_Info *MBoot)
                size = (size + (base & 63) + 63) >> 6;
                base = base >> 6;
                if(base & 63) {
-                       Uint64  val = -1 << (base & 63);
+                       Uint64  val = -1LL << (base & 63);
                        gaSuperBitmap[base / 64] &= ~val;
                        size -= (base & 63);
                        base += 64 - (base & 63);
@@ -258,7 +259,7 @@ void MM_InitPhys_Multiboot(tMBoot_Info *MBoot)
        size = firstFreePage >> 12;
        memset( &gaMainBitmap[base / 64], -1, size/8 );
        if( size & 7 ) {
-               Uint64  val = -1 << (size & 7);
+               Uint64  val = -1LL << (size & 7);
                val <<= (size/8)&7;
                gaMainBitmap[base / 64] |= val;
        }
@@ -268,7 +269,7 @@ void MM_InitPhys_Multiboot(tMBoot_Info *MBoot)
                if(gaiStaticAllocPages[i] != 0)
                        continue;
                gaMainBitmap[ gaiStaticAllocPages[i] >> (12+6) ]
-                       &= ~(1 << ((gaiStaticAllocPages[i]>>12)&63));
+                       &= ~(1LL << ((gaiStaticAllocPages[i]>>12)&63));
        }
        
        // Fill the super bitmap
@@ -276,8 +277,8 @@ void MM_InitPhys_Multiboot(tMBoot_Info *MBoot)
        memset(gaSuperBitmap, 0, superPages<<12);
        for( base = 0; base < (size+63)/64; base ++)
        {
-               if( gaMainBitmap[ base ] == -1 )
-                       gaSuperBitmap[ base/64 ] |= 1 << (base&63);
+               if( gaMainBitmap[ base ] + 1 == 0 )
+                       gaSuperBitmap[ base/64 ] |= 1LL << (base&63);
        }
        
        // Set free page counts
@@ -285,7 +286,7 @@ void MM_InitPhys_Multiboot(tMBoot_Info *MBoot)
        {
                 int    rangeID;
                // Skip allocated
-               if( gaMainBitmap[ base >> 6 ] & (1 << (base&63))  )     continue;
+               if( gaMainBitmap[ base >> 6 ] & (1LL << (base&63))  )   continue;
                
                // Get range ID
                rangeID = MM_int_GetRangeID( base << 12 );
@@ -323,11 +324,11 @@ tPAddr MM_AllocPhysRange(int Num, int Bits)
        if( Bits <= 0 || Bits >= 64 )   // Speedup for the common case
                rangeID = MM_PHYS_MAX;
        else
-               rangeID = MM_int_GetRangeID( (1 << Bits) -1 );
+               rangeID = MM_int_GetRangeID( (1LL << Bits) - 1 );
        
        LOG("rangeID = %i", rangeID);
        
-       LOCK(&glPhysicalPages);
+       Mutex_Acquire(&glPhysicalPages);
        
        // Check if the range actually has any free pages
        while(giPhysRangeFree[rangeID] == 0 && rangeID)
@@ -337,7 +338,7 @@ tPAddr MM_AllocPhysRange(int Num, int Bits)
        
        // What the? Oh, man. No free pages
        if(giPhysRangeFree[rangeID] == 0) {
-               RELEASE(&glPhysicalPages);
+               Mutex_Release(&glPhysicalPages);
                // TODO: Page out
                // ATM. Just Warning
                Warning(" MM_AllocPhysRange: Out of free pages");
@@ -364,23 +365,23 @@ tPAddr MM_AllocPhysRange(int Num, int Bits)
                {
                        //Log(" MM_AllocPhysRange: addr = 0x%x", addr);
                        // Check the super bitmap
-                       if( gaSuperBitmap[addr >> (6+6)] == -1 ) {
+                       if( gaSuperBitmap[addr >> (6+6)] + 1 == 0 ) {
                                LOG("nFree = %i = 0 (super) (0x%x)", nFree, addr);
                                nFree = 0;
-                               addr += 1 << (6+6);
-                               addr &= (1 << (6+6)) - 1;
+                               addr += 1LL << (6+6);
+                               addr &= ~0xFFF; // (1LL << 6+6) - 1
                                continue;
                        }
                        // Check page block (64 pages)
-                       if( gaSuperBitmap[addr >> (6+6)] & (1 << (addr>>6)&63)) {
+                       if( gaMainBitmap[addr >> 6] + 1 == 0) {
                                LOG("nFree = %i = 0 (main) (0x%x)", nFree, addr);
                                nFree = 0;
-                               addr += 1 << (12+6);
-                               addr &= (1 << (12+6)) - 1;
+                               addr += 1LL << (6);
+                               addr &= ~0x3F;
                                continue;
                        }
                        // Check individual page
-                       if( gaMainBitmap[addr >> 6] & (1 << (addr & 63)) ) {
+                       if( gaMainBitmap[addr >> 6] & (1LL << (addr & 63)) ) {
                                LOG("nFree = %i = 0 (page) (0x%x)", nFree, addr);
                                nFree = 0;
                                addr ++;
@@ -405,7 +406,7 @@ tPAddr MM_AllocPhysRange(int Num, int Bits)
                nFree = 1;
                addr = giPhysRangeLast[ rangeID ];
                // TODO
-               RELEASE(&glPhysicalPages);
+               Mutex_Release(&glPhysicalPages);
                // TODO: Page out
                // ATM. Just Warning
                Warning(" MM_AllocPhysRange: Out of memory (unable to fulfil request for %i pages)", Num);
@@ -422,9 +423,12 @@ tPAddr MM_AllocPhysRange(int Num, int Bits)
        addr -= Num;
        for( i = 0; i < Num; i++, addr++ )
        {
-               gaMainBitmap[addr >> 6] |= 1 << (addr & 63);
-               rangeID = MM_int_GetRangeID(addr);
+               gaMainBitmap[addr >> 6] |= 1LL << (addr & 63);
+               rangeID = MM_int_GetRangeID(addr << 12);
                giPhysRangeFree[ rangeID ] --;
+               LOG("%x == %x", addr, giPhysRangeFirst[ rangeID ]);
+               if(addr == giPhysRangeFirst[ rangeID ])
+                       giPhysRangeFirst[ rangeID ] += 1;
        }
        ret = addr;     // Save the return address
        
@@ -434,11 +438,11 @@ tPAddr MM_AllocPhysRange(int Num, int Bits)
        Num = (Num + (64-1)) & ~(64-1);
        for( i = 0; i < Num/64; i++ )
        {
-               if( gaMainBitmap[ addr >> 6 ] == -1 )
-                       gaSuperBitmap[addr>>12] |= 1 << ((addr >> 6) & 64);
+               if( gaMainBitmap[ addr >> 6 ] + 1 == 0 )
+                       gaSuperBitmap[addr>>12] |= 1LL << ((addr >> 6) & 63);
        }
        
-       RELEASE(&glPhysicalPages);
+       Mutex_Release(&glPhysicalPages);
        LEAVE('x', ret << 12);
        return ret << 12;
 }
@@ -473,18 +477,18 @@ void MM_RefPhys(tPAddr PAddr)
        
        if( PAddr >> 12 > giMaxPhysPage )       return ;
        
-       if( gaMainBitmap[ page >> 6 ] & (1 << (page&63)) )
+       if( gaMainBitmap[ page >> 6 ] & (1LL << (page&63)) )
        {
                // Reference again
-               gaMultiBitmap[ page >> 6 ] |= 1 << (page&63);
+               gaMultiBitmap[ page >> 6 ] |= 1LL << (page&63);
                gaiPageReferences[ page ] ++;
        }
        else
        {
                // Allocate
-               gaMainBitmap[page >> 6] |= 1 << (page&63);
-               if( gaMainBitmap[page >> 6 ] == -1 )
-                       gaSuperBitmap[page>> 12] |= 1 << ((page >> 6) & 63);
+               gaMainBitmap[page >> 6] |= 1LL << (page&63);
+               if( gaMainBitmap[page >> 6 ] + 1 == 0 )
+                       gaSuperBitmap[page>> 12] |= 1LL << ((page >> 6) & 63);
        }
 }
 
@@ -497,26 +501,31 @@ void MM_DerefPhys(tPAddr PAddr)
        
        if( PAddr >> 12 > giMaxPhysPage )       return ;
        
-       if( gaMultiBitmap[ page >> 6 ] & (1 << (page&63)) ) {
+       if( gaMultiBitmap[ page >> 6 ] & (1LL << (page&63)) ) {
                gaiPageReferences[ page ] --;
                if( gaiPageReferences[ page ] == 1 )
-                       gaMultiBitmap[ page >> 6 ] &= ~(1 << (page&63));
+                       gaMultiBitmap[ page >> 6 ] &= ~(1LL << (page&63));
                if( gaiPageReferences[ page ] == 0 )
-                       gaMainBitmap[ page >> 6 ] &= ~(1 << (page&63));
+                       gaMainBitmap[ page >> 6 ] &= ~(1LL << (page&63));
        }
        else
-               gaMainBitmap[ page >> 6 ] &= ~(1 << (page&63));
+               gaMainBitmap[ page >> 6 ] &= ~(1LL << (page&63));
        
-       // TODO: Update free counts
-       if( !(gaMainBitmap[ page >> 6 ] & (1 << (page&63))) )
+       // Update the free counts if the page was freed
+       if( !(gaMainBitmap[ page >> 6 ] & (1LL << (page&63))) )
        {
                 int    rangeID;
                rangeID = MM_int_GetRangeID( PAddr );
                giPhysRangeFree[ rangeID ] ++;
+               if( giPhysRangeFirst[rangeID] > page )
+                       giPhysRangeFirst[rangeID] = page;
+               if( giPhysRangeLast[rangeID] < page )
+                       giPhysRangeLast[rangeID] = page;
        }
        
-       if(gaMainBitmap[ page >> 6 ] == 0) {
-               gaSuperBitmap[page >> 12] &= ~(1 << ((page >> 6) & 63));
+       // If the bitmap entry is not -1, unset the bit in the super bitmap
+       if(gaMainBitmap[ page >> 6 ] + 1 != 0 ) {
+               gaSuperBitmap[page >> 12] &= ~(1LL << ((page >> 6) & 63));
        }
 }
 

UCC git Repository :: git.ucc.asn.au