2 This file is part of GNUnet.
3 Copyright (C) 2011 - 2013 GNUnet e.V.
5 GNUnet is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published
7 by the Free Software Foundation; either version 3, or (at your
8 option) any later version.
10 GNUnet is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with GNUnet; see the file COPYING. If not, write to the
17 Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
18 Boston, MA 02110-1301, USA.
22 * @file regex/gnunet-regex-profiler.c
23 * @brief Regex profiler for testing distributed regex use.
24 * @author Bartlomiej Polot
25 * @author Maximilian Szengel
32 #include "gnunet_applications.h"
33 #include "gnunet_util_lib.h"
34 #include "regex_internal_lib.h"
35 #include "gnunet_arm_service.h"
36 #include "gnunet_dht_service.h"
37 #include "gnunet_testbed_service.h"
39 #define FIND_TIMEOUT \
40 GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 90)
48 * The testbed operation handle
50 struct GNUNET_TESTBED_Operation *op;
58 * The next pointer for DLL
60 struct DLLOperation *next;
63 * The prev pointer for DLL
65 struct DLLOperation *prev;
70 * Available states during profiling
82 STATE_SLAVES_STARTING,
100 * Matching strings against announced regexes
105 * Destroying peers; we can do this as the controller takes care of stopping a
106 * peer if it is running
108 STATE_PEERS_DESTROYING
123 * Peer configuration handle.
125 struct GNUNET_CONFIGURATION_Handle *cfg;
128 * The actual testbed peer handle.
130 struct GNUNET_TESTBED_Peer *peer_handle;
133 * Peer's search string.
135 const char *search_str;
138 * Set to GNUNET_YES if the peer successfully matched the above
139 * search string. GNUNET_NO if the string could not be matched
140 * during the profiler run. GNUNET_SYSERR if the string matching
141 * timed out. Undefined if search_str is NULL
143 int search_str_matched;
148 struct GNUNET_DHT_Handle *dht_handle;
151 * Handle to a running regex search.
153 struct REGEX_INTERNAL_Search *search_handle;
156 * Testbed operation handle for DHT.
158 struct GNUNET_TESTBED_Operation *op_handle;
161 * Peers's statistics handle.
163 struct GNUNET_STATISTICS_Handle *stats_handle;
166 * The starting time of a profiling step.
168 struct GNUNET_TIME_Absolute prof_start_time;
173 struct GNUNET_SCHEDULER_Task * timeout;
178 struct GNUNET_TESTBED_Operation *daemon_op;
182 * Set when shutting down to avoid making more queries.
184 static int in_shutdown;
187 * The array of peers; we fill this as the peers are given to us by the testbed
189 static struct RegexPeer *peers;
192 * Host registration handle
194 static struct GNUNET_TESTBED_HostRegistrationHandle *reg_handle;
197 * Handle to the master controller process
199 static struct GNUNET_TESTBED_ControllerProc *mc_proc;
202 * Handle to the master controller
204 static struct GNUNET_TESTBED_Controller *mc;
207 * Handle to global configuration
209 static struct GNUNET_CONFIGURATION_Handle *cfg;
212 * Abort task identifier
214 static struct GNUNET_SCHEDULER_Task * abort_task;
217 * Host registration task identifier
219 static struct GNUNET_SCHEDULER_Task * register_hosts_task;
222 * Global event mask for all testbed events
224 static uint64_t event_mask;
227 * The starting time of a profiling step
229 static struct GNUNET_TIME_Absolute prof_start_time;
232 * Duration profiling step has taken
234 static struct GNUNET_TIME_Relative prof_time;
237 * Number of peers to be started by the profiler
239 static unsigned int num_peers;
242 * Global testing status
247 * current state of profiling
252 * Folder where policy files are stored.
254 static char * policy_dir;
257 * File with hostnames where to execute the test.
259 static char *hosts_file;
262 * File with the strings to look for.
264 static char *strings_file;
267 * Search strings (num_peers of them).
269 static char **search_strings;
272 * How many searches are we going to start in parallel
274 static long long unsigned int init_parallel_searches;
277 * How many searches are running in parallel
279 static unsigned int parallel_searches;
282 * Number of strings found in the published regexes.
284 static unsigned int strings_found;
287 * Index of peer to start next announce/search.
289 static unsigned int next_search;
292 * Search timeout task identifier.
294 static struct GNUNET_SCHEDULER_Task * search_timeout_task;
297 * Search timeout in seconds.
299 static struct GNUNET_TIME_Relative search_timeout_time = { 60000 };
302 * File to log statistics to.
304 static struct GNUNET_DISK_FileHandle *data_file;
307 * Filename to log statistics to.
309 static char *data_filename;
312 * Prefix used for regex announcing. We need to prefix the search
313 * strings with it, in order to find something.
315 static char * regex_prefix;
318 * What's the maximum regex reannounce period.
320 static struct GNUNET_TIME_Relative reannounce_period_max;
323 /******************************************************************************/
324 /****************************** DECLARATIONS ********************************/
325 /******************************************************************************/
328 * DHT connect callback.
330 * @param cls internal peer id.
331 * @param op operation handle.
332 * @param ca_result connect adapter result.
333 * @param emsg error message.
336 dht_connect_cb (void *cls, struct GNUNET_TESTBED_Operation *op,
337 void *ca_result, const char *emsg);
340 * DHT connect adapter.
342 * @param cls not used.
343 * @param cfg configuration handle.
348 dht_ca (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg);
352 * Adapter function called to destroy a connection to
356 * @param op_result service handle returned from the connect adapter
359 dht_da (void *cls, void *op_result);
363 * Function called by testbed once we are connected to stats
364 * service. Get the statistics for the services of interest.
366 * @param cls the 'struct RegexPeer' for which we connected to stats
367 * @param op connect operation handle
368 * @param ca_result handle to stats service
369 * @param emsg error message on failure
372 stats_connect_cb (void *cls,
373 struct GNUNET_TESTBED_Operation *op,
379 * Start announcing the next regex in the DHT.
381 * @param cls Index of the next peer in the peers array.
384 announce_next_regex (void *cls);
387 /******************************************************************************/
388 /******************************** SHUTDOWN **********************************/
389 /******************************************************************************/
398 do_shutdown (void *cls)
400 struct RegexPeer *peer;
401 unsigned int peer_cnt;
402 unsigned int search_str_cnt;
403 char output_buffer[512];
406 if (NULL != abort_task)
408 GNUNET_SCHEDULER_cancel (abort_task);
411 if (NULL != register_hosts_task)
413 GNUNET_SCHEDULER_cancel (register_hosts_task);
414 register_hosts_task = NULL;
416 for (peer_cnt = 0; peer_cnt < num_peers; peer_cnt++)
418 peer = &peers[peer_cnt];
420 if (GNUNET_YES != peer->search_str_matched && NULL != data_file)
422 prof_time = GNUNET_TIME_absolute_get_duration (peer->prof_start_time);
424 GNUNET_snprintf (output_buffer,
425 sizeof (output_buffer),
426 "%p Search string not found: %s (%d)\n"
427 "%p On peer: %u (%p)\n"
429 peer, peer->search_str, peer->search_str_matched,
430 peer, peer->id, peer,
432 GNUNET_STRINGS_relative_time_to_string (prof_time,
434 if (size != GNUNET_DISK_file_write (data_file, output_buffer, size))
435 GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Unable to write to file!\n");
438 if (NULL != peers[peer_cnt].op_handle)
439 GNUNET_TESTBED_operation_done (peers[peer_cnt].op_handle);
442 if (NULL != data_file)
444 GNUNET_DISK_file_close (data_file);
447 for (search_str_cnt = 0;
448 search_str_cnt < num_peers && NULL != search_strings;
451 GNUNET_free_non_null (search_strings[search_str_cnt]);
453 GNUNET_free_non_null (search_strings);
454 search_strings = NULL;
456 if (NULL != reg_handle)
458 GNUNET_TESTBED_cancel_registration (reg_handle);
463 GNUNET_TESTBED_controller_disconnect (mc);
468 GNUNET_TESTBED_controller_stop (mc_proc);
473 GNUNET_CONFIGURATION_destroy (cfg);
480 * abort task to run on test timed out
487 unsigned long i = (unsigned long) cls;
489 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
490 "Aborting from line %lu...\n", i);
492 result = GNUNET_SYSERR;
493 GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
497 /******************************************************************************/
498 /********************* STATISTICS SERVICE CONNECTIONS ***********************/
499 /******************************************************************************/
502 * Adapter function called to establish a connection to
503 * statistics service.
506 * @param cfg configuration of the peer to connect to; will be available until
507 * GNUNET_TESTBED_operation_done() is called on the operation returned
508 * from GNUNET_TESTBED_service_connect()
509 * @return service handle to return in 'op_result', NULL on error
513 const struct GNUNET_CONFIGURATION_Handle *cfg)
515 return GNUNET_STATISTICS_create ("<driver>", cfg);
520 * Adapter function called to destroy a connection to
521 * statistics service.
524 * @param op_result service handle returned from the connect adapter
527 stats_da (void *cls, void *op_result)
529 struct RegexPeer *peer = cls;
531 GNUNET_assert (op_result == peer->stats_handle);
533 GNUNET_STATISTICS_destroy (peer->stats_handle, GNUNET_NO);
534 peer->stats_handle = NULL;
539 * Process statistic values. Write all values to global 'data_file', if present.
542 * @param subsystem name of subsystem that created the statistic
543 * @param name the name of the datum
544 * @param value the current value
545 * @param is_persistent GNUNET_YES if the value is persistent, GNUNET_NO if not
546 * @return #GNUNET_OK to continue, #GNUNET_SYSERR to abort iteration
549 stats_iterator (void *cls,
550 const char *subsystem,
552 uint64_t value, int is_persistent)
554 struct RegexPeer *peer = cls;
555 char output_buffer[512];
558 if (NULL == data_file)
560 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
561 "%p -> %s [%s]: %llu\n",
562 peer, subsystem, name, value);
566 GNUNET_snprintf (output_buffer,
567 sizeof (output_buffer),
570 subsystem, value, name);
571 if (size != GNUNET_DISK_file_write (data_file, output_buffer, size))
572 GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Unable to write to file!\n");
579 * Stats callback. Finish the stats testbed operation and when all stats have
580 * been iterated, shutdown the profiler.
583 * @param success GNUNET_OK if statistics were
584 * successfully obtained, GNUNET_SYSERR if not.
590 static unsigned int peer_cnt;
591 struct RegexPeer *peer = cls;
593 if (GNUNET_OK != success)
595 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
596 "Getting statistics for peer %u failed!\n",
601 GNUNET_assert (NULL != peer->op_handle);
603 GNUNET_TESTBED_operation_done (peer->op_handle);
604 peer->op_handle = NULL;
607 peer = &peers[peer_cnt];
609 fprintf (stderr, "s");
610 if (peer_cnt == num_peers)
612 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
613 "\nCollecting stats finished. Shutting down.\n");
614 GNUNET_SCHEDULER_shutdown ();
620 GNUNET_TESTBED_service_connect (NULL,
633 * Function called by testbed once we are connected to stats
634 * service. Get the statistics for the services of interest.
636 * @param cls the 'struct RegexPeer' for which we connected to stats
637 * @param op connect operation handle
638 * @param ca_result handle to stats service
639 * @param emsg error message on failure
642 stats_connect_cb (void *cls,
643 struct GNUNET_TESTBED_Operation *op,
647 struct RegexPeer *peer = cls;
649 if (NULL == ca_result || NULL != emsg)
651 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
652 "Failed to connect to statistics service on peer %u: %s\n",
655 peer->stats_handle = NULL;
659 peer->stats_handle = ca_result;
661 if (NULL == GNUNET_STATISTICS_get (peer->stats_handle, NULL, NULL,
662 GNUNET_TIME_UNIT_FOREVER_REL,
664 &stats_iterator, peer))
666 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
667 "Could not get statistics of peer %u!\n", peer->id);
673 * Task to collect all statistics from all peers, will shutdown the
674 * profiler, when done.
679 do_collect_stats (void *cls)
681 struct RegexPeer *peer = &peers[0];
683 GNUNET_assert (NULL != peer->peer_handle);
686 GNUNET_TESTBED_service_connect (NULL,
697 /******************************************************************************/
698 /************************ REGEX FIND CONNECTIONS **************************/
699 /******************************************************************************/
703 * Start searching for the next string in the DHT.
705 * @param cls Index of the next peer in the peers array.
708 find_string (void *cls);
712 * Method called when we've found a peer that announced a regex
713 * that matches our search string. Now get the statistics.
715 * @param cls Closure provided in REGEX_INTERNAL_search.
716 * @param id Peer providing a regex that matches the string.
717 * @param get_path Path of the get request.
718 * @param get_path_length Lenght of get_path.
719 * @param put_path Path of the put request.
720 * @param put_path_length Length of the put_path.
723 regex_found_handler (void *cls,
724 const struct GNUNET_PeerIdentity *id,
725 const struct GNUNET_PeerIdentity *get_path,
726 unsigned int get_path_length,
727 const struct GNUNET_PeerIdentity *put_path,
728 unsigned int put_path_length)
730 struct RegexPeer *peer = cls;
731 char output_buffer[512];
734 if (GNUNET_YES == peer->search_str_matched)
736 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
737 "String %s on peer %u already matched!\n",
738 peer->search_str, peer->id);
745 if (NULL != peer->timeout)
747 GNUNET_SCHEDULER_cancel (peer->timeout);
748 peer->timeout = NULL;
749 if (GNUNET_NO == in_shutdown)
750 GNUNET_SCHEDULER_add_now (&announce_next_regex, NULL);
755 // FIXME not possible right now
756 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
757 "String matching timed out for string %s on peer %u (%i/%i)\n",
758 peer->search_str, peer->id, strings_found, num_peers);
759 peer->search_str_matched = GNUNET_SYSERR;
763 prof_time = GNUNET_TIME_absolute_get_duration (peer->prof_start_time);
765 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
766 "String %s found on peer %u after %s (%i/%i) (%u||)\n",
767 peer->search_str, peer->id,
768 GNUNET_STRINGS_relative_time_to_string (prof_time, GNUNET_NO),
769 strings_found, num_peers, parallel_searches);
771 peer->search_str_matched = GNUNET_YES;
773 if (NULL != data_file)
776 GNUNET_snprintf (output_buffer,
777 sizeof (output_buffer),
779 "%p Search string: %s\n"
780 "%p Search duration: %s\n\n",
782 peer, peer->search_str,
784 GNUNET_STRINGS_relative_time_to_string (prof_time,
787 if (size != GNUNET_DISK_file_write (data_file, output_buffer, size))
788 GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Unable to write to file!\n");
792 GNUNET_TESTBED_operation_done (peer->op_handle);
793 peer->op_handle = NULL;
795 if (strings_found == num_peers)
797 prof_time = GNUNET_TIME_absolute_get_duration (prof_start_time);
798 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
799 "All strings successfully matched in %s\n",
800 GNUNET_STRINGS_relative_time_to_string (prof_time, GNUNET_NO));
802 if (NULL != search_timeout_task)
804 GNUNET_SCHEDULER_cancel (search_timeout_task);
805 search_timeout_task = NULL;
808 GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Collecting stats.\n");
809 GNUNET_SCHEDULER_add_now (&do_collect_stats, NULL);
815 * Connect by string timeout task. This will cancel the profiler after the
816 * specified timeout 'search_timeout'.
821 search_timed_out (void *cls)
825 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
826 "Finding matches to all strings did not succeed after %s.\n",
827 GNUNET_STRINGS_relative_time_to_string (search_timeout_time,
829 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
830 "Found %i of %i strings\n", strings_found, num_peers);
832 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
833 "Search timed out after %s."
834 "Collecting stats and shutting down.\n",
835 GNUNET_STRINGS_relative_time_to_string (search_timeout_time,
838 in_shutdown = GNUNET_YES;
839 for (i = 0; i < num_peers; i++)
841 if (NULL != peers[i].op_handle)
843 GNUNET_TESTBED_operation_done (peers[i].op_handle);
844 peers[i].op_handle = NULL;
847 GNUNET_SCHEDULER_add_now (&do_collect_stats, NULL);
852 * Search timed out. It might still complete in the future,
853 * but we should start another one.
855 * @param cls Index of the next peer in the peers array.
858 find_timed_out (void *cls)
860 struct RegexPeer *p = cls;
863 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
864 "Searching for string \"%s\" on peer %d timed out.\n",
867 if (GNUNET_NO == in_shutdown)
868 GNUNET_SCHEDULER_add_now (&announce_next_regex, NULL);
873 * Start searching for a string in the DHT.
875 * @param cls Index of the next peer in the peers array.
878 find_string (void *cls)
880 unsigned int search_peer = (unsigned int) (long) cls;
882 if ( (search_peer >= num_peers) ||
883 (GNUNET_YES == in_shutdown) )
886 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
887 "Searching for string \"%s\" on peer %d (%u||)\n",
888 peers[search_peer].search_str,
892 peers[search_peer].op_handle =
893 GNUNET_TESTBED_service_connect (NULL,
894 peers[search_peer].peer_handle,
900 &peers[search_peer]);
901 GNUNET_assert (NULL != peers[search_peer].op_handle);
902 peers[search_peer].timeout
903 = GNUNET_SCHEDULER_add_delayed (FIND_TIMEOUT,
905 &peers[search_peer]);
910 * Callback called when testbed has started the daemon we asked for.
913 * @param op the operation handle
914 * @param emsg NULL on success; otherwise an error description
917 daemon_started (void *cls,
918 struct GNUNET_TESTBED_Operation *op,
921 struct RegexPeer *peer = (struct RegexPeer *) cls;
922 unsigned long search_peer;
925 GNUNET_TESTBED_operation_done (peer->daemon_op);
926 peer->daemon_op = NULL;
929 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
930 "Failed to start/stop daemon at peer %u: %s\n", peer->id, emsg);
935 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
936 "Deamon %u started successfully\n", peer->id);
939 /* Find a peer to look for a string matching the regex announced */
940 search_peer = GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK,
942 for (i = 0; peers[search_peer].search_str != NULL; i++)
944 search_peer = (search_peer + 1) % num_peers;
946 GNUNET_assert (0); /* we ran out of peers, must be a bug */
948 peers[search_peer].search_str = search_strings[peer->id];
949 peers[search_peer].search_str_matched = GNUNET_NO;
950 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply(
951 reannounce_period_max,
954 (void *) search_peer);
959 * Task to start the daemons on each peer so that the regexes are announced
963 * @param tc the task context
966 do_announce (void *cls)
970 if (GNUNET_YES == in_shutdown)
972 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
973 "Starting announce.\n");
974 for (i = 0; i < init_parallel_searches; i++)
976 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
977 " scheduling announce %u\n",
979 (void) GNUNET_SCHEDULER_add_now (&announce_next_regex, NULL);
985 * Start announcing the next regex in the DHT.
987 * @param cls Closure (unused).
990 announce_next_regex (void *cls)
992 struct RegexPeer *peer;
994 if (GNUNET_YES == in_shutdown)
996 if (next_search >= num_peers)
998 if (strings_found != num_peers)
1000 struct GNUNET_TIME_Relative new_delay;
1001 if (NULL != search_timeout_task)
1002 GNUNET_SCHEDULER_cancel (search_timeout_task);
1003 new_delay = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 15);
1004 search_timeout_task = GNUNET_SCHEDULER_add_delayed (new_delay,
1011 GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Starting daemon %u\n", next_search);
1012 peer = &peers[next_search];
1014 GNUNET_TESTBED_peer_manage_service (NULL,
1021 parallel_searches++;
1026 * DHT connect callback. Called when we are connected to the dht service for
1027 * the peer in 'cls'. If successfull we connect to the stats service of this
1028 * peer and then try to match the search string of this peer.
1030 * @param cls internal peer id.
1031 * @param op operation handle.
1032 * @param ca_result connect adapter result.
1033 * @param emsg error message.
1036 dht_connect_cb (void *cls,
1037 struct GNUNET_TESTBED_Operation *op,
1041 struct RegexPeer *peer = (struct RegexPeer *) cls;
1043 if (NULL != emsg || NULL == op || NULL == ca_result)
1045 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "DHT connect failed: %s\n", emsg);
1049 GNUNET_assert (NULL != peer->dht_handle);
1050 GNUNET_assert (peer->op_handle == op);
1051 GNUNET_assert (peer->dht_handle == ca_result);
1053 peer->search_str_matched = GNUNET_NO;
1054 peer->search_handle = REGEX_INTERNAL_search (peer->dht_handle,
1056 ®ex_found_handler, peer,
1058 peer->prof_start_time = GNUNET_TIME_absolute_get ();
1063 * DHT connect adapter. Opens a connection to the dht service.
1065 * @param cls Closure (peer).
1066 * @param cfg Configuration handle.
1071 dht_ca (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg)
1073 struct RegexPeer *peer = cls;
1075 peer->dht_handle = GNUNET_DHT_connect (cfg, 32);
1077 return peer->dht_handle;
1082 * Adapter function called to destroy a connection to the dht service.
1084 * @param cls Closure (peer).
1085 * @param op_result Service handle returned from the connect adapter.
1088 dht_da (void *cls, void *op_result)
1090 struct RegexPeer *peer = (struct RegexPeer *) cls;
1092 GNUNET_assert (peer->dht_handle == op_result);
1094 if (NULL != peer->search_handle)
1096 REGEX_INTERNAL_search_cancel (peer->search_handle);
1097 peer->search_handle = NULL;
1100 if (NULL != peer->dht_handle)
1102 GNUNET_DHT_disconnect (peer->dht_handle);
1103 peer->dht_handle = NULL;
1109 * Signature of a main function for a testcase.
1112 * @param h the run handle
1113 * @param num_peers_ number of peers in 'peers'
1114 * @param testbed_peers handle to peers run in the testbed. NULL upon timeout (see
1115 * GNUNET_TESTBED_test_run()).
1116 * @param links_succeeded the number of overlay link connection attempts that
1118 * @param links_failed the number of overlay link connection attempts that
1122 test_master (void *cls,
1123 struct GNUNET_TESTBED_RunHandle *h,
1124 unsigned int num_peers_,
1125 struct GNUNET_TESTBED_Peer **testbed_peers,
1126 unsigned int links_succeeded,
1127 unsigned int links_failed)
1131 GNUNET_assert (num_peers_ == num_peers);
1133 prof_time = GNUNET_TIME_absolute_get_duration (prof_start_time);
1134 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1135 "Testbed started in %s\n",
1136 GNUNET_STRINGS_relative_time_to_string (prof_time, GNUNET_NO));
1138 if (NULL != abort_task)
1140 GNUNET_SCHEDULER_cancel (abort_task);
1144 for (i = 0; i < num_peers; i++)
1146 peers[i].peer_handle = testbed_peers[i];
1149 GNUNET_CONFIGURATION_get_value_yesno (cfg, "DHT", "DISABLE_TRY_CONNECT"))
1151 struct GNUNET_TIME_Relative settle_time;
1154 GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS,
1156 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1157 "Waiting for DHT for %s to settle new connections.\n\n",
1158 GNUNET_STRINGS_relative_time_to_string(settle_time, GNUNET_NO));
1159 GNUNET_SCHEDULER_add_delayed (settle_time, &do_announce, NULL);
1163 GNUNET_SCHEDULER_add_now (&do_announce, NULL);
1165 search_timeout_task =
1166 GNUNET_SCHEDULER_add_delayed (search_timeout_time, &search_timed_out, NULL);
1170 * Function that will be called whenever something in the testbed changes.
1172 * @param cls closure, NULL
1173 * @param event information on what is happening
1176 master_controller_cb (void *cls,
1177 const struct GNUNET_TESTBED_EventInformation *event)
1179 switch (event->type)
1181 case GNUNET_TESTBED_ET_CONNECT:
1184 case GNUNET_TESTBED_ET_PEER_START:
1194 /******************************************************************************/
1195 /*************************** TESTBED PEER SETUP *****************************/
1196 /******************************************************************************/
1200 * Load search strings from given filename. One search string per line.
1202 * @param filename filename of the file containing the search strings.
1203 * @param strings set of strings loaded from file. Caller needs to free this
1204 * if number returned is greater than zero.
1205 * @param limit upper limit on the number of strings read from the file
1206 * @return number of strings found in the file. #GNUNET_SYSERR on error.
1209 load_search_strings (const char *filename,
1216 unsigned int offset;
1220 if (NULL == filename)
1222 return GNUNET_SYSERR;
1225 if (GNUNET_YES != GNUNET_DISK_file_test (filename))
1227 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1228 "Could not find search strings file %s\n", filename);
1229 return GNUNET_SYSERR;
1231 if (GNUNET_OK != GNUNET_DISK_file_size (filename, &filesize, GNUNET_YES, GNUNET_YES))
1235 GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Search strings file %s is empty.\n", filename);
1236 return GNUNET_SYSERR;
1238 data = GNUNET_malloc (filesize);
1239 if (filesize != GNUNET_DISK_fn_read (filename, data, filesize))
1242 GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Could not read search strings file %s.\n",
1244 return GNUNET_SYSERR;
1249 while (offset < (filesize - 1) && str_cnt < limit)
1252 if (((data[offset] == '\n')) && (buf != &data[offset]))
1254 data[offset] = '\0';
1256 buf = &data[offset + 1];
1258 else if ((data[offset] == '\n') || (data[offset] == '\0'))
1259 buf = &data[offset + 1];
1261 *strings = GNUNET_malloc (sizeof (char *) * str_cnt);
1263 for (i = 0; i < str_cnt; i++)
1265 GNUNET_asprintf (&(*strings)[i], "%s%s", regex_prefix, &data[offset]);
1266 offset += strlen (&data[offset]) + 1;
1274 * Main function that will be run by the scheduler.
1276 * @param cls closure
1277 * @param args remaining command-line arguments
1278 * @param cfgfile name of the configuration file used (for saving, can be NULL!)
1279 * @param config configuration
1284 const char *cfgfile,
1285 const struct GNUNET_CONFIGURATION_Handle *config)
1287 unsigned int nsearchstrs;
1289 struct GNUNET_TIME_Relative abort_time;
1291 in_shutdown = GNUNET_NO;
1296 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1297 _("No configuration file given. Exiting\n"));
1298 GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
1301 cfg = GNUNET_CONFIGURATION_dup (config);
1303 GNUNET_CONFIGURATION_get_value_string (cfg, "REGEXPROFILER",
1307 GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR,
1310 GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
1314 GNUNET_CONFIGURATION_get_value_number (cfg, "REGEXPROFILER",
1315 "PARALLEL_SEARCHES",
1316 &init_parallel_searches))
1318 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1319 "Configuration option \"PARALLEL_SEARCHES\" missing."
1320 " Using default (%d)\n", 10);
1321 init_parallel_searches = 10;
1324 GNUNET_CONFIGURATION_get_value_time (cfg, "REGEXPROFILER",
1325 "REANNOUNCE_PERIOD_MAX",
1326 &reannounce_period_max))
1328 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1329 "reannounce_period_max not given. Using 10 minutes.\n");
1330 reannounce_period_max =
1331 GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 10);
1334 /* Check arguments */
1335 if (NULL == policy_dir)
1337 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1338 _("No policy directory specified on command line. Exiting.\n"));
1341 if (GNUNET_YES != GNUNET_DISK_directory_test (policy_dir, GNUNET_YES))
1343 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1344 _("Specified policies directory does not exist. Exiting.\n"));
1345 GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
1348 if (0 >= (int) (num_peers = GNUNET_DISK_directory_scan (policy_dir, NULL, NULL)))
1350 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1351 _("No files found in `%s'\n"),
1355 GNUNET_CONFIGURATION_set_value_string (cfg, "REGEXPROFILER",
1356 "POLICY_DIR", policy_dir);
1357 if (GNUNET_YES != GNUNET_DISK_file_test (strings_file))
1359 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1360 _("No search strings file given. Exiting.\n"));
1361 GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
1364 nsearchstrs = load_search_strings (strings_file,
1367 if (num_peers != nsearchstrs)
1369 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1370 "Error loading search strings.\n");
1371 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1372 "File (%s) does not contain enough strings (%u/%u).\n",
1373 strings_file, nsearchstrs, num_peers);
1374 GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
1377 if ( (0 == num_peers) || (NULL == search_strings))
1379 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1380 _("Error loading search strings. Exiting.\n"));
1381 GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
1384 for (i = 0; i < num_peers; i++)
1385 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1386 "search string: %s\n",
1390 if ( (NULL != data_filename) &&
1391 (NULL == (data_file =
1392 GNUNET_DISK_file_open (data_filename,
1393 GNUNET_DISK_OPEN_READWRITE |
1394 GNUNET_DISK_OPEN_TRUNCATE |
1395 GNUNET_DISK_OPEN_CREATE,
1396 GNUNET_DISK_PERM_USER_READ |
1397 GNUNET_DISK_PERM_USER_WRITE))) )
1399 GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_ERROR,
1405 /* Initialize peers */
1406 peers = GNUNET_malloc (sizeof (struct RegexPeer) * num_peers);
1407 for (i = 0; i < num_peers; i++)
1410 GNUNET_CONFIGURATION_set_value_number (cfg,
1411 "TESTBED", "OVERLAY_RANDOM_LINKS",
1413 GNUNET_CONFIGURATION_set_value_number (cfg,
1415 (long long unsigned)
1416 (log (num_peers) / log (2.0)));
1418 /* For feedback about the start process activate these and pass master_cb */
1419 event_mask |= (1LL << GNUNET_TESTBED_ET_PEER_START);
1420 // event_mask |= (1LL << GNUNET_TESTBED_ET_PEER_STOP);
1421 event_mask |= (1LL << GNUNET_TESTBED_ET_CONNECT);
1422 // event_mask |= (1LL << GNUNET_TESTBED_ET_DISCONNECT);
1423 prof_start_time = GNUNET_TIME_absolute_get ();
1424 GNUNET_TESTBED_run (hosts_file,
1428 &master_controller_cb,
1429 NULL, /* master_controller_cb cls */
1431 NULL); /* test_master cls */
1433 GNUNET_CONFIGURATION_get_value_time (cfg, "TESTBED",
1437 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1438 "SETUP_TIMEOUT not given. Using 15 minutes.\n");
1440 GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 15);
1442 abort_time = GNUNET_TIME_relative_add (abort_time, GNUNET_TIME_UNIT_MINUTES);
1444 GNUNET_SCHEDULER_add_delayed (abort_time,
1447 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1448 "setup_timeout: %s\n",
1449 GNUNET_STRINGS_relative_time_to_string (abort_time, GNUNET_YES));
1456 * @param argc argument count
1457 * @param argv argument values
1458 * @return 0 on success
1461 main (int argc, char *const *argv)
1463 static const struct GNUNET_GETOPT_CommandLineOption options[] = {
1464 {'o', "output-file", "FILENAME",
1465 gettext_noop ("name of the file for writing statistics"),
1466 GNUNET_YES, &GNUNET_GETOPT_set_string, &data_filename},
1467 {'t', "matching-timeout", "TIMEOUT",
1468 gettext_noop ("wait TIMEOUT before ending the experiment"),
1469 GNUNET_YES, &GNUNET_GETOPT_set_relative_time, &search_timeout_time},
1470 {'p', "policy-dir", "DIRECTORY",
1471 gettext_noop ("directory with policy files"),
1472 GNUNET_YES, &GNUNET_GETOPT_set_filename, &policy_dir},
1473 {'s', "strings-file", "FILENAME",
1474 gettext_noop ("name of file with input strings"),
1475 GNUNET_YES, &GNUNET_GETOPT_set_filename, &strings_file},
1476 {'H', "hosts-file", "FILENAME",
1477 gettext_noop ("name of file with hosts' names"),
1478 GNUNET_YES, &GNUNET_GETOPT_set_filename, &hosts_file},
1479 GNUNET_GETOPT_OPTION_END
1483 if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
1485 result = GNUNET_SYSERR;
1487 GNUNET_PROGRAM_run (argc, argv,
1488 "gnunet-regex-profiler",
1489 _("Profiler for regex"),
1490 options, &run, NULL);
1491 if (GNUNET_OK != ret)
1493 if (GNUNET_OK != result)