X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/a5338b60e111972364a8bc6f07011c6defd213d2..34f6af4f61e0eec9c96113e07f140b609b4113c8:/lunaix-os/kernel/fs/twifs/twifs.c

diff --git a/lunaix-os/kernel/fs/twifs/twifs.c b/lunaix-os/kernel/fs/twifs/twifs.c
index f535084..165d0bd 100644
--- a/lunaix-os/kernel/fs/twifs/twifs.c
+++ b/lunaix-os/kernel/fs/twifs/twifs.c
@@ -8,73 +8,41 @@
  * @copyright Copyright (c) 2022
  *
  */
+#include <klibc/strfmt.h>
 #include <klibc/string.h>
 #include <lunaix/clock.h>
-#include <lunaix/fs.h>
+#include <lunaix/fs/api.h>
 #include <lunaix/fs/twifs.h>
+#include <lunaix/fs/twimap.h>
 #include <lunaix/mm/cake.h>
 #include <lunaix/mm/valloc.h>
+#include <lunaix/spike.h>
+
+#include <asm/pagetable.h>
 
 static struct twifs_node* fs_root;
 
 static struct cake_pile* twi_pile;
 
-int
-__twifs_dirlookup(struct v_inode* inode, struct v_dnode* dnode);
+static volatile u32_t inode_id = 0;
 
-int
-__twifs_openfile(struct v_inode* inode, struct v_file* file);
-
-struct twifs_node*
-__twifs_get_node(struct twifs_node* parent, struct hstr* name);
-
-struct v_inode*
-__twifs_create_inode(struct twifs_node* twi_node);
-
-int
-__twifs_iterate_dir(struct v_file* file, struct dir_context* dctx);
-
-int
-__twifs_mount(struct v_superblock* vsb, struct v_dnode* mount_point);
-
-int
-__twifs_mkdir(struct v_inode* inode, struct v_dnode* dnode);
-
-int
-__twifs_rmstuff(struct v_inode* inode);
-
-int
-__twifs_fwrite(struct v_file* file, void* buffer, size_t len, size_t fpos);
-
-int
-__twifs_fread(struct v_file* file, void* buffer, size_t len, size_t fpos);
-
-void
-twifs_init()
-{
-    twi_pile = cake_new_pile("twifs_node", sizeof(struct twifs_node), 1, 0);
-
-    struct filesystem* twifs = vzalloc(sizeof(struct filesystem));
-    twifs->fs_name = HSTR("twifs", 5);
-    twifs->mount = __twifs_mount;
-    twifs->types = FSTYPE_ROFS;
-
-    fsm_register(twifs);
-
-    fs_root = twifs_dir_node(NULL, NULL, 0, 0);
-}
+extern const struct v_file_ops twifs_file_ops;
+extern const struct v_inode_ops twifs_inode_ops;
 
 struct twifs_node*
 __twifs_new_node(struct twifs_node* parent,
                  const char* name,
                  int name_len,
-                 uint32_t itype)
+                 u32_t itype)
 {
     struct twifs_node* node = cake_grab(twi_pile);
     memset(node, 0, sizeof(*node));
 
-    node->name = HSTR(name, name_len);
+    strncpy(node->name_val, name, VFS_NAME_MAXLEN);
+
+    node->name = HSTR(node->name_val, MIN(name_len, VFS_NAME_MAXLEN));
     node->itype = itype;
+    node->ino_id = inode_id++;
     hstr_rehash(&node->name, HSTR_FULL_HASH);
     llist_init_head(&node->children);
 
@@ -85,124 +53,68 @@ __twifs_new_node(struct twifs_node* parent,
     return node;
 }
 
-int
-twifs_rm_node(struct twifs_node* node)
+void
+__twifs_init_inode(struct v_superblock* vsb, struct v_inode* inode)
 {
-    if ((node->itype & VFS_IFDIR) && !llist_empty(&node->children)) {
-        return ENOTEMPTY;
-    }
-    llist_delete(&node->siblings);
-    vfs_i_free(node->inode);
-    cake_release(twi_pile, node);
-    return 0;
+    inode->ops = (struct v_inode_ops*)&twifs_inode_ops;
+    inode->default_fops = (struct v_file_ops*)&twifs_file_ops;
 }
 
-struct twifs_node*
-twifs_file_node(struct twifs_node* parent,
-                const char* name,
-                int name_len,
-                uint32_t itype)
+int
+__twifs_mount(struct v_superblock* vsb, struct v_dnode* mount_point)
 {
-    struct twifs_node* twi_node =
-      __twifs_new_node(parent, name, name_len, VFS_IFFILE | itype);
-
-    struct v_inode* twi_inode = __twifs_create_inode(twi_node);
-    twi_node->inode = twi_inode;
-
-    return twi_node;
-}
+    vsb->ops.init_inode = __twifs_init_inode;
 
-struct twifs_node*
-twifs_dir_node(struct twifs_node* parent,
-               const char* name,
-               int name_len,
-               uint32_t itype)
-{
-    struct hstr hname = HSTR(name, name_len);
-    hstr_rehash(&hname, HSTR_FULL_HASH);
-    struct twifs_node* node = __twifs_get_node(parent, &hname);
-    if (node) {
-        return node;
+    struct v_inode* inode = vfs_i_alloc(vsb);
+    if (!inode) {
+        return ENOMEM;
     }
 
-    struct twifs_node* twi_node =
-      __twifs_new_node(parent, name, name_len, VFS_IFDIR | itype);
-
-    struct v_inode* twi_inode = __twifs_create_inode(twi_node);
-    twi_node->inode = twi_inode;
+    inode->id = fs_root->ino_id;
+    inode->itype = fs_root->itype;
+    inode->data = fs_root;
 
-    return twi_node;
-}
-
-struct twifs_node*
-twifs_toplevel_node(const char* name, int name_len, uint32_t itype)
-{
-    return twifs_dir_node(fs_root, name, name_len, itype);
+    vfs_i_addhash(inode);
+    vfs_assign_inode(mount_point, inode);
+    return 0;
 }
 
 int
-__twifs_mkdir(struct v_inode* inode, struct v_dnode* dnode)
+__twifs_unmount(struct v_superblock* vsb)
 {
-    struct twifs_node* parent_node = (struct twifs_node*)inode->data;
-    if (!(parent_node->itype & VFS_IFDIR)) {
-        return ENOTDIR;
-    }
-    struct twifs_node* new_node =
-      twifs_dir_node(parent_node, dnode->name.value, dnode->name.len, 0);
-    dnode->inode = new_node->inode;
-
     return 0;
 }
 
 int
-__twifs_mount(struct v_superblock* vsb, struct v_dnode* mount_point)
+__twifs_fwrite(struct v_inode* inode, void* buffer, size_t len, size_t fpos)
 {
-    mount_point->inode = fs_root->inode;
-    return 0;
+    struct twifs_node* twi_node = (struct twifs_node*)inode->data;
+    if (!twi_node || !twi_node->ops.write) {
+        return ENOTSUP;
+    }
+    return twi_node->ops.write(inode, buffer, len, fpos);
 }
 
-struct v_inode*
-__twifs_create_inode(struct twifs_node* twi_node)
+int
+__twifs_fwrite_pg(struct v_inode* inode, void* buffer, size_t fpos)
 {
-    struct v_inode* inode = vfs_i_alloc();
-    *inode = (struct v_inode){ .ctime = 0,
-                               .itype = twi_node->itype,
-                               .lb_addr = 0,
-                               .lb_usage = 0,
-                               .data = twi_node,
-                               .mtime = 0,
-                               .open_count = 0 };
-    inode->ops.dir_lookup = __twifs_dirlookup;
-    inode->ops.mkdir = __twifs_mkdir;
-    inode->ops.unlink = __twifs_rmstuff;
-    inode->ops.rmdir = __twifs_rmstuff;
-    inode->ops.open = __twifs_openfile;
-
-    inode->default_fops = (struct v_file_ops){ .read = __twifs_fread,
-                                               .write = __twifs_fwrite,
-                                               .readdir = __twifs_iterate_dir };
-
-    return inode;
+    return __twifs_fwrite(inode, buffer, PAGE_SIZE, fpos);
 }
 
 int
-__twifs_fwrite(struct v_file* file, void* buffer, size_t len, size_t fpos)
+__twifs_fread(struct v_inode* inode, void* buffer, size_t len, size_t fpos)
 {
-    struct twifs_node* twi_node = (struct twifs_node*)file->inode->data;
-    if (!twi_node || !twi_node->ops.write) {
+    struct twifs_node* twi_node = (struct twifs_node*)inode->data;
+    if (!twi_node || !twi_node->ops.read) {
         return ENOTSUP;
     }
-    return twi_node->ops.write(file, buffer, len, fpos);
+    return twi_node->ops.read(inode, buffer, len, fpos);
 }
 
 int
-__twifs_fread(struct v_file* file, void* buffer, size_t len, size_t fpos)
+__twifs_fread_pg(struct v_inode* inode, void* buffer, size_t fpos)
 {
-    struct twifs_node* twi_node = (struct twifs_node*)file->inode->data;
-    if (!twi_node || !twi_node->ops.read) {
-        return ENOTSUP;
-    }
-    return twi_node->ops.read(file, buffer, len, fpos);
+    return __twifs_fread(inode, buffer, PAGE_SIZE, fpos);
 }
 
 struct twifs_node*
@@ -221,25 +133,35 @@ __twifs_get_node(struct twifs_node* parent, struct hstr* name)
     return NULL;
 }
 
-int
-__twifs_rmstuff(struct v_inode* inode)
-{
-    struct twifs_node* twi_node = (struct twifs_node*)inode->data;
-    return twifs_rm_node(twi_node);
-}
-
 int
 __twifs_dirlookup(struct v_inode* inode, struct v_dnode* dnode)
 {
     struct twifs_node* twi_node = (struct twifs_node*)inode->data;
 
-    if (!(twi_node->itype & VFS_IFDIR)) {
+    if (!check_directory_node(inode)) {
         return ENOTDIR;
     }
 
     struct twifs_node* child_node = __twifs_get_node(twi_node, &dnode->name);
     if (child_node) {
-        dnode->inode = child_node->inode;
+        struct v_inode* child_inode = vfs_i_find(inode->sb, child_node->ino_id);
+        if (child_inode) {
+            goto done;
+        }
+
+        if (!(child_inode = vfs_i_alloc(inode->sb))) {
+            return ENOENT;
+        }
+
+        child_inode->id = child_node->ino_id;
+        child_inode->itype = child_node->itype;
+        child_inode->data = child_node;
+
+        vfs_i_addhash(child_inode);
+
+    done:
+        dnode->data = child_node->data;
+        vfs_assign_inode(dnode, child_inode);
         return 0;
     }
     return ENOENT;
@@ -249,20 +171,23 @@ int
 __twifs_iterate_dir(struct v_file* file, struct dir_context* dctx)
 {
     struct twifs_node* twi_node = (struct twifs_node*)(file->inode->data);
-    int counter = 0;
+    unsigned int counter = 2;
     struct twifs_node *pos, *n;
 
+    if (fsapi_handle_pseudo_dirent(file, dctx)) {
+        return 1;
+    }
+
     llist_for_each(pos, n, &twi_node->children, siblings)
     {
-        if (counter++ >= dctx->index) {
-            dctx->index = counter;
-            dctx->read_complete_callback(
-              dctx, pos->name.value, pos->name.len, pos->itype);
-            return 0;
+        if (counter++ >= file->f_pos) {
+            fsapi_dir_report(
+              dctx, pos->name.value, pos->name.len, vfs_get_dtype(pos->itype));
+            return 1;
         }
     }
 
-    return 1;
+    return 0;
 }
 
 int
@@ -274,3 +199,108 @@ __twifs_openfile(struct v_inode* inode, struct v_file* file)
     }
     return ENOTSUP;
 }
+
+int
+twifs_rm_node(struct twifs_node* node)
+{
+    if (check_itype(node->itype, VFS_IFDIR) && !llist_empty(&node->children)) {
+        return ENOTEMPTY;
+    }
+    llist_delete(&node->siblings);
+    cake_release(twi_pile, node);
+    return 0;
+}
+
+struct twifs_node*
+twifs_file_node_vargs(struct twifs_node* parent, const char* fmt, va_list args)
+{
+    char buf[VFS_NAME_MAXLEN];
+    size_t len = ksnprintfv(buf, fmt, VFS_NAME_MAXLEN, args);
+
+    return __twifs_new_node(parent ? parent : fs_root, buf, len, F_FILE);
+}
+
+struct twifs_node*
+twifs_file_node(struct twifs_node* parent, const char* fmt, ...)
+{
+    va_list args;
+    va_start(args, fmt);
+
+    struct twifs_node* twi_node = twifs_file_node_vargs(parent, fmt, args);
+
+    va_end(args);
+
+    return twi_node;
+}
+
+struct twifs_node*
+twifs_dir_node(struct twifs_node* parent, const char* fmt, ...)
+{
+    va_list args;
+    va_start(args, fmt);
+
+    char buf[VFS_NAME_MAXLEN];
+    size_t len = ksnprintfv(buf, fmt, VFS_NAME_MAXLEN, args);
+    struct twifs_node* twi_node =
+      __twifs_new_node(parent ? parent : fs_root, buf, len, VFS_IFDIR);
+
+    va_end(args);
+
+    return twi_node;
+}
+
+void
+twifs_init()
+{
+    struct filesystem* fs;
+    fs = fsapi_fs_declare("twifs", FSTYPE_PSEUDO | FSTYPE_ROFS);
+    
+    fsapi_fs_set_mntops(fs, __twifs_mount, __twifs_unmount);
+    fsapi_fs_finalise(fs);
+
+    twi_pile = cake_new_pile("twifs_node", sizeof(struct twifs_node), 1, 0);
+    fs_root = twifs_dir_node(NULL, NULL, 0, 0);
+}
+EXPORT_FILE_SYSTEM(twifs, twifs_init);
+
+void
+twifs_register_plugins()
+{
+    ldga_invoke_fn0(twiplugin_inits);
+}
+
+int
+__twifs_twimap_file_read(struct v_inode* inode,
+                         void* buf,
+                         size_t len,
+                         size_t fpos)
+{
+    struct twimap* map = twinode_getdata(inode, struct twimap*);
+    return twimap_read(map, buf, len, fpos);
+}
+
+struct twimap*
+twifs_mapping(struct twifs_node* parent, void* data, const char* fmt, ...)
+{
+    va_list args;
+    va_start(args, fmt);
+
+    struct twimap* map = twimap_create(data);
+    struct twifs_node* node = twifs_file_node_vargs(parent, fmt, args);
+    node->ops.read = __twifs_twimap_file_read;
+    node->data = map;
+
+    return map;
+}
+
+const struct v_file_ops twifs_file_ops = { .close = default_file_close,
+                                           .read = __twifs_fread,
+                                           .read_page = __twifs_fread_pg,
+                                           .write = __twifs_fwrite,
+                                           .write_page = __twifs_fwrite_pg,
+                                           .readdir = __twifs_iterate_dir };
+
+const struct v_inode_ops twifs_inode_ops = { .dir_lookup = __twifs_dirlookup,
+                                             .mkdir = default_inode_mkdir,
+                                             .rmdir = default_inode_rmdir,
+                                             .open = __twifs_openfile };
\ No newline at end of file