[EXT2] Upgrade to 0.69
[reactos.git] / drivers / filesystems / ext2 / src / ext4 / ext4_xattr.c
diff --git a/drivers/filesystems/ext2/src/ext4/ext4_xattr.c b/drivers/filesystems/ext2/src/ext4/ext4_xattr.c
new file mode 100644 (file)
index 0000000..e1a26e9
--- /dev/null
@@ -0,0 +1,1295 @@
+/*
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public Licens
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-
+ */
+
+#include <ext2fs.h>
+#include <linux/module.h>
+#include <linux/ext4_xattr.h>
+
+static ext4_fsblk_t ext4_new_meta_blocks(void *icb, struct inode *inode,
+       ext4_fsblk_t goal,
+       unsigned int flags,
+       unsigned long *count, int *errp)
+{
+       NTSTATUS status;
+       ULONG blockcnt = (count) ? *count : 1;
+       ULONG block = 0;
+
+       status = Ext2NewBlock((PEXT2_IRP_CONTEXT)icb,
+               inode->i_sb->s_priv,
+               0, (ULONG)goal,
+               &block,
+               &blockcnt);
+       if (count)
+               *count = blockcnt;
+
+       if (!NT_SUCCESS(status)) {
+               *errp = Ext2LinuxError(status);
+               return 0;
+       }
+       inode->i_blocks += (blockcnt * (inode->i_sb->s_blocksize >> 9));
+       return block;
+}
+
+static void ext4_free_blocks(void *icb, struct inode *inode,
+       ext4_fsblk_t block, int count, int flags)
+{
+       Ext2FreeBlock((PEXT2_IRP_CONTEXT)icb, inode->i_sb->s_priv, (ULONG)block, count);
+       inode->i_blocks -= count * (inode->i_sb->s_blocksize >> 9);
+       return;
+}
+
+static inline ext4_fsblk_t ext4_inode_to_goal_block(struct inode *inode)
+{
+       PEXT2_VCB Vcb;
+       Vcb = inode->i_sb->s_priv;
+       return (inode->i_ino - 1) / BLOCKS_PER_GROUP;
+}
+
+#define NAME_HASH_SHIFT 5
+#define VALUE_HASH_SHIFT 16
+
+static inline void ext4_xattr_compute_hash(struct ext4_xattr_header *header,
+                                          struct ext4_xattr_entry *entry)
+{
+       __u32 hash = 0;
+       char *name = EXT4_XATTR_NAME(entry);
+       int n;
+
+       for (n = 0; n < entry->e_name_len; n++) {
+               hash = (hash << NAME_HASH_SHIFT) ^
+                      (hash >> (8 * sizeof(hash) - NAME_HASH_SHIFT)) ^ *name++;
+       }
+
+       if (entry->e_value_block == 0 && entry->e_value_size != 0) {
+               __le32 *value =
+                   (__le32 *)((char *)header + le16_to_cpu(entry->e_value_offs));
+               for (n = (le32_to_cpu(entry->e_value_size) + EXT4_XATTR_ROUND) >>
+                        EXT4_XATTR_PAD_BITS;
+                    n; n--) {
+                       hash = (hash << VALUE_HASH_SHIFT) ^
+                              (hash >> (8 * sizeof(hash) - VALUE_HASH_SHIFT)) ^
+                              le32_to_cpu(*value++);
+               }
+       }
+       entry->e_hash = cpu_to_le32(hash);
+}
+
+#define BLOCK_HASH_SHIFT 16
+
+/*
+ * ext4_xattr_rehash()
+ *
+ * Re-compute the extended attribute hash value after an entry has changed.
+ */
+static void ext4_xattr_rehash(struct ext4_xattr_header *header,
+                             struct ext4_xattr_entry *entry)
+{
+       struct ext4_xattr_entry *here;
+       __u32 hash = 0;
+
+       ext4_xattr_compute_hash(header, entry);
+       here = EXT4_XATTR_ENTRY(header + 1);
+       while (!EXT4_XATTR_IS_LAST_ENTRY(here)) {
+               if (!here->e_hash) {
+                       /* Block is not shared if an entry's hash value == 0 */
+                       hash = 0;
+                       break;
+               }
+               hash = (hash << BLOCK_HASH_SHIFT) ^
+                      (hash >> (8 * sizeof(hash) - BLOCK_HASH_SHIFT)) ^
+                      le32_to_cpu(here->e_hash);
+               here = EXT4_XATTR_NEXT(here);
+       }
+       header->h_hash = cpu_to_le32(hash);
+}
+
+#if CONFIG_META_CSUM_ENABLE
+static __u32
+ext4_xattr_block_checksum(PEXT2_MCB inode_ref,
+                         ext4_fsblk_t blocknr,
+                         struct ext4_xattr_header *header)
+{
+       __u32 checksum = 0;
+       __u64 le64_blocknr = blocknr;
+       struct ext4_sblock *sb = &inode_ref->fs->sb;
+
+       if (ext4_sb_feature_ro_com(sb, EXT4_FRO_COM_METADATA_CSUM)) {
+               __u32 orig_checksum;
+
+               /* Preparation: temporarily set bg checksum to 0 */
+               orig_checksum = header->h_checksum;
+               header->h_checksum = 0;
+               /* First calculate crc32 checksum against fs uuid */
+               checksum = ext4_crc32c(EXT4_CRC32_INIT, sb->uuid,
+                               sizeof(sb->uuid));
+               /* Then calculate crc32 checksum block number */
+               checksum = ext4_crc32c(checksum, &le64_blocknr,
+                                    sizeof(le64_blocknr));
+               /* Finally calculate crc32 checksum against 
+                * the entire xattr block */
+               checksum = ext4_crc32c(checksum, header,
+                                  ext4_sb_get_block_size(sb));
+               header->h_checksum = orig_checksum;
+       }
+       return checksum;
+}
+#else
+#define ext4_xattr_block_checksum(...) 0
+#endif
+
+static void
+ext4_xattr_set_block_checksum(PEXT2_MCB inode_ref,
+                             ext4_fsblk_t blocknr,
+                             struct ext4_xattr_header *header)
+{
+       /* TODO: Need METADATA_CSUM supports. */
+       header->h_checksum = 0;
+}
+
+static int ext4_xattr_item_cmp(struct rb_node *_a,
+                              struct rb_node *_b)
+{
+       int result;
+       struct ext4_xattr_item *a, *b;
+       a = container_of(_a, struct ext4_xattr_item, node);
+       a = container_of(_a, struct ext4_xattr_item, node);
+       b = container_of(_b, struct ext4_xattr_item, node);
+
+       if (a->is_data && !b->is_data)
+               return -1;
+       
+       if (!a->is_data && b->is_data)
+               return 1;
+
+       result = a->name_index - b->name_index;
+       if (result)
+               return result;
+
+       if (a->name_len < b->name_len)
+               return -1;
+
+       if (a->name_len > b->name_len)
+               return 1;
+
+       return memcmp(a->name, b->name, a->name_len);
+}
+
+//
+// Red-black tree insert routine.
+//
+
+static struct ext4_xattr_item *
+ext4_xattr_item_search(struct ext4_xattr_ref *xattr_ref,
+                      struct ext4_xattr_item *name)
+{
+       struct rb_node *new = xattr_ref->root.rb_node;
+
+       while (new) {
+               struct ext4_xattr_item *node =
+                       container_of(new, struct ext4_xattr_item, node);
+               int result = ext4_xattr_item_cmp(&name->node, new);
+
+               if (result < 0)
+                       new = new->rb_left;
+               else if (result > 0)
+                       new = new->rb_right;
+               else
+                       return node;
+
+       }
+
+       return NULL;
+}
+
+static void ext4_xattr_item_insert(struct ext4_xattr_ref *xattr_ref,
+                                  struct ext4_xattr_item *item)
+{
+       rb_insert(&xattr_ref->root, &item->node,
+             ext4_xattr_item_cmp);
+       list_add_tail(&item->list_node, &xattr_ref->ordered_list);
+}
+
+static void ext4_xattr_item_remove(struct ext4_xattr_ref *xattr_ref,
+                                  struct ext4_xattr_item *item)
+{
+       rb_erase(&item->node, &xattr_ref->root);
+       list_del_init(&item->list_node);
+}
+
+static struct ext4_xattr_item *
+ext4_xattr_item_alloc(__u8 name_index, const char *name, size_t name_len)
+{
+       struct ext4_xattr_item *item;
+       item = kzalloc(sizeof(struct ext4_xattr_item) + name_len, GFP_NOFS);
+       if (!item)
+               return NULL;
+
+       item->name_index = name_index;
+       item->name = (char *)(item + 1);
+       item->name_len = name_len;
+       item->data = NULL;
+       item->data_size = 0;
+       INIT_LIST_HEAD(&item->list_node);
+
+       memcpy(item->name, name, name_len);
+
+       if (name_index == EXT4_XATTR_INDEX_SYSTEM &&
+           name_len == 4 &&
+           !memcmp(name, "data", 4))
+               item->is_data = TRUE;
+       else
+               item->is_data = FALSE;
+
+       return item;
+}
+
+static int ext4_xattr_item_alloc_data(struct ext4_xattr_item *item,
+                                     const void *orig_data, size_t data_size)
+{
+       void *data = NULL;
+       ASSERT(!item->data);
+       data = kmalloc(data_size, GFP_NOFS);
+       if (!data)
+               return -ENOMEM;
+
+       if (orig_data)
+               memcpy(data, orig_data, data_size);
+
+       item->data = data;
+       item->data_size = data_size;
+       return 0;
+}
+
+static void ext4_xattr_item_free_data(struct ext4_xattr_item *item)
+{
+       ASSERT(item->data);
+       kfree(item->data);
+       item->data = NULL;
+       item->data_size = 0;
+}
+
+static int ext4_xattr_item_resize_data(struct ext4_xattr_item *item,
+                                      size_t new_data_size)
+{
+       if (new_data_size != item->data_size) {
+               void *new_data;
+               new_data = kmalloc(new_data_size, GFP_NOFS);
+               if (!new_data)
+                       return -ENOMEM;
+
+               memcpy(new_data, item->data, item->data_size);
+               kfree(item->data);
+
+               item->data = new_data;
+               item->data_size = new_data_size;
+       }
+       return 0;
+}
+
+static void ext4_xattr_item_free(struct ext4_xattr_item *item)
+{
+       if (item->data)
+               ext4_xattr_item_free_data(item);
+
+       kfree(item);
+}
+
+static void *ext4_xattr_entry_data(struct ext4_xattr_ref *xattr_ref,
+                                  struct ext4_xattr_entry *entry,
+                                  BOOL in_inode)
+{
+       char *ret;
+       int block_size;
+       if (in_inode) {
+               struct ext4_xattr_ibody_header *header;
+               struct ext4_xattr_entry *first_entry;
+               int inode_size = xattr_ref->fs->InodeSize;
+               header = EXT4_XATTR_IHDR(xattr_ref->OnDiskInode);
+               first_entry = EXT4_XATTR_IFIRST(header);
+
+               ret = ((char *)first_entry + le16_to_cpu(entry->e_value_offs));
+               if (ret + EXT4_XATTR_SIZE(le32_to_cpu(entry->e_value_size)) -
+                       (char *)xattr_ref->OnDiskInode > inode_size)
+                       ret = NULL;
+
+               return ret;
+
+       }
+       block_size = xattr_ref->fs->BlockSize;
+       ret = ((char *)xattr_ref->block_bh->b_data + le16_to_cpu(entry->e_value_offs));
+       if (ret + EXT4_XATTR_SIZE(le32_to_cpu(entry->e_value_size)) -
+                       (char *)xattr_ref->block_bh->b_data > block_size)
+               ret = NULL;
+       return ret;
+}
+
+static int ext4_xattr_block_fetch(struct ext4_xattr_ref *xattr_ref)
+{
+       int ret = 0;
+       size_t size_rem;
+       void *data;
+       struct ext4_xattr_entry *entry = NULL;
+
+       ASSERT(xattr_ref->block_bh->b_data);
+       entry = EXT4_XATTR_BFIRST(xattr_ref->block_bh);
+
+       size_rem = xattr_ref->fs->BlockSize;
+       for (; size_rem > 0 && !EXT4_XATTR_IS_LAST_ENTRY(entry);
+            entry = EXT4_XATTR_NEXT(entry),
+            size_rem -= EXT4_XATTR_LEN(entry->e_name_len)) {
+               struct ext4_xattr_item *item;
+               char *e_name = EXT4_XATTR_NAME(entry);
+
+               data = ext4_xattr_entry_data(xattr_ref, entry, FALSE);
+               if (!data) {
+                       ret = -EIO;
+                       goto Finish;
+               }
+
+               item = ext4_xattr_item_alloc(entry->e_name_index, e_name,
+                                            (size_t)entry->e_name_len);
+               if (!item) {
+                       ret = -ENOMEM;
+                       goto Finish;
+               }
+               if (ext4_xattr_item_alloc_data(
+                       item, data, le32_to_cpu(entry->e_value_size)) != 0) {
+                       ext4_xattr_item_free(item);
+                       ret = -ENOMEM;
+                       goto Finish;
+               }
+               ext4_xattr_item_insert(xattr_ref, item);
+               xattr_ref->block_size_rem -=
+                       EXT4_XATTR_SIZE(item->data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+               xattr_ref->ea_size += EXT4_XATTR_SIZE(item->data_size) +
+                                     EXT4_XATTR_LEN(item->name_len);
+       }
+
+Finish:
+       return ret;
+}
+
+static int ext4_xattr_inode_fetch(struct ext4_xattr_ref *xattr_ref)
+{
+       void *data;
+       size_t size_rem;
+       int ret = 0;
+       struct ext4_xattr_ibody_header *header = NULL;
+       struct ext4_xattr_entry *entry = NULL;
+       int inode_size = xattr_ref->fs->InodeSize;
+
+       header = EXT4_XATTR_IHDR(xattr_ref->OnDiskInode);
+       entry = EXT4_XATTR_IFIRST(header);
+
+       size_rem = inode_size - EXT4_GOOD_OLD_INODE_SIZE -
+                  xattr_ref->OnDiskInode->i_extra_isize;
+       for (; size_rem > 0 && !EXT4_XATTR_IS_LAST_ENTRY(entry);
+            entry = EXT4_XATTR_NEXT(entry),
+            size_rem -= EXT4_XATTR_LEN(entry->e_name_len)) {
+               struct ext4_xattr_item *item;
+               char *e_name = EXT4_XATTR_NAME(entry);
+
+               data = ext4_xattr_entry_data(xattr_ref, entry, TRUE);
+               if (!data) {
+                       ret = -EIO;
+                       goto Finish;
+               }
+
+               item = ext4_xattr_item_alloc(entry->e_name_index, e_name,
+                                            (size_t)entry->e_name_len);
+               if (!item) {
+                       ret = -ENOMEM;
+                       goto Finish;
+               }
+               if (ext4_xattr_item_alloc_data(
+                       item, data, le32_to_cpu(entry->e_value_size)) != 0) {
+                       ext4_xattr_item_free(item);
+                       ret = -ENOMEM;
+                       goto Finish;
+               }
+               item->in_inode = TRUE;
+               ext4_xattr_item_insert(xattr_ref, item);
+               xattr_ref->inode_size_rem -=
+                       EXT4_XATTR_SIZE(item->data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+               xattr_ref->ea_size += EXT4_XATTR_SIZE(item->data_size) +
+                                     EXT4_XATTR_LEN(item->name_len);
+       }
+
+Finish:
+       return ret;
+}
+
+static __s32 ext4_xattr_inode_space(struct ext4_xattr_ref *xattr_ref)
+{
+       int inode_size = xattr_ref->fs->InodeSize;
+       int size_rem = inode_size - EXT4_GOOD_OLD_INODE_SIZE -
+                           xattr_ref->OnDiskInode->i_extra_isize;
+       return size_rem;
+}
+
+static __s32 ext4_xattr_block_space(struct ext4_xattr_ref *xattr_ref)
+{
+       return xattr_ref->fs->BlockSize;
+}
+
+static int ext4_xattr_fetch(struct ext4_xattr_ref *xattr_ref)
+{
+       int ret = 0;
+       int inode_size = xattr_ref->fs->InodeSize;
+       if (inode_size > EXT4_GOOD_OLD_INODE_SIZE) {
+               ret = ext4_xattr_inode_fetch(xattr_ref);
+               if (ret != 0)
+                       return ret;
+       }
+
+       if (xattr_ref->block_loaded)
+               ret = ext4_xattr_block_fetch(xattr_ref);
+
+       xattr_ref->dirty = FALSE;
+       return ret;
+}
+
+static struct ext4_xattr_item *
+ext4_xattr_lookup_item(struct ext4_xattr_ref *xattr_ref, __u8 name_index,
+                      const char *name, size_t name_len)
+{
+       struct ext4_xattr_item tmp = {
+               FALSE,
+               FALSE,
+               name_index,
+               (char *)name, /*won't touch this string*/
+               name_len,
+       };
+       if (name_index == EXT4_XATTR_INDEX_SYSTEM &&
+           name_len == 4 &&
+           !memcmp(name, "data", 4))
+               tmp.is_data = TRUE;
+
+       return ext4_xattr_item_search(xattr_ref, &tmp);
+}
+
+static struct ext4_xattr_item *
+ext4_xattr_insert_item(struct ext4_xattr_ref *xattr_ref, __u8 name_index,
+                      const char *name, size_t name_len, const void *data,
+                      size_t data_size,
+                      int *err)
+{
+       struct ext4_xattr_item *item;
+       item = ext4_xattr_item_alloc(name_index, name, name_len);
+       if (!item) {
+               if (err)
+                       *err = -ENOMEM;
+
+               return NULL;
+       }
+
+       item->in_inode = TRUE;
+       if (xattr_ref->inode_size_rem <
+          EXT4_XATTR_SIZE(data_size) +
+          EXT4_XATTR_LEN(item->name_len)) {
+               if (xattr_ref->block_size_rem <
+                  EXT4_XATTR_SIZE(data_size) +
+                  EXT4_XATTR_LEN(item->name_len)) {
+                       if (err)
+                               *err = -ENOSPC;
+
+                       return NULL;
+               }
+
+               item->in_inode = FALSE;
+       }
+       if (ext4_xattr_item_alloc_data(item, data, data_size) != 0) {
+               ext4_xattr_item_free(item);
+               if (err)
+                       *err = -ENOMEM;
+
+               return NULL;
+       }
+       ext4_xattr_item_insert(xattr_ref, item);
+       xattr_ref->ea_size +=
+           EXT4_XATTR_SIZE(item->data_size) + EXT4_XATTR_LEN(item->name_len);
+       if (item->in_inode) {
+               xattr_ref->inode_size_rem -=
+                       EXT4_XATTR_SIZE(item->data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+       } else {
+               xattr_ref->block_size_rem -=
+                       EXT4_XATTR_SIZE(item->data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+       }
+       xattr_ref->dirty = TRUE;
+       if (err)
+               *err = 0;
+
+       return item;
+}
+
+static struct ext4_xattr_item *
+ext4_xattr_insert_item_ordered(struct ext4_xattr_ref *xattr_ref, __u8 name_index,
+       const char *name, size_t name_len, const void *data,
+       size_t data_size,
+       int *err)
+{
+       struct ext4_xattr_item *item, *last_item = NULL;
+       item = ext4_xattr_item_alloc(name_index, name, name_len);
+       if (!item) {
+               if (err)
+                       *err = -ENOMEM;
+
+               return NULL;
+       }
+
+       if (!list_empty(&xattr_ref->ordered_list))
+               last_item = list_entry(xattr_ref->ordered_list.prev,
+                                       struct ext4_xattr_item,
+                                       list_node);
+
+       item->in_inode = TRUE;
+       if ((xattr_ref->inode_size_rem <
+               EXT4_XATTR_SIZE(data_size) +
+               EXT4_XATTR_LEN(item->name_len))
+                       ||
+               (last_item && !last_item->in_inode)) {
+               if (xattr_ref->block_size_rem <
+                       EXT4_XATTR_SIZE(data_size) +
+                       EXT4_XATTR_LEN(item->name_len)) {
+                       if (err)
+                               *err = -ENOSPC;
+
+                       return NULL;
+               }
+
+               item->in_inode = FALSE;
+       }
+       if (ext4_xattr_item_alloc_data(item, data, data_size) != 0) {
+               ext4_xattr_item_free(item);
+               if (err)
+                       *err = -ENOMEM;
+
+               return NULL;
+       }
+       ext4_xattr_item_insert(xattr_ref, item);
+       xattr_ref->ea_size +=
+               EXT4_XATTR_SIZE(item->data_size) + EXT4_XATTR_LEN(item->name_len);
+       if (item->in_inode) {
+               xattr_ref->inode_size_rem -=
+                       EXT4_XATTR_SIZE(item->data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+       }
+       else {
+               xattr_ref->block_size_rem -=
+                       EXT4_XATTR_SIZE(item->data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+       }
+       xattr_ref->dirty = TRUE;
+       if (err)
+               *err = 0;
+
+       return item;
+}
+
+static int ext4_xattr_remove_item(struct ext4_xattr_ref *xattr_ref,
+                                 __u8 name_index, const char *name,
+                                 size_t name_len)
+{
+       int ret = -ENOENT;
+       struct ext4_xattr_item *item =
+           ext4_xattr_lookup_item(xattr_ref, name_index, name, name_len);
+       if (item) {
+               if (item == xattr_ref->iter_from) {
+                       struct rb_node *next_node;
+                       next_node = rb_next(&item->node);
+                       if (next_node)
+                               xattr_ref->iter_from =
+                                       container_of(next_node,
+                                                    struct ext4_xattr_item,
+                                                    node);
+                       else
+                               xattr_ref->iter_from = NULL;
+               }
+
+               xattr_ref->ea_size -= EXT4_XATTR_SIZE(item->data_size) +
+                                     EXT4_XATTR_LEN(item->name_len);
+
+               if (item->in_inode) {
+                       xattr_ref->inode_size_rem +=
+                               EXT4_XATTR_SIZE(item->data_size) +
+                               EXT4_XATTR_LEN(item->name_len);
+               } else {
+                       xattr_ref->block_size_rem +=
+                               EXT4_XATTR_SIZE(item->data_size) +
+                               EXT4_XATTR_LEN(item->name_len);
+               }
+
+               ext4_xattr_item_remove(xattr_ref, item);
+               ext4_xattr_item_free(item);
+               xattr_ref->dirty = TRUE;
+               ret = 0;
+       }
+       return ret;
+}
+
+static int ext4_xattr_resize_item(struct ext4_xattr_ref *xattr_ref,
+                                 struct ext4_xattr_item *item,
+                                 size_t new_data_size)
+{
+       int ret = 0;
+       BOOL to_inode = FALSE, to_block = FALSE;
+       size_t old_data_size = item->data_size;
+       size_t orig_room_size = item->in_inode ?
+               xattr_ref->inode_size_rem :
+               xattr_ref->block_size_rem;
+
+       /*
+        * Check if we can hold this entry in both in-inode and
+        * on-block form
+        *
+        * More complicated case: we do not allow entries stucking in
+        * the middle between in-inode space and on-block space, so
+        * the entry has to stay in either inode space or block space.
+        */
+       if (item->in_inode) {
+               if (xattr_ref->inode_size_rem +
+                              EXT4_XATTR_SIZE(old_data_size) <
+                              EXT4_XATTR_SIZE(new_data_size)) {
+                       if (xattr_ref->block_size_rem <
+                                      EXT4_XATTR_SIZE(new_data_size) +
+                                      EXT4_XATTR_LEN(item->name_len))
+                               return -ENOSPC;
+
+                       to_block = TRUE;
+               }
+       } else {
+               if (xattr_ref->block_size_rem +
+                               EXT4_XATTR_SIZE(old_data_size) <
+                               EXT4_XATTR_SIZE(new_data_size)) {
+                       if (xattr_ref->inode_size_rem <
+                                       EXT4_XATTR_SIZE(new_data_size) +
+                                       EXT4_XATTR_LEN(item->name_len))
+                               return -ENOSPC;
+
+                       to_inode = TRUE;
+               }
+       }
+       ret = ext4_xattr_item_resize_data(item, new_data_size);
+       if (ret)
+               return ret;
+
+       xattr_ref->ea_size =
+           xattr_ref->ea_size -
+           EXT4_XATTR_SIZE(old_data_size) +
+           EXT4_XATTR_SIZE(new_data_size);
+
+       /*
+        * This entry may originally lie in inode space or block space,
+        * and it is going to be transferred to another place.
+        */
+       if (to_block) {
+               xattr_ref->inode_size_rem +=
+                       EXT4_XATTR_SIZE(old_data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+               xattr_ref->block_size_rem -=
+                       EXT4_XATTR_SIZE(new_data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+               item->in_inode = FALSE;
+       } else if (to_inode) {
+               xattr_ref->block_size_rem +=
+                       EXT4_XATTR_SIZE(old_data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+               xattr_ref->inode_size_rem -=
+                       EXT4_XATTR_SIZE(new_data_size) +
+                       EXT4_XATTR_LEN(item->name_len);
+               item->in_inode = TRUE;
+       } else {
+               /*
+                * No need to transfer as there is enough space for the entry
+                * to stay in inode space or block space it used to be.
+                */
+               orig_room_size +=
+                       EXT4_XATTR_SIZE(old_data_size);
+               orig_room_size -=
+                       EXT4_XATTR_SIZE(new_data_size);
+               if (item->in_inode)
+                       xattr_ref->inode_size_rem = orig_room_size;
+               else
+                       xattr_ref->block_size_rem = orig_room_size;
+
+       }
+       xattr_ref->dirty = TRUE;
+       return ret;
+}
+
+void ext4_xattr_purge_items(struct ext4_xattr_ref *xattr_ref)
+{
+       struct rb_node *first_node;
+       struct ext4_xattr_item *item = NULL;
+       first_node = rb_first(&xattr_ref->root);
+       if (first_node)
+               item = container_of(first_node, struct ext4_xattr_item,
+                                   node);
+
+       while (item) {
+               struct rb_node *next_node;
+               struct ext4_xattr_item *next_item = NULL;
+               next_node = rb_next(&item->node);
+               if (next_node)
+                       next_item = container_of(next_node, struct ext4_xattr_item,
+                                                node);
+               else
+                       next_item = NULL;
+
+               ext4_xattr_item_remove(xattr_ref, item);
+               ext4_xattr_item_free(item);
+
+               item = next_item;
+       }
+       xattr_ref->ea_size = 0;
+       if (ext4_xattr_inode_space(xattr_ref) <
+          sizeof(struct ext4_xattr_ibody_header))
+               xattr_ref->inode_size_rem = 0;
+       else
+               xattr_ref->inode_size_rem =
+                      ext4_xattr_inode_space(xattr_ref) -
+                      sizeof(struct ext4_xattr_ibody_header);
+
+       xattr_ref->block_size_rem =
+               ext4_xattr_block_space(xattr_ref) -
+               sizeof(struct ext4_xattr_header);
+}
+
+static int ext4_xattr_try_alloc_block(struct ext4_xattr_ref *xattr_ref)
+{
+       int ret = 0;
+
+       ext4_fsblk_t xattr_block = 0;
+       xattr_block = xattr_ref->inode_ref->Inode.i_file_acl;
+       if (!xattr_block) {
+               ext4_fsblk_t goal =
+                       ext4_inode_to_goal_block(&xattr_ref->inode_ref->Inode);
+
+               xattr_block = ext4_new_meta_blocks(xattr_ref->IrpContext,
+                                                 &xattr_ref->inode_ref->Inode,
+                                             goal, 0, NULL,
+                                             &ret);
+               if (ret != 0)
+                       goto Finish;
+
+               xattr_ref->block_bh = extents_bwrite(&xattr_ref->fs->sb, xattr_block);
+               if (!xattr_ref->block_bh) {
+                       ext4_free_blocks(xattr_ref->IrpContext, &xattr_ref->inode_ref->Inode,
+                                              xattr_block, 1, 0);
+                       ret = -ENOMEM;
+                       goto Finish;
+               }
+
+               xattr_ref->inode_ref->Inode.i_file_acl = xattr_block;
+               xattr_ref->IsOnDiskInodeDirty = TRUE;
+               xattr_ref->block_loaded = TRUE;
+       }
+
+Finish:
+       return ret;
+}
+
+static void ext4_xattr_try_free_block(struct ext4_xattr_ref *xattr_ref)
+{
+       ext4_fsblk_t xattr_block;
+       xattr_block = xattr_ref->inode_ref->Inode.i_file_acl;
+       xattr_ref->inode_ref->Inode.i_file_acl = 0;
+       extents_brelse(xattr_ref->block_bh);
+       xattr_ref->block_bh = NULL;
+       ext4_free_blocks(xattr_ref->IrpContext, &xattr_ref->inode_ref->Inode,
+               xattr_block, 1, 0);
+       xattr_ref->IsOnDiskInodeDirty = TRUE;
+       xattr_ref->block_loaded = FALSE;
+}
+
+static void ext4_xattr_set_block_header(struct ext4_xattr_ref *xattr_ref)
+{
+       struct ext4_xattr_header *block_header = NULL;
+       block_header = EXT4_XATTR_BHDR(xattr_ref->block_bh);
+
+       memset(block_header, 0, sizeof(struct ext4_xattr_header));
+       block_header->h_magic = EXT4_XATTR_MAGIC;
+       block_header->h_refcount = cpu_to_le32(1);
+       block_header->h_blocks = cpu_to_le32(1);
+}
+
+static void
+ext4_xattr_set_inode_entry(struct ext4_xattr_item *item,
+                          struct ext4_xattr_ibody_header *ibody_header,
+                          struct ext4_xattr_entry *entry, void *ibody_data_ptr)
+{
+       entry->e_name_len = (__u8)item->name_len;
+       entry->e_name_index = item->name_index;
+       entry->e_value_offs =
+          cpu_to_le16((char *)ibody_data_ptr - (char *)EXT4_XATTR_IFIRST(ibody_header));
+       entry->e_value_block = 0;
+       entry->e_value_size = cpu_to_le32(item->data_size);
+}
+
+static void ext4_xattr_set_block_entry(struct ext4_xattr_item *item,
+                                      struct ext4_xattr_header *block_header,
+                                      struct ext4_xattr_entry *block_entry,
+                                      void *block_data_ptr)
+{
+       block_entry->e_name_len = (__u8)item->name_len;
+       block_entry->e_name_index = item->name_index;
+       block_entry->e_value_offs =
+           cpu_to_le16((char *)block_data_ptr - (char *)block_header);
+       block_entry->e_value_block = 0;
+       block_entry->e_value_size = cpu_to_le32(item->data_size);
+}
+
+static int ext4_xattr_write_to_disk(struct ext4_xattr_ref *xattr_ref)
+{
+       int ret = 0;
+       BOOL block_modified = FALSE;
+       void *ibody_data = NULL;
+       void *block_data = NULL;
+       size_t inode_size_rem, block_size_rem;
+       struct ext4_xattr_ibody_header *ibody_header = NULL;
+       struct ext4_xattr_header *block_header = NULL;
+       struct ext4_xattr_entry *entry = NULL;
+       struct ext4_xattr_entry *block_entry = NULL;
+       struct ext4_xattr_item *item = NULL;
+
+       inode_size_rem = ext4_xattr_inode_space(xattr_ref);
+       block_size_rem = ext4_xattr_block_space(xattr_ref);
+       if (inode_size_rem > sizeof(struct ext4_xattr_ibody_header)) {
+               ibody_header = EXT4_XATTR_IHDR(xattr_ref->OnDiskInode);
+               entry = EXT4_XATTR_IFIRST(ibody_header);
+       }
+
+       if (!xattr_ref->dirty)
+               goto Finish;
+       /* If there are enough spaces in the ibody EA table.*/
+       if (inode_size_rem > sizeof(struct ext4_xattr_ibody_header)) {
+               memset(ibody_header, 0, inode_size_rem);
+               ibody_header->h_magic = EXT4_XATTR_MAGIC;
+               ibody_data = (char *)ibody_header + inode_size_rem;
+               inode_size_rem -= sizeof(struct ext4_xattr_ibody_header);
+
+               xattr_ref->IsOnDiskInodeDirty = TRUE;
+       }
+       /* If we need an extra block to hold the EA entries*/
+       if (xattr_ref->ea_size > inode_size_rem) {
+               if (!xattr_ref->block_loaded) {
+                       ret = ext4_xattr_try_alloc_block(xattr_ref);
+                       if (ret != 0)
+                               goto Finish;
+               }
+               memset(xattr_ref->block_bh->b_data, 0, xattr_ref->fs->BlockSize);
+               block_header = EXT4_XATTR_BHDR(xattr_ref->block_bh);
+               block_entry = EXT4_XATTR_BFIRST(xattr_ref->block_bh);
+               ext4_xattr_set_block_header(xattr_ref);
+               block_data = (char *)block_header + block_size_rem;
+               block_size_rem -= sizeof(struct ext4_xattr_header);
+
+               extents_mark_buffer_dirty(xattr_ref->block_bh);
+       } else {
+               /* We don't need an extra block.*/
+               if (xattr_ref->block_loaded) {
+                       block_header = EXT4_XATTR_BHDR(xattr_ref->block_bh);
+                       le32_add_cpu(&block_header->h_refcount, -1);
+                       if (!block_header->h_refcount) {
+                               ext4_xattr_try_free_block(xattr_ref);
+                               block_header = NULL;
+                       } else {
+                               block_entry =
+                                   EXT4_XATTR_BFIRST(xattr_ref->block_bh);
+                               block_data =
+                                   (char *)block_header + block_size_rem;
+                               block_size_rem -=
+                                   sizeof(struct ext4_xattr_header);
+                               xattr_ref->inode_ref->Inode.i_file_acl = 0;
+
+                               xattr_ref->IsOnDiskInodeDirty = TRUE;
+                               extents_mark_buffer_dirty(xattr_ref->block_bh);
+                       }
+               }
+       }
+
+       list_for_each_entry(item, &xattr_ref->ordered_list, struct ext4_xattr_item, list_node) {
+               if (item->in_inode) {
+                       ibody_data = (char *)ibody_data -
+                                    EXT4_XATTR_SIZE(item->data_size);
+                       ext4_xattr_set_inode_entry(item, ibody_header, entry,
+                                                  ibody_data);
+                       memcpy(EXT4_XATTR_NAME(entry), item->name,
+                              item->name_len);
+                       memcpy(ibody_data, item->data, item->data_size);
+                       entry = EXT4_XATTR_NEXT(entry);
+                       inode_size_rem -= EXT4_XATTR_SIZE(item->data_size) +
+                                         EXT4_XATTR_LEN(item->name_len);
+
+                       xattr_ref->IsOnDiskInodeDirty = TRUE;
+                       continue;
+               }
+               if (EXT4_XATTR_SIZE(item->data_size) +
+                       EXT4_XATTR_LEN(item->name_len) >
+                   block_size_rem) {
+                       ret = -ENOSPC;
+                       DbgPrint("ext4_xattr.c: IMPOSSIBLE -ENOSPC AS WE DID INSPECTION!\n");
+                       ASSERT(0);
+               }
+               block_data =
+                   (char *)block_data - EXT4_XATTR_SIZE(item->data_size);
+               ext4_xattr_set_block_entry(item, block_header, block_entry,
+                                          block_data);
+               memcpy(EXT4_XATTR_NAME(block_entry), item->name,
+                      item->name_len);
+               memcpy(block_data, item->data, item->data_size);
+               ext4_xattr_compute_hash(block_header, block_entry);
+               block_entry = EXT4_XATTR_NEXT(block_entry);
+               block_size_rem -= EXT4_XATTR_SIZE(item->data_size) +
+                                 EXT4_XATTR_LEN(item->name_len);
+
+               block_modified = TRUE;
+       }
+       xattr_ref->dirty = FALSE;
+       if (block_modified) {
+               ext4_xattr_rehash(block_header,
+                                 EXT4_XATTR_BFIRST(xattr_ref->block_bh));
+               ext4_xattr_set_block_checksum(xattr_ref->inode_ref,
+                                             xattr_ref->block_bh->b_blocknr,
+                                             block_header);
+               extents_mark_buffer_dirty(xattr_ref->block_bh);
+       }
+
+Finish:
+       return ret;
+}
+
+void ext4_fs_xattr_iterate(struct ext4_xattr_ref *ref,
+                          int (*iter)(struct ext4_xattr_ref *ref,
+                                    struct ext4_xattr_item *item,
+                                        BOOL is_last))
+{
+       struct ext4_xattr_item *item;
+       if (!ref->iter_from) {
+               struct list_head *first_node;
+               first_node = ref->ordered_list.next;
+               if (first_node && first_node != &ref->ordered_list) {
+                       ref->iter_from =
+                               list_entry(first_node,
+                                            struct ext4_xattr_item,
+                                            list_node);
+               }
+       }
+
+       item = ref->iter_from;
+       while (item) {
+               struct list_head *next_node;
+               struct ext4_xattr_item *next_item;
+               int ret = EXT4_XATTR_ITERATE_CONT;
+               next_node = item->list_node.next;
+               if (next_node && next_node != &ref->ordered_list)
+                       next_item = list_entry(next_node, struct ext4_xattr_item,
+                                                list_node);
+               else
+                       next_item = NULL;
+               if (iter)
+                       ret = iter(ref, item, !next_item);
+
+               if (ret != EXT4_XATTR_ITERATE_CONT) {
+                       if (ret == EXT4_XATTR_ITERATE_STOP)
+                               ref->iter_from = NULL;
+
+                       break;
+               }
+               item = next_item;
+       }
+}
+
+void ext4_fs_xattr_iterate_reset(struct ext4_xattr_ref *ref)
+{
+       ref->iter_from = NULL;
+}
+
+int ext4_fs_set_xattr(struct ext4_xattr_ref *ref, __u8 name_index,
+                     const char *name, size_t name_len, const void *data,
+                     size_t data_size, BOOL replace)
+{
+       int ret = 0;
+       struct ext4_xattr_item *item =
+           ext4_xattr_lookup_item(ref, name_index, name, name_len);
+       if (replace) {
+               if (!item) {
+                       ret = -ENODATA;
+                       goto Finish;
+               }
+               if (item->data_size != data_size)
+                       ret = ext4_xattr_resize_item(ref, item, data_size);
+
+               if (ret != 0) {
+                       goto Finish;
+               }
+               memcpy(item->data, data, data_size);
+       } else {
+               if (item) {
+                       ret = -EEXIST;
+                       goto Finish;
+               }
+               item = ext4_xattr_insert_item(ref, name_index, name, name_len,
+                                             data, data_size, &ret);
+       }
+Finish:
+       return ret;
+}
+
+int ext4_fs_set_xattr_ordered(struct ext4_xattr_ref *ref, __u8 name_index,
+       const char *name, size_t name_len, const void *data,
+       size_t data_size)
+{
+       int ret = 0;
+       struct ext4_xattr_item *item =
+               ext4_xattr_lookup_item(ref, name_index, name, name_len);
+       if (item) {
+               ret = -EEXIST;
+               goto Finish;
+       }
+       item = ext4_xattr_insert_item_ordered(ref, name_index, name, name_len,
+               data, data_size, &ret);
+Finish:
+       return ret;
+}
+
+int ext4_fs_remove_xattr(struct ext4_xattr_ref *ref, __u8 name_index,
+                        const char *name, size_t name_len)
+{
+       return ext4_xattr_remove_item(ref, name_index, name, name_len);
+}
+
+int ext4_fs_get_xattr(struct ext4_xattr_ref *ref, __u8 name_index,
+                     const char *name, size_t name_len, void *buf,
+                     size_t buf_size, size_t *data_size)
+{
+       int ret = 0;
+       size_t item_size = 0;
+       struct ext4_xattr_item *item =
+           ext4_xattr_lookup_item(ref, name_index, name, name_len);
+
+       if (!item) {
+               ret = -ENODATA;
+               goto Finish;
+       }
+       item_size = item->data_size;
+       if (buf_size > item_size)
+               buf_size = item_size;
+
+       if (buf)
+               memcpy(buf, item->data, buf_size);
+
+Finish:
+       if (data_size)
+               *data_size = item_size;
+
+       return ret;
+}
+
+int ext4_fs_get_xattr_ref(PEXT2_IRP_CONTEXT IrpContext, PEXT2_VCB fs, PEXT2_MCB inode_ref,
+                         struct ext4_xattr_ref *ref)
+{
+       int rc;
+       ext4_fsblk_t xattr_block;
+       xattr_block = inode_ref->Inode.i_file_acl;
+       memset(&ref->root, 0, sizeof(struct rb_root));
+       ref->ea_size = 0;
+       ref->iter_from = NULL;
+       if (xattr_block) {
+               ref->block_bh = extents_bread(&fs->sb, xattr_block);
+               if (!ref->block_bh)
+                       return -EIO;
+
+               ref->block_loaded = TRUE;
+       } else
+               ref->block_loaded = FALSE;
+
+       ref->inode_ref = inode_ref;
+       ref->fs = fs;
+       INIT_LIST_HEAD(&ref->ordered_list);
+
+       ref->OnDiskInode = Ext2AllocateInode(fs);
+       if (!ref->OnDiskInode) {
+               if (xattr_block) {
+                       extents_brelse(ref->block_bh);
+                       ref->block_bh = NULL;
+               }
+               return -ENOMEM;
+       }
+       if (!Ext2LoadInodeXattr(fs, &inode_ref->Inode, ref->OnDiskInode)) {
+               if (xattr_block) {
+                       extents_brelse(ref->block_bh);
+                       ref->block_bh = NULL;
+               }
+
+               Ext2DestroyInode(fs, ref->OnDiskInode);
+               return -EIO;
+       }
+       ref->IsOnDiskInodeDirty = FALSE;
+
+       if (ext4_xattr_inode_space(ref) <
+          sizeof(struct ext4_xattr_ibody_header) +
+          sizeof(__u32))
+               ref->inode_size_rem = 0;
+       else {
+               ref->inode_size_rem =
+                       ext4_xattr_inode_space(ref) -
+                       sizeof(struct ext4_xattr_ibody_header);
+       }
+
+       ref->block_size_rem =
+               ext4_xattr_block_space(ref) -
+               sizeof(struct ext4_xattr_header) -
+               sizeof(__u32);
+
+       rc = ext4_xattr_fetch(ref);
+       if (rc != 0) {
+               ext4_xattr_purge_items(ref);
+               if (xattr_block) {
+                       extents_brelse(ref->block_bh);
+                       ref->block_bh = NULL;
+               }
+
+               Ext2DestroyInode(fs, ref->OnDiskInode);
+               return rc;
+       }
+       ref->IrpContext = IrpContext;
+       return 0;
+}
+
+int ext4_fs_put_xattr_ref(struct ext4_xattr_ref *ref)
+{
+       int ret;
+       sector_t orig_file_acl = ref->inode_ref->Inode.i_file_acl;
+       ret = ext4_xattr_write_to_disk(ref);
+       if (ref->IsOnDiskInodeDirty) {
+               ASSERT(ref->fs->InodeSize > EXT4_GOOD_OLD_INODE_SIZE);
+
+               /* As we may do block allocation in ext4_xattr_write_to_disk */
+               if (ret)
+                       ref->inode_ref->Inode.i_file_acl = orig_file_acl;
+
+               if (!ret) {
+                       ret = Ext2SaveInode(ref->IrpContext, ref->fs, &ref->inode_ref->Inode)
+                               ? 0 : -EIO;
+                       if (!ret) {
+                               ret = Ext2SaveInodeXattr(ref->IrpContext,
+                                               ref->fs,
+                                               &ref->inode_ref->Inode,
+                                               ref->OnDiskInode)
+                                       ? 0 : -EIO;
+                       }
+               }
+               ref->IsOnDiskInodeDirty = FALSE;
+       }
+       if (ref->block_loaded) {
+               if (!ret)
+                       extents_brelse(ref->block_bh);
+               else
+                       extents_bforget(ref->block_bh);
+
+               ref->block_bh = NULL;
+               ref->block_loaded = FALSE;
+       }
+       ext4_xattr_purge_items(ref);
+       Ext2DestroyInode(ref->fs, ref->OnDiskInode);
+       ref->OnDiskInode = NULL;
+       ref->inode_ref = NULL;
+       ref->fs = NULL;
+       return ret;
+}
+
+struct xattr_prefix {
+       const char *prefix;
+       __u8 name_index;
+};
+
+static const struct xattr_prefix prefix_tbl[] = {
+    {"user.", EXT4_XATTR_INDEX_USER},
+    {"system.posix_acl_access", EXT4_XATTR_INDEX_POSIX_ACL_ACCESS},
+    {"system.posix_acl_default", EXT4_XATTR_INDEX_POSIX_ACL_DEFAULT},
+    {"trusted.", EXT4_XATTR_INDEX_TRUSTED},
+    {"security.", EXT4_XATTR_INDEX_SECURITY},
+    {"system.", EXT4_XATTR_INDEX_SYSTEM},
+    {"system.richacl", EXT4_XATTR_INDEX_RICHACL},
+    {NULL, 0},
+};
+
+const char *ext4_extract_xattr_name(const char *full_name, size_t full_name_len,
+                             __u8 *name_index, size_t *name_len,
+                             BOOL *found)
+{
+       int i;
+       ASSERT(name_index);
+       ASSERT(found);
+
+       *found = FALSE;
+
+       if (!full_name_len) {
+               if (name_len)
+                       *name_len = 0;
+
+               return NULL;
+       }
+
+       for (i = 0; prefix_tbl[i].prefix; i++) {
+               size_t prefix_len = strlen(prefix_tbl[i].prefix);
+               if (full_name_len >= prefix_len &&
+                   !memcmp(full_name, prefix_tbl[i].prefix, prefix_len)) {
+                       BOOL require_name =
+                               prefix_tbl[i].prefix[prefix_len - 1] == '.';
+                       *name_index = prefix_tbl[i].name_index;
+                       if (name_len)
+                               *name_len = full_name_len - prefix_len;
+
+                       if (!(full_name_len - prefix_len) && require_name)
+                               return NULL;
+
+                       *found = TRUE;
+                       if (require_name)
+                               return full_name + prefix_len;
+
+                       return NULL;
+               }
+       }
+       if (name_len)
+               *name_len = 0;
+
+       return NULL;
+}
+
+const char *ext4_get_xattr_name_prefix(__u8 name_index,
+                                      size_t *ret_prefix_len)
+{
+       int i;
+
+       for (i = 0; prefix_tbl[i].prefix; i++) {
+               size_t prefix_len = strlen(prefix_tbl[i].prefix);
+               if (prefix_tbl[i].name_index == name_index) {
+                       if (ret_prefix_len)
+                               *ret_prefix_len = prefix_len;
+
+                       return prefix_tbl[i].prefix;
+               }
+       }
+       if (ret_prefix_len)
+               *ret_prefix_len = 0;
+
+       return NULL;
+}