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       }
 885 #if DEBUG
 886       else
 887       {
 888         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 889                     "MESH: callback completed successfully\n");
 890
 891       }
 892 #endif
 893     }
 894   }
 895 }
 896
 897
 898 /**
 899  * Function to process all messages received from the service
 900  *
 901  * @param cls closure
 902  * @param msg message received, NULL on timeout or fatal error
 903  */
 904 static void
 905 msg_received (void *cls, const struct GNUNET_MessageHeader *msg)
 906 {
 907   struct GNUNET_MESH_Handle *h = cls;
 908
 909   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 910               "mesh: received a message type %hu from MESH\n",
 911               ntohs (msg->type));
 912   if (msg == NULL)
 913   {
 914     reconnect (h);
 915     return;
 916   }
 917   switch (ntohs (msg->type))
 918   {
 919     /* Notify of a new incoming tunnel */
 920   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE:
 921     process_tunnel_created (h, (struct GNUNET_MESH_TunnelNotification *) msg);
 922     break;
 923     /* Notify of a tunnel disconnection */
 924   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY:
 925     process_tunnel_destroy (h, (struct GNUNET_MESH_TunnelMessage *) msg);
 926     break;
 927     /* Notify of a new peer or a peer disconnect in the tunnel */
 928   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD:
 929   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL:
 930     process_peer_event (h, (struct GNUNET_MESH_PeerControl *) msg);
 931     break;
 932     /* Notify of a new data packet in the tunnel */
 933   case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
 934   case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
 935   case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
 936     process_incoming_data (h, msg);
 937     break;
 938     /* We shouldn't get any other packages, log and ignore */
 939   default:
 940     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 941                 "MESH: unsolicited message form service (type %d)\n",
 942                 ntohs (msg->type));
 943   }
 944   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: message processed\n");
 945   GNUNET_CLIENT_receive (h->client, &msg_received, h,
 946                          GNUNET_TIME_UNIT_FOREVER_REL);
 947 }
 948
 949
 950 /******************************************************************************/
 951 /************************       SEND FUNCTIONS     ****************************/
 952 /******************************************************************************/
 953
 954 /**
 955  * Function called to send a message to the service.
 956  * "buf" will be NULL and "size" zero if the socket was closed for writing in
 957  * the meantime.
 958  *
 959  * @param cls closure, the mesh handle
 960  * @param size number of bytes available in buf
 961  * @param buf where the callee should write the connect message
 962  * @return number of bytes written to buf
 963  */
 964 static size_t
 965 send_callback (void *cls, size_t size, void *buf)
 966 {
 967   struct GNUNET_MESH_Handle *h = cls;
 968   struct GNUNET_MESH_TransmitHandle *th;
 969   char *cbuf = buf;
 970   size_t tsize;
 971   size_t psize;
 972
 973   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Send packet() Buffer %u\n", size);
 974   h->th = NULL;
 975   if ((0 == size) || (NULL == buf))
 976   {
 977     reconnect (h);
 978     return 0;
 979   }
 980   tsize = 0;
 981   while ((NULL != (th = h->th_head)) && (size >= th->size))
 982   {
 983 #if DEBUG
 984     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:     type: %u\n",
 985                 ntohs (((struct GNUNET_MessageHeader *) &th[1])->type));
 986     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:     size: %u\n",
 987                 ntohs (((struct GNUNET_MessageHeader *) &th[1])->size));
 988 #endif
 989     if (NULL != th->notify)
 990     {
 991       if (th->target == 0)
 992       {
 993         /* multicast */
 994         struct GNUNET_MESH_Multicast mc;
 995
 996         GNUNET_assert (size >= sizeof (mc) + th->size);
 997         psize =
 998             th->notify (th->notify_cls, size - sizeof (mc), &cbuf[sizeof (mc)]);
 999         if (psize > 0)
1000         {
1001           mc.header.size = htons (sizeof (mc) + th->size);
1002           mc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_MULTICAST);
1003           mc.tid = htonl (th->tunnel->tid);
1004           memset (&mc.oid, 0, sizeof (struct GNUNET_PeerIdentity));
1005           memcpy (cbuf, &mc, sizeof (mc));
1006           psize = th->size + sizeof (mc);
1007         }
1008       }
1009       else
1010       {
1011         /* unicast */
1012         struct GNUNET_MESH_Unicast uc;
1013
1014         GNUNET_assert (size >= sizeof (uc) + th->size);
1015         psize =
1016             th->notify (th->notify_cls, size - sizeof (uc), &cbuf[sizeof (uc)]);
1017         if (psize > 0)
1018         {
1019           uc.header.size = htons (sizeof (uc) + th->size);
1020           uc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_UNICAST);
1021           uc.tid = htonl (th->tunnel->tid);
1022           memset (&uc.oid, 0, sizeof (struct GNUNET_PeerIdentity));
1023           GNUNET_PEER_resolve (th->target, &uc.destination);
1024           memcpy (cbuf, &uc, sizeof (uc));
1025           psize = th->size + sizeof (uc);
1026         }
1027       }
1028     }
1029     else
1030     {
1031       memcpy (cbuf, &th[1], th->size);
1032       psize = th->size;
1033     }
1034     if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1035       GNUNET_SCHEDULER_cancel (th->timeout_task);
1036     GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th);
1037     GNUNET_free (th);
1038     cbuf += psize;
1039     size -= psize;
1040     tsize += psize;
1041   }
1042   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:   total size: %u\n", tsize);
1043   if (NULL != (th = h->th_head))
1044   {
1045     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:   next size: %u\n", th->size);
1046     h->th =
1047         GNUNET_CLIENT_notify_transmit_ready (h->client, th->size,
1048                                              GNUNET_TIME_UNIT_FOREVER_REL,
1049                                              GNUNET_YES, &send_callback, h);
1050   }
1051   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Send packet() END\n");
1052   if (GNUNET_NO == h->in_receive)
1053   {
1054     h->in_receive = GNUNET_YES;
1055     GNUNET_CLIENT_receive (h->client, &msg_received, h,
1056                            GNUNET_TIME_UNIT_FOREVER_REL);
1057   }
1058   return tsize;
1059 }
1060
1061
1062 /**
1063  * Auxiliary function to send an already constructed packet to the service.
1064  * Takes care of creating a new queue element, copying the message and
1065  * calling the tmt_rdy function if necessary.
1066  * @param h mesh handle
1067  * @param msg message to transmit
1068  */
1069 static void
1070 send_packet (struct GNUNET_MESH_Handle *h,
1071              const struct GNUNET_MessageHeader *msg)
1072 {
1073   struct GNUNET_MESH_TransmitHandle *th;
1074   size_t msize;
1075
1076   msize = ntohs (msg->size);
1077   th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle) + msize);
1078   th->priority = UINT32_MAX;
1079   th->timeout = GNUNET_TIME_UNIT_FOREVER_ABS;
1080   th->size = msize;
1081   memcpy (&th[1], msg, msize);
1082   add_to_queue (h, th);
1083   if (NULL != h->th)
1084     return;
1085   h->th =
1086       GNUNET_CLIENT_notify_transmit_ready (h->client, msize,
1087                                            GNUNET_TIME_UNIT_FOREVER_REL,
1088                                            GNUNET_YES, &send_callback, h);
1089 }
1090
1091
1092 /******************************************************************************/
1093 /**********************      API CALL DEFINITIONS     *************************/
1094 /******************************************************************************/
1095
1096 /**
1097  * Connect to the mesh service.
1098  *
1099  * @param cfg configuration to use
1100  * @param queue_size size of the data message queue, shared among all tunnels
1101  *                   (each tunnel is guaranteed to accept at least one message,
1102  *                    no matter what is the status of other tunnels)
1103  * @param cls closure for the various callbacks that follow
1104  *            (including handlers in the handlers array)
1105  * @param new_tunnel function called when an *inbound* tunnel is created
1106  * @param cleaner function called when an *inbound* tunnel is destroyed
1107  * @param handlers callbacks for messages we care about, NULL-terminated
1108  *                note that the mesh is allowed to drop notifications about
1109  *                inbound messages if the client does not process them fast
1110  *                enough (for this notification type, a bounded queue is used)
1111  * @param stypes list of the applications that this client claims to provide
1112  * @return handle to the mesh service NULL on error
1113  *         (in this case, init is never called)
1114  */
1115 struct GNUNET_MESH_Handle *
1116 GNUNET_MESH_connect (const struct GNUNET_CONFIGURATION_Handle *cfg,
1117                      unsigned int queue_size, void *cls,
1118                      GNUNET_MESH_InboundTunnelNotificationHandler new_tunnel,
1119                      GNUNET_MESH_TunnelEndHandler cleaner,
1120                      const struct GNUNET_MESH_MessageHandler *handlers,
1121                      const GNUNET_MESH_ApplicationType *stypes)
1122 {
1123   struct GNUNET_MESH_Handle *h;
1124   struct GNUNET_MESH_ClientConnect *msg;
1125   GNUNET_MESH_ApplicationType *apps;
1126   uint16_t napps;
1127   uint16_t *types;
1128   uint16_t ntypes;
1129   size_t size;
1130
1131   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: GNUNET_MESH_connect()\n");
1132   h = GNUNET_malloc (sizeof (struct GNUNET_MESH_Handle));
1133   h->cfg = cfg;
1134   h->max_queue_size = queue_size;
1135   h->new_tunnel = new_tunnel;
1136   h->cleaner = cleaner;
1137   h->client = GNUNET_CLIENT_connect ("mesh", cfg);
1138   if (h->client == NULL)
1139   {
1140     GNUNET_break (0);
1141     GNUNET_free (h);
1142     return NULL;
1143   }
1144   h->cls = cls;
1145   /* FIXME memdup? */
1146   h->applications = stypes;
1147   h->message_handlers = handlers;
1148   h->next_tid = GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
1149   h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS;
1150
1151   /* count handlers and apps, calculate size */
1152   for (h->n_applications = 0; stypes[h->n_applications]; h->n_applications++) ;
1153   for (h->n_handlers = 0; handlers[h->n_handlers].type; h->n_handlers++) ;
1154   size = sizeof (struct GNUNET_MESH_ClientConnect);
1155   size += h->n_applications * sizeof (GNUNET_MESH_ApplicationType);
1156   size += h->n_handlers * sizeof (uint16_t);
1157
1158   {
1159     char buf[size];
1160
1161     /* build connection packet */
1162     msg = (struct GNUNET_MESH_ClientConnect *) buf;
1163     msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT);
1164     msg->header.size = htons (size);
1165     apps = (GNUNET_MESH_ApplicationType *) &msg[1];
1166     for (napps = 0; napps < h->n_applications; napps++)
1167     {
1168       apps[napps] = htonl(h->applications[napps]);
1169       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "mesh:  app %u\n", h->applications[napps]);
1170     }
1171     types = (uint16_t *) &apps[napps];
1172     for (ntypes = 0; ntypes < h->n_handlers; ntypes++)
1173       types[ntypes] = htons(h->message_handlers[ntypes].type);
1174     msg->applications = htons (napps);
1175     msg->types = htons (ntypes);
1176 #if DEBUG
1177     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1178                 "mesh: Sending %lu bytes long message %d types and %d apps\n",
1179                 ntohs (msg->header.size), ntypes, napps);
1180 #endif
1181     send_packet (h, &msg->header);
1182   }
1183   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: GNUNET_MESH_connect() END\n");
1184   return h;
1185 }
1186
1187
1188 /**
1189  * Disconnect from the mesh service.
1190  *
1191  * @param handle connection to mesh to disconnect
1192  */
1193 void
1194 GNUNET_MESH_disconnect (struct GNUNET_MESH_Handle *handle)
1195 {
1196   struct GNUNET_MESH_Tunnel *t;
1197   struct GNUNET_MESH_Tunnel *aux;
1198
1199   t = handle->tunnels_head;
1200   while (NULL != t)
1201   {
1202     aux = t->next;
1203     destroy_tunnel (t);
1204     t = aux;
1205   }
1206   if (NULL != handle->th)
1207   {
1208     GNUNET_CLIENT_notify_transmit_ready_cancel (handle->th);
1209   }
1210   if (NULL != handle->client)
1211   {
1212     GNUNET_CLIENT_disconnect (handle->client, GNUNET_NO);
1213   }
1214   GNUNET_free (handle);
1215 }
1216
1217
1218 /**
1219  * Create a new tunnel (we're initiator and will be allowed to add/remove peers
1220  * and to broadcast).
1221  *
1222  * @param h mesh handle
1223  * @param tunnel_ctx client's tunnel context to associate with the tunnel
1224  * @param connect_handler function to call when peers are actually connected
1225  * @param disconnect_handler function to call when peers are disconnected
1226  * @param handler_cls closure for connect/disconnect handlers
1227  */
1228 struct GNUNET_MESH_Tunnel *
1229 GNUNET_MESH_tunnel_create (struct GNUNET_MESH_Handle *h, void *tunnel_ctx,
1230                            GNUNET_MESH_PeerConnectHandler connect_handler,
1231                            GNUNET_MESH_PeerDisconnectHandler disconnect_handler,
1232                            void *handler_cls)
1233 {
1234   struct GNUNET_MESH_Tunnel *t;
1235   struct GNUNET_MESH_TunnelMessage msg;
1236
1237   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Creating new tunnel\n");
1238   t = create_tunnel (h, 0);
1239   t->connect_handler = connect_handler;
1240   t->disconnect_handler = disconnect_handler;
1241   t->cls = handler_cls;
1242   t->ctx = tunnel_ctx;
1243   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
1244   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1245   msg.tunnel_id = htonl (t->tid);
1246   send_packet (h, &msg.header);
1247   return t;
1248 }
1249
1250
1251 /**
1252  * Destroy an existing tunnel.
1253  *
1254  * @param tun tunnel handle
1255  */
1256 void
1257 GNUNET_MESH_tunnel_destroy (struct GNUNET_MESH_Tunnel *tunnel)
1258 {
1259   struct GNUNET_MESH_Handle *h;
1260   struct GNUNET_MESH_TunnelMessage msg;
1261
1262   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Destroying tunnel\n");
1263   h = tunnel->mesh;
1264
1265   if (0 != tunnel->owner)
1266     GNUNET_PEER_change_rc (tunnel->owner, -1);
1267
1268   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY);
1269   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1270   msg.tunnel_id = htonl (tunnel->tid);
1271   destroy_tunnel (tunnel);
1272   send_packet (h, &msg.header);
1273 }
1274
1275
1276 /**
1277  * Request that a peer should be added to the tunnel.  The existing
1278  * connect handler will be called ONCE with either success or failure.
1279  * This function should NOT be called again with the same peer before the
1280  * connect handler is called.
1281  *
1282  * @param tunnel handle to existing tunnel
1283  * @param peer peer to add
1284  */
1285 void
1286 GNUNET_MESH_peer_request_connect_add (struct GNUNET_MESH_Tunnel *tunnel,
1287                                       const struct GNUNET_PeerIdentity *peer)
1288 {
1289   struct GNUNET_MESH_PeerControl msg;
1290   GNUNET_PEER_Id peer_id;
1291   unsigned int i;
1292
1293   peer_id = GNUNET_PEER_intern (peer);
1294   for (i = 0; i < tunnel->npeers; i++)
1295   {
1296     if (tunnel->peers[i]->id == peer_id)
1297     {
1298       GNUNET_PEER_change_rc (peer_id, -1);
1299       GNUNET_break (0);
1300       return;
1301     }
1302   }
1303   if (NULL == add_peer_to_tunnel (tunnel, peer))
1304     return;
1305
1306   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1307   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
1308   msg.tunnel_id = htonl (tunnel->tid);
1309   msg.peer = *peer;
1310   send_packet (tunnel->mesh, &msg.header);
1311
1312   return;
1313 }
1314
1315
1316 /**
1317  * Request that a peer should be removed from the tunnel.  The existing
1318  * disconnect handler will be called ONCE if we were connected.
1319  *
1320  * @param tunnel handle to existing tunnel
1321  * @param peer peer to remove
1322  */
1323 void
1324 GNUNET_MESH_peer_request_connect_del (struct GNUNET_MESH_Tunnel *tunnel,
1325                                       const struct GNUNET_PeerIdentity *peer)
1326 {
1327   struct GNUNET_MESH_PeerControl msg;
1328   GNUNET_PEER_Id peer_id;
1329   unsigned int i;
1330
1331   peer_id = GNUNET_PEER_search (peer);
1332   if (0 == peer_id)
1333   {
1334     GNUNET_break (0);
1335     return;
1336   }
1337   for (i = 0; i < tunnel->npeers; i++)
1338     if (tunnel->peers[i]->id == peer_id)
1339       break;
1340   if (i == tunnel->npeers)
1341   {
1342     GNUNET_break (0);
1343     return;
1344   }
1345   if (NULL != tunnel->disconnect_handler && tunnel->peers[i]->connected == 1)
1346     tunnel->disconnect_handler (tunnel->cls, peer);
1347   GNUNET_PEER_change_rc (peer_id, -1);
1348   GNUNET_free (tunnel->peers[i]);
1349   tunnel->peers[i] = tunnel->peers[tunnel->npeers - 1];
1350   GNUNET_array_grow (tunnel->peers, tunnel->npeers, tunnel->npeers - 1);
1351
1352   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1353   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL);
1354   msg.tunnel_id = htonl (tunnel->tid);
1355   memcpy (&msg.peer, peer, sizeof (struct GNUNET_PeerIdentity));
1356   send_packet (tunnel->mesh, &msg.header);
1357 }
1358
1359
1360 /**
1361  * Request that the mesh should try to connect to a peer supporting the given
1362  * message type.
1363  *
1364  * @param tunnel handle to existing tunnel
1365  * @param app_type application type that must be supported by the peer (MESH
1366  *                 should discover peer in proximity handling this type)
1367  */
1368 void
1369 GNUNET_MESH_peer_request_connect_by_type (struct GNUNET_MESH_Tunnel *tunnel,
1370                                           GNUNET_MESH_ApplicationType app_type)
1371 {
1372   struct GNUNET_MESH_ConnectPeerByType msg;
1373
1374   GNUNET_array_append (tunnel->apps, tunnel->napps, app_type);
1375
1376   msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
1377   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE);
1378   msg.tunnel_id = htonl (tunnel->tid);
1379   msg.type = htonl (app_type);
1380   send_packet (tunnel->mesh, &msg.header);
1381 }
1382
1383
1384 /**
1385  * Ask the mesh to call "notify" once it is ready to transmit the
1386  * given number of bytes to the specified "target".  If we are not yet
1387  * connected to the specified peer, a call to this function will cause
1388  * us to try to establish a connection.
1389  *
1390  * @param tunnel tunnel to use for transmission
1391  * @param cork is corking allowed for this transmission?
1392  * @param priority how important is the message?
1393  * @param maxdelay how long can the message wait?
1394  * @param target destination for the message,
1395  *               NULL for multicast to all tunnel targets
1396  * @param notify_size how many bytes of buffer space does notify want?
1397  * @param notify function to call when buffer space is available;
1398  *        will be called with NULL on timeout or if the overall queue
1399  *        for this peer is larger than queue_size and this is currently
1400  *        the message with the lowest priority
1401  * @param notify_cls closure for notify
1402  * @return non-NULL if the notify callback was queued,
1403  *         NULL if we can not even queue the request (insufficient
1404  *         memory); if NULL is returned, "notify" will NOT be called.
1405  */
1406 struct GNUNET_MESH_TransmitHandle *
1407 GNUNET_MESH_notify_transmit_ready (struct GNUNET_MESH_Tunnel *tunnel, int cork,
1408                                    uint32_t priority,
1409                                    struct GNUNET_TIME_Relative maxdelay,
1410                                    const struct GNUNET_PeerIdentity *target,
1411                                    size_t notify_size,
1412                                    GNUNET_CONNECTION_TransmitReadyNotify notify,
1413                                    void *notify_cls)
1414 {
1415   struct GNUNET_MESH_TransmitHandle *th;
1416   struct GNUNET_MESH_TransmitHandle *least_priority_th;
1417   uint32_t least_priority;
1418   size_t overhead;
1419
1420   GNUNET_assert (NULL != notify);
1421   if (tunnel->mesh->npackets >= tunnel->mesh->max_queue_size &&
1422       tunnel->npackets > 0)
1423   {
1424     /* queue full */
1425     if (0 == priority)
1426       return NULL;
1427     th = tunnel->mesh->th_tail;
1428     least_priority = priority;
1429     least_priority_th = NULL;
1430     while (NULL != th)
1431     {
1432       if (th->priority < least_priority && th->tunnel->npackets > 1)
1433       {
1434         least_priority_th = th;
1435         least_priority = th->priority;
1436       }
1437       th = th->prev;
1438     }
1439     if (NULL == least_priority_th)
1440       return NULL;
1441     /* Can't be a control message */
1442     GNUNET_assert (NULL != least_priority_th->notify);
1443     least_priority_th->notify (notify_cls, 0, NULL);
1444     least_priority_th->tunnel->npackets--;
1445     tunnel->mesh->npackets--;
1446     GNUNET_CONTAINER_DLL_remove (tunnel->mesh->th_head, tunnel->mesh->th_tail,
1447                                  least_priority_th);
1448     if (GNUNET_SCHEDULER_NO_TASK != least_priority_th->timeout_task)
1449       GNUNET_SCHEDULER_cancel (least_priority_th->timeout_task);
1450     GNUNET_free (least_priority_th);
1451   }
1452   tunnel->npackets++;
1453   tunnel->mesh->npackets++;
1454   th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle));
1455   th->tunnel = tunnel;
1456   th->priority = priority;
1457   th->timeout = GNUNET_TIME_relative_to_absolute (maxdelay);
1458   th->target = GNUNET_PEER_intern (target);
1459   if (NULL == target)
1460     overhead = sizeof (struct GNUNET_MESH_Multicast);
1461   else
1462     overhead = sizeof (struct GNUNET_MESH_Unicast);
1463   th->size = notify_size + overhead;
1464   th->notify = notify;
1465   th->notify_cls = notify_cls;
1466   add_to_queue (tunnel->mesh, th);
1467   return th;
1468 }
1469
1470
1471 /**
1472  * Cancel the specified transmission-ready notification.
1473  *
1474  * @param th handle that was returned by "notify_transmit_ready".
1475  */
1476 void
1477 GNUNET_MESH_notify_transmit_ready_cancel (struct GNUNET_MESH_TransmitHandle *th)
1478 {
1479   struct GNUNET_MESH_Handle *mesh;
1480
1481   mesh = th->tunnel->mesh;
1482   if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1483     GNUNET_SCHEDULER_cancel (th->timeout_task);
1484   GNUNET_CONTAINER_DLL_remove (mesh->th_head, mesh->th_tail, th);
1485   GNUNET_free (th);
1486   if ((NULL == mesh->th_head) && (NULL != mesh->th))
1487   {
1488     /* queue empty, no point in asking for transmission */
1489     GNUNET_CLIENT_notify_transmit_ready_cancel (mesh->th);
1490     mesh->th = NULL;
1491   }
1492 }
1493
1494
1495 #if 0                           /* keep Emacsens' auto-indent happy */
1496 {
1497 #endif
1498 #ifdef __cplusplus
1499 }
1500 #endif