Rename RCU CB context to 'cannot block' context
[akaros.git] / kern / src / trap.c
1 /* Copyright (c) 2012 The Regents of the University of California
2  * Barret Rhoden <brho@cs.berkeley.edu>
3  * See LICENSE for details.
4  *
5  * Arch-independent trap handling and kernel messaging */
6
7 #include <arch/arch.h>
8 #include <smp.h>
9 #include <trap.h>
10 #include <stdio.h>
11 #include <slab.h>
12 #include <assert.h>
13 #include <kdebug.h>
14 #include <kmalloc.h>
15 #include <rcu.h>
16
17 static void print_unhandled_trap(struct proc *p, struct user_context *ctx,
18                                  unsigned int trap_nr, unsigned int err,
19                                  unsigned long aux)
20 {
21         struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
22         uint32_t vcoreid = pcpui->owning_vcoreid;
23         struct preempt_data *vcpd = &p->procdata->vcore_preempt_data[vcoreid];
24         static spinlock_t print_trap_lock = SPINLOCK_INITIALIZER;
25
26         spin_lock(&print_trap_lock);
27         if (!proc_is_vcctx_ready(p))
28                 printk("Unhandled user trap from early SCP\n");
29         else if (vcpd->notif_disabled)
30                 printk("Unhandled user trap in vcore context from VC %d\n", vcoreid);
31         print_user_ctx(ctx);
32         printk("err 0x%x (for PFs: User 4, Wr 2, Rd 1), aux %p\n", err, aux);
33         debug_addr_proc(p, get_user_ctx_pc(ctx));
34         print_vmrs(p);
35         backtrace_user_ctx(p, ctx);
36         spin_unlock(&print_trap_lock);
37 }
38
39 /* Traps that are considered normal operations. */
40 static bool benign_trap(unsigned int err)
41 {
42         return err & PF_VMR_BACKED;
43 }
44
45 static void printx_unhandled_trap(struct proc *p, struct user_context *ctx,
46                                   unsigned int trap_nr, unsigned int err,
47                                   unsigned long aux)
48 {
49         if (printx_on && !benign_trap(err))
50                 print_unhandled_trap(p, ctx, trap_nr, err, aux);
51 }
52
53 /* Helper, reflects the current context back to the 2LS.  Returns 0 on success,
54  * -1 on failure. */
55 int reflect_current_context(void)
56 {
57         uint32_t coreid = core_id();
58         struct per_cpu_info *pcpui = &per_cpu_info[coreid];
59         struct proc *p = pcpui->cur_proc;
60         uint32_t vcoreid = pcpui->owning_vcoreid;
61         struct preempt_data *vcpd = &p->procdata->vcore_preempt_data[vcoreid];
62
63         assert(pcpui->cur_proc == pcpui->owning_proc);
64         if (!proc_is_vcctx_ready(p))
65                 return -1;
66         if (vcpd->notif_disabled)
67                 return -1;
68         /* the guts of a __notify */
69         vcpd->notif_disabled = TRUE;
70         copy_current_ctx_to(&vcpd->uthread_ctx);
71         memset(pcpui->cur_ctx, 0, sizeof(struct user_context));
72         proc_init_ctx(pcpui->cur_ctx, vcoreid, vcpd->vcore_entry,
73                       vcpd->vcore_stack, vcpd->vcore_tls_desc);
74         return 0;
75 }
76
77 void reflect_unhandled_trap(unsigned int trap_nr, unsigned int err,
78                             unsigned long aux)
79 {
80         uint32_t coreid = core_id();
81         struct per_cpu_info *pcpui = &per_cpu_info[coreid];
82         struct proc *p = pcpui->cur_proc;
83
84         assert(p);
85         assert(pcpui->cur_ctx && (pcpui->cur_ctx->type == ROS_HW_CTX));
86         /* need to store trap_nr, err code, and aux into the tf so that it can get
87          * extracted on the other end, and we need to flag the TF in some way so we
88          * can tell it was reflected.  for example, on a PF, we need some number (14
89          * on x86), the prot violation (write, read, etc), and the virt addr (aux).
90          * parlib will know how to extract this info. */
91         __arch_reflect_trap_hwtf(&pcpui->cur_ctx->tf.hw_tf, trap_nr, err, aux);
92         printx_unhandled_trap(p, pcpui->cur_ctx, trap_nr, err, aux);
93         if (reflect_current_context()) {
94                 print_unhandled_trap(p, pcpui->cur_ctx, trap_nr, err, aux);
95                 proc_destroy(p);
96         }
97 }
98
99 /* Helper, copies the current context to to_ctx. */
100 void copy_current_ctx_to(struct user_context *to_ctx)
101 {
102         struct user_context *cur_ctx = current_ctx;
103
104         /* Be sure to finalize into cur_ctx, not the to_ctx.  o/w the arch could get
105          * confused by other calls to finalize. */
106         arch_finalize_ctx(cur_ctx);
107         *to_ctx = *cur_ctx;
108 }
109
110 struct kmem_cache *kernel_msg_cache;
111
112 void kernel_msg_init(void)
113 {
114         kernel_msg_cache = kmem_cache_create("kernel_msgs",
115                                              sizeof(struct kernel_message),
116                                              ARCH_CL_SIZE, 0, NULL, 0, 0, NULL);
117 }
118
119 uint32_t send_kernel_message(uint32_t dst, amr_t pc, long arg0, long arg1,
120                              long arg2, int type)
121 {
122         kernel_message_t *k_msg;
123         assert(pc);
124         // note this will be freed on the destination core
125         k_msg = kmem_cache_alloc(kernel_msg_cache, 0);
126         k_msg->srcid = core_id();
127         k_msg->dstid = dst;
128         k_msg->pc = pc;
129         k_msg->arg0 = arg0;
130         k_msg->arg1 = arg1;
131         k_msg->arg2 = arg2;
132         switch (type) {
133                 case KMSG_IMMEDIATE:
134                         spin_lock_irqsave(&per_cpu_info[dst].immed_amsg_lock);
135                         STAILQ_INSERT_TAIL(&per_cpu_info[dst].immed_amsgs, k_msg, link);
136                         spin_unlock_irqsave(&per_cpu_info[dst].immed_amsg_lock);
137                         break;
138                 case KMSG_ROUTINE:
139                         spin_lock_irqsave(&per_cpu_info[dst].routine_amsg_lock);
140                         STAILQ_INSERT_TAIL(&per_cpu_info[dst].routine_amsgs, k_msg, link);
141                         spin_unlock_irqsave(&per_cpu_info[dst].routine_amsg_lock);
142                         break;
143                 default:
144                         panic("Unknown type of kernel message!");
145         }
146         /* since we touched memory the other core will touch (the lock), we don't
147          * need an wmb_f() */
148         /* if we're sending a routine message locally, we don't want/need an IPI */
149         if ((dst != k_msg->srcid) || (type == KMSG_IMMEDIATE))
150                 send_ipi(dst, I_KERNEL_MSG);
151         return 0;
152 }
153
154 /* Kernel message IPI/IRQ handler.
155  *
156  * This processes immediate messages, and that's it (it used to handle routines
157  * too, if it came in from userspace).  Routine messages will get processed when
158  * the kernel has a chance (right before popping to userspace or in smp_idle
159  * before halting).
160  *
161  * Note that all of this happens from interrupt context, and interrupts are
162  * disabled. */
163 void handle_kmsg_ipi(struct hw_trapframe *hw_tf, void *data)
164 {
165         struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
166         struct kernel_message *kmsg_i, *temp;
167         /* Avoid locking if the list appears empty (lockless peek is okay) */
168         if (STAILQ_EMPTY(&pcpui->immed_amsgs))
169                 return;
170         /* The lock serves as a cmb to force a re-read of the head of the list */
171         spin_lock_irqsave(&pcpui->immed_amsg_lock);
172         STAILQ_FOREACH_SAFE(kmsg_i, &pcpui->immed_amsgs, link, temp) {
173                 pcpui_trace_kmsg(pcpui, (uintptr_t)kmsg_i->pc);
174                 kmsg_i->pc(kmsg_i->srcid, kmsg_i->arg0, kmsg_i->arg1, kmsg_i->arg2);
175                 STAILQ_REMOVE(&pcpui->immed_amsgs, kmsg_i, kernel_message, link);
176                 kmem_cache_free(kernel_msg_cache, (void*)kmsg_i);
177         }
178         spin_unlock_irqsave(&pcpui->immed_amsg_lock);
179 }
180
181 bool has_routine_kmsg(void)
182 {
183         struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
184         /* lockless peek */
185         return !STAILQ_EMPTY(&pcpui->routine_amsgs);
186 }
187
188 /* Helper function, gets the next routine KMSG (RKM).  Returns 0 if the list was
189  * empty. */
190 static kernel_message_t *get_next_rkmsg(struct per_cpu_info *pcpui)
191 {
192         struct kernel_message *kmsg;
193         /* Avoid locking if the list appears empty (lockless peek is okay) */
194         if (STAILQ_EMPTY(&pcpui->routine_amsgs))
195                 return 0;
196         /* The lock serves as a cmb to force a re-read of the head of the list.
197          * IRQs are disabled by our caller. */
198         spin_lock(&pcpui->routine_amsg_lock);
199         kmsg = STAILQ_FIRST(&pcpui->routine_amsgs);
200         if (kmsg)
201                 STAILQ_REMOVE_HEAD(&pcpui->routine_amsgs, link);
202         spin_unlock(&pcpui->routine_amsg_lock);
203         return kmsg;
204 }
205
206 /* Runs a routine kernel message.  If we execute a message, this does not
207  * return, but instead will call smp_idle().
208  *
209  * Note that routine messages do not have to return, but almost all of them do.
210  * If you're thinking of changing this, take a look at __launch_kthread. */
211 void process_routine_kmsg(void)
212 {
213         uint32_t pcoreid = core_id();
214         struct per_cpu_info *pcpui = &per_cpu_info[pcoreid];
215         struct kernel_message msg_cp, *kmsg;
216
217         /* Important that callers have IRQs disabled when checking for RKMs.  When
218          * sending cross-core RKMs, the IPI is used to keep the core from going to
219          * sleep - even though RKMs aren't handled in the kmsg handler. */
220         assert(!irq_is_enabled());
221         kmsg = get_next_rkmsg(pcpui);
222         if (!kmsg)
223                 return;
224         msg_cp = *kmsg;
225         kmem_cache_free(kernel_msg_cache, kmsg);
226         assert(msg_cp.dstid == pcoreid);
227         /* The kmsg could block.  If it does, we want the kthread code to know it's
228          * not running on behalf of a process, and we're actually spawning a kernel
229          * task.  While we do have a syscall that does work in an RKM (change_to),
230          * it's not really the rest of the syscall context.  When we return or
231          * otherwise call smp_idle, smp_idle will reset these flags. */
232         pcpui->cur_kthread->flags = KTH_KTASK_FLAGS;
233         pcpui_trace_kmsg(pcpui, (uintptr_t)msg_cp.pc);
234         msg_cp.pc(msg_cp.srcid, msg_cp.arg0, msg_cp.arg1, msg_cp.arg2);
235         smp_idle();
236 }
237
238 /* extremely dangerous and racy: prints out the immed and routine kmsgs for a
239  * specific core (so possibly remotely) */
240 void print_kmsgs(uint32_t coreid)
241 {
242         struct per_cpu_info *pcpui = &per_cpu_info[coreid];
243         void __print_kmsgs(struct kernel_msg_list *list, char *type)
244         {
245                 struct kernel_message *kmsg_i;
246
247                 STAILQ_FOREACH(kmsg_i, list, link) {
248                         printk("%s KMSG on %d from %d to run %p(%s)(%p, %p, %p)\n", type,
249                                kmsg_i->dstid, kmsg_i->srcid, kmsg_i->pc,
250                                get_fn_name((long)kmsg_i->pc),
251                                kmsg_i->arg0, kmsg_i->arg1, kmsg_i->arg2);
252                 }
253         }
254         __print_kmsgs(&pcpui->immed_amsgs, "Immedte");
255         __print_kmsgs(&pcpui->routine_amsgs, "Routine");
256 }
257
258 void __kmsg_trampoline(uint32_t srcid, long a0, long a1, long a2)
259 {
260         ((void (*)(long arg0, long arg1))a0)(a1, a2);
261 }
262
263 /* Debugging stuff */
264 void kmsg_queue_stat(void)
265 {
266         struct kernel_message *kmsg;
267         bool immed_emp, routine_emp;
268         for (int i = 0; i < num_cores; i++) {
269                 spin_lock_irqsave(&per_cpu_info[i].immed_amsg_lock);
270                 immed_emp = STAILQ_EMPTY(&per_cpu_info[i].immed_amsgs);
271                 spin_unlock_irqsave(&per_cpu_info[i].immed_amsg_lock);
272                 spin_lock_irqsave(&per_cpu_info[i].routine_amsg_lock);
273                 routine_emp = STAILQ_EMPTY(&per_cpu_info[i].routine_amsgs);
274                 spin_unlock_irqsave(&per_cpu_info[i].routine_amsg_lock);
275                 printk("Core %d's immed_emp: %d, routine_emp %d\n", i, immed_emp,
276                routine_emp);
277                 if (!immed_emp) {
278                         kmsg = STAILQ_FIRST(&per_cpu_info[i].immed_amsgs);
279                         printk("Immed msg on core %d:\n", i);
280                         printk("\tsrc:  %d\n", kmsg->srcid);
281                         printk("\tdst:  %d\n", kmsg->dstid);
282                         printk("\tpc:   %p\n", kmsg->pc);
283                         printk("\targ0: %p\n", kmsg->arg0);
284                         printk("\targ1: %p\n", kmsg->arg1);
285                         printk("\targ2: %p\n", kmsg->arg2);
286                 }
287                 if (!routine_emp) {
288                         kmsg = STAILQ_FIRST(&per_cpu_info[i].routine_amsgs);
289                         printk("Routine msg on core %d:\n", i);
290                         printk("\tsrc:  %d\n", kmsg->srcid);
291                         printk("\tdst:  %d\n", kmsg->dstid);
292                         printk("\tpc:   %p\n", kmsg->pc);
293                         printk("\targ0: %p\n", kmsg->arg0);
294                         printk("\targ1: %p\n", kmsg->arg1);
295                         printk("\targ2: %p\n", kmsg->arg2);
296                 }
297
298         }
299 }
300
301 void print_kctx_depths(const char *str)
302 {
303         uint32_t coreid = core_id();
304         struct per_cpu_info *pcpui = &per_cpu_info[coreid];
305
306         if (!str)
307                 str = "(none)";
308         printk("%s: Core %d, irq depth %d, ktrap depth %d, irqon %d\n", str, coreid,
309                irq_depth(pcpui), ktrap_depth(pcpui), irq_is_enabled());
310 }
311
312 void print_user_ctx(struct user_context *ctx)
313 {
314         switch (ctx->type) {
315         case ROS_HW_CTX:
316                 print_trapframe(&ctx->tf.hw_tf);
317                 break;
318         case ROS_SW_CTX:
319                 print_swtrapframe(&ctx->tf.sw_tf);
320                 break;
321         case ROS_VM_CTX:
322                 print_vmtrapframe(&ctx->tf.vm_tf);
323                 break;
324         default:
325                 printk("Bad TF %p type %d!\n", ctx, ctx->type);
326         }
327 }