libdtrace/common/dt_strtab.c

43 dt_strtab_grow(dt_strtab_t *sp)
47 	if ((ptr = malloc(sp->str_bufsz)) == NULL)
50 	bufs = realloc(sp->str_bufs, (sp->str_nbufs + 1) * sizeof (char *));
57 	sp->str_nbufs++;
58 	sp->str_bufs = bufs;
59 	sp->str_ptr = ptr;
60 	sp->str_bufs[sp->str_nbufs - 1] = sp->str_ptr;
68 	dt_strtab_t *sp = malloc(sizeof (dt_strtab_t));
73 	if (sp == NULL)
76 	bzero(sp, sizeof (dt_strtab_t));
77 	sp->str_hash = calloc(nbuckets, sizeof (dt_strhash_t *));
79 	if (sp->str_hash == NULL)
82 	sp->str_hashsz = nbuckets;
83 	sp->str_bufs = NULL;
84 	sp->str_ptr = NULL;
85 	sp->str_nbufs = 0;
86 	sp->str_bufsz = bufsz;
87 	sp->str_nstrs = 1;
88 	sp->str_size = 1;
90 	if (dt_strtab_grow(sp) == -1)
93 	*sp->str_ptr++ = '\0';
94 	return (sp);
97 	dt_strtab_destroy(sp);
102 dt_strtab_destroy(dt_strtab_t *sp)
107 	for (i = 0; i < sp->str_hashsz; i++) {
108 		for (hp = sp->str_hash[i]; hp != NULL; hp = hq) {
114 	for (i = 0; i < sp->str_nbufs; i++)
115 		free(sp->str_bufs[i]);
117 	if (sp->str_hash != NULL)
118 		free(sp->str_hash);
119 	if (sp->str_bufs != NULL)
120 		free(sp->str_bufs);
122 	free(sp);
148 dt_strtab_compare(dt_strtab_t *sp, dt_strhash_t *hp,
157 		if (buf == sp->str_bufs[b] + sp->str_bufsz)
158 			buf = sp->str_bufs[++b];
160 		resid = sp->str_bufs[b] + sp->str_bufsz - buf;
175 dt_strtab_copyin(dt_strtab_t *sp, const char *str, size_t len)
177 	char *old_p = sp->str_ptr;
178 	ulong_t old_n = sp->str_nbufs;
180 	ulong_t b = sp->str_nbufs - 1;
184 		if (sp->str_ptr == sp->str_bufs[b] + sp->str_bufsz) {
185 			if (dt_strtab_grow(sp) == -1)
190 		resid = sp->str_bufs[b] + sp->str_bufsz - sp->str_ptr;
192 		bcopy(str, sp->str_ptr, n);
194 		sp->str_ptr += n;
202 	while (sp->str_nbufs != old_n)
203 		free(sp->str_bufs[--sp->str_nbufs]);
205 	sp->str_ptr = old_p;
210 dt_strtab_index(dt_strtab_t *sp, const char *str)
219 	h = dt_strtab_hash(str, &len) % sp->str_hashsz;
221 	for (hp = sp->str_hash[h]; hp != NULL; hp = hp->str_next) {
222 		if (dt_strtab_compare(sp, hp, str, len + 1) == 0)
230 dt_strtab_insert(dt_strtab_t *sp, const char *str)
237 	if ((off = dt_strtab_index(sp, str)) != -1)
240 	h = dt_strtab_hash(str, &len) % sp->str_hashsz;
249 	hp->str_data = sp->str_ptr;
250 	hp->str_buf = sp->str_nbufs - 1;
251 	hp->str_off = sp->str_size;
253 	hp->str_next = sp->str_hash[h];
259 	if (dt_strtab_copyin(sp, str, len + 1) == -1)
262 	sp->str_nstrs++;
263 	sp->str_size += len + 1;
264 	sp->str_hash[h] = hp;
270 dt_strtab_size(const dt_strtab_t *sp)
272 	return (sp->str_size);
276 dt_strtab_write(const dt_strtab_t *sp, dt_strtab_write_f *func, void *private)
282 	for (i = 0; i < sp->str_nbufs; i++, total += res) {
283 		if (i == sp->str_nbufs - 1)
284 			n = sp->str_ptr - sp->str_bufs[i];
286 			n = sp->str_bufsz;
288 		if ((res = func(sp->str_bufs[i], n, total, private)) <= 0)
292 	if (total == 0 && sp->str_size != 0)