This library uses the macros in the obj_pool.h and trp.h to create a memory pool for strings and expose an API for handling them. Signed-off-by: David Barr <david.barr@xxxxxxxxxxxx> Signed-off-by: Ramkumar Ramachandra <artagnon@xxxxxxxxx> --- vcs-svn/string_pool.c | 110 +++++++++++++++++++++++++++++++++++++++++++++++++ vcs-svn/string_pool.h | 14 ++++++ 2 files changed, 124 insertions(+), 0 deletions(-) create mode 100644 vcs-svn/string_pool.c create mode 100644 vcs-svn/string_pool.h diff --git a/vcs-svn/string_pool.c b/vcs-svn/string_pool.c new file mode 100644 index 0000000..cfcf127 --- /dev/null +++ b/vcs-svn/string_pool.c @@ -0,0 +1,110 @@ +#include "git-compat-util.h" + +#include "trp.h" +#include "obj_pool.h" +#include "string_pool.h" + +typedef struct node_s node_t; +static struct trp_root tree = { ~0 }; + +struct node_s { + uint32_t offset; + struct trp_node children; +}; + +/* Create two memory pools: one for node_t, and another for strings */ +obj_pool_gen(node, node_t, 4096); +obj_pool_gen(string, char, 4096); + +static char *node_value(node_t *node) +{ + return node ? string_pointer(node->offset) : NULL; +} + +static int node_value_cmp(node_t *a, node_t *b) +{ + return strcmp(node_value(a), node_value(b)); +} + +static int node_indentity_cmp(node_t *a, node_t *b) +{ + int r = node_value_cmp(a, b); + return r ? r : (((uintptr_t) a) > ((uintptr_t) b)) + - (((uintptr_t) a) < ((uintptr_t) b)); +} + +/* Build a Treap from the node_s structure (a trp_node w/ offset) */ +trp_gen(static, tree_, node_t, children, node, node_indentity_cmp); + +char *pool_fetch(uint32_t entry) +{ + return node_value(node_pointer(entry)); +} + +uint32_t pool_intern(char *key) +{ + /* Canonicalize key */ + node_t *match = NULL; + uint32_t key_len; + if (key == NULL) + return ~0; + key_len = strlen(key) + 1; + node_t *node = node_pointer(node_alloc(1)); + node->offset = string_alloc(key_len); + strcpy(node_value(node), key); + match = tree_psearch(&tree, node); + if (!match || node_value_cmp(node, match)) { + tree_insert(&tree, node); + } else { + node_free(1); + string_free(key_len); + node = match; + } + return node_offset(node); +} + +uint32_t pool_tok_r(char *str, const char *delim, char **saveptr) +{ + char *token = strtok_r(str, delim, saveptr); + return token ? pool_intern(token) : ~0; +} + +void pool_print_seq(uint32_t len, uint32_t *seq, char delim, FILE *stream) +{ + uint32_t i; + for (i = 0; i < len && ~seq[i]; i++) { + fputs(pool_fetch(seq[i]), stream); + if (i < len - 1 && ~seq[i + 1]) + fputc(delim, stream); + } +} + +uint32_t pool_tok_seq(uint32_t max, uint32_t *seq, char *delim, char *str) +{ + char *context = NULL; + uint32_t length = 0, token = str ? pool_tok_r(str, delim, &context) : ~0; + while (length < max) { + seq[length++] = token; + if (token == ~0) + break; + token = pool_tok_r(NULL, delim, &context); + } + seq[length ? length - 1 : 0] = ~0; + return length; +} + +void pool_init(void) +{ + uint32_t node; + node_init(); + string_init(); + for (node = 0; node < node_pool.size; node++) { + tree_insert(&tree, node_pointer(node)); + } +} + +void pool_reset(void) +{ + node_reset(); + string_reset(); +} diff --git a/vcs-svn/string_pool.h b/vcs-svn/string_pool.h new file mode 100644 index 0000000..e2cc447 --- /dev/null +++ b/vcs-svn/string_pool.h @@ -0,0 +1,14 @@ +#ifndef STRING_POOL_H_ +#define STRING_POOL_H_ + +#include "git-compat-util.h" + +uint32_t pool_intern(char *key); +char *pool_fetch(uint32_t entry); +uint32_t pool_tok_r(char *str, const char *delim, char **saveptr); +void pool_print_seq(uint32_t len, uint32_t *seq, char delim, FILE *stream); +uint32_t pool_tok_seq(uint32_t max, uint32_t *seq, char *delim, char *str); +void pool_init(void); +void pool_reset(void); + +#endif -- 1.7.1 -- To unsubscribe from this list: send the line "unsubscribe git" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html