Second Extended Filesystem (ext2) and other improvements (#33)

[lunaix-os.git] / lunaix-os / kernel / fs / ext2 / alloc.c

diff --git a/lunaix-os/kernel/fs/ext2/alloc.c b/lunaix-os/kernel/fs/ext2/alloc.c
new file mode 100644 (file)
index 0000000..2996141
--- /dev/null
+++ b/lunaix-os/kernel/fs/ext2/alloc.c
@@ -0,0 +1,99 @@
+#include "ext2.h"
+
+static inline unsigned int
+__ext2_global_slot_alloc(struct v_superblock* vsb, int type_sel, 
+                         struct ext2_gdesc** gd_out)
+{
+    struct ext2_sbinfo* sb;
+    struct ext2_gdesc *pos;
+    struct llist_header *header;
+    
+    sb = EXT2_SB(vsb);
+    header = &sb->free_list_sel[type_sel];
+    
+    if (type_sel == GDESC_INO_SEL) {
+        pos = list_entry(header->next, struct ext2_gdesc, free_grps_ino);
+    } 
+    else {
+        pos = list_entry(header->next, struct ext2_gdesc, free_grps_blk);
+    }
+
+    int alloc = ext2gd_alloc_slot(pos, type_sel);
+
+    if (valid_bmp_slot(alloc)) {
+        *gd_out = pos;
+    }
+
+    return alloc;
+}
+
+int
+ext2ino_alloc_slot(struct v_superblock* vsb, struct ext2_gdesc** gd_out)
+{
+    return __ext2_global_slot_alloc(vsb, GDESC_INO_SEL, gd_out);
+}
+
+int
+ext2db_alloc_slot(struct v_superblock* vsb, struct ext2_gdesc** gd_out)
+{
+    return __ext2_global_slot_alloc(vsb, GDESC_BLK_SEL, gd_out);
+}
+
+int
+ext2gd_alloc_slot(struct ext2_gdesc* gd, int type_sel) 
+{
+    struct ext2_bmp* bmp;
+    struct ext2_sbinfo *sb;
+    int alloc;
+    
+    sb = gd->sb;
+    bmp = &gd->bmps[type_sel];
+    alloc = ext2bmp_alloc_one(bmp);
+    
+    if (alloc < 0) {
+        return alloc;
+    }
+
+    if (!ext2bmp_check_free(bmp)) {
+        llist_delete(&gd->free_list_sel[type_sel]);
+    }
+
+    if (type_sel == GDESC_INO_SEL) {
+        gd->info->bg_free_ino_cnt--;
+        sb->raw->s_free_ino_cnt--;
+    } else {
+        gd->info->bg_free_blk_cnt--;
+        sb->raw->s_free_blk_cnt--;
+    }
+
+    fsblock_dirty(gd->buf);
+    fsblock_dirty(sb->buf);
+    return alloc;
+}
+
+void
+ext2gd_free_slot(struct ext2_gdesc* gd, int type_sel, int slot)
+{
+    struct llist_header *free_ent, *free_list;
+    struct ext2_sbinfo *sb;
+
+    ext2bmp_free_one(&gd->bmps[type_sel], slot);
+
+    sb = gd->sb;
+    free_ent  = &gd->free_list_sel[slot];
+    free_list = &gd->sb->free_list_sel[slot];
+    if (llist_empty(free_ent)) {
+        llist_append(free_list, free_ent);
+    }
+
+    if (type_sel == GDESC_INO_SEL) {
+        gd->info->bg_free_ino_cnt++;
+        sb->raw->s_free_ino_cnt++;
+    } else {
+        gd->info->bg_free_blk_cnt++;
+        sb->raw->s_free_blk_cnt++;
+    }
+
+    fsblock_dirty(gd->buf);
+    fsblock_dirty(sb->buf);
+}
\ No newline at end of file