aboutsummaryrefslogtreecommitdiff
path: root/lib/util
diff options
context:
space:
mode:
Diffstat (limited to 'lib/util')
-rw-r--r--lib/util/str_table.c198
1 files changed, 72 insertions, 126 deletions
diff --git a/lib/util/str_table.c b/lib/util/str_table.c
index 1463546..2816ff8 100644
--- a/lib/util/str_table.c
+++ b/lib/util/str_table.c
@@ -30,187 +30,133 @@ static sqfs_u32 strhash(const char *s)
return a;
}
-static int strings_grow(str_table_t *table)
+static bool key_equals_function(void *user, const void *a, const void *b)
{
- size_t newsz;
- void *new;
-
- if (table->num_strings < table->max_strings)
- return 0;
-
- newsz = table->max_strings ? (table->max_strings * 2) : 16;
- new = realloc(table->strings, sizeof(table->strings[0]) * newsz);
-
- if (new == NULL)
- return SQFS_ERROR_ALLOC;
-
- table->strings = new;
- table->max_strings = newsz;
- return 0;
+ (void)user;
+ return strcmp(a, b) == 0;
}
-int str_table_init(str_table_t *table, size_t size)
+int str_table_init(str_table_t *table)
{
memset(table, 0, sizeof(*table));
- table->buckets = alloc_array(size, sizeof(table->buckets[0]));
- table->num_buckets = size;
+ if (array_init(&table->bucket_ptrs, sizeof(str_bucket_t *), 0))
+ goto fail_arr;
- if (table->buckets == NULL)
- return SQFS_ERROR_ALLOC;
+ table->ht = hash_table_create(NULL, key_equals_function);
+ if (table->ht == NULL)
+ goto fail_ht;
return 0;
+fail_ht:
+ array_cleanup(&table->bucket_ptrs);
+fail_arr:
+ memset(table, 0, sizeof(*table));
+ return SQFS_ERROR_ALLOC;
}
int str_table_copy(str_table_t *dst, const str_table_t *src)
{
- str_bucket_t *bucket, *it, **next;
- size_t i;
+ str_bucket_t *bucket, **array;
+ int ret;
- dst->num_strings = src->num_strings;
- dst->max_strings = src->max_strings;
- dst->strings = calloc(sizeof(dst->strings[0]), src->num_strings);
+ ret = array_init_copy(&dst->bucket_ptrs, &src->bucket_ptrs);
+ if (ret != 0)
+ return ret;
- if (dst->strings == NULL)
- return -1;
-
- for (i = 0; i < src->num_strings; ++i) {
- dst->strings[i] = strdup(src->strings[i]);
-
- if (dst->strings[i] == NULL)
- goto fail_str;
+ dst->ht = hash_table_clone(src->ht);
+ if (dst->ht == NULL) {
+ array_cleanup(&dst->bucket_ptrs);
+ return SQFS_ERROR_ALLOC;
}
- dst->num_buckets = src->num_buckets;
- dst->buckets = calloc(sizeof(dst->buckets[0]), src->num_buckets);
+ array = (str_bucket_t **)dst->bucket_ptrs.data;
- if (dst->buckets == NULL)
- goto fail_str;
-
- for (i = 0; i < src->num_buckets; ++i) {
- next = &(dst->buckets[i]);
+ hash_table_foreach(dst->ht, ent) {
+ bucket = alloc_flex(sizeof(*bucket), 1, strlen(ent->key) + 1);
+ if (bucket == NULL) {
+ str_table_cleanup(dst);
+ return SQFS_ERROR_ALLOC;
+ }
- for (it = src->buckets[i]; it != NULL; it = it->next) {
- bucket = malloc(sizeof(*bucket));
- if (bucket == NULL)
- goto fail_buck;
+ memcpy(bucket, ent->data,
+ sizeof(*bucket) + strlen(ent->key) + 1);
- bucket->next = NULL;
- bucket->index = it->index;
- bucket->refcount = it->refcount;
- bucket->str = dst->strings[bucket->index];
+ ent->data = bucket;
+ ent->key = bucket->string;
- *next = bucket;
- next = &(bucket->next);
- }
+ array[bucket->index] = bucket;
}
return 0;
-fail_buck:
- for (i = 0; i < dst->num_buckets; ++i) {
- while (dst->buckets[i] != NULL) {
- bucket = dst->buckets[i];
- dst->buckets[i] = bucket->next;
-
- free(bucket);
- }
- }
- free(dst->buckets);
-fail_str:
- for (i = 0; i < dst->num_strings; ++i)
- free(dst->strings[i]);
-
- free(dst->strings);
- dst->strings = NULL;
- return -1;
}
void str_table_cleanup(str_table_t *table)
{
- str_bucket_t *bucket;
- size_t i;
-
- for (i = 0; i < table->num_buckets; ++i) {
- while (table->buckets[i] != NULL) {
- bucket = table->buckets[i];
- table->buckets[i] = bucket->next;
-
- free(bucket->str);
- free(bucket);
- }
+ hash_table_foreach(table->ht, ent) {
+ free(ent->data);
+ ent->data = NULL;
+ ent->key = NULL;
}
- free(table->buckets);
- free(table->strings);
+ hash_table_destroy(table->ht, NULL);
+ array_cleanup(&table->bucket_ptrs);
memset(table, 0, sizeof(*table));
}
int str_table_get_index(str_table_t *table, const char *str, size_t *idx)
{
- str_bucket_t *bucket;
+ struct hash_entry *ent;
+ str_bucket_t *new;
sqfs_u32 hash;
- size_t index;
- int err;
hash = strhash(str);
- index = hash % table->num_buckets;
- bucket = table->buckets[index];
-
- while (bucket != NULL) {
- if (strcmp(bucket->str, str) == 0) {
- *idx = bucket->index;
- return 0;
- }
+ ent = hash_table_search_pre_hashed(table->ht, hash, str);
- bucket = bucket->next;
+ if (ent != NULL) {
+ *idx = ((str_bucket_t *)ent->data)->index;
+ return 0;
}
- err = strings_grow(table);
- if (err)
- return err;
+ new = alloc_flex(sizeof(*new), 1, strlen(str) + 1);
+ if (new == NULL)
+ return SQFS_ERROR_ALLOC;
- bucket = calloc(1, sizeof(*bucket));
- if (bucket == NULL)
- goto fail_oom;
+ new->index = table->next_index;
+ strcpy(new->string, str);
- bucket->str = strdup(str);
- if (bucket->str == NULL)
- goto fail_oom;
+ ent = hash_table_insert_pre_hashed(table->ht, hash, str, new);
+ if (ent == NULL) {
+ free(new);
+ return SQFS_ERROR_ALLOC;
+ }
+
+ ent->key = new->string;
- bucket->index = table->num_strings;
- table->strings[table->num_strings++] = bucket->str;
- *idx = bucket->index;
+ if (array_append(&table->bucket_ptrs, &new) != 0) {
+ free(new);
+ ent->key = NULL;
+ ent->data = NULL;
+ return SQFS_ERROR_ALLOC;
+ }
- bucket->next = table->buckets[index];
- table->buckets[index] = bucket;
+ *idx = table->next_index++;
return 0;
-fail_oom:
- free(bucket);
- return SQFS_ERROR_ALLOC;
}
-const char *str_table_get_string(const str_table_t *table, size_t index)
+static str_bucket_t *bucket_by_index(const str_table_t *table, size_t index)
{
- if (index >= table->num_strings)
+ if (index >= table->bucket_ptrs.used)
return NULL;
- return table->strings[index];
+ return ((str_bucket_t **)table->bucket_ptrs.data)[index];
}
-static str_bucket_t *bucket_by_index(const str_table_t *table, size_t index)
+const char *str_table_get_string(const str_table_t *table, size_t index)
{
- str_bucket_t *bucket = NULL;
- sqfs_u32 hash;
-
- if (index < table->num_strings) {
- hash = strhash(table->strings[index]);
- bucket = table->buckets[hash % table->num_buckets];
-
- while (bucket != NULL && bucket->index != index)
- bucket = bucket->next;
- }
+ str_bucket_t *bucket = bucket_by_index(table, index);
- return bucket;
+ return bucket == NULL ? NULL : bucket->string;
}
void str_table_add_ref(str_table_t *table, size_t index)