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 == th->timeout.abs_value)
 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,
 688          "Next retry in %s\n",
 689          GNUNET_STRINGS_relative_time_to_string (h->reconnect_time,
 690                                                  GNUNET_NO));
 691     GNUNET_break (0);
 692     return GNUNET_NO;
 693   }
 694   else
 695   {
 696     h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS;
 697   }
 698   send_connect (h);
 699   /* Rebuild all channels */
 700   for (ch = h->channels_head; NULL != ch; ch = ch->next)
 701   {
 702     struct GNUNET_MESH_ChannelMessage tmsg;
 703     uint32_t options;
 704
 705     if (ch->chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV)
 706     {
 707       /* Channel was created by service (incoming channel) */
 708       /* TODO: Notify service of missing channel, to request
 709        * creator to recreate path (find a path to him via DHT?)
 710        */
 711       continue;
 712     }
 713         ch->allow_send = GNUNET_NO;
 714     tmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CHANNEL_CREATE);
 715     tmsg.header.size = htons (sizeof (struct GNUNET_MESH_ChannelMessage));
 716     tmsg.channel_id = htonl (ch->chid);
 717     tmsg.port = htonl (ch->port);
 718     GNUNET_PEER_resolve (ch->peer, &tmsg.peer);
 719
 720     options = 0;
 721     if (GNUNET_YES == ch->nobuffer)
 722       options |= GNUNET_MESH_OPTION_NOBUFFER;
 723
 724     if (GNUNET_YES == ch->reliable)
 725       options |= GNUNET_MESH_OPTION_RELIABLE;
 726
 727     tmsg.opt = htonl (options);
 728     send_packet (h, &tmsg.header, ch);
 729   }
 730   return GNUNET_YES;
 731 }
 732
 733 /**
 734  * Reconnect callback: tries to reconnect again after a failer previous
 735  * reconnecttion
 736  * @param cls closure (mesh handle)
 737  * @param tc task context
 738  */
 739 static void
 740 reconnect_cbk (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 741 {
 742   struct GNUNET_MESH_Handle *h = cls;
 743
 744   h->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
 745   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 746     return;
 747   do_reconnect (h);
 748 }
 749
 750
 751 /**
 752  * Reconnect to the service, retransmit all infomation to try to restore the
 753  * original state.
 754  *
 755  * @param h handle to the mesh
 756  *
 757  * @return GNUNET_YES in case of sucess, GNUNET_NO otherwise (service down...)
 758  */
 759 static void
 760 reconnect (struct GNUNET_MESH_Handle *h)
 761 {
 762   LOG (GNUNET_ERROR_TYPE_DEBUG, "Requested RECONNECT\n");
 763   h->in_receive = GNUNET_NO;
 764   if (GNUNET_SCHEDULER_NO_TASK == h->reconnect_task)
 765     h->reconnect_task = GNUNET_SCHEDULER_add_delayed (h->reconnect_time,
 766                                                       &reconnect_cbk, h);
 767 }
 768
 769
 770 /******************************************************************************/
 771 /***********************      RECEIVE HANDLERS     ****************************/
 772 /******************************************************************************/
 773
 774 /**
 775  * Process the new channel notification and add it to the channels in the handle
 776  *
 777  * @param h     The mesh handle
 778  * @param msg   A message with the details of the new incoming channel
 779  */
 780 static void
 781 process_channel_created (struct GNUNET_MESH_Handle *h,
 782                         const struct GNUNET_MESH_ChannelMessage *msg)
 783 {
 784   struct GNUNET_MESH_Channel *ch;
 785   MESH_ChannelNumber chid;
 786   uint32_t port;
 787
 788   chid = ntohl (msg->channel_id);
 789   port = ntohl (msg->port);
 790   LOG (GNUNET_ERROR_TYPE_DEBUG, "Creating incoming channel %X:%u\n", chid, port);
 791   if (chid < GNUNET_MESH_LOCAL_CHANNEL_ID_SERV)
 792   {
 793     GNUNET_break (0);
 794     return;
 795   }
 796   if (NULL != h->new_channel)
 797   {
 798     ch = create_channel (h, chid);
 799     ch->allow_send = GNUNET_NO;
 800     ch->peer = GNUNET_PEER_intern (&msg->peer);
 801     ch->mesh = h;
 802     ch->chid = chid;
 803     ch->port = port;
 804     if (0 != (msg->opt & GNUNET_MESH_OPTION_NOBUFFER))
 805       ch->nobuffer = GNUNET_YES;
 806     else
 807       ch->nobuffer = GNUNET_NO;
 808
 809     if (0 != (msg->opt & GNUNET_MESH_OPTION_RELIABLE))
 810       ch->reliable = GNUNET_YES;
 811     else
 812       ch->reliable = GNUNET_NO;
 813
 814     if (GNUNET_YES == ch->reliable &&
 815         0 != (msg->opt & GNUNET_MESH_OPTION_OOORDER))
 816       ch->ooorder = GNUNET_YES;
 817     else
 818       ch->ooorder = GNUNET_NO;
 819
 820     LOG (GNUNET_ERROR_TYPE_DEBUG, "  created channel %p\n", t);
 821     ch->ctx = h->new_channel (h->cls, ch, &msg->peer, ch->port);
 822     LOG (GNUNET_ERROR_TYPE_DEBUG, "User notified\n");
 823   }
 824   else
 825   {
 826     struct GNUNET_MESH_ChannelMessage d_msg;
 827
 828     LOG (GNUNET_ERROR_TYPE_DEBUG, "No handler for incoming channels\n");
 829
 830     d_msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CHANNEL_DESTROY);
 831     d_msg.header.size = htons (sizeof (struct GNUNET_MESH_ChannelMessage));
 832     d_msg.channel_id = msg->channel_id;
 833     memset (&d_msg.peer, 0, sizeof (struct GNUNET_PeerIdentity));
 834     d_msg.port = 0;
 835     d_msg.opt = 0;
 836
 837     send_packet (h, &d_msg.header, NULL);
 838   }
 839   return;
 840 }
 841
 842
 843 /**
 844  * Process the channel destroy notification and free associated resources
 845  *
 846  * @param h     The mesh handle
 847  * @param msg   A message with the details of the channel being destroyed
 848  */
 849 static void
 850 process_channel_destroy (struct GNUNET_MESH_Handle *h,
 851                          const struct GNUNET_MESH_ChannelMessage *msg)
 852 {
 853   struct GNUNET_MESH_Channel *ch;
 854   MESH_ChannelNumber chid;
 855
 856   LOG (GNUNET_ERROR_TYPE_DEBUG, "Destroying channel from service\n");
 857   chid = ntohl (msg->channel_id);
 858   ch = retrieve_channel (h, chid);
 859
 860   if (NULL == ch)
 861   {
 862     LOG (GNUNET_ERROR_TYPE_DEBUG, "channel %X unknown\n", chid);
 863     return;
 864   }
 865   LOG (GNUNET_ERROR_TYPE_DEBUG, "channel %X destroyed\n", ch->chid);
 866   destroy_channel (ch, GNUNET_YES);
 867 }
 868
 869
 870 /**
 871  * Process the incoming data packets, call appropriate handlers.
 872  *
 873  * @param h         The mesh handle
 874  * @param message   A message encapsulating the data
 875  */
 876 static void
 877 process_incoming_data (struct GNUNET_MESH_Handle *h,
 878                        const struct GNUNET_MessageHeader *message)
 879 {
 880   const struct GNUNET_MessageHeader *payload;
 881   const struct GNUNET_MESH_MessageHandler *handler;
 882   struct GNUNET_MESH_LocalData *dmsg;
 883   struct GNUNET_MESH_Channel *ch;
 884   unsigned int i;
 885   uint16_t type;
 886
 887   LOG (GNUNET_ERROR_TYPE_DEBUG, "Got a data message!\n");
 888
 889   dmsg = (struct GNUNET_MESH_LocalData *) message;
 890
 891   ch = retrieve_channel (h, ntohl (dmsg->chid));
 892   payload = (struct GNUNET_MessageHeader *) &dmsg[1];
 893   LOG (GNUNET_ERROR_TYPE_DEBUG, "  %s data on channel %s [%X]\n",
 894        ch->chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV ? "fwd" : "bck",
 895        GNUNET_i2s (GNUNET_PEER_resolve2 (ch->peer)), ntohl (dmsg->chid));
 896   if (NULL == ch)
 897   {
 898     /* Channel was ignored/destroyed, probably service didn't get it yet */
 899     LOG (GNUNET_ERROR_TYPE_DEBUG, "  ignored!\n");
 900     return;
 901   }
 902   type = ntohs (payload->type);
 903   LOG (GNUNET_ERROR_TYPE_DEBUG, "  payload type %u\n", type);
 904   for (i = 0; i < h->n_handlers; i++)
 905   {
 906     handler = &h->message_handlers[i];
 907     LOG (GNUNET_ERROR_TYPE_DEBUG,
 908          "    checking handler for type %u\n",
 909          handler->type);
 910     if (handler->type == type)
 911     {
 912       if (GNUNET_OK !=
 913           handler->callback (h->cls, ch, &ch->ctx, payload))
 914       {
 915         LOG (GNUNET_ERROR_TYPE_DEBUG, "callback caused disconnection\n");
 916         GNUNET_MESH_channel_destroy (ch);
 917         return;
 918       }
 919       else
 920       {
 921         LOG (GNUNET_ERROR_TYPE_DEBUG,
 922              "callback completed successfully\n");
 923       }
 924     }
 925   }
 926 }
 927
 928
 929 /**
 930  * Process a local ACK message, enabling the client to send
 931  * more data to the service.
 932  *
 933  * @param h Mesh handle.
 934  * @param message Message itself.
 935  */
 936 static void
 937 process_ack (struct GNUNET_MESH_Handle *h,
 938              const struct GNUNET_MessageHeader *message)
 939 {
 940   struct GNUNET_MESH_LocalAck *msg;
 941   struct GNUNET_MESH_Channel *ch;
 942   MESH_ChannelNumber chid;
 943
 944   LOG (GNUNET_ERROR_TYPE_DEBUG, "Got an ACK!\n");
 945   h->acks_recv++;
 946   msg = (struct GNUNET_MESH_LocalAck *) message;
 947   chid = ntohl (msg->channel_id);
 948     ch = retrieve_channel (h, chid);
 949   if (NULL == ch)
 950   {
 951     LOG (GNUNET_ERROR_TYPE_WARNING, "ACK on unknown channel %X\n", chid);
 952     return;
 953   }
 954   LOG (GNUNET_ERROR_TYPE_DEBUG, "  on channel %X!\n", ch->chid);
 955   ch->allow_send = GNUNET_YES;
 956   if (NULL == h->th && 0 < ch->packet_size)
 957   {
 958     LOG (GNUNET_ERROR_TYPE_DEBUG, "  tmt rdy was NULL, requesting!\n");
 959     h->th =
 960         GNUNET_CLIENT_notify_transmit_ready (h->client, ch->packet_size,
 961                                              GNUNET_TIME_UNIT_FOREVER_REL,
 962                                              GNUNET_YES, &send_callback, h);
 963   }
 964 }
 965
 966
 967 /**
 968  * Process a local reply about info on all channels, pass info to the user.
 969  *
 970  * @param h Mesh handle.
 971  * @param message Message itself.
 972  */
 973 static void
 974 process_get_channels (struct GNUNET_MESH_Handle *h,
 975                      const struct GNUNET_MessageHeader *message)
 976 {
 977   struct GNUNET_MESH_LocalMonitor *msg;
 978
 979   GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Get Channels messasge received\n");
 980
 981   if (NULL == h->channels_cb)
 982   {
 983     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "  ignored\n");
 984     return;
 985   }
 986
 987   msg = (struct GNUNET_MESH_LocalMonitor *) message;
 988   if (ntohs (message->size) !=
 989       (sizeof (struct GNUNET_MESH_LocalMonitor) +
 990        sizeof (struct GNUNET_PeerIdentity)))
 991   {
 992     GNUNET_break_op (0);
 993     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 994                 "Get channels message: size %hu - expected %u\n",
 995                 ntohs (message->size),
 996                 sizeof (struct GNUNET_MESH_LocalMonitor));
 997     return;
 998   }
 999   h->channels_cb (h->channels_cls,
1000                   ntohl (msg->channel_id),
1001                   &msg->owner,
1002                   &msg->destination);
1003 }
1004
1005
1006
1007 /**
1008  * Process a local monitor_channel reply, pass info to the user.
1009  *
1010  * @param h Mesh handle.
1011  * @param message Message itself.
1012  */
1013 static void
1014 process_show_channel (struct GNUNET_MESH_Handle *h,
1015                      const struct GNUNET_MessageHeader *message)
1016 {
1017   struct GNUNET_MESH_LocalMonitor *msg;
1018   size_t esize;
1019
1020   GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Show Channel messasge received\n");
1021
1022   if (NULL == h->channel_cb)
1023   {
1024     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "  ignored\n");
1025     return;
1026   }
1027
1028   /* Verify message sanity */
1029   msg = (struct GNUNET_MESH_LocalMonitor *) message;
1030   esize = sizeof (struct GNUNET_MESH_LocalMonitor);
1031   if (ntohs (message->size) != esize)
1032   {
1033     GNUNET_break_op (0);
1034     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1035                 "Show channel message: size %hu - expected %u\n",
1036                 ntohs (message->size),
1037                 esize);
1038
1039     h->channel_cb (h->channel_cls, NULL, NULL);
1040     h->channel_cb = NULL;
1041     h->channel_cls = NULL;
1042
1043     return;
1044   }
1045
1046   h->channel_cb (h->channel_cls,
1047                  &msg->destination,
1048                  &msg->owner);
1049 }
1050
1051
1052 /**
1053  * Function to process all messages received from the service
1054  *
1055  * @param cls closure
1056  * @param msg message received, NULL on timeout or fatal error
1057  */
1058 static void
1059 msg_received (void *cls, const struct GNUNET_MessageHeader *msg)
1060 {
1061   struct GNUNET_MESH_Handle *h = cls;
1062   uint16_t type;
1063
1064   if (msg == NULL)
1065   {
1066     LOG (GNUNET_ERROR_TYPE_DEBUG,
1067          "Mesh service disconnected, reconnecting\n", h);
1068     reconnect (h);
1069     return;
1070   }
1071   type = ntohs (msg->type);
1072   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n");
1073   LOG (GNUNET_ERROR_TYPE_DEBUG, "Received a message: %s\n",
1074        GNUNET_MESH_DEBUG_M2S (type));
1075   switch (type)
1076   {
1077     /* Notify of a new incoming channel */
1078   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_CHANNEL_CREATE:
1079     process_channel_created (h, (struct GNUNET_MESH_ChannelMessage *) msg);
1080     break;
1081     /* Notify of a channel disconnection */
1082   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_CHANNEL_DESTROY:
1083     process_channel_destroy (h, (struct GNUNET_MESH_ChannelMessage *) msg);
1084     break;
1085     /* Notify of a new data packet in the channel */
1086   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA:
1087     process_incoming_data (h, msg);
1088     break;
1089   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK:
1090     process_ack (h, msg);
1091     break;
1092   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_CHANNELS:
1093     process_get_channels (h, msg);
1094     break;
1095   case GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_CHANNEL:
1096     process_show_channel (h, msg);
1097     break;
1098   default:
1099     /* We shouldn't get any other packages, log and ignore */
1100     LOG (GNUNET_ERROR_TYPE_WARNING,
1101          "unsolicited message form service (type %s)\n",
1102          GNUNET_MESH_DEBUG_M2S (ntohs (msg->type)));
1103   }
1104   LOG (GNUNET_ERROR_TYPE_DEBUG, "message processed\n");
1105   if (GNUNET_YES == h->in_receive)
1106   {
1107     GNUNET_CLIENT_receive (h->client, &msg_received, h,
1108                            GNUNET_TIME_UNIT_FOREVER_REL);
1109   }
1110   else
1111   {
1112     LOG (GNUNET_ERROR_TYPE_DEBUG,
1113          "in receive off, not calling CLIENT_receive\n");
1114   }
1115 }
1116
1117
1118 /******************************************************************************/
1119 /************************       SEND FUNCTIONS     ****************************/
1120 /******************************************************************************/
1121
1122 /**
1123  * Function called to send a message to the service.
1124  * "buf" will be NULL and "size" zero if the socket was closed for writing in
1125  * the meantime.
1126  *
1127  * @param cls closure, the mesh handle
1128  * @param size number of bytes available in buf
1129  * @param buf where the callee should write the connect message
1130  * @return number of bytes written to buf
1131  */
1132 static size_t
1133 send_callback (void *cls, size_t size, void *buf)
1134 {
1135   struct GNUNET_MESH_Handle *h = cls;
1136   struct GNUNET_MESH_TransmitHandle *th;
1137   struct GNUNET_MESH_TransmitHandle *next;
1138   struct GNUNET_MESH_Channel *ch;
1139   char *cbuf = buf;
1140   size_t tsize;
1141   size_t psize;
1142   size_t nsize;
1143
1144   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n");
1145   LOG (GNUNET_ERROR_TYPE_DEBUG, "# Send packet() Buffer %u\n", size);
1146   if ((0 == size) || (NULL == buf))
1147   {
1148     LOG (GNUNET_ERROR_TYPE_DEBUG, "# Received NULL send callback on %p\n", h);
1149     reconnect (h);
1150     h->th = NULL;
1151     return 0;
1152   }
1153   tsize = 0;
1154   next = h->th_head;
1155   nsize = message_ready_size (h);
1156   while ((NULL != (th = next)) && (0 < nsize) && (size >= nsize))
1157   {
1158     ch = th->channel;
1159     if (GNUNET_YES == th_is_payload (th))
1160     {
1161       struct GNUNET_MESH_LocalData *dmsg;
1162       struct GNUNET_MessageHeader *mh;
1163
1164       LOG (GNUNET_ERROR_TYPE_DEBUG, "#  payload\n");
1165       if (GNUNET_NO == ch->allow_send)
1166       {
1167         /* This channel is not ready to transmit yet, try next message */
1168         next = th->next;
1169         continue;
1170       }
1171       ch->packet_size = 0;
1172       GNUNET_assert (size >= th->size);
1173       dmsg = (struct GNUNET_MESH_LocalData *) cbuf;
1174       mh = (struct GNUNET_MessageHeader *) &dmsg[1];
1175       psize = th->notify (th->notify_cls,
1176                           size - sizeof (struct GNUNET_MESH_LocalData),
1177                           mh);
1178       if (psize > 0)
1179       {
1180         psize += sizeof (struct GNUNET_MESH_LocalData);
1181         GNUNET_assert (size >= psize);
1182         dmsg->header.size = htons (psize);
1183         dmsg->chid = htonl (ch->chid);
1184         dmsg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA);
1185         LOG (GNUNET_ERROR_TYPE_DEBUG, "#  payload type %s\n",
1186              GNUNET_MESH_DEBUG_M2S (ntohs (mh->type)));
1187                 ch->allow_send = GNUNET_NO;
1188       }
1189       else
1190       {
1191         LOG (GNUNET_ERROR_TYPE_DEBUG,
1192              "#  callback returned size 0, "
1193              "application canceled transmission\n");
1194       }
1195     }
1196     else
1197     {
1198       struct GNUNET_MessageHeader *mh = (struct GNUNET_MessageHeader *) &th[1];
1199
1200       LOG (GNUNET_ERROR_TYPE_DEBUG, "#  mesh internal traffic, type %s\n",
1201            GNUNET_MESH_DEBUG_M2S (ntohs (mh->type)));
1202       memcpy (cbuf, &th[1], th->size);
1203       psize = th->size;
1204     }
1205     if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1206       GNUNET_SCHEDULER_cancel (th->timeout_task);
1207     GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th);
1208     GNUNET_free (th);
1209     next = h->th_head;
1210     nsize = message_ready_size (h);
1211     cbuf += psize;
1212     size -= psize;
1213     tsize += psize;
1214   }
1215   LOG (GNUNET_ERROR_TYPE_DEBUG, "#  total size: %u\n", tsize);
1216   h->th = NULL;
1217   size = message_ready_size (h);
1218   if (0 != size)
1219   {
1220     LOG (GNUNET_ERROR_TYPE_DEBUG, "#  next size: %u\n", size);
1221     h->th =
1222         GNUNET_CLIENT_notify_transmit_ready (h->client, size,
1223                                              GNUNET_TIME_UNIT_FOREVER_REL,
1224                                              GNUNET_YES, &send_callback, h);
1225   }
1226   else
1227   {
1228     if (NULL != h->th_head)
1229       LOG (GNUNET_ERROR_TYPE_DEBUG, "#  can't transmit any more\n");
1230     else
1231       LOG (GNUNET_ERROR_TYPE_DEBUG, "#  nothing left to transmit\n");
1232   }
1233   if (GNUNET_NO == h->in_receive)
1234   {
1235     LOG (GNUNET_ERROR_TYPE_DEBUG, "# start receiving from service\n");
1236     h->in_receive = GNUNET_YES;
1237     GNUNET_CLIENT_receive (h->client, &msg_received, h,
1238                            GNUNET_TIME_UNIT_FOREVER_REL);
1239   }
1240   LOG (GNUNET_ERROR_TYPE_DEBUG, "# Send packet() END\n");
1241   return tsize;
1242 }
1243
1244
1245 /**
1246  * Auxiliary function to send an already constructed packet to the service.
1247  * Takes care of creating a new queue element, copying the message and
1248  * calling the tmt_rdy function if necessary.
1249  *
1250  * @param h mesh handle
1251  * @param msg message to transmit
1252  * @param channel channel this send is related to (NULL if N/A)
1253  */
1254 static void
1255 send_packet (struct GNUNET_MESH_Handle *h,
1256              const struct GNUNET_MessageHeader *msg,
1257              struct GNUNET_MESH_Channel *channel)
1258 {
1259   struct GNUNET_MESH_TransmitHandle *th;
1260   size_t msize;
1261
1262   LOG (GNUNET_ERROR_TYPE_DEBUG, " Sending message to service: %s\n",
1263        GNUNET_MESH_DEBUG_M2S(ntohs(msg->type)));
1264   msize = ntohs (msg->size);
1265   th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle) + msize);
1266   th->timeout = GNUNET_TIME_UNIT_FOREVER_ABS;
1267   th->size = msize;
1268   th->channel = channel;
1269   memcpy (&th[1], msg, msize);
1270   add_to_queue (h, th);
1271   LOG (GNUNET_ERROR_TYPE_DEBUG, "  queued\n");
1272   if (NULL != h->th)
1273     return;
1274   LOG (GNUNET_ERROR_TYPE_DEBUG, "  calling ntfy tmt rdy for %u bytes\n", msize);
1275   h->th =
1276       GNUNET_CLIENT_notify_transmit_ready (h->client, msize,
1277                                            GNUNET_TIME_UNIT_FOREVER_REL,
1278                                            GNUNET_YES, &send_callback, h);
1279 }
1280
1281
1282 /******************************************************************************/
1283 /**********************      API CALL DEFINITIONS     *************************/
1284 /******************************************************************************/
1285
1286 struct GNUNET_MESH_Handle *
1287 GNUNET_MESH_connect (const struct GNUNET_CONFIGURATION_Handle *cfg, void *cls,
1288                      GNUNET_MESH_InboundChannelNotificationHandler new_channel,
1289                      GNUNET_MESH_ChannelEndHandler cleaner,
1290                      const struct GNUNET_MESH_MessageHandler *handlers,
1291                      const uint32_t *ports)
1292 {
1293   struct GNUNET_MESH_Handle *h;
1294
1295   LOG (GNUNET_ERROR_TYPE_DEBUG, "GNUNET_MESH_connect()\n");
1296   h = GNUNET_malloc (sizeof (struct GNUNET_MESH_Handle));
1297   LOG (GNUNET_ERROR_TYPE_DEBUG, " addr %p\n", h);
1298   h->cfg = cfg;
1299   h->new_channel = new_channel;
1300   h->cleaner = cleaner;
1301   h->client = GNUNET_CLIENT_connect ("mesh", cfg);
1302   if (h->client == NULL)
1303   {
1304     GNUNET_break (0);
1305     GNUNET_free (h);
1306     return NULL;
1307   }
1308   h->cls = cls;
1309   h->message_handlers = handlers;
1310   h->ports = ports;
1311   h->next_chid = GNUNET_MESH_LOCAL_CHANNEL_ID_CLI;
1312   h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS;
1313   h->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
1314
1315   if (NULL != ports && ports[0] != 0 && NULL == new_channel)
1316   {
1317     GNUNET_break (0);
1318     LOG (GNUNET_ERROR_TYPE_DEBUG,
1319          "no new channel handler given, ports parameter is useless!!\n");
1320   }
1321   if ((NULL == ports || ports[0] == 0) && NULL != new_channel)
1322   {
1323     GNUNET_break (0);
1324     LOG (GNUNET_ERROR_TYPE_DEBUG,
1325          "no ports given, new channel handler will never be called!!\n");
1326   }
1327   /* count handlers */
1328   for (h->n_handlers = 0;
1329        handlers && handlers[h->n_handlers].type;
1330        h->n_handlers++) ;
1331   for (h->n_ports = 0;
1332        ports && ports[h->n_ports];
1333        h->n_ports++) ;
1334   send_connect (h);
1335   LOG (GNUNET_ERROR_TYPE_DEBUG, "GNUNET_MESH_connect() END\n");
1336   return h;
1337 }
1338
1339
1340 void
1341 GNUNET_MESH_disconnect (struct GNUNET_MESH_Handle *handle)
1342 {
1343   struct GNUNET_MESH_Channel *ch;
1344   struct GNUNET_MESH_Channel *aux;
1345   struct GNUNET_MESH_TransmitHandle *th;
1346
1347   LOG (GNUNET_ERROR_TYPE_DEBUG, "MESH DISCONNECT\n");
1348
1349   ch = handle->channels_head;
1350   while (NULL != ch)
1351   {
1352     aux = ch->next;
1353     if (ch->chid < GNUNET_MESH_LOCAL_CHANNEL_ID_SERV)
1354     {
1355       GNUNET_break (0);
1356       LOG (GNUNET_ERROR_TYPE_DEBUG, "channel %X not destroyed\n", ch->chid);
1357     }
1358     destroy_channel (ch, GNUNET_YES);
1359     ch = aux;
1360   }
1361   while ( (th = handle->th_head) != NULL)
1362   {
1363     struct GNUNET_MessageHeader *msg;
1364
1365     /* Make sure it is an allowed packet (everything else should have been
1366      * already canceled).
1367      */
1368     GNUNET_break (GNUNET_NO == th_is_payload (th));
1369     msg = (struct GNUNET_MessageHeader *) &th[1];
1370     switch (ntohs(msg->type))
1371     {
1372       case GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT:
1373       case GNUNET_MESSAGE_TYPE_MESH_LOCAL_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_LOCAL_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_LOCAL_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