X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Ffs%2Fgnunet-service-fs_push.c;h=270188ef4595b9e7bc6458f70df3286e3bb94a97;hb=1b49510e7736620d7e10a33326076e2d75c86943;hp=28406085b3fcb5bc0cb2ae760bab8de41f1d76ef;hpb=c79de770dedd58ca9138e802bd055c7657ebfc7a;p=oweals%2Fgnunet.git diff --git a/src/fs/gnunet-service-fs_push.c b/src/fs/gnunet-service-fs_push.c index 28406085b..270188ef4 100644 --- a/src/fs/gnunet-service-fs_push.c +++ b/src/fs/gnunet-service-fs_push.c @@ -1,6 +1,6 @@ /* This file is part of GNUnet. - (C) 2011 Christian Grothoff (and other contributing authors) + Copyright (C) 2011 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 @@ -14,8 +14,8 @@ You should have received a copy of the GNU General Public License along with GNUnet; see the file COPYING. If not, write to the - Free Software Foundation, Inc., 59 Temple Place - Suite 330, - Boston, MA 02111-1307, USA. + Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, + Boston, MA 02110-1301, USA. */ /** @@ -32,7 +32,18 @@ /** - * How long must content remain valid for us to consider it for migration? + * Maximum number of blocks we keep in memory for migration. + */ +#define MAX_MIGRATION_QUEUE 8 + +/** + * Blocks are at most migrated to this number of peers + * plus one, each time they are fetched from the database. + */ +#define MIGRATION_LIST_SIZE 2 + +/** + * How long must content remain valid for us to consider it for migration? * If content will expire too soon, there is clearly no point in pushing * it to other peers. This value gives the threshold for migration. Note * that if this value is increased, the migration testcase may need to be @@ -60,10 +71,10 @@ struct MigrationReadyBlock /** * Query for the block. */ - GNUNET_HashCode query; + struct GNUNET_HashCode query; /** - * When does this block expire? + * When does this block expire? */ struct GNUNET_TIME_Absolute expiration; @@ -110,7 +121,7 @@ struct MigrationReadyPeer * Handle to peer. */ struct GSF_ConnectedPeer *peer; - + /** * Handle for current transmission request, * or NULL for none. @@ -152,7 +163,7 @@ static struct GNUNET_DATASTORE_QueueEntry *mig_qe; /** * ID of task that collects blocks for migration. */ -static GNUNET_SCHEDULER_TaskIdentifier mig_task; +static struct GNUNET_SCHEDULER_Task * mig_task; /** * What is the maximum frequency at which we are allowed to @@ -170,6 +181,11 @@ static unsigned int mig_size; */ static int enabled; +/** + * Did we find anything in the datastore? + */ +static int value_found; + /** * Delete the given migration block. @@ -179,11 +195,8 @@ static int enabled; static void delete_migration_block (struct MigrationReadyBlock *mb) { - GNUNET_CONTAINER_DLL_remove (mig_head, - mig_tail, - mb); - GNUNET_PEER_decrement_rcs (mb->target_list, - MIGRATION_LIST_SIZE); + GNUNET_CONTAINER_DLL_remove (mig_head, mig_tail, mb); + GNUNET_PEER_decrement_rcs (mb->target_list, MIGRATION_LIST_SIZE); mig_size--; GNUNET_free (mb); } @@ -191,24 +204,23 @@ delete_migration_block (struct MigrationReadyBlock *mb) /** * Find content for migration to this peer. - */ + */ static void find_content (struct MigrationReadyPeer *mrp); /** - * Transmit the message currently scheduled for - * transmission. + * Transmit the message currently scheduled for transmission. * - * @param cls the 'struct MigrationReadyPeer' - * @param buf_size number of bytes available in buf + * @param cls the `struct MigrationReadyPeer` + * @param buf_size number of bytes available in @a buf * @param buf where to copy the message, NULL on error (peer disconnect) - * @return number of bytes copied to 'buf', can be 0 (without indicating an error) + * @return number of bytes copied to @a buf, can be 0 (without indicating an error) */ static size_t transmit_message (void *cls, - size_t buf_size, - void *buf) + size_t buf_size, + void *buf) { struct MigrationReadyPeer *peer = cls; struct PutMessage *msg; @@ -217,15 +229,21 @@ transmit_message (void *cls, peer->th = NULL; msg = peer->msg; peer->msg = NULL; - if (buf == NULL) - { - GNUNET_free (msg); - return 0; - } + if (NULL == buf) + { + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Failed to migrate content to another peer (disconnect)\n"); + GNUNET_free (msg); + return 0; + } msize = ntohs (msg->header.size); GNUNET_assert (msize <= buf_size); memcpy (buf, msg, msize); GNUNET_free (msg); + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Pushing %u bytes to %s\n", + msize, + GNUNET_i2s (GSF_connected_peer_get_identity2_(peer->peer))); find_content (peer); return msize; } @@ -236,15 +254,15 @@ transmit_message (void *cls, * * @param peer target peer * @param block the block - * @return GNUNET_YES if the block was deleted (!) + * @return #GNUNET_YES if the block was deleted (!) */ static int transmit_content (struct MigrationReadyPeer *peer, - struct MigrationReadyBlock *block) + struct MigrationReadyBlock *block) { size_t msize; struct PutMessage *msg; - unsigned int i; + unsigned int i; struct GSF_PeerPerformanceData *ppd; int ret; @@ -252,38 +270,39 @@ transmit_content (struct MigrationReadyPeer *peer, GNUNET_assert (NULL == peer->th); msize = sizeof (struct PutMessage) + block->size; msg = GNUNET_malloc (msize); - msg->header.type = htons (42); + msg->header.type = htons (GNUNET_MESSAGE_TYPE_FS_PUT); msg->header.size = htons (msize); - - memcpy (&msg[1], - &block[1], - block->size); + msg->type = htonl (block->type); + msg->expiration = GNUNET_TIME_absolute_hton (block->expiration); + memcpy (&msg[1], &block[1], block->size); peer->msg = msg; - for (i=0;itarget_list[i] == 0) { - if (block->target_list[i] == 0) - { - block->target_list[i] = ppd->pid; - GNUNET_PEER_change_rc (block->target_list[i], 1); - break; - } + block->target_list[i] = ppd->pid; + GNUNET_PEER_change_rc (block->target_list[i], 1); + break; } + } if (MIGRATION_LIST_SIZE == i) - { - delete_migration_block (block); - ret = GNUNET_YES; - } + { + delete_migration_block (block); + ret = GNUNET_YES; + } else - { - ret = GNUNET_NO; - } + { + ret = GNUNET_NO; + } + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Asking for transmission of %u bytes to %s for migration\n", + msize, + GNUNET_i2s (GSF_connected_peer_get_identity2_(peer->peer))); peer->th = GSF_peer_transmit_ (peer->peer, - GNUNET_NO, - 0 /* priority */, - GNUNET_TIME_UNIT_FOREVER_REL, - msize, - &transmit_message, - peer); + GNUNET_NO, 0 /* priority */ , + GNUNET_TIME_UNIT_FOREVER_REL, + msize, + &transmit_message, peer); return ret; } @@ -300,7 +319,7 @@ count_targets (struct MigrationReadyBlock *block) { unsigned int i; - for (i=0;itarget_list[i] == 0) return i; return i; @@ -309,7 +328,7 @@ count_targets (struct MigrationReadyBlock *block) /** * Check if sending this block to this peer would - * be a good idea. + * be a good idea. * * @param peer target peer * @param block the block @@ -317,21 +336,22 @@ count_targets (struct MigrationReadyBlock *block) */ static long score_content (struct MigrationReadyPeer *peer, - struct MigrationReadyBlock *block) + struct MigrationReadyBlock *block) { unsigned int i; struct GSF_PeerPerformanceData *ppd; struct GNUNET_PeerIdentity id; + struct GNUNET_HashCode hc; uint32_t dist; ppd = GSF_get_peer_performance_data_ (peer->peer); - for (i=0;itarget_list[i] == ppd->pid) return -1; - GNUNET_PEER_resolve (ppd->pid, - &id); - dist = GNUNET_CRYPTO_hash_distance_u32 (&block->query, - &id.hashPubKey); + GNUNET_assert (0 != ppd->pid); + GNUNET_PEER_resolve (ppd->pid, &id); + GNUNET_CRYPTO_hash (&id, sizeof (struct GNUNET_PeerIdentity), &hc); + dist = GNUNET_CRYPTO_hash_distance_u32 (&block->query, &hc); /* closer distance, higher score: */ return UINT32_MAX - dist; } @@ -349,7 +369,7 @@ consider_gathering (void); * Find content for migration to this peer. * * @param mrp peer to find content for - */ + */ static void find_content (struct MigrationReadyPeer *mrp) { @@ -363,38 +383,46 @@ find_content (struct MigrationReadyPeer *mrp) best_score = -1; pos = mig_head; while (NULL != pos) + { + score = score_content (mrp, pos); + if (score > best_score) { - score = score_content (mrp, pos); - if (score > best_score) - { - best_score = score; - best = pos; - } - pos = pos->next; + best_score = score; + best = pos; } - if ( (NULL == best) && - (mig_size >= MAX_MIGRATION_QUEUE) ) + pos = pos->next; + } + if (NULL == best) + { + if (mig_size < MAX_MIGRATION_QUEUE) { - /* failed to find migration target AND - queue is full, purge most-forwarded - block from queue to make room for more */ - score = 0; - pos = mig_head; - while (NULL != pos) - { - score = count_targets (pos); - if (score >= best_score) - { - best_score = score; - best = pos; - } - pos = pos->next; - } - GNUNET_assert (NULL != best); - delete_migration_block (best); - consider_gathering (); - return; + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "No content found for pushing, waiting for queue to fill\n"); + return; /* will fill up eventually... */ } + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "No suitable content found, purging content from full queue\n"); + /* failed to find migration target AND + * queue is full, purge most-forwarded + * block from queue to make room for more */ + pos = mig_head; + while (NULL != pos) + { + score = count_targets (pos); + if (score >= best_score) + { + best_score = score; + best = pos; + } + pos = pos->next; + } + GNUNET_assert (NULL != best); + delete_migration_block (best); + consider_gathering (); + return; + } + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Preparing to push best content to peer\n"); transmit_content (mrp, best); } @@ -402,13 +430,13 @@ find_content (struct MigrationReadyPeer *mrp) /** * Task that is run periodically to obtain blocks for content * migration - * + * * @param cls unused * @param tc scheduler context (also unused) */ static void gather_migration_blocks (void *cls, - const struct GNUNET_SCHEDULER_TaskContext *tc); + const struct GNUNET_SCHEDULER_TaskContext *tc); /** @@ -420,23 +448,29 @@ consider_gathering () { struct GNUNET_TIME_Relative delay; - if (GSF_dsh == NULL) + if (NULL == GSF_dsh) return; - if (mig_qe != NULL) + if (NULL != mig_qe) return; - if (mig_task != GNUNET_SCHEDULER_NO_TASK) + if (NULL != mig_task) return; - if (mig_size >= MAX_MIGRATION_QUEUE) + if (mig_size >= MAX_MIGRATION_QUEUE) return; - delay = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, - mig_size); - delay = GNUNET_TIME_relative_divide (delay, - MAX_MIGRATION_QUEUE); - delay = GNUNET_TIME_relative_max (delay, - min_migration_delay); - mig_task = GNUNET_SCHEDULER_add_delayed (delay, - &gather_migration_blocks, - NULL); + delay = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, mig_size); + delay = GNUNET_TIME_relative_divide (delay, MAX_MIGRATION_QUEUE); + delay = GNUNET_TIME_relative_max (delay, min_migration_delay); + if (GNUNET_NO == value_found) + { + /* wait at least 5s if the datastore is empty */ + delay = GNUNET_TIME_relative_max (delay, + GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, + 5)); + } + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Scheduling gathering task (queue size: %u)\n", + mig_size); + mig_task = + GNUNET_SCHEDULER_add_delayed (delay, &gather_migration_blocks, NULL); } @@ -456,50 +490,54 @@ consider_gathering () */ static void process_migration_content (void *cls, - const GNUNET_HashCode * key, - size_t size, - const void *data, - enum GNUNET_BLOCK_Type type, - uint32_t priority, - uint32_t anonymity, - struct GNUNET_TIME_Absolute - expiration, uint64_t uid) + const struct GNUNET_HashCode *key, + size_t size, + const void *data, + enum GNUNET_BLOCK_Type type, + uint32_t priority, + uint32_t anonymity, + struct GNUNET_TIME_Absolute expiration, + uint64_t uid) { struct MigrationReadyBlock *mb; struct MigrationReadyPeer *pos; - - if (key == NULL) - { - mig_qe = NULL; - consider_gathering (); - return; - } - if (GNUNET_TIME_absolute_get_remaining (expiration).rel_value < - MIN_MIGRATION_CONTENT_LIFETIME.rel_value) - { - /* content will expire soon, don't bother */ - GNUNET_DATASTORE_get_next (GSF_dsh, GNUNET_YES); - return; - } + + mig_qe = NULL; + if (NULL == key) + { + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "No content found for migration...\n"); + consider_gathering (); + return; + } + value_found = GNUNET_YES; + if (GNUNET_TIME_absolute_get_remaining (expiration).rel_value_us < + MIN_MIGRATION_CONTENT_LIFETIME.rel_value_us) + { + /* content will expire soon, don't bother */ + consider_gathering (); + return; + } if (type == GNUNET_BLOCK_TYPE_FS_ONDEMAND) - { - if (GNUNET_OK != - GNUNET_FS_handle_on_demand_block (key, size, data, - type, priority, anonymity, - expiration, uid, - &process_migration_content, - NULL)) - { - GNUNET_DATASTORE_get_next (GSF_dsh, GNUNET_YES); - } - return; - } -#if DEBUG_FS + { + if (GNUNET_OK != + GNUNET_FS_handle_on_demand_block (key, + size, + data, + type, + priority, + anonymity, + expiration, + uid, + &process_migration_content, NULL)) + consider_gathering (); + return; + } GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, - "Retrieved block `%s' of type %u for migration\n", - GNUNET_h2s (key), - type); -#endif + "Retrieved block `%s' of type %u for migration (queue size: %u/%u)\n", + GNUNET_h2s (key), + type, mig_size + 1, + MAX_MIGRATION_QUEUE); mb = GNUNET_malloc (sizeof (struct MigrationReadyBlock) + size); mb->query = *key; mb->expiration = expiration; @@ -507,44 +545,51 @@ process_migration_content (void *cls, mb->type = type; memcpy (&mb[1], data, size); GNUNET_CONTAINER_DLL_insert_after (mig_head, - mig_tail, - mig_tail, - mb); + mig_tail, + mig_tail, + mb); mig_size++; - pos = peer_head; - while (pos != NULL) + for (pos = peer_head; NULL != pos; pos = pos->next) + { + if (NULL == pos->th) { - if (NULL == pos->th) - { - if (GNUNET_YES == transmit_content (pos, mb)) - break; /* 'mb' was freed! */ - } - pos = pos->next; + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Preparing to push best content to peer %s\n", + GNUNET_i2s (GSF_connected_peer_get_identity2_(pos->peer))); + if (GNUNET_YES == transmit_content (pos, mb)) + break; /* 'mb' was freed! */ } - GNUNET_DATASTORE_get_next (GSF_dsh, GNUNET_YES); + } + consider_gathering (); } /** * Task that is run periodically to obtain blocks for content * migration - * + * * @param cls unused * @param tc scheduler context (also unused) */ static void gather_migration_blocks (void *cls, - const struct GNUNET_SCHEDULER_TaskContext *tc) + const struct GNUNET_SCHEDULER_TaskContext *tc) { - mig_task = GNUNET_SCHEDULER_NO_TASK; - if (GSF_dsh != NULL) - { - mig_qe = GNUNET_DATASTORE_get_random (GSF_dsh, - 0, UINT_MAX, - GNUNET_TIME_UNIT_FOREVER_REL, - &process_migration_content, NULL); - GNUNET_assert (mig_qe != NULL); - } + mig_task = NULL; + if (mig_size >= MAX_MIGRATION_QUEUE) + return; + if (NULL == GSF_dsh) + return; + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Asking datastore for content for replication (queue size: %u)\n", + mig_size); + value_found = GNUNET_NO; + mig_qe = + GNUNET_DATASTORE_get_for_replication (GSF_dsh, 0, UINT_MAX, + GNUNET_TIME_UNIT_FOREVER_REL, + &process_migration_content, NULL); + if (NULL == mig_qe) + consider_gathering (); } @@ -561,12 +606,26 @@ GSF_push_start_ (struct GSF_ConnectedPeer *peer) if (GNUNET_YES != enabled) return; - mrp = GNUNET_malloc (sizeof (struct MigrationReadyPeer)); + for (mrp = peer_head; NULL != mrp; mrp = mrp->next) + if (mrp->peer == peer) + break; + if (NULL != mrp) + { + /* same peer added twice, must not happen */ + GNUNET_break (0); + return; + } + + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Adding peer %s to list for pushing\n", + GNUNET_i2s (GSF_connected_peer_get_identity2_(peer))); + + mrp = GNUNET_new (struct MigrationReadyPeer); mrp->peer = peer; find_content (mrp); - GNUNET_CONTAINER_DLL_insert (peer_head, - peer_tail, - mrp); + GNUNET_CONTAINER_DLL_insert (peer_head, + peer_tail, + mrp); } @@ -581,21 +640,25 @@ GSF_push_stop_ (struct GSF_ConnectedPeer *peer) { struct MigrationReadyPeer *pos; - pos = peer_head; - while (pos != NULL) - { - if (pos->peer == peer) - { - GNUNET_CONTAINER_DLL_remove (peer_head, - peer_tail, - pos); - if (NULL != pos->th) - GSF_peer_transmit_cancel_ (pos->th); - GNUNET_free (pos); - return; - } - pos = pos->next; - } + for (pos = peer_head; NULL != pos; pos = pos->next) + if (pos->peer == peer) + break; + if (NULL == pos) + return; + GNUNET_CONTAINER_DLL_remove (peer_head, + peer_tail, + pos); + if (NULL != pos->th) + { + GSF_peer_transmit_cancel_ (pos->th); + pos->th = NULL; + } + if (NULL != pos->msg) + { + GNUNET_free (pos->msg); + pos->msg = NULL; + } + GNUNET_free (pos); } @@ -605,24 +668,20 @@ GSF_push_stop_ (struct GSF_ConnectedPeer *peer) void GSF_push_init_ () { - enabled = GNUNET_CONFIGURATION_get_value_yesno (GSF_cfg, - "FS", - "CONTENT_PUSHING"); + enabled = + GNUNET_CONFIGURATION_get_value_yesno (GSF_cfg, "FS", "CONTENT_PUSHING"); if (GNUNET_YES != enabled) return; - - if (GNUNET_OK != - GNUNET_CONFIGURATION_get_value_time (GSF_cfg, - "fs", - "MIN_MIGRATION_DELAY", - &min_migration_delay)) - { - GNUNET_log (GNUNET_ERROR_TYPE_WARNING, - _("Invalid value specified for option `%s' in section `%s', content pushing disabled\n"), - "MIN_MIGRATION_DELAY", - "fs"); - return; - } + + if (GNUNET_OK != + GNUNET_CONFIGURATION_get_value_time (GSF_cfg, "fs", "MIN_MIGRATION_DELAY", + &min_migration_delay)) + { + GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_WARNING, + "fs", "MIN_MIGRATION_DELAY", + _("time required, content pushing disabled")); + return; + } consider_gathering (); } @@ -633,16 +692,16 @@ GSF_push_init_ () void GSF_push_done_ () { - if (GNUNET_SCHEDULER_NO_TASK != mig_task) - { - GNUNET_SCHEDULER_cancel (mig_task); - mig_task = GNUNET_SCHEDULER_NO_TASK; - } + if (NULL != mig_task) + { + GNUNET_SCHEDULER_cancel (mig_task); + mig_task = NULL; + } if (NULL != mig_qe) - { - GNUNET_DATASTORE_cancel (mig_qe); - mig_qe = NULL; - } + { + GNUNET_DATASTORE_cancel (mig_qe); + mig_qe = NULL; + } while (NULL != mig_head) delete_migration_block (mig_head); GNUNET_assert (0 == mig_size);