src/ats/test_ats_api_scheduling_destroy_session.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2010,2011 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20 /**
  21  * @file ats/test_ats_api_scheduling_destroy_session.c
  22  * @brief test destroying sessions with unknown address (address NULL, length 0)
  23  *        in automatic transport selection scheduling API
  24  * @author Christian Grothoff
  25  * @author Matthias Wachs
  26  *
  27  */
  28 #include "platform.h"
  29 #include "gnunet_ats_service.h"
  30 #include "gnunet_testing_lib-new.h"
  31 #include "ats.h"
  32 #include "test_ats_api_common.h"
  33
  34
  35 static GNUNET_SCHEDULER_TaskIdentifier die_task;
  36
  37 static struct GNUNET_ATS_SchedulingHandle *ats;
  38
  39 static int ret;
  40
  41 static int stage;
  42
  43 static struct Test_Address test_addr;
  44
  45 static struct PeerContext p;
  46
  47 static struct GNUNET_HELLO_Address hello_address;
  48
  49
  50 static void
  51 end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  52 {
  53   die_task = GNUNET_SCHEDULER_NO_TASK;
  54   if (ats != NULL)
  55     GNUNET_ATS_scheduling_done (ats);
  56   ret = GNUNET_SYSERR;
  57 }
  58
  59
  60 static void
  61 end ()
  62 {
  63   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutting down\n");
  64   if (die_task != GNUNET_SCHEDULER_NO_TASK)
  65   {
  66     GNUNET_SCHEDULER_cancel (die_task);
  67     die_task = GNUNET_SCHEDULER_NO_TASK;
  68   }
  69   GNUNET_ATS_scheduling_done (ats);
  70   if (2 == stage)
  71     ret = 0;
  72   else
  73   {
  74     GNUNET_break (0);
  75     ret = 1;
  76   }
  77 }
  78
  79
  80 static void
  81 address_suggest_cb (void *cls, const struct GNUNET_HELLO_Address *address,
  82                     struct Session *session,
  83                     struct GNUNET_BANDWIDTH_Value32NBO bandwidth_out,
  84                     struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in,
  85                     const struct GNUNET_ATS_Information *atsi,
  86                     uint32_t ats_count)
  87 {
  88   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Stage %u: ATS suggests address `%s' session %p\n",
  89               stage, GNUNET_i2s (&address->peer), session);
  90   GNUNET_ATS_reset_backoff(ats, &address->peer);
  91
  92   GNUNET_assert (0 ==
  93                  memcmp (&address->peer, &p.id,
  94                          sizeof (struct GNUNET_PeerIdentity)));
  95   GNUNET_assert (0 == strcmp (address->transport_name, test_addr.plugin));
  96   GNUNET_assert (address->address_length == test_addr.addr_len);
  97   GNUNET_assert (0 ==
  98                  memcmp (address->address, test_addr.plugin,
  99                          address->address_length));
 100   GNUNET_assert (test_addr.session == session);
 101
 102   if (0 == stage)
 103   {
 104     /* Delete session without the address */
 105     struct GNUNET_HELLO_Address hello_address_2;
 106     hello_address_2.peer = p.id;
 107     hello_address_2.transport_name = test_addr.plugin;
 108     hello_address_2.address = NULL;
 109     hello_address_2.address_length = 0;
 110
 111     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 112                 "Stage %u: Destroying address for peer `%s' address %p length %u session %p\n",
 113                 stage,
 114                 GNUNET_i2s (&hello_address_2.peer),
 115                 hello_address_2.address,
 116                 hello_address_2.address_length,
 117                 session);
 118
 119     GNUNET_ATS_address_destroyed (ats, &hello_address_2, test_addr.session);
 120     test_addr.session = NULL;
 121     GNUNET_ATS_suggest_address (ats, &p.id);
 122   }
 123   if (1 == stage)
 124   {
 125     /* End */
 126     GNUNET_SCHEDULER_add_now (&end, NULL);
 127   }
 128   stage++;
 129 }
 130
 131
 132 static void
 133 run (void *cls,
 134      const struct GNUNET_CONFIGURATION_Handle *cfg,
 135      struct GNUNET_TESTING_Peer *peer)
 136 {
 137   ret = GNUNET_SYSERR;
 138
 139   die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL);
 140   ats = GNUNET_ATS_scheduling_init (cfg, &address_suggest_cb, NULL);
 141   if (ats == NULL)
 142   {
 143     ret = GNUNET_SYSERR;
 144     end ();
 145     return;
 146   }
 147
 148   /* set up peer */
 149   GNUNET_CRYPTO_hash_create_random (GNUNET_CRYPTO_QUALITY_WEAK,
 150                                     &p.id.hashPubKey);
 151   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Created peer `%s'\n",
 152               GNUNET_i2s (&p.id));
 153
 154   test_addr.plugin = "test";
 155   test_addr.session = &test_addr;
 156   test_addr.addr = GNUNET_strdup ("test");
 157   test_addr.addr_len = 4;
 158
 159   /* Adding address with session */
 160   hello_address.peer = p.id;
 161   hello_address.transport_name = test_addr.plugin;
 162   hello_address.address = test_addr.addr;
 163   hello_address.address_length = test_addr.addr_len;
 164   GNUNET_ATS_address_add (ats, &hello_address, test_addr.session, NULL, 0);
 165   GNUNET_ATS_suggest_address (ats, &p.id);
 166 }
 167
 168
 169 int
 170 main (int argc, char *argv[])
 171 {
 172   if (0 != GNUNET_TESTING_peer_run ("test_ats_api_scheduling_destroy_session",
 173                                     "test_ats_api.conf",
 174                                     &run, NULL))
 175     return 1;
 176   return ret;
 177 }
 178
 179 /* end of file test_ats_api_scheduling_destroy_session.c */