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