2N/A/* Mmap management. */
2N/A/*
2N/A * GRUB -- GRand Unified Bootloader
2N/A * Copyright (C) 2009 Free Software Foundation, Inc.
2N/A *
2N/A * GRUB is free software: you can redistribute it and/or modify
2N/A * it under the terms of the GNU General Public License as published by
2N/A * the Free Software Foundation, either version 3 of the License, or
2N/A * (at your option) any later version.
2N/A *
2N/A * GRUB is distributed in the hope that it will be useful,
2N/A * but WITHOUT ANY WARRANTY; without even the implied warranty of
2N/A * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
2N/A * GNU General Public License for more details.
2N/A *
2N/A * You should have received a copy of the GNU General Public License
2N/A * along with GRUB. If not, see <http://www.gnu.org/licenses/>.
2N/A */
2N/A
2N/A#include <grub/machine/memory.h>
2N/A#include <grub/memory.h>
2N/A#include <grub/misc.h>
2N/A#include <grub/term.h>
2N/A#include <grub/loader.h>
2N/A
2N/A#define min(a,b) (((a) < (b)) ? (a) : (b))
2N/A
2N/Astatic void *hooktarget = 0;
2N/A
2N/Aextern grub_uint8_t grub_machine_mmaphook_start;
2N/Aextern grub_uint8_t grub_machine_mmaphook_end;
2N/Aextern grub_uint8_t grub_machine_mmaphook_int12;
2N/Aextern grub_uint8_t grub_machine_mmaphook_int15;
2N/A
2N/Astatic grub_uint16_t grub_machine_mmaphook_int12offset = 0;
2N/Astatic grub_uint16_t grub_machine_mmaphook_int12segment = 0;
2N/Aextern grub_uint16_t grub_machine_mmaphook_int15offset;
2N/Aextern grub_uint16_t grub_machine_mmaphook_int15segment;
2N/A
2N/Aextern grub_uint16_t grub_machine_mmaphook_mmap_num;
2N/Aextern grub_uint16_t grub_machine_mmaphook_kblow;
2N/Aextern grub_uint16_t grub_machine_mmaphook_kbin16mb;
2N/Aextern grub_uint16_t grub_machine_mmaphook_64kbin4gb;
2N/A
2N/Astruct grub_e820_mmap_entry
2N/A{
2N/A grub_uint64_t addr;
2N/A grub_uint64_t len;
2N/A grub_uint32_t type;
2N/A} __attribute__((packed));
2N/A
2N/A
2N/Astatic grub_err_t
2N/Apreboot (int noreturn __attribute__ ((unused)))
2N/A{
2N/A struct grub_e820_mmap_entry *hookmmap, *hookmmapcur;
2N/A auto int NESTED_FUNC_ATTR fill_hook (grub_uint64_t, grub_uint64_t,
2N/A grub_uint32_t);
2N/A int NESTED_FUNC_ATTR fill_hook (grub_uint64_t addr, grub_uint64_t size,
2N/A grub_memory_type_t type)
2N/A {
2N/A grub_dprintf ("mmap", "mmap chunk %llx-%llx:%x\n", addr, addr + size, type);
2N/A hookmmapcur->addr = addr;
2N/A hookmmapcur->len = size;
2N/A hookmmapcur->type = type;
2N/A hookmmapcur++;
2N/A return 0;
2N/A }
2N/A
2N/A if (! hooktarget)
2N/A return grub_error (GRUB_ERR_OUT_OF_MEMORY,
2N/A "no space is allocated for memory hook");
2N/A
2N/A grub_dprintf ("mmap", "installing preboot handlers\n");
2N/A
2N/A hookmmapcur = hookmmap = (struct grub_e820_mmap_entry *)
2N/A ((grub_uint8_t *) hooktarget + (&grub_machine_mmaphook_end
2N/A - &grub_machine_mmaphook_start));
2N/A
2N/A grub_mmap_iterate (fill_hook);
2N/A grub_machine_mmaphook_mmap_num = hookmmapcur - hookmmap;
2N/A
2N/A grub_machine_mmaphook_kblow = grub_mmap_get_lower () >> 10;
2N/A grub_machine_mmaphook_kbin16mb
2N/A = min (grub_mmap_get_upper (),0x3f00000ULL) >> 10;
2N/A grub_machine_mmaphook_64kbin4gb
2N/A = min (grub_mmap_get_post64 (), 0xfc000000ULL) >> 16;
2N/A
2N/A /* Correct BDA. */
2N/A *((grub_uint16_t *) 0x413) = grub_mmap_get_lower () >> 10;
2N/A
2N/A /* Save old interrupt handlers. */
2N/A grub_machine_mmaphook_int12offset = *((grub_uint16_t *) 0x48);
2N/A grub_machine_mmaphook_int12segment = *((grub_uint16_t *) 0x4a);
2N/A grub_machine_mmaphook_int15offset = *((grub_uint16_t *) 0x54);
2N/A grub_machine_mmaphook_int15segment = *((grub_uint16_t *) 0x56);
2N/A
2N/A grub_dprintf ("mmap", "hooktarget = %p\n", hooktarget);
2N/A
2N/A /* Install the interrupt handlers. */
2N/A grub_memcpy (hooktarget, &grub_machine_mmaphook_start,
2N/A &grub_machine_mmaphook_end - &grub_machine_mmaphook_start);
2N/A
2N/A *((grub_uint16_t *) 0x4a) = ((grub_addr_t) hooktarget) >> 4;
2N/A *((grub_uint16_t *) 0x56) = ((grub_addr_t) hooktarget) >> 4;
2N/A *((grub_uint16_t *) 0x48) = &grub_machine_mmaphook_int12
2N/A - &grub_machine_mmaphook_start;
2N/A *((grub_uint16_t *) 0x54) = &grub_machine_mmaphook_int15
2N/A - &grub_machine_mmaphook_start;
2N/A
2N/A return GRUB_ERR_NONE;
2N/A}
2N/A
2N/Astatic grub_err_t
2N/Apreboot_rest (void)
2N/A{
2N/A /* Restore old interrupt handlers. */
2N/A *((grub_uint16_t *) 0x48) = grub_machine_mmaphook_int12offset;
2N/A *((grub_uint16_t *) 0x4a) = grub_machine_mmaphook_int12segment;
2N/A *((grub_uint16_t *) 0x54) = grub_machine_mmaphook_int15offset;
2N/A *((grub_uint16_t *) 0x56) = grub_machine_mmaphook_int15segment;
2N/A
2N/A return GRUB_ERR_NONE;
2N/A}
2N/A
2N/Astatic grub_err_t
2N/Amalloc_hook (void)
2N/A{
2N/A static int reentry = 0;
2N/A static int mmapregion = 0;
2N/A static int slots_available = 0;
2N/A int hooksize;
2N/A int regcount = 0;
2N/A auto int NESTED_FUNC_ATTR count_hook (grub_uint64_t, grub_uint64_t,
2N/A grub_uint32_t);
2N/A int NESTED_FUNC_ATTR count_hook (grub_uint64_t addr __attribute__ ((unused)),
2N/A grub_uint64_t size __attribute__ ((unused)),
2N/A grub_memory_type_t type __attribute__ ((unused)))
2N/A {
2N/A regcount++;
2N/A return 0;
2N/A }
2N/A
2N/A if (reentry)
2N/A return GRUB_ERR_NONE;
2N/A
2N/A grub_dprintf ("mmap", "registering\n");
2N/A
2N/A grub_mmap_iterate (count_hook);
2N/A
2N/A /* Mapping hook itself may introduce up to 2 additional regions. */
2N/A regcount += 2;
2N/A
2N/A if (regcount <= slots_available)
2N/A return GRUB_ERR_NONE;
2N/A
2N/A if (mmapregion)
2N/A {
2N/A grub_mmap_free_and_unregister (mmapregion);
2N/A mmapregion = 0;
2N/A hooktarget = 0;
2N/A }
2N/A
2N/A hooksize = &grub_machine_mmaphook_end - &grub_machine_mmaphook_start
2N/A + regcount * sizeof (struct grub_e820_mmap_entry);
2N/A /* Allocate an integer number of KiB. */
2N/A hooksize = ((hooksize - 1) | 0x3ff) + 1;
2N/A slots_available = (hooksize - (&grub_machine_mmaphook_end
2N/A - &grub_machine_mmaphook_start))
2N/A / sizeof (struct grub_e820_mmap_entry);
2N/A
2N/A reentry = 1;
2N/A hooktarget
2N/A = grub_mmap_malign_and_register (16, hooksize, &mmapregion,
2N/A GRUB_MEMORY_RESERVED,
2N/A GRUB_MMAP_MALLOC_LOW);
2N/A reentry = 0;
2N/A
2N/A if (! hooktarget)
2N/A {
2N/A slots_available = 0;
2N/A return grub_error (GRUB_ERR_OUT_OF_MEMORY, "no space for mmap hook");
2N/A }
2N/A return GRUB_ERR_NONE;
2N/A}
2N/A
2N/Agrub_err_t
2N/Agrub_machine_mmap_register (grub_uint64_t start __attribute__ ((unused)),
2N/A grub_uint64_t size __attribute__ ((unused)),
2N/A int type __attribute__ ((unused)),
2N/A int handle __attribute__ ((unused)))
2N/A{
2N/A grub_err_t err;
2N/A static struct grub_preboot *preb_handle = 0;
2N/A
2N/A err = malloc_hook ();
2N/A if (err)
2N/A return err;
2N/A
2N/A if (! preb_handle)
2N/A {
2N/A grub_dprintf ("mmap", "adding preboot\n");
2N/A preb_handle
2N/A = grub_loader_register_preboot_hook (preboot, preboot_rest,
2N/A GRUB_LOADER_PREBOOT_HOOK_PRIO_MEMORY);
2N/A if (! preb_handle)
2N/A return grub_errno;
2N/A }
2N/A return GRUB_ERR_NONE;
2N/A}
2N/A
2N/Agrub_err_t
2N/Agrub_machine_mmap_unregister (int handle __attribute__ ((unused)))
2N/A{
2N/A return GRUB_ERR_NONE;
2N/A}