src/mesh/mesh_api.c

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