Readdir fixes, ext2_readdir()
[akaros.git] / kern / src / kfs.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  * Implementation of the KFS file system.  It is a RAM based, read-only FS
6  * consisting of files that are added to the kernel binary image.  Might turn
7  * this into a read/write FS with directories someday. */
8
9 #ifdef __SHARC__
10 #pragma nosharc
11 #endif
12
13 #ifdef __DEPUTY__
14 #pragma nodeputy
15 #endif
16
17 #include <vfs.h>
18 #include <kfs.h>
19 #include <slab.h>
20 #include <kmalloc.h>
21 #include <string.h>
22 #include <stdio.h>
23 #include <assert.h>
24 #include <error.h>
25 #include <cpio.h>
26 #include <pmap.h>
27 #include <smp.h>
28
29 #define KFS_MAX_FILE_SIZE 1024*1024*128
30 #define KFS_MAGIC 0xdead0001
31
32 /* VFS required Functions */
33 /* These structs are declared again and initialized farther down */
34 struct page_map_operations kfs_pm_op;
35 struct super_operations kfs_s_op;
36 struct inode_operations kfs_i_op;
37 struct dentry_operations kfs_d_op;
38 struct file_operations kfs_f_op_file;
39 struct file_operations kfs_f_op_dir;
40 struct file_operations kfs_f_op_sym;
41
42 /* TODO: something more better.  Prob something like the vmem cache, for this,
43  * pids, etc.  Good enough for now.  This also means we can only have one
44  * KFS instance, and we also aren't synchronizing access. */
45 static unsigned long kfs_get_free_ino(void)
46 {
47         static unsigned long last_ino = 1;       /* 1 is reserved for the root */
48         last_ino++;
49         if (!last_ino)
50                 panic("Out of inos in KFS!");
51         return last_ino;
52 }
53
54 /* Slabs for KFS specific info chunks */
55 struct kmem_cache *kfs_i_kcache;
56
57 static void kfs_init(void)
58 {
59         kfs_i_kcache = kmem_cache_create("kfs_ino_info", sizeof(struct kfs_i_info),
60                                          __alignof__(struct kfs_i_info), 0, 0, 0);
61 }
62
63 /* Creates the SB (normally would read in from disc and create).  Passes it's
64  * ref out to whoever consumes this.  Returns 0 on failure.
65  * TODO: consider pulling out more of the FS-independent stuff, if possible.
66  * There are only two things, but the pain in the ass is that you'd need to read
67  * the disc to get that first inode, and it's a FS-specific thing. */
68 struct super_block *kfs_get_sb(struct fs_type *fs, int flags,
69                                char *dev_name, struct vfsmount *vmnt)
70 {
71         /* Ought to check that dev_name has our FS on it.  in this case, it's
72          * irrelevant. */
73         //if (something_bad)
74         //      return 0;
75         static bool ran_once = FALSE;
76         if (!ran_once) {
77                 ran_once = TRUE;
78                 kfs_init();
79         }
80
81         /* Build and init the SB.  No need to read off disc. */
82         struct super_block *sb = get_sb();
83         sb->s_dev = 0;
84         sb->s_blocksize = 1;
85         sb->s_maxbytes = KFS_MAX_FILE_SIZE;
86         sb->s_type = &kfs_fs_type;
87         sb->s_op = &kfs_s_op;
88         sb->s_flags = flags;
89         sb->s_magic = KFS_MAGIC;
90         sb->s_mount = vmnt;
91         sb->s_syncing = FALSE;
92         sb->s_bdev = 0;
93         strlcpy(sb->s_name, "KFS", 32);
94         /* store the location of the CPIO archive.  make this more generic later. */
95         extern uint8_t _binary_obj_kern_initramfs_cpio_size[];
96         extern uint8_t _binary_obj_kern_initramfs_cpio_start[];
97         sb->s_fs_info = (void*)_binary_obj_kern_initramfs_cpio_start;
98
99         /* Final stages of initializing the sb, mostly FS-independent */
100         /* 1 is the KFS root ino (inode number) */
101         init_sb(sb, vmnt, &kfs_d_op, 1, 0);
102         /* Parses the CPIO entries and builds the in-memory KFS tree. */
103         parse_cpio_entries(sb, sb->s_fs_info);
104         printk("KFS superblock loaded\n");
105         return sb;
106 }
107
108 void kfs_kill_sb(struct super_block *sb)
109 {
110         panic("Killing KFS is not supported!");
111 }
112
113 /* Every FS must have a static FS Type, with which the VFS code can bootstrap */
114 struct fs_type kfs_fs_type = {"KFS", 0, kfs_get_sb, kfs_kill_sb, {0, 0},
115                TAILQ_HEAD_INITIALIZER(kfs_fs_type.fs_supers)};
116
117 /* Page Map Operations */
118
119 /* Fills page with its contents from its backing store file.  Note that we do
120  * the zero padding here, instead of higher in the VFS.  Might change in the
121  * future. */
122 int kfs_readpage(struct file *file, struct page *page)
123 {
124         size_t pg_idx_byte = page->pg_index * PGSIZE;
125         struct kfs_i_info *k_i_info = (struct kfs_i_info*)
126                                       file->f_dentry->d_inode->i_fs_info;
127         uintptr_t begin = (size_t)k_i_info->filestart + pg_idx_byte;
128         /* If we're beyond the initial start point, we just need a zero page.  This
129          * is for a hole or for extending a file (even though it won't be saved).
130          * Otherwise, we want the data from KFS, being careful to not copy from
131          * beyond the original EOF (and zero padding anything extra). */
132         if (pg_idx_byte >= k_i_info->init_size) {
133                 memset(page2kva(page), 0, PGSIZE);
134         } else {
135                 size_t copy_amt = MIN(PGSIZE, k_i_info->init_size - pg_idx_byte);
136                 memcpy(page2kva(page), (void*)begin, copy_amt);
137                 memset(page2kva(page) + copy_amt, 0, PGSIZE - copy_amt);
138         }
139         /* This is supposed to be done in the IO system when the operation is
140          * complete.  Since we aren't doing a real IO request, and it is already
141          * done, we can do it here. */
142         page->pg_flags |= PG_UPTODATE;
143         unlock_page(page);
144         return 0;
145 }
146
147 /* Super Operations */
148
149 /* Creates and initializes a new inode.  FS specific, yet inode-generic fields
150  * are filled in.  inode-specific fields are filled in in read_inode() based on
151  * what's on the disk for a given i_no.  i_no and i_fop are set by the caller.
152  *
153  * Note that this means this inode can be for an inode that is already on disk,
154  * or it can be used when creating.  The i_fop depends on the type of file
155  * (file, directory, symlink, etc). */
156 struct inode *kfs_alloc_inode(struct super_block *sb)
157 {
158         struct inode *inode = kmem_cache_alloc(inode_kcache, 0);
159         memset(inode, 0, sizeof(struct inode));
160         inode->i_op = &kfs_i_op;
161         inode->i_pm.pm_op = &kfs_pm_op;
162         inode->i_fs_info = kmem_cache_alloc(kfs_i_kcache, 0);
163         TAILQ_INIT(&((struct kfs_i_info*)inode->i_fs_info)->children);
164         ((struct kfs_i_info*)inode->i_fs_info)->filestart = 0;
165         return inode;
166 }
167
168 /* deallocs and cleans up after an inode. */
169 void kfs_dealloc_inode(struct inode *inode)
170 {
171         /* If we're a symlink, give up our storage for the symname */
172         if (S_ISLNK(inode->i_mode))
173                 kfree(((struct kfs_i_info*)inode->i_fs_info)->filestart);
174         kmem_cache_free(kfs_i_kcache, inode->i_fs_info);
175 }
176
177 /* reads the inode data on disk specified by inode->i_ino into the inode.
178  * basically, it's a "make this inode the one for i_ino (i number)" */
179 void kfs_read_inode(struct inode *inode)
180 {
181         /* need to do something to link this inode/file to the actual "blocks" on
182          * "disk". */
183
184         /* TODO: what does it mean to ask for an inode->i_ino that doesn't exist?
185          *      possibly a bug, since these inos come from directories */
186         if (inode->i_ino == 1) {
187                 inode->i_mode = S_IRWXU | S_IRWXG | S_IRWXO;
188                 SET_FTYPE(inode->i_mode, __S_IFDIR);
189                 inode->i_fop = &kfs_f_op_dir;
190                 inode->i_nlink = 1;                             /* assuming only one hardlink */
191                 inode->i_uid = 0;
192                 inode->i_gid = 0;
193                 inode->i_size = 0;                              /* make sense for KFS? */
194                 inode->i_atime.tv_sec = 0;
195                 inode->i_atime.tv_nsec = 0;
196                 inode->i_mtime.tv_sec = 0;
197                 inode->i_mtime.tv_nsec = 0;
198                 inode->i_ctime.tv_sec = 0;
199                 inode->i_ctime.tv_nsec = 0;
200                 inode->i_blocks = 0;
201                 inode->i_flags = 0;
202                 inode->i_socket = FALSE;
203         } else {
204                 panic("Not implemented");
205         }
206         /* TODO: unused: inode->i_hash add to hash (saves on disc reading) */
207 }
208
209 /* called when an inode in memory is modified (journalling FS's care) */
210 void kfs_dirty_inode(struct inode *inode)
211 {       // KFS doesn't care
212 }
213
214 /* write the inode to disk (specifically, to inode inode->i_ino), synchronously
215  * if we're asked to wait */
216 void kfs_write_inode(struct inode *inode, bool wait)
217 {       // KFS doesn't care
218 }
219
220 /* called when an inode is decref'd, to do any FS specific work */
221 void kfs_put_inode(struct inode *inode)
222 {       // KFS doesn't care
223 }
224
225 /* called when an inode is about to be destroyed.  the generic version ought to
226  * remove every reference to the inode from the VFS, and if the inode isn't in
227  * any directory, calls delete_inode */
228 void kfs_drop_inode(struct inode *inode)
229 { // TODO: should call a generic one instead.  or at least do something...
230         // remove from lists
231 }
232
233 /* delete the inode from disk (all data) and deallocs the in memory inode */
234 void kfs_delete_inode(struct inode *inode)
235 {
236         // would remove from "disk" here
237         kfs_dealloc_inode(inode);
238         /* TODO: give up our i_ino */
239 }
240
241 /* unmount and release the super block */
242 void kfs_put_super(struct super_block *sb)
243 {
244         panic("Shazbot! KFS can't be unmounted yet!");
245 }
246
247 /* updates the on-disk SB with the in-memory SB */
248 void kfs_write_super(struct super_block *sb)
249 {       // KFS doesn't care
250 }
251
252 /* syncs FS metadata with the disc, synchronously if we're waiting.  this info
253  * also includes anything pointed to by s_fs_info. */
254 int kfs_sync_fs(struct super_block *sb, bool wait)
255 {
256         return 0;
257 }
258
259 /* remount the FS with the new flags */
260 int kfs_remount_fs(struct super_block *sb, int flags, char *data)
261 {
262         warn("KFS will not remount.");
263         return -1; // can't remount
264 }
265
266 /* interrupts a mount operation - used by NFS and friends */
267 void kfs_umount_begin(struct super_block *sb)
268 {
269         panic("Cannot abort a KFS mount, and why would you?");
270 }
271
272 /* inode_operations */
273
274 /* Little helper, used for initializing new inodes for file-like objects (files,
275  * symlinks, etc).  We pass the dentry, since we need to up it. */
276 static void kfs_init_inode(struct inode *dir, struct dentry *dentry)
277 {
278         struct inode *inode = dentry->d_inode;
279         kref_get(&dentry->d_kref, 1);   /* to pin the dentry in RAM, KFS-style... */
280         inode->i_ino = kfs_get_free_ino();
281         /* our parent dentry's inode tracks our dentry info.  We do this
282          * since it's all in memory and we aren't using the dcache yet.
283          * We're reusing the subdirs link, which is used by the VFS when
284          * we're a directory.  But since we're a file, it's okay to reuse
285          * it. */
286         TAILQ_INSERT_TAIL(&((struct kfs_i_info*)dir->i_fs_info)->children,
287                           dentry, d_subdirs_link);
288 }
289
290 /* Called when creating a new disk inode in dir associated with dentry.  We need
291  * to fill out the i_ino, set the type, and do whatever else we need */
292 int kfs_create(struct inode *dir, struct dentry *dentry, int mode,
293                struct nameidata *nd)
294 {
295         struct inode *inode = dentry->d_inode;
296         kfs_init_inode(dir, dentry);
297         SET_FTYPE(inode->i_mode, __S_IFREG);
298         inode->i_fop = &kfs_f_op_file;
299         /* fs_info->filestart is set by the caller, or else when first written (for
300          * new files.  it was set to 0 in alloc_inode(). */
301         return 0;
302 }
303
304 /* Searches the directory for the filename in the dentry, filling in the dentry
305  * with the FS specific info of this file.  If it succeeds, it will pass back
306  * the *dentry you should use.  If this fails, it will return 0 and will take
307  * the ref to the dentry for you.  Either way, you shouldn't use the ref you
308  * passed in anymore.  Still, there are issues with refcnting with this.
309  *
310  * Callers, make sure you alloc and fill out the name parts of the dentry, and
311  * an initialized nameidata. TODO: not sure why we need an ND.  Don't use it in
312  * a fs_lookup for now!
313  *
314  * Because of the way KFS currently works, if there is ever a dentry, it's
315  * already in memory, along with its inode (all path's pinned).  So we just find
316  * it and return it, freeing the one that came in. */
317 struct dentry *kfs_lookup(struct inode *dir, struct dentry *dentry,
318                           struct nameidata *nd)
319 {
320         struct kfs_i_info *k_i_info = (struct kfs_i_info*)dir->i_fs_info;
321         struct dentry *dir_dent = TAILQ_FIRST(&dir->i_dentry);
322         struct dentry *d_i;
323
324         assert(dir_dent && dir_dent == TAILQ_LAST(&dir->i_dentry, dentry_tailq));
325         assert(S_ISDIR(dir->i_mode));
326         assert(kref_refcnt(&dentry->d_kref) == 1);
327         TAILQ_FOREACH(d_i, &dir_dent->d_subdirs, d_subdirs_link) {
328                 if (!strcmp(d_i->d_name.name, dentry->d_name.name)) {
329                         /* since this dentry is already in memory (that's how KFS works), we
330                          * can free the one that came in and return the real one */
331                         kref_put(&dentry->d_kref);
332                         kref_get(&d_i->d_kref, 1);
333                         return d_i;
334                 }
335         }
336         TAILQ_FOREACH(d_i, &k_i_info->children, d_subdirs_link) {
337                 if (!strcmp(d_i->d_name.name, dentry->d_name.name)) {
338                         /* since this dentry is already in memory (that's how KFS works), we
339                          * can free the one that came in and return the real one */
340                         kref_put(&dentry->d_kref);
341                         kref_get(&d_i->d_kref, 1);
342                         return d_i;
343                 }
344         }
345         /* no match, consider caching the negative result, freeing the
346          * dentry, etc */
347         printd("Not Found %s!!\n", dentry->d_name.name);
348         kref_put(&dentry->d_kref);
349         return 0;
350 }
351
352 /* Hard link to old_dentry in directory dir with a name specified by new_dentry.
353  * At the very least, set the new_dentry's FS-specific fields. */
354 int kfs_link(struct dentry *old_dentry, struct inode *dir,
355              struct dentry *new_dentry)
356 {
357         assert(new_dentry->d_op = &kfs_d_op);
358         kref_get(&new_dentry->d_kref, 1);               /* pin the dentry, KFS-style */
359         /* KFS-style directory-tracking-of-kids */
360         TAILQ_INSERT_TAIL(&((struct kfs_i_info*)dir->i_fs_info)->children,
361                           new_dentry, d_subdirs_link);
362         return 0;
363 }
364
365 /* Removes the link from the dentry in the directory */
366 int kfs_unlink(struct inode *dir, struct dentry *dentry)
367 {
368         /* Stop tracking our child */
369         TAILQ_REMOVE(&((struct kfs_i_info*)dir->i_fs_info)->children, dentry,
370                      d_subdirs_link);
371         return 0;
372 }
373
374 /* Creates a new inode for a symlink dir, linking to / containing the name
375  * symname.  dentry is the controlling dentry of the inode. */
376 int kfs_symlink(struct inode *dir, struct dentry *dentry, const char *symname)
377 {
378         struct inode *inode = dentry->d_inode;
379         struct kfs_i_info *k_i_info = (struct kfs_i_info*)inode->i_fs_info;
380         size_t len = strlen(symname);
381         char *string = kmalloc(len + 1, 0);
382
383         kfs_init_inode(dir, dentry);
384         SET_FTYPE(inode->i_mode, __S_IFLNK);
385         inode->i_fop = &kfs_f_op_sym;
386         strncpy(string, symname, len);
387         string[len] = '\0';             /* symname should be \0d anyway, but just in case */
388         k_i_info->filestart = string;   /* reusing this void* to hold the char* */
389         return 0;
390 }
391
392 /* Called when creating a new inode for a directory associated with dentry in
393  * dir with the given mode.  Note, we might (later) need to track subdirs within
394  * the parent inode, like we do with regular files.  I'd rather not, so we'll
395  * see if we need it. */
396 int kfs_mkdir(struct inode *dir, struct dentry *dentry, int mode)
397 {
398         struct inode *inode = dentry->d_inode;
399         kref_get(&dentry->d_kref, 1);   /* to pin the dentry in RAM, KFS-style... */
400         inode->i_ino = kfs_get_free_ino();
401         SET_FTYPE(inode->i_mode, __S_IFDIR);
402         inode->i_fop = &kfs_f_op_dir;
403         /* get ready to have our own kids */
404         TAILQ_INIT(&((struct kfs_i_info*)inode->i_fs_info)->children);
405         ((struct kfs_i_info*)inode->i_fs_info)->filestart = 0;
406         return 0;
407 }
408
409 /* Removes from dir the directory 'dentry.'  KFS doesn't store anything in the
410  * inode for which children it has.  It probably should, but since everything is
411  * pinned, it just relies on the dentry connections. */
412 int kfs_rmdir(struct inode *dir, struct dentry *dentry)
413 {
414         struct kfs_i_info *d_info = (struct kfs_i_info*)dentry->d_inode->i_fs_info;
415         struct dentry *d_i;
416         bool empty = TRUE;
417         /* Check if we are empty.  If not, error out, need to check the sub-dirs as
418          * well as the sub-"files" */
419         TAILQ_FOREACH(d_i, &dentry->d_subdirs, d_subdirs_link) {
420                 empty = FALSE;
421                 break;
422         }
423         TAILQ_FOREACH(d_i, &d_info->children, d_subdirs_link) {
424                 empty = FALSE;
425                 break;
426         }
427         if (!empty)
428                 return -ENOTEMPTY;
429         return 0;
430 }
431
432 /* Used to make a generic file, based on the type and the major/minor numbers
433  * (in rdev), with the given mode.  As with others, this creates a new disk
434  * inode for the file */
435 int kfs_mknod(struct inode *dir, struct dentry *dentry, int mode, dev_t rdev)
436 {
437         return -1;
438 }
439
440 /* Moves old_dentry from old_dir to new_dentry in new_dir */
441 int kfs_rename(struct inode *old_dir, struct dentry *old_dentry,
442                struct inode *new_dir, struct dentry *new_dentry)
443 {
444         return -1;
445 }
446
447 /* Returns the char* for the symname for the given dentry.  The VFS code that
448  * calls this for real FS's might assume it's already read in, so if the char *
449  * isn't already in memory, we'd need to read it in here.  Regarding the char*
450  * storage, the char* only will last as long as the dentry and inode are in
451  * memory. */
452 char *kfs_readlink(struct dentry *dentry)
453 {
454         struct inode *inode = dentry->d_inode;
455         struct kfs_i_info *k_i_info = (struct kfs_i_info*)inode->i_fs_info;
456         if (!S_ISLNK(inode->i_mode))
457                 return 0;
458         return k_i_info->filestart;
459 }
460
461 /* Modifies the size of the file of inode to whatever its i_size is set to */
462 void kfs_truncate(struct inode *inode)
463 {
464 }
465
466 /* Checks whether the the access mode is allowed for the file belonging to the
467  * inode.  Implies that the permissions are on the file, and not the hardlink */
468 int kfs_permission(struct inode *inode, int mode, struct nameidata *nd)
469 {
470         return -1;
471 }
472
473
474 /* dentry_operations */
475 /* Determines if the dentry is still valid before using it to translate a path.
476  * Network FS's need to deal with this. */
477 int kfs_d_revalidate(struct dentry *dir, struct nameidata *nd)
478 { // default, nothing
479         return -1;
480 }
481
482 /* Produces the hash to lookup this dentry from the dcache */
483 int kfs_d_hash(struct dentry *dentry, struct qstr *name)
484 {
485         return -1;
486 }
487
488 /* Compares name1 and name2.  name1 should be a member of dir. */
489 int kfs_d_compare(struct dentry *dir, struct qstr *name1, struct qstr *name2)
490 { // default, string comp (case sensitive)
491         return -1;
492 }
493
494 /* Called when the last ref is deleted (refcnt == 0) */
495 int kfs_d_delete(struct dentry *dentry)
496 { // default, nothin
497         return -1;
498 }
499
500 /* Called when it's about to be slab-freed */
501 int kfs_d_release(struct dentry *dentry)
502 {
503         return -1;
504 }
505
506 /* Called when the dentry loses it's inode (becomes "negative") */
507 void kfs_d_iput(struct dentry *dentry, struct inode *inode)
508 { // default, call i_put to release the inode object
509 }
510
511
512 /* file_operations */
513
514 /* Updates the file pointer.  KFS doesn't let you go past the end of a file
515  * yet, so it won't let you seek past either.  TODO: think about locking. */
516 off_t kfs_llseek(struct file *file, off_t offset, int whence)
517 {
518         off_t temp_off = 0;
519         switch (whence) {
520                 case SEEK_SET:
521                         temp_off = offset;
522                         break;
523                 case SEEK_CUR:
524                         temp_off = file->f_pos + offset;
525                         break;
526                 case SEEK_END:
527                         temp_off = file->f_dentry->d_inode->i_size + offset;
528                         break;
529                 default:
530                         set_errno(EINVAL);
531                         warn("Unknown 'whence' in llseek()!\n");
532                         return -1;
533         }
534         /* make sure the f_pos isn't outside the limits of the existing file.
535          * techincally, if they go too far, we should return EINVAL */
536         temp_off = MAX(MIN(temp_off, file->f_dentry->d_inode->i_size), 0);
537         file->f_pos = temp_off;
538         return temp_off;
539 }
540
541 /* Fills in the next directory entry (dirent), starting with d_off.  KFS treats
542  * the size of each dirent as 1 byte, which we can get away with since the d_off
543  * is a way of communicating with future calls to readdir (FS-specific).
544  *
545  * Like with read and write, there will be issues with userspace and the *dirent
546  * buf.  TODO: we don't really do anything with userspace concerns here, in part
547  * because memcpy_to doesn't work well.  When we fix how we want to handle the
548  * userbuffers, we can write this accordingly. (UMEM)  */
549 int kfs_readdir(struct file *dir, struct dirent *dirent)
550 {
551         int count = 2;  /* total num dirents, gets incremented in check_entry() */
552         int desired_file = dirent->d_off;
553         bool found = FALSE;
554         struct dentry *subent;
555         struct dentry *dir_d = dir->f_dentry;
556         struct kfs_i_info *k_i_info = (struct kfs_i_info*)dir_d->d_inode->i_fs_info;
557
558         /* how we check inside the for loops below.  moderately ghetto. */
559         void check_entry(void)
560         {
561                 if (count++ == desired_file) {
562                         dirent->d_ino = subent->d_inode->i_ino;
563                         dirent->d_off = count;
564                         dirent->d_reclen = subent->d_name.len;
565                         /* d_name.name is null terminated, the byte after d_name.len */
566                         assert(subent->d_name.len <= MAX_FILENAME_SZ);
567                         strncpy(dirent->d_name, subent->d_name.name, subent->d_name.len +1);
568                         found = TRUE;
569                 }
570         }
571
572         /* Handle . and .. (first two dirents) */
573         if (desired_file == 0) {
574                 dirent->d_ino = dir_d->d_inode->i_ino;
575                 dirent->d_off = 1;
576                 dirent->d_reclen = 1;
577                 strncpy(dirent->d_name, ".", 2);        /* the extra is for the null term */
578                 found = TRUE;
579         } else if (desired_file == 1) {
580                 dirent->d_ino = dir_d->d_parent->d_inode->i_ino;
581                 dirent->d_off = 2;
582                 dirent->d_reclen = 2;
583                 strncpy(dirent->d_name, "..", 3);       /* the extra is for the null term */
584                 found = TRUE;
585         }
586         /* need to check the sub-dirs as well as the sub-"files".  The main
587          * ghetto-ness with this is that we check even though we have our result,
588          * simply to figure out how big our directory is.  It's just not worth
589          * changing at this point. */
590         TAILQ_FOREACH(subent, &dir_d->d_subdirs, d_subdirs_link)
591                 check_entry();
592         TAILQ_FOREACH(subent, &k_i_info->children, d_subdirs_link)
593                 check_entry();
594         if (!found)
595                 return -ENOENT;
596         if (count - 1 == desired_file)          /* found the last dir in the list */
597                 return 0;
598         return 1;                                                       /* normal success for readdir */
599 }
600
601 /* This is called when a VMR is mapping a particular file.  The FS needs to do
602  * whatever it needs so that faults can be handled by read_page(), and handle all
603  * of the cases of MAP_SHARED, MAP_PRIVATE, whatever.  It also needs to ensure
604  * the file is not being mmaped in a way that conflicts with the manner in which
605  * the file was opened or the file type. */
606 int kfs_mmap(struct file *file, struct vm_region *vmr)
607 {
608         if (S_ISREG(file->f_dentry->d_inode->i_mode))
609                 return 0;
610         return -1;
611 }
612
613 /* Called by the VFS while opening the file, which corresponds to inode,  for
614  * the FS to do whatever it needs. */
615 int kfs_open(struct inode *inode, struct file *file)
616 {
617         return 0;
618 }
619
620 /* Called when a file descriptor is closed. */
621 int kfs_flush(struct file *file)
622 {
623         return -1;
624 }
625
626 /* Called when the file is about to be closed (file obj freed) */
627 int kfs_release(struct inode *inode, struct file *file)
628 {
629         return 0;
630 }
631
632 /* Flushes the file's dirty contents to disc */
633 int kfs_fsync(struct file *file, struct dentry *dentry, int datasync)
634 {
635         return -1;
636 }
637
638 /* Traditionally, sleeps until there is file activity.  We probably won't
639  * support this, or we'll handle it differently. */
640 unsigned int kfs_poll(struct file *file, struct poll_table_struct *poll_table)
641 {
642         return -1;
643 }
644
645 /* Reads count bytes from a file, starting from (and modifiying) offset, and
646  * putting the bytes into buffers described by vector */
647 ssize_t kfs_readv(struct file *file, const struct iovec *vector,
648                   unsigned long count, off_t *offset)
649 {
650         return -1;
651 }
652
653 /* Writes count bytes to a file, starting from (and modifiying) offset, and
654  * taking the bytes from buffers described by vector */
655 ssize_t kfs_writev(struct file *file, const struct iovec *vector,
656                   unsigned long count, off_t *offset)
657 {
658         return -1;
659 }
660
661 /* Write the contents of file to the page.  Will sort the params later */
662 ssize_t kfs_sendpage(struct file *file, struct page *page, int offset,
663                      size_t size, off_t pos, int more)
664 {
665         return -1;
666 }
667
668 /* Checks random FS flags.  Used by NFS. */
669 int kfs_check_flags(int flags)
670 { // default, nothing
671         return -1;
672 }
673
674 /* Redeclaration and initialization of the FS ops structures */
675 struct page_map_operations kfs_pm_op = {
676         kfs_readpage,
677 };
678
679 struct super_operations kfs_s_op = {
680         kfs_alloc_inode,
681         kfs_dealloc_inode,
682         kfs_read_inode,
683         kfs_dirty_inode,
684         kfs_write_inode,
685         kfs_put_inode,
686         kfs_drop_inode,
687         kfs_delete_inode,
688         kfs_put_super,
689         kfs_write_super,
690         kfs_sync_fs,
691         kfs_remount_fs,
692         kfs_umount_begin,
693 };
694
695 struct inode_operations kfs_i_op = {
696         kfs_create,
697         kfs_lookup,
698         kfs_link,
699         kfs_unlink,
700         kfs_symlink,
701         kfs_mkdir,
702         kfs_rmdir,
703         kfs_mknod,
704         kfs_rename,
705         kfs_readlink,
706         kfs_truncate,
707         kfs_permission,
708 };
709
710 struct dentry_operations kfs_d_op = {
711         kfs_d_revalidate,
712         kfs_d_hash,
713         kfs_d_compare,
714         kfs_d_delete,
715         kfs_d_release,
716         kfs_d_iput,
717 };
718
719 struct file_operations kfs_f_op_file = {
720         kfs_llseek,
721         generic_file_read,
722         generic_file_write,
723         kfs_readdir,
724         kfs_mmap,
725         kfs_open,
726         kfs_flush,
727         kfs_release,
728         kfs_fsync,
729         kfs_poll,
730         kfs_readv,
731         kfs_writev,
732         kfs_sendpage,
733         kfs_check_flags,
734 };
735
736 struct file_operations kfs_f_op_dir = {
737         kfs_llseek,
738         generic_dir_read,
739         0,
740         kfs_readdir,
741         kfs_mmap,
742         kfs_open,
743         kfs_flush,
744         kfs_release,
745         kfs_fsync,
746         kfs_poll,
747         kfs_readv,
748         kfs_writev,
749         kfs_sendpage,
750         kfs_check_flags,
751 };
752
753 struct file_operations kfs_f_op_sym = {
754         kfs_llseek,
755         generic_file_read,
756         generic_file_write,
757         kfs_readdir,
758         kfs_mmap,
759         kfs_open,
760         kfs_flush,
761         kfs_release,
762         kfs_fsync,
763         kfs_poll,
764         kfs_readv,
765         kfs_writev,
766         kfs_sendpage,
767         kfs_check_flags,
768 };
769
770 /* KFS Specific Internal Functions */
771
772 /* Need to pass path separately, since we'll recurse on it.  TODO: this recurses,
773  * and takes up a lot of stack space (~270 bytes).  Core 0's KSTACK is 8 pages,
774  * which can handle about 120 levels deep...  Other cores are not so fortunate.
775  * Can rework this if it becomes an issue. */
776 static int __add_kfs_entry(struct dentry *parent, char *path,
777                            struct cpio_bin_hdr *c_bhdr)
778 {
779         char *first_slash = strchr(path, '/');  
780         char dir[MAX_FILENAME_SZ + 1];  /* room for the \0 */
781         size_t dirname_sz;                              /* not counting the \0 */
782         struct dentry *dentry = 0;
783         struct inode *inode;
784         int err;
785         char *symname, old_end;                 /* for symlink manipulation */
786
787         if (first_slash) {
788                 /* get the first part, find that dentry, pass in the second part,
789                  * recurse.  this isn't being smart about extra slashes, dots, or
790                  * anything like that. */
791                 dirname_sz = first_slash - path;
792                 assert(dirname_sz <= MAX_FILENAME_SZ);
793                 strncpy(dir, path, dirname_sz);
794                 dir[dirname_sz] = '\0';
795                 printd("Finding DIR %s in dentry %s (start: %p, size %d)\n", dir,
796                        parent->d_name.name, c_bhdr->c_filestart, c_bhdr->c_filesize);
797                 /* Need to create a dentry for the lookup, and fill in the basic nd */
798                 dentry = get_dentry(parent->d_sb, parent, dir);
799                 /* TODO: use a VFS lookup instead, to use the dcache, thought its not a
800                  * big deal since KFS currently pins all metadata. */
801                 dentry = kfs_lookup(parent->d_inode, dentry, 0);
802                 if (!dentry) {
803                         printk("Missing dir in CPIO archive or something, aborting.\n");
804                         return -1;
805                 }
806                 return __add_kfs_entry(dentry, first_slash + 1, c_bhdr);
807         } else {
808                 /* no directories left in the path.  add the 'file' to the dentry */
809                 printd("Adding file/dir %s to dentry %s (start: %p, size %d)\n", path,
810                        parent->d_name.name, c_bhdr->c_filestart, c_bhdr->c_filesize);
811                 /* Init the dentry for this path */
812                 dentry = get_dentry(parent->d_sb, parent, path);
813                 dcache_put(dentry);                     /* TODO: should set a d_flag too */
814                 /* build the inode */
815                 switch (c_bhdr->c_mode & CPIO_FILE_MASK) {
816                         case (CPIO_DIRECTORY):
817                                 err = create_dir(parent->d_inode, dentry, c_bhdr->c_mode);
818                                 assert(!err);
819                                 break;
820                         case (CPIO_SYMLINK):
821                                 /* writing the '\0' is safe since the next entry is always still
822                                  * in the CPIO (and we are processing sequentially). */
823                                 symname = c_bhdr->c_filestart;
824                                 old_end = symname[c_bhdr->c_filesize];
825                                 symname[c_bhdr->c_filesize] = '\0';
826                                 err = create_symlink(parent->d_inode, dentry, symname,
827                                                      c_bhdr->c_mode & CPIO_PERM_MASK);
828                                 assert(!err);
829                                 symname[c_bhdr->c_filesize] = old_end;
830                                 break;
831                         case (CPIO_REG_FILE):
832                                 err = create_file(parent->d_inode, dentry,
833                                                   c_bhdr->c_mode & CPIO_PERM_MASK);
834                                 assert(!err);
835                                 ((struct kfs_i_info*)dentry->d_inode->i_fs_info)->filestart =
836                                                                                                                 c_bhdr->c_filestart;
837                                 ((struct kfs_i_info*)dentry->d_inode->i_fs_info)->init_size =
838                                                                                                                 c_bhdr->c_filesize;
839                                 break;
840                         default:
841                                 printk("Unknown file type %d in the CPIO!",
842                                        c_bhdr->c_mode & CPIO_FILE_MASK);
843                                 return -1;
844                 }
845                 inode = dentry->d_inode;
846                 /* Set other info from the CPIO entry */
847                 inode->i_uid = c_bhdr->c_uid;
848                 inode->i_gid = c_bhdr->c_gid;
849                 inode->i_atime.tv_sec = c_bhdr->c_mtime;
850                 inode->i_ctime.tv_sec = c_bhdr->c_mtime;
851                 inode->i_mtime.tv_sec = c_bhdr->c_mtime;
852                 inode->i_size = c_bhdr->c_filesize;
853                 //inode->i_XXX = c_bhdr->c_dev;                 /* and friends */
854                 inode->i_bdev = 0;                                              /* assuming blockdev? */
855                 inode->i_socket = FALSE;
856                 inode->i_blocks = c_bhdr->c_filesize;   /* blocksize == 1 */
857         }
858         return 0;
859 }
860
861 /* Adds an entry (from a CPIO archive) to KFS.  This will put all the FS
862  * metadata in memory, instead of having to reparse the entire archive each time
863  * we need to traverse.
864  *
865  * The other option is to just maintain a LL of {FN, FS}, and O(n) scan it.
866  *
867  * The path is a complete path, interpreted from the root of the mount point.
868  * Directories have a size of 0.  so do symlinks, but we don't handle those yet.
869  *
870  * If a directory does not exist for a file, this will return an error.  Don't
871  * use the -depth flag to find when building the CPIO archive, and this won't be
872  * a problem.  (Maybe) */
873 static int add_kfs_entry(struct super_block *sb, struct cpio_bin_hdr *c_bhdr)
874 {
875         char *path = c_bhdr->c_filename;
876         /* Root of the FS, already part of KFS */
877         if (!strcmp(path, "."))
878                 return 0;
879         return __add_kfs_entry(sb->s_mount->mnt_root, path, c_bhdr);
880 }
881
882 void parse_cpio_entries(struct super_block *sb, void *cpio_b)
883 {
884         struct cpio_newc_header *c_hdr = (struct cpio_newc_header*)cpio_b;
885
886         char buf[9] = {0};      /* temp space for strol conversions */
887         size_t namesize = 0;
888         int offset = 0;         /* offset in the cpio archive */
889         struct cpio_bin_hdr *c_bhdr = kmalloc(sizeof(*c_bhdr), 0);
890         memset(c_bhdr, 0, sizeof(*c_bhdr));
891
892         /* read all files and paths */
893         for (; ; c_hdr = (struct cpio_newc_header*)(cpio_b + offset)) {
894                 offset += sizeof(*c_hdr);
895                 if (strncmp(c_hdr->c_magic, "070701", 6)) {
896                         printk("Invalid magic number in CPIO header, aborting.\n");
897                         return;
898                 }
899                 c_bhdr->c_filename = (char*)c_hdr + sizeof(*c_hdr);
900                 namesize = cpio_strntol(buf, c_hdr->c_namesize, 8);
901                 printd("Namesize: %d\n", size);
902                 if (!strcmp(c_bhdr->c_filename, "TRAILER!!!"))
903                         break;
904                 c_bhdr->c_ino = cpio_strntol(buf, c_hdr->c_ino, 8);
905                 c_bhdr->c_mode = (int)cpio_strntol(buf, c_hdr->c_mode, 8);
906                 c_bhdr->c_uid = cpio_strntol(buf, c_hdr->c_uid, 8);
907                 c_bhdr->c_gid = cpio_strntol(buf, c_hdr->c_gid, 8);
908                 c_bhdr->c_nlink = (unsigned int)cpio_strntol(buf, c_hdr->c_nlink, 8);
909                 c_bhdr->c_mtime = cpio_strntol(buf, c_hdr->c_mtime, 8);
910                 c_bhdr->c_filesize = cpio_strntol(buf, c_hdr->c_filesize, 8);
911                 c_bhdr->c_dev_maj = cpio_strntol(buf, c_hdr->c_dev_maj, 8);
912                 c_bhdr->c_dev_min = cpio_strntol(buf, c_hdr->c_dev_min, 8);
913                 c_bhdr->c_rdev_maj = cpio_strntol(buf, c_hdr->c_rdev_maj, 8);
914                 c_bhdr->c_rdev_min = cpio_strntol(buf, c_hdr->c_rdev_min, 8);
915                 printd("File: %s: %d Bytes\n", c_bhdr->c_filename, c_bhdr->c_filesize);
916                 offset += namesize;
917                 /* header + name will be padded out to 4-byte alignment */
918                 offset = ROUNDUP(offset, 4);
919                 c_bhdr->c_filestart = cpio_b + offset;
920                 /* make this a function pointer or something */
921                 if (add_kfs_entry(sb, c_bhdr)) {
922                         printk("Failed to add an entry to KFS!\n");
923                         break;
924                 }
925                 offset += c_bhdr->c_filesize;
926                 offset = ROUNDUP(offset, 4);
927                 //printk("offset is %d bytes\n", offset);
928                 c_hdr = (struct cpio_newc_header*)(cpio_b + offset);
929         }
930         kfree(c_bhdr);
931 }