2 This file is part of GNUnet
3 (C) 2008, 2009 Christian Grothoff (and other contributing authors)
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., 59 Temple Place - Suite 330,
18 Boston, MA 02111-1307, USA.
22 * @file include/gnunet_testing_lib.h
23 * @brief convenience API for writing testcases for GNUnet
24 * Many testcases need to start and stop gnunetd,
25 * and this library is supposed to make that easier
26 * for TESTCASES. Normal programs should always
27 * use functions from gnunet_{util,arm}_lib.h. This API is
28 * ONLY for writing testcases!
29 * @author Christian Grothoff
32 #ifndef GNUNET_TESTING_LIB_H
33 #define GNUNET_TESTING_LIB_H
35 #include "gnunet_util_lib.h"
36 #include "gnunet_statistics_service.h"
41 #if 0 /* keep Emacsens' auto-indent happy */
47 * Handle for a GNUnet daemon (technically a set of
48 * daemons; the handle is really for the master ARM
49 * daemon) started by the testing library.
51 struct GNUNET_TESTING_Daemon;
54 * Linked list of hostnames and ports to use for starting daemons.
56 struct GNUNET_TESTING_Host
59 * Pointer to next item in the list.
61 struct GNUNET_TESTING_Host *next;
64 * Hostname to connect to.
69 * Username to use when connecting (may be null).
74 * Port to use for SSH connection (used for ssh
75 * connection forwarding, 0 to let ssh decide)
81 * Prototype of a function that will be called whenever
82 * a daemon was started by the testing library.
85 * @param id identifier for the daemon, NULL on error
86 * @param d handle for the daemon
87 * @param emsg error message (NULL on success)
89 typedef void (*GNUNET_TESTING_NotifyHostkeyCreated)(void *cls,
90 const struct GNUNET_PeerIdentity *id,
91 struct GNUNET_TESTING_Daemon *d,
95 * Prototype of a function that will be called whenever
96 * a daemon was started by the testing library.
99 * @param id identifier for the daemon, NULL on error
100 * @param cfg configuration used by this daemon
101 * @param d handle for the daemon
102 * @param emsg error message (NULL on success)
104 typedef void (*GNUNET_TESTING_NotifyDaemonRunning)(void *cls,
105 const struct GNUNET_PeerIdentity *id,
106 const struct GNUNET_CONFIGURATION_Handle *cfg,
107 struct GNUNET_TESTING_Daemon *d,
112 * Handle to an entire testbed of GNUnet peers.
114 struct GNUNET_TESTING_Testbed;
117 * Phases of starting GNUnet on a system.
119 enum GNUNET_TESTING_StartPhase
122 * Copy the configuration file to the target system.
127 * Configuration file has been copied, generate hostkey.
132 * Create the hostkey for the peer.
137 * Hostkey generated, wait for topology to be finished.
142 * Topology has been created, now start ARM.
147 * ARM has been started, check that it has properly daemonized and
148 * then try to connect to the CORE service (which should be
149 * auto-started by ARM).
154 * We're waiting for CORE to start.
159 * Core has notified us that we've established a connection to the service.
160 * The main FSM halts here and waits to be moved to UPDATE or CLEANUP.
165 * We've been asked to terminate the instance and are now waiting for
166 * the remote command to stop the gnunet-arm process and delete temporary
172 * We've received a configuration update and are currently waiting for
173 * the copy process for the update to complete. Once it is, we will
174 * return to "SP_START_DONE" (and rely on ARM to restart all affected
181 * Prototype of a function that will be called when a
182 * particular operation was completed the testing library.
185 * @param emsg NULL on success
187 typedef void (*GNUNET_TESTING_NotifyCompletion)(void *cls,
191 * Prototype of a function that will be called with the
192 * number of connections created for a particular topology.
195 * @param num_connections the number of connections created
197 typedef void (*GNUNET_TESTING_NotifyConnections)(void *cls,
198 unsigned int num_connections);
201 * Handle for a GNUnet daemon (technically a set of
202 * daemons; the handle is really for the master ARM
203 * daemon) started by the testing library.
205 struct GNUNET_TESTING_Daemon
210 struct GNUNET_SCHEDULER_Handle *sched;
215 struct GNUNET_CONFIGURATION_Handle *cfg;
218 * At what time to give up starting the peer
220 struct GNUNET_TIME_Absolute max_timeout;
223 * Host to run GNUnet on.
228 * Port to use for ssh, NULL to let system choose default.
233 * Result of GNUNET_i2s of this peer,
239 * Username we are using.
244 * Name of the configuration file
249 * Callback to inform initiator that the peer's
250 * hostkey has been created.
252 GNUNET_TESTING_NotifyHostkeyCreated hostkey_callback;
255 * Closure for hostkey creation callback.
260 * Function to call when the peer is running.
262 GNUNET_TESTING_NotifyDaemonRunning cb;
270 * Arguments from "daemon_stop" call.
272 GNUNET_TESTING_NotifyCompletion dead_cb;
275 * Closure for 'dead_cb'.
280 * Arguments from "daemon_stop" call.
282 GNUNET_TESTING_NotifyCompletion update_cb;
285 * Closure for 'update_cb'.
290 * Identity of this peer (once started).
292 struct GNUNET_PeerIdentity id;
295 * Flag to indicate that we've already been asked
296 * to terminate (but could not because some action
297 * was still pending).
302 * PID of the process that we started last.
307 * In which phase are we during the start of
310 enum GNUNET_TESTING_StartPhase phase;
313 * ID of the current task.
315 GNUNET_SCHEDULER_TaskIdentifier task;
318 * Handle to the server.
320 struct GNUNET_CORE_Handle *server;
323 * Handle to the transport service of this peer
325 struct GNUNET_TRANSPORT_Handle *th;
328 * HELLO message for this peer
330 struct GNUNET_HELLO_Message *hello;
333 * Handle to a pipe for reading the hostkey.
335 struct GNUNET_DISK_PipeHandle *pipe_stdout;
338 * Output from gnunet-peerinfo is read into this buffer.
340 char hostkeybuf[105];
343 * Current position in 'hostkeybuf' (for reading from gnunet-peerinfo)
345 unsigned int hostkeybufpos;
348 * Set to GNUNET_YES once the peer is up.
353 * Used to tell shutdown not to remove configuration for the peer
354 * (if it's going to be restarted later)
361 * Handle to a group of GNUnet peers.
363 struct GNUNET_TESTING_PeerGroup;
367 * Prototype of a function that will be called whenever
368 * two daemons are connected by the testing library.
371 * @param first peer id for first daemon
372 * @param second peer id for the second daemon
373 * @param distance distance between the connected peers
374 * @param first_cfg config for the first daemon
375 * @param second_cfg config for the second daemon
376 * @param first_daemon handle for the first daemon
377 * @param second_daemon handle for the second daemon
378 * @param emsg error message (NULL on success)
380 typedef void (*GNUNET_TESTING_NotifyConnection)(void *cls,
381 const struct GNUNET_PeerIdentity *first,
382 const struct GNUNET_PeerIdentity *second,
384 const struct GNUNET_CONFIGURATION_Handle *first_cfg,
385 const struct GNUNET_CONFIGURATION_Handle *second_cfg,
386 struct GNUNET_TESTING_Daemon *first_daemon,
387 struct GNUNET_TESTING_Daemon *second_daemon,
391 * Prototype of a callback function indicating that two peers
392 * are currently connected.
395 * @param first peer id for first daemon
396 * @param second peer id for the second daemon
397 * @param distance distance between the connected peers
398 * @param emsg error message (NULL on success)
400 typedef void (*GNUNET_TESTING_NotifyTopology)(void *cls,
401 const struct GNUNET_PeerIdentity *first,
402 const struct GNUNET_PeerIdentity *second,
403 struct GNUNET_TIME_Relative latency,
408 * Starts a GNUnet daemon. GNUnet must be installed on the target
409 * system and available in the PATH. The machine must furthermore be
410 * reachable via "ssh" (unless the hostname is "NULL") without the
411 * need to enter a password.
413 * @param sched scheduler to use
414 * @param cfg configuration to use
415 * @param timeout how long to wait starting up peers
416 * @param hostname name of the machine where to run GNUnet
417 * (use NULL for localhost).
418 * @param ssh_username ssh username to use when connecting to hostname
419 * @param sshport port to pass to ssh process when connecting to hostname
420 * @param hostkey_callback function to call once the hostkey has been
421 * generated for this peer, but it hasn't yet been started
422 * (NULL to start immediately, otherwise waits on GNUNET_TESTING_daemon_continue_start)
423 * @param hostkey_cls closure for hostkey callback
424 * @param cb function to call once peer is up, or failed to start
425 * @param cb_cls closure for cb
426 * @return handle to the daemon (actual start will be completed asynchronously)
428 struct GNUNET_TESTING_Daemon *
429 GNUNET_TESTING_daemon_start (struct GNUNET_SCHEDULER_Handle *sched,
430 const struct GNUNET_CONFIGURATION_Handle *cfg,
431 struct GNUNET_TIME_Relative timeout,
432 const char *hostname,
433 const char *ssh_username,
435 GNUNET_TESTING_NotifyHostkeyCreated hostkey_callback,
437 GNUNET_TESTING_NotifyDaemonRunning cb,
441 * Continues GNUnet daemon startup when user wanted to be notified
442 * once a hostkey was generated (for creating friends files, blacklists,
445 * @param daemon the daemon to finish starting
448 GNUNET_TESTING_daemon_continue_startup(struct GNUNET_TESTING_Daemon *daemon);
451 * Check whether the given daemon is running.
453 * @param daemon the daemon to check
455 * @return GNUNET_YES if the daemon is up, GNUNET_NO if the
456 * daemon is down, GNUNET_SYSERR on error.
459 GNUNET_TESTING_daemon_running (struct GNUNET_TESTING_Daemon *daemon);
462 * Restart (stop and start) a GNUnet daemon.
464 * @param d the daemon that should be restarted
465 * @param cb function called once the daemon is (re)started
466 * @param cb_cls closure for cb
469 GNUNET_TESTING_daemon_restart (struct GNUNET_TESTING_Daemon *d,
470 GNUNET_TESTING_NotifyDaemonRunning cb, void *cb_cls);
473 * Start a peer that has previously been stopped using the daemon_stop
474 * call (and files weren't deleted and the allow restart flag)
476 * @param daemon the daemon to start (has been previously stopped)
477 * @param timeout how long to wait for restart
478 * @param cb the callback for notification when the peer is running
479 * @param cb_cls closure for the callback
482 GNUNET_TESTING_daemon_start_stopped (struct GNUNET_TESTING_Daemon *daemon,
483 struct GNUNET_TIME_Relative timeout,
484 GNUNET_TESTING_NotifyDaemonRunning cb,
488 * Get a certain testing daemon handle.
490 * @param pg handle to the set of running peers
491 * @param position the number of the peer to return
493 struct GNUNET_TESTING_Daemon *
494 GNUNET_TESTING_daemon_get (struct GNUNET_TESTING_PeerGroup *pg,
495 unsigned int position);
498 * Get a daemon by peer identity, so callers can
499 * retrieve the daemon without knowing it's offset.
501 * @param pg the peer group to retrieve the daemon from
502 * @param peer_id the peer identity of the daemon to retrieve
504 * @return the daemon on success, or NULL if no such peer identity is found
506 struct GNUNET_TESTING_Daemon *
507 GNUNET_TESTING_daemon_get_by_id (struct GNUNET_TESTING_PeerGroup *pg,
508 struct GNUNET_PeerIdentity *peer_id);
511 * Stops a GNUnet daemon.
513 * @param d the daemon that should be stopped
514 * @param timeout how long to wait for process for shutdown to complete
515 * @param cb function called once the daemon was stopped
516 * @param cb_cls closure for cb
517 * @param delete_files GNUNET_YES to remove files, GNUNET_NO
518 * to leave them (i.e. for restarting at a later time,
519 * or logfile inspection once finished)
520 * @param allow_restart GNUNET_YES to restart peer later (using this API)
521 * GNUNET_NO to kill off and clean up for good
524 GNUNET_TESTING_daemon_stop (struct GNUNET_TESTING_Daemon *d,
525 struct GNUNET_TIME_Relative timeout,
526 GNUNET_TESTING_NotifyCompletion cb, void *cb_cls,
527 int delete_files, int allow_restart);
531 * Changes the configuration of a GNUnet daemon.
533 * @param d the daemon that should be modified
534 * @param cfg the new configuration for the daemon
535 * @param cb function called once the configuration was changed
536 * @param cb_cls closure for cb
538 void GNUNET_TESTING_daemon_reconfigure (struct GNUNET_TESTING_Daemon *d,
539 struct GNUNET_CONFIGURATION_Handle *cfg,
540 GNUNET_TESTING_NotifyCompletion cb,
545 * Establish a connection between two GNUnet daemons.
547 * @param d1 handle for the first daemon
548 * @param d2 handle for the second daemon
549 * @param timeout how long is the connection attempt
551 * @param max_connect_attempts how many times should we try to reconnect
553 * @param cb function to call at the end
554 * @param cb_cls closure for cb
556 void GNUNET_TESTING_daemons_connect (struct GNUNET_TESTING_Daemon *d1,
557 struct GNUNET_TESTING_Daemon *d2,
558 struct GNUNET_TIME_Relative timeout,
559 unsigned int max_connect_attempts,
560 GNUNET_TESTING_NotifyConnection cb,
567 * Start count gnunetd processes with the same set of transports and
568 * applications. The port numbers (any option called "PORT") will be
569 * adjusted to ensure that no two peers running on the same system
570 * have the same port(s) in their respective configurations.
572 * @param sched scheduler to use
573 * @param cfg configuration template to use
574 * @param total number of daemons to start
575 * @param timeout total time allowed for peers to start
576 * @param hostkey_callback function to call on each peers hostkey generation
577 * if NULL, peers will be started by this call, if non-null,
578 * GNUNET_TESTING_daemons_continue_startup must be called after
579 * successful hostkey generation
580 * @param hostkey_cls closure for hostkey callback
581 * @param cb function to call on each daemon that was started
582 * @param cb_cls closure for cb
583 * @param connect_callback function to call each time two hosts are connected
584 * @param connect_callback_cls closure for connect_callback
585 * @param hostnames linked list of hosts to use to start peers on (NULL to run on localhost only)
587 * @return NULL on error, otherwise handle to control peer group
589 struct GNUNET_TESTING_PeerGroup *
590 GNUNET_TESTING_daemons_start (struct GNUNET_SCHEDULER_Handle *sched,
591 const struct GNUNET_CONFIGURATION_Handle *cfg,
593 struct GNUNET_TIME_Relative timeout,
594 GNUNET_TESTING_NotifyHostkeyCreated hostkey_callback,
596 GNUNET_TESTING_NotifyDaemonRunning cb,
598 GNUNET_TESTING_NotifyConnection
599 connect_callback, void *connect_callback_cls,
600 const struct GNUNET_TESTING_Host *hostnames);
603 * Function which continues a peer group starting up
604 * after successfully generating hostkeys for each peer.
606 * @param pg the peer group to continue starting
609 GNUNET_TESTING_daemons_continue_startup(struct GNUNET_TESTING_PeerGroup *pg);
612 * Restart all peers in the given group.
614 * @param pg the handle to the peer group
615 * @param callback function to call on completion (or failure)
616 * @param callback_cls closure for the callback function
619 GNUNET_TESTING_daemons_restart (struct GNUNET_TESTING_PeerGroup *pg,
620 GNUNET_TESTING_NotifyCompletion callback,
625 * Shutdown all peers started in the given group.
627 * @param pg handle to the peer group
628 * @param timeout how long to wait for shutdown
629 * @param cb callback to notify upon success or failure
630 * @param cb_cls closure for cb
633 GNUNET_TESTING_daemons_stop (struct GNUNET_TESTING_PeerGroup *pg,
634 struct GNUNET_TIME_Relative timeout,
635 GNUNET_TESTING_NotifyCompletion cb,
640 * Count the number of running peers.
642 * @param pg handle for the peer group
644 * @return the number of currently running peers in the peer group
647 GNUNET_TESTING_daemons_running (struct GNUNET_TESTING_PeerGroup *pg);
650 * Simulate churn by stopping some peers (and possibly
651 * re-starting others if churn is called multiple times). This
652 * function can only be used to create leave-join churn (peers "never"
653 * leave for good). First "voff" random peers that are currently
654 * online will be taken offline; then "von" random peers that are then
655 * offline will be put back online. No notifications will be
656 * generated for any of these operations except for the callback upon
657 * completion. Note that the implementation is at liberty to keep
658 * the ARM service itself (but none of the other services or daemons)
659 * running even though the "peer" is being varied offline.
661 * @param pg handle for the peer group
662 * @param voff number of peers that should go offline
663 * @param von number of peers that should come back online;
664 * must be zero on first call (since "testbed_start"
665 * always starts all of the peers)
666 * @param timeout how long to wait for operations to finish before
668 * @param cb function to call at the end
669 * @param cb_cls closure for cb
672 GNUNET_TESTING_daemons_churn (struct GNUNET_TESTING_PeerGroup *pg,
675 struct GNUNET_TIME_Relative timeout,
676 GNUNET_TESTING_NotifyCompletion cb,
680 * Callback function to process statistic values.
683 * @param peer the peer the statistics belong to
684 * @param subsystem name of subsystem that created the statistic
685 * @param name the name of the datum
686 * @param value the current value
687 * @param is_persistent GNUNET_YES if the value is persistent, GNUNET_NO if not
688 * @return GNUNET_OK to continue, GNUNET_SYSERR to abort iteration
690 typedef int (*GNUNET_TESTING_STATISTICS_Iterator) (void *cls,
691 const struct GNUNET_PeerIdentity *peer,
692 const char *subsystem,
698 * Iterate over all (running) peers in the peer group, retrieve
699 * all statistics from each.
702 GNUNET_TESTING_get_statistics (struct GNUNET_TESTING_PeerGroup *pg,
703 GNUNET_STATISTICS_Callback cont,
704 GNUNET_TESTING_STATISTICS_Iterator proc, void *cls);
707 * Topologies supported for testbeds.
709 enum GNUNET_TESTING_Topology
712 * A clique (everyone connected to everyone else).
714 GNUNET_TESTING_TOPOLOGY_CLIQUE,
717 * Small-world network (2d torus plus random links).
719 GNUNET_TESTING_TOPOLOGY_SMALL_WORLD,
722 * Small-world network (ring plus random links).
724 GNUNET_TESTING_TOPOLOGY_SMALL_WORLD_RING,
729 GNUNET_TESTING_TOPOLOGY_RING,
734 GNUNET_TESTING_TOPOLOGY_2D_TORUS,
739 GNUNET_TESTING_TOPOLOGY_ERDOS_RENYI,
742 * Certain percentage of peers are unable to communicate directly
743 * replicating NAT conditions
745 GNUNET_TESTING_TOPOLOGY_INTERNAT,
748 * Scale free topology.
750 GNUNET_TESTING_TOPOLOGY_SCALE_FREE,
753 * Straight line topology.
755 GNUNET_TESTING_TOPOLOGY_LINE,
758 * All peers are disconnected.
760 GNUNET_TESTING_TOPOLOGY_NONE
764 * Options for connecting a topology.
766 enum GNUNET_TESTING_TopologyOption
769 * Try to connect all peers specified in the topology.
771 GNUNET_TESTING_TOPOLOGY_OPTION_ALL,
774 * Choose a random subset of connections to create.
776 GNUNET_TESTING_TOPOLOGY_OPTION_RANDOM,
779 * Create at least X connections for each peer.
781 GNUNET_TESTING_TOPOLOGY_OPTION_MINIMUM,
784 * Using a depth first search, create one connection
785 * per peer. If any are missed (graph disconnected)
786 * start over at those peers until all have at least one
789 GNUNET_TESTING_TOPOLOGY_OPTION_DFS,
792 * Find the N closest peers to each allowed peer in the
793 * topology and make sure a connection to those peers
794 * exists in the connect topology.
796 GNUNET_TESTING_TOPOLOGY_OPTION_ADD_CLOSEST,
799 * No options specified.
801 GNUNET_TESTING_TOPOLOGY_OPTION_NONE
806 * Get a topology from a string input.
808 * @param topology where to write the retrieved topology
809 * @param topology_string The string to attempt to
810 * get a configuration value from
811 * @return GNUNET_YES if topology string matched a
812 * known topology, GNUNET_NO if not
815 GNUNET_TESTING_topology_get(enum GNUNET_TESTING_Topology *topology, char * topology_string);
818 * Get connect topology option from string input.
820 * @param topology_option where to write the retrieved topology
821 * @param topology_string The string to attempt to
822 * get a configuration value from
823 * @return GNUNET_YES if topology string matched a
824 * known topology, GNUNET_NO if not
827 GNUNET_TESTING_topology_option_get(enum GNUNET_TESTING_TopologyOption *topology_option,
828 char * topology_string);
832 * Takes a peer group and creates a topology based on the
833 * one specified. Creates a topology means generates friend
834 * files for the peers so they can only connect to those allowed
835 * by the topology. This will only have an effect once peers
836 * are started if the FRIENDS_ONLY option is set in the base
837 * config. Also takes an optional restrict topology which
838 * disallows direct TCP connections UNLESS they are specified in
839 * the restricted topology.
841 * @param pg the peer group struct representing the running peers
842 * @param topology which topology to connect the peers in
843 * @param restrict_topology allow only direct TCP connections in this topology
844 * @param restrict_transports space delimited list of transports to blacklist
845 * to create restricted topology
847 * @return the maximum number of connections were all allowed peers
848 * connected to each other
851 GNUNET_TESTING_create_topology (struct GNUNET_TESTING_PeerGroup *pg,
852 enum GNUNET_TESTING_Topology topology,
853 enum GNUNET_TESTING_Topology restrict_topology,
854 char *restrict_transports);
857 * Iterate over all (running) peers in the peer group, retrieve
858 * all connections that each currently has.
860 * @param pg the peer group we are concerned with
861 * @param cb callback for topology information
862 * @param cls closure for callback
865 GNUNET_TESTING_get_topology (struct GNUNET_TESTING_PeerGroup *pg, GNUNET_TESTING_NotifyTopology cb, void *cls);
868 * There are many ways to connect peers that are supported by this function.
869 * To connect peers in the same topology that was created via the
870 * GNUNET_TESTING_create_topology, the topology variable must be set to
871 * GNUNET_TESTING_TOPOLOGY_NONE. If the topology variable is specified,
872 * a new instance of that topology will be generated and attempted to be
873 * connected. This could result in some connections being impossible,
874 * because some topologies are non-deterministic.
876 * @param pg the peer group struct representing the running peers
877 * @param topology which topology to connect the peers in
878 * @param options options for connecting the topology
879 * @param option_modifier modifier for options that take a parameter
880 * @return the number of connections that will be attempted, GNUNET_SYSERR on error
883 GNUNET_TESTING_connect_topology (struct GNUNET_TESTING_PeerGroup *pg,
884 enum GNUNET_TESTING_Topology topology,
885 enum GNUNET_TESTING_TopologyOption options,
886 double option_modifier);
889 * Start or stop an individual peer from the given group.
891 * @param pg handle to the peer group
892 * @param offset which peer to start or stop
893 * @param desired_status GNUNET_YES to have it running, GNUNET_NO to stop it
894 * @param timeout how long to wait for shutdown
895 * @param cb function to call at the end
896 * @param cb_cls closure for cb
899 GNUNET_TESTING_daemons_vary (struct GNUNET_TESTING_PeerGroup *pg,
902 struct GNUNET_TIME_Relative timeout,
903 GNUNET_TESTING_NotifyCompletion cb,
907 #if 0 /* keep Emacsens' auto-indent happy */