Sorted out proc refcounting with ARCs
[akaros.git] / kern / src / kfs.c
index 6e8a983..71fed7f 100644 (file)
 #include <assert.h>
 #include <error.h>
 #include <cpio.h>
+#include <pmap.h>
+#include <smp.h>
 
 #define KFS_MAX_FILE_SIZE 1024*1024*128
 #define KFS_MAGIC 0xdead0001
 
 /* VFS required Functions */
 /* These structs are declared again and initialized farther down */
+struct page_map_operations kfs_pm_op;
 struct super_operations kfs_s_op;
 struct inode_operations kfs_i_op;
 struct dentry_operations kfs_d_op;
@@ -55,8 +58,8 @@ static void kfs_init(void)
                                         __alignof__(struct kfs_i_info), 0, 0, 0);
 }
 
-/* Creates the SB (normally would read in from disc and create).  Ups the refcnt
- * for whoever consumes this.  Returns 0 on failure.
+/* Creates the SB (normally would read in from disc and create).  Passes it's
+ * ref out to whoever consumes this.  Returns 0 on failure.
  * TODO: consider pulling out more of the FS-independent stuff, if possible.
  * There are only two things, but the pain in the ass is that you'd need to read
  * the disc to get that first inode, and it's a FS-specific thing. */
@@ -109,40 +112,59 @@ void kfs_kill_sb(struct super_block *sb)
 struct fs_type kfs_fs_type = {"KFS", 0, kfs_get_sb, kfs_kill_sb, {0, 0},
                TAILQ_HEAD_INITIALIZER(kfs_fs_type.fs_supers)};
 
+/* Page Map Operations */
+
+/* Fills page with its contents from its backing store file.  Note that we do
+ * the zero padding here, instead of higher in the VFS.  Might change in the
+ * future. */
+int kfs_readpage(struct file *file, struct page *page)
+{
+       size_t pg_idx_byte = page->pg_index * PGSIZE;
+       struct kfs_i_info *k_i_info = (struct kfs_i_info*)
+                                     file->f_dentry->d_inode->i_fs_info;
+       uintptr_t begin = (size_t)k_i_info->filestart + pg_idx_byte;
+       /* If we're beyond the initial start point, we just need a zero page.  This
+        * is for a hole or for extending a file (even though it won't be saved).
+        * Otherwise, we want the data from KFS, being careful to not copy from
+        * beyond the original EOF (and zero padding anything extra). */
+       if (pg_idx_byte >= k_i_info->init_size) {
+               memset(page2kva(page), 0, PGSIZE);
+       } else {
+               size_t copy_amt = MIN(PGSIZE, k_i_info->init_size - pg_idx_byte);
+               memcpy(page2kva(page), (void*)begin, copy_amt);
+               memset(page2kva(page) + copy_amt, 0, PGSIZE - copy_amt);
+       }
+       /* This is supposed to be done in the IO system when the operation is
+        * complete.  Since we aren't doing a real IO request, and it is already
+        * done, we can do it here. */
+       page->pg_flags |= PG_UPTODATE;
+       unlock_page(page);
+       return 0;
+}
+
 /* Super Operations */
 
 /* creates and initializes a new inode.  generic fields are filled in.  specific
  * fields are filled in in read_inode() based on what's on the disk for a given
- * i_no.  i_no is set by the caller. */
+ * i_no.  i_no is set by the caller.  Note that this means this inode can be for
+ * an inode that is already on disk, or it can be used when creating. */
 struct inode *kfs_alloc_inode(struct super_block *sb)
 {
-       /* arguably, we can avoid some of this init by using the slab/cache */
        struct inode *inode = kmem_cache_alloc(inode_kcache, 0);
        memset(inode, 0, sizeof(struct inode));
-       TAILQ_INSERT_HEAD(&sb->s_inodes, inode, i_sb_list);
-       TAILQ_INIT(&inode->i_dentry);
-       inode->i_ino = 0;                                       /* set by caller later */
-       atomic_set(&inode->i_refcnt, 1);
-       inode->i_blksize = 1;                           /* keep in sync with get_sb() */
-       spinlock_init(&inode->i_lock);
        inode->i_op = &kfs_i_op;
        inode->i_fop = &kfs_f_op;
-       inode->i_sb = sb;
-       inode->i_state = 0;                                     /* need real states, want I_NEW */
-       inode->dirtied_when = 0;
-       atomic_set(&inode->i_writecount, 0);
+       inode->i_pm.pm_op = &kfs_pm_op;
        inode->i_fs_info = kmem_cache_alloc(kfs_i_kcache, 0);
        TAILQ_INIT(&((struct kfs_i_info*)inode->i_fs_info)->children);
        ((struct kfs_i_info*)inode->i_fs_info)->filestart = 0;
        return inode;
-       /* caller sets i_ino, i_list set when applicable */
 }
 
 /* deallocs and cleans up after an inode. */
 void kfs_destroy_inode(struct inode *inode)
 {
        kmem_cache_free(kfs_i_kcache, inode->i_fs_info);
-       kmem_cache_free(inode_kcache, inode);
 }
 
 /* reads the inode data on disk specified by inode->i_ino into the inode.
@@ -155,7 +177,8 @@ void kfs_read_inode(struct inode *inode)
        /* TODO: what does it mean to ask for an inode->i_ino that doesn't exist?
         *      possibly a bug, since these inos come from directories */
        if (inode->i_ino == 1) {
-               inode->i_mode = 0x777;                  /* TODO: use something appropriate */
+               inode->i_mode = S_IRWXU | S_IRWXG | S_IRWXO;
+               inode->i_type = FS_I_DIR;
                inode->i_nlink = 1;                             /* assuming only one hardlink */
                inode->i_uid = 0;
                inode->i_gid = 0;
@@ -169,13 +192,12 @@ void kfs_read_inode(struct inode *inode)
                inode->i_ctime.tv_nsec = 0;
                inode->i_blocks = 0;
                inode->i_bdev = 0;                              /* assuming blockdev? */
-               inode->i_flags = FS_I_DIR;
+               inode->i_flags = 0;
                inode->i_socket = FALSE;
        } else {
                panic("Not implemented");
        }
-       /* TODO: unused: inode->i_hash add to hash (saves on disc reading)
-        * i_mapping, i_data, when they mean something */
+       /* TODO: unused: inode->i_hash add to hash (saves on disc reading) */
 }
 
 /* called when an inode in memory is modified (journalling FS's care) */
@@ -242,41 +264,15 @@ void kfs_umount_begin(struct super_block *sb)
 
 /* inode_operations */
 
-/* Helper op, used when creating regular files (kfs_create()) and when making
- * directories (kfs_mkdir()).  References are a bit ugly.  We're passing out a
- * ref that is already stored/accounted for.  Might change that...  Also, this
- * needs to handle having nd == 0. */
-struct inode *kfs_create_generic(struct inode *dir, struct dentry *dentry,
-                                 int mode, struct nameidata *nd)
-{
-       /* note it is the i_ino that uniquely identifies a file in the system.
-        * there's a diff between creating an inode (even for an in-use ino) and
-        * then filling it in, and vs creating a brand new one */
-       struct inode *inode = kfs_alloc_inode(dentry->d_sb);
-       dentry->d_inode = inode;                /* inode ref stored here */
-       TAILQ_INSERT_TAIL(&inode->i_dentry, dentry, d_alias); /* stored dentry ref*/
-       inode->i_mode = mode;
-       inode->i_ino = kfs_get_free_ino();
-       inode->i_nlink = 1;
-       inode->i_atime.tv_sec = 0;              /* TODO: now! */
-       inode->i_ctime.tv_sec = 0;              /* TODO: now! */
-       inode->i_mtime.tv_sec = 0;              /* TODO: now! */
-       inode->i_atime.tv_nsec = 0;             /* are these supposed to be the extra ns? */
-       inode->i_ctime.tv_nsec = 0;
-       inode->i_mtime.tv_nsec = 0;
-       return inode;
-}
-
-/* Create a new disk inode in dir associated with dentry, with the given mode.
- * called when creating a regular file.  dir is the directory/parent.  dentry is
- * the dentry of the inode we are creating. */
+/* Called when creating a new disk inode in dir associated with dentry.  We need
+ * to fill out the i_ino, set the type, and do whatever else we need */
 int kfs_create(struct inode *dir, struct dentry *dentry, int mode,
                struct nameidata *nd)
 {
-       struct inode *inode = kfs_create_generic(dir, dentry, mode, nd);        
-       if (!inode)
-               return -1;
-       inode->i_flags = FS_I_FILE;
+       struct inode *inode = dentry->d_inode;
+       kref_get(&dentry->d_kref, 1);   /* to pin the dentry in RAM, KFS-style... */
+       inode->i_ino = kfs_get_free_ino();
+       inode->i_type = FS_I_FILE;
        /* our parent dentry's inode tracks our dentry info.  We do this
         * since it's all in memory and we aren't using the dcache yet.
         * We're reusing the subdirs link, which is used by the VFS when
@@ -290,16 +286,50 @@ int kfs_create(struct inode *dir, struct dentry *dentry, int mode,
 }
 
 /* Searches the directory for the filename in the dentry, filling in the dentry
- * with the FS specific info of this file */
+ * with the FS specific info of this file.  If it succeeds, it will pass back
+ * the *dentry you should use.  If this fails, it will return 0 and will take
+ * the ref to the dentry for you.  Either way, you shouldn't use the ref you
+ * passed in anymore.  Still, there are issues with refcnting with this.
+ *
+ * Callers, make sure you alloc and fill out the name parts of the dentry, and
+ * an initialized nameidata. TODO: not sure why we need an ND.  Don't use it in
+ * a fs_lookup for now!
+ *
+ * Because of the way KFS currently works, if there is ever a dentry, it's
+ * already in memory, along with its inode (all path's pinned).  So we just find
+ * it and return it, freeing the one that came in. */
 struct dentry *kfs_lookup(struct inode *dir, struct dentry *dentry,
                           struct nameidata *nd)
 {
-       // TODO: does this mean the inode too?
-       // what the hell are we returning?  the passed in dentry?  an error code?
-       // this will have to read the directory, then find the ino, then creates a
-       // dentry for it
-       //
-       // linux now has a nameidata for this
+       struct kfs_i_info *k_i_info = (struct kfs_i_info*)dir->i_fs_info;
+       struct dentry *dir_dent = TAILQ_FIRST(&dir->i_dentry);
+       struct dentry *d_i;
+
+       assert(dir_dent && dir_dent == TAILQ_LAST(&dir->i_dentry, dentry_tailq));
+       assert(dir->i_type & FS_I_DIR);
+       assert(kref_refcnt(&dentry->d_kref) == 1);
+       TAILQ_FOREACH(d_i, &dir_dent->d_subdirs, d_subdirs_link) {
+               if (!strcmp(d_i->d_name.name, dentry->d_name.name)) {
+                       /* since this dentry is already in memory (that's how KFS works), we
+                        * can free the one that came in and return the real one */
+                       kref_put(&dentry->d_kref);
+                       kref_get(&d_i->d_kref, 1);
+                       return d_i;
+               }
+       }
+       TAILQ_FOREACH(d_i, &k_i_info->children, d_subdirs_link) {
+               if (!strcmp(d_i->d_name.name, dentry->d_name.name)) {
+                       /* since this dentry is already in memory (that's how KFS works), we
+                        * can free the one that came in and return the real one */
+                       kref_put(&dentry->d_kref);
+                       kref_get(&d_i->d_kref, 1);
+                       return d_i;
+               }
+       }
+       /* no match, consider caching the negative result, freeing the
+        * dentry, etc */
+       printd("Not Found %s!!\n", dentry->d_name.name);
+       kref_put(&dentry->d_kref);
        return 0;
 }
 
@@ -324,21 +354,16 @@ int kfs_symlink(struct inode *dir, struct dentry *dentry, const char *symname)
        return -1;
 }
 
-/* Creates a new inode for a directory associated with dentry in dir with the
- * given mode.  Note, we might (later) need to track subdirs within the parent
- * inode, like we do with regular files.  I'd rather not, so we'll see if we
- * need it. */
+/* Called when creating a new inode for a directory associated with dentry in
+ * dir with the given mode.  Note, we might (later) need to track subdirs within
+ * the parent inode, like we do with regular files.  I'd rather not, so we'll
+ * see if we need it. */
 int kfs_mkdir(struct inode *dir, struct dentry *dentry, int mode)
 {
-       struct inode *inode = kfs_create_generic(dir, dentry, mode, 0); 
-       if (!inode)
-               return -1;
-       struct dentry *parent = TAILQ_FIRST(&dir->i_dentry);
-       assert(parent && parent == TAILQ_LAST(&dir->i_dentry, dentry_tailq));
-       inode->i_flags = FS_I_DIR;
-       /* parent dentry tracks dentry as a subdir */
-       TAILQ_INSERT_TAIL(&parent->d_subdirs, dentry, d_subdirs_link);
-       atomic_inc(&dentry->d_refcnt);
+       struct inode *inode = dentry->d_inode;
+       kref_get(&dentry->d_kref, 1);   /* to pin the dentry in RAM, KFS-style... */
+       inode->i_ino = kfs_get_free_ino();
+       inode->i_type = FS_I_DIR;               /* this might be FS specific in the future */
        /* get ready to have our own kids */
        TAILQ_INIT(&((struct kfs_i_info*)inode->i_fs_info)->children);
        ((struct kfs_i_info*)inode->i_fs_info)->filestart = 0;
@@ -429,7 +454,7 @@ int kfs_d_delete(struct dentry *dentry)
 
 /* Called when it's about to be slab-freed */
 int kfs_d_release(struct dentry *dentry)
-{ // default, nothin
+{
        return -1;
 }
 
@@ -441,76 +466,109 @@ void kfs_d_iput(struct dentry *dentry, struct inode *inode)
 
 /* file_operations */
 
-/* Updates the file pointer */
+/* Updates the file pointer.  KFS doesn't let you go past the end of a file
+ * yet, so it won't let you seek past either.  TODO: think about locking. */
 off_t kfs_llseek(struct file *file, off_t offset, int whence)
 {
-       return -1;
-}
-
-/* Read cound bytes from the file into buf, starting at *offset, which is increased
- * accordingly */
-ssize_t kfs_read(struct file *file, char *buf, size_t count, off_t *offset)
+       off_t temp_off = 0;
+       switch (whence) {
+               case SEEK_SET:
+                       temp_off = offset;
+                       break;
+               case SEEK_CUR:
+                       temp_off = file->f_pos + offset;
+                       break;
+               case SEEK_END:
+                       temp_off = file->f_dentry->d_inode->i_size + offset;
+                       break;
+               default:
+                       set_errno(EINVAL);
+                       warn("Unknown 'whence' in llseek()!\n");
+                       return -1;
+       }
+       /* make sure the f_pos isn't outside the limits of the existing file.
+        * techincally, if they go too far, we should return EINVAL */
+       temp_off = MAX(MIN(temp_off, file->f_dentry->d_inode->i_size), 0);
+       file->f_pos = temp_off;
+       return temp_off;
+}
+
+/* Fills in the next directory entry (dirent), starting with d_off.  Like with
+ * read and write, there will be issues with userspace and the *dirent buf.
+ * TODO: we don't really do anything with userspace concerns here, in part
+ * because memcpy_to doesn't work well.  When we fix how we want to handle the
+ * userbuffers, we can write this accordingly. (UMEM)  */
+int kfs_readdir(struct file *dir, struct dirent *dirent)
 {
-       return -1;
-}
+       int count = 0;
+       bool found = FALSE;
+       struct dentry *subent;
+       struct dentry *dir_d = dir->f_dentry;
+       struct kfs_i_info *k_i_info = (struct kfs_i_info*)dir_d->d_inode->i_fs_info;
+
+       /* how we check inside the for loops below.  moderately ghetto. */
+       void check_entry(void)
+       {
+               if (count++ == dirent->d_off) {
+                       dirent->d_ino = subent->d_inode->i_ino;
+                       dirent->d_reclen = subent->d_name.len;
+                       /* d_name.name is null terminated, the byte after d_name.len */
+                       assert(subent->d_name.len <= MAX_FILENAME_SZ);
+                       strncpy(dirent->d_name, subent->d_name.name, subent->d_name.len +1);
+                       found = TRUE;
+               }
+       }
+       /* some of this error handling can be done by the VFS.  The syscall should
+        * handle EBADF, EFAULT, and EINVAL (TODO, memory related). */
+       if (!(dir_d->d_inode->i_type & FS_I_DIR)) {
+               set_errno(ENOTDIR);
+               return -1;
+       }
 
-/* Writes count bytes from buf to the file, starting at *offset, which is
- * increased accordingly */
-ssize_t kfs_write(struct file *file, const char *buf, size_t count,
-                  off_t *offset)
-{
-       return -1;
-}
+       /* need to check the sub-dirs as well as the sub-"files" */
+       TAILQ_FOREACH(subent, &dir_d->d_subdirs, d_subdirs_link)
+               check_entry();
+       TAILQ_FOREACH(subent, &k_i_info->children, d_subdirs_link)
+               check_entry();
 
-/* Fills in the next directory entry (dirent), starting with d_off */
-int kfs_readdir(struct file *dir, struct dirent *dirent)
-{
-       return -1;
+       if (!found) {
+               set_errno(ENOENT);
+               return -1;
+       }
+       if (count - 1 == dirent->d_off)         /* found the last dir in the list */
+               return 0;
+       return 1;                                                       /* normal success for readdir */
 }
 
-/* Memory maps the file into the virtual memory area */
-int kfs_mmap(struct file *file, struct vm_area_struct *vma)
+/* This is called when a VMR is mapping a particular file.  The FS needs to do
+ * whatever it needs so that faults can be handled by read_page(), and handle all
+ * of the cases of MAP_SHARED, MAP_PRIVATE, whatever.  It also needs to ensure
+ * the file is not being mmaped in a way that conflicts with the manner in which
+ * the file was opened or the file type. */
+int kfs_mmap(struct file *file, struct vm_region *vmr)
 {
+       if (file->f_dentry->d_inode->i_type & FS_I_FILE)
+               return 0;
        return -1;
 }
 
-/* Opens the file specified by the inode, creating and filling in the file */
-/* TODO: fill out the other // entries, sort vmnt refcnting */
+/* Called by the VFS while opening the file, which corresponds to inode,  for
+ * the FS to do whatever it needs. */
 int kfs_open(struct inode *inode, struct file *file)
 {
-       /* This is mostly FS-agnostic, consider a helper */
-       file = kmem_cache_alloc(file_kcache, 0);
-       /* Add to the list of all files of this SB */
-       TAILQ_INSERT_TAIL(&inode->i_sb->s_files, file, f_list);
-       file->f_inode = inode;
-       atomic_inc(&inode->i_refcnt);
-       file->f_vfsmnt = inode->i_sb->s_mount;          /* saving a ref to the vmnt...*/
-       file->f_op = &kfs_f_op;
-       atomic_set(&file->f_refcnt, 1);                         /* ref passed out */
-       file->f_flags = inode->i_flags;                         /* just taking the inode vals */
-       file->f_mode = inode->i_mode;
-       file->f_pos = 0;
-       file->f_uid = inode->i_uid;
-       file->f_gid = inode->i_gid;
-       file->f_error = 0;
-//     struct event_poll_tailq         f_ep_links;
-       spinlock_init(&file->f_ep_lock);
-       file->f_fs_info = 0;
-//     struct address_space            *f_mapping;             /* page cache mapping */
        return 0;
 }
 
 /* Called when a file descriptor is closed. */
 int kfs_flush(struct file *file)
 {
-       kmem_cache_free(file_kcache, file);
        return -1;
 }
 
-/* Called when the file refcnt == 0 */
+/* Called when the file is about to be closed (file obj freed) */
 int kfs_release(struct inode *inode, struct file *file)
 {
-       return -1;
+       return 0;
 }
 
 /* Flushes the file's dirty contents to disc */
@@ -556,6 +614,10 @@ int kfs_check_flags(int flags)
 }
 
 /* Redeclaration and initialization of the FS ops structures */
+struct page_map_operations kfs_pm_op = {
+       kfs_readpage,
+};
+
 struct super_operations kfs_s_op = {
        kfs_alloc_inode,
        kfs_destroy_inode,
@@ -600,8 +662,8 @@ struct dentry_operations kfs_d_op = {
 
 struct file_operations kfs_f_op = {
        kfs_llseek,
-       kfs_read,
-       kfs_write,
+       generic_file_read,
+       generic_file_write,
        kfs_readdir,
        kfs_mmap,
        kfs_open,
@@ -617,97 +679,18 @@ struct file_operations kfs_f_op = {
 
 /* KFS Specific Internal Functions */
 
-/* For obj files compiled with the kernel */
-#define DECL_PROG(x) \
-    extern uint8_t (COUNT(sizeof(size_t)) _binary_obj_tests_##x##_size)[],\
-        (COUNT(_binary_obj_user_apps_##x##_size)_binary_obj_tests_##x##_start)[];
-
-#define KFS_PENTRY(x) {#x, _binary_obj_tests_##x##_start, (size_t) _binary_obj_tests_##x##_size},
-
-/*
- * Hardcode the files included in the KFS.  PROGs need to be in sync with the
- * userapps in kern/src/Makefrag.  Files need to be in kern/kfs/
- * Make sure to declare it, and add an entry.  Keep MAX_KFS_FILES big enough too
- * Note that files with a . in their name will have an _ instead.
- */
-#ifdef __CONFIG_KFS__
-DECL_PROG(tlstest);
-DECL_PROG(fp_test);
-DECL_PROG(mproctests);
-DECL_PROG(hello);
-DECL_PROG(mhello);
-DECL_PROG(pthread_test);
-DECL_PROG(pthread_barrier_test);
-DECL_PROG(idle);
-DECL_PROG(tsc_spitter);
-DECL_PROG(msr_get_cores);
-DECL_PROG(msr_get_singlecore);
-DECL_PROG(msr_dumb_while);
-DECL_PROG(msr_nice_while);
-DECL_PROG(msr_single_while);
-DECL_PROG(msr_cycling_vcores);
-#endif
-
-struct kfs_entry kfs[MAX_KFS_FILES] = {
-#ifdef __CONFIG_KFS__
-       KFS_PENTRY(tlstest)
-       KFS_PENTRY(fp_test)
-       KFS_PENTRY(mproctests)
-       KFS_PENTRY(hello)
-       KFS_PENTRY(mhello)
-       KFS_PENTRY(pthread_test)
-       KFS_PENTRY(pthread_barrier_test)
-       KFS_PENTRY(idle)
-       KFS_PENTRY(tsc_spitter)
-       KFS_PENTRY(msr_get_cores)
-       KFS_PENTRY(msr_get_singlecore)
-       KFS_PENTRY(msr_dumb_while)
-       KFS_PENTRY(msr_nice_while)
-       KFS_PENTRY(msr_single_while)
-       KFS_PENTRY(msr_cycling_vcores)
-#endif
-};
-
-ssize_t kfs_lookup_path(char* path)
-{
-       for (int i = 0; i < MAX_KFS_FILES; i++)
-               // need to think about how to copy-in something of unknown length
-               if (!strncmp(kfs[i].name, path, strlen(path)))
-                       return i;
-       return -EINVAL;
-}
-
-/*
- * Creates a process from the file pointed to by the KFS inode (index)
- * This should take a real inode or something to point to the real location,
- * and proc_create shouldn't assume everything is contiguous
- */
-struct proc *kfs_proc_create(int kfs_inode)
-{
-       if (kfs_inode < 0 || kfs_inode >= MAX_KFS_FILES)
-               panic("Invalid kfs_inode.  Check you error codes!");
-       return proc_create(kfs[kfs_inode].start, kfs[kfs_inode].size);
-}
-
-/* Dumps the contents of the KFS file to the console.  Not meant to be taken
- * too seriously - just dumps each char. */
-void kfs_cat(int kfs_inode)
-{
-       if (kfs_inode < 0 || kfs_inode >= MAX_KFS_FILES)
-               panic("Invalid kfs_inode.  Check you error codes!");
-       uint8_t *end = kfs[kfs_inode].start + kfs[kfs_inode].size;
-       for (uint8_t *ptr = kfs[kfs_inode].start; ptr < end; ptr++)
-               cputchar(*ptr);
-}
-
-/* Need to pass path separately, since we'll recurse on it */
+/* Need to pass path separately, since we'll recurse on it.  TODO: this recurses,
+ * and takes up a lot of stack space (~270 bytes).  Core 0's KSTACK is 8 pages,
+ * which can handle about 120 levels deep...  Other cores are not so fortunate.
+ * Can rework this if it becomes an issue. */
 static int __add_kfs_entry(struct dentry *parent, char *path,
                            struct cpio_bin_hdr *c_bhdr)
 {
        char *first_slash = strchr(path, '/');  
        char dir[MAX_FILENAME_SZ + 1];  /* room for the \0 */
        size_t dirname_sz;                              /* not counting the \0 */
-       struct dentry *dentry;
+       struct dentry *dentry = 0;
+       struct nameidata nd = {0};
        struct inode *inode;
 
        if (first_slash) {
@@ -718,31 +701,44 @@ static int __add_kfs_entry(struct dentry *parent, char *path,
                assert(dirname_sz <= MAX_FILENAME_SZ);
                strncpy(dir, path, dirname_sz);
                dir[dirname_sz] = '\0';
-               printk("Finding DIR %s in dentry %s (start: %p, size %d)\n", dir,
+               printd("Finding DIR %s in dentry %s (start: %p, size %d)\n", dir,
                       parent->d_name.name, c_bhdr->c_filestart, c_bhdr->c_filesize);
-               // send to the real dentry
-               __add_kfs_entry(parent, first_slash + 1, c_bhdr);
-
+               /* Need to create a dentry for the lookup, and fill in the basic nd */
+               dentry = get_dentry(parent->d_sb, parent, dir);
+               nd.dentry = dentry;
+               nd.mnt = dentry->d_sb->s_mount;
+               //nd.flags = 0;                 /* TODO: once we have lookup flags */
+               //nd.last_type = 0;             /* TODO: should be a DIR */
+               //nd.intent = 0;                /* TODO: RW, prob irrelevant*/
+               /* TODO: use a VFS lookup instead, to use the dcache, thought its not a
+                * big deal since KFS currently pins all metadata. */
+               dentry = kfs_lookup(parent->d_inode, dentry, &nd);
+               if (!dentry) {
+                       printk("Missing dir in CPIO archive or something, aborting.\n");
+                       return -1;
+               }
+               return __add_kfs_entry(dentry, first_slash + 1, c_bhdr);
        } else {
                /* no directories left in the path.  add the 'file' to the dentry */
-               printk("Adding file/dir %s to dentry %s (start: %p, size %d)\n", path,
+               printd("Adding file/dir %s to dentry %s (start: %p, size %d)\n", path,
                       parent->d_name.name, c_bhdr->c_filestart, c_bhdr->c_filesize);
-
                /* Init the dentry for this path */
                dentry = get_dentry(parent->d_sb, parent, path);
-               dentry->d_op = &kfs_d_op;
                dcache_put(dentry);                     /* TODO: should set a d_flag too */
                /* build the inode */
+               /* XXX: note we use an unrefcnt'd inode here (grabbing the dentry's) */
                if (!c_bhdr->c_filesize) {
                        /* we are a directory.  Note that fifos might look like dirs... */
-                       kfs_mkdir(parent->d_inode, dentry, c_bhdr->c_mode);
+                       create_dir(parent->d_inode, dentry, c_bhdr->c_mode);
                        inode = dentry->d_inode;
                } else {
                        /* we are a file */
-                       kfs_create(parent->d_inode, dentry, c_bhdr->c_mode, 0);
+                       create_file(parent->d_inode, dentry, c_bhdr->c_mode);
                        inode = dentry->d_inode;
                        ((struct kfs_i_info*)inode->i_fs_info)->filestart =
                                                                c_bhdr->c_filestart;
+                       ((struct kfs_i_info*)inode->i_fs_info)->init_size =
+                                                               c_bhdr->c_filesize;
                }
                /* Set other info from the CPIO entry */
                inode->i_uid = c_bhdr->c_uid;
@@ -813,10 +809,11 @@ void parse_cpio_entries(struct super_block *sb, void *cpio_b)
                c_bhdr->c_dev_min = cpio_strntol(buf, c_hdr->c_dev_min, 8);
                c_bhdr->c_rdev_maj = cpio_strntol(buf, c_hdr->c_rdev_maj, 8);
                c_bhdr->c_rdev_min = cpio_strntol(buf, c_hdr->c_rdev_min, 8);
-               printk("File: %s: %d Bytes\n", c_bhdr->c_filename, c_bhdr->c_filesize);
+               printd("File: %s: %d Bytes\n", c_bhdr->c_filename, c_bhdr->c_filesize);
                offset += namesize;
                /* header + name will be padded out to 4-byte alignment */
                offset = ROUNDUP(offset, 4);
+               c_bhdr->c_filestart = cpio_b + offset;
                /* make this a function pointer or something */
                if (add_kfs_entry(sb, c_bhdr)) {
                        printk("Failed to add an entry to KFS!\n");
@@ -829,25 +826,3 @@ void parse_cpio_entries(struct super_block *sb, void *cpio_b)
        }
        kfree(c_bhdr);
 }
-
-/* Debugging */
-void print_dir_tree(struct dentry *dentry, int depth)
-{
-       struct inode *inode = dentry->d_inode;
-       struct kfs_i_info *k_i_info = (struct kfs_i_info*)inode->i_fs_info;
-       struct dentry *d_i;
-       assert(dentry && inode && inode->i_flags & FS_I_DIR);
-       char buf[32] = {0};
-
-       for (int i = 0; i < depth; i++)
-               buf[i] = '\t';
-
-       TAILQ_FOREACH(d_i, &dentry->d_subdirs, d_subdirs_link) {
-               printk("%sDir %s has child dir: %s\n", buf, dentry->d_name.name,
-                      d_i->d_name.name);
-               print_dir_tree(d_i, depth + 1);
-       }
-       TAILQ_FOREACH(d_i, &k_i_info->children, d_subdirs_link)
-               printk("%sDir %s has child file: %s\n", buf, dentry->d_name.name,
-                      d_i->d_name.name);
-}