Bring a bit more apic infrastructure up.
[akaros.git] / kern / arch / riscv / trap.c
1 #include <arch/arch.h>
2 #include <assert.h>
3 #include <trap.h>
4 #include <arch/console.h>
5 #include <console.h>
6 #include <string.h>
7 #include <process.h>
8 #include <syscall.h>
9 #include <monitor.h>
10 #include <manager.h>
11 #include <stdio.h>
12 #include <smp.h>
13 #include <slab.h>
14 #include <mm.h>
15 #include <umem.h>
16 #include <pmap.h>
17
18 /* These are the stacks the kernel will load when it receives a trap from user
19  * space.  The deal is that they get set right away in entry.S, and can always
20  * be used for finding the top of the stack (from which you should subtract the
21  * sizeof the trapframe.  Note, we need to have a junk value in the array so
22  * that this is NOT part of the BSS.  If it is in the BSS, it will get 0'd in
23  * kernel_init(), which is after these values get set.
24  *
25  * TODO: if these end up becoming contended cache lines, move this to
26  * per_cpu_info. */
27 uintptr_t core_stacktops[MAX_NUM_CPUS] = {0xcafebabe, 0};
28
29 void
30 advance_pc(struct hw_trapframe *state)
31 {
32         state->epc += 4;
33 }
34
35 /* Set stacktop for the current core to be the stack the kernel will start on
36  * when trapping/interrupting from userspace */
37 void set_stack_top(uintptr_t stacktop)
38 {
39         core_stacktops[core_id()] = stacktop;
40 }
41
42 /* Note the assertion assumes we are in the top page of the stack. */
43 uintptr_t get_stack_top(void)
44 {
45         register uintptr_t sp asm ("sp");
46         uintptr_t stacktop = core_stacktops[core_id()];
47         assert(ROUNDUP(sp, PGSIZE) == stacktop);
48         return stacktop;
49 }
50
51 void
52 idt_init(void)
53 {
54 }
55
56 /* Helper.  For now, this copies out the TF to pcpui, and sets cur_ctx to point
57  * to it. */
58 static void set_current_ctx_hw(struct per_cpu_info *pcpui,
59                                struct hw_trapframe *hw_tf)
60 {
61         if (irq_is_enabled())
62                 warn("Turn off IRQs until cur_ctx is set!");
63         assert(!pcpui->cur_ctx);
64         pcpui->actual_ctx.type = ROS_HW_CTX;
65         pcpui->actual_ctx.tf.hw_tf = *hw_tf;
66         pcpui->cur_ctx = &pcpui->actual_ctx;
67 }
68
69 static void set_current_ctx_sw(struct per_cpu_info *pcpui,
70                                struct sw_trapframe *sw_tf)
71 {
72         if (irq_is_enabled())
73                 warn("Turn off IRQs until cur_ctx is set!");
74         assert(!pcpui->cur_ctx);
75         pcpui->actual_ctx.type = ROS_SW_CTX;
76         pcpui->actual_ctx.tf.sw_tf = *sw_tf;
77         pcpui->cur_ctx = &pcpui->actual_ctx;
78 }
79
80 static int
81 format_trapframe(struct hw_trapframe *hw_tf, char* buf, int bufsz)
82 {
83         // slightly hackish way to read out the instruction that faulted.
84         // not guaranteed to be right 100% of the time
85         uint32_t insn;
86         if(!(current && !memcpy_from_user(current,&insn,(void*)hw_tf->epc,4)))
87                 insn = -1;
88
89         int len = snprintf(buf,bufsz,"TRAP frame at %p on core %d\n",
90                            hw_tf, core_id());
91         static const char* regnames[] = {
92           "z ", "ra", "s0", "s1", "s2", "s3", "s4", "s5",
93           "s6", "s7", "s8", "s9", "sA", "sB", "sp", "tp",
94           "v0", "v1", "a0", "a1", "a2", "a3", "a4", "a5",
95           "a6", "a7", "a8", "a9", "aA", "aB", "aC", "aD"
96         };
97         
98         hw_tf->gpr[0] = 0;
99         
100         for(int i = 0; i < 32; i+=4)
101         {
102                 for(int j = 0; j < 4; j++)
103                         len += snprintf(buf+len, bufsz-len,
104                                         "%s %016lx%c", regnames[i+j], hw_tf->gpr[i+j], 
105                                         j < 3 ? ' ' : '\n');
106         }
107         len += snprintf(buf+len, bufsz-len,
108                         "sr %016lx pc %016lx va %016lx insn       %08x\n",
109                                         hw_tf->sr, hw_tf->epc, hw_tf->badvaddr, insn);
110
111         buf[bufsz-1] = 0;
112         return len;
113 }
114
115 void
116 print_trapframe(struct hw_trapframe *hw_tf)
117 {
118         char buf[1024];
119         int len = format_trapframe(hw_tf, buf, sizeof(buf));
120         cputbuf(buf,len);
121 }
122
123 static void exit_halt_loop(struct hw_trapframe *hw_tf)
124 {
125         extern char after_cpu_halt;
126         if ((char*)hw_tf->epc >= (char*)&cpu_halt &&
127             (char*)hw_tf->epc < &after_cpu_halt)
128                 hw_tf->epc = hw_tf->gpr[GPR_RA];
129 }
130
131 static void handle_keypress(char c)
132 {
133         amr_t handler = c == 'G' ? __run_mon : __cons_add_char;
134         send_kernel_message(core_id(), handler, (long)&cons_buf, (long)c, 0,
135                             KMSG_ROUTINE);
136         cons_init();
137 }
138
139 static void handle_host_interrupt(struct hw_trapframe *hw_tf)
140 {
141         uintptr_t fh = mtpcr(PCR_FROMHOST, 0);
142         switch (fh >> 56)
143         {
144           case 0x00: return;
145           case 0x01: handle_keypress(fh); return;
146           default: assert(0);
147         }
148 }
149
150 static void handle_timer_interrupt(struct hw_trapframe *hw_tf)
151 {
152         timer_interrupt(hw_tf, NULL);
153 }
154
155 /* Assumes that any IPI you get is really a kernel message */
156 static void handle_interprocessor_interrupt(struct hw_trapframe *hw_tf)
157 {
158         clear_ipi();
159         handle_kmsg_ipi(hw_tf, 0);
160 }
161
162 static void
163 unhandled_trap(struct hw_trapframe *state, const char* name)
164 {
165         static spinlock_t screwup_lock = SPINLOCK_INITIALIZER;
166         spin_lock(&screwup_lock);
167
168         if(in_kernel(state))
169         {
170                 print_trapframe(state);
171                 panic("Unhandled trap in kernel!\nTrap type: %s", name);
172         }
173         else
174         {
175                 char tf_buf[1024];
176                 format_trapframe(state, tf_buf, sizeof(tf_buf));
177
178                 warn("Unhandled trap in user!\nTrap type: %s\n%s", name, tf_buf);
179                 backtrace();
180                 spin_unlock(&screwup_lock);
181
182                 assert(current);
183                 enable_irq();
184                 proc_destroy(current);
185         }
186 }
187
188 static void
189 handle_misaligned_fetch(struct hw_trapframe *state)
190 {
191         unhandled_trap(state, "Misaligned Fetch");
192 }
193
194 static void
195 handle_misaligned_load(struct hw_trapframe *state)
196 {
197         unhandled_trap(state, "Misaligned Load");
198 }
199
200 static void
201 handle_misaligned_store(struct hw_trapframe *state)
202 {
203         unhandled_trap(state, "Misaligned Store");
204 }
205
206 static void
207 handle_fault_fetch(struct hw_trapframe *state)
208 {
209         if(in_kernel(state))
210         {
211                 print_trapframe(state);
212                 panic("Instruction Page Fault in the Kernel at %p!", state->epc);
213         }
214
215         set_current_ctx_hw(&per_cpu_info[core_id()], state);
216
217 #warning "returns EAGAIN if you should reflect the fault"
218         if(handle_page_fault(current, state->epc, PROT_EXEC))
219                 unhandled_trap(state, "Instruction Page Fault");
220 }
221
222 static void
223 handle_fault_load(struct hw_trapframe *state)
224 {
225         if(in_kernel(state))
226         {
227                 print_trapframe(state);
228                 panic("Load Page Fault in the Kernel at %p!", state->badvaddr);
229         }
230
231         set_current_ctx_hw(&per_cpu_info[core_id()], state);
232
233 #warning "returns EAGAIN if you should reflect the fault"
234         if(handle_page_fault(current, state->badvaddr, PROT_READ))
235                 unhandled_trap(state, "Load Page Fault");
236 }
237
238 static void
239 handle_fault_store(struct hw_trapframe *state)
240 {
241         if(in_kernel(state))
242         {
243                 print_trapframe(state);
244                 panic("Store Page Fault in the Kernel at %p!", state->badvaddr);
245         }
246
247         set_current_ctx_hw(&per_cpu_info[core_id()], state);
248
249         if(handle_page_fault(current, state->badvaddr, PROT_WRITE))
250                 unhandled_trap(state, "Store Page Fault");
251 }
252
253 static void
254 handle_illegal_instruction(struct hw_trapframe *state)
255 {
256         assert(!in_kernel(state));
257
258         struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
259         set_current_ctx_hw(pcpui, state);
260         if (emulate_fpu(state) == 0)
261         {
262                 advance_pc(&pcpui->cur_ctx->tf.hw_tf);
263                 return;
264         }
265
266         unhandled_trap(state, "Illegal Instruction");
267 }
268
269 static void
270 handle_syscall(struct hw_trapframe *state)
271 {
272         uintptr_t a0 = state->gpr[GPR_A0];
273         uintptr_t a1 = state->gpr[GPR_A1];
274
275         advance_pc(state);
276         set_current_ctx_hw(&per_cpu_info[core_id()], state);
277         enable_irq();
278         prep_syscalls(current, (struct syscall*)a0, a1);
279 }
280
281 static void
282 handle_breakpoint(struct hw_trapframe *state)
283 {
284         advance_pc(state);
285         monitor(state);
286 }
287
288 void
289 handle_trap(struct hw_trapframe *hw_tf)
290 {
291         static void (*const trap_handlers[])(struct hw_trapframe *) = {
292           [CAUSE_MISALIGNED_FETCH] = handle_misaligned_fetch,
293           [CAUSE_FAULT_FETCH] = handle_fault_fetch,
294           [CAUSE_ILLEGAL_INSTRUCTION] = handle_illegal_instruction,
295           [CAUSE_PRIVILEGED_INSTRUCTION] = handle_illegal_instruction,
296           [CAUSE_SYSCALL] = handle_syscall,
297           [CAUSE_BREAKPOINT] = handle_breakpoint,
298           [CAUSE_MISALIGNED_LOAD] = handle_misaligned_load,
299           [CAUSE_MISALIGNED_STORE] = handle_misaligned_store,
300           [CAUSE_FAULT_LOAD] = handle_fault_load,
301           [CAUSE_FAULT_STORE] = handle_fault_store,
302         };
303
304         static void (*const irq_handlers[])(struct hw_trapframe *) = {
305           [IRQ_TIMER] = handle_timer_interrupt,
306           [IRQ_HOST] = handle_host_interrupt,
307           [IRQ_IPI] = handle_interprocessor_interrupt,
308         };
309         
310         struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
311         if (hw_tf->cause < 0)
312         {
313                 uint8_t irq = hw_tf->cause;
314                 assert(irq < sizeof(irq_handlers)/sizeof(irq_handlers[0]) &&
315                        irq_handlers[irq]);
316
317                 if (in_kernel(hw_tf))
318                         exit_halt_loop(hw_tf);
319                 else
320                         set_current_ctx_hw(&per_cpu_info[core_id()], hw_tf);
321
322                 inc_irq_depth(pcpui);
323                 irq_handlers[irq](hw_tf);
324                 dec_irq_depth(pcpui);
325         }
326         else
327         {
328                 assert(hw_tf->cause < sizeof(trap_handlers)/sizeof(trap_handlers[0]) &&
329                        trap_handlers[hw_tf->cause]);
330                 if (in_kernel(hw_tf)) {
331                         inc_ktrap_depth(pcpui);
332                         trap_handlers[hw_tf->cause](hw_tf);
333                         dec_ktrap_depth(pcpui);
334                 } else {
335                         trap_handlers[hw_tf->cause](hw_tf);
336                 }
337                 #warning "if a trap wasn't handled fully, like an MCP pf, reflect it
338                 reflect_unhandled_trap(hw_tf->tf_trapno, hw_tf->tf_err, aux);
339         }
340         
341         extern void pop_hw_tf(struct hw_trapframe *tf); /* in asm */
342         /* Return to the current process, which should be runnable.  If we're the
343          * kernel, we should just return naturally.  Note that current and tf need
344          * to still be okay (might not be after blocking) */
345         if (in_kernel(hw_tf))
346                 pop_hw_tf(hw_tf);
347         else
348                 proc_restartcore();
349 }
350
351 /* We don't have NMIs now. */
352 void send_nmi(uint32_t os_coreid)
353 {
354         printk("%s not implemented\n", __FUNCTION);
355 }
356
357 void register_raw_irq(unsigned int vector, isr_t handler, void *data)
358 {
359         printk("%s not implemented\n", __FUNCTION);
360 }
361
362 void unregister_raw_irq(unsigned int vector, isr_t handler, void *data)
363 {
364         printk("%s not implemented\n", __FUNCTION);
365 }
366
367 int register_dev_irq(int irq, void (*handler)(struct hw_trapframe *, void *),
368                      void *irq_arg)
369 {
370         printk("%s not implemented\n", __FUNCTION);
371         return -1;
372 }
373
374 void __arch_reflect_trap_hwtf(struct hw_trapframe *hw_tf, unsigned int trap_nr,
375                               unsigned int err, unsigned long aux)
376 {
377         printk("%s not implemented\n", __FUNCTION);
378 }