File opening revisited
[akaros.git] / kern / src / syscall.c
index 457ec0b..19cb653 100644 (file)
@@ -19,6 +19,7 @@
 #include <process.h>
 #include <schedule.h>
 #include <pmap.h>
+#include <umem.h>
 #include <mm.h>
 #include <trap.h>
 #include <syscall.h>
@@ -193,38 +194,48 @@ static pid_t sys_getpid(struct proc *p)
        return p->pid;
 }
 
-/*
- * Creates a process found at the user string 'path'.  Currently uses KFS.
- * Not runnable by default, so it needs it's status to be changed so that the
- * next call to schedule() will try to run it.
- * TODO: once we have a decent VFS, consider splitting this up
- * and once there's an mmap, can have most of this in process.c
- */
-static int sys_proc_create(struct proc *p, const char *DANGEROUS path)
+/* Creates a process from the file 'path'.  The process is not runnable by
+ * default, so it needs it's status to be changed so that the next call to
+ * schedule() will try to run it.  TODO: take args/envs from userspace. */
+static int sys_proc_create(struct proc *p, char *path, size_t path_l,
+                           struct procinfo *pi)
 {
        int pid = 0;
-       char tpath[MAX_PATH_LEN];
-       /*
-        * There's a bunch of issues with reading in the path, which we'll
-        * need to sort properly in the VFS.  Main concerns are TOCTOU (copy-in),
-        * whether or not it's a big deal that the pointer could be into kernel
-        * space, and resolving both of these without knowing the length of the
-        * string. (TODO)
-        * Change this so that all syscalls with a pointer take a length.
-        *
-        * zra: I've added this user_mem_strlcpy, which I think eliminates the
-     * the TOCTOU issue. Adding a length arg to this call would allow a more
-        * efficient implementation, though, since only one call to user_mem_check
-        * would be required.
-        */
-       int ret = user_mem_strlcpy(p,tpath, path, MAX_PATH_LEN, PTE_USER_RO);
-       int kfs_inode = kfs_lookup_path(tpath);
-       if (kfs_inode < 0)
-               return -EINVAL;
-       struct proc *new_p = kfs_proc_create(kfs_inode);
+       char *t_path;
+       struct file *program;
+       struct proc *new_p;
+
+       /* Copy in the path.  Consider putting an upper bound. */
+       t_path = user_strdup_errno(p, path, path_l);
+       if (IS_ERR(t_path))
+               return -1;
+       program = do_file_open(t_path, 0, 0);
+       user_memdup_free(p, t_path);
+       if (!program)
+               return -1;                      /* presumably, errno is already set */
+       /* TODO: need to split the proc creation, since you must load after setting
+        * args/env, since auxp gets set up there. */
+       //new_p = proc_create(program, 0, 0);
+       if (proc_alloc(&new_p, current))
+               goto mid_error;
+       /* Set the argument stuff needed by glibc */
+       if (memcpy_from_user_errno(p, new_p->procinfo->argp, pi->argp,
+                                  sizeof(pi->argp)))
+               goto late_error;
+       if (memcpy_from_user_errno(p, new_p->procinfo->argbuf, pi->argbuf,
+                                  sizeof(pi->argbuf)))
+               goto late_error;
+       if (load_elf(new_p, program))
+               goto late_error;
+       kref_put(&program->f_kref);
        pid = new_p->pid;
-       proc_decref(new_p, 1); // let go of the reference created in proc_create()
+       kref_put(&new_p->kref); /* give up the reference created in proc_create() */
        return pid;
+late_error:
+       proc_destroy(new_p);
+mid_error:
+       kref_put(&program->f_kref);
+       return -1;
 }
 
 /* Makes process PID runnable.  Consider moving the functionality to process.c */
@@ -239,17 +250,17 @@ static error_t sys_proc_run(struct proc *p, unsigned pid)
        spin_lock(&p->proc_lock);
        // make sure we have access and it's in the right state to be activated
        if (!proc_controls(p, target)) {
-               proc_decref(target, 1);
+               kref_put(&target->kref);
                retval = -EPERM;
        } else if (target->state != PROC_CREATED) {
-               proc_decref(target, 1);
+               kref_put(&target->kref);
                retval = -EINVAL;
        } else {
                __proc_set_state(target, PROC_RUNNABLE_S);
                schedule_proc(target);
        }
        spin_unlock(&p->proc_lock);
-       proc_decref(target, 1);
+       kref_put(&target->kref);
        return retval;
 }
 
@@ -267,20 +278,20 @@ static error_t sys_proc_destroy(struct proc *p, pid_t pid, int exitcode)
                return -1;
        }
        if (!proc_controls(p, p_to_die)) {
-               proc_decref(p_to_die, 1);
+               kref_put(&p_to_die->kref);
                set_errno(current_tf, EPERM);
                return -1;
        }
        if (p_to_die == p) {
                // syscall code and pid2proc both have edible references, only need 1.
                p->exitcode = exitcode;
-               proc_decref(p, 1);
+               kref_put(&p_to_die->kref);
                printd("[PID %d] proc exiting gracefully (code %d)\n", p->pid,exitcode);
        } else {
                printd("[%d] destroying proc %d\n", p->pid, p_to_die->pid);
        }
        proc_destroy(p_to_die);
-       proc_decref(p_to_die, 1);
+       kref_put(&p_to_die->kref);
        return ESUCCESS;
 }
 
@@ -290,31 +301,6 @@ static int sys_proc_yield(struct proc *p, bool being_nice)
        return 0;
 }
 
-static ssize_t sys_run_binary(env_t* e, void *DANGEROUS binary_buf, size_t len,
-                              procinfo_t*DANGEROUS procinfo, size_t num_colors)
-{
-       env_t* env = proc_create(NULL,0);
-       assert(env != NULL);
-
-       // let me know if you use this.  we need to sort process creation better.
-       printk("sys_run_binary() is deprecated.  Use at your own risk.");
-       if(memcpy_from_user(e,e->procinfo,procinfo,sizeof(*procinfo)))
-               return -1;
-       proc_init_procinfo(e);
-
-       env_load_icode(env,e,binary_buf,len);
-       __proc_set_state(env, PROC_RUNNABLE_S);
-       schedule_proc(env);
-       if(num_colors > 0) {
-               env->cache_colors_map = cache_colors_map_alloc();
-               for(int i=0; i<num_colors; i++)
-                       cache_color_alloc(llc_cache, env->cache_colors_map);
-       }
-       proc_decref(env, 1);
-       proc_yield(e, 0);
-       return 0;
-}
-
 static ssize_t sys_fork(env_t* e)
 {
        // TODO: right now we only support fork for single-core processes
@@ -324,7 +310,8 @@ static ssize_t sys_fork(env_t* e)
                return -1;
        }
 
-       env_t* env = proc_create(NULL,0);
+       env_t* env;
+       assert(!proc_alloc(&env, current));
        assert(env != NULL);
 
        env->heap_top = e->heap_top;
@@ -359,6 +346,7 @@ static ssize_t sys_fork(env_t* e)
                        /* 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;
@@ -377,62 +365,73 @@ static ssize_t sys_fork(env_t* e)
        /* 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_decref(env,2);
+               proc_destroy(env);      /* this is prob what you want, not decref by 2 */
                set_errno(current_tf,ENOMEM);
                return -1;
        }
+       
+       /* TODO: copy all open files, except O_CLOEXEC */
 
        __proc_set_state(env, PROC_RUNNABLE_S);
        schedule_proc(env);
 
        // don't decref the new process.
        // that will happen when the parent waits for it.
+       // TODO: if the parent doesn't wait, we need to change the child's parent
+       // when the parent dies, or at least decref it
 
        printd("[PID %d] fork PID %d\n",e->pid,env->pid);
 
        return env->pid;
 }
 
-intreg_t sys_exec(struct proc* p, int fd, procinfo_t* pi)
+/* Load the binary "path" into the current process, and start executing it.
+ * argv and envp are magically bundled in procinfo for now.  Keep in sync with
+ * glibc's sysdeps/ros/execve.c */
+static int sys_exec(struct proc *p, char *path, size_t path_l,
+                    struct procinfo *pi)
 {
-       if(p->state != PROC_RUNNING_S)
-               return -1;
-
        int ret = -1;
-       struct file* f = file_open_from_fd(p,fd);
-       if(f == NULL) {
-               set_errno(current_tf, EBADF);
-               goto out;
-       }
-
-       // Set the argument stuff needed by glibc
-       if(memcpy_from_user(p,p->procinfo->argp,pi->argp,sizeof(pi->argp))) {
-               proc_destroy(p);
-               goto out;
-       }
-       if(memcpy_from_user(p,p->procinfo->argbuf,pi->argbuf,sizeof(pi->argbuf))) {
-               proc_destroy(p);
-               goto out;
-       }
+       char *t_path;
+       struct file *program;
 
-       // TODO: don't do this either (PC)
+       /* We probably want it to never be allowed to exec if it ever was _M */
+       if(p->state != PROC_RUNNING_S)
+               return -1;
+       /* Copy in the path.  Consider putting an upper bound. */
+       t_path = user_strdup_errno(p, path, path_l);
+       if (IS_ERR(t_path))
+               return -1;
+       program = do_file_open(t_path, 0, 0);
+       user_memdup_free(p, t_path);
+       if (!program)
+               return -1;                      /* presumably, errno is already set */
+       /* Set the argument stuff needed by glibc */
+       if (memcpy_from_user_errno(p, p->procinfo->argp, pi->argp,
+                                  sizeof(pi->argp)))
+               goto mid_error;
+       if (memcpy_from_user_errno(p, p->procinfo->argbuf, pi->argbuf,
+                                  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));
-
-       env_user_mem_free(p,0,USTACKTOP);
-
-       if(load_elf(p,f))
-       {
+       env_user_mem_free(p, 0, UMAPTOP);
+       if (load_elf(p, program)) {
+               kref_put(&program->f_kref);
                proc_destroy(p);
-               goto out;
+               smp_idle();             /* syscall can't return on failure now */
        }
-       file_decref(f);
+       printd("[PID %d] exec %s\n", p->pid, file_name(program));
+       kref_put(&program->f_kref);
        *current_tf = p->env_tf;
-       ret = 0;
-
-       printd("[PID %d] exec fd %d\n",p->pid,fd);
-
-out:
-       return ret;
+       return 0;
+mid_error:
+       kref_put(&program->f_kref);
+       return -1;
 }
 
 static ssize_t sys_trywait(env_t* e, pid_t pid, int* status)
@@ -471,7 +470,9 @@ static ssize_t sys_trywait(env_t* e, pid_t pid, int* status)
                }
 
                // if the wait succeeded, decref twice
-               proc_decref(p,1 + (ret == 0));
+               if (ret == 0)
+                       kref_put(&p->kref);
+               kref_put(&p->kref);
                return ret;
        }
 
@@ -536,6 +537,8 @@ static ssize_t sys_shared_page_alloc(env_t* p1,
                                      int p1_flags, int p2_flags
                                     )
 {
+       /* When we remove/change this, also get rid of page_insert_in_range() */
+       printk("[kernel] the current shared page alloc is deprecated.\n");
        //if (!VALID_USER_PERMS(p1_flags)) return -EPERM;
        //if (!VALID_USER_PERMS(p2_flags)) return -EPERM;
 
@@ -548,7 +551,7 @@ static ssize_t sys_shared_page_alloc(env_t* p1,
        page_t* page;
        error_t e = upage_alloc(p1, &page,1);
        if (e < 0) {
-               proc_decref(p2, 1);
+               kref_put(&p2->kref);
                return e;
        }
 
@@ -556,7 +559,7 @@ static ssize_t sys_shared_page_alloc(env_t* p1,
                        (void*SNT)UTEXT, (void*SNT)UTOP, p2_flags);
        if (p2_addr == NULL) {
                page_free(page);
-               proc_decref(p2, 1);
+               kref_put(&p2->kref);
                return -EFAIL;
        }
 
@@ -565,11 +568,11 @@ static ssize_t sys_shared_page_alloc(env_t* p1,
        if(p1_addr == NULL) {
                page_remove(p2->env_pgdir, p2_addr);
                page_free(page);
-               proc_decref(p2, 1);
+               kref_put(&p2->kref);
                return -EFAIL;
        }
        *addr = p1_addr;
-       proc_decref(p2, 1);
+       kref_put(&p2->kref);
        return ESUCCESS;
 }
 
@@ -594,14 +597,14 @@ static int sys_notify(struct proc *p, int target_pid, unsigned int notif,
                return -1;
        }
        if (!proc_controls(p, target)) {
-               proc_decref(target, 1);
+               kref_put(&target->kref);
                set_errno(current_tf, EPERM);
                return -1;
        }
        /* if the user provided a notif_event, copy it in and use that */
        if (u_ne) {
                if (memcpy_from_user(p, &local_ne, u_ne, sizeof(struct notif_event))) {
-                       proc_decref(target, 1);
+                       kref_put(&target->kref);
                        set_errno(current_tf, EINVAL);
                        return -1;
                }
@@ -609,7 +612,7 @@ static int sys_notify(struct proc *p, int target_pid, unsigned int notif,
        } else {
                proc_notify(target, notif, 0);
        }
-       proc_decref(target, 1);
+       kref_put(&target->kref);
        return 0;
 }
 
@@ -849,23 +852,46 @@ intreg_t sys_pread(struct proc* p, int fd, void* buf, int len, int offset)
        return ret;
 }
 
-intreg_t sys_open(struct proc* p, const char* path, int oflag, int mode)
+/* Checks args/reads in the path, opens the file, and inserts it into the
+ * process's open file list. 
+ *
+ * TODO: take the path length */
+intreg_t sys_open(struct proc *p, const char *path, int oflag, int mode)
 {
-       printd("File Open, p: %p, path: %s, oflag: %d, mode: 0x%x\n", p, path, oflag, mode);
-       char* fn = user_strdup_errno(p,path,PGSIZE);
-       if(fn == NULL) {
-               printd("File Open, user_strdup_errno failed\n");
+       int fd = 0;
+       struct file *file;
+
+       char *t_path = user_strdup_errno(p, path, PGSIZE);
+       if (t_path == NULL)
+               return -1;
+       file = do_file_open(t_path, oflag, mode);
+       user_memdup_free(p, t_path);
+       if (!file)
+               return -1;
+       fd = insert_file(&p->open_files, file); /* stores the ref to file */
+       kref_put(&file->f_kref);
+       if (fd < 0) {
+               warn("File insertion failed");
                return -1;
        }
-       printd("File Open, About to open\n");
-       int ret = ufe(open,PADDR(fn),oflag,mode,0);
-       printd("File Open, res=%d\n", ret);
-       user_memdup_free(p,fn);
-       return ret;
+       printd("File Open, res=%d\n", fd);
+       return fd;
 }
-intreg_t sys_close(struct proc* p, int fd)
+
+intreg_t sys_close(struct proc *p, int fd)
 {
-       return ufe(close,fd,0,0,0);
+       struct file *file = put_file_from_fd(&p->open_files, fd);
+       if (!file) {
+               set_errno(current_tf, EBADF);
+               return -1;
+       }
+       /* TEMP TEST */
+       if (kref_refcnt(&file->f_kref)) {
+               printk("sys_close: Detected positive refcnt %d for file %s\n",
+                      kref_refcnt(&file->f_kref), file_name(file));
+               panic("Idiot.");
+       }
+       return 0;
 }
 
 #define NEWLIB_STAT_SIZE 64
@@ -1076,7 +1102,6 @@ intreg_t syscall(struct proc *p, uintreg_t syscallno, uintreg_t a1,
                [SYS_proc_run] = (syscall_t)sys_proc_run,
                [SYS_proc_destroy] = (syscall_t)sys_proc_destroy,
                [SYS_yield] = (syscall_t)sys_proc_yield,
-               [SYS_run_binary] = (syscall_t)sys_run_binary,
                [SYS_fork] = (syscall_t)sys_fork,
                [SYS_exec] = (syscall_t)sys_exec,
                [SYS_trywait] = (syscall_t)sys_trywait,
@@ -1176,7 +1201,7 @@ intreg_t process_generic_syscalls(struct proc *p, size_t max)
         * the *p). */
        // TODO: ought to be unnecessary, if you called this right, kept here for
        // now in case anyone actually uses the ARSCs.
-       proc_incref(p, 1);
+       kref_get(&p->kref, 1);
 
        // max is the most we'll process.  max = 0 means do as many as possible
        while (RING_HAS_UNCONSUMED_REQUESTS(sysbr) && ((!max)||(count < max)) ) {
@@ -1206,7 +1231,7 @@ intreg_t process_generic_syscalls(struct proc *p, size_t max)
        }
        // load sane page tables (and don't rely on decref to do it for you).
        lcr3(boot_cr3);
-       proc_decref(p, 1);
+       kref_put(&p->kref);
        return (intreg_t)count;
 }