/*
This file is part of GNUnet.
- (C) 2012 Christian Grothoff
+ Copyright (C) 2012 Christian Grothoff
GNUnet is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published
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.
*/
/**
* @file fs/gnunet-daemon-fsprofiler.c
* @brief daemon that publishes and downloads (random) files
* @author Christian Grothoff
+ *
+ * TODO:
+ * - how to signal driver that we're done?
*/
#include "platform.h"
#include "gnunet_fs_service.h"
/**
* Task to run the operation.
*/
- GNUNET_SCHEDULER_TaskIdentifier task;
+ struct GNUNET_SCHEDULER_Task * task;
+
+ /**
+ * Secondary task to run the operation.
+ */
+ struct GNUNET_SCHEDULER_Task * stask;
/**
* X-value.
/**
- * Parse a pattern string and store the corresponding
+ * Parse a pattern string and store the corresponding
* 'struct Pattern' in the given head/tail.
*
* @param head where to store the head
"(%llu,%llu,%llu)",
&x, &y, &t))
{
- p = GNUNET_malloc (sizeof (struct Pattern));
+ p = GNUNET_new (struct Pattern);
p->x = x;
p->y = y;
- p->delay.rel_value = (uint64_t) t;
+ p->delay.rel_value_us = (uint64_t) t;
GNUNET_CONTAINER_DLL_insert (*head, *tail, p);
pattern = strstr (pattern, ")");
GNUNET_assert (NULL != pattern);
}
+/**
+ * Create a KSK URI from a number.
+ *
+ * @param kval the number
+ * @return corresponding KSK URI
+ */
+static struct GNUNET_FS_Uri *
+make_keywords (uint64_t kval)
+{
+ char kw[128];
+
+ GNUNET_snprintf (kw, sizeof (kw),
+ "%llu", (unsigned long long) kval);
+ return GNUNET_FS_uri_ksk_create (kw, NULL);
+}
+
+
/**
* Create a file of the given length with a deterministic amount
* of data to be published under keyword 'kval'.
struct GNUNET_FS_BlockOptions bo;
char *data;
struct GNUNET_FS_Uri *keywords;
- char kw[128];
unsigned long long i;
uint64_t xor;
return NULL;
/* initialize data with 'unique' data only depending on 'kval' and 'size',
making sure that blocks do not repeat */
- for (i=0;i<length; i+=8)
+ for (i=0;i<length; i+=8)
{
xor = length ^ kval ^ (uint64_t) (i / 32 / 1024);
- memcpy (&data[i], &xor, GNUNET_MIN (length - i, sizeof (uint64_t)));
+ GNUNET_memcpy (&data[i], &xor, GNUNET_MIN (length - i, sizeof (uint64_t)));
}
bo.expiration_time = GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_DAYS);
bo.anonymity_level = (uint32_t) anonymity_level;
bo.content_priority = 128;
bo.replication_level = (uint32_t) replication_level;
- GNUNET_snprintf (kw, sizeof (kw),
- "%llu", (unsigned long long) kval);
- keywords = GNUNET_FS_uri_ksk_create (kw, NULL);
+ keywords = make_keywords (kval);
fi = GNUNET_FS_file_information_create_from_data (fs_handle,
ctx,
length,
* Task run during shutdown.
*
* @param cls unused
- * @param tc unused
*/
static void
-shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+shutdown_task (void *cls)
{
struct Pattern *p;
while (NULL != (p = publish_head))
{
- if (GNUNET_SCHEDULER_NO_TASK != p->task)
+ if (NULL != p->task)
GNUNET_SCHEDULER_cancel (p->task);
if (NULL != p->ctx)
GNUNET_FS_publish_stop (p->ctx);
}
while (NULL != (p = download_head))
{
- if (GNUNET_SCHEDULER_NO_TASK != p->task)
+ if (NULL != p->task)
GNUNET_SCHEDULER_cancel (p->task);
+ if (NULL != p->stask)
+ GNUNET_SCHEDULER_cancel (p->stask);
if (NULL != p->ctx)
GNUNET_FS_download_stop (p->ctx, GNUNET_YES);
if (NULL != p->sctx)
}
+/**
+ * Task run when a publish operation should be stopped.
+ *
+ * @param cls the 'struct Pattern' of the publish operation to stop
+ */
+static void
+publish_stop_task (void *cls)
+{
+ struct Pattern *p = cls;
+
+ p->task = NULL;
+ GNUNET_FS_publish_stop (p->ctx);
+}
+
+
+/**
+ * Task run when a download operation should be stopped.
+ *
+ * @param cls the 'struct Pattern' of the download operation to stop
+ */
+static void
+download_stop_task (void *cls)
+{
+ struct Pattern *p = cls;
+
+ p->task = NULL;
+ GNUNET_FS_download_stop (p->ctx, GNUNET_YES);
+}
+
+
+/**
+ * Task run when a download operation should be stopped.
+ *
+ * @param cls the 'struct Pattern' of the download operation to stop
+ */
+static void
+search_stop_task (void *cls)
+{
+ struct Pattern *p = cls;
+
+ p->stask = NULL;
+ GNUNET_FS_search_stop (p->sctx);
+}
+
+
/**
* Notification of FS to a client about the progress of an
* operation. Callbacks of this type will be used for uploads,
progress_cb (void *cls,
const struct GNUNET_FS_ProgressInfo *info)
{
- // FIXME:
- // - search result => start download
- // - publishing done => staitstic, terminate 'struct Pattern'
- // - download done => statistic, terminate 'struct Pattern'
- // => all patterns done => then what!? (how do we tell the
- // drive that we are done!?)
- return NULL;
+ struct Pattern *p;
+ const struct GNUNET_FS_Uri *uri;
+
+ switch (info->status)
+ {
+ case GNUNET_FS_STATUS_PUBLISH_START:
+ case GNUNET_FS_STATUS_PUBLISH_PROGRESS:
+ p = info->value.publish.cctx;
+ return p;
+ case GNUNET_FS_STATUS_PUBLISH_PROGRESS_DIRECTORY:
+ p = info->value.publish.cctx;
+ return p;
+ case GNUNET_FS_STATUS_PUBLISH_ERROR:
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+ "Publishing failed\n");
+ GNUNET_STATISTICS_update (stats_handle,
+ "# failed publish operations", 1, GNUNET_NO);
+ p = info->value.publish.cctx;
+ p->task = GNUNET_SCHEDULER_add_now (&publish_stop_task, p);
+ return p;
+ case GNUNET_FS_STATUS_PUBLISH_COMPLETED:
+ p = info->value.publish.cctx;
+ GNUNET_STATISTICS_update (stats_handle,
+ "# publishing time (ms)",
+ (long long) GNUNET_TIME_absolute_get_duration (p->start_time).rel_value_us / 1000LL,
+ GNUNET_NO);
+ p->task = GNUNET_SCHEDULER_add_now (&publish_stop_task, p);
+ return p;
+ case GNUNET_FS_STATUS_PUBLISH_STOPPED:
+ p = info->value.publish.cctx;
+ p->ctx = NULL;
+ GNUNET_CONTAINER_DLL_remove (publish_head, publish_tail, p);
+ GNUNET_free (p);
+ return NULL;
+ case GNUNET_FS_STATUS_DOWNLOAD_START:
+ case GNUNET_FS_STATUS_DOWNLOAD_PROGRESS:
+ case GNUNET_FS_STATUS_DOWNLOAD_ACTIVE:
+ case GNUNET_FS_STATUS_DOWNLOAD_INACTIVE:
+ p = info->value.download.cctx;
+ return p;
+ case GNUNET_FS_STATUS_DOWNLOAD_ERROR:
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+ "Download failed\n");
+ GNUNET_STATISTICS_update (stats_handle,
+ "# failed downloads", 1, GNUNET_NO);
+ p = info->value.download.cctx;
+ p->task = GNUNET_SCHEDULER_add_now (&download_stop_task, p);
+ return p;
+ case GNUNET_FS_STATUS_DOWNLOAD_COMPLETED:
+ p = info->value.download.cctx;
+ GNUNET_STATISTICS_update (stats_handle,
+ "# download time (ms)",
+ (long long) GNUNET_TIME_absolute_get_duration (p->start_time).rel_value_us / 1000LL,
+ GNUNET_NO);
+ p->task = GNUNET_SCHEDULER_add_now (&download_stop_task, p);
+ return p;
+ case GNUNET_FS_STATUS_DOWNLOAD_STOPPED:
+ p = info->value.download.cctx;
+ p->ctx = NULL;
+ if (NULL == p->sctx)
+ {
+ GNUNET_CONTAINER_DLL_remove (download_head, download_tail, p);
+ GNUNET_free (p);
+ }
+ return NULL;
+ case GNUNET_FS_STATUS_SEARCH_START:
+ case GNUNET_FS_STATUS_SEARCH_RESULT_NAMESPACE:
+ p = info->value.search.cctx;
+ return p;
+ case GNUNET_FS_STATUS_SEARCH_RESULT:
+ p = info->value.search.cctx;
+ uri = info->value.search.specifics.result.uri;
+ if (GNUNET_YES != GNUNET_FS_uri_test_chk (uri))
+ return NULL; /* not what we want */
+ if (p->y != GNUNET_FS_uri_chk_get_file_size (uri))
+ return NULL; /* not what we want */
+ GNUNET_STATISTICS_update (stats_handle,
+ "# search time (ms)",
+ (long long) GNUNET_TIME_absolute_get_duration (p->start_time).rel_value_us / 1000LL,
+ GNUNET_NO);
+ p->start_time = GNUNET_TIME_absolute_get ();
+ p->ctx = GNUNET_FS_download_start (fs_handle, uri,
+ NULL, NULL, NULL,
+ 0, GNUNET_FS_uri_chk_get_file_size (uri),
+ anonymity_level,
+ GNUNET_FS_DOWNLOAD_NO_TEMPORARIES,
+ p,
+ NULL);
+ p->stask = GNUNET_SCHEDULER_add_now (&search_stop_task, p);
+ return NULL;
+ case GNUNET_FS_STATUS_SEARCH_UPDATE:
+ case GNUNET_FS_STATUS_SEARCH_RESULT_STOPPED:
+ return NULL; /* don't care */
+ case GNUNET_FS_STATUS_SEARCH_ERROR:
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+ "Search failed\n");
+ GNUNET_STATISTICS_update (stats_handle,
+ "# failed searches", 1, GNUNET_NO);
+ p = info->value.search.cctx;
+ p->stask = GNUNET_SCHEDULER_add_now (&search_stop_task, p);
+ return p;
+ case GNUNET_FS_STATUS_SEARCH_STOPPED:
+ p = info->value.search.cctx;
+ p->sctx = NULL;
+ if (NULL == p->ctx)
+ {
+ GNUNET_CONTAINER_DLL_remove (download_head, download_tail, p);
+ GNUNET_free (p);
+ }
+ return NULL;
+ default:
+ /* unexpected event during profiling */
+ GNUNET_break (0);
+ return NULL;
+ }
}
* Start publish operation.
*
* @param cls the 'struct Pattern' specifying the operation to perform
- * @param tc scheduler context
*/
-static void
-start_publish (void *cls,
- const struct GNUNET_SCHEDULER_TaskContext *tc)
+static void
+start_publish (void *cls)
{
struct Pattern *p = cls;
struct GNUNET_FS_FileInformation *fi;
- p->task = GNUNET_SCHEDULER_NO_TASK;
- if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
- return;
+ p->task = NULL;
fi = make_file (p->x, p->y, p);
+ p->start_time = GNUNET_TIME_absolute_get ();
p->ctx = GNUNET_FS_publish_start (fs_handle,
fi,
NULL, NULL, NULL,
* Start download operation.
*
* @param cls the 'struct Pattern' specifying the operation to perform
- * @param tc scheduler context
*/
-static void
-start_download (void *cls,
- const struct GNUNET_SCHEDULER_TaskContext *tc)
+static void
+start_download (void *cls)
{
struct Pattern *p = cls;
+ struct GNUNET_FS_Uri *keywords;
- p->task = GNUNET_SCHEDULER_NO_TASK;
- if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
- return;
- // FIXME: start search operation
+ p->task = NULL;
+ keywords = make_keywords (p->x);
+ p->start_time = GNUNET_TIME_absolute_get ();
+ p->sctx = GNUNET_FS_search_start (fs_handle, keywords,
+ anonymity_level,
+ GNUNET_FS_SEARCH_OPTION_NONE,
+ p);
}
cfg = cfg_;
/* Scheduled the task to clean up when shutdown is called */
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
- NULL);
+ GNUNET_SCHEDULER_add_shutdown (&shutdown_task,
+ NULL);
if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_number (cfg,
+ GNUNET_CONFIGURATION_get_value_number (cfg,
"TESTBED", "PEERID",
&my_peerid))
{
return;
}
if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_number (cfg,
+ GNUNET_CONFIGURATION_get_value_number (cfg,
"FSPROFILER", "ANONYMITY_LEVEL",
&anonymity_level))
anonymity_level = 1;
if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_number (cfg,
+ GNUNET_CONFIGURATION_get_value_number (cfg,
"FSPROFILER", "REPLICATION_LEVEL",
- &replication_level))
+ &replication_level))
replication_level = 1;
GNUNET_snprintf (myoptname, sizeof (myoptname),
"DOWNLOAD-PATTERN-%u", my_peerid);
if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_string (cfg,
+ GNUNET_CONFIGURATION_get_value_string (cfg,
"FSPROFILER", myoptname,
- &download_pattern))
+ &download_pattern))
download_pattern = GNUNET_strdup ("");
GNUNET_snprintf (myoptname, sizeof (myoptname),
"PUBLISH-PATTERN-%u", my_peerid);
if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_string (cfg,
+ GNUNET_CONFIGURATION_get_value_string (cfg,
"FSPROFILER", myoptname,
- &publish_pattern))
+ &publish_pattern))
publish_pattern = GNUNET_strdup ("");
if ( (GNUNET_OK !=
parse_pattern (&download_head,