Fork and exec handle files better
[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
18 struct sb_tailq super_blocks = TAILQ_HEAD_INITIALIZER(super_blocks);
19 spinlock_t super_blocks_lock = SPINLOCK_INITIALIZER;
20 struct fs_type_tailq file_systems = TAILQ_HEAD_INITIALIZER(file_systems);
21 struct namespace default_ns;
22 // TODO: temp dcache, holds all dentries ever for now
23 struct dentry_slist dcache = SLIST_HEAD_INITIALIZER(dcache);
24 spinlock_t dcache_lock = SPINLOCK_INITIALIZER;
25
26 struct kmem_cache *dentry_kcache; // not to be confused with the dcache
27 struct kmem_cache *inode_kcache;
28 struct kmem_cache *file_kcache;
29
30 /* Mounts fs from dev_name at mnt_pt in namespace ns.  There could be no mnt_pt,
31  * such as with the root of (the default) namespace.  Not sure how it would work
32  * with multiple namespaces on the same FS yet.  Note if you mount the same FS
33  * multiple times, you only have one FS still (and one SB).  If we ever support
34  * that... */
35 struct vfsmount *mount_fs(struct fs_type *fs, char *dev_name,
36                           struct dentry *mnt_pt, int flags,
37                           struct namespace *ns)
38 {
39         struct super_block *sb;
40         struct vfsmount *vmnt = kmalloc(sizeof(struct vfsmount), 0);
41
42         /* this first ref is stored in the NS tailq below */
43         kref_init(&vmnt->mnt_kref, fake_release, 1);
44         /* Build the vfsmount, if there is no mnt_pt, mnt is the root vfsmount (for now).
45          * fields related to the actual FS, like the sb and the mnt_root are set in
46          * the fs-specific get_sb() call. */
47         if (!mnt_pt) {
48                 vmnt->mnt_parent = NULL;
49                 vmnt->mnt_mountpoint = NULL;
50         } else { /* common case, but won't be tested til we try to mount another FS */
51                 mnt_pt->d_mount_point = TRUE;
52                 mnt_pt->d_mounted_fs = vmnt;
53                 kref_get(&vmnt->mnt_kref, 1); /* held by mnt_pt */
54                 vmnt->mnt_parent = mnt_pt->d_sb->s_mount;
55                 vmnt->mnt_mountpoint = mnt_pt;
56         }
57         TAILQ_INIT(&vmnt->mnt_child_mounts);
58         vmnt->mnt_flags = flags;
59         vmnt->mnt_devname = dev_name;
60         vmnt->mnt_namespace = ns;
61         kref_get(&ns->kref, 1); /* held by vmnt */
62
63         /* Read in / create the SB */
64         sb = fs->get_sb(fs, flags, dev_name, vmnt);
65         if (!sb)
66                 panic("You're FS sucks");
67
68         /* TODO: consider moving this into get_sb or something, in case the SB
69          * already exists (mounting again) (if we support that) */
70         spin_lock(&super_blocks_lock);
71         TAILQ_INSERT_TAIL(&super_blocks, sb, s_list); /* storing a ref here... */
72         spin_unlock(&super_blocks_lock);
73
74         /* Update holding NS */
75         spin_lock(&ns->lock);
76         TAILQ_INSERT_TAIL(&ns->vfsmounts, vmnt, mnt_list);
77         spin_unlock(&ns->lock);
78         /* note to self: so, right after this point, the NS points to the root FS
79          * mount (we return the mnt, which gets assigned), the root mnt has a dentry
80          * for /, backed by an inode, with a SB prepped and in memory. */
81         return vmnt;
82 }
83
84 void vfs_init(void)
85 {
86         struct fs_type *fs;
87
88         dentry_kcache = kmem_cache_create("dentry", sizeof(struct dentry),
89                                           __alignof__(struct dentry), 0, 0, 0);
90         inode_kcache = kmem_cache_create("inode", sizeof(struct inode),
91                                          __alignof__(struct inode), 0, 0, 0);
92         file_kcache = kmem_cache_create("file", sizeof(struct file),
93                                         __alignof__(struct file), 0, 0, 0);
94         /* default NS never dies, +1 to exist */
95         kref_init(&default_ns.kref, fake_release, 1);
96         spinlock_init(&default_ns.lock);
97         default_ns.root = NULL;
98         TAILQ_INIT(&default_ns.vfsmounts);
99
100         /* build list of all FS's in the system.  put yours here.  if this is ever
101          * done on the fly, we'll need to lock. */
102         TAILQ_INSERT_TAIL(&file_systems, &kfs_fs_type, list);
103         TAILQ_FOREACH(fs, &file_systems, list)
104                 printk("Supports the %s Filesystem\n", fs->name);
105
106         /* mounting KFS at the root (/), pending root= parameters */
107         // TODO: linux creates a temp root_fs, then mounts the real root onto that
108         default_ns.root = mount_fs(&kfs_fs_type, "RAM", NULL, 0, &default_ns);
109
110         printk("vfs_init() completed\n");
111 }
112
113 /* Builds / populates the qstr of a dentry based on its d_iname.  If there is an
114  * l_name, (long), it will use that instead of the inline name.  This will
115  * probably change a bit. */
116 void qstr_builder(struct dentry *dentry, char *l_name)
117 {
118         dentry->d_name.name = l_name ? l_name : dentry->d_iname;
119         // TODO: pending what we actually do in d_hash
120         //dentry->d_name.hash = dentry->d_op->d_hash(dentry, &dentry->d_name); 
121         dentry->d_name.hash = 0xcafebabe;
122         dentry->d_name.len = strnlen(dentry->d_name.name, MAX_FILENAME_SZ);
123 }
124
125 /* Useful little helper - return the string ptr for a given file */
126 char *file_name(struct file *file)
127 {
128         return file->f_dentry->d_name.name;
129 }
130
131 /* Some issues with this, coupled closely to fs_lookup.  This assumes that
132  * negative dentries are not returned (might differ from linux) */
133 static struct dentry *do_lookup(struct dentry *parent, char *name)
134 {
135         struct dentry *dentry;
136         /* TODO: look up in the dentry cache first */
137         dentry = get_dentry(parent->d_sb, parent, name);
138         dentry = parent->d_inode->i_op->lookup(parent->d_inode, dentry, 0);
139         /* insert in dentry cache */
140         /* TODO: if the following are done by us, how do we know the i_ino?
141          * also need to handle inodes that are already read in!  For now, we're
142          * going to have the FS handle it in it's lookup() method: 
143          * - get a new inode
144          * - read in the inode
145          * - put in the inode cache */
146         return dentry;
147 }
148
149 /* Walk up one directory, being careful of mountpoints, namespaces, and the top
150  * of the FS */
151 static int climb_up(struct nameidata *nd)
152 {
153         // TODO
154         warn("Climbing up (../) in path lookup not supported yet!");
155         return 0;
156 }
157
158 /* Update ND such that it represents having followed dentry.  IAW the nd
159  * refcnting rules, we need to decref any references that were in there before
160  * they get clobbered. */
161 static int next_link(struct dentry *dentry, struct nameidata *nd)
162 {
163         assert(nd->dentry && nd->mnt);
164         /* update the dentry */
165         kref_get(&dentry->d_kref, 1);
166         kref_put(&nd->dentry->d_kref);
167         nd->dentry = dentry;
168         /* update the mount, if we need to */
169         if (dentry->d_sb->s_mount != nd->mnt) {
170                 kref_get(&dentry->d_sb->s_mount->mnt_kref, 1);
171                 kref_put(&nd->mnt->mnt_kref);
172                 nd->mnt = dentry->d_sb->s_mount;
173         }
174         return 0;
175 }
176
177 static int follow_mount(struct nameidata *nd)
178 {
179         /* Detect mount, follow, etc... (TODO!) */
180         return 0;
181 }
182
183 static int follow_symlink(struct nameidata *nd)
184 {
185         /* Detect symlink, LOOKUP_FOLLOW, follow it, etc... (TODO!) */
186         return 0;
187 }
188
189 /* Resolves the links in a basic path walk.  0 for success, -EWHATEVER
190  * otherwise.  The final lookup is returned via nd. */
191 static int link_path_walk(char *path, struct nameidata *nd)
192 {
193         struct dentry *link_dentry;
194         struct inode *link_inode, *nd_inode;
195         char *next_slash;
196         char *link = path;
197         int error;
198
199         /* skip all leading /'s */
200         while (*link == '/')
201                 link++;
202         /* if there's nothing left (null terminated), we're done */
203         if (*link == '\0')
204                 return 0;
205         /* TODO: deal with depth and LOOKUP_FOLLOW, important for symlinks */
206
207         /* iterate through each intermediate link of the path.  in general, nd
208          * tracks where we are in the path, as far as dentries go.  once we have the
209          * next dentry, we try to update nd based on that dentry.  link is the part
210          * of the path string that we are looking up */
211         while (1) {
212                 nd_inode = nd->dentry->d_inode;
213                 if ((error = check_perms(nd_inode, nd->intent)))
214                         return error;
215                 /* find the next link, break out if it is the end */
216                 next_slash = strchr(link, '/');
217                 if (!next_slash)
218                         break;
219                 else
220                         if (*(next_slash + 1) == '\0') {
221                                 /* trailing slash on the path meant the target is a dir */
222                                 nd->flags |= LOOKUP_DIRECTORY;
223                                 *next_slash = '\0';
224                                 break;
225                         }
226                 /* skip over any interim ./ */
227                 if (!strncmp("./", link, 2)) {
228                         link = next_slash + 1;
229                         continue;
230                 }
231                 /* Check for "../", walk up */
232                 if (!strncmp("../", link, 3)) {
233                         climb_up(nd);
234                         link = next_slash + 2;
235                         continue;
236                 }
237                 *next_slash = '\0';
238                 link_dentry = do_lookup(nd->dentry, link);
239                 *next_slash = '/';
240                 if (!link_dentry)
241                         return -ENOENT;
242                 /* make link_dentry the current step/answer */
243                 next_link(link_dentry, nd);
244                 kref_put(&link_dentry->d_kref); /* do_lookup gave us a refcnt dentry */
245                 /* we could be on a mountpoint or a symlink - need to follow them */
246                 follow_mount(nd);
247                 follow_symlink(nd);
248                 if (!(nd->dentry->d_inode->i_type & FS_I_DIR))
249                         return -ENOTDIR;
250                 /* move through the path string to the next entry */
251                 link = next_slash + 1;
252         }
253         /* now, we're on the last link of the path */
254         /* if we just want the parent, leave now.  and save the name of the link
255          * (last) and the type (last_type).  Note that using the qstr in this manner
256          * only allows us to use the qstr as long as the path is a valid string. */
257         if (nd->flags & LOOKUP_PARENT) {
258                 /* consider using a slimmer qstr_builder for this */
259                 nd->last.name = link;
260                 nd->last.len = strlen(link);
261                 nd->last.hash = nd->dentry->d_op->d_hash(nd->dentry, &nd->last);
262                 return 0;
263         }
264         /* deal with some weird cases with . and .. (completely untested) */
265         if (!strcmp(".", link))
266                 return 0;
267         if (!strcmp("..", link))
268                 return climb_up(nd);
269         link_dentry = do_lookup(nd->dentry, link);
270         if (!link_dentry)
271                 return -ENOENT;
272         next_link(link_dentry, nd);
273         kref_put(&link_dentry->d_kref); /* do_lookup gave us a refcnt dentry */
274         follow_mount(nd);
275         follow_symlink(nd);
276         /* If we wanted a directory, but didn't get one, error out */
277         if ((nd->flags & LOOKUP_DIRECTORY) &&
278            !(nd->dentry->d_inode->i_type & FS_I_DIR))
279                 return -ENOTDIR;
280         return 0;
281 }
282
283 /* Given path, return the inode for the final dentry.  The ND should be
284  * initialized for the first call - specifically, we need the intent. 
285  * LOOKUP_PARENT and friends go in this flags var.
286  *
287  * TODO: this should consider the intent.  Note that creating requires writing
288  * to the last directory.
289  *
290  * Need to be careful too.  While the path has been copied-in to the kernel,
291  * it's still user input.  */
292 int path_lookup(char *path, int flags, struct nameidata *nd)
293 {
294         printd("Path lookup for %s\n", path);
295         /* we allow absolute lookups with no process context */
296         if (path[0] == '/') {                   /* absolute lookup */
297                 if (!current)
298                         nd->dentry = default_ns.root->mnt_root;
299                 else
300                         nd->dentry = current->fs_env.root;      
301         } else {                                                /* relative lookup */
302                 assert(current);
303                 /* Don't need to lock on the fs_env since we're reading one item */
304                 nd->dentry = current->fs_env.pwd;       
305         }
306         nd->mnt = nd->dentry->d_sb->s_mount;
307         /* Whenever references get put in the nd, incref them.  Whenever they are
308          * removed, decref them. */
309         kref_get(&nd->mnt->mnt_kref, 1);
310         kref_get(&nd->dentry->d_kref, 1);
311         nd->flags = flags;
312         nd->depth = 0;                                  /* used in symlink following */
313         return link_path_walk(path, nd);        
314 }
315
316 /* Call this after any use of path_lookup when you are done with its results,
317  * regardless of whether it succeeded or not.  It will free any references */
318 void path_release(struct nameidata *nd)
319 {
320         kref_put(&nd->dentry->d_kref);
321         kref_put(&nd->mnt->mnt_kref);
322 }
323
324 /* Superblock functions */
325
326 /* Helper to alloc and initialize a generic superblock.  This handles all the
327  * VFS related things, like lists.  Each FS will need to handle its own things
328  * in it's *_get_sb(), usually involving reading off the disc. */
329 struct super_block *get_sb(void)
330 {
331         struct super_block *sb = kmalloc(sizeof(struct super_block), 0);
332         sb->s_dirty = FALSE;
333         spinlock_init(&sb->s_lock);
334         kref_init(&sb->s_kref, fake_release, 1); /* for the ref passed out */
335         TAILQ_INIT(&sb->s_inodes);
336         TAILQ_INIT(&sb->s_dirty_i);
337         TAILQ_INIT(&sb->s_io_wb);
338         SLIST_INIT(&sb->s_anon_d);
339         TAILQ_INIT(&sb->s_files);
340         sb->s_fs_info = 0; // can override somewhere else
341         return sb;
342 }
343
344 /* Final stages of initializing a super block, including creating and linking
345  * the root dentry, root inode, vmnt, and sb.  The d_op and root_ino are
346  * FS-specific, but otherwise it's FS-independent, tricky, and not worth having
347  * around multiple times.
348  *
349  * Not the world's best interface, so it's subject to change, esp since we're
350  * passing (now 3) FS-specific things. */
351 void init_sb(struct super_block *sb, struct vfsmount *vmnt,
352              struct dentry_operations *d_op, unsigned long root_ino,
353              void *d_fs_info)
354 {
355         /* Build and init the first dentry / inode.  The dentry ref is stored later
356          * by vfsmount's mnt_root.  The parent is dealt with later. */
357         struct dentry *d_root = get_dentry(sb, 0,  "/");        /* probably right */
358
359         /* a lot of here on down is normally done in lookup() or create, since
360          * get_dentry isn't a fully usable dentry.  The two FS-specific settings are
361          * normally inherited from a parent within the same FS in get_dentry, but we
362          * have none here. */
363         d_root->d_op = d_op;
364         d_root->d_fs_info = d_fs_info;
365         struct inode *inode = get_inode(d_root);
366         if (!inode)
367                 panic("This FS sucks!");
368         d_root->d_inode = inode;                                /* storing the inode's kref here */
369         TAILQ_INSERT_TAIL(&inode->i_dentry, d_root, d_alias);   /* weak ref */
370         inode->i_ino = root_ino;
371         /* TODO: add the inode to the appropriate list (off i_list) */
372         /* TODO: do we need to read in the inode?  can we do this on demand? */
373         /* if this FS is already mounted, we'll need to do something different. */
374         sb->s_op->read_inode(inode);
375         /* Link the dentry and SB to the VFS mount */
376         vmnt->mnt_root = d_root;                                /* ref comes from get_dentry */
377         vmnt->mnt_sb = sb;
378         /* If there is no mount point, there is no parent.  This is true only for
379          * the rootfs. */
380         if (vmnt->mnt_mountpoint) {
381                 kref_get(&vmnt->mnt_mountpoint->d_kref, 1);     /* held by d_root */
382                 d_root->d_parent = vmnt->mnt_mountpoint;        /* dentry of the root */
383         }
384         /* insert the dentry into the dentry cache.  when's the earliest we can?
385          * when's the earliest we should?  what about concurrent accesses to the
386          * same dentry?  should be locking the dentry... */
387         dcache_put(d_root); // TODO: should set a d_flag too
388 }
389
390 /* Dentry Functions */
391
392 /* Helper to alloc and initialize a generic dentry.  The following needs to be
393  * set still: d_op (if no parent), d_fs_info (opt), d_inode, connect the inode
394  * to the dentry (and up the d_kref again), maybe dcache_put().  The inode
395  * stitching is done in get_inode() or lookup (depending on the FS).
396  *
397  * If the name is longer than the inline name, it will kmalloc a buffer, so
398  * don't worry about the storage for *name after calling this. */
399 struct dentry *get_dentry(struct super_block *sb, struct dentry *parent,
400                           char *name)
401 {
402         assert(name);
403         size_t name_len = strnlen(name, MAX_FILENAME_SZ);       /* not including \0! */
404         struct dentry *dentry = kmem_cache_alloc(dentry_kcache, 0);
405         char *l_name = 0;
406
407         //memset(dentry, 0, sizeof(struct dentry));
408         kref_init(&dentry->d_kref, dentry_release, 1);  /* this ref is returned */
409         spinlock_init(&dentry->d_lock);
410         TAILQ_INIT(&dentry->d_subdirs);
411         dentry->d_time = 0;
412         kref_get(&sb->s_kref, 1);
413         dentry->d_sb = sb;                                      /* storing a ref here... */
414         dentry->d_mount_point = FALSE;
415         dentry->d_mounted_fs = 0;
416         if (parent)     {                                               /* no parent for rootfs mount */
417                 kref_get(&parent->d_kref, 1);
418                 dentry->d_op = parent->d_op;    /* d_op set in init_sb for parentless */
419         }
420         dentry->d_parent = parent;
421         dentry->d_flags = 0;                            /* related to its dcache state */
422         dentry->d_fs_info = 0;
423         SLIST_INIT(&dentry->d_bucket);
424         if (name_len < DNAME_INLINE_LEN) {
425                 strncpy(dentry->d_iname, name, name_len);
426                 dentry->d_iname[name_len] = '\0';
427                 qstr_builder(dentry, 0);
428         } else {
429                 l_name = kmalloc(name_len + 1, 0);
430                 assert(l_name);
431                 strncpy(l_name, name, name_len);
432                 l_name[name_len] = '\0';
433                 qstr_builder(dentry, l_name);
434         }
435         /* Catch bugs by aggressively zeroing this (o/w we use old stuff) */
436         dentry->d_inode = 0;
437         return dentry;
438 }
439
440 /* Adds a dentry to the dcache. */
441 void dcache_put(struct dentry *dentry)
442 {
443 #if 0 /* pending a more thorough review of the dcache */
444         /* TODO: should set a d_flag too */
445         spin_lock(&dcache_lock);
446         SLIST_INSERT_HEAD(&dcache, dentry, d_hash);
447         spin_unlock(&dcache_lock);
448 #endif
449 }
450
451 /* Cleans up the dentry (after ref == 0).  We still may want it, and this is
452  * where we should add it to the dentry cache.  (TODO).  For now, we do nothing,
453  * since we don't have a dcache.
454  * 
455  * This has to handle two types of dentries: full ones (ones that had been used)
456  * and ones that had been just for lookups - hence the check for d_inode.
457  *
458  * Note that dentries pin and kref their inodes.  When all the dentries are
459  * gone, we want the inode to be released via kref.  The inode has internal /
460  * weak references to the dentry, which are not refcounted. */
461 void dentry_release(struct kref *kref)
462 {
463         struct dentry *dentry = container_of(kref, struct dentry, d_kref);
464         printd("Freeing dentry %08p: %s\n", dentry, dentry->d_name.name);
465         assert(dentry->d_op);   /* catch bugs.  a while back, some lacked d_op */
466         dentry->d_op->d_release(dentry);
467         /* TODO: check/test the boundaries on this. */
468         if (dentry->d_name.len > DNAME_INLINE_LEN)
469                 kfree((void*)dentry->d_name.name);
470         kref_put(&dentry->d_sb->s_kref);
471         if (dentry->d_mounted_fs)
472                 kref_put(&dentry->d_mounted_fs->mnt_kref);
473         if (dentry->d_inode) {
474                 TAILQ_REMOVE(&dentry->d_inode->i_dentry, dentry, d_alias);
475                 kref_put(&dentry->d_inode->i_kref);     /* but dentries kref inodes */
476         }
477         kmem_cache_free(dentry_kcache, dentry);
478 }
479
480 /* Inode Functions */
481
482 /* Creates and initializes a new inode.  Generic fields are filled in.
483  * FS-specific fields are filled in by the callout.  Specific fields are filled
484  * in in read_inode() based on what's on the disk for a given i_no, or when the
485  * inode is created (for new objects).
486  *
487  * i_no is set by the caller.  Note that this means this inode can be for an
488  * inode that is already on disk, or it can be used when creating. */
489 struct inode *get_inode(struct dentry *dentry)
490 {
491         struct super_block *sb = dentry->d_sb;
492         /* FS allocs and sets the following: i_op, i_fop, i_pm.pm_op, and any FS
493          * specific stuff. */
494         struct inode *inode = sb->s_op->alloc_inode(sb);
495         if (!inode) {
496                 set_errno(current_tf, ENOMEM);
497                 return 0;
498         }
499         TAILQ_INSERT_HEAD(&sb->s_inodes, inode, i_sb_list);             /* weak inode ref */
500         TAILQ_INIT(&inode->i_dentry);
501         TAILQ_INSERT_TAIL(&inode->i_dentry, dentry, d_alias);   /* weak dentry ref*/
502         /* one for the dentry->d_inode, one passed out */
503         kref_init(&inode->i_kref, inode_release, 2);
504         dentry->d_inode = inode;
505         inode->i_ino = 0;                                       /* set by caller later */
506         inode->i_blksize = sb->s_blocksize;
507         spinlock_init(&inode->i_lock);
508         inode->i_sb = sb;
509         inode->i_state = 0;                                     /* need real states, like I_NEW */
510         inode->dirtied_when = 0;
511         atomic_set(&inode->i_writecount, 0);
512         /* Set up the page_map structures.  Default is to use the embedded one.
513          * Might push some of this back into specific FSs.  For now, the FS tells us
514          * what pm_op they want via i_pm.pm_op, which we use when we point i_mapping
515          * to i_pm. */
516         inode->i_mapping = &inode->i_pm;
517         inode->i_mapping->pm_host = inode;
518         radix_tree_init(&inode->i_mapping->pm_tree);
519         spinlock_init(&inode->i_mapping->pm_tree_lock);
520         inode->i_mapping->pm_flags = 0;
521         return inode;
522 }
523
524 /* Helper op, used when creating regular files and directories.  Note we make a
525  * distinction between the mode and the file type (for now).  After calling
526  * this, call the FS specific version (create or mkdir), which will set the
527  * i_ino, the filetype, and do any other FS-specific stuff.  Also note that a
528  * lot of inode stuff was initialized in get_inode/alloc_inode.  The stuff here
529  * is pertinent to the specific creator (user), mode, and time.  Also note we
530  * don't pass this an nd, like Linux does... */
531 static struct inode *create_inode(struct dentry *dentry, int mode)
532 {
533         /* note it is the i_ino that uniquely identifies a file in the system.
534          * there's a diff between creating an inode (even for an in-use ino) and
535          * then filling it in, and vs creating a brand new one */
536         struct inode *inode = get_inode(dentry);
537         if (!inode)
538                 return 0;
539         inode->i_mode = mode;
540         inode->i_nlink = 1;
541         inode->i_size = 0;
542         inode->i_blocks = 0;
543         inode->i_atime.tv_sec = 0;              /* TODO: now! */
544         inode->i_ctime.tv_sec = 0;
545         inode->i_mtime.tv_sec = 0;
546         inode->i_atime.tv_nsec = 0;             /* are these supposed to be the extra ns? */
547         inode->i_ctime.tv_nsec = 0;
548         inode->i_mtime.tv_nsec = 0;
549         inode->i_bdev = inode->i_sb->s_bdev;
550         return inode;
551 }
552
553 /* Create a new disk inode in dir associated with dentry, with the given mode.
554  * called when creating a regular file.  dir is the directory/parent.  dentry is
555  * the dentry of the inode we are creating.  Note the lack of the nd... */
556 int create_file(struct inode *dir, struct dentry *dentry, int flags, int mode)
557 {
558         struct inode *new_file = create_inode(dentry, mode);
559         if (!new_file)
560                 return -1;
561         dir->i_op->create(dir, dentry, mode, 0);
562         /* when we have notions of users, do something here: */
563         new_file->i_uid = 0;
564         new_file->i_gid = 0;
565         /* Not supposed to keep these creation flags */
566         new_file->i_flags = flags & ~(O_CREAT|O_TRUNC|O_EXCL|O_NOCTTY);
567         kref_put(&new_file->i_kref);
568         return 0;
569 }
570
571 /* Creates a new inode for a directory associated with dentry in dir with the
572  * given mode. */
573 int create_dir(struct inode *dir, struct dentry *dentry, int mode)
574 {
575         struct inode *new_dir = create_inode(dentry, mode);
576         if (!new_dir)
577                 return -1;
578         dir->i_op->mkdir(dir, dentry, mode);
579         /* Make sure my parent tracks me.  This is okay, since no directory (dir)
580          * can have more than one dentry */
581         struct dentry *parent = TAILQ_FIRST(&dir->i_dentry);
582         assert(parent && parent == TAILQ_LAST(&dir->i_dentry, dentry_tailq));
583         /* parent dentry tracks dentry as a subdir, weak reference */
584         TAILQ_INSERT_TAIL(&parent->d_subdirs, dentry, d_subdirs_link);
585         kref_put(&new_dir->i_kref);
586         return 0;
587 }
588
589 /* Returns 0 if the given mode is acceptable for the inode, and an appropriate
590  * error code if not.  Needs to be writen, based on some sensible rules, and
591  * will also probably use 'current' */
592 int check_perms(struct inode *inode, int access_mode)
593 {
594         return 0;       /* anything goes! */
595 }
596
597 /* Called after all external refs are gone to clean up the inode.  Once this is
598  * called, all dentries pointing here are already done (one of them triggered
599  * this via kref_put(). */
600 void inode_release(struct kref *kref)
601 {
602         struct inode *inode = container_of(kref, struct inode, i_kref);
603         inode->i_sb->s_op->destroy_inode(inode);
604         kref_put(&inode->i_sb->s_kref);
605         assert(inode->i_mapping == &inode->i_pm);
606         kmem_cache_free(inode_kcache, inode);
607         /* TODO: (BDEV) */
608         // kref_put(inode->i_bdev->kref); /* assuming it's a bdev */
609 }
610
611 /* File functions */
612
613 /* Read count bytes from the file into buf, starting at *offset, which is increased
614  * accordingly, returning the number of bytes transfered.  Most filesystems will
615  * use this function for their f_op->read.  Note, this uses the page cache.
616  * Want to try out page remapping later on... */
617 ssize_t generic_file_read(struct file *file, char *buf, size_t count,
618                           off_t *offset)
619 {
620         struct page *page;
621         int error;
622         off_t page_off;
623         unsigned long first_idx, last_idx;
624         size_t copy_amt;
625         char *buf_end;
626
627         /* Consider pushing some error checking higher in the VFS */
628         if (!count)
629                 return 0;
630         if (*offset == file->f_dentry->d_inode->i_size)
631                 return 0; /* EOF */
632         /* Make sure we don't go past the end of the file */
633         if (*offset + count > file->f_dentry->d_inode->i_size) {
634                 count = file->f_dentry->d_inode->i_size - *offset;
635         }
636         page_off = *offset & (PGSIZE - 1);
637         first_idx = *offset >> PGSHIFT;
638         last_idx = (*offset + count) >> PGSHIFT;
639         buf_end = buf + count;
640         /* For each file page, make sure it's in the page cache, then copy it out.
641          * TODO: will probably need to consider concurrently truncated files here.*/
642         for (int i = first_idx; i <= last_idx; i++) {
643                 error = file_load_page(file, i, &page);
644                 assert(!error); /* TODO: handle ENOMEM and friends */
645                 copy_amt = MIN(PGSIZE - page_off, buf_end - buf);
646                 /* TODO: think about this.  if it's a user buffer, we're relying on
647                  * current to detect whose it is (which should work for async calls).
648                  * Also, need to propagate errors properly...  Probably should do a
649                  * user_mem_check, then free, and also to make a distinction between
650                  * when the kernel wants a read/write (TODO: KFOP) */
651                 if (current) {
652                         memcpy_to_user(current, buf, page2kva(page) + page_off, copy_amt);
653                 } else {
654                         memcpy(buf, page2kva(page) + page_off, copy_amt);
655                 }
656                 buf += copy_amt;
657                 page_off = 0;
658                 page_decref(page);      /* it's still in the cache, we just don't need it */
659         }
660         assert(buf == buf_end);
661         *offset += count;
662         return count;
663 }
664
665 /* Write count bytes from buf to the file, starting at *offset, which is increased
666  * accordingly, returning the number of bytes transfered.  Most filesystems will
667  * use this function for their f_op->write.  Note, this uses the page cache.
668  * Changes don't get flushed to disc til there is an fsync, page cache eviction,
669  * or other means of trying to writeback the pages. */
670 ssize_t generic_file_write(struct file *file, const char *buf, size_t count,
671                            off_t *offset)
672 {
673         struct page *page;
674         int error;
675         off_t page_off;
676         unsigned long first_idx, last_idx;
677         size_t copy_amt;
678         const char *buf_end;
679
680         /* Consider pushing some error checking higher in the VFS */
681         if (!count)
682                 return 0;
683         /* Extend the file.  Should put more checks in here, and maybe do this per
684          * page in the for loop below. */
685         if (*offset + count > file->f_dentry->d_inode->i_size)
686                 file->f_dentry->d_inode->i_size = *offset + count;
687         page_off = *offset & (PGSIZE - 1);
688         first_idx = *offset >> PGSHIFT;
689         last_idx = (*offset + count) >> PGSHIFT;
690         buf_end = buf + count;
691         /* For each file page, make sure it's in the page cache, then write it.*/
692         for (int i = first_idx; i <= last_idx; i++) {
693                 error = file_load_page(file, i, &page);
694                 assert(!error); /* TODO: handle ENOMEM and friends */
695                 copy_amt = MIN(PGSIZE - page_off, buf_end - buf);
696                 /* TODO: think about this.  if it's a user buffer, we're relying on
697                  * current to detect whose it is (which should work for async calls). */
698                 if (current) {
699                         memcpy_to_user(current, page2kva(page) + page_off, buf, copy_amt);
700                 } else {
701                         memcpy(page2kva(page) + page_off, buf, copy_amt);
702                 }
703                 buf += copy_amt;
704                 page_off = 0;
705                 page_decref(page);      /* it's still in the cache, we just don't need it */
706         }
707         assert(buf == buf_end);
708         *offset += count;
709         return count;
710 }
711
712 /* Opens the file, using permissions from current for lack of a better option.
713  * It will attempt to create the file if it does not exist and O_CREAT is
714  * specified.  This will return 0 on failure, and set errno.
715  * TODO: There's a lot of stuff that we don't do, esp related to permission
716  * checking and file truncating.  Create should set errno and propagate it up.*/
717 struct file *do_file_open(char *path, int flags, int mode)
718 {
719         struct file *file = 0;
720         struct dentry *file_d;
721         struct inode *parent_i;
722         struct nameidata nd_r = {0}, *nd = &nd_r;
723         int lookup_flags = LOOKUP_PARENT;
724         int error = 0;
725
726         /* lookup the parent */
727         nd->intent = flags & (O_RDONLY|O_WRONLY|O_RDWR);
728         if (flags & O_CREAT)
729                 lookup_flags |= LOOKUP_CREATE;
730         error = path_lookup(path, lookup_flags, nd);
731         if (error) {
732                 set_errno(current_tf, -error);
733                 return 0;
734         }
735         /* see if the target is there, handle accordingly */
736         file_d = do_lookup(nd->dentry, nd->last.name); 
737         if (!file_d) {
738                 if (!(flags & O_CREAT)) {
739                         path_release(nd);
740                         set_errno(current_tf, ENOENT);
741                         return 0;
742                 }
743                 /* Create the inode/file.  get a fresh dentry too: */
744                 file_d = get_dentry(nd->dentry->d_sb, nd->dentry, nd->last.name);
745                 parent_i = nd->dentry->d_inode;
746                 /* TODO: mode should be & ~umask.  Note that mode only applies to future
747                  * opens. */
748                 if (create_file(parent_i, file_d, flags, mode)) {
749                         kref_put(&file_d->d_kref);
750                         path_release(nd);
751                         return 0;
752                 }
753                 dcache_put(file_d);
754         } else {        /* the file exists */
755                 if ((flags & O_CREAT) && (flags & O_EXCL)) {
756                         /* wanted to create, not open, bail out */
757                         kref_put(&file_d->d_kref);
758                         path_release(nd);
759                         set_errno(current_tf, EACCES);
760                         return 0;
761                 }
762         }
763         /* now open the file (freshly created or if it already existed).  At this
764          * point, file_d is a refcnt'd dentry, regardless of which branch we took.*/
765         if (flags & O_TRUNC)
766                 warn("File truncation not supported yet.");
767         file = dentry_open(file_d);             /* sets errno */
768         if (!file) {
769                 kref_put(&file_d->d_kref);
770                 path_release(nd);
771                 return 0;
772         }
773         /* TODO: check the inode's mode (S_XXX) against the flags O_RDWR */
774         /* f_mode stores how the FILE is open, regardless of the mode */
775         file->f_mode = flags & (O_RDONLY|O_WRONLY|O_RDWR);
776         kref_put(&file_d->d_kref);
777         path_release(nd);
778         return file;
779 }
780
781 /* Opens and returns the file specified by dentry */
782 struct file *dentry_open(struct dentry *dentry)
783 {
784         struct inode *inode;
785         struct file *file = kmem_cache_alloc(file_kcache, 0);
786         if (!file) {
787                 set_errno(current_tf, ENOMEM);
788                 return 0;
789         }
790         inode = dentry->d_inode;
791         /* one for the ref passed out, and *none* for the sb TAILQ */
792         kref_init(&file->f_kref, file_release, 1);
793         /* Add to the list of all files of this SB */
794         TAILQ_INSERT_TAIL(&inode->i_sb->s_files, file, f_list);
795         kref_get(&dentry->d_kref, 1);
796         file->f_dentry = dentry;
797         kref_get(&inode->i_sb->s_mount->mnt_kref, 1);
798         file->f_vfsmnt = inode->i_sb->s_mount;          /* saving a ref to the vmnt...*/
799         file->f_op = inode->i_fop;
800         file->f_flags = inode->i_flags;                         /* just taking the inode vals */
801         file->f_mode = inode->i_mode;
802         file->f_pos = 0;
803         file->f_uid = inode->i_uid;
804         file->f_gid = inode->i_gid;
805         file->f_error = 0;
806 //      struct event_poll_tailq         f_ep_links;
807         spinlock_init(&file->f_ep_lock);
808         file->f_fs_info = 0;                                            /* prob overriden by the fs */
809         file->f_mapping = inode->i_mapping;
810         file->f_op->open(inode, file);
811         return file;
812 }
813
814 /* Closes a file, fsync, whatever else is necessary.  Called when the kref hits
815  * 0.  Note that the file is not refcounted on the s_files list, nor is the
816  * f_mapping refcounted (it is pinned by the i_mapping). */
817 void file_release(struct kref *kref)
818 {
819         struct file *file = container_of(kref, struct file, f_kref);
820
821         struct super_block *sb = file->f_dentry->d_sb;
822         spin_lock(&sb->s_lock);
823         TAILQ_REMOVE(&sb->s_files, file, f_list);
824         spin_unlock(&sb->s_lock);
825
826         /* TODO: fsync (BLK).  also, we may want to parallelize the blocking that
827          * could happen in here (spawn kernel threads)... */
828         file->f_op->release(file->f_dentry->d_inode, file);
829         /* Clean up the other refs we hold */
830         kref_put(&file->f_dentry->d_kref);
831         kref_put(&file->f_vfsmnt->mnt_kref);
832         kmem_cache_free(file_kcache, file);
833 }
834
835 /* Page cache functions */
836
837 /* Looks up the index'th page in the page map, returning an incref'd reference,
838  * or 0 if it was not in the map. */
839 struct page *pm_find_page(struct page_map *pm, unsigned long index)
840 {
841         spin_lock(&pm->pm_tree_lock);
842         struct page *page = (struct page*)radix_lookup(&pm->pm_tree, index);
843         if (page)
844                 page_incref(page);
845         spin_unlock(&pm->pm_tree_lock);
846         return page;
847 }
848
849 /* Attempts to insert the page into the page_map, returns 0 for success, or an
850  * error code if there was one already (EEXIST) or we ran out of memory
851  * (ENOMEM).  On success, this will preemptively lock the page, and will also
852  * store a reference to the page in the pm. */
853 int pm_insert_page(struct page_map *pm, unsigned long index, struct page *page)
854 {
855         int error = 0;
856         spin_lock(&pm->pm_tree_lock);
857         error = radix_insert(&pm->pm_tree, index, page);
858         if (!error) {
859                 page_incref(page);
860                 page->pg_flags |= PG_LOCKED;
861                 page->pg_mapping = pm;
862                 page->pg_index = index;
863                 pm->pm_num_pages++;
864         }
865         spin_unlock(&pm->pm_tree_lock);
866         return error;
867 }
868
869 /* Removes the page, including its reference.  Not sure yet what interface we
870  * want to this (pm and index or page), and this has never been used.  There are
871  * also issues with when you want to call this, since a page in the cache may be
872  * mmap'd by someone else. */
873 int pm_remove_page(struct page_map *pm, struct page *page)
874 {
875         void *retval;
876         warn("pm_remove_page() hasn't been thought through or tested.");
877         spin_lock(&pm->pm_tree_lock);
878         retval = radix_delete(&pm->pm_tree, page->pg_index);
879         spin_unlock(&pm->pm_tree_lock);
880         assert(retval == (void*)page);
881         page_decref(page);
882         page->pg_mapping = 0;
883         page->pg_index = 0;
884         pm->pm_num_pages--;
885         return 0;
886 }
887
888 /* Makes sure the index'th page from file is loaded in the page cache and
889  * returns its location via **pp.  Note this will give you a refcnt'd reference.
890  * This may block! TODO: (BLK) */
891 int file_load_page(struct file *file, unsigned long index, struct page **pp)
892 {
893         struct page_map *pm = file->f_mapping;
894         struct page *page;
895         int error;
896         bool page_was_mapped = TRUE;
897
898         page = pm_find_page(pm, index);
899         while (!page) {
900                 /* kpage_alloc, since we want the page to persist after the proc
901                  * dies (can be used by others, until the inode shuts down). */
902                 if (kpage_alloc(&page))
903                         return -ENOMEM;
904                 /* might want to initialize other things, perhaps in page_alloc() */
905                 page->pg_flags = 0;
906                 error = pm_insert_page(pm, index, page);
907                 switch (error) {
908                         case 0:
909                                 page_was_mapped = FALSE;
910                                 break;
911                         case -EEXIST:
912                                 /* the page was mapped already (benign race), just get rid of
913                                  * our page and try again (the only case that uses the while) */
914                                 page_decref(page);
915                                 page = pm_find_page(pm, index);
916                                 break;
917                         default:
918                                 /* something is wrong, bail out! */
919                                 page_decref(page);
920                                 return error;
921                 }
922         }
923         *pp = page;
924         /* if the page was in the map, we need to do some checks, and might have to
925          * read in the page later.  If the page was freshly inserted to the pm by
926          * us, we skip this since we are the one doing the readpage(). */
927         if (page_was_mapped) {
928                 /* is it already here and up to date?  if so, we're done */
929                 if (page->pg_flags & PG_UPTODATE)
930                         return 0;
931                 /* if not, try to lock the page (could BLOCK) */
932                 lock_page(page);
933                 /* we got it, is our page still in the cache?  check the mapping.  if
934                  * not, start over, perhaps with EAGAIN and outside support */
935                 if (!page->pg_mapping)
936                         panic("Page is not in the mapping!  Haven't implemented this!");
937                 /* double check, are we up to date?  if so, we're done */
938                 if (page->pg_flags & PG_UPTODATE) {
939                         unlock_page(page);
940                         return 0;
941                 }
942         }
943         /* if we're here, the page is locked by us, and it needs to be read in */
944         assert(page->pg_mapping == pm);
945         error = pm->pm_op->readpage(file, page);
946         assert(!error);
947         /* Try to sleep on the IO.  The page will be unlocked when the IO is done */
948         lock_page(page);
949         unlock_page(page);
950         assert(page->pg_flags & PG_UPTODATE);
951         return 0;
952 }
953
954 /* Process-related File management functions */
955
956 /* Given any FD, get the appropriate file, 0 o/w */
957 struct file *get_file_from_fd(struct files_struct *open_files, int file_desc)
958 {
959         struct file *retval = 0;
960         spin_lock(&open_files->lock);
961         if (file_desc < open_files->max_fdset) {
962                 if (GET_BITMASK_BIT(open_files->open_fds->fds_bits, file_desc)) {
963                         /* while max_files and max_fdset might not line up, we should never
964                          * have a valid fdset higher than files */
965                         assert(file_desc < open_files->max_files);
966                         retval = open_files->fd[file_desc];
967                         assert(retval);
968                         kref_get(&retval->f_kref, 1);
969                 }
970         }
971         spin_unlock(&open_files->lock);
972         return retval;
973 }
974
975 /* Remove FD from the open files, if it was there, and return f.  Currently,
976  * this decref's f, so the return value is not consumable or even usable.  This
977  * hasn't been thought through yet. */
978 struct file *put_file_from_fd(struct files_struct *open_files, int file_desc)
979 {
980         struct file *file = 0;
981         spin_lock(&open_files->lock);
982         if (file_desc < open_files->max_fdset) {
983                 if (GET_BITMASK_BIT(open_files->open_fds->fds_bits, file_desc)) {
984                         /* while max_files and max_fdset might not line up, we should never
985                          * have a valid fdset higher than files */
986                         assert(file_desc < open_files->max_files);
987                         file = open_files->fd[file_desc];
988                         open_files->fd[file_desc] = 0;
989                         CLR_BITMASK_BIT(open_files->open_fds->fds_bits, file_desc);
990                         /* the if case is due to files (stdin) without a *file yet */
991                         if (file)
992                                 kref_put(&file->f_kref);
993                 }
994         }
995         spin_unlock(&open_files->lock);
996         return file;
997 }
998
999 /* Inserts the file in the files_struct, returning the corresponding new file
1000  * descriptor, or an error code.  We currently grab the first open FD. */
1001 int insert_file(struct files_struct *open_files, struct file *file)
1002 {
1003         int slot = -1;
1004         spin_lock(&open_files->lock);
1005         for (int i = 0; i < open_files->max_fdset; i++) {
1006                 if (GET_BITMASK_BIT(open_files->open_fds->fds_bits, i))
1007                         continue;
1008                 slot = i;
1009                 SET_BITMASK_BIT(open_files->open_fds->fds_bits, slot);
1010                 assert(slot < open_files->max_files && open_files->fd[slot] == 0);
1011                 kref_get(&file->f_kref, 1);
1012                 open_files->fd[slot] = file;
1013                 if (slot >= open_files->next_fd)
1014                         open_files->next_fd = slot + 1;
1015                 break;
1016         }
1017         if (slot == -1) /* should expand the FD array and fd_set */
1018                 warn("Ran out of file descriptors, deal with me!");
1019         spin_unlock(&open_files->lock);
1020         return slot;
1021 }
1022
1023 /* Closes all open files.  Mostly just a "put" for all files.  If cloexec, it
1024  * will only close files that are opened with O_CLOEXEC. */
1025 void close_all_files(struct files_struct *open_files, bool cloexec)
1026 {
1027         struct file *file;
1028         spin_lock(&open_files->lock);
1029         for (int i = 0; i < open_files->max_fdset; i++) {
1030                 if (GET_BITMASK_BIT(open_files->open_fds->fds_bits, i)) {
1031                         /* while max_files and max_fdset might not line up, we should never
1032                          * have a valid fdset higher than files */
1033                         assert(i < open_files->max_files);
1034                         file = open_files->fd[i];
1035                         if (cloexec && !(file->f_flags | O_CLOEXEC))
1036                                 continue;
1037                         open_files->fd[i] = 0;
1038                         /* the if case is due to files (stdin) without a *file yet */
1039                         if (file)
1040                                 kref_put(&file->f_kref);
1041                         CLR_BITMASK_BIT(open_files->open_fds->fds_bits, i);
1042                 }
1043         }
1044         spin_unlock(&open_files->lock);
1045 }
1046
1047 /* Inserts all of the files from src into dst, used by sys_fork(). */
1048 void clone_files(struct files_struct *src, struct files_struct *dst)
1049 {
1050         struct file *file;
1051         spin_lock(&src->lock);
1052         spin_lock(&dst->lock);
1053         for (int i = 0; i < src->max_fdset; i++) {
1054                 if (GET_BITMASK_BIT(src->open_fds->fds_bits, i)) {
1055                         /* while max_files and max_fdset might not line up, we should never
1056                          * have a valid fdset higher than files */
1057                         assert(i < src->max_files);
1058                         file = src->fd[i];
1059                         SET_BITMASK_BIT(dst->open_fds->fds_bits, i);
1060                         assert(i < dst->max_files && dst->fd[i] == 0);
1061                         dst->fd[i] = file;
1062                         /* the if case is due to files (stdin) without a *file yet */
1063                         if (file)
1064                                 kref_get(&file->f_kref, 1);
1065                 }
1066         }
1067         spin_unlock(&dst->lock);
1068         spin_unlock(&src->lock);
1069 }