2009-05-02 Vladimir Serbinenko <phcoder@gmail.com>
Mmap services * loader/i386/efi/linux.c (grub_linux_boot): use grub_mmap_iterate * loader/i386/linux.c (find_mmap_size): likewise (allocate_pages): likewise * loader/i386/multiboot.c (grub_get_multiboot_mmap_len): likewise (grub_fill_multiboot_mmap): likewise (grub_multiboot): use grub_mmap_get_lower and grub_mmap_get_upper * loader/i386/pc/linux.c (grub_cmd_linux): use grub_mmap_get_lower * include/grub/i386/bsd.h (OPENBSD_MMAP_AVAILABLE): new definition (OPENBSD_MMAP_RESERVED): likewise * include/grub/i386/pc/memory.h: include grub/memory.h (grub_lower_mem): removed (grub_upper_mem): likewise (GRUB_MACHINE_MEMORY_ACPI): new definition (GRUB_MACHINE_MEMORY_NVS): likewise (GRUB_MACHINE_MEMORY_MAX_TYPE): likewise (GRUB_MACHINE_MEMORY_HOLE): likewise (grub_machine_mmap_register): likewise (grub_machine_mmap_unregister): likewise (grub_machine_get_upper): likewise (grub_machine_get_lower): likewise (grub_machine_get_post64): likewise * include/grub/i386/efi/memory.h: new file * include/grub/x86_64/efi/memory.h: likewise * include/grub/efi/memory.h: likewise * conf/i386-pc.rmk (pkglib_MODULES): added mmap.mod (mmap_mod_SOURCES): new variable (mmap_mod_LDFLAGS): likewise (mmap_mod_ASFLAGS): likewise * conf/i386-coreboot.rmk: likewise * conf/i386-ieee1275.rmk: likewise * conf/i386-efi.rmk: likewise * conf/x86_64-efi.rmk: likewise * include/grub/types.h (UINT_TO_PTR): new macro (PTR_TO_UINT32): likewise (PTR_TO_UINT64): likewise * include/grub/memory.h: new file * mmap/i386/pc/mmap.c: likewise * mmap/i386/pc/mmap_helper.S: likewise * mmap/i386/uppermem.c: likewise * mmap/mmap.c: likewise * mmap/efi/mmap.c: likewise * kern/i386/coreboot/init.c (grub_machine_init): don't use grub_upper_mem * kern/i386/pc/init.c (grub_lower_mem): removed variable (grub_upper_mem): likewise (grub_machine_init): don't use grub_upper_mem, make grub_lower_mem local * loader/i386/bsd.c (grub_openbsd_boot): use grub_mmap_get_lower, grub_mmap_iterate and grub_mmap_get_upper (grub_netbsd_boot): use grub_mmap_get_lower and grub_mmap_get_upper
This commit is contained in:
phcoder
parent
d558e6b5ac
commit
09d842b9c6
27 changed files with 1579 additions and 128 deletions
284
mmap/efi/mmap.c
Normal file
284
mmap/efi/mmap.c
Normal file
|
|
@ -0,0 +1,284 @@
|
|||
/* Mmap management. */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2009 Free Software Foundation, Inc.
|
||||
*
|
||||
* GRUB is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* GRUB is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <grub/machine/memory.h>
|
||||
#include <grub/memory.h>
|
||||
#include <grub/err.h>
|
||||
#include <grub/efi/api.h>
|
||||
#include <grub/efi/efi.h>
|
||||
#include <grub/mm.h>
|
||||
#include <grub/misc.h>
|
||||
|
||||
#define NEXT_MEMORY_DESCRIPTOR(desc, size) \
|
||||
((grub_efi_memory_descriptor_t *) ((char *) (desc) + (size)))
|
||||
|
||||
grub_err_t
|
||||
grub_machine_mmap_iterate (int NESTED_FUNC_ATTR (*hook) (grub_uint64_t,
|
||||
grub_uint64_t,
|
||||
grub_uint32_t))
|
||||
{
|
||||
grub_efi_uintn_t mmap_size = 0;
|
||||
grub_efi_memory_descriptor_t *map_buf = 0;
|
||||
grub_efi_uintn_t map_key = 0;
|
||||
grub_efi_uintn_t desc_size = 0;
|
||||
grub_efi_uint32_t desc_version = 0;
|
||||
grub_efi_memory_descriptor_t *desc;
|
||||
|
||||
if (grub_efi_get_memory_map (&mmap_size, map_buf,
|
||||
&map_key, &desc_size,
|
||||
&desc_version) < 0)
|
||||
return grub_errno;
|
||||
|
||||
map_buf = grub_malloc (mmap_size);
|
||||
if (! map_buf)
|
||||
return grub_errno;
|
||||
|
||||
if (grub_efi_get_memory_map (&mmap_size, map_buf,
|
||||
&map_key, &desc_size,
|
||||
&desc_version) <= 0)
|
||||
{
|
||||
grub_free (map_buf);
|
||||
return grub_errno;
|
||||
}
|
||||
|
||||
for (desc = map_buf;
|
||||
desc < NEXT_MEMORY_DESCRIPTOR (map_buf, mmap_size);
|
||||
desc = NEXT_MEMORY_DESCRIPTOR (desc, desc_size))
|
||||
{
|
||||
grub_dprintf ("mmap", "EFI memory region 0x%llx-0x%llx: %d\n",
|
||||
(unsigned long long) desc->physical_start,
|
||||
(unsigned long long) desc->physical_start
|
||||
+ desc->num_pages * 4096, desc->type);
|
||||
switch (desc->type)
|
||||
{
|
||||
case GRUB_EFI_RUNTIME_SERVICES_CODE:
|
||||
hook (desc->physical_start, desc->num_pages * 4096,
|
||||
GRUB_MACHINE_MEMORY_CODE);
|
||||
break;
|
||||
|
||||
default:
|
||||
grub_printf ("Unknown memory type %d, considering reserved\n",
|
||||
desc->type);
|
||||
|
||||
case GRUB_EFI_RESERVED_MEMORY_TYPE:
|
||||
case GRUB_EFI_RUNTIME_SERVICES_DATA:
|
||||
case GRUB_EFI_UNUSABLE_MEMORY:
|
||||
case GRUB_EFI_MEMORY_MAPPED_IO:
|
||||
case GRUB_EFI_MEMORY_MAPPED_IO_PORT_SPACE:
|
||||
case GRUB_EFI_PAL_CODE:
|
||||
hook (desc->physical_start, desc->num_pages * 4096,
|
||||
GRUB_MACHINE_MEMORY_RESERVED);
|
||||
break;
|
||||
|
||||
case GRUB_EFI_LOADER_CODE:
|
||||
case GRUB_EFI_LOADER_DATA:
|
||||
case GRUB_EFI_BOOT_SERVICES_CODE:
|
||||
case GRUB_EFI_BOOT_SERVICES_DATA:
|
||||
case GRUB_EFI_CONVENTIONAL_MEMORY:
|
||||
hook (desc->physical_start, desc->num_pages * 4096,
|
||||
GRUB_MACHINE_MEMORY_AVAILABLE);
|
||||
break;
|
||||
|
||||
case GRUB_EFI_ACPI_RECLAIM_MEMORY:
|
||||
hook (desc->physical_start, desc->num_pages * 4096,
|
||||
GRUB_MACHINE_MEMORY_ACPI);
|
||||
break;
|
||||
|
||||
case GRUB_EFI_ACPI_MEMORY_NVS:
|
||||
hook (desc->physical_start, desc->num_pages * 4096,
|
||||
GRUB_MACHINE_MEMORY_NVS);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
|
||||
static inline grub_efi_memory_type_t
|
||||
make_efi_memtype (int type)
|
||||
{
|
||||
switch (type)
|
||||
{
|
||||
case GRUB_MACHINE_MEMORY_CODE:
|
||||
return GRUB_EFI_RUNTIME_SERVICES_CODE;
|
||||
|
||||
/* No way to remove a chunk of memory from EFI mmap.
|
||||
So mark it as unusable. */
|
||||
case GRUB_MACHINE_MEMORY_HOLE:
|
||||
|
||||
default:
|
||||
|
||||
case GRUB_MACHINE_MEMORY_RESERVED:
|
||||
return GRUB_EFI_UNUSABLE_MEMORY;
|
||||
|
||||
case GRUB_MACHINE_MEMORY_AVAILABLE:
|
||||
return GRUB_EFI_CONVENTIONAL_MEMORY;
|
||||
|
||||
case GRUB_MACHINE_MEMORY_ACPI:
|
||||
return GRUB_EFI_ACPI_RECLAIM_MEMORY;
|
||||
|
||||
case GRUB_MACHINE_MEMORY_NVS:
|
||||
return GRUB_EFI_ACPI_RECLAIM_MEMORY;
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
struct overlay
|
||||
{
|
||||
struct overlay *next;
|
||||
grub_efi_physical_address_t address;
|
||||
grub_efi_uintn_t pages;
|
||||
int handle;
|
||||
};
|
||||
|
||||
static struct overlay *overlays = 0;
|
||||
static int curhandle = 1;
|
||||
|
||||
int
|
||||
grub_mmap_register (grub_uint64_t start, grub_uint64_t size, int type)
|
||||
{
|
||||
grub_uint64_t end = start + size;
|
||||
grub_efi_physical_address_t address;
|
||||
grub_efi_boot_services_t *b;
|
||||
grub_efi_uintn_t pages;
|
||||
grub_efi_status_t status;
|
||||
struct overlay *curover;
|
||||
|
||||
curover = (struct overlay *) grub_malloc (sizeof (struct overlay));
|
||||
if (! curover)
|
||||
return 0;
|
||||
|
||||
b = grub_efi_system_table->boot_services;
|
||||
address = start & (~0x3ffULL);
|
||||
pages = (end - address + 0x3ff) >> 12;
|
||||
status = efi_call_2 (b->free_pages, address, pages);
|
||||
if (status != GRUB_EFI_SUCCESS && status != GRUB_EFI_NOT_FOUND)
|
||||
{
|
||||
grub_free (curover);
|
||||
return 0;
|
||||
}
|
||||
status = efi_call_4 (b->allocate_pages, GRUB_EFI_ALLOCATE_ADDRESS,
|
||||
make_efi_memtype (type), pages, &address);
|
||||
if (status != GRUB_EFI_SUCCESS)
|
||||
{
|
||||
grub_free (curover);
|
||||
return 0;
|
||||
}
|
||||
curover->next = overlays;
|
||||
curover->handle = curhandle++;
|
||||
curover->address = address;
|
||||
curover->pages = pages;
|
||||
overlays = curover;
|
||||
|
||||
return curover->handle;
|
||||
}
|
||||
|
||||
grub_err_t
|
||||
grub_machine_mmap_unregister (int handle)
|
||||
{
|
||||
struct overlay *curover, *prevover;
|
||||
grub_efi_boot_services_t *b;
|
||||
grub_efi_status_t status;
|
||||
|
||||
b = grub_efi_system_table->boot_services;
|
||||
|
||||
|
||||
for (curover = overlays, prevover = 0; curover;
|
||||
prevover = curover, curover = curover->next)
|
||||
{
|
||||
if (curover->handle == handle)
|
||||
{
|
||||
status = efi_call_2 (b->free_pages, curover->address, curover->pages);
|
||||
if (prevover != 0)
|
||||
prevover->next = curover->next;
|
||||
else
|
||||
overlays = curover->next;
|
||||
grub_free (curover);
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
}
|
||||
return grub_error (GRUB_ERR_BAD_ARGUMENT, "handle %d not found", handle);
|
||||
}
|
||||
|
||||
/* Result is always page-aligned. */
|
||||
char *
|
||||
grub_mmap_malign_and_register (grub_uint64_t align __attribute__ ((unused)),
|
||||
grub_uint64_t size,
|
||||
int *handle, int type,
|
||||
int flags __attribute__ ((unused)))
|
||||
{
|
||||
grub_efi_physical_address_t address;
|
||||
grub_efi_boot_services_t *b;
|
||||
grub_efi_uintn_t pages;
|
||||
grub_efi_status_t status;
|
||||
struct overlay *curover;
|
||||
grub_efi_allocate_type_t atype;
|
||||
|
||||
curover = (struct overlay *) grub_malloc (sizeof (struct overlay));
|
||||
if (! curover)
|
||||
return 0;
|
||||
|
||||
b = grub_efi_system_table->boot_services;
|
||||
|
||||
address = 0xffffffff;
|
||||
|
||||
#if GRUB_TARGET_SIZEOF_VOID_P < 8
|
||||
/* Limit the memory access to less than 4GB for 32-bit platforms. */
|
||||
atype = GRUB_EFI_ALLOCATE_MAX_ADDRESS;
|
||||
#else
|
||||
atype = GRUB_EFI_ALLOCATE_ANY_PAGES;
|
||||
#endif
|
||||
|
||||
pages = (size + 0x3ff) >> 12;
|
||||
status = efi_call_4 (b->allocate_pages, atype,
|
||||
make_efi_memtype (type), pages, &address);
|
||||
if (status != GRUB_EFI_SUCCESS)
|
||||
{
|
||||
grub_free (curover);
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (address == 0)
|
||||
{
|
||||
/* Uggh, the address 0 was allocated... This is too annoying,
|
||||
so reallocate another one. */
|
||||
address = 0xffffffff;
|
||||
status = efi_call_4 (b->allocate_pages, atype,
|
||||
make_efi_memtype (type), pages, &address);
|
||||
grub_efi_free_pages (0, pages);
|
||||
if (status != GRUB_EFI_SUCCESS)
|
||||
return 0;
|
||||
}
|
||||
|
||||
curover->next = overlays;
|
||||
curover->handle = curhandle++;
|
||||
curover->address = address;
|
||||
curover->pages = pages;
|
||||
overlays = curover;
|
||||
*handle = curover->handle;
|
||||
|
||||
return UINT_TO_PTR (curover->address);
|
||||
}
|
||||
|
||||
void
|
||||
grub_mmap_free_and_unregister (int handle)
|
||||
{
|
||||
grub_machine_mmap_unregister (handle);
|
||||
}
|
||||
98
mmap/i386/mmap.c
Normal file
98
mmap/i386/mmap.c
Normal file
|
|
@ -0,0 +1,98 @@
|
|||
/* Mmap management. */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2009 Free Software Foundation, Inc.
|
||||
*
|
||||
* GRUB is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* GRUB is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <grub/machine/memory.h>
|
||||
#include <grub/memory.h>
|
||||
#include <grub/err.h>
|
||||
#include <grub/misc.h>
|
||||
#include <grub/mm.h>
|
||||
|
||||
|
||||
#ifndef GRUB_MMAP_REGISTER_BY_FIRMWARE
|
||||
|
||||
char *
|
||||
grub_mmap_malign_and_register (grub_uint64_t align, grub_uint64_t size,
|
||||
int *handle, int type, int flags)
|
||||
{
|
||||
grub_uint64_t highestlow = 0;
|
||||
|
||||
auto int NESTED_FUNC_ATTR find_hook (grub_uint64_t, grub_uint64_t,
|
||||
grub_uint32_t);
|
||||
int NESTED_FUNC_ATTR find_hook (grub_uint64_t start, grub_uint64_t rangesize,
|
||||
grub_uint32_t memtype)
|
||||
{
|
||||
grub_uint64_t end = start + rangesize;
|
||||
if (memtype != GRUB_MACHINE_MEMORY_AVAILABLE)
|
||||
return 0;
|
||||
if (end > 0x100000)
|
||||
end = 0x100000;
|
||||
if (end > start + size
|
||||
&& highestlow < ((end - size) - ((end - size) & (align - 1))))
|
||||
highestlow = (end - size) - ((end - size) & (align - 1));
|
||||
return 0;
|
||||
}
|
||||
|
||||
char * ret;
|
||||
if (flags & GRUB_MMAP_MALLOC_LOW)
|
||||
{
|
||||
/* FIXME: use low-memory mm allocation once it's available. */
|
||||
grub_mmap_iterate (find_hook);
|
||||
ret = UINT_TO_PTR (highestlow);
|
||||
}
|
||||
else
|
||||
ret = grub_memalign (align, size);
|
||||
|
||||
if (! ret)
|
||||
{
|
||||
*handle = 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
*handle = grub_mmap_register (PTR_TO_UINT64 (ret), size, type);
|
||||
if (! *handle)
|
||||
{
|
||||
grub_free (ret);
|
||||
return 0;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void
|
||||
grub_mmap_free_and_unregister (int handle)
|
||||
{
|
||||
struct grub_mmap_region *cur;
|
||||
grub_uint64_t addr;
|
||||
|
||||
for (cur = grub_mmap_overlays; cur; cur = cur->next)
|
||||
if (cur->handle == handle)
|
||||
break;
|
||||
|
||||
if (! cur)
|
||||
return;
|
||||
|
||||
addr = cur->start;
|
||||
|
||||
grub_mmap_unregister (handle);
|
||||
|
||||
if (addr >= 0x100000)
|
||||
grub_free (UINT_TO_PTR (addr));
|
||||
}
|
||||
|
||||
#endif
|
||||
216
mmap/i386/pc/mmap.c
Normal file
216
mmap/i386/pc/mmap.c
Normal file
|
|
@ -0,0 +1,216 @@
|
|||
/* Mmap management. */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2009 Free Software Foundation, Inc.
|
||||
*
|
||||
* GRUB is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* GRUB is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <grub/machine/memory.h>
|
||||
#include <grub/memory.h>
|
||||
#include <grub/misc.h>
|
||||
#include <grub/term.h>
|
||||
#include <grub/loader.h>
|
||||
|
||||
#define min(a,b) (((a) < (b)) ? (a) : (b))
|
||||
|
||||
static void *hooktarget = 0;
|
||||
|
||||
extern grub_uint8_t grub_machine_mmaphook_start;
|
||||
extern grub_uint8_t grub_machine_mmaphook_end;
|
||||
extern grub_uint8_t grub_machine_mmaphook_int12;
|
||||
extern grub_uint8_t grub_machine_mmaphook_int15;
|
||||
|
||||
static grub_uint16_t grub_machine_mmaphook_int12offset = 0;
|
||||
static grub_uint16_t grub_machine_mmaphook_int12segment = 0;
|
||||
extern grub_uint16_t grub_machine_mmaphook_int15offset;
|
||||
extern grub_uint16_t grub_machine_mmaphook_int15segment;
|
||||
|
||||
extern grub_uint16_t grub_machine_mmaphook_mmap_num;
|
||||
extern grub_uint16_t grub_machine_mmaphook_kblow;
|
||||
extern grub_uint16_t grub_machine_mmaphook_kbin16mb;
|
||||
extern grub_uint16_t grub_machine_mmaphook_64kbin4gb;
|
||||
|
||||
struct grub_e820_mmap_entry
|
||||
{
|
||||
grub_uint64_t addr;
|
||||
grub_uint64_t len;
|
||||
grub_uint32_t type;
|
||||
} __attribute__((packed));
|
||||
|
||||
|
||||
static grub_err_t
|
||||
preboot (int noreturn __attribute__ ((unused)))
|
||||
{
|
||||
struct grub_e820_mmap_entry *hookmmap, *hookmmapcur;
|
||||
auto int NESTED_FUNC_ATTR fill_hook (grub_uint64_t, grub_uint64_t,
|
||||
grub_uint32_t);
|
||||
int NESTED_FUNC_ATTR fill_hook (grub_uint64_t addr, grub_uint64_t size,
|
||||
grub_uint32_t type)
|
||||
{
|
||||
grub_dprintf ("mmap", "mmap chunk %llx-%llx:%x\n", addr, addr + size, type);
|
||||
hookmmapcur->addr = addr;
|
||||
hookmmapcur->len = size;
|
||||
hookmmapcur->type = type;
|
||||
hookmmapcur++;
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (! hooktarget)
|
||||
return grub_error (GRUB_ERR_OUT_OF_MEMORY,
|
||||
"no space is allocated for memory hook");
|
||||
|
||||
grub_dprintf ("mmap", "installing preboot handlers\n");
|
||||
|
||||
hookmmapcur = hookmmap = (struct grub_e820_mmap_entry *)
|
||||
((grub_uint8_t *) hooktarget + (&grub_machine_mmaphook_end
|
||||
- &grub_machine_mmaphook_start));
|
||||
|
||||
grub_mmap_iterate (fill_hook);
|
||||
grub_machine_mmaphook_mmap_num = hookmmapcur - hookmmap;
|
||||
|
||||
grub_machine_mmaphook_kblow = grub_mmap_get_lower () >> 10;
|
||||
grub_machine_mmaphook_kbin16mb
|
||||
= min (grub_mmap_get_upper (),0x3f00000ULL) >> 10;
|
||||
grub_machine_mmaphook_64kbin4gb
|
||||
= min (grub_mmap_get_post64 (), 0xfc000000ULL) >> 16;
|
||||
|
||||
/* Correct BDA. */
|
||||
*((grub_uint16_t *) 0x413) = grub_mmap_get_lower () >> 10;
|
||||
|
||||
/* Save old interrupt handlers. */
|
||||
grub_machine_mmaphook_int12offset = *((grub_uint16_t *) 0x48);
|
||||
grub_machine_mmaphook_int12segment = *((grub_uint16_t *) 0x4a);
|
||||
grub_machine_mmaphook_int15offset = *((grub_uint16_t *) 0x54);
|
||||
grub_machine_mmaphook_int15segment = *((grub_uint16_t *) 0x56);
|
||||
|
||||
grub_dprintf ("mmap", "hooktarget = %p\n", hooktarget);
|
||||
|
||||
/* Install the interrupt handlers. */
|
||||
grub_memcpy (hooktarget, &grub_machine_mmaphook_start,
|
||||
&grub_machine_mmaphook_end - &grub_machine_mmaphook_start);
|
||||
|
||||
*((grub_uint16_t *) 0x4a) = PTR_TO_UINT32 (hooktarget) >> 4;
|
||||
*((grub_uint16_t *) 0x56) = PTR_TO_UINT32 (hooktarget) >> 4;
|
||||
*((grub_uint16_t *) 0x48) = &grub_machine_mmaphook_int12
|
||||
- &grub_machine_mmaphook_start;
|
||||
*((grub_uint16_t *) 0x54) = &grub_machine_mmaphook_int15
|
||||
- &grub_machine_mmaphook_start;
|
||||
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
|
||||
static grub_err_t
|
||||
preboot_rest (void)
|
||||
{
|
||||
/* Restore old interrupt handlers. */
|
||||
*((grub_uint16_t *) 0x48) = grub_machine_mmaphook_int12offset;
|
||||
*((grub_uint16_t *) 0x4a) = grub_machine_mmaphook_int12segment;
|
||||
*((grub_uint16_t *) 0x54) = grub_machine_mmaphook_int15offset;
|
||||
*((grub_uint16_t *) 0x56) = grub_machine_mmaphook_int15segment;
|
||||
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
|
||||
static grub_err_t
|
||||
malloc_hook (void)
|
||||
{
|
||||
static int reentry = 0;
|
||||
static int mmapregion = 0;
|
||||
static int slots_available = 0;
|
||||
int hooksize;
|
||||
int regcount = 0;
|
||||
auto int NESTED_FUNC_ATTR count_hook (grub_uint64_t, grub_uint64_t,
|
||||
grub_uint32_t);
|
||||
int NESTED_FUNC_ATTR count_hook (grub_uint64_t addr __attribute__ ((unused)),
|
||||
grub_uint64_t size __attribute__ ((unused)),
|
||||
grub_uint32_t type __attribute__ ((unused)))
|
||||
{
|
||||
regcount++;
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (reentry)
|
||||
return GRUB_ERR_NONE;
|
||||
|
||||
grub_dprintf ("mmap", "registering\n");
|
||||
|
||||
grub_mmap_iterate (count_hook);
|
||||
|
||||
/* Mapping hook itself may introduce up to 2 additional regions. */
|
||||
regcount += 2;
|
||||
|
||||
if (regcount <= slots_available)
|
||||
return GRUB_ERR_NONE;
|
||||
|
||||
if (mmapregion)
|
||||
{
|
||||
grub_mmap_free_and_unregister (mmapregion);
|
||||
mmapregion = 0;
|
||||
hooktarget = 0;
|
||||
}
|
||||
|
||||
hooksize = &grub_machine_mmaphook_end - &grub_machine_mmaphook_start
|
||||
+ regcount * sizeof (struct grub_e820_mmap_entry);
|
||||
/* Allocate an integer number of KiB. */
|
||||
hooksize = ((hooksize - 1) | 0x3ff) + 1;
|
||||
slots_available = (hooksize - (&grub_machine_mmaphook_end
|
||||
- &grub_machine_mmaphook_start))
|
||||
/ sizeof (struct grub_e820_mmap_entry);
|
||||
|
||||
reentry = 1;
|
||||
hooktarget
|
||||
= grub_mmap_malign_and_register (16, hooksize, &mmapregion,
|
||||
GRUB_MACHINE_MEMORY_RESERVED,
|
||||
GRUB_MMAP_MALLOC_LOW);
|
||||
reentry = 0;
|
||||
|
||||
if (! hooktarget)
|
||||
{
|
||||
slots_available = 0;
|
||||
return grub_error (GRUB_ERR_OUT_OF_MEMORY, "No space for mmap hook");
|
||||
}
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
|
||||
grub_err_t
|
||||
grub_machine_mmap_register (grub_uint64_t start __attribute__ ((unused)),
|
||||
grub_uint64_t size __attribute__ ((unused)),
|
||||
int type __attribute__ ((unused)),
|
||||
int handle __attribute__ ((unused)))
|
||||
{
|
||||
grub_err_t err;
|
||||
static void *preb_handle = 0;
|
||||
|
||||
err = malloc_hook ();
|
||||
if (err)
|
||||
return err;
|
||||
|
||||
if (! preb_handle)
|
||||
{
|
||||
grub_dprintf ("mmap", "adding preboot\n");
|
||||
preb_handle
|
||||
= grub_loader_register_preboot_hook (preboot, preboot_rest,
|
||||
GRUB_LOADER_PREBOOT_HOOK_PRIO_MEMORY);
|
||||
if (! preb_handle)
|
||||
return grub_errno;
|
||||
}
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
|
||||
grub_err_t
|
||||
grub_machine_mmap_unregister (int handle __attribute__ ((unused)))
|
||||
{
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
122
mmap/i386/pc/mmap_helper.S
Normal file
122
mmap/i386/pc/mmap_helper.S
Normal file
|
|
@ -0,0 +1,122 @@
|
|||
/* Mmap management. */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2009 Free Software Foundation, Inc.
|
||||
*
|
||||
* GRUB is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* GRUB is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <grub/symbol.h>
|
||||
|
||||
#define DS(x) ((x) - segstart)
|
||||
|
||||
segstart:
|
||||
VARIABLE(grub_machine_mmaphook_start)
|
||||
.code16
|
||||
VARIABLE(grub_machine_mmaphook_int12)
|
||||
push %ds
|
||||
push %cs
|
||||
pop %ds
|
||||
movw DS (EXT_C (grub_machine_mmaphook_kblow)), %ax
|
||||
pop %ds
|
||||
iret
|
||||
|
||||
VARIABLE(grub_machine_mmaphook_int15)
|
||||
pushf
|
||||
cmpw $0xe801, %ax
|
||||
jz e801
|
||||
cmpw $0xe820, %ax
|
||||
jz e820
|
||||
cmpb $0x88, %ah
|
||||
jz h88
|
||||
popf
|
||||
/* ljmp */
|
||||
.byte 0xea
|
||||
VARIABLE (grub_machine_mmaphook_int15offset)
|
||||
.word 0
|
||||
VARIABLE (grub_machine_mmaphook_int15segment)
|
||||
.word 0
|
||||
|
||||
e801:
|
||||
popf
|
||||
push %ds
|
||||
push %cs
|
||||
pop %ds
|
||||
movw DS (EXT_C (grub_machine_mmaphook_kbin16mb)), %ax
|
||||
movw DS (EXT_C (grub_machine_mmaphook_64kbin4gb)), %bx
|
||||
movw %ax, %cx
|
||||
movw %bx, %dx
|
||||
pop %ds
|
||||
clc
|
||||
iret
|
||||
|
||||
h88:
|
||||
popf
|
||||
push %ds
|
||||
push %cs
|
||||
pop %ds
|
||||
movw DS (EXT_C (grub_machine_mmaphook_kbin16mb)), %ax
|
||||
pop %ds
|
||||
clc
|
||||
iret
|
||||
|
||||
e820:
|
||||
popf
|
||||
push %ds
|
||||
push %cs
|
||||
pop %ds
|
||||
cmpw $20, %cx
|
||||
jb errexit
|
||||
cmpw DS (EXT_C (grub_machine_mmaphook_mmap_num)), %bx
|
||||
jae errexit
|
||||
cmp $0x534d4150, %edx
|
||||
jne errexit
|
||||
push %si
|
||||
push %di
|
||||
movw $20, %cx
|
||||
movw $(DS(mmaphook_mmap)), %si
|
||||
mov %bx, %ax
|
||||
imul $20, %ax
|
||||
add %ax, %si
|
||||
rep movsb
|
||||
pop %di
|
||||
pop %si
|
||||
movl $20, %ecx
|
||||
inc %bx
|
||||
cmpw DS(EXT_C(grub_machine_mmaphook_mmap_num)), %bx
|
||||
jb noclean
|
||||
xor %bx, %bx
|
||||
noclean:
|
||||
mov $0x534d4150, %eax
|
||||
pop %ds
|
||||
clc
|
||||
iret
|
||||
errexit:
|
||||
mov $0x534d4150, %eax
|
||||
pop %ds
|
||||
stc
|
||||
xor %bx, %bx
|
||||
iret
|
||||
|
||||
VARIABLE(grub_machine_mmaphook_mmap_num)
|
||||
.word 0
|
||||
VARIABLE(grub_machine_mmaphook_kblow)
|
||||
.word 0
|
||||
VARIABLE (grub_machine_mmaphook_kbin16mb)
|
||||
.word 0
|
||||
VARIABLE (grub_machine_mmaphook_64kbin4gb)
|
||||
.word 0
|
||||
mmaphook_mmap:
|
||||
/* Memory map is placed just after the interrupt handlers. */
|
||||
VARIABLE(grub_machine_mmaphook_end)
|
||||
85
mmap/i386/uppermem.c
Normal file
85
mmap/i386/uppermem.c
Normal file
|
|
@ -0,0 +1,85 @@
|
|||
/* Compute amount of lower and upper memory till the first hole. */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2009 Free Software Foundation, Inc.
|
||||
*
|
||||
* GRUB is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* GRUB is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <grub/memory.h>
|
||||
#include <grub/mm.h>
|
||||
#include <grub/misc.h>
|
||||
|
||||
grub_uint64_t
|
||||
grub_mmap_get_lower (void)
|
||||
{
|
||||
grub_uint64_t lower = 0;
|
||||
|
||||
auto int NESTED_FUNC_ATTR hook (grub_uint64_t, grub_uint64_t, grub_uint32_t);
|
||||
int NESTED_FUNC_ATTR hook (grub_uint64_t addr, grub_uint64_t size,
|
||||
grub_uint32_t type)
|
||||
{
|
||||
if (type != GRUB_MACHINE_MEMORY_AVAILABLE)
|
||||
return 0;
|
||||
if (addr == 0)
|
||||
lower = size;
|
||||
return 0;
|
||||
}
|
||||
|
||||
grub_mmap_iterate (hook);
|
||||
if (lower > 0x100000)
|
||||
lower = 0x100000;
|
||||
return lower;
|
||||
}
|
||||
|
||||
grub_uint64_t
|
||||
grub_mmap_get_upper (void)
|
||||
{
|
||||
grub_uint64_t upper = 0;
|
||||
|
||||
auto int NESTED_FUNC_ATTR hook (grub_uint64_t, grub_uint64_t, grub_uint32_t);
|
||||
int NESTED_FUNC_ATTR hook (grub_uint64_t addr, grub_uint64_t size,
|
||||
grub_uint32_t type)
|
||||
{
|
||||
if (type != GRUB_MACHINE_MEMORY_AVAILABLE)
|
||||
return 0;
|
||||
if (addr <= 0x100000 && addr + size > 0x100000)
|
||||
upper = addr + size - 0x100000;
|
||||
return 0;
|
||||
}
|
||||
|
||||
grub_mmap_iterate (hook);
|
||||
return upper;
|
||||
}
|
||||
|
||||
/* Count the continous bytes after 64 MiB. */
|
||||
grub_uint64_t
|
||||
grub_mmap_get_post64 (void)
|
||||
{
|
||||
grub_uint64_t post64 = 0;
|
||||
|
||||
auto int NESTED_FUNC_ATTR hook (grub_uint64_t, grub_uint64_t, grub_uint32_t);
|
||||
int NESTED_FUNC_ATTR hook (grub_uint64_t addr, grub_uint64_t size,
|
||||
grub_uint32_t type)
|
||||
{
|
||||
if (type != GRUB_MACHINE_MEMORY_AVAILABLE)
|
||||
return 0;
|
||||
if (addr <= 0x4000000 && addr + size > 0x4000000)
|
||||
post64 = addr + size - 0x4000000;
|
||||
return 0;
|
||||
}
|
||||
|
||||
grub_mmap_iterate (hook);
|
||||
return post64;
|
||||
}
|
||||
426
mmap/mmap.c
Normal file
426
mmap/mmap.c
Normal file
|
|
@ -0,0 +1,426 @@
|
|||
/* Mmap management. */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2009 Free Software Foundation, Inc.
|
||||
*
|
||||
* GRUB is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* GRUB is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <grub/machine/memory.h>
|
||||
#include <grub/memory.h>
|
||||
#include <grub/err.h>
|
||||
#include <grub/misc.h>
|
||||
#include <grub/mm.h>
|
||||
#include <grub/command.h>
|
||||
#include <grub/dl.h>
|
||||
|
||||
#ifndef GRUB_MMAP_REGISTER_BY_FIRMWARE
|
||||
|
||||
struct grub_mmap_region *grub_mmap_overlays = 0;
|
||||
static int curhandle = 1;
|
||||
|
||||
#endif
|
||||
|
||||
grub_err_t
|
||||
grub_mmap_iterate (int NESTED_FUNC_ATTR (*hook) (grub_uint64_t,
|
||||
grub_uint64_t, grub_uint32_t))
|
||||
{
|
||||
|
||||
/* This function resolves overlapping regions and sorts the memory map.
|
||||
It uses scanline (sweeping) algorithm.
|
||||
*/
|
||||
/* If same page is used by multiple types it's resolved
|
||||
according to priority:
|
||||
1 - free memory
|
||||
2 - memory usable by firmware-aware code
|
||||
3 - unusable memory
|
||||
4 - a range deliberately empty
|
||||
*/
|
||||
int priority[GRUB_MACHINE_MEMORY_MAX_TYPE + 2] =
|
||||
{
|
||||
#ifdef GRUB_MACHINE_MEMORY_AVAILABLE
|
||||
[GRUB_MACHINE_MEMORY_AVAILABLE] = 1,
|
||||
#endif
|
||||
#ifdef GRUB_MACHINE_MEMORY_RESERVED
|
||||
[GRUB_MACHINE_MEMORY_RESERVED] = 3,
|
||||
#endif
|
||||
#ifdef GRUB_MACHINE_MEMORY_ACPI
|
||||
[GRUB_MACHINE_MEMORY_ACPI] = 2,
|
||||
#endif
|
||||
#ifdef GRUB_MACHINE_MEMORY_CODE
|
||||
[GRUB_MACHINE_MEMORY_CODE] = 3,
|
||||
#endif
|
||||
#ifdef GRUB_MACHINE_MEMORY_NVS
|
||||
[GRUB_MACHINE_MEMORY_NVS] = 3,
|
||||
#endif
|
||||
[GRUB_MACHINE_MEMORY_HOLE] = 4,
|
||||
};
|
||||
|
||||
int i, k, done;
|
||||
|
||||
/* Scanline events. */
|
||||
struct grub_mmap_scan
|
||||
{
|
||||
/* At which memory address. */
|
||||
grub_uint64_t pos;
|
||||
/* 0 = region starts, 1 = region ends. */
|
||||
int type;
|
||||
/* Which type of memory region? */
|
||||
int memtype;
|
||||
};
|
||||
struct grub_mmap_scan *scanline_events;
|
||||
struct grub_mmap_scan t;
|
||||
|
||||
/* Previous scanline event. */
|
||||
grub_uint64_t lastaddr;
|
||||
int lasttype;
|
||||
/* Current scanline event. */
|
||||
int curtype;
|
||||
/* How many regions of given type overlap at current location? */
|
||||
int present[GRUB_MACHINE_MEMORY_MAX_TYPE + 2];
|
||||
/* Number of mmap chunks. */
|
||||
int mmap_num;
|
||||
|
||||
#ifndef GRUB_MMAP_REGISTER_BY_FIRMWARE
|
||||
struct grub_mmap_region *cur;
|
||||
#endif
|
||||
|
||||
auto int NESTED_FUNC_ATTR count_hook (grub_uint64_t, grub_uint64_t,
|
||||
grub_uint32_t);
|
||||
int NESTED_FUNC_ATTR count_hook (grub_uint64_t addr __attribute__ ((unused)),
|
||||
grub_uint64_t size __attribute__ ((unused)),
|
||||
grub_uint32_t type __attribute__ ((unused)))
|
||||
{
|
||||
mmap_num++;
|
||||
return 0;
|
||||
}
|
||||
|
||||
auto int NESTED_FUNC_ATTR fill_hook (grub_uint64_t, grub_uint64_t,
|
||||
grub_uint32_t);
|
||||
int NESTED_FUNC_ATTR fill_hook (grub_uint64_t addr,
|
||||
grub_uint64_t size,
|
||||
grub_uint32_t type)
|
||||
{
|
||||
scanline_events[i].pos = addr;
|
||||
scanline_events[i].type = 0;
|
||||
if (type <= GRUB_MACHINE_MEMORY_MAX_TYPE && priority[type])
|
||||
scanline_events[i].memtype = type;
|
||||
else
|
||||
{
|
||||
grub_dprintf ("mmap", "Unknown memory type %d. Assuming unusable\n",
|
||||
type);
|
||||
scanline_events[i].memtype = GRUB_MACHINE_MEMORY_RESERVED;
|
||||
}
|
||||
i++;
|
||||
|
||||
scanline_events[i].pos = addr + size;
|
||||
scanline_events[i].type = 1;
|
||||
scanline_events[i].memtype = scanline_events[i - 1].memtype;
|
||||
i++;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
mmap_num = 0;
|
||||
|
||||
#ifndef GRUB_MMAP_REGISTER_BY_FIRMWARE
|
||||
for (cur = grub_mmap_overlays; cur; cur = cur->next)
|
||||
mmap_num++;
|
||||
#endif
|
||||
|
||||
grub_machine_mmap_iterate (count_hook);
|
||||
|
||||
/* Initialize variables. */
|
||||
grub_memset (present, 0, sizeof (present));
|
||||
scanline_events = (struct grub_mmap_scan *)
|
||||
grub_malloc (sizeof (struct grub_mmap_scan) * 2 * mmap_num);
|
||||
|
||||
if (! scanline_events)
|
||||
{
|
||||
return grub_error (GRUB_ERR_OUT_OF_MEMORY,
|
||||
"couldn't allocate space for new memory map");
|
||||
}
|
||||
|
||||
i = 0;
|
||||
#ifndef GRUB_MMAP_REGISTER_BY_FIRMWARE
|
||||
/* Register scanline events. */
|
||||
for (cur = grub_mmap_overlays; cur; cur = cur->next)
|
||||
{
|
||||
scanline_events[i].pos = cur->start;
|
||||
scanline_events[i].type = 0;
|
||||
if (cur->type == GRUB_MACHINE_MEMORY_HOLE
|
||||
|| (cur->type >= 0 && cur->type <= GRUB_MACHINE_MEMORY_MAX_TYPE
|
||||
&& priority[cur->type]))
|
||||
scanline_events[i].memtype = cur->type;
|
||||
else
|
||||
scanline_events[i].memtype = GRUB_MACHINE_MEMORY_RESERVED;
|
||||
i++;
|
||||
|
||||
scanline_events[i].pos = cur->end;
|
||||
scanline_events[i].type = 1;
|
||||
scanline_events[i].memtype = scanline_events[i - 1].memtype;
|
||||
i++;
|
||||
}
|
||||
#endif /* ! GRUB_MMAP_REGISTER_BY_FIRMWARE */
|
||||
|
||||
grub_machine_mmap_iterate (fill_hook);
|
||||
|
||||
/* Primitive bubble sort. It has complexity O(n^2) but since we're
|
||||
unlikely to have more than 100 chunks it's probably one of the
|
||||
fastest for one purpose. */
|
||||
done = 1;
|
||||
while (done)
|
||||
{
|
||||
done = 0;
|
||||
for (i = 0; i < 2 * mmap_num - 1; i++)
|
||||
if (scanline_events[i + 1].pos < scanline_events[i].pos
|
||||
|| (scanline_events[i + 1].pos == scanline_events[i].pos
|
||||
&& scanline_events[i + 1].type == 0
|
||||
&& scanline_events[i].type == 1))
|
||||
{
|
||||
t = scanline_events[i + 1];
|
||||
scanline_events[i + 1] = scanline_events[i];
|
||||
scanline_events[i] = t;
|
||||
done = 1;
|
||||
}
|
||||
}
|
||||
|
||||
lastaddr = scanline_events[0].pos;
|
||||
lasttype = scanline_events[0].memtype;
|
||||
for (i = 0; i < 2 * mmap_num; i++)
|
||||
{
|
||||
/* Process event. */
|
||||
if (scanline_events[i].type)
|
||||
present[scanline_events[i].memtype]--;
|
||||
else
|
||||
present[scanline_events[i].memtype]++;
|
||||
|
||||
/* Determine current region type. */
|
||||
curtype = -1;
|
||||
for (k = 0; k <= GRUB_MACHINE_MEMORY_MAX_TYPE + 1; k++)
|
||||
if (present[k] && (curtype == -1 || priority[k] > priority[curtype]))
|
||||
curtype = k;
|
||||
|
||||
/* Anounce region to the hook if necessary. */
|
||||
if ((curtype == -1 || curtype != lasttype)
|
||||
&& lastaddr != scanline_events[i].pos
|
||||
&& lasttype != -1
|
||||
&& lasttype != GRUB_MACHINE_MEMORY_HOLE
|
||||
&& hook (lastaddr, scanline_events[i].pos - lastaddr, lasttype))
|
||||
{
|
||||
grub_free (scanline_events);
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
|
||||
/* Update last values if necessary. */
|
||||
if (curtype == -1 || curtype != lasttype)
|
||||
{
|
||||
lasttype = curtype;
|
||||
lastaddr = scanline_events[i].pos;
|
||||
}
|
||||
}
|
||||
|
||||
grub_free (scanline_events);
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
|
||||
#ifndef GRUB_MMAP_REGISTER_BY_FIRMWARE
|
||||
int
|
||||
grub_mmap_register (grub_uint64_t start, grub_uint64_t size, int type)
|
||||
{
|
||||
struct grub_mmap_region *cur;
|
||||
|
||||
grub_dprintf ("mmap", "registering\n");
|
||||
|
||||
cur = (struct grub_mmap_region *)
|
||||
grub_malloc (sizeof (struct grub_mmap_region));
|
||||
if (! cur)
|
||||
{
|
||||
grub_error (GRUB_ERR_OUT_OF_MEMORY,
|
||||
"couldn't allocate memory map overlay");
|
||||
return 0;
|
||||
}
|
||||
|
||||
cur->next = grub_mmap_overlays;
|
||||
cur->start = start;
|
||||
cur->end = start + size;
|
||||
cur->type = type;
|
||||
cur->handle = curhandle++;
|
||||
grub_mmap_overlays = cur;
|
||||
|
||||
if (grub_machine_mmap_register (start, size, type, curhandle))
|
||||
{
|
||||
grub_mmap_overlays = cur->next;
|
||||
grub_free (cur);
|
||||
return 0;
|
||||
}
|
||||
|
||||
return cur->handle;
|
||||
}
|
||||
|
||||
grub_err_t
|
||||
grub_mmap_unregister (int handle)
|
||||
{
|
||||
struct grub_mmap_region *cur, *prev;
|
||||
|
||||
for (cur = grub_mmap_overlays, prev = 0; cur; prev= cur, cur = cur->next)
|
||||
if (handle == cur->handle)
|
||||
{
|
||||
grub_err_t err;
|
||||
if ((err = grub_machine_mmap_unregister (handle)))
|
||||
return err;
|
||||
|
||||
if (prev)
|
||||
prev->next = cur->next;
|
||||
else
|
||||
grub_mmap_overlays = cur->next;
|
||||
grub_free (cur);
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
return grub_error (GRUB_ERR_BAD_ARGUMENT, "mmap overlay not found");
|
||||
}
|
||||
|
||||
#endif /* ! GRUB_MMAP_REGISTER_BY_FIRMWARE */
|
||||
|
||||
#define CHUNK_SIZE 0x400
|
||||
|
||||
static inline grub_uint64_t
|
||||
fill_mask (grub_uint64_t addr, grub_uint64_t mask, grub_uint64_t iterator)
|
||||
{
|
||||
int i, j;
|
||||
grub_uint64_t ret = (addr & mask);
|
||||
|
||||
/* Find first fixed bit. */
|
||||
for (i = 0; i < 64; i++)
|
||||
if ((mask & (1ULL << i)) != 0)
|
||||
break;
|
||||
j = 0;
|
||||
for (; i < 64; i++)
|
||||
if ((mask & (1ULL << i)) == 0)
|
||||
{
|
||||
if ((iterator & (1ULL << j)) != 0)
|
||||
ret |= 1ULL << i;
|
||||
j++;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
static grub_err_t
|
||||
grub_cmd_badram (grub_command_t cmd __attribute__ ((unused)),
|
||||
int argc, char **args)
|
||||
{
|
||||
char * str;
|
||||
grub_uint64_t badaddr, badmask;
|
||||
|
||||
auto int NESTED_FUNC_ATTR hook (grub_uint64_t, grub_uint64_t, grub_uint32_t);
|
||||
int NESTED_FUNC_ATTR hook (grub_uint64_t addr,
|
||||
grub_uint64_t size,
|
||||
grub_uint32_t type __attribute__ ((unused)))
|
||||
{
|
||||
grub_uint64_t iterator, low, high, cur;
|
||||
int tail, var;
|
||||
int i;
|
||||
grub_dprintf ("badram", "hook %llx+%llx\n", (unsigned long long) addr,
|
||||
(unsigned long long) size);
|
||||
|
||||
/* How many trailing zeros? */
|
||||
for (tail = 0; ! (badmask & (1ULL << tail)); tail++);
|
||||
|
||||
/* How many zeros in mask? */
|
||||
var = 0;
|
||||
for (i = 0; i < 64; i++)
|
||||
if (! (badmask & (1ULL << i)))
|
||||
var++;
|
||||
|
||||
if (fill_mask (badaddr, badmask, 0) >= addr)
|
||||
iterator = 0;
|
||||
else
|
||||
{
|
||||
low = 0;
|
||||
high = ~0ULL;
|
||||
/* Find starting value. Keep low and high such that
|
||||
fill_mask (low) < addr and fill_mask (high) >= addr;
|
||||
*/
|
||||
while (high - low > 1)
|
||||
{
|
||||
cur = (low + high) / 2;
|
||||
if (fill_mask (badaddr, badmask, cur) >= addr)
|
||||
high = cur;
|
||||
else
|
||||
low = cur;
|
||||
}
|
||||
iterator = high;
|
||||
}
|
||||
|
||||
for (; iterator < (1ULL << (var - tail))
|
||||
&& (cur = fill_mask (badaddr, badmask, iterator)) < addr + size;
|
||||
iterator++)
|
||||
{
|
||||
grub_dprintf ("badram", "%llx (size %llx) is a badram range\n",
|
||||
(long long) cur, (long long) (1ULL << tail) - 1);
|
||||
grub_mmap_register (cur, (1ULL << tail) - 1, GRUB_MACHINE_MEMORY_HOLE);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (argc != 1)
|
||||
return grub_error (GRUB_ERR_BAD_ARGUMENT, "badram string required");
|
||||
|
||||
grub_dprintf ("badram", "executing badram\n");
|
||||
|
||||
str = args[0];
|
||||
|
||||
while (1)
|
||||
{
|
||||
/* Parse address and mask. */
|
||||
badaddr = grub_strtoull (str, &str, 16);
|
||||
if (*str == ',')
|
||||
str++;
|
||||
badmask = grub_strtoull (str, &str, 16);
|
||||
if (*str == ',')
|
||||
str++;
|
||||
|
||||
if (grub_errno == GRUB_ERR_BAD_NUMBER)
|
||||
{
|
||||
grub_errno = 0;
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
|
||||
/* When part of a page is tainted, we discard the whole of it. There's
|
||||
no point in providing sub-page chunks. */
|
||||
badmask &= ~(CHUNK_SIZE - 1);
|
||||
|
||||
grub_dprintf ("badram", "badram %llx:%llx\n",
|
||||
(unsigned long long) badaddr, (unsigned long long) badmask);
|
||||
|
||||
grub_mmap_iterate (hook);
|
||||
}
|
||||
}
|
||||
|
||||
static grub_command_t cmd;
|
||||
|
||||
|
||||
GRUB_MOD_INIT(mmap)
|
||||
{
|
||||
(void) mod; /* To stop warning. */
|
||||
cmd = grub_register_command ("badram", grub_cmd_badram,
|
||||
"badram ADDR1,MASK1[,ADDR2,MASK2[,...]]",
|
||||
"declare memory regions as badram");
|
||||
}
|
||||
|
||||
GRUB_MOD_FINI(mmap)
|
||||
{
|
||||
grub_unregister_command (cmd);
|
||||
}
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue