proc_run() now returns
[akaros.git] / kern / src / syscall.c
index 0d785ac..ef1ec78 100644 (file)
@@ -9,7 +9,7 @@
 #include <arch/arch.h>
 #include <arch/mmu.h>
 #include <arch/console.h>
-#include <ros/timer.h>
+#include <ros/time.h>
 #include <error.h>
 
 #include <elf.h>
@@ -234,8 +234,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 +299,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;
@@ -373,8 +374,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 +404,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,52 +424,26 @@ 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;
-       }
-
-       // TODO: (PC) this won't work.  Needs revisiting.
-       // copy procdata and procinfo
-       memcpy(env->procdata,e->procdata,sizeof(struct procdata));
-       memcpy(env->procinfo,e->procinfo,sizeof(struct procinfo));
-       env->procinfo->pid = env->pid;
-       env->procinfo->ppid = env->ppid;
-
-       /* 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)) {
+       /* 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
+        * whatever stuff is needed between procinfo/procdata. */
+       /* Copy over the procinfo argument stuff in case they don't exec */
+       memcpy(env->procinfo->argp, e->procinfo->argp, sizeof(e->procinfo->argp));
+       memcpy(env->procinfo->argbuf, e->procinfo->argbuf,
+              sizeof(e->procinfo->argbuf));
+       #ifdef __i386__
+       /* new guy needs to know about ldt (everything else in procdata is fresh */
+       env->procdata->ldt = e->procdata->ldt;
+       #endif
+
        clone_files(&e->open_files, &env->open_files);
        __proc_ready(env);
        __proc_set_state(env, PROC_RUNNABLE_S);
@@ -478,7 +472,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) {
@@ -489,21 +483,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);
@@ -517,21 +517,17 @@ static int sys_exec(struct proc *p, char *path, size_t path_l,
                                   sizeof(pi->argbuf)))
                goto mid_error;
        /* This is the point of no return for the process. */
-       /* TODO: issues with this: Need to also assert there are no outstanding
-        * users of the sysrings.  the ldt page will get freed shortly, so that's
-        * okay.  Potentially issues with the nm and vcpd if we were in _M before
-        * and someone is trying to notify. */
-       memset(p->procdata, 0, sizeof(procdata_t));
+       #ifdef __i386__
+       /* clear this, so the new program knows to get an LDT */
+       p->procdata->ldt = 0;
+       #endif
        destroy_vmrs(p);
        close_all_files(&p->open_files, TRUE);
        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;
@@ -547,21 +543,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 */
        __proc_set_state(p, PROC_RUNNABLE_S);
        schedule_proc(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,13 +643,10 @@ static int sys_shared_page_free(env_t* p1, void*DANGEROUS addr, pid_t p2)
 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;
+       /* resource_req 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. */
+       return resource_req(p, type, amt_wanted, amt_wanted_min, flags);
 }
 
 /* Untested.  Will notify the target on the given vcore, if the caller controls
@@ -687,7 +681,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};
 
@@ -699,9 +694,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;
 }
@@ -1264,6 +1261,7 @@ intreg_t sys_gettimeofday(struct proc *p, int *buf)
        spin_unlock(&gtod_lock);
 
        long long dt = read_tsc();
+       /* TODO: This probably wants its own function, using a struct timeval */
        int kbuf[2] = {t0+dt/system_timing.tsc_freq,
            (dt%system_timing.tsc_freq)*1000000/system_timing.tsc_freq};
 
@@ -1316,13 +1314,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"},
@@ -1472,7 +1471,7 @@ void __signal_syscall(struct syscall *sysc, struct proc *p)
        struct event_msg local_msg;
        /* User sets the ev_q then atomically sets the flag (races with SC_DONE) */
        if (atomic_read(&sysc->flags) & SC_UEVENT) {
-               rmb();
+               rmb();  /* read the ev_q after reading the flag */
                ev_q = sysc->ev_q;
                if (ev_q) {
                        memset(&local_msg, 0, sizeof(struct event_msg));