include/mm.h

   1 /*
   2  *  FreeLoader
   3  *  Copyright (C) 1998-2003  Brian Palmer  <brianp@sginet.com>
   4  *
   5  *  This program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with this program; if not, write to the Free Software
  17  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  */
  19
  20
  21 #ifndef __MEMORY_H
  22 #define __MEMORY_H
  23
  24 typedef enum
  25 {
  26         BiosMemoryUsable=1,
  27         BiosMemoryReserved,
  28         BiosMemoryAcpiReclaim,
  29         BiosMemoryAcpiNvs
  30 } BIOS_MEMORY_TYPE;
  31
  32 typedef struct
  33 {
  34         ULONGLONG               BaseAddress;
  35         ULONGLONG               Length;
  36         ULONG           Type;
  37         ULONG           Reserved;
  38 } PACKED BIOS_MEMORY_MAP, *PBIOS_MEMORY_MAP;
  39
  40 #if  defined(__i386__) || defined(_PPC_)
  41
  42 #define MM_PAGE_SIZE    4096
  43 #define MM_PAGE_MASK    0xFFF
  44 #define MM_PAGE_SHIFT   12
  45
  46 #define MM_SIZE_TO_PAGES(a)  \
  47         ( ((a) >> MM_PAGE_SHIFT) + ((a) & MM_PAGE_MASK ? 1 : 0) )
  48
  49 #endif // defined __i386__ or _PPC_
  50
  51 typedef struct
  52 {
  53         TYPE_OF_MEMORY  PageAllocated;                                  // Type of allocated memory (LoaderFree if this memory is free)
  54         ULONG                   PageAllocationLength;                   // Number of pages allocated (or zero if this isn't the first page in the chain)
  55 } PACKED PAGE_LOOKUP_TABLE_ITEM, *PPAGE_LOOKUP_TABLE_ITEM;
  56
  57 //
  58 // Define this to 1 if you want the entire contents
  59 // of the memory allocation bitmap displayed
  60 // when a chunk is allocated or freed
  61 //
  62 #define DUMP_MEM_MAP_ON_VERIFY  0
  63
  64
  65
  66 extern  PVOID   PageLookupTableAddress;
  67 extern  ULONG           TotalPagesInLookupTable;
  68 extern  ULONG           FreePagesInLookupTable;
  69 extern  ULONG           LastFreePageHint;
  70
  71 #ifdef DEBUG
  72 PUCHAR  MmGetSystemMemoryMapTypeString(ULONG Type);
  73 #endif
  74
  75 ULONG           MmGetPageNumberFromAddress(PVOID Address);      // Returns the page number that contains a linear address
  76 PVOID   MmGetEndAddressOfAnyMemory(PBIOS_MEMORY_MAP BiosMemoryMap, ULONG MapCount);     // Returns the last address of memory from the memory map
  77 ULONG           MmGetAddressablePageCountIncludingHoles(PBIOS_MEMORY_MAP BiosMemoryMap, ULONG MapCount);        // Returns the count of addressable pages from address zero including any memory holes and reserved memory regions
  78 PVOID   MmFindLocationForPageLookupTable(PBIOS_MEMORY_MAP BiosMemoryMap, ULONG MapCount);       // Returns the address for a memory chunk big enough to hold the page lookup table (starts search from end of memory)
  79 VOID    MmSortBiosMemoryMap(PBIOS_MEMORY_MAP BiosMemoryMap, ULONG MapCount);    // Sorts the BIOS_MEMORY_MAP array so the first element corresponds to the first address in memory
  80 VOID    MmInitPageLookupTable(PVOID PageLookupTable, ULONG TotalPageCount, PBIOS_MEMORY_MAP BiosMemoryMap, ULONG MapCount);     // Inits the page lookup table according to the memory types in the memory map
  81 VOID    MmMarkPagesInLookupTable(PVOID PageLookupTable, ULONG StartPage, ULONG PageCount, TYPE_OF_MEMORY PageAllocated);        // Marks the specified pages as allocated or free in the lookup table
  82 VOID    MmAllocatePagesInLookupTable(PVOID PageLookupTable, ULONG StartPage, ULONG PageCount);  // Allocates the specified pages in the lookup table
  83 ULONG           MmCountFreePagesInLookupTable(PVOID PageLookupTable, ULONG TotalPageCount);     // Returns the number of free pages in the lookup table
  84 ULONG           MmFindAvailablePages(PVOID PageLookupTable, ULONG TotalPageCount, ULONG PagesNeeded, BOOLEAN FromEnd);  // Returns the page number of the first available page range from the beginning or end of memory
  85 ULONG           MmFindAvailablePagesBeforePage(PVOID PageLookupTable, ULONG TotalPageCount, ULONG PagesNeeded, ULONG LastPage); // Returns the page number of the first available page range before the specified page
  86 VOID    MmFixupSystemMemoryMap(PBIOS_MEMORY_MAP BiosMemoryMap, ULONG* MapCount);        // Removes entries in the memory map that describe memory above 4G
  87 VOID    MmUpdateLastFreePageHint(PVOID PageLookupTable, ULONG TotalPageCount);  // Sets the LastFreePageHint to the last usable page of memory
  88 BOOLEAN MmAreMemoryPagesAvailable(PVOID PageLookupTable, ULONG TotalPageCount, PVOID PageAddress, ULONG PageCount);     // Returns TRUE if the specified pages of memory are available, otherwise FALSE
  89
  90 ULONG           GetSystemMemorySize(VOID);                                                              // Returns the amount of total memory in the system
  91 PPAGE_LOOKUP_TABLE_ITEM MmGetMemoryMap(ULONG *NoEntries);                       // Returns a pointer to the memory mapping table and a number of entries in it
  92
  93
  94 //BOOLEAN       MmInitializeMemoryManager(ULONG LowMemoryStart, ULONG LowMemoryLength);
  95 BOOLEAN MmInitializeMemoryManager(VOID);
  96 PVOID   MmAllocateMemory(ULONG MemorySize);
  97 VOID    MmFreeMemory(PVOID MemoryPointer);
  98 VOID    MmChangeAllocationPolicy(BOOLEAN PolicyAllocatePagesFromEnd);
  99 //PVOID MmAllocateLowMemory(ULONG MemorySize);
 100 //VOID  MmFreeLowMemory(PVOID MemoryPointer);
 101 PVOID   MmAllocateMemoryAtAddress(ULONG MemorySize, PVOID DesiredAddress);
 102 PVOID   MmAllocateHighestMemoryBelowAddress(ULONG MemorySize, PVOID DesiredAddress);
 103
 104 #endif // defined __MEMORY_H