src/mesh/mesh_api_new.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_new.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 #ifdef __cplusplus
  32 extern "C"
  33 {
  34 #if 0                           /* keep Emacsens' auto-indent happy */
  35 }
  36 #endif
  37 #endif
  38
  39 #include "platform.h"
  40 #include "gnunet_common.h"
  41 #include "gnunet_client_lib.h"
  42 #include "gnunet_util_lib.h"
  43 #include "gnunet_peer_lib.h"
  44 #include "gnunet_mesh_service_new.h"
  45 #include "mesh.h"
  46 #include "mesh_protocol.h"
  47
  48 #define DEBUG GNUNET_YES
  49
  50 /******************************************************************************/
  51 /************************      DATA STRUCTURES     ****************************/
  52 /******************************************************************************/
  53
  54 /**
  55  * Transmission queue to the service
  56  */
  57 struct GNUNET_MESH_TransmitHandle
  58 {
  59
  60     /**
  61      * Double Linked list
  62      */
  63   struct GNUNET_MESH_TransmitHandle *next;
  64
  65     /**
  66      * Double Linked list
  67      */
  68   struct GNUNET_MESH_TransmitHandle *prev;
  69
  70     /**
  71      * Tunnel this message is sent over (may be NULL for control messages).
  72      */
  73   struct GNUNET_MESH_Tunnel *tunnel;
  74
  75     /**
  76      * Callback to obtain the message to transmit, or NULL if we
  77      * got the message in 'data'.  Notice that messages built
  78      * by 'notify' need to be encapsulated with information about
  79      * the 'target'.
  80      */
  81   GNUNET_CONNECTION_TransmitReadyNotify notify;
  82
  83     /**
  84      * Closure for 'notify'
  85      */
  86   void *notify_cls;
  87
  88     /**
  89      * How long is this message valid.  Once the timeout has been
  90      * reached, the message must no longer be sent.  If this
  91      * is a message with a 'notify' callback set, the 'notify'
  92      * function should be called with 'buf' NULL and size 0.
  93      */
  94   struct GNUNET_TIME_Absolute timeout;
  95
  96     /**
  97      * Task triggering a timeout, can be NO_TASK if the timeout is FOREVER.
  98      */
  99   GNUNET_SCHEDULER_TaskIdentifier timeout_task;
 100
 101     /**
 102      * Priority of the message.  The queue is sorted by priority,
 103      * control messages have the maximum priority (UINT32_MAX).
 104      */
 105   uint32_t priority;
 106
 107     /**
 108      * Target of the message, 0 for broadcast.  This field
 109      * is only valid if 'notify' is non-NULL.
 110      */
 111   GNUNET_PEER_Id target;
 112
 113     /**
 114      * Size of 'data' -- or the desired size of 'notify' if 'data' is NULL.
 115      */
 116   size_t size;
 117 };
 118
 119
 120 /**
 121  * Opaque handle to the service.
 122  */
 123 struct GNUNET_MESH_Handle
 124 {
 125
 126     /**
 127      * Handle to the server connection, to send messages later
 128      */
 129   struct GNUNET_CLIENT_Connection *client;
 130
 131     /**
 132      * Set of handlers used for processing incoming messages in the tunnels
 133      */
 134   const struct GNUNET_MESH_MessageHandler *message_handlers;
 135
 136     /**
 137      * Set of applications that should be claimed to be offered at this node.
 138      * Note that this is just informative, the appropiate handlers must be
 139      * registered independently and the mapping is up to the developer of the
 140      * client application.
 141      */
 142   const GNUNET_MESH_ApplicationType *applications;
 143
 144     /**
 145      * Double linked list of the tunnels this client is connected to.
 146      */
 147   struct GNUNET_MESH_Tunnel *tunnels_head;
 148   struct GNUNET_MESH_Tunnel *tunnels_tail;
 149
 150     /**
 151      * Callback for inbound tunnel creation
 152      */
 153   GNUNET_MESH_InboundTunnelNotificationHandler *new_tunnel;
 154
 155     /**
 156      * Callback for inbound tunnel disconnection
 157      */
 158   GNUNET_MESH_TunnelEndHandler *cleaner;
 159
 160     /**
 161      * Handle to cancel pending transmissions in case of disconnection
 162      */
 163   struct GNUNET_CLIENT_TransmitHandle *th;
 164
 165     /**
 166      * Closure for all the handlers given by the client
 167      */
 168   void *cls;
 169
 170     /**
 171      * Messages to send to the service
 172      */
 173   struct GNUNET_MESH_TransmitHandle *th_head;
 174   struct GNUNET_MESH_TransmitHandle *th_tail;
 175
 176     /**
 177      * tid of the next tunnel to create (to avoid reusing IDs often)
 178      */
 179   MESH_TunnelNumber next_tid;
 180   unsigned int n_handlers;
 181   unsigned int n_applications;
 182   unsigned int max_queue_size;
 183
 184     /**
 185      * Have we started the task to receive messages from the service
 186      * yet? We do this after we send the 'MESH_LOCAL_CONNECT' message.
 187      */
 188   int in_receive;
 189
 190     /**
 191      * Number of packets queued
 192      */
 193   unsigned int npackets;
 194
 195   /**
 196    * Configuration given by the client, in case of reconnection
 197    */
 198   const struct GNUNET_CONFIGURATION_Handle *cfg;
 199
 200   /**
 201    * Time to the next reconnect in case one reconnect fails
 202    */
 203   struct GNUNET_TIME_Relative reconnect_time;
 204 };
 205
 206
 207 /**
 208  * Description of a peer
 209  */
 210 struct GNUNET_MESH_Peer
 211 {
 212     /**
 213      * ID of the peer in short form
 214      */
 215   GNUNET_PEER_Id id;
 216
 217   /**
 218    * Tunnel this peer belongs to
 219    */
 220   struct GNUNET_MESH_Tunnel *t;
 221
 222   /**
 223    * Flag indicating whether service has informed about its connection
 224    */
 225   int connected;
 226
 227 };
 228
 229
 230 /**
 231  * Opaque handle to a tunnel.
 232  */
 233 struct GNUNET_MESH_Tunnel
 234 {
 235
 236     /**
 237      * DLL
 238      */
 239   struct GNUNET_MESH_Tunnel *next;
 240   struct GNUNET_MESH_Tunnel *prev;
 241
 242     /**
 243      * Callback to execute when peers connect to the tunnel
 244      */
 245   GNUNET_MESH_PeerConnectHandler connect_handler;
 246
 247     /**
 248      * Callback to execute when peers disconnect from the tunnel
 249      */
 250   GNUNET_MESH_PeerDisconnectHandler disconnect_handler;
 251
 252     /**
 253      * Closure for the connect/disconnect handlers
 254      */
 255   void *cls;
 256
 257     /**
 258      * Handle to the mesh this tunnel belongs to
 259      */
 260   struct GNUNET_MESH_Handle *mesh;
 261
 262     /**
 263      * Local ID of the tunnel
 264      */
 265   MESH_TunnelNumber tid;
 266
 267     /**
 268      * Owner of the tunnel
 269      */
 270   GNUNET_PEER_Id owner;
 271
 272     /**
 273      * All peers added to the tunnel
 274      */
 275   struct GNUNET_MESH_Peer **peers;
 276
 277   /**
 278    * List of application types that have been requested for this tunnel
 279    */
 280   GNUNET_MESH_ApplicationType *apps;
 281
 282   /**
 283    * Any data the caller wants to put in here
 284    */
 285   void *ctx;
 286
 287   /**
 288      * Number of peers added to the tunnel
 289      */
 290   unsigned int npeers;
 291
 292     /**
 293      * Number of packets queued in this tunnel
 294      */
 295   unsigned int npackets;
 296
 297     /**
 298      * Number of applications requested this tunnel
 299      */
 300   unsigned int napps;
 301
 302 };
 303
 304
 305 /******************************************************************************/
 306 /***********************     AUXILIARY FUNCTIONS      *************************/
 307 /******************************************************************************/
 308
 309 /**
 310  * Get the tunnel handler for the tunnel specified by id from the given handle
 311  * @param h Mesh handle
 312  * @param tid ID of the wanted tunnel
 313  * @return handle to the required tunnel or NULL if not found
 314  */
 315 static struct GNUNET_MESH_Tunnel *
 316 retrieve_tunnel (struct GNUNET_MESH_Handle *h, MESH_TunnelNumber tid)
 317 {
 318   struct GNUNET_MESH_Tunnel *t;
 319
 320   t = h->tunnels_head;
 321   while (t != NULL)
 322   {
 323     if (t->tid == tid)
 324       return t;
 325     t = t->next;
 326   }
 327   return NULL;
 328 }
 329
 330
 331 /**
 332  * Create a new tunnel and insert it in the tunnel list of the mesh handle
 333  * @param h Mesh handle
 334  * @param tid desired tid of the tunnel, 0 to assign one automatically
 335  * @return handle to the created tunnel
 336  */
 337 static struct GNUNET_MESH_Tunnel *
 338 create_tunnel (struct GNUNET_MESH_Handle *h, MESH_TunnelNumber tid)
 339 {
 340   struct GNUNET_MESH_Tunnel *t;
 341
 342   t = GNUNET_malloc (sizeof (struct GNUNET_MESH_Tunnel));
 343   GNUNET_CONTAINER_DLL_insert (h->tunnels_head, h->tunnels_tail, t);
 344   t->mesh = h;
 345   if (0 == tid)
 346   {
 347     t->tid = h->next_tid;
 348     while (NULL != retrieve_tunnel (h, h->next_tid))
 349     {
 350       h->next_tid++;
 351       h->next_tid &= ~GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
 352       h->next_tid |= GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
 353     }
 354   }
 355   else
 356   {
 357     t->tid = tid;
 358   }
 359   return t;
 360 }
 361
 362
 363 /**
 364  * Destroy the specified tunnel.
 365  * - Destroys all peers, calling the disconnect callback on each if needed
 366  * - Cancels all outgoing traffic for that tunnel, calling respective notifys
 367  * - Calls cleaner if tunnel was inbound
 368  * - Frees all memory used
 369  * @param tid ID of the wanted tunnel
 370  * @return handle to the required tunnel or NULL if not found
 371  */
 372 static void
 373 destroy_tunnel (struct GNUNET_MESH_Tunnel *t)
 374 {
 375   struct GNUNET_MESH_Handle *h;
 376   struct GNUNET_PeerIdentity pi;
 377   struct GNUNET_MESH_TransmitHandle *th;
 378   struct GNUNET_MESH_TransmitHandle *aux;
 379   unsigned int i;
 380
 381   if (NULL == t)
 382   {
 383     GNUNET_break (0);
 384     return;
 385   }
 386   h = t->mesh;
 387   th = h->th_head;
 388   while (NULL != th)
 389   {
 390     if (th->tunnel == t)
 391     {
 392       aux = th->next;
 393       GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th);
 394       if (NULL == h->th_head && NULL != h->th)
 395       {
 396         GNUNET_CLIENT_notify_transmit_ready_cancel (h->th);
 397         h->th = NULL;
 398       }
 399       if (NULL != th->notify)
 400         th->notify (th->notify_cls, 0, NULL);
 401       if (GNUNET_SCHEDULER_NO_TASK != th->timeout_task)
 402         GNUNET_SCHEDULER_cancel (th->timeout_task);
 403       GNUNET_free (th);
 404       th = aux;
 405     }
 406     else
 407     {
 408       th = th->next;
 409     }
 410   }
 411   GNUNET_CONTAINER_DLL_remove (h->tunnels_head, h->tunnels_tail, t);
 412   for (i = 0; i < t->npeers; i++)
 413   {
 414     if (NULL != t->disconnect_handler && t->peers[i]->connected)
 415     {
 416       GNUNET_PEER_resolve (t->peers[i]->id, &pi);
 417       t->disconnect_handler (t->cls, &pi);
 418     }
 419     GNUNET_PEER_change_rc (t->peers[i]->id, -1);
 420     GNUNET_free (t->peers[i]);
 421   }
 422   if (t->npeers > 0)
 423     GNUNET_free (t->peers);
 424   if (NULL != h->cleaner && 0 != t->owner)
 425     h->cleaner (h->cls, t, t->ctx);
 426   if (0 != t->owner)
 427     GNUNET_PEER_change_rc (t->owner, -1);
 428   if (0 != t->napps && t->apps)
 429     GNUNET_free (t->apps);
 430   GNUNET_free (t);
 431   return;
 432 }
 433
 434
 435 /**
 436  * Get the peer descriptor for the peer with id from the given tunnel
 437  * @param t Tunnel handle
 438  * @param id Short form ID of the wanted peer
 439  * @return handle to the requested peer or NULL if not found
 440  */
 441 static struct GNUNET_MESH_Peer *
 442 retrieve_peer (struct GNUNET_MESH_Tunnel *t, GNUNET_PEER_Id id)
 443 {
 444   unsigned int i;
 445
 446   for (i = 0; i < t->npeers; i++)
 447     if (t->peers[i]->id == id)
 448       return t->peers[i];
 449   return NULL;
 450 }
 451
 452
 453 /**
 454  * Add a peer into a tunnel
 455  * @param t Tunnel handle
 456  * @param pi Full ID of the new peer
 457  * @return handle to the newly created peer
 458  */
 459 static struct GNUNET_MESH_Peer *
 460 add_peer_to_tunnel (struct GNUNET_MESH_Tunnel *t,
 461                     const struct GNUNET_PeerIdentity *pi)
 462 {
 463   struct GNUNET_MESH_Peer *p;
 464   GNUNET_PEER_Id id;
 465
 466   if (0 != t->owner)
 467   {
 468     GNUNET_break (0);
 469     return NULL;
 470   }
 471   id = GNUNET_PEER_intern (pi);
 472
 473   p = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer));
 474   p->id = id;
 475   p->t = t;
 476   GNUNET_array_append (t->peers, t->npeers, p);
 477   return p;
 478 }
 479
 480
 481 /**
 482  * Remove a peer from a tunnel
 483  * @param p Peer handle
 484  */
 485 static void
 486 remove_peer_from_tunnel (struct GNUNET_MESH_Peer *p)
 487 {
 488   unsigned int i;
 489
 490   for (i = 0; i < p->t->npeers; i++)
 491   {
 492     if (p->t->peers[i] == p)
 493       break;
 494   }
 495   if (i == p->t->npeers)
 496   {
 497     GNUNET_break (0);
 498     return;
 499   }
 500   p->t->peers[i] = p->t->peers[p->t->npeers - 1];
 501   GNUNET_array_grow (p->t->peers, p->t->npeers, p->t->npeers - 1);
 502 }
 503
 504
 505 /**
 506  * Notify client that the transmission has timed out
 507  * @param cls closure
 508  * @param tc task context
 509  */
 510 static void
 511 timeout_transmission (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 512 {
 513   struct GNUNET_MESH_TransmitHandle *th = cls;
 514   struct GNUNET_MESH_Handle *mesh;
 515
 516   mesh = th->tunnel->mesh;
 517   GNUNET_CONTAINER_DLL_remove (mesh->th_head, mesh->th_tail, th);
 518   if (th->notify != NULL)
 519     th->notify (th->notify_cls, 0, NULL);
 520   GNUNET_free (th);
 521   if ((NULL == mesh->th_head) && (NULL != mesh->th))
 522   {
 523     /* queue empty, no point in asking for transmission */
 524     GNUNET_CLIENT_notify_transmit_ready_cancel (mesh->th);
 525     mesh->th = NULL;
 526   }
 527 }
 528
 529
 530 /**
 531  * Add a transmit handle to the transmission queue by priority and set the
 532  * timeout if needed.
 533  *
 534  * @param h mesh handle with the queue head and tail
 535  * @param th handle to the packet to be transmitted
 536  */
 537 static void
 538 add_to_queue (struct GNUNET_MESH_Handle *h,
 539               struct GNUNET_MESH_TransmitHandle *th)
 540 {
 541   struct GNUNET_MESH_TransmitHandle *p;
 542
 543   p = h->th_head;
 544   while ((NULL != p) && (th->priority <= p->priority))
 545     p = p->next;
 546   if (NULL == p)
 547     p = h->th_tail;
 548   else
 549     p = p->prev;
 550   GNUNET_CONTAINER_DLL_insert_after (h->th_head, h->th_tail, p, th);
 551   if (GNUNET_TIME_UNIT_FOREVER_ABS.abs_value == th->timeout.abs_value)
 552     return;
 553   th->timeout_task =
 554       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_absolute_get_remaining
 555                                     (th->timeout), &timeout_transmission, th);
 556 }
 557
 558
 559 /**
 560  * Auxiliary function to send an already constructed packet to the service.
 561  * Takes care of creating a new queue element, copying the message and
 562  * calling the tmt_rdy function if necessary.
 563  * @param h mesh handle
 564  * @param msg message to transmit
 565  */
 566 static void
 567 send_packet (struct GNUNET_MESH_Handle *h,
 568              const struct GNUNET_MessageHeader *msg);
 569
 570
 571 /**
 572  * Reconnect callback: tries to reconnect again after a failer previous
 573  * reconnecttion
 574  * @param cls closure (mesh handle)
 575  * @param tc task context
 576  */
 577 static void
 578 reconnect_cbk (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
 579
 580
 581 /**
 582  * Reconnect to the service, retransmit all infomation to try to restore the
 583  * original state.
 584  *
 585  * @param h handle to the mesh
 586  *
 587  * @return GNUNET_YES in case of sucess, GNUNET_NO otherwise (service down...)
 588  */
 589 static int
 590 reconnect (struct GNUNET_MESH_Handle *h)
 591 {
 592   struct GNUNET_MESH_Tunnel *t;
 593   unsigned int i;
 594
 595 #if DEBUG
 596   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: *****************************\n");
 597   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: *******   RECONNECT   *******\n");
 598   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: *****************************\n");
 599 #endif
 600   h->in_receive = GNUNET_NO;
 601   /* disconnect */
 602   if (NULL != h->th)
 603   {
 604     GNUNET_CLIENT_notify_transmit_ready_cancel (h->th);
 605   }
 606   if (NULL != h->client)
 607   {
 608     GNUNET_CLIENT_disconnect (h->client, GNUNET_NO);
 609   }
 610
 611   /* connect again */
 612   h->client = GNUNET_CLIENT_connect ("mesh", h->cfg);
 613   if (h->client == NULL)
 614   {
 615     GNUNET_SCHEDULER_add_delayed (h->reconnect_time, &reconnect_cbk, h);
 616     h->reconnect_time =
 617         GNUNET_TIME_relative_min (GNUNET_TIME_UNIT_HOURS,
 618                                   GNUNET_TIME_relative_multiply
 619                                   (h->reconnect_time, 2));
 620     GNUNET_break (0);
 621     return GNUNET_NO;
 622   }
 623   else
 624   {
 625     h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS;
 626   }
 627   /* Rebuild all tunnels */
 628   for (t = h->tunnels_head; NULL != t; t = t->next)
 629   {
 630     struct GNUNET_MESH_TunnelMessage tmsg;
 631     struct GNUNET_MESH_PeerControl pmsg;
 632
 633     tmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
 634     tmsg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
 635     tmsg.tunnel_id = htonl (t->tid);
 636     send_packet (h, &tmsg.header);
 637
 638     pmsg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
 639     pmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
 640     pmsg.tunnel_id = htonl (t->tid);
 641
 642     /* Reconnect all peers */
 643     for (i = 0; i < t->npeers; i++)
 644     {
 645       GNUNET_PEER_resolve (t->peers[i]->id, &pmsg.peer);
 646       if (NULL != t->disconnect_handler && t->peers[i]->connected)
 647         t->disconnect_handler (t->cls, &pmsg.peer);
 648       /* If the tunnel was "by type", dont connect individual peers */
 649       if (0 == t->napps)
 650         send_packet (t->mesh, &pmsg.header);
 651     }
 652     /* Reconnect all types, if any  */
 653     for (i = 0; i < t->napps; i++)
 654     {
 655       struct GNUNET_MESH_ConnectPeerByType msg;
 656
 657       msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
 658       msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE);
 659       msg.tunnel_id = htonl (t->tid);
 660       msg.type = htonl (t->apps[i]);
 661       send_packet (t->mesh, &msg.header);
 662     }
 663   }
 664   return GNUNET_YES;
 665 }
 666
 667 /**
 668  * Reconnect callback: tries to reconnect again after a failer previous
 669  * reconnecttion
 670  * @param cls closure (mesh handle)
 671  * @param tc task context
 672  */
 673 static void
 674 reconnect_cbk (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 675 {
 676   struct GNUNET_MESH_Handle *h = cls;
 677
 678   if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
 679     return;
 680   reconnect (h);
 681 }
 682
 683
 684 /******************************************************************************/
 685 /***********************      RECEIVE HANDLERS     ****************************/
 686 /******************************************************************************/
 687
 688 /**
 689  * Process the new tunnel notification and add it to the tunnels in the handle
 690  *
 691  * @param h     The mesh handle
 692  * @param msg   A message with the details of the new incoming tunnel
 693  */
 694 static void
 695 process_tunnel_created (struct GNUNET_MESH_Handle *h,
 696                         const struct GNUNET_MESH_TunnelNotification *msg)
 697 {
 698   struct GNUNET_MESH_Tunnel *t;
 699   struct GNUNET_TRANSPORT_ATS_Information atsi;
 700   MESH_TunnelNumber tid;
 701
 702   tid = ntohl (msg->tunnel_id);
 703   if (tid <= GNUNET_MESH_LOCAL_TUNNEL_ID_CLI)
 704   {
 705     GNUNET_break (0);
 706     return;
 707   }
 708   t = create_tunnel (h, tid);
 709   t->owner = GNUNET_PEER_intern (&msg->peer);
 710   t->npeers = 1;
 711   t->peers = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer *));
 712   t->peers[0] = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer));
 713   t->peers[0]->t = t;
 714   t->peers[0]->connected = 1;
 715   t->peers[0]->id = t->owner;
 716   t->mesh = h;
 717   t->tid = tid;
 718   if (NULL != h->new_tunnel)
 719   {
 720     atsi.type = 0;
 721     atsi.value = 0;
 722     t->ctx = h->new_tunnel (h->cls, t, &msg->peer, &atsi);
 723   }
 724   GNUNET_CONTAINER_DLL_insert (h->tunnels_head, h->tunnels_tail, t);
 725   return;
 726 }
 727
 728
 729 /**
 730  * Process the tunnel destroy notification and free associated resources
 731  *
 732  * @param h     The mesh handle
 733  * @param msg   A message with the details of the tunnel being destroyed
 734  */
 735 static void
 736 process_tunnel_destroy (struct GNUNET_MESH_Handle *h,
 737                         const struct GNUNET_MESH_TunnelMessage *msg)
 738 {
 739   struct GNUNET_MESH_Tunnel *t;
 740   MESH_TunnelNumber tid;
 741
 742   tid = ntohl (msg->tunnel_id);
 743   t = retrieve_tunnel (h, tid);
 744
 745   if (NULL == t)
 746   {
 747     GNUNET_break (0);
 748     return;
 749   }
 750   if (0 == t->owner)
 751   {
 752     GNUNET_break (0);
 753   }
 754
 755   destroy_tunnel (t);
 756   return;
 757 }
 758
 759
 760 /**
 761  * Process the new peer event and notify the upper level of it
 762  *
 763  * @param h     The mesh handle
 764  * @param msg   A message with the details of the peer event
 765  */
 766 static void
 767 process_peer_event (struct GNUNET_MESH_Handle *h,
 768                     const struct GNUNET_MESH_PeerControl *msg)
 769 {
 770   struct GNUNET_MESH_Tunnel *t;
 771   struct GNUNET_MESH_Peer *p;
 772   struct GNUNET_TRANSPORT_ATS_Information atsi;
 773   GNUNET_PEER_Id id;
 774   uint16_t size;
 775
 776   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: processig peer event\n");
 777   size = ntohs (msg->header.size);
 778   if (size != sizeof (struct GNUNET_MESH_PeerControl))
 779   {
 780     GNUNET_break (0);
 781     return;
 782   }
 783   t = retrieve_tunnel (h, ntohl (msg->tunnel_id));
 784   if (NULL == t)
 785   {
 786     GNUNET_break (0);
 787     return;
 788   }
 789   id = GNUNET_PEER_search (&msg->peer);
 790   if ((p = retrieve_peer (t, id)) == NULL)
 791     p = add_peer_to_tunnel (t, &msg->peer);
 792   if (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD == ntohs (msg->header.type))
 793   {
 794     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: adding peer\n");
 795     if (NULL != t->connect_handler)
 796     {
 797       atsi.type = 0;
 798       atsi.value = 0;
 799       t->connect_handler (t->cls, &msg->peer, &atsi);
 800     }
 801     p->connected = 1;
 802   }
 803   else
 804   {
 805     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: removing peer\n");
 806     if (NULL != t->disconnect_handler && p->connected)
 807     {
 808       t->disconnect_handler (t->cls, &msg->peer);
 809     }
 810     remove_peer_from_tunnel (p);
 811     GNUNET_free (p);
 812   }
 813   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: processing peer event END\n");
 814 }
 815
 816
 817 /**
 818  * Process the incoming data packets
 819  *
 820  * @param h         The mesh handle
 821  * @param message   A message encapsulating the data
 822  */
 823 static void
 824 process_incoming_data (struct GNUNET_MESH_Handle *h,
 825                        const struct GNUNET_MessageHeader *message)
 826 {
 827   const struct GNUNET_MessageHeader *payload;
 828   const struct GNUNET_MESH_MessageHandler *handler;
 829   const struct GNUNET_PeerIdentity *peer;
 830   struct GNUNET_MESH_Unicast *ucast;
 831   struct GNUNET_MESH_Multicast *mcast;
 832   struct GNUNET_MESH_ToOrigin *to_orig;
 833   struct GNUNET_MESH_Tunnel *t;
 834   unsigned int i;
 835   uint16_t type;
 836
 837
 838   type = ntohs (message->type);
 839   switch (type)
 840   {
 841   case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
 842     ucast = (struct GNUNET_MESH_Unicast *) message;
 843     t = retrieve_tunnel (h, ntohl (ucast->tid));
 844     payload = (struct GNUNET_MessageHeader *) &ucast[1];
 845     peer = &ucast->oid;
 846     break;
 847   case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
 848     mcast = (struct GNUNET_MESH_Multicast *) message;
 849     t = retrieve_tunnel (h, ntohl (mcast->tid));
 850     payload = (struct GNUNET_MessageHeader *) &mcast[1];
 851     peer = &mcast->oid;
 852     break;
 853   case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
 854     to_orig = (struct GNUNET_MESH_ToOrigin *) message;
 855     t = retrieve_tunnel (h, ntohl (to_orig->tid));
 856     payload = (struct GNUNET_MessageHeader *) &to_orig[1];
 857     peer = &to_orig->sender;
 858     break;
 859   default:
 860     GNUNET_break (0);
 861     return;
 862   }
 863   if (NULL == t)
 864   {
 865     GNUNET_break (0);
 866     return;
 867   }
 868   type = ntohs (payload->type);
 869   for (i = 0; i < h->n_handlers; i++)
 870   {
 871     handler = &h->message_handlers[i];
 872     if (handler->type == type)
 873     {
 874       struct GNUNET_TRANSPORT_ATS_Information atsi;
 875
 876       atsi.type = 0;
 877       atsi.value = 0;
 878       if (GNUNET_OK !=
 879           handler->callback (h->cls, t, &t->ctx, peer, payload, &atsi))
 880       {
 881         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 882                     "MESH: callback caused disconnection\n");
 883         GNUNET_MESH_disconnect (h);
 884         break;
 885       }
 886 #if DEBUG
 887       else
 888       {
 889         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 890                     "MESH: callback completed successfully\n");
 891
 892       }
 893 #endif
 894     }
 895   }
 896 }
 897
 898
 899 /**
 900  * Function to process all messages received from the service
 901  *
 902  * @param cls closure
 903  * @param msg message received, NULL on timeout or fatal error
 904  */
 905 static void
 906 msg_received (void *cls, const struct GNUNET_MessageHeader *msg)
 907 {
 908   struct GNUNET_MESH_Handle *h = cls;
 909
 910   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 911               "mesh: received a message type %hu from MESH\n",
 912               ntohs (msg->type));
 913   if (msg == NULL)
 914   {
 915     reconnect (h);
 916     return;
 917   }
 918   switch (ntohs (msg->type))
 919   {
 920     /* Notify of a new incoming tunnel */
 921   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE:
 922     process_tunnel_created (h, (struct GNUNET_MESH_TunnelNotification *) msg);
 923     break;
 924     /* Notify of a tunnel disconnection */
 925   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY:
 926     process_tunnel_destroy (h, (struct GNUNET_MESH_TunnelMessage *) msg);
 927     break;
 928     /* Notify of a new peer or a peer disconnect in the tunnel */
 929   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD:
 930   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL:
 931     process_peer_event (h, (struct GNUNET_MESH_PeerControl *) msg);
 932     break;
 933     /* Notify of a new data packet in the tunnel */
 934   case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
 935   case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
 936   case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
 937     process_incoming_data (h, msg);
 938     break;
 939     /* We shouldn't get any other packages, log and ignore */
 940   default:
 941     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 942                 "MESH: unsolicited message form service (type %d)\n",
 943                 ntohs (msg->type));
 944   }
 945   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: message processed\n");
 946   GNUNET_CLIENT_receive (h->client, &msg_received, h,
 947                          GNUNET_TIME_UNIT_FOREVER_REL);
 948 }
 949
 950
 951 /******************************************************************************/
 952 /************************       SEND FUNCTIONS     ****************************/
 953 /******************************************************************************/
 954
 955 /**
 956  * Function called to send a message to the service.
 957  * "buf" will be NULL and "size" zero if the socket was closed for writing in
 958  * the meantime.
 959  *
 960  * @param cls closure, the mesh handle
 961  * @param size number of bytes available in buf
 962  * @param buf where the callee should write the connect message
 963  * @return number of bytes written to buf
 964  */
 965 static size_t
 966 send_callback (void *cls, size_t size, void *buf)
 967 {
 968   struct GNUNET_MESH_Handle *h = cls;
 969   struct GNUNET_MESH_TransmitHandle *th;
 970   char *cbuf = buf;
 971   size_t tsize;
 972   size_t psize;
 973
 974   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Send packet() Buffer %u\n", size);
 975   h->th = NULL;
 976   if ((0 == size) || (NULL == buf))
 977   {
 978     reconnect (h);
 979     return 0;
 980   }
 981   tsize = 0;
 982   while ((NULL != (th = h->th_head)) && (size >= th->size))
 983   {
 984 #if DEBUG
 985     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:     type: %u\n",
 986                 ntohs (((struct GNUNET_MessageHeader *) &th[1])->type));
 987     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:     size: %u\n",
 988                 ntohs (((struct GNUNET_MessageHeader *) &th[1])->size));
 989 #endif
 990     if (NULL != th->notify)
 991     {
 992       if (th->target == 0)
 993       {
 994         /* multicast */
 995         struct GNUNET_MESH_Multicast mc;
 996
 997         GNUNET_assert (size >= sizeof (mc) + th->size);
 998         psize =
 999             th->notify (th->notify_cls, size - sizeof (mc), &cbuf[sizeof (mc)]);
1000         if (psize > 0)
1001         {
1002           mc.header.size = htons (sizeof (mc) + th->size);
1003           mc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_MULTICAST);
1004           mc.tid = htonl (th->tunnel->tid);
1005           memset (&mc.oid, 0, sizeof (struct GNUNET_PeerIdentity));
1006           memcpy (cbuf, &mc, sizeof (mc));
1007           psize = th->size + sizeof (mc);
1008         }
1009       }
1010       else
1011       {
1012         /* unicast */
1013         struct GNUNET_MESH_Unicast uc;
1014
1015         GNUNET_assert (size >= sizeof (uc) + th->size);
1016         psize =
1017             th->notify (th->notify_cls, size - sizeof (uc), &cbuf[sizeof (uc)]);
1018         if (psize > 0)
1019         {
1020           uc.header.size = htons (sizeof (uc) + th->size);
1021           uc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_UNICAST);
1022           uc.tid = htonl (th->tunnel->tid);
1023           memset (&uc.oid, 0, sizeof (struct GNUNET_PeerIdentity));
1024           GNUNET_PEER_resolve (th->target, &uc.destination);
1025           memcpy (cbuf, &uc, sizeof (uc));
1026           psize = th->size + sizeof (uc);
1027         }
1028       }
1029     }
1030     else
1031     {
1032       memcpy (cbuf, &th[1], th->size);
1033       psize = th->size;
1034     }
1035     if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1036       GNUNET_SCHEDULER_cancel (th->timeout_task);
1037     GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th);
1038     GNUNET_free (th);
1039     cbuf += psize;
1040     size -= psize;
1041     tsize += psize;
1042   }
1043   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:   total size: %u\n", tsize);
1044   if (NULL != (th = h->th_head))
1045   {
1046     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:   next size: %u\n", th->size);
1047     h->th =
1048         GNUNET_CLIENT_notify_transmit_ready (h->client, th->size,
1049                                              GNUNET_TIME_UNIT_FOREVER_REL,
1050                                              GNUNET_YES, &send_callback, h);
1051   }
1052   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Send packet() END\n");
1053   if (GNUNET_NO == h->in_receive)
1054   {
1055     h->in_receive = GNUNET_YES;
1056     GNUNET_CLIENT_receive (h->client, &msg_received, h,
1057                            GNUNET_TIME_UNIT_FOREVER_REL);
1058   }
1059   return tsize;
1060 }
1061
1062
1063 /**
1064  * Auxiliary function to send an already constructed packet to the service.
1065  * Takes care of creating a new queue element, copying the message and
1066  * calling the tmt_rdy function if necessary.
1067  * @param h mesh handle
1068  * @param msg message to transmit
1069  */
1070 static void
1071 send_packet (struct GNUNET_MESH_Handle *h,
1072              const struct GNUNET_MessageHeader *msg)
1073 {
1074   struct GNUNET_MESH_TransmitHandle *th;
1075   size_t msize;
1076
1077   msize = ntohs (msg->size);
1078   th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle) + msize);
1079   th->priority = UINT32_MAX;
1080   th->timeout = GNUNET_TIME_UNIT_FOREVER_ABS;
1081   th->size = msize;
1082   memcpy (&th[1], msg, msize);
1083   add_to_queue (h, th);
1084   if (NULL != h->th)
1085     return;
1086   h->th =
1087       GNUNET_CLIENT_notify_transmit_ready (h->client, msize,
1088                                            GNUNET_TIME_UNIT_FOREVER_REL,
1089                                            GNUNET_YES, &send_callback, h);
1090 }
1091
1092
1093 /******************************************************************************/
1094 /**********************      API CALL DEFINITIONS     *************************/
1095 /******************************************************************************/
1096
1097 /**
1098  * Connect to the mesh service.
1099  *
1100  * @param cfg configuration to use
1101  * @param queue_size size of the data message queue, shared among all tunnels
1102  *                   (each tunnel is guaranteed to accept at least one message,
1103  *                    no matter what is the status of other tunnels)
1104  * @param cls closure for the various callbacks that follow
1105  *            (including handlers in the handlers array)
1106  * @param new_tunnel function called when an *inbound* tunnel is created
1107  * @param cleaner function called when an *inbound* tunnel is destroyed
1108  * @param handlers callbacks for messages we care about, NULL-terminated
1109  *                note that the mesh is allowed to drop notifications about
1110  *                inbound messages if the client does not process them fast
1111  *                enough (for this notification type, a bounded queue is used)
1112  * @param stypes list of the applications that this client claims to provide
1113  * @return handle to the mesh service NULL on error
1114  *         (in this case, init is never called)
1115  */
1116 struct GNUNET_MESH_Handle *
1117 GNUNET_MESH_connect (const struct GNUNET_CONFIGURATION_Handle *cfg,
1118                      unsigned int queue_size, void *cls,
1119                      GNUNET_MESH_InboundTunnelNotificationHandler new_tunnel,
1120                      GNUNET_MESH_TunnelEndHandler cleaner,
1121                      const struct GNUNET_MESH_MessageHandler *handlers,
1122                      const GNUNET_MESH_ApplicationType *stypes)
1123 {
1124   struct GNUNET_MESH_Handle *h;
1125   struct GNUNET_MESH_ClientConnect *msg;
1126   GNUNET_MESH_ApplicationType *apps;
1127   uint16_t napps;
1128   uint16_t *types;
1129   uint16_t ntypes;
1130   size_t size;
1131
1132   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: GNUNET_MESH_connect()\n");
1133   h = GNUNET_malloc (sizeof (struct GNUNET_MESH_Handle));
1134   h->cfg = cfg;
1135   h->max_queue_size = queue_size;
1136   h->new_tunnel = new_tunnel;
1137   h->cleaner = cleaner;
1138   h->client = GNUNET_CLIENT_connect ("mesh", cfg);
1139   if (h->client == NULL)
1140   {
1141     GNUNET_break (0);
1142     GNUNET_free (h);
1143     return NULL;
1144   }
1145   h->cls = cls;
1146   /* FIXME memdup? */
1147   h->applications = stypes;
1148   h->message_handlers = handlers;
1149   h->next_tid = GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
1150   h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS;
1151
1152   /* count handlers and apps, calculate size */
1153   for (h->n_applications = 0; stypes[h->n_applications]; h->n_applications++) ;
1154   for (h->n_handlers = 0; handlers[h->n_handlers].type; h->n_handlers++) ;
1155   size = sizeof (struct GNUNET_MESH_ClientConnect);
1156   size += h->n_applications * sizeof (GNUNET_MESH_ApplicationType);
1157   size += h->n_handlers * sizeof (uint16_t);
1158
1159   {
1160     char buf[size];
1161
1162     /* build connection packet */
1163     msg = (struct GNUNET_MESH_ClientConnect *) buf;
1164     msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT);
1165     msg->header.size = htons (size);
1166     apps = (GNUNET_MESH_ApplicationType *) &msg[1];
1167     for (napps = 0; napps < h->n_applications; napps++)
1168     {
1169       apps[napps] = htonl (h->applications[napps]);
1170       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:  app %u\n",
1171                   h->applications[napps]);
1172     }
1173     types = (uint16_t *) & apps[napps];
1174     for (ntypes = 0; ntypes < h->n_handlers; ntypes++)
1175       types[ntypes] = htons (h->message_handlers[ntypes].type);
1176     msg->applications = htons (napps);
1177     msg->types = htons (ntypes);
1178 #if DEBUG
1179     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1180                 "mesh: Sending %lu bytes long message %d types and %d apps\n",
1181                 ntohs (msg->header.size), ntypes, napps);
1182 #endif
1183     send_packet (h, &msg->header);
1184   }
1185   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: GNUNET_MESH_connect() END\n");
1186   return h;
1187 }
1188
1189
1190 /**
1191  * Disconnect from the mesh service.
1192  *
1193  * @param handle connection to mesh to disconnect
1194  */
1195 void
1196 GNUNET_MESH_disconnect (struct GNUNET_MESH_Handle *handle)
1197 {
1198   struct GNUNET_MESH_Tunnel *t;
1199   struct GNUNET_MESH_Tunnel *aux;
1200
1201   t = handle->tunnels_head;
1202   while (NULL != t)
1203   {
1204     aux = t->next;
1205     destroy_tunnel (t);
1206     t = aux;
1207   }
1208   if (NULL != handle->th)
1209   {
1210     GNUNET_CLIENT_notify_transmit_ready_cancel (handle->th);
1211   }
1212   if (NULL != handle->client)
1213   {
1214     GNUNET_CLIENT_disconnect (handle->client, GNUNET_NO);
1215   }
1216   GNUNET_free (handle);
1217 }
1218
1219
1220 /**
1221  * Create a new tunnel (we're initiator and will be allowed to add/remove peers
1222  * and to broadcast).
1223  *
1224  * @param h mesh handle
1225  * @param tunnel_ctx client's tunnel context to associate with the tunnel
1226  * @param connect_handler function to call when peers are actually connected
1227  * @param disconnect_handler function to call when peers are disconnected
1228  * @param handler_cls closure for connect/disconnect handlers
1229  */
1230 struct GNUNET_MESH_Tunnel *
1231 GNUNET_MESH_tunnel_create (struct GNUNET_MESH_Handle *h, void *tunnel_ctx,
1232                            GNUNET_MESH_PeerConnectHandler connect_handler,
1233                            GNUNET_MESH_PeerDisconnectHandler disconnect_handler,
1234                            void *handler_cls)
1235 {
1236   struct GNUNET_MESH_Tunnel *t;
1237   struct GNUNET_MESH_TunnelMessage msg;
1238
1239   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Creating new tunnel\n");
1240   t = create_tunnel (h, 0);
1241   t->connect_handler = connect_handler;
1242   t->disconnect_handler = disconnect_handler;
1243   t->cls = handler_cls;
1244   t->ctx = tunnel_ctx;
1245   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
1246   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1247   msg.tunnel_id = htonl (t->tid);
1248   send_packet (h, &msg.header);
1249   return t;
1250 }
1251
1252
1253 /**
1254  * Destroy an existing tunnel.
1255  *
1256  * @param tun tunnel handle
1257  */
1258 void
1259 GNUNET_MESH_tunnel_destroy (struct GNUNET_MESH_Tunnel *tunnel)
1260 {
1261   struct GNUNET_MESH_Handle *h;
1262   struct GNUNET_MESH_TunnelMessage msg;
1263
1264   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Destroying tunnel\n");
1265   h = tunnel->mesh;
1266
1267   if (0 != tunnel->owner)
1268     GNUNET_PEER_change_rc (tunnel->owner, -1);
1269
1270   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY);
1271   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1272   msg.tunnel_id = htonl (tunnel->tid);
1273   destroy_tunnel (tunnel);
1274   send_packet (h, &msg.header);
1275 }
1276
1277
1278 /**
1279  * Request that a peer should be added to the tunnel.  The existing
1280  * connect handler will be called ONCE with either success or failure.
1281  * This function should NOT be called again with the same peer before the
1282  * connect handler is called.
1283  *
1284  * @param tunnel handle to existing tunnel
1285  * @param peer peer to add
1286  */
1287 void
1288 GNUNET_MESH_peer_request_connect_add (struct GNUNET_MESH_Tunnel *tunnel,
1289                                       const struct GNUNET_PeerIdentity *peer)
1290 {
1291   struct GNUNET_MESH_PeerControl msg;
1292   GNUNET_PEER_Id peer_id;
1293   unsigned int i;
1294
1295   peer_id = GNUNET_PEER_intern (peer);
1296   for (i = 0; i < tunnel->npeers; i++)
1297   {
1298     if (tunnel->peers[i]->id == peer_id)
1299     {
1300       GNUNET_PEER_change_rc (peer_id, -1);
1301       GNUNET_break (0);
1302       return;
1303     }
1304   }
1305   if (NULL == add_peer_to_tunnel (tunnel, peer))
1306     return;
1307
1308   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1309   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
1310   msg.tunnel_id = htonl (tunnel->tid);
1311   msg.peer = *peer;
1312   send_packet (tunnel->mesh, &msg.header);
1313
1314   return;
1315 }
1316
1317
1318 /**
1319  * Request that a peer should be removed from the tunnel.  The existing
1320  * disconnect handler will be called ONCE if we were connected.
1321  *
1322  * @param tunnel handle to existing tunnel
1323  * @param peer peer to remove
1324  */
1325 void
1326 GNUNET_MESH_peer_request_connect_del (struct GNUNET_MESH_Tunnel *tunnel,
1327                                       const struct GNUNET_PeerIdentity *peer)
1328 {
1329   struct GNUNET_MESH_PeerControl msg;
1330   GNUNET_PEER_Id peer_id;
1331   unsigned int i;
1332
1333   peer_id = GNUNET_PEER_search (peer);
1334   if (0 == peer_id)
1335   {
1336     GNUNET_break (0);
1337     return;
1338   }
1339   for (i = 0; i < tunnel->npeers; i++)
1340     if (tunnel->peers[i]->id == peer_id)
1341       break;
1342   if (i == tunnel->npeers)
1343   {
1344     GNUNET_break (0);
1345     return;
1346   }
1347   if (NULL != tunnel->disconnect_handler && tunnel->peers[i]->connected == 1)
1348     tunnel->disconnect_handler (tunnel->cls, peer);
1349   GNUNET_PEER_change_rc (peer_id, -1);
1350   GNUNET_free (tunnel->peers[i]);
1351   tunnel->peers[i] = tunnel->peers[tunnel->npeers - 1];
1352   GNUNET_array_grow (tunnel->peers, tunnel->npeers, tunnel->npeers - 1);
1353
1354   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1355   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL);
1356   msg.tunnel_id = htonl (tunnel->tid);
1357   memcpy (&msg.peer, peer, sizeof (struct GNUNET_PeerIdentity));
1358   send_packet (tunnel->mesh, &msg.header);
1359 }
1360
1361
1362 /**
1363  * Request that the mesh should try to connect to a peer supporting the given
1364  * message type.
1365  *
1366  * @param tunnel handle to existing tunnel
1367  * @param app_type application type that must be supported by the peer (MESH
1368  *                 should discover peer in proximity handling this type)
1369  */
1370 void
1371 GNUNET_MESH_peer_request_connect_by_type (struct GNUNET_MESH_Tunnel *tunnel,
1372                                           GNUNET_MESH_ApplicationType app_type)
1373 {
1374   struct GNUNET_MESH_ConnectPeerByType msg;
1375
1376   GNUNET_array_append (tunnel->apps, tunnel->napps, app_type);
1377
1378   msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
1379   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE);
1380   msg.tunnel_id = htonl (tunnel->tid);
1381   msg.type = htonl (app_type);
1382   send_packet (tunnel->mesh, &msg.header);
1383 }
1384
1385
1386 /**
1387  * Ask the mesh to call "notify" once it is ready to transmit the
1388  * given number of bytes to the specified "target".  If we are not yet
1389  * connected to the specified peer, a call to this function will cause
1390  * us to try to establish a connection.
1391  *
1392  * @param tunnel tunnel to use for transmission
1393  * @param cork is corking allowed for this transmission?
1394  * @param priority how important is the message?
1395  * @param maxdelay how long can the message wait?
1396  * @param target destination for the message,
1397  *               NULL for multicast to all tunnel targets
1398  * @param notify_size how many bytes of buffer space does notify want?
1399  * @param notify function to call when buffer space is available;
1400  *        will be called with NULL on timeout or if the overall queue
1401  *        for this peer is larger than queue_size and this is currently
1402  *        the message with the lowest priority
1403  * @param notify_cls closure for notify
1404  * @return non-NULL if the notify callback was queued,
1405  *         NULL if we can not even queue the request (insufficient
1406  *         memory); if NULL is returned, "notify" will NOT be called.
1407  */
1408 struct GNUNET_MESH_TransmitHandle *
1409 GNUNET_MESH_notify_transmit_ready (struct GNUNET_MESH_Tunnel *tunnel, int cork,
1410                                    uint32_t priority,
1411                                    struct GNUNET_TIME_Relative maxdelay,
1412                                    const struct GNUNET_PeerIdentity *target,
1413                                    size_t notify_size,
1414                                    GNUNET_CONNECTION_TransmitReadyNotify notify,
1415                                    void *notify_cls)
1416 {
1417   struct GNUNET_MESH_TransmitHandle *th;
1418   struct GNUNET_MESH_TransmitHandle *least_priority_th;
1419   uint32_t least_priority;
1420   size_t overhead;
1421
1422   GNUNET_assert (NULL != notify);
1423   if (tunnel->mesh->npackets >= tunnel->mesh->max_queue_size &&
1424       tunnel->npackets > 0)
1425   {
1426     /* queue full */
1427     if (0 == priority)
1428       return NULL;
1429     th = tunnel->mesh->th_tail;
1430     least_priority = priority;
1431     least_priority_th = NULL;
1432     while (NULL != th)
1433     {
1434       if (th->priority < least_priority && th->tunnel->npackets > 1)
1435       {
1436         least_priority_th = th;
1437         least_priority = th->priority;
1438       }
1439       th = th->prev;
1440     }
1441     if (NULL == least_priority_th)
1442       return NULL;
1443     /* Can't be a control message */
1444     GNUNET_assert (NULL != least_priority_th->notify);
1445     least_priority_th->notify (notify_cls, 0, NULL);
1446     least_priority_th->tunnel->npackets--;
1447     tunnel->mesh->npackets--;
1448     GNUNET_CONTAINER_DLL_remove (tunnel->mesh->th_head, tunnel->mesh->th_tail,
1449                                  least_priority_th);
1450     if (GNUNET_SCHEDULER_NO_TASK != least_priority_th->timeout_task)
1451       GNUNET_SCHEDULER_cancel (least_priority_th->timeout_task);
1452     GNUNET_free (least_priority_th);
1453   }
1454   tunnel->npackets++;
1455   tunnel->mesh->npackets++;
1456   th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle));
1457   th->tunnel = tunnel;
1458   th->priority = priority;
1459   th->timeout = GNUNET_TIME_relative_to_absolute (maxdelay);
1460   th->target = GNUNET_PEER_intern (target);
1461   if (NULL == target)
1462     overhead = sizeof (struct GNUNET_MESH_Multicast);
1463   else
1464     overhead = sizeof (struct GNUNET_MESH_Unicast);
1465   th->size = notify_size + overhead;
1466   th->notify = notify;
1467   th->notify_cls = notify_cls;
1468   add_to_queue (tunnel->mesh, th);
1469   return th;
1470 }
1471
1472
1473 /**
1474  * Cancel the specified transmission-ready notification.
1475  *
1476  * @param th handle that was returned by "notify_transmit_ready".
1477  */
1478 void
1479 GNUNET_MESH_notify_transmit_ready_cancel (struct GNUNET_MESH_TransmitHandle *th)
1480 {
1481   struct GNUNET_MESH_Handle *mesh;
1482
1483   mesh = th->tunnel->mesh;
1484   if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1485     GNUNET_SCHEDULER_cancel (th->timeout_task);
1486   GNUNET_CONTAINER_DLL_remove (mesh->th_head, mesh->th_tail, th);
1487   GNUNET_free (th);
1488   if ((NULL == mesh->th_head) && (NULL != mesh->th))
1489   {
1490     /* queue empty, no point in asking for transmission */
1491     GNUNET_CLIENT_notify_transmit_ready_cancel (mesh->th);
1492     mesh->th = NULL;
1493   }
1494 }
1495
1496
1497 #if 0                           /* keep Emacsens' auto-indent happy */
1498 {
1499 #endif
1500 #ifdef __cplusplus
1501 }
1502 #endif