/*
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
*/
#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);
}