src/fs/fs_publish.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 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 2, 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/fs_publish.c
  23  * @brief publish a file or directory in GNUnet
  24  * @see http://gnunet.org/encoding.php3
  25  * @author Krista Bennett
  26  * @author Christian Grothoff
  27  *
  28  * TODO:
  29  * - indexing cleanup: unindex on failure (can wait)
  30  * - persistence support (can wait)
  31  * - datastore reservation support (optimization)
  32  * - location URIs (publish with anonymity-level zero)
  33  */
  34
  35 #include "platform.h"
  36 #include "gnunet_constants.h"
  37 #include "gnunet_signatures.h"
  38 #include "gnunet_util_lib.h"
  39 #include "gnunet_fs_service.h"
  40 #include "fs.h"
  41 #include "fs_tree.h"
  42
  43 #define DEBUG_PUBLISH GNUNET_NO
  44
  45 /**
  46  * Maximum allowed size for a KBlock.
  47  */
  48 #define MAX_KBLOCK_SIZE 60000
  49
  50 /**
  51  * Maximum allowed size for an SBlock.
  52  */
  53 #define MAX_SBLOCK_SIZE 60000
  54
  55 /**
  56  * Main function that performs the upload.
  57  * @param cls "struct GNUNET_FS_PublishContext" identifies the upload
  58  * @param tc task context
  59  */
  60 static void
  61 do_upload (void *cls,
  62            const struct GNUNET_SCHEDULER_TaskContext *tc);
  63
  64
  65 /**
  66  * Context for "ds_put_cont".
  67  */
  68 struct PutContCtx
  69 {
  70   /**
  71    * Current publishing context.
  72    */
  73   struct GNUNET_FS_PublishContext *sc;
  74
  75   /**
  76    * Specific file with the block.
  77    */
  78   struct GNUNET_FS_FileInformation *p;
  79
  80   /**
  81    * Function to run next, if any (can be NULL).
  82    */
  83   GNUNET_SCHEDULER_Task cont;
  84
  85   /**
  86    * Closure for cont.
  87    */
  88   void *cont_cls;
  89 };
  90
  91
  92 /**
  93  * Fill in all of the generic fields for
  94  * a publish event.
  95  *
  96  * @param pi structure to fill in
  97  * @param sc overall publishing context
  98  * @param p file information for the file being published
  99  * @param offset where in the file are we so far
 100  */
 101 static void
 102 make_publish_status (struct GNUNET_FS_ProgressInfo *pi,
 103                      struct GNUNET_FS_PublishContext *sc,
 104                      const struct GNUNET_FS_FileInformation *p,
 105                      uint64_t offset)
 106 {
 107   pi->value.publish.sc = sc;
 108   pi->value.publish.fi = p;
 109   pi->value.publish.cctx
 110     = p->client_info;
 111   pi->value.publish.pctx
 112     = (NULL == p->dir) ? NULL : p->dir->client_info;
 113   pi->value.publish.filename
 114     = (p->is_directory) ? p->data.dir.dirname : p->data.file.filename;
 115   pi->value.publish.size
 116     = (p->is_directory) ? p->data.dir.dir_size : p->data.file.file_size;
 117   pi->value.publish.eta
 118     = GNUNET_TIME_calculate_eta (p->start_time,
 119                                  offset,
 120                                  pi->value.publish.size);
 121   pi->value.publish.completed = offset;
 122   pi->value.publish.duration = GNUNET_TIME_absolute_get_duration (p->start_time);
 123   pi->value.publish.anonymity = p->anonymity;
 124 }
 125
 126
 127 /**
 128  * Cleanup the publish context, we're done
 129  * with it.
 130  *
 131  * @param sc struct to clean up after
 132  */
 133 static void
 134 publish_cleanup (struct GNUNET_FS_PublishContext *sc)
 135 {
 136   GNUNET_FS_file_information_destroy (sc->fi, NULL, NULL);
 137   if (sc->namespace != NULL)
 138     GNUNET_FS_namespace_delete (sc->namespace, GNUNET_NO);
 139   GNUNET_free_non_null (sc->nid);
 140   GNUNET_free_non_null (sc->nuid);
 141   GNUNET_DATASTORE_disconnect (sc->dsh, GNUNET_NO);
 142   GNUNET_free (sc);
 143 }
 144
 145
 146 /**
 147  * Function called by the datastore API with
 148  * the result from the PUT request.
 149  *
 150  * @param cls our closure
 151  * @param success GNUNET_OK on success
 152  * @param msg error message (or NULL)
 153  */
 154 static void
 155 ds_put_cont (void *cls,
 156              int success,
 157              const char *msg)
 158 {
 159   struct PutContCtx *pcc = cls;
 160   struct GNUNET_FS_ProgressInfo pi;
 161
 162   if (GNUNET_SYSERR == pcc->sc->in_network_wait)
 163     {
 164       /* we were aborted in the meantime,
 165          finish shutdown! */
 166       publish_cleanup (pcc->sc);
 167       return;
 168     }
 169   GNUNET_assert (GNUNET_YES == pcc->sc->in_network_wait);
 170   pcc->sc->in_network_wait = GNUNET_NO;
 171   if (GNUNET_OK != success)
 172     {
 173       GNUNET_asprintf (&pcc->p->emsg,
 174                        _("Upload failed: %s"),
 175                        msg);
 176       GNUNET_FS_file_information_sync (pcc->p);
 177       pi.status = GNUNET_FS_STATUS_PUBLISH_ERROR;
 178       make_publish_status (&pi, pcc->sc, pcc->p, 0);
 179       pi.value.publish.eta = GNUNET_TIME_UNIT_FOREVER_REL;
 180       pi.value.publish.specifics.error.message = pcc->p->emsg;
 181       pcc->p->client_info
 182         = pcc->sc->h->upcb (pcc->sc->h->upcb_cls,
 183                             &pi);
 184     }
 185   GNUNET_FS_file_information_sync (pcc->p);
 186   if (NULL != pcc->cont)
 187     pcc->sc->upload_task
 188       = GNUNET_SCHEDULER_add_with_priority (pcc->sc->h->sched,
 189                                             GNUNET_SCHEDULER_PRIORITY_BACKGROUND,
 190                                             pcc->cont,
 191                                             pcc->cont_cls);
 192   GNUNET_free (pcc);
 193 }
 194
 195
 196 /**
 197  * Generate the callback that signals clients
 198  * that a file (or directory) has been completely
 199  * published.
 200  *
 201  * @param p the completed upload
 202  * @param sc context of the publication
 203  */
 204 static void
 205 signal_publish_completion (struct GNUNET_FS_FileInformation *p,
 206                            struct GNUNET_FS_PublishContext *sc)
 207 {
 208   struct GNUNET_FS_ProgressInfo pi;
 209
 210   pi.status = GNUNET_FS_STATUS_PUBLISH_COMPLETED;
 211   make_publish_status (&pi, sc, p,
 212                        GNUNET_ntohll (p->chk_uri->data.chk.file_length));
 213   pi.value.publish.eta = GNUNET_TIME_UNIT_ZERO;
 214   pi.value.publish.specifics.completed.chk_uri = p->chk_uri;
 215   p->client_info
 216     = sc->h->upcb (sc->h->upcb_cls,
 217                   &pi);
 218 }
 219
 220
 221 /**
 222  * Generate the callback that signals clients
 223  * that a file (or directory) has encountered
 224  * a problem during publication.
 225  *
 226  * @param p the upload that had trouble
 227  * @param sc context of the publication
 228  * @param emsg error message
 229  */
 230 static void
 231 signal_publish_error (struct GNUNET_FS_FileInformation *p,
 232                       struct GNUNET_FS_PublishContext *sc,
 233                       const char *emsg)
 234 {
 235   struct GNUNET_FS_ProgressInfo pi;
 236
 237   p->emsg = GNUNET_strdup (emsg);
 238   pi.status = GNUNET_FS_STATUS_PUBLISH_ERROR;
 239   make_publish_status (&pi, sc, p, 0);
 240   pi.value.publish.eta = GNUNET_TIME_UNIT_FOREVER_REL;
 241   pi.value.publish.specifics.error.message =emsg;
 242   p->client_info
 243     = sc->h->upcb (sc->h->upcb_cls,
 244                   &pi);
 245 }
 246
 247
 248 /**
 249  * We've finished publishing the SBlock as part of a larger upload.
 250  * Check the result and complete the larger upload.
 251  *
 252  * @param cls the "struct GNUNET_FS_PublishContext*" of the larger upload
 253  * @param uri URI of the published SBlock
 254  * @param emsg NULL on success, otherwise error message
 255  */
 256 static void
 257 publish_sblocks_cont (void *cls,
 258                       const struct GNUNET_FS_Uri *uri,
 259                       const char *emsg)
 260 {
 261   struct GNUNET_FS_PublishContext *sc = cls;
 262   if (NULL != emsg)
 263     {
 264       signal_publish_error (sc->fi,
 265                             sc,
 266                             emsg);
 267       return;
 268     }
 269   // FIXME: release the datastore reserve here!
 270   signal_publish_completion (sc->fi, sc);
 271   sc->all_done = GNUNET_YES;
 272 }
 273
 274
 275 /**
 276  * We are almost done publishing the structure,
 277  * add SBlocks (if needed).
 278  *
 279  * @param sc overall upload data
 280  */
 281 static void
 282 publish_sblock (struct GNUNET_FS_PublishContext *sc)
 283 {
 284   if (NULL != sc->namespace)
 285     GNUNET_FS_publish_sks (sc->h,
 286                            sc->namespace,
 287                            sc->nid,
 288                            sc->nuid,
 289                            sc->fi->meta,
 290                            sc->fi->chk_uri,
 291                            sc->fi->expirationTime,
 292                            sc->fi->anonymity,
 293                            sc->fi->priority,
 294                            sc->options,
 295                            &publish_sblocks_cont,
 296                            sc);
 297   else
 298     publish_sblocks_cont (sc, NULL, NULL);
 299 }
 300
 301
 302 /**
 303  * We've finished publishing a KBlock as part of a larger upload.
 304  * Check the result and continue the larger upload.
 305  *
 306  * @param cls the "struct GNUNET_FS_PublishContext*"
 307  *        of the larger upload
 308  * @param uri URI of the published blocks
 309  * @param emsg NULL on success, otherwise error message
 310  */
 311 static void
 312 publish_kblocks_cont (void *cls,
 313                       const struct GNUNET_FS_Uri *uri,
 314                       const char *emsg)
 315 {
 316   struct GNUNET_FS_PublishContext *sc = cls;
 317   struct GNUNET_FS_FileInformation *p = sc->fi_pos;
 318
 319   if (NULL != emsg)
 320     {
 321       signal_publish_error (p, sc, emsg);
 322       sc->upload_task
 323         = GNUNET_SCHEDULER_add_with_priority (sc->h->sched,
 324                                               GNUNET_SCHEDULER_PRIORITY_BACKGROUND,
 325                                               &do_upload,
 326                                               sc);
 327       return;
 328     }
 329   GNUNET_FS_file_information_sync (p);
 330   if (NULL != p->dir)
 331     signal_publish_completion (p, sc);
 332   /* move on to next file */
 333   if (NULL != p->next)
 334     sc->fi_pos = p->next;
 335   else
 336     sc->fi_pos = p->dir;
 337   sc->upload_task
 338     = GNUNET_SCHEDULER_add_with_priority (sc->h->sched,
 339                                           GNUNET_SCHEDULER_PRIORITY_BACKGROUND,
 340                                           &do_upload,
 341                                           sc);
 342 }
 343
 344
 345 /**
 346  * Function called by the tree encoder to obtain
 347  * a block of plaintext data (for the lowest level
 348  * of the tree).
 349  *
 350  * @param cls our publishing context
 351  * @param offset identifies which block to get
 352  * @param max (maximum) number of bytes to get; returning
 353  *        fewer will also cause errors
 354  * @param buf where to copy the plaintext buffer
 355  * @param emsg location to store an error message (on error)
 356  * @return number of bytes copied to buf, 0 on error
 357  */
 358 static size_t
 359 block_reader (void *cls,
 360               uint64_t offset,
 361               size_t max,
 362               void *buf,
 363               char **emsg)
 364 {
 365   struct GNUNET_FS_PublishContext *sc = cls;
 366   struct GNUNET_FS_FileInformation *p;
 367   size_t pt_size;
 368   const char *dd;
 369
 370   p = sc->fi_pos;
 371   if (p->is_directory)
 372     {
 373       pt_size = GNUNET_MIN(max,
 374                            p->data.dir.dir_size - offset);
 375       dd = p->data.dir.dir_data;
 376       memcpy (buf,
 377               &dd[offset],
 378               pt_size);
 379     }
 380   else
 381     {
 382       pt_size = GNUNET_MIN(max,
 383                            p->data.file.file_size - offset);
 384       if (pt_size == 0)
 385         return 0; /* calling reader with pt_size==0
 386                      might free buf, so don't! */
 387       if (pt_size !=
 388           p->data.file.reader (p->data.file.reader_cls,
 389                                offset,
 390                                pt_size,
 391                                buf,
 392                                emsg))
 393         return 0;
 394     }
 395   return pt_size;
 396 }
 397
 398
 399 /**
 400  * The tree encoder has finished processing a
 401  * file.   Call it's finish method and deal with
 402  * the final result.
 403  *
 404  * @param cls our publishing context
 405  * @param tc scheduler's task context (not used)
 406  */
 407 static void
 408 encode_cont (void *cls,
 409              const struct GNUNET_SCHEDULER_TaskContext *tc)
 410 {
 411   struct GNUNET_FS_PublishContext *sc = cls;
 412   struct GNUNET_FS_FileInformation *p;
 413   struct GNUNET_FS_ProgressInfo pi;
 414   char *emsg;
 415
 416   p = sc->fi_pos;
 417   GNUNET_FS_tree_encoder_finish (p->te,
 418                                  &p->chk_uri,
 419                                  &emsg);
 420   p->te = NULL;
 421   if (NULL != emsg)
 422     {
 423       GNUNET_asprintf (&p->emsg,
 424                        _("Upload failed: %s"),
 425                        emsg);
 426       GNUNET_free (emsg);
 427       GNUNET_FS_file_information_sync (p);
 428       pi.status = GNUNET_FS_STATUS_PUBLISH_ERROR;
 429       make_publish_status (&pi, sc, p, 0);
 430       pi.value.publish.eta = GNUNET_TIME_UNIT_FOREVER_REL;
 431       pi.value.publish.specifics.error.message = p->emsg;
 432       p->client_info
 433         = sc->h->upcb (sc->h->upcb_cls,
 434                        &pi);
 435     }
 436   /* continue with main */
 437   sc->upload_task
 438     = GNUNET_SCHEDULER_add_with_priority (sc->h->sched,
 439                                           GNUNET_SCHEDULER_PRIORITY_BACKGROUND,
 440                                     &do_upload,
 441                                     sc);
 442 }
 443
 444
 445 /**
 446  * Function called asking for the current (encoded)
 447  * block to be processed.  After processing the
 448  * client should either call "GNUNET_FS_tree_encode_next"
 449  * or (on error) "GNUNET_FS_tree_encode_finish".
 450  *
 451  * @param cls closure
 452  * @param query the query for the block (key for lookup in the datastore)
 453  * @param offset offset of the block in the file
 454  * @param type type of the block (IBLOCK or DBLOCK)
 455  * @param block the (encrypted) block
 456  * @param block_size size of block (in bytes)
 457  */
 458 static void
 459 block_proc (void *cls,
 460             const GNUNET_HashCode *query,
 461             uint64_t offset,
 462             uint32_t type,
 463             const void *block,
 464             uint16_t block_size)
 465 {
 466   struct GNUNET_FS_PublishContext *sc = cls;
 467   struct GNUNET_FS_FileInformation *p;
 468   struct PutContCtx * dpc_cls;
 469   struct OnDemandBlock odb;
 470
 471   p = sc->fi_pos;
 472   if (NULL == sc->dsh)
 473     {
 474       sc->upload_task
 475         = GNUNET_SCHEDULER_add_with_priority (sc->h->sched,
 476                                         GNUNET_SCHEDULER_PRIORITY_BACKGROUND,
 477                                         &do_upload,
 478                                         sc);
 479       return;
 480     }
 481
 482   GNUNET_assert (GNUNET_NO == sc->in_network_wait);
 483   sc->in_network_wait = GNUNET_YES;
 484   dpc_cls = GNUNET_malloc(sizeof(struct PutContCtx));
 485   dpc_cls->cont = &do_upload;
 486   dpc_cls->cont_cls = sc;
 487   dpc_cls->sc = sc;
 488   dpc_cls->p = p;
 489   if ( (! p->is_directory) &&
 490        (GNUNET_YES == p->data.file.do_index) &&
 491        (type == GNUNET_DATASTORE_BLOCKTYPE_DBLOCK) )
 492     {
 493 #if DEBUG_PUBLISH
 494       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 495                   "Indexing block `%s' for offset %llu with index size %u\n",
 496                   GNUNET_h2s (query),
 497                   (unsigned long long) offset,
 498                   sizeof (struct OnDemandBlock));
 499 #endif
 500       odb.offset = GNUNET_htonll (offset);
 501       odb.file_id = p->data.file.file_id;
 502       GNUNET_DATASTORE_put (sc->dsh,
 503                             sc->rid,
 504                             query,
 505                             sizeof(struct OnDemandBlock),
 506                             &odb,
 507                             GNUNET_DATASTORE_BLOCKTYPE_ONDEMAND,
 508                             p->priority,
 509                             p->anonymity,
 510                             p->expirationTime,
 511                             GNUNET_CONSTANTS_SERVICE_TIMEOUT,
 512                             &ds_put_cont,
 513                             dpc_cls);
 514       return;
 515     }
 516 #if DEBUG_PUBLISH
 517   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 518               "Publishing block `%s' for offset %llu with size %u\n",
 519               GNUNET_h2s (query),
 520               (unsigned long long) offset,
 521               (unsigned int) block_size);
 522 #endif
 523   GNUNET_DATASTORE_put (sc->dsh,
 524                         sc->rid,
 525                         query,
 526                         block_size,
 527                         block,
 528                         type,
 529                         p->priority,
 530                         p->anonymity,
 531                         p->expirationTime,
 532                         GNUNET_CONSTANTS_SERVICE_TIMEOUT,
 533                         &ds_put_cont,
 534                         dpc_cls);
 535 }
 536
 537
 538 /**
 539  * Function called with information about our
 540  * progress in computing the tree encoding.
 541  *
 542  * @param cls closure
 543  * @param offset where are we in the file
 544  * @param pt_block plaintext of the currently processed block
 545  * @param pt_size size of pt_block
 546  * @param depth depth of the block in the tree
 547  */
 548 static void
 549 progress_proc (void *cls,
 550                uint64_t offset,
 551                const void *pt_block,
 552                size_t pt_size,
 553                unsigned int depth)
 554 {
 555   struct GNUNET_FS_PublishContext *sc = cls;
 556   struct GNUNET_FS_FileInformation *p;
 557   struct GNUNET_FS_ProgressInfo pi;
 558
 559   p = sc->fi_pos;
 560   pi.status = GNUNET_FS_STATUS_PUBLISH_PROGRESS;
 561   make_publish_status (&pi, sc, p, offset);
 562   pi.value.publish.specifics.progress.data = pt_block;
 563   pi.value.publish.specifics.progress.offset = offset;
 564   pi.value.publish.specifics.progress.data_len = pt_size;
 565   pi.value.publish.specifics.progress.depth = depth;
 566   p->client_info
 567     = sc->h->upcb (sc->h->upcb_cls,
 568                    &pi);
 569 }
 570
 571
 572 /**
 573  * We are uploading a file or directory; load (if necessary) the next
 574  * block into memory, encrypt it and send it to the FS service.  Then
 575  * continue with the main task.
 576  *
 577  * @param sc overall upload data
 578  */
 579 static void
 580 publish_content (struct GNUNET_FS_PublishContext *sc)
 581 {
 582   struct GNUNET_FS_FileInformation *p;
 583   char *emsg;
 584   struct GNUNET_FS_DirectoryBuilder *db;
 585   struct GNUNET_FS_FileInformation *dirpos;
 586   void *raw_data;
 587   uint64_t size;
 588
 589   p = sc->fi_pos;
 590   if (NULL == p->te)
 591     {
 592       if (p->is_directory)
 593         {
 594           db = GNUNET_FS_directory_builder_create (p->meta);
 595           dirpos = p->data.dir.entries;
 596           while (NULL != dirpos)
 597             {
 598               if (dirpos->is_directory)
 599                 {
 600                   raw_data = dirpos->data.dir.dir_data;
 601                   dirpos->data.dir.dir_data = NULL;
 602                 }
 603               else
 604                 {
 605                   raw_data = NULL;
 606                   if ( (dirpos->data.file.file_size < MAX_INLINE_SIZE) &&
 607                        (dirpos->data.file.file_size > 0) )
 608                     {
 609                       raw_data = GNUNET_malloc (dirpos->data.file.file_size);
 610                       emsg = NULL;
 611                       if (dirpos->data.file.file_size !=
 612                           dirpos->data.file.reader (dirpos->data.file.reader_cls,
 613                                                     0,
 614                                                     dirpos->data.file.file_size,
 615                                                     raw_data,
 616                                                     &emsg))
 617                         {
 618                           GNUNET_free_non_null (emsg);
 619                           GNUNET_free (raw_data);
 620                           raw_data = NULL;
 621                         }
 622                     }
 623                 }
 624               GNUNET_FS_directory_builder_add (db,
 625                                                dirpos->chk_uri,
 626                                                dirpos->meta,
 627                                                raw_data);
 628               GNUNET_free_non_null (raw_data);
 629               dirpos = dirpos->next;
 630             }
 631           GNUNET_FS_directory_builder_finish (db,
 632                                               &p->data.dir.dir_size,
 633                                               &p->data.dir.dir_data);
 634         }
 635       size = (p->is_directory)
 636         ? p->data.dir.dir_size
 637         : p->data.file.file_size;
 638       p->te = GNUNET_FS_tree_encoder_create (sc->h,
 639                                              size,
 640                                              sc,
 641                                              &block_reader,
 642                                              &block_proc,
 643                                              &progress_proc,
 644                                              &encode_cont);
 645
 646     }
 647   GNUNET_FS_tree_encoder_next (p->te);
 648 }
 649
 650
 651 /**
 652  * Process the response (or lack thereof) from
 653  * the "fs" service to our 'start index' request.
 654  *
 655  * @param cls closure (of type "struct GNUNET_FS_PublishContext*"_)
 656  * @param msg the response we got
 657  */
 658 static void
 659 process_index_start_response (void *cls,
 660                               const struct GNUNET_MessageHeader *msg)
 661 {
 662   struct GNUNET_FS_PublishContext *sc = cls;
 663   struct GNUNET_FS_FileInformation *p;
 664   const char *emsg;
 665   uint16_t msize;
 666
 667   GNUNET_CLIENT_disconnect (sc->client, GNUNET_NO);
 668   sc->client = NULL;
 669   p = sc->fi_pos;
 670   if (msg == NULL)
 671     {
 672       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 673                   _("Can not index file `%s': %s.  Will try to insert instead.\n"),
 674                   p->data.file.filename,
 675                   _("timeout on index-start request to `fs' service"));
 676       p->data.file.do_index = GNUNET_NO;
 677       publish_content (sc);
 678       return;
 679     }
 680   if (ntohs (msg->type) != GNUNET_MESSAGE_TYPE_FS_INDEX_START_OK)
 681     {
 682       msize = ntohs (msg->size);
 683       emsg = (const char *) &msg[1];
 684       if ( (msize <= sizeof (struct GNUNET_MessageHeader)) ||
 685            (emsg[msize - sizeof(struct GNUNET_MessageHeader) - 1] != '\0') )
 686         emsg = gettext_noop ("unknown error");
 687       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 688                   _("Can not index file `%s': %s.  Will try to insert instead.\n"),
 689                   p->data.file.filename,
 690                   gettext (emsg));
 691       p->data.file.do_index = GNUNET_NO;
 692       publish_content (sc);
 693       return;
 694     }
 695   p->data.file.index_start_confirmed = GNUNET_YES;
 696   /* success! continue with indexing */
 697   publish_content (sc);
 698 }
 699
 700
 701 /**
 702  * Function called once the hash computation over an
 703  * indexed file has completed.
 704  *
 705  * @param cls closure, our publishing context
 706  * @param res resulting hash, NULL on error
 707  */
 708 static void
 709 hash_for_index_cb (void *cls,
 710                    const GNUNET_HashCode *
 711                    res)
 712 {
 713   struct GNUNET_FS_PublishContext *sc = cls;
 714   struct GNUNET_FS_FileInformation *p;
 715   struct IndexStartMessage *ism;
 716   size_t slen;
 717   struct GNUNET_CLIENT_Connection *client;
 718   uint32_t dev;
 719   uint64_t ino;
 720
 721   p = sc->fi_pos;
 722   if (NULL == res)
 723     {
 724       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 725                   _("Can not index file `%s': %s.  Will try to insert instead.\n"),
 726                   p->data.file.filename,
 727                   _("failed to compute hash"));
 728       p->data.file.do_index = GNUNET_NO;
 729       publish_content (sc);
 730       return;
 731     }
 732   if (GNUNET_YES == p->data.file.index_start_confirmed)
 733     {
 734       publish_content (sc);
 735       return;
 736     }
 737   slen = strlen (p->data.file.filename) + 1;
 738   if (slen > GNUNET_SERVER_MAX_MESSAGE_SIZE - sizeof(struct IndexStartMessage))
 739     {
 740       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 741                   _("Can not index file `%s': %s.  Will try to insert instead.\n"),
 742                   p->data.file.filename,
 743                   _("filename too long"));
 744       p->data.file.do_index = GNUNET_NO;
 745       publish_content (sc);
 746       return;
 747     }
 748 #if DEBUG_PUBLISH
 749   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 750               "Hash of indexed file `%s' is `%s'\n",
 751               p->data.file.filename,
 752               GNUNET_h2s (res));
 753 #endif
 754   client = GNUNET_CLIENT_connect (sc->h->sched,
 755                                   "fs",
 756                                   sc->h->cfg);
 757   if (NULL == client)
 758     {
 759       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 760                   _("Can not index file `%s': %s.  Will try to insert instead.\n"),
 761                   p->data.file.filename,
 762                   _("could not connect to `fs' service"));
 763       p->data.file.do_index = GNUNET_NO;
 764       publish_content (sc);
 765       return;
 766     }
 767   p->data.file.file_id = *res;
 768   p->data.file.have_hash = GNUNET_YES;
 769   ism = GNUNET_malloc (sizeof(struct IndexStartMessage) +
 770                        slen);
 771   ism->header.size = htons(sizeof(struct IndexStartMessage) +
 772                            slen);
 773   ism->header.type = htons(GNUNET_MESSAGE_TYPE_FS_INDEX_START);
 774   if (GNUNET_OK ==
 775       GNUNET_DISK_file_get_identifiers (p->data.file.filename,
 776                                         &dev,
 777                                         &ino))
 778     {
 779       ism->device = htonl (dev);
 780       ism->inode = GNUNET_htonll(ino);
 781     }
 782   else
 783     {
 784       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 785                   _("Failed to get file identifiers for `%s'\n"),
 786                   p->data.file.filename);
 787     }
 788   ism->file_id = *res;
 789   memcpy (&ism[1],
 790           p->data.file.filename,
 791           slen);
 792   sc->client = client;
 793   GNUNET_break (GNUNET_YES ==
 794                 GNUNET_CLIENT_transmit_and_get_response (client,
 795                                                          &ism->header,
 796                                                          GNUNET_TIME_UNIT_FOREVER_REL,
 797                                                          GNUNET_YES,
 798                                                          &process_index_start_response,
 799                                                          sc));
 800   GNUNET_free (ism);
 801 }
 802
 803
 804 /**
 805  * Main function that performs the upload.
 806  * @param cls "struct GNUNET_FS_PublishContext" identifies the upload
 807  * @param tc task context
 808  */
 809 static void
 810 do_upload (void *cls,
 811            const struct GNUNET_SCHEDULER_TaskContext *tc)
 812 {
 813   struct GNUNET_FS_PublishContext *sc = cls;
 814   struct GNUNET_FS_ProgressInfo pi;
 815   struct GNUNET_FS_FileInformation *p;
 816   char *fn;
 817
 818   sc->upload_task = GNUNET_SCHEDULER_NO_TASK;
 819   p = sc->fi_pos;
 820   if (NULL == p)
 821     {
 822       /* upload of entire hierarchy complete,
 823          publish namespace entries */
 824       publish_sblock (sc);
 825       return;
 826     }
 827   /* find starting position */
 828   while ( (p->is_directory) &&
 829           (NULL != p->data.dir.entries) &&
 830           (NULL == p->emsg) &&
 831           (NULL == p->data.dir.entries->chk_uri) )
 832     {
 833       p = p->data.dir.entries;
 834       sc->fi_pos = p;
 835     }
 836   /* abort on error */
 837   if (NULL != p->emsg)
 838     {
 839       /* error with current file, abort all
 840          related files as well! */
 841       while (NULL != p->dir)
 842         {
 843           fn = GNUNET_CONTAINER_meta_data_get_by_type (p->meta,
 844                                                        EXTRACTOR_METATYPE_FILENAME);
 845           p = p->dir;
 846           GNUNET_asprintf (&p->emsg,
 847                            _("Recursive upload failed at `%s'"),
 848                            fn);
 849           GNUNET_free (fn);
 850           GNUNET_FS_file_information_sync (p);
 851           pi.status = GNUNET_FS_STATUS_PUBLISH_ERROR;
 852           make_publish_status (&pi, sc, p, 0);
 853           pi.value.publish.eta = GNUNET_TIME_UNIT_FOREVER_REL;
 854           pi.value.publish.specifics.error.message = p->emsg;
 855           p->client_info
 856             = sc->h->upcb (sc->h->upcb_cls,
 857                            &pi);
 858         }
 859       sc->all_done = GNUNET_YES;
 860       return;
 861     }
 862   /* handle completion */
 863   if (NULL != p->chk_uri)
 864     {
 865       /* upload of "p" complete, publish KBlocks! */
 866       if (p->keywords != NULL)
 867         {
 868           GNUNET_FS_publish_ksk (sc->h,
 869                                  p->keywords,
 870                                  p->meta,
 871                                  p->chk_uri,
 872                                  p->expirationTime,
 873                                  p->anonymity,
 874                                  p->priority,
 875                                  sc->options,
 876                                  &publish_kblocks_cont,
 877                                  sc);
 878         }
 879       else
 880         {
 881           publish_kblocks_cont (sc,
 882                                 p->chk_uri,
 883                                 NULL);
 884         }
 885       return;
 886     }
 887   if ( (!p->is_directory) &&
 888        (p->data.file.do_index) )
 889     {
 890       if (NULL == p->data.file.filename)
 891         {
 892           p->data.file.do_index = GNUNET_NO;
 893           GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 894                       _("Can not index file `%s': %s.  Will try to insert instead.\n"),
 895                       "<no-name>",
 896                       _("needs to be an actual file"));
 897           publish_content (sc);
 898           return;
 899         }
 900       if (p->data.file.have_hash)
 901         hash_for_index_cb (sc,
 902                            &p->data.file.file_id);
 903       else
 904         {
 905           p->start_time = GNUNET_TIME_absolute_get ();
 906           GNUNET_CRYPTO_hash_file (sc->h->sched,
 907                                    GNUNET_SCHEDULER_PRIORITY_IDLE,
 908                                    p->data.file.filename,
 909                                    HASHING_BLOCKSIZE,
 910                                    &hash_for_index_cb,
 911                                    sc);
 912         }
 913       return;
 914     }
 915   publish_content (sc);
 916 }
 917
 918
 919 /**
 920  * Signal the FS's progress function that we are starting
 921  * an upload.
 922  *
 923  * @param cls closure (of type "struct GNUNET_FS_PublishContext*")
 924  * @param fi the entry in the publish-structure
 925  * @param length length of the file or directory
 926  * @param meta metadata for the file or directory (can be modified)
 927  * @param uri pointer to the keywords that will be used for this entry (can be modified)
 928  * @param anonymity pointer to selected anonymity level (can be modified)
 929  * @param priority pointer to selected priority (can be modified)
 930  * @param expirationTime pointer to selected expiration time (can be modified)
 931  * @param client_info pointer to client context set upon creation (can be modified)
 932  * @return GNUNET_OK to continue (always)
 933  */
 934 static int
 935 fip_signal_start(void *cls,
 936                  struct GNUNET_FS_FileInformation *fi,
 937                  uint64_t length,
 938                  struct GNUNET_CONTAINER_MetaData *meta,
 939                  struct GNUNET_FS_Uri **uri,
 940                  uint32_t *anonymity,
 941                  uint32_t *priority,
 942                  struct GNUNET_TIME_Absolute *expirationTime,
 943                  void **client_info)
 944 {
 945   struct GNUNET_FS_PublishContext *sc = cls;
 946   struct GNUNET_FS_ProgressInfo pi;
 947
 948   pi.status = GNUNET_FS_STATUS_PUBLISH_START;
 949   make_publish_status (&pi, sc, fi, 0);
 950   *client_info = sc->h->upcb (sc->h->upcb_cls,
 951                               &pi);
 952   return GNUNET_OK;
 953 }
 954
 955
 956 /**
 957  * Publish a file or directory.
 958  *
 959  * @param h handle to the file sharing subsystem
 960  * @param fi information about the file or directory structure to publish
 961  * @param namespace namespace to publish the file in, NULL for no namespace
 962  * @param nid identifier to use for the publishd content in the namespace
 963  *        (can be NULL, must be NULL if namespace is NULL)
 964  * @param nuid update-identifier that will be used for future updates
 965  *        (can be NULL, must be NULL if namespace or nid is NULL)
 966  * @param options options for the publication
 967  * @return context that can be used to control the publish operation
 968  */
 969 struct GNUNET_FS_PublishContext *
 970 GNUNET_FS_publish_start (struct GNUNET_FS_Handle *h,
 971                          struct GNUNET_FS_FileInformation *fi,
 972                          struct GNUNET_FS_Namespace *namespace,
 973                          const char *nid,
 974                          const char *nuid,
 975                          enum GNUNET_FS_PublishOptions options)
 976 {
 977   struct GNUNET_FS_PublishContext *ret;
 978   struct GNUNET_DATASTORE_Handle *dsh;
 979
 980   if (0 == (options & GNUNET_FS_PUBLISH_OPTION_SIMULATE_ONLY))
 981     {
 982       dsh = GNUNET_DATASTORE_connect (h->cfg,
 983                                       h->sched);
 984       if (NULL == dsh)
 985         return NULL;
 986     }
 987   else
 988     {
 989       dsh = NULL;
 990     }
 991   ret = GNUNET_malloc (sizeof (struct GNUNET_FS_PublishContext));
 992   ret->dsh = dsh;
 993   ret->h = h;
 994   ret->fi = fi;
 995   ret->namespace = namespace;
 996   if (namespace != NULL)
 997     {
 998       namespace->rc++;
 999       GNUNET_assert (NULL != nid);
1000       ret->nid = GNUNET_strdup (nid);
1001       if (NULL != nuid)
1002         ret->nuid = GNUNET_strdup (nuid);
1003     }
1004   // FIXME: make upload persistent!
1005
1006   /* signal start */
1007   GNUNET_FS_file_information_inspect (ret->fi,
1008                                       &fip_signal_start,
1009                                       ret);
1010   ret->fi_pos = ret->fi;
1011
1012   // FIXME: calculate space needed for "fi"
1013   // and reserve as first task (then trigger
1014   // "do_upload" from that continuation)!
1015   ret->upload_task
1016     = GNUNET_SCHEDULER_add_with_priority (h->sched,
1017                                     GNUNET_SCHEDULER_PRIORITY_BACKGROUND,
1018                                     &do_upload,
1019                                     ret);
1020   return ret;
1021 }
1022
1023
1024 /**
1025  * Signal the FS's progress function that we are stopping
1026  * an upload.
1027  *
1028  * @param cls closure (of type "struct GNUNET_FS_PublishContext*")
1029  * @param fi the entry in the publish-structure
1030  * @param length length of the file or directory
1031  * @param meta metadata for the file or directory (can be modified)
1032  * @param uri pointer to the keywords that will be used for this entry (can be modified)
1033  * @param anonymity pointer to selected anonymity level (can be modified)
1034  * @param priority pointer to selected priority (can be modified)
1035  * @param expirationTime pointer to selected expiration time (can be modified)
1036  * @param client_info pointer to client context set upon creation (can be modified)
1037  * @return GNUNET_OK to continue (always)
1038  */
1039 static int
1040 fip_signal_stop(void *cls,
1041                 struct GNUNET_FS_FileInformation *fi,
1042                 uint64_t length,
1043                 struct GNUNET_CONTAINER_MetaData *meta,
1044                 struct GNUNET_FS_Uri **uri,
1045                 uint32_t *anonymity,
1046                 uint32_t *priority,
1047                 struct GNUNET_TIME_Absolute *expirationTime,
1048                 void **client_info)
1049 {
1050   struct GNUNET_FS_PublishContext*sc = cls;
1051   struct GNUNET_FS_ProgressInfo pi;
1052   uint64_t off;
1053
1054   off = (fi->chk_uri == NULL) ? 0 : length;
1055   pi.status = GNUNET_FS_STATUS_PUBLISH_STOPPED;
1056   make_publish_status (&pi, sc, fi, off);
1057   GNUNET_break (NULL ==
1058                 sc->h->upcb (sc->h->upcb_cls,
1059                              &pi));
1060   *client_info = NULL;
1061   return GNUNET_OK;
1062 }
1063
1064
1065 /**
1066  * Stop an upload.  Will abort incomplete uploads (but
1067  * not remove blocks that have already been publishd) or
1068  * simply clean up the state for completed uploads.
1069  * Must NOT be called from within the event callback!
1070  *
1071  * @param sc context for the upload to stop
1072  */
1073 void
1074 GNUNET_FS_publish_stop (struct GNUNET_FS_PublishContext *sc)
1075 {
1076   if (GNUNET_SCHEDULER_NO_TASK != sc->upload_task)
1077     GNUNET_SCHEDULER_cancel (sc->h->sched, sc->upload_task);
1078   else
1079     GNUNET_assert (sc->all_done == GNUNET_YES);
1080   // FIXME: remove from persistence DB (?) --- think more about
1081   //        shutdown / persistent-resume APIs!!!
1082   GNUNET_FS_file_information_inspect (sc->fi,
1083                                       &fip_signal_stop,
1084                                       sc);
1085   if (GNUNET_YES == sc->in_network_wait)
1086     {
1087       sc->in_network_wait = GNUNET_SYSERR;
1088       return;
1089     }
1090   publish_cleanup (sc);
1091 }
1092
1093
1094 /**
1095  * Context for the KSK publication.
1096  */
1097 struct PublishKskContext
1098 {
1099
1100   /**
1101    * Keywords to use.
1102    */
1103   struct GNUNET_FS_Uri *ksk_uri;
1104
1105   /**
1106    * Global FS context.
1107    */
1108   struct GNUNET_FS_Handle *h;
1109
1110   /**
1111    * The master block that we are sending
1112    * (in plaintext), has "mdsize+slen" more
1113    * bytes than the struct would suggest.
1114    */
1115   struct KBlock *kb;
1116
1117   /**
1118    * Buffer of the same size as "kb" for
1119    * the encrypted version.
1120    */
1121   struct KBlock *cpy;
1122
1123   /**
1124    * Handle to the datastore, NULL if we are just
1125    * simulating.
1126    */
1127   struct GNUNET_DATASTORE_Handle *dsh;
1128
1129   /**
1130    * Function to call once we're done.
1131    */
1132   GNUNET_FS_PublishContinuation cont;
1133
1134   /**
1135    * Closure for cont.
1136    */
1137   void *cont_cls;
1138
1139   /**
1140    * When should the KBlocks expire?
1141    */
1142   struct GNUNET_TIME_Absolute expirationTime;
1143
1144   /**
1145    * Size of the serialized metadata.
1146    */
1147   ssize_t mdsize;
1148
1149   /**
1150    * Size of the (CHK) URI as a string.
1151    */
1152   size_t slen;
1153
1154   /**
1155    * Keyword that we are currently processing.
1156    */
1157   unsigned int i;
1158
1159   /**
1160    * Anonymity level for the KBlocks.
1161    */
1162   uint32_t anonymity;
1163
1164   /**
1165    * Priority for the KBlocks.
1166    */
1167   uint32_t priority;
1168 };
1169
1170
1171 /**
1172  * Continuation of "GNUNET_FS_publish_ksk" that performs
1173  * the actual publishing operation (iterating over all
1174  * of the keywords).
1175  *
1176  * @param cls closure of type "struct PublishKskContext*"
1177  * @param tc unused
1178  */
1179 static void
1180 publish_ksk_cont (void *cls,
1181                   const struct GNUNET_SCHEDULER_TaskContext *tc);
1182
1183
1184 /**
1185  * Function called by the datastore API with
1186  * the result from the PUT request.
1187  *
1188  * @param cls closure of type "struct PublishKskContext*"
1189  * @param success GNUNET_OK on success
1190  * @param msg error message (or NULL)
1191  */
1192 static void
1193 kb_put_cont (void *cls,
1194              int success,
1195              const char *msg)
1196 {
1197   struct PublishKskContext *pkc = cls;
1198
1199   if (GNUNET_OK != success)
1200     {
1201       GNUNET_DATASTORE_disconnect (pkc->dsh, GNUNET_NO);
1202       GNUNET_free (pkc->cpy);
1203       GNUNET_free (pkc->kb);
1204       pkc->cont (pkc->cont_cls,
1205                  NULL,
1206                  msg);
1207       GNUNET_FS_uri_destroy (pkc->ksk_uri);
1208       GNUNET_free (pkc);
1209       return;
1210     }
1211   GNUNET_SCHEDULER_add_continuation (pkc->h->sched,
1212                                      &publish_ksk_cont,
1213                                      pkc,
1214                                      GNUNET_SCHEDULER_REASON_PREREQ_DONE);
1215 }
1216
1217
1218 /**
1219  * Continuation of "GNUNET_FS_publish_ksk" that performs the actual
1220  * publishing operation (iterating over all of the keywords).
1221  *
1222  * @param cls closure of type "struct PublishKskContext*"
1223  * @param tc unused
1224  */
1225 static void
1226 publish_ksk_cont (void *cls,
1227                   const struct GNUNET_SCHEDULER_TaskContext *tc)
1228 {
1229   struct PublishKskContext *pkc = cls;
1230   const char *keyword;
1231   GNUNET_HashCode key;
1232   GNUNET_HashCode query;
1233   struct GNUNET_CRYPTO_AesSessionKey skey;
1234   struct GNUNET_CRYPTO_AesInitializationVector iv;
1235   struct GNUNET_CRYPTO_RsaPrivateKey *pk;
1236
1237
1238   if ( (pkc->i == pkc->ksk_uri->data.ksk.keywordCount) ||
1239        (NULL == pkc->dsh) )
1240     {
1241       if (NULL != pkc->dsh)
1242         GNUNET_DATASTORE_disconnect (pkc->dsh, GNUNET_NO);
1243       GNUNET_free (pkc->cpy);
1244       GNUNET_free (pkc->kb);
1245       pkc->cont (pkc->cont_cls,
1246                  pkc->ksk_uri,
1247                  NULL);
1248       GNUNET_FS_uri_destroy (pkc->ksk_uri);
1249       GNUNET_free (pkc);
1250       return;
1251     }
1252   keyword = pkc->ksk_uri->data.ksk.keywords[pkc->i++];
1253   /* first character of keyword indicates if it is
1254      mandatory or not -- ignore for hashing */
1255   GNUNET_CRYPTO_hash (&keyword[1], strlen (&keyword[1]), &key);
1256   GNUNET_CRYPTO_hash_to_aes_key (&key, &skey, &iv);
1257   GNUNET_CRYPTO_aes_encrypt (&pkc->kb[1],
1258                              pkc->slen + pkc->mdsize,
1259                              &skey,
1260                              &iv,
1261                              &pkc->cpy[1]);
1262   pk = GNUNET_CRYPTO_rsa_key_create_from_hash (&key);
1263   GNUNET_CRYPTO_rsa_key_get_public (pk, &pkc->cpy->keyspace);
1264   GNUNET_CRYPTO_hash (&pkc->cpy->keyspace,
1265                       sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded),
1266                       &query);
1267   GNUNET_assert (GNUNET_OK ==
1268                  GNUNET_CRYPTO_rsa_sign (pk,
1269                                          &pkc->cpy->purpose,
1270                                          &pkc->cpy->signature));
1271   GNUNET_CRYPTO_rsa_key_free (pk);
1272   GNUNET_DATASTORE_put (pkc->dsh,
1273                         0,
1274                         &query,
1275                         pkc->mdsize +
1276                         sizeof (struct KBlock) +
1277                         pkc->slen,
1278                         pkc->cpy,
1279                         GNUNET_DATASTORE_BLOCKTYPE_KBLOCK,
1280                         pkc->priority,
1281                         pkc->anonymity,
1282                         pkc->expirationTime,
1283                         GNUNET_CONSTANTS_SERVICE_TIMEOUT,
1284                         &kb_put_cont,
1285                         pkc);
1286 }
1287
1288
1289 /**
1290  * Publish a CHK under various keywords on GNUnet.
1291  *
1292  * @param h handle to the file sharing subsystem
1293  * @param ksk_uri keywords to use
1294  * @param meta metadata to use
1295  * @param uri URI to refer to in the KBlock
1296  * @param expirationTime when the KBlock expires
1297  * @param anonymity anonymity level for the KBlock
1298  * @param priority priority for the KBlock
1299  * @param options publication options
1300  * @param cont continuation
1301  * @param cont_cls closure for cont
1302  */
1303 void
1304 GNUNET_FS_publish_ksk (struct GNUNET_FS_Handle *h,
1305                        struct GNUNET_FS_Uri *ksk_uri,
1306                        struct GNUNET_CONTAINER_MetaData *meta,
1307                        struct GNUNET_FS_Uri *uri,
1308                        struct GNUNET_TIME_Absolute expirationTime,
1309                        uint32_t anonymity,
1310                        uint32_t priority,
1311                        enum GNUNET_FS_PublishOptions options,
1312                        GNUNET_FS_PublishContinuation cont,
1313                        void *cont_cls)
1314 {
1315   struct PublishKskContext *pkc;
1316   char *uris;
1317   size_t size;
1318   char *kbe;
1319   char *sptr;
1320
1321   pkc = GNUNET_malloc (sizeof (struct PublishKskContext));
1322   pkc->h = h;
1323   pkc->expirationTime = expirationTime;
1324   pkc->anonymity = anonymity;
1325   pkc->priority = priority;
1326   pkc->cont = cont;
1327   pkc->cont_cls = cont_cls;
1328   if (0 == (options & GNUNET_FS_PUBLISH_OPTION_SIMULATE_ONLY))
1329     {
1330       pkc->dsh = GNUNET_DATASTORE_connect (h->cfg,
1331                                            h->sched);
1332       if (pkc->dsh == NULL)
1333         {
1334           cont (cont_cls, NULL, _("Could not connect to datastore."));
1335           GNUNET_free (pkc);
1336           return;
1337         }
1338     }
1339   if (meta == NULL)
1340     pkc->mdsize = 0;
1341   else
1342     pkc->mdsize = GNUNET_CONTAINER_meta_data_get_serialized_size (meta);
1343   GNUNET_assert (pkc->mdsize >= 0);
1344   uris = GNUNET_FS_uri_to_string (uri);
1345   pkc->slen = strlen (uris) + 1;
1346   size = pkc->mdsize + sizeof (struct KBlock) + pkc->slen;
1347   if (size > MAX_KBLOCK_SIZE)
1348     {
1349       size = MAX_KBLOCK_SIZE;
1350       pkc->mdsize = size - sizeof (struct KBlock) - pkc->slen;
1351     }
1352   pkc->kb = GNUNET_malloc (size);
1353   kbe = (char *) &pkc->kb[1];
1354   memcpy (kbe, uris, pkc->slen);
1355   GNUNET_free (uris);
1356   sptr = &kbe[pkc->slen];
1357   if (meta != NULL)
1358     pkc->mdsize = GNUNET_CONTAINER_meta_data_serialize (meta,
1359                                                         &sptr,
1360                                                         pkc->mdsize,
1361                                                         GNUNET_CONTAINER_META_DATA_SERIALIZE_PART);
1362   if (pkc->mdsize == -1)
1363     {
1364       GNUNET_break (0);
1365       GNUNET_free (pkc->kb);
1366       if (pkc->dsh != NULL)
1367         GNUNET_DATASTORE_disconnect (pkc->dsh, GNUNET_NO);
1368       cont (cont_cls, NULL, _("Internal error."));
1369       GNUNET_free (pkc);
1370       return;
1371     }
1372   size = sizeof (struct KBlock) + pkc->slen + pkc->mdsize;
1373
1374   pkc->cpy = GNUNET_malloc (size);
1375   pkc->cpy->purpose.size = htonl (sizeof (struct GNUNET_CRYPTO_RsaSignaturePurpose) +
1376                                   sizeof(struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded) +
1377                                   pkc->mdsize +
1378                                   pkc->slen);
1379   pkc->cpy->purpose.purpose = htonl(GNUNET_SIGNATURE_PURPOSE_FS_KBLOCK);
1380   pkc->ksk_uri = GNUNET_FS_uri_dup (ksk_uri);
1381   GNUNET_SCHEDULER_add_continuation (h->sched,
1382                                      &publish_ksk_cont,
1383                                      pkc,
1384                                      GNUNET_SCHEDULER_REASON_PREREQ_DONE);
1385 }
1386
1387
1388 /**
1389  * Context for the SKS publication.
1390  */
1391 struct PublishSksContext
1392 {
1393
1394   /**
1395    * Global FS context.
1396    */
1397   struct GNUNET_FS_Uri *uri;
1398
1399   /**
1400    * Handle to the datastore.
1401    */
1402   struct GNUNET_DATASTORE_Handle *dsh;
1403
1404   /**
1405    * Function to call once we're done.
1406    */
1407   GNUNET_FS_PublishContinuation cont;
1408
1409   /**
1410    * Closure for cont.
1411    */
1412   void *cont_cls;
1413
1414 };
1415
1416
1417 /**
1418  * Function called by the datastore API with
1419  * the result from the PUT (SBlock) request.
1420  *
1421  * @param cls closure of type "struct PublishSksContext*"
1422  * @param success GNUNET_OK on success
1423  * @param msg error message (or NULL)
1424  */
1425 static void
1426 sb_put_cont (void *cls,
1427              int success,
1428              const char *msg)
1429 {
1430   struct PublishSksContext *psc = cls;
1431
1432   if (NULL != psc->dsh)
1433     GNUNET_DATASTORE_disconnect (psc->dsh, GNUNET_NO);
1434   if (GNUNET_OK != success)
1435     psc->cont (psc->cont_cls,
1436                NULL,
1437                msg);
1438   else
1439     psc->cont (psc->cont_cls,
1440                psc->uri,
1441                NULL);
1442   GNUNET_FS_uri_destroy (psc->uri);
1443   GNUNET_free (psc);
1444 }
1445
1446
1447 /**
1448  * Publish an SBlock on GNUnet.
1449  *
1450  * @param h handle to the file sharing subsystem
1451  * @param namespace namespace to publish in
1452  * @param identifier identifier to use
1453  * @param update update identifier to use
1454  * @param meta metadata to use
1455  * @param uri URI to refer to in the SBlock
1456  * @param expirationTime when the SBlock expires
1457  * @param anonymity anonymity level for the SBlock
1458  * @param priority priority for the SBlock
1459  * @param options publication options
1460  * @param cont continuation
1461  * @param cont_cls closure for cont
1462  */
1463 void
1464 GNUNET_FS_publish_sks (struct GNUNET_FS_Handle *h,
1465                        struct GNUNET_FS_Namespace *namespace,
1466                        const char *identifier,
1467                        const char *update,
1468                        struct GNUNET_CONTAINER_MetaData *meta,
1469                        struct GNUNET_FS_Uri *uri,
1470                        struct GNUNET_TIME_Absolute expirationTime,
1471                        uint32_t anonymity,
1472                        uint32_t priority,
1473                        enum GNUNET_FS_PublishOptions options,
1474                        GNUNET_FS_PublishContinuation cont,
1475                        void *cont_cls)
1476 {
1477   struct PublishSksContext *psc;
1478   struct GNUNET_CRYPTO_AesSessionKey sk;
1479   struct GNUNET_CRYPTO_AesInitializationVector iv;
1480   struct GNUNET_FS_Uri *sks_uri;
1481   char *uris;
1482   size_t size;
1483   size_t slen;
1484   size_t nidlen;
1485   size_t idlen;
1486   ssize_t mdsize;
1487   struct SBlock *sb;
1488   struct SBlock *sb_enc;
1489   char *dest;
1490   GNUNET_HashCode key;           /* hash of thisId = key */
1491   GNUNET_HashCode id;          /* hash of hc = identifier */
1492
1493   uris = GNUNET_FS_uri_to_string (uri);
1494   slen = strlen (uris) + 1;
1495   idlen = strlen (identifier);
1496   if (update == NULL)
1497     update = "";
1498   nidlen = strlen (update) + 1;
1499   mdsize = GNUNET_CONTAINER_meta_data_get_serialized_size (meta);
1500
1501   size = sizeof (struct SBlock) + slen + nidlen + mdsize;
1502   if (size > MAX_SBLOCK_SIZE)
1503     {
1504       size = MAX_SBLOCK_SIZE;
1505       mdsize = size - (sizeof (struct SBlock) + slen + nidlen);
1506     }
1507   sb = GNUNET_malloc (sizeof (struct SBlock) + size);
1508   dest = (char *) &sb[1];
1509   memcpy (dest, update, nidlen);
1510   dest += nidlen;
1511   memcpy (dest, uris, slen);
1512   dest += slen;
1513   mdsize = GNUNET_CONTAINER_meta_data_serialize (meta,
1514                                                  &dest,
1515                                                  mdsize,
1516                                                  GNUNET_CONTAINER_META_DATA_SERIALIZE_PART);
1517   if (mdsize == -1)
1518     {
1519       GNUNET_break (0);
1520       GNUNET_free (uris);
1521       GNUNET_free (sb);
1522       cont (cont_cls,
1523             NULL,
1524             _("Internal error."));
1525       return;
1526     }
1527   size = sizeof (struct SBlock) + mdsize + slen + nidlen;
1528   sb_enc = GNUNET_malloc (sizeof (struct SBlock) + size);
1529   GNUNET_CRYPTO_hash (identifier, idlen, &key);
1530   GNUNET_CRYPTO_hash (&key, sizeof (GNUNET_HashCode), &id);
1531   sks_uri = GNUNET_malloc (sizeof (struct GNUNET_FS_Uri));
1532   sks_uri->type = sks;
1533   GNUNET_CRYPTO_rsa_key_get_public (namespace->key, &sb_enc->subspace);
1534   GNUNET_CRYPTO_hash (&sb_enc->subspace,
1535                       sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded),
1536                       &sks_uri->data.sks.namespace);
1537   sks_uri->data.sks.identifier = GNUNET_strdup (identifier);
1538   GNUNET_CRYPTO_hash_xor (&id,
1539                           &sks_uri->data.sks.namespace,
1540                           &sb_enc->identifier);
1541   GNUNET_CRYPTO_hash_to_aes_key (&key, &sk, &iv);
1542   GNUNET_CRYPTO_aes_encrypt (&sb[1],
1543                              size - sizeof (struct SBlock),
1544                              &sk,
1545                              &iv,
1546                              &sb_enc[1]);
1547   sb_enc->purpose.purpose = htonl (GNUNET_SIGNATURE_PURPOSE_FS_SBLOCK);
1548   sb_enc->purpose.size = htonl(slen + mdsize + nidlen
1549                                + sizeof(struct SBlock)
1550                                - sizeof(struct GNUNET_CRYPTO_RsaSignature));
1551   GNUNET_assert (GNUNET_OK ==
1552                  GNUNET_CRYPTO_rsa_sign (namespace->key,
1553                                          &sb_enc->purpose,
1554                                          &sb_enc->signature));
1555   psc = GNUNET_malloc (sizeof(struct PublishSksContext));
1556   psc->uri = sks_uri;
1557   psc->cont = cont;
1558   psc->cont_cls = cont_cls;
1559   if (0 != (options & GNUNET_FS_PUBLISH_OPTION_SIMULATE_ONLY))
1560     {
1561       GNUNET_free (sb_enc);
1562       GNUNET_free (sb);
1563       sb_put_cont (psc,
1564                    GNUNET_OK,
1565                    NULL);
1566       return;
1567     }
1568   psc->dsh = GNUNET_DATASTORE_connect (h->cfg, h->sched);
1569   if (NULL == psc->dsh)
1570     {
1571       GNUNET_free (sb_enc);
1572       GNUNET_free (sb);
1573       sb_put_cont (psc,
1574                    GNUNET_NO,
1575                    _("Failed to connect to datastore."));
1576       return;
1577     }
1578
1579   GNUNET_DATASTORE_put (psc->dsh,
1580                         0,
1581                         &sb->identifier,
1582                         size,
1583                         sb_enc,
1584                         GNUNET_DATASTORE_BLOCKTYPE_SBLOCK,
1585                         priority,
1586                         anonymity,
1587                         expirationTime,
1588                         GNUNET_CONSTANTS_SERVICE_TIMEOUT,
1589                         &sb_put_cont,
1590                         psc);
1591   GNUNET_free (sb);
1592   GNUNET_free (sb_enc);
1593 }
1594
1595
1596 /* end of fs_publish.c */