-do not modify 'const' argument
[oweals/gnunet.git] / src / multicast / gnunet-service-multicast.c
1 /*
2      This file is part of GNUnet.
3      (C) 2009 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 multicast/gnunet-service-multicast.c
23  * @brief program that does multicast
24  * @author Christian Grothoff
25  */
26 #include "platform.h"
27 #include "gnunet_util_lib.h"
28 #include "gnunet_signatures.h"
29 #include "gnunet_applications.h"
30 #include "gnunet_statistics_service.h"
31 #include "gnunet_core_service.h"
32 #include "gnunet_cadet_service.h"
33 #include "gnunet_multicast_service.h"
34 #include "multicast.h"
35
36 /**
37  * Handle to our current configuration.
38  */
39 static const struct GNUNET_CONFIGURATION_Handle *cfg;
40
41 /**
42  * Server handle.
43  */
44 static struct GNUNET_SERVER_Handle *server;
45
46 /**
47  * Core handle.
48  * Only used during initialization.
49  */
50 static struct GNUNET_CORE_Handle *core;
51
52 /**
53  * CADET handle.
54  */
55 static struct GNUNET_CADET_Handle *cadet;
56
57 /**
58  * Identity of this peer.
59  */
60 static struct GNUNET_PeerIdentity this_peer;
61
62 /**
63  * Handle to the statistics service.
64  */
65 static struct GNUNET_STATISTICS_Handle *stats;
66
67 /**
68  * Notification context, simplifies client broadcasts.
69  */
70 static struct GNUNET_SERVER_NotificationContext *nc;
71
72 /**
73  * All connected origin clients.
74  * Group's pub_key_hash -> struct Origin * (uniq)
75  */
76 static struct GNUNET_CONTAINER_MultiHashMap *origins;
77
78 /**
79  * All connected member clients.
80  * Group's pub_key_hash -> struct Member * (multi)
81  */
82 static struct GNUNET_CONTAINER_MultiHashMap *members;
83
84 /**
85  * Connected member clients per group.
86  * Group's pub_key_hash -> Member's pub_key_hash (uniq) -> struct Member * (uniq)
87  */
88 static struct GNUNET_CONTAINER_MultiHashMap *group_members;
89
90 /**
91  * Incoming CADET channels.
92  * Group's pub_key_hash -> struct Channel * (multi)
93  */
94 static struct GNUNET_CONTAINER_MultiHashMap *channels_in;
95
96 /**
97  * Outgoing CADET channels.
98  * Group's pub_key_hash -> struct Channel * (multi)
99  */
100 static struct GNUNET_CONTAINER_MultiHashMap *channels_out;
101
102 /**
103  * Join status of a remote peer.
104  */
105 enum JoinStatus
106 {
107   JOIN_REFUSED  = -1,
108   JOIN_NOT_ASKED = 0,
109   JOIN_WAITING   = 1,
110   JOIN_ADMITTED  = 2,
111 };
112
113 enum ChannelDirection
114 {
115   DIR_INCOMING = 0,
116   DIR_OUTGOING = 1,
117 };
118
119
120 /**
121  * Context for a CADET channel.
122  */
123 struct Channel
124 {
125   /**
126    * Group the channel belongs to.
127    *
128    * Only set for outgoing channels.
129    */
130   struct Group *grp;
131
132   /**
133    * CADET channel.
134    */
135   struct GNUNET_CADET_Channel *channel;
136
137   /**
138    * CADET transmission handle.
139    */
140   struct GNUNET_CADET_TransmitHandle *tmit_handle;
141
142   /**
143    * Public key of the target group.
144    */
145   struct GNUNET_CRYPTO_EddsaPublicKey group_key;
146
147   /**
148    * Hash of @a group_key.
149    */
150   struct GNUNET_HashCode group_key_hash;
151
152   /**
153    * Public key of the joining member.
154    */
155   struct GNUNET_CRYPTO_EcdsaPublicKey member_key;
156
157   /**
158    * Remote peer identity.
159    */
160   struct GNUNET_PeerIdentity peer;
161
162   /**
163    * Is the remote peer admitted to the group?
164    * @see enum JoinStatus
165    */
166   int8_t join_status;
167
168   /**
169    * Channel direction.
170    * @see enum ChannelDirection
171    */
172   uint8_t direction;
173 };
174
175
176 /**
177  * List of connected clients.
178  */
179 struct ClientList
180 {
181   struct ClientList *prev;
182   struct ClientList *next;
183   struct GNUNET_SERVER_Client *client;
184 };
185
186 /**
187  * Common part of the client context for both an origin and member.
188  */
189 struct Group
190 {
191   struct ClientList *clients_head;
192   struct ClientList *clients_tail;
193
194   /**
195    * Public key of the group.
196    */
197   struct GNUNET_CRYPTO_EddsaPublicKey pub_key;
198
199   /**
200    * Hash of @a pub_key.
201    */
202   struct GNUNET_HashCode pub_key_hash;
203
204   /**
205    * Is this an origin (#GNUNET_YES), or member (#GNUNET_NO)?
206    */
207   uint8_t is_origin;
208
209   /**
210    * Is the client disconnected? #GNUNET_YES or #GNUNET_NO
211    */
212   uint8_t disconnected;
213 };
214
215
216 /**
217  * Client context for a group's origin.
218  */
219 struct Origin
220 {
221   struct Group grp;
222
223   /**
224    * Private key of the group.
225    */
226   struct GNUNET_CRYPTO_EddsaPrivateKey priv_key;
227
228   /**
229    * Last message fragment ID sent to the group.
230    */
231   uint64_t max_fragment_id;
232 };
233
234
235 /**
236  * Client context for a group member.
237  */
238 struct Member
239 {
240   struct Group grp;
241
242   /**
243    * Private key of the member.
244    */
245   struct GNUNET_CRYPTO_EcdsaPrivateKey priv_key;
246
247   /**
248    * Public key of the member.
249    */
250   struct GNUNET_CRYPTO_EcdsaPublicKey pub_key;
251
252   /**
253    * Hash of @a pub_key.
254    */
255   struct GNUNET_HashCode pub_key_hash;
256
257   /**
258    * Join request sent to the origin / members.
259    */
260   struct MulticastJoinRequestMessage *join_req;
261
262   /**
263    * Join decision sent in reply to our request.
264    *
265    * Only a positive decision is stored here, in case of a negative decision the
266    * client is disconnected.
267    */
268   struct MulticastJoinDecisionMessageHeader *join_dcsn;
269
270   /**
271    * CADET channel to the origin.
272    */
273   struct Channel *origin_channel;
274
275   /**
276    * Peer identity of origin.
277    */
278   struct GNUNET_PeerIdentity origin;
279
280   /**
281    * Peer identity of relays (other members to connect).
282    */
283   struct GNUNET_PeerIdentity *relays;
284
285   /**
286    * Last request fragment ID sent to the origin.
287    */
288   uint64_t max_fragment_id;
289
290   /**
291    * Number of @a relays.
292    */
293   uint32_t relay_count;
294 };
295
296
297 /**
298  * Task run during shutdown.
299  *
300  * @param cls unused
301  * @param tc unused
302  */
303 static void
304 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
305 {
306   if (NULL != core)
307   {
308     GNUNET_CORE_disconnect (core);
309     core = NULL;
310   }
311   if (NULL != cadet)
312   {
313     GNUNET_CADET_disconnect (cadet);
314     cadet = NULL;
315   }
316   if (NULL != stats)
317   {
318     GNUNET_STATISTICS_destroy (stats, GNUNET_YES);
319     stats = NULL;
320   }
321   /* FIXME: do more clean up here */
322 }
323
324
325 /**
326  * Clean up origin data structures after a client disconnected.
327  */
328 static void
329 cleanup_origin (struct Origin *orig)
330 {
331   struct Group *grp = &orig->grp;
332   GNUNET_CONTAINER_multihashmap_remove (origins, &grp->pub_key_hash, orig);
333 }
334
335
336 /**
337  * Clean up member data structures after a client disconnected.
338  */
339 static void
340 cleanup_member (struct Member *mem)
341 {
342   struct Group *grp = &mem->grp;
343   struct GNUNET_CONTAINER_MultiHashMap *
344     grp_mem = GNUNET_CONTAINER_multihashmap_get (group_members,
345                                                  &grp->pub_key_hash);
346   GNUNET_assert (NULL != grp_mem);
347   GNUNET_CONTAINER_multihashmap_remove (grp_mem, &mem->pub_key_hash, mem);
348
349   if (0 == GNUNET_CONTAINER_multihashmap_size (grp_mem))
350   {
351     GNUNET_CONTAINER_multihashmap_remove (group_members, &grp->pub_key_hash,
352                                           grp_mem);
353     GNUNET_CONTAINER_multihashmap_destroy (grp_mem);
354   }
355   if (NULL != mem->join_dcsn)
356   {
357     GNUNET_free (mem->join_dcsn);
358     mem->join_dcsn = NULL;
359   }
360   GNUNET_CONTAINER_multihashmap_remove (members, &grp->pub_key_hash, mem);
361 }
362
363
364 /**
365  * Clean up group data structures after a client disconnected.
366  */
367 static void
368 cleanup_group (struct Group *grp)
369 {
370   (GNUNET_YES == grp->is_origin)
371     ? cleanup_origin ((struct Origin *) grp)
372     : cleanup_member ((struct Member *) grp);
373
374   GNUNET_free (grp);
375 }
376
377
378 /**
379  * Called whenever a client is disconnected.
380  *
381  * Frees our resources associated with that client.
382  *
383  * @param cls  Closure.
384  * @param client  Client handle.
385  */
386 static void
387 client_notify_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
388 {
389   if (NULL == client)
390     return;
391
392   struct Group *grp
393     = GNUNET_SERVER_client_get_user_context (client, struct Group);
394
395   if (NULL == grp)
396   {
397     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
398                 "%p User context is NULL in client_disconnect()\n", grp);
399     GNUNET_assert (0);
400     return;
401   }
402
403   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
404               "%p Client (%s) disconnected from group %s\n",
405               grp, (GNUNET_YES == grp->is_origin) ? "origin" : "member",
406               GNUNET_h2s (&grp->pub_key_hash));
407
408   struct ClientList *cl = grp->clients_head;
409   while (NULL != cl)
410   {
411     if (cl->client == client)
412     {
413       GNUNET_CONTAINER_DLL_remove (grp->clients_head, grp->clients_tail, cl);
414       GNUNET_free (cl);
415       break;
416     }
417     cl = cl->next;
418   }
419
420   if (NULL == grp->clients_head)
421   { /* Last client disconnected. */
422 #if FIXME
423     if (NULL != grp->tmit_head)
424     { /* Send pending messages via CADET before cleanup. */
425       transmit_message (grp);
426     }
427     else
428 #endif
429     {
430       cleanup_group (grp);
431     }
432   }
433 }
434
435
436 /**
437  * Send message to all clients connected to the group.
438  */
439 static void
440 client_send_msg (const struct Group *grp,
441                  const struct GNUNET_MessageHeader *msg)
442 {
443   GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
444               "%p Sending message to clients.\n", grp);
445
446   struct ClientList *cl = grp->clients_head;
447   while (NULL != cl)
448   {
449     GNUNET_SERVER_notification_context_add (nc, cl->client);
450     GNUNET_SERVER_notification_context_unicast (nc, cl->client, msg, GNUNET_NO);
451     cl = cl->next;
452   }
453 }
454
455
456 /**
457  * Iterator callback for sending a message to origin clients.
458  */
459 static int
460 client_send_origin_cb (void *cls, const struct GNUNET_HashCode *pub_key_hash,
461                        void *origin)
462 {
463   const struct GNUNET_MessageHeader *msg = cls;
464   struct Member *orig = origin;
465
466   client_send_msg (&orig->grp, msg);
467   return GNUNET_YES;
468 }
469
470
471 /**
472  * Iterator callback for sending a message to member clients.
473  */
474 static int
475 client_send_member_cb (void *cls, const struct GNUNET_HashCode *pub_key_hash,
476                        void *member)
477 {
478   const struct GNUNET_MessageHeader *msg = cls;
479   struct Member *mem = member;
480
481   if (NULL != mem->join_dcsn)
482   { /* Only send message to admitted members */
483     client_send_msg (&mem->grp, msg);
484   }
485   return GNUNET_YES;
486 }
487
488
489 /**
490  * Send message to all origin and member clients connected to the group.
491  *
492  * @param grp  The group to send @a msg to.
493  * @param msg  Message to send.
494  */
495 static int
496 client_send_all (struct GNUNET_HashCode *pub_key_hash,
497                  const struct GNUNET_MessageHeader *msg)
498 {
499   int n = 0;
500   if (origins != NULL)
501     n += GNUNET_CONTAINER_multihashmap_get_multiple (origins, pub_key_hash,
502                                                      client_send_origin_cb,
503                                                      (void *) msg);
504   if (members != NULL)
505     n += GNUNET_CONTAINER_multihashmap_get_multiple (members, pub_key_hash,
506                                                      client_send_member_cb,
507                                                      (void *) msg);
508   return n;
509 }
510
511
512 /**
513  * Send message to all origin clients connected to the group.
514  *
515  * @param grp  The group to send @a msg to.
516  * @param msg  Message to send.
517  */
518 static int
519 client_send_origin (struct GNUNET_HashCode *pub_key_hash,
520                     const struct GNUNET_MessageHeader *msg)
521 {
522   int n = 0;
523   if (origins != NULL)
524     n += GNUNET_CONTAINER_multihashmap_get_multiple (origins, pub_key_hash,
525                                                      client_send_origin_cb,
526                                                      (void *) msg);
527   return n;
528 }
529
530
531 /**
532  * CADET is ready to transmit a message.
533  */
534 size_t
535 cadet_notify_transmit_ready (void *cls, size_t buf_size, void *buf)
536 {
537   if (0 == buf_size)
538   {
539     /* FIXME: connection closed */
540     return 0;
541   }
542   const struct GNUNET_MessageHeader *msg = cls;
543   uint16_t msg_size = ntohs (msg->size);
544   GNUNET_assert (msg_size <= buf_size);
545   memcpy (buf, msg, msg_size);
546   return msg_size;
547 }
548
549
550 /**
551  * Send a message to a CADET channel.
552  *
553  * @param chn  Channel.
554  * @param msg  Message.
555  */
556 static void
557 cadet_send_msg (struct Channel *chn, const struct GNUNET_MessageHeader *msg)
558 {
559   chn->tmit_handle
560     = GNUNET_CADET_notify_transmit_ready (chn->channel, GNUNET_NO,
561                                           GNUNET_TIME_UNIT_FOREVER_REL,
562                                           ntohs (msg->size),
563                                           &cadet_notify_transmit_ready,
564                                           (void *) msg);
565   GNUNET_assert (NULL != chn->tmit_handle);
566 }
567
568
569 /**
570  * Create new outgoing CADET channel.
571  *
572  * @param peer
573  *        Peer to connect to.
574  * @param group_key
575  *        Public key of group the channel belongs to.
576  * @param group_key_hash
577  *        Hash of @a group_key.
578  *
579  * @return Channel.
580  */
581 static struct Channel *
582 cadet_channel_create (struct Group *grp, struct GNUNET_PeerIdentity *peer)
583 {
584   struct Channel *chn = GNUNET_malloc (sizeof (*chn));
585   chn->grp = grp;
586   chn->group_key = grp->pub_key;
587   chn->group_key_hash = grp->pub_key_hash;
588   chn->peer = *peer;
589   chn->direction = DIR_OUTGOING;
590   chn->join_status = JOIN_WAITING;
591   chn->channel = GNUNET_CADET_channel_create (cadet, chn, &chn->peer,
592                                               GNUNET_APPLICATION_TYPE_MULTICAST,
593                                               GNUNET_CADET_OPTION_RELIABLE);
594   GNUNET_CONTAINER_multihashmap_put (channels_out, &chn->group_key_hash, chn,
595                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
596   return chn;
597 }
598
599
600 /**
601  * Create CADET channel and send a join request.
602  */
603 static void
604 cadet_send_join_request (struct Member *mem)
605 {
606   mem->origin_channel = cadet_channel_create (&mem->grp, &mem->origin);
607   cadet_send_msg (mem->origin_channel, &mem->join_req->header);
608
609   uint32_t i;
610   for (i = 0; i < mem->relay_count; i++)
611   {
612     struct Channel *
613       chn = cadet_channel_create (&mem->grp, &mem->relays[i]);
614     cadet_send_msg (chn, &mem->join_req->header);
615   }
616 }
617
618
619 static int
620 cadet_send_join_decision_cb (void *cls,
621                              const struct GNUNET_HashCode *group_key_hash,
622                              void *channel)
623 {
624   const struct MulticastJoinDecisionMessageHeader *hdcsn = cls;
625   struct Channel *chn = channel;
626
627   if (0 == memcmp (&hdcsn->member_key, &chn->member_key, sizeof (chn->member_key))
628       && 0 == memcmp (&hdcsn->peer, &chn->peer, sizeof (chn->peer)))
629   {
630     cadet_send_msg (chn, &hdcsn->header);
631     return GNUNET_NO;
632   }
633   return GNUNET_YES;
634 }
635
636
637 /**
638  * Send join decision to a remote peer.
639  */
640 static void
641 cadet_send_join_decision (struct Group *grp,
642                           const struct MulticastJoinDecisionMessageHeader *hdcsn)
643 {
644   GNUNET_CONTAINER_multihashmap_get_multiple (channels_in, &grp->pub_key_hash,
645                                               &cadet_send_join_decision_cb,
646                                               (void *) hdcsn);
647 }
648
649
650 /**
651  * Iterator callback for sending a message to origin clients.
652  */
653 static int
654 cadet_send_members_cb (void *cls, const struct GNUNET_HashCode *pub_key_hash,
655                        void *channel)
656 {
657   const struct GNUNET_MessageHeader *msg = cls;
658   struct Channel *chn = channel;
659   if (JOIN_ADMITTED == chn->join_status)
660     cadet_send_msg (chn, msg);
661   return GNUNET_YES;
662 }
663
664
665 static int
666 cadet_send_members (struct GNUNET_HashCode *pub_key_hash,
667                     const struct GNUNET_MessageHeader *msg)
668 {
669   int n = 0;
670   if (channels_in != NULL)
671     n += GNUNET_CONTAINER_multihashmap_get_multiple (channels_in, pub_key_hash,
672                                                      cadet_send_members_cb,
673                                                      (void *) msg);
674   return n;
675 }
676
677 /**
678  * Handle a connecting client starting an origin.
679  */
680 static void
681 client_recv_origin_start (void *cls, struct GNUNET_SERVER_Client *client,
682                           const struct GNUNET_MessageHeader *m)
683 {
684   const struct MulticastOriginStartMessage *
685     msg = (const struct MulticastOriginStartMessage *) m;
686
687   struct GNUNET_CRYPTO_EddsaPublicKey pub_key;
688   struct GNUNET_HashCode pub_key_hash;
689
690   GNUNET_CRYPTO_eddsa_key_get_public (&msg->group_key, &pub_key);
691   GNUNET_CRYPTO_hash (&pub_key, sizeof (pub_key), &pub_key_hash);
692
693   struct Origin *
694     orig = GNUNET_CONTAINER_multihashmap_get (origins, &pub_key_hash);
695   struct Group *grp;
696
697   if (NULL == orig)
698   {
699     orig = GNUNET_new (struct Origin);
700     orig->priv_key = msg->group_key;
701     orig->max_fragment_id = GNUNET_ntohll (msg->max_fragment_id);
702     grp = &orig->grp;
703     grp->is_origin = GNUNET_YES;
704     grp->pub_key = pub_key;
705     grp->pub_key_hash = pub_key_hash;
706
707     GNUNET_CONTAINER_multihashmap_put (origins, &grp->pub_key_hash, orig,
708                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
709   }
710   else
711   {
712     grp = &orig->grp;
713   }
714
715   struct ClientList *cl = GNUNET_new (struct ClientList);
716   cl->client = client;
717   GNUNET_CONTAINER_DLL_insert (grp->clients_head, grp->clients_tail, cl);
718
719   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
720               "%p Client connected as origin to group %s.\n",
721               orig, GNUNET_h2s (&grp->pub_key_hash));
722
723   GNUNET_SERVER_client_set_user_context (client, grp);
724   GNUNET_SERVER_receive_done (client, GNUNET_OK);
725 }
726
727
728 /**
729  * Handle a connecting client joining a group.
730  */
731 static void
732 client_recv_member_join (void *cls, struct GNUNET_SERVER_Client *client,
733                          const struct GNUNET_MessageHeader *m)
734 {
735   const struct MulticastMemberJoinMessage *
736     msg = (const struct MulticastMemberJoinMessage *) m;
737   uint16_t msg_size = ntohs (msg->header.size);
738
739   struct GNUNET_CRYPTO_EcdsaPublicKey mem_pub_key;
740   struct GNUNET_HashCode pub_key_hash, mem_pub_key_hash;
741
742   GNUNET_CRYPTO_ecdsa_key_get_public (&msg->member_key, &mem_pub_key);
743   GNUNET_CRYPTO_hash (&mem_pub_key, sizeof (mem_pub_key), &mem_pub_key_hash);
744   GNUNET_CRYPTO_hash (&msg->group_key, sizeof (msg->group_key), &pub_key_hash);
745
746   struct GNUNET_CONTAINER_MultiHashMap *
747     grp_mem = GNUNET_CONTAINER_multihashmap_get (group_members, &pub_key_hash);
748   struct Member *mem = NULL;
749   struct Group *grp;
750
751   if (NULL != grp_mem)
752   {
753     mem = GNUNET_CONTAINER_multihashmap_get (grp_mem, &mem_pub_key_hash);
754   }
755   if (NULL == mem)
756   {
757     mem = GNUNET_new (struct Member);
758     mem->priv_key = msg->member_key;
759     mem->pub_key = mem_pub_key;
760     mem->pub_key_hash = mem_pub_key_hash;
761     mem->max_fragment_id = 0; // FIXME
762
763     grp = &mem->grp;
764     grp->is_origin = GNUNET_NO;
765     grp->pub_key = msg->group_key;
766     grp->pub_key_hash = pub_key_hash;
767
768     if (NULL == grp_mem)
769     {
770       grp_mem = GNUNET_CONTAINER_multihashmap_create (1, GNUNET_YES);
771       GNUNET_CONTAINER_multihashmap_put (group_members, &grp->pub_key_hash, grp_mem,
772                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
773     }
774     GNUNET_CONTAINER_multihashmap_put (grp_mem, &mem->pub_key_hash, mem,
775                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
776     GNUNET_CONTAINER_multihashmap_put (members, &grp->pub_key_hash, mem,
777                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
778   }
779   else
780   {
781     grp = &mem->grp;
782   }
783
784   struct ClientList *cl = GNUNET_new (struct ClientList);
785   cl->client = client;
786   GNUNET_CONTAINER_DLL_insert (grp->clients_head, grp->clients_tail, cl);
787
788   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
789               "%p Client connected to group %s..\n",
790               mem, GNUNET_h2s (&grp->pub_key_hash));
791   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
792               "%p ..as member %s.\n",
793               mem, GNUNET_h2s (&mem_pub_key_hash));
794
795   GNUNET_SERVER_client_set_user_context (client, grp);
796
797   if (NULL != mem->join_dcsn)
798   { /* Already got a join decision, send it to client. */
799     GNUNET_SERVER_notification_context_add (nc, client);
800     GNUNET_SERVER_notification_context_unicast (nc, client,
801                                                 (struct GNUNET_MessageHeader *)
802                                                 mem->join_dcsn,
803                                                 GNUNET_NO);
804   }
805   else if (grp->clients_head == grp->clients_tail)
806   { /* First client of the group, send join request. */
807     struct GNUNET_PeerIdentity *relays = (struct GNUNET_PeerIdentity *) &msg[1];
808     uint32_t relay_count = ntohs (msg->relay_count);
809     uint16_t relay_size = relay_count * sizeof (*relays);
810     struct GNUNET_MessageHeader *join_msg = NULL;
811     uint16_t join_msg_size = 0;
812     if (sizeof (*msg) + relay_size + sizeof (struct GNUNET_MessageHeader)
813         <= msg_size)
814     {
815       join_msg = (struct GNUNET_MessageHeader *)
816         (((char *) &msg[1]) + relay_size);
817       join_msg_size = ntohs (join_msg->size);
818     }
819     if (sizeof (*msg) + relay_size + join_msg_size != msg_size)
820     {
821       GNUNET_break (0);
822       GNUNET_SERVER_client_disconnect (client);
823       return;
824     }
825
826     struct MulticastJoinRequestMessage *
827       req = GNUNET_malloc (sizeof (*req) + join_msg_size);
828     req->header.size = htons (sizeof (*req) + join_msg_size);
829     req->header.type = htons (GNUNET_MESSAGE_TYPE_MULTICAST_JOIN_REQUEST);
830     req->group_key = grp->pub_key;
831     req->peer = this_peer;
832     GNUNET_CRYPTO_ecdsa_key_get_public (&mem->priv_key, &req->member_key);
833     if (0 < join_msg_size)
834       memcpy (&req[1], join_msg, join_msg_size);
835
836     req->purpose.size = htonl (msg_size
837                                - sizeof (req->header)
838                                - sizeof (req->reserved)
839                                - sizeof (req->signature));
840     req->purpose.purpose = htonl (GNUNET_SIGNATURE_PURPOSE_MULTICAST_REQUEST);
841
842     if (GNUNET_OK != GNUNET_CRYPTO_ecdsa_sign (&mem->priv_key, &req->purpose,
843                                                &req->signature))
844     {
845       /* FIXME: handle error */
846       GNUNET_assert (0);
847     }
848
849     if (NULL != mem->join_req)
850       GNUNET_free (mem->join_req);
851     mem->join_req = req;
852
853     if (0 == client_send_origin (&grp->pub_key_hash, &mem->join_req->header))
854     { /* No local origins, send to remote origin */
855       cadet_send_join_request (mem);
856     }
857   }
858   GNUNET_SERVER_receive_done (client, GNUNET_OK);
859 }
860
861
862 static void
863 client_send_join_decision (struct Member *mem,
864                            const struct MulticastJoinDecisionMessageHeader *hdcsn)
865 {
866   client_send_msg (&mem->grp, &hdcsn->header);
867
868   const struct MulticastJoinDecisionMessage *
869     dcsn = (const struct MulticastJoinDecisionMessage *) &hdcsn[1];
870   if (GNUNET_YES == ntohl (dcsn->is_admitted))
871   { /* Member admitted, store join_decision. */
872     uint16_t dcsn_size = ntohs (dcsn->header.size);
873     mem->join_dcsn = GNUNET_malloc (dcsn_size);
874     memcpy (mem->join_dcsn, dcsn, dcsn_size);
875   }
876   else
877   { /* Refused entry, disconnect clients. */
878     struct ClientList *cl = mem->grp.clients_head;
879     while (NULL != cl)
880     {
881       struct GNUNET_SERVER_Client *client = cl->client;
882       cl = cl->next;
883       GNUNET_SERVER_client_disconnect (client);
884     }
885   }
886 }
887
888
889 /**
890  * Join decision from client.
891  */
892 static void
893 client_recv_join_decision (void *cls, struct GNUNET_SERVER_Client *client,
894                            const struct GNUNET_MessageHeader *m)
895 {
896   struct Group *
897     grp = GNUNET_SERVER_client_get_user_context (client, struct Group);
898   const struct MulticastJoinDecisionMessageHeader *
899     hdcsn = (const struct MulticastJoinDecisionMessageHeader *) m;
900
901   if (NULL == grp)
902   {
903     GNUNET_break (0);
904     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
905     return;
906   }
907   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
908               "%p Got join decision from client for group %s..\n",
909               grp, GNUNET_h2s (&grp->pub_key_hash));
910
911   struct GNUNET_CONTAINER_MultiHashMap *
912     grp_mem = GNUNET_CONTAINER_multihashmap_get (group_members,
913                                                  &grp->pub_key_hash);
914   struct Member *mem = NULL;
915   if (NULL != grp_mem)
916   {
917     struct GNUNET_HashCode member_key_hash;
918     GNUNET_CRYPTO_hash (&hdcsn->member_key, sizeof (hdcsn->member_key),
919                         &member_key_hash);
920     mem = GNUNET_CONTAINER_multihashmap_get (grp_mem, &member_key_hash);
921     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
922                 "%p ..and member %s: %p\n",
923                 grp, GNUNET_h2s (&member_key_hash), mem);
924   }
925   if (NULL != mem)
926   { /* Found local member */
927     client_send_join_decision (mem, hdcsn);
928   }
929   else
930   { /* Look for remote member */
931     cadet_send_join_decision (grp, hdcsn);
932   }
933   GNUNET_SERVER_receive_done (client, GNUNET_OK);
934 }
935
936
937 /**
938  * Incoming message from a client.
939  */
940 static void
941 client_recv_multicast_message (void *cls, struct GNUNET_SERVER_Client *client,
942                                const struct GNUNET_MessageHeader *m)
943 {
944   struct Group *
945     grp = GNUNET_SERVER_client_get_user_context (client, struct Group);
946   struct GNUNET_MULTICAST_MessageHeader *out;
947   struct Origin *orig;
948
949   if (NULL == grp)
950   {
951     GNUNET_break (0);
952     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
953     return;
954   }
955   GNUNET_assert (GNUNET_YES == grp->is_origin);
956   orig = (struct Origin *) grp;
957   /* FIXME: yucky, should use separate message structs for P2P and CS! */
958   out = (struct GNUNET_MULTICAST_MessageHeader *) GNUNET_copy_message (m);
959
960   out->fragment_id = GNUNET_htonll (++orig->max_fragment_id);
961   out->purpose.size = htonl (ntohs (out->header.size)
962                              - sizeof (out->header)
963                              - sizeof (out->hop_counter)
964                              - sizeof (out->signature));
965   out->purpose.purpose = htonl (GNUNET_SIGNATURE_PURPOSE_MULTICAST_MESSAGE);
966
967   if (GNUNET_OK != GNUNET_CRYPTO_eddsa_sign (&orig->priv_key, &out->purpose,
968                                              &out->signature))
969   {
970     GNUNET_assert (0);
971   }
972
973   client_send_all (&grp->pub_key_hash, &out->header);
974   cadet_send_members (&grp->pub_key_hash, &out->header);
975   GNUNET_free (out);
976
977   GNUNET_SERVER_receive_done (client, GNUNET_OK);
978 }
979
980
981 /**
982  * Incoming request from a client.
983  */
984 static void
985 client_recv_multicast_request (void *cls, struct GNUNET_SERVER_Client *client,
986                                const struct GNUNET_MessageHeader *m)
987 {
988   struct Group *grp = GNUNET_SERVER_client_get_user_context (client, struct Group);
989   struct Member *mem;
990   struct GNUNET_MULTICAST_RequestHeader *out;
991
992   if (NULL == grp)
993   {
994     GNUNET_break (0);
995     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
996     return;
997   }
998   GNUNET_assert (GNUNET_NO == grp->is_origin);
999   mem = (struct Member *) grp;
1000   /* FIXME: yucky, should use separate message structs for P2P and CS! */
1001   out = (struct GNUNET_MULTICAST_RequestHeader *) GNUNET_copy_message (m);
1002
1003   out->fragment_id = GNUNET_ntohll (++mem->max_fragment_id);
1004   out->purpose.size = htonl (ntohs (out->header.size)
1005                              - sizeof (out->header)
1006                              - sizeof (out->member_key)
1007                              - sizeof (out->signature));
1008   out->purpose.purpose = htonl (GNUNET_SIGNATURE_PURPOSE_MULTICAST_REQUEST);
1009
1010   if (GNUNET_OK != GNUNET_CRYPTO_ecdsa_sign (&mem->priv_key, &out->purpose,
1011                                              &out->signature))
1012   {
1013     GNUNET_assert (0);
1014   }
1015
1016   if (0 == client_send_origin (&grp->pub_key_hash, &out->header))
1017   { /* No local origins, send to remote origin */
1018     if (NULL != mem->origin_channel)
1019     {
1020       cadet_send_msg (mem->origin_channel, &out->header);
1021     }
1022     else
1023     {
1024       /* FIXME: not yet connected to origin */
1025       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1026       GNUNET_free (out);
1027       return;
1028     }
1029   }
1030   GNUNET_free (out);
1031   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1032 }
1033
1034
1035 /**
1036  * A new client connected.
1037  */
1038 static void
1039 client_notify_connect (void *cls, struct GNUNET_SERVER_Client *client)
1040 {
1041   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Client connected: %p\n", client);
1042   /* FIXME: send connect ACK */
1043 }
1044
1045
1046 /**
1047  * Message handlers for the server.
1048  */
1049 static const struct GNUNET_SERVER_MessageHandler server_handlers[] = {
1050   { &client_recv_origin_start, NULL,
1051     GNUNET_MESSAGE_TYPE_MULTICAST_ORIGIN_START, 0 },
1052
1053   { &client_recv_member_join, NULL,
1054     GNUNET_MESSAGE_TYPE_MULTICAST_MEMBER_JOIN, 0 },
1055
1056   { &client_recv_join_decision, NULL,
1057     GNUNET_MESSAGE_TYPE_MULTICAST_JOIN_DECISION, 0 },
1058
1059   { &client_recv_multicast_message, NULL,
1060     GNUNET_MESSAGE_TYPE_MULTICAST_MESSAGE, 0 },
1061
1062   { &client_recv_multicast_request, NULL,
1063     GNUNET_MESSAGE_TYPE_MULTICAST_REQUEST, 0 },
1064
1065   {NULL, NULL, 0, 0}
1066 };
1067
1068
1069 /**
1070  * New incoming CADET channel.
1071  */
1072 static void *
1073 cadet_notify_channel_new (void *cls,
1074                           struct GNUNET_CADET_Channel *channel,
1075                           const struct GNUNET_PeerIdentity *initiator,
1076                           uint32_t port,
1077                           enum GNUNET_CADET_ChannelOption options)
1078 {
1079   return NULL;
1080 }
1081
1082
1083 /**
1084  * CADET channel is being destroyed.
1085  */
1086 static void
1087 cadet_notify_channel_end (void *cls,
1088                           const struct GNUNET_CADET_Channel *channel,
1089                           void *ctx)
1090 {
1091   if (NULL == ctx)
1092     return;
1093
1094   struct Channel *chn = ctx;
1095   if (NULL != chn->grp)
1096   {
1097     if (GNUNET_NO == chn->grp->is_origin)
1098     {
1099       struct Member *mem = (struct Member *) chn->grp;
1100       if (chn == mem->origin_channel)
1101         mem->origin_channel = NULL;
1102     }
1103   }
1104   GNUNET_free (chn);
1105 }
1106
1107
1108 /**
1109  * Incoming join request message from CADET.
1110  */
1111 int
1112 cadet_recv_join_request (void *cls,
1113                          struct GNUNET_CADET_Channel *channel,
1114                          void **ctx,
1115                          const struct GNUNET_MessageHeader *m)
1116 {
1117   const struct MulticastJoinRequestMessage *
1118     req = (const struct MulticastJoinRequestMessage *) m;
1119   uint16_t size = ntohs (m->size);
1120   if (size < sizeof (*req))
1121   {
1122     GNUNET_break_op (0);
1123     return GNUNET_SYSERR;
1124   }
1125   if (NULL != *ctx)
1126   {
1127     GNUNET_break_op (0);
1128     return GNUNET_SYSERR;
1129   }
1130   if (ntohl (req->purpose.size) != (size
1131                                     - sizeof (req->header)
1132                                     - sizeof (req->reserved)
1133                                     - sizeof (req->signature)))
1134   {
1135     GNUNET_break_op (0);
1136     return GNUNET_SYSERR;
1137   }
1138   if (GNUNET_OK !=
1139       GNUNET_CRYPTO_ecdsa_verify (GNUNET_SIGNATURE_PURPOSE_MULTICAST_REQUEST,
1140                                   &req->purpose, &req->signature,
1141                                   &req->member_key))
1142   {
1143     GNUNET_break_op (0);
1144     return GNUNET_SYSERR;
1145   }
1146
1147   struct GNUNET_HashCode group_key_hash;
1148   GNUNET_CRYPTO_hash (&req->group_key, sizeof (req->group_key), &group_key_hash);
1149
1150   struct Channel *chn = GNUNET_malloc (sizeof *chn);
1151   chn->channel = channel;
1152   chn->group_key = req->group_key;
1153   chn->group_key_hash = group_key_hash;
1154   chn->member_key = req->member_key;
1155   chn->peer = req->peer;
1156   chn->join_status = JOIN_WAITING;
1157   GNUNET_CONTAINER_multihashmap_put (channels_in, &chn->group_key_hash, chn,
1158                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
1159
1160   client_send_all (&group_key_hash, m);
1161   return GNUNET_OK;
1162 }
1163
1164
1165 /**
1166  * Incoming join decision message from CADET.
1167  */
1168 int
1169 cadet_recv_join_decision (void *cls,
1170                           struct GNUNET_CADET_Channel *channel,
1171                           void **ctx,
1172                           const struct GNUNET_MessageHeader *m)
1173 {
1174   const struct MulticastJoinDecisionMessage *
1175     dcsn = (const struct MulticastJoinDecisionMessage *) m;
1176   uint16_t size = ntohs (m->size);
1177   if (size < sizeof (*dcsn))
1178   {
1179     GNUNET_break_op (0);
1180     return GNUNET_SYSERR;
1181   }
1182   struct Channel *chn = *ctx;
1183   if (NULL == chn)
1184   {
1185     GNUNET_break_op (0);
1186     return GNUNET_SYSERR;
1187   }
1188   if (NULL == chn->grp || GNUNET_NO != chn->grp->is_origin)
1189   {
1190     GNUNET_break_op (0);
1191     return GNUNET_SYSERR;
1192   }
1193   switch (chn->join_status)
1194   {
1195   case JOIN_REFUSED:
1196     return GNUNET_SYSERR;
1197
1198   case JOIN_ADMITTED:
1199     return GNUNET_OK;
1200
1201   case JOIN_NOT_ASKED:
1202   case JOIN_WAITING:
1203     break;
1204   }
1205
1206   struct MulticastJoinDecisionMessageHeader *
1207     hdcsn = GNUNET_malloc (sizeof (*hdcsn) + size);
1208   hdcsn->peer = chn->peer;
1209   memcpy (&hdcsn[1], dcsn, sizeof (*hdcsn) + size);
1210
1211   struct Member *mem = (struct Member *) chn->grp;
1212   client_send_join_decision (mem, hdcsn);
1213   GNUNET_free (hdcsn);
1214   if (GNUNET_YES == ntohs (dcsn->is_admitted))
1215   {
1216     chn->join_status = JOIN_ADMITTED;
1217     return GNUNET_OK;
1218   }
1219   else
1220   {
1221     chn->join_status = JOIN_REFUSED;
1222     return GNUNET_SYSERR;
1223   }
1224 }
1225
1226 /**
1227  * Incoming multicast message from CADET.
1228  */
1229 int
1230 cadet_recv_message (void *cls,
1231                     struct GNUNET_CADET_Channel *channel,
1232                     void **ctx,
1233                     const struct GNUNET_MessageHeader *m)
1234 {
1235   const struct GNUNET_MULTICAST_MessageHeader *
1236     msg = (const struct GNUNET_MULTICAST_MessageHeader *) m;
1237   uint16_t size = ntohs (m->size);
1238   if (size < sizeof (*msg))
1239   {
1240     GNUNET_break_op (0);
1241     return GNUNET_SYSERR;
1242   }
1243   struct Channel *chn = *ctx;
1244   if (NULL == chn)
1245   {
1246     GNUNET_break_op (0);
1247     return GNUNET_SYSERR;
1248   }
1249   if (ntohl (msg->purpose.size) != (size
1250                                     - sizeof (msg->header)
1251                                     - sizeof (msg->hop_counter)
1252                                     - sizeof (msg->signature)))
1253   {
1254     GNUNET_break_op (0);
1255     return GNUNET_SYSERR;
1256   }
1257   if (GNUNET_OK !=
1258       GNUNET_CRYPTO_eddsa_verify (GNUNET_SIGNATURE_PURPOSE_MULTICAST_MESSAGE,
1259                                   &msg->purpose, &msg->signature,
1260                                   &chn->group_key))
1261   {
1262     GNUNET_break_op (0);
1263     return GNUNET_SYSERR;
1264   }
1265
1266   client_send_all (&chn->group_key_hash, m);
1267   return GNUNET_OK;
1268 }
1269
1270
1271 /**
1272  * Incoming multicast request message from CADET.
1273  */
1274 int
1275 cadet_recv_request (void *cls,
1276                     struct GNUNET_CADET_Channel *channel,
1277                     void **ctx,
1278                     const struct GNUNET_MessageHeader *m)
1279 {
1280   const struct GNUNET_MULTICAST_RequestHeader *
1281     req = (const struct GNUNET_MULTICAST_RequestHeader *) m;
1282   uint16_t size = ntohs (m->size);
1283   if (size < sizeof (*req))
1284   {
1285     GNUNET_break_op (0);
1286     return GNUNET_SYSERR;
1287   }
1288   struct Channel *chn = *ctx;
1289   if (NULL == chn)
1290   {
1291     GNUNET_break_op (0);
1292     return GNUNET_SYSERR;
1293   }
1294   if (ntohl (req->purpose.size) != (size
1295                                     - sizeof (req->header)
1296                                     - sizeof (req->member_key)
1297                                     - sizeof (req->signature)))
1298   {
1299     GNUNET_break_op (0);
1300     return GNUNET_SYSERR;
1301   }
1302   if (GNUNET_OK !=
1303       GNUNET_CRYPTO_ecdsa_verify (GNUNET_SIGNATURE_PURPOSE_MULTICAST_REQUEST,
1304                                   &req->purpose, &req->signature,
1305                                   &req->member_key))
1306   {
1307     GNUNET_break_op (0);
1308     return GNUNET_SYSERR;
1309   }
1310
1311   client_send_origin (&chn->group_key_hash, m);
1312   return GNUNET_OK;
1313 }
1314
1315
1316 /**
1317  * Message handlers for CADET.
1318  */
1319 static const struct GNUNET_CADET_MessageHandler cadet_handlers[] = {
1320   { &cadet_recv_join_request, GNUNET_MESSAGE_TYPE_MULTICAST_JOIN_REQUEST, 0 },
1321   { &cadet_recv_message, GNUNET_MESSAGE_TYPE_MULTICAST_MESSAGE, 0 },
1322   { &cadet_recv_request, GNUNET_MESSAGE_TYPE_MULTICAST_REQUEST, 0 },
1323   { NULL, 0, 0 }
1324 };
1325
1326
1327 /**
1328  * Listening ports for CADET.
1329  */
1330 static const uint32_t cadet_ports[] = { GNUNET_APPLICATION_TYPE_MULTICAST, 0 };
1331
1332
1333 /**
1334  * Connected to core service.
1335  */
1336 static void
1337 core_connected_cb  (void *cls, const struct GNUNET_PeerIdentity *my_identity)
1338 {
1339   this_peer = *my_identity;
1340
1341   stats = GNUNET_STATISTICS_create ("multicast", cfg);
1342   origins = GNUNET_CONTAINER_multihashmap_create (1, GNUNET_YES);
1343   members = GNUNET_CONTAINER_multihashmap_create (1, GNUNET_YES);
1344   group_members = GNUNET_CONTAINER_multihashmap_create (1, GNUNET_NO);
1345   channels_in = GNUNET_CONTAINER_multihashmap_create (1, GNUNET_YES);
1346   channels_out = GNUNET_CONTAINER_multihashmap_create (1, GNUNET_YES);
1347
1348   cadet = GNUNET_CADET_connect (cfg, NULL,
1349                                 &cadet_notify_channel_new,
1350                                 &cadet_notify_channel_end,
1351                                 cadet_handlers, cadet_ports);
1352
1353   nc = GNUNET_SERVER_notification_context_create (server, 1);
1354   GNUNET_SERVER_add_handlers (server, server_handlers);
1355   GNUNET_SERVER_disconnect_notify (server, &client_notify_disconnect, NULL);
1356
1357   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
1358                                 NULL);
1359 }
1360
1361
1362 /**
1363  * Service started.
1364  *
1365  * @param cls closure
1366  * @param server the initialized server
1367  * @param cfg configuration to use
1368  */
1369 static void
1370 run (void *cls, struct GNUNET_SERVER_Handle *srv,
1371      const struct GNUNET_CONFIGURATION_Handle *c)
1372 {
1373   cfg = c;
1374   server = srv;
1375   GNUNET_SERVER_connect_notify (server, &client_notify_connect, NULL);
1376   core = GNUNET_CORE_connect (cfg, NULL, &core_connected_cb, NULL, NULL,
1377                               NULL, GNUNET_NO, NULL, GNUNET_NO, NULL);
1378 }
1379
1380
1381 /**
1382  * The main function for the multicast service.
1383  *
1384  * @param argc number of arguments from the command line
1385  * @param argv command line arguments
1386  * @return 0 ok, 1 on error
1387  */
1388 int
1389 main (int argc, char *const *argv)
1390 {
1391   return (GNUNET_OK ==
1392           GNUNET_SERVICE_run (argc, argv, "multicast",
1393                               GNUNET_SERVICE_OPTION_NONE, &run, NULL)) ? 0 : 1;
1394 }
1395
1396 /* end of gnunet-service-multicast.c */