/**
- * Process a search result.
+ * Shutdown any existing connection to the FS
+ * service and try to establish a fresh one
+ * (and then re-transmit our search request).
*
- * @param sc our search context
- * @param type type of the result
- * @param expiration when it will expire
- * @param data the (encrypted) response
- * @param size size of @a data
+ * @param sc the search to reconnec
*/
static void
-process_result (struct GNUNET_FS_SearchContext *sc,
- enum GNUNET_BLOCK_Type type,
- struct GNUNET_TIME_Absolute expiration,
- const void *data,
- size_t size)
+try_reconnect (struct GNUNET_FS_SearchContext *sc);
+
+
+/**
+ * We check a result message from the service.
+ *
+ * @param cls closure
+ * @param msg result message received
+ */
+static int
+check_result (void *cls,
+ const struct ClientPutMessage *cm)
{
- if (GNUNET_TIME_absolute_get_duration (expiration).rel_value_us > 0)
+ /* payload of any variable size is OK */
+ return GNUNET_OK;
+}
+
+
+/**
+ * We process a search result from the service.
+ *
+ * @param cls closure
+ * @param msg result message received
+ */
+static void
+handle_result (void *cls,
+ const struct ClientPutMessage *cm)
+{
+ struct GNUNET_FS_SearchContext *sc = cls;
+ uint16_t msize = ntohs (cm->header.size) - sizeof (*cm);
+ enum GNUNET_BLOCK_Type type = ntohl (cm->type);
+
+ if (GNUNET_TIME_absolute_get_duration (GNUNET_TIME_absolute_ntoh (cm->expiration)).rel_value_us > 0)
{
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
"Result received has already expired.\n");
{
case GNUNET_BLOCK_TYPE_FS_UBLOCK:
if (GNUNET_FS_URI_SKS == sc->uri->type)
- process_sblock (sc, data, size);
+ process_sblock (sc,
+ (const struct UBlock *) &cm[1],
+ msize);
else
- process_kblock (sc, data, size);
+ process_kblock (sc,
+ (const struct UBlock *) &cm[1],
+ msize);
break;
case GNUNET_BLOCK_TYPE_ANY:
GNUNET_break (0);
break;
default:
GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Got result with unknown block type `%d', ignoring"), type);
+ _("Got result with unknown block type `%d', ignoring"),
+ type);
break;
}
}
-/**
- * Shutdown any existing connection to the FS
- * service and try to establish a fresh one
- * (and then re-transmit our search request).
- *
- * @param sc the search to reconnec
- */
-static void
-try_reconnect (struct GNUNET_FS_SearchContext *sc);
-
-
-/**
- * Type of a function to call when we receive a message
- * from the service.
- *
- * @param cls closure
- * @param msg message received, NULL on timeout or fatal error
- */
-static void
-receive_results (void *cls,
- const struct GNUNET_MessageHeader *msg)
-{
- struct GNUNET_FS_SearchContext *sc = cls;
- const struct ClientPutMessage *cm;
- uint16_t msize;
-
- if ((NULL == msg) || (ntohs (msg->type) != GNUNET_MESSAGE_TYPE_FS_PUT) ||
- (ntohs (msg->size) <= sizeof (struct ClientPutMessage)))
- {
- try_reconnect (sc);
- return;
- }
- msize = ntohs (msg->size);
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Receiving %u bytes of result from fs service\n",
- msize);
- cm = (const struct ClientPutMessage *) msg;
- process_result (sc, ntohl (cm->type),
- GNUNET_TIME_absolute_ntoh (cm->expiration), &cm[1],
- msize - sizeof (struct ClientPutMessage));
- /* continue receiving */
- GNUNET_CLIENT_receive (sc->client,
- &receive_results,
- sc,
- GNUNET_TIME_UNIT_FOREVER_REL);
-}
-
-
/**
* Schedule the transmission of the (next) search request
* to the service.
}
if (0 == mbc->put_cnt)
return GNUNET_SYSERR;
- mbc->sc->search_request_map_offset++;
mbc->xoff[--mbc->put_cnt] = *key;
return GNUNET_OK;
/**
- * We're ready to transmit the search request to the file-sharing
- * service. Do it. If the request is too large to fit into a single
- * message, transmit in increments.
+ * Schedule the transmission of the (next) search request
+ * to the service.
*
- * @param cls closure
- * @param size number of bytes available in @a buf
- * @param buf where the callee should write the message
- * @return number of bytes written to @a buf
+ * @param sc context for the search
*/
-static size_t
-transmit_search_request (void *cls,
- size_t size,
- void *buf)
+static void
+schedule_transmit_search_request (struct GNUNET_FS_SearchContext *sc)
{
- struct GNUNET_FS_SearchContext *sc = cls;
struct MessageBuilderContext mbc;
- size_t msize;
+ struct GNUNET_MQ_Envelope *env;
struct SearchMessage *sm;
struct GNUNET_CRYPTO_EcdsaPublicKey dpub;
unsigned int total_seen_results; /* total number of result hashes to send */
- unsigned int message_size_limit;
uint32_t options;
+ unsigned int left;
+ unsigned int todo;
+ unsigned int fit;
+ int first_call;
+ unsigned int search_request_map_offset;
+ unsigned int keyword_offset;
- if (NULL == buf)
- {
- try_reconnect (sc);
- return 0;
- }
+ memset (&mbc, 0, sizeof (mbc));
mbc.sc = sc;
- mbc.skip_cnt = sc->search_request_map_offset;
- sm = buf;
- sm->header.type = htons (GNUNET_MESSAGE_TYPE_FS_START_SEARCH);
- mbc.xoff = (struct GNUNET_HashCode *) &sm[1];
- options = SEARCH_MESSAGE_OPTION_NONE;
- if (0 != (sc->options & GNUNET_FS_SEARCH_OPTION_LOOPBACK_ONLY))
- options |= SEARCH_MESSAGE_OPTION_LOOPBACK_ONLY;
if (GNUNET_FS_uri_test_ksk (sc->uri))
{
- msize = sizeof (struct SearchMessage);
- GNUNET_assert (size >= msize);
- mbc.keyword_offset = sc->keyword_offset;
- /* calculate total number of known results (in put_cnt => total_seen_results) */
mbc.put_cnt = 0;
GNUNET_CONTAINER_multihashmap_iterate (sc->master_result_map,
- &find_result_set, &mbc);
+ &find_result_set,
+ &mbc);
total_seen_results = mbc.put_cnt;
- /* calculate how many results we can send in this message */
- message_size_limit = (size - msize) / sizeof (struct GNUNET_HashCode);
- mbc.put_cnt = GNUNET_MIN (message_size_limit,
- total_seen_results - mbc.skip_cnt);
- if (sc->search_request_map_offset < total_seen_results)
- GNUNET_assert (mbc.put_cnt > 0);
-
- /* now build message */
- msize += sizeof (struct GNUNET_HashCode) * mbc.put_cnt;
- sm->header.size = htons (msize);
- sm->type = htonl (GNUNET_BLOCK_TYPE_FS_UBLOCK);
- sm->anonymity_level = htonl (sc->anonymity);
- memset (&sm->target, 0, sizeof (struct GNUNET_PeerIdentity));
- sm->query = sc->requests[sc->keyword_offset].uquery;
- GNUNET_CONTAINER_multihashmap_iterate (sc->master_result_map,
- &build_result_set, &mbc);
- GNUNET_assert (0 == mbc.put_cnt);
- GNUNET_assert (total_seen_results >= sc->search_request_map_offset);
- if (total_seen_results != sc->search_request_map_offset)
- {
- /* more requesting to be done... */
- sm->options = htonl (options | SEARCH_MESSAGE_OPTION_CONTINUED);
- schedule_transmit_search_request (sc);
- return msize;
- }
- sm->options = htonl (options);
- sc->keyword_offset++;
- sc->search_request_map_offset = 0;
- if (sc->uri->data.ksk.keywordCount != sc->keyword_offset)
- {
- /* more requesting to be done... */
- schedule_transmit_search_request (sc);
- return msize;
- }
}
else
{
- GNUNET_assert (GNUNET_FS_uri_test_sks (sc->uri));
- msize = sizeof (struct SearchMessage);
- GNUNET_assert (size >= msize);
- sm->type = htonl (GNUNET_BLOCK_TYPE_FS_UBLOCK);
- sm->anonymity_level = htonl (sc->anonymity);
- memset (&sm->target, 0, sizeof (struct GNUNET_PeerIdentity));
- GNUNET_CRYPTO_ecdsa_public_key_derive (&sc->uri->data.sks.ns,
- sc->uri->data.sks.identifier,
- "fs-ublock",
- &dpub);
- GNUNET_CRYPTO_hash (&dpub,
- sizeof (dpub),
- &sm->query);
- message_size_limit = (size - msize) / sizeof (struct GNUNET_HashCode);
- total_seen_results = GNUNET_CONTAINER_multihashmap_size (sc->master_result_map);
- mbc.put_cnt = GNUNET_MIN (message_size_limit,
- total_seen_results - mbc.skip_cnt);
- mbc.keyword_offset = 0;
- if (sc->search_request_map_offset < total_seen_results)
- GNUNET_assert (mbc.put_cnt > 0);
- msize += sizeof (struct GNUNET_HashCode) * mbc.put_cnt;
- GNUNET_CONTAINER_multihashmap_iterate (sc->master_result_map,
- &build_result_set, &mbc);
- sm->header.size = htons (msize);
- GNUNET_assert (total_seen_results >= sc->search_request_map_offset);
- if (total_seen_results != sc->search_request_map_offset)
+ total_seen_results
+ = GNUNET_CONTAINER_multihashmap_size (sc->master_result_map);
+ }
+ search_request_map_offset = 0;
+ keyword_offset = 0;
+
+ first_call = GNUNET_YES;
+ while ( (0 != (left =
+ (total_seen_results - search_request_map_offset))) ||
+ (GNUNET_YES == first_call) )
+ {
+ first_call = GNUNET_NO;
+ options = SEARCH_MESSAGE_OPTION_NONE;
+ if (0 != (sc->options & GNUNET_FS_SEARCH_OPTION_LOOPBACK_ONLY))
+ options |= SEARCH_MESSAGE_OPTION_LOOPBACK_ONLY;
+
+ fit = (GNUNET_SERVER_MAX_MESSAGE_SIZE - 1 - sizeof (*sm)) / sizeof (struct GNUNET_HashCode);
+ todo = GNUNET_MIN (fit,
+ left);
+ env = GNUNET_MQ_msg_extra (sm,
+ sizeof (struct GNUNET_HashCode) * todo,
+ GNUNET_MESSAGE_TYPE_FS_START_SEARCH);
+ mbc.skip_cnt = search_request_map_offset;
+ mbc.xoff = (struct GNUNET_HashCode *) &sm[1];
+
+ if (GNUNET_FS_uri_test_ksk (sc->uri))
+ {
+ mbc.keyword_offset = keyword_offset;
+ /* calculate how many results we can send in this message */
+ mbc.put_cnt = todo;
+ /* now build message */
+ sm->type = htonl (GNUNET_BLOCK_TYPE_FS_UBLOCK);
+ sm->anonymity_level = htonl (sc->anonymity);
+ memset (&sm->target,
+ 0,
+ sizeof (struct GNUNET_PeerIdentity));
+ sm->query = sc->requests[keyword_offset].uquery;
+ GNUNET_CONTAINER_multihashmap_iterate (sc->master_result_map,
+ &build_result_set,
+ &mbc);
+ search_request_map_offset += todo;
+ GNUNET_assert (0 == mbc.put_cnt);
+ GNUNET_assert (total_seen_results >= search_request_map_offset);
+ if (total_seen_results != search_request_map_offset)
+ {
+ /* more requesting to be done... */
+ sm->options = htonl (options | SEARCH_MESSAGE_OPTION_CONTINUED);
+ }
+ else
+ {
+ sm->options = htonl (options);
+ keyword_offset++;
+ search_request_map_offset = 0;
+ if (sc->uri->data.ksk.keywordCount != keyword_offset)
+ {
+ /* more keywords => more requesting to be done... */
+ first_call = GNUNET_YES;
+ }
+ }
+ }
+ else
{
- /* more requesting to be done... */
- sm->options = htonl (options | SEARCH_MESSAGE_OPTION_CONTINUED);
- schedule_transmit_search_request (sc);
- return msize;
+ GNUNET_assert (GNUNET_FS_uri_test_sks (sc->uri));
+
+ sm->type = htonl (GNUNET_BLOCK_TYPE_FS_UBLOCK);
+ sm->anonymity_level = htonl (sc->anonymity);
+ memset (&sm->target,
+ 0,
+ sizeof (struct GNUNET_PeerIdentity));
+ GNUNET_CRYPTO_ecdsa_public_key_derive (&sc->uri->data.sks.ns,
+ sc->uri->data.sks.identifier,
+ "fs-ublock",
+ &dpub);
+ GNUNET_CRYPTO_hash (&dpub,
+ sizeof (dpub),
+ &sm->query);
+ mbc.put_cnt = todo;
+ mbc.keyword_offset = 0;
+ GNUNET_CONTAINER_multihashmap_iterate (sc->master_result_map,
+ &build_result_set,
+ &mbc);
+ GNUNET_assert (total_seen_results >= search_request_map_offset);
+ if (total_seen_results != search_request_map_offset)
+ {
+ /* more requesting to be done... */
+ sm->options = htonl (options | SEARCH_MESSAGE_OPTION_CONTINUED);
+ }
+ else
+ {
+ sm->options = htonl (options);
+ }
}
- sm->options = htonl (options);
+ GNUNET_MQ_send (sc->mq,
+ env);
}
- GNUNET_CLIENT_receive (sc->client,
- &receive_results, sc,
- GNUNET_TIME_UNIT_FOREVER_REL);
- return msize;
}
/**
- * Schedule the transmission of the (next) search request
- * to the service.
+ * Generic error handler, called with the appropriate error code and
+ * the same closure specified at the creation of the message queue.
+ * Not every message queue implementation supports an error handler.
*
- * @param sc context for the search
+ * @param cls closure with the `struct GNUNET_FS_SearchContext *`
+ * @param error error code
*/
static void
-schedule_transmit_search_request (struct GNUNET_FS_SearchContext *sc)
+search_mq_error_handler (void *cls,
+ enum GNUNET_MQ_Error error)
{
- size_t size;
- unsigned int left;
- unsigned int fit;
- unsigned int request;
-
- size = sizeof (struct SearchMessage);
- left =
- GNUNET_CONTAINER_multihashmap_size (sc->master_result_map) -
- sc->search_request_map_offset;
- fit = (GNUNET_SERVER_MAX_MESSAGE_SIZE - 1 - size) / sizeof (struct GNUNET_HashCode);
- request = GNUNET_MIN (fit, left);
- size += sizeof (struct GNUNET_HashCode) * request;
- GNUNET_CLIENT_notify_transmit_ready (sc->client, size,
- GNUNET_CONSTANTS_SERVICE_TIMEOUT,
- GNUNET_NO,
- &transmit_search_request, sc);
+ struct GNUNET_FS_SearchContext *sc = cls;
+
+ if (NULL != sc->mq)
+ {
+ GNUNET_MQ_destroy (sc->mq);
+ sc->mq = NULL;
+ }
+ try_reconnect (sc);
}
static void
do_reconnect (void *cls)
{
+ GNUNET_MQ_hd_var_size (result,
+ GNUNET_MESSAGE_TYPE_FS_PUT,
+ struct ClientPutMessage);
struct GNUNET_FS_SearchContext *sc = cls;
- struct GNUNET_CLIENT_Connection *client;
+ struct GNUNET_MQ_MessageHandler handlers[] = {
+ make_result_handler (sc),
+ GNUNET_MQ_handler_end ()
+ };
sc->task = NULL;
- client = GNUNET_CLIENT_connect ("fs", sc->h->cfg);
- if (NULL == client)
+ sc->mq = GNUNET_CLIENT_connecT (sc->h->cfg,
+ "fs",
+ handlers,
+ &search_mq_error_handler,
+ sc);
+ if (NULL == sc->mq)
{
try_reconnect (sc);
return;
}
- sc->client = client;
- sc->search_request_map_offset = 0;
- sc->keyword_offset = 0;
schedule_transmit_search_request (sc);
}
static void
try_reconnect (struct GNUNET_FS_SearchContext *sc)
{
- if (NULL != sc->client)
+ if (NULL != sc->mq)
{
- GNUNET_CLIENT_disconnect (sc->client);
- sc->client = NULL;
+ GNUNET_MQ_destroy (sc->mq);
+ sc->mq = NULL;
}
sc->reconnect_backoff = GNUNET_TIME_STD_BACKOFF (sc->reconnect_backoff);
sc->task =
struct GNUNET_CRYPTO_EcdsaPublicKey anon_pub;
struct SearchRequestEntry *sre;
- GNUNET_assert (NULL == sc->client);
+ GNUNET_assert (NULL == sc->mq);
if (GNUNET_FS_uri_test_ksk (sc->uri))
{
GNUNET_assert (0 != sc->uri->data.ksk.keywordCount);
&update_sre_result_maps,
sc);
}
- sc->client = GNUNET_CLIENT_connect ("fs", sc->h->cfg);
- if (NULL == sc->client)
+ GNUNET_assert (NULL == sc->task);
+ do_reconnect (sc);
+ if (NULL == sc->mq)
+ {
+ GNUNET_SCHEDULER_cancel (sc->task);
+ sc->task = NULL;
return GNUNET_SYSERR;
- sc->search_request_map_offset = 0;
- schedule_transmit_search_request (sc);
+ }
return GNUNET_OK;
}
GNUNET_SCHEDULER_cancel (sc->task);
sc->task = NULL;
}
- if (NULL != sc->client)
+ if (NULL != sc->mq)
{
- GNUNET_CLIENT_disconnect (sc->client);
- sc->client = NULL;
+ GNUNET_MQ_destroy (sc->mq);
+ sc->mq = NULL;
}
GNUNET_CONTAINER_multihashmap_destroy (sc->master_result_map);
if (NULL != sc->requests)
GNUNET_SCHEDULER_cancel (sc->task);
sc->task = NULL;
}
- if (NULL != sc->client)
- GNUNET_CLIENT_disconnect (sc->client);
- sc->client = NULL;
+ if (NULL != sc->mq)
+ {
+ GNUNET_MQ_destroy (sc->mq);
+ sc->mq = NULL;
+ }
GNUNET_FS_search_sync_ (sc);
GNUNET_CONTAINER_multihashmap_iterate (sc->master_result_map,
- &search_result_freeze_probes, sc);
+ &search_result_freeze_probes,
+ sc);
pi.status = GNUNET_FS_STATUS_SEARCH_PAUSED;
- sc->client_info = GNUNET_FS_search_make_status_ (&pi, sc->h, sc);
+ sc->client_info = GNUNET_FS_search_make_status_ (&pi,
+ sc->h,
+ sc);
}
{
struct GNUNET_FS_ProgressInfo pi;
- GNUNET_assert (NULL == sc->client);
+ GNUNET_assert (NULL == sc->mq);
GNUNET_assert (NULL == sc->task);
do_reconnect (sc);
GNUNET_FS_search_sync_ (sc);
sc->client_info = GNUNET_FS_search_make_status_ (&pi, sc->h, sc);
GNUNET_break (NULL == sc->client_info);
if (NULL != sc->task)
+ {
GNUNET_SCHEDULER_cancel (sc->task);
- if (NULL != sc->client)
- GNUNET_CLIENT_disconnect (sc->client);
+ sc->task = NULL;
+ }
+ if (NULL != sc->mq)
+ {
+ GNUNET_MQ_destroy (sc->mq);
+ sc->mq = NULL;
+ }
GNUNET_CONTAINER_multihashmap_iterate (sc->master_result_map,
&search_result_free, sc);
GNUNET_CONTAINER_multihashmap_destroy (sc->master_result_map);
projects / oweals / gnunet.git / commitdiff