lunaix-os/kernel/asm/x86/pfault.c

   1 #include <arch/x86/interrupts.h>
   2 #include <lunaix/common.h>
   3 #include <lunaix/lxsignal.h>
   4 #include <lunaix/mm/mm.h>
   5 #include <lunaix/mm/pmm.h>
   6 #include <lunaix/mm/region.h>
   7 #include <lunaix/mm/vmm.h>
   8 #include <lunaix/sched.h>
   9 #include <lunaix/status.h>
  10 #include <lunaix/syslog.h>
  11
  12 static void
  13 kprintf(const char* fmt, ...)
  14 {
  15     va_list args;
  16     va_start(args, fmt);
  17     __kprintf("PFAULT", fmt, args);
  18     va_end(args);
  19 }
  20
  21 #define COW_MASK (REGION_RSHARED | REGION_READ | REGION_WRITE)
  22
  23 extern void
  24 __print_panic_msg(const char* msg, const isr_param* param);
  25
  26 void
  27 intr_routine_page_fault(const isr_param* param)
  28 {
  29     uintptr_t ptr = cpu_rcr2();
  30     if (!ptr) {
  31         goto segv_term;
  32     }
  33
  34     v_mapping mapping;
  35     if (!vmm_lookup(ptr, &mapping)) {
  36         goto segv_term;
  37     }
  38
  39     if (!SEL_RPL(param->cs)) {
  40         // 如果是内核页错误……
  41         if (do_kernel(&mapping)) {
  42             return;
  43         }
  44         goto segv_term;
  45     }
  46
  47     struct mm_region* hit_region = region_get(&__current->mm.regions, ptr);
  48
  49     if (!hit_region) {
  50         // Into the void...
  51         goto segv_term;
  52     }
  53
  54     x86_pte_t* pte = &PTE_MOUNTED(PD_REFERENCED, ptr >> 12);
  55     if ((*pte & PG_PRESENT)) {
  56         if ((hit_region->attr & COW_MASK) == COW_MASK) {
  57             // normal page fault, do COW
  58             cpu_invplg(pte);
  59             uintptr_t pa =
  60               (uintptr_t)vmm_dup_page(__current->pid, PG_ENTRY_ADDR(*pte));
  61             pmm_free_page(__current->pid, *pte & ~0xFFF);
  62             *pte = (*pte & 0xFFF) | pa | PG_WRITE;
  63             goto resolved;
  64         }
  65         // impossible cases or accessing privileged page
  66         goto segv_term;
  67     }
  68
  69     if (!(*pte)) {
  70         // Invalid location
  71         goto segv_term;
  72     }
  73
  74     uintptr_t loc = *pte & ~0xfff;
  75
  76     // a writable page, not present, not cached, pte attr is not null
  77     //   -> a new page need to be alloc
  78     if ((hit_region->attr & REGION_WRITE) && (*pte & 0xfff) && !loc) {
  79         cpu_invplg(pte);
  80         uintptr_t pa = pmm_alloc_page(__current->pid, 0);
  81         *pte = *pte | pa | PG_PRESENT;
  82         goto resolved;
  83     }
  84
  85     // page not present, bring it from disk or somewhere else
  86     __print_panic_msg("WIP page fault route", param);
  87     while (1)
  88         ;
  89
  90 segv_term:
  91     kprintf(KERROR "(pid: %d) Segmentation fault on %p (%p:%p)\n",
  92             __current->pid,
  93             ptr,
  94             param->cs,
  95             param->eip);
  96     __SIGSET(__current->sig_pending, _SIGSEGV);
  97     schedule();
  98     // should not reach
  99     while (1)
 100         ;
 101
 102 resolved:
 103     cpu_invplg(ptr);
 104     return;
 105 }
 106
 107 int
 108 do_kernel(v_mapping* mapping)
 109 {
 110     uintptr_t addr = mapping->va;
 111     if (addr >= KHEAP_START && addr < PROC_START) {
 112         // This is kernel heap page
 113         uintptr_t pa = pmm_alloc_page(KERNEL_PID, 0);
 114         *mapping->pte = (*mapping->pte & 0xfff) | pa | PG_PRESENT;
 115         cpu_invplg(mapping->pte);
 116         cpu_invplg(addr);
 117         goto done;
 118     }
 119
 120     return 0;
 121 done:
 122     return 1;
 123 }