X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Ffragmentation%2Ftest_fragmentation.c;h=3248e1b0031da0fb57e2f7c415e895c9c1efa6c8;hb=392baa82fc240d76b31a353c2d729a6f83f10b2c;hp=8a9af04e794fe065451b679fb538755b70019094;hpb=6ce274c7e9fb01c76dc904d907fce9c893d0fc44;p=oweals%2Fgnunet.git

diff --git a/src/fragmentation/test_fragmentation.c b/src/fragmentation/test_fragmentation.c
index 8a9af04e7..3248e1b00 100644
--- a/src/fragmentation/test_fragmentation.c
+++ b/src/fragmentation/test_fragmentation.c
@@ -1,6 +1,6 @@
 /*
      This file is part of GNUnet
-     (C) 2004, 2009 Christian Grothoff (and other contributing authors)
+     Copyright (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
@@ -14,8 +14,8 @@
 
      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.
+     Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+     Boston, MA 02110-1301, USA.
 */
 /**
  * @file fragmentation/test_fragmentation.c
@@ -25,14 +25,12 @@
 #include "platform.h"
 #include "gnunet_fragmentation_lib.h"
 
-#define VERBOSE GNUNET_NO
-
 #define DETAILS GNUNET_NO
 
 /**
  * Number of messages to transmit (note: each uses ~32k memory!)
  */
-#define NUM_MSGS 500
+#define NUM_MSGS 1000
 
 /**
  * MTU to force on fragmentation (must be > 1k + 12)
@@ -42,9 +40,9 @@
 /**
  * Simulate dropping of 1 out of how many messages? (must be > 1)
  */
-#define DROPRATE 2
+#define DROPRATE 15
 
-static int ret = 1; 
+static int ret = 1;
 
 static unsigned int dups;
 
@@ -60,123 +58,190 @@ static struct GNUNET_DEFRAGMENT_Context *defrag;
 
 static struct GNUNET_BANDWIDTH_Tracker trackers[NUM_MSGS];
 
-static struct GNUNET_FRAGMENT_Context *frags[NUM_MSGS];
+static struct GNUNET_FRAGMENT_Context *frag;
+
+static struct GNUNET_SCHEDULER_Task * shutdown_task;
+
+static struct GNUNET_TIME_Relative msg_delay;
+
+static struct GNUNET_TIME_Relative ack_delay;
+
+
+static void
+do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  ret = 0;
+  shutdown_task = NULL;
+  GNUNET_DEFRAGMENT_context_destroy (defrag);
+  defrag = NULL;
+  if (NULL != frag)
+  {
+    GNUNET_FRAGMENT_context_destroy (frag, &msg_delay, &ack_delay);
+    frag = NULL;
+  }
+  fprintf (stderr,
+           "\nFinal message-delay: %s\n",
+           GNUNET_STRINGS_relative_time_to_string (msg_delay,
+                                                   GNUNET_YES));
+  fprintf (stderr,
+           "Final ack-delay: %s\n",
+           GNUNET_STRINGS_relative_time_to_string (ack_delay,
+                                                   GNUNET_YES));
+}
+
 
 static void
-proc_msgs (void *cls,
-	   const struct GNUNET_MessageHeader *hdr)
+proc_msgs (void *cls, const struct GNUNET_MessageHeader *hdr)
 {
   static unsigned int total;
   unsigned int i;
   const char *buf;
 
 #if DETAILS
-  fprintf (stderr, "!"); /* message complete, good! */
+  FPRINTF (stderr, "%s",  "M! ");        /* message complete, good! */
 #endif
-  buf = (const char*) hdr;
-  for (i=sizeof (struct GNUNET_MessageHeader);i<ntohs(hdr->size);i++)
+  buf = (const char *) hdr;
+  for (i = sizeof (struct GNUNET_MessageHeader); i < ntohs (hdr->size); i++)
     GNUNET_assert (buf[i] == (char) i);
   total++;
 #if ! DETAILS
   if (0 == (total % (NUM_MSGS / 100)))
-    fprintf (stderr, ".");
+    FPRINTF (stderr, "%s",  ".");
 #endif
-  if (total == NUM_MSGS)
-    {
-      ret = 0;
-      GNUNET_DEFRAGMENT_context_destroy (defrag);
-      defrag = NULL;
-      for (i=0;i<NUM_MSGS;i++)
-	{
-	  if (frags[i] == NULL)
-	    continue;
-	  GNUNET_FRAGMENT_context_destroy (frags[i]);
-	  frags[i] = NULL;
-	}
-    }
+  /* tolerate 10% loss, i.e. due to duplicate fragment IDs */
+  if ((total >= NUM_MSGS - (NUM_MSGS / 10)) && (ret != 0))
+  {
+    if (NULL == shutdown_task)
+      shutdown_task = GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
+  }
 }
 
 
 /**
- * Process ACK (by passing to fragmenter)
+ * Process fragment (by passing to defrag).
  */
 static void
-proc_acks (void *cls,
-	   const struct GNUNET_MessageHeader *hdr)
+proc_frac (void *cls, const struct GNUNET_MessageHeader *hdr)
 {
-  unsigned int i;
+  struct GNUNET_FRAGMENT_Context **fc = cls;
   int ret;
 
+  GNUNET_FRAGMENT_context_transmission_done (*fc);
   if (0 == GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, DROPRATE))
-    {						
-      ack_drops++;
-      return; /* random drop */
-    }
-  for (i=0;i<NUM_MSGS;i++)
-    {
-      if (frags[i] == NULL)
-	continue;     
-      ret = GNUNET_FRAGMENT_process_ack (frags[i],
-					 hdr);
-      if (ret == GNUNET_OK)
-	{
+  {
+    frag_drops++;
 #if DETAILS
-	  fprintf (stderr, "@"); /* good ACK */
+    FPRINTF (stderr, "%s",  "DF ");    /* dropped Frag */
 #endif
-	  GNUNET_FRAGMENT_context_destroy (frags[i]);
-	  frags[i] = NULL;
-	  acks++;
-	  return;
-	}
-      if (ret == GNUNET_NO)
-	{
+    return;                     /* random drop */
+  }
+  if (NULL == defrag)
+  {
+    FPRINTF (stderr, "%s",  "?E ");      /* Error: frag after shutdown!? */
+    return;
+  }
+  ret = GNUNET_DEFRAGMENT_process_fragment (defrag, hdr);
+  if (ret == GNUNET_NO)
+  {
 #if DETAILS
-	  fprintf (stderr, "@"); /* good ACK */
+    FPRINTF (stderr, "%s",  "FF ");      /* duplicate fragment */
 #endif
-	  acks++;
-	  return;
-	}
-    }
+    dups++;
+  }
+  else if (ret == GNUNET_OK)
+  {
 #if DETAILS
-  fprintf (stderr, "_"); /* BAD: ack that nobody feels responsible for... */
+    FPRINTF (stderr, "%s",  "F! ");      /* good fragment */
 #endif
+    fragc++;
+  }
+}
+
+
+static void
+next_transmission ()
+{
+  static unsigned int i;
+  struct GNUNET_MessageHeader *msg;
+  static char buf[MTU + 32 * 1024];
+  unsigned int j;
+
+  if (0 == i)
+  {
+    for (j = 0; j < sizeof (buf); j++)
+      buf[j] = (char) j;
+  }
+  else
+  {
+    GNUNET_FRAGMENT_context_destroy (frag,
+                                     &msg_delay,
+                                     &ack_delay);
+    frag = NULL;
+  }
+  if (i == NUM_MSGS)
+    return;
+#if DETAILS
+  FPRINTF (stderr, "%s",  "T! ");        /* sending message */
+#endif
+  msg = (struct GNUNET_MessageHeader *) buf;
+  msg->type = htons ((uint16_t) i);
+  msg->size =
+    htons (sizeof (struct GNUNET_MessageHeader) + (17 * i) % (32 * 1024));
+  frag = GNUNET_FRAGMENT_context_create (NULL /* no stats */ ,
+                                         MTU, &trackers[i],
+                                         msg_delay,
+                                         ack_delay,
+                                         msg,
+                                         &proc_frac, &frag);
+  i++;
 }
 
 
 /**
- * Process fragment (by passing to defrag).
+ * Process ACK (by passing to fragmenter)
  */
 static void
-proc_frac (void *cls,
-	   const struct GNUNET_MessageHeader *hdr)
+proc_acks (void *cls,
+           uint32_t msg_id,
+           const struct GNUNET_MessageHeader *hdr)
 {
+  unsigned int i;
   int ret;
 
   if (0 == GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, DROPRATE))
+  {
+    ack_drops++;
+#if DETAILS
+    FPRINTF (stderr, "%s",  "DA ");    /* dropped ACK */
+#endif
+    return;                     /* random drop */
+  }
+  for (i = 0; i < NUM_MSGS; i++)
+  {
+    if (NULL == frag)
+      continue;
+    ret = GNUNET_FRAGMENT_process_ack (frag, hdr);
+    if (ret == GNUNET_OK)
     {
-      frag_drops++;
-      return; /* random drop */
-    }
-  if (NULL == defrag)
-    {
-      fprintf (stderr, "E"); /* Error: frag after shutdown!? */
+#if DETAILS
+      FPRINTF (stderr, "%s",  "GA ");    /* good ACK */
+#endif
+      next_transmission ();
+      acks++;
       return;
     }
-  ret = GNUNET_DEFRAGMENT_process_fragment (defrag, hdr);
-  if (ret == GNUNET_NO)
+    if (ret == GNUNET_NO)
     {
 #if DETAILS
-      fprintf (stderr, "?"); /* duplicate fragment */
+      FPRINTF (stderr, "%s",  "AA ");    /* duplciate ACK */
 #endif
-      dups++;
+      acks++;
+      return;
     }
-  else if (ret == GNUNET_OK)
-    {
+  }
 #if DETAILS
-      fprintf (stderr, "."); /* good fragment */
+  FPRINTF (stderr, "%s",  "?A ");        /* BAD: ack that nobody feels responsible for... */
 #endif
-      fragc++;
-    }
 }
 
 
@@ -186,33 +251,15 @@ proc_frac (void *cls,
 static void
 run (void *cls,
      char *const *args,
-     const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *cfg)
+     const char *cfgfile,
+     const struct GNUNET_CONFIGURATION_Handle *cfg)
 {
-  unsigned int i;
-  struct GNUNET_MessageHeader *msg;
-  char buf[MTU + 32 * 1024];
-
-  defrag = GNUNET_DEFRAGMENT_context_create (NULL,
-					     MTU,
-					     NUM_MSGS /* enough space for all */,
-					     NULL,
-					     &proc_msgs,
-					     &proc_acks);
-  for (i=0;i<sizeof(buf);i++)
-    buf[i] = (char) i;
-  msg = (struct GNUNET_MessageHeader* ) buf;
-  for (i=0;i<NUM_MSGS;i++)
-    {
-      msg->type = htons ((uint16_t) i);
-      msg->size = htons (MTU + 1 + (17 * i) % (32 * 1024));
-      frags[i] = GNUNET_FRAGMENT_context_create (NULL /* no stats */, 
-						 MTU,
-						 &trackers[i],
-						 GNUNET_TIME_UNIT_SECONDS,
-						 msg,
-						 &proc_frac,
-						 NULL);
-    }
+  defrag = GNUNET_DEFRAGMENT_context_create (NULL, MTU,
+                                             3,
+                                             NULL,
+                                             &proc_msgs,
+                                             &proc_acks);
+  next_transmission ();
 }
 
 
@@ -227,32 +274,30 @@ main (int argc, char *argv[])
     "-c",
     "test_fragmentation_data.conf",
     "-L",
-#if VERBOSE
-    "DEBUG",
-#else
     "WARNING",
-#endif
     NULL
   };
   unsigned int i;
 
+  msg_delay = GNUNET_TIME_UNIT_MILLISECONDS;
+  ack_delay = GNUNET_TIME_UNIT_SECONDS;
   GNUNET_log_setup ("test-fragmentation",
-#if VERBOSE
-                    "DEBUG",
-#else
                     "WARNING",
-#endif
                     NULL);
-  for (i=0;i<NUM_MSGS;i++)
-    GNUNET_BANDWIDTH_tracker_init (&trackers[i],
-				   GNUNET_BANDWIDTH_value_init ((i+1) * 1024),
-				   100);
-  GNUNET_PROGRAM_run (5, argv_prog, "test-fragmentation", "nohelp", options, &run, NULL);
-  fprintf (stderr, 
-	   "\nHad %u good fragments, %u duplicate fragments, %u acks and %u simulated drops of acks\n",
-	   fragc,
-	   dups,
-	   acks,
-	   ack_drops);
+  for (i = 0; i < NUM_MSGS; i++)
+    GNUNET_BANDWIDTH_tracker_init (&trackers[i], NULL, NULL,
+                                   GNUNET_BANDWIDTH_value_init ((i + 1) * 1024),
+                                   100);
+  GNUNET_PROGRAM_run (5,
+                      argv_prog,
+                      "test-fragmentation", "nohelp",
+                      options,
+                      &run, NULL);
+  FPRINTF (stderr,
+           "\nHad %u good fragments, %u duplicate fragments, %u acks and %u simulated drops of acks\n",
+           fragc,
+           dups,
+           acks,
+           ack_drops);
   return ret;
 }