src/fs/test_fs_namespace.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2005, 2006, 2008, 2009 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 /**
  22  * @file fs/test_fs_namespace.c
  23  * @brief Test for fs_namespace.c
  24  * @author Christian Grothoff
  25  */
  26 #include "platform.h"
  27 #include "gnunet_util_lib.h"
  28 #include "gnunet_testing_lib.h"
  29 #include "gnunet_fs_service.h"
  30
  31
  32 static struct GNUNET_FS_PseudonymIdentifier nsid;
  33
  34 static struct GNUNET_FS_Uri *sks_expect_uri;
  35
  36 static struct GNUNET_FS_Uri *ksk_expect_uri;
  37
  38 static struct GNUNET_FS_Handle *fs;
  39
  40 static struct GNUNET_FS_SearchContext *sks_search;
  41
  42 static struct GNUNET_FS_SearchContext *ksk_search;
  43
  44 static GNUNET_SCHEDULER_TaskIdentifier kill_task;
  45
  46 static int update_started;
  47
  48 static int err;
  49
  50
  51 static void
  52 abort_ksk_search_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  53 {
  54   if (ksk_search != NULL)
  55   {
  56     GNUNET_FS_search_stop (ksk_search);
  57     ksk_search = NULL;
  58     if (sks_search == NULL)
  59     {
  60       GNUNET_FS_stop (fs);
  61       if (GNUNET_SCHEDULER_NO_TASK != kill_task)
  62         GNUNET_SCHEDULER_cancel (kill_task);
  63     }
  64   }
  65 }
  66
  67
  68 static void
  69 abort_sks_search_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  70 {
  71   struct GNUNET_FS_Namespace *ns;
  72
  73   if (sks_search == NULL)
  74     return;
  75   GNUNET_FS_search_stop (sks_search);
  76   sks_search = NULL;
  77   ns = GNUNET_FS_namespace_create (fs, "testNamespace");
  78   GNUNET_assert (NULL != ns);
  79   GNUNET_assert (GNUNET_OK == GNUNET_FS_namespace_delete (ns, GNUNET_YES));
  80   if (ksk_search == NULL)
  81   {
  82     GNUNET_FS_stop (fs);
  83     if (GNUNET_SCHEDULER_NO_TASK != kill_task)
  84       GNUNET_SCHEDULER_cancel (kill_task);
  85   }
  86 }
  87
  88
  89 static void
  90 do_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  91 {
  92   FPRINTF (stderr, "%s",  "Operation timed out\n");
  93   kill_task = GNUNET_SCHEDULER_NO_TASK;
  94   abort_sks_search_task (NULL, tc);
  95   abort_ksk_search_task (NULL, tc);
  96 }
  97
  98
  99 static void *
 100 progress_cb (void *cls, const struct GNUNET_FS_ProgressInfo *event)
 101 {
 102   switch (event->status)
 103   {
 104   case GNUNET_FS_STATUS_SEARCH_RESULT:
 105     if (sks_search == event->value.search.sc)
 106     {
 107       if (!GNUNET_FS_uri_test_equal
 108           (sks_expect_uri, event->value.search.specifics.result.uri))
 109       {
 110         FPRINTF (stderr, "%s",  "Wrong result for sks search!\n");
 111         err = 1;
 112       }
 113       /* give system 1ms to initiate update search! */
 114       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MILLISECONDS,
 115                                     &abort_sks_search_task, NULL);
 116     }
 117     else if (ksk_search == event->value.search.sc)
 118     {
 119       if (!GNUNET_FS_uri_test_equal
 120           (ksk_expect_uri, event->value.search.specifics.result.uri))
 121       {
 122         FPRINTF (stderr, "%s",  "Wrong result for ksk search!\n");
 123         err = 1;
 124       }
 125       GNUNET_SCHEDULER_add_continuation (&abort_ksk_search_task, NULL,
 126                                          GNUNET_SCHEDULER_REASON_PREREQ_DONE);
 127     }
 128     else
 129     {
 130       FPRINTF (stderr, "%s",  "Unexpected search result received!\n");
 131       GNUNET_break (0);
 132     }
 133     break;
 134   case GNUNET_FS_STATUS_SEARCH_ERROR:
 135     FPRINTF (stderr, "Error searching file: %s\n",
 136              event->value.search.specifics.error.message);
 137     if (sks_search == event->value.search.sc)
 138       GNUNET_SCHEDULER_add_continuation (&abort_sks_search_task, NULL,
 139                                          GNUNET_SCHEDULER_REASON_PREREQ_DONE);
 140     else if (ksk_search == event->value.search.sc)
 141       GNUNET_SCHEDULER_add_continuation (&abort_ksk_search_task, NULL,
 142                                          GNUNET_SCHEDULER_REASON_PREREQ_DONE);
 143     else
 144       GNUNET_break (0);
 145     break;
 146   case GNUNET_FS_STATUS_SEARCH_START:
 147     GNUNET_assert ((NULL == event->value.search.cctx) ||
 148                    (0 == strcmp ("sks_search", event->value.search.cctx)) ||
 149                    (0 == strcmp ("ksk_search", event->value.search.cctx)));
 150     if (NULL == event->value.search.cctx)
 151     {
 152       GNUNET_assert (0 == strcmp ("sks_search", event->value.search.pctx));
 153       update_started = GNUNET_YES;
 154     }
 155     GNUNET_assert (1 == event->value.search.anonymity);
 156     break;
 157   case GNUNET_FS_STATUS_SEARCH_RESULT_STOPPED:
 158     return NULL;
 159   case GNUNET_FS_STATUS_SEARCH_STOPPED:
 160     return NULL;
 161   default:
 162     FPRINTF (stderr, "Unexpected event: %d\n", event->status);
 163     break;
 164   }
 165   return event->value.search.cctx;
 166 }
 167
 168
 169 static void
 170 publish_cont (void *cls, const struct GNUNET_FS_Uri *ksk_uri, const char *emsg)
 171 {
 172   char *msg;
 173   struct GNUNET_FS_Uri *sks_uri;
 174   char sbuf[1024];
 175   char buf[1024];
 176   char *ret;
 177
 178   if (NULL != emsg)
 179   {
 180     FPRINTF (stderr, "Error publishing: %s\n", emsg);
 181     err = 1;
 182     GNUNET_FS_stop (fs);
 183     return;
 184   }
 185   ret = GNUNET_STRINGS_data_to_string (&nsid, sizeof (nsid), buf, sizeof (buf));
 186   GNUNET_assert (NULL != ret);
 187   ret[0] = '\0';
 188   GNUNET_snprintf (sbuf, sizeof (sbuf), "gnunet://fs/sks/%s/this", buf);
 189   sks_uri = GNUNET_FS_uri_parse (sbuf, &msg);
 190   if (NULL == sks_uri)
 191   {
 192     FPRINTF (stderr, "failed to parse URI `%s': %s\n", sbuf, msg);
 193     err = 1;
 194     GNUNET_FS_stop (fs);
 195     GNUNET_free_non_null (msg);
 196     return;
 197   }
 198   ksk_search =
 199       GNUNET_FS_search_start (fs, ksk_uri, 1, GNUNET_FS_SEARCH_OPTION_NONE,
 200                               "ksk_search");
 201   sks_search =
 202       GNUNET_FS_search_start (fs, sks_uri, 1, GNUNET_FS_SEARCH_OPTION_NONE,
 203                               "sks_search");
 204   GNUNET_FS_uri_destroy (sks_uri);
 205 }
 206
 207
 208 static void
 209 sks_cont (void *cls, const struct GNUNET_FS_Uri *uri, const char *emsg)
 210 {
 211   struct GNUNET_CONTAINER_MetaData *meta;
 212   struct GNUNET_FS_Uri *ksk_uri;
 213   char *msg;
 214   struct GNUNET_FS_BlockOptions bo;
 215
 216   if (NULL == uri)
 217   {
 218     fprintf (stderr, "Error publishing: %s\n", emsg);
 219     err = 1;
 220     GNUNET_FS_stop (fs);
 221     return;
 222   }
 223   meta = GNUNET_CONTAINER_meta_data_create ();
 224   msg = NULL;
 225   ksk_uri = GNUNET_FS_uri_parse ("gnunet://fs/ksk/ns-search", &msg);
 226   GNUNET_assert (NULL == msg);
 227   ksk_expect_uri = GNUNET_FS_uri_dup (uri);
 228   bo.content_priority = 1;
 229   bo.anonymity_level = 1;
 230   bo.replication_level = 0;
 231   bo.expiration_time =
 232       GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_MINUTES);
 233   GNUNET_FS_publish_ksk (fs, ksk_uri, meta, uri, &bo,
 234                          GNUNET_FS_PUBLISH_OPTION_NONE, &publish_cont, NULL);
 235   GNUNET_FS_uri_destroy (ksk_uri);
 236   GNUNET_CONTAINER_meta_data_destroy (meta);
 237 }
 238
 239
 240 static void
 241 adv_cont (void *cls, const struct GNUNET_FS_Uri *uri, const char *emsg)
 242 {
 243   struct GNUNET_CONTAINER_MetaData *meta;
 244   struct GNUNET_FS_Namespace *ns;
 245   struct GNUNET_FS_BlockOptions bo;
 246
 247   if (NULL != emsg)
 248   {
 249     FPRINTF (stderr, "Error publishing: %s\n", emsg);
 250     err = 1;
 251     GNUNET_FS_stop (fs);
 252     return;
 253   }
 254   ns = GNUNET_FS_namespace_create (fs, "testNamespace");
 255   GNUNET_assert (NULL != ns);
 256   meta = GNUNET_CONTAINER_meta_data_create ();
 257   GNUNET_assert (NULL == emsg);
 258   sks_expect_uri = GNUNET_FS_uri_dup (uri);
 259   bo.content_priority = 1;
 260   bo.anonymity_level = 1;
 261   bo.replication_level = 0;
 262   bo.expiration_time =
 263       GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_MINUTES);
 264   GNUNET_FS_publish_sks (fs, ns, "this", "next", meta, uri,
 265                          &bo, GNUNET_FS_PUBLISH_OPTION_NONE, &sks_cont, NULL);
 266   GNUNET_CONTAINER_meta_data_destroy (meta);
 267   GNUNET_FS_namespace_delete (ns, GNUNET_NO);
 268 }
 269
 270
 271 static void
 272 ns_iterator (void *cls, const char *name, const struct GNUNET_FS_PseudonymIdentifier *id)
 273 {
 274   int *ok = cls;
 275
 276   if (0 != strcmp (name, "testNamespace"))
 277     return;
 278   *ok = GNUNET_YES;
 279   nsid = *id;
 280 }
 281
 282
 283 static void
 284 testNamespace ()
 285 {
 286   struct GNUNET_FS_Namespace *ns;
 287   struct GNUNET_FS_BlockOptions bo;
 288   struct GNUNET_CONTAINER_MetaData *meta;
 289   struct GNUNET_FS_Uri *ksk_uri;
 290   struct GNUNET_FS_Uri *sks_uri;
 291   int ok;
 292
 293   ns = GNUNET_FS_namespace_create (fs, "testNamespace");
 294   GNUNET_assert (NULL != ns);
 295   ok = GNUNET_NO;
 296   GNUNET_FS_namespace_list (fs, &ns_iterator, &ok);
 297   if (GNUNET_NO == ok)
 298   {
 299     FPRINTF (stderr, "%s",  "namespace_list failed to find namespace!\n");
 300     GNUNET_FS_namespace_delete (ns, GNUNET_YES);
 301     GNUNET_FS_stop (fs);
 302     err = 1;
 303     return;
 304   }
 305   meta = GNUNET_CONTAINER_meta_data_create ();
 306   ksk_uri = GNUNET_FS_uri_parse ("gnunet://fs/ksk/testnsa", NULL);
 307   bo.content_priority = 1;
 308   bo.anonymity_level = 1;
 309   bo.replication_level = 0;
 310   bo.expiration_time =
 311       GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_MINUTES);
 312   sks_uri = GNUNET_FS_uri_sks_create (ns, "root", NULL);
 313   GNUNET_FS_publish_ksk (fs,
 314                          ksk_uri, meta, sks_uri, &bo, GNUNET_FS_PUBLISH_OPTION_NONE,
 315                          &adv_cont, NULL);
 316   GNUNET_FS_uri_destroy (sks_uri);
 317   kill_task =
 318       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES, &do_timeout,
 319                                     NULL);
 320   GNUNET_FS_uri_destroy (ksk_uri);
 321   GNUNET_FS_namespace_delete (ns, GNUNET_NO);
 322   GNUNET_CONTAINER_meta_data_destroy (meta);
 323 }
 324
 325
 326 static void
 327 run (void *cls,
 328      const struct GNUNET_CONFIGURATION_Handle *cfg,
 329      struct GNUNET_TESTING_Peer *peer)
 330 {
 331   fs = GNUNET_FS_start (cfg, "test-fs-namespace", &progress_cb, NULL,
 332                         GNUNET_FS_FLAGS_NONE, GNUNET_FS_OPTIONS_END);
 333   testNamespace ();
 334 }
 335
 336
 337 int
 338 main (int argc, char *argv[])
 339 {
 340   if (0 != GNUNET_TESTING_peer_run ("test-fs-namespace",
 341                                     "test_fs_namespace_data.conf",
 342                                     &run, NULL))
 343     return 1;
 344   return err;
 345 }
 346
 347
 348 /* end of test_fs_namespace.c */