2 This file is part of GNUnet.
3 (C) 2004, 2009 Christian Grothoff (and other contributing authors)
5 GNUnet is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published
7 by the Free Software Foundation; either version 3, or (at your
8 option) any later version.
10 GNUnet is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with GNUnet; see the file COPYING. If not, write to the
17 Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18 Boston, MA 02111-1307, USA.
21 * @file util/test_container_bloomfilter.c
22 * @brief Testcase for the bloomfilter.
23 * @author Christian Grothoff
24 * @author Igor Wronsky
28 #include "gnunet_common.h"
29 #include "gnunet_container_lib.h"
33 #define TESTFILE "/tmp/bloomtest.dat"
36 * Generate a random hashcode.
39 nextHC (GNUNET_HashCode * hc)
41 GNUNET_CRYPTO_hash_create_random (GNUNET_CRYPTO_QUALITY_WEAK, hc);
45 add_iterator (void *cls, GNUNET_HashCode * next)
58 main (int argc, char *argv[])
60 struct GNUNET_CONTAINER_BloomFilter *bf;
61 struct GNUNET_CONTAINER_BloomFilter *bfi;
70 GNUNET_log_setup ("test-container-bloomfilter", "WARNING", NULL);
71 GNUNET_CRYPTO_seed_weak_random (1);
72 if (0 == STAT (TESTFILE, &sbuf))
73 if (0 != UNLINK (TESTFILE))
74 GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_ERROR, "unlink", TESTFILE);
75 bf = GNUNET_CONTAINER_bloomfilter_load (TESTFILE, SIZE, K);
77 for (i = 0; i < 200; i++)
80 GNUNET_CONTAINER_bloomfilter_add (bf, &tmp);
82 GNUNET_CRYPTO_seed_weak_random (1);
84 for (i = 0; i < 200; i++)
87 if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
92 printf ("Got %d elements out of" "200 expected after insertion.\n", ok1);
93 GNUNET_CONTAINER_bloomfilter_free (bf);
96 if (GNUNET_OK != GNUNET_CONTAINER_bloomfilter_get_raw_data (bf, buf, SIZE))
98 GNUNET_CONTAINER_bloomfilter_free (bf);
102 GNUNET_CONTAINER_bloomfilter_free (bf);
104 bf = GNUNET_CONTAINER_bloomfilter_load (TESTFILE, SIZE, K);
105 GNUNET_assert (bf != NULL);
106 bfi = GNUNET_CONTAINER_bloomfilter_init (buf, SIZE, K);
107 GNUNET_assert (bfi != NULL);
109 GNUNET_CRYPTO_seed_weak_random (1);
112 for (i = 0; i < 200; i++)
115 if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
117 if (GNUNET_CONTAINER_bloomfilter_test (bfi, &tmp) == GNUNET_YES)
122 printf ("Got %d elements out of 200 " "expected after reloading.\n", ok1);
123 GNUNET_CONTAINER_bloomfilter_free (bf);
124 GNUNET_CONTAINER_bloomfilter_free (bfi);
130 printf ("Got %d elements out of 200 " "expected after initialization.\n",
132 GNUNET_CONTAINER_bloomfilter_free (bf);
133 GNUNET_CONTAINER_bloomfilter_free (bfi);
137 GNUNET_CRYPTO_seed_weak_random (1);
138 for (i = 0; i < 100; i++)
141 GNUNET_CONTAINER_bloomfilter_remove (bf, &tmp);
142 GNUNET_CONTAINER_bloomfilter_remove (bfi, &tmp);
145 GNUNET_CRYPTO_seed_weak_random (1);
149 for (i = 0; i < 200; i++)
152 if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
154 if (GNUNET_CONTAINER_bloomfilter_test (bfi, &tmp) == GNUNET_YES)
160 printf ("Expected 100 elements in loaded filter"
161 " after adding 200 and deleting 100, got %d\n", ok1);
162 GNUNET_CONTAINER_bloomfilter_free (bf);
163 GNUNET_CONTAINER_bloomfilter_free (bfi);
168 printf ("Expected 200 elements in initialized filter"
169 " after adding 200 and deleting 100 "
170 "(which should do nothing for a filter not backed by a file), got %d\n",
172 GNUNET_CONTAINER_bloomfilter_free (bf);
173 GNUNET_CONTAINER_bloomfilter_free (bfi);
177 GNUNET_CRYPTO_seed_weak_random (3);
179 GNUNET_CONTAINER_bloomfilter_clear (bf);
181 for (i = 0; i < 1000; i++)
184 if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
189 GNUNET_CONTAINER_bloomfilter_free (bf);
190 GNUNET_CONTAINER_bloomfilter_free (bfi);
194 if (GNUNET_OK != GNUNET_CONTAINER_bloomfilter_or (bf, buf, SIZE))
196 GNUNET_CONTAINER_bloomfilter_free (bf);
197 GNUNET_CONTAINER_bloomfilter_free (bfi);
201 GNUNET_CRYPTO_seed_weak_random (2);
203 GNUNET_CONTAINER_bloomfilter_resize (bfi, &add_iterator, &i, SIZE * 2, K);
205 GNUNET_CRYPTO_seed_weak_random (2);
207 GNUNET_CONTAINER_bloomfilter_resize (bf, &add_iterator, &i, SIZE * 2, K);
208 GNUNET_CRYPTO_seed_weak_random (2);
212 for (i = 0; i < 20; i++)
215 if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
217 if (GNUNET_CONTAINER_bloomfilter_test (bfi, &tmp) == GNUNET_YES)
223 printf ("Expected 20 elements in resized file-backed filter"
224 " after adding 20, got %d\n", ok1);
225 GNUNET_CONTAINER_bloomfilter_free (bf);
226 GNUNET_CONTAINER_bloomfilter_free (bfi);
231 printf ("Expected 20 elements in resized filter"
232 " after adding 20, got %d\n", ok2);
233 GNUNET_CONTAINER_bloomfilter_free (bf);
234 GNUNET_CONTAINER_bloomfilter_free (bfi);
239 GNUNET_CONTAINER_bloomfilter_free (bf);
240 GNUNET_CONTAINER_bloomfilter_free (bfi);
242 GNUNET_break (0 == UNLINK (TESTFILE));