SCPs can sleep on events (XCC)
[akaros.git] / kern / src / syscall.c
index e135f0d..d1beec4 100644 (file)
@@ -24,7 +24,6 @@
 #include <syscall.h>
 #include <kmalloc.h>
 #include <stdio.h>
-#include <resource.h>
 #include <frontend.h>
 #include <colored_caches.h>
 #include <hashtable.h>
@@ -74,8 +73,8 @@ static void finish_sysc(struct syscall *sysc, struct proc *p)
 
 /* Helper that "finishes" the current async syscall.  This should be used when
  * we are calling a function in a syscall that might not return and won't be
- * able to use the normal syscall return path, such as proc_yield() and
- * resource_req().  Call this from within syscall.c (I don't want it global).
+ * able to use the normal syscall return path, such as proc_yield().  Call this
+ * from within syscall.c (I don't want it global).
  *
  * It is possible for another user thread to see the syscall being done early -
  * they just need to be careful with the weird proc management calls (as in,
@@ -234,8 +233,8 @@ static uint16_t sys_cgetc(struct proc *p)
        return c;
 }
 
-/* Returns the id of the cpu this syscall is executed on. */
-static uint32_t sys_getcpuid(void)
+/* Returns the id of the physical core this syscall is executed on. */
+static uint32_t sys_getpcoreid(void)
 {
        return core_id();
 }
@@ -299,6 +298,7 @@ static int sys_proc_create(struct proc *p, char *path, size_t path_l,
        return pid;
 late_error:
        proc_destroy(new_p);
+       proc_decref(new_p);     /* give up the reference created in proc_create() */
 mid_error:
        kref_put(&program->f_kref);
        return -1;
@@ -323,7 +323,7 @@ static error_t sys_proc_run(struct proc *p, unsigned pid)
                retval = -EINVAL;
        } else {
                __proc_set_state(target, PROC_RUNNABLE_S);
-               schedule_proc(target);
+               schedule_scp(target);
        }
        spin_unlock(&p->proc_lock);
        proc_decref(target);
@@ -373,8 +373,25 @@ static int sys_proc_yield(struct proc *p, bool being_nice)
        return 0;
 }
 
+static void sys_change_vcore(struct proc *p, uint32_t vcoreid,
+                             bool enable_my_notif)
+{
+       struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
+       /* Change to vcore may start the vcore up remotely before we can finish the
+        * async syscall, so we need to finish the sysc and not touch the struct.
+        * Note this sysc has no return value. */
+       finish_sysc(pcpui->cur_sysc, pcpui->cur_proc);
+       pcpui->cur_sysc = 0;    /* don't touch sysc again */
+       proc_change_to_vcore(p, vcoreid, enable_my_notif);
+       /* Shouldn't return, to prevent the chance of mucking with cur_sysc.
+        * smp_idle will make sure we run the appropriate cur_tf (which will be the
+        * new vcore for successful calls). */
+       smp_idle();
+}
+
 static ssize_t sys_fork(env_t* e)
 {
+       int8_t state = 0;
        // TODO: right now we only support fork for single-core processes
        if (e->state != PROC_RUNNING_S) {
                set_errno(EINVAL);
@@ -386,12 +403,14 @@ static ssize_t sys_fork(env_t* e)
 
        env->heap_top = e->heap_top;
        env->ppid = e->pid;
+       disable_irqsave(&state);        /* protect cur_tf */
        /* Can't really fork if we don't have a current_tf to fork */
        if (!current_tf) {
                set_errno(EINVAL);
                return -1;
        }
        env->env_tf = *current_tf;
+       enable_irqsave(&state);
 
        /* We need to speculatively say the syscall worked before copying the memory
         * out, since the 'forked' process's call never actually goes through the
@@ -404,36 +423,13 @@ static ssize_t sys_fork(env_t* e)
                if(GET_BITMASK_BIT(e->cache_colors_map,i))
                        cache_color_alloc(llc_cache, env->cache_colors_map);
 
-       duplicate_vmrs(e, env);
-
-       int copy_page(env_t* e, pte_t* pte, void* va, void* arg)
-       {
-               env_t* env = (env_t*)arg;
-
-               if(PAGE_PRESENT(*pte))
-               {
-                       page_t* pp;
-                       if(upage_alloc(env,&pp,0))
-                               return -1;
-                       if(page_insert(env->env_pgdir,pp,va,*pte & PTE_PERM))
-                       {
-                               page_decref(pp);
-                               return -1;
-                       }
-                       pagecopy(page2kva(pp),ppn2kva(PTE2PPN(*pte)));
-                       page_decref(pp);
-               } else {
-                       assert(PAGE_PAGED_OUT(*pte));
-                       /* TODO: (SWAP) will need to either make a copy or CoW/refcnt the
-                        * backend store.  For now, this PTE will be the same as the
-                        * original PTE */
-                       panic("Swapping not supported!");
-                       pte_t* newpte = pgdir_walk(env->env_pgdir,va,1);
-                       if(!newpte)
-                               return -1;
-                       *newpte = *pte;
-               }
-               return 0;
+       /* Make the new process have the same VMRs as the older.  This will copy the
+        * contents of non MAP_SHARED pages to the new VMRs. */
+       if (duplicate_vmrs(e, env)) {
+               proc_destroy(env);      /* this is prob what you want, not decref by 2 */
+               proc_decref(env);
+               set_errno(ENOMEM);
+               return -1;
        }
 
        /* In general, a forked process should be a fresh process, and we copy over
@@ -447,17 +443,10 @@ static ssize_t sys_fork(env_t* e)
        env->procdata->ldt = e->procdata->ldt;
        #endif
 
-       /* for now, just copy the contents of every present page in the entire
-        * address space. */
-       if (env_user_mem_walk(e, 0, UMAPTOP, &copy_page, env)) {
-               proc_destroy(env);      /* this is prob what you want, not decref by 2 */
-               set_errno(ENOMEM);
-               return -1;
-       }
        clone_files(&e->open_files, &env->open_files);
        __proc_ready(env);
        __proc_set_state(env, PROC_RUNNABLE_S);
-       schedule_proc(env);
+       schedule_scp(env);
 
        // don't decref the new process.
        // that will happen when the parent waits for it.
@@ -482,7 +471,7 @@ static int sys_exec(struct proc *p, char *path, size_t path_l,
        char *t_path;
        struct file *program;
        struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
-       struct trapframe *old_cur_tf = pcpui->cur_tf;
+       int8_t state = 0;
 
        /* We probably want it to never be allowed to exec if it ever was _M */
        if (p->state != PROC_RUNNING_S) {
@@ -493,21 +482,27 @@ static int sys_exec(struct proc *p, char *path, size_t path_l,
                set_errno(EINVAL);
                return -1;
        }
+       /* Copy in the path.  Consider putting an upper bound on path_l. */
+       t_path = user_strdup_errno(p, path, path_l);
+       if (!t_path)
+               return -1;
+       disable_irqsave(&state);        /* protect cur_tf */
        /* Can't exec if we don't have a current_tf to restart (if we fail).  This
         * isn't 100% true, but I'm okay with it. */
-       if (!old_cur_tf) {
+       if (!pcpui->cur_tf) {
+               enable_irqsave(&state);
                set_errno(EINVAL);
                return -1;
        }
-       /* Copy in the path.  Consider putting an upper bound on path_l. */
-       t_path = user_strdup_errno(p, path, path_l);
-       if (!t_path)
-               return -1;
+       /* Preemptively copy out the cur_tf, in case we fail later (easier on cur_tf
+        * if we do this now) */
+       p->env_tf = *pcpui->cur_tf;
        /* Clear the current_tf.  We won't be returning the 'normal' way.  Even if
         * we want to return with an error, we need to go back differently in case
         * we succeed.  This needs to be done before we could possibly block, but
         * unfortunately happens before the point of no return. */
        pcpui->cur_tf = 0;
+       enable_irqsave(&state);
        /* This could block: */
        program = do_file_open(t_path, 0, 0);
        user_memdup_free(p, t_path);
@@ -530,11 +525,8 @@ static int sys_exec(struct proc *p, char *path, size_t path_l,
        env_user_mem_free(p, 0, UMAPTOP);
        if (load_elf(p, program)) {
                kref_put(&program->f_kref);
-               /* Need an edible reference for proc_destroy in case it doesn't return.
-                * sys_exec was given current's ref (counted once just for current) */
-               proc_incref(p, 1);
+               /* Note this is an inedible reference, but proc_destroy now returns */
                proc_destroy(p);
-               proc_decref(p);
                /* We don't want to do anything else - we just need to not accidentally
                 * return to the user (hence the all_out) */
                goto all_out;
@@ -550,22 +542,22 @@ mid_error:
         * error value (errno is already set). */
        kref_put(&program->f_kref);
 early_error:
-       p->env_tf = *old_cur_tf;
        finish_current_sysc(-1);
 success:
        /* Here's how we'll restart the new (or old) process: */
        spin_lock(&p->proc_lock);
-       __unmap_vcore(p, 0);    /* VC# keep in sync with proc_run _S */
+       __unmap_vcore(p, 0);    /* VC# keep in sync with proc_run_s */
        __proc_set_state(p, PROC_RUNNABLE_S);
-       schedule_proc(p);
+       schedule_scp(p);
        spin_unlock(&p->proc_lock);
 all_out:
        /* we can't return, since we'd write retvals to the old location of the
-        * sycall struct (which has been freed and is in the old userspace) (or has
+        * syscall struct (which has been freed and is in the old userspace) (or has
         * already been written to).*/
+       disable_irq();                  /* abandon_core/clear_own wants irqs disabled */
+       clear_owning_proc(core_id());
        abandon_core();
-       smp_idle();
-       assert(0);
+       smp_idle();                             /* will reenable interrupts */
 }
 
 static ssize_t sys_trywait(env_t* e, pid_t pid, int* status)
@@ -646,19 +638,6 @@ static int sys_shared_page_free(env_t* p1, void*DANGEROUS addr, pid_t p2)
        return -1;
 }
 
-
-static int sys_resource_req(struct proc *p, int type, unsigned int amt_wanted,
-                            unsigned int amt_wanted_min, int flags)
-{
-       int retval;
-       finish_current_sysc(0);
-       /* this might not return (if it's a _S -> _M transition) */
-       proc_incref(p, 1);
-       retval = resource_req(p, type, amt_wanted, amt_wanted_min, flags);
-       proc_decref(p);
-       return retval;
-}
-
 /* Untested.  Will notify the target on the given vcore, if the caller controls
  * the target.  Will honor the target's wanted/vcoreid.  u_ne can be NULL. */
 static int sys_notify(struct proc *p, int target_pid, unsigned int ev_type,
@@ -691,7 +670,8 @@ static int sys_notify(struct proc *p, int target_pid, unsigned int ev_type,
 /* Will notify the calling process on the given vcore, independently of WANTED
  * or advertised vcoreid.  If you change the parameters, change pop_ros_tf() */
 static int sys_self_notify(struct proc *p, uint32_t vcoreid,
-                           unsigned int ev_type, struct event_msg *u_msg)
+                           unsigned int ev_type, struct event_msg *u_msg,
+                           bool priv)
 {
        struct event_msg local_msg = {0};
 
@@ -703,9 +683,11 @@ static int sys_self_notify(struct proc *p, uint32_t vcoreid,
                        set_errno(EINVAL);
                        return -1;
                }
+       } else {
+               local_msg.ev_type = ev_type;
        }
        /* this will post a message and IPI, regardless of wants/needs/debutantes.*/
-       post_vcore_event(p, &local_msg, vcoreid);
+       post_vcore_event(p, &local_msg, vcoreid, priv ? EVENT_VCORE_PRIVATE : 0);
        proc_notify(p, vcoreid);
        return 0;
 }
@@ -738,6 +720,40 @@ static int sys_halt_core(struct proc *p, unsigned int usec)
        return 0;
 }
 
+/* Changes a process into _M mode, or -EINVAL if it already is an mcp.
+ * __proc_change_to_m() returns and we'll eventually finish the sysc later.  The
+ * original context may restart on a remote core before we return and finish,
+ * but that's fine thanks to the async kernel interface. */
+static int sys_change_to_m(struct proc *p)
+{
+       int retval = 0;
+       spin_lock(&p->proc_lock);
+       if (!__proc_is_mcp(p)) {
+               /* Catch user bugs */
+               if (!p->procdata->res_req[RES_CORES].amt_wanted) {
+                       printk("[kernel] process needs to specify amt_wanted\n");
+                       p->procdata->res_req[RES_CORES].amt_wanted = 1;
+               }
+               __proc_change_to_m(p);
+               /* Tell the ksched about us */
+               register_mcp(p);
+       } else {
+               set_errno(EINVAL);
+               retval = -1;
+       }
+       spin_unlock(&p->proc_lock);
+       return retval;
+}
+
+/* Not sure what people will need.  For now, they can send in the resource they
+ * want.  Up to the ksched to support this, and other things (like -1 for all
+ * resources).  Might have this info go in via procdata instead. */
+static int sys_poke_ksched(struct proc *p, int res_type)
+{
+       poke_ksched(p, res_type);
+       return 0;
+}
+
 /************** Platform Specific Syscalls **************/
 
 //Read a buffer over the serial port
@@ -1321,13 +1337,14 @@ const static struct sys_table_entry syscall_table[] = {
        [SYS_reboot] = {(syscall_t)reboot, "reboot!"},
        [SYS_cputs] = {(syscall_t)sys_cputs, "cputs"},
        [SYS_cgetc] = {(syscall_t)sys_cgetc, "cgetc"},
-       [SYS_getcpuid] = {(syscall_t)sys_getcpuid, "getcpuid"},
+       [SYS_getpcoreid] = {(syscall_t)sys_getpcoreid, "getpcoreid"},
        [SYS_getvcoreid] = {(syscall_t)sys_getvcoreid, "getvcoreid"},
        [SYS_getpid] = {(syscall_t)sys_getpid, "getpid"},
        [SYS_proc_create] = {(syscall_t)sys_proc_create, "proc_create"},
        [SYS_proc_run] = {(syscall_t)sys_proc_run, "proc_run"},
        [SYS_proc_destroy] = {(syscall_t)sys_proc_destroy, "proc_destroy"},
        [SYS_yield] = {(syscall_t)sys_proc_yield, "proc_yield"},
+       [SYS_change_vcore] = {(syscall_t)sys_change_vcore, "change_vcore"},
        [SYS_fork] = {(syscall_t)sys_fork, "fork"},
        [SYS_exec] = {(syscall_t)sys_exec, "exec"},
        [SYS_trywait] = {(syscall_t)sys_trywait, "trywait"},
@@ -1336,7 +1353,6 @@ const static struct sys_table_entry syscall_table[] = {
        [SYS_mprotect] = {(syscall_t)sys_mprotect, "mprotect"},
        [SYS_shared_page_alloc] = {(syscall_t)sys_shared_page_alloc, "pa"},
        [SYS_shared_page_free] = {(syscall_t)sys_shared_page_free, "pf"},
-       [SYS_resource_req] = {(syscall_t)sys_resource_req, "resource_req"},
        [SYS_notify] = {(syscall_t)sys_notify, "notify"},
        [SYS_self_notify] = {(syscall_t)sys_self_notify, "self_notify"},
        [SYS_halt_core] = {(syscall_t)sys_halt_core, "halt_core"},
@@ -1353,6 +1369,8 @@ const static struct sys_table_entry syscall_table[] = {
 #ifdef __CONFIG_ARSC_SERVER__
        [SYS_init_arsc] = {(syscall_t)sys_init_arsc, "init_arsc"},
 #endif
+       [SYS_change_to_m] = {(syscall_t)sys_change_to_m, "change_to_m"},
+       [SYS_poke_ksched] = {(syscall_t)sys_poke_ksched, "poke_ksched"},
        [SYS_read] = {(syscall_t)sys_read, "read"},
        [SYS_write] = {(syscall_t)sys_write, "write"},
        [SYS_open] = {(syscall_t)sys_open, "open"},