ng

1 files changed, 246 insertions, 0 deletions

diff --git a/src/util/test_container_bloomfilter.c b/src/util/test_container_bloomfilter.c
new file mode 100644
index 000000000..9beb11298
--- /dev/null
+++ b/src/util/test_container_bloomfilter.c
@@ -0,0 +1,246 @@
+/*
+     This file is part of GNUnet.
+     (C) 2004, 2009 Christian Grothoff (and other contributing authors)
+
+     GNUnet is free software; you can redistribute it and/or modify
+     it under the terms of the GNU General Public License as published
+     by the Free Software Foundation; either version 2, 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
+     General Public License for more details.
+
+     You should have received a copy of the GNU General Public License
+     along with GNUnet; see the file COPYING.  If not, write to the
+     Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+     Boston, MA 02111-1307, USA.
+*/
+/**
+ * @file util/test_container_bloomfilter.c
+ * @brief Testcase for the bloomfilter.
+ * @author Christian Grothoff
+ * @author Igor Wronsky
+ */
+
+#include "platform.h"
+#include "gnunet_common.h"
+#include "gnunet_container_lib.h"
+
+#define K 4
+#define SIZE 65536
+#define TESTFILE "/tmp/bloomtest.dat"
+
+/**
+ * Generate a random hashcode.
+ */
+static void
+nextHC (GNUNET_HashCode * hc)
+{
+  GNUNET_CRYPTO_hash_create_random (hc);
+}
+
+static int
+add_iterator (GNUNET_HashCode * next, void *arg)
+{
+  int *ret = arg;
+  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;
+  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);
+  srand (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);
+    }
+  srand (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);
+
+  srand (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;
+    }
+
+  srand (1);
+  for (i = 0; i < 100; i++)
+    {
+      nextHC (&tmp);
+      GNUNET_CONTAINER_bloomfilter_remove (bf, &tmp);
+      GNUNET_CONTAINER_bloomfilter_remove (bfi, &tmp);
+    }
+
+  srand (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;
+    }
+
+  srand (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;
+    }
+
+  srand (2);
+  i = 20;
+  GNUNET_CONTAINER_bloomfilter_resize (bfi, &add_iterator, &i, SIZE * 2, K);
+
+  srand (2);
+  i = 20;
+  GNUNET_CONTAINER_bloomfilter_resize (bf, &add_iterator, &i, SIZE * 2, K);
+  srand (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;
+}