Eliminate mesh priority option
[oweals/gnunet.git] / src / mesh / mesh_api.c
1 /*
2      This file is part of GNUnet.
3      (C) 2011 Christian Grothoff (and other contributing authors)
4      GNUnet is free software; you can redistribute it and/or modify
5      it under the terms of the GNU General Public License as published
6      by the Free Software Foundation; either version 3, or (at your
7      option) any later version.
8      GNUnet is distributed in the hope that it will be useful, but
9      WITHOUT ANY WARRANTY; without even the implied warranty of
10      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11      General Public License for more details.
12      You should have received a copy of the GNU General Public License
13      along with GNUnet; see the file COPYING.  If not, write to the
14      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
15      Boston, MA 02111-1307, USA.
16 */
17
18 /**
19  * @file mesh/mesh_api.c
20  * @brief mesh api: client implementation of mesh service
21  * @author Bartlomiej Polot
22  *
23  * STRUCTURE:
24  * - CONSTANTS
25  * - DATA STRUCTURES
26  * - AUXILIARY FUNCTIONS
27  * - RECEIVE HANDLERS
28  * - SEND FUNCTIONS
29  * - API CALL DEFINITIONS
30  *
31  * TODO: add regex to reconnect
32  */
33 #include "platform.h"
34 #include "gnunet_common.h"
35 #include "gnunet_client_lib.h"
36 #include "gnunet_util_lib.h"
37 #include "gnunet_peer_lib.h"
38 #include "gnunet_mesh_service.h"
39 #include "mesh.h"
40 #include "mesh_protocol.h"
41
42 #define LOG(kind,...) GNUNET_log_from (kind, "mesh-api",__VA_ARGS__)
43
44
45 /******************************************************************************/
46 /************************        CONSTANTS         ****************************/
47 /******************************************************************************/
48
49 #define HIGH_PID 0xFFFF0000
50 #define LOW_PID 0x0000FFFF
51
52 #define PID_OVERFLOW(pid, max) (pid > HIGH_PID && max < LOW_PID)
53
54 /******************************************************************************/
55 /************************      DATA STRUCTURES     ****************************/
56 /******************************************************************************/
57
58 /**
59  * Transmission queue to the service
60  */
61 struct GNUNET_MESH_TransmitHandle
62 {
63
64     /**
65      * Double Linked list
66      */
67   struct GNUNET_MESH_TransmitHandle *next;
68
69     /**
70      * Double Linked list
71      */
72   struct GNUNET_MESH_TransmitHandle *prev;
73
74     /**
75      * Tunnel this message is sent on / for (may be NULL for control messages).
76      */
77   struct GNUNET_MESH_Tunnel *tunnel;
78
79     /**
80      * Callback to obtain the message to transmit, or NULL if we
81      * got the message in 'data'.  Notice that messages built
82      * by 'notify' need to be encapsulated with information about
83      * the 'target'.
84      */
85   GNUNET_CONNECTION_TransmitReadyNotify notify;
86
87     /**
88      * Closure for 'notify'
89      */
90   void *notify_cls;
91
92     /**
93      * How long is this message valid.  Once the timeout has been
94      * reached, the message must no longer be sent.  If this
95      * is a message with a 'notify' callback set, the 'notify'
96      * function should be called with 'buf' NULL and size 0.
97      */
98   struct GNUNET_TIME_Absolute timeout;
99
100     /**
101      * Task triggering a timeout, can be NO_TASK if the timeout is FOREVER.
102      */
103   GNUNET_SCHEDULER_TaskIdentifier timeout_task;
104
105     /**
106      * Target of the message, 0 for multicast.  This field
107      * is only valid if 'notify' is non-NULL.
108      */
109   GNUNET_PEER_Id target;
110
111     /**
112      * Size of 'data' -- or the desired size of 'notify' if 'data' is NULL.
113      */
114   size_t size;
115 };
116
117
118 /**
119  * Opaque handle to the service.
120  */
121 struct GNUNET_MESH_Handle
122 {
123
124     /**
125      * Handle to the server connection, to send messages later
126      */
127   struct GNUNET_CLIENT_Connection *client;
128
129     /**
130      * Set of handlers used for processing incoming messages in the tunnels
131      */
132   const struct GNUNET_MESH_MessageHandler *message_handlers;
133
134     /**
135      * Set of applications that should be claimed to be offered at this node.
136      * Note that this is just informative, the appropiate handlers must be
137      * registered independently and the mapping is up to the developer of the
138      * client application.
139      */
140   const GNUNET_MESH_ApplicationType *applications;
141
142     /**
143      * Double linked list of the tunnels this client is connected to, head.
144      */
145   struct GNUNET_MESH_Tunnel *tunnels_head;
146
147     /**
148      * Double linked list of the tunnels this client is connected to, tail.
149      */
150   struct GNUNET_MESH_Tunnel *tunnels_tail;
151
152     /**
153      * Callback for inbound tunnel creation
154      */
155   GNUNET_MESH_InboundTunnelNotificationHandler *new_tunnel;
156
157     /**
158      * Callback for inbound tunnel disconnection
159      */
160   GNUNET_MESH_TunnelEndHandler *cleaner;
161
162     /**
163      * Handle to cancel pending transmissions in case of disconnection
164      */
165   struct GNUNET_CLIENT_TransmitHandle *th;
166
167     /**
168      * Closure for all the handlers given by the client
169      */
170   void *cls;
171
172     /**
173      * Messages to send to the service, head.
174      */
175   struct GNUNET_MESH_TransmitHandle *th_head;
176
177     /**
178      * Messages to send to the service, tail.
179      */
180   struct GNUNET_MESH_TransmitHandle *th_tail;
181
182     /**
183      * tid of the next tunnel to create (to avoid reusing IDs often)
184      */
185   MESH_TunnelNumber next_tid;
186
187     /**
188      * Number of handlers in the handlers array.
189      */
190   unsigned int n_handlers;
191
192     /**
193      * Number of applications in the applications array.
194      */
195   unsigned int n_applications;
196
197     /**
198      * Have we started the task to receive messages from the service
199      * yet? We do this after we send the 'MESH_LOCAL_CONNECT' message.
200      */
201   int in_receive;
202
203   /**
204    * Configuration given by the client, in case of reconnection
205    */
206   const struct GNUNET_CONFIGURATION_Handle *cfg;
207
208   /**
209    * Time to the next reconnect in case one reconnect fails
210    */
211   struct GNUNET_TIME_Relative reconnect_time;
212   
213   /**
214    * Task for trying to reconnect.
215    */
216   GNUNET_SCHEDULER_TaskIdentifier reconnect_task;
217 };
218
219
220 /**
221  * Description of a peer
222  */
223 struct GNUNET_MESH_Peer
224 {
225     /**
226      * ID of the peer in short form
227      */
228   GNUNET_PEER_Id id;
229
230   /**
231    * Tunnel this peer belongs to
232    */
233   struct GNUNET_MESH_Tunnel *t;
234
235   /**
236    * Flag indicating whether service has informed about its connection
237    */
238   int connected;
239
240 };
241
242
243 /**
244  * Opaque handle to a tunnel.
245  */
246 struct GNUNET_MESH_Tunnel
247 {
248
249     /**
250      * DLL next
251      */
252   struct GNUNET_MESH_Tunnel *next;
253
254     /**
255      * DLL prev
256      */
257   struct GNUNET_MESH_Tunnel *prev;
258
259     /**
260      * Callback to execute when peers connect to the tunnel
261      */
262   GNUNET_MESH_PeerConnectHandler connect_handler;
263
264     /**
265      * Callback to execute when peers disconnect from the tunnel
266      */
267   GNUNET_MESH_PeerDisconnectHandler disconnect_handler;
268
269     /**
270      * Closure for the connect/disconnect handlers
271      */
272   void *cls;
273
274     /**
275      * Handle to the mesh this tunnel belongs to
276      */
277   struct GNUNET_MESH_Handle *mesh;
278
279     /**
280      * Local ID of the tunnel
281      */
282   MESH_TunnelNumber tid;
283
284     /**
285      * Owner of the tunnel. 0 if the tunnel is the local client.
286      */
287   GNUNET_PEER_Id owner;
288
289     /**
290      * All peers added to the tunnel
291      */
292   struct GNUNET_MESH_Peer **peers;
293
294   /**
295    * List of application types that have been requested for this tunnel
296    */
297   GNUNET_MESH_ApplicationType *apps;
298
299   /**
300    * Any data the caller wants to put in here
301    */
302   void *ctx;
303
304   /**
305      * Number of peers added to the tunnel
306      */
307   unsigned int npeers;
308
309     /**
310      * Number of packets queued in this tunnel
311      */
312   unsigned int npackets;
313
314     /**
315      * Number of applications requested this tunnel
316      */
317   unsigned int napps;
318
319     /**
320      * Is the tunnel throttled to the slowest peer?
321      */
322   int speed_min;
323
324     /**
325      * Is the tunnel allowed to buffer?
326      */
327   int buffering;
328
329     /**
330      * Next packet PID.
331      */
332   uint32_t pid;
333
334     /**
335      * Maximum allowed PID.
336      */
337   uint32_t max_pid;
338
339
340 };
341
342
343 /******************************************************************************/
344 /***********************     AUXILIARY FUNCTIONS      *************************/
345 /******************************************************************************/
346
347 /**
348  * Get the tunnel handler for the tunnel specified by id from the given handle
349  * @param h Mesh handle
350  * @param tid ID of the wanted tunnel
351  * @return handle to the required tunnel or NULL if not found
352  */
353 static struct GNUNET_MESH_Tunnel *
354 retrieve_tunnel (struct GNUNET_MESH_Handle *h, MESH_TunnelNumber tid)
355 {
356   struct GNUNET_MESH_Tunnel *t;
357
358   t = h->tunnels_head;
359   while (t != NULL)
360   {
361     if (t->tid == tid)
362       return t;
363     t = t->next;
364   }
365   return NULL;
366 }
367
368
369 /**
370  * Create a new tunnel and insert it in the tunnel list of the mesh handle
371  * @param h Mesh handle
372  * @param tid desired tid of the tunnel, 0 to assign one automatically
373  * @return handle to the created tunnel
374  */
375 static struct GNUNET_MESH_Tunnel *
376 create_tunnel (struct GNUNET_MESH_Handle *h, MESH_TunnelNumber tid)
377 {
378   struct GNUNET_MESH_Tunnel *t;
379
380   t = GNUNET_malloc (sizeof (struct GNUNET_MESH_Tunnel));
381   GNUNET_CONTAINER_DLL_insert (h->tunnels_head, h->tunnels_tail, t);
382   t->mesh = h;
383   if (0 == tid)
384   {
385     t->tid = h->next_tid;
386     while (NULL != retrieve_tunnel (h, h->next_tid))
387     {
388       h->next_tid++;
389       h->next_tid &= ~GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
390       h->next_tid |= GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
391     }
392   }
393   else
394   {
395     t->tid = tid;
396   }
397   return t;
398 }
399
400
401 /**
402  * Destroy the specified tunnel.
403  * - Destroys all peers, calling the disconnect callback on each if needed
404  * - Cancels all outgoing traffic for that tunnel, calling respective notifys
405  * - Calls cleaner if tunnel was inbound
406  * - Frees all memory used
407  *
408  * @param t Pointer to the tunnel.
409  * @param call_cleaner Whether to call the cleaner handler.
410  *
411  * @return Handle to the required tunnel or NULL if not found.
412  */
413 static void
414 destroy_tunnel (struct GNUNET_MESH_Tunnel *t, int call_cleaner)
415 {
416   struct GNUNET_MESH_Handle *h;
417   struct GNUNET_PeerIdentity pi;
418   struct GNUNET_MESH_TransmitHandle *th;
419   struct GNUNET_MESH_TransmitHandle *next;
420   unsigned int i;
421
422   LOG (GNUNET_ERROR_TYPE_DEBUG, "destroy_tunnel %X\n", t->tid);
423
424   if (NULL == t)
425   {
426     GNUNET_break (0);
427     return;
428   }
429   h = t->mesh;
430
431   /* disconnect all peers */
432   GNUNET_CONTAINER_DLL_remove (h->tunnels_head, h->tunnels_tail, t);
433   for (i = 0; i < t->npeers; i++)
434   {
435     if ( (NULL != t->disconnect_handler) && t->peers[i]->connected)
436     {
437       GNUNET_PEER_resolve (t->peers[i]->id, &pi);
438       t->disconnect_handler (t->cls, &pi);
439     }
440     GNUNET_PEER_change_rc (t->peers[i]->id, -1);
441     GNUNET_free (t->peers[i]);
442   }
443
444   /* signal tunnel destruction */
445   if ( (NULL != h->cleaner) && (0 != t->owner) && (GNUNET_YES == call_cleaner) )
446     h->cleaner (h->cls, t, t->ctx);
447
448   /* check that clients did not leave messages behind in the queue */
449   for (th = h->th_head; NULL != th; th = next)
450   {
451     next = th->next;
452     if (th->tunnel != t)
453       continue;
454     /* Clients should have aborted their requests already.
455      * Management traffic should be ok, as clients can't cancel that */
456     GNUNET_break (NULL == th->notify);
457     GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th);
458
459     /* clean up request */
460     if (GNUNET_SCHEDULER_NO_TASK != th->timeout_task)
461       GNUNET_SCHEDULER_cancel (th->timeout_task);
462     GNUNET_free (th);    
463   }
464
465   /* if there are no more pending requests with mesh service, cancel active request */
466   /* Note: this should be unnecessary... */
467   if ( (NULL == h->th_head) && (NULL != h->th))
468   {
469     GNUNET_CLIENT_notify_transmit_ready_cancel (h->th);
470     h->th = NULL;
471   }
472
473
474   if (t->npeers > 0)
475     GNUNET_free (t->peers);
476   if (0 != t->owner)
477     GNUNET_PEER_change_rc (t->owner, -1);
478   if (0 != t->napps && t->apps)
479     GNUNET_free (t->apps);
480   GNUNET_free (t);
481   return;
482 }
483
484
485 /**
486  * Get the peer descriptor for the peer with id from the given tunnel
487  * @param t Tunnel handle
488  * @param id Short form ID of the wanted peer
489  * @return handle to the requested peer or NULL if not found
490  */
491 static struct GNUNET_MESH_Peer *
492 retrieve_peer (struct GNUNET_MESH_Tunnel *t, GNUNET_PEER_Id id)
493 {
494   unsigned int i;
495
496   for (i = 0; i < t->npeers; i++)
497     if (t->peers[i]->id == id)
498       return t->peers[i];
499   return NULL;
500 }
501
502
503 /**
504  * Add a peer into a tunnel
505  * @param t Tunnel handle
506  * @param pi Full ID of the new peer
507  * @return handle to the newly created peer
508  */
509 static struct GNUNET_MESH_Peer *
510 add_peer_to_tunnel (struct GNUNET_MESH_Tunnel *t,
511                     const struct GNUNET_PeerIdentity *pi)
512 {
513   struct GNUNET_MESH_Peer *p;
514   GNUNET_PEER_Id id;
515
516   if (0 != t->owner)
517   {
518     GNUNET_break (0);
519     return NULL;
520   }
521   id = GNUNET_PEER_intern (pi);
522
523   p = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer));
524   p->id = id;
525   p->t = t;
526   GNUNET_array_append (t->peers, t->npeers, p);
527   return p;
528 }
529
530
531 /**
532  * Remove a peer from a tunnel
533  * @param p Peer handle
534  */
535 static void
536 remove_peer_from_tunnel (struct GNUNET_MESH_Peer *p)
537 {
538   unsigned int i;
539
540   for (i = 0; i < p->t->npeers; i++)
541   {
542     if (p->t->peers[i] == p)
543       break;
544   }
545   if (i == p->t->npeers)
546   {
547     GNUNET_break (0);
548     return;
549   }
550   p->t->peers[i] = p->t->peers[p->t->npeers - 1];
551   GNUNET_array_grow (p->t->peers, p->t->npeers, p->t->npeers - 1);
552 }
553
554
555 /**
556  * Notify client that the transmission has timed out
557  * @param cls closure
558  * @param tc task context
559  */
560 static void
561 timeout_transmission (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
562 {
563   struct GNUNET_MESH_TransmitHandle *th = cls;
564   struct GNUNET_MESH_Handle *mesh;
565
566   mesh = th->tunnel->mesh;
567   GNUNET_CONTAINER_DLL_remove (mesh->th_head, mesh->th_tail, th);
568   if (th->notify != NULL)
569     th->notify (th->notify_cls, 0, NULL);
570   GNUNET_free (th);
571   if ((NULL == mesh->th_head) && (NULL != mesh->th))
572   {
573     /* queue empty, no point in asking for transmission */
574     GNUNET_CLIENT_notify_transmit_ready_cancel (mesh->th);
575     mesh->th = NULL;
576   }
577 }
578
579
580 /**
581  * Add a transmit handle to the transmission queue by priority and set the
582  * timeout if needed.
583  *
584  * @param h mesh handle with the queue head and tail
585  * @param th handle to the packet to be transmitted
586  */
587 static void
588 add_to_queue (struct GNUNET_MESH_Handle *h,
589               struct GNUNET_MESH_TransmitHandle *th)
590 {
591   struct GNUNET_MESH_TransmitHandle *p;
592
593   p = h->th_head;
594   while ((NULL != p))
595     p = p->next;
596   if (NULL == p)
597     p = h->th_tail;
598   else
599     p = p->prev;
600   GNUNET_CONTAINER_DLL_insert_after (h->th_head, h->th_tail, p, th);
601   if (GNUNET_TIME_UNIT_FOREVER_ABS.abs_value == th->timeout.abs_value)
602     return;
603   th->timeout_task =
604       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_absolute_get_remaining
605                                     (th->timeout), &timeout_transmission, th);
606 }
607
608
609 /**
610  * Auxiliary function to send an already constructed packet to the service.
611  * Takes care of creating a new queue element, copying the message and
612  * calling the tmt_rdy function if necessary.
613  *
614  * @param h mesh handle
615  * @param msg message to transmit
616  * @param tunnel tunnel this send is related to (NULL if N/A)
617  */
618 static void
619 send_packet (struct GNUNET_MESH_Handle *h,
620              const struct GNUNET_MessageHeader *msg,
621              struct GNUNET_MESH_Tunnel *tunnel);
622
623
624 /**
625  * Reconnect callback: tries to reconnect again after a failer previous
626  * reconnecttion
627  * @param cls closure (mesh handle)
628  * @param tc task context
629  */
630 static void
631 reconnect_cbk (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
632
633
634 /**
635  * Send a connect packet to the service with the applications and types
636  * requested by the user.
637  *
638  * @param h The mesh handle.
639  *
640  */
641 static void
642 send_connect (struct GNUNET_MESH_Handle *h)
643 {
644   size_t size;
645
646   size = sizeof (struct GNUNET_MESH_ClientConnect);
647   size += h->n_applications * sizeof (GNUNET_MESH_ApplicationType);
648   size += h->n_handlers * sizeof (uint16_t);
649   {
650     char buf[size] GNUNET_ALIGN;
651     struct GNUNET_MESH_ClientConnect *msg;
652     GNUNET_MESH_ApplicationType *apps;
653     uint16_t napps;
654     uint16_t *types;
655     uint16_t ntypes;
656
657     /* build connection packet */
658     msg = (struct GNUNET_MESH_ClientConnect *) buf;
659     msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT);
660     msg->header.size = htons (size);
661     apps = (GNUNET_MESH_ApplicationType *) &msg[1];
662     for (napps = 0; napps < h->n_applications; napps++)
663     {
664       apps[napps] = htonl (h->applications[napps]);
665       LOG (GNUNET_ERROR_TYPE_DEBUG, " app %u\n", h->applications[napps]);
666     }
667     types = (uint16_t *) & apps[napps];
668     for (ntypes = 0; ntypes < h->n_handlers; ntypes++)
669       types[ntypes] = htons (h->message_handlers[ntypes].type);
670     msg->applications = htons (napps);
671     msg->types = htons (ntypes);
672     LOG (GNUNET_ERROR_TYPE_DEBUG,
673          "Sending %lu bytes long message %d types and %d apps\n",
674          ntohs (msg->header.size), ntypes, napps);
675     send_packet (h, &msg->header, NULL);
676   }
677 }
678
679
680 /**
681  * Reconnect to the service, retransmit all infomation to try to restore the
682  * original state.
683  *
684  * @param h handle to the mesh
685  *
686  * @return GNUNET_YES in case of sucess, GNUNET_NO otherwise (service down...)
687  */
688 static int
689 do_reconnect (struct GNUNET_MESH_Handle *h)
690 {
691   struct GNUNET_MESH_Tunnel *t;
692   unsigned int i;
693
694   LOG (GNUNET_ERROR_TYPE_DEBUG, "*****************************\n");
695   LOG (GNUNET_ERROR_TYPE_DEBUG, "*******   RECONNECT   *******\n");
696   LOG (GNUNET_ERROR_TYPE_DEBUG, "*****************************\n");
697
698   h->in_receive = GNUNET_NO;
699   /* disconnect */
700   if (NULL != h->th)
701   {
702     GNUNET_CLIENT_notify_transmit_ready_cancel (h->th);
703     h->th = NULL;
704   }
705   if (NULL != h->client)
706   {
707     GNUNET_CLIENT_disconnect (h->client);
708   }
709
710   /* connect again */
711   h->client = GNUNET_CLIENT_connect ("mesh", h->cfg);
712   if (h->client == NULL)
713   {
714     h->reconnect_task = GNUNET_SCHEDULER_add_delayed (h->reconnect_time,
715                                                       &reconnect_cbk, h);
716     h->reconnect_time =
717         GNUNET_TIME_relative_min (GNUNET_TIME_UNIT_SECONDS,
718                                   GNUNET_TIME_relative_multiply
719                                   (h->reconnect_time, 2));
720     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Next retry in %sms\n",
721          GNUNET_TIME_relative_to_string (h->reconnect_time));
722     GNUNET_break (0);
723     return GNUNET_NO;
724   }
725   else
726   {
727     h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS;
728   }
729   send_connect (h);
730   /* Rebuild all tunnels */
731   for (t = h->tunnels_head; NULL != t; t = t->next)
732   {
733     struct GNUNET_MESH_TunnelMessage tmsg;
734     struct GNUNET_MESH_PeerControl pmsg;
735
736     if (t->tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
737     {
738       /* Tunnel was created by service (incoming tunnel) */
739       /* TODO: Notify service of missing tunnel, to request
740        * creator to recreate path (find a path to him via DHT?)
741        */
742       continue;
743     }
744     tmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
745     tmsg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
746     tmsg.tunnel_id = htonl (t->tid);
747     send_packet (h, &tmsg.header, t);
748
749     pmsg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
750     pmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
751     pmsg.tunnel_id = htonl (t->tid);
752
753     /* Reconnect all peers */
754     /* If the tunnel was "by type", dont connect individual peers */
755     for (i = 0; i < t->npeers && 0 == t->napps; i++)
756     {
757       GNUNET_PEER_resolve (t->peers[i]->id, &pmsg.peer);
758       if (NULL != t->disconnect_handler && t->peers[i]->connected)
759         t->disconnect_handler (t->cls, &pmsg.peer);
760       send_packet (t->mesh, &pmsg.header, t);
761     }
762     /* Reconnect all types, if any  */
763     for (i = 0; i < t->napps; i++)
764     {
765       struct GNUNET_MESH_ConnectPeerByType msg;
766
767       msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
768       msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE);
769       msg.tunnel_id = htonl (t->tid);
770       msg.type = htonl (t->apps[i]);
771       send_packet (t->mesh, &msg.header, t);
772     }
773     if (GNUNET_NO == t->buffering)
774       GNUNET_MESH_tunnel_buffer (t, GNUNET_NO);
775     if (GNUNET_YES == t->speed_min)
776       GNUNET_MESH_tunnel_speed_min (t);
777   }
778   return GNUNET_YES;
779 }
780
781 /**
782  * Reconnect callback: tries to reconnect again after a failer previous
783  * reconnecttion
784  * @param cls closure (mesh handle)
785  * @param tc task context
786  */
787 static void
788 reconnect_cbk (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
789 {
790   struct GNUNET_MESH_Handle *h = cls;
791
792   h->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
793   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
794     return;
795   do_reconnect (h);
796 }
797
798
799 /**
800  * Reconnect to the service, retransmit all infomation to try to restore the
801  * original state.
802  *
803  * @param h handle to the mesh
804  *
805  * @return GNUNET_YES in case of sucess, GNUNET_NO otherwise (service down...)
806  */
807 static void
808 reconnect (struct GNUNET_MESH_Handle *h)
809 {
810   LOG (GNUNET_ERROR_TYPE_DEBUG, "Requested RECONNECT\n");
811   if (GNUNET_SCHEDULER_NO_TASK == h->reconnect_task)
812     h->reconnect_task = GNUNET_SCHEDULER_add_delayed (h->reconnect_time,
813                                                       &reconnect_cbk, h);
814 }
815
816
817 /******************************************************************************/
818 /***********************      RECEIVE HANDLERS     ****************************/
819 /******************************************************************************/
820
821 /**
822  * Process the new tunnel notification and add it to the tunnels in the handle
823  *
824  * @param h     The mesh handle
825  * @param msg   A message with the details of the new incoming tunnel
826  */
827 static void
828 process_tunnel_created (struct GNUNET_MESH_Handle *h,
829                         const struct GNUNET_MESH_TunnelNotification *msg)
830 {
831   struct GNUNET_MESH_Tunnel *t;
832   MESH_TunnelNumber tid;
833
834   tid = ntohl (msg->tunnel_id);
835   if (tid < GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
836   {
837     GNUNET_break (0);
838     return;
839   }
840   if (NULL != h->new_tunnel)
841   {
842     struct GNUNET_ATS_Information atsi;
843
844     t = create_tunnel (h, tid);
845     t->owner = GNUNET_PEER_intern (&msg->peer);
846     t->npeers = 1;
847     t->peers = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer *));
848     t->peers[0] = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer));
849     t->peers[0]->t = t;
850     t->peers[0]->connected = 1;
851     t->peers[0]->id = t->owner;
852     GNUNET_PEER_change_rc (t->owner, 1);
853     t->mesh = h;
854     t->tid = tid;
855     atsi.type = 0;
856     atsi.value = 0;
857     t->ctx = h->new_tunnel (h->cls, t, &msg->peer, &atsi);
858     LOG (GNUNET_ERROR_TYPE_DEBUG, "new incoming tunnel %X\n", t->tid);
859   }
860   else
861   {
862     struct GNUNET_MESH_TunnelMessage d_msg;
863
864     LOG (GNUNET_ERROR_TYPE_DEBUG, "No handler for incoming tunnels\n");
865
866     d_msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY);
867     d_msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
868     d_msg.tunnel_id = msg->tunnel_id;
869
870     send_packet (h, &d_msg.header, NULL);
871   }
872   return;
873 }
874
875
876 /**
877  * Process the tunnel destroy notification and free associated resources
878  *
879  * @param h     The mesh handle
880  * @param msg   A message with the details of the tunnel being destroyed
881  */
882 static void
883 process_tunnel_destroy (struct GNUNET_MESH_Handle *h,
884                         const struct GNUNET_MESH_TunnelMessage *msg)
885 {
886   struct GNUNET_MESH_Tunnel *t;
887   MESH_TunnelNumber tid;
888
889   tid = ntohl (msg->tunnel_id);
890   t = retrieve_tunnel (h, tid);
891
892   if (NULL == t)
893   {
894     return;
895   }
896   if (0 == t->owner)
897   {
898     GNUNET_break (0);
899   }
900   LOG (GNUNET_ERROR_TYPE_DEBUG, "tunnel %X destroyed\n", t->tid);
901   destroy_tunnel (t, GNUNET_YES);
902   return;
903 }
904
905
906 /**
907  * Process the new peer event and notify the upper level of it
908  *
909  * @param h     The mesh handle
910  * @param msg   A message with the details of the peer event
911  */
912 static void
913 process_peer_event (struct GNUNET_MESH_Handle *h,
914                     const struct GNUNET_MESH_PeerControl *msg)
915 {
916   struct GNUNET_MESH_Tunnel *t;
917   struct GNUNET_MESH_Peer *p;
918   struct GNUNET_ATS_Information atsi;
919   GNUNET_PEER_Id id;
920   uint16_t size;
921
922   LOG (GNUNET_ERROR_TYPE_DEBUG, "processig peer event\n");
923   size = ntohs (msg->header.size);
924   if (size != sizeof (struct GNUNET_MESH_PeerControl))
925   {
926     GNUNET_break (0);
927     return;
928   }
929   t = retrieve_tunnel (h, ntohl (msg->tunnel_id));
930   if (NULL == t)
931   {
932     GNUNET_break (0);
933     return;
934   }
935   id = GNUNET_PEER_search (&msg->peer);
936   if ((p = retrieve_peer (t, id)) == NULL)
937     p = add_peer_to_tunnel (t, &msg->peer);
938   if (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD == ntohs (msg->header.type))
939   {
940     LOG (GNUNET_ERROR_TYPE_DEBUG, "adding peer\n");
941     if (NULL != t->connect_handler)
942     {
943       atsi.type = 0;
944       atsi.value = 0;
945       t->connect_handler (t->cls, &msg->peer, &atsi);
946     }
947     p->connected = 1;
948   }
949   else
950   {
951     LOG (GNUNET_ERROR_TYPE_DEBUG, "removing peer\n");
952     if (NULL != t->disconnect_handler && p->connected)
953     {
954       t->disconnect_handler (t->cls, &msg->peer);
955     }
956     remove_peer_from_tunnel (p);
957     GNUNET_free (p);
958   }
959   LOG (GNUNET_ERROR_TYPE_DEBUG, "processing peer event END\n");
960 }
961
962
963 /**
964  * Process the incoming data packets
965  *
966  * @param h         The mesh handle
967  * @param message   A message encapsulating the data
968  * 
969  * @return GNUNET_YES if everything went fine
970  *         GNUNET_NO if client closed connection (h no longer valid)
971  */
972 static int
973 process_incoming_data (struct GNUNET_MESH_Handle *h,
974                        const struct GNUNET_MessageHeader *message)
975 {
976   const struct GNUNET_MessageHeader *payload;
977   const struct GNUNET_MESH_MessageHandler *handler;
978   const struct GNUNET_PeerIdentity *peer;
979   struct GNUNET_MESH_Unicast *ucast;
980   struct GNUNET_MESH_Multicast *mcast;
981   struct GNUNET_MESH_ToOrigin *to_orig;
982   struct GNUNET_MESH_Tunnel *t;
983   unsigned int i;
984   uint16_t type;
985
986   LOG (GNUNET_ERROR_TYPE_DEBUG, "Got a data message!\n");
987   type = ntohs (message->type);
988   switch (type)
989   {
990   case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
991     ucast = (struct GNUNET_MESH_Unicast *) message;
992
993     t = retrieve_tunnel (h, ntohl (ucast->tid));
994     payload = (struct GNUNET_MessageHeader *) &ucast[1];
995     peer = &ucast->oid;
996     LOG (GNUNET_ERROR_TYPE_DEBUG, "  ucast on tunnel %s [%X]\n",
997          GNUNET_i2s (peer), ntohl (ucast->tid));
998     break;
999   case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
1000     mcast = (struct GNUNET_MESH_Multicast *) message;
1001     t = retrieve_tunnel (h, ntohl (mcast->tid));
1002     payload = (struct GNUNET_MessageHeader *) &mcast[1];
1003     peer = &mcast->oid;
1004     LOG (GNUNET_ERROR_TYPE_DEBUG, "  mcast on tunnel %s [%X]\n",
1005          GNUNET_i2s (peer), ntohl (mcast->tid));
1006     break;
1007   case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
1008     to_orig = (struct GNUNET_MESH_ToOrigin *) message;
1009     t = retrieve_tunnel (h, ntohl (to_orig->tid));
1010     payload = (struct GNUNET_MessageHeader *) &to_orig[1];
1011     peer = &to_orig->sender;
1012     LOG (GNUNET_ERROR_TYPE_DEBUG, "  torig on tunnel %s [%X]\n",
1013          GNUNET_i2s (peer), ntohl (to_orig->tid));
1014     break;
1015   default:
1016     GNUNET_break (0);
1017     return GNUNET_YES;
1018   }
1019   if (NULL == t)
1020   {
1021     /* Tunnel was ignored, probably service didn't get it yet */
1022     return GNUNET_YES;
1023   }
1024   type = ntohs (payload->type);
1025   for (i = 0; i < h->n_handlers; i++)
1026   {
1027     handler = &h->message_handlers[i];
1028     if (handler->type == type)
1029     {
1030       struct GNUNET_ATS_Information atsi;
1031
1032       atsi.type = 0;
1033       atsi.value = 0;
1034       if (GNUNET_OK !=
1035           handler->callback (h->cls, t, &t->ctx, peer, payload, &atsi))
1036       {
1037         LOG (GNUNET_ERROR_TYPE_DEBUG, "callback caused disconnection\n");
1038         GNUNET_MESH_disconnect (h);
1039         return GNUNET_NO;
1040       }
1041       else
1042       {
1043         LOG (GNUNET_ERROR_TYPE_DEBUG,
1044              "callback completed successfully\n");
1045       }
1046     }
1047   }
1048   return GNUNET_YES;
1049 }
1050
1051
1052 /**
1053  * Process a local ACK message, enabling the client to send
1054  * more data to the service.
1055  * 
1056  * @param h Mesh handle.
1057  * @param message Message itself.
1058  */
1059 static void
1060 process_ack (struct GNUNET_MESH_Handle *h,
1061              const struct GNUNET_MessageHeader *message)
1062 {
1063   struct GNUNET_MESH_LocalAck *msg;
1064   struct GNUNET_MESH_Tunnel *t;
1065   uint32_t ack;
1066
1067   LOG (GNUNET_ERROR_TYPE_DEBUG, "Got an ACK!\n");
1068   msg = (struct GNUNET_MESH_LocalAck *) message;
1069
1070   t = retrieve_tunnel (h, ntohl (msg->tunnel_id));
1071
1072   if (NULL == t)
1073   {
1074     LOG (GNUNET_ERROR_TYPE_WARNING,
1075          "ACK on unknown tunnel %X\n",
1076          ntohl (msg->tunnel_id));
1077     return;
1078   }
1079   ack = ntohl (msg->max_pid);
1080   if (ack > t->max_pid || PID_OVERFLOW (t->max_pid, ack))
1081     t->max_pid = ack;
1082 }
1083
1084
1085 /**
1086  * Function to process all messages received from the service
1087  *
1088  * @param cls closure
1089  * @param msg message received, NULL on timeout or fatal error
1090  */
1091 static void
1092 msg_received (void *cls, const struct GNUNET_MessageHeader *msg)
1093 {
1094   struct GNUNET_MESH_Handle *h = cls;
1095
1096   if (msg == NULL)
1097   {
1098     LOG (GNUNET_ERROR_TYPE_DEBUG, "Received NULL msg\n");
1099     reconnect (h);
1100     return;
1101   }
1102   LOG (GNUNET_ERROR_TYPE_DEBUG, "received a message type %hu from MESH\n",
1103        ntohs (msg->type));
1104   switch (ntohs (msg->type))
1105   {
1106     /* Notify of a new incoming tunnel */
1107   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE:
1108     process_tunnel_created (h, (struct GNUNET_MESH_TunnelNotification *) msg);
1109     break;
1110     /* Notify of a tunnel disconnection */
1111   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY:
1112     process_tunnel_destroy (h, (struct GNUNET_MESH_TunnelMessage *) msg);
1113     break;
1114     /* Notify of a new peer or a peer disconnect in the tunnel */
1115   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD:
1116   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL:
1117     process_peer_event (h, (struct GNUNET_MESH_PeerControl *) msg);
1118     break;
1119     /* Notify of a new data packet in the tunnel */
1120   case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
1121   case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
1122   case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
1123     if (GNUNET_NO == process_incoming_data (h, msg))
1124       return;
1125     break;
1126   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK:
1127     process_ack (h, msg);
1128     break;
1129   default:
1130     /* We shouldn't get any other packages, log and ignore */
1131     LOG (GNUNET_ERROR_TYPE_WARNING,
1132          "unsolicited message form service (type %d)\n",
1133          ntohs (msg->type));
1134   }
1135   LOG (GNUNET_ERROR_TYPE_DEBUG, "message processed\n");
1136   GNUNET_CLIENT_receive (h->client, &msg_received, h,
1137                          GNUNET_TIME_UNIT_FOREVER_REL);
1138 }
1139
1140
1141 /******************************************************************************/
1142 /************************       SEND FUNCTIONS     ****************************/
1143 /******************************************************************************/
1144
1145 /**
1146  * Function called to send a message to the service.
1147  * "buf" will be NULL and "size" zero if the socket was closed for writing in
1148  * the meantime.
1149  *
1150  * @param cls closure, the mesh handle
1151  * @param size number of bytes available in buf
1152  * @param buf where the callee should write the connect message
1153  * @return number of bytes written to buf
1154  */
1155 static size_t
1156 send_callback (void *cls, size_t size, void *buf)
1157 {
1158   struct GNUNET_MESH_Handle *h = cls;
1159   struct GNUNET_MESH_TransmitHandle *th;
1160   char *cbuf = buf;
1161   size_t tsize;
1162   size_t psize;
1163
1164   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send packet() Buffer %u\n", size);
1165   h->th = NULL;
1166   if ((0 == size) || (NULL == buf))
1167   {
1168     LOG (GNUNET_ERROR_TYPE_DEBUG, "Received NULL callback\n");
1169     reconnect (h);
1170     return 0;
1171   }
1172   tsize = 0;
1173   while ((NULL != (th = h->th_head)) && (size >= th->size))
1174   {
1175     if (NULL != th->notify)
1176     {
1177       if (th->tunnel->tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
1178       {
1179         /* traffic to origin */
1180         struct GNUNET_MESH_ToOrigin to;
1181         struct GNUNET_MessageHeader *mh;
1182
1183         GNUNET_assert (size >= th->size);
1184         mh = (struct GNUNET_MessageHeader *) &cbuf[sizeof (to)];
1185         psize = th->notify (th->notify_cls, size - sizeof (to), mh);
1186         LOG (GNUNET_ERROR_TYPE_DEBUG, "  to origin, type %u\n",
1187              ntohs (mh->type));
1188         if (psize > 0)
1189         {
1190           psize += sizeof (to);
1191           GNUNET_assert (size >= psize);
1192           to.header.size = htons (psize);
1193           to.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN);
1194           to.tid = htonl (th->tunnel->tid);
1195           memset (&to.oid, 0, sizeof (struct GNUNET_PeerIdentity));
1196           memset (&to.sender, 0, sizeof (struct GNUNET_PeerIdentity));
1197           memcpy (cbuf, &to, sizeof (to));
1198         }
1199       }
1200       else if (th->target == 0)
1201       {
1202         /* multicast */
1203         struct GNUNET_MESH_Multicast mc;
1204         struct GNUNET_MessageHeader *mh;
1205
1206         GNUNET_assert (size >= th->size);
1207         mh = (struct GNUNET_MessageHeader *) &cbuf[sizeof (mc)];
1208         psize = th->notify (th->notify_cls, size - sizeof (mc), mh);
1209         LOG (GNUNET_ERROR_TYPE_DEBUG, "  multicast, type %u\n",
1210              ntohs (mh->type));
1211         if (psize > 0)
1212         {
1213           psize += sizeof (mc);
1214           GNUNET_assert (size >= psize);
1215           mc.header.size = htons (psize);
1216           mc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_MULTICAST);
1217           mc.tid = htonl (th->tunnel->tid);
1218           mc.pid = 0;
1219           mc.ttl = 0;
1220           memset (&mc.oid, 0, sizeof (struct GNUNET_PeerIdentity));
1221           memcpy (cbuf, &mc, sizeof (mc));
1222         }
1223       }
1224       else
1225       {
1226         /* unicast */
1227         struct GNUNET_MESH_Unicast uc;
1228         struct GNUNET_MessageHeader *mh;
1229
1230         GNUNET_assert (size >= th->size);
1231         mh = (struct GNUNET_MessageHeader *) &cbuf[sizeof (uc)];
1232         psize = th->notify (th->notify_cls, size - sizeof (uc), mh);
1233         LOG (GNUNET_ERROR_TYPE_DEBUG, "  unicast, type %u\n",
1234              ntohs (mh->type));
1235         if (psize > 0)
1236         {
1237           psize += sizeof (uc);
1238           GNUNET_assert (size >= psize);
1239           uc.header.size = htons (psize);
1240           uc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_UNICAST);
1241           uc.tid = htonl (th->tunnel->tid);
1242           memset (&uc.oid, 0, sizeof (struct GNUNET_PeerIdentity));
1243           GNUNET_PEER_resolve (th->target, &uc.destination);
1244           memcpy (cbuf, &uc, sizeof (uc));
1245         }
1246       }
1247     }
1248     else
1249     {
1250       struct GNUNET_MessageHeader *mh = (struct GNUNET_MessageHeader *) &th[1];
1251       LOG (GNUNET_ERROR_TYPE_DEBUG, "  mesh traffic, type %u\n",
1252              ntohs (mh->type));
1253       memcpy (cbuf, &th[1], th->size);
1254       psize = th->size;
1255     }
1256     if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1257       GNUNET_SCHEDULER_cancel (th->timeout_task);
1258     if (NULL != th->notify)
1259     {
1260       th->tunnel->npackets--;
1261     }
1262     GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th);
1263     GNUNET_free (th);
1264     cbuf += psize;
1265     size -= psize;
1266     tsize += psize;
1267   }
1268   LOG (GNUNET_ERROR_TYPE_DEBUG, "  total size: %u\n", tsize);
1269   if (NULL != (th = h->th_head))
1270   {
1271     LOG (GNUNET_ERROR_TYPE_DEBUG, "  next size: %u\n", th->size);
1272     if (NULL == h->th)
1273       h->th =
1274           GNUNET_CLIENT_notify_transmit_ready (h->client, th->size,
1275                                                GNUNET_TIME_UNIT_FOREVER_REL,
1276                                                GNUNET_YES, &send_callback, h);
1277   }
1278   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send packet() END\n");
1279   if (GNUNET_NO == h->in_receive)
1280   {
1281     h->in_receive = GNUNET_YES;
1282     GNUNET_CLIENT_receive (h->client, &msg_received, h,
1283                            GNUNET_TIME_UNIT_FOREVER_REL);
1284   }
1285   return tsize;
1286 }
1287
1288
1289 /**
1290  * Auxiliary function to send an already constructed packet to the service.
1291  * Takes care of creating a new queue element, copying the message and
1292  * calling the tmt_rdy function if necessary.
1293  * 
1294  * @param h mesh handle
1295  * @param msg message to transmit
1296  * @param tunnel tunnel this send is related to (NULL if N/A)
1297  */
1298 static void
1299 send_packet (struct GNUNET_MESH_Handle *h,
1300              const struct GNUNET_MessageHeader *msg,
1301              struct GNUNET_MESH_Tunnel *tunnel)
1302 {
1303   struct GNUNET_MESH_TransmitHandle *th;
1304   size_t msize;
1305
1306   msize = ntohs (msg->size);
1307   th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle) + msize);
1308   th->timeout = GNUNET_TIME_UNIT_FOREVER_ABS;
1309   th->size = msize;
1310   th->tunnel = tunnel;
1311   memcpy (&th[1], msg, msize);
1312   add_to_queue (h, th);
1313   if (NULL != h->th)
1314     return;
1315   h->th =
1316       GNUNET_CLIENT_notify_transmit_ready (h->client, msize,
1317                                            GNUNET_TIME_UNIT_FOREVER_REL,
1318                                            GNUNET_YES, &send_callback, h);
1319 }
1320
1321
1322 /******************************************************************************/
1323 /**********************      API CALL DEFINITIONS     *************************/
1324 /******************************************************************************/
1325
1326 /**
1327  * Connect to the mesh service.
1328  *
1329  * @param cfg configuration to use
1330  * @param cls closure for the various callbacks that follow
1331  *            (including handlers in the handlers array)
1332  * @param new_tunnel function called when an *inbound* tunnel is created
1333  * @param cleaner function called when an *inbound* tunnel is destroyed by the
1334  *                remote peer, it is *not* called if GNUNET_MESH_tunnel_destroy
1335  *                is called on the tunnel
1336  * @param handlers callbacks for messages we care about, NULL-terminated
1337  *                note that the mesh is allowed to drop notifications about
1338  *                inbound messages if the client does not process them fast
1339  *                enough (for this notification type, a bounded queue is used)
1340  * @param stypes list of the applications that this client claims to provide
1341  * @return handle to the mesh service NULL on error
1342  *         (in this case, init is never called)
1343  */
1344 struct GNUNET_MESH_Handle *
1345 GNUNET_MESH_connect (const struct GNUNET_CONFIGURATION_Handle *cfg, void *cls,
1346                      GNUNET_MESH_InboundTunnelNotificationHandler new_tunnel,
1347                      GNUNET_MESH_TunnelEndHandler cleaner,
1348                      const struct GNUNET_MESH_MessageHandler *handlers,
1349                      const GNUNET_MESH_ApplicationType *stypes)
1350 {
1351   struct GNUNET_MESH_Handle *h;
1352
1353   LOG (GNUNET_ERROR_TYPE_DEBUG, "GNUNET_MESH_connect()\n");
1354   h = GNUNET_malloc (sizeof (struct GNUNET_MESH_Handle));
1355   h->cfg = cfg;
1356   h->new_tunnel = new_tunnel;
1357   h->cleaner = cleaner;
1358   h->client = GNUNET_CLIENT_connect ("mesh", cfg);
1359   if (h->client == NULL)
1360   {
1361     GNUNET_break (0);
1362     GNUNET_free (h);
1363     return NULL;
1364   }
1365   h->cls = cls;
1366   /* FIXME memdup? */
1367   h->applications = stypes;
1368   h->message_handlers = handlers;
1369   h->next_tid = GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
1370   h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS;
1371   h->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
1372
1373   /* count handlers and apps, calculate size */
1374   for (h->n_applications = 0; stypes[h->n_applications]; h->n_applications++) ;
1375   for (h->n_handlers = 0; handlers[h->n_handlers].type; h->n_handlers++) ;
1376   send_connect (h);
1377   LOG (GNUNET_ERROR_TYPE_DEBUG, "GNUNET_MESH_connect() END\n");
1378   return h;
1379 }
1380
1381
1382 /**
1383  * Disconnect from the mesh service. All tunnels will be destroyed. All tunnel
1384  * disconnect callbacks will be called on any still connected peers, notifying
1385  * about their disconnection. The registered inbound tunnel cleaner will be
1386  * called should any inbound tunnels still exist.
1387  *
1388  * @param handle connection to mesh to disconnect
1389  */
1390 void
1391 GNUNET_MESH_disconnect (struct GNUNET_MESH_Handle *handle)
1392 {
1393   struct GNUNET_MESH_Tunnel *t;
1394   struct GNUNET_MESH_Tunnel *aux;
1395   struct GNUNET_MESH_TransmitHandle *th;
1396
1397   LOG (GNUNET_ERROR_TYPE_DEBUG, "MESH DISCONNECT\n");
1398
1399   t = handle->tunnels_head;
1400   while (NULL != t)
1401   {
1402     aux = t->next;
1403     if (t->tid < GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
1404     {
1405       GNUNET_break (0);
1406       LOG (GNUNET_ERROR_TYPE_DEBUG, "tunnel %X not destroyed\n", t->tid);
1407     }
1408     destroy_tunnel (t, GNUNET_YES);
1409     t = aux;
1410   }
1411   while ( (th = handle->th_head) != NULL)
1412   {
1413     struct GNUNET_MessageHeader *msg;
1414
1415     /* Make sure it is an allowed packet (everything else should have been
1416      * already canceled).
1417      */
1418     GNUNET_break (NULL == th->notify);
1419     msg = (struct GNUNET_MessageHeader *) &th[1];
1420     switch (ntohs(msg->type))
1421     {
1422       case GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT:
1423       case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY:
1424         break;
1425       default:
1426         GNUNET_break (0);
1427         LOG (GNUNET_ERROR_TYPE_DEBUG, "unexpected msg %u\n",
1428              ntohs(msg->type));
1429     }
1430
1431     GNUNET_CONTAINER_DLL_remove (handle->th_head, handle->th_tail, th);
1432     GNUNET_free (th);
1433   }
1434
1435   if (NULL != handle->th)
1436   {
1437     GNUNET_CLIENT_notify_transmit_ready_cancel (handle->th);
1438     handle->th = NULL;
1439   }
1440   if (NULL != handle->client)
1441   {
1442     GNUNET_CLIENT_disconnect (handle->client);
1443     handle->client = NULL;
1444   }
1445   if (GNUNET_SCHEDULER_NO_TASK != handle->reconnect_task)
1446   {
1447     GNUNET_SCHEDULER_cancel(handle->reconnect_task);
1448     handle->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
1449   }
1450   GNUNET_free (handle);
1451 }
1452
1453
1454 /**
1455  * Announce to ther peer the availability of services described by the regex,
1456  * in order to be reachable to other peers via connect_by_string.
1457  * 
1458  * Note that the first 8 characters are considered to be part of a prefix,
1459  * (for instance 'gnunet://'). If you put a variable part in there (*, +. ()),
1460  * all matching strings will be stored in the DHT.
1461  *
1462  * @param h handle to mesh.
1463  * @param regex string with the regular expression describing local services.
1464  */
1465 void
1466 GNUNET_MESH_announce_regex (struct GNUNET_MESH_Handle *h,
1467                             const char *regex)
1468 {
1469   struct GNUNET_MessageHeader *msg;
1470   size_t len;
1471   size_t msgsize;
1472
1473   len = strlen (regex);
1474   msgsize = sizeof(struct GNUNET_MessageHeader) + len;
1475   GNUNET_assert (UINT16_MAX > msgsize);
1476
1477   {
1478     char buffer[msgsize];
1479
1480     msg = (struct GNUNET_MessageHeader *) buffer;
1481     msg->size = htons (msgsize);
1482     msg->type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_ANNOUNCE_REGEX);
1483     memcpy (&msg[1], regex, len);
1484
1485     send_packet(h, msg, NULL);
1486   }
1487 }
1488
1489 /**
1490  * Create a new tunnel (we're initiator and will be allowed to add/remove peers
1491  * and to broadcast).
1492  *
1493  * @param h mesh handle
1494  * @param tunnel_ctx client's tunnel context to associate with the tunnel
1495  * @param connect_handler function to call when peers are actually connected
1496  * @param disconnect_handler function to call when peers are disconnected
1497  * @param handler_cls closure for connect/disconnect handlers
1498  */
1499 struct GNUNET_MESH_Tunnel *
1500 GNUNET_MESH_tunnel_create (struct GNUNET_MESH_Handle *h, void *tunnel_ctx,
1501                            GNUNET_MESH_PeerConnectHandler connect_handler,
1502                            GNUNET_MESH_PeerDisconnectHandler disconnect_handler,
1503                            void *handler_cls)
1504 {
1505   struct GNUNET_MESH_Tunnel *t;
1506   struct GNUNET_MESH_TunnelMessage msg;
1507
1508   LOG (GNUNET_ERROR_TYPE_DEBUG, "Creating new tunnel\n");
1509   t = create_tunnel (h, 0);
1510   t->connect_handler = connect_handler;
1511   t->disconnect_handler = disconnect_handler;
1512   t->cls = handler_cls;
1513   t->ctx = tunnel_ctx;
1514   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
1515   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1516   msg.tunnel_id = htonl (t->tid);
1517   send_packet (h, &msg.header, t);
1518   return t;
1519 }
1520
1521
1522 /**
1523  * Destroy an existing tunnel. The existing callback for the tunnel will NOT
1524  * be called.
1525  *
1526  * @param tunnel tunnel handle
1527  */
1528 void
1529 GNUNET_MESH_tunnel_destroy (struct GNUNET_MESH_Tunnel *tunnel)
1530 {
1531   struct GNUNET_MESH_Handle *h;
1532   struct GNUNET_MESH_TunnelMessage msg;
1533   struct GNUNET_MESH_TransmitHandle *th;
1534
1535   LOG (GNUNET_ERROR_TYPE_DEBUG, "Destroying tunnel\n");
1536   h = tunnel->mesh;
1537
1538   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY);
1539   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1540   msg.tunnel_id = htonl (tunnel->tid);
1541   th = h->th_head;
1542   while (th != NULL)
1543   {
1544     struct GNUNET_MESH_TransmitHandle *aux;
1545     if (th->tunnel == tunnel)
1546     {
1547       aux = th->next;
1548       /* FIXME call the handler? */
1549       if (NULL != th->notify)
1550         th->notify (th->notify_cls, 0, NULL);
1551       GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th);
1552       GNUNET_free (th);
1553       th = aux;
1554     }
1555     else
1556       th = th->next;
1557   }
1558
1559   destroy_tunnel (tunnel, GNUNET_NO);
1560   send_packet (h, &msg.header, NULL);
1561 }
1562
1563 /**
1564  * Request that the tunnel data rate is limited to the speed of the slowest
1565  * receiver.
1566  *
1567  * @param tunnel Tunnel affected.
1568  */
1569 void
1570 GNUNET_MESH_tunnel_speed_min (struct GNUNET_MESH_Tunnel *tunnel)
1571 {
1572   struct GNUNET_MESH_TunnelMessage msg;
1573   struct GNUNET_MESH_Handle *h;
1574
1575   h = tunnel->mesh;
1576   tunnel->speed_min = GNUNET_YES;
1577
1578   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_MIN);
1579   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1580   msg.tunnel_id = htonl (tunnel->tid);
1581
1582   send_packet (h, &msg.header, NULL);
1583 }
1584
1585
1586 /**
1587  * Request that the tunnel data rate is limited to the speed of the fastest
1588  * receiver. This is the default behavior.
1589  *
1590  * @param tunnel Tunnel affected.
1591  */
1592 void
1593 GNUNET_MESH_tunnel_speed_max (struct GNUNET_MESH_Tunnel *tunnel)
1594 {
1595   struct GNUNET_MESH_TunnelMessage msg;
1596   struct GNUNET_MESH_Handle *h;
1597
1598   h = tunnel->mesh;
1599   tunnel->speed_min = GNUNET_NO;
1600
1601   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_MAX);
1602   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1603   msg.tunnel_id = htonl (tunnel->tid);
1604
1605   send_packet (h, &msg.header, NULL);
1606 }
1607
1608 /**
1609  * Turn on/off the buffering status of the tunnel.
1610  * 
1611  * @param tunnel Tunnel affected.
1612  * @param buffer GNUNET_YES to turn buffering on (default),
1613  *               GNUNET_NO otherwise.
1614  */
1615 void
1616 GNUNET_MESH_tunnel_buffer (struct GNUNET_MESH_Tunnel *tunnel, int buffer)
1617 {
1618   struct GNUNET_MESH_TunnelMessage msg;
1619   struct GNUNET_MESH_Handle *h;
1620
1621   h = tunnel->mesh;
1622   tunnel->buffering = buffer;
1623
1624   if (GNUNET_YES == buffer)
1625     msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_BUFFER);
1626   else
1627     msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_NOBUFFER);
1628   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1629   msg.tunnel_id = htonl (tunnel->tid);
1630
1631   send_packet (h, &msg.header, NULL);
1632 }
1633
1634 /**
1635  * Request that a peer should be added to the tunnel.  The existing
1636  * connect handler will be called ONCE with either success or failure.
1637  * This function should NOT be called again with the same peer before the
1638  * connect handler is called.
1639  *
1640  * @param tunnel handle to existing tunnel
1641  * @param peer peer to add
1642  */
1643 void
1644 GNUNET_MESH_peer_request_connect_add (struct GNUNET_MESH_Tunnel *tunnel,
1645                                       const struct GNUNET_PeerIdentity *peer)
1646 {
1647   struct GNUNET_MESH_PeerControl msg;
1648   GNUNET_PEER_Id peer_id;
1649   unsigned int i;
1650
1651   peer_id = GNUNET_PEER_intern (peer);
1652   for (i = 0; i < tunnel->npeers; i++)
1653   {
1654     if (tunnel->peers[i]->id == peer_id)
1655     {
1656       /* Peer already exists in tunnel */
1657       GNUNET_PEER_change_rc (peer_id, -1);
1658       GNUNET_break (0);
1659       return;
1660     }
1661   }
1662   if (NULL == add_peer_to_tunnel (tunnel, peer))
1663     return;
1664
1665   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1666   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
1667   msg.tunnel_id = htonl (tunnel->tid);
1668   msg.peer = *peer;
1669   send_packet (tunnel->mesh, &msg.header, tunnel);
1670
1671   return;
1672 }
1673
1674
1675 /**
1676  * Request that a peer should be removed from the tunnel.  The existing
1677  * disconnect handler will be called ONCE if we were connected.
1678  *
1679  * @param tunnel handle to existing tunnel
1680  * @param peer peer to remove
1681  */
1682 void
1683 GNUNET_MESH_peer_request_connect_del (struct GNUNET_MESH_Tunnel *tunnel,
1684                                       const struct GNUNET_PeerIdentity *peer)
1685 {
1686   struct GNUNET_MESH_PeerControl msg;
1687   GNUNET_PEER_Id peer_id;
1688   unsigned int i;
1689
1690   peer_id = GNUNET_PEER_search (peer);
1691   if (0 == peer_id)
1692   {
1693     GNUNET_break (0);
1694     return;
1695   }
1696   for (i = 0; i < tunnel->npeers; i++)
1697     if (tunnel->peers[i]->id == peer_id)
1698       break;
1699   if (i == tunnel->npeers)
1700   {
1701     GNUNET_break (0);
1702     return;
1703   }
1704   if (NULL != tunnel->disconnect_handler && tunnel->peers[i]->connected == 1)
1705     tunnel->disconnect_handler (tunnel->cls, peer);
1706   GNUNET_PEER_change_rc (peer_id, -1);
1707   GNUNET_free (tunnel->peers[i]);
1708   tunnel->peers[i] = tunnel->peers[tunnel->npeers - 1];
1709   GNUNET_array_grow (tunnel->peers, tunnel->npeers, tunnel->npeers - 1);
1710
1711   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1712   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL);
1713   msg.tunnel_id = htonl (tunnel->tid);
1714   memcpy (&msg.peer, peer, sizeof (struct GNUNET_PeerIdentity));
1715   send_packet (tunnel->mesh, &msg.header, tunnel);
1716 }
1717
1718
1719 /**
1720  * Request that the mesh should try to connect to a peer supporting the given
1721  * message type.
1722  *
1723  * @param tunnel handle to existing tunnel
1724  * @param app_type application type that must be supported by the peer (MESH
1725  *                 should discover peer in proximity handling this type)
1726  */
1727 void
1728 GNUNET_MESH_peer_request_connect_by_type (struct GNUNET_MESH_Tunnel *tunnel,
1729                                           GNUNET_MESH_ApplicationType app_type)
1730 {
1731   struct GNUNET_MESH_ConnectPeerByType msg;
1732
1733   GNUNET_array_append (tunnel->apps, tunnel->napps, app_type);
1734
1735   LOG (GNUNET_ERROR_TYPE_DEBUG, "* CONNECT BY TYPE *\n");
1736   msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
1737   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE);
1738   msg.tunnel_id = htonl (tunnel->tid);
1739   msg.type = htonl (app_type);
1740   send_packet (tunnel->mesh, &msg.header, tunnel);
1741 }
1742
1743
1744 /**
1745  * Request that the mesh should try to connect to a peer matching the
1746  * description given in the service string.
1747  * 
1748  * FIXME: allow multiple? how to deal with reconnect?
1749  *
1750  * @param tunnel handle to existing tunnel
1751  * @param description string describing the destination node requirements
1752  */
1753 void
1754 GNUNET_MESH_peer_request_connect_by_string (struct GNUNET_MESH_Tunnel *tunnel,
1755                                             const char *description)
1756 {
1757   struct GNUNET_MESH_ConnectPeerByString *m;
1758   size_t len;
1759   size_t msgsize;
1760
1761   len = strlen (description);
1762   msgsize = sizeof(struct GNUNET_MESH_ConnectPeerByString) + len;
1763   GNUNET_assert (UINT16_MAX > msgsize);
1764   {
1765     char buffer[msgsize];
1766
1767     m = (struct GNUNET_MESH_ConnectPeerByString *) buffer;
1768     m->header.size = htons (msgsize);
1769     m->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_STRING);
1770     m->tunnel_id = htonl (tunnel->tid);
1771     memcpy(&m[1], description, len);
1772
1773     send_packet (tunnel->mesh, &m->header, tunnel);
1774   }
1775 }
1776
1777
1778 /**
1779  * Request that the given peer isn't added to this tunnel in calls to
1780  * connect_by_* calls, (due to misbehaviour, bad performance, ...).
1781  *
1782  * @param tunnel handle to existing tunnel.
1783  * @param peer peer identity of the peer which should be blacklisted
1784  *                  for the tunnel.
1785  */
1786 void
1787 GNUNET_MESH_peer_blacklist (struct GNUNET_MESH_Tunnel *tunnel,
1788                             const struct GNUNET_PeerIdentity *peer)
1789 {
1790   struct GNUNET_MESH_PeerControl msg;
1791
1792   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1793   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_BLACKLIST);
1794   msg.tunnel_id = htonl (tunnel->tid);
1795   msg.peer = *peer;
1796   send_packet (tunnel->mesh, &msg.header, tunnel);
1797
1798   return;
1799 }
1800
1801
1802 /**
1803  * Request that the given peer isn't blacklisted anymore from this tunnel,
1804  * and therefore can be added in future calls to connect_by_*.
1805  * The peer must have been previously blacklisted for this tunnel.
1806  *
1807  * @param tunnel handle to existing tunnel.
1808  * @param peer peer identity of the peer which shouldn't be blacklisted
1809  *                  for the tunnel anymore.
1810  */
1811 void
1812 GNUNET_MESH_peer_unblacklist (struct GNUNET_MESH_Tunnel *tunnel,
1813                               const struct GNUNET_PeerIdentity *peer)
1814 {
1815   struct GNUNET_MESH_PeerControl msg;
1816
1817   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1818   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_UNBLACKLIST);
1819   msg.tunnel_id = htonl (tunnel->tid);
1820   msg.peer = *peer;
1821   send_packet (tunnel->mesh, &msg.header, tunnel);
1822
1823   return;
1824 }
1825
1826
1827 /**
1828  * Ask the mesh to call "notify" once it is ready to transmit the
1829  * given number of bytes to the specified tunnel or target.
1830  * Only one call can be active at any time, to issue another request,
1831  * wait for the callback or cancel the current request.
1832  *
1833  * @param tunnel tunnel to use for transmission
1834  * @param cork is corking allowed for this transmission?
1835  * @param maxdelay how long can the message wait?
1836  * @param target destination for the message
1837  *               NULL for multicast to all tunnel targets
1838  * @param notify_size how many bytes of buffer space does notify want?
1839  * @param notify function to call when buffer space is available;
1840  *        will be called with NULL on timeout or if the overall queue
1841  *        for this peer is larger than queue_size and this is currently
1842  *        the message with the lowest priority
1843  * @param notify_cls closure for notify
1844  * @return non-NULL if the notify callback was queued,
1845  *         NULL if we can not even queue the request (insufficient
1846  *         memory); if NULL is returned, "notify" will NOT be called.
1847  */
1848 struct GNUNET_MESH_TransmitHandle *
1849 GNUNET_MESH_notify_transmit_ready (struct GNUNET_MESH_Tunnel *tunnel, int cork,
1850                                    struct GNUNET_TIME_Relative maxdelay,
1851                                    const struct GNUNET_PeerIdentity *target,
1852                                    size_t notify_size,
1853                                    GNUNET_CONNECTION_TransmitReadyNotify notify,
1854                                    void *notify_cls)
1855 {
1856   struct GNUNET_MESH_TransmitHandle *th;
1857   size_t overhead;
1858
1859   GNUNET_assert (NULL != tunnel);
1860   LOG (GNUNET_ERROR_TYPE_DEBUG, "mesh notify transmit ready called\n");
1861   if (NULL != target)
1862     LOG (GNUNET_ERROR_TYPE_DEBUG, "    target %s\n", GNUNET_i2s (target));
1863   else
1864     LOG (GNUNET_ERROR_TYPE_DEBUG, "    target multicast\n");
1865   GNUNET_assert (NULL != notify);
1866   GNUNET_assert (0 == tunnel->npackets);
1867   tunnel->npackets++;
1868   th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle));
1869   th->tunnel = tunnel;
1870   th->timeout = GNUNET_TIME_relative_to_absolute (maxdelay);
1871   th->target = GNUNET_PEER_intern (target);
1872   if (tunnel->tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
1873     overhead = sizeof (struct GNUNET_MESH_ToOrigin);
1874   else if (NULL == target)
1875     overhead = sizeof (struct GNUNET_MESH_Multicast);
1876   else
1877     overhead = sizeof (struct GNUNET_MESH_Unicast);
1878   th->size = notify_size + overhead;
1879   th->notify = notify;
1880   th->notify_cls = notify_cls;
1881   add_to_queue (tunnel->mesh, th);
1882   if (NULL != tunnel->mesh->th)
1883     return th;
1884   tunnel->mesh->th =
1885       GNUNET_CLIENT_notify_transmit_ready (tunnel->mesh->client, th->size,
1886                                            GNUNET_TIME_UNIT_FOREVER_REL,
1887                                            GNUNET_YES, &send_callback,
1888                                            tunnel->mesh);
1889   return th;
1890 }
1891
1892
1893 /**
1894  * Cancel the specified transmission-ready notification.
1895  *
1896  * @param th handle that was returned by "notify_transmit_ready".
1897  */
1898 void
1899 GNUNET_MESH_notify_transmit_ready_cancel (struct GNUNET_MESH_TransmitHandle *th)
1900 {
1901   struct GNUNET_MESH_Handle *mesh;
1902
1903   mesh = th->tunnel->mesh;
1904   if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1905     GNUNET_SCHEDULER_cancel (th->timeout_task);
1906   GNUNET_CONTAINER_DLL_remove (mesh->th_head, mesh->th_tail, th);
1907   GNUNET_free (th);
1908   if ((NULL == mesh->th_head) && (NULL != mesh->th))
1909   {
1910     /* queue empty, no point in asking for transmission */
1911     GNUNET_CLIENT_notify_transmit_ready_cancel (mesh->th);
1912     mesh->th = NULL;
1913   }
1914 }
1915
1916
1917 /**
1918  * Transition API for tunnel ctx management
1919  */
1920 void
1921 GNUNET_MESH_tunnel_set_data (struct GNUNET_MESH_Tunnel *tunnel, void *data)
1922 {
1923   tunnel->ctx = data;
1924 }
1925
1926 /**
1927  * Transition API for tunnel ctx management
1928  */
1929 void *
1930 GNUNET_MESH_tunnel_get_data (struct GNUNET_MESH_Tunnel *tunnel)
1931 {
1932   return tunnel->ctx;
1933 }
1934
1935