/*
- This file is part of GNUnet.
- (C) 2010,2011 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.
-*/
+ This file is part of GNUnet.
+ (C) 2010,2011 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 ats/test_ats_api_performance_monitor.c
- * @brief test performance monitoring
+ * @brief test performance API's address monitor feature
* @author Christian Grothoff
* @author Matthias Wachs
*/
#include "gnunet_testing_lib.h"
#include "ats.h"
-#define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 20)
-#define SHUTDOWN_CORRECT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
-
-#define ATS_COUNT 3
+#define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10)
+#define WAIT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
static GNUNET_SCHEDULER_TaskIdentifier die_task;
-static GNUNET_SCHEDULER_TaskIdentifier stage_task;
-
-struct GNUNET_CONFIGURATION_Handle *cfg;
-static struct GNUNET_ATS_SchedulingHandle *sh;
+/**
+ * Statistics handle
+ */
+struct GNUNET_STATISTICS_Handle *stats;
-static struct GNUNET_ATS_PerformanceHandle *ph;
+/**
+ * Configuration handle
+ */
+struct GNUNET_CONFIGURATION_Handle *cfg;
-static struct GNUNET_ATS_PerformanceMonitorHandle *phm;
+/**
+ * ATS scheduling handle
+ */
+static struct GNUNET_ATS_SchedulingHandle *sched_ats;
-static struct GNUNET_HELLO_Address addr;
+/**
+ * ATS performance handle
+ */
+static struct GNUNET_ATS_PerformanceHandle *perf_ats;
-static struct GNUNET_ATS_Information atsi[ATS_COUNT];
+struct GNUNET_ATS_AddressListHandle* phal;
static int ret;
-static void cleanup_addresses ()
+struct Address
{
- GNUNET_ATS_address_destroyed (sh, &addr, NULL);
-}
+ char *plugin;
+ size_t plugin_len;
+
+ void *addr;
+ size_t addr_len;
+
+ struct GNUNET_ATS_Information *ats;
+ int ats_count;
-static void setup_addresses ()
+ void *session;
+};
+
+struct PeerContext
{
- memset (&addr.peer,'\0', sizeof (addr.peer));
- addr.transport_name = "test";
- addr.address = NULL;
- addr.address_length = 0;
- atsi[0].type = htonl(GNUNET_ATS_NETWORK_TYPE);
- atsi[0].value = htonl(GNUNET_ATS_NET_LAN);
+ struct GNUNET_PeerIdentity id;
- atsi[1].type = htonl(GNUNET_ATS_QUALITY_NET_DELAY);
- atsi[1].value = htonl(100);
+ struct Address *addr;
+};
- atsi[2].type = htonl(GNUNET_ATS_QUALITY_NET_DISTANCE);
- atsi[2].value = htonl(5);
+static struct PeerContext p[2];
- GNUNET_ATS_address_add (sh, &addr, NULL, atsi, ATS_COUNT);
-}
+static struct Address p0_addresses[2];
+static struct Address p1_addresses[2];
+struct GNUNET_HELLO_Address p0_ha[2];
+struct GNUNET_HELLO_Address p1_ha[2];
+struct GNUNET_HELLO_Address *s_ha[2];
static void
-end_now (int res)
-{
- if (GNUNET_SCHEDULER_NO_TASK != stage_task)
- {
- GNUNET_SCHEDULER_cancel (stage_task);
- stage_task = GNUNET_SCHEDULER_NO_TASK;
- }
- if (GNUNET_SCHEDULER_NO_TASK != die_task)
- {
- GNUNET_SCHEDULER_cancel (die_task);
- die_task = GNUNET_SCHEDULER_NO_TASK;
- }
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutdown\n");
-
- cleanup_addresses ();
-
- if (NULL != phm)
- {
- GNUNET_ATS_performance_monitor_stop (phm);
- phm = NULL;
- }
-
- if (NULL != ph)
- {
- GNUNET_ATS_performance_done (ph);
- ph = NULL;
- }
-
- if (NULL != sh)
- {
- GNUNET_ATS_scheduling_done (sh);
- sh = NULL;
- }
- ret = res;
-}
+end (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
static void
-end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
+
+
+static void
+ats_perf_cb (void *cls,
+ const struct GNUNET_HELLO_Address *address,
+ int address_active,
+ struct GNUNET_BANDWIDTH_Value32NBO bandwidth_out,
+ struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in,
+ const struct GNUNET_ATS_Information *ats,
+ uint32_t ats_count)
{
- die_task = GNUNET_SCHEDULER_NO_TASK;
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Timeout\n");
- end_now (1);
+ static int peer0 = GNUNET_NO;
+ static int peer1 = GNUNET_NO;
+ static int done = GNUNET_NO;
+ if ((GNUNET_NO == peer0) && (0 == memcmp (address, &p[0].id, sizeof (p[0].id))))
+ {
+ peer0 = GNUNET_YES;
+ }
+ if ((GNUNET_NO == peer0) && (0 == memcmp (address, &p[1].id, sizeof (p[1].id))))
+ {
+ peer1 = GNUNET_YES;
+ }
+ if ((peer0 == GNUNET_YES) && (peer1 = GNUNET_YES) && (GNUNET_NO == done))
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+ "Done\n");
+ done = GNUNET_YES;
+ GNUNET_SCHEDULER_add_now (&end, NULL);
+
+ }
}
-static void
-next_stage (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+
+static int
+stat_cb(void *cls, const char *subsystem,
+ const char *name, uint64_t value,
+ int is_persistent)
{
- static int stage_counter = 0;
-
- stage_task = GNUNET_SCHEDULER_NO_TASK;
- if (0 == stage_counter)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Stop performance monitoring\n");
-
- GNUNET_ATS_performance_monitor_stop (phm);
- phm = NULL;
-
- stage_task = GNUNET_SCHEDULER_add_delayed (SHUTDOWN_CORRECT, &next_stage, NULL);
- stage_counter++;
- return;
- }
- else
- {
- end_now (0);
- }
+
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO, "ATS statistics: `%s' `%s' %llu\n",
+ subsystem,name, value);
+ if (4 == value)
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+ "All addresses added\n");
+ }
+
+ return GNUNET_OK;
+
}
-static void end (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+static void
+address_suggest_cb (void *cls, const struct GNUNET_HELLO_Address *address,
+ struct Session *session,
+ struct GNUNET_BANDWIDTH_Value32NBO bandwidth_out,
+ struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in,
+ const struct GNUNET_ATS_Information *atsi,
+ uint32_t ats_count)
{
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Success\n");
- end_now (0);
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Did not expect suggestion callback!\n");
+ GNUNET_SCHEDULER_add_now (&end_badly, NULL);
+ return;
}
static void
-perf_mon_cb (void *cls,
- struct GNUNET_PeerIdentity *peer,
- struct GNUNET_ATS_Information *ats,
- uint32_t ats_count)
+end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
{
- static int stage_counter = 0;
- int c1;
- int c2;
- int c3;
-
- if (0 == stage_counter)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received initial callback for peer `%s' with %u information\n",
- GNUNET_i2s (&addr.peer), ats_count);
- if ((0 != memcmp (peer, &addr.peer, sizeof (addr.peer))) ||
- (ats_count < ATS_COUNT))
- {
- GNUNET_break (0);
- GNUNET_SCHEDULER_add_now (&end_badly, NULL);
- return;
- }
- c3 = 0;
- for (c1 = 0; c1 < ats_count; c1++)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "ATS information [%u] %u : %u \n", c1, ntohl (ats[c1].type), ntohl (ats[c1].value));
- for (c2 = 0; c2 < ATS_COUNT; c2++)
- {
- if (ats[c1].type == atsi[c2].type)
- {
- if (ats[c1].value == atsi[c2].value)
- {
- c3++;
- }
- else
- {
- GNUNET_break (0);
- }
- }
- }
- }
-
- if (ATS_COUNT != c3)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received only %u correct ATS information \n", c3);
- GNUNET_break (0);
- GNUNET_SCHEDULER_add_now (&end_badly, NULL);
- return;
- }
- stage_counter ++;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received %u correct ATS information \n", c3);
-
- atsi[0].type = htonl(GNUNET_ATS_NETWORK_TYPE);
- atsi[0].value = htonl(GNUNET_ATS_NET_WAN);
-
- atsi[1].type = htonl(GNUNET_ATS_QUALITY_NET_DELAY);
- atsi[1].value = htonl(1000);
-
- atsi[2].type = htonl(GNUNET_ATS_QUALITY_NET_DISTANCE);
- atsi[2].value = htonl(50);
-
- GNUNET_ATS_address_update (sh, &addr, NULL, atsi, ATS_COUNT);
- }
- else if (1 == stage_counter)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received updated callback for peer `%s' with %u information\n",
- GNUNET_i2s (&addr.peer), ats_count);
-
- if ((0 != memcmp (peer, &addr.peer, sizeof (addr.peer))) ||
- (ats_count < ATS_COUNT))
- {
- GNUNET_break (0);
- GNUNET_SCHEDULER_add_now (&end_badly, NULL);
- return;
- }
- c3 = 0;
- for (c1 = 0; c1 < ats_count; c1++)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "ATS information [%u] %u : %u \n", c1, ntohl (ats[c1].type), ntohl (ats[c1].value));
- for (c2 = 0; c2 < ATS_COUNT; c2++)
- {
- if (ats[c1].type == atsi[c2].type)
- {
- if (ats[c1].value == atsi[c2].value)
- {
- c3++;
- }
- else
- {
- GNUNET_break (0);
- }
- }
- }
- }
-
- if (ATS_COUNT != c3)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received only %u correct ATS information \n", c3);
- GNUNET_break (0);
- GNUNET_SCHEDULER_add_now (&end_badly, NULL);
- return;
- }
- stage_counter ++;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received %u correct ATS information, shutdown... \n", c3);
- GNUNET_SCHEDULER_add_now (&end, NULL);
- return;
- }
+ die_task = GNUNET_SCHEDULER_NO_TASK;
+ end ( NULL, NULL);
+ ret = GNUNET_SYSERR;
}
+static void
+end (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+ GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Shutting down\n");
+ if (die_task != GNUNET_SCHEDULER_NO_TASK )
+ {
+ GNUNET_SCHEDULER_cancel (die_task);
+ die_task = GNUNET_SCHEDULER_NO_TASK;
+ }
+
+ if (NULL != sched_ats)
+ {
+ GNUNET_ATS_scheduling_done (sched_ats);
+ sched_ats = NULL;
+ }
+
+ if (NULL != perf_ats)
+ {
+ GNUNET_ATS_performance_done (perf_ats);
+ perf_ats = NULL;
+ }
+
+ GNUNET_STATISTICS_watch_cancel (stats, "ats", "# addresses", &stat_cb, NULL);
+ if (NULL != stats)
+ {
+ GNUNET_STATISTICS_destroy (stats, GNUNET_NO);
+ stats = NULL;
+ }
+
+
+ GNUNET_free_non_null(p0_addresses[0].addr);
+ GNUNET_free_non_null(p0_addresses[1].addr);
+ GNUNET_free_non_null(p1_addresses[0].addr);
+ GNUNET_free_non_null(p1_addresses[1].addr);
+
+ ret = 0;
+}
static void
-run (void *cls,
- const struct GNUNET_CONFIGURATION_Handle *mycfg,
- struct GNUNET_TESTING_Peer *peer)
+run (void *cls, const struct GNUNET_CONFIGURATION_Handle *mycfg,
+ struct GNUNET_TESTING_Peer *peer)
{
ret = 1;
cfg = (struct GNUNET_CONFIGURATION_Handle *) mycfg;
- die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL);
-
- sh = GNUNET_ATS_scheduling_init (cfg, NULL, NULL);
- GNUNET_assert (NULL != sh);
-
- setup_addresses ();
-
- ph = GNUNET_ATS_performance_init (cfg, NULL, NULL);
- GNUNET_assert (NULL != ph);
-
- phm = GNUNET_ATS_performance_monitor_start (ph, &perf_mon_cb, &ret);
- GNUNET_assert (NULL != phm);
-
- stage_task = GNUNET_SCHEDULER_add_delayed (SHUTDOWN_CORRECT, &next_stage, NULL);
+ die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL );
+
+ if (NULL == (perf_ats = GNUNET_ATS_performance_init (cfg, &ats_perf_cb, NULL)))
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+ "Failed to connect to performance API\n");
+ GNUNET_SCHEDULER_add_now (end_badly, NULL);
+ }
+
+
+ stats = GNUNET_STATISTICS_create ("ats", cfg);
+ GNUNET_STATISTICS_watch (stats, "ats", "# addresses", &stat_cb, NULL);
+
+ /* set up peer 0 */
+ memset (&p[0].id, '1', sizeof (p[0].id));
+ p0_addresses[0].plugin = "test";
+ p0_addresses[0].session = NULL;
+ p0_addresses[0].addr = GNUNET_strdup ("test_p0_a0");
+ p0_addresses[0].addr_len = strlen (p0_addresses[0].addr) + 1;
+
+ p0_ha[0].address = p0_addresses[0].addr;
+ p0_ha[0].address_length = p0_addresses[0].addr_len;
+ p0_ha[0].peer = p[0].id;
+ p0_ha[0].transport_name = p0_addresses[0].plugin;
+
+ p0_addresses[1].plugin = "test";
+ p0_addresses[1].session = NULL;
+ p0_addresses[1].addr = GNUNET_strdup ("test_p0_a1");
+ p0_addresses[1].addr_len = strlen (p0_addresses[1].addr) + 1;
+
+ p0_ha[1].address = p0_addresses[1].addr;
+ p0_ha[1].address_length = p0_addresses[1].addr_len;
+ p0_ha[1].peer = p[0].id;
+ p0_ha[1].transport_name = p0_addresses[1].plugin;
+
+ GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Created peer 0: `%s'\n",
+ GNUNET_i2s (&p[0].id));
+
+ memset (&p[1].id, '2', sizeof (p[1].id));
+ p1_addresses[0].plugin = "test";
+ p1_addresses[0].session = NULL;
+ p1_addresses[0].addr = GNUNET_strdup ("test_p1_a0");
+ p1_addresses[0].addr_len = strlen (p1_addresses[0].addr) + 1;
+
+ p1_ha[0].address = p1_addresses[0].addr;
+ p1_ha[0].address_length = p1_addresses[0].addr_len;
+ p1_ha[0].peer = p[1].id;
+ p1_ha[0].transport_name = p1_addresses[0].plugin;
+
+ p1_addresses[1].plugin = "test";
+ p1_addresses[1].session = NULL;
+ p1_addresses[1].addr = GNUNET_strdup ("test_p1_a1");
+ p1_addresses[1].addr_len = strlen (p1_addresses[1].addr) + 1;
+
+ p1_ha[1].address = p1_addresses[1].addr;
+ p1_ha[1].address_length = p1_addresses[1].addr_len;
+ p1_ha[1].peer = p[1].id;
+ p1_ha[1].transport_name = p1_addresses[1].plugin;
+
+ GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Created peer 1: `%s'\n",
+ GNUNET_i2s (&p[1].id));
+
+ /* Add addresses */
+ sched_ats = GNUNET_ATS_scheduling_init (cfg, &address_suggest_cb, NULL );
+ if (sched_ats == NULL )
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Could not setup peer!\n");
+ GNUNET_SCHEDULER_add_now (&end_badly, NULL);
+ return;
+ }
+
+ GNUNET_ATS_address_add (sched_ats, &p0_ha[0], NULL, NULL, 0);
+ GNUNET_ATS_address_add (sched_ats, &p0_ha[1], NULL, NULL, 0);
+
+ GNUNET_ATS_address_add (sched_ats, &p1_ha[0], NULL, NULL, 0);
+ GNUNET_ATS_address_add (sched_ats, &p1_ha[1], NULL, NULL, 0);
}
-
int
main (int argc, char *argv[])
{
- if (0 != GNUNET_TESTING_peer_run ("test_ats_api_performance_monitor",
- "test_ats_api.conf",
- &run, NULL))
+ if (0
+ != GNUNET_TESTING_peer_run ("test_ats_api_performance",
+ "test_ats_api.conf", &run, NULL ))
return 1;
return ret;
}