1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
#include "cbit/htab.h"
#include <string.h>
#include <stdio.h>
#include <assert.h>
struct teststr {
bool valid;
const char *what;
};
#define ts_null(ts) ({ if (0) printf("null? %p\n", *ts); !*ts; })
#define ts_eq(ts, cp) ({ if (0) printf("eq? %p %p\n", *ts, *cp); !strcmp(*(ts), *(cp)); })
#define ts_hash(strp) strlen(*(strp))
DECL_EXTERN_HTAB_KEY(teststr, const char *);
DECL_HTAB(teststr_int, teststr, int);
int main() {
struct htab_teststr_int *hp;
HTAB_STORAGE(teststr_int) stor;
HTAB_STORAGE_INIT(&stor, teststr_int);
hp = &stor.h;
for(int i = 0; i < 100; i++) {
const char *k;
asprintf((char **) &k, "foo%d", i);
bool new;
*htab_setp_teststr_int(hp, &k, &new) = i;
assert(new);
}
{
const char *k = "foo31";
bool new;
htab_setp_teststr_int(hp, &k, NULL);
htab_setp_teststr_int(hp, &k, &new);
assert(!new);
htab_remove_teststr_int(hp, &k);
}
HTAB_FOREACH(hp, const char **k, int *v, teststr_int) {
if(*v % 10 == 1)
printf("%s -> %d\n", *k, *v);
}
htab_free_storage_teststr_int(hp);
}
/*
expect-output<<
foo91 -> 91
foo21 -> 21
foo1 -> 1
foo11 -> 11
foo31 -> 31
foo41 -> 41
foo51 -> 51
foo61 -> 61
foo71 -> 71
foo81 -> 81
>>
expect-exit 0
*/
IMPL_HTAB_KEY(teststr, ts_hash, ts_eq, ts_null, /*nil_byte*/ 0);
|
