Yield plays nice with preemption
[akaros.git] / kern / include / process.h
index cf87bc3..2fcb1dd 100644 (file)
@@ -11,6 +11,7 @@
 #define ROS_KERN_PROCESS_H
 
 #include <ros/common.h>
+#include <ros/notification.h>
 #include <trap.h>
 #include <atomic.h>
 
 #define PROC_RUNNABLE_M                        0x20
 #define PROC_RUNNING_M                 0x40
 
-#define procstate2str(state) ((state)==PROC_CREATED    ? "CREATED   " : \
+#define procstate2str(state) ((state)==PROC_CREATED    ? "CREATED"    : \
                               (state)==PROC_RUNNABLE_S ? "RUNNABLE_S" : \
-                              (state)==PROC_RUNNING_S  ? "RUNNING_S " : \
-                              (state)==PROC_WAITING    ? "WAITING   " : \
-                              (state)==PROC_DYING      ? "DYING     " : \
+                              (state)==PROC_RUNNING_S  ? "RUNNING_S : \
+                              (state)==PROC_WAITING    ? "WAITING"    : \
+                              (state)==PROC_DYING      ? "DYING"      : \
                               (state)==PROC_RUNNABLE_M ? "RUNNABLE_M" : \
-                              (state)==PROC_RUNNING_M  ? "RUNNING_M " : \
-                                                         "UNKNOWN   ")
+                              (state)==PROC_RUNNING_M  ? "RUNNING_M : \
+                                                         "UNKNOWN")
 
 #include <env.h>
 
@@ -78,10 +79,15 @@ int __proc_set_state(struct proc *p, uint32_t state) WRITES(p->state);
 struct proc *pid2proc(pid_t pid);
 bool proc_controls(struct proc *SAFE actor, struct proc *SAFE target);
 void proc_run(struct proc *SAFE p);
-void proc_startcore(struct proc *SAFE p, trapframe_t *SAFE tf)
-     __attribute__((noreturn));
+void proc_restartcore(struct proc *SAFE p, trapframe_t *SAFE tf);
 void proc_destroy(struct proc *SAFE p);
-void proc_yield(struct proc *SAFE p);
+void proc_yield(struct proc *SAFE p, bool being_nice);
+void do_notify(struct proc *p, uint32_t vcoreid, unsigned int notif,
+               struct notif_event *ne);
+void proc_notify(struct proc *p, unsigned int notif, struct notif_event *ne);
+
+/* Exposed for sys_getvcoreid(), til it's unnecessary */
+uint32_t proc_get_vcoreid(struct proc *SAFE p, uint32_t pcoreid);
 
 /* Process core management.  Only call these if you are RUNNING_M or RUNNABLE_M.
  * These all adjust the vcoremap and take appropriate actions (like __startcore
@@ -111,6 +117,9 @@ bool __proc_take_cores(struct proc *SAFE p, uint32_t *pcorelist,
 bool __proc_take_allcores(struct proc *SAFE p, amr_t message, TV(a0t) arg0,
                           TV(a1t) arg1, TV(a2t) arg2);
 void __proc_unlock_ipi_pending(struct proc *p, bool ipi_pending);
+/* Exposed for kern/src/resource.c for now */
+void __map_vcore(struct proc *p, uint32_t vcoreid, uint32_t pcoreid);
+void __unmap_vcore(struct proc *p, uint32_t vcoreid);
 
 /* Will probably have generic versions of these later. */
 void proc_incref(struct proc *SAFE p, size_t count);
@@ -123,31 +132,28 @@ void proc_decref(struct proc *SAFE p, size_t count);
 #define current per_cpu_info[core_id()].cur_proc
 #define set_current_proc(p) per_cpu_info[core_id()].cur_proc = (p)
 
-/* Allows the kernel to figure out what tf is on this core's stack.  Can be used
+/* Allows the kernel to figure out what *user* tf is on this core's stack.  Can be used
  * just like a pointer to a struct Trapframe.  Need these to be macros due to
  * some circular dependencies with smp.h.  This is done here instead of
- * elsewhere (like trap.h) for other elliptical reasons. */
+ * elsewhere (like trap.h) for other elliptical reasons.  Note the distinction
+ * between kernel and user contexts.  The kernel always returns to its nested,
+ * interrupted contexts via iret/etc.  We don't always do that for user
+ * contexts. */
 #define current_tf per_cpu_info[core_id()].cur_tf
 #define set_current_tf(tf) per_cpu_info[core_id()].cur_tf = (tf)
 
 void abandon_core(void);
 
-/* Active message handlers for process management */
-#ifdef __IVY__
-void __startcore(trapframe_t *tf, uint32_t srcid, struct proc *CT(1) a0,
-                 trapframe_t *CT(1) a1, void *SNT a2);
-void __death(trapframe_t *tf, uint32_t srcid, void *SNT a0, void *SNT a1,
-             void *SNT a2);
-#else
-void __startcore(trapframe_t *tf, uint32_t srcid, void * a0, void * a1,
-                 void * a2);
-void __death(trapframe_t *tf, uint32_t srcid, void * a0, void * a1,
-             void * a2);
-#endif
+/* Kernel message handlers for process management */
+void __startcore(trapframe_t *tf, uint32_t srcid, void *a0, void *a1, void *a2);
+void __notify(trapframe_t *tf, uint32_t srcid, void *a0, void *a1, void *a2);
+void __preempt(trapframe_t *tf, uint32_t srcid, void *a0, void *a1, void *a2);
+void __death(trapframe_t *tf, uint32_t srcid, void *a0, void *a1, void *a2);
 
 /* Arch Specific */
 void proc_init_trapframe(trapframe_t *SAFE tf, uint32_t vcoreid,
                          uint32_t entryp, uint32_t stack_top);
+void proc_secure_trapframe(struct trapframe *tf);
 void proc_set_syscall_retval(trapframe_t *SAFE tf, intreg_t value);
 void __abandon_core(void);