diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/testbed/gnunet-service-testbed.c | 39 | ||||
-rw-r--r-- | src/testbed/gnunet-service-testbed.c.new | 3019 | ||||
-rw-r--r-- | src/testbed/testbed.h | 24 | ||||
-rw-r--r-- | src/testbed/testbed_api.c | 48 | ||||
-rw-r--r-- | src/testbed/testbed_api.c.new | 2222 |
5 files changed, 5336 insertions, 16 deletions
diff --git a/src/testbed/gnunet-service-testbed.c b/src/testbed/gnunet-service-testbed.c index 1bd22958e..05d296fee 100644 --- a/src/testbed/gnunet-service-testbed.c +++ b/src/testbed/gnunet-service-testbed.c | |||
@@ -180,8 +180,7 @@ struct Slave | |||
180 | struct GNUNET_TESTBED_Controller *controller; | 180 | struct GNUNET_TESTBED_Controller *controller; |
181 | 181 | ||
182 | /** | 182 | /** |
183 | * The configuration of the slave. Will be NULL for slave which we didn't | 183 | * The configuration of the slave. Cannot be NULL |
184 | * directly start | ||
185 | */ | 184 | */ |
186 | struct GNUNET_CONFIGURATION_Handle *cfg; | 185 | struct GNUNET_CONFIGURATION_Handle *cfg; |
187 | 186 | ||
@@ -1546,7 +1545,7 @@ handle_link_controllers (void *cls, struct GNUNET_SERVER_Client *client, | |||
1546 | GNUNET_TESTBED_controller_connect (cfg, host_list[slave->host_id], | 1545 | GNUNET_TESTBED_controller_connect (cfg, host_list[slave->host_id], |
1547 | master_context->event_mask, | 1546 | master_context->event_mask, |
1548 | &slave_event_callback, slave); | 1547 | &slave_event_callback, slave); |
1549 | GNUNET_CONFIGURATION_destroy (cfg); | 1548 | slave->cfg = cfg; |
1550 | if (NULL != slave->controller) | 1549 | if (NULL != slave->controller) |
1551 | send_operation_success_msg (client, GNUNET_ntohll (msg->operation_id)); | 1550 | send_operation_success_msg (client, GNUNET_ntohll (msg->operation_id)); |
1552 | else | 1551 | else |
@@ -2474,23 +2473,31 @@ handle_overlay_connect (void *cls, struct GNUNET_SERVER_Client *client, | |||
2474 | if ((peer2_host_id >= slave_list_size) | 2473 | if ((peer2_host_id >= slave_list_size) |
2475 | || (NULL ==slave_list[peer2_host_id])) | 2474 | || (NULL ==slave_list[peer2_host_id])) |
2476 | { | 2475 | { |
2477 | /* Peer2's host is located above us or lateral to us.. | 2476 | struct GNUNET_TESTBED_NeedControllerConfig *reply; |
2478 | FIXME: should try to ask for the configuration of peer2's host by | 2477 | |
2479 | sending a reply asking for its configuration */ | ||
2480 | GNUNET_break (0); | ||
2481 | GNUNET_SERVER_client_drop (client); | ||
2482 | GNUNET_free (occ); | 2478 | GNUNET_free (occ); |
2483 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | 2479 | reply = GNUNET_malloc (sizeof (struct |
2480 | GNUNET_TESTBED_NeedControllerConfig)); | ||
2481 | reply->header.size = htons (sizeof (struct | ||
2482 | GNUNET_TESTBED_NeedControllerConfig)); | ||
2483 | reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_NEEDCONTROLLERCONFIG); | ||
2484 | reply->controller_host_id = msg->peer2_host_id; | ||
2485 | reply->operation_id = msg->operation_id; | ||
2486 | queue_message (client, &reply->header); | ||
2487 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2484 | return; | 2488 | return; |
2485 | } | 2489 | } |
2486 | occ->peer2_controller = slave_list[peer2_host_id]->controller; | 2490 | else |
2487 | if (NULL == occ->peer2_controller) | ||
2488 | { | 2491 | { |
2489 | GNUNET_break (0); | 2492 | occ->peer2_controller = slave_list[peer2_host_id]->controller; |
2490 | GNUNET_SERVER_client_drop (client); | 2493 | if (NULL == occ->peer2_controller) |
2491 | GNUNET_free (occ); | 2494 | { |
2492 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | 2495 | GNUNET_break (0); /* What's going on? */ |
2493 | return; | 2496 | GNUNET_SERVER_client_drop (client); |
2497 | GNUNET_free (occ); | ||
2498 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
2499 | return; | ||
2500 | } | ||
2494 | } | 2501 | } |
2495 | } | 2502 | } |
2496 | else | 2503 | else |
diff --git a/src/testbed/gnunet-service-testbed.c.new b/src/testbed/gnunet-service-testbed.c.new new file mode 100644 index 000000000..655885a29 --- /dev/null +++ b/src/testbed/gnunet-service-testbed.c.new | |||
@@ -0,0 +1,3019 @@ | |||
1 | /* | ||
2 | This file is part of GNUnet. | ||
3 | (C) 2012 Christian Grothoff (and other contributing authors) | ||
4 | |||
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. | ||
9 | |||
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. | ||
14 | |||
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. | ||
19 | */ | ||
20 | |||
21 | /** | ||
22 | * @file testbed/gnunet-service-testbed.c | ||
23 | * @brief implementation of the TESTBED service | ||
24 | * @author Sree Harsha Totakura | ||
25 | */ | ||
26 | |||
27 | #include "platform.h" | ||
28 | #include "gnunet_service_lib.h" | ||
29 | #include "gnunet_server_lib.h" | ||
30 | #include "gnunet_transport_service.h" | ||
31 | #include "gnunet_core_service.h" | ||
32 | #include "gnunet_hello_lib.h" | ||
33 | #include <zlib.h> | ||
34 | |||
35 | #include "gnunet_testbed_service.h" | ||
36 | #include "testbed.h" | ||
37 | #include "testbed_api.h" | ||
38 | #include "testbed_api_hosts.h" | ||
39 | #include "gnunet_testing_lib-new.h" | ||
40 | |||
41 | /** | ||
42 | * Generic logging | ||
43 | */ | ||
44 | #define LOG(kind,...) \ | ||
45 | GNUNET_log (kind, __VA_ARGS__) | ||
46 | |||
47 | /** | ||
48 | * Debug logging | ||
49 | */ | ||
50 | #define LOG_DEBUG(...) \ | ||
51 | LOG (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__) | ||
52 | |||
53 | /** | ||
54 | * By how much should the arrays lists grow | ||
55 | */ | ||
56 | #define LIST_GROW_STEP 10 | ||
57 | |||
58 | /** | ||
59 | * Default timeout for operations which may take some time | ||
60 | */ | ||
61 | #define TIMEOUT GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 30) | ||
62 | |||
63 | /** | ||
64 | * The main context information associated with the client which started us | ||
65 | */ | ||
66 | struct Context | ||
67 | { | ||
68 | /** | ||
69 | * The client handle associated with this context | ||
70 | */ | ||
71 | struct GNUNET_SERVER_Client *client; | ||
72 | |||
73 | /** | ||
74 | * The network address of the master controller | ||
75 | */ | ||
76 | char *master_ip; | ||
77 | |||
78 | /** | ||
79 | * The TESTING system handle for starting peers locally | ||
80 | */ | ||
81 | struct GNUNET_TESTING_System *system; | ||
82 | |||
83 | /** | ||
84 | * Event mask of event to be responded in this context | ||
85 | */ | ||
86 | uint64_t event_mask; | ||
87 | |||
88 | /** | ||
89 | * Our host id according to this context | ||
90 | */ | ||
91 | uint32_t host_id; | ||
92 | }; | ||
93 | |||
94 | |||
95 | /** | ||
96 | * The message queue for sending messages to clients | ||
97 | */ | ||
98 | struct MessageQueue | ||
99 | { | ||
100 | /** | ||
101 | * The message to be sent | ||
102 | */ | ||
103 | struct GNUNET_MessageHeader *msg; | ||
104 | |||
105 | /** | ||
106 | * The client to send the message to | ||
107 | */ | ||
108 | struct GNUNET_SERVER_Client *client; | ||
109 | |||
110 | /** | ||
111 | * next pointer for DLL | ||
112 | */ | ||
113 | struct MessageQueue *next; | ||
114 | |||
115 | /** | ||
116 | * prev pointer for DLL | ||
117 | */ | ||
118 | struct MessageQueue *prev; | ||
119 | }; | ||
120 | |||
121 | |||
122 | /** | ||
123 | * The structure for identifying a shared service | ||
124 | */ | ||
125 | struct SharedService | ||
126 | { | ||
127 | /** | ||
128 | * The name of the shared service | ||
129 | */ | ||
130 | char *name; | ||
131 | |||
132 | /** | ||
133 | * Number of shared peers per instance of the shared service | ||
134 | */ | ||
135 | uint32_t num_shared; | ||
136 | |||
137 | /** | ||
138 | * Number of peers currently sharing the service | ||
139 | */ | ||
140 | uint32_t num_sharing; | ||
141 | }; | ||
142 | |||
143 | |||
144 | /** | ||
145 | * A routing entry | ||
146 | */ | ||
147 | struct Route | ||
148 | { | ||
149 | /** | ||
150 | * destination host | ||
151 | */ | ||
152 | uint32_t dest; | ||
153 | |||
154 | /** | ||
155 | * The destination host is reachable thru | ||
156 | */ | ||
157 | uint32_t thru; | ||
158 | }; | ||
159 | |||
160 | |||
161 | /** | ||
162 | * Context information used while linking controllers | ||
163 | */ | ||
164 | struct LinkControllersContext; | ||
165 | |||
166 | |||
167 | /** | ||
168 | * Structure representing a connected(directly-linked) controller | ||
169 | */ | ||
170 | struct Slave | ||
171 | { | ||
172 | /** | ||
173 | * The controller process handle if we had started the controller | ||
174 | */ | ||
175 | struct GNUNET_TESTBED_ControllerProc *controller_proc; | ||
176 | |||
177 | /** | ||
178 | * The controller handle | ||
179 | */ | ||
180 | struct GNUNET_TESTBED_Controller *controller; | ||
181 | |||
182 | /** | ||
183 | * The configuration of the slave. Cannot be NULL | ||
184 | */ | ||
185 | struct GNUNET_CONFIGURATION_Handle *cfg; | ||
186 | |||
187 | /** | ||
188 | * handle to lcc which is associated with this slave startup. Should be set to | ||
189 | * NULL when the slave has successfully started up | ||
190 | */ | ||
191 | struct LinkControllersContext *lcc; | ||
192 | |||
193 | /** | ||
194 | * The id of the host this controller is running on | ||
195 | */ | ||
196 | uint32_t host_id; | ||
197 | }; | ||
198 | |||
199 | |||
200 | /** | ||
201 | * States of LCFContext | ||
202 | */ | ||
203 | enum LCFContextState | ||
204 | { | ||
205 | /** | ||
206 | * The Context has been initialized; Nothing has been done on it | ||
207 | */ | ||
208 | INIT, | ||
209 | |||
210 | /** | ||
211 | * Delegated host has been registered at the forwarding controller | ||
212 | */ | ||
213 | DELEGATED_HOST_REGISTERED, | ||
214 | |||
215 | /** | ||
216 | * The slave host has been registred at the forwarding controller | ||
217 | */ | ||
218 | SLAVE_HOST_REGISTERED, | ||
219 | |||
220 | /** | ||
221 | * The context has been finished (may have error) | ||
222 | */ | ||
223 | FINISHED | ||
224 | }; | ||
225 | |||
226 | |||
227 | /** | ||
228 | * Link controllers request forwarding context | ||
229 | */ | ||
230 | struct LCFContext | ||
231 | { | ||
232 | /** | ||
233 | * The gateway which will pass the link message to delegated host | ||
234 | */ | ||
235 | struct Slave *gateway; | ||
236 | |||
237 | /** | ||
238 | * The controller link message that has to be forwarded to | ||
239 | */ | ||
240 | struct GNUNET_TESTBED_ControllerLinkMessage *msg; | ||
241 | |||
242 | /** | ||
243 | * The client which has asked to perform this operation | ||
244 | */ | ||
245 | struct GNUNET_SERVER_Client *client; | ||
246 | |||
247 | /** | ||
248 | * The host registration handle while registered hosts in this context | ||
249 | */ | ||
250 | struct GNUNET_TESTBED_HostRegistrationHandle *rhandle; | ||
251 | |||
252 | /** | ||
253 | * The id of the operation which created this context | ||
254 | */ | ||
255 | uint64_t operation_id; | ||
256 | |||
257 | /** | ||
258 | * The state of this context | ||
259 | */ | ||
260 | enum LCFContextState state; | ||
261 | |||
262 | /** | ||
263 | * The delegated host | ||
264 | */ | ||
265 | uint32_t delegated_host_id; | ||
266 | |||
267 | /** | ||
268 | * The slave host | ||
269 | */ | ||
270 | uint32_t slave_host_id; | ||
271 | |||
272 | }; | ||
273 | |||
274 | |||
275 | /** | ||
276 | * Structure of a queue entry in LCFContext request queue | ||
277 | */ | ||
278 | struct LCFContextQueue | ||
279 | { | ||
280 | /** | ||
281 | * The LCFContext | ||
282 | */ | ||
283 | struct LCFContext *lcf; | ||
284 | |||
285 | /** | ||
286 | * Head prt for DLL | ||
287 | */ | ||
288 | struct LCFContextQueue *next; | ||
289 | |||
290 | /** | ||
291 | * Tail ptr for DLL | ||
292 | */ | ||
293 | struct LCFContextQueue *prev; | ||
294 | }; | ||
295 | |||
296 | |||
297 | /** | ||
298 | * A peer | ||
299 | */ | ||
300 | struct Peer | ||
301 | { | ||
302 | union | ||
303 | { | ||
304 | struct | ||
305 | { | ||
306 | /** | ||
307 | * The peer handle from testing API | ||
308 | */ | ||
309 | struct GNUNET_TESTING_Peer *peer; | ||
310 | |||
311 | /** | ||
312 | * The modified (by GNUNET_TESTING_peer_configure) configuration this | ||
313 | * peer is configured with | ||
314 | */ | ||
315 | struct GNUNET_CONFIGURATION_Handle *cfg; | ||
316 | |||
317 | /** | ||
318 | * Is the peer running | ||
319 | */ | ||
320 | int is_running; | ||
321 | |||
322 | } local; | ||
323 | |||
324 | struct | ||
325 | { | ||
326 | /** | ||
327 | * The controller this peer is started through | ||
328 | */ | ||
329 | struct GNUNET_TESTBED_Controller *controller; | ||
330 | |||
331 | } remote; | ||
332 | |||
333 | } details; | ||
334 | |||
335 | /** | ||
336 | * Is this peer locally created? | ||
337 | */ | ||
338 | int is_remote; | ||
339 | |||
340 | /** | ||
341 | * Our local reference id for this peer | ||
342 | */ | ||
343 | uint32_t id; | ||
344 | |||
345 | }; | ||
346 | |||
347 | |||
348 | /** | ||
349 | * Context information for connecting 2 peers in overlay | ||
350 | */ | ||
351 | struct OverlayConnectContext | ||
352 | { | ||
353 | /** | ||
354 | * The client which has requested for overlay connection | ||
355 | */ | ||
356 | struct GNUNET_SERVER_Client *client; | ||
357 | |||
358 | /** | ||
359 | * the peer which has to connect to the other peer | ||
360 | */ | ||
361 | struct Peer *peer; | ||
362 | |||
363 | /** | ||
364 | * Transport handle of the first peer to get its HELLO | ||
365 | */ | ||
366 | struct GNUNET_TRANSPORT_Handle *p1th; | ||
367 | |||
368 | /** | ||
369 | * Transport handle of other peer to offer first peer's HELLO | ||
370 | */ | ||
371 | struct GNUNET_TRANSPORT_Handle *p2th; | ||
372 | |||
373 | /** | ||
374 | * Core handles of the first peer; used to notify when second peer connects to it | ||
375 | */ | ||
376 | struct GNUNET_CORE_Handle *ch; | ||
377 | |||
378 | /** | ||
379 | * HELLO of the other peer | ||
380 | */ | ||
381 | struct GNUNET_MessageHeader *hello; | ||
382 | |||
383 | /** | ||
384 | * Get hello handle to acquire HELLO of first peer | ||
385 | */ | ||
386 | struct GNUNET_TRANSPORT_GetHelloHandle *ghh; | ||
387 | |||
388 | /** | ||
389 | * The error message we send if this overlay connect operation has timed out | ||
390 | */ | ||
391 | char *emsg; | ||
392 | |||
393 | /** | ||
394 | * Operation context for suboperations | ||
395 | */ | ||
396 | struct OperationContext *opc; | ||
397 | |||
398 | /** | ||
399 | * Controller of peer 2; NULL if the peer is local | ||
400 | */ | ||
401 | struct GNUNET_TESTBED_Controller *peer2_controller; | ||
402 | |||
403 | /** | ||
404 | * The peer identity of the first peer | ||
405 | */ | ||
406 | struct GNUNET_PeerIdentity peer_identity; | ||
407 | |||
408 | /** | ||
409 | * The peer identity of the other peer | ||
410 | */ | ||
411 | struct GNUNET_PeerIdentity other_peer_identity; | ||
412 | |||
413 | /** | ||
414 | * The id of the operation responsible for creating this context | ||
415 | */ | ||
416 | uint64_t op_id; | ||
417 | |||
418 | /** | ||
419 | * The id of the task for sending HELLO of peer 2 to peer 1 and ask peer 1 to | ||
420 | * connect to peer 2 | ||
421 | */ | ||
422 | GNUNET_SCHEDULER_TaskIdentifier send_hello_task; | ||
423 | |||
424 | /** | ||
425 | * The id of the overlay connect timeout task | ||
426 | */ | ||
427 | GNUNET_SCHEDULER_TaskIdentifier timeout_task; | ||
428 | |||
429 | /** | ||
430 | * The id of peer A | ||
431 | */ | ||
432 | uint32_t peer_id; | ||
433 | |||
434 | /** | ||
435 | * The id of peer B | ||
436 | */ | ||
437 | uint32_t other_peer_id; | ||
438 | |||
439 | /** | ||
440 | * Number of times we tried to send hello; used to increase delay in offering | ||
441 | * hellos | ||
442 | */ | ||
443 | uint16_t retries; | ||
444 | }; | ||
445 | |||
446 | |||
447 | /** | ||
448 | * Context information to used during operations which forward the overlay | ||
449 | * connect message | ||
450 | */ | ||
451 | struct ForwardedOverlayConnectContext | ||
452 | { | ||
453 | /** | ||
454 | * A copy of the original message which resulted in creation of this context | ||
455 | */ | ||
456 | struct GNUNET_TESTBED_OverlayConnectMessage *oc_msg; | ||
457 | |||
458 | /** | ||
459 | * The gateway controller to which this operation is forwarded to | ||
460 | */ | ||
461 | struct GNUNET_TESTBED_Controller *gateway; | ||
462 | }; | ||
463 | |||
464 | |||
465 | /** | ||
466 | * Context information for RequestOverlayConnect | ||
467 | * operations. RequestOverlayConnect is used when peers A, B reside on different | ||
468 | * hosts and the host controller for peer B is asked by the host controller of | ||
469 | * peer A to make peer B connect to peer A | ||
470 | */ | ||
471 | struct RequestOverlayConnectContext | ||
472 | { | ||
473 | /** | ||
474 | * The transport handle of peer B | ||
475 | */ | ||
476 | struct GNUNET_TRANSPORT_Handle *th; | ||
477 | |||
478 | /** | ||
479 | * Peer A's HELLO | ||
480 | */ | ||
481 | struct GNUNET_MessageHeader *hello; | ||
482 | |||
483 | /** | ||
484 | * The peer identity of peer A | ||
485 | */ | ||
486 | struct GNUNET_PeerIdentity a_id; | ||
487 | |||
488 | /** | ||
489 | * Task for offering HELLO of A to B and doing try_connect | ||
490 | */ | ||
491 | GNUNET_SCHEDULER_TaskIdentifier attempt_connect_task_id; | ||
492 | |||
493 | /** | ||
494 | * Task to timeout RequestOverlayConnect | ||
495 | */ | ||
496 | GNUNET_SCHEDULER_TaskIdentifier timeout_rocc_task_id; | ||
497 | |||
498 | /** | ||
499 | * Number of times we tried to send hello; used to increase delay in offering | ||
500 | * hellos | ||
501 | */ | ||
502 | uint16_t retries; | ||
503 | |||
504 | }; | ||
505 | |||
506 | |||
507 | /** | ||
508 | * Context information for operations forwarded to subcontrollers | ||
509 | */ | ||
510 | struct ForwardedOperationContext | ||
511 | { | ||
512 | /** | ||
513 | * The generated operation context | ||
514 | */ | ||
515 | struct OperationContext *opc; | ||
516 | |||
517 | /** | ||
518 | * The client to which we have to reply | ||
519 | */ | ||
520 | struct GNUNET_SERVER_Client *client; | ||
521 | |||
522 | /** | ||
523 | * Closure pointer | ||
524 | */ | ||
525 | void *cls; | ||
526 | |||
527 | /** | ||
528 | * Task ID for the timeout task | ||
529 | */ | ||
530 | GNUNET_SCHEDULER_TaskIdentifier timeout_task; | ||
531 | |||
532 | /** | ||
533 | * The id of the operation that has been forwarded | ||
534 | */ | ||
535 | uint64_t operation_id; | ||
536 | |||
537 | }; | ||
538 | |||
539 | |||
540 | /** | ||
541 | * Context information used while linking controllers | ||
542 | */ | ||
543 | struct LinkControllersContext | ||
544 | { | ||
545 | /** | ||
546 | * The client which initiated the link controller operation | ||
547 | */ | ||
548 | struct GNUNET_SERVER_Client *client; | ||
549 | |||
550 | /** | ||
551 | * The ID of the operation | ||
552 | */ | ||
553 | uint64_t operation_id; | ||
554 | |||
555 | }; | ||
556 | |||
557 | |||
558 | |||
559 | /** | ||
560 | * The master context; generated with the first INIT message | ||
561 | */ | ||
562 | static struct Context *master_context; | ||
563 | |||
564 | /** | ||
565 | * Our hostname; we give this to all the peers we start | ||
566 | */ | ||
567 | static char *hostname; | ||
568 | |||
569 | /***********/ | ||
570 | /* Handles */ | ||
571 | /***********/ | ||
572 | |||
573 | /** | ||
574 | * Current Transmit Handle; NULL if no notify transmit exists currently | ||
575 | */ | ||
576 | static struct GNUNET_SERVER_TransmitHandle *transmit_handle; | ||
577 | |||
578 | /****************/ | ||
579 | /* Lists & Maps */ | ||
580 | /****************/ | ||
581 | |||
582 | /** | ||
583 | * The head for the LCF queue | ||
584 | */ | ||
585 | static struct LCFContextQueue *lcfq_head; | ||
586 | |||
587 | /** | ||
588 | * The tail for the LCF queue | ||
589 | */ | ||
590 | static struct LCFContextQueue *lcfq_tail; | ||
591 | |||
592 | /** | ||
593 | * The message queue head | ||
594 | */ | ||
595 | static struct MessageQueue *mq_head; | ||
596 | |||
597 | /** | ||
598 | * The message queue tail | ||
599 | */ | ||
600 | static struct MessageQueue *mq_tail; | ||
601 | |||
602 | /** | ||
603 | * Array of host list | ||
604 | */ | ||
605 | static struct GNUNET_TESTBED_Host **host_list; | ||
606 | |||
607 | /** | ||
608 | * A list of routes | ||
609 | */ | ||
610 | static struct Route **route_list; | ||
611 | |||
612 | /** | ||
613 | * A list of directly linked neighbours | ||
614 | */ | ||
615 | static struct Slave **slave_list; | ||
616 | |||
617 | /** | ||
618 | * A list of peers we know about | ||
619 | */ | ||
620 | static struct Peer **peer_list; | ||
621 | |||
622 | /** | ||
623 | * The hashmap of shared services | ||
624 | */ | ||
625 | static struct GNUNET_CONTAINER_MultiHashMap *ss_map; | ||
626 | |||
627 | /** | ||
628 | * The size of the host list | ||
629 | */ | ||
630 | static uint32_t host_list_size; | ||
631 | |||
632 | /** | ||
633 | * The size of the route list | ||
634 | */ | ||
635 | static uint32_t route_list_size; | ||
636 | |||
637 | /** | ||
638 | * The size of directly linked neighbours list | ||
639 | */ | ||
640 | static uint32_t slave_list_size; | ||
641 | |||
642 | /** | ||
643 | * The size of the peer list | ||
644 | */ | ||
645 | static uint32_t peer_list_size; | ||
646 | |||
647 | /*********/ | ||
648 | /* Tasks */ | ||
649 | /*********/ | ||
650 | |||
651 | /** | ||
652 | * The lcf_task handle | ||
653 | */ | ||
654 | static GNUNET_SCHEDULER_TaskIdentifier lcf_proc_task_id; | ||
655 | |||
656 | /** | ||
657 | * The shutdown task handle | ||
658 | */ | ||
659 | static GNUNET_SCHEDULER_TaskIdentifier shutdown_task_id; | ||
660 | |||
661 | |||
662 | /** | ||
663 | * Function called to notify a client about the connection begin ready to queue | ||
664 | * more data. "buf" will be NULL and "size" zero if the connection was closed | ||
665 | * for writing in the meantime. | ||
666 | * | ||
667 | * @param cls NULL | ||
668 | * @param size number of bytes available in buf | ||
669 | * @param buf where the callee should write the message | ||
670 | * @return number of bytes written to buf | ||
671 | */ | ||
672 | static size_t | ||
673 | transmit_ready_notify (void *cls, size_t size, void *buf) | ||
674 | { | ||
675 | struct MessageQueue *mq_entry; | ||
676 | |||
677 | transmit_handle = NULL; | ||
678 | mq_entry = mq_head; | ||
679 | GNUNET_assert (NULL != mq_entry); | ||
680 | if (0 == size) | ||
681 | return 0; | ||
682 | GNUNET_assert (ntohs (mq_entry->msg->size) <= size); | ||
683 | size = ntohs (mq_entry->msg->size); | ||
684 | memcpy (buf, mq_entry->msg, size); | ||
685 | GNUNET_free (mq_entry->msg); | ||
686 | GNUNET_CONTAINER_DLL_remove (mq_head, mq_tail, mq_entry); | ||
687 | GNUNET_free (mq_entry); | ||
688 | mq_entry = mq_head; | ||
689 | if (NULL != mq_entry) | ||
690 | transmit_handle = | ||
691 | GNUNET_SERVER_notify_transmit_ready (mq_entry->client, | ||
692 | ntohs (mq_entry->msg->size), | ||
693 | GNUNET_TIME_UNIT_FOREVER_REL, | ||
694 | &transmit_ready_notify, NULL); | ||
695 | return size; | ||
696 | } | ||
697 | |||
698 | |||
699 | /** | ||
700 | * Queues a message in send queue for sending to the service | ||
701 | * | ||
702 | * @param client the client to whom the queued message has to be sent | ||
703 | * @param msg the message to queue | ||
704 | */ | ||
705 | static void | ||
706 | queue_message (struct GNUNET_SERVER_Client *client, | ||
707 | struct GNUNET_MessageHeader *msg) | ||
708 | { | ||
709 | struct MessageQueue *mq_entry; | ||
710 | uint16_t type; | ||
711 | uint16_t size; | ||
712 | |||
713 | type = ntohs (msg->type); | ||
714 | size = ntohs (msg->size); | ||
715 | GNUNET_assert ((GNUNET_MESSAGE_TYPE_TESTBED_INIT <= type) && | ||
716 | (GNUNET_MESSAGE_TYPE_TESTBED_MAX > type)); | ||
717 | mq_entry = GNUNET_malloc (sizeof (struct MessageQueue)); | ||
718 | mq_entry->msg = msg; | ||
719 | mq_entry->client = client; | ||
720 | LOG_DEBUG ("Queueing message of type %u, size %u for sending\n", type, | ||
721 | ntohs (msg->size)); | ||
722 | GNUNET_CONTAINER_DLL_insert_tail (mq_head, mq_tail, mq_entry); | ||
723 | if (NULL == transmit_handle) | ||
724 | transmit_handle = | ||
725 | GNUNET_SERVER_notify_transmit_ready (client, size, | ||
726 | GNUNET_TIME_UNIT_FOREVER_REL, | ||
727 | &transmit_ready_notify, NULL); | ||
728 | } | ||
729 | |||
730 | |||
731 | /** | ||
732 | * Similar to GNUNET_realloc; however clears tail part of newly allocated memory | ||
733 | * | ||
734 | * @param ptr the memory block to realloc | ||
735 | * @param size the size of ptr | ||
736 | * @param new_size the size to which ptr has to be realloc'ed | ||
737 | * @return the newly reallocated memory block | ||
738 | */ | ||
739 | static void * | ||
740 | TESTBED_realloc (void *ptr, size_t size, size_t new_size) | ||
741 | { | ||
742 | ptr = GNUNET_realloc (ptr, new_size); | ||
743 | if (new_size > size) | ||
744 | (void) memset (ptr + size, 0, new_size - size); | ||
745 | return ptr; | ||
746 | } | ||
747 | |||
748 | |||
749 | /** | ||
750 | * Function to add a host to the current list of known hosts | ||
751 | * | ||
752 | * @param host the host to add | ||
753 | * @return GNUNET_OK on success; GNUNET_SYSERR on failure due to host-id | ||
754 | * already in use | ||
755 | */ | ||
756 | static int | ||
757 | host_list_add (struct GNUNET_TESTBED_Host *host) | ||
758 | { | ||
759 | uint32_t host_id; | ||
760 | uint32_t orig_size; | ||
761 | |||
762 | host_id = GNUNET_TESTBED_host_get_id_ (host); | ||
763 | orig_size = host_list_size; | ||
764 | if (host_list_size <= host_id) | ||
765 | { | ||
766 | while (host_list_size <= host_id) | ||
767 | host_list_size += LIST_GROW_STEP; | ||
768 | host_list = | ||
769 | TESTBED_realloc (host_list, | ||
770 | sizeof (struct GNUNET_TESTBED_Host *) * orig_size, | ||
771 | sizeof (struct GNUNET_TESTBED_Host *) | ||
772 | * host_list_size); | ||
773 | } | ||
774 | if (NULL != host_list[host_id]) | ||
775 | { | ||
776 | LOG_DEBUG ("A host with id: %u already exists\n", host_id); | ||
777 | return GNUNET_SYSERR; | ||
778 | } | ||
779 | host_list[host_id] = host; | ||
780 | return GNUNET_OK; | ||
781 | } | ||
782 | |||
783 | |||
784 | /** | ||
785 | * Adds a route to the route list | ||
786 | * | ||
787 | * @param route the route to add | ||
788 | */ | ||
789 | static void | ||
790 | route_list_add (struct Route *route) | ||
791 | { | ||
792 | uint32_t orig_size; | ||
793 | |||
794 | orig_size = route_list_size; | ||
795 | if (route->dest >= route_list_size) | ||
796 | { | ||
797 | while (route->dest >= route_list_size) | ||
798 | route_list_size += LIST_GROW_STEP; | ||
799 | route_list = | ||
800 | TESTBED_realloc (route_list, | ||
801 | sizeof (struct Route *) * orig_size, | ||
802 | sizeof (struct Route *) * route_list_size); | ||
803 | } | ||
804 | GNUNET_assert (NULL == route_list[route->dest]); | ||
805 | route_list[route->dest] = route; | ||
806 | } | ||
807 | |||
808 | |||
809 | /** | ||
810 | * Adds a slave to the slave array | ||
811 | * | ||
812 | * @param slave the slave controller to add | ||
813 | */ | ||
814 | static void | ||
815 | slave_list_add (struct Slave *slave) | ||
816 | { | ||
817 | if (slave->host_id >= slave_list_size) | ||
818 | { | ||
819 | slave_list = | ||
820 | TESTBED_realloc (slave_list, sizeof (struct Slave *) * slave_list_size, | ||
821 | sizeof (struct Slave *) * (slave_list_size + | ||
822 | LIST_GROW_STEP)); | ||
823 | slave_list_size += LIST_GROW_STEP; | ||
824 | } | ||
825 | GNUNET_assert (NULL == slave_list[slave->host_id]); | ||
826 | slave_list[slave->host_id] = slave; | ||
827 | } | ||
828 | |||
829 | |||
830 | /** | ||
831 | * Adds a peer to the peer array | ||
832 | * | ||
833 | * @param peer the peer to add | ||
834 | */ | ||
835 | static void | ||
836 | peer_list_add (struct Peer *peer) | ||
837 | { | ||
838 | uint32_t orig_size; | ||
839 | |||
840 | orig_size = peer_list_size; | ||
841 | if (peer->id >= peer_list_size) | ||
842 | { | ||
843 | while (peer->id >= peer_list_size) | ||
844 | peer_list_size += LIST_GROW_STEP; | ||
845 | peer_list = | ||
846 | TESTBED_realloc (peer_list, sizeof (struct Peer *) * orig_size, | ||
847 | sizeof (struct Peer *) * peer_list_size); | ||
848 | } | ||
849 | GNUNET_assert (NULL == peer_list[peer->id]); | ||
850 | peer_list[peer->id] = peer; | ||
851 | } | ||
852 | |||
853 | |||
854 | /** | ||
855 | * Removes a the give peer from the peer array | ||
856 | * | ||
857 | * @param peer the peer to be removed | ||
858 | */ | ||
859 | static void | ||
860 | peer_list_remove (struct Peer *peer) | ||
861 | { | ||
862 | uint32_t id; | ||
863 | uint32_t orig_size; | ||
864 | |||
865 | peer_list[peer->id] = NULL; | ||
866 | orig_size = peer_list_size; | ||
867 | while (peer_list_size >= LIST_GROW_STEP) | ||
868 | { | ||
869 | for (id = peer_list_size - 1; | ||
870 | (id >= peer_list_size - LIST_GROW_STEP) && (id != UINT32_MAX); id--) | ||
871 | if (NULL != peer_list[id]) | ||
872 | break; | ||
873 | if (id != ((peer_list_size - LIST_GROW_STEP) - 1)) | ||
874 | break; | ||
875 | peer_list_size -= LIST_GROW_STEP; | ||
876 | } | ||
877 | if (orig_size == peer_list_size) | ||
878 | return; | ||
879 | peer_list = | ||
880 | GNUNET_realloc (peer_list, sizeof (struct Peer *) * peer_list_size); | ||
881 | } | ||
882 | |||
883 | |||
884 | /** | ||
885 | * Finds the route with directly connected host as destination through which | ||
886 | * the destination host can be reached | ||
887 | * | ||
888 | * @param host_id the id of the destination host | ||
889 | * @return the route with directly connected destination host; NULL if no route | ||
890 | * is found | ||
891 | */ | ||
892 | static struct Route * | ||
893 | find_dest_route (uint32_t host_id) | ||
894 | { | ||
895 | struct Route *route; | ||
896 | |||
897 | while (NULL != (route = route_list[host_id])) | ||
898 | { | ||
899 | if (route->thru == master_context->host_id) | ||
900 | break; | ||
901 | host_id = route->thru; | ||
902 | } | ||
903 | return route; | ||
904 | } | ||
905 | |||
906 | |||
907 | /** | ||
908 | * Routes message to a host given its host_id | ||
909 | * | ||
910 | * @param host_id the id of the destination host | ||
911 | * @param msg the message to be routed | ||
912 | */ | ||
913 | static void | ||
914 | route_message (uint32_t host_id, const struct GNUNET_MessageHeader *msg) | ||
915 | { | ||
916 | GNUNET_break (0); | ||
917 | } | ||
918 | |||
919 | |||
920 | /** | ||
921 | * Send operation failure message to client | ||
922 | * | ||
923 | * @param client the client to which the failure message has to be sent to | ||
924 | * @param operation_id the id of the failed operation | ||
925 | * @param emsg the error message; can be NULL | ||
926 | */ | ||
927 | static void | ||
928 | send_operation_fail_msg (struct GNUNET_SERVER_Client *client, | ||
929 | uint64_t operation_id, const char *emsg) | ||
930 | { | ||
931 | struct GNUNET_TESTBED_OperationFailureEventMessage *msg; | ||
932 | uint16_t msize; | ||
933 | uint16_t emsg_len; | ||
934 | |||
935 | msize = sizeof (struct GNUNET_TESTBED_OperationFailureEventMessage); | ||
936 | emsg_len = (NULL == emsg) ? 0 : strlen (emsg) + 1; | ||
937 | msize += emsg_len; | ||
938 | msg = GNUNET_malloc (msize); | ||
939 | msg->header.size = htons (msize); | ||
940 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_OPERATIONFAILEVENT); | ||
941 | msg->event_type = htonl (GNUNET_TESTBED_ET_OPERATION_FINISHED); | ||
942 | msg->operation_id = GNUNET_htonll (operation_id); | ||
943 | if (0 != emsg_len) | ||
944 | memcpy (&msg[1], emsg, emsg_len); | ||
945 | queue_message (client, &msg->header); | ||
946 | } | ||
947 | |||
948 | |||
949 | /** | ||
950 | * Function to send generic operation success message to given client | ||
951 | * | ||
952 | * @param client the client to send the message to | ||
953 | * @param operation_id the id of the operation which was successful | ||
954 | */ | ||
955 | static void | ||
956 | send_operation_success_msg (struct GNUNET_SERVER_Client *client, | ||
957 | uint64_t operation_id) | ||
958 | { | ||
959 | struct GNUNET_TESTBED_GenericOperationSuccessEventMessage *msg; | ||
960 | uint16_t msize; | ||
961 | |||
962 | msize = sizeof (struct GNUNET_TESTBED_GenericOperationSuccessEventMessage); | ||
963 | msg = GNUNET_malloc (msize); | ||
964 | msg->header.size = htons (msize); | ||
965 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_GENERICOPSUCCESS); | ||
966 | msg->operation_id = GNUNET_htonll (operation_id); | ||
967 | msg->event_type = htonl (GNUNET_TESTBED_ET_OPERATION_FINISHED); | ||
968 | queue_message (client, &msg->header); | ||
969 | } | ||
970 | |||
971 | |||
972 | /** | ||
973 | * The Link Controller forwarding task | ||
974 | * | ||
975 | * @param cls the LCFContext | ||
976 | * @param tc the Task context from scheduler | ||
977 | */ | ||
978 | static void | ||
979 | lcf_proc_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc); | ||
980 | |||
981 | |||
982 | /** | ||
983 | * Completion callback for host registrations while forwarding Link Controller messages | ||
984 | * | ||
985 | * @param cls the LCFContext | ||
986 | * @param emsg the error message; NULL if host registration is successful | ||
987 | */ | ||
988 | static void | ||
989 | lcf_proc_cc (void *cls, const char *emsg) | ||
990 | { | ||
991 | struct LCFContext *lcf = cls; | ||
992 | |||
993 | lcf->rhandle = NULL; | ||
994 | GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == lcf_proc_task_id); | ||
995 | switch (lcf->state) | ||
996 | { | ||
997 | case INIT: | ||
998 | if (NULL != emsg) | ||
999 | goto registration_error; | ||
1000 | lcf->state = DELEGATED_HOST_REGISTERED; | ||
1001 | lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf); | ||
1002 | break; | ||
1003 | case DELEGATED_HOST_REGISTERED: | ||
1004 | if (NULL != emsg) | ||
1005 | goto registration_error; | ||
1006 | lcf->state = SLAVE_HOST_REGISTERED; | ||
1007 | lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf); | ||
1008 | break; | ||
1009 | default: | ||
1010 | GNUNET_assert (0); /* Shouldn't reach here */ | ||
1011 | } | ||
1012 | return; | ||
1013 | |||
1014 | registration_error: | ||
1015 | LOG (GNUNET_ERROR_TYPE_WARNING, "Host registration failed with message: %s\n", | ||
1016 | emsg); | ||
1017 | lcf->state = FINISHED; | ||
1018 | lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf); | ||
1019 | } | ||
1020 | |||
1021 | |||
1022 | /** | ||
1023 | * Callback to be called when forwarded link controllers operation is | ||
1024 | * successfull. We have to relay the reply msg back to the client | ||
1025 | * | ||
1026 | * @param cls ForwardedOperationContext | ||
1027 | * @param msg the peer create success message | ||
1028 | */ | ||
1029 | static void | ||
1030 | forwarded_operation_reply_relay (void *cls, | ||
1031 | const struct GNUNET_MessageHeader *msg) | ||
1032 | { | ||
1033 | struct ForwardedOperationContext *fopc = cls; | ||
1034 | struct GNUNET_MessageHeader *dup_msg; | ||
1035 | uint16_t msize; | ||
1036 | |||
1037 | msize = ntohs (msg->size); | ||
1038 | LOG_DEBUG ("Relaying message with type: %u, size: %u\n", ntohs (msg->type), | ||
1039 | msize); | ||
1040 | dup_msg = GNUNET_malloc (msize); | ||
1041 | (void) memcpy (dup_msg, msg, msize); | ||
1042 | queue_message (fopc->client, dup_msg); | ||
1043 | GNUNET_SERVER_client_drop (fopc->client); | ||
1044 | GNUNET_SCHEDULER_cancel (fopc->timeout_task); | ||
1045 | GNUNET_free (fopc); | ||
1046 | } | ||
1047 | |||
1048 | |||
1049 | /** | ||
1050 | * Task to free resources when forwarded link controllers has been timedout | ||
1051 | * | ||
1052 | * @param cls the ForwardedOperationContext | ||
1053 | * @param tc the task context from scheduler | ||
1054 | */ | ||
1055 | static void | ||
1056 | forwarded_operation_timeout (void *cls, | ||
1057 | const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
1058 | { | ||
1059 | struct ForwardedOperationContext *fopc = cls; | ||
1060 | |||
1061 | GNUNET_TESTBED_forward_operation_msg_cancel_ (fopc->opc); | ||
1062 | send_operation_fail_msg (fopc->client, fopc->operation_id, "Timeout"); | ||
1063 | GNUNET_SERVER_client_drop (fopc->client); | ||
1064 | GNUNET_free (fopc); | ||
1065 | } | ||
1066 | |||
1067 | |||
1068 | /** | ||
1069 | * The Link Controller forwarding task | ||
1070 | * | ||
1071 | * @param cls the LCFContext | ||
1072 | * @param tc the Task context from scheduler | ||
1073 | */ | ||
1074 | static void | ||
1075 | lcf_proc_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
1076 | { | ||
1077 | struct LCFContext *lcf = cls; | ||
1078 | struct LCFContextQueue *lcfq; | ||
1079 | struct ForwardedOperationContext *fopc; | ||
1080 | |||
1081 | lcf_proc_task_id = GNUNET_SCHEDULER_NO_TASK; | ||
1082 | switch (lcf->state) | ||
1083 | { | ||
1084 | case INIT: | ||
1085 | if (GNUNET_NO == | ||
1086 | GNUNET_TESTBED_is_host_registered_ (host_list[lcf->delegated_host_id], | ||
1087 | lcf->gateway->controller)) | ||
1088 | { | ||
1089 | lcf->rhandle = | ||
1090 | GNUNET_TESTBED_register_host (lcf->gateway->controller, | ||
1091 | host_list[lcf->delegated_host_id], | ||
1092 | lcf_proc_cc, lcf); | ||
1093 | } | ||
1094 | else | ||
1095 | { | ||
1096 | lcf->state = DELEGATED_HOST_REGISTERED; | ||
1097 | lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf); | ||
1098 | } | ||
1099 | break; | ||
1100 | case DELEGATED_HOST_REGISTERED: | ||
1101 | if (GNUNET_NO == | ||
1102 | GNUNET_TESTBED_is_host_registered_ (host_list[lcf->slave_host_id], | ||
1103 | lcf->gateway->controller)) | ||
1104 | { | ||
1105 | lcf->rhandle = | ||
1106 | GNUNET_TESTBED_register_host (lcf->gateway->controller, | ||
1107 | host_list[lcf->slave_host_id], | ||
1108 | lcf_proc_cc, lcf); | ||
1109 | } | ||
1110 | else | ||
1111 | { | ||
1112 | lcf->state = SLAVE_HOST_REGISTERED; | ||
1113 | lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf); | ||
1114 | } | ||
1115 | break; | ||
1116 | case SLAVE_HOST_REGISTERED: | ||
1117 | fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext)); | ||
1118 | fopc->client = lcf->client; | ||
1119 | fopc->operation_id = lcf->operation_id; | ||
1120 | fopc->opc = | ||
1121 | GNUNET_TESTBED_forward_operation_msg_ (lcf->gateway->controller, | ||
1122 | lcf->operation_id, | ||
1123 | &lcf->msg->header, | ||
1124 | &forwarded_operation_reply_relay, | ||
1125 | fopc); | ||
1126 | fopc->timeout_task = | ||
1127 | GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout, | ||
1128 | fopc); | ||
1129 | lcf->state = FINISHED; | ||
1130 | lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf); | ||
1131 | break; | ||
1132 | case FINISHED: | ||
1133 | lcfq = lcfq_head; | ||
1134 | GNUNET_assert (lcfq->lcf == lcf); | ||
1135 | GNUNET_free (lcf->msg); | ||
1136 | GNUNET_free (lcf); | ||
1137 | GNUNET_CONTAINER_DLL_remove (lcfq_head, lcfq_tail, lcfq); | ||
1138 | GNUNET_free (lcfq); | ||
1139 | if (NULL != lcfq_head) | ||
1140 | lcf_proc_task_id = | ||
1141 | GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcfq_head->lcf); | ||
1142 | } | ||
1143 | } | ||
1144 | |||
1145 | |||
1146 | /** | ||
1147 | * Callback for event from slave controllers | ||
1148 | * | ||
1149 | * @param cls struct Slave * | ||
1150 | * @param event information about the event | ||
1151 | */ | ||
1152 | static void | ||
1153 | slave_event_callback (void *cls, | ||
1154 | const struct GNUNET_TESTBED_EventInformation *event) | ||
1155 | { | ||
1156 | GNUNET_break (0); | ||
1157 | } | ||
1158 | |||
1159 | |||
1160 | /** | ||
1161 | * Callback to signal successfull startup of the controller process | ||
1162 | * | ||
1163 | * @param cls the handle to the slave whose status is to be found here | ||
1164 | * @param cfg the configuration with which the controller has been started; | ||
1165 | * NULL if status is not GNUNET_OK | ||
1166 | * @param status GNUNET_OK if the startup is successfull; GNUNET_SYSERR if not, | ||
1167 | * GNUNET_TESTBED_controller_stop() shouldn't be called in this case | ||
1168 | */ | ||
1169 | static void | ||
1170 | slave_status_callback (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg, | ||
1171 | int status) | ||
1172 | { | ||
1173 | struct Slave *slave = cls; | ||
1174 | struct LinkControllersContext *lcc; | ||
1175 | |||
1176 | lcc = slave->lcc; | ||
1177 | if (GNUNET_SYSERR == status) | ||
1178 | { | ||
1179 | slave->controller_proc = NULL; | ||
1180 | slave_list[slave->host_id] = NULL; | ||
1181 | if (NULL != slave->cfg) | ||
1182 | GNUNET_CONFIGURATION_destroy (slave->cfg); | ||
1183 | GNUNET_free (slave); | ||
1184 | slave = NULL; | ||
1185 | LOG (GNUNET_ERROR_TYPE_WARNING, "Unexpected slave shutdown\n"); | ||
1186 | GNUNET_SCHEDULER_shutdown (); /* We too shutdown */ | ||
1187 | goto clean_lcc; | ||
1188 | } | ||
1189 | slave->controller = | ||
1190 | GNUNET_TESTBED_controller_connect (cfg, host_list[slave->host_id], | ||
1191 | master_context->event_mask, | ||
1192 | &slave_event_callback, slave); | ||
1193 | if (NULL != slave->controller) | ||
1194 | { | ||
1195 | send_operation_success_msg (lcc->client, lcc->operation_id); | ||
1196 | slave->cfg = GNUNET_CONFIGURATION_dup (cfg); | ||
1197 | } | ||
1198 | else | ||
1199 | { | ||
1200 | send_operation_fail_msg (lcc->client, lcc->operation_id, | ||
1201 | "Could not connect to delegated controller"); | ||
1202 | GNUNET_TESTBED_controller_stop (slave->controller_proc); | ||
1203 | slave_list[slave->host_id] = NULL; | ||
1204 | GNUNET_free (slave); | ||
1205 | slave = NULL; | ||
1206 | } | ||
1207 | |||
1208 | clean_lcc: | ||
1209 | if (NULL != lcc) | ||
1210 | { | ||
1211 | if (NULL != lcc->client) | ||
1212 | { | ||
1213 | GNUNET_SERVER_receive_done (lcc->client, GNUNET_OK); | ||
1214 | GNUNET_SERVER_client_drop (lcc->client); | ||
1215 | lcc->client = NULL; | ||
1216 | } | ||
1217 | GNUNET_free (lcc); | ||
1218 | } | ||
1219 | if (NULL != slave) | ||
1220 | slave->lcc = NULL; | ||
1221 | } | ||
1222 | |||
1223 | |||
1224 | /** | ||
1225 | * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_INIT messages | ||
1226 | * | ||
1227 | * @param cls NULL | ||
1228 | * @param client identification of the client | ||
1229 | * @param message the actual message | ||
1230 | */ | ||
1231 | static void | ||
1232 | handle_init (void *cls, struct GNUNET_SERVER_Client *client, | ||
1233 | const struct GNUNET_MessageHeader *message) | ||
1234 | { | ||
1235 | const struct GNUNET_TESTBED_InitMessage *msg; | ||
1236 | struct GNUNET_TESTBED_Host *host; | ||
1237 | const char *controller_hostname; | ||
1238 | uint16_t msize; | ||
1239 | |||
1240 | if (NULL != master_context) | ||
1241 | { | ||
1242 | LOG_DEBUG ("We are being connected to laterally\n"); | ||
1243 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1244 | return; | ||
1245 | } | ||
1246 | msg = (const struct GNUNET_TESTBED_InitMessage *) message; | ||
1247 | msize = ntohs (message->size); | ||
1248 | if (msize <= sizeof (struct GNUNET_TESTBED_InitMessage)) | ||
1249 | { | ||
1250 | GNUNET_break (0); | ||
1251 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1252 | return; | ||
1253 | } | ||
1254 | msize -= sizeof (struct GNUNET_TESTBED_InitMessage); | ||
1255 | controller_hostname = (const char *) &msg[1]; | ||
1256 | if ('\0' != controller_hostname[msize - 1]) | ||
1257 | { | ||
1258 | GNUNET_break (0); | ||
1259 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1260 | return; | ||
1261 | } | ||
1262 | master_context = GNUNET_malloc (sizeof (struct Context)); | ||
1263 | master_context->client = client; | ||
1264 | master_context->host_id = ntohl (msg->host_id); | ||
1265 | master_context->master_ip = GNUNET_strdup (controller_hostname); | ||
1266 | LOG_DEBUG ("Master Controller IP: %s\n", master_context->master_ip); | ||
1267 | master_context->system = | ||
1268 | GNUNET_TESTING_system_create ("testbed", master_context->master_ip, hostname); | ||
1269 | host = | ||
1270 | GNUNET_TESTBED_host_create_with_id (master_context->host_id, NULL, NULL, | ||
1271 | 0); | ||
1272 | host_list_add (host); | ||
1273 | master_context->event_mask = GNUNET_ntohll (msg->event_mask); | ||
1274 | GNUNET_SERVER_client_keep (client); | ||
1275 | LOG_DEBUG ("Created master context with host ID: %u\n", | ||
1276 | master_context->host_id); | ||
1277 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1278 | } | ||
1279 | |||
1280 | |||
1281 | /** | ||
1282 | * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST messages | ||
1283 | * | ||
1284 | * @param cls NULL | ||
1285 | * @param client identification of the client | ||
1286 | * @param message the actual message | ||
1287 | */ | ||
1288 | static void | ||
1289 | handle_add_host (void *cls, struct GNUNET_SERVER_Client *client, | ||
1290 | const struct GNUNET_MessageHeader *message) | ||
1291 | { | ||
1292 | struct GNUNET_TESTBED_Host *host; | ||
1293 | const struct GNUNET_TESTBED_AddHostMessage *msg; | ||
1294 | struct GNUNET_TESTBED_HostConfirmedMessage *reply; | ||
1295 | char *username; | ||
1296 | char *hostname; | ||
1297 | char *emsg; | ||
1298 | uint32_t host_id; | ||
1299 | uint16_t username_length; | ||
1300 | uint16_t hostname_length; | ||
1301 | uint16_t reply_size; | ||
1302 | uint16_t msize; | ||
1303 | |||
1304 | msg = (const struct GNUNET_TESTBED_AddHostMessage *) message; | ||
1305 | msize = ntohs (msg->header.size); | ||
1306 | username = (char *) &(msg[1]); | ||
1307 | username_length = ntohs (msg->user_name_length); | ||
1308 | GNUNET_assert (msize > (sizeof (struct GNUNET_TESTBED_AddHostMessage) + username_length + 1)); /* msg must contain hostname */ | ||
1309 | if (0 != username_length) | ||
1310 | GNUNET_assert ('\0' == username[username_length]); | ||
1311 | username_length = (0 == username_length) ? 0 : username_length + 1; | ||
1312 | hostname = username + username_length; | ||
1313 | hostname_length = | ||
1314 | msize - (sizeof (struct GNUNET_TESTBED_AddHostMessage) + username_length); | ||
1315 | GNUNET_assert ('\0' == hostname[hostname_length - 1]); | ||
1316 | GNUNET_assert (strlen (hostname) == hostname_length - 1); | ||
1317 | host_id = ntohl (msg->host_id); | ||
1318 | LOG_DEBUG ("Received ADDHOST message\n"); | ||
1319 | LOG_DEBUG ("-------host id: %u\n", host_id); | ||
1320 | LOG_DEBUG ("-------hostname: %s\n", hostname); | ||
1321 | if (0 != username_length) | ||
1322 | LOG_DEBUG ("-------username: %s\n", username); | ||
1323 | else | ||
1324 | { | ||
1325 | LOG_DEBUG ("-------username: NULL\n"); | ||
1326 | username = NULL; | ||
1327 | } | ||
1328 | LOG_DEBUG ("-------ssh port: %u\n", ntohs (msg->ssh_port)); | ||
1329 | host = | ||
1330 | GNUNET_TESTBED_host_create_with_id (host_id, hostname, username, | ||
1331 | ntohs (msg->ssh_port)); | ||
1332 | GNUNET_assert (NULL != host); | ||
1333 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1334 | reply_size = sizeof (struct GNUNET_TESTBED_HostConfirmedMessage); | ||
1335 | if (GNUNET_OK != host_list_add (host)) | ||
1336 | { | ||
1337 | /* We are unable to add a host */ | ||
1338 | emsg = "A host exists with given host-id"; | ||
1339 | LOG_DEBUG ("%s: %u", emsg, host_id); | ||
1340 | GNUNET_TESTBED_host_destroy (host); | ||
1341 | reply_size += strlen (emsg) + 1; | ||
1342 | reply = GNUNET_malloc (reply_size); | ||
1343 | memcpy (&reply[1], emsg, strlen (emsg) + 1); | ||
1344 | } | ||
1345 | else | ||
1346 | reply = GNUNET_malloc (reply_size); | ||
1347 | reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_ADDHOSTCONFIRM); | ||
1348 | reply->header.size = htons (reply_size); | ||
1349 | reply->host_id = htonl (host_id); | ||
1350 | queue_message (client, &reply->header); | ||
1351 | } | ||
1352 | |||
1353 | |||
1354 | /** | ||
1355 | * Iterator over hash map entries. | ||
1356 | * | ||
1357 | * @param cls closure | ||
1358 | * @param key current key code | ||
1359 | * @param value value in the hash map | ||
1360 | * @return GNUNET_YES if we should continue to | ||
1361 | * iterate, | ||
1362 | * GNUNET_NO if not. | ||
1363 | */ | ||
1364 | int | ||
1365 | ss_exists_iterator (void *cls, const struct GNUNET_HashCode *key, void *value) | ||
1366 | { | ||
1367 | struct SharedService *queried_ss = cls; | ||
1368 | struct SharedService *ss = value; | ||
1369 | |||
1370 | if (0 == strcmp (ss->name, queried_ss->name)) | ||
1371 | return GNUNET_NO; | ||
1372 | else | ||
1373 | return GNUNET_YES; | ||
1374 | } | ||
1375 | |||
1376 | |||
1377 | /** | ||
1378 | * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST messages | ||
1379 | * | ||
1380 | * @param cls NULL | ||
1381 | * @param client identification of the client | ||
1382 | * @param message the actual message | ||
1383 | */ | ||
1384 | static void | ||
1385 | handle_configure_shared_service (void *cls, struct GNUNET_SERVER_Client *client, | ||
1386 | const struct GNUNET_MessageHeader *message) | ||
1387 | { | ||
1388 | const struct GNUNET_TESTBED_ConfigureSharedServiceMessage *msg; | ||
1389 | struct SharedService *ss; | ||
1390 | char *service_name; | ||
1391 | struct GNUNET_HashCode hash; | ||
1392 | uint16_t msg_size; | ||
1393 | uint16_t service_name_size; | ||
1394 | |||
1395 | msg = (const struct GNUNET_TESTBED_ConfigureSharedServiceMessage *) message; | ||
1396 | msg_size = ntohs (message->size); | ||
1397 | if (msg_size <= sizeof (struct GNUNET_TESTBED_ConfigureSharedServiceMessage)) | ||
1398 | { | ||
1399 | GNUNET_break (0); | ||
1400 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1401 | return; | ||
1402 | } | ||
1403 | service_name_size = | ||
1404 | msg_size - sizeof (struct GNUNET_TESTBED_ConfigureSharedServiceMessage); | ||
1405 | service_name = (char *) &msg[1]; | ||
1406 | if ('\0' != service_name[service_name_size - 1]) | ||
1407 | { | ||
1408 | GNUNET_break (0); | ||
1409 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1410 | return; | ||
1411 | } | ||
1412 | LOG_DEBUG ("Received service sharing request for %s, with %d peers\n", | ||
1413 | service_name, ntohl (msg->num_peers)); | ||
1414 | if (ntohl (msg->host_id) != master_context->host_id) | ||
1415 | { | ||
1416 | route_message (ntohl (msg->host_id), message); | ||
1417 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1418 | return; | ||
1419 | } | ||
1420 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1421 | ss = GNUNET_malloc (sizeof (struct SharedService)); | ||
1422 | ss->name = strdup (service_name); | ||
1423 | ss->num_shared = ntohl (msg->num_peers); | ||
1424 | GNUNET_CRYPTO_hash (ss->name, service_name_size, &hash); | ||
1425 | if (GNUNET_SYSERR == | ||
1426 | GNUNET_CONTAINER_multihashmap_get_multiple (ss_map, &hash, | ||
1427 | &ss_exists_iterator, ss)) | ||
1428 | { | ||
1429 | LOG (GNUNET_ERROR_TYPE_WARNING, | ||
1430 | "Service %s already configured as a shared service. " | ||
1431 | "Ignoring service sharing request \n", ss->name); | ||
1432 | GNUNET_free (ss->name); | ||
1433 | GNUNET_free (ss); | ||
1434 | return; | ||
1435 | } | ||
1436 | GNUNET_CONTAINER_multihashmap_put (ss_map, &hash, ss, | ||
1437 | GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE); | ||
1438 | } | ||
1439 | |||
1440 | |||
1441 | /** | ||
1442 | * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_LCONTROLLERS message | ||
1443 | * | ||
1444 | * @param cls NULL | ||
1445 | * @param client identification of the client | ||
1446 | * @param message the actual message | ||
1447 | */ | ||
1448 | static void | ||
1449 | handle_link_controllers (void *cls, struct GNUNET_SERVER_Client *client, | ||
1450 | const struct GNUNET_MessageHeader *message) | ||
1451 | { | ||
1452 | const struct GNUNET_TESTBED_ControllerLinkMessage *msg; | ||
1453 | struct GNUNET_CONFIGURATION_Handle *cfg; | ||
1454 | struct LCFContextQueue *lcfq; | ||
1455 | struct Route *route; | ||
1456 | struct Route *new_route; | ||
1457 | char *config; | ||
1458 | uLongf dest_size; | ||
1459 | size_t config_size; | ||
1460 | uint32_t delegated_host_id; | ||
1461 | uint32_t slave_host_id; | ||
1462 | uint16_t msize; | ||
1463 | |||
1464 | if (NULL == master_context) | ||
1465 | { | ||
1466 | GNUNET_break (0); | ||
1467 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1468 | return; | ||
1469 | } | ||
1470 | msize = ntohs (message->size); | ||
1471 | if (sizeof (struct GNUNET_TESTBED_ControllerLinkMessage) >= msize) | ||
1472 | { | ||
1473 | GNUNET_break (0); | ||
1474 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1475 | return; | ||
1476 | } | ||
1477 | msg = (const struct GNUNET_TESTBED_ControllerLinkMessage *) message; | ||
1478 | delegated_host_id = ntohl (msg->delegated_host_id); | ||
1479 | if (delegated_host_id == master_context->host_id) | ||
1480 | { | ||
1481 | GNUNET_break (0); | ||
1482 | LOG (GNUNET_ERROR_TYPE_WARNING, "Trying to link ourselves\n"); | ||
1483 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1484 | return; | ||
1485 | } | ||
1486 | if ((delegated_host_id >= host_list_size) || | ||
1487 | (NULL == host_list[delegated_host_id])) | ||
1488 | { | ||
1489 | LOG (GNUNET_ERROR_TYPE_WARNING, | ||
1490 | "Delegated host %u not registered with us\n", delegated_host_id); | ||
1491 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1492 | return; | ||
1493 | } | ||
1494 | slave_host_id = ntohl (msg->slave_host_id); | ||
1495 | if ((slave_host_id >= host_list_size) || (NULL == host_list[slave_host_id])) | ||
1496 | { | ||
1497 | LOG (GNUNET_ERROR_TYPE_WARNING, "Slave host not registered with us\n"); | ||
1498 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1499 | return; | ||
1500 | } | ||
1501 | if (slave_host_id == delegated_host_id) | ||
1502 | { | ||
1503 | LOG (GNUNET_ERROR_TYPE_WARNING, "Slave and delegated host are same\n"); | ||
1504 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1505 | return; | ||
1506 | } | ||
1507 | |||
1508 | if (slave_host_id == master_context->host_id) /* Link from us */ | ||
1509 | { | ||
1510 | struct Slave *slave; | ||
1511 | struct LinkControllersContext *lcc; | ||
1512 | |||
1513 | msize -= sizeof (struct GNUNET_TESTBED_ControllerLinkMessage); | ||
1514 | config_size = ntohs (msg->config_size); | ||
1515 | if ((delegated_host_id < slave_list_size) && (NULL != slave_list[delegated_host_id])) /* We have already added */ | ||
1516 | { | ||
1517 | LOG (GNUNET_ERROR_TYPE_WARNING, "Host %u already connected\n", | ||
1518 | delegated_host_id); | ||
1519 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1520 | return; | ||
1521 | } | ||
1522 | config = GNUNET_malloc (config_size); | ||
1523 | dest_size = (uLongf) config_size; | ||
1524 | if (Z_OK != | ||
1525 | uncompress ((Bytef *) config, &dest_size, (const Bytef *) &msg[1], | ||
1526 | (uLong) msize)) | ||
1527 | { | ||
1528 | GNUNET_break (0); /* Compression error */ | ||
1529 | GNUNET_free (config); | ||
1530 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1531 | return; | ||
1532 | } | ||
1533 | if (config_size != dest_size) | ||
1534 | { | ||
1535 | LOG (GNUNET_ERROR_TYPE_WARNING, "Uncompressed config size mismatch\n"); | ||
1536 | GNUNET_free (config); | ||
1537 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1538 | return; | ||
1539 | } | ||
1540 | cfg = GNUNET_CONFIGURATION_create (); /* Free here or in lcfcontext */ | ||
1541 | if (GNUNET_OK != | ||
1542 | GNUNET_CONFIGURATION_deserialize (cfg, config, config_size, GNUNET_NO)) | ||
1543 | { | ||
1544 | GNUNET_break (0); /* Configuration parsing error */ | ||
1545 | GNUNET_free (config); | ||
1546 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1547 | return; | ||
1548 | } | ||
1549 | GNUNET_free (config); | ||
1550 | if ((delegated_host_id < slave_list_size) && | ||
1551 | (NULL != slave_list[delegated_host_id])) | ||
1552 | { | ||
1553 | GNUNET_break (0); /* Configuration parsing error */ | ||
1554 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1555 | return; | ||
1556 | } | ||
1557 | slave = GNUNET_malloc (sizeof (struct Slave)); | ||
1558 | slave->host_id = delegated_host_id; | ||
1559 | slave_list_add (slave); | ||
1560 | if (1 != msg->is_subordinate) | ||
1561 | { | ||
1562 | slave->controller = | ||
1563 | GNUNET_TESTBED_controller_connect (cfg, host_list[slave->host_id], | ||
1564 | master_context->event_mask, | ||
1565 | &slave_event_callback, slave); | ||
1566 | slave->cfg = cfg; | ||
1567 | if (NULL != slave->controller) | ||
1568 | send_operation_success_msg (client, GNUNET_ntohll (msg->operation_id)); | ||
1569 | else | ||
1570 | send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id), | ||
1571 | "Could not connect to delegated controller"); | ||
1572 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1573 | return; | ||
1574 | } | ||
1575 | lcc = GNUNET_malloc (sizeof (struct LinkControllersContext)); | ||
1576 | lcc->operation_id = GNUNET_ntohll (msg->operation_id); | ||
1577 | GNUNET_SERVER_client_keep (client); | ||
1578 | lcc->client = client; | ||
1579 | slave->lcc = lcc; | ||
1580 | slave->controller_proc = | ||
1581 | GNUNET_TESTBED_controller_start (master_context->master_ip, | ||
1582 | host_list[slave->host_id], cfg, | ||
1583 | &slave_status_callback, slave); | ||
1584 | GNUNET_CONFIGURATION_destroy (cfg); | ||
1585 | new_route = GNUNET_malloc (sizeof (struct Route)); | ||
1586 | new_route->dest = delegated_host_id; | ||
1587 | new_route->thru = master_context->host_id; | ||
1588 | route_list_add (new_route); | ||
1589 | return; | ||
1590 | } | ||
1591 | |||
1592 | /* Route the request */ | ||
1593 | if (slave_host_id >= route_list_size) | ||
1594 | { | ||
1595 | LOG (GNUNET_ERROR_TYPE_WARNING, "No route towards slave host"); | ||
1596 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1597 | return; | ||
1598 | } | ||
1599 | lcfq = GNUNET_malloc (sizeof (struct LCFContextQueue)); | ||
1600 | lcfq->lcf = GNUNET_malloc (sizeof (struct LCFContext)); | ||
1601 | lcfq->lcf->delegated_host_id = delegated_host_id; | ||
1602 | lcfq->lcf->slave_host_id = slave_host_id; | ||
1603 | route = find_dest_route (slave_host_id); | ||
1604 | GNUNET_assert (NULL != route); /* because we add routes carefully */ | ||
1605 | GNUNET_assert (route->dest < slave_list_size); | ||
1606 | GNUNET_assert (NULL != slave_list[route->dest]); | ||
1607 | lcfq->lcf->state = INIT; | ||
1608 | lcfq->lcf->operation_id = GNUNET_ntohll (msg->operation_id); | ||
1609 | lcfq->lcf->gateway = slave_list[route->dest]; | ||
1610 | lcfq->lcf->msg = GNUNET_malloc (msize); | ||
1611 | (void) memcpy (lcfq->lcf->msg, msg, msize); | ||
1612 | GNUNET_SERVER_client_keep (client); | ||
1613 | lcfq->lcf->client = client; | ||
1614 | if (NULL == lcfq_head) | ||
1615 | { | ||
1616 | GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == lcf_proc_task_id); | ||
1617 | GNUNET_CONTAINER_DLL_insert_tail (lcfq_head, lcfq_tail, lcfq); | ||
1618 | lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcfq->lcf); | ||
1619 | } | ||
1620 | else | ||
1621 | GNUNET_CONTAINER_DLL_insert_tail (lcfq_head, lcfq_tail, lcfq); | ||
1622 | /* FIXME: Adding a new route should happen after the controllers are linked | ||
1623 | * successfully */ | ||
1624 | if (1 != msg->is_subordinate) | ||
1625 | { | ||
1626 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1627 | return; | ||
1628 | } | ||
1629 | if ((delegated_host_id < route_list_size) | ||
1630 | && (NULL != route_list[delegated_host_id])) | ||
1631 | { | ||
1632 | GNUNET_break_op (0); /* Are you trying to link delegated host twice | ||
1633 | with is subordinate flag set to GNUNET_YES? */ | ||
1634 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1635 | return; | ||
1636 | } | ||
1637 | new_route = GNUNET_malloc (sizeof (struct Route)); | ||
1638 | new_route->dest = delegated_host_id; | ||
1639 | new_route->thru = route->dest; | ||
1640 | route_list_add (new_route); | ||
1641 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1642 | } | ||
1643 | |||
1644 | |||
1645 | /** | ||
1646 | * The task to be executed if the forwarded peer create operation has been | ||
1647 | * timed out | ||
1648 | * | ||
1649 | * @param cls the FowardedOperationContext | ||
1650 | * @param tc the TaskContext from the scheduler | ||
1651 | */ | ||
1652 | static void | ||
1653 | peer_create_forward_timeout (void *cls, | ||
1654 | const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
1655 | { | ||
1656 | struct ForwardedOperationContext *fo_ctxt = cls; | ||
1657 | |||
1658 | /* send error msg to client */ | ||
1659 | send_operation_fail_msg (fo_ctxt->client, fo_ctxt->operation_id, "Timedout"); | ||
1660 | GNUNET_SERVER_client_drop (fo_ctxt->client); | ||
1661 | GNUNET_TESTBED_forward_operation_msg_cancel_ (fo_ctxt->opc); | ||
1662 | GNUNET_free (fo_ctxt); | ||
1663 | } | ||
1664 | |||
1665 | |||
1666 | /** | ||
1667 | * Callback to be called when forwarded peer create operation is | ||
1668 | * successfull. We have to relay the reply msg back to the client | ||
1669 | * | ||
1670 | * @param cls ForwardedOperationContext | ||
1671 | * @param msg the peer create success message | ||
1672 | */ | ||
1673 | static void | ||
1674 | peer_create_success_cb (void *cls, const struct GNUNET_MessageHeader *msg) | ||
1675 | { | ||
1676 | struct ForwardedOperationContext *fo_ctxt = cls; | ||
1677 | const struct GNUNET_TESTBED_PeerCreateSuccessEventMessage *success_msg; | ||
1678 | struct GNUNET_MessageHeader *dup_msg; | ||
1679 | struct Peer *peer; | ||
1680 | uint16_t msize; | ||
1681 | |||
1682 | GNUNET_SCHEDULER_cancel (fo_ctxt->timeout_task); | ||
1683 | if (ntohs (msg->type) == GNUNET_MESSAGE_TYPE_TESTBED_PEERCREATESUCCESS) | ||
1684 | { | ||
1685 | success_msg = | ||
1686 | (const struct GNUNET_TESTBED_PeerCreateSuccessEventMessage *) msg; | ||
1687 | peer = GNUNET_malloc (sizeof (struct Peer)); | ||
1688 | peer->is_remote = GNUNET_YES; | ||
1689 | peer->id = ntohl (success_msg->peer_id); | ||
1690 | GNUNET_assert (NULL != fo_ctxt->cls); | ||
1691 | peer->details.remote.controller = fo_ctxt->cls; | ||
1692 | peer_list_add (peer); | ||
1693 | } | ||
1694 | msize = ntohs (msg->size); | ||
1695 | dup_msg = GNUNET_malloc (msize); | ||
1696 | (void) memcpy (dup_msg, msg, msize); | ||
1697 | queue_message (fo_ctxt->client, dup_msg); | ||
1698 | GNUNET_SERVER_client_drop (fo_ctxt->client); | ||
1699 | GNUNET_free (fo_ctxt); | ||
1700 | } | ||
1701 | |||
1702 | |||
1703 | |||
1704 | /** | ||
1705 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_CREATEPEER messages | ||
1706 | * | ||
1707 | * @param cls NULL | ||
1708 | * @param client identification of the client | ||
1709 | * @param message the actual message | ||
1710 | */ | ||
1711 | static void | ||
1712 | handle_peer_create (void *cls, struct GNUNET_SERVER_Client *client, | ||
1713 | const struct GNUNET_MessageHeader *message) | ||
1714 | { | ||
1715 | const struct GNUNET_TESTBED_PeerCreateMessage *msg; | ||
1716 | struct GNUNET_TESTBED_PeerCreateSuccessEventMessage *reply; | ||
1717 | struct GNUNET_CONFIGURATION_Handle *cfg; | ||
1718 | struct ForwardedOperationContext *fo_ctxt; | ||
1719 | struct Route *route; | ||
1720 | struct Peer *peer; | ||
1721 | char *config; | ||
1722 | size_t dest_size; | ||
1723 | int ret; | ||
1724 | uint32_t config_size; | ||
1725 | uint32_t host_id; | ||
1726 | uint32_t peer_id; | ||
1727 | uint16_t msize; | ||
1728 | |||
1729 | |||
1730 | msize = ntohs (message->size); | ||
1731 | if (msize <= sizeof (struct GNUNET_TESTBED_PeerCreateMessage)) | ||
1732 | { | ||
1733 | GNUNET_break (0); /* We need configuration */ | ||
1734 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1735 | return; | ||
1736 | } | ||
1737 | msg = (const struct GNUNET_TESTBED_PeerCreateMessage *) message; | ||
1738 | host_id = ntohl (msg->host_id); | ||
1739 | peer_id = ntohl (msg->peer_id); | ||
1740 | if (UINT32_MAX == peer_id) | ||
1741 | { | ||
1742 | send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id), | ||
1743 | "Cannot create peer with given ID"); | ||
1744 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1745 | return; | ||
1746 | } | ||
1747 | if (host_id == master_context->host_id) | ||
1748 | { | ||
1749 | char *emsg; | ||
1750 | |||
1751 | /* We are responsible for this peer */ | ||
1752 | msize -= sizeof (struct GNUNET_TESTBED_PeerCreateMessage); | ||
1753 | config_size = ntohl (msg->config_size); | ||
1754 | config = GNUNET_malloc (config_size); | ||
1755 | dest_size = config_size; | ||
1756 | if (Z_OK != | ||
1757 | (ret = | ||
1758 | uncompress ((Bytef *) config, (uLongf *) & dest_size, | ||
1759 | (const Bytef *) &msg[1], (uLong) msize))) | ||
1760 | { | ||
1761 | GNUNET_break (0); /* uncompression error */ | ||
1762 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1763 | return; | ||
1764 | } | ||
1765 | if (config_size != dest_size) | ||
1766 | { | ||
1767 | GNUNET_break (0); /* Uncompressed config size mismatch */ | ||
1768 | GNUNET_free (config); | ||
1769 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1770 | return; | ||
1771 | } | ||
1772 | cfg = GNUNET_CONFIGURATION_create (); | ||
1773 | if (GNUNET_OK != | ||
1774 | GNUNET_CONFIGURATION_deserialize (cfg, config, config_size, GNUNET_NO)) | ||
1775 | { | ||
1776 | GNUNET_break (0); /* Configuration parsing error */ | ||
1777 | GNUNET_free (config); | ||
1778 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1779 | return; | ||
1780 | } | ||
1781 | GNUNET_free (config); | ||
1782 | peer = GNUNET_malloc (sizeof (struct Peer)); | ||
1783 | peer->is_remote = GNUNET_NO; | ||
1784 | peer->details.local.cfg = cfg; | ||
1785 | peer->id = peer_id; | ||
1786 | LOG_DEBUG ("Creating peer with id: %u\n", peer->id); | ||
1787 | peer->details.local.peer = | ||
1788 | GNUNET_TESTING_peer_configure (master_context->system, | ||
1789 | peer->details.local.cfg, peer->id, | ||
1790 | NULL /* Peer id */ , | ||
1791 | &emsg); | ||
1792 | if (NULL == peer->details.local.peer) | ||
1793 | { | ||
1794 | LOG (GNUNET_ERROR_TYPE_WARNING, "Configuring peer failed: %s\n", emsg); | ||
1795 | GNUNET_free (emsg); | ||
1796 | GNUNET_free (peer); | ||
1797 | GNUNET_break (0); | ||
1798 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
1799 | return; | ||
1800 | } | ||
1801 | peer->details.local.is_running = GNUNET_NO; | ||
1802 | peer_list_add (peer); | ||
1803 | reply = | ||
1804 | GNUNET_malloc (sizeof | ||
1805 | (struct GNUNET_TESTBED_PeerCreateSuccessEventMessage)); | ||
1806 | reply->header.size = | ||
1807 | htons (sizeof (struct GNUNET_TESTBED_PeerCreateSuccessEventMessage)); | ||
1808 | reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEERCREATESUCCESS); | ||
1809 | reply->peer_id = msg->peer_id; | ||
1810 | reply->operation_id = msg->operation_id; | ||
1811 | queue_message (client, &reply->header); | ||
1812 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1813 | return; | ||
1814 | } | ||
1815 | |||
1816 | /* Forward peer create request */ | ||
1817 | route = find_dest_route (host_id); | ||
1818 | if (NULL == route) | ||
1819 | { | ||
1820 | GNUNET_break (0); | ||
1821 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1822 | return; | ||
1823 | } | ||
1824 | fo_ctxt = GNUNET_malloc (sizeof (struct ForwardedOperationContext)); | ||
1825 | GNUNET_SERVER_client_keep (client); | ||
1826 | fo_ctxt->client = client; | ||
1827 | fo_ctxt->operation_id = GNUNET_ntohll (msg->operation_id); | ||
1828 | fo_ctxt->cls = slave_list[route->dest]->controller; | ||
1829 | fo_ctxt->opc = | ||
1830 | GNUNET_TESTBED_forward_operation_msg_ (slave_list | ||
1831 | [route->dest]->controller, | ||
1832 | fo_ctxt->operation_id, | ||
1833 | &msg->header, | ||
1834 | peer_create_success_cb, fo_ctxt); | ||
1835 | fo_ctxt->timeout_task = | ||
1836 | GNUNET_SCHEDULER_add_delayed (TIMEOUT, &peer_create_forward_timeout, | ||
1837 | fo_ctxt); | ||
1838 | |||
1839 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1840 | } | ||
1841 | |||
1842 | |||
1843 | /** | ||
1844 | * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER messages | ||
1845 | * | ||
1846 | * @param cls NULL | ||
1847 | * @param client identification of the client | ||
1848 | * @param message the actual message | ||
1849 | */ | ||
1850 | static void | ||
1851 | handle_peer_destroy (void *cls, struct GNUNET_SERVER_Client *client, | ||
1852 | const struct GNUNET_MessageHeader *message) | ||
1853 | { | ||
1854 | const struct GNUNET_TESTBED_PeerDestroyMessage *msg; | ||
1855 | struct ForwardedOperationContext *fopc; | ||
1856 | struct Peer *peer; | ||
1857 | uint32_t peer_id; | ||
1858 | |||
1859 | msg = (const struct GNUNET_TESTBED_PeerDestroyMessage *) message; | ||
1860 | peer_id = ntohl (msg->peer_id); | ||
1861 | LOG_DEBUG ("Received peer destory on peer: %u and operation id: %ul\n", | ||
1862 | peer_id, GNUNET_ntohll (msg->operation_id)); | ||
1863 | if ((peer_list_size <= peer_id) || (NULL == peer_list[peer_id])) | ||
1864 | { | ||
1865 | LOG (GNUNET_ERROR_TYPE_ERROR, | ||
1866 | "Asked to destroy a non existent peer with id: %u\n", peer_id); | ||
1867 | send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id), | ||
1868 | "Peer doesn't exist"); | ||
1869 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1870 | return; | ||
1871 | } | ||
1872 | peer = peer_list[peer_id]; | ||
1873 | if (GNUNET_YES == peer->is_remote) | ||
1874 | { | ||
1875 | /* Forward the destory message to sub controller */ | ||
1876 | fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext)); | ||
1877 | GNUNET_SERVER_client_keep (client); | ||
1878 | fopc->client = client; | ||
1879 | fopc->operation_id = GNUNET_ntohll (msg->operation_id); | ||
1880 | fopc->opc = | ||
1881 | GNUNET_TESTBED_forward_operation_msg_ (peer->details.remote.controller, | ||
1882 | fopc->operation_id, &msg->header, | ||
1883 | &forwarded_operation_reply_relay, | ||
1884 | fopc); | ||
1885 | fopc->timeout_task = | ||
1886 | GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout, | ||
1887 | fopc); | ||
1888 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1889 | return; | ||
1890 | } | ||
1891 | GNUNET_TESTING_peer_destroy (peer->details.local.peer); | ||
1892 | GNUNET_CONFIGURATION_destroy (peer->details.local.cfg); | ||
1893 | peer_list_remove (peer); | ||
1894 | GNUNET_free (peer); | ||
1895 | send_operation_success_msg (client, GNUNET_ntohll (msg->operation_id)); | ||
1896 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1897 | } | ||
1898 | |||
1899 | |||
1900 | /** | ||
1901 | * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER messages | ||
1902 | * | ||
1903 | * @param cls NULL | ||
1904 | * @param client identification of the client | ||
1905 | * @param message the actual message | ||
1906 | */ | ||
1907 | static void | ||
1908 | handle_peer_start (void *cls, struct GNUNET_SERVER_Client *client, | ||
1909 | const struct GNUNET_MessageHeader *message) | ||
1910 | { | ||
1911 | const struct GNUNET_TESTBED_PeerStartMessage *msg; | ||
1912 | struct GNUNET_TESTBED_PeerEventMessage *reply; | ||
1913 | struct ForwardedOperationContext *fopc; | ||
1914 | struct Peer *peer; | ||
1915 | uint32_t peer_id; | ||
1916 | |||
1917 | msg = (const struct GNUNET_TESTBED_PeerStartMessage *) message; | ||
1918 | peer_id = ntohl (msg->peer_id); | ||
1919 | if ((peer_id >= peer_list_size) || (NULL == peer_list[peer_id])) | ||
1920 | { | ||
1921 | GNUNET_break (0); | ||
1922 | LOG (GNUNET_ERROR_TYPE_ERROR, | ||
1923 | "Asked to start a non existent peer with id: %u\n", peer_id); | ||
1924 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1925 | return; | ||
1926 | } | ||
1927 | peer = peer_list[peer_id]; | ||
1928 | if (GNUNET_YES == peer->is_remote) | ||
1929 | { | ||
1930 | fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext)); | ||
1931 | GNUNET_SERVER_client_keep (client); | ||
1932 | fopc->client = client; | ||
1933 | fopc->operation_id = GNUNET_ntohll (msg->operation_id); | ||
1934 | fopc->opc = | ||
1935 | GNUNET_TESTBED_forward_operation_msg_ (peer->details.remote.controller, | ||
1936 | fopc->operation_id, &msg->header, | ||
1937 | &forwarded_operation_reply_relay, | ||
1938 | fopc); | ||
1939 | fopc->timeout_task = | ||
1940 | GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout, | ||
1941 | fopc); | ||
1942 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1943 | return; | ||
1944 | } | ||
1945 | if (GNUNET_OK != GNUNET_TESTING_peer_start (peer->details.local.peer)) | ||
1946 | { | ||
1947 | send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id), | ||
1948 | "Failed to start"); | ||
1949 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1950 | return; | ||
1951 | } | ||
1952 | peer->details.local.is_running = GNUNET_YES; | ||
1953 | reply = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_PeerEventMessage)); | ||
1954 | reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEEREVENT); | ||
1955 | reply->header.size = htons (sizeof (struct GNUNET_TESTBED_PeerEventMessage)); | ||
1956 | reply->event_type = htonl (GNUNET_TESTBED_ET_PEER_START); | ||
1957 | reply->host_id = htonl (master_context->host_id); | ||
1958 | reply->peer_id = msg->peer_id; | ||
1959 | reply->operation_id = msg->operation_id; | ||
1960 | queue_message (client, &reply->header); | ||
1961 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1962 | } | ||
1963 | |||
1964 | |||
1965 | /** | ||
1966 | * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER messages | ||
1967 | * | ||
1968 | * @param cls NULL | ||
1969 | * @param client identification of the client | ||
1970 | * @param message the actual message | ||
1971 | */ | ||
1972 | static void | ||
1973 | handle_peer_stop (void *cls, struct GNUNET_SERVER_Client *client, | ||
1974 | const struct GNUNET_MessageHeader *message) | ||
1975 | { | ||
1976 | const struct GNUNET_TESTBED_PeerStopMessage *msg; | ||
1977 | struct GNUNET_TESTBED_PeerEventMessage *reply; | ||
1978 | struct ForwardedOperationContext *fopc; | ||
1979 | struct Peer *peer; | ||
1980 | uint32_t peer_id; | ||
1981 | |||
1982 | msg = (const struct GNUNET_TESTBED_PeerStopMessage *) message; | ||
1983 | peer_id = ntohl (msg->peer_id); | ||
1984 | if ((peer_id >= peer_list_size) || (NULL == peer_list[peer_id])) | ||
1985 | { | ||
1986 | send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id), | ||
1987 | "Peer not found"); | ||
1988 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
1989 | return; | ||
1990 | } | ||
1991 | peer = peer_list[peer_id]; | ||
1992 | if (GNUNET_YES == peer->is_remote) | ||
1993 | { | ||
1994 | fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext)); | ||
1995 | GNUNET_SERVER_client_keep (client); | ||
1996 | fopc->client = client; | ||
1997 | fopc->operation_id = GNUNET_ntohll (msg->operation_id); | ||
1998 | fopc->opc = | ||
1999 | GNUNET_TESTBED_forward_operation_msg_ (peer->details.remote.controller, | ||
2000 | fopc->operation_id, &msg->header, | ||
2001 | &forwarded_operation_reply_relay, | ||
2002 | fopc); | ||
2003 | fopc->timeout_task = | ||
2004 | GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout, | ||
2005 | fopc); | ||
2006 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2007 | return; | ||
2008 | } | ||
2009 | if (GNUNET_OK != GNUNET_TESTING_peer_stop (peer->details.local.peer)) | ||
2010 | { | ||
2011 | send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id), | ||
2012 | "Peer not running"); | ||
2013 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2014 | return; | ||
2015 | } | ||
2016 | peer->details.local.is_running = GNUNET_NO; | ||
2017 | reply = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_PeerEventMessage)); | ||
2018 | reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEEREVENT); | ||
2019 | reply->header.size = htons (sizeof (struct GNUNET_TESTBED_PeerEventMessage)); | ||
2020 | reply->event_type = htonl (GNUNET_TESTBED_ET_PEER_STOP); | ||
2021 | reply->host_id = htonl (master_context->host_id); | ||
2022 | reply->peer_id = msg->peer_id; | ||
2023 | reply->operation_id = msg->operation_id; | ||
2024 | queue_message (client, &reply->header); | ||
2025 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2026 | } | ||
2027 | |||
2028 | |||
2029 | /** | ||
2030 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_GETPEERCONFIG messages | ||
2031 | * | ||
2032 | * @param cls NULL | ||
2033 | * @param client identification of the client | ||
2034 | * @param message the actual message | ||
2035 | */ | ||
2036 | static void | ||
2037 | handle_peer_get_config (void *cls, struct GNUNET_SERVER_Client *client, | ||
2038 | const struct GNUNET_MessageHeader *message) | ||
2039 | { | ||
2040 | const struct GNUNET_TESTBED_PeerGetConfigurationMessage *msg; | ||
2041 | struct GNUNET_TESTBED_PeerConfigurationInformationMessage *reply; | ||
2042 | struct Peer *peer; | ||
2043 | char *config; | ||
2044 | char *xconfig; | ||
2045 | size_t c_size; | ||
2046 | size_t xc_size; | ||
2047 | uint32_t peer_id; | ||
2048 | uint16_t msize; | ||
2049 | |||
2050 | msg = (const struct GNUNET_TESTBED_PeerGetConfigurationMessage *) message; | ||
2051 | peer_id = ntohl (msg->peer_id); | ||
2052 | if ((peer_id >= peer_list_size) || (NULL == peer_list[peer_id])) | ||
2053 | { | ||
2054 | send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id), | ||
2055 | "Peer not found"); | ||
2056 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2057 | return; | ||
2058 | } | ||
2059 | peer = peer_list[peer_id]; | ||
2060 | if (GNUNET_YES == peer->is_remote) | ||
2061 | { | ||
2062 | struct ForwardedOperationContext *fopc; | ||
2063 | |||
2064 | fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext)); | ||
2065 | GNUNET_SERVER_client_keep (client); | ||
2066 | fopc->client = client; | ||
2067 | fopc->operation_id = GNUNET_ntohll (msg->operation_id); | ||
2068 | fopc->opc = | ||
2069 | GNUNET_TESTBED_forward_operation_msg_ (peer->details.remote.controller, | ||
2070 | fopc->operation_id, &msg->header, | ||
2071 | &forwarded_operation_reply_relay, | ||
2072 | fopc); | ||
2073 | fopc->timeout_task = | ||
2074 | GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout, | ||
2075 | fopc); | ||
2076 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2077 | return; | ||
2078 | } | ||
2079 | config = | ||
2080 | GNUNET_CONFIGURATION_serialize (peer_list[peer_id]->details.local.cfg, | ||
2081 | &c_size); | ||
2082 | xc_size = GNUNET_TESTBED_compress_config_ (config, c_size, &xconfig); | ||
2083 | GNUNET_free (config); | ||
2084 | msize = | ||
2085 | xc_size + | ||
2086 | sizeof (struct GNUNET_TESTBED_PeerConfigurationInformationMessage); | ||
2087 | reply = GNUNET_realloc (xconfig, msize); | ||
2088 | (void) memmove (&reply[1], reply, xc_size); | ||
2089 | reply->header.size = htons (msize); | ||
2090 | reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEERCONFIG); | ||
2091 | reply->peer_id = msg->peer_id; | ||
2092 | reply->operation_id = msg->operation_id; | ||
2093 | GNUNET_TESTING_peer_get_identity (peer_list[peer_id]->details.local.peer, | ||
2094 | &reply->peer_identity); | ||
2095 | reply->config_size = htons ((uint16_t) c_size); | ||
2096 | queue_message (client, &reply->header); | ||
2097 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2098 | } | ||
2099 | |||
2100 | |||
2101 | /** | ||
2102 | * Task for cleaing up overlay connect context structure | ||
2103 | * | ||
2104 | * @param cls the overlay connect context | ||
2105 | * @param tc the task context | ||
2106 | */ | ||
2107 | static void | ||
2108 | occ_cleanup (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
2109 | { | ||
2110 | struct OverlayConnectContext *occ = cls; | ||
2111 | |||
2112 | LOG_DEBUG ("Cleaning up occ\n"); | ||
2113 | GNUNET_free_non_null (occ->emsg); | ||
2114 | GNUNET_free_non_null (occ->hello); | ||
2115 | GNUNET_SERVER_client_drop (occ->client); | ||
2116 | if (NULL != occ->opc) | ||
2117 | GNUNET_TESTBED_forward_operation_msg_cancel_ (occ->opc); | ||
2118 | if (GNUNET_SCHEDULER_NO_TASK != occ->send_hello_task) | ||
2119 | GNUNET_SCHEDULER_cancel (occ->send_hello_task); | ||
2120 | if (NULL != occ->ch) | ||
2121 | GNUNET_CORE_disconnect (occ->ch); | ||
2122 | if (NULL != occ->ghh) | ||
2123 | GNUNET_TRANSPORT_get_hello_cancel (occ->ghh); | ||
2124 | if (NULL != occ->p1th) | ||
2125 | GNUNET_TRANSPORT_disconnect (occ->p1th); | ||
2126 | if (NULL != occ->p2th) | ||
2127 | GNUNET_TRANSPORT_disconnect (occ->p2th); | ||
2128 | GNUNET_free (occ); | ||
2129 | } | ||
2130 | |||
2131 | |||
2132 | /** | ||
2133 | * Task which will be run when overlay connect request has been timed out | ||
2134 | * | ||
2135 | * @param cls the OverlayConnectContext | ||
2136 | * @param tc the TaskContext | ||
2137 | */ | ||
2138 | static void | ||
2139 | timeout_overlay_connect (void *cls, | ||
2140 | const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
2141 | { | ||
2142 | struct OverlayConnectContext *occ = cls; | ||
2143 | |||
2144 | occ->timeout_task = GNUNET_SCHEDULER_NO_TASK; | ||
2145 | send_operation_fail_msg (occ->client, occ->op_id, occ->emsg); | ||
2146 | occ_cleanup (occ, tc); | ||
2147 | } | ||
2148 | |||
2149 | |||
2150 | |||
2151 | /** | ||
2152 | * Function called to notify transport users that another | ||
2153 | * peer connected to us. | ||
2154 | * | ||
2155 | * @param cls closure | ||
2156 | * @param new_peer the peer that connected | ||
2157 | * @param ats performance data | ||
2158 | * @param ats_count number of entries in ats (excluding 0-termination) | ||
2159 | */ | ||
2160 | static void | ||
2161 | overlay_connect_notify (void *cls, const struct GNUNET_PeerIdentity *new_peer, | ||
2162 | const struct GNUNET_ATS_Information *ats, | ||
2163 | unsigned int ats_count) | ||
2164 | { | ||
2165 | struct OverlayConnectContext *occ = cls; | ||
2166 | struct GNUNET_TESTBED_ConnectionEventMessage *msg; | ||
2167 | char *new_peer_str; | ||
2168 | char *other_peer_str; | ||
2169 | |||
2170 | LOG_DEBUG ("Overlay connect notify\n"); | ||
2171 | if (0 == | ||
2172 | memcmp (new_peer, &occ->peer_identity, | ||
2173 | sizeof (struct GNUNET_PeerIdentity))) | ||
2174 | return; | ||
2175 | new_peer_str = GNUNET_strdup (GNUNET_i2s (new_peer)); | ||
2176 | other_peer_str = GNUNET_strdup (GNUNET_i2s (&occ->other_peer_identity)); | ||
2177 | if (0 != | ||
2178 | memcmp (new_peer, &occ->other_peer_identity, | ||
2179 | sizeof (struct GNUNET_PeerIdentity))) | ||
2180 | { | ||
2181 | LOG_DEBUG ("Unexpected peer %4s connected when expecting peer %4s\n", | ||
2182 | new_peer_str, other_peer_str); | ||
2183 | GNUNET_free (new_peer_str); | ||
2184 | GNUNET_free (other_peer_str); | ||
2185 | return; | ||
2186 | } | ||
2187 | GNUNET_free (new_peer_str); | ||
2188 | LOG_DEBUG ("Peer %4s connected to peer %4s\n", other_peer_str, | ||
2189 | GNUNET_i2s (&occ->peer_identity)); | ||
2190 | GNUNET_free (other_peer_str); | ||
2191 | if (GNUNET_SCHEDULER_NO_TASK != occ->send_hello_task) | ||
2192 | { | ||
2193 | GNUNET_SCHEDULER_cancel (occ->send_hello_task); | ||
2194 | occ->send_hello_task = GNUNET_SCHEDULER_NO_TASK; | ||
2195 | } | ||
2196 | GNUNET_assert (GNUNET_SCHEDULER_NO_TASK != occ->timeout_task); | ||
2197 | GNUNET_SCHEDULER_cancel (occ->timeout_task); | ||
2198 | occ->timeout_task = GNUNET_SCHEDULER_NO_TASK; | ||
2199 | GNUNET_free_non_null (occ->emsg); | ||
2200 | occ->emsg = NULL; | ||
2201 | if (NULL != occ->p2th) | ||
2202 | GNUNET_TRANSPORT_disconnect (occ->p2th); | ||
2203 | occ->p2th = NULL; | ||
2204 | LOG_DEBUG ("Peers connected - Sending overlay connect success\n"); | ||
2205 | msg = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_ConnectionEventMessage)); | ||
2206 | msg->header.size = | ||
2207 | htons (sizeof (struct GNUNET_TESTBED_ConnectionEventMessage)); | ||
2208 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEERCONEVENT); | ||
2209 | msg->event_type = htonl (GNUNET_TESTBED_ET_CONNECT); | ||
2210 | msg->peer1 = htonl (occ->peer_id); | ||
2211 | msg->peer2 = htonl (occ->other_peer_id); | ||
2212 | msg->operation_id = GNUNET_htonll (occ->op_id); | ||
2213 | queue_message (occ->client, &msg->header); | ||
2214 | GNUNET_SCHEDULER_add_now (&occ_cleanup, occ); | ||
2215 | } | ||
2216 | |||
2217 | |||
2218 | /** | ||
2219 | * Task to offer HELLO of peer 1 to peer 2 and try to make peer 2 to connect to | ||
2220 | * peer 1. | ||
2221 | * | ||
2222 | * @param cls the OverlayConnectContext | ||
2223 | * @param tc the TaskContext from scheduler | ||
2224 | */ | ||
2225 | static void | ||
2226 | send_hello (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
2227 | { | ||
2228 | struct OverlayConnectContext *occ = cls; | ||
2229 | char *other_peer_str; | ||
2230 | |||
2231 | occ->send_hello_task = GNUNET_SCHEDULER_NO_TASK; | ||
2232 | if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN)) | ||
2233 | return; | ||
2234 | GNUNET_assert (NULL != occ->hello); | ||
2235 | other_peer_str = GNUNET_strdup (GNUNET_i2s (&occ->other_peer_identity)); | ||
2236 | if (NULL != occ->peer2_controller) | ||
2237 | { | ||
2238 | struct GNUNET_TESTBED_RequestConnectMessage *msg; | ||
2239 | uint16_t msize; | ||
2240 | uint16_t hello_size; | ||
2241 | |||
2242 | LOG_DEBUG ("Offering HELLO of %s to %s via Remote Overlay Request\n", | ||
2243 | GNUNET_i2s (&occ->peer_identity), other_peer_str); | ||
2244 | hello_size = ntohs (occ->hello->size); | ||
2245 | msize = sizeof (struct GNUNET_TESTBED_RequestConnectMessage) + hello_size; | ||
2246 | msg = GNUNET_malloc (msize); | ||
2247 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_REQUESTCONNECT); | ||
2248 | msg->header.size = htons (msize); | ||
2249 | msg->peer = htonl (occ->other_peer_id); | ||
2250 | msg->operation_id = GNUNET_htonll (occ->op_id); | ||
2251 | (void) memcpy (&msg->peer_identity, &occ->peer_identity, | ||
2252 | sizeof (struct GNUNET_PeerIdentity)); | ||
2253 | memcpy (msg->hello, occ->hello, hello_size); | ||
2254 | GNUNET_TESTBED_queue_message_ (occ->peer2_controller, &msg->header); | ||
2255 | } | ||
2256 | else | ||
2257 | { | ||
2258 | LOG_DEBUG ("Offering HELLO of %s to %s\n", | ||
2259 | GNUNET_i2s (&occ->peer_identity), other_peer_str); | ||
2260 | GNUNET_TRANSPORT_offer_hello (occ->p2th, occ->hello, NULL, NULL); | ||
2261 | GNUNET_TRANSPORT_try_connect (occ->p2th, &occ->peer_identity); | ||
2262 | occ->send_hello_task = | ||
2263 | GNUNET_SCHEDULER_add_delayed | ||
2264 | (GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, | ||
2265 | 100 * (pow (2, occ->retries++))), | ||
2266 | &send_hello, occ); | ||
2267 | } | ||
2268 | GNUNET_free (other_peer_str); | ||
2269 | } | ||
2270 | |||
2271 | /** | ||
2272 | * Test for checking whether HELLO message is empty | ||
2273 | * | ||
2274 | * @param cls empty flag to set | ||
2275 | * @param address the HELLO | ||
2276 | * @param expiration expiration of the HELLO | ||
2277 | * @return | ||
2278 | */ | ||
2279 | static int | ||
2280 | test_address (void *cls, const struct GNUNET_HELLO_Address *address, | ||
2281 | struct GNUNET_TIME_Absolute expiration) | ||
2282 | { | ||
2283 | int *empty = cls; | ||
2284 | |||
2285 | *empty = GNUNET_NO; | ||
2286 | return GNUNET_OK; | ||
2287 | } | ||
2288 | |||
2289 | |||
2290 | /** | ||
2291 | * Function called whenever there is an update to the HELLO of peers in the | ||
2292 | * OverlayConnectClosure. If we have a valid HELLO, we connect to the peer 2's | ||
2293 | * transport and offer peer 1's HELLO and ask peer 2 to connect to peer 1 | ||
2294 | * | ||
2295 | * @param cls closure | ||
2296 | * @param hello our updated HELLO | ||
2297 | */ | ||
2298 | static void | ||
2299 | hello_update_cb (void *cls, const struct GNUNET_MessageHeader *hello) | ||
2300 | { | ||
2301 | struct OverlayConnectContext *occ = cls; | ||
2302 | int empty; | ||
2303 | uint16_t msize; | ||
2304 | |||
2305 | msize = ntohs (hello->size); | ||
2306 | empty = GNUNET_YES; | ||
2307 | (void) GNUNET_HELLO_iterate_addresses ((const struct GNUNET_HELLO_Message *) | ||
2308 | hello, GNUNET_NO, &test_address, | ||
2309 | &empty); | ||
2310 | if (GNUNET_YES == empty) | ||
2311 | { | ||
2312 | LOG_DEBUG ("HELLO of %s is empty\n", GNUNET_i2s (&occ->peer_identity)); | ||
2313 | return; | ||
2314 | } | ||
2315 | LOG_DEBUG ("Received HELLO of %s\n", GNUNET_i2s (&occ->peer_identity)); | ||
2316 | occ->hello = GNUNET_malloc (msize); | ||
2317 | memcpy (occ->hello, hello, msize); | ||
2318 | GNUNET_TRANSPORT_get_hello_cancel (occ->ghh); | ||
2319 | occ->ghh = NULL; | ||
2320 | GNUNET_TRANSPORT_disconnect (occ->p1th); | ||
2321 | occ->p1th = NULL; | ||
2322 | GNUNET_free_non_null (occ->emsg); | ||
2323 | if (NULL == occ->peer2_controller) | ||
2324 | { | ||
2325 | occ->p2th = | ||
2326 | GNUNET_TRANSPORT_connect (peer_list[occ->other_peer_id]->details.local.cfg, | ||
2327 | &occ->other_peer_identity, NULL, NULL, NULL, | ||
2328 | NULL); | ||
2329 | if (NULL == occ->p2th) | ||
2330 | { | ||
2331 | GNUNET_asprintf (&occ->emsg, "Cannot connect to TRANSPORT of %s\n", | ||
2332 | GNUNET_i2s (&occ->other_peer_identity)); | ||
2333 | GNUNET_SCHEDULER_cancel (occ->timeout_task); | ||
2334 | occ->timeout_task = GNUNET_SCHEDULER_add_now (&timeout_overlay_connect, occ); | ||
2335 | return; | ||
2336 | } | ||
2337 | } | ||
2338 | occ->emsg = GNUNET_strdup ("Timeout while offering HELLO to other peer"); | ||
2339 | occ->send_hello_task = GNUNET_SCHEDULER_add_now (&send_hello, occ); | ||
2340 | } | ||
2341 | |||
2342 | |||
2343 | /** | ||
2344 | * Function called after GNUNET_CORE_connect has succeeded (or failed | ||
2345 | * for good). Note that the private key of the peer is intentionally | ||
2346 | * not exposed here; if you need it, your process should try to read | ||
2347 | * the private key file directly (which should work if you are | ||
2348 | * authorized...). | ||
2349 | * | ||
2350 | * @param cls closure | ||
2351 | * @param server handle to the server, NULL if we failed | ||
2352 | * @param my_identity ID of this peer, NULL if we failed | ||
2353 | */ | ||
2354 | static void | ||
2355 | core_startup_cb (void *cls, struct GNUNET_CORE_Handle *server, | ||
2356 | const struct GNUNET_PeerIdentity *my_identity) | ||
2357 | { | ||
2358 | struct OverlayConnectContext *occ = cls; | ||
2359 | |||
2360 | GNUNET_free_non_null (occ->emsg); | ||
2361 | occ->emsg = GNUNET_strdup ("Failed to connect to CORE\n"); | ||
2362 | if ((NULL == server) || (NULL == my_identity)) | ||
2363 | goto error_return; | ||
2364 | GNUNET_free (occ->emsg); | ||
2365 | occ->ch = server; | ||
2366 | occ->emsg = NULL; | ||
2367 | memcpy (&occ->peer_identity, my_identity, | ||
2368 | sizeof (struct GNUNET_PeerIdentity)); | ||
2369 | occ->p1th = | ||
2370 | GNUNET_TRANSPORT_connect (occ->peer->details.local.cfg, | ||
2371 | &occ->peer_identity, NULL, NULL, NULL, NULL); | ||
2372 | if (NULL == occ->p1th) | ||
2373 | { | ||
2374 | GNUNET_asprintf (&occ->emsg, "Cannot connect to TRANSPORT of peers %4s", | ||
2375 | GNUNET_i2s (&occ->peer_identity)); | ||
2376 | goto error_return; | ||
2377 | } | ||
2378 | LOG_DEBUG ("Acquiring HELLO of peer %s\n", GNUNET_i2s (&occ->peer_identity)); | ||
2379 | occ->emsg = GNUNET_strdup ("Timeout while acquiring HELLO message"); | ||
2380 | occ->ghh = GNUNET_TRANSPORT_get_hello (occ->p1th, &hello_update_cb, occ); | ||
2381 | return; | ||
2382 | |||
2383 | error_return: | ||
2384 | GNUNET_SCHEDULER_cancel (occ->timeout_task); | ||
2385 | occ->timeout_task = GNUNET_SCHEDULER_add_now (&timeout_overlay_connect, occ); | ||
2386 | return; | ||
2387 | } | ||
2388 | |||
2389 | |||
2390 | /** | ||
2391 | * Callback to be called when forwarded get peer config operation as part of | ||
2392 | * overlay connect is successfull. Connection to Peer 1's core is made and is | ||
2393 | * checked for new connection from peer 2 | ||
2394 | * | ||
2395 | * @param cls ForwardedOperationContext | ||
2396 | * @param msg the peer create success message | ||
2397 | */ | ||
2398 | static void | ||
2399 | overlay_connect_get_config (void *cls, const struct GNUNET_MessageHeader *msg) | ||
2400 | { | ||
2401 | struct OverlayConnectContext *occ = cls; | ||
2402 | const struct GNUNET_TESTBED_PeerConfigurationInformationMessage *cmsg; | ||
2403 | const struct GNUNET_CORE_MessageHandler no_handlers[] = { | ||
2404 | {NULL, 0, 0} | ||
2405 | }; | ||
2406 | |||
2407 | occ->opc = NULL; | ||
2408 | if (GNUNET_MESSAGE_TYPE_TESTBED_PEERCONFIG != ntohs (msg->type)) | ||
2409 | goto error_return; | ||
2410 | cmsg = (const struct GNUNET_TESTBED_PeerConfigurationInformationMessage *) | ||
2411 | msg; | ||
2412 | memcpy (&occ->other_peer_identity, &cmsg->peer_identity, | ||
2413 | sizeof (struct GNUNET_PeerIdentity)); | ||
2414 | GNUNET_free_non_null (occ->emsg); | ||
2415 | occ->emsg = GNUNET_strdup ("Timeout while connecting to CORE"); | ||
2416 | occ->ch = | ||
2417 | GNUNET_CORE_connect (occ->peer->details.local.cfg, occ, &core_startup_cb, | ||
2418 | &overlay_connect_notify, NULL, NULL, GNUNET_NO, NULL, | ||
2419 | GNUNET_NO, no_handlers); | ||
2420 | if (NULL == occ->ch) | ||
2421 | goto error_return; | ||
2422 | return; | ||
2423 | |||
2424 | error_return: | ||
2425 | GNUNET_SCHEDULER_cancel (occ->timeout_task); | ||
2426 | occ->timeout_task = | ||
2427 | GNUNET_SCHEDULER_add_now (&timeout_overlay_connect, occ); | ||
2428 | } | ||
2429 | |||
2430 | |||
2431 | /** | ||
2432 | * Callback to be called when forwarded overlay connection operation has a reply | ||
2433 | * from the sub-controller successfull. We have to relay the reply msg back to | ||
2434 | * the client | ||
2435 | * | ||
2436 | * @param cls ForwardedOperationContext | ||
2437 | * @param msg the peer create success message | ||
2438 | */ | ||
2439 | static void | ||
2440 | forwarded_overlay_connect_listener (void *cls, | ||
2441 | const struct GNUNET_MessageHeader *msg) | ||
2442 | { | ||
2443 | struct ForwardedOperationContext *fopc = cls; | ||
2444 | struct ForwardedOverlayConnectContext *focc; | ||
2445 | struct GNUNET_TESTBED_NeedControllerConfig *rmsg; | ||
2446 | struct Slave *slave; | ||
2447 | uint32_t host_id; | ||
2448 | |||
2449 | if (GNUNET_MESSAGE_TYPE_TESTBED_NEEDCONTROLLERCONFIG != ntohs (msg->type)) | ||
2450 | { | ||
2451 | forwarded_operation_reply_relay (cls, msg); | ||
2452 | return; | ||
2453 | } | ||
2454 | rmsg = (struct GNUNET_TESTBED_NeedControllerConfig *) msg; | ||
2455 | host_id = ntohl (rmsg->controller_host_id); | ||
2456 | if ((host_id >= slave_list_size) || (NULL == slave_list[host_id])) | ||
2457 | { | ||
2458 | /* We too do not have the requested controllers configuration; escalate it | ||
2459 | to our boss */ | ||
2460 | forwarded_operation_reply_relay (cls, msg); | ||
2461 | return; | ||
2462 | } | ||
2463 | slave = slave_list[host_id]; | ||
2464 | GNUNET_assert (NULL != slave->cfg); | ||
2465 | //GNUNET_TESTBED_controller_link_with_opid ( | ||
2466 | } | ||
2467 | |||
2468 | |||
2469 | /** | ||
2470 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_OLCONNECT messages | ||
2471 | * | ||
2472 | * @param cls NULL | ||
2473 | * @param client identification of the client | ||
2474 | * @param message the actual message | ||
2475 | */ | ||
2476 | static void | ||
2477 | handle_overlay_connect (void *cls, struct GNUNET_SERVER_Client *client, | ||
2478 | const struct GNUNET_MessageHeader *message) | ||
2479 | { | ||
2480 | const struct GNUNET_TESTBED_OverlayConnectMessage *msg; | ||
2481 | struct OverlayConnectContext *occ; | ||
2482 | const struct GNUNET_CORE_MessageHandler no_handlers[] = { | ||
2483 | {NULL, 0, 0} | ||
2484 | }; | ||
2485 | struct Peer *peer; | ||
2486 | uint64_t operation_id; | ||
2487 | uint32_t p1; | ||
2488 | uint32_t p2; | ||
2489 | |||
2490 | msg = (const struct GNUNET_TESTBED_OverlayConnectMessage *) message; | ||
2491 | p1 = ntohl (msg->peer1); | ||
2492 | p2 = ntohl (msg->peer2); | ||
2493 | GNUNET_assert (p1 < peer_list_size); | ||
2494 | GNUNET_assert (NULL != peer_list[p1]); | ||
2495 | peer = peer_list[p1]; | ||
2496 | operation_id = GNUNET_ntohll (msg->operation_id); | ||
2497 | if (GNUNET_YES == peer->is_remote) | ||
2498 | { | ||
2499 | struct ForwardedOperationContext *fopc; | ||
2500 | |||
2501 | LOG_DEBUG ("Forwarding overlay connect\n"); | ||
2502 | fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext)); | ||
2503 | GNUNET_SERVER_client_keep (client); | ||
2504 | fopc->client = client; | ||
2505 | fopc->operation_id = operation_id; | ||
2506 | fopc->opc = | ||
2507 | GNUNET_TESTBED_forward_operation_msg_ (peer->details.remote.controller, | ||
2508 | operation_id, message, | ||
2509 | &forwarded_operation_reply_relay, | ||
2510 | fopc); | ||
2511 | fopc->timeout_task = | ||
2512 | GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout, | ||
2513 | fopc); | ||
2514 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2515 | return; | ||
2516 | } | ||
2517 | occ = GNUNET_malloc (sizeof (struct OverlayConnectContext)); | ||
2518 | GNUNET_SERVER_client_keep (client); | ||
2519 | occ->client = client; | ||
2520 | occ->peer_id = p1; | ||
2521 | occ->other_peer_id = p2; | ||
2522 | occ->peer = peer_list[p1]; | ||
2523 | occ->op_id = GNUNET_ntohll (msg->operation_id); | ||
2524 | if ((p2 >= peer_list_size) || (NULL == peer_list[p2])) | ||
2525 | { | ||
2526 | uint32_t peer2_host_id; | ||
2527 | |||
2528 | peer2_host_id = ntohl (msg->peer2_host_id); | ||
2529 | if ((peer2_host_id >= slave_list_size) | ||
2530 | || (NULL ==slave_list[peer2_host_id])) | ||
2531 | { | ||
2532 | struct GNUNET_TESTBED_NeedControllerConfig *reply; | ||
2533 | |||
2534 | GNUNET_free (occ); | ||
2535 | reply = GNUNET_malloc (sizeof (struct | ||
2536 | GNUNET_TESTBED_NeedControllerConfig)); | ||
2537 | reply->header.size = htons (sizeof (struct | ||
2538 | GNUNET_TESTBED_NeedControllerConfig)); | ||
2539 | reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_NEEDCONTROLLERCONFIG); | ||
2540 | reply->controller_host_id = msg->peer2_host_id; | ||
2541 | reply->operation_id = msg->operation_id; | ||
2542 | queue_message (client, &reply->header); | ||
2543 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2544 | return; | ||
2545 | } | ||
2546 | else | ||
2547 | { | ||
2548 | occ->peer2_controller = slave_list[peer2_host_id]->controller; | ||
2549 | if (NULL == occ->peer2_controller) | ||
2550 | { | ||
2551 | GNUNET_break (0); /* What's going on? */ | ||
2552 | GNUNET_SERVER_client_drop (client); | ||
2553 | GNUNET_free (occ); | ||
2554 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
2555 | return; | ||
2556 | } | ||
2557 | } | ||
2558 | } | ||
2559 | else | ||
2560 | { | ||
2561 | if (GNUNET_YES == peer_list[occ->other_peer_id]->is_remote) | ||
2562 | occ->peer2_controller = peer_list[occ->other_peer_id]->details.remote.controller; | ||
2563 | } | ||
2564 | /* Get the identity of the second peer */ | ||
2565 | if (NULL != occ->peer2_controller) | ||
2566 | { | ||
2567 | struct GNUNET_TESTBED_PeerGetConfigurationMessage cmsg; | ||
2568 | |||
2569 | cmsg.header.size = | ||
2570 | htons (sizeof (struct GNUNET_TESTBED_PeerGetConfigurationMessage)); | ||
2571 | cmsg.header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_GETPEERCONFIG); | ||
2572 | cmsg.peer_id = msg->peer2; | ||
2573 | cmsg.operation_id = msg->operation_id; | ||
2574 | occ->opc = | ||
2575 | GNUNET_TESTBED_forward_operation_msg_ (occ->peer2_controller, | ||
2576 | occ->op_id, &cmsg.header, | ||
2577 | &overlay_connect_get_config, | ||
2578 | occ); | ||
2579 | occ->emsg = | ||
2580 | GNUNET_strdup ("Timeout while getting peer identity of peer B\n"); | ||
2581 | occ->timeout_task = | ||
2582 | GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply | ||
2583 | (GNUNET_TIME_UNIT_SECONDS, 30), | ||
2584 | &timeout_overlay_connect, occ); | ||
2585 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2586 | return; | ||
2587 | } | ||
2588 | GNUNET_TESTING_peer_get_identity (peer_list[occ->other_peer_id]->details.local.peer, | ||
2589 | &occ->other_peer_identity); | ||
2590 | /* Connect to the core of 1st peer and wait for the 2nd peer to connect */ | ||
2591 | occ->emsg = GNUNET_strdup ("Timeout while connecting to CORE"); | ||
2592 | occ->ch = | ||
2593 | GNUNET_CORE_connect (occ->peer->details.local.cfg, occ, &core_startup_cb, | ||
2594 | &overlay_connect_notify, NULL, NULL, GNUNET_NO, NULL, | ||
2595 | GNUNET_NO, no_handlers); | ||
2596 | if (NULL == occ->ch) | ||
2597 | occ->timeout_task = | ||
2598 | GNUNET_SCHEDULER_add_now (&timeout_overlay_connect, occ); | ||
2599 | else | ||
2600 | occ->timeout_task = | ||
2601 | GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply | ||
2602 | (GNUNET_TIME_UNIT_SECONDS, 30), | ||
2603 | &timeout_overlay_connect, occ); | ||
2604 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2605 | } | ||
2606 | |||
2607 | |||
2608 | /** | ||
2609 | * Function to cleanup RequestOverlayConnectContext and any associated tasks | ||
2610 | * with it | ||
2611 | * | ||
2612 | * @param rocc the RequestOverlayConnectContext | ||
2613 | */ | ||
2614 | static void | ||
2615 | cleanup_rocc (struct RequestOverlayConnectContext *rocc) | ||
2616 | { | ||
2617 | if (GNUNET_SCHEDULER_NO_TASK != rocc->attempt_connect_task_id) | ||
2618 | GNUNET_SCHEDULER_cancel (rocc->attempt_connect_task_id); | ||
2619 | if (GNUNET_SCHEDULER_NO_TASK != rocc->timeout_rocc_task_id) | ||
2620 | GNUNET_SCHEDULER_cancel (rocc->timeout_rocc_task_id); | ||
2621 | GNUNET_TRANSPORT_disconnect (rocc->th); | ||
2622 | GNUNET_free_non_null (rocc->hello); | ||
2623 | GNUNET_free (rocc); | ||
2624 | } | ||
2625 | |||
2626 | |||
2627 | /** | ||
2628 | * Task to timeout rocc and cleanit up | ||
2629 | * | ||
2630 | * @param cls the RequestOverlayConnectContext | ||
2631 | * @param tc the TaskContext from scheduler | ||
2632 | */ | ||
2633 | static void | ||
2634 | timeout_rocc_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
2635 | { | ||
2636 | struct RequestOverlayConnectContext *rocc = cls; | ||
2637 | |||
2638 | rocc->timeout_rocc_task_id = GNUNET_SCHEDULER_NO_TASK; | ||
2639 | cleanup_rocc (rocc); | ||
2640 | } | ||
2641 | |||
2642 | |||
2643 | /** | ||
2644 | * Function called to notify transport users that another | ||
2645 | * peer connected to us. | ||
2646 | * | ||
2647 | * @param cls closure | ||
2648 | * @param new_peer the peer that connected | ||
2649 | * @param ats performance data | ||
2650 | * @param ats_count number of entries in ats (excluding 0-termination) | ||
2651 | */ | ||
2652 | static void | ||
2653 | transport_connect_notify (void *cls, const struct GNUNET_PeerIdentity *new_peer, | ||
2654 | const struct GNUNET_ATS_Information * ats, | ||
2655 | uint32_t ats_count) | ||
2656 | { | ||
2657 | struct RequestOverlayConnectContext *rocc = cls; | ||
2658 | |||
2659 | LOG_DEBUG ("Request Overlay connect notify\n"); | ||
2660 | if (0 != memcmp (new_peer, &rocc->a_id, sizeof (struct GNUNET_PeerIdentity))) | ||
2661 | return; | ||
2662 | LOG_DEBUG ("Peer %4s connected\n", GNUNET_i2s (&rocc->a_id)); | ||
2663 | cleanup_rocc (rocc); | ||
2664 | } | ||
2665 | |||
2666 | |||
2667 | /** | ||
2668 | * Task to offer the HELLO message to the peer and ask it to connect to the peer | ||
2669 | * whose identity is in RequestOverlayConnectContext | ||
2670 | * | ||
2671 | * @param cls the RequestOverlayConnectContext | ||
2672 | * @param tc the TaskContext from scheduler | ||
2673 | */ | ||
2674 | static void | ||
2675 | attempt_connect_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
2676 | { | ||
2677 | struct RequestOverlayConnectContext *rocc = cls; | ||
2678 | |||
2679 | rocc->attempt_connect_task_id = GNUNET_SCHEDULER_NO_TASK; | ||
2680 | GNUNET_TRANSPORT_offer_hello (rocc->th, rocc->hello, NULL, NULL); | ||
2681 | GNUNET_TRANSPORT_try_connect (rocc->th, &rocc->a_id); | ||
2682 | rocc->attempt_connect_task_id = | ||
2683 | GNUNET_SCHEDULER_add_delayed | ||
2684 | (GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, | ||
2685 | 100 * (pow (2, rocc->retries++))), | ||
2686 | &attempt_connect_task, rocc); | ||
2687 | } | ||
2688 | |||
2689 | |||
2690 | /** | ||
2691 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_REQUESTCONNECT messages | ||
2692 | * | ||
2693 | * @param cls NULL | ||
2694 | * @param client identification of the client | ||
2695 | * @param message the actual message | ||
2696 | */ | ||
2697 | static void | ||
2698 | handle_overlay_request_connect (void *cls, struct GNUNET_SERVER_Client *client, | ||
2699 | const struct GNUNET_MessageHeader *message) | ||
2700 | { | ||
2701 | const struct GNUNET_TESTBED_RequestConnectMessage *msg; | ||
2702 | struct RequestOverlayConnectContext *rocc; | ||
2703 | struct Peer *peer; | ||
2704 | uint32_t peer_id; | ||
2705 | uint16_t msize; | ||
2706 | uint16_t hsize; | ||
2707 | |||
2708 | msize = ntohs (message->size); | ||
2709 | if (sizeof (struct GNUNET_TESTBED_RequestConnectMessage) >= msize) | ||
2710 | { | ||
2711 | GNUNET_break (0); | ||
2712 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
2713 | return; | ||
2714 | } | ||
2715 | msg = (const struct GNUNET_TESTBED_RequestConnectMessage *) message; | ||
2716 | if ((NULL == msg->hello) || | ||
2717 | (GNUNET_MESSAGE_TYPE_HELLO != ntohs (msg->hello->type))) | ||
2718 | { | ||
2719 | GNUNET_break (0); | ||
2720 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
2721 | return; | ||
2722 | } | ||
2723 | hsize = ntohs (msg->hello->size); | ||
2724 | if ((sizeof (struct GNUNET_TESTBED_RequestConnectMessage) + hsize) != msize) | ||
2725 | { | ||
2726 | GNUNET_break (0); | ||
2727 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
2728 | return; | ||
2729 | } | ||
2730 | peer_id = ntohl (msg->peer); | ||
2731 | if ((peer_id >= peer_list_size) || (NULL == (peer = peer_list[peer_id]))) | ||
2732 | { | ||
2733 | GNUNET_break_op (0); | ||
2734 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
2735 | return; | ||
2736 | } | ||
2737 | if (GNUNET_YES == peer->is_remote) | ||
2738 | { | ||
2739 | struct GNUNET_MessageHeader *msg2; | ||
2740 | |||
2741 | msg2 = GNUNET_malloc (msize); | ||
2742 | (void) memcpy (msg2, message, msize); | ||
2743 | GNUNET_TESTBED_queue_message_ (peer->details.remote.controller, msg2); | ||
2744 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2745 | return; | ||
2746 | } | ||
2747 | rocc = GNUNET_malloc (sizeof (struct RequestOverlayConnectContext)); | ||
2748 | rocc->th = GNUNET_TRANSPORT_connect (peer->details.local.cfg, NULL, rocc, | ||
2749 | NULL, &transport_connect_notify, NULL); | ||
2750 | if (NULL == rocc->th) | ||
2751 | { | ||
2752 | GNUNET_break (0); | ||
2753 | GNUNET_free (rocc); | ||
2754 | GNUNET_SERVER_receive_done (client, GNUNET_SYSERR); | ||
2755 | return; | ||
2756 | } | ||
2757 | memcpy (&rocc->a_id, &msg->peer_identity, | ||
2758 | sizeof (struct GNUNET_PeerIdentity)); | ||
2759 | rocc->hello = GNUNET_malloc (hsize); | ||
2760 | memcpy (rocc->hello, msg->hello, hsize); | ||
2761 | rocc->attempt_connect_task_id = | ||
2762 | GNUNET_SCHEDULER_add_now (&attempt_connect_task, rocc); | ||
2763 | rocc->timeout_rocc_task_id = | ||
2764 | GNUNET_SCHEDULER_add_delayed (TIMEOUT, &timeout_rocc_task, rocc); | ||
2765 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2766 | } | ||
2767 | |||
2768 | |||
2769 | /** | ||
2770 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_GETSLAVECONFIG messages | ||
2771 | * | ||
2772 | * @param cls NULL | ||
2773 | * @param client identification of the client | ||
2774 | * @param message the actual message | ||
2775 | */ | ||
2776 | static void | ||
2777 | handle_slave_get_config (void *cls, struct GNUNET_SERVER_Client *client, | ||
2778 | const struct GNUNET_MessageHeader *message) | ||
2779 | { | ||
2780 | struct GNUNET_TESTBED_SlaveGetConfigurationMessage *msg; | ||
2781 | struct Slave *slave; | ||
2782 | struct GNUNET_TESTBED_SlaveConfiguration *reply; | ||
2783 | char *config; | ||
2784 | char *xconfig; | ||
2785 | size_t config_size; | ||
2786 | size_t xconfig_size; | ||
2787 | size_t reply_size; | ||
2788 | uint64_t op_id; | ||
2789 | uint32_t slave_id; | ||
2790 | |||
2791 | msg = (struct GNUNET_TESTBED_SlaveGetConfigurationMessage *) message; | ||
2792 | slave_id = ntohl (msg->slave_id); | ||
2793 | op_id = GNUNET_ntohll (msg->operation_id); | ||
2794 | if ((slave_list_size <= slave_id) || (NULL == slave_list[slave_id])) | ||
2795 | { | ||
2796 | send_operation_fail_msg (client, op_id, "Slave not found"); | ||
2797 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2798 | return; | ||
2799 | } | ||
2800 | slave = slave_list[slave_id]; | ||
2801 | if (NULL == slave->cfg) | ||
2802 | { | ||
2803 | send_operation_fail_msg (client, op_id, | ||
2804 | "Configuration not found (slave not started by me)"); | ||
2805 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2806 | return; | ||
2807 | } | ||
2808 | config = GNUNET_CONFIGURATION_serialize (slave->cfg, &config_size); | ||
2809 | xconfig_size = GNUNET_TESTBED_compress_config_ (config, config_size, | ||
2810 | &xconfig); | ||
2811 | GNUNET_free (config); | ||
2812 | reply_size = xconfig_size + sizeof (struct GNUNET_TESTBED_SlaveConfiguration); | ||
2813 | GNUNET_break (reply_size <= UINT16_MAX); | ||
2814 | GNUNET_break (config_size <= UINT16_MAX); | ||
2815 | reply = GNUNET_realloc (xconfig, reply_size); | ||
2816 | (void) memmove (&reply[1], reply, xconfig_size); | ||
2817 | reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_SLAVECONFIG); | ||
2818 | reply->header.size = htons ((uint16_t) reply_size); | ||
2819 | reply->slave_id = msg->slave_id; | ||
2820 | reply->operation_id = msg->operation_id; | ||
2821 | reply->config_size = htons ((uint16_t) config_size); | ||
2822 | queue_message (client, &reply->header); | ||
2823 | GNUNET_SERVER_receive_done (client, GNUNET_OK); | ||
2824 | } | ||
2825 | |||
2826 | |||
2827 | /** | ||
2828 | * Iterator over hash map entries. | ||
2829 | * | ||
2830 | * @param cls closure | ||
2831 | * @param key current key code | ||
2832 | * @param value value in the hash map | ||
2833 | * @return GNUNET_YES if we should continue to | ||
2834 | * iterate, | ||
2835 | * GNUNET_NO if not. | ||
2836 | */ | ||
2837 | static int | ||
2838 | ss_map_free_iterator (void *cls, const struct GNUNET_HashCode *key, void *value) | ||
2839 | { | ||
2840 | struct SharedService *ss = value; | ||
2841 | |||
2842 | GNUNET_assert (GNUNET_YES == | ||
2843 | GNUNET_CONTAINER_multihashmap_remove (ss_map, key, value)); | ||
2844 | GNUNET_free (ss->name); | ||
2845 | GNUNET_free (ss); | ||
2846 | return GNUNET_YES; | ||
2847 | } | ||
2848 | |||
2849 | |||
2850 | /** | ||
2851 | * Task to clean up and shutdown nicely | ||
2852 | * | ||
2853 | * @param cls NULL | ||
2854 | * @param tc the TaskContext from scheduler | ||
2855 | */ | ||
2856 | static void | ||
2857 | shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
2858 | { | ||
2859 | struct LCFContextQueue *lcfq; | ||
2860 | uint32_t id; | ||
2861 | |||
2862 | shutdown_task_id = GNUNET_SCHEDULER_NO_TASK; | ||
2863 | LOG (GNUNET_ERROR_TYPE_DEBUG, "Shutting down testbed service\n"); | ||
2864 | (void) GNUNET_CONTAINER_multihashmap_iterate (ss_map, &ss_map_free_iterator, | ||
2865 | NULL); | ||
2866 | GNUNET_CONTAINER_multihashmap_destroy (ss_map); | ||
2867 | if (NULL != lcfq_head) | ||
2868 | { | ||
2869 | if (GNUNET_SCHEDULER_NO_TASK != lcf_proc_task_id) | ||
2870 | { | ||
2871 | GNUNET_SCHEDULER_cancel (lcf_proc_task_id); | ||
2872 | lcf_proc_task_id = GNUNET_SCHEDULER_NO_TASK; | ||
2873 | } | ||
2874 | if (NULL != lcfq_head->lcf->rhandle) | ||
2875 | GNUNET_TESTBED_cancel_registration (lcfq_head->lcf->rhandle); | ||
2876 | } | ||
2877 | GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == lcf_proc_task_id); | ||
2878 | for (lcfq = lcfq_head; NULL != lcfq; lcfq = lcfq_head) | ||
2879 | { | ||
2880 | GNUNET_free (lcfq->lcf->msg); | ||
2881 | GNUNET_free (lcfq->lcf); | ||
2882 | GNUNET_CONTAINER_DLL_remove (lcfq_head, lcfq_tail, lcfq); | ||
2883 | GNUNET_free (lcfq); | ||
2884 | } | ||
2885 | /* Clear peer list */ | ||
2886 | for (id = 0; id < peer_list_size; id++) | ||
2887 | if (NULL != peer_list[id]) | ||
2888 | { | ||
2889 | if (GNUNET_NO == peer_list[id]->is_remote) | ||
2890 | { | ||
2891 | if (GNUNET_YES == peer_list[id]->details.local.is_running) | ||
2892 | GNUNET_TESTING_peer_stop (peer_list[id]->details.local.peer); | ||
2893 | GNUNET_TESTING_peer_destroy (peer_list[id]->details.local.peer); | ||
2894 | GNUNET_CONFIGURATION_destroy (peer_list[id]->details.local.cfg); | ||
2895 | } | ||
2896 | GNUNET_free (peer_list[id]); | ||
2897 | } | ||
2898 | GNUNET_free_non_null (peer_list); | ||
2899 | /* Clear host list */ | ||
2900 | for (id = 0; id < host_list_size; id++) | ||
2901 | if (NULL != host_list[id]) | ||
2902 | GNUNET_TESTBED_host_destroy (host_list[id]); | ||
2903 | GNUNET_free_non_null (host_list); | ||
2904 | /* Clear route list */ | ||
2905 | for (id = 0; id < route_list_size; id++) | ||
2906 | if (NULL != route_list[id]) | ||
2907 | GNUNET_free (route_list[id]); | ||
2908 | GNUNET_free_non_null (route_list); | ||
2909 | /* Clear slave_list */ | ||
2910 | for (id = 0; id < slave_list_size; id++) | ||
2911 | if (NULL != slave_list[id]) | ||
2912 | { | ||
2913 | if (NULL != slave_list[id]->cfg) | ||
2914 | GNUNET_CONFIGURATION_destroy (slave_list[id]->cfg); | ||
2915 | if (NULL != slave_list[id]->controller) | ||
2916 | GNUNET_TESTBED_controller_disconnect (slave_list[id]->controller); | ||
2917 | if (NULL != slave_list[id]->controller_proc) | ||
2918 | GNUNET_TESTBED_controller_stop (slave_list[id]->controller_proc); | ||
2919 | GNUNET_free (slave_list[id]); | ||
2920 | } | ||
2921 | GNUNET_free_non_null (slave_list); | ||
2922 | if (NULL != master_context) | ||
2923 | { | ||
2924 | GNUNET_free_non_null (master_context->master_ip); | ||
2925 | if (NULL != master_context->system) | ||
2926 | GNUNET_TESTING_system_destroy (master_context->system, GNUNET_YES); | ||
2927 | GNUNET_free (master_context); | ||
2928 | master_context = NULL; | ||
2929 | } | ||
2930 | GNUNET_free_non_null (hostname); | ||
2931 | } | ||
2932 | |||
2933 | |||
2934 | /** | ||
2935 | * Callback for client disconnect | ||
2936 | * | ||
2937 | * @param cls NULL | ||
2938 | * @param client the client which has disconnected | ||
2939 | */ | ||
2940 | static void | ||
2941 | client_disconnect_cb (void *cls, struct GNUNET_SERVER_Client *client) | ||
2942 | { | ||
2943 | if (NULL == master_context) | ||
2944 | return; | ||
2945 | if (client == master_context->client) | ||
2946 | { | ||
2947 | LOG (GNUNET_ERROR_TYPE_DEBUG, "Master client disconnected\n"); | ||
2948 | GNUNET_SERVER_client_drop (client); | ||
2949 | /* should not be needed as we're terminated by failure to read | ||
2950 | * from stdin, but if stdin fails for some reason, this shouldn't | ||
2951 | * hurt for now --- might need to revise this later if we ever | ||
2952 | * decide that master connections might be temporarily down | ||
2953 | * for some reason */ | ||
2954 | //GNUNET_SCHEDULER_shutdown (); | ||
2955 | } | ||
2956 | } | ||
2957 | |||
2958 | |||
2959 | /** | ||
2960 | * Testbed setup | ||
2961 | * | ||
2962 | * @param cls closure | ||
2963 | * @param server the initialized server | ||
2964 | * @param cfg configuration to use | ||
2965 | */ | ||
2966 | static void | ||
2967 | testbed_run (void *cls, struct GNUNET_SERVER_Handle *server, | ||
2968 | const struct GNUNET_CONFIGURATION_Handle *cfg) | ||
2969 | { | ||
2970 | static const struct GNUNET_SERVER_MessageHandler message_handlers[] = { | ||
2971 | {&handle_init, NULL, GNUNET_MESSAGE_TYPE_TESTBED_INIT, 0}, | ||
2972 | {&handle_add_host, NULL, GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST, 0}, | ||
2973 | {&handle_configure_shared_service, NULL, | ||
2974 | GNUNET_MESSAGE_TYPE_TESTBED_SERVICESHARE, 0}, | ||
2975 | {&handle_link_controllers, NULL, | ||
2976 | GNUNET_MESSAGE_TYPE_TESTBED_LCONTROLLERS, 0}, | ||
2977 | {&handle_peer_create, NULL, GNUNET_MESSAGE_TYPE_TESTBED_CREATEPEER, 0}, | ||
2978 | {&handle_peer_destroy, NULL, GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER, | ||
2979 | sizeof (struct GNUNET_TESTBED_PeerDestroyMessage)}, | ||
2980 | {&handle_peer_start, NULL, GNUNET_MESSAGE_TYPE_TESTBED_STARTPEER, | ||
2981 | sizeof (struct GNUNET_TESTBED_PeerStartMessage)}, | ||
2982 | {&handle_peer_stop, NULL, GNUNET_MESSAGE_TYPE_TESTBED_STOPPEER, | ||
2983 | sizeof (struct GNUNET_TESTBED_PeerStopMessage)}, | ||
2984 | {&handle_peer_get_config, NULL, GNUNET_MESSAGE_TYPE_TESTBED_GETPEERCONFIG, | ||
2985 | sizeof (struct GNUNET_TESTBED_PeerGetConfigurationMessage)}, | ||
2986 | {&handle_overlay_connect, NULL, GNUNET_MESSAGE_TYPE_TESTBED_OLCONNECT, | ||
2987 | sizeof (struct GNUNET_TESTBED_OverlayConnectMessage)}, | ||
2988 | {&handle_overlay_request_connect, NULL, GNUNET_MESSAGE_TYPE_TESTBED_REQUESTCONNECT, | ||
2989 | 0}, | ||
2990 | {handle_slave_get_config, NULL, GNUNET_MESSAGE_TYPE_TESTBED_GETSLAVECONFIG, | ||
2991 | sizeof (struct GNUNET_TESTBED_SlaveGetConfigurationMessage)}, | ||
2992 | {NULL} | ||
2993 | }; | ||
2994 | |||
2995 | GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_get_value_string | ||
2996 | (cfg, "testbed", "HOSTNAME", &hostname)); | ||
2997 | GNUNET_SERVER_add_handlers (server, message_handlers); | ||
2998 | GNUNET_SERVER_disconnect_notify (server, &client_disconnect_cb, NULL); | ||
2999 | ss_map = GNUNET_CONTAINER_multihashmap_create (5, GNUNET_NO); | ||
3000 | shutdown_task_id = | ||
3001 | GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, | ||
3002 | &shutdown_task, NULL); | ||
3003 | LOG_DEBUG ("Testbed startup complete\n"); | ||
3004 | } | ||
3005 | |||
3006 | |||
3007 | /** | ||
3008 | * The starting point of execution | ||
3009 | */ | ||
3010 | int | ||
3011 | main (int argc, char *const *argv) | ||
3012 | { | ||
3013 | //sleep (15); /* Debugging */ | ||
3014 | return (GNUNET_OK == | ||
3015 | GNUNET_SERVICE_run (argc, argv, "testbed", GNUNET_SERVICE_OPTION_NONE, | ||
3016 | &testbed_run, NULL)) ? 0 : 1; | ||
3017 | } | ||
3018 | |||
3019 | /* end of gnunet-service-testbed.c */ | ||
diff --git a/src/testbed/testbed.h b/src/testbed/testbed.h index ac8fed0a6..198f764b3 100644 --- a/src/testbed/testbed.h +++ b/src/testbed/testbed.h | |||
@@ -701,6 +701,30 @@ struct GNUNET_TESTBED_SlaveConfiguration | |||
701 | 701 | ||
702 | }; | 702 | }; |
703 | 703 | ||
704 | |||
705 | /** | ||
706 | * Message sent from a controller to the testbed API seeking the configuration | ||
707 | * of the host whose id is contained in the message. | ||
708 | */ | ||
709 | struct GNUNET_TESTBED_NeedControllerConfig | ||
710 | { | ||
711 | /** | ||
712 | * Type is GNUNET_MESSAGE_TYPE_TESTBED_NEEDCONTROLLERCONFIG | ||
713 | */ | ||
714 | struct GNUNET_MessageHeader header; | ||
715 | |||
716 | /** | ||
717 | * The id of the controller host | ||
718 | */ | ||
719 | uint32_t controller_host_id GNUNET_PACKED; | ||
720 | |||
721 | /** | ||
722 | * Operation ID | ||
723 | */ | ||
724 | uint64_t operation_id GNUNET_PACKED; | ||
725 | |||
726 | }; | ||
727 | |||
704 | GNUNET_NETWORK_STRUCT_END | 728 | GNUNET_NETWORK_STRUCT_END |
705 | 729 | ||
706 | #endif | 730 | #endif |
diff --git a/src/testbed/testbed_api.c b/src/testbed/testbed_api.c index 1d99c7e3f..35956f852 100644 --- a/src/testbed/testbed_api.c +++ b/src/testbed/testbed_api.c | |||
@@ -859,6 +859,46 @@ handle_slave_config (struct GNUNET_TESTBED_Controller *c, | |||
859 | 859 | ||
860 | 860 | ||
861 | /** | 861 | /** |
862 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_NEEDCONTROLLERCONFIG message from | ||
863 | * controller (testbed service) | ||
864 | * | ||
865 | * @param c the controller handler | ||
866 | * @param msg message received | ||
867 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
868 | * not | ||
869 | */ | ||
870 | static int | ||
871 | handle_need_controller_config (struct GNUNET_TESTBED_Controller *c, | ||
872 | const struct GNUNET_TESTBED_NeedControllerConfig * msg) | ||
873 | { | ||
874 | struct OperationContext *opc; | ||
875 | struct OverlayConnectData *oc_data; | ||
876 | uint64_t op_id; | ||
877 | |||
878 | op_id = GNUNET_ntohll (msg->operation_id); | ||
879 | if (NULL == (opc = find_opc (c, op_id))) | ||
880 | { | ||
881 | LOG_DEBUG ("Operation not found\n"); | ||
882 | return GNUNET_YES; | ||
883 | } | ||
884 | if (OP_FORWARDED == opc->type) | ||
885 | { | ||
886 | handle_forwarded_operation_msg (c, opc, | ||
887 | (const struct GNUNET_MessageHeader *) msg); | ||
888 | return GNUNET_YES; | ||
889 | } | ||
890 | oc_data = opc->data; | ||
891 | /* FIXME: Should spawn operations to: | ||
892 | 1. Acquire configuration of peer2's controller, | ||
893 | 2. link peer1's controller to peer2's controller | ||
894 | 3. ask them to attempt overlay connect on peer1 and peer2 again */ | ||
895 | GNUNET_break (NULL == oc_data); | ||
896 | GNUNET_break (0); | ||
897 | return GNUNET_YES; | ||
898 | } | ||
899 | |||
900 | |||
901 | /** | ||
862 | * Handler for messages from controller (testbed service) | 902 | * Handler for messages from controller (testbed service) |
863 | * | 903 | * |
864 | * @param cls the controller handler | 904 | * @param cls the controller handler |
@@ -952,6 +992,14 @@ message_handler (void *cls, const struct GNUNET_MessageHeader *msg) | |||
952 | handle_slave_config (c, (const struct | 992 | handle_slave_config (c, (const struct |
953 | GNUNET_TESTBED_SlaveConfiguration *) msg); | 993 | GNUNET_TESTBED_SlaveConfiguration *) msg); |
954 | break; | 994 | break; |
995 | case GNUNET_MESSAGE_TYPE_TESTBED_NEEDCONTROLLERCONFIG: | ||
996 | GNUNET_assert (msize == sizeof (struct | ||
997 | GNUNET_TESTBED_NeedControllerConfig)); | ||
998 | status = | ||
999 | handle_need_controller_config (c, (const struct | ||
1000 | GNUNET_TESTBED_NeedControllerConfig | ||
1001 | *) msg); | ||
1002 | break; | ||
955 | default: | 1003 | default: |
956 | GNUNET_assert (0); | 1004 | GNUNET_assert (0); |
957 | } | 1005 | } |
diff --git a/src/testbed/testbed_api.c.new b/src/testbed/testbed_api.c.new new file mode 100644 index 000000000..0bc6830ec --- /dev/null +++ b/src/testbed/testbed_api.c.new | |||
@@ -0,0 +1,2222 @@ | |||
1 | /* | ||
2 | This file is part of GNUnet | ||
3 | (C) 2008--2012 Christian Grothoff (and other contributing authors) | ||
4 | |||
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. | ||
9 | |||
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. | ||
14 | |||
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. | ||
19 | */ | ||
20 | |||
21 | /** | ||
22 | * @file testbed/testbed_api.c | ||
23 | * @brief API for accessing the GNUnet testing service. | ||
24 | * This library is supposed to make it easier to write | ||
25 | * testcases and script large-scale benchmarks. | ||
26 | * @author Christian Grothoff | ||
27 | * @author Sree Harsha Totakura | ||
28 | */ | ||
29 | |||
30 | |||
31 | #include "platform.h" | ||
32 | #include "gnunet_testbed_service.h" | ||
33 | #include "gnunet_core_service.h" | ||
34 | #include "gnunet_constants.h" | ||
35 | #include "gnunet_transport_service.h" | ||
36 | #include "gnunet_hello_lib.h" | ||
37 | #include <zlib.h> | ||
38 | |||
39 | #include "testbed.h" | ||
40 | #include "testbed_api.h" | ||
41 | #include "testbed_api_hosts.h" | ||
42 | #include "testbed_api_peers.h" | ||
43 | #include "testbed_api_operations.h" | ||
44 | |||
45 | /** | ||
46 | * Generic logging shorthand | ||
47 | */ | ||
48 | #define LOG(kind, ...) \ | ||
49 | GNUNET_log_from (kind, "testbed-api", __VA_ARGS__); | ||
50 | |||
51 | /** | ||
52 | * Debug logging | ||
53 | */ | ||
54 | #define LOG_DEBUG(...) \ | ||
55 | LOG (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__); | ||
56 | |||
57 | /** | ||
58 | * Relative time seconds shorthand | ||
59 | */ | ||
60 | #define TIME_REL_SECS(sec) \ | ||
61 | GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, sec) | ||
62 | |||
63 | |||
64 | /** | ||
65 | * Default server message sending retry timeout | ||
66 | */ | ||
67 | #define TIMEOUT_REL TIME_REL_SECS(1) | ||
68 | |||
69 | |||
70 | /** | ||
71 | * Testbed Helper binary name | ||
72 | */ | ||
73 | #define HELPER_TESTBED_BINARY "gnunet-helper-testbed" | ||
74 | #define HELPER_TESTBED_BINARY_SSH ". ~/.bashrc; gnunet-helper-testbed" | ||
75 | |||
76 | |||
77 | /** | ||
78 | * Handle for controller process | ||
79 | */ | ||
80 | struct GNUNET_TESTBED_ControllerProc | ||
81 | { | ||
82 | /** | ||
83 | * The process handle | ||
84 | */ | ||
85 | struct GNUNET_HELPER_Handle *helper; | ||
86 | |||
87 | /** | ||
88 | * The host where the helper is run | ||
89 | */ | ||
90 | struct GNUNET_TESTBED_Host *host; | ||
91 | |||
92 | /** | ||
93 | * The controller error callback | ||
94 | */ | ||
95 | GNUNET_TESTBED_ControllerStatusCallback cb; | ||
96 | |||
97 | /** | ||
98 | * The closure for the above callback | ||
99 | */ | ||
100 | void *cls; | ||
101 | |||
102 | /** | ||
103 | * The send handle for the helper | ||
104 | */ | ||
105 | struct GNUNET_HELPER_SendHandle *shandle; | ||
106 | |||
107 | /** | ||
108 | * The message corresponding to send handle | ||
109 | */ | ||
110 | struct GNUNET_MessageHeader *msg; | ||
111 | |||
112 | /** | ||
113 | * The port number for ssh; used for helpers starting ssh | ||
114 | */ | ||
115 | char *port; | ||
116 | |||
117 | /** | ||
118 | * The ssh destination string; used for helpers starting ssh | ||
119 | */ | ||
120 | char *dst; | ||
121 | |||
122 | /** | ||
123 | * The configuration of the running testbed service | ||
124 | */ | ||
125 | struct GNUNET_CONFIGURATION_Handle *cfg; | ||
126 | |||
127 | }; | ||
128 | |||
129 | |||
130 | /** | ||
131 | * The message queue for sending messages to the controller service | ||
132 | */ | ||
133 | struct MessageQueue | ||
134 | { | ||
135 | /** | ||
136 | * The message to be sent | ||
137 | */ | ||
138 | struct GNUNET_MessageHeader *msg; | ||
139 | |||
140 | /** | ||
141 | * next pointer for DLL | ||
142 | */ | ||
143 | struct MessageQueue *next; | ||
144 | |||
145 | /** | ||
146 | * prev pointer for DLL | ||
147 | */ | ||
148 | struct MessageQueue *prev; | ||
149 | }; | ||
150 | |||
151 | |||
152 | /** | ||
153 | * Structure for a controller link | ||
154 | */ | ||
155 | struct ControllerLink | ||
156 | { | ||
157 | /** | ||
158 | * The next ptr for DLL | ||
159 | */ | ||
160 | struct ControllerLink *next; | ||
161 | |||
162 | /** | ||
163 | * The prev ptr for DLL | ||
164 | */ | ||
165 | struct ControllerLink *prev; | ||
166 | |||
167 | /** | ||
168 | * The host which will be referred in the peer start request. This is the | ||
169 | * host where the peer should be started | ||
170 | */ | ||
171 | struct GNUNET_TESTBED_Host *delegated_host; | ||
172 | |||
173 | /** | ||
174 | * The host which will contacted to delegate the peer start request | ||
175 | */ | ||
176 | struct GNUNET_TESTBED_Host *slave_host; | ||
177 | |||
178 | /** | ||
179 | * The configuration to be used to connect to slave host | ||
180 | */ | ||
181 | const struct GNUNET_CONFIGURATION_Handle *slave_cfg; | ||
182 | |||
183 | /** | ||
184 | * GNUNET_YES if the slave should be started (and stopped) by us; GNUNET_NO | ||
185 | * if we are just allowed to use the slave via TCP/IP | ||
186 | */ | ||
187 | int is_subordinate; | ||
188 | }; | ||
189 | |||
190 | |||
191 | /** | ||
192 | * handle for host registration | ||
193 | */ | ||
194 | struct GNUNET_TESTBED_HostRegistrationHandle | ||
195 | { | ||
196 | /** | ||
197 | * The host being registered | ||
198 | */ | ||
199 | struct GNUNET_TESTBED_Host *host; | ||
200 | |||
201 | /** | ||
202 | * The controller at which this host is being registered | ||
203 | */ | ||
204 | struct GNUNET_TESTBED_Controller *c; | ||
205 | |||
206 | /** | ||
207 | * The Registartion completion callback | ||
208 | */ | ||
209 | GNUNET_TESTBED_HostRegistrationCompletion cc; | ||
210 | |||
211 | /** | ||
212 | * The closure for above callback | ||
213 | */ | ||
214 | void *cc_cls; | ||
215 | }; | ||
216 | |||
217 | |||
218 | /** | ||
219 | * Context data for forwarded Operation | ||
220 | */ | ||
221 | struct ForwardedOperationData | ||
222 | { | ||
223 | |||
224 | /** | ||
225 | * The callback to call when reply is available | ||
226 | */ | ||
227 | GNUNET_CLIENT_MessageHandler cc; | ||
228 | |||
229 | /** | ||
230 | * The closure for the above callback | ||
231 | */ | ||
232 | void *cc_cls; | ||
233 | |||
234 | }; | ||
235 | |||
236 | |||
237 | /** | ||
238 | * Context data for get slave config operations | ||
239 | */ | ||
240 | struct GetSlaveConfigData | ||
241 | { | ||
242 | /** | ||
243 | * The operation closure | ||
244 | */ | ||
245 | void *op_cls; | ||
246 | |||
247 | /** | ||
248 | * The id of the slave controller | ||
249 | */ | ||
250 | uint32_t slave_id; | ||
251 | |||
252 | }; | ||
253 | |||
254 | |||
255 | /** | ||
256 | * Context data for controller link operations | ||
257 | */ | ||
258 | struct ControllerLinkData | ||
259 | { | ||
260 | /** | ||
261 | * The controller link message | ||
262 | */ | ||
263 | struct GNUNET_TESTBED_ControllerLinkMessage *msg; | ||
264 | |||
265 | /** | ||
266 | * The operation closure | ||
267 | */ | ||
268 | void *op_cls; | ||
269 | |||
270 | }; | ||
271 | |||
272 | |||
273 | /** | ||
274 | * Returns the operation context with the given id if found in the Operation | ||
275 | * context queues of the controller | ||
276 | * | ||
277 | * @param c the controller whose queues are searched | ||
278 | * @param id the id which has to be checked | ||
279 | * @return the matching operation context; NULL if no match found | ||
280 | */ | ||
281 | static struct OperationContext * | ||
282 | find_opc (const struct GNUNET_TESTBED_Controller *c, const uint64_t id) | ||
283 | { | ||
284 | struct OperationContext *opc; | ||
285 | |||
286 | for (opc = c->ocq_head; NULL != opc; opc = opc->next) | ||
287 | { | ||
288 | if (id == opc->id) | ||
289 | return opc; | ||
290 | } | ||
291 | return NULL; | ||
292 | } | ||
293 | |||
294 | |||
295 | /** | ||
296 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_ADDHOSTCONFIRM message from | ||
297 | * controller (testbed service) | ||
298 | * | ||
299 | * @param c the controller handler | ||
300 | * @param msg message received | ||
301 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
302 | * not | ||
303 | */ | ||
304 | static int | ||
305 | handle_addhostconfirm (struct GNUNET_TESTBED_Controller *c, | ||
306 | const struct GNUNET_TESTBED_HostConfirmedMessage *msg) | ||
307 | { | ||
308 | struct GNUNET_TESTBED_HostRegistrationHandle *rh; | ||
309 | char *emsg; | ||
310 | uint16_t msg_size; | ||
311 | |||
312 | rh = c->rh; | ||
313 | if (NULL == rh) | ||
314 | { | ||
315 | return GNUNET_OK; | ||
316 | } | ||
317 | if (GNUNET_TESTBED_host_get_id_ (rh->host) != ntohl (msg->host_id)) | ||
318 | { | ||
319 | LOG_DEBUG ("Mismatch in host id's %u, %u of host confirm msg\n", | ||
320 | GNUNET_TESTBED_host_get_id_ (rh->host), ntohl (msg->host_id)); | ||
321 | return GNUNET_OK; | ||
322 | } | ||
323 | c->rh = NULL; | ||
324 | msg_size = ntohs (msg->header.size); | ||
325 | if (sizeof (struct GNUNET_TESTBED_HostConfirmedMessage) == msg_size) | ||
326 | { | ||
327 | LOG_DEBUG ("Host %u successfully registered\n", ntohl (msg->host_id)); | ||
328 | GNUNET_TESTBED_mark_host_registered_at_ (rh->host, c); | ||
329 | rh->cc (rh->cc_cls, NULL); | ||
330 | GNUNET_free (rh); | ||
331 | return GNUNET_OK; | ||
332 | } | ||
333 | /* We have an error message */ | ||
334 | emsg = (char *) &msg[1]; | ||
335 | if ('\0' != | ||
336 | emsg[msg_size - sizeof (struct GNUNET_TESTBED_HostConfirmedMessage)]) | ||
337 | { | ||
338 | GNUNET_break (0); | ||
339 | GNUNET_free (rh); | ||
340 | return GNUNET_NO; | ||
341 | } | ||
342 | LOG (GNUNET_ERROR_TYPE_ERROR, _("Adding host %u failed with error: %s\n"), | ||
343 | ntohl (msg->host_id), emsg); | ||
344 | rh->cc (rh->cc_cls, emsg); | ||
345 | GNUNET_free (rh); | ||
346 | return GNUNET_OK; | ||
347 | } | ||
348 | |||
349 | |||
350 | /** | ||
351 | * Handler for forwarded operations | ||
352 | * | ||
353 | * @param c the controller handle | ||
354 | * @param opc the opearation context | ||
355 | * @param msg the message | ||
356 | */ | ||
357 | static void | ||
358 | handle_forwarded_operation_msg (struct GNUNET_TESTBED_Controller *c, | ||
359 | struct OperationContext *opc, | ||
360 | const struct GNUNET_MessageHeader *msg) | ||
361 | { | ||
362 | struct ForwardedOperationData *fo_data; | ||
363 | |||
364 | fo_data = opc->data; | ||
365 | if (NULL != fo_data->cc) | ||
366 | fo_data->cc (fo_data->cc_cls, msg); | ||
367 | GNUNET_CONTAINER_DLL_remove (c->ocq_head, c->ocq_tail, opc); | ||
368 | GNUNET_free (fo_data); | ||
369 | GNUNET_free (opc); | ||
370 | } | ||
371 | |||
372 | |||
373 | /** | ||
374 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_ADDHOSTCONFIRM message from | ||
375 | * controller (testbed service) | ||
376 | * | ||
377 | * @param c the controller handler | ||
378 | * @param msg message received | ||
379 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
380 | * not | ||
381 | */ | ||
382 | static int | ||
383 | handle_opsuccess (struct GNUNET_TESTBED_Controller *c, | ||
384 | const struct | ||
385 | GNUNET_TESTBED_GenericOperationSuccessEventMessage *msg) | ||
386 | { | ||
387 | struct OperationContext *opc; | ||
388 | struct GNUNET_TESTBED_EventInformation event; | ||
389 | uint64_t op_id; | ||
390 | |||
391 | op_id = GNUNET_ntohll (msg->operation_id); | ||
392 | LOG_DEBUG ("Operation %ul successful\n", op_id); | ||
393 | if (NULL == (opc = find_opc (c, op_id))) | ||
394 | { | ||
395 | LOG_DEBUG ("Operation not found\n"); | ||
396 | return GNUNET_YES; | ||
397 | } | ||
398 | event.type = GNUNET_TESTBED_ET_OPERATION_FINISHED; | ||
399 | event.details.operation_finished.operation = opc->op; | ||
400 | event.details.operation_finished.op_cls = NULL; | ||
401 | event.details.operation_finished.emsg = NULL; | ||
402 | event.details.operation_finished.generic = NULL; | ||
403 | switch (opc->type) | ||
404 | { | ||
405 | case OP_FORWARDED: | ||
406 | { | ||
407 | handle_forwarded_operation_msg | ||
408 | (c, opc, (const struct GNUNET_MessageHeader *) msg); | ||
409 | return GNUNET_YES; | ||
410 | } | ||
411 | break; | ||
412 | case OP_PEER_DESTROY: | ||
413 | { | ||
414 | struct GNUNET_TESTBED_Peer *peer; | ||
415 | |||
416 | peer = opc->data; | ||
417 | GNUNET_free (peer); | ||
418 | opc->data = NULL; | ||
419 | //PEERDESTROYDATA | ||
420 | } | ||
421 | break; | ||
422 | case OP_LINK_CONTROLLERS: | ||
423 | { | ||
424 | struct ControllerLinkData *data; | ||
425 | |||
426 | data = opc->data; | ||
427 | GNUNET_assert (NULL != data); | ||
428 | event.details.operation_finished.op_cls = data->op_cls; | ||
429 | GNUNET_free (data); | ||
430 | opc->data = NULL; | ||
431 | } | ||
432 | break; | ||
433 | default: | ||
434 | GNUNET_assert (0); | ||
435 | } | ||
436 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
437 | opc->state = OPC_STATE_FINISHED; | ||
438 | if (0 != (c->event_mask & (1L << GNUNET_TESTBED_ET_OPERATION_FINISHED))) | ||
439 | { | ||
440 | if (NULL != c->cc) | ||
441 | c->cc (c->cc_cls, &event); | ||
442 | } | ||
443 | return GNUNET_YES; | ||
444 | } | ||
445 | |||
446 | |||
447 | /** | ||
448 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_PEERCREATESUCCESS message from | ||
449 | * controller (testbed service) | ||
450 | * | ||
451 | * @param c the controller handle | ||
452 | * @param msg message received | ||
453 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
454 | * not | ||
455 | */ | ||
456 | static int | ||
457 | handle_peer_create_success (struct GNUNET_TESTBED_Controller *c, | ||
458 | const struct | ||
459 | GNUNET_TESTBED_PeerCreateSuccessEventMessage *msg) | ||
460 | { | ||
461 | struct OperationContext *opc; | ||
462 | struct PeerCreateData *data; | ||
463 | struct GNUNET_TESTBED_Peer *peer; | ||
464 | GNUNET_TESTBED_PeerCreateCallback cb; | ||
465 | void *cls; | ||
466 | uint64_t op_id; | ||
467 | |||
468 | GNUNET_assert (sizeof (struct GNUNET_TESTBED_PeerCreateSuccessEventMessage) == | ||
469 | ntohs (msg->header.size)); | ||
470 | op_id = GNUNET_ntohll (msg->operation_id); | ||
471 | if (NULL == (opc = find_opc (c, op_id))) | ||
472 | { | ||
473 | LOG_DEBUG ("Operation context for PeerCreateSuccessEvent not found\n"); | ||
474 | return GNUNET_YES; | ||
475 | } | ||
476 | if (OP_FORWARDED == opc->type) | ||
477 | { | ||
478 | handle_forwarded_operation_msg (c, opc, | ||
479 | (const struct GNUNET_MessageHeader *) msg); | ||
480 | return GNUNET_YES; | ||
481 | } | ||
482 | GNUNET_assert (OP_PEER_CREATE == opc->type); | ||
483 | GNUNET_assert (NULL != opc->data); | ||
484 | data = opc->data; | ||
485 | GNUNET_assert (NULL != data->peer); | ||
486 | peer = data->peer; | ||
487 | GNUNET_assert (peer->unique_id == ntohl (msg->peer_id)); | ||
488 | peer->state = PS_CREATED; | ||
489 | cb = data->cb; | ||
490 | cls = data->cls; | ||
491 | GNUNET_free (opc->data); | ||
492 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
493 | opc->state = OPC_STATE_FINISHED; | ||
494 | if (NULL != cb) | ||
495 | cb (cls, peer, NULL); | ||
496 | return GNUNET_YES; | ||
497 | } | ||
498 | |||
499 | |||
500 | /** | ||
501 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_PEEREVENT message from | ||
502 | * controller (testbed service) | ||
503 | * | ||
504 | * @param c the controller handler | ||
505 | * @param msg message received | ||
506 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
507 | * not | ||
508 | */ | ||
509 | static int | ||
510 | handle_peer_event (struct GNUNET_TESTBED_Controller *c, | ||
511 | const struct GNUNET_TESTBED_PeerEventMessage *msg) | ||
512 | { | ||
513 | struct OperationContext *opc; | ||
514 | struct GNUNET_TESTBED_Peer *peer; | ||
515 | struct PeerEventData *data; | ||
516 | GNUNET_TESTBED_PeerChurnCallback pcc; | ||
517 | void *pcc_cls; | ||
518 | struct GNUNET_TESTBED_EventInformation event; | ||
519 | uint64_t op_id; | ||
520 | |||
521 | GNUNET_assert (sizeof (struct GNUNET_TESTBED_PeerEventMessage) == | ||
522 | ntohs (msg->header.size)); | ||
523 | op_id = GNUNET_ntohll (msg->operation_id); | ||
524 | if (NULL == (opc = find_opc (c, op_id))) | ||
525 | { | ||
526 | LOG_DEBUG ("Operation not found\n"); | ||
527 | return GNUNET_YES; | ||
528 | } | ||
529 | if (OP_FORWARDED == opc->type) | ||
530 | { | ||
531 | handle_forwarded_operation_msg (c, opc, | ||
532 | (const struct GNUNET_MessageHeader *) msg); | ||
533 | return GNUNET_YES; | ||
534 | } | ||
535 | GNUNET_assert ((OP_PEER_START == opc->type) || (OP_PEER_STOP == opc->type)); | ||
536 | data = opc->data; | ||
537 | GNUNET_assert (NULL != data); | ||
538 | peer = data->peer; | ||
539 | GNUNET_assert (NULL != peer); | ||
540 | event.type = (enum GNUNET_TESTBED_EventType) ntohl (msg->event_type); | ||
541 | switch (event.type) | ||
542 | { | ||
543 | case GNUNET_TESTBED_ET_PEER_START: | ||
544 | peer->state = PS_STARTED; | ||
545 | event.details.peer_start.host = peer->host; | ||
546 | event.details.peer_start.peer = peer; | ||
547 | break; | ||
548 | case GNUNET_TESTBED_ET_PEER_STOP: | ||
549 | peer->state = PS_STOPPED; | ||
550 | event.details.peer_stop.peer = peer; | ||
551 | break; | ||
552 | default: | ||
553 | GNUNET_assert (0); /* We should never reach this state */ | ||
554 | } | ||
555 | pcc = data->pcc; | ||
556 | pcc_cls = data->pcc_cls; | ||
557 | GNUNET_free (data); | ||
558 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
559 | opc->state = OPC_STATE_FINISHED; | ||
560 | if (0 != | ||
561 | ((GNUNET_TESTBED_ET_PEER_START | GNUNET_TESTBED_ET_PEER_STOP) & | ||
562 | c->event_mask)) | ||
563 | { | ||
564 | if (NULL != c->cc) | ||
565 | c->cc (c->cc_cls, &event); | ||
566 | } | ||
567 | if (NULL != pcc) | ||
568 | pcc (pcc_cls, NULL); | ||
569 | return GNUNET_YES; | ||
570 | } | ||
571 | |||
572 | |||
573 | /** | ||
574 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_PEERCONEVENT message from | ||
575 | * controller (testbed service) | ||
576 | * | ||
577 | * @param c the controller handler | ||
578 | * @param msg message received | ||
579 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
580 | * not | ||
581 | */ | ||
582 | static int | ||
583 | handle_peer_conevent (struct GNUNET_TESTBED_Controller *c, | ||
584 | const struct GNUNET_TESTBED_ConnectionEventMessage *msg) | ||
585 | { | ||
586 | struct OperationContext *opc; | ||
587 | struct OverlayConnectData *data; | ||
588 | GNUNET_TESTBED_OperationCompletionCallback cb; | ||
589 | void *cb_cls; | ||
590 | struct GNUNET_TESTBED_EventInformation event; | ||
591 | uint64_t op_id; | ||
592 | |||
593 | op_id = GNUNET_ntohll (msg->operation_id); | ||
594 | if (NULL == (opc = find_opc (c, op_id))) | ||
595 | { | ||
596 | LOG_DEBUG ("Operation not found\n"); | ||
597 | return GNUNET_YES; | ||
598 | } | ||
599 | if (OP_FORWARDED == opc->type) | ||
600 | { | ||
601 | handle_forwarded_operation_msg (c, opc, | ||
602 | (const struct GNUNET_MessageHeader *) msg); | ||
603 | return GNUNET_YES; | ||
604 | } | ||
605 | GNUNET_assert (OP_OVERLAY_CONNECT == opc->type); | ||
606 | data = opc->data; | ||
607 | GNUNET_assert (NULL != data); | ||
608 | GNUNET_assert ((ntohl (msg->peer1) == data->p1->unique_id) && | ||
609 | (ntohl (msg->peer2) == data->p2->unique_id)); | ||
610 | event.type = (enum GNUNET_TESTBED_EventType) ntohl (msg->event_type); | ||
611 | switch (event.type) | ||
612 | { | ||
613 | case GNUNET_TESTBED_ET_CONNECT: | ||
614 | event.details.peer_connect.peer1 = data->p1; | ||
615 | event.details.peer_connect.peer2 = data->p2; | ||
616 | break; | ||
617 | case GNUNET_TESTBED_ET_DISCONNECT: | ||
618 | GNUNET_assert (0); /* FIXME: implement */ | ||
619 | break; | ||
620 | default: | ||
621 | GNUNET_assert (0); /* Should never reach here */ | ||
622 | break; | ||
623 | } | ||
624 | cb = data->cb; | ||
625 | cb_cls = data->cb_cls; | ||
626 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
627 | opc->state = OPC_STATE_FINISHED; | ||
628 | GNUNET_free (data); | ||
629 | if (0 != | ||
630 | ((GNUNET_TESTBED_ET_CONNECT | GNUNET_TESTBED_ET_DISCONNECT) & | ||
631 | c->event_mask)) | ||
632 | { | ||
633 | if (NULL != c->cc) | ||
634 | c->cc (c->cc_cls, &event); | ||
635 | } | ||
636 | if (NULL != cb) | ||
637 | cb (cb_cls, opc->op, NULL); | ||
638 | return GNUNET_YES; | ||
639 | } | ||
640 | |||
641 | |||
642 | /** | ||
643 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_PEERCONFIG message from | ||
644 | * controller (testbed service) | ||
645 | * | ||
646 | * @param c the controller handler | ||
647 | * @param msg message received | ||
648 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
649 | * not | ||
650 | */ | ||
651 | static int | ||
652 | handle_peer_config (struct GNUNET_TESTBED_Controller *c, | ||
653 | const struct | ||
654 | GNUNET_TESTBED_PeerConfigurationInformationMessage *msg) | ||
655 | { | ||
656 | struct OperationContext *opc; | ||
657 | struct GNUNET_TESTBED_Peer *peer; | ||
658 | struct PeerInfoData *data; | ||
659 | struct GNUNET_TESTBED_PeerInformation *pinfo; | ||
660 | GNUNET_TESTBED_PeerInfoCallback cb; | ||
661 | void *cb_cls; | ||
662 | uint64_t op_id; | ||
663 | |||
664 | op_id = GNUNET_ntohll (msg->operation_id); | ||
665 | if (NULL == (opc = find_opc (c, op_id))) | ||
666 | { | ||
667 | LOG_DEBUG ("Operation not found\n"); | ||
668 | return GNUNET_YES; | ||
669 | } | ||
670 | if (OP_FORWARDED == opc->type) | ||
671 | { | ||
672 | handle_forwarded_operation_msg (c, opc, | ||
673 | (const struct GNUNET_MessageHeader *) msg); | ||
674 | return GNUNET_YES; | ||
675 | } | ||
676 | data = opc->data; | ||
677 | GNUNET_assert (NULL != data); | ||
678 | peer = data->peer; | ||
679 | GNUNET_assert (NULL != peer); | ||
680 | GNUNET_assert (ntohl (msg->peer_id) == peer->unique_id); | ||
681 | pinfo = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_PeerInformation)); | ||
682 | pinfo->pit = data->pit; | ||
683 | cb = data->cb; | ||
684 | cb_cls = data->cb_cls; | ||
685 | GNUNET_free (data); | ||
686 | opc->data = NULL; | ||
687 | switch (pinfo->pit) | ||
688 | { | ||
689 | case GNUNET_TESTBED_PIT_IDENTITY: | ||
690 | pinfo->result.id = GNUNET_malloc (sizeof (struct GNUNET_PeerIdentity)); | ||
691 | (void) memcpy (pinfo->result.id, &msg->peer_identity, | ||
692 | sizeof (struct GNUNET_PeerIdentity)); | ||
693 | break; | ||
694 | case GNUNET_TESTBED_PIT_CONFIGURATION: | ||
695 | pinfo->result.cfg = /* Freed in oprelease_peer_getinfo */ | ||
696 | GNUNET_TESTBED_extract_config_ (&msg->header); | ||
697 | break; | ||
698 | case GNUNET_TESTBED_PIT_GENERIC: | ||
699 | GNUNET_assert (0); /* never reach here */ | ||
700 | break; | ||
701 | } | ||
702 | opc->data = pinfo; | ||
703 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
704 | opc->state = OPC_STATE_FINISHED; | ||
705 | if (NULL != cb) | ||
706 | cb (cb_cls, opc->op, pinfo, NULL); | ||
707 | return GNUNET_YES; | ||
708 | } | ||
709 | |||
710 | |||
711 | /** | ||
712 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_OPERATIONFAILEVENT message from | ||
713 | * controller (testbed service) | ||
714 | * | ||
715 | * @param c the controller handler | ||
716 | * @param msg message received | ||
717 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
718 | * not | ||
719 | */ | ||
720 | static int | ||
721 | handle_op_fail_event (struct GNUNET_TESTBED_Controller *c, | ||
722 | const struct GNUNET_TESTBED_OperationFailureEventMessage | ||
723 | *msg) | ||
724 | { | ||
725 | struct OperationContext *opc; | ||
726 | const char *emsg; | ||
727 | uint64_t op_id; | ||
728 | struct GNUNET_TESTBED_EventInformation event; | ||
729 | |||
730 | op_id = GNUNET_ntohll (msg->operation_id); | ||
731 | if (NULL == (opc = find_opc (c, op_id))) | ||
732 | { | ||
733 | LOG_DEBUG ("Operation not found\n"); | ||
734 | return GNUNET_YES; | ||
735 | } | ||
736 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
737 | if (OP_FORWARDED == opc->type) | ||
738 | { | ||
739 | handle_forwarded_operation_msg (c, opc, | ||
740 | (const struct GNUNET_MessageHeader *) msg); | ||
741 | return GNUNET_YES; | ||
742 | } | ||
743 | opc->state = OPC_STATE_FINISHED; | ||
744 | emsg = GNUNET_TESTBED_parse_error_string_ (msg); | ||
745 | if (NULL == emsg) | ||
746 | emsg = "Unknown error"; | ||
747 | if (OP_PEER_INFO == opc->type) | ||
748 | { | ||
749 | struct PeerInfoData *data; | ||
750 | data = opc->data; | ||
751 | if (NULL != data->cb) | ||
752 | data->cb (data->cb_cls, opc->op, NULL, emsg); | ||
753 | GNUNET_free (data); | ||
754 | return GNUNET_YES; /* We do not call controller callback for peer info */ | ||
755 | } | ||
756 | if ((0 != (GNUNET_TESTBED_ET_OPERATION_FINISHED & c->event_mask)) && | ||
757 | (NULL != c->cc)) | ||
758 | { | ||
759 | event.type = GNUNET_TESTBED_ET_OPERATION_FINISHED; | ||
760 | event.details.operation_finished.operation = opc->op; | ||
761 | event.details.operation_finished.op_cls = NULL; | ||
762 | event.details.operation_finished.emsg = emsg; | ||
763 | event.details.operation_finished.generic = NULL; | ||
764 | c->cc (c->cc_cls, &event); | ||
765 | } | ||
766 | switch (opc->type) | ||
767 | { | ||
768 | case OP_PEER_CREATE: | ||
769 | { | ||
770 | struct PeerCreateData *data; | ||
771 | data = opc->data; | ||
772 | GNUNET_free (data->peer); | ||
773 | if (NULL != data->cb) | ||
774 | data->cb (data->cls, NULL, emsg); | ||
775 | GNUNET_free (data); | ||
776 | } | ||
777 | break; | ||
778 | case OP_PEER_START: | ||
779 | case OP_PEER_STOP: | ||
780 | { | ||
781 | struct PeerEventData *data; | ||
782 | data = opc->data; | ||
783 | if (NULL != data->pcc) | ||
784 | data->pcc (data->pcc_cls, emsg); | ||
785 | GNUNET_free (data); | ||
786 | } | ||
787 | break; | ||
788 | case OP_PEER_DESTROY: | ||
789 | break; | ||
790 | case OP_PEER_INFO: | ||
791 | GNUNET_assert (0); | ||
792 | case OP_OVERLAY_CONNECT: | ||
793 | { | ||
794 | struct OverlayConnectData *data; | ||
795 | data = opc->data; | ||
796 | if (NULL != data->cb) | ||
797 | data->cb (data->cb_cls, opc->op, emsg); | ||
798 | GNUNET_free (data); | ||
799 | } | ||
800 | break; | ||
801 | case OP_FORWARDED: | ||
802 | GNUNET_assert (0); | ||
803 | case OP_LINK_CONTROLLERS: /* No secondary callback */ | ||
804 | break; | ||
805 | default: | ||
806 | GNUNET_break (0); | ||
807 | } | ||
808 | return GNUNET_YES; | ||
809 | } | ||
810 | |||
811 | |||
812 | /** | ||
813 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_SLAVECONFIG message from controller | ||
814 | * (testbed service) | ||
815 | * | ||
816 | * @param c the controller handler | ||
817 | * @param msg message received | ||
818 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
819 | * not | ||
820 | */ | ||
821 | static int | ||
822 | handle_slave_config (struct GNUNET_TESTBED_Controller *c, | ||
823 | const struct GNUNET_TESTBED_SlaveConfiguration * msg) | ||
824 | { | ||
825 | struct OperationContext *opc; | ||
826 | void *op_cls; | ||
827 | uint64_t op_id; | ||
828 | struct GNUNET_TESTBED_EventInformation event; | ||
829 | |||
830 | op_id = GNUNET_ntohll (msg->operation_id); | ||
831 | if (NULL == (opc = find_opc (c, op_id))) | ||
832 | { | ||
833 | LOG_DEBUG ("Operation not found\n"); | ||
834 | return GNUNET_YES; | ||
835 | } | ||
836 | if (OP_GET_SLAVE_CONFIG != opc->type) | ||
837 | { | ||
838 | GNUNET_break (0); | ||
839 | return GNUNET_YES; | ||
840 | } | ||
841 | op_cls = ((struct GetSlaveConfigData *) opc->data)->op_cls; | ||
842 | GNUNET_free (opc->data); | ||
843 | opc->data = NULL; | ||
844 | opc->state = OPC_STATE_FINISHED; | ||
845 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
846 | if ((0 != (GNUNET_TESTBED_ET_OPERATION_FINISHED & c->event_mask)) && | ||
847 | (NULL != c->cc)) | ||
848 | { | ||
849 | opc->data = GNUNET_TESTBED_extract_config_ (&msg->header); | ||
850 | event.type = GNUNET_TESTBED_ET_OPERATION_FINISHED; | ||
851 | event.details.operation_finished.generic = opc->data; | ||
852 | event.details.operation_finished.operation = opc->op; | ||
853 | event.details.operation_finished.op_cls = op_cls; | ||
854 | event.details.operation_finished.emsg = NULL; | ||
855 | c->cc (c->cc_cls, &event); | ||
856 | } | ||
857 | return GNUNET_YES; | ||
858 | } | ||
859 | |||
860 | |||
861 | /** | ||
862 | * Handler for GNUNET_MESSAGE_TYPE_TESTBED_NEEDCONTROLLERCONFIG message from | ||
863 | * controller (testbed service) | ||
864 | * | ||
865 | * @param c the controller handler | ||
866 | * @param msg message received | ||
867 | * @return GNUNET_YES if we can continue receiving from service; GNUNET_NO if | ||
868 | * not | ||
869 | */ | ||
870 | static int | ||
871 | handle_need_controller_config (struct GNUNET_TESTBED_Controller *c, | ||
872 | const struct GNUNET_TESTBED_NeedControllerConfig * msg) | ||
873 | { | ||
874 | struct OperationContext *opc; | ||
875 | uint64_t op_id; | ||
876 | |||
877 | op_id = GNUNET_ntohll (msg->operation_id); | ||
878 | if (NULL == (opc = find_opc (c, op_id))) | ||
879 | { | ||
880 | LOG_DEBUG ("Operation not found\n"); | ||
881 | return GNUNET_YES; | ||
882 | } | ||
883 | /* This has to be a forwarded operation since there is no function in the | ||
884 | testbed API to create an operation context for this message type */ | ||
885 | GNUNET_assert (OP_FORWARDED == opc->type); | ||
886 | handle_forwarded_operation_msg (c, opc, | ||
887 | (const struct GNUNET_MessageHeader *) msg); | ||
888 | return GNUNET_YES; | ||
889 | } | ||
890 | |||
891 | |||
892 | /** | ||
893 | * Handler for messages from controller (testbed service) | ||
894 | * | ||
895 | * @param cls the controller handler | ||
896 | * @param msg message received, NULL on timeout or fatal error | ||
897 | */ | ||
898 | static void | ||
899 | message_handler (void *cls, const struct GNUNET_MessageHeader *msg) | ||
900 | { | ||
901 | struct GNUNET_TESTBED_Controller *c = cls; | ||
902 | int status; | ||
903 | uint16_t msize; | ||
904 | |||
905 | c->in_receive = GNUNET_NO; | ||
906 | /* FIXME: Add checks for message integrity */ | ||
907 | if (NULL == msg) | ||
908 | { | ||
909 | LOG_DEBUG ("Receive timed out or connection to service dropped\n"); | ||
910 | return; | ||
911 | } | ||
912 | status = GNUNET_OK; | ||
913 | msize = ntohs (msg->size); | ||
914 | switch (ntohs (msg->type)) | ||
915 | { | ||
916 | case GNUNET_MESSAGE_TYPE_TESTBED_ADDHOSTCONFIRM: | ||
917 | GNUNET_assert (msize >= | ||
918 | sizeof (struct GNUNET_TESTBED_HostConfirmedMessage)); | ||
919 | status = | ||
920 | handle_addhostconfirm (c, | ||
921 | (const struct GNUNET_TESTBED_HostConfirmedMessage | ||
922 | *) msg); | ||
923 | break; | ||
924 | case GNUNET_MESSAGE_TYPE_TESTBED_GENERICOPSUCCESS: | ||
925 | GNUNET_assert (msize == | ||
926 | sizeof (struct | ||
927 | GNUNET_TESTBED_GenericOperationSuccessEventMessage)); | ||
928 | status = | ||
929 | handle_opsuccess (c, | ||
930 | (const struct | ||
931 | GNUNET_TESTBED_GenericOperationSuccessEventMessage *) | ||
932 | msg); | ||
933 | break; | ||
934 | case GNUNET_MESSAGE_TYPE_TESTBED_PEERCREATESUCCESS: | ||
935 | GNUNET_assert (msize == | ||
936 | sizeof (struct | ||
937 | GNUNET_TESTBED_PeerCreateSuccessEventMessage)); | ||
938 | status = | ||
939 | handle_peer_create_success (c, | ||
940 | (const struct | ||
941 | GNUNET_TESTBED_PeerCreateSuccessEventMessage | ||
942 | *) msg); | ||
943 | break; | ||
944 | case GNUNET_MESSAGE_TYPE_TESTBED_PEEREVENT: | ||
945 | GNUNET_assert (msize == sizeof (struct GNUNET_TESTBED_PeerEventMessage)); | ||
946 | status = | ||
947 | handle_peer_event (c, | ||
948 | (const struct GNUNET_TESTBED_PeerEventMessage *) | ||
949 | msg); | ||
950 | |||
951 | break; | ||
952 | case GNUNET_MESSAGE_TYPE_TESTBED_PEERCONFIG: | ||
953 | GNUNET_assert (msize >= | ||
954 | sizeof (struct | ||
955 | GNUNET_TESTBED_PeerConfigurationInformationMessage)); | ||
956 | status = | ||
957 | handle_peer_config (c, | ||
958 | (const struct | ||
959 | GNUNET_TESTBED_PeerConfigurationInformationMessage | ||
960 | *) msg); | ||
961 | break; | ||
962 | case GNUNET_MESSAGE_TYPE_TESTBED_PEERCONEVENT: | ||
963 | GNUNET_assert (msize == | ||
964 | sizeof (struct GNUNET_TESTBED_ConnectionEventMessage)); | ||
965 | status = | ||
966 | handle_peer_conevent (c, | ||
967 | (const struct | ||
968 | GNUNET_TESTBED_ConnectionEventMessage *) msg); | ||
969 | break; | ||
970 | case GNUNET_MESSAGE_TYPE_TESTBED_OPERATIONFAILEVENT: | ||
971 | GNUNET_assert (msize >= | ||
972 | sizeof (struct GNUNET_TESTBED_OperationFailureEventMessage)); | ||
973 | status = | ||
974 | handle_op_fail_event (c, | ||
975 | (const struct | ||
976 | GNUNET_TESTBED_OperationFailureEventMessage *) | ||
977 | msg); | ||
978 | break; | ||
979 | case GNUNET_MESSAGE_TYPE_TESTBED_SLAVECONFIG: | ||
980 | GNUNET_assert (msize > | ||
981 | sizeof (struct GNUNET_TESTBED_SlaveConfiguration)); | ||
982 | status = | ||
983 | handle_slave_config (c, (const struct | ||
984 | GNUNET_TESTBED_SlaveConfiguration *) msg); | ||
985 | break; | ||
986 | case GNUNET_MESSAGE_TYPE_TESTBED_NEEDCONTROLLERCONFIG: | ||
987 | GNUNET_assert (msize == sizeof (struct | ||
988 | GNUNET_TESTBED_NeedControllerConfig)); | ||
989 | status = | ||
990 | handle_need_controller_config (c, (const struct | ||
991 | GNUNET_TESTBED_NeedControllerConfig | ||
992 | *) msg); | ||
993 | break; | ||
994 | default: | ||
995 | GNUNET_assert (0); | ||
996 | } | ||
997 | if ((GNUNET_OK == status) && (GNUNET_NO == c->in_receive)) | ||
998 | { | ||
999 | c->in_receive = GNUNET_YES; | ||
1000 | GNUNET_CLIENT_receive (c->client, &message_handler, c, | ||
1001 | GNUNET_TIME_UNIT_FOREVER_REL); | ||
1002 | } | ||
1003 | } | ||
1004 | |||
1005 | |||
1006 | /** | ||
1007 | * Function called to notify a client about the connection begin ready to queue | ||
1008 | * more data. "buf" will be NULL and "size" zero if the connection was closed | ||
1009 | * for writing in the meantime. | ||
1010 | * | ||
1011 | * @param cls closure | ||
1012 | * @param size number of bytes available in buf | ||
1013 | * @param buf where the callee should write the message | ||
1014 | * @return number of bytes written to buf | ||
1015 | */ | ||
1016 | static size_t | ||
1017 | transmit_ready_notify (void *cls, size_t size, void *buf) | ||
1018 | { | ||
1019 | struct GNUNET_TESTBED_Controller *c = cls; | ||
1020 | struct MessageQueue *mq_entry; | ||
1021 | |||
1022 | c->th = NULL; | ||
1023 | mq_entry = c->mq_head; | ||
1024 | GNUNET_assert (NULL != mq_entry); | ||
1025 | if ((0 == size) && (NULL == buf)) /* Timeout */ | ||
1026 | { | ||
1027 | LOG_DEBUG ("Message sending timed out -- retrying\n"); | ||
1028 | c->th = | ||
1029 | GNUNET_CLIENT_notify_transmit_ready (c->client, | ||
1030 | ntohs (mq_entry->msg->size), | ||
1031 | TIMEOUT_REL, GNUNET_YES, | ||
1032 | &transmit_ready_notify, c); | ||
1033 | return 0; | ||
1034 | } | ||
1035 | GNUNET_assert (ntohs (mq_entry->msg->size) <= size); | ||
1036 | size = ntohs (mq_entry->msg->size); | ||
1037 | memcpy (buf, mq_entry->msg, size); | ||
1038 | LOG_DEBUG ("Message of type: %u and size: %u sent\n", | ||
1039 | ntohs (mq_entry->msg->type), size); | ||
1040 | GNUNET_free (mq_entry->msg); | ||
1041 | GNUNET_CONTAINER_DLL_remove (c->mq_head, c->mq_tail, mq_entry); | ||
1042 | GNUNET_free (mq_entry); | ||
1043 | mq_entry = c->mq_head; | ||
1044 | if (NULL != mq_entry) | ||
1045 | c->th = | ||
1046 | GNUNET_CLIENT_notify_transmit_ready (c->client, | ||
1047 | ntohs (mq_entry->msg->size), | ||
1048 | TIMEOUT_REL, GNUNET_YES, | ||
1049 | &transmit_ready_notify, c); | ||
1050 | if (GNUNET_NO == c->in_receive) | ||
1051 | { | ||
1052 | c->in_receive = GNUNET_YES; | ||
1053 | GNUNET_CLIENT_receive (c->client, &message_handler, c, | ||
1054 | GNUNET_TIME_UNIT_FOREVER_REL); | ||
1055 | } | ||
1056 | return size; | ||
1057 | } | ||
1058 | |||
1059 | |||
1060 | /** | ||
1061 | * Queues a message in send queue for sending to the service | ||
1062 | * | ||
1063 | * @param controller the handle to the controller | ||
1064 | * @param msg the message to queue | ||
1065 | */ | ||
1066 | void | ||
1067 | GNUNET_TESTBED_queue_message_ (struct GNUNET_TESTBED_Controller *controller, | ||
1068 | struct GNUNET_MessageHeader *msg) | ||
1069 | { | ||
1070 | struct MessageQueue *mq_entry; | ||
1071 | uint16_t type; | ||
1072 | uint16_t size; | ||
1073 | |||
1074 | type = ntohs (msg->type); | ||
1075 | size = ntohs (msg->size); | ||
1076 | GNUNET_assert ((GNUNET_MESSAGE_TYPE_TESTBED_INIT <= type) && | ||
1077 | (GNUNET_MESSAGE_TYPE_TESTBED_MAX > type)); | ||
1078 | mq_entry = GNUNET_malloc (sizeof (struct MessageQueue)); | ||
1079 | mq_entry->msg = msg; | ||
1080 | LOG (GNUNET_ERROR_TYPE_DEBUG, | ||
1081 | "Queueing message of type %u, size %u for sending\n", type, | ||
1082 | ntohs (msg->size)); | ||
1083 | GNUNET_CONTAINER_DLL_insert_tail (controller->mq_head, controller->mq_tail, | ||
1084 | mq_entry); | ||
1085 | if (NULL == controller->th) | ||
1086 | controller->th = | ||
1087 | GNUNET_CLIENT_notify_transmit_ready (controller->client, size, | ||
1088 | TIMEOUT_REL, GNUNET_YES, | ||
1089 | &transmit_ready_notify, | ||
1090 | controller); | ||
1091 | } | ||
1092 | |||
1093 | |||
1094 | /** | ||
1095 | * Sends the given message as an operation. The given callback is called when a | ||
1096 | * reply for the operation is available. Call | ||
1097 | * GNUNET_TESTBED_forward_operation_msg_cancel_() to cleanup the returned | ||
1098 | * operation context if the cc hasn't been called | ||
1099 | * | ||
1100 | * @param controller the controller to which the message has to be sent | ||
1101 | * @param operation_id the operation id of the message | ||
1102 | * @param msg the message to send | ||
1103 | * @param cc the callback to call when reply is available | ||
1104 | * @param cc_cls the closure for the above callback | ||
1105 | * @return the operation context which can be used to cancel the forwarded | ||
1106 | * operation | ||
1107 | */ | ||
1108 | struct OperationContext * | ||
1109 | GNUNET_TESTBED_forward_operation_msg_ (struct GNUNET_TESTBED_Controller | ||
1110 | *controller, uint64_t operation_id, | ||
1111 | const struct GNUNET_MessageHeader *msg, | ||
1112 | GNUNET_CLIENT_MessageHandler cc, | ||
1113 | void *cc_cls) | ||
1114 | { | ||
1115 | struct OperationContext *opc; | ||
1116 | struct ForwardedOperationData *data; | ||
1117 | struct GNUNET_MessageHeader *dup_msg; | ||
1118 | uint16_t msize; | ||
1119 | |||
1120 | data = GNUNET_malloc (sizeof (struct ForwardedOperationData)); | ||
1121 | data->cc = cc; | ||
1122 | data->cc_cls = cc_cls; | ||
1123 | opc = GNUNET_malloc (sizeof (struct OperationContext)); | ||
1124 | opc->c = controller; | ||
1125 | opc->type = OP_FORWARDED; | ||
1126 | opc->data = data; | ||
1127 | opc->id = operation_id; | ||
1128 | msize = ntohs (msg->size); | ||
1129 | dup_msg = GNUNET_malloc (msize); | ||
1130 | (void) memcpy (dup_msg, msg, msize); | ||
1131 | GNUNET_TESTBED_queue_message_ (opc->c, dup_msg); | ||
1132 | GNUNET_CONTAINER_DLL_insert_tail (controller->ocq_head, controller->ocq_tail, | ||
1133 | opc); | ||
1134 | return opc; | ||
1135 | } | ||
1136 | |||
1137 | |||
1138 | /** | ||
1139 | * Function to cancel an operation created by simply forwarding an operation | ||
1140 | * message. | ||
1141 | * | ||
1142 | * @param opc the operation context from GNUNET_TESTBED_forward_operation_msg_() | ||
1143 | */ | ||
1144 | void | ||
1145 | GNUNET_TESTBED_forward_operation_msg_cancel_ (struct OperationContext *opc) | ||
1146 | { | ||
1147 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
1148 | GNUNET_free (opc->data); | ||
1149 | GNUNET_free (opc); | ||
1150 | } | ||
1151 | |||
1152 | |||
1153 | /** | ||
1154 | * Functions with this signature are called whenever a | ||
1155 | * complete message is received by the tokenizer. | ||
1156 | * | ||
1157 | * Do not call GNUNET_SERVER_mst_destroy in callback | ||
1158 | * | ||
1159 | * @param cls closure | ||
1160 | * @param client identification of the client | ||
1161 | * @param message the actual message | ||
1162 | * | ||
1163 | * @return GNUNET_OK on success, GNUNET_SYSERR to stop further processing | ||
1164 | */ | ||
1165 | static int | ||
1166 | helper_mst (void *cls, void *client, const struct GNUNET_MessageHeader *message) | ||
1167 | { | ||
1168 | struct GNUNET_TESTBED_ControllerProc *cp = cls; | ||
1169 | const struct GNUNET_TESTBED_HelperReply *msg; | ||
1170 | const char *hostname; | ||
1171 | char *config; | ||
1172 | uLongf config_size; | ||
1173 | uLongf xconfig_size; | ||
1174 | |||
1175 | msg = (const struct GNUNET_TESTBED_HelperReply *) message; | ||
1176 | GNUNET_assert (sizeof (struct GNUNET_TESTBED_HelperReply) < | ||
1177 | ntohs (msg->header.size)); | ||
1178 | GNUNET_assert (GNUNET_MESSAGE_TYPE_TESTBED_HELPER_REPLY == | ||
1179 | ntohs (msg->header.type)); | ||
1180 | config_size = (uLongf) ntohs (msg->config_size); | ||
1181 | xconfig_size = | ||
1182 | (uLongf) (ntohs (msg->header.size) - | ||
1183 | sizeof (struct GNUNET_TESTBED_HelperReply)); | ||
1184 | config = GNUNET_malloc (config_size); | ||
1185 | GNUNET_assert (Z_OK == | ||
1186 | uncompress ((Bytef *) config, &config_size, | ||
1187 | (const Bytef *) &msg[1], xconfig_size)); | ||
1188 | GNUNET_assert (NULL == cp->cfg); | ||
1189 | cp->cfg = GNUNET_CONFIGURATION_create (); | ||
1190 | GNUNET_assert (GNUNET_CONFIGURATION_deserialize | ||
1191 | (cp->cfg, config, config_size, GNUNET_NO)); | ||
1192 | GNUNET_free (config); | ||
1193 | if ((NULL == cp->host) || | ||
1194 | (NULL == (hostname = GNUNET_TESTBED_host_get_hostname_ (cp->host)))) | ||
1195 | hostname = "localhost"; | ||
1196 | /* Change the hostname so that we can connect to it */ | ||
1197 | GNUNET_CONFIGURATION_set_value_string (cp->cfg, "testbed", "hostname", | ||
1198 | hostname); | ||
1199 | cp->cb (cp->cls, cp->cfg, GNUNET_OK); | ||
1200 | return GNUNET_OK; | ||
1201 | } | ||
1202 | |||
1203 | |||
1204 | /** | ||
1205 | * Continuation function from GNUNET_HELPER_send() | ||
1206 | * | ||
1207 | * @param cls closure | ||
1208 | * @param result GNUNET_OK on success, | ||
1209 | * GNUNET_NO if helper process died | ||
1210 | * GNUNET_SYSERR during GNUNET_HELPER_stop | ||
1211 | */ | ||
1212 | static void | ||
1213 | clear_msg (void *cls, int result) | ||
1214 | { | ||
1215 | struct GNUNET_TESTBED_ControllerProc *cp = cls; | ||
1216 | |||
1217 | GNUNET_assert (NULL != cp->shandle); | ||
1218 | cp->shandle = NULL; | ||
1219 | GNUNET_free (cp->msg); | ||
1220 | } | ||
1221 | |||
1222 | |||
1223 | /** | ||
1224 | * Callback that will be called when the helper process dies. This is not called | ||
1225 | * when the helper process is stoped using GNUNET_HELPER_stop() | ||
1226 | * | ||
1227 | * @param cls the closure from GNUNET_HELPER_start() | ||
1228 | */ | ||
1229 | static void | ||
1230 | helper_exp_cb (void *cls) | ||
1231 | { | ||
1232 | struct GNUNET_TESTBED_ControllerProc *cp = cls; | ||
1233 | GNUNET_TESTBED_ControllerStatusCallback cb; | ||
1234 | void *cb_cls; | ||
1235 | |||
1236 | cb = cp->cb; | ||
1237 | cb_cls = cp->cls; | ||
1238 | cp->helper = NULL; | ||
1239 | GNUNET_TESTBED_controller_stop (cp); | ||
1240 | if (NULL != cb) | ||
1241 | cb (cb_cls, NULL, GNUNET_SYSERR); | ||
1242 | } | ||
1243 | |||
1244 | |||
1245 | /** | ||
1246 | * Function to call to start a link-controllers type operation once all queues | ||
1247 | * the operation is part of declare that the operation can be activated. | ||
1248 | * | ||
1249 | * @param cls the closure from GNUNET_TESTBED_operation_create_() | ||
1250 | */ | ||
1251 | static void | ||
1252 | opstart_link_controllers (void *cls) | ||
1253 | { | ||
1254 | struct OperationContext *opc = cls; | ||
1255 | struct ControllerLinkData *data; | ||
1256 | struct GNUNET_TESTBED_ControllerLinkMessage *msg; | ||
1257 | |||
1258 | GNUNET_assert (NULL != opc->data); | ||
1259 | data = opc->data; | ||
1260 | msg = data->msg; | ||
1261 | data->msg = NULL; | ||
1262 | opc->state = OPC_STATE_STARTED; | ||
1263 | GNUNET_CONTAINER_DLL_insert_tail (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
1264 | GNUNET_TESTBED_queue_message_ (opc->c, &msg->header); | ||
1265 | } | ||
1266 | |||
1267 | |||
1268 | /** | ||
1269 | * Callback which will be called when link-controllers type operation is released | ||
1270 | * | ||
1271 | * @param cls the closure from GNUNET_TESTBED_operation_create_() | ||
1272 | */ | ||
1273 | static void | ||
1274 | oprelease_link_controllers (void *cls) | ||
1275 | { | ||
1276 | struct OperationContext *opc = cls; | ||
1277 | struct ControllerLinkData *data; | ||
1278 | |||
1279 | data = opc->data; | ||
1280 | switch (opc->state) | ||
1281 | { | ||
1282 | case OPC_STATE_INIT: | ||
1283 | GNUNET_free (data->msg); | ||
1284 | break; | ||
1285 | case OPC_STATE_STARTED: | ||
1286 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
1287 | break; | ||
1288 | case OPC_STATE_FINISHED: | ||
1289 | break; | ||
1290 | } | ||
1291 | GNUNET_free_non_null (data); | ||
1292 | GNUNET_free (opc); | ||
1293 | } | ||
1294 | |||
1295 | |||
1296 | /** | ||
1297 | * Function to be called when get slave config operation is ready | ||
1298 | * | ||
1299 | * @param cls the OperationContext of type OP_GET_SLAVE_CONFIG | ||
1300 | */ | ||
1301 | static void | ||
1302 | opstart_get_slave_config (void *cls) | ||
1303 | { | ||
1304 | struct OperationContext *opc = cls; | ||
1305 | struct GetSlaveConfigData *data; | ||
1306 | struct GNUNET_TESTBED_SlaveGetConfigurationMessage *msg; | ||
1307 | uint16_t msize; | ||
1308 | |||
1309 | data = opc->data; | ||
1310 | msize = sizeof (struct GNUNET_TESTBED_SlaveGetConfigurationMessage); | ||
1311 | msg = GNUNET_malloc (msize); | ||
1312 | msg->header.size = htons (msize); | ||
1313 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_GETSLAVECONFIG); | ||
1314 | msg->operation_id = GNUNET_htonll (opc->id); | ||
1315 | msg->slave_id = htonl (data->slave_id); | ||
1316 | GNUNET_CONTAINER_DLL_insert_tail (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
1317 | GNUNET_TESTBED_queue_message_ (opc->c, &msg->header); | ||
1318 | opc->state = OPC_STATE_STARTED; | ||
1319 | } | ||
1320 | |||
1321 | |||
1322 | /** | ||
1323 | * Function to be called when get slave config operation is cancelled or finished | ||
1324 | * | ||
1325 | * @param cls the OperationContext of type OP_GET_SLAVE_CONFIG | ||
1326 | */ | ||
1327 | static void | ||
1328 | oprelease_get_slave_config (void *cls) | ||
1329 | { | ||
1330 | struct OperationContext *opc = cls; | ||
1331 | |||
1332 | switch (opc->state) | ||
1333 | { | ||
1334 | case OPC_STATE_INIT: | ||
1335 | GNUNET_free (opc->data); | ||
1336 | break; | ||
1337 | case OPC_STATE_STARTED: | ||
1338 | GNUNET_free (opc->data); | ||
1339 | GNUNET_CONTAINER_DLL_remove (opc->c->ocq_head, opc->c->ocq_tail, opc); | ||
1340 | break; | ||
1341 | case OPC_STATE_FINISHED: | ||
1342 | if (NULL != opc->data) | ||
1343 | GNUNET_CONFIGURATION_destroy (opc->data); | ||
1344 | break; | ||
1345 | } | ||
1346 | GNUNET_free (opc); | ||
1347 | } | ||
1348 | |||
1349 | |||
1350 | /** | ||
1351 | * Starts a controller process at the host. FIXME: add controller start callback | ||
1352 | * with the configuration with which the controller is started | ||
1353 | * | ||
1354 | * @param controller_ip the ip address of the controller. Will be set as TRUSTED | ||
1355 | * host when starting testbed controller at host | ||
1356 | * @param host the host where the controller has to be started; NULL for | ||
1357 | * localhost | ||
1358 | * @param cfg template configuration to use for the remote controller; the | ||
1359 | * remote controller will be started with a slightly modified | ||
1360 | * configuration (port numbers, unix domain sockets and service home | ||
1361 | * values are changed as per TESTING library on the remote host) | ||
1362 | * @param cb function called when the controller is successfully started or | ||
1363 | * dies unexpectedly; GNUNET_TESTBED_controller_stop shouldn't be | ||
1364 | * called if cb is called with GNUNET_SYSERR as status. Will never be | ||
1365 | * called in the same task as 'GNUNET_TESTBED_controller_start' | ||
1366 | * (synchronous errors will be signalled by returning NULL). This | ||
1367 | * parameter cannot be NULL. | ||
1368 | * @param cls closure for above callbacks | ||
1369 | * @return the controller process handle, NULL on errors | ||
1370 | */ | ||
1371 | struct GNUNET_TESTBED_ControllerProc * | ||
1372 | GNUNET_TESTBED_controller_start (const char *controller_ip, | ||
1373 | struct GNUNET_TESTBED_Host *host, | ||
1374 | const struct GNUNET_CONFIGURATION_Handle *cfg, | ||
1375 | GNUNET_TESTBED_ControllerStatusCallback cb, | ||
1376 | void *cls) | ||
1377 | { | ||
1378 | struct GNUNET_TESTBED_ControllerProc *cp; | ||
1379 | struct GNUNET_TESTBED_HelperInit *msg; | ||
1380 | const char *hostname; | ||
1381 | static char *const binary_argv[] = { | ||
1382 | HELPER_TESTBED_BINARY, NULL | ||
1383 | }; | ||
1384 | |||
1385 | hostname = NULL; | ||
1386 | cp = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_ControllerProc)); | ||
1387 | if ((NULL == host) || (0 == GNUNET_TESTBED_host_get_id_ (host))) | ||
1388 | cp->helper = | ||
1389 | GNUNET_HELPER_start (GNUNET_YES, HELPER_TESTBED_BINARY, binary_argv, | ||
1390 | &helper_mst, &helper_exp_cb, cp); | ||
1391 | else | ||
1392 | { | ||
1393 | char *remote_args[8]; | ||
1394 | unsigned int argp; | ||
1395 | const char *username; | ||
1396 | |||
1397 | username = GNUNET_TESTBED_host_get_username_ (host); | ||
1398 | hostname = GNUNET_TESTBED_host_get_hostname_ (host); | ||
1399 | GNUNET_asprintf (&cp->port, "%u", GNUNET_TESTBED_host_get_ssh_port_ (host)); | ||
1400 | if (NULL == username) | ||
1401 | GNUNET_asprintf (&cp->dst, "%s", hostname); | ||
1402 | else | ||
1403 | GNUNET_asprintf (&cp->dst, "%s@%s", username, hostname); | ||
1404 | LOG_DEBUG ("Starting SSH to destination %s\n", cp->dst); | ||
1405 | argp = 0; | ||
1406 | remote_args[argp++] = "ssh"; | ||
1407 | remote_args[argp++] = "-p"; | ||
1408 | remote_args[argp++] = cp->port; | ||
1409 | remote_args[argp++] = "-o"; | ||
1410 | remote_args[argp++] = "BatchMode=yes"; | ||
1411 | remote_args[argp++] = cp->dst; | ||
1412 | remote_args[argp++] = HELPER_TESTBED_BINARY_SSH; | ||
1413 | remote_args[argp++] = NULL; | ||
1414 | GNUNET_assert (argp == 8); | ||
1415 | cp->helper = | ||
1416 | GNUNET_HELPER_start (GNUNET_NO, "ssh", remote_args, &helper_mst, | ||
1417 | &helper_exp_cb, cp); | ||
1418 | } | ||
1419 | if (NULL == cp->helper) | ||
1420 | { | ||
1421 | GNUNET_free_non_null (cp->port); | ||
1422 | GNUNET_free_non_null (cp->dst); | ||
1423 | GNUNET_free (cp); | ||
1424 | return NULL; | ||
1425 | } | ||
1426 | cp->host = host; | ||
1427 | cp->cb = cb; | ||
1428 | cp->cls = cls; | ||
1429 | msg = GNUNET_TESTBED_create_helper_init_msg_ (controller_ip, hostname, cfg); | ||
1430 | cp->msg = &msg->header; | ||
1431 | cp->shandle = | ||
1432 | GNUNET_HELPER_send (cp->helper, &msg->header, GNUNET_NO, &clear_msg, cp); | ||
1433 | if (NULL == cp->shandle) | ||
1434 | { | ||
1435 | GNUNET_free (msg); | ||
1436 | GNUNET_TESTBED_controller_stop (cp); | ||
1437 | return NULL; | ||
1438 | } | ||
1439 | return cp; | ||
1440 | } | ||
1441 | |||
1442 | |||
1443 | /** | ||
1444 | * Stop the controller process (also will terminate all peers and controllers | ||
1445 | * dependent on this controller). This function blocks until the testbed has | ||
1446 | * been fully terminated (!). The controller status cb from | ||
1447 | * GNUNET_TESTBED_controller_start() will not be called. | ||
1448 | * | ||
1449 | * @param cproc the controller process handle | ||
1450 | */ | ||
1451 | void | ||
1452 | GNUNET_TESTBED_controller_stop (struct GNUNET_TESTBED_ControllerProc *cproc) | ||
1453 | { | ||
1454 | if (NULL != cproc->shandle) | ||
1455 | GNUNET_HELPER_send_cancel (cproc->shandle); | ||
1456 | if (NULL != cproc->helper) | ||
1457 | GNUNET_HELPER_stop (cproc->helper); | ||
1458 | if (NULL != cproc->cfg) | ||
1459 | GNUNET_CONFIGURATION_destroy (cproc->cfg); | ||
1460 | GNUNET_free_non_null (cproc->port); | ||
1461 | GNUNET_free_non_null (cproc->dst); | ||
1462 | GNUNET_free (cproc); | ||
1463 | } | ||
1464 | |||
1465 | |||
1466 | /** | ||
1467 | * Start a controller process using the given configuration at the | ||
1468 | * given host. | ||
1469 | * | ||
1470 | * @param cfg configuration to use | ||
1471 | * @param host host to run the controller on; This should be the same host if | ||
1472 | * the controller was previously started with | ||
1473 | * GNUNET_TESTBED_controller_start; NULL for localhost | ||
1474 | * @param event_mask bit mask with set of events to call 'cc' for; | ||
1475 | * or-ed values of "1LL" shifted by the | ||
1476 | * respective 'enum GNUNET_TESTBED_EventType' | ||
1477 | * (i.e. "(1LL << GNUNET_TESTBED_ET_CONNECT) | ...") | ||
1478 | * @param cc controller callback to invoke on events | ||
1479 | * @param cc_cls closure for cc | ||
1480 | * @return handle to the controller | ||
1481 | */ | ||
1482 | struct GNUNET_TESTBED_Controller * | ||
1483 | GNUNET_TESTBED_controller_connect (const struct GNUNET_CONFIGURATION_Handle | ||
1484 | *cfg, struct GNUNET_TESTBED_Host *host, | ||
1485 | uint64_t event_mask, | ||
1486 | GNUNET_TESTBED_ControllerCallback cc, | ||
1487 | void *cc_cls) | ||
1488 | { | ||
1489 | struct GNUNET_TESTBED_Controller *controller; | ||
1490 | struct GNUNET_TESTBED_InitMessage *msg; | ||
1491 | const char *controller_hostname; | ||
1492 | unsigned long long max_parallel_operations; | ||
1493 | unsigned long long max_parallel_service_connections; | ||
1494 | unsigned long long max_parallel_topology_config_operations; | ||
1495 | |||
1496 | if (GNUNET_OK != | ||
1497 | GNUNET_CONFIGURATION_get_value_number (cfg, "testbed", | ||
1498 | "MAX_PARALLEL_OPERATIONS", | ||
1499 | &max_parallel_operations)) | ||
1500 | { | ||
1501 | GNUNET_break (0); | ||
1502 | return NULL; | ||
1503 | } | ||
1504 | if (GNUNET_OK != | ||
1505 | GNUNET_CONFIGURATION_get_value_number (cfg, "testbed", | ||
1506 | "MAX_PARALLEL_SERVICE_CONNECTIONS", | ||
1507 | &max_parallel_service_connections)) | ||
1508 | { | ||
1509 | GNUNET_break (0); | ||
1510 | return NULL; | ||
1511 | } | ||
1512 | if (GNUNET_OK != | ||
1513 | GNUNET_CONFIGURATION_get_value_number (cfg, "testbed", | ||
1514 | "MAX_PARALLEL_TOPOLOGY_CONFIG_OPERATIONS", | ||
1515 | &max_parallel_topology_config_operations)) | ||
1516 | { | ||
1517 | GNUNET_break (0); | ||
1518 | return NULL; | ||
1519 | } | ||
1520 | controller = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_Controller)); | ||
1521 | controller->cc = cc; | ||
1522 | controller->cc_cls = cc_cls; | ||
1523 | controller->event_mask = event_mask; | ||
1524 | controller->cfg = GNUNET_CONFIGURATION_dup (cfg); | ||
1525 | controller->client = GNUNET_CLIENT_connect ("testbed", controller->cfg); | ||
1526 | if (NULL == controller->client) | ||
1527 | { | ||
1528 | GNUNET_TESTBED_controller_disconnect (controller); | ||
1529 | return NULL; | ||
1530 | } | ||
1531 | if (NULL == host) | ||
1532 | { | ||
1533 | host = GNUNET_TESTBED_host_create_by_id_ (0); | ||
1534 | if (NULL == host) /* If the above host create fails */ | ||
1535 | { | ||
1536 | LOG (GNUNET_ERROR_TYPE_WARNING, | ||
1537 | "Treating NULL host as localhost. Multiple references to localhost " | ||
1538 | "may break when localhost freed before calling disconnect \n"); | ||
1539 | host = GNUNET_TESTBED_host_lookup_by_id_ (0); | ||
1540 | } | ||
1541 | else | ||
1542 | { | ||
1543 | controller->aux_host = GNUNET_YES; | ||
1544 | } | ||
1545 | } | ||
1546 | GNUNET_assert (NULL != host); | ||
1547 | GNUNET_TESTBED_mark_host_registered_at_ (host, controller); | ||
1548 | controller->host = host; | ||
1549 | controller->opq_parallel_operations = | ||
1550 | GNUNET_TESTBED_operation_queue_create_ ((unsigned int) | ||
1551 | max_parallel_operations); | ||
1552 | controller->opq_parallel_service_connections = | ||
1553 | GNUNET_TESTBED_operation_queue_create_ ((unsigned int) | ||
1554 | max_parallel_service_connections); | ||
1555 | controller->opq_parallel_topology_config_operations= | ||
1556 | GNUNET_TESTBED_operation_queue_create_ ((unsigned int) | ||
1557 | max_parallel_service_connections); | ||
1558 | controller_hostname = GNUNET_TESTBED_host_get_hostname_ (host); | ||
1559 | if (NULL == controller_hostname) | ||
1560 | controller_hostname = "127.0.0.1"; | ||
1561 | msg = | ||
1562 | GNUNET_malloc (sizeof (struct GNUNET_TESTBED_InitMessage) + | ||
1563 | strlen (controller_hostname) + 1); | ||
1564 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_INIT); | ||
1565 | msg->header.size = | ||
1566 | htons (sizeof (struct GNUNET_TESTBED_InitMessage) + | ||
1567 | strlen (controller_hostname) + 1); | ||
1568 | msg->host_id = htonl (GNUNET_TESTBED_host_get_id_ (host)); | ||
1569 | msg->event_mask = GNUNET_htonll (controller->event_mask); | ||
1570 | strcpy ((char *) &msg[1], controller_hostname); | ||
1571 | GNUNET_TESTBED_queue_message_ (controller, | ||
1572 | (struct GNUNET_MessageHeader *) msg); | ||
1573 | return controller; | ||
1574 | } | ||
1575 | |||
1576 | |||
1577 | /** | ||
1578 | * Configure shared services at a controller. Using this function, | ||
1579 | * you can specify that certain services (such as "resolver") | ||
1580 | * should not be run for each peer but instead be shared | ||
1581 | * across N peers on the specified host. This function | ||
1582 | * must be called before any peers are created at the host. | ||
1583 | * | ||
1584 | * @param controller controller to configure | ||
1585 | * @param service_name name of the service to share | ||
1586 | * @param num_peers number of peers that should share one instance | ||
1587 | * of the specified service (1 for no sharing is the default), | ||
1588 | * use 0 to disable the service | ||
1589 | */ | ||
1590 | void | ||
1591 | GNUNET_TESTBED_controller_configure_sharing (struct GNUNET_TESTBED_Controller | ||
1592 | *controller, | ||
1593 | const char *service_name, | ||
1594 | uint32_t num_peers) | ||
1595 | { | ||
1596 | struct GNUNET_TESTBED_ConfigureSharedServiceMessage *msg; | ||
1597 | uint16_t service_name_size; | ||
1598 | uint16_t msg_size; | ||
1599 | |||
1600 | service_name_size = strlen (service_name) + 1; | ||
1601 | msg_size = | ||
1602 | sizeof (struct GNUNET_TESTBED_ConfigureSharedServiceMessage) + | ||
1603 | service_name_size; | ||
1604 | msg = GNUNET_malloc (msg_size); | ||
1605 | msg->header.size = htons (msg_size); | ||
1606 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_SERVICESHARE); | ||
1607 | msg->host_id = htonl (GNUNET_TESTBED_host_get_id_ (controller->host)); | ||
1608 | msg->num_peers = htonl (num_peers); | ||
1609 | memcpy (&msg[1], service_name, service_name_size); | ||
1610 | GNUNET_TESTBED_queue_message_ (controller, | ||
1611 | (struct GNUNET_MessageHeader *) msg); | ||
1612 | } | ||
1613 | |||
1614 | |||
1615 | /** | ||
1616 | * disconnects from the controller. | ||
1617 | * | ||
1618 | * @param controller handle to controller to stop | ||
1619 | */ | ||
1620 | void | ||
1621 | GNUNET_TESTBED_controller_disconnect (struct GNUNET_TESTBED_Controller | ||
1622 | *controller) | ||
1623 | { | ||
1624 | struct MessageQueue *mq_entry; | ||
1625 | |||
1626 | if (NULL != controller->th) | ||
1627 | GNUNET_CLIENT_notify_transmit_ready_cancel (controller->th); | ||
1628 | /* Clear the message queue */ | ||
1629 | while (NULL != (mq_entry = controller->mq_head)) | ||
1630 | { | ||
1631 | GNUNET_CONTAINER_DLL_remove (controller->mq_head, controller->mq_tail, | ||
1632 | mq_entry); | ||
1633 | GNUNET_free (mq_entry->msg); | ||
1634 | GNUNET_free (mq_entry); | ||
1635 | } | ||
1636 | if (NULL != controller->client) | ||
1637 | GNUNET_CLIENT_disconnect (controller->client); | ||
1638 | GNUNET_CONFIGURATION_destroy (controller->cfg); | ||
1639 | if (GNUNET_YES == controller->aux_host) | ||
1640 | GNUNET_TESTBED_host_destroy (controller->host); | ||
1641 | GNUNET_TESTBED_operation_queue_destroy_ (controller->opq_parallel_operations); | ||
1642 | GNUNET_TESTBED_operation_queue_destroy_ | ||
1643 | (controller->opq_parallel_service_connections); | ||
1644 | GNUNET_TESTBED_operation_queue_destroy_ | ||
1645 | (controller->opq_parallel_topology_config_operations); | ||
1646 | GNUNET_free (controller); | ||
1647 | } | ||
1648 | |||
1649 | |||
1650 | /** | ||
1651 | * Register a host with the controller | ||
1652 | * | ||
1653 | * @param controller the controller handle | ||
1654 | * @param host the host to register | ||
1655 | * @param cc the completion callback to call to inform the status of | ||
1656 | * registration. After calling this callback the registration handle | ||
1657 | * will be invalid. Cannot be NULL. | ||
1658 | * @param cc_cls the closure for the cc | ||
1659 | * @return handle to the host registration which can be used to cancel the | ||
1660 | * registration | ||
1661 | */ | ||
1662 | struct GNUNET_TESTBED_HostRegistrationHandle * | ||
1663 | GNUNET_TESTBED_register_host (struct GNUNET_TESTBED_Controller *controller, | ||
1664 | struct GNUNET_TESTBED_Host *host, | ||
1665 | GNUNET_TESTBED_HostRegistrationCompletion cc, | ||
1666 | void *cc_cls) | ||
1667 | { | ||
1668 | struct GNUNET_TESTBED_HostRegistrationHandle *rh; | ||
1669 | struct GNUNET_TESTBED_AddHostMessage *msg; | ||
1670 | const char *username; | ||
1671 | const char *hostname; | ||
1672 | uint16_t msg_size; | ||
1673 | uint16_t user_name_length; | ||
1674 | |||
1675 | if (NULL != controller->rh) | ||
1676 | return NULL; | ||
1677 | hostname = GNUNET_TESTBED_host_get_hostname_ (host); | ||
1678 | if (GNUNET_YES == GNUNET_TESTBED_is_host_registered_ (host, controller)) | ||
1679 | { | ||
1680 | LOG (GNUNET_ERROR_TYPE_WARNING, "Host hostname: %s already registered\n", | ||
1681 | (NULL == hostname) ? "localhost" : hostname); | ||
1682 | return NULL; | ||
1683 | } | ||
1684 | rh = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_HostRegistrationHandle)); | ||
1685 | rh->host = host; | ||
1686 | rh->c = controller; | ||
1687 | GNUNET_assert (NULL != cc); | ||
1688 | rh->cc = cc; | ||
1689 | rh->cc_cls = cc_cls; | ||
1690 | controller->rh = rh; | ||
1691 | username = GNUNET_TESTBED_host_get_username_ (host); | ||
1692 | msg_size = (sizeof (struct GNUNET_TESTBED_AddHostMessage)); | ||
1693 | user_name_length = 0; | ||
1694 | if (NULL != username) | ||
1695 | { | ||
1696 | user_name_length = strlen (username) + 1; | ||
1697 | msg_size += user_name_length; | ||
1698 | } | ||
1699 | /* FIXME: what happens when hostname is NULL? localhost */ | ||
1700 | GNUNET_assert (NULL != hostname); | ||
1701 | msg_size += strlen (hostname) + 1; | ||
1702 | msg = GNUNET_malloc (msg_size); | ||
1703 | msg->header.size = htons (msg_size); | ||
1704 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST); | ||
1705 | msg->host_id = htonl (GNUNET_TESTBED_host_get_id_ (host)); | ||
1706 | msg->ssh_port = htons (GNUNET_TESTBED_host_get_ssh_port_ (host)); | ||
1707 | msg->user_name_length = htons (user_name_length); | ||
1708 | if (NULL != username) | ||
1709 | memcpy (&msg[1], username, user_name_length); | ||
1710 | strcpy (((void *) &msg[1]) + user_name_length, hostname); | ||
1711 | GNUNET_TESTBED_queue_message_ (controller, | ||
1712 | (struct GNUNET_MessageHeader *) msg); | ||
1713 | return rh; | ||
1714 | } | ||
1715 | |||
1716 | |||
1717 | /** | ||
1718 | * Cancel the pending registration. Note that if the registration message is | ||
1719 | * already sent to the service the cancellation has only the effect that the | ||
1720 | * registration completion callback for the registration is never called. | ||
1721 | * | ||
1722 | * @param handle the registration handle to cancel | ||
1723 | */ | ||
1724 | void | ||
1725 | GNUNET_TESTBED_cancel_registration (struct GNUNET_TESTBED_HostRegistrationHandle | ||
1726 | *handle) | ||
1727 | { | ||
1728 | if (handle != handle->c->rh) | ||
1729 | { | ||
1730 | GNUNET_break (0); | ||
1731 | return; | ||
1732 | } | ||
1733 | handle->c->rh = NULL; | ||
1734 | GNUNET_free (handle); | ||
1735 | } | ||
1736 | |||
1737 | |||
1738 | /** | ||
1739 | * Same as the GNUNET_TESTBED_controller_link_2, however sets the operation id | ||
1740 | * of the generation operation to the given id | ||
1741 | * | ||
1742 | * @param op_cls the operation closure for the event which is generated to | ||
1743 | * signal success or failure of this operation | ||
1744 | * @param operation_id the id to use for the generated operation's id | ||
1745 | * @param master handle to the master controller who creates the association | ||
1746 | * @param delegated_host requests to which host should be delegated; cannot be NULL | ||
1747 | * @param slave_host which host is used to run the slave controller; use NULL to | ||
1748 | * make the master controller connect to the delegated host | ||
1749 | * @param sxcfg serialized and compressed configuration | ||
1750 | * @param sxcfg_size the size sxcfg | ||
1751 | * @param scfg_size the size of uncompressed serialized configuration | ||
1752 | * @param is_subordinate GNUNET_YES if the controller at delegated_host should | ||
1753 | * be started by the slave controller; GNUNET_NO if the slave | ||
1754 | * controller has to connect to the already started delegated | ||
1755 | * controller via TCP/IP | ||
1756 | * @return the operation handle | ||
1757 | */ | ||
1758 | static struct GNUNET_TESTBED_Operation * | ||
1759 | GNUNET_TESTBED_controller_link_2_with_opid (void *op_cls, | ||
1760 | uint64_t operation_id, | ||
1761 | struct GNUNET_TESTBED_Controller *master, | ||
1762 | struct GNUNET_TESTBED_Host | ||
1763 | * delegated_host, | ||
1764 | struct GNUNET_TESTBED_Host *slave_host, | ||
1765 | const char *sxcfg, | ||
1766 | size_t sxcfg_size, | ||
1767 | size_t scfg_size, int is_subordinate) | ||
1768 | { | ||
1769 | struct OperationContext *opc; | ||
1770 | struct GNUNET_TESTBED_ControllerLinkMessage *msg; | ||
1771 | struct ControllerLinkData *data; | ||
1772 | uint16_t msg_size; | ||
1773 | |||
1774 | GNUNET_assert (GNUNET_YES == | ||
1775 | GNUNET_TESTBED_is_host_registered_ (delegated_host, master)); | ||
1776 | if ((NULL != slave_host) && (0 != GNUNET_TESTBED_host_get_id_ (slave_host))) | ||
1777 | GNUNET_assert (GNUNET_YES == | ||
1778 | GNUNET_TESTBED_is_host_registered_ (slave_host, master)); | ||
1779 | msg_size = sxcfg_size + sizeof (struct GNUNET_TESTBED_ControllerLinkMessage); | ||
1780 | msg = GNUNET_malloc (msg_size); | ||
1781 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_LCONTROLLERS); | ||
1782 | msg->header.size = htons (msg_size); | ||
1783 | msg->delegated_host_id = htonl (GNUNET_TESTBED_host_get_id_ (delegated_host)); | ||
1784 | msg->slave_host_id = | ||
1785 | htonl (GNUNET_TESTBED_host_get_id_ | ||
1786 | ((NULL != slave_host) ? slave_host : master->host)); | ||
1787 | msg->config_size = htons ((uint16_t) scfg_size); | ||
1788 | msg->is_subordinate = (GNUNET_YES == is_subordinate) ? 1 : 0; | ||
1789 | memcpy (&msg[1], sxcfg, sxcfg_size); | ||
1790 | data = GNUNET_malloc (sizeof (struct ControllerLinkData)); | ||
1791 | data->msg = msg; | ||
1792 | data->op_cls = op_cls; | ||
1793 | opc = GNUNET_malloc (sizeof (struct OperationContext)); | ||
1794 | opc->c = master; | ||
1795 | opc->data = data; | ||
1796 | opc->type = OP_LINK_CONTROLLERS; | ||
1797 | opc->id = master->operation_counter++; | ||
1798 | opc->state = OPC_STATE_INIT; | ||
1799 | msg->operation_id = GNUNET_htonll (opc->id); | ||
1800 | opc->op = | ||
1801 | GNUNET_TESTBED_operation_create_ (opc, &opstart_link_controllers, | ||
1802 | &oprelease_link_controllers); | ||
1803 | GNUNET_TESTBED_operation_queue_insert_ (master->opq_parallel_operations, | ||
1804 | opc->op); | ||
1805 | return opc->op; | ||
1806 | } | ||
1807 | |||
1808 | |||
1809 | /** | ||
1810 | * Same as the GNUNET_TESTBED_controller_link, however expects configuration in | ||
1811 | * serialized and compressed | ||
1812 | * | ||
1813 | * @param op_cls the operation closure for the event which is generated to | ||
1814 | * signal success or failure of this operation | ||
1815 | * @param master handle to the master controller who creates the association | ||
1816 | * @param delegated_host requests to which host should be delegated; cannot be NULL | ||
1817 | * @param slave_host which host is used to run the slave controller; use NULL to | ||
1818 | * make the master controller connect to the delegated host | ||
1819 | * @param sxcfg serialized and compressed configuration | ||
1820 | * @param sxcfg_size the size sxcfg | ||
1821 | * @param scfg_size the size of uncompressed serialized configuration | ||
1822 | * @param is_subordinate GNUNET_YES if the controller at delegated_host should | ||
1823 | * be started by the slave controller; GNUNET_NO if the slave | ||
1824 | * controller has to connect to the already started delegated | ||
1825 | * controller via TCP/IP | ||
1826 | * @return the operation handle | ||
1827 | */ | ||
1828 | struct GNUNET_TESTBED_Operation * | ||
1829 | GNUNET_TESTBED_controller_link_2 (void *op_cls, | ||
1830 | struct GNUNET_TESTBED_Controller *master, | ||
1831 | struct GNUNET_TESTBED_Host *delegated_host, | ||
1832 | struct GNUNET_TESTBED_Host *slave_host, | ||
1833 | const char *sxcfg, size_t sxcfg_size, | ||
1834 | size_t scfg_size, int is_subordinate) | ||
1835 | { | ||
1836 | return | ||
1837 | GNUNET_TESTBED_controller_link_2_with_opid (op_cls, | ||
1838 | master->operation_counter++, | ||
1839 | master, | ||
1840 | delegated_host, | ||
1841 | slave_host, | ||
1842 | sxcfg, sxcfg_size, | ||
1843 | scfg_size, is_subordinate); | ||
1844 | } | ||
1845 | |||
1846 | |||
1847 | /** | ||
1848 | * Compresses given configuration using zlib compress | ||
1849 | * | ||
1850 | * @param config the serialized configuration | ||
1851 | * @param size the size of config | ||
1852 | * @param xconfig will be set to the compressed configuration (memory is fresly | ||
1853 | * allocated) | ||
1854 | * @return the size of the xconfig | ||
1855 | */ | ||
1856 | size_t | ||
1857 | GNUNET_TESTBED_compress_config_ (const char *config, size_t size, | ||
1858 | char **xconfig) | ||
1859 | { | ||
1860 | size_t xsize; | ||
1861 | |||
1862 | xsize = compressBound ((uLong) size); | ||
1863 | *xconfig = GNUNET_malloc (xsize); | ||
1864 | GNUNET_assert (Z_OK == | ||
1865 | compress2 ((Bytef *) * xconfig, (uLongf *) & xsize, | ||
1866 | (const Bytef *) config, (uLongf) size, | ||
1867 | Z_BEST_SPEED)); | ||
1868 | return xsize; | ||
1869 | } | ||
1870 | |||
1871 | |||
1872 | /** | ||
1873 | * Same as the GNUNET_TESTBED_controller_link, however sets the operation id | ||
1874 | * of the generation operation to the given id | ||
1875 | * | ||
1876 | * @param op_cls the operation closure for the event which is generated to | ||
1877 | * signal success or failure of this operation | ||
1878 | * @param operation_id the id to use for the generated operation's id | ||
1879 | * @param master handle to the master controller who creates the association | ||
1880 | * @param delegated_host requests to which host should be delegated; cannot be NULL | ||
1881 | * @param slave_host which host is used to run the slave controller; use NULL to | ||
1882 | * make the master controller connect to the delegated host | ||
1883 | * @param slave_cfg configuration to use for the slave controller | ||
1884 | * @param is_subordinate GNUNET_YES if the controller at delegated_host should | ||
1885 | * be started by the slave controller; GNUNET_NO if the slave | ||
1886 | * controller has to connect to the already started delegated | ||
1887 | * controller via TCP/IP | ||
1888 | * @return the operation handle | ||
1889 | */ | ||
1890 | struct GNUNET_TESTBED_Operation * | ||
1891 | GNUNET_TESTBED_controller_link_with_opid (void *op_cls, | ||
1892 | uint64_t operation_id, | ||
1893 | struct GNUNET_TESTBED_Controller *master, | ||
1894 | struct GNUNET_TESTBED_Host *delegated_host, | ||
1895 | struct GNUNET_TESTBED_Host *slave_host, | ||
1896 | const struct GNUNET_CONFIGURATION_Handle | ||
1897 | *slave_cfg, int is_subordinate) | ||
1898 | { | ||
1899 | struct GNUNET_TESTBED_Operation *op; | ||
1900 | char *config; | ||
1901 | char *cconfig; | ||
1902 | size_t cc_size; | ||
1903 | size_t config_size; | ||
1904 | |||
1905 | GNUNET_assert (GNUNET_YES == | ||
1906 | GNUNET_TESTBED_is_host_registered_ (delegated_host, master)); | ||
1907 | if ((NULL != slave_host) && (0 != GNUNET_TESTBED_host_get_id_ (slave_host))) | ||
1908 | GNUNET_assert (GNUNET_YES == | ||
1909 | GNUNET_TESTBED_is_host_registered_ (slave_host, master)); | ||
1910 | config = GNUNET_CONFIGURATION_serialize (slave_cfg, &config_size); | ||
1911 | cc_size = GNUNET_TESTBED_compress_config_ (config, config_size, &cconfig); | ||
1912 | GNUNET_free (config); | ||
1913 | /* Configuration doesn't fit in 1 message */ | ||
1914 | GNUNET_assert ((UINT16_MAX - | ||
1915 | sizeof (struct GNUNET_TESTBED_ControllerLinkMessage)) >= cc_size); | ||
1916 | op = GNUNET_TESTBED_controller_link_2_with_opid (op_cls, | ||
1917 | master->operation_counter++, | ||
1918 | master, delegated_host, | ||
1919 | slave_host, | ||
1920 | (const char *) cconfig, | ||
1921 | cc_size, config_size, | ||
1922 | is_subordinate); | ||
1923 | GNUNET_free (cconfig); | ||
1924 | return op; | ||
1925 | } | ||
1926 | |||
1927 | |||
1928 | /** | ||
1929 | * Create a link from slave controller to delegated controller. Whenever the | ||
1930 | * master controller is asked to start a peer at the delegated controller the | ||
1931 | * request will be routed towards slave controller (if a route exists). The | ||
1932 | * slave controller will then route it to the delegated controller. The | ||
1933 | * configuration of the delegated controller is given and is used to either | ||
1934 | * create the delegated controller or to connect to an existing controller. Note | ||
1935 | * that while starting the delegated controller the configuration will be | ||
1936 | * modified to accommodate available free ports. the 'is_subordinate' specifies | ||
1937 | * if the given delegated controller should be started and managed by the slave | ||
1938 | * controller, or if the delegated controller already has a master and the slave | ||
1939 | * controller connects to it as a non master controller. The success or failure | ||
1940 | * of this operation will be signalled through the | ||
1941 | * GNUNET_TESTBED_ControllerCallback() with an event of type | ||
1942 | * GNUNET_TESTBED_ET_OPERATION_FINISHED | ||
1943 | * | ||
1944 | * @param op_cls the operation closure for the event which is generated to | ||
1945 | * signal success or failure of this operation | ||
1946 | * @param master handle to the master controller who creates the association | ||
1947 | * @param delegated_host requests to which host should be delegated; cannot be NULL | ||
1948 | * @param slave_host which host is used to run the slave controller; use NULL to | ||
1949 | * make the master controller connect to the delegated host | ||
1950 | * @param slave_cfg configuration to use for the slave controller | ||
1951 | * @param is_subordinate GNUNET_YES if the controller at delegated_host should | ||
1952 | * be started by the slave controller; GNUNET_NO if the slave | ||
1953 | * controller has to connect to the already started delegated | ||
1954 | * controller via TCP/IP | ||
1955 | * @return the operation handle | ||
1956 | */ | ||
1957 | struct GNUNET_TESTBED_Operation * | ||
1958 | GNUNET_TESTBED_controller_link (void *op_cls, | ||
1959 | struct GNUNET_TESTBED_Controller *master, | ||
1960 | struct GNUNET_TESTBED_Host *delegated_host, | ||
1961 | struct GNUNET_TESTBED_Host *slave_host, | ||
1962 | const struct GNUNET_CONFIGURATION_Handle | ||
1963 | *slave_cfg, int is_subordinate) | ||
1964 | { | ||
1965 | return | ||
1966 | GNUNET_TESTBED_controller_link_with_opid (op_cls, | ||
1967 | master->operation_counter++, | ||
1968 | master, delegated_host, | ||
1969 | slave_host, slave_cfg, | ||
1970 | is_subordinate); | ||
1971 | } | ||
1972 | |||
1973 | |||
1974 | /** | ||
1975 | * Function to acquire the configuration of a running slave controller. The | ||
1976 | * completion of the operation is signalled through the controller_cb from | ||
1977 | * GNUNET_TESTBED_controller_connect(). If the operation is successful the | ||
1978 | * handle to the configuration is available in the generic pointer of | ||
1979 | * operation_finished field of struct GNUNET_TESTBED_EventInformation. | ||
1980 | * | ||
1981 | * @param op_cls the closure for the operation | ||
1982 | * @param master the handle to master controller | ||
1983 | * @param slave_host the host where the slave controller is running; the handle | ||
1984 | * to the slave_host should remain valid until this operation is | ||
1985 | * cancelled or marked as finished | ||
1986 | * @return the operation handle; NULL if the slave_host is not registered at | ||
1987 | * master | ||
1988 | */ | ||
1989 | struct GNUNET_TESTBED_Operation * | ||
1990 | GNUNET_TESTBED_get_slave_config (void *op_cls, | ||
1991 | struct GNUNET_TESTBED_Controller *master, | ||
1992 | struct GNUNET_TESTBED_Host *slave_host) | ||
1993 | { | ||
1994 | struct OperationContext *opc; | ||
1995 | struct GetSlaveConfigData *data; | ||
1996 | |||
1997 | if (GNUNET_NO == GNUNET_TESTBED_is_host_registered_ (slave_host, master)) | ||
1998 | return NULL; | ||
1999 | data = GNUNET_malloc (sizeof (struct GetSlaveConfigData)); | ||
2000 | data->slave_id = GNUNET_TESTBED_host_get_id_ (slave_host); | ||
2001 | data->op_cls = op_cls; | ||
2002 | opc = GNUNET_malloc (sizeof (struct OperationContext)); | ||
2003 | opc->state = OPC_STATE_INIT; | ||
2004 | opc->c = master; | ||
2005 | opc->id = master->operation_counter++; | ||
2006 | opc->type = OP_GET_SLAVE_CONFIG; | ||
2007 | opc->data = data; | ||
2008 | opc->op = | ||
2009 | GNUNET_TESTBED_operation_create_ (opc, &opstart_get_slave_config, | ||
2010 | &oprelease_get_slave_config); | ||
2011 | GNUNET_TESTBED_operation_queue_insert_ (master->opq_parallel_operations, | ||
2012 | opc->op); | ||
2013 | return opc->op; | ||
2014 | } | ||
2015 | |||
2016 | |||
2017 | /** | ||
2018 | * Ask the testbed controller to write the current overlay topology to | ||
2019 | * a file. Naturally, the file will only contain a snapshot as the | ||
2020 | * topology may evolve all the time. | ||
2021 | * | ||
2022 | * @param controller overlay controller to inspect | ||
2023 | * @param filename name of the file the topology should | ||
2024 | * be written to. | ||
2025 | */ | ||
2026 | void | ||
2027 | GNUNET_TESTBED_overlay_write_topology_to_file (struct GNUNET_TESTBED_Controller | ||
2028 | *controller, | ||
2029 | const char *filename) | ||
2030 | { | ||
2031 | GNUNET_break (0); | ||
2032 | } | ||
2033 | |||
2034 | |||
2035 | /** | ||
2036 | * Creates a helper initialization message. This function is here because we | ||
2037 | * want to use this in testing | ||
2038 | * | ||
2039 | * @param cname the ip address of the controlling host | ||
2040 | * @param hostname the hostname of the destination this message is intended for | ||
2041 | * @param cfg the configuration that has to used to start the testbed service | ||
2042 | * thru helper | ||
2043 | * @return the initialization message | ||
2044 | */ | ||
2045 | struct GNUNET_TESTBED_HelperInit * | ||
2046 | GNUNET_TESTBED_create_helper_init_msg_ (const char *cname, | ||
2047 | const char *hostname, | ||
2048 | const struct GNUNET_CONFIGURATION_Handle | ||
2049 | *cfg) | ||
2050 | { | ||
2051 | struct GNUNET_TESTBED_HelperInit *msg; | ||
2052 | char *config; | ||
2053 | char *xconfig; | ||
2054 | size_t config_size; | ||
2055 | size_t xconfig_size; | ||
2056 | uint16_t cname_len; | ||
2057 | uint16_t hostname_len; | ||
2058 | uint16_t msg_size; | ||
2059 | |||
2060 | config = GNUNET_CONFIGURATION_serialize (cfg, &config_size); | ||
2061 | GNUNET_assert (NULL != config); | ||
2062 | xconfig_size = | ||
2063 | GNUNET_TESTBED_compress_config_ (config, config_size, &xconfig); | ||
2064 | GNUNET_free (config); | ||
2065 | cname_len = strlen (cname); | ||
2066 | hostname_len = (NULL == hostname) ? 0 : strlen (hostname); | ||
2067 | msg_size = | ||
2068 | xconfig_size + cname_len + 1 + sizeof (struct GNUNET_TESTBED_HelperInit); | ||
2069 | msg_size += hostname_len; | ||
2070 | msg = GNUNET_realloc (xconfig, msg_size); | ||
2071 | (void) memmove (((void *) &msg[1]) + cname_len + 1 + hostname_len, | ||
2072 | msg, | ||
2073 | xconfig_size); | ||
2074 | msg->header.size = htons (msg_size); | ||
2075 | msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_HELPER_INIT); | ||
2076 | msg->cname_size = htons (cname_len); | ||
2077 | msg->hostname_size = htons (hostname_len); | ||
2078 | msg->config_size = htons (config_size); | ||
2079 | (void) strcpy ((char *) &msg[1], cname); | ||
2080 | if (0 != hostname_len) | ||
2081 | (void) strncpy (((char *) &msg[1]) + cname_len + 1, hostname, hostname_len); | ||
2082 | return msg; | ||
2083 | } | ||
2084 | |||
2085 | |||
2086 | /** | ||
2087 | * Cancel a pending operation. Releases all resources | ||
2088 | * of the operation and will ensure that no event | ||
2089 | * is generated for the operation. Does NOT guarantee | ||
2090 | * that the operation will be fully undone (or that | ||
2091 | * nothing ever happened). | ||
2092 | * | ||
2093 | * @param operation operation to cancel | ||
2094 | */ | ||
2095 | void | ||
2096 | GNUNET_TESTBED_operation_cancel (struct GNUNET_TESTBED_Operation *operation) | ||
2097 | { | ||
2098 | GNUNET_TESTBED_operation_done (operation); | ||
2099 | } | ||
2100 | |||
2101 | |||
2102 | /** | ||
2103 | * Signal that the information from an operation has been fully | ||
2104 | * processed. This function MUST be called for each event | ||
2105 | * of type 'operation_finished' to fully remove the operation | ||
2106 | * from the operation queue. After calling this function, the | ||
2107 | * 'op_result' becomes invalid (!). | ||
2108 | * | ||
2109 | * @param operation operation to signal completion for | ||
2110 | */ | ||
2111 | void | ||
2112 | GNUNET_TESTBED_operation_done (struct GNUNET_TESTBED_Operation *operation) | ||
2113 | { | ||
2114 | switch (operation->type) | ||
2115 | { | ||
2116 | case OP_PEER_CREATE: | ||
2117 | case OP_PEER_DESTROY: | ||
2118 | case OP_PEER_START: | ||
2119 | case OP_PEER_STOP: | ||
2120 | case OP_PEER_INFO: | ||
2121 | case OP_OVERLAY_CONNECT: | ||
2122 | case OP_LINK_CONTROLLERS: | ||
2123 | GNUNET_TESTBED_operation_release_ (operation); | ||
2124 | return; | ||
2125 | default: | ||
2126 | GNUNET_assert (0); | ||
2127 | break; | ||
2128 | } | ||
2129 | } | ||
2130 | |||
2131 | |||
2132 | /** | ||
2133 | * Generates configuration by uncompressing configuration in given message. The | ||
2134 | * given message should be of the following types: | ||
2135 | * GNUNET_MESSAGE_TYPE_TESTBED_PEERCONFIG, | ||
2136 | * GNUNET_MESSAGE_TYPE_TESTBED_SLAVECONFIG | ||
2137 | * | ||
2138 | * @param msg the message containing compressed configuration | ||
2139 | * @return handle to the parsed configuration | ||
2140 | */ | ||
2141 | struct GNUNET_CONFIGURATION_Handle * | ||
2142 | GNUNET_TESTBED_extract_config_ (const struct GNUNET_MessageHeader *msg) | ||
2143 | { | ||
2144 | struct GNUNET_CONFIGURATION_Handle *cfg; | ||
2145 | Bytef *data; | ||
2146 | const Bytef *xdata; | ||
2147 | uLong data_len; | ||
2148 | uLong xdata_len; | ||
2149 | int ret; | ||
2150 | |||
2151 | switch (ntohs (msg->type)) | ||
2152 | { | ||
2153 | case GNUNET_MESSAGE_TYPE_TESTBED_PEERCONFIG: | ||
2154 | { | ||
2155 | const struct GNUNET_TESTBED_PeerConfigurationInformationMessage *imsg; | ||
2156 | |||
2157 | imsg = (const struct GNUNET_TESTBED_PeerConfigurationInformationMessage *) | ||
2158 | msg; | ||
2159 | data_len = (uLong) ntohs (imsg->config_size); | ||
2160 | xdata_len = ntohs (imsg->header.size) | ||
2161 | - sizeof (struct GNUNET_TESTBED_PeerConfigurationInformationMessage); | ||
2162 | xdata = (const Bytef *) &imsg[1]; | ||
2163 | } | ||
2164 | break; | ||
2165 | case GNUNET_MESSAGE_TYPE_TESTBED_SLAVECONFIG: | ||
2166 | { | ||
2167 | const struct GNUNET_TESTBED_SlaveConfiguration *imsg; | ||
2168 | |||
2169 | imsg = (const struct GNUNET_TESTBED_SlaveConfiguration *) msg; | ||
2170 | data_len = (uLong) ntohs (imsg->config_size); | ||
2171 | xdata_len = ntohs (imsg->header.size) | ||
2172 | - sizeof (struct GNUNET_TESTBED_SlaveConfiguration); | ||
2173 | xdata = (const Bytef *) &imsg[1]; | ||
2174 | } | ||
2175 | break; | ||
2176 | default: | ||
2177 | GNUNET_assert (0); | ||
2178 | } | ||
2179 | data = GNUNET_malloc (data_len); | ||
2180 | if (Z_OK != | ||
2181 | (ret = | ||
2182 | uncompress (data, &data_len, xdata, xdata_len))) | ||
2183 | GNUNET_assert (0); | ||
2184 | cfg = GNUNET_CONFIGURATION_create (); | ||
2185 | GNUNET_assert (GNUNET_OK == | ||
2186 | GNUNET_CONFIGURATION_deserialize (cfg, (const char *) data, | ||
2187 | (size_t) data_len, | ||
2188 | GNUNET_NO)); | ||
2189 | GNUNET_free (data); | ||
2190 | return cfg; | ||
2191 | } | ||
2192 | |||
2193 | |||
2194 | /** | ||
2195 | * Checks the integrity of the OperationFailureEventMessage and if good returns | ||
2196 | * the error message it contains. | ||
2197 | * | ||
2198 | * @param msg the OperationFailureEventMessage | ||
2199 | * @return the error message | ||
2200 | */ | ||
2201 | const char * | ||
2202 | GNUNET_TESTBED_parse_error_string_ (const struct | ||
2203 | GNUNET_TESTBED_OperationFailureEventMessage | ||
2204 | *msg) | ||
2205 | { | ||
2206 | uint16_t msize; | ||
2207 | const char *emsg; | ||
2208 | |||
2209 | msize = ntohs (msg->header.size); | ||
2210 | if (sizeof (struct GNUNET_TESTBED_OperationFailureEventMessage) >= msize) | ||
2211 | return NULL; | ||
2212 | msize -= sizeof (struct GNUNET_TESTBED_OperationFailureEventMessage); | ||
2213 | emsg = (const char *) &msg[1]; | ||
2214 | if ('\0' != emsg[msize - 1]) | ||
2215 | { | ||
2216 | GNUNET_break (0); | ||
2217 | return NULL; | ||
2218 | } | ||
2219 | return emsg; | ||
2220 | } | ||
2221 | |||
2222 | /* end of testbed_api.c */ | ||