9 #include <stringbucket.h>
11 #define STRINGBUCKET_STRINGSIZE 64
13 const char * strnchr(const char *s, int c, int n) {
24 struct stringbucket * stringbucket_open(const char *filename) {
26 struct stringbucket *obj = malloc(sizeof(struct stringbucket));
29 perror("stringbucket allocate");
33 obj->fd = open(filename, O_RDWR | O_APPEND | O_CREAT, 0600);
34 flock(obj->fd, LOCK_SH);
36 flock(obj->fd, LOCK_UN);
37 obj->size = st.st_size;
38 obj->list = mmap(NULL, obj->size, PROT_READ | PROT_WRITE, MAP_SHARED, obj->fd, 0);
39 if (obj->list == NULL) {
40 perror("stringbucket mmap");
49 void stringbucket_close(struct stringbucket *sb) {
50 munmap(sb->list, sb->size);
55 int stringbucket_find(struct stringbucket *sb, const char *string) {
56 char * end = sb->list + sb->size;
57 int string_len = strlen(string);
59 char * ptr = sb->list;
61 char * next = (char *) strnchr(ptr, '\n', end - ptr);
65 if (len > STRINGBUCKET_STRINGSIZE)
66 len = STRINGBUCKET_STRINGSIZE;
67 if (memcmp(ptr, string, (len < string_len ? string_len : len)) == 0) {
68 return (ptr - sb->list);
76 int stringbucket_add(struct stringbucket *sb, const char *string) {
77 if (stringbucket_find(sb, string) != -1) return 0;
78 flock(sb->fd, LOCK_EX);
79 write(sb->fd, string, strlen(string));
80 write(sb->fd, "\n", 1);
81 flock(sb->fd, LOCK_UN);
85 int stringbucket_delete(struct stringbucket *sb, const char *string) {
86 int pos = stringbucket_find(sb, string);
87 if (pos == -1) return 0;
89 /* We doin' it DOS style! */
93 void stringbucket_iterate(struct stringbucket *sb, void (*iter)(char *, void *), void *stuff) {
94 char string[STRINGBUCKET_STRINGSIZE + 1];
95 char * ptr = sb->list;
96 char * end = sb->list + sb->size;
99 char * next = (char *) strnchr(ptr, '\n', end - ptr);
106 int len = next - ptr;
107 memcpy(string, ptr, len);