ext4_xattr: rework the EA submodule
[lwext4.git] / src / ext4.c
index e0923e4700c0ceef31ec9d226e25e4426d0b964f..b529bbde24e047b8b48b7ddfef065d3e719cb317 100644 (file)
  */
 
 #include "ext4_config.h"
-#include "ext4.h"
-#include "ext4_blockdev.h"
 #include "ext4_types.h"
-#include "ext4_debug.h"
+#include "ext4_misc.h"
 #include "ext4_errno.h"
+#include "ext4_oflags.h"
+#include "ext4_debug.h"
+
+#include "ext4.h"
+#include "ext4_trans.h"
+#include "ext4_blockdev.h"
 #include "ext4_fs.h"
 #include "ext4_dir.h"
 #include "ext4_inode.h"
 #include "ext4_super.h"
+#include "ext4_block_group.h"
 #include "ext4_dir_idx.h"
 #include "ext4_xattr.h"
 #include "ext4_journal.h"
@@ -561,7 +566,31 @@ static int __ext4_recover(const char *mount_point)
                jbd_put_fs(jbd_fs);
                free(jbd_fs);
        }
+       if (r == EOK && !mp->fs.read_only) {
+               uint32_t bgid;
+               uint64_t free_blocks_count = 0;
+               uint32_t free_inodes_count = 0;
+               struct ext4_block_group_ref bg_ref;
 
+               /* Update superblock's stats */
+               for (bgid = 0;bgid < ext4_block_group_cnt(&mp->fs.sb);bgid++) {
+                       r = ext4_fs_get_block_group_ref(&mp->fs, bgid, &bg_ref);
+                       if (r != EOK)
+                               goto Finish;
+
+                       free_blocks_count +=
+                               ext4_bg_get_free_blocks_count(bg_ref.block_group,
+                                               &mp->fs.sb);
+                       free_inodes_count +=
+                               ext4_bg_get_free_inodes_count(bg_ref.block_group,
+                                               &mp->fs.sb);
+
+                       ext4_fs_put_block_group_ref(&bg_ref);
+               }
+               ext4_sb_set_free_blocks_cnt(&mp->fs.sb, free_blocks_count);
+               ext4_set32(&mp->fs.sb, free_inodes_count, free_inodes_count);
+               /* We don't need to save the superblock stats immediately. */
+       }
 
 Finish:
        EXT4_MP_UNLOCK(mp);
@@ -1645,9 +1674,6 @@ int ext4_fread(ext4_file *f, void *buf, size_t size, size_t *rcnt)
 
        ext4_assert(f && f->mp);
 
-       if (f->mp->fs.read_only)
-               return EROFS;
-
        if (f->flags & O_WRONLY)
                return EPERM;
 
@@ -2390,13 +2416,13 @@ Finish:
 int ext4_setxattr(const char *path, const char *name, size_t name_len,
                  const void *data, size_t data_size, bool replace)
 {
+       bool found;
        int r = EOK;
        ext4_file f;
        uint32_t inode;
        uint8_t name_index;
        const char *dissected_name = NULL;
        size_t dissected_len = 0;
-       struct ext4_xattr_ref xattr_ref;
        struct ext4_inode_ref inode_ref;
        struct ext4_mountpoint *mp = ext4_get_mount(path);
        if (!mp)
@@ -2406,8 +2432,9 @@ int ext4_setxattr(const char *path, const char *name, size_t name_len,
                return EROFS;
 
        dissected_name = ext4_extract_xattr_name(name, name_len,
-                               &name_index, &dissected_len);
-       if (!dissected_len)
+                               &name_index, &dissected_len,
+                               &found);
+       if (!found)
                return EINVAL;
 
        EXT4_MP_LOCK(mp);
@@ -2423,16 +2450,9 @@ int ext4_setxattr(const char *path, const char *name, size_t name_len,
        if (r != EOK)
                goto Finish;
 
-       r = ext4_fs_get_xattr_ref(&mp->fs, &inode_ref, &xattr_ref);
-       if (r != EOK) {
-               ext4_fs_put_inode_ref(&inode_ref);
-               goto Finish;
-       }
-
-       r = ext4_fs_set_xattr(&xattr_ref, name_index, dissected_name,
-                       dissected_len, data, data_size, replace);
+       r = ext4_xattr_set(&inode_ref, name_index, dissected_name,
+                       dissected_len, data, data_size);
 
-       ext4_fs_put_xattr_ref(&xattr_ref);
        ext4_fs_put_inode_ref(&inode_ref);
 Finish:
        if (r != EOK)
@@ -2447,21 +2467,22 @@ Finish:
 int ext4_getxattr(const char *path, const char *name, size_t name_len,
                  void *buf, size_t buf_size, size_t *data_size)
 {
+       bool found;
        int r = EOK;
        ext4_file f;
        uint32_t inode;
        uint8_t name_index;
        const char *dissected_name = NULL;
        size_t dissected_len = 0;
-       struct ext4_xattr_ref xattr_ref;
        struct ext4_inode_ref inode_ref;
        struct ext4_mountpoint *mp = ext4_get_mount(path);
        if (!mp)
                return ENOENT;
 
        dissected_name = ext4_extract_xattr_name(name, name_len,
-                               &name_index, &dissected_len);
-       if (!dissected_len)
+                               &name_index, &dissected_len,
+                               &found);
+       if (!found)
                return EINVAL;
 
        EXT4_MP_LOCK(mp);
@@ -2475,76 +2496,28 @@ int ext4_getxattr(const char *path, const char *name, size_t name_len,
        if (r != EOK)
                goto Finish;
 
-       r = ext4_fs_get_xattr_ref(&mp->fs, &inode_ref, &xattr_ref);
-       if (r != EOK) {
-               ext4_fs_put_inode_ref(&inode_ref);
-               goto Finish;
-       }
-
-       r = ext4_fs_get_xattr(&xattr_ref, name_index, dissected_name,
+       r = ext4_xattr_get(&inode_ref, name_index, dissected_name,
                                dissected_len, buf, buf_size, data_size);
 
-       ext4_fs_put_xattr_ref(&xattr_ref);
        ext4_fs_put_inode_ref(&inode_ref);
 Finish:
        EXT4_MP_UNLOCK(mp);
        return r;
 }
 
-struct ext4_listxattr_iterator {
-       char *list;
-       char *list_ptr;
-       size_t size;
-       size_t ret_size;
-       bool list_too_small;
-       bool get_required_size;
-};
-
-static int ext4_iterate_ea_list(struct ext4_xattr_ref *ref,
-                               struct ext4_xattr_item *item)
-{
-       struct ext4_listxattr_iterator *lxi;
-       lxi = ref->iter_arg;
-       if (!lxi->get_required_size) {
-               size_t plen;
-               const char *prefix;
-               prefix = ext4_get_xattr_name_prefix(item->name_index, &plen);
-               if (lxi->ret_size + plen + item->name_len + 1 > lxi->size) {
-                       lxi->list_too_small = 1;
-                       return EXT4_XATTR_ITERATE_STOP;
-               }
-               if (prefix) {
-                       memcpy(lxi->list_ptr, prefix, plen);
-                       lxi->list_ptr += plen;
-                       lxi->ret_size += plen;
-               }
-               memcpy(lxi->list_ptr, item->name, item->name_len);
-               lxi->list_ptr[item->name_len] = 0;
-               lxi->list_ptr += item->name_len + 1;
-       }
-       lxi->ret_size += item->name_len + 1;
-       return EXT4_XATTR_ITERATE_CONT;
-}
-
 int ext4_listxattr(const char *path, char *list, size_t size, size_t *ret_size)
 {
        int r = EOK;
        ext4_file f;
        uint32_t inode;
-       struct ext4_xattr_ref xattr_ref;
+       size_t list_len, list_size = 0;
        struct ext4_inode_ref inode_ref;
-       struct ext4_listxattr_iterator lxi;
+       struct ext4_xattr_list_entry *xattr_list = NULL,
+                                    *entry = NULL;
        struct ext4_mountpoint *mp = ext4_get_mount(path);
        if (!mp)
                return ENOENT;
 
-       lxi.list = list;
-       lxi.list_ptr = list;
-       lxi.size = size;
-       lxi.ret_size = 0;
-       lxi.list_too_small = false;
-       lxi.get_required_size = (!size) ? true : false;
-
        EXT4_MP_LOCK(mp);
        r = ext4_generic_open2(&f, path, O_RDWR, EXT4_DE_UNKNOWN, NULL, NULL);
        if (r != EOK)
@@ -2556,39 +2529,70 @@ int ext4_listxattr(const char *path, char *list, size_t size, size_t *ret_size)
        if (r != EOK)
                goto Finish;
 
-       r = ext4_fs_get_xattr_ref(&mp->fs, &inode_ref, &xattr_ref);
-       if (r != EOK) {
-               ext4_fs_put_inode_ref(&inode_ref);
-               goto Finish;
-       }
+       r = ext4_xattr_list(&inode_ref, NULL, &list_len);
+       if (r == EOK && list_len) {
+               xattr_list = malloc(list_len);
+               if (!xattr_list) {
+                       ext4_fs_put_inode_ref(&inode_ref);
+                       r = ENOMEM;
+                       goto Finish;
+               }
+               entry = xattr_list;
+               r = ext4_xattr_list(&inode_ref, entry, &list_len);
+               if (r != EOK) {
+                       ext4_fs_put_inode_ref(&inode_ref);
+                       goto Finish;
+               }
 
-       xattr_ref.iter_arg = &lxi;
-       ext4_fs_xattr_iterate(&xattr_ref, ext4_iterate_ea_list);
-       if (lxi.list_too_small)
-               r = ERANGE;
+               for (;entry;entry = entry->next) {
+                       size_t prefix_len;
+                       const char *prefix =
+                               ext4_get_xattr_name_prefix(entry->name_index,
+                                                          &prefix_len);
+                       if (size) {
+                               if (prefix_len + entry->name_len + 1 > size) {
+                                       ext4_fs_put_inode_ref(&inode_ref);
+                                       r = ERANGE;
+                                       goto Finish;
+                               }
+                       }
 
-       if (r == EOK) {
+                       if (list && size) {
+                               memcpy(list, prefix, prefix_len);
+                               list += prefix_len;
+                               memcpy(list, entry->name,
+                                       entry->name_len);
+                               list[entry->name_len] = 0;
+                               list += entry->name_len + 1;
+
+                               size -= prefix_len + entry->name_len + 1;
+                       }
+
+                       list_size += prefix_len + entry->name_len + 1;
+               }
                if (ret_size)
-                       *ret_size = lxi.ret_size;
+                       *ret_size = list_size;
 
        }
-       ext4_fs_put_xattr_ref(&xattr_ref);
        ext4_fs_put_inode_ref(&inode_ref);
 Finish:
        EXT4_MP_UNLOCK(mp);
+       if (xattr_list)
+               free(xattr_list);
+
        return r;
 
 }
 
 int ext4_removexattr(const char *path, const char *name, size_t name_len)
 {
+       bool found;
        int r = EOK;
        ext4_file f;
        uint32_t inode;
        uint8_t name_index;
        const char *dissected_name = NULL;
        size_t dissected_len = 0;
-       struct ext4_xattr_ref xattr_ref;
        struct ext4_inode_ref inode_ref;
        struct ext4_mountpoint *mp = ext4_get_mount(path);
        if (!mp)
@@ -2598,8 +2602,9 @@ int ext4_removexattr(const char *path, const char *name, size_t name_len)
                return EROFS;
 
        dissected_name = ext4_extract_xattr_name(name, name_len,
-                                               &name_index, &dissected_len);
-       if (!dissected_len)
+                                               &name_index, &dissected_len,
+                                               &found);
+       if (!found)
                return EINVAL;
 
        EXT4_MP_LOCK(mp);
@@ -2615,16 +2620,9 @@ int ext4_removexattr(const char *path, const char *name, size_t name_len)
        if (r != EOK)
                goto Finish;
 
-       r = ext4_fs_get_xattr_ref(&mp->fs, &inode_ref, &xattr_ref);
-       if (r != EOK) {
-               ext4_fs_put_inode_ref(&inode_ref);
-               goto Finish;
-       }
-
-       r = ext4_fs_remove_xattr(&xattr_ref, name_index, dissected_name,
-                               dissected_len);
+       r = ext4_xattr_remove(&inode_ref, name_index, dissected_name,
+                             dissected_len);
 
-       ext4_fs_put_xattr_ref(&xattr_ref);
        ext4_fs_put_inode_ref(&inode_ref);
 Finish:
        if (r != EOK)
@@ -2883,6 +2881,11 @@ End:
        return r;
 }
 
+int ext4_dir_mv(const char *path, const char *new_path)
+{
+       return ext4_frename(path, new_path);
+}
+
 int ext4_dir_mk(const char *path)
 {
        int r;