ng

1 files changed, 185 insertions, 0 deletions

diff --git a/src/hello/test_hello.c b/src/hello/test_hello.c
new file mode 100644
index 000000000..728afd63f
--- /dev/null
+++ b/src/hello/test_hello.c
@@ -0,0 +1,185 @@
+/*
+     This file is part of GNUnet
+     (C) 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 hello/test_hello.c
+ * @brief test for hello.c
+ * @author Christian Grothoff
+ */
+
+/**
+ * Testcase for HELLO code.
+ */
+#include "platform.h"
+#include "gnunet_hello_lib.h"
+
+#define DEBUG 0
+
+static size_t
+my_addr_gen (void *cls, size_t max, void *buf)
+{
+  unsigned int *i = cls;
+  size_t ret;
+
+#if DEBUG
+  fprintf (stderr, "DEBUG: my_addr_gen called with i = %d\n", *i);
+#endif
+  if (0 == *i)
+    return 0;
+  ret = GNUNET_HELLO_add_address ("test",
+                                  GNUNET_TIME_absolute_get (),
+                                  "address_information", *i, buf, max);
+  (*i)--;
+  return ret;
+}
+
+
+static int
+check_addr (void *cls,
+            const char *tname,
+            struct GNUNET_TIME_Absolute expiration,
+            const void *addr, size_t addrlen)
+{
+  unsigned int *i = cls;
+
+#if DEBUG
+  fprintf (stderr, "DEBUG: check_addr called with i = %d and addrlen = %u\n",
+           *i, addrlen);
+#endif
+  GNUNET_assert (addrlen > 0);
+  GNUNET_assert (*i & (1 << (addrlen - 1)));
+  *i -= (1 << (addrlen - 1));
+  GNUNET_assert (0 == strncmp ("address_information", addr, addrlen));
+  GNUNET_assert (0 == strcmp ("test", tname));
+  return GNUNET_OK;
+}
+
+
+static int
+remove_some (void *cls,
+             const char *tname,
+             struct GNUNET_TIME_Absolute expiration,
+             const void *addr, size_t addrlen)
+{
+  unsigned int *i = cls;
+
+#if DEBUG
+  fprintf (stderr, "DEBUG: remove_some called with i = %d and addrlen = %u\n",
+           *i, addrlen);
+#endif
+  GNUNET_assert (addrlen > 0);
+  if (*i & (1 << (addrlen - 1)))
+    {
+      *i -= (1 << (addrlen - 1));
+      return GNUNET_NO;
+    }
+  return GNUNET_OK;
+}
+
+
+int
+main (int argc, char *argv[])
+{
+  struct GNUNET_HELLO_Message *msg1;
+  struct GNUNET_HELLO_Message *msg2;
+  struct GNUNET_HELLO_Message *msg3;
+  struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded publicKey;
+  struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded pk;
+  struct GNUNET_TIME_Absolute startup_time;
+  int ok;
+  unsigned int i;
+
+  GNUNET_log_setup ("test-hello", "DEBUG", NULL);
+  startup_time = GNUNET_TIME_absolute_get ();
+  ok = 0;
+  memset (&publicKey, 42, sizeof (publicKey));
+  fprintf (stderr, "Testing HELLO creation (without addresses)...\n");
+  i = 0;
+  msg1 = GNUNET_HELLO_create (&publicKey, &my_addr_gen, &i);
+  GNUNET_assert (msg1 != NULL);
+  GNUNET_assert (0 < GNUNET_HELLO_size (msg1));
+
+  fprintf (stderr, "Testing address iteration (empty set)...\n");
+  GNUNET_assert (NULL ==
+                 GNUNET_HELLO_iterate_addresses (msg1,
+                                                 GNUNET_NO, &check_addr, &i));
+
+  fprintf (stderr, "Testing HELLO creation (with one address)...\n");
+  i = 1;
+  msg2 = GNUNET_HELLO_create (&publicKey, &my_addr_gen, &i);
+  GNUNET_assert (msg2 != NULL);
+  GNUNET_assert (GNUNET_HELLO_size (msg1) < GNUNET_HELLO_size (msg2));
+
+  fprintf (stderr, "Testing address iteration (one address)...\n");
+  i = 1;
+  GNUNET_assert (NULL ==
+                 GNUNET_HELLO_iterate_addresses (msg2,
+                                                 GNUNET_NO, &check_addr, &i));
+  GNUNET_assert (i == 0);
+
+  fprintf (stderr, "Testing get_key from HELLO...\n");
+  GNUNET_assert (GNUNET_OK == GNUNET_HELLO_get_key (msg2, &pk));
+  GNUNET_assert (0 == memcmp (&publicKey, &pk, sizeof (pk)));
+  GNUNET_free (msg1);
+
+  fprintf (stderr, "Testing HELLO creation (with two addresses)...\n");
+  i = 2;
+  msg3 = GNUNET_HELLO_create (&publicKey, &my_addr_gen, &i);
+  GNUNET_assert (msg3 != NULL);
+  GNUNET_assert (GNUNET_HELLO_size (msg2) < GNUNET_HELLO_size (msg3));
+
+  fprintf (stderr, "Testing address iteration (two addresses)...\n");
+  i = 3;
+  GNUNET_assert (NULL ==
+                 GNUNET_HELLO_iterate_addresses (msg3,
+                                                 GNUNET_NO, &check_addr, &i));
+  GNUNET_assert (i == 0);
+
+  fprintf (stderr, "Testing HELLO merge...\n");
+  msg1 = GNUNET_HELLO_merge (msg2, msg3);
+  GNUNET_assert (GNUNET_HELLO_size (msg1) == GNUNET_HELLO_size (msg3));
+
+  i = 3;
+  GNUNET_assert (NULL ==
+                 GNUNET_HELLO_iterate_addresses (msg1,
+                                                 GNUNET_NO, &check_addr, &i));
+  GNUNET_assert (i == 0);
+  GNUNET_free (msg1);
+
+  fprintf (stderr, "Testing address iteration to copy HELLO...\n");
+  i = 2;
+  msg1 = GNUNET_HELLO_iterate_addresses (msg3, GNUNET_YES, &remove_some, &i);
+  GNUNET_assert (msg1 != NULL);
+  GNUNET_assert (i == 0);
+  i = 1;
+  GNUNET_assert (NULL ==
+                 GNUNET_HELLO_iterate_addresses (msg1,
+                                                 GNUNET_NO, &check_addr, &i));
+  GNUNET_assert (i == 0);
+  GNUNET_free (msg1);
+
+  fprintf (stderr, "Testing delta address iteration...\n");
+  i = 2;
+  GNUNET_HELLO_iterate_new_addresses (msg3,
+                                      msg2, startup_time, &check_addr, &i);
+  GNUNET_assert (i == 0);
+  GNUNET_free (msg2);
+  GNUNET_free (msg3);
+  return 0;                     /* testcase passed */
+}