1 |
|
|
/* $OpenBSD: hcreate.c,v 1.7 2016/05/29 20:47:49 guenther Exp $ */ |
2 |
|
|
/* $NetBSD: hcreate.c,v 1.5 2004/04/23 02:48:12 simonb Exp $ */ |
3 |
|
|
|
4 |
|
|
/* |
5 |
|
|
* Copyright (c) 2001 Christopher G. Demetriou |
6 |
|
|
* All rights reserved. |
7 |
|
|
* |
8 |
|
|
* Redistribution and use in source and binary forms, with or without |
9 |
|
|
* modification, are permitted provided that the following conditions |
10 |
|
|
* are met: |
11 |
|
|
* 1. Redistributions of source code must retain the above copyright |
12 |
|
|
* notice, this list of conditions and the following disclaimer. |
13 |
|
|
* 2. Redistributions in binary form must reproduce the above copyright |
14 |
|
|
* notice, this list of conditions and the following disclaimer in the |
15 |
|
|
* documentation and/or other materials provided with the distribution. |
16 |
|
|
* 3. All advertising materials mentioning features or use of this software |
17 |
|
|
* must display the following acknowledgement: |
18 |
|
|
* This product includes software developed for the |
19 |
|
|
* NetBSD Project. See http://www.NetBSD.org/ for |
20 |
|
|
* information about NetBSD. |
21 |
|
|
* 4. The name of the author may not be used to endorse or promote products |
22 |
|
|
* derived from this software without specific prior written permission. |
23 |
|
|
* |
24 |
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
25 |
|
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
26 |
|
|
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
27 |
|
|
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
28 |
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
29 |
|
|
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
30 |
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
31 |
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
32 |
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
33 |
|
|
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
34 |
|
|
* |
35 |
|
|
* <<Id: LICENSE,v 1.2 2000/06/14 15:57:33 cgd Exp>> |
36 |
|
|
*/ |
37 |
|
|
|
38 |
|
|
/* |
39 |
|
|
* hcreate() / hsearch() / hdestroy() |
40 |
|
|
* |
41 |
|
|
* SysV/XPG4 hash table functions. |
42 |
|
|
* |
43 |
|
|
* Implementation done based on NetBSD manual page and Solaris manual page, |
44 |
|
|
* plus my own personal experience about how they're supposed to work. |
45 |
|
|
* |
46 |
|
|
* I tried to look at Knuth (as cited by the Solaris manual page), but |
47 |
|
|
* nobody had a copy in the office, so... |
48 |
|
|
*/ |
49 |
|
|
|
50 |
|
|
#include <assert.h> |
51 |
|
|
#include <errno.h> |
52 |
|
|
#include <stdint.h> |
53 |
|
|
#include <search.h> |
54 |
|
|
#include <stdlib.h> |
55 |
|
|
#include <string.h> |
56 |
|
|
#include <sys/queue.h> |
57 |
|
|
|
58 |
|
|
#include <db.h> /* for __default_hash */ |
59 |
|
|
|
60 |
|
|
#ifndef _DIAGASSERT |
61 |
|
|
#define _DIAGASSERT(x) |
62 |
|
|
#endif |
63 |
|
|
|
64 |
|
|
/* |
65 |
|
|
* DO NOT MAKE THIS STRUCTURE LARGER THAN 32 BYTES (4 ptrs on 64-bit |
66 |
|
|
* ptr machine) without adjusting MAX_BUCKETS_LG2 below. |
67 |
|
|
*/ |
68 |
|
|
struct internal_entry { |
69 |
|
|
SLIST_ENTRY(internal_entry) link; |
70 |
|
|
ENTRY ent; |
71 |
|
|
}; |
72 |
|
|
SLIST_HEAD(internal_head, internal_entry); |
73 |
|
|
|
74 |
|
|
#define MIN_BUCKETS_LG2 4 |
75 |
|
|
#define MIN_BUCKETS (1 << MIN_BUCKETS_LG2) |
76 |
|
|
|
77 |
|
|
/* |
78 |
|
|
* max * sizeof internal_entry must fit into size_t. |
79 |
|
|
* assumes internal_entry is <= 32 (2^5) bytes. |
80 |
|
|
*/ |
81 |
|
|
#define MAX_BUCKETS_LG2 (sizeof (size_t) * 8 - 1 - 5) |
82 |
|
|
#define MAX_BUCKETS ((size_t)1 << MAX_BUCKETS_LG2) |
83 |
|
|
|
84 |
|
|
static struct internal_head *htable; |
85 |
|
|
static size_t htablesize; |
86 |
|
|
|
87 |
|
|
int |
88 |
|
|
hcreate(size_t nel) |
89 |
|
|
{ |
90 |
|
|
size_t idx; |
91 |
|
|
unsigned int p2; |
92 |
|
|
|
93 |
|
|
/* Make sure this isn't called when a table already exists. */ |
94 |
|
|
_DIAGASSERT(htable == NULL); |
95 |
|
|
if (htable != NULL) { |
96 |
|
|
errno = EINVAL; |
97 |
|
|
return 0; |
98 |
|
|
} |
99 |
|
|
|
100 |
|
|
/* If nel is too small, make it min sized. */ |
101 |
|
|
if (nel < MIN_BUCKETS) |
102 |
|
|
nel = MIN_BUCKETS; |
103 |
|
|
|
104 |
|
|
/* If it's too large, cap it. */ |
105 |
|
|
if (nel > MAX_BUCKETS) |
106 |
|
|
nel = MAX_BUCKETS; |
107 |
|
|
|
108 |
|
|
/* If it's is not a power of two in size, round up. */ |
109 |
|
|
if ((nel & (nel - 1)) != 0) { |
110 |
|
|
for (p2 = 0; nel != 0; p2++) |
111 |
|
|
nel >>= 1; |
112 |
|
|
_DIAGASSERT(p2 <= MAX_BUCKETS_LG2); |
113 |
|
|
nel = 1 << p2; |
114 |
|
|
} |
115 |
|
|
|
116 |
|
|
/* Allocate the table. */ |
117 |
|
|
htablesize = nel; |
118 |
|
|
htable = calloc(htablesize, sizeof htable[0]); |
119 |
|
|
if (htable == NULL) { |
120 |
|
|
errno = ENOMEM; |
121 |
|
|
return 0; |
122 |
|
|
} |
123 |
|
|
|
124 |
|
|
/* Initialize it. */ |
125 |
|
|
for (idx = 0; idx < htablesize; idx++) |
126 |
|
|
SLIST_INIT(&htable[idx]); |
127 |
|
|
|
128 |
|
|
return 1; |
129 |
|
|
} |
130 |
|
|
|
131 |
|
|
void |
132 |
|
|
hdestroy(void) |
133 |
|
|
{ |
134 |
|
|
struct internal_entry *ie; |
135 |
|
|
size_t idx; |
136 |
|
|
|
137 |
|
|
_DIAGASSERT(htable != NULL); |
138 |
|
|
if (htable == NULL) |
139 |
|
|
return; |
140 |
|
|
|
141 |
|
|
for (idx = 0; idx < htablesize; idx++) { |
142 |
|
|
while (!SLIST_EMPTY(&htable[idx])) { |
143 |
|
|
ie = SLIST_FIRST(&htable[idx]); |
144 |
|
|
SLIST_REMOVE_HEAD(&htable[idx], link); |
145 |
|
|
free(ie->ent.key); |
146 |
|
|
free(ie); |
147 |
|
|
} |
148 |
|
|
} |
149 |
|
|
free(htable); |
150 |
|
|
htable = NULL; |
151 |
|
|
} |
152 |
|
|
|
153 |
|
|
ENTRY * |
154 |
|
|
hsearch(ENTRY item, ACTION action) |
155 |
|
|
{ |
156 |
|
|
struct internal_head *head; |
157 |
|
|
struct internal_entry *ie; |
158 |
|
|
uint32_t hashval; |
159 |
|
|
size_t len; |
160 |
|
|
|
161 |
|
|
_DIAGASSERT(htable != NULL); |
162 |
|
|
_DIAGASSERT(item.key != NULL); |
163 |
|
|
_DIAGASSERT(action == ENTER || action == FIND); |
164 |
|
|
|
165 |
|
|
len = strlen(item.key); |
166 |
|
|
hashval = __default_hash(item.key, len); |
167 |
|
|
|
168 |
|
|
head = &htable[hashval & (htablesize - 1)]; |
169 |
|
|
ie = SLIST_FIRST(head); |
170 |
|
|
while (ie != NULL) { |
171 |
|
|
if (strcmp(ie->ent.key, item.key) == 0) |
172 |
|
|
break; |
173 |
|
|
ie = SLIST_NEXT(ie, link); |
174 |
|
|
} |
175 |
|
|
|
176 |
|
|
if (ie != NULL) |
177 |
|
|
return &ie->ent; |
178 |
|
|
else if (action == FIND) |
179 |
|
|
return NULL; |
180 |
|
|
|
181 |
|
|
ie = malloc(sizeof *ie); |
182 |
|
|
if (ie == NULL) |
183 |
|
|
return NULL; |
184 |
|
|
ie->ent.key = item.key; |
185 |
|
|
ie->ent.data = item.data; |
186 |
|
|
|
187 |
|
|
SLIST_INSERT_HEAD(head, ie, link); |
188 |
|
|
return &ie->ent; |
189 |
|
|
} |