book3s_32.c 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. // SPDX-License-Identifier: GPL-2.0
  2. #define DISABLE_BRANCH_PROFILING
  3. #include <linux/kasan.h>
  4. #include <linux/memblock.h>
  5. #include <mm/mmu_decl.h>
  6. int __init kasan_init_region(void *start, size_t size)
  7. {
  8. unsigned long k_start = (unsigned long)kasan_mem_to_shadow(start);
  9. unsigned long k_end = (unsigned long)kasan_mem_to_shadow(start + size);
  10. unsigned long k_nobat = k_start;
  11. unsigned long k_cur;
  12. phys_addr_t phys;
  13. int ret;
  14. while (k_nobat < k_end) {
  15. unsigned int k_size = bat_block_size(k_nobat, k_end);
  16. int idx = find_free_bat();
  17. if (idx == -1)
  18. break;
  19. if (k_size < SZ_128K)
  20. break;
  21. phys = memblock_phys_alloc_range(k_size, k_size, 0,
  22. MEMBLOCK_ALLOC_ANYWHERE);
  23. if (!phys)
  24. break;
  25. setbat(idx, k_nobat, phys, k_size, PAGE_KERNEL);
  26. k_nobat += k_size;
  27. }
  28. if (k_nobat != k_start)
  29. update_bats();
  30. if (k_nobat < k_end) {
  31. phys = memblock_phys_alloc_range(k_end - k_nobat, PAGE_SIZE, 0,
  32. MEMBLOCK_ALLOC_ANYWHERE);
  33. if (!phys)
  34. return -ENOMEM;
  35. }
  36. ret = kasan_init_shadow_page_tables(k_start, k_end);
  37. if (ret)
  38. return ret;
  39. kasan_update_early_region(k_start, k_nobat, __pte(0));
  40. for (k_cur = k_nobat; k_cur < k_end; k_cur += PAGE_SIZE) {
  41. pmd_t *pmd = pmd_off_k(k_cur);
  42. pte_t pte = pfn_pte(PHYS_PFN(phys + k_cur - k_nobat), PAGE_KERNEL);
  43. __set_pte_at(&init_mm, k_cur, pte_offset_kernel(pmd, k_cur), pte, 0);
  44. }
  45. flush_tlb_kernel_range(k_start, k_end);
  46. memset(kasan_mem_to_shadow(start), 0, k_end - k_start);
  47. return 0;
  48. }