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 multicast.  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  * Send a connect packet to the service with the applications and types
 583  * requested by the user.
 584  *
 585  * @param h The mesh handle.
 586  *
 587  */
 588 static void
 589 send_connect (struct GNUNET_MESH_Handle *h)
 590 {
 591   size_t size;
 592
 593   size = sizeof (struct GNUNET_MESH_ClientConnect);
 594   size += h->n_applications * sizeof (GNUNET_MESH_ApplicationType);
 595   size += h->n_handlers * sizeof (uint16_t);
 596   {
 597     char buf[size];
 598     struct GNUNET_MESH_ClientConnect *msg;
 599     GNUNET_MESH_ApplicationType *apps;
 600     uint16_t napps;
 601     uint16_t *types;
 602     uint16_t ntypes;
 603
 604     /* build connection packet */
 605     msg = (struct GNUNET_MESH_ClientConnect *) buf;
 606     msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT);
 607     msg->header.size = htons (size);
 608     apps = (GNUNET_MESH_ApplicationType *) &msg[1];
 609     for (napps = 0; napps < h->n_applications; napps++)
 610     {
 611       apps[napps] = htonl (h->applications[napps]);
 612       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:  app %u\n",
 613                   h->applications[napps]);
 614     }
 615     types = (uint16_t *) & apps[napps];
 616     for (ntypes = 0; ntypes < h->n_handlers; ntypes++)
 617       types[ntypes] = htons (h->message_handlers[ntypes].type);
 618     msg->applications = htons (napps);
 619     msg->types = htons (ntypes);
 620 #if DEBUG
 621     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 622                 "mesh: Sending %lu bytes long message %d types and %d apps\n",
 623                 ntohs (msg->header.size), ntypes, napps);
 624 #endif
 625     send_packet (h, &msg->header);
 626   }
 627 }
 628
 629
 630 /**
 631  * Reconnect to the service, retransmit all infomation to try to restore the
 632  * original state.
 633  *
 634  * @param h handle to the mesh
 635  *
 636  * @return GNUNET_YES in case of sucess, GNUNET_NO otherwise (service down...)
 637  */
 638 static int
 639 reconnect (struct GNUNET_MESH_Handle *h)
 640 {
 641   struct GNUNET_MESH_Tunnel *t;
 642   unsigned int i;
 643
 644 #if DEBUG
 645   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: *****************************\n");
 646   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: *******   RECONNECT   *******\n");
 647   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: *****************************\n");
 648 #endif
 649   h->in_receive = GNUNET_NO;
 650   /* disconnect */
 651   if (NULL != h->th)
 652   {
 653     GNUNET_CLIENT_notify_transmit_ready_cancel (h->th);
 654   }
 655   if (NULL != h->client)
 656   {
 657     GNUNET_CLIENT_disconnect (h->client, GNUNET_NO);
 658   }
 659
 660   /* connect again */
 661   h->client = GNUNET_CLIENT_connect ("mesh", h->cfg);
 662   if (h->client == NULL)
 663   {
 664     GNUNET_SCHEDULER_add_delayed (h->reconnect_time, &reconnect_cbk, h);
 665     h->reconnect_time =
 666         GNUNET_TIME_relative_min (GNUNET_TIME_UNIT_HOURS,
 667                                   GNUNET_TIME_relative_multiply
 668                                   (h->reconnect_time, 2));
 669     GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
 670                "mesh:   Next retry in %sms\n",
 671                GNUNET_TIME_relative_to_string(h->reconnect_time));
 672     GNUNET_break (0);
 673     return GNUNET_NO;
 674   }
 675   else
 676   {
 677     h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS;
 678   }
 679   send_connect(h);
 680   /* Rebuild all tunnels */
 681   for (t = h->tunnels_head; NULL != t; t = t->next)
 682   {
 683     struct GNUNET_MESH_TunnelMessage tmsg;
 684     struct GNUNET_MESH_PeerControl pmsg;
 685
 686     tmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
 687     tmsg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
 688     tmsg.tunnel_id = htonl (t->tid);
 689     send_packet (h, &tmsg.header);
 690
 691     pmsg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
 692     pmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
 693     pmsg.tunnel_id = htonl (t->tid);
 694
 695     /* Reconnect all peers */
 696     for (i = 0; i < t->npeers; i++)
 697     {
 698       GNUNET_PEER_resolve (t->peers[i]->id, &pmsg.peer);
 699       if (NULL != t->disconnect_handler && t->peers[i]->connected)
 700         t->disconnect_handler (t->cls, &pmsg.peer);
 701       /* If the tunnel was "by type", dont connect individual peers */
 702       if (0 == t->napps)
 703         send_packet (t->mesh, &pmsg.header);
 704     }
 705     /* Reconnect all types, if any  */
 706     for (i = 0; i < t->napps; i++)
 707     {
 708       struct GNUNET_MESH_ConnectPeerByType msg;
 709
 710       msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
 711       msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE);
 712       msg.tunnel_id = htonl (t->tid);
 713       msg.type = htonl (t->apps[i]);
 714       send_packet (t->mesh, &msg.header);
 715     }
 716   }
 717   return GNUNET_YES;
 718 }
 719
 720 /**
 721  * Reconnect callback: tries to reconnect again after a failer previous
 722  * reconnecttion
 723  * @param cls closure (mesh handle)
 724  * @param tc task context
 725  */
 726 static void
 727 reconnect_cbk (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 728 {
 729   struct GNUNET_MESH_Handle *h = cls;
 730
 731   if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
 732     return;
 733   reconnect (h);
 734 }
 735
 736
 737 /******************************************************************************/
 738 /***********************      RECEIVE HANDLERS     ****************************/
 739 /******************************************************************************/
 740
 741 /**
 742  * Process the new tunnel notification and add it to the tunnels in the handle
 743  *
 744  * @param h     The mesh handle
 745  * @param msg   A message with the details of the new incoming tunnel
 746  */
 747 static void
 748 process_tunnel_created (struct GNUNET_MESH_Handle *h,
 749                         const struct GNUNET_MESH_TunnelNotification *msg)
 750 {
 751   struct GNUNET_MESH_Tunnel *t;
 752   struct GNUNET_TRANSPORT_ATS_Information atsi;
 753   MESH_TunnelNumber tid;
 754
 755   tid = ntohl (msg->tunnel_id);
 756   if (tid <= GNUNET_MESH_LOCAL_TUNNEL_ID_CLI)
 757   {
 758     GNUNET_break (0);
 759     return;
 760   }
 761   t = create_tunnel (h, tid);
 762   t->owner = GNUNET_PEER_intern (&msg->peer);
 763   t->npeers = 1;
 764   t->peers = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer *));
 765   t->peers[0] = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer));
 766   t->peers[0]->t = t;
 767   t->peers[0]->connected = 1;
 768   t->peers[0]->id = t->owner;
 769   t->mesh = h;
 770   t->tid = tid;
 771   if (NULL != h->new_tunnel)
 772   {
 773     atsi.type = 0;
 774     atsi.value = 0;
 775     t->ctx = h->new_tunnel (h->cls, t, &msg->peer, &atsi);
 776   }
 777   return;
 778 }
 779
 780
 781 /**
 782  * Process the tunnel destroy notification and free associated resources
 783  *
 784  * @param h     The mesh handle
 785  * @param msg   A message with the details of the tunnel being destroyed
 786  */
 787 static void
 788 process_tunnel_destroy (struct GNUNET_MESH_Handle *h,
 789                         const struct GNUNET_MESH_TunnelMessage *msg)
 790 {
 791   struct GNUNET_MESH_Tunnel *t;
 792   MESH_TunnelNumber tid;
 793
 794   tid = ntohl (msg->tunnel_id);
 795   t = retrieve_tunnel (h, tid);
 796
 797   if (NULL == t)
 798   {
 799     GNUNET_break (0);
 800     return;
 801   }
 802   if (0 == t->owner)
 803   {
 804     GNUNET_break (0);
 805   }
 806
 807   destroy_tunnel (t);
 808   return;
 809 }
 810
 811
 812 /**
 813  * Process the new peer event and notify the upper level of it
 814  *
 815  * @param h     The mesh handle
 816  * @param msg   A message with the details of the peer event
 817  */
 818 static void
 819 process_peer_event (struct GNUNET_MESH_Handle *h,
 820                     const struct GNUNET_MESH_PeerControl *msg)
 821 {
 822   struct GNUNET_MESH_Tunnel *t;
 823   struct GNUNET_MESH_Peer *p;
 824   struct GNUNET_TRANSPORT_ATS_Information atsi;
 825   GNUNET_PEER_Id id;
 826   uint16_t size;
 827
 828   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: processig peer event\n");
 829   size = ntohs (msg->header.size);
 830   if (size != sizeof (struct GNUNET_MESH_PeerControl))
 831   {
 832     GNUNET_break (0);
 833     return;
 834   }
 835   t = retrieve_tunnel (h, ntohl (msg->tunnel_id));
 836   if (NULL == t)
 837   {
 838     GNUNET_break (0);
 839     return;
 840   }
 841   id = GNUNET_PEER_search (&msg->peer);
 842   if ((p = retrieve_peer (t, id)) == NULL)
 843     p = add_peer_to_tunnel (t, &msg->peer);
 844   if (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD == ntohs (msg->header.type))
 845   {
 846     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: adding peer\n");
 847     if (NULL != t->connect_handler)
 848     {
 849       atsi.type = 0;
 850       atsi.value = 0;
 851       t->connect_handler (t->cls, &msg->peer, &atsi);
 852     }
 853     p->connected = 1;
 854   }
 855   else
 856   {
 857     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: removing peer\n");
 858     if (NULL != t->disconnect_handler && p->connected)
 859     {
 860       t->disconnect_handler (t->cls, &msg->peer);
 861     }
 862     remove_peer_from_tunnel (p);
 863     GNUNET_free (p);
 864   }
 865   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: processing peer event END\n");
 866 }
 867
 868
 869 /**
 870  * Process the incoming data packets
 871  *
 872  * @param h         The mesh handle
 873  * @param message   A message encapsulating the data
 874  */
 875 static void
 876 process_incoming_data (struct GNUNET_MESH_Handle *h,
 877                        const struct GNUNET_MessageHeader *message)
 878 {
 879   const struct GNUNET_MessageHeader *payload;
 880   const struct GNUNET_MESH_MessageHandler *handler;
 881   const struct GNUNET_PeerIdentity *peer;
 882   struct GNUNET_MESH_Unicast *ucast;
 883   struct GNUNET_MESH_Multicast *mcast;
 884   struct GNUNET_MESH_ToOrigin *to_orig;
 885   struct GNUNET_MESH_Tunnel *t;
 886   unsigned int i;
 887   uint16_t type;
 888
 889
 890   type = ntohs (message->type);
 891   switch (type)
 892   {
 893   case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
 894     ucast = (struct GNUNET_MESH_Unicast *) message;
 895     t = retrieve_tunnel (h, ntohl (ucast->tid));
 896     payload = (struct GNUNET_MessageHeader *) &ucast[1];
 897     peer = &ucast->oid;
 898     break;
 899   case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
 900     mcast = (struct GNUNET_MESH_Multicast *) message;
 901     t = retrieve_tunnel (h, ntohl (mcast->tid));
 902     payload = (struct GNUNET_MessageHeader *) &mcast[1];
 903     peer = &mcast->oid;
 904     break;
 905   case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
 906     to_orig = (struct GNUNET_MESH_ToOrigin *) message;
 907     t = retrieve_tunnel (h, ntohl (to_orig->tid));
 908     payload = (struct GNUNET_MessageHeader *) &to_orig[1];
 909     peer = &to_orig->sender;
 910     break;
 911   default:
 912     GNUNET_break (0);
 913     return;
 914   }
 915   if (NULL == t)
 916   {
 917     GNUNET_break (0);
 918     return;
 919   }
 920   type = ntohs (payload->type);
 921   for (i = 0; i < h->n_handlers; i++)
 922   {
 923     handler = &h->message_handlers[i];
 924     if (handler->type == type)
 925     {
 926       struct GNUNET_TRANSPORT_ATS_Information atsi;
 927
 928       atsi.type = 0;
 929       atsi.value = 0;
 930       if (GNUNET_OK !=
 931           handler->callback (h->cls, t, &t->ctx, peer, payload, &atsi))
 932       {
 933         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 934                     "MESH: callback caused disconnection\n");
 935         GNUNET_MESH_disconnect (h);
 936         return;
 937       }
 938 #if DEBUG
 939       else
 940       {
 941         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 942                     "MESH: callback completed successfully\n");
 943
 944       }
 945 #endif
 946     }
 947   }
 948 }
 949
 950
 951 /**
 952  * Function to process all messages received from the service
 953  *
 954  * @param cls closure
 955  * @param msg message received, NULL on timeout or fatal error
 956  */
 957 static void
 958 msg_received (void *cls, const struct GNUNET_MessageHeader *msg)
 959 {
 960   struct GNUNET_MESH_Handle *h = cls;
 961
 962   if (msg == NULL)
 963   {
 964     reconnect (h);
 965     return;
 966   }
 967   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 968               "mesh: received a message type %hu from MESH\n",
 969               ntohs (msg->type));
 970   switch (ntohs (msg->type))
 971   {
 972     /* Notify of a new incoming tunnel */
 973   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE:
 974     process_tunnel_created (h, (struct GNUNET_MESH_TunnelNotification *) msg);
 975     break;
 976     /* Notify of a tunnel disconnection */
 977   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY:
 978     process_tunnel_destroy (h, (struct GNUNET_MESH_TunnelMessage *) msg);
 979     break;
 980     /* Notify of a new peer or a peer disconnect in the tunnel */
 981   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD:
 982   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL:
 983     process_peer_event (h, (struct GNUNET_MESH_PeerControl *) msg);
 984     break;
 985     /* Notify of a new data packet in the tunnel */
 986   case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
 987   case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
 988   case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
 989     process_incoming_data (h, msg);
 990     break;
 991     /* We shouldn't get any other packages, log and ignore */
 992   default:
 993     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 994                 "MESH: unsolicited message form service (type %d)\n",
 995                 ntohs (msg->type));
 996   }
 997   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: message processed\n");
 998   GNUNET_CLIENT_receive (h->client, &msg_received, h,
 999                          GNUNET_TIME_UNIT_FOREVER_REL);
1000 }
1001
1002
1003 /******************************************************************************/
1004 /************************       SEND FUNCTIONS     ****************************/
1005 /******************************************************************************/
1006
1007 /**
1008  * Function called to send a message to the service.
1009  * "buf" will be NULL and "size" zero if the socket was closed for writing in
1010  * the meantime.
1011  *
1012  * @param cls closure, the mesh handle
1013  * @param size number of bytes available in buf
1014  * @param buf where the callee should write the connect message
1015  * @return number of bytes written to buf
1016  */
1017 static size_t
1018 send_callback (void *cls, size_t size, void *buf)
1019 {
1020   struct GNUNET_MESH_Handle *h = cls;
1021   struct GNUNET_MESH_TransmitHandle *th;
1022   char *cbuf = buf;
1023   size_t tsize;
1024   size_t psize;
1025
1026   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Send packet() Buffer %u\n", size);
1027   h->th = NULL;
1028   if ((0 == size) || (NULL == buf))
1029   {
1030     reconnect (h);
1031     return 0;
1032   }
1033   tsize = 0;
1034   while ((NULL != (th = h->th_head)) && (size >= th->size))
1035   {
1036 #if DEBUG
1037     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:     type: %u\n",
1038                 ntohs (((struct GNUNET_MessageHeader *) &th[1])->type));
1039     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:     size: %u\n",
1040                 ntohs (((struct GNUNET_MessageHeader *) &th[1])->size));
1041 #endif
1042     if (NULL != th->notify)
1043     {
1044       if (th->target == 0)
1045       {
1046         /* multicast */
1047         struct GNUNET_MESH_Multicast mc;
1048
1049         GNUNET_assert (size >= sizeof (mc) + th->size);
1050         psize =
1051             th->notify (th->notify_cls, size - sizeof (mc), &cbuf[sizeof (mc)]);
1052         if (psize > 0)
1053         {
1054           mc.header.size = htons (sizeof (mc) + th->size);
1055           mc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_MULTICAST);
1056           mc.tid = htonl (th->tunnel->tid);
1057           memset (&mc.oid, 0, sizeof (struct GNUNET_PeerIdentity));
1058           memcpy (cbuf, &mc, sizeof (mc));
1059           psize = th->size + sizeof (mc);
1060         }
1061       }
1062       else
1063       {
1064         /* unicast */
1065         struct GNUNET_MESH_Unicast uc;
1066
1067         GNUNET_assert (size >= sizeof (uc) + th->size);
1068         psize =
1069             th->notify (th->notify_cls, size - sizeof (uc), &cbuf[sizeof (uc)]);
1070         if (psize > 0)
1071         {
1072           uc.header.size = htons (sizeof (uc) + th->size);
1073           uc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_UNICAST);
1074           uc.tid = htonl (th->tunnel->tid);
1075           memset (&uc.oid, 0, sizeof (struct GNUNET_PeerIdentity));
1076           GNUNET_PEER_resolve (th->target, &uc.destination);
1077           memcpy (cbuf, &uc, sizeof (uc));
1078           psize = th->size + sizeof (uc);
1079         }
1080       }
1081     }
1082     else
1083     {
1084       memcpy (cbuf, &th[1], th->size);
1085       psize = th->size;
1086     }
1087     if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1088       GNUNET_SCHEDULER_cancel (th->timeout_task);
1089     GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th);
1090     GNUNET_free (th);
1091     cbuf += psize;
1092     size -= psize;
1093     tsize += psize;
1094   }
1095   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:   total size: %u\n", tsize);
1096   if (NULL != (th = h->th_head))
1097   {
1098     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh:   next size: %u\n", th->size);
1099     h->th =
1100         GNUNET_CLIENT_notify_transmit_ready (h->client, th->size,
1101                                              GNUNET_TIME_UNIT_FOREVER_REL,
1102                                              GNUNET_YES, &send_callback, h);
1103   }
1104   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Send packet() END\n");
1105   if (GNUNET_NO == h->in_receive)
1106   {
1107     h->in_receive = GNUNET_YES;
1108     GNUNET_CLIENT_receive (h->client, &msg_received, h,
1109                            GNUNET_TIME_UNIT_FOREVER_REL);
1110   }
1111   return tsize;
1112 }
1113
1114
1115 /**
1116  * Auxiliary function to send an already constructed packet to the service.
1117  * Takes care of creating a new queue element, copying the message and
1118  * calling the tmt_rdy function if necessary.
1119  * @param h mesh handle
1120  * @param msg message to transmit
1121  */
1122 static void
1123 send_packet (struct GNUNET_MESH_Handle *h,
1124              const struct GNUNET_MessageHeader *msg)
1125 {
1126   struct GNUNET_MESH_TransmitHandle *th;
1127   size_t msize;
1128
1129   msize = ntohs (msg->size);
1130   th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle) + msize);
1131   th->priority = UINT32_MAX;
1132   th->timeout = GNUNET_TIME_UNIT_FOREVER_ABS;
1133   th->size = msize;
1134   memcpy (&th[1], msg, msize);
1135   add_to_queue (h, th);
1136   if (NULL != h->th)
1137     return;
1138   h->th =
1139       GNUNET_CLIENT_notify_transmit_ready (h->client, msize,
1140                                            GNUNET_TIME_UNIT_FOREVER_REL,
1141                                            GNUNET_YES, &send_callback, h);
1142 }
1143
1144
1145 /******************************************************************************/
1146 /**********************      API CALL DEFINITIONS     *************************/
1147 /******************************************************************************/
1148
1149 /**
1150  * Connect to the mesh service.
1151  *
1152  * @param cfg configuration to use
1153  * @param queue_size size of the data message queue, shared among all tunnels
1154  *                   (each tunnel is guaranteed to accept at least one message,
1155  *                    no matter what is the status of other tunnels)
1156  * @param cls closure for the various callbacks that follow
1157  *            (including handlers in the handlers array)
1158  * @param new_tunnel function called when an *inbound* tunnel is created
1159  * @param cleaner function called when an *inbound* tunnel is destroyed
1160  * @param handlers callbacks for messages we care about, NULL-terminated
1161  *                note that the mesh is allowed to drop notifications about
1162  *                inbound messages if the client does not process them fast
1163  *                enough (for this notification type, a bounded queue is used)
1164  * @param stypes list of the applications that this client claims to provide
1165  * @return handle to the mesh service NULL on error
1166  *         (in this case, init is never called)
1167  */
1168 struct GNUNET_MESH_Handle *
1169 GNUNET_MESH_connect (const struct GNUNET_CONFIGURATION_Handle *cfg,
1170                      unsigned int queue_size, void *cls,
1171                      GNUNET_MESH_InboundTunnelNotificationHandler new_tunnel,
1172                      GNUNET_MESH_TunnelEndHandler cleaner,
1173                      const struct GNUNET_MESH_MessageHandler *handlers,
1174                      const GNUNET_MESH_ApplicationType *stypes)
1175 {
1176   struct GNUNET_MESH_Handle *h;
1177
1178   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: GNUNET_MESH_connect()\n");
1179   h = GNUNET_malloc (sizeof (struct GNUNET_MESH_Handle));
1180   h->cfg = cfg;
1181   h->max_queue_size = queue_size;
1182   h->new_tunnel = new_tunnel;
1183   h->cleaner = cleaner;
1184   h->client = GNUNET_CLIENT_connect ("mesh", cfg);
1185   if (h->client == NULL)
1186   {
1187     GNUNET_break (0);
1188     GNUNET_free (h);
1189     return NULL;
1190   }
1191   h->cls = cls;
1192   /* FIXME memdup? */
1193   h->applications = stypes;
1194   h->message_handlers = handlers;
1195   h->next_tid = GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
1196   h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS;
1197
1198   /* count handlers and apps, calculate size */
1199   for (h->n_applications = 0; stypes[h->n_applications]; h->n_applications++) ;
1200   for (h->n_handlers = 0; handlers[h->n_handlers].type; h->n_handlers++) ;
1201   send_connect(h);
1202   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: GNUNET_MESH_connect() END\n");
1203   return h;
1204 }
1205
1206
1207 /**
1208  * Disconnect from the mesh service.
1209  *
1210  * @param handle connection to mesh to disconnect
1211  */
1212 void
1213 GNUNET_MESH_disconnect (struct GNUNET_MESH_Handle *handle)
1214 {
1215   struct GNUNET_MESH_Tunnel *t;
1216   struct GNUNET_MESH_Tunnel *aux;
1217
1218   t = handle->tunnels_head;
1219   while (NULL != t)
1220   {
1221     aux = t->next;
1222     destroy_tunnel (t);
1223     t = aux;
1224   }
1225   if (NULL != handle->th)
1226   {
1227     GNUNET_CLIENT_notify_transmit_ready_cancel (handle->th);
1228   }
1229   if (NULL != handle->client)
1230   {
1231     GNUNET_CLIENT_disconnect (handle->client, GNUNET_NO);
1232   }
1233   GNUNET_free (handle);
1234 }
1235
1236
1237 /**
1238  * Create a new tunnel (we're initiator and will be allowed to add/remove peers
1239  * and to broadcast).
1240  *
1241  * @param h mesh handle
1242  * @param tunnel_ctx client's tunnel context to associate with the tunnel
1243  * @param connect_handler function to call when peers are actually connected
1244  * @param disconnect_handler function to call when peers are disconnected
1245  * @param handler_cls closure for connect/disconnect handlers
1246  */
1247 struct GNUNET_MESH_Tunnel *
1248 GNUNET_MESH_tunnel_create (struct GNUNET_MESH_Handle *h, void *tunnel_ctx,
1249                            GNUNET_MESH_PeerConnectHandler connect_handler,
1250                            GNUNET_MESH_PeerDisconnectHandler disconnect_handler,
1251                            void *handler_cls)
1252 {
1253   struct GNUNET_MESH_Tunnel *t;
1254   struct GNUNET_MESH_TunnelMessage msg;
1255
1256   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Creating new tunnel\n");
1257   t = create_tunnel (h, 0);
1258   t->connect_handler = connect_handler;
1259   t->disconnect_handler = disconnect_handler;
1260   t->cls = handler_cls;
1261   t->ctx = tunnel_ctx;
1262   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
1263   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1264   msg.tunnel_id = htonl (t->tid);
1265   send_packet (h, &msg.header);
1266   return t;
1267 }
1268
1269
1270 /**
1271  * Destroy an existing tunnel.
1272  *
1273  * @param tun tunnel handle
1274  */
1275 void
1276 GNUNET_MESH_tunnel_destroy (struct GNUNET_MESH_Tunnel *tunnel)
1277 {
1278   struct GNUNET_MESH_Handle *h;
1279   struct GNUNET_MESH_TunnelMessage msg;
1280
1281   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Destroying tunnel\n");
1282   h = tunnel->mesh;
1283
1284   if (0 != tunnel->owner)
1285     GNUNET_PEER_change_rc (tunnel->owner, -1);
1286
1287   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY);
1288   msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1289   msg.tunnel_id = htonl (tunnel->tid);
1290   destroy_tunnel (tunnel);
1291   send_packet (h, &msg.header);
1292 }
1293
1294
1295 /**
1296  * Request that a peer should be added to the tunnel.  The existing
1297  * connect handler will be called ONCE with either success or failure.
1298  * This function should NOT be called again with the same peer before the
1299  * connect handler is called.
1300  *
1301  * @param tunnel handle to existing tunnel
1302  * @param peer peer to add
1303  */
1304 void
1305 GNUNET_MESH_peer_request_connect_add (struct GNUNET_MESH_Tunnel *tunnel,
1306                                       const struct GNUNET_PeerIdentity *peer)
1307 {
1308   struct GNUNET_MESH_PeerControl msg;
1309   GNUNET_PEER_Id peer_id;
1310   unsigned int i;
1311
1312   peer_id = GNUNET_PEER_intern (peer);
1313   for (i = 0; i < tunnel->npeers; i++)
1314   {
1315     if (tunnel->peers[i]->id == peer_id)
1316     {
1317       GNUNET_PEER_change_rc (peer_id, -1);
1318       GNUNET_break (0);
1319       return;
1320     }
1321   }
1322   if (NULL == add_peer_to_tunnel (tunnel, peer))
1323     return;
1324
1325   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1326   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
1327   msg.tunnel_id = htonl (tunnel->tid);
1328   msg.peer = *peer;
1329   send_packet (tunnel->mesh, &msg.header);
1330
1331   return;
1332 }
1333
1334
1335 /**
1336  * Request that a peer should be removed from the tunnel.  The existing
1337  * disconnect handler will be called ONCE if we were connected.
1338  *
1339  * @param tunnel handle to existing tunnel
1340  * @param peer peer to remove
1341  */
1342 void
1343 GNUNET_MESH_peer_request_connect_del (struct GNUNET_MESH_Tunnel *tunnel,
1344                                       const struct GNUNET_PeerIdentity *peer)
1345 {
1346   struct GNUNET_MESH_PeerControl msg;
1347   GNUNET_PEER_Id peer_id;
1348   unsigned int i;
1349
1350   peer_id = GNUNET_PEER_search (peer);
1351   if (0 == peer_id)
1352   {
1353     GNUNET_break (0);
1354     return;
1355   }
1356   for (i = 0; i < tunnel->npeers; i++)
1357     if (tunnel->peers[i]->id == peer_id)
1358       break;
1359   if (i == tunnel->npeers)
1360   {
1361     GNUNET_break (0);
1362     return;
1363   }
1364   if (NULL != tunnel->disconnect_handler && tunnel->peers[i]->connected == 1)
1365     tunnel->disconnect_handler (tunnel->cls, peer);
1366   GNUNET_PEER_change_rc (peer_id, -1);
1367   GNUNET_free (tunnel->peers[i]);
1368   tunnel->peers[i] = tunnel->peers[tunnel->npeers - 1];
1369   GNUNET_array_grow (tunnel->peers, tunnel->npeers, tunnel->npeers - 1);
1370
1371   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1372   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL);
1373   msg.tunnel_id = htonl (tunnel->tid);
1374   memcpy (&msg.peer, peer, sizeof (struct GNUNET_PeerIdentity));
1375   send_packet (tunnel->mesh, &msg.header);
1376 }
1377
1378
1379 /**
1380  * Request that the mesh should try to connect to a peer supporting the given
1381  * message type.
1382  *
1383  * @param tunnel handle to existing tunnel
1384  * @param app_type application type that must be supported by the peer (MESH
1385  *                 should discover peer in proximity handling this type)
1386  */
1387 void
1388 GNUNET_MESH_peer_request_connect_by_type (struct GNUNET_MESH_Tunnel *tunnel,
1389                                           GNUNET_MESH_ApplicationType app_type)
1390 {
1391   struct GNUNET_MESH_ConnectPeerByType msg;
1392
1393   GNUNET_array_append (tunnel->apps, tunnel->napps, app_type);
1394
1395   msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
1396   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE);
1397   msg.tunnel_id = htonl (tunnel->tid);
1398   msg.type = htonl (app_type);
1399   send_packet (tunnel->mesh, &msg.header);
1400 }
1401
1402
1403 /**
1404  * Ask the mesh to call "notify" once it is ready to transmit the
1405  * given number of bytes to the specified "target".  If we are not yet
1406  * connected to the specified peer, a call to this function will cause
1407  * us to try to establish a connection.
1408  *
1409  * @param tunnel tunnel to use for transmission
1410  * @param cork is corking allowed for this transmission?
1411  * @param priority how important is the message?
1412  * @param maxdelay how long can the message wait?
1413  * @param target destination for the message,
1414  *               NULL for multicast to all tunnel targets
1415  * @param notify_size how many bytes of buffer space does notify want?
1416  * @param notify function to call when buffer space is available;
1417  *        will be called with NULL on timeout or if the overall queue
1418  *        for this peer is larger than queue_size and this is currently
1419  *        the message with the lowest priority
1420  * @param notify_cls closure for notify
1421  * @return non-NULL if the notify callback was queued,
1422  *         NULL if we can not even queue the request (insufficient
1423  *         memory); if NULL is returned, "notify" will NOT be called.
1424  */
1425 struct GNUNET_MESH_TransmitHandle *
1426 GNUNET_MESH_notify_transmit_ready (struct GNUNET_MESH_Tunnel *tunnel, int cork,
1427                                    uint32_t priority,
1428                                    struct GNUNET_TIME_Relative maxdelay,
1429                                    const struct GNUNET_PeerIdentity *target,
1430                                    size_t notify_size,
1431                                    GNUNET_CONNECTION_TransmitReadyNotify notify,
1432                                    void *notify_cls)
1433 {
1434   struct GNUNET_MESH_TransmitHandle *th;
1435   struct GNUNET_MESH_TransmitHandle *least_priority_th;
1436   uint32_t least_priority;
1437   size_t overhead;
1438
1439   GNUNET_assert (NULL != notify);
1440   if (tunnel->mesh->npackets >= tunnel->mesh->max_queue_size &&
1441       tunnel->npackets > 0)
1442   {
1443     /* queue full */
1444     if (0 == priority)
1445       return NULL;
1446     th = tunnel->mesh->th_tail;
1447     least_priority = priority;
1448     least_priority_th = NULL;
1449     while (NULL != th)
1450     {
1451       if (th->priority < least_priority && th->tunnel->npackets > 1)
1452       {
1453         least_priority_th = th;
1454         least_priority = th->priority;
1455       }
1456       th = th->prev;
1457     }
1458     if (NULL == least_priority_th)
1459       return NULL;
1460     /* Can't be a control message */
1461     GNUNET_assert (NULL != least_priority_th->notify);
1462     least_priority_th->notify (notify_cls, 0, NULL);
1463     least_priority_th->tunnel->npackets--;
1464     tunnel->mesh->npackets--;
1465     GNUNET_CONTAINER_DLL_remove (tunnel->mesh->th_head, tunnel->mesh->th_tail,
1466                                  least_priority_th);
1467     if (GNUNET_SCHEDULER_NO_TASK != least_priority_th->timeout_task)
1468       GNUNET_SCHEDULER_cancel (least_priority_th->timeout_task);
1469     GNUNET_free (least_priority_th);
1470   }
1471   tunnel->npackets++;
1472   tunnel->mesh->npackets++;
1473   th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle));
1474   th->tunnel = tunnel;
1475   th->priority = priority;
1476   th->timeout = GNUNET_TIME_relative_to_absolute (maxdelay);
1477   th->target = GNUNET_PEER_intern (target);
1478   if (NULL == target)
1479     overhead = sizeof (struct GNUNET_MESH_Multicast);
1480   else
1481     overhead = sizeof (struct GNUNET_MESH_Unicast);
1482   th->size = notify_size + overhead;
1483   th->notify = notify;
1484   th->notify_cls = notify_cls;
1485   add_to_queue (tunnel->mesh, th);
1486   return th;
1487 }
1488
1489
1490 /**
1491  * Cancel the specified transmission-ready notification.
1492  *
1493  * @param th handle that was returned by "notify_transmit_ready".
1494  */
1495 void
1496 GNUNET_MESH_notify_transmit_ready_cancel (struct GNUNET_MESH_TransmitHandle *th)
1497 {
1498   struct GNUNET_MESH_Handle *mesh;
1499
1500   mesh = th->tunnel->mesh;
1501   if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1502     GNUNET_SCHEDULER_cancel (th->timeout_task);
1503   GNUNET_CONTAINER_DLL_remove (mesh->th_head, mesh->th_tail, th);
1504   GNUNET_free (th);
1505   if ((NULL == mesh->th_head) && (NULL != mesh->th))
1506   {
1507     /* queue empty, no point in asking for transmission */
1508     GNUNET_CLIENT_notify_transmit_ready_cancel (mesh->th);
1509     mesh->th = NULL;
1510   }
1511 }
1512
1513
1514 #if 0                           /* keep Emacsens' auto-indent happy */
1515 {
1516 #endif
1517 #ifdef __cplusplus
1518 }
1519 #endif