2 This file is part of GNUnet.
3 (C) 2006, 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 2, 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 transport_testing.c
23 * @brief testing lib for transport service
25 * @author Matthias Wachs
28 #include "transport-testing.h"
30 struct ConnectingContext
32 struct PeerContext * p1;
33 struct PeerContext * p2;
34 GNUNET_SCHEDULER_TaskIdentifier tct;
38 notify_connect (void *cls,
39 const struct GNUNET_PeerIdentity *peer,
40 const struct GNUNET_TRANSPORT_ATS_Information *ats,
43 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer `%s' connected \n",
46 struct PeerContext * p = cls;
50 p->nc (p->cb_cls, peer, ats, ats_count);
54 notify_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
56 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer `%s' disconnected \n",
59 struct PeerContext * p = cls;
63 p->nd (p->cb_cls, peer);
67 notify_receive (void *cls,
68 const struct GNUNET_PeerIdentity *peer,
69 const struct GNUNET_MessageHeader *message,
70 const struct GNUNET_TRANSPORT_ATS_Information *ats,
73 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Receiving\n");
75 struct PeerContext * p = cls;
79 p->rec (p->cb_cls, peer, message, ats, ats_count);
84 exchange_hello_last (void *cls,
85 const struct GNUNET_MessageHeader *message)
87 struct ConnectingContext * cc = cls;
88 struct PeerContext *me = cc->p2;
89 struct PeerContext *p1 = cc->p1;
91 GNUNET_assert (message != NULL);
92 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
93 "Exchanging HELLO of size %d with peer (%s)!\n",
94 (int) GNUNET_HELLO_size((const struct GNUNET_HELLO_Message *)message),
95 GNUNET_i2s (&me->id));
96 GNUNET_assert (GNUNET_OK ==
97 GNUNET_HELLO_get_id ((const struct GNUNET_HELLO_Message *)
99 GNUNET_TRANSPORT_offer_hello (p1->th, message, NULL, NULL);
104 exchange_hello (void *cls,
105 const struct GNUNET_MessageHeader *message)
107 struct ConnectingContext * cc = cls;
108 struct PeerContext *me = cc->p1;
109 struct PeerContext *p2 = cc->p2;
111 GNUNET_assert (message != NULL);
112 GNUNET_assert (GNUNET_OK ==
113 GNUNET_HELLO_get_id ((const struct GNUNET_HELLO_Message *)
115 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
116 "Exchanging HELLO of size %d from peer %s!\n",
117 (int) GNUNET_HELLO_size((const struct GNUNET_HELLO_Message *)message),
118 GNUNET_i2s (&me->id));
119 GNUNET_TRANSPORT_offer_hello (p2->th, message, NULL, NULL);
123 try_connect (void *cls,
124 const struct GNUNET_SCHEDULER_TaskContext *tc)
126 struct ConnectingContext * cc = cls;
127 struct PeerContext *p1 = cc->p1;
128 struct PeerContext *p2 = cc->p2;
130 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
131 "Asking peers to connect...\n");
132 /* FIXME: 'pX.id' may still be all-zeros here... */
133 GNUNET_TRANSPORT_try_connect (p2->th,
135 GNUNET_TRANSPORT_try_connect (p1->th,
137 cc->tct = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
143 GNUNET_TRANSPORT_TESTING_start_peer (const char * cfgname,
144 GNUNET_TRANSPORT_ReceiveCallback rec,
145 GNUNET_TRANSPORT_NotifyConnect nc,
146 GNUNET_TRANSPORT_NotifyDisconnect nd,
149 struct PeerContext * p = GNUNET_malloc (sizeof (struct PeerContext));
151 p->cfg = GNUNET_CONFIGURATION_create ();
153 GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_load (p->cfg, cfgname));
154 if (GNUNET_CONFIGURATION_have_value (p->cfg,"PATHS", "SERVICEHOME"))
155 GNUNET_CONFIGURATION_get_value_string (p->cfg, "PATHS", "SERVICEHOME", &p->servicehome);
156 if (NULL != p->servicehome)
157 GNUNET_DISK_directory_remove (p->servicehome);
158 p->arm_proc = GNUNET_OS_start_process (NULL, NULL, "gnunet-service-arm",
159 "gnunet-service-arm",
160 "-c", cfgname, NULL);
166 p->th = GNUNET_TRANSPORT_connect(p->cfg, NULL,
171 GNUNET_assert (p->th != NULL);
176 GNUNET_TRANSPORT_TESTING_stop_peer (struct PeerContext * p)
179 GNUNET_TRANSPORT_disconnect(p->th);
181 if (NULL != p->arm_proc)
183 if (0 != GNUNET_OS_process_kill (p->arm_proc, SIGTERM))
184 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
185 GNUNET_OS_process_wait (p->arm_proc);
186 GNUNET_OS_process_close (p->arm_proc);
189 GNUNET_CONFIGURATION_destroy (p->cfg);
190 if (p->servicehome != NULL)
192 GNUNET_DISK_directory_remove (p->servicehome);
193 GNUNET_free(p->servicehome);
199 GNUNET_TRANSPORT_TESTING_connect_peers (struct PeerContext * p1,
200 struct PeerContext * p2,
201 GNUNET_TRANSPORT_TESTING_connect_cb * cb,
204 struct ConnectingContext * cc = GNUNET_malloc (sizeof (struct ConnectingContext));
206 GNUNET_assert (p1 != NULL);
207 GNUNET_assert (p1->th != NULL);
209 GNUNET_assert (p2 != NULL);
210 GNUNET_assert (p2->th != NULL);
214 GNUNET_TRANSPORT_get_hello (p1->th, &exchange_hello, cc);
215 GNUNET_TRANSPORT_get_hello (p2->th, &exchange_hello_last, cc);
217 cc->tct = GNUNET_SCHEDULER_add_now (&try_connect, cc);
222 /* end of transport_testing.h */