3ac619cfb8c7a43dc4633eb5a1b616341cca214f
[akaros.git] / kern / src / vfs.c
1 /* Copyright (c) 2009, 2010 The Regents of the University of California
2  * Barret Rhoden <brho@cs.berkeley.edu>
3  * See LICENSE for details.
4  *
5  * Default implementations and global values for the VFS. */
6
7 #include <vfs.h> // keep this first
8 #include <sys/queue.h>
9 #include <assert.h>
10 #include <stdio.h>
11 #include <atomic.h>
12 #include <slab.h>
13 #include <kmalloc.h>
14 #include <kfs.h>
15 #include <pmap.h>
16 #include <umem.h>
17 #include <smp.h>
18
19 struct sb_tailq super_blocks = TAILQ_HEAD_INITIALIZER(super_blocks);
20 spinlock_t super_blocks_lock = SPINLOCK_INITIALIZER;
21 struct fs_type_tailq file_systems = TAILQ_HEAD_INITIALIZER(file_systems);
22 struct namespace default_ns;
23 // TODO: temp dcache, holds all dentries ever for now
24 struct dentry_slist dcache = SLIST_HEAD_INITIALIZER(dcache);
25 spinlock_t dcache_lock = SPINLOCK_INITIALIZER;
26
27 struct kmem_cache *dentry_kcache; // not to be confused with the dcache
28 struct kmem_cache *inode_kcache;
29 struct kmem_cache *file_kcache;
30
31 /* Mounts fs from dev_name at mnt_pt in namespace ns.  There could be no mnt_pt,
32  * such as with the root of (the default) namespace.  Not sure how it would work
33  * with multiple namespaces on the same FS yet.  Note if you mount the same FS
34  * multiple times, you only have one FS still (and one SB).  If we ever support
35  * that... */
36 struct vfsmount *mount_fs(struct fs_type *fs, char *dev_name,
37                           struct dentry *mnt_pt, int flags,
38                           struct namespace *ns)
39 {
40         struct super_block *sb;
41         struct vfsmount *vmnt = kmalloc(sizeof(struct vfsmount), 0);
42
43         /* this first ref is stored in the NS tailq below */
44         kref_init(&vmnt->mnt_kref, fake_release, 1);
45         /* Build the vfsmount, if there is no mnt_pt, mnt is the root vfsmount (for now).
46          * fields related to the actual FS, like the sb and the mnt_root are set in
47          * the fs-specific get_sb() call. */
48         if (!mnt_pt) {
49                 vmnt->mnt_parent = NULL;
50                 vmnt->mnt_mountpoint = NULL;
51         } else { /* common case, but won't be tested til we try to mount another FS */
52                 mnt_pt->d_mount_point = TRUE;
53                 mnt_pt->d_mounted_fs = vmnt;
54                 kref_get(&vmnt->mnt_kref, 1); /* held by mnt_pt */
55                 vmnt->mnt_parent = mnt_pt->d_sb->s_mount;
56                 vmnt->mnt_mountpoint = mnt_pt;
57         }
58         TAILQ_INIT(&vmnt->mnt_child_mounts);
59         vmnt->mnt_flags = flags;
60         vmnt->mnt_devname = dev_name;
61         vmnt->mnt_namespace = ns;
62         kref_get(&ns->kref, 1); /* held by vmnt */
63
64         /* Read in / create the SB */
65         sb = fs->get_sb(fs, flags, dev_name, vmnt);
66         if (!sb)
67                 panic("You're FS sucks");
68
69         /* TODO: consider moving this into get_sb or something, in case the SB
70          * already exists (mounting again) (if we support that) */
71         spin_lock(&super_blocks_lock);
72         TAILQ_INSERT_TAIL(&super_blocks, sb, s_list); /* storing a ref here... */
73         spin_unlock(&super_blocks_lock);
74
75         /* Update holding NS */
76         spin_lock(&ns->lock);
77         TAILQ_INSERT_TAIL(&ns->vfsmounts, vmnt, mnt_list);
78         spin_unlock(&ns->lock);
79         /* note to self: so, right after this point, the NS points to the root FS
80          * mount (we return the mnt, which gets assigned), the root mnt has a dentry
81          * for /, backed by an inode, with a SB prepped and in memory. */
82         return vmnt;
83 }
84
85 void vfs_init(void)
86 {
87         struct fs_type *fs;
88
89         dentry_kcache = kmem_cache_create("dentry", sizeof(struct dentry),
90                                           __alignof__(struct dentry), 0, 0, 0);
91         inode_kcache = kmem_cache_create("inode", sizeof(struct inode),
92                                          __alignof__(struct inode), 0, 0, 0);
93         file_kcache = kmem_cache_create("file", sizeof(struct file),
94                                         __alignof__(struct file), 0, 0, 0);
95         /* default NS never dies, +1 to exist */
96         kref_init(&default_ns.kref, fake_release, 1);
97         spinlock_init(&default_ns.lock);
98         default_ns.root = NULL;
99         TAILQ_INIT(&default_ns.vfsmounts);
100
101         /* build list of all FS's in the system.  put yours here.  if this is ever
102          * done on the fly, we'll need to lock. */
103         TAILQ_INSERT_TAIL(&file_systems, &kfs_fs_type, list);
104         TAILQ_FOREACH(fs, &file_systems, list)
105                 printk("Supports the %s Filesystem\n", fs->name);
106
107         /* mounting KFS at the root (/), pending root= parameters */
108         // TODO: linux creates a temp root_fs, then mounts the real root onto that
109         default_ns.root = mount_fs(&kfs_fs_type, "RAM", NULL, 0, &default_ns);
110
111         printk("vfs_init() completed\n");
112 }
113
114 /* Builds / populates the qstr of a dentry based on its d_iname.  If there is an
115  * l_name, (long), it will use that instead of the inline name.  This will
116  * probably change a bit. */
117 void qstr_builder(struct dentry *dentry, char *l_name)
118 {
119         dentry->d_name.name = l_name ? l_name : dentry->d_iname;
120         // TODO: pending what we actually do in d_hash
121         //dentry->d_name.hash = dentry->d_op->d_hash(dentry, &dentry->d_name); 
122         dentry->d_name.hash = 0xcafebabe;
123         dentry->d_name.len = strnlen(dentry->d_name.name, MAX_FILENAME_SZ);
124 }
125
126 /* Useful little helper - return the string ptr for a given file */
127 char *file_name(struct file *file)
128 {
129         return file->f_dentry->d_name.name;
130 }
131
132 /* Some issues with this, coupled closely to fs_lookup.  This assumes that
133  * negative dentries are not returned (might differ from linux) */
134 static struct dentry *do_lookup(struct dentry *parent, char *name)
135 {
136         struct dentry *dentry;
137         /* TODO: look up in the dentry cache first */
138         dentry = get_dentry(parent->d_sb, parent, name);
139         dentry = parent->d_inode->i_op->lookup(parent->d_inode, dentry, 0);
140         /* insert in dentry cache */
141         /* TODO: if the following are done by us, how do we know the i_ino?
142          * also need to handle inodes that are already read in!  For now, we're
143          * going to have the FS handle it in it's lookup() method: 
144          * - get a new inode
145          * - read in the inode
146          * - put in the inode cache */
147         return dentry;
148 }
149
150 /* Update ND such that it represents having followed dentry.  IAW the nd
151  * refcnting rules, we need to decref any references that were in there before
152  * they get clobbered. */
153 static int next_link(struct dentry *dentry, struct nameidata *nd)
154 {
155         assert(nd->dentry && nd->mnt);
156         /* update the dentry */
157         kref_get(&dentry->d_kref, 1);
158         kref_put(&nd->dentry->d_kref);
159         nd->dentry = dentry;
160         /* update the mount, if we need to */
161         if (dentry->d_sb->s_mount != nd->mnt) {
162                 kref_get(&dentry->d_sb->s_mount->mnt_kref, 1);
163                 kref_put(&nd->mnt->mnt_kref);
164                 nd->mnt = dentry->d_sb->s_mount;
165         }
166         return 0;
167 }
168
169 /* Walk up one directory, being careful of mountpoints, namespaces, and the top
170  * of the FS */
171 static int climb_up(struct nameidata *nd)
172 {
173         printd("CLIMB_UP, from %s\n", nd->dentry->d_name.name);
174         /* Top of the world, just return.  Should also check for being at the top of
175          * the current process's namespace (TODO) */
176         if (!nd->dentry->d_parent)
177                 return -1;
178         /* Check if we are at the top of a mount, if so, we need to follow
179          * backwards, and then climb_up from that one.  We might need to climb
180          * multiple times if we mount multiple FSs at the same spot (highly
181          * unlikely).  This is completely untested.  Might recurse instead. */
182         while (nd->mnt->mnt_root == nd->dentry) {
183                 if (!nd->mnt->mnt_parent) {
184                         warn("Might have expected a parent vfsmount (dentry had a parent)");
185                         return -1;
186                 }
187                 next_link(nd->mnt->mnt_mountpoint, nd);
188         }
189         /* Backwards walk (no mounts or any other issues now). */
190         next_link(nd->dentry->d_parent, nd);
191         printd("CLIMB_UP, to   %s\n", nd->dentry->d_name.name);
192         return 0;
193 }
194
195 static int follow_mount(struct nameidata *nd)
196 {
197         /* Detect mount, follow, etc... (TODO!) */
198         return 0;
199 }
200
201 static int link_path_walk(char *path, struct nameidata *nd);
202
203 /* When nd->dentry is for a symlink, this will recurse and follow that symlink,
204  * so that nd contains the results of following the symlink (dentry and mnt).
205  * Returns when it isn't a symlink, 1 on following a link, and < 0 on error. */
206 static int follow_symlink(struct nameidata *nd)
207 {
208         int retval;
209         char *symname;
210         if (nd->dentry->d_inode->i_type != FS_I_SYMLINK)
211                 return 0;
212         if (nd->depth > MAX_SYMLINK_DEPTH)
213                 return -ELOOP;
214         printd("Following symlink for dentry %08p %s\n", nd->dentry,
215                nd->dentry->d_name.name);
216         nd->depth++;
217         symname = nd->dentry->d_inode->i_op->readlink(nd->dentry);
218         /* We need to pin in nd->dentry (the dentry of the symlink), since we need
219          * it's symname's storage to stay in memory throughout the upcoming
220          * link_path_walk().  The last_sym gets decreffed when we path_release() or
221          * follow another symlink. */
222         if (nd->last_sym)
223                 kref_put(&nd->last_sym->d_kref);
224         kref_get(&nd->dentry->d_kref, 1);
225         nd->last_sym = nd->dentry;
226         /* If this an absolute path in the symlink, we need to free the old path and
227          * start over, otherwise, we continue from the PARENT of nd (the symlink) */
228         if (symname[0] == '/') {
229                 path_release(nd);
230                 if (!current)
231                         nd->dentry = default_ns.root->mnt_root;
232                 else
233                         nd->dentry = current->fs_env.root;      
234                 nd->mnt = nd->dentry->d_sb->s_mount;
235                 kref_get(&nd->mnt->mnt_kref, 1);
236                 kref_get(&nd->dentry->d_kref, 1);
237         } else {
238                 climb_up(nd);
239         }
240         /* either way, keep on walking in the free world! */
241         retval = link_path_walk(symname, nd);
242         return (retval == 0 ? 1 : retval);
243 }
244
245 /* Little helper, to make it easier to break out of the nested loops.  Will also
246  * '\0' out the first slash if it's slashes all the way down.  Or turtles. */
247 static bool packed_trailing_slashes(char *first_slash)
248 {
249         for (char *i = first_slash; *i == '/'; i++) {
250                 if (*(i + 1) == '\0') {
251                         *first_slash = '\0';
252                         return TRUE;
253                 }
254         }
255         return FALSE;
256 }
257
258 /* Simple helper to set nd to track it's last name to be Name.  Also be careful
259  * with the storage of name.  Don't use and nd's name past the lifetime of the
260  * string used in the path_lookup()/link_path_walk/whatever.  Consider replacing
261  * parts of this with a qstr builder.  Note this uses the dentry's d_op, which
262  * might not be the dentry we care about. */
263 static void stash_nd_name(struct nameidata *nd, char *name)
264 {
265         nd->last.name = name;
266         nd->last.len = strlen(name);
267         nd->last.hash = nd->dentry->d_op->d_hash(nd->dentry, &nd->last);
268 }
269
270 /* Resolves the links in a basic path walk.  0 for success, -EWHATEVER
271  * otherwise.  The final lookup is returned via nd. */
272 static int link_path_walk(char *path, struct nameidata *nd)
273 {
274         struct dentry *link_dentry;
275         struct inode *link_inode, *nd_inode;
276         char *next_slash;
277         char *link = path;
278         int error;
279
280         /* Prevent crazy recursion */
281         if (nd->depth > MAX_SYMLINK_DEPTH)
282                 return -ELOOP;
283         /* skip all leading /'s */
284         while (*link == '/')
285                 link++;
286         /* if there's nothing left (null terminated), we're done */
287         if (*link == '\0')
288                 return 0;
289         /* iterate through each intermediate link of the path.  in general, nd
290          * tracks where we are in the path, as far as dentries go.  once we have the
291          * next dentry, we try to update nd based on that dentry.  link is the part
292          * of the path string that we are looking up */
293         while (1) {
294                 nd_inode = nd->dentry->d_inode;
295                 if ((error = check_perms(nd_inode, nd->intent)))
296                         return error;
297                 /* find the next link, break out if it is the end */
298                 next_slash = strchr(link, '/');
299                 if (!next_slash) {
300                         break;
301                 } else {
302                         if (packed_trailing_slashes(next_slash)) {
303                                 nd->flags |= LOOKUP_DIRECTORY;
304                                 break;
305                         }
306                 }
307                 /* skip over any interim ./ */
308                 if (!strncmp("./", link, 2))
309                         goto next_loop;
310                 /* Check for "../", walk up */
311                 if (!strncmp("../", link, 3)) {
312                         climb_up(nd);
313                         goto next_loop;
314                 }
315                 *next_slash = '\0';
316                 link_dentry = do_lookup(nd->dentry, link);
317                 *next_slash = '/';
318                 if (!link_dentry)
319                         return -ENOENT;
320                 /* make link_dentry the current step/answer */
321                 next_link(link_dentry, nd);
322                 kref_put(&link_dentry->d_kref); /* do_lookup gave us a refcnt dentry */
323                 /* we could be on a mountpoint or a symlink - need to follow them */
324                 follow_mount(nd);
325                 if ((error = follow_symlink(nd)) < 0)
326                         return error;
327                 /* Turn off a possible DIRECTORY lookup, which could have been set
328                  * during the follow_symlink (a symlink could have had a directory at
329                  * the end), though it was in the middle of the real path. */
330                 nd->flags &= ~LOOKUP_DIRECTORY;
331                 if (!(nd->dentry->d_inode->i_type & FS_I_DIR))
332                         return -ENOTDIR;
333 next_loop:
334                 /* move through the path string to the next entry */
335                 link = next_slash + 1;
336                 /* advance past any other interim slashes.  we know we won't hit the end
337                  * due to the for loop check above */
338                 while (*link == '/')
339                         link++;
340         }
341         /* Now, we're on the last link of the path.  We need to deal with with . and
342          * .. .  This might be weird with PARENT lookups - not sure what semantics
343          * we want exactly.  This will give the parent of whatever the PATH was
344          * supposed to look like.  Note that ND currently points to the parent of
345          * the last item (link). */
346         if (!strcmp(".", link)) {
347                 if (nd->flags & LOOKUP_PARENT) {
348                         stash_nd_name(nd, nd->dentry->d_name.name);
349                         climb_up(nd);
350                 }
351                 return 0;
352         }
353         if (!strcmp("..", link)) {
354                 climb_up(nd);
355                 if (nd->flags & LOOKUP_PARENT) {
356                         stash_nd_name(nd, nd->dentry->d_name.name);
357                         climb_up(nd);
358                 }
359                 return 0;
360         }
361         /* need to attempt to look it up, in case it's a symlink */
362         link_dentry = do_lookup(nd->dentry, link);
363         if (!link_dentry) {
364                 /* if there's no dentry, we are okay if we are looking for the parent */
365                 if (nd->flags & LOOKUP_PARENT) {
366                         stash_nd_name(nd, link);
367                         return 0;
368                 } else {
369                         return -ENOENT;
370                 }
371         }
372         next_link(link_dentry, nd);
373         kref_put(&link_dentry->d_kref); /* do_lookup gave us a refcnt'd dentry */
374         /* at this point, nd is on the final link, but it might be a symlink */
375         if (nd->flags & LOOKUP_FOLLOW) {
376                 error = follow_symlink(nd);
377                 if (error < 0)
378                         return error;
379                 /* if we actually followed a symlink, then nd is set and we're done */
380                 if (error > 0)
381                         return 0;
382         }
383         /* One way or another, nd is on the last element of the path, symlinks and
384          * all.  Now we need to climb up to set nd back on the parent, if that's
385          * what we wanted */
386         if (nd->flags & LOOKUP_PARENT) {
387                 stash_nd_name(nd, link_dentry->d_name.name);
388                 climb_up(nd);
389                 return 0;
390         }
391         /* now, we have the dentry set, and don't want the parent, but might be on a
392          * mountpoint still.  FYI: this hasn't been thought through completely. */
393         follow_mount(nd);
394         /* If we wanted a directory, but didn't get one, error out */
395         if ((nd->flags & LOOKUP_DIRECTORY) &&
396            !(nd->dentry->d_inode->i_type & FS_I_DIR))
397                 return -ENOTDIR;
398         return 0;
399 }
400
401 /* Given path, return the inode for the final dentry.  The ND should be
402  * initialized for the first call - specifically, we need the intent. 
403  * LOOKUP_PARENT and friends go in the flags var, which is not the intent.
404  *
405  * Need to be careful too.  While the path has been copied-in to the kernel,
406  * it's still user input.  */
407 int path_lookup(char *path, int flags, struct nameidata *nd)
408 {
409         printd("Path lookup for %s\n", path);
410         /* we allow absolute lookups with no process context */
411         if (path[0] == '/') {                   /* absolute lookup */
412                 if (!current)
413                         nd->dentry = default_ns.root->mnt_root;
414                 else
415                         nd->dentry = current->fs_env.root;      
416         } else {                                                /* relative lookup */
417                 assert(current);
418                 /* Don't need to lock on the fs_env since we're reading one item */
419                 nd->dentry = current->fs_env.pwd;       
420         }
421         nd->mnt = nd->dentry->d_sb->s_mount;
422         /* Whenever references get put in the nd, incref them.  Whenever they are
423          * removed, decref them. */
424         kref_get(&nd->mnt->mnt_kref, 1);
425         kref_get(&nd->dentry->d_kref, 1);
426         nd->flags = flags;
427         nd->depth = 0;                                  /* used in symlink following */
428         return link_path_walk(path, nd);        
429 }
430
431 /* Call this after any use of path_lookup when you are done with its results,
432  * regardless of whether it succeeded or not.  It will free any references */
433 void path_release(struct nameidata *nd)
434 {
435         kref_put(&nd->dentry->d_kref);
436         kref_put(&nd->mnt->mnt_kref);
437         /* Free the last symlink dentry used, if there was one */
438         if (nd->last_sym) {
439                 kref_put(&nd->last_sym->d_kref);
440                 nd->last_sym = 0;                       /* catch reuse bugs */
441         }
442 }
443
444 /* Superblock functions */
445
446 /* Helper to alloc and initialize a generic superblock.  This handles all the
447  * VFS related things, like lists.  Each FS will need to handle its own things
448  * in it's *_get_sb(), usually involving reading off the disc. */
449 struct super_block *get_sb(void)
450 {
451         struct super_block *sb = kmalloc(sizeof(struct super_block), 0);
452         sb->s_dirty = FALSE;
453         spinlock_init(&sb->s_lock);
454         kref_init(&sb->s_kref, fake_release, 1); /* for the ref passed out */
455         TAILQ_INIT(&sb->s_inodes);
456         TAILQ_INIT(&sb->s_dirty_i);
457         TAILQ_INIT(&sb->s_io_wb);
458         SLIST_INIT(&sb->s_anon_d);
459         TAILQ_INIT(&sb->s_files);
460         sb->s_fs_info = 0; // can override somewhere else
461         return sb;
462 }
463
464 /* Final stages of initializing a super block, including creating and linking
465  * the root dentry, root inode, vmnt, and sb.  The d_op and root_ino are
466  * FS-specific, but otherwise it's FS-independent, tricky, and not worth having
467  * around multiple times.
468  *
469  * Not the world's best interface, so it's subject to change, esp since we're
470  * passing (now 3) FS-specific things. */
471 void init_sb(struct super_block *sb, struct vfsmount *vmnt,
472              struct dentry_operations *d_op, unsigned long root_ino,
473              void *d_fs_info)
474 {
475         /* Build and init the first dentry / inode.  The dentry ref is stored later
476          * by vfsmount's mnt_root.  The parent is dealt with later. */
477         struct dentry *d_root = get_dentry(sb, 0,  "/");        /* probably right */
478
479         /* a lot of here on down is normally done in lookup() or create, since
480          * get_dentry isn't a fully usable dentry.  The two FS-specific settings are
481          * normally inherited from a parent within the same FS in get_dentry, but we
482          * have none here. */
483         d_root->d_op = d_op;
484         d_root->d_fs_info = d_fs_info;
485         struct inode *inode = get_inode(d_root);
486         if (!inode)
487                 panic("This FS sucks!");
488         d_root->d_inode = inode;                                /* storing the inode's kref here */
489         TAILQ_INSERT_TAIL(&inode->i_dentry, d_root, d_alias);   /* weak ref */
490         inode->i_ino = root_ino;
491         /* TODO: add the inode to the appropriate list (off i_list) */
492         /* TODO: do we need to read in the inode?  can we do this on demand? */
493         /* if this FS is already mounted, we'll need to do something different. */
494         sb->s_op->read_inode(inode);
495         /* Link the dentry and SB to the VFS mount */
496         vmnt->mnt_root = d_root;                                /* ref comes from get_dentry */
497         vmnt->mnt_sb = sb;
498         /* If there is no mount point, there is no parent.  This is true only for
499          * the rootfs. */
500         if (vmnt->mnt_mountpoint) {
501                 kref_get(&vmnt->mnt_mountpoint->d_kref, 1);     /* held by d_root */
502                 d_root->d_parent = vmnt->mnt_mountpoint;        /* dentry of the root */
503         }
504         /* insert the dentry into the dentry cache.  when's the earliest we can?
505          * when's the earliest we should?  what about concurrent accesses to the
506          * same dentry?  should be locking the dentry... */
507         dcache_put(d_root); // TODO: should set a d_flag too
508 }
509
510 /* Dentry Functions */
511
512 /* Helper to alloc and initialize a generic dentry.  The following needs to be
513  * set still: d_op (if no parent), d_fs_info (opt), d_inode, connect the inode
514  * to the dentry (and up the d_kref again), maybe dcache_put().  The inode
515  * stitching is done in get_inode() or lookup (depending on the FS).
516  * The setting of the d_op might be problematic when dealing with mounts.  Just
517  * overwrite it.
518  *
519  * If the name is longer than the inline name, it will kmalloc a buffer, so
520  * don't worry about the storage for *name after calling this. */
521 struct dentry *get_dentry(struct super_block *sb, struct dentry *parent,
522                           char *name)
523 {
524         assert(name);
525         size_t name_len = strnlen(name, MAX_FILENAME_SZ);       /* not including \0! */
526         struct dentry *dentry = kmem_cache_alloc(dentry_kcache, 0);
527         char *l_name = 0;
528
529         if (!dentry)
530                 return 0;
531         //memset(dentry, 0, sizeof(struct dentry));
532         kref_init(&dentry->d_kref, dentry_release, 1);  /* this ref is returned */
533         spinlock_init(&dentry->d_lock);
534         TAILQ_INIT(&dentry->d_subdirs);
535         dentry->d_time = 0;
536         kref_get(&sb->s_kref, 1);
537         dentry->d_sb = sb;                                      /* storing a ref here... */
538         dentry->d_mount_point = FALSE;
539         dentry->d_mounted_fs = 0;
540         if (parent)     {                                               /* no parent for rootfs mount */
541                 kref_get(&parent->d_kref, 1);
542                 dentry->d_op = parent->d_op;    /* d_op set in init_sb for parentless */
543         }
544         dentry->d_parent = parent;
545         dentry->d_flags = 0;                            /* related to its dcache state */
546         dentry->d_fs_info = 0;
547         SLIST_INIT(&dentry->d_bucket);
548         if (name_len < DNAME_INLINE_LEN) {
549                 strncpy(dentry->d_iname, name, name_len);
550                 dentry->d_iname[name_len] = '\0';
551                 qstr_builder(dentry, 0);
552         } else {
553                 l_name = kmalloc(name_len + 1, 0);
554                 assert(l_name);
555                 strncpy(l_name, name, name_len);
556                 l_name[name_len] = '\0';
557                 qstr_builder(dentry, l_name);
558         }
559         /* Catch bugs by aggressively zeroing this (o/w we use old stuff) */
560         dentry->d_inode = 0;
561         return dentry;
562 }
563
564 /* Adds a dentry to the dcache. */
565 void dcache_put(struct dentry *dentry)
566 {
567 #if 0 /* pending a more thorough review of the dcache */
568         /* TODO: should set a d_flag too */
569         spin_lock(&dcache_lock);
570         SLIST_INSERT_HEAD(&dcache, dentry, d_hash);
571         spin_unlock(&dcache_lock);
572 #endif
573 }
574
575 /* Cleans up the dentry (after ref == 0).  We still may want it, and this is
576  * where we should add it to the dentry cache.  (TODO).  For now, we do nothing,
577  * since we don't have a dcache.
578  * 
579  * This has to handle two types of dentries: full ones (ones that had been used)
580  * and ones that had been just for lookups - hence the check for d_inode.
581  *
582  * Note that dentries pin and kref their inodes.  When all the dentries are
583  * gone, we want the inode to be released via kref.  The inode has internal /
584  * weak references to the dentry, which are not refcounted. */
585 void dentry_release(struct kref *kref)
586 {
587         struct dentry *dentry = container_of(kref, struct dentry, d_kref);
588         printd("Freeing dentry %08p: %s\n", dentry, dentry->d_name.name);
589         assert(dentry->d_op);   /* catch bugs.  a while back, some lacked d_op */
590         dentry->d_op->d_release(dentry);
591         /* TODO: check/test the boundaries on this. */
592         if (dentry->d_name.len > DNAME_INLINE_LEN)
593                 kfree((void*)dentry->d_name.name);
594         kref_put(&dentry->d_sb->s_kref);
595         if (dentry->d_mounted_fs)
596                 kref_put(&dentry->d_mounted_fs->mnt_kref);
597         if (dentry->d_inode) {
598                 TAILQ_REMOVE(&dentry->d_inode->i_dentry, dentry, d_alias);
599                 kref_put(&dentry->d_inode->i_kref);     /* but dentries kref inodes */
600         }
601         kmem_cache_free(dentry_kcache, dentry);
602 }
603
604 /* Looks up the dentry for the given path, returning a refcnt'd dentry (or 0).
605  * Permissions are applied for the current user, which is quite a broken system
606  * at the moment.  Flags are lookup flags. */
607 struct dentry *lookup_dentry(char *path, int flags)
608 {
609         struct dentry *dentry;
610         struct nameidata nd_r = {0}, *nd = &nd_r;
611         int error;
612
613         error = path_lookup(path, flags, nd);
614         if (error) {
615                 path_release(nd);
616                 set_errno(-error);
617                 return 0;
618         }
619         dentry = nd->dentry;
620         kref_get(&dentry->d_kref, 1);
621         path_release(nd);
622         return dentry;
623 }
624
625 /* Inode Functions */
626
627 /* Creates and initializes a new inode.  Generic fields are filled in.
628  * FS-specific fields are filled in by the callout.  Specific fields are filled
629  * in in read_inode() based on what's on the disk for a given i_no, or when the
630  * inode is created (for new objects).
631  *
632  * i_no is set by the caller.  Note that this means this inode can be for an
633  * inode that is already on disk, or it can be used when creating. */
634 struct inode *get_inode(struct dentry *dentry)
635 {
636         struct super_block *sb = dentry->d_sb;
637         /* FS allocs and sets the following: i_op, i_fop, i_pm.pm_op, and any FS
638          * specific stuff. */
639         struct inode *inode = sb->s_op->alloc_inode(sb);
640         if (!inode) {
641                 set_errno(ENOMEM);
642                 return 0;
643         }
644         TAILQ_INSERT_HEAD(&sb->s_inodes, inode, i_sb_list);             /* weak inode ref */
645         TAILQ_INIT(&inode->i_dentry);
646         TAILQ_INSERT_TAIL(&inode->i_dentry, dentry, d_alias);   /* weak dentry ref*/
647         /* one for the dentry->d_inode, one passed out */
648         kref_init(&inode->i_kref, inode_release, 2);
649         dentry->d_inode = inode;
650         inode->i_ino = 0;                                       /* set by caller later */
651         inode->i_blksize = sb->s_blocksize;
652         spinlock_init(&inode->i_lock);
653         inode->i_sb = sb;
654         inode->i_state = 0;                                     /* need real states, like I_NEW */
655         inode->dirtied_when = 0;
656         inode->i_flags = 0;
657         atomic_set(&inode->i_writecount, 0);
658         /* Set up the page_map structures.  Default is to use the embedded one.
659          * Might push some of this back into specific FSs.  For now, the FS tells us
660          * what pm_op they want via i_pm.pm_op, which we use when we point i_mapping
661          * to i_pm. */
662         inode->i_mapping = &inode->i_pm;
663         inode->i_mapping->pm_host = inode;
664         radix_tree_init(&inode->i_mapping->pm_tree);
665         spinlock_init(&inode->i_mapping->pm_tree_lock);
666         inode->i_mapping->pm_flags = 0;
667         return inode;
668 }
669
670 /* Helper op, used when creating regular files, directories, symlinks, etc.
671  * Note we make a distinction between the mode and the file type (for now).
672  * After calling this, call the FS specific version (create or mkdir), which
673  * will set the i_ino, the filetype, and do any other FS-specific stuff.  Also
674  * note that a lot of inode stuff was initialized in get_inode/alloc_inode.  The
675  * stuff here is pertinent to the specific creator (user), mode, and time.  Also
676  * note we don't pass this an nd, like Linux does... */
677 static struct inode *create_inode(struct dentry *dentry, int mode)
678 {
679         /* note it is the i_ino that uniquely identifies a file in the system.
680          * there's a diff between creating an inode (even for an in-use ino) and
681          * then filling it in, and vs creating a brand new one */
682         struct inode *inode = get_inode(dentry);
683         if (!inode)
684                 return 0;
685         inode->i_mode = mode;
686         inode->i_nlink = 1;
687         inode->i_size = 0;
688         inode->i_blocks = 0;
689         inode->i_atime.tv_sec = 0;              /* TODO: now! */
690         inode->i_ctime.tv_sec = 0;
691         inode->i_mtime.tv_sec = 0;
692         inode->i_atime.tv_nsec = 0;             /* are these supposed to be the extra ns? */
693         inode->i_ctime.tv_nsec = 0;
694         inode->i_mtime.tv_nsec = 0;
695         inode->i_bdev = inode->i_sb->s_bdev;
696         /* when we have notions of users, do something here: */
697         inode->i_uid = 0;
698         inode->i_gid = 0;
699         return inode;
700 }
701
702 /* Create a new disk inode in dir associated with dentry, with the given mode.
703  * called when creating a regular file.  dir is the directory/parent.  dentry is
704  * the dentry of the inode we are creating.  Note the lack of the nd... */
705 int create_file(struct inode *dir, struct dentry *dentry, int mode)
706 {
707         struct inode *new_file = create_inode(dentry, mode);
708         if (!new_file)
709                 return -1;
710         dir->i_op->create(dir, dentry, mode, 0);
711         kref_put(&new_file->i_kref);
712         return 0;
713 }
714
715 /* Creates a new inode for a directory associated with dentry in dir with the
716  * given mode. */
717 int create_dir(struct inode *dir, struct dentry *dentry, int mode)
718 {
719         struct inode *new_dir = create_inode(dentry, mode);
720         if (!new_dir)
721                 return -1;
722         dir->i_op->mkdir(dir, dentry, mode);
723         /* Make sure my parent tracks me.  This is okay, since no directory (dir)
724          * can have more than one dentry */
725         struct dentry *parent = TAILQ_FIRST(&dir->i_dentry);
726         assert(parent && parent == TAILQ_LAST(&dir->i_dentry, dentry_tailq));
727         /* parent dentry tracks dentry as a subdir, weak reference */
728         TAILQ_INSERT_TAIL(&parent->d_subdirs, dentry, d_subdirs_link);
729         kref_put(&new_dir->i_kref);
730         return 0;
731 }
732
733 /* Creates a new inode for a symlink associated with dentry in dir, containing
734  * the symlink symname */
735 int create_symlink(struct inode *dir, struct dentry *dentry,
736                    const char *symname, int mode)
737 {
738         struct inode *new_sym = create_inode(dentry, mode);
739         if (!new_sym)
740                 return -1;
741         dir->i_op->symlink(dir, dentry, symname);
742         kref_put(&new_sym->i_kref);
743         return 0;
744 }
745
746 /* Returns 0 if the given mode is acceptable for the inode, and an appropriate
747  * error code if not.  Needs to be writen, based on some sensible rules, and
748  * will also probably use 'current' */
749 int check_perms(struct inode *inode, int access_mode)
750 {
751         return 0;       /* anything goes! */
752 }
753
754 /* Called after all external refs are gone to clean up the inode.  Once this is
755  * called, all dentries pointing here are already done (one of them triggered
756  * this via kref_put(). */
757 void inode_release(struct kref *kref)
758 {
759         struct inode *inode = container_of(kref, struct inode, i_kref);
760         inode->i_sb->s_op->destroy_inode(inode);
761         kref_put(&inode->i_sb->s_kref);
762         assert(inode->i_mapping == &inode->i_pm);
763         kmem_cache_free(inode_kcache, inode);
764         /* TODO: (BDEV) */
765         // kref_put(inode->i_bdev->kref); /* assuming it's a bdev */
766 }
767
768 /* Fills in kstat with the stat information for the inode */
769 void stat_inode(struct inode *inode, struct kstat *kstat)
770 {
771         kstat->st_dev = inode->i_sb->s_dev;
772         kstat->st_ino = inode->i_ino;
773         kstat->st_mode = inode->i_mode;
774         kstat->st_nlink = inode->i_nlink;
775         kstat->st_uid = inode->i_uid;
776         kstat->st_gid = inode->i_gid;
777         kstat->st_rdev = inode->i_rdev;
778         kstat->st_size = inode->i_size;
779         kstat->st_blksize = inode->i_blksize;
780         kstat->st_blocks = inode->i_blocks;
781         kstat->st_atime = inode->i_atime;
782         kstat->st_mtime = inode->i_mtime;
783         kstat->st_ctime = inode->i_ctime;
784 }
785
786 /* File functions */
787
788 /* Read count bytes from the file into buf, starting at *offset, which is increased
789  * accordingly, returning the number of bytes transfered.  Most filesystems will
790  * use this function for their f_op->read.  Note, this uses the page cache.
791  * Want to try out page remapping later on... */
792 ssize_t generic_file_read(struct file *file, char *buf, size_t count,
793                           off_t *offset)
794 {
795         struct page *page;
796         int error;
797         off_t page_off;
798         unsigned long first_idx, last_idx;
799         size_t copy_amt;
800         char *buf_end;
801
802         /* Consider pushing some error checking higher in the VFS */
803         if (!count)
804                 return 0;
805         if (*offset == file->f_dentry->d_inode->i_size)
806                 return 0; /* EOF */
807         /* Make sure we don't go past the end of the file */
808         if (*offset + count > file->f_dentry->d_inode->i_size) {
809                 count = file->f_dentry->d_inode->i_size - *offset;
810         }
811         page_off = *offset & (PGSIZE - 1);
812         first_idx = *offset >> PGSHIFT;
813         last_idx = (*offset + count) >> PGSHIFT;
814         buf_end = buf + count;
815         /* For each file page, make sure it's in the page cache, then copy it out.
816          * TODO: will probably need to consider concurrently truncated files here.*/
817         for (int i = first_idx; i <= last_idx; i++) {
818                 error = file_load_page(file, i, &page);
819                 assert(!error); /* TODO: handle ENOMEM and friends */
820                 copy_amt = MIN(PGSIZE - page_off, buf_end - buf);
821                 /* TODO: (UMEM) think about this.  if it's a user buffer, we're relying
822                  * on current to detect whose it is (which should work for async calls).
823                  * Also, need to propagate errors properly...  Probably should do a
824                  * user_mem_check, then free, and also to make a distinction between
825                  * when the kernel wants a read/write (TODO: KFOP) */
826                 if (current) {
827                         memcpy_to_user(current, buf, page2kva(page) + page_off, copy_amt);
828                 } else {
829                         memcpy(buf, page2kva(page) + page_off, copy_amt);
830                 }
831                 buf += copy_amt;
832                 page_off = 0;
833                 page_decref(page);      /* it's still in the cache, we just don't need it */
834         }
835         assert(buf == buf_end);
836         *offset += count;
837         return count;
838 }
839
840 /* Write count bytes from buf to the file, starting at *offset, which is increased
841  * accordingly, returning the number of bytes transfered.  Most filesystems will
842  * use this function for their f_op->write.  Note, this uses the page cache.
843  * Changes don't get flushed to disc til there is an fsync, page cache eviction,
844  * or other means of trying to writeback the pages. */
845 ssize_t generic_file_write(struct file *file, const char *buf, size_t count,
846                            off_t *offset)
847 {
848         struct page *page;
849         int error;
850         off_t page_off;
851         unsigned long first_idx, last_idx;
852         size_t copy_amt;
853         const char *buf_end;
854
855         /* Consider pushing some error checking higher in the VFS */
856         if (!count)
857                 return 0;
858         /* Extend the file.  Should put more checks in here, and maybe do this per
859          * page in the for loop below. */
860         if (*offset + count > file->f_dentry->d_inode->i_size)
861                 file->f_dentry->d_inode->i_size = *offset + count;
862         page_off = *offset & (PGSIZE - 1);
863         first_idx = *offset >> PGSHIFT;
864         last_idx = (*offset + count) >> PGSHIFT;
865         buf_end = buf + count;
866         /* For each file page, make sure it's in the page cache, then write it.*/
867         for (int i = first_idx; i <= last_idx; i++) {
868                 error = file_load_page(file, i, &page);
869                 assert(!error); /* TODO: handle ENOMEM and friends */
870                 copy_amt = MIN(PGSIZE - page_off, buf_end - buf);
871                 /* TODO: (UMEM) (KFOP) think about this.  if it's a user buffer, we're
872                  * relying on current to detect whose it is (which should work for async
873                  * calls). */
874                 if (current) {
875                         memcpy_from_user(current, page2kva(page) + page_off, buf, copy_amt);
876                 } else {
877                         memcpy(page2kva(page) + page_off, buf, copy_amt);
878                 }
879                 buf += copy_amt;
880                 page_off = 0;
881                 page_decref(page);      /* it's still in the cache, we just don't need it */
882         }
883         assert(buf == buf_end);
884         *offset += count;
885         return count;
886 }
887
888 /* Directories usually use this for their read method, which is the way glibc
889  * currently expects us to do a readdir (short of doing linux's getdents).  Will
890  * probably need work, based on whatever real programs want. */
891 ssize_t generic_dir_read(struct file *file, char *u_buf, size_t count,
892                          off_t *offset)
893 {
894         struct kdirent dir_r = {0}, *dirent = &dir_r;
895         unsigned int num_dirents = count / sizeof(struct kdirent);
896         int retval = 1;
897         size_t amt_copied = 0;
898         char *buf_end = u_buf + count;
899
900         if (!count)
901                 return 0;
902         if (*offset % sizeof(struct kdirent)) {
903                 printk("[kernel] the f_pos for a directory should be dirent-aligned\n");
904                 set_errno(EINVAL);
905                 return -1;
906         }
907         /* for now, we need to tell readdir which dirent we want */
908         dirent->d_off = *offset / sizeof(struct kdirent);
909         for (; (u_buf < buf_end) && (retval == 1); u_buf += sizeof(struct kdirent)){
910                 /* TODO: UMEM/KFOP (pin the u_buf in the syscall, ditch the local copy,
911                  * get rid of this memcpy and reliance on current, etc).  Might be
912                  * tricky with the dirent->d_off */
913                 retval = file->f_op->readdir(file, dirent);
914                 if (retval < 0)
915                         break;
916                 if (current) {
917                         memcpy_to_user(current, u_buf, dirent, sizeof(struct dirent));
918                 } else {
919                         memcpy(u_buf, dirent, sizeof(struct dirent));
920                 }
921                 amt_copied += sizeof(struct dirent);
922                 dirent->d_off++;
923         }
924         *offset += amt_copied;
925         return amt_copied;
926 }
927
928 /* Opens the file, using permissions from current for lack of a better option.
929  * It will attempt to create the file if it does not exist and O_CREAT is
930  * specified.  This will return 0 on failure, and set errno.  TODO: There's some
931  * stuff that we don't do, esp related file truncating/creation.  flags are for
932  * opening, the mode is for creating.  The flags related to how to create
933  * (O_CREAT_FLAGS) are handled in this function, not in create_file().
934  *
935  * It's tempting to split this into a do_file_create and a do_file_open, based
936  * on the O_CREAT flag, but the O_CREAT flag can be ignored if the file exists
937  * already and O_EXCL isn't specified.  We could have open call create if it
938  * fails, but for now we'll keep it as is. */
939 struct file *do_file_open(char *path, int flags, int mode)
940 {
941         struct file *file = 0;
942         struct dentry *file_d;
943         struct inode *parent_i;
944         struct nameidata nd_r = {0}, *nd = &nd_r;
945         int error;
946
947         /* this isn't quite right, due to the nature of O_CREAT */
948         if (flags & O_CREAT)
949                 nd->intent = LOOKUP_CREATE;
950         else
951                 nd->intent = LOOKUP_OPEN;
952         /* get the parent, following links.  this means you get the parent of the
953          * final link (which may not be in 'path' in the first place. */
954         error = path_lookup(path, LOOKUP_PARENT | LOOKUP_FOLLOW, nd);
955         if (error) {
956                 path_release(nd);
957                 set_errno(-error);
958                 return 0;
959         }
960         /* see if the target is there, handle accordingly */
961         file_d = do_lookup(nd->dentry, nd->last.name); 
962         if (!file_d) {
963                 if (!(flags & O_CREAT)) {
964                         path_release(nd);
965                         set_errno(ENOENT);
966                         return 0;
967                 }
968                 /* Create the inode/file.  get a fresh dentry too: */
969                 file_d = get_dentry(nd->dentry->d_sb, nd->dentry, nd->last.name);
970                 parent_i = nd->dentry->d_inode;
971                 /* TODO: mode should be & ~umask.  Note that mode technically should
972                  * only apply to future opens, though we apply it immediately. */
973                 if (create_file(parent_i, file_d, mode)) {
974                         kref_put(&file_d->d_kref);
975                         path_release(nd);
976                         return 0;
977                 }
978                 dcache_put(file_d);
979         } else {        /* something already exists (might be a dir) */
980                 if ((flags & O_CREAT) && (flags & O_EXCL)) {
981                         /* wanted to create, not open, bail out */
982                         kref_put(&file_d->d_kref);
983                         path_release(nd);
984                         set_errno(EEXIST);
985                         return 0;
986                 }
987         }
988         /* now open the file (freshly created or if it already existed).  At this
989          * point, file_d is a refcnt'd dentry, regardless of which branch we took.*/
990         if (flags & O_TRUNC)
991                 warn("File truncation not supported yet.");
992         file = dentry_open(file_d, flags);              /* sets errno */
993         if (!file) {
994                 kref_put(&file_d->d_kref);
995                 path_release(nd);
996                 return 0;
997         }
998         kref_put(&file_d->d_kref);
999         path_release(nd);
1000         return file;
1001 }
1002
1003 /* Path is the location of the symlink, sometimes called the "new path", and
1004  * symname is who we link to, sometimes called the "old path". */
1005 int do_symlink(char *path, const char *symname, int mode)
1006 {
1007         struct dentry *sym_d;
1008         struct inode *parent_i;
1009         struct nameidata nd_r = {0}, *nd = &nd_r;
1010         int error;
1011
1012         nd->intent = LOOKUP_CREATE;
1013         /* get the parent, but don't follow links */
1014         error = path_lookup(path, LOOKUP_PARENT, nd);
1015         if (error) {
1016                 set_errno(-error);
1017                 path_release(nd);
1018                 return -1;
1019         }
1020         /* see if the target is already there, handle accordingly */
1021         sym_d = do_lookup(nd->dentry, nd->last.name); 
1022         if (sym_d) {
1023                 set_errno(EEXIST);
1024                 kref_put(&sym_d->d_kref);
1025                 path_release(nd);
1026                 return -1;
1027         }
1028         /* Doesn't already exist, let's try to make it: */
1029         sym_d = get_dentry(nd->dentry->d_sb, nd->dentry, nd->last.name);
1030         if (!sym_d) {
1031                 set_errno(ENOMEM);
1032                 path_release(nd);
1033                 return -1;
1034         }
1035         parent_i = nd->dentry->d_inode;
1036         /* TODO: mode should be & ~umask. */
1037         if (create_symlink(parent_i, sym_d, symname, mode)) {
1038                 kref_put(&sym_d->d_kref);
1039                 path_release(nd);
1040                 return -1;
1041         }
1042         dcache_put(sym_d);
1043         kref_put(&sym_d->d_kref);
1044         path_release(nd);
1045         return 0;
1046 }
1047
1048 /* Makes a hard link for the file behind old_path to new_path */
1049 int do_link(char *old_path, char *new_path)
1050 {
1051         struct dentry *link_d, *old_d;
1052         struct inode *inode, *parent_dir;
1053         struct nameidata nd_r = {0}, *nd = &nd_r;
1054         int error;
1055         int retval = -1;
1056
1057         nd->intent = LOOKUP_CREATE;
1058         /* get the absolute parent of the new_path */
1059         error = path_lookup(new_path, LOOKUP_PARENT | LOOKUP_FOLLOW, nd);
1060         if (error) {
1061                 set_errno(-error);
1062                 goto out_path_only;
1063         }
1064         parent_dir = nd->dentry->d_inode;
1065         /* see if the new target is already there, handle accordingly */
1066         link_d = do_lookup(nd->dentry, nd->last.name); 
1067         if (link_d) {
1068                 set_errno(EEXIST);
1069                 goto out_link_d;
1070         }
1071         /* Doesn't already exist, let's try to make it.  Still need to stitch it to
1072          * an inode and set its FS-specific stuff after this.*/
1073         link_d = get_dentry(nd->dentry->d_sb, nd->dentry, nd->last.name);
1074         if (!link_d) {
1075                 set_errno(ENOMEM);
1076                 goto out_path_only;
1077         }
1078         /* Now let's get the old_path target */
1079         old_d = lookup_dentry(old_path, LOOKUP_FOLLOW);
1080         if (!old_d)                                     /* errno set by lookup_dentry */
1081                 goto out_link_d;
1082         /* For now, can only link to files */
1083         if (old_d->d_inode->i_type != FS_I_FILE) {
1084                 set_errno(EPERM);
1085                 goto out_both_ds;
1086         }
1087         /* Must be on the same FS */
1088         if (old_d->d_sb != link_d->d_sb) {
1089                 set_errno(EXDEV);
1090                 goto out_both_ds;
1091         }
1092         /* Do whatever FS specific stuff there is first (which is also a chance to
1093          * bail out). */
1094         error = parent_dir->i_op->link(old_d, parent_dir, link_d);
1095         if (error) {
1096                 set_errno(-error);
1097                 goto out_both_ds;
1098         }
1099         /* Finally stitch it up */
1100         inode = old_d->d_inode;
1101         kref_get(&inode->i_kref, 1);
1102         link_d->d_inode = inode;
1103         inode->i_nlink++;
1104         TAILQ_INSERT_TAIL(&inode->i_dentry, link_d, d_alias);   /* weak ref */
1105         dcache_put(link_d);
1106         retval = 0;                             /* Note the fall through to the exit paths */
1107 out_both_ds:
1108         kref_put(&old_d->d_kref);
1109 out_link_d:
1110         kref_put(&link_d->d_kref);
1111 out_path_only:
1112         path_release(nd);
1113         return retval;
1114 }
1115
1116 /* Checks to see if path can be accessed via mode.  Doesn't do much now.  This
1117  * is an example of decent error propagation from the lower levels via int
1118  * retvals. */
1119 int do_file_access(char *path, int mode)
1120 {
1121         struct nameidata nd_r = {0}, *nd = &nd_r;
1122         int retval = 0;
1123         nd->intent = LOOKUP_ACCESS;
1124         retval = path_lookup(path, 0, nd);
1125         path_release(nd);       
1126         return retval;
1127 }
1128
1129 /* Opens and returns the file specified by dentry */
1130 struct file *dentry_open(struct dentry *dentry, int flags)
1131 {
1132         struct inode *inode;
1133         int desired_mode;
1134         struct file *file = kmem_cache_alloc(file_kcache, 0);
1135         if (!file) {
1136                 set_errno(ENOMEM);
1137                 return 0;
1138         }
1139         inode = dentry->d_inode;
1140         /* Do the mode first, since we can still error out.  f_mode stores how the
1141          * OS file is open, which can be more restrictive than the i_mode */
1142         switch (flags & (O_RDONLY | O_WRONLY | O_RDWR)) {
1143                 case O_RDONLY:
1144                         desired_mode = S_IRUSR;
1145                         break;
1146                 case O_WRONLY:
1147                         desired_mode = S_IWUSR;
1148                         break;
1149                 case O_RDWR:
1150                         desired_mode = S_IRUSR | S_IWUSR;
1151                         break;
1152                 default:
1153                         goto error_access;
1154         }
1155         if (check_perms(inode, desired_mode))
1156                 goto error_access;
1157         file->f_mode = desired_mode;
1158         /* one for the ref passed out, and *none* for the sb TAILQ */
1159         kref_init(&file->f_kref, file_release, 1);
1160         /* Add to the list of all files of this SB */
1161         TAILQ_INSERT_TAIL(&inode->i_sb->s_files, file, f_list);
1162         kref_get(&dentry->d_kref, 1);
1163         file->f_dentry = dentry;
1164         kref_get(&inode->i_sb->s_mount->mnt_kref, 1);
1165         file->f_vfsmnt = inode->i_sb->s_mount;          /* saving a ref to the vmnt...*/
1166         file->f_op = inode->i_fop;
1167         /* Don't store open mode or creation flags */
1168         file->f_flags = flags & ~(O_ACCMODE | O_CREAT_FLAGS);
1169         file->f_pos = 0;
1170         file->f_uid = inode->i_uid;
1171         file->f_gid = inode->i_gid;
1172         file->f_error = 0;
1173 //      struct event_poll_tailq         f_ep_links;
1174         spinlock_init(&file->f_ep_lock);
1175         file->f_fs_info = 0;                                            /* prob overriden by the fs */
1176         file->f_mapping = inode->i_mapping;
1177         file->f_op->open(inode, file);
1178         return file;
1179 error_access:
1180         set_errno(EACCES);
1181         kmem_cache_free(file_kcache, file);
1182         return 0;
1183 }
1184
1185 /* Closes a file, fsync, whatever else is necessary.  Called when the kref hits
1186  * 0.  Note that the file is not refcounted on the s_files list, nor is the
1187  * f_mapping refcounted (it is pinned by the i_mapping). */
1188 void file_release(struct kref *kref)
1189 {
1190         struct file *file = container_of(kref, struct file, f_kref);
1191
1192         struct super_block *sb = file->f_dentry->d_sb;
1193         spin_lock(&sb->s_lock);
1194         TAILQ_REMOVE(&sb->s_files, file, f_list);
1195         spin_unlock(&sb->s_lock);
1196
1197         /* TODO: fsync (BLK).  also, we may want to parallelize the blocking that
1198          * could happen in here (spawn kernel threads)... */
1199         file->f_op->release(file->f_dentry->d_inode, file);
1200         /* Clean up the other refs we hold */
1201         kref_put(&file->f_dentry->d_kref);
1202         kref_put(&file->f_vfsmnt->mnt_kref);
1203         kmem_cache_free(file_kcache, file);
1204 }
1205
1206 /* Page cache functions */
1207
1208 /* Looks up the index'th page in the page map, returning an incref'd reference,
1209  * or 0 if it was not in the map. */
1210 struct page *pm_find_page(struct page_map *pm, unsigned long index)
1211 {
1212         spin_lock(&pm->pm_tree_lock);
1213         struct page *page = (struct page*)radix_lookup(&pm->pm_tree, index);
1214         if (page)
1215                 page_incref(page);
1216         spin_unlock(&pm->pm_tree_lock);
1217         return page;
1218 }
1219
1220 /* Attempts to insert the page into the page_map, returns 0 for success, or an
1221  * error code if there was one already (EEXIST) or we ran out of memory
1222  * (ENOMEM).  On success, this will preemptively lock the page, and will also
1223  * store a reference to the page in the pm. */
1224 int pm_insert_page(struct page_map *pm, unsigned long index, struct page *page)
1225 {
1226         int error = 0;
1227         spin_lock(&pm->pm_tree_lock);
1228         error = radix_insert(&pm->pm_tree, index, page);
1229         if (!error) {
1230                 page_incref(page);
1231                 page->pg_flags |= PG_LOCKED;
1232                 page->pg_mapping = pm;
1233                 page->pg_index = index;
1234                 pm->pm_num_pages++;
1235         }
1236         spin_unlock(&pm->pm_tree_lock);
1237         return error;
1238 }
1239
1240 /* Removes the page, including its reference.  Not sure yet what interface we
1241  * want to this (pm and index or page), and this has never been used.  There are
1242  * also issues with when you want to call this, since a page in the cache may be
1243  * mmap'd by someone else. */
1244 int pm_remove_page(struct page_map *pm, struct page *page)
1245 {
1246         void *retval;
1247         warn("pm_remove_page() hasn't been thought through or tested.");
1248         spin_lock(&pm->pm_tree_lock);
1249         retval = radix_delete(&pm->pm_tree, page->pg_index);
1250         spin_unlock(&pm->pm_tree_lock);
1251         assert(retval == (void*)page);
1252         page_decref(page);
1253         page->pg_mapping = 0;
1254         page->pg_index = 0;
1255         pm->pm_num_pages--;
1256         return 0;
1257 }
1258
1259 /* Makes sure the index'th page from file is loaded in the page cache and
1260  * returns its location via **pp.  Note this will give you a refcnt'd reference.
1261  * This may block! TODO: (BLK) */
1262 int file_load_page(struct file *file, unsigned long index, struct page **pp)
1263 {
1264         struct page_map *pm = file->f_mapping;
1265         struct page *page;
1266         int error;
1267         bool page_was_mapped = TRUE;
1268
1269         page = pm_find_page(pm, index);
1270         while (!page) {
1271                 /* kpage_alloc, since we want the page to persist after the proc
1272                  * dies (can be used by others, until the inode shuts down). */
1273                 if (kpage_alloc(&page))
1274                         return -ENOMEM;
1275                 /* might want to initialize other things, perhaps in page_alloc() */
1276                 page->pg_flags = 0;
1277                 error = pm_insert_page(pm, index, page);
1278                 switch (error) {
1279                         case 0:
1280                                 page_was_mapped = FALSE;
1281                                 break;
1282                         case -EEXIST:
1283                                 /* the page was mapped already (benign race), just get rid of
1284                                  * our page and try again (the only case that uses the while) */
1285                                 page_decref(page);
1286                                 page = pm_find_page(pm, index);
1287                                 break;
1288                         default:
1289                                 /* something is wrong, bail out! */
1290                                 page_decref(page);
1291                                 return error;
1292                 }
1293         }
1294         *pp = page;
1295         /* if the page was in the map, we need to do some checks, and might have to
1296          * read in the page later.  If the page was freshly inserted to the pm by
1297          * us, we skip this since we are the one doing the readpage(). */
1298         if (page_was_mapped) {
1299                 /* is it already here and up to date?  if so, we're done */
1300                 if (page->pg_flags & PG_UPTODATE)
1301                         return 0;
1302                 /* if not, try to lock the page (could BLOCK) */
1303                 lock_page(page);
1304                 /* we got it, is our page still in the cache?  check the mapping.  if
1305                  * not, start over, perhaps with EAGAIN and outside support */
1306                 if (!page->pg_mapping)
1307                         panic("Page is not in the mapping!  Haven't implemented this!");
1308                 /* double check, are we up to date?  if so, we're done */
1309                 if (page->pg_flags & PG_UPTODATE) {
1310                         unlock_page(page);
1311                         return 0;
1312                 }
1313         }
1314         /* if we're here, the page is locked by us, and it needs to be read in */
1315         assert(page->pg_mapping == pm);
1316         error = pm->pm_op->readpage(file, page);
1317         assert(!error);
1318         /* Try to sleep on the IO.  The page will be unlocked when the IO is done */
1319         lock_page(page);
1320         unlock_page(page);
1321         assert(page->pg_flags & PG_UPTODATE);
1322         return 0;
1323 }
1324
1325 /* Process-related File management functions */
1326
1327 /* Given any FD, get the appropriate file, 0 o/w */
1328 struct file *get_file_from_fd(struct files_struct *open_files, int file_desc)
1329 {
1330         struct file *retval = 0;
1331         if (file_desc < 0)
1332                 return 0;
1333         spin_lock(&open_files->lock);
1334         if (file_desc < open_files->max_fdset) {
1335                 if (GET_BITMASK_BIT(open_files->open_fds->fds_bits, file_desc)) {
1336                         /* while max_files and max_fdset might not line up, we should never
1337                          * have a valid fdset higher than files */
1338                         assert(file_desc < open_files->max_files);
1339                         retval = open_files->fd[file_desc];
1340                         assert(retval);
1341                         kref_get(&retval->f_kref, 1);
1342                 }
1343         }
1344         spin_unlock(&open_files->lock);
1345         return retval;
1346 }
1347
1348 /* Remove FD from the open files, if it was there, and return f.  Currently,
1349  * this decref's f, so the return value is not consumable or even usable.  This
1350  * hasn't been thought through yet. */
1351 struct file *put_file_from_fd(struct files_struct *open_files, int file_desc)
1352 {
1353         struct file *file = 0;
1354         if (file_desc < 0)
1355                 return 0;
1356         spin_lock(&open_files->lock);
1357         if (file_desc < open_files->max_fdset) {
1358                 if (GET_BITMASK_BIT(open_files->open_fds->fds_bits, file_desc)) {
1359                         /* while max_files and max_fdset might not line up, we should never
1360                          * have a valid fdset higher than files */
1361                         assert(file_desc < open_files->max_files);
1362                         file = open_files->fd[file_desc];
1363                         open_files->fd[file_desc] = 0;
1364                         CLR_BITMASK_BIT(open_files->open_fds->fds_bits, file_desc);
1365                         /* the if case is due to files (stdin) without a *file yet */
1366                         if (file)
1367                                 kref_put(&file->f_kref);
1368                 }
1369         }
1370         spin_unlock(&open_files->lock);
1371         return file;
1372 }
1373
1374 /* Inserts the file in the files_struct, returning the corresponding new file
1375  * descriptor, or an error code.  We currently grab the first open FD. */
1376 int insert_file(struct files_struct *open_files, struct file *file)
1377 {
1378         int slot = -1;
1379         spin_lock(&open_files->lock);
1380         for (int i = 0; i < open_files->max_fdset; i++) {
1381                 if (GET_BITMASK_BIT(open_files->open_fds->fds_bits, i))
1382                         continue;
1383                 slot = i;
1384                 SET_BITMASK_BIT(open_files->open_fds->fds_bits, slot);
1385                 assert(slot < open_files->max_files && open_files->fd[slot] == 0);
1386                 kref_get(&file->f_kref, 1);
1387                 open_files->fd[slot] = file;
1388                 if (slot >= open_files->next_fd)
1389                         open_files->next_fd = slot + 1;
1390                 break;
1391         }
1392         if (slot == -1) /* should expand the FD array and fd_set */
1393                 warn("Ran out of file descriptors, deal with me!");
1394         spin_unlock(&open_files->lock);
1395         return slot;
1396 }
1397
1398 /* Closes all open files.  Mostly just a "put" for all files.  If cloexec, it
1399  * will only close files that are opened with O_CLOEXEC. */
1400 void close_all_files(struct files_struct *open_files, bool cloexec)
1401 {
1402         struct file *file;
1403         spin_lock(&open_files->lock);
1404         for (int i = 0; i < open_files->max_fdset; i++) {
1405                 if (GET_BITMASK_BIT(open_files->open_fds->fds_bits, i)) {
1406                         /* while max_files and max_fdset might not line up, we should never
1407                          * have a valid fdset higher than files */
1408                         assert(i < open_files->max_files);
1409                         file = open_files->fd[i];
1410                         if (cloexec && !(file->f_flags | O_CLOEXEC))
1411                                 continue;
1412                         open_files->fd[i] = 0;
1413                         /* the if case is due to files (stdin) without a *file yet */
1414                         if (file)
1415                                 kref_put(&file->f_kref);
1416                         CLR_BITMASK_BIT(open_files->open_fds->fds_bits, i);
1417                 }
1418         }
1419         spin_unlock(&open_files->lock);
1420 }
1421
1422 /* Inserts all of the files from src into dst, used by sys_fork(). */
1423 void clone_files(struct files_struct *src, struct files_struct *dst)
1424 {
1425         struct file *file;
1426         spin_lock(&src->lock);
1427         spin_lock(&dst->lock);
1428         for (int i = 0; i < src->max_fdset; i++) {
1429                 if (GET_BITMASK_BIT(src->open_fds->fds_bits, i)) {
1430                         /* while max_files and max_fdset might not line up, we should never
1431                          * have a valid fdset higher than files */
1432                         assert(i < src->max_files);
1433                         file = src->fd[i];
1434                         SET_BITMASK_BIT(dst->open_fds->fds_bits, i);
1435                         assert(i < dst->max_files && dst->fd[i] == 0);
1436                         dst->fd[i] = file;
1437                         /* the if case is due to files (stdin) without a *file yet */
1438                         if (file)
1439                                 kref_get(&file->f_kref, 1);
1440                 }
1441         }
1442         spin_unlock(&dst->lock);
1443         spin_unlock(&src->lock);
1444 }
1445
1446 static void print_dir(struct dentry *dentry, char *buf, int depth)
1447 {
1448         struct dentry *child_d;
1449         struct dirent next;
1450         struct file *dir;
1451         int retval;
1452         int child_num = 0;
1453
1454         if (!dentry->d_inode->i_type & FS_I_DIR) {
1455                 warn("Thought this was only directories!!");
1456                 return;
1457         }
1458         /* Print this dentry */
1459         printk("%s%s/\n", buf, dentry->d_name.name);
1460         if (depth >= 32)
1461                 return;
1462         /* Set buffer for our kids */
1463         buf[depth] = '\t';
1464         dir = dentry_open(dentry, 0);
1465         if (!dir)
1466                 panic("Filesystem seems inconsistent - unable to open a dir!");
1467         /* Process every child, recursing on directories */
1468         while (1) {
1469                 next.d_off = child_num++;
1470                 retval = dir->f_op->readdir(dir, &next);
1471                 if (retval >= 0) {
1472                         /* there is an entry, now get its dentry */
1473                         child_d = do_lookup(dentry, next.d_name);
1474                         if (!child_d)
1475                                 panic("Inconsistent FS, dirent doesn't have a dentry!");
1476                         /* Recurse for directories, or just print the name for others */
1477                         switch (child_d->d_inode->i_type) {
1478                                 case (FS_I_DIR):
1479                                         print_dir(child_d, buf, depth + 1);
1480                                         break;
1481                                 case (FS_I_FILE):
1482                                         printk("%s%s size(B): %d\n", buf, next.d_name,
1483                                                child_d->d_inode->i_size);
1484                                         break;
1485                                 case (FS_I_SYMLINK):
1486                                         printk("%s%s -> %s\n", buf, next.d_name,
1487                                                child_d->d_inode->i_op->readlink(child_d));
1488                                         break;
1489                                 default:
1490                                         warn("Look around you!  Unknown filetype!");
1491                         }
1492                         kref_put(&child_d->d_kref);     
1493                 }
1494                 if (retval <= 0)
1495                         break;
1496         }
1497         /* Reset buffer to the way it was */
1498         buf[depth] = '\0';
1499         kref_put(&dir->f_kref);
1500 }
1501
1502 /* Debugging */
1503 int ls_dash_r(char *path)
1504 {
1505         struct nameidata nd_r = {0}, *nd = &nd_r;
1506         int error;
1507         char buf[32] = {0};
1508
1509         error = path_lookup(path, LOOKUP_ACCESS | LOOKUP_DIRECTORY, nd);
1510         if (error) {
1511                 path_release(nd);
1512                 return error;
1513         }
1514         print_dir(nd->dentry, buf, 0);
1515         path_release(nd);
1516         return 0;
1517 }