/*
This file is part of GNUnet
- (C) 2004, 2009 Christian Grothoff (and other contributing authors)
+ 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 General Public License as published
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
#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 5000
+#define NUM_MSGS 1000
/**
* MTU to force on fragmentation (must be > 1k + 12)
/**
* Simulate dropping of 1 out of how many messages? (must be > 1)
*/
-#define DROPRATE 10
+#define DROPRATE 15
static int ret = 1;
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)
+{
+ 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)
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++)
total++;
#if ! DETAILS
if (0 == (total % (NUM_MSGS / 100)))
- fprintf (stderr, ".");
+ FPRINTF (stderr, "%s", ".");
#endif
/* tolerate 10% loss, i.e. due to duplicate fragment IDs */
- if (total >= NUM_MSGS - (NUM_MSGS / 10))
+ if ((total >= NUM_MSGS - (NUM_MSGS / 10)) && (ret != 0))
{
- 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;
- }
+ 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, uint32_t msg_id, 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++;
+ frag_drops++;
+#if DETAILS
+ FPRINTF (stderr, "%s", "DF "); /* dropped Frag */
+#endif
return; /* random drop */
}
- for (i = 0; i < NUM_MSGS; i++)
+ if (NULL == defrag)
+ {
+ FPRINTF (stderr, "%s", "?E "); /* Error: frag after shutdown!? */
+ return;
+ }
+ ret = GNUNET_DEFRAGMENT_process_fragment (defrag, hdr);
+ if (ret == GNUNET_NO)
{
- if (frags[i] == NULL)
- continue;
- ret = GNUNET_FRAGMENT_process_ack (frags[i], hdr);
- if (ret == GNUNET_OK)
- {
#if DETAILS
- fprintf (stderr, "@"); /* good ACK */
+ FPRINTF (stderr, "%s", "FF "); /* duplicate fragment */
#endif
- GNUNET_FRAGMENT_context_destroy (frags[i]);
- frags[i] = NULL;
- acks++;
- return;
- }
- if (ret == GNUNET_NO)
- {
+ dups++;
+ }
+ else if (ret == GNUNET_OK)
+ {
#if DETAILS
- fprintf (stderr, "@"); /* good ACK */
+ FPRINTF (stderr, "%s", "F! "); /* good fragment */
#endif
- acks++;
- return;
- }
+ 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, "_"); /* BAD: ack that nobody feels responsible for... */
+ 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)
{
- struct GNUNET_FRAGMENT_Context **fc = cls;
+ unsigned int i;
int ret;
- GNUNET_FRAGMENT_context_transmission_done (*fc);
if (0 == GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, DROPRATE))
{
- frag_drops++;
+ ack_drops++;
+#if DETAILS
+ FPRINTF (stderr, "%s", "DA "); /* dropped ACK */
+#endif
return; /* random drop */
}
- if (NULL == defrag)
- {
- fprintf (stderr, "E"); /* Error: frag after shutdown!? */
- return;
- }
- ret = GNUNET_DEFRAGMENT_process_fragment (defrag, hdr);
- if (ret == GNUNET_NO)
+ for (i = 0; i < NUM_MSGS; i++)
{
+ if (NULL == frag)
+ continue;
+ ret = GNUNET_FRAGMENT_process_ack (frag, hdr);
+ if (ret == GNUNET_OK)
+ {
#if DETAILS
- fprintf (stderr, "?"); /* duplicate fragment */
+ FPRINTF (stderr, "%s", "GA "); /* good ACK */
#endif
- dups++;
- }
- else if (ret == GNUNET_OK)
- {
+ next_transmission ();
+ acks++;
+ return;
+ }
+ if (ret == GNUNET_NO)
+ {
#if DETAILS
- fprintf (stderr, "."); /* good fragment */
+ FPRINTF (stderr, "%s", "AA "); /* duplciate ACK */
#endif
- fragc++;
+ acks++;
+ return;
+ }
}
+#if DETAILS
+ FPRINTF (stderr, "%s", "?A "); /* BAD: ack that nobody feels responsible for... */
+#endif
}
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 (sizeof (struct GNUNET_MessageHeader) + (17 * i) % (32 * 1024));
- frags[i] = GNUNET_FRAGMENT_context_create (NULL /* no stats */ ,
- MTU,
- &trackers[i],
- GNUNET_TIME_UNIT_SECONDS,
- msg, &proc_frac, &frags[i]);
- }
+ defrag = GNUNET_DEFRAGMENT_context_create (NULL, MTU,
+ 3,
+ NULL,
+ &proc_msgs,
+ &proc_acks);
+ next_transmission ();
}
"-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_tracker_init (&trackers[i], NULL, NULL,
GNUNET_BANDWIDTH_value_init ((i + 1) * 1024),
100);
- GNUNET_PROGRAM_run (5, argv_prog, "test-fragmentation", "nohelp", options,
+ GNUNET_PROGRAM_run (5,
+ argv_prog,
+ "test-fragmentation", "nohelp",
+ options,
&run, NULL);
- fprintf (stderr,
+ FPRINTF (stderr,
"\nHad %u good fragments, %u duplicate fragments, %u acks and %u simulated drops of acks\n",
- fragc, dups, acks, ack_drops);
+ fragc,
+ dups,
+ acks,
+ ack_drops);
return ret;
}