/*
     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 3, 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 (GNUNET_CRYPTO_QUALITY_WEAK, hc);
}

static int
add_iterator (void *cls, GNUNET_HashCode * next)
{
  int *ret = cls;
  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);
  SRANDOM (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);
    }
  SRANDOM (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);

  SRANDOM (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;
    }

  SRANDOM (1);
  for (i = 0; i < 100; i++)
    {
      nextHC (&tmp);
      GNUNET_CONTAINER_bloomfilter_remove (bf, &tmp);
      GNUNET_CONTAINER_bloomfilter_remove (bfi, &tmp);
    }

  SRANDOM (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;
    }

  SRANDOM (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;
    }

  SRANDOM (2);
  i = 20;
  GNUNET_CONTAINER_bloomfilter_resize (bfi, &add_iterator, &i, SIZE * 2, K);

  SRANDOM (2);
  i = 20;
  GNUNET_CONTAINER_bloomfilter_resize (bf, &add_iterator, &i, SIZE * 2, K);
  SRANDOM (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;
}