1 files changed, 302 insertions, 0 deletions

diff --git a/src/lib/util/test_container_bloomfilter.c b/src/lib/util/test_container_bloomfilter.c
new file mode 100644
index 000000000..244733dd9
--- /dev/null
+++ b/src/lib/util/test_container_bloomfilter.c
@@ -0,0 +1,302 @@
+/*
+     This file is part of GNUnet.
+     Copyright (C) 2004, 2009 GNUnet e.V.
+
+     GNUnet is free software: you can redistribute it and/or modify it
+     under the terms of the GNU Affero General Public License as published
+     by the Free Software Foundation, either version 3 of the License,
+     or (at your option) any later version.
+
+     GNUnet is distributed in the hope that it will be useful, but
+     WITHOUT ANY WARRANTY; without even the implied warranty of
+     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+     Affero General Public License for more details.
+
+     You should have received a copy of the GNU Affero General Public License
+     along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+     SPDX-License-Identifier: AGPL3.0-or-later
+ */
+/**
+ * @file util/test_container_bloomfilter.c
+ * @brief Testcase for the bloomfilter.
+ * @author Christian Grothoff
+ * @author Igor Wronsky
+ */
+
+
+#include "platform.h"
+#include "gnunet_util_lib.h"
+
+#define K 4
+#define SIZE 65536
+#define TESTFILE "/tmp/bloomtest.dat"
+
+
+static void
+bernd_interop (void)
+{
+  struct GNUNET_HashCode hc;
+  char val[128];
+  size_t len;
+  struct GNUNET_CONTAINER_BloomFilter *bf;
+
+  len = GNUNET_DNSPARSER_hex_to_bin (
+    "ac4d46b62f8ddaf3cefbc1c01e47536b7ff297cb081e27a396362b1e92e5729b",
+    val);
+  GNUNET_assert (len < 128);
+  GNUNET_CRYPTO_hash (val,
+                      len,
+                      &hc);
+  fprintf (stderr,
+           "sha512: %s\n",
+           GNUNET_DNSPARSER_bin_to_hex (&hc,
+                                        sizeof (hc)));
+  bf = GNUNET_CONTAINER_bloomfilter_init (NULL,
+                                          128,
+                                          16);
+  GNUNET_CONTAINER_bloomfilter_add (bf,
+                                    &hc);
+  len = GNUNET_CONTAINER_bloomfilter_get_size (bf);
+  {
+    char raw[len];
+
+    GNUNET_CONTAINER_bloomfilter_get_raw_data (bf,
+                                               raw,
+                                               len);
+    fprintf (stderr,
+             "BF: %s\n",
+             GNUNET_DNSPARSER_bin_to_hex (raw,
+                                          len));
+  }
+
+}
+
+
+/**
+ * Generate a random hashcode.
+ */
+static void
+nextHC (struct GNUNET_HashCode *hc)
+{
+  GNUNET_CRYPTO_hash_create_random (GNUNET_CRYPTO_QUALITY_WEAK, hc);
+}
+
+
+static int
+add_iterator (void *cls, struct GNUNET_HashCode *next)
+{
+  int *ret = cls;
+  struct GNUNET_HashCode pos;
+
+  if (0 == (*ret)--)
+    return GNUNET_NO;
+  nextHC (&pos);
+  *next = pos;
+  return GNUNET_YES;
+}
+
+
+int
+main (int argc, char *argv[])
+{
+  struct GNUNET_CONTAINER_BloomFilter *bf;
+  struct GNUNET_CONTAINER_BloomFilter *bfi;
+  struct GNUNET_HashCode tmp;
+  int i;
+  int ok1;
+  int ok2;
+  int falseok;
+  char buf[SIZE];
+  struct stat sbuf;
+
+  GNUNET_log_setup ("test-container-bloomfilter",
+                    "WARNING",
+                    NULL);
+  if (0)
+  {
+    bernd_interop ();
+    return 0;
+  }
+  GNUNET_CRYPTO_seed_weak_random (1);
+  if (0 == stat (TESTFILE, &sbuf))
+    if (0 != unlink (TESTFILE))
+      GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_ERROR, "unlink", TESTFILE);
+  bf = GNUNET_CONTAINER_bloomfilter_load (TESTFILE, SIZE, K);
+
+  for (i = 0; i < 200; i++)
+  {
+    nextHC (&tmp);
+    GNUNET_CONTAINER_bloomfilter_add (bf, &tmp);
+  }
+  GNUNET_CRYPTO_seed_weak_random (1);
+  ok1 = 0;
+  for (i = 0; i < 200; i++)
+  {
+    nextHC (&tmp);
+    if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
+      ok1++;
+  }
+  if (ok1 != 200)
+  {
+    printf ("Got %d elements out of"
+            "200 expected after insertion.\n",
+            ok1);
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    return -1;
+  }
+  if (GNUNET_OK != GNUNET_CONTAINER_bloomfilter_get_raw_data (bf, buf, SIZE))
+  {
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    return -1;
+  }
+
+  GNUNET_CONTAINER_bloomfilter_free (bf);
+
+  bf = GNUNET_CONTAINER_bloomfilter_load (TESTFILE, SIZE, K);
+  GNUNET_assert (bf != NULL);
+  bfi = GNUNET_CONTAINER_bloomfilter_init (buf, SIZE, K);
+  GNUNET_assert (bfi != NULL);
+
+  GNUNET_CRYPTO_seed_weak_random (1);
+  ok1 = 0;
+  ok2 = 0;
+  for (i = 0; i < 200; i++)
+  {
+    nextHC (&tmp);
+    if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
+      ok1++;
+    if (GNUNET_CONTAINER_bloomfilter_test (bfi, &tmp) == GNUNET_YES)
+      ok2++;
+  }
+  if (ok1 != 200)
+  {
+    printf ("Got %d elements out of 200 "
+            "expected after reloading.\n",
+            ok1);
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    GNUNET_CONTAINER_bloomfilter_free (bfi);
+    return -1;
+  }
+
+  if (ok2 != 200)
+  {
+    printf ("Got %d elements out of 200 "
+            "expected after initialization.\n",
+            ok2);
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    GNUNET_CONTAINER_bloomfilter_free (bfi);
+    return -1;
+  }
+
+  GNUNET_CRYPTO_seed_weak_random (1);
+  for (i = 0; i < 100; i++)
+  {
+    nextHC (&tmp);
+    GNUNET_CONTAINER_bloomfilter_remove (bf, &tmp);
+    GNUNET_CONTAINER_bloomfilter_remove (bfi, &tmp);
+  }
+
+  GNUNET_CRYPTO_seed_weak_random (1);
+
+  ok1 = 0;
+  ok2 = 0;
+  for (i = 0; i < 200; i++)
+  {
+    nextHC (&tmp);
+    if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
+      ok1++;
+    if (GNUNET_CONTAINER_bloomfilter_test (bfi, &tmp) == GNUNET_YES)
+      ok2++;
+  }
+
+  if (ok1 != 100)
+  {
+    printf ("Expected 100 elements in loaded filter"
+            " after adding 200 and deleting 100, got %d\n",
+            ok1);
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    GNUNET_CONTAINER_bloomfilter_free (bfi);
+    return -1;
+  }
+  if (ok2 != 200)
+  {
+    printf ("Expected 200 elements in initialized filter"
+            " after adding 200 and deleting 100 "
+            "(which should do nothing for a filter not backed by a file), got %d\n",
+            ok2);
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    GNUNET_CONTAINER_bloomfilter_free (bfi);
+    return -1;
+  }
+
+  GNUNET_CRYPTO_seed_weak_random (3);
+
+  GNUNET_CONTAINER_bloomfilter_clear (bf);
+  falseok = 0;
+  for (i = 0; i < 1000; i++)
+  {
+    nextHC (&tmp);
+    if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
+      falseok++;
+  }
+  if (falseok > 0)
+  {
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    GNUNET_CONTAINER_bloomfilter_free (bfi);
+    return -1;
+  }
+
+  if (GNUNET_OK != GNUNET_CONTAINER_bloomfilter_or (bf, buf, SIZE))
+  {
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    GNUNET_CONTAINER_bloomfilter_free (bfi);
+    return -1;
+  }
+
+  GNUNET_CRYPTO_seed_weak_random (2);
+  i = 20;
+  GNUNET_CONTAINER_bloomfilter_resize (bfi, &add_iterator, &i, SIZE * 2, K);
+
+  GNUNET_CRYPTO_seed_weak_random (2);
+  i = 20;
+  GNUNET_CONTAINER_bloomfilter_resize (bf, &add_iterator, &i, SIZE * 2, K);
+  GNUNET_CRYPTO_seed_weak_random (2);
+
+  ok1 = 0;
+  ok2 = 0;
+  for (i = 0; i < 20; i++)
+  {
+    nextHC (&tmp);
+    if (GNUNET_CONTAINER_bloomfilter_test (bf, &tmp) == GNUNET_YES)
+      ok1++;
+    if (GNUNET_CONTAINER_bloomfilter_test (bfi, &tmp) == GNUNET_YES)
+      ok2++;
+  }
+
+  if (ok1 != 20)
+  {
+    printf ("Expected 20 elements in resized file-backed filter"
+            " after adding 20, got %d\n",
+            ok1);
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    GNUNET_CONTAINER_bloomfilter_free (bfi);
+    return -1;
+  }
+  if (ok2 != 20)
+  {
+    printf ("Expected 20 elements in resized filter"
+            " after adding 20, got %d\n",
+            ok2);
+    GNUNET_CONTAINER_bloomfilter_free (bf);
+    GNUNET_CONTAINER_bloomfilter_free (bfi);
+    return -1;
+  }
+
+
+  GNUNET_CONTAINER_bloomfilter_free (bf);
+  GNUNET_CONTAINER_bloomfilter_free (bfi);
+
+  GNUNET_break (0 == unlink (TESTFILE));
+  return 0;
+}