#include "gnunet_peerstore_service.h"
#include "gnunet_sensor_service.h"
#include "gnunet_sensor_util_lib.h"
+#include "gnunet_transport_service.h"
/**
* Time to wait for the peer to startup completely
*/
struct GNUNET_TESTBED_Peer *testbed_peer;
-};
+ /**
+ * Index of this peer within our list
+ */
+ int index;
+ /**
+ * TESTBED operation used to connect to statistics service
+ */
+ struct GNUNET_TESTBED_Operation *statistics_op;
+
+ /**
+ * Handle to the peer's statistics service
+ */
+ struct GNUNET_STATISTICS_Handle *statistics;
+
+};
/**
* Name of the configuration file used
static const char *sensor_dst_dir = "/tmp/gnunet-sensor-profiler";
/**
- * GNUnet configuration
+ * Scheduled task to shutdown
*/
-struct GNUNET_CONFIGURATION_Handle *cfg;
+static GNUNET_SCHEDULER_TaskIdentifier shutdown_task = GNUNET_SCHEDULER_NO_TASK;
/**
- * Return value of the program
+ * GNUnet configuration
*/
-static int ok = 1;
+static struct GNUNET_CONFIGURATION_Handle *cfg;
/**
* Number of peers to run (Option -p)
*/
static char *topology_file;
+/**
+ * Number of peers to simulate anomalies on (Option -a)
+ */
+static unsigned int anomalous_peers = 0;
+
/**
* Array of peer info for all peers
*/
*/
static unsigned int sensor_names_size = 0;
+/**
+ * Task run after any waiting period
+ */
+static GNUNET_SCHEDULER_TaskIdentifier delayed_task = GNUNET_SCHEDULER_NO_TASK;
+
/**
* Copy directory recursively
int i;
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutting down.\n");
- if (NULL != cfg)
+ if (GNUNET_SCHEDULER_NO_TASK != delayed_task)
{
- GNUNET_CONFIGURATION_destroy (cfg);
- cfg = NULL;
+ GNUNET_SCHEDULER_cancel (delayed_task);
+ delayed_task = GNUNET_SCHEDULER_NO_TASK;
}
- if (NULL != sensor_names)
+ for (i = 0; i < num_peers; i++)
{
- for (i = 0; i < sensor_names_size; i++)
- GNUNET_free (sensor_names[i]);
- GNUNET_array_grow (sensor_names, sensor_names_size, 0);
+ if (NULL != all_peers_info[i].statistics_op)
+ {
+ GNUNET_TESTBED_operation_done (all_peers_info[i].statistics_op);
+ all_peers_info[i].statistics_op = NULL;
+ }
}
if (NULL != peerstore_op)
{
GNUNET_free (all_peers_info);
all_peers_info = NULL;
}
+ if (NULL != cfg)
+ {
+ GNUNET_CONFIGURATION_destroy (cfg);
+ cfg = NULL;
+ }
+ if (NULL != sensor_names)
+ {
+ for (i = 0; i < sensor_names_size; i++)
+ GNUNET_free (sensor_names[i]);
+ GNUNET_array_grow (sensor_names, sensor_names_size, 0);
+ }
GNUNET_SCHEDULER_shutdown ();
}
if (GNUNET_YES == GNUNET_DISK_file_test (dst))
GNUNET_DISK_directory_remove (dst);
copy_result = GNUNET_DISK_file_copy (filename, dst);
+ if (GNUNET_OK == copy_result)
+ GNUNET_DISK_fix_permissions (dst, GNUNET_NO, GNUNET_NO);
}
GNUNET_free (dst);
return copy_result;
* @return #GNUNET_YES to continue iterating, #GNUNET_NO to stop
*/
static int
-peerstore_watch_cb (void *cls, struct GNUNET_PEERSTORE_Record *record,
- char *emsg)
+peerstore_watch_cb (void *cls,
+ const struct GNUNET_PEERSTORE_Record *record,
+ const char *emsg)
{
struct PeerInfo *peer = cls;
struct GNUNET_SENSOR_DashboardAnomalyEntry *anomaly;
GNUNET_assert (0 ==
GNUNET_CRYPTO_cmp_peer_identity (&peer->peer_id,
record->peer));
- printf ("Anomaly report:\n" " Peer: `%s'\n" " Sensor: `%s'\n"
- " Anomalous: `%d'\n" " Anomalous neighbors: %f.\n\n",
- GNUNET_i2s (&peer->peer_id), record->key, anomaly->anomalous,
- anomaly->anomalous_neighbors);
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Anomaly report:{'peerid': '%s'," "'peer': %d," "'sensor': '%s',"
+ "'anomalous': %d," "'neighbors': %f}\n",
+ GNUNET_i2s (&peer->peer_id), peer->index, record->key,
+ anomaly->anomalous, anomaly->anomalous_neighbors);
return GNUNET_YES;
}
}
+/**
+ * Callback to be called when statistics service connect operation is completed
+ *
+ * @param cls the callback closure from functions generating an operation
+ * @param op the operation that has been finished
+ * @param ca_result the service handle returned from GNUNET_TESTBED_ConnectAdapter()
+ * @param emsg error message in case the operation has failed; will be NULL if
+ * operation has executed successfully.
+ */
+static void
+statistics_connect_cb (void *cls, struct GNUNET_TESTBED_Operation *op,
+ void *ca_result, const char *emsg)
+{
+ struct PeerInfo *peer = cls;
+
+ if (NULL != emsg)
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "ERROR: %s.\n", emsg);
+ GNUNET_assert (0);
+ }
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Connected to statistics service on peer `%s'.\n", GNUNET_i2s (&peer->peer_id));
+ GNUNET_STATISTICS_set (peer->statistics, "# peers connected", 0, GNUNET_NO);
+}
+
+
+/**
+ * Adapter function called to establish a connection to statistics service.
+ *
+ * @param cls closure
+ * @param cfg configuration of the peer to connect to; will be available until
+ * GNUNET_TESTBED_operation_done() is called on the operation returned
+ * from GNUNET_TESTBED_service_connect()
+ * @return service handle to return in 'op_result', NULL on error
+ */
+static void *
+statistics_connect_adapter (void *cls,
+ const struct GNUNET_CONFIGURATION_Handle *cfg)
+{
+ struct PeerInfo *peer = cls;
+
+ peer->statistics = GNUNET_STATISTICS_create ("core", cfg);
+ GNUNET_assert (NULL != peer->statistics);
+ return peer->statistics;
+}
+
+
+/**
+ * Adapter function called to destroy a connection to statistics service.
+ *
+ * @param cls closure
+ * @param op_result service handle returned from the connect adapter
+ */
+static void
+statistics_disconnect_adapter (void *cls, void *op_result)
+{
+ struct PeerInfo *peer = cls;
+
+ GNUNET_STATISTICS_destroy (peer->statistics, GNUNET_NO);
+ peer->statistics = NULL;
+}
+
+
+/**
+ * This function is called after the estimated training period is over.
+ */
+static void
+simulate_anomalies (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+ int i;
+ uint32_t an_peer;
+ struct GNUNET_TIME_Relative shutdown_delay;
+
+ delayed_task = GNUNET_SCHEDULER_NO_TASK;
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Training period over, simulating anomalies now.\n");
+ GNUNET_assert (anomalous_peers <= num_peers);
+ for (i = 0; i < anomalous_peers; i++)
+ {
+ an_peer = GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, num_peers);
+ if (NULL != all_peers_info[an_peer].statistics_op)
+ {
+ i--;
+ continue;
+ }
+ all_peers_info[an_peer].statistics_op =
+ GNUNET_TESTBED_service_connect (NULL, all_peers_info[an_peer].testbed_peer,
+ "statistics", &statistics_connect_cb,
+ &all_peers_info[an_peer], &statistics_connect_adapter,
+ &statistics_disconnect_adapter, &all_peers_info[an_peer]);
+ }
+ shutdown_delay = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, num_peers * 6);
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutting down in %s\n",
+ GNUNET_STRINGS_relative_time_to_string (shutdown_delay, GNUNET_NO));
+ GNUNET_SCHEDULER_cancel (shutdown_task);
+ shutdown_task = GNUNET_SCHEDULER_add_delayed (shutdown_delay, &do_shutdown, NULL);
+}
+
+
/**
* This function is called after a delay which ensures that all peers are
* properly initialized
unsigned long long int training_points;
struct GNUNET_TIME_Relative training_period;
+ delayed_task = GNUNET_SCHEDULER_NO_TASK;
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "All peers are ready.\n");
GNUNET_assert (GNUNET_OK ==
GNUNET_CONFIGURATION_get_value_number (cfg,
"Sleeping for a training period of %s.\n",
GNUNET_STRINGS_relative_time_to_string (training_period,
GNUNET_NO));
- //TODO
+ delayed_task =
+ GNUNET_SCHEDULER_add_delayed (training_period, &simulate_anomalies, NULL);
}
sensor_services_started++;
if (sensor_services_started == num_peers)
{
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
- (PEER_STARTUP_TIME, num_peers), &peers_ready,
- NULL);
+ delayed_task =
+ GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
+ (PEER_STARTUP_TIME, num_peers),
+ &peers_ready, NULL);
}
}
}
peer->testbed_peer = testbed_peer;
GNUNET_CRYPTO_get_peer_identity (pinfo->result.cfg, &peer->peer_id);
+ peer->index = peers_known;
peers_known++;
if (1 == peers_known) /* First peer is collection point */
{
"%d peers started. %d links succeeded. %d links failed.\n",
num_peers, links_succeeded, links_failed);
GNUNET_assert (num == num_peers);
- GNUNET_assert (0 == links_failed);
/* Collect peer information */
all_peers_info = GNUNET_new_array (num_peers, struct PeerInfo);
static int
verify_args ()
{
- if (num_peers < 3)
+ if (num_peers < 2)
{
GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
_
- ("Invalid or missing number of peers. Set at least 3 peers.\n"));
+ ("Invalid or missing number of peers. Set at least 2 peers.\n"));
return GNUNET_SYSERR;
}
if (NULL == topology_file ||
cfg, "TESTBED",
"OVERLAY_TOPOLOGY_FILE",
topology_file);
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &do_shutdown,
- NULL);
+ shutdown_task =
+ GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &do_shutdown,
+ NULL);
GNUNET_TESTBED_run (NULL, cfg, num_peers, 0, NULL, NULL, &test_master, NULL);
}
{'t', "topology-file", "FILEPATH", gettext_noop ("Path to topology file"),
GNUNET_YES, &GNUNET_GETOPT_set_filename, &topology_file},
{'i', "sensors-interval", "INTERVAL",
- gettext_noop ("Change the interval or running sensors to given value"),
+ gettext_noop ("Change the interval of running sensors to given value"),
GNUNET_YES, &GNUNET_GETOPT_set_uint, &sensors_interval},
+ {'a', "anomalous-peers", "COUNT",
+ gettext_noop ("Number of peers to simulate anomalies on"), GNUNET_YES,
+ &GNUNET_GETOPT_set_uint, &anomalous_peers},
GNUNET_GETOPT_OPTION_END
};
return (GNUNET_OK ==
GNUNET_PROGRAM_run (argc, argv, "gnunet-sensor-profiler",
gettext_noop ("Profiler for sensor service"),
- options, &run, NULL)) ? ok : 1;
+ options, &run, NULL)) ? 0 : 1;
}
/* end of gnunet-sensor-profiler.c */
projects / oweals / gnunet.git / blobdiff