OSDev.org

Code: Select all

uint64_t *find_pte(uint64_t *pt, uint64_t virt) {
  uint64_t shift = 48;
  for (int i = 0; i < 4; i++) {
    shift -= 9;
    uint64_t idx = (virt >> shift) & 0x1ff;
    uint64_t *page_table =
        (uint64_t *)((uint64_t)pt + hhdm_request.response->offset);
    if (i == 2) {
      return page_table + idx;
    }
    if (!(page_table[idx] & PRESENT)) {
      if (page_table[idx] & PAGE2MB) {
        panic("This shall not happen.");
      }
      return page_table + idx;
    } else {
      pt = (uint64_t *)(page_table[idx] & 0x000ffffffffff000);
    }
  }
  return 0;
}

Code: Select all

    if (i == 2) {
      return page_table + idx;
    }

Code: Select all

    default:
      uint64_t disalignz = entry->base % 2097152;
      entry->base = align_down(entry->base, 2097152);
      uint64_t page_amountz =
          align_up(entry->length - disalignz, 2097152) / 2097152;
      for (uint64_t j = 0; j != page_amountz; j++) {
        map2mb(ker_map.pml4, entry->base + (j * 2097152),
               hhdm_request.response->offset + entry->base + (j * 2097152),
               PRESENT | RWALLOWED);
      }
      hhdm_pages += page_amount;
      break;
    }