1 /******************************************************************************/
2 /* pfixtools: a collection of postfix related tools */
4 /* ________________________________________________________________________ */
6 /* Redistribution and use in source and binary forms, with or without */
7 /* modification, are permitted provided that the following conditions */
10 /* 1. Redistributions of source code must retain the above copyright */
11 /* notice, this list of conditions and the following disclaimer. */
12 /* 2. Redistributions in binary form must reproduce the above copyright */
13 /* notice, this list of conditions and the following disclaimer in the */
14 /* documentation and/or other materials provided with the distribution. */
15 /* 3. The names of its contributors may not be used to endorse or promote */
16 /* products derived from this software without specific prior written */
19 /* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND */
20 /* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE */
21 /* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR */
22 /* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS */
23 /* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR */
24 /* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF */
25 /* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS */
26 /* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN */
27 /* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) */
28 /* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF */
29 /* THE POSSIBILITY OF SUCH DAMAGE. */
30 /******************************************************************************/
33 * Copyright © 2008 Florent Bruneau
36 #include <arpa/inet.h>
38 #include <netinet/in.h>
46 static trie_t *create_trie_from_file(const char *file)
49 const char *map, *p, *end;
54 fd = open(file, O_RDONLY, 0000);
60 if (fstat(fd, &st) < 0) {
66 p = map = mmap(NULL, st.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
67 if (map == MAP_FAILED) {
74 end = map + st.st_size;
75 while (end > map && end[-1] != '\n') {
78 if (end != map + st.st_size) {
79 syslog(LOG_WARNING, "file %s miss a final \\n, ignoring last line",
84 while (p < end && p != NULL) {
85 const char *eol = (char *)memchr(p, '\n', end - p);
89 if (eol - p > BUFSIZ) {
94 for (const char *s = eol - 1 ; s >= p ; --s) {
95 line[i++] = ascii_tolower(*s);
98 memcpy(line, p, eol - p);
102 trie_insert(db, line);
105 munmap((void*)map, st.st_size);
106 trie_compile(db, false);
111 int main(int argc, char *argv[])
115 trie_t *trie = trie_new();
116 trie_insert(trie, "abcde123456789");
117 trie_insert(trie, "abcde123654789");
118 trie_insert(trie, "abcdefghi");
119 trie_insert(trie, "coucou");
120 trie_insert(trie, "coucou chez vous");
121 trie_insert(trie, "debout !");
122 trie_compile(trie, false);
123 trie_inspect(trie, true);
125 #define ASSERT_TRUE(str) \
126 if (!trie_lookup(trie, str)) { \
127 printf("\"%s\" not found in trie\n", str); \
130 #define ASSERT_FALSE(str) \
131 if (trie_lookup(trie, str)) { \
132 printf("\"%s\" found in trie\n", str); \
136 ASSERT_FALSE("coucou ");
137 ASSERT_FALSE("abcde123");
138 ASSERT_FALSE("abcde");
139 ASSERT_FALSE("coucou chez vous tous");
140 ASSERT_TRUE("abcde123456789");
141 ASSERT_TRUE("abcde123456789");
142 ASSERT_TRUE("abcde123654789");
143 ASSERT_TRUE("abcdefghi");
144 ASSERT_TRUE("coucou");
145 ASSERT_TRUE("coucou chez vous");
146 ASSERT_TRUE("debout !");
153 trie = create_trie_from_file(argv[1]);
155 trie_inspect(trie, false);