^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) * fake_mem.c
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) * Copyright (C) 2015 FUJITSU LIMITED
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) * Author: Taku Izumi <izumi.taku@jp.fujitsu.com>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) * This code introduces new boot option named "efi_fake_mem"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) * By specifying this parameter, you can add arbitrary attribute to
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) * specific memory range by updating original (firmware provided) EFI
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) * memmap.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) #include <linux/kernel.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) #include <linux/efi.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) #include <linux/init.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) #include <linux/memblock.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) #include <linux/types.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) #include <linux/sort.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20) #include "fake_mem.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) struct efi_mem_range efi_fake_mems[EFI_MAX_FAKEMEM];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) int nr_fake_mem;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) static int __init cmp_fake_mem(const void *x1, const void *x2)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) const struct efi_mem_range *m1 = x1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) const struct efi_mem_range *m2 = x2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) if (m1->range.start < m2->range.start)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) if (m1->range.start > m2->range.start)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) return 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) static void __init efi_fake_range(struct efi_mem_range *efi_range)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) struct efi_memory_map_data data = { 0 };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) int new_nr_map = efi.memmap.nr_map;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41) efi_memory_desc_t *md;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) void *new_memmap;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44) /* count up the number of EFI memory descriptor */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) for_each_efi_memory_desc(md)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) new_nr_map += efi_memmap_split_count(md, &efi_range->range);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48) /* allocate memory for new EFI memmap */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49) if (efi_memmap_alloc(new_nr_map, &data) != 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52) /* create new EFI memmap */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) new_memmap = early_memremap(data.phys_map, data.size);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54) if (!new_memmap) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55) __efi_memmap_free(data.phys_map, data.size, data.flags);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56) return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59) efi_memmap_insert(&efi.memmap, new_memmap, efi_range);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 61) /* swap into new EFI memmap */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 62) early_memunmap(new_memmap, data.size);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 63)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 64) efi_memmap_install(&data);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 65) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 66)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 67) void __init efi_fake_memmap(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 68) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 69) int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 70)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 71) if (!efi_enabled(EFI_MEMMAP) || !nr_fake_mem)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 72) return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 73)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 74) for (i = 0; i < nr_fake_mem; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 75) efi_fake_range(&efi_fake_mems[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 76)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 77) /* print new EFI memmap */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 78) efi_print_memmap();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 79) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 80)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 81) static int __init setup_fake_mem(char *p)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 82) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 83) u64 start = 0, mem_size = 0, attribute = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 84) int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 85)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 86) if (!p)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 87) return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 88)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 89) while (*p != '\0') {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 90) mem_size = memparse(p, &p);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 91) if (*p == '@')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 92) start = memparse(p+1, &p);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 93) else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 94) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 95)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 96) if (*p == ':')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 97) attribute = simple_strtoull(p+1, &p, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 98) else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 99) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 100)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 101) if (nr_fake_mem >= EFI_MAX_FAKEMEM)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 102) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 103)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 104) efi_fake_mems[nr_fake_mem].range.start = start;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 105) efi_fake_mems[nr_fake_mem].range.end = start + mem_size - 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 106) efi_fake_mems[nr_fake_mem].attribute = attribute;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 107) nr_fake_mem++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 108)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 109) if (*p == ',')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 110) p++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 111) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 112)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 113) sort(efi_fake_mems, nr_fake_mem, sizeof(struct efi_mem_range),
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 114) cmp_fake_mem, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 115)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 116) for (i = 0; i < nr_fake_mem; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 117) pr_info("efi_fake_mem: add attr=0x%016llx to [mem 0x%016llx-0x%016llx]",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 118) efi_fake_mems[i].attribute, efi_fake_mems[i].range.start,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 119) efi_fake_mems[i].range.end);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 120)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 121) return *p == '\0' ? 0 : -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 122) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 123)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 124) early_param("efi_fake_mem", setup_fake_mem);