From 9cd1a30affa7354db1db0897469efa9969b3a79b Mon Sep 17 00:00:00 2001
From: Matthias Wachs <wachs@net.in.tum.de>
Date: Wed, 21 Apr 2010 14:10:10 +0000
Subject: [PATCH]

---
 src/hostlist/hostlist-client.c                |  2 +-
 src/hostlist/hostlist-server.c                |  2 +-
 .../test_gnunet_daemon_hostlist_learning.c    | 37 ++++++++++++++++++-
 3 files changed, 38 insertions(+), 3 deletions(-)

diff --git a/src/hostlist/hostlist-client.c b/src/hostlist/hostlist-client.c
index 39f975cc0..7d7472544 100644
--- a/src/hostlist/hostlist-client.c
+++ b/src/hostlist/hostlist-client.c
@@ -499,7 +499,7 @@ static void update_hostlist ( )
        current_hostlist->times_used++;
        current_hostlist->quality = checked_add ( current_hostlist->quality, HOSTLIST_SUCCESSFUL_DOWNLOAD);
        GNUNET_asprintf (&stat,
-                        gettext_noop("Learned URI `%s' downloaded"),
+                        gettext_noop("Advertised URI `%s' downloaded"),
                         current_hostlist->hostlist_uri);
 
        GNUNET_STATISTICS_update ( stats,
diff --git a/src/hostlist/hostlist-server.c b/src/hostlist/hostlist-server.c
index 61e757bf4..4a28cdfc3 100644
--- a/src/hostlist/hostlist-server.c
+++ b/src/hostlist/hostlist-server.c
@@ -410,7 +410,7 @@ adv_transmit_ready ( void *cls, size_t size, void *buf)
   GNUNET_STATISTICS_set (stats,
                          gettext_noop("# hostlist advertisements send"),
                          hostlist_adv_count,
-                         GNUNET_YES);
+                         GNUNET_NO);
   return transmission_size;
 }
 
diff --git a/src/hostlist/test_gnunet_daemon_hostlist_learning.c b/src/hostlist/test_gnunet_daemon_hostlist_learning.c
index 58f02bbd9..468ee5b17 100644
--- a/src/hostlist/test_gnunet_daemon_hostlist_learning.c
+++ b/src/hostlist/test_gnunet_daemon_hostlist_learning.c
@@ -43,6 +43,7 @@
 
 static int timeout;
 static int adv_arrived;
+static int adv_sent;
 static int learned_hostlist_saved;
 static int learned_hostlist_downloaded;
 
@@ -110,6 +111,7 @@ static void shutdown_testcase()
         check_task);
     check_task = GNUNET_SCHEDULER_NO_TASK;
   }
+  GNUNET_free (current_adv_uri);
   GNUNET_SCHEDULER_add_now (sched,
                             &clean_up, NULL);
 }
@@ -161,6 +163,22 @@ process_uris_recv (void *cls,
   return GNUNET_OK;
 }
 
+static int
+process_adv_sent (void *cls,
+              const char *subsystem,
+              const char *name,
+              uint64_t value,
+              int is_persistent)
+{
+  if ( (value == 1) && (adv_sent == GNUNET_NO))
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+                _("Server has successfully sent advertisement\n"));
+    adv_sent = GNUNET_YES;
+  }
+  return GNUNET_OK;
+}
+
 /**
  * Check the server statistics regularly
  */
@@ -169,7 +187,7 @@ check_statistics (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
   char *stat;
   GNUNET_asprintf (&stat,
-                   gettext_noop("Learned URI `%s' downloaded"),
+                   gettext_noop("Advertised URI `%s' downloaded"),
                    current_adv_uri);
   GNUNET_STATISTICS_get (learn_peer.stats,
                          "hostlist",
@@ -186,6 +204,13 @@ check_statistics (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
                          NULL,
                          &process_uris_recv,
                          NULL);
+  GNUNET_STATISTICS_get (adv_peer.stats,
+                         "hostlist",
+                         gettext_noop("# hostlist advertisements send"),
+                         GNUNET_TIME_UNIT_MINUTES,
+                         NULL,
+                         &process_adv_sent,
+                         NULL);
   check_task = GNUNET_SCHEDULER_add_delayed (sched,
                                 CHECK_INTERVALL,
                                 &check_statistics,
@@ -318,6 +343,8 @@ setup_adv_peer (struct PeerContext *p, const char *cfgname)
 #endif
   GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_load (p->cfg, cfgname));
   GNUNET_ARM_start_services (p->cfg, sched, "core", NULL);
+  p->stats = GNUNET_STATISTICS_create (sched, "hostlist", p->cfg);
+  GNUNET_assert ( NULL != p->stats );
 }
 
 
@@ -392,6 +419,7 @@ run (void *cls,
 {
   timeout = GNUNET_NO;
   adv_arrived = GNUNET_NO;
+  adv_sent =GNUNET_NO;
   learned_hostlist_downloaded = GNUNET_NO;
   sched = s;
   timeout_task = GNUNET_SCHEDULER_add_delayed (sched,
@@ -453,6 +481,13 @@ check ()
                 "Advertisement hostlist could not be downloaded from server\n");
     return GNUNET_YES;
   }
+
+  if (adv_sent == GNUNET_NO)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                "Advertisement was not sent from server to client\n");
+    return GNUNET_YES;
+  }
   return GNUNET_NO;
 }
 
-- 
2.25.1