#include "ext4_debug.h"
#include "ext4_block_group.h"
#include "ext4_balloc.h"
-#include "ext4_bitmap.h"
#include "ext4_inode.h"
-#include "ext4_ialloc.h"
#include "ext4_extent.h"
#include <string.h>
+#include <stdlib.h>
+
+static int ext4_xattr_item_cmp(struct ext4_xattr_item *a,
+ struct ext4_xattr_item *b)
+{
+ int result;
+ result = a->name_index - b->name_index;
+ if (result)
+ return result;
+
+ result = a->name_len - b->name_len;
+ if (result)
+ return result;
+
+ return memcmp(a->name, b->name, a->name_len);
+}
+
+RB_GENERATE_INTERNAL(ext4_xattr_tree,
+ ext4_xattr_item,
+ node,
+ ext4_xattr_item_cmp,
+ static inline)
+
+static struct ext4_xattr_item *
+ext4_xattr_item_alloc(uint8_t name_index,
+ char *name,
+ size_t name_len)
+{
+ struct ext4_xattr_item *item;
+ item = malloc(sizeof(struct ext4_xattr_item) +
+ name_len);
+ 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;
+
+ memset(&item->node, 0, sizeof(item->node));
+ memcpy(item->name, name, name_len);
+
+ return item;
+}
+
+static int
+ext4_xattr_item_alloc_data(struct ext4_xattr_item *item,
+ void *orig_data,
+ size_t data_size)
+{
+ void *data = NULL;
+ ext4_assert(!item->data);
+ data = malloc(data_size);
+ if (!data)
+ return ENOMEM;
+
+ if (orig_data)
+ memcpy(data, orig_data, data_size);
+
+ item->data = data;
+ item->data_size = data_size;
+ return EOK;
+}
+
+static void
+ext4_xattr_item_free_data(struct ext4_xattr_item *item)
+{
+ ext4_assert(item->data);
+ free(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 = realloc(item->data, new_data_size);
+ if (!new_data)
+ return ENOMEM;
+
+ item->data = new_data;
+ item->data_size = new_data_size;
+ }
+ return EOK;
+}
+
+static void
+ext4_xattr_item_free(struct ext4_xattr_item *item)
+{
+ if (item->data)
+ ext4_xattr_item_free_data(item);
+
+ free(item);
+}
+
static void *ext4_xattr_entry_data(struct ext4_xattr_ref *xattr_ref,
struct ext4_xattr_entry *entry,
return ret;
}
-static int ext4_xattr_block_lookup(struct ext4_xattr_ref *xattr_ref,
- uint8_t name_index,
- char *name,
- size_t name_len,
- struct ext4_xattr_entry **found_entry,
- void **out_data,
- size_t *out_len)
+static int ext4_xattr_block_fetch(struct ext4_xattr_ref *xattr_ref)
{
+ int ret = EOK;
size_t size_rem;
- bool entry_found = false;
void *data;
struct ext4_xattr_entry *entry = NULL;
for(;size_rem > 0 && !EXT4_XATTR_IS_LAST_ENTRY(entry);
entry = EXT4_XATTR_NEXT(entry),
size_rem -= EXT4_XATTR_LEN(entry->e_name_len)) {
- int diff = -1;
- if (name_index == entry->e_name_index &&
- name_len == entry->e_name_len) {
- char *e_name = (char *)(entry + 1);
- diff = memcmp(e_name,
- name,
- name_len);
- }
- if (!diff) {
- entry_found = true;
- break;
- }
- }
- if (!entry_found)
- return ENOENT;
+ struct ext4_xattr_item *item;
+ char *e_name = (char *)(entry + 1);
- data = ext4_xattr_entry_data(xattr_ref, entry,
- false);
- if (!data)
- return EIO;
+ data = ext4_xattr_entry_data(xattr_ref, entry,
+ false);
+ if (!data) {
+ ret = EIO;
+ goto Finish;
+ }
- if (found_entry)
- *found_entry = entry;
+ 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,
+ to_le32(entry->e_value_size))
+ != EOK) {
+ ext4_xattr_item_free(item);
+ ret = ENOMEM;
+ goto Finish;
+ }
+ RB_INSERT(ext4_xattr_tree, &xattr_ref->root, item);
- if (out_data && out_len) {
- *out_data = data;
- *out_len = to_le32(entry->e_value_size);
}
- return EOK;
+Finish:
+ return ret;
}
-static int ext4_xattr_inode_lookup(struct ext4_xattr_ref *xattr_ref,
- uint8_t name_index,
- char *name,
- size_t name_len,
- struct ext4_xattr_entry **found_entry,
- void **out_data,
- size_t *out_len)
+static int ext4_xattr_inode_fetch(struct ext4_xattr_ref *xattr_ref)
{
void *data;
size_t size_rem;
- bool entry_found = false;
+ int ret = EOK;
struct ext4_xattr_ibody_header *header = NULL;
struct ext4_xattr_entry *entry = NULL;
uint16_t inode_size = ext4_get16(&xattr_ref->fs->sb,
for(;size_rem > 0 && !EXT4_XATTR_IS_LAST_ENTRY(entry);
entry = EXT4_XATTR_NEXT(entry),
size_rem -= EXT4_XATTR_LEN(entry->e_name_len)) {
- int diff = -1;
- if (name_index == entry->e_name_index &&
- name_len == entry->e_name_len) {
- char *e_name = (char *)(entry + 1);
- diff = memcmp(e_name,
- name,
- name_len);
- }
- if (!diff) {
- entry_found = true;
- break;
- }
- }
- if (!entry_found)
- return ENOENT;
-
- data = ext4_xattr_entry_data(xattr_ref, entry,
- true);
- if (!data)
- return EIO;
+ struct ext4_xattr_item *item;
+ char *e_name = (char *)(entry + 1);
- if (found_entry)
- *found_entry = entry;
+ data = ext4_xattr_entry_data(xattr_ref, entry,
+ true);
+ if (!data) {
+ ret = EIO;
+ goto Finish;
+ }
- if (out_data && out_len) {
- *out_data = data;
- *out_len = to_le32(entry->e_value_size);
+ 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,
+ to_le32(entry->e_value_size))
+ != EOK) {
+ ext4_xattr_item_free(item);
+ ret = ENOMEM;
+ goto Finish;
+ }
+ RB_INSERT(ext4_xattr_tree, &xattr_ref->root, item);
}
- return EOK;
+Finish:
+ return ret;
}
-int ext4_xattr_lookup(struct ext4_xattr_ref *xattr_ref,
- uint8_t name_index,
- char *name,
- size_t name_len,
- struct ext4_xattr_entry **found_entry,
- void **out_data,
- size_t *out_len)
+static int ext4_xattr_fetch(struct ext4_xattr_ref *xattr_ref)
{
- int ret = ENOENT;
+ int ret = EOK;
uint16_t inode_size = ext4_get16(&xattr_ref->fs->sb,
inode_size);
if (inode_size > EXT4_GOOD_OLD_INODE_SIZE) {
- ret = ext4_xattr_inode_lookup(xattr_ref,
- name_index,
- name,
- name_len,
- found_entry,
- out_data,
- out_len);
- if (ret != ENOENT)
+ ret = ext4_xattr_inode_fetch(xattr_ref);
+ if (ret != EOK)
return ret;
}
if (xattr_ref->block_loaded)
- ret = ext4_xattr_block_lookup(xattr_ref,
- name_index,
- name,
- name_len,
- found_entry,
- out_data,
- out_len);
+ ret = ext4_xattr_block_fetch(xattr_ref);
+
+ xattr_ref->dirty = false;
return ret;
}
+static void
+ext4_xattr_purge_items(struct ext4_xattr_ref *xattr_ref)
+{
+ struct ext4_xattr_item *item, *save_item;
+ RB_FOREACH_SAFE(item,
+ ext4_xattr_tree,
+ &xattr_ref->root,
+ save_item) {
+ RB_REMOVE(ext4_xattr_tree, &xattr_ref->root, item);
+ ext4_xattr_item_free(item);
+ }
+ xattr_ref->ea_size = 0;
+}
+
int ext4_fs_get_xattr_ref(struct ext4_fs *fs,
struct ext4_inode_ref *inode_ref,
struct ext4_xattr_ref *ref)
uint64_t xattr_block;
xattr_block = ext4_inode_get_file_acl(inode_ref->inode,
&fs->sb);
+ RB_INIT(&ref->root);
if (xattr_block) {
rc = ext4_block_get(fs->bdev,
&inode_ref->block, xattr_block);
ref->inode_ref = inode_ref;
ref->fs = fs;
+
+ rc = ext4_xattr_fetch(ref);
+ if (rc != EOK) {
+ ext4_xattr_purge_items(ref);
+ if (xattr_block)
+ ext4_block_set(fs->bdev, &inode_ref->block);
+
+ ref->block_loaded = false;
+ return rc;
+ }
return EOK;
}
ext4_block_set(ref->fs->bdev, &ref->block);
ref->block_loaded = false;
}
+ ext4_xattr_purge_items(ref);
ref->inode_ref = NULL;
ref->fs = NULL;
}