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