summaryrefslogtreecommitdiff
path: root/src/libdatastruct
diff options
context:
space:
mode:
Diffstat (limited to 'src/libdatastruct')
-rw-r--r--src/libdatastruct/bigkv_fill.c74
-rw-r--r--src/libdatastruct/bigkv_free.c1
-rw-r--r--src/libdatastruct/bigkv_init.c63
3 files changed, 75 insertions, 63 deletions
diff --git a/src/libdatastruct/bigkv_fill.c b/src/libdatastruct/bigkv_fill.c
new file mode 100644
index 0000000..06747c6
--- /dev/null
+++ b/src/libdatastruct/bigkv_fill.c
@@ -0,0 +1,74 @@
+/* ISC license. */
+
+#include <stdint.h>
+#include <string.h>
+#include <errno.h>
+
+#include <skalibs/bytestr.h>
+#include <skalibs/stralloc.h>
+#include <skalibs/genalloc.h>
+#include <skalibs/avltree.h>
+#include <skalibs/bigkv.h>
+
+static void *bigkv_dtok (uint32_t d, void *p)
+{
+ bigkv_t *b = p ;
+ return b->storage.s + genalloc_s(bigkv_node_t, &b->nodes)[d].k ;
+}
+
+static int bigkv_cmp (void const *a, void const *b, void *p)
+{
+ (void)p ;
+ return strcmp((char const *)a, (char const *)b) ;
+}
+
+int bigkv_fill (bigkv_t *b, char const *const *argv, char delim, char const *prefix, char const *stop, uint32_t options)
+{
+ int i = 0 ;
+ size_t prefixlen = prefix ? strlen(prefix) : 0 ;
+ avltree_init(&b->map, 3, 3, 8, &bigkv_dtok, &bigkv_cmp, b) ;
+ for (; argv[i] && !(stop && !strcmp(argv[i], stop)) ; i++)
+ {
+ bigkv_node_t node = { .k = b->storage.len } ;
+ char const *s = argv[i] ;
+ size_t len = strlen(s) ;
+ size_t pos ;
+ int isdup ;
+ uint32_t d ;
+ if (prefixlen)
+ {
+ if (!strncmp(s, prefix, prefixlen)) return i+1 ;
+ s += prefixlen ;
+ len -= prefixlen ;
+ }
+ pos = byte_chr(s, len, delim) ;
+ if (!stralloc_catb(&b->storage, s, pos+1)) goto err ;
+ b->storage.s[pos] = 0 ;
+ isdup = avltree_search(&b->map, s, &d) ;
+ if (isdup)
+ {
+ if (options & BIGKV_OPTIONS_NODUP) goto invalid ;
+ b->storage.len = node.k ;
+ }
+ if (pos < len)
+ {
+ node.v = b->storage.len ;
+ if (!stralloc_catb(&b->storage, s + pos + 1, len - pos)) goto err ;
+ }
+ else node.v = b->storage.len - 1 ;
+ if (isdup) genalloc_s(bigkv_node_t, &b->nodes)[d].v = node.v ;
+ else
+ {
+ d = genalloc_len(bigkv_node_t, &b->nodes) ;
+ if (!genalloc_append(bigkv_node_t, &b->nodes, &node)) goto err ;
+ if (!avltree_insert(&b->map, d)) goto err ;
+ }
+ }
+ return i ;
+
+ invalid:
+ errno = EINVAL ;
+ err:
+ bigkv_free(b) ;
+ return -1 ;
+}
diff --git a/src/libdatastruct/bigkv_free.c b/src/libdatastruct/bigkv_free.c
index 9b43c52..848c614 100644
--- a/src/libdatastruct/bigkv_free.c
+++ b/src/libdatastruct/bigkv_free.c
@@ -10,4 +10,5 @@ void bigkv_free (bigkv_t *b)
avltree_free(&b->map) ;
genalloc_free(bigkv_node_t, &b->nodes) ;
stralloc_free(&b->storage) ;
+ *b = bigkv_zero ;
}
diff --git a/src/libdatastruct/bigkv_init.c b/src/libdatastruct/bigkv_init.c
deleted file mode 100644
index aa7f296..0000000
--- a/src/libdatastruct/bigkv_init.c
+++ /dev/null
@@ -1,63 +0,0 @@
-/* ISC license. */
-
-#include <stdint.h>
-#include <string.h>
-#include <errno.h>
-
-#include <skalibs/stralloc.h>
-#include <skalibs/genalloc.h>
-#include <skalibs/avltree.h>
-#include <skalibs/bigkv.h>
-
-static void *bigkv_dtok (uint32_t d, void *p)
-{
- bigkv_t *b = p ;
- return b->storage.s + genalloc_s(bigkv_node_t, &b->nodes)[d].k ;
-}
-
-static int bigkv_cmp (void const *a, void const *b, void *p)
-{
- (void)p ;
- return strcmp((char const *)a, (char const *)b) ;
-}
-
-int bigkv_init (bigkv_t *b, char const *const *argv, char delim, char const *prefix, char const *stop, uint32_t options)
-{
- size_t prefixlen = prefix ? strlen(prefix) : 0 ;
- avltree_init(&b->map, 3, 3, 8, &bigkv_dtok, &bigkv_cmp, b) ;
- for (; *argv && !(stop && !strcmp(*argv, stop)) ; argv++)
- {
- bigkv_node_t node = { .k = b->storage.len } ;
- uint32_t d ;
- char const *s = *argv ;
- char *p ;
- if (prefixlen)
- {
- if (!strncmp(s, prefix, prefixlen)) break ;
- s += prefixlen ;
- }
- if (avltree_search(&b->map, s, &d))
- {
- if (options & BIGKV_OPTIONS_NODUP) goto invalid ;
- else avltree_delete(&b->map, s) ;
- }
- if (!stralloc_cats(&b->storage, s) || !stralloc_0(&b->storage)) goto err ;
- p = strchr(s, delim) ;
- if (p)
- {
- node.v = node.k + (p - s) + 1 ;
- b->storage.s[node.k + (p - s)] = 0 ;
- }
- else node.v = b->storage.len - 1 ;
- d = genalloc_len(bigkv_node_t, &b->nodes) ;
- if (!genalloc_append(bigkv_node_t, &b->nodes, &node)) goto err ;
- if (!avltree_insert(&b->map, d)) goto err ;
- }
- return 1 ;
-
- invalid:
- errno = EINVAL ;
- err:
- bigkv_free(b) ;
- return 0 ;
-}