| /* |
| Copyright (C) 2008 Arnaldo Carvalho de Melo <acme@redhat.com> |
| |
| This program is free software; you can redistribute it and/or modify it |
| under the terms of version 2 of the GNU General Public License as |
| published by the Free Software Foundation. |
| */ |
| |
| #include "strings.h" |
| #include "gobuffer.h" |
| |
| #include <search.h> |
| #include <stdint.h> |
| #include <stdlib.h> |
| #include <stdio.h> |
| #include <string.h> |
| #include <zlib.h> |
| |
| #include "dutil.h" |
| |
| struct strings *strings__new(void) |
| { |
| struct strings *self = malloc(sizeof(*self)); |
| |
| if (self != NULL) { |
| self->tree = NULL; |
| gobuffer__init(&self->gb); |
| } |
| |
| return self; |
| |
| } |
| |
| static void do_nothing(void *ptr __unused) |
| { |
| } |
| |
| void strings__delete(struct strings *self) |
| { |
| tdestroy(self->tree, do_nothing); |
| __gobuffer__delete(&self->gb); |
| free(self); |
| } |
| |
| static strings_t strings__insert(struct strings *self, const char *s) |
| { |
| return gobuffer__add(&self->gb, s, strlen(s) + 1); |
| } |
| |
| struct search_key { |
| struct strings *self; |
| const char *str; |
| }; |
| |
| static int strings__compare(const void *a, const void *b) |
| { |
| const struct search_key *key = a; |
| |
| return strcmp(key->str, key->self->gb.entries + *(strings_t *)&b); |
| } |
| |
| strings_t strings__add(struct strings *self, const char *str) |
| { |
| strings_t *s; |
| strings_t index; |
| struct search_key key = { |
| .self = self, |
| .str = str, |
| }; |
| |
| if (str == NULL) |
| return 0; |
| |
| s = tsearch(&key, &self->tree, strings__compare); |
| if (s != NULL) { |
| if (*(struct search_key **)s == (void *)&key) { /* Not found, replace with the right key */ |
| index = strings__insert(self, str); |
| if (index != 0) |
| *s = index; |
| else { |
| tdelete(&key, &self->tree, strings__compare); |
| return 0; |
| } |
| } else /* Found! */ |
| index = *s; |
| } else |
| return 0; |
| |
| return index; |
| } |
| |
| strings_t strings__find(struct strings *self, const char *str) |
| { |
| strings_t *s; |
| struct search_key key = { |
| .self = self, |
| .str = str, |
| }; |
| |
| if (str == NULL) |
| return 0; |
| |
| s = tfind(&key, &self->tree, strings__compare); |
| return s ? *s : 0; |
| } |
| |
| int strings__cmp(const struct strings *self, strings_t a, strings_t b) |
| { |
| return a == b ? 0 : strcmp(strings__ptr(self, a), |
| strings__ptr(self, b)); |
| } |