adding listing of indexed files

1 files changed, 209 insertions, 2 deletions

diff --git a/src/fs/fs_unindex.c b/src/fs/fs_unindex.c
index 8fb13a644..c32a10cbf 100644
--- a/src/fs/fs_unindex.c
+++ b/src/fs/fs_unindex.c
@@ -1,6 +1,6 @@
 /*
      This file is part of GNUnet.
-     (C) 2003, 2004, 2006 Christian Grothoff (and other contributing authors)
+     (C) 2003, 2004, 2006, 2009 Christian Grothoff (and other contributing authors)
 
      GNUnet is free software; you can redistribute it and/or modify
      it under the terms of the GNU General Public License as published
@@ -29,21 +29,228 @@
  */
 
 #include "platform.h"
+#include "gnunet_constants.h"
 #include "gnunet_fs_service.h"
+#include "gnunet_protocols.h"
 #include "fs.h"
 
+
+/**
+ * Context for "GNUNET_FS_get_indexed_files".
+ */
+struct GetIndexedContext
+{
+  /**
+   * Handle to global FS context.
+   */
+  struct GNUNET_FS_Handle *h;
+
+  /**
+   * Connection to the FS service.
+   */
+  struct GNUNET_CLIENT_Connection *client;
+
+  /**
+   * Function to call for each indexed file.
+   */
+  GNUNET_FS_IndexedFileProcessor iterator;
+
+  /**
+   * Closure for iterator.
+   */
+  void *iterator_cls;
+
+  /**
+   * Continuation to trigger at the end.
+   */
+  GNUNET_SCHEDULER_Task cont;
+
+  /**
+   * Closure for cont.
+   */
+  void *cont_cls;
+};
+
+
+/**
+ * Function called on each response from the FS
+ * service with information about indexed files.
+ *
+ * @param cls closure (of type "struct GetIndexedContext*")
+ * @param msg message with indexing information
+ */
+static void
+handle_index_info (void *cls,
+                   const struct GNUNET_MessageHeader *msg)
+{
+  struct GetIndexedContext *gic = cls;
+  const struct IndexInfoMessage *iim;
+  uint16_t msize;
+  const char *filename;
+
+  if (NULL == msg)
+    {
+      GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+                  _("Failed to receive response for `%s' request from `%s' service.\n"),
+                  "GET_INDEXED",
+                  "fs");
+      GNUNET_SCHEDULER_add_continuation (gic->h->sched,
+                                         GNUNET_NO,
+                                         gic->cont,
+                                         gic->cont_cls,
+                                         GNUNET_SCHEDULER_REASON_TIMEOUT);
+      GNUNET_CLIENT_disconnect (gic->client);
+      GNUNET_free (gic);
+      return;
+    }
+  if (ntohs (msg->type) == GNUNET_MESSAGE_TYPE_FS_INDEX_LIST_END)
+    {
+      /* normal end-of-list */
+      GNUNET_SCHEDULER_add_continuation (gic->h->sched,
+                                         GNUNET_NO,
+                                         gic->cont,
+                                         gic->cont_cls,
+                                         GNUNET_SCHEDULER_REASON_PREREQ_DONE);
+      GNUNET_CLIENT_disconnect (gic->client);
+      GNUNET_free (gic);
+      return;
+    }
+  msize = ntohs (msg->size);
+  iim = (const struct IndexInfoMessage*) msg;
+  filename = (const char*) &iim[1];
+  if ( (ntohs (msg->type) != GNUNET_MESSAGE_TYPE_FS_INDEX_LIST_ENTRY) ||
+       (msize <= sizeof (struct IndexInfoMessage)) ||
+       (filename[msize-sizeof (struct IndexInfoMessage) -1] != '\0') )
+    {
+      /* bogus reply */
+      GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+                  _("Failed to receive valid response for `%s' request from `%s' service.\n"),
+                  "GET_INDEXED",
+                  "fs");
+      GNUNET_SCHEDULER_add_continuation (gic->h->sched,
+                                         GNUNET_NO,
+                                         gic->cont,
+                                         gic->cont_cls,
+                                         GNUNET_SCHEDULER_REASON_TIMEOUT);
+      GNUNET_CLIENT_disconnect (gic->client);
+      GNUNET_free (gic);
+      return;
+    }
+  if (GNUNET_OK !=
+      gic->iterator (gic->iterator_cls,
+                     filename,
+                     &iim->file_id))
+    {
+      GNUNET_SCHEDULER_add_continuation (gic->h->sched,
+                                         GNUNET_NO,
+                                         gic->cont,
+                                         gic->cont_cls,
+                                         GNUNET_SCHEDULER_REASON_PREREQ_DONE);
+      GNUNET_CLIENT_disconnect (gic->client);
+      GNUNET_free (gic);
+      return;
+    }
+  /* get more */
+  GNUNET_CLIENT_receive (gic->client,
+                         &handle_index_info,
+                         gic,
+                         GNUNET_CONSTANTS_SERVICE_TIMEOUT);  
+}
+
+
+/**
+ * Transmit the request to get a list of all 
+ * indexed files to the "FS" service.
+ *
+ * @param cls closure (of type "struct GetIndexedContext*")
+ * @param size number of bytes availabe in buf
+ * @param buf where to write the message, NULL on error
+ * @return number of bytes written to buf
+ */
+static size_t
+transmit_get_indexed (void *cls,
+                      size_t size,
+                      void *buf)
+{
+  struct GetIndexedContext *gic = cls;
+  struct GNUNET_MessageHeader *hdr;
+
+  if (NULL == buf)
+    {
+      GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+                  _("Failed to transmit `%s' request to `%s' service.\n"),
+                  "GET_INDEXED",
+                  "fs");
+      GNUNET_SCHEDULER_add_continuation (gic->h->sched,
+                                         GNUNET_NO,
+                                         gic->cont,
+                                         gic->cont_cls,
+                                         GNUNET_SCHEDULER_REASON_TIMEOUT);
+      GNUNET_CLIENT_disconnect (gic->client);
+      GNUNET_free (gic);
+      return 0;
+    }
+  GNUNET_assert (size >= sizeof (struct GNUNET_MessageHeader));
+  hdr = buf;
+  hdr->size = htons (sizeof (struct GNUNET_MessageHeader));
+  hdr->type = htons (GNUNET_MESSAGE_TYPE_FS_INDEX_LIST_GET);
+  GNUNET_CLIENT_receive (gic->client,
+                         &handle_index_info,
+                         gic,
+                         GNUNET_CONSTANTS_SERVICE_TIMEOUT);
+  return sizeof (struct GNUNET_MessageHeader);
+}
+
+
 /**
  * Iterate over all indexed files.
  *
  * @param h handle to the file sharing subsystem
  * @param iterator function to call on each indexed file
  * @param iterator_cls closure for iterator
+ * @param cont continuation to call when done;
+ *             reason should be "TIMEOUT" (on
+ *             error) or  "PREREQ_DONE" (on success)
+ * @param cont_cls closure for cont
  */
 void 
 GNUNET_FS_get_indexed_files (struct GNUNET_FS_Handle *h,
                              GNUNET_FS_IndexedFileProcessor iterator,
-                             void *iterator_cls)
+                             void *iterator_cls,
+                             GNUNET_SCHEDULER_Task cont,
+                             void *cont_cls)
 {
+  struct GNUNET_CLIENT_Connection *client;
+  struct GetIndexedContext *gic;
+
+  client = GNUNET_CLIENT_connect (h->sched,
+                                  "fs",
+                                  h->cfg);
+  if (NULL == client)
+    {
+      GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+                  _("Failed to not connect to `%s' service.\n"),
+                  "fs");
+      GNUNET_SCHEDULER_add_continuation (h->sched,
+                                         GNUNET_NO,
+                                         cont,
+                                         cont_cls,
+                                         GNUNET_SCHEDULER_REASON_TIMEOUT);
+      return;
+    }
+
+  gic = GNUNET_malloc (sizeof (struct GetIndexedContext));
+  gic->h = h;
+  gic->client = client;
+  gic->iterator = iterator;
+  gic->iterator_cls = iterator_cls;
+  gic->cont = cont;
+  gic->cont_cls = cont_cls;
+  GNUNET_CLIENT_notify_transmit_ready (client,
+                                       sizeof (struct GNUNET_MessageHeader),
+                                       GNUNET_CONSTANTS_SERVICE_TIMEOUT,
+                                       &transmit_get_indexed,
+                                       gic);
 }