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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
#include "cbit/htab.h"
#include <string.h>
#include <stdio.h>
#include <stdlib.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);
#define u32_hash(up) (*(up) % 100)
#define u32_null(up) (!*(up))
#define u32_eq(u1p, u2p) (*(u1p) == *(u2p))
DECL_STATIC_HTAB_KEY(u32, uint32_t, u32_hash, u32_eq, u32_null, 0);
DECL_HTAB(u32_u32, u32, uint32_t);
int main() {
/* test loop crap */
LET(int y = 5)
printf("5=%d\n", y);
for (int i = 0; ; i++) {
LET_LOOP(int x = 5) {
printf("*%d.%d\n", i, x);
if (i == 4)
break;
else
continue;
}
abort();
}
struct htab_teststr_int *hp;
HTAB_STORAGE(teststr_int) stor = HTAB_STORAGE_INIT_STATIC(&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);
assert(htab_getbucket_teststr_int(hp, &k)->value == i);
assert(*htab_getp_teststr_int(hp, &k) == i);
}
{
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);
HTAB_STORAGE(u32_u32) h2;
HTAB_STORAGE_INIT(&h2, u32_u32);
uint32_t h2_raw[501];
memset(h2_raw, 0xff, sizeof(h2_raw));
for (int i = 0; i < 3000; i++) {
uint32_t op = arc4random() & 1;
uint32_t key = (arc4random() % 500) + 1;
if (op == 0) { /* set */
uint32_t val = arc4random() & 0x7fffffff;
*htab_setp_u32_u32(&h2.h, &key, NULL) = val;
h2_raw[key] = val;
} else { /* delete */
htab_remove_u32_u32(&h2.h, &key);
h2_raw[key] = -1;
}
}
for (uint32_t k = 1; k <= 500; k++) {
uint32_t raw = h2_raw[k];
uint32_t *hashedp = htab_getp_u32_u32(&h2.h, &k);
uint32_t hashed = hashedp ? *hashedp : -1;
/* printf("%d %x %x\n", k, raw, hashed); */
assert(hashed == raw);
}
}
/*
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);
|
