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.
19 * @file mesh/mesh_api_new.c
20 * @brief mesh api: client implementation of mesh service
21 * @author Bartlomiej Polot
24 * - callbacks to client missing on certain events
25 * - processing messages from service is incomplete
30 * - AUXILIARY FUNCTIONS
33 * - API CALL DEFINITIONS
38 #if 0 /* keep Emacsens' auto-indent happy */
44 #include "gnunet_common.h"
45 #include "gnunet_client_lib.h"
46 #include "gnunet_util_lib.h"
47 #include "gnunet_peer_lib.h"
48 #include "gnunet_mesh_service_new.h"
50 #include "mesh_protocol.h"
53 /******************************************************************************/
54 /************************ DATA STRUCTURES ****************************/
55 /******************************************************************************/
58 * Transmission queue to the service
60 struct GNUNET_MESH_TransmitHandle
66 struct GNUNET_MESH_TransmitHandle *next;
71 struct GNUNET_MESH_TransmitHandle *prev;
74 * Tunnel this message is sent over (may be NULL for control messages).
76 struct GNUNET_MESH_Tunnel *tunnel;
79 * Data itself, currently points to the end of this struct if
80 * we have a message already, NULL if the message is to be
81 * obtained from the callback.
83 const struct GNUNET_MessageHeader *data;
86 * Callback to obtain the message to transmit, or NULL if we
87 * got the message in 'data'. Notice that messages built
88 * by 'notify' need to be encapsulated with information about
91 GNUNET_CONNECTION_TransmitReadyNotify notify;
94 * Closure for 'notify'
99 * How long is this message valid. Once the timeout has been
100 * reached, the message must no longer be sent. If this
101 * is a message with a 'notify' callback set, the 'notify'
102 * function should be called with 'buf' NULL and size 0.
104 struct GNUNET_TIME_Absolute timeout;
107 * Task triggering a timeout, can be NO_TASK if the timeout is FOREVER.
109 GNUNET_SCHEDULER_TaskIdentifier timeout_task;
112 * Priority of the message. The queue is sorted by priority,
113 * control messages have the maximum priority (UINT32_MAX).
118 * Target of the message, 0 for broadcast. This field
119 * is only valid if 'notify' is non-NULL.
121 GNUNET_PEER_Id target;
124 * Size of 'data' -- or the desired size of 'notify' if 'data' is NULL.
131 * Opaque handle to the service.
133 struct GNUNET_MESH_Handle
137 * Handle to the server connection, to send messages later
139 struct GNUNET_CLIENT_Connection *client;
142 * Set of handlers used for processing incoming messages in the tunnels
144 const struct GNUNET_MESH_MessageHandler *message_handlers;
147 * Set of applications that should be claimed to be offered at this node.
148 * Note that this is just informative, the appropiate handlers must be
149 * registered independently and the mapping is up to the developer of the
150 * client application.
152 const GNUNET_MESH_ApplicationType *applications;
155 * Double linked list of the tunnels this client is connected to.
157 struct GNUNET_MESH_Tunnel *tunnels_head;
158 struct GNUNET_MESH_Tunnel *tunnels_tail;
161 * Callback for inbound tunnel creation
163 GNUNET_MESH_InboundTunnelNotificationHandler *new_tunnel;
166 * Callback for inbound tunnel disconnection
168 GNUNET_MESH_TunnelEndHandler *cleaner;
171 * Handle to cancel pending transmissions in case of disconnection
173 struct GNUNET_CLIENT_TransmitHandle *th;
176 * Closure for all the handlers given by the client
181 * Messages to send to the service
183 struct GNUNET_MESH_TransmitHandle *th_head;
184 struct GNUNET_MESH_TransmitHandle *th_tail;
187 * tid of the next tunnel to create (to avoid reusing IDs often)
189 MESH_TunnelNumber next_tid;
190 unsigned int n_handlers;
191 unsigned int n_applications;
192 unsigned int max_queue_size;
195 * Have we started the task to receive messages from the service
196 * yet? We do this after we send the 'MESH_LOCAL_CONNECT' message.
201 * Number of packets queued
203 unsigned int npackets;
206 * Configuration given by the client, in case of reconnection
208 const struct GNUNET_CONFIGURATION_Handle *cfg;
213 * Description of a peer
215 struct GNUNET_MESH_Peer
218 * ID of the peer in short form
223 * Tunnel this peer belongs to
225 struct GNUNET_MESH_Tunnel *t;
228 * Flag indicating whether service has informed about its connection
236 * Opaque handle to a tunnel.
238 struct GNUNET_MESH_Tunnel
244 struct GNUNET_MESH_Tunnel *next;
245 struct GNUNET_MESH_Tunnel *prev;
248 * Callback to execute when peers connect to the tunnel
250 GNUNET_MESH_TunnelConnectHandler connect_handler;
253 * Callback to execute when peers disconnect from the tunnel
255 GNUNET_MESH_TunnelDisconnectHandler disconnect_handler;
258 * Closure for the connect/disconnect handlers
263 * Handle to the mesh this tunnel belongs to
265 struct GNUNET_MESH_Handle *mesh;
268 * Local ID of the tunnel
270 MESH_TunnelNumber tid;
273 * Owner of the tunnel
275 GNUNET_PEER_Id owner;
278 * All peers added to the tunnel
280 struct GNUNET_MESH_Peer **peers;
283 * List of application types that have been requested for this tunnel
285 GNUNET_MESH_ApplicationType *apps;
288 * Any data the caller wants to put in here
293 * Number of peers added to the tunnel
298 * Number of packets queued in this tunnel
300 unsigned int npackets;
303 * Number of applications requested this tunnel
310 /******************************************************************************/
311 /*********************** AUXILIARY FUNCTIONS *************************/
312 /******************************************************************************/
315 * Get the tunnel handler for the tunnel specified by id from the given handle
316 * @param h Mesh handle
317 * @param tid ID of the wanted tunnel
318 * @return handle to the required tunnel or NULL if not found
320 static struct GNUNET_MESH_Tunnel *
321 retrieve_tunnel (struct GNUNET_MESH_Handle *h, MESH_TunnelNumber tid)
323 struct GNUNET_MESH_Tunnel *t;
337 * Create a new tunnel and insert it in the tunnel list of the mesh handle
338 * @param h Mesh handle
339 * @param tid desired tid of the tunnel, 0 to assign one automatically
340 * @return handle to the created tunnel
342 static struct GNUNET_MESH_Tunnel *
343 create_tunnel (struct GNUNET_MESH_Handle *h, MESH_TunnelNumber tid)
345 struct GNUNET_MESH_Tunnel *t;
347 t = GNUNET_malloc (sizeof (struct GNUNET_MESH_Tunnel));
348 GNUNET_CONTAINER_DLL_insert (h->tunnels_head, h->tunnels_tail, t);
352 t->tid = h->next_tid;
353 while (NULL != retrieve_tunnel (h, h->next_tid))
356 h->next_tid &= ~GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
357 h->next_tid |= GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
369 * Get the tunnel handler for the tunnel specified by id from the given handle
370 * @param h Mesh handle
371 * @param tid ID of the wanted tunnel
372 * @return handle to the required tunnel or NULL if not found
375 destroy_tunnel (struct GNUNET_MESH_Handle *h, MESH_TunnelNumber tid)
377 struct GNUNET_MESH_Tunnel *t;
378 struct GNUNET_PeerIdentity pi;
381 t = retrieve_tunnel (h, tid);
387 /* TODO remove data packets from queue */
388 GNUNET_CONTAINER_DLL_remove (h->tunnels_head, h->tunnels_tail, t);
389 for (i = 0; i < t->npeers; i++)
391 if (NULL != t->disconnect_handler && t->peers[i]->connected)
393 GNUNET_PEER_resolve (t->peers[i]->id, &pi);
394 t->disconnect_handler (t->cls, &pi);
396 GNUNET_PEER_change_rc (t->peers[i]->id, -1);
397 GNUNET_free (t->peers[i]);
399 if (NULL != h->cleaner && 0 != t->owner)
400 h->cleaner (h->cls, t, t->ctx);
402 GNUNET_PEER_change_rc (t->owner, -1);
403 if (0 != t->napps && t->apps)
404 GNUNET_free (t->apps);
406 GNUNET_free (t->peers);
412 * Get the peer descriptor for the peer with id from the given tunnel
413 * @param t Tunnel handle
414 * @param id Short form ID of the wanted peer
415 * @return handle to the requested peer or NULL if not found
417 static struct GNUNET_MESH_Peer *
418 retrieve_peer (struct GNUNET_MESH_Tunnel *t, GNUNET_PEER_Id id)
422 for (i = 0; i < t->npeers; i++)
423 if (t->peers[i]->id == id)
430 * Add a peer into a tunnel
431 * @param t Tunnel handle
432 * @param pi Full ID of the new peer
433 * @return handle to the newly created peer
435 static struct GNUNET_MESH_Peer *
436 add_peer_to_tunnel (struct GNUNET_MESH_Tunnel *t,
437 const struct GNUNET_PeerIdentity *pi)
439 struct GNUNET_MESH_Peer *p;
442 id = GNUNET_PEER_intern (pi);
444 p = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer));
447 GNUNET_array_append (t->peers, t->npeers, p);
453 * Remove a peer from a tunnel
454 * @param t Tunnel handle
455 * @param p Peer handle
458 remove_peer_from_tunnel (struct GNUNET_MESH_Peer *p)
462 for (i = 0; i < p->t->npeers; i++)
464 if (p->t->peers[i] == p)
467 if (i == p->t->npeers)
472 p->t->peers[i] = p->t->peers[p->t->npeers - 1];
473 GNUNET_array_grow (p->t->peers, p->t->npeers, p->t->npeers - 1);
478 * Notify client that the transmission has timed out
480 * @param tc task context
483 timeout_transmission (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
485 struct GNUNET_MESH_TransmitHandle *th = cls;
486 struct GNUNET_MESH_Handle *mesh;
488 mesh = th->tunnel->mesh;
489 GNUNET_CONTAINER_DLL_remove (mesh->th_head, mesh->th_tail, th);
490 if (th->notify != NULL)
491 th->notify (th->notify_cls, 0, NULL);
493 if ((NULL == mesh->th_head) && (NULL != mesh->th))
495 /* queue empty, no point in asking for transmission */
496 GNUNET_CLIENT_notify_transmit_ready_cancel (mesh->th);
503 * Add a transmit handle to the transmission queue by priority and set the
506 * @param h mesh handle with the queue head and tail
507 * @param q handle to the packet to be transmitted
510 add_to_queue (struct GNUNET_MESH_Handle *h,
511 struct GNUNET_MESH_TransmitHandle *th)
513 struct GNUNET_MESH_TransmitHandle *p;
516 while ((NULL != p) && (th->priority <= p->priority))
522 GNUNET_CONTAINER_DLL_insert_after (h->th_head, h->th_tail, p, th);
523 if (GNUNET_TIME_UNIT_FOREVER_ABS.abs_value == th->timeout.abs_value)
526 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_absolute_get_remaining
527 (th->timeout), &timeout_transmission, th);
532 send_packet (struct GNUNET_MESH_Handle *h,
533 const struct GNUNET_MessageHeader *msg);
537 * Reconnect to the service, retransmit all infomation to try to restore the
540 * @param h handle to the mesh
542 * @return GNUNET_YES in case of sucess, GNUNET_NO otherwise (service down...)
545 reconnect (struct GNUNET_MESH_Handle *h)
547 struct GNUNET_MESH_Tunnel *t;
550 h->in_receive = GNUNET_NO;
554 GNUNET_CLIENT_notify_transmit_ready_cancel (h->th);
556 if (NULL != h->client)
558 GNUNET_CLIENT_disconnect (h->client, GNUNET_NO);
562 h->client = GNUNET_CLIENT_connect ("mesh", h->cfg);
563 if (h->client == NULL)
565 /* FIXME: panic? exponential backoff retry? */
569 /* Rebuild all tunnels */
570 for (t = h->tunnels_head; NULL != t; t = t->next)
572 struct GNUNET_MESH_TunnelMessage tmsg;
573 struct GNUNET_MESH_PeerControl pmsg;
575 tmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
576 tmsg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
577 tmsg.tunnel_id = htonl (t->tid);
578 send_packet (h, &tmsg.header);
580 pmsg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
581 pmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
582 pmsg.tunnel_id = htonl (t->tid);
584 /* Reconnect all peers */
585 for (i = 0; i < t->npeers; i++)
587 GNUNET_PEER_resolve (t->peers[i]->id, &pmsg.peer);
588 if (NULL != t->disconnect_handler && t->peers[i]->connected)
589 t->disconnect_handler (t->cls, &pmsg.peer);
590 /* If the tunnel was "by type", dont connect individual peers */
592 send_packet (t->mesh, &pmsg.header);
594 /* Reconnect all types, if any */
595 for (i = 0; i < t->napps; i++)
597 struct GNUNET_MESH_ConnectPeerByType msg;
599 msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
600 msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE);
601 msg.tunnel_id = htonl (t->tid);
602 msg.type = htonl (t->apps[i]);
603 send_packet (t->mesh, &msg.header);
610 /******************************************************************************/
611 /*********************** RECEIVE HANDLERS ****************************/
612 /******************************************************************************/
615 * Process the new tunnel notification and add it to the tunnels in the handle
617 * @param h The mesh handle
618 * @param msg A message with the details of the new incoming tunnel
621 process_tunnel_create (struct GNUNET_MESH_Handle *h,
622 const struct GNUNET_MESH_TunnelMessage *msg)
624 struct GNUNET_MESH_Tunnel *t;
625 MESH_TunnelNumber tid;
627 tid = ntohl (msg->tunnel_id);
628 if (tid <= GNUNET_MESH_LOCAL_TUNNEL_ID_CLI)
633 t = create_tunnel (h, tid);
636 GNUNET_CONTAINER_DLL_insert (h->tunnels_head, h->tunnels_tail, t);
642 * Process the tunnel destroy notification and free associated resources
644 * @param h The mesh handle
645 * @param msg A message with the details of the tunnel being destroyed
648 process_tunnel_destroy (struct GNUNET_MESH_Handle *h,
649 const struct GNUNET_MESH_TunnelMessage *msg)
651 struct GNUNET_MESH_Tunnel *t;
652 MESH_TunnelNumber tid;
654 tid = ntohl (msg->tunnel_id);
655 t = retrieve_tunnel (h, tid);
660 GNUNET_CONTAINER_DLL_insert (h->tunnels_head, h->tunnels_tail, t);
666 * Process the new peer event and notify the upper level of it
668 * @param h The mesh handle
669 * @param msg A message with the details of the peer event
672 process_peer_event (struct GNUNET_MESH_Handle *h,
673 const struct GNUNET_MESH_PeerControl *msg)
675 struct GNUNET_MESH_Tunnel *t;
676 struct GNUNET_MESH_Peer *p;
677 struct GNUNET_TRANSPORT_ATS_Information atsi;
681 size = ntohs (msg->header.size);
682 if (size != sizeof (struct GNUNET_MESH_PeerControl))
687 t = retrieve_tunnel (h, ntohl (msg->tunnel_id));
690 t = create_tunnel (h, msg->tunnel_id);
691 t->owner = GNUNET_PEER_intern (&msg->peer);
693 t->peers = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer *));
694 t->peers[0] = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer));
696 t->peers[0]->connected = 1;
697 t->peers[0]->id = t->owner;
700 id = GNUNET_PEER_search (&msg->peer);
701 if ((p = retrieve_peer (t, id)) == NULL)
702 p = add_peer_to_tunnel (t, &msg->peer);
705 if (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD == msg->header.type)
707 if (NULL != t->connect_handler)
709 t->connect_handler (t->cls, &msg->peer, &atsi);
715 if (NULL != t->disconnect_handler && p->connected)
717 t->disconnect_handler (t->cls, &msg->peer);
719 remove_peer_from_tunnel (p);
726 * Process the incoming data packets
728 * @param h The mesh handle
729 * @param msh A message encapsulating the data
732 process_incoming_data (struct GNUNET_MESH_Handle *h,
733 const struct GNUNET_MessageHeader *message)
735 const struct GNUNET_MessageHeader *payload;
736 const struct GNUNET_MESH_MessageHandler *handler;
737 const struct GNUNET_PeerIdentity *peer;
738 struct GNUNET_MESH_Unicast *ucast;
739 struct GNUNET_MESH_Multicast *mcast;
740 struct GNUNET_MESH_ToOrigin *to_orig;
741 struct GNUNET_MESH_Tunnel *t;
745 type = ntohs (message->type);
748 case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
749 ucast = (struct GNUNET_MESH_Unicast *) message;
750 t = retrieve_tunnel (h, ntohl (ucast->tid));
751 payload = (struct GNUNET_MessageHeader *) &ucast[1];
754 case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
755 mcast = (struct GNUNET_MESH_Multicast *) message;
756 t = retrieve_tunnel (h, ntohl (mcast->tid));
757 payload = (struct GNUNET_MessageHeader *) &mcast[1];
760 case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
761 to_orig = (struct GNUNET_MESH_ToOrigin *) message;
762 t = retrieve_tunnel (h, ntohl (to_orig->tid));
763 payload = (struct GNUNET_MessageHeader *) &to_orig[1];
764 peer = &to_orig->sender;
775 for (i = 0; i < h->n_handlers; i++)
777 handler = &h->message_handlers[i];
778 if (handler->type == type)
780 struct GNUNET_TRANSPORT_ATS_Information atsi;
785 handler->callback (h->cls, t, &t->ctx, peer, payload, &atsi))
787 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
788 "MESH: callback completed successfully\n");
792 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
793 "MESH: callback caused disconnection\n");
794 GNUNET_MESH_disconnect (h);
802 * Function to process all messages received from the service
805 * @param msg message received, NULL on timeout or fatal error
808 msg_received (void *cls, const struct GNUNET_MessageHeader *msg)
810 struct GNUNET_MESH_Handle *h = cls;
812 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: received a message from MESH\n");
818 switch (ntohs (msg->type))
820 /* Notify of a new incoming tunnel */
821 case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE:
822 process_tunnel_create (h, (struct GNUNET_MESH_TunnelMessage *) msg);
824 /* Notify of a tunnel disconnection */
825 case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY:
826 process_tunnel_destroy (h, (struct GNUNET_MESH_TunnelMessage *) msg);
828 /* Notify of a new peer or a peer disconnect in the tunnel */
829 case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD:
830 case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL:
831 process_peer_event (h, (struct GNUNET_MESH_PeerControl *) msg);
833 /* Notify of a new data packet in the tunnel */
834 case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
835 case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
836 case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
837 process_incoming_data (h, msg);
839 /* We shouldn't get any other packages, log and ignore */
841 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
842 "MESH: unsolicited message form service (type %d)\n",
845 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: message processed\n");
846 GNUNET_CLIENT_receive (h->client, &msg_received, h,
847 GNUNET_TIME_UNIT_FOREVER_REL);
851 /******************************************************************************/
852 /************************ SEND FUNCTIONS ****************************/
853 /******************************************************************************/
856 * Function called to send a message to the service.
857 * "buf" will be NULL and "size" zero if the socket was closed for writing in
860 * @param cls closure, the mesh handle
861 * @param size number of bytes available in buf
862 * @param buf where the callee should write the connect message
863 * @return number of bytes written to buf
866 send_callback (void *cls, size_t size, void *buf)
868 struct GNUNET_MESH_Handle *h = cls;
869 struct GNUNET_MESH_TransmitHandle *th;
874 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Send packet() Buffer %u\n", size);
876 if ((0 == size) || (NULL == buf))
882 while ((NULL != (th = h->th_head)) && (size >= th->size))
884 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: type: %u\n",
885 ntohs (th->data->type));
886 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: size: %u\n",
887 ntohs (th->data->size));
888 if (NULL == th->data)
890 GNUNET_assert (NULL != th->notify);
894 struct GNUNET_MESH_Multicast mc;
896 GNUNET_assert (size >= sizeof (mc) + th->size);
898 th->notify (th->notify_cls, size - sizeof (mc), &cbuf[sizeof (mc)]);
901 mc.header.size = htons (sizeof (mc) + th->size);
902 mc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_MULTICAST);
903 mc.tid = htonl (th->tunnel->tid);
904 memset (&mc.oid, 0, sizeof (struct GNUNET_PeerIdentity)); /* myself */
905 memcpy (cbuf, &mc, sizeof (mc));
906 psize = th->size + sizeof (mc);
912 struct GNUNET_MESH_Unicast uc;
914 GNUNET_assert (size >= sizeof (uc) + th->size);
916 th->notify (th->notify_cls, size - sizeof (uc), &cbuf[sizeof (uc)]);
919 uc.header.size = htons (sizeof (uc) + th->size);
920 uc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_UNICAST);
921 uc.tid = htonl (th->tunnel->tid);
922 memset (&uc.oid, 0, sizeof (struct GNUNET_PeerIdentity)); /* myself */
923 GNUNET_PEER_resolve (th->target, &uc.destination);
924 memcpy (cbuf, &uc, sizeof (uc));
925 psize = th->size + sizeof (uc);
931 memcpy (cbuf, th->data, th->size);
934 if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
935 GNUNET_SCHEDULER_cancel (th->timeout_task);
936 GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th);
942 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: total size: %u\n", tsize);
943 if (NULL != (th = h->th_head))
945 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: next size: %u\n", th->size);
947 GNUNET_CLIENT_notify_transmit_ready (h->client, th->size,
948 GNUNET_TIME_UNIT_FOREVER_REL,
949 GNUNET_YES, &send_callback, h);
951 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Send packet() END\n");
952 if (GNUNET_NO == h->in_receive)
954 h->in_receive = GNUNET_YES;
955 GNUNET_CLIENT_receive (h->client, &msg_received, h,
956 GNUNET_TIME_UNIT_FOREVER_REL);
963 * Auxiliary function to send an already constructed packet to the service.
964 * Takes care of creating a new queue element, copying the message and
965 * calling the tmt_rdy function if necessary.
966 * @param h mesh handle
967 * @param msg message to transmit
970 send_packet (struct GNUNET_MESH_Handle *h,
971 const struct GNUNET_MessageHeader *msg)
973 struct GNUNET_MESH_TransmitHandle *th;
976 msize = ntohs (msg->size);
977 th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle) + msize);
978 th->priority = UINT32_MAX;
979 th->timeout = GNUNET_TIME_UNIT_FOREVER_ABS;
981 th->data = (void *) &th[1];
982 memcpy (&th[1], msg, msize);
983 add_to_queue (h, th);
987 GNUNET_CLIENT_notify_transmit_ready (h->client, msize,
988 GNUNET_TIME_UNIT_FOREVER_REL,
989 GNUNET_YES, &send_callback, h);
993 /******************************************************************************/
994 /********************** API CALL DEFINITIONS *************************/
995 /******************************************************************************/
998 * Connect to the mesh service.
1000 * @param cfg configuration to use
1001 * @param queue_size size of the data message queue, shared among all tunnels
1002 * (each tunnel is guaranteed to accept at least one message,
1003 * no matter what is the status of other tunnels)
1004 * @param cls closure for the various callbacks that follow
1005 * (including handlers in the handlers array)
1006 * @param new_tunnel function called when an *inbound* tunnel is created
1007 * @param cleaner function called when an *inbound* tunnel is destroyed
1008 * @param handlers callbacks for messages we care about, NULL-terminated
1009 * note that the mesh is allowed to drop notifications about
1010 * inbound messages if the client does not process them fast
1011 * enough (for this notification type, a bounded queue is used)
1012 * @param stypes list of the applications that this client claims to provide
1013 * @return handle to the mesh service NULL on error
1014 * (in this case, init is never called)
1016 struct GNUNET_MESH_Handle *
1017 GNUNET_MESH_connect (const struct GNUNET_CONFIGURATION_Handle *cfg,
1018 unsigned int queue_size, void *cls,
1019 GNUNET_MESH_InboundTunnelNotificationHandler new_tunnel,
1020 GNUNET_MESH_TunnelEndHandler cleaner,
1021 const struct GNUNET_MESH_MessageHandler *handlers,
1022 const GNUNET_MESH_ApplicationType *stypes)
1024 struct GNUNET_MESH_Handle *h;
1025 struct GNUNET_MESH_ClientConnect *msg;
1026 GNUNET_MESH_ApplicationType *apps;
1032 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: GNUNET_MESH_connect()\n");
1033 h = GNUNET_malloc (sizeof (struct GNUNET_MESH_Handle));
1035 h->max_queue_size = queue_size;
1036 h->new_tunnel = new_tunnel;
1037 h->cleaner = cleaner;
1038 h->client = GNUNET_CLIENT_connect ("mesh", cfg);
1039 if (h->client == NULL)
1046 h->message_handlers = handlers;
1047 h->applications = stypes;
1048 h->next_tid = GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
1050 /* count handlers and apps, calculate size */
1051 for (h->n_handlers = 0; handlers[h->n_handlers].type; h->n_handlers++) ;
1052 for (h->n_applications = 0; stypes[h->n_applications]; h->n_applications++) ;
1053 size = sizeof (struct GNUNET_MESH_ClientConnect);
1054 size += h->n_handlers * sizeof (uint16_t);
1055 size += h->n_applications * sizeof (GNUNET_MESH_ApplicationType);
1060 /* build connection packet */
1061 msg = (struct GNUNET_MESH_ClientConnect *) buf;
1062 msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT);
1063 msg->header.size = htons (size);
1064 types = (uint16_t *) & msg[1];
1065 for (ntypes = 0; ntypes < h->n_handlers; ntypes++)
1066 types[ntypes] = h->message_handlers[ntypes].type;
1067 apps = (GNUNET_MESH_ApplicationType *) &types[ntypes];
1068 for (napps = 0; napps < h->n_applications; napps++)
1069 apps[napps] = h->applications[napps];
1070 msg->applications = htons (napps);
1071 msg->types = htons (ntypes);
1072 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1073 "mesh: Sending %lu bytes long message %d types and %d apps\n",
1074 ntohs (msg->header.size), ntypes, napps);
1075 send_packet (h, &msg->header);
1077 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: GNUNET_MESH_connect() END\n");
1083 * Disconnect from the mesh service.
1085 * @param handle connection to mesh to disconnect
1088 GNUNET_MESH_disconnect (struct GNUNET_MESH_Handle *handle)
1090 struct GNUNET_MESH_Tunnel *t;
1092 for (t = handle->tunnels_head; NULL != t; t = t->next)
1094 destroy_tunnel (handle, t->tid);
1096 if (NULL != handle->th)
1098 GNUNET_CLIENT_notify_transmit_ready_cancel (handle->th);
1100 if (NULL != handle->client)
1102 GNUNET_CLIENT_disconnect (handle->client, GNUNET_NO);
1104 GNUNET_free (handle);
1109 * Create a new tunnel (we're initiator and will be allowed to add/remove peers
1110 * and to broadcast).
1112 * @param h mesh handle
1113 * @param tunnel_ctx client's tunnel context to associate with the tunnel
1114 * @param connect_handler function to call when peers are actually connected
1115 * @param disconnect_handler function to call when peers are disconnected
1116 * @param handler_cls closure for connect/disconnect handlers
1118 struct GNUNET_MESH_Tunnel *
1119 GNUNET_MESH_tunnel_create (struct GNUNET_MESH_Handle *h, void *tunnel_ctx,
1120 GNUNET_MESH_TunnelConnectHandler connect_handler,
1121 GNUNET_MESH_TunnelDisconnectHandler
1122 disconnect_handler, void *handler_cls)
1124 struct GNUNET_MESH_Tunnel *t;
1125 struct GNUNET_MESH_TunnelMessage msg;
1127 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Creating new tunnel\n");
1128 t = create_tunnel (h, 0);
1129 t->connect_handler = connect_handler;
1130 t->disconnect_handler = disconnect_handler;
1131 t->cls = handler_cls;
1132 t->ctx = tunnel_ctx;
1133 msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
1134 msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1135 msg.tunnel_id = htonl (t->tid);
1136 send_packet (h, &msg.header);
1142 * Destroy an existing tunnel.
1144 * @param tun tunnel handle
1147 GNUNET_MESH_tunnel_destroy (struct GNUNET_MESH_Tunnel *tun)
1149 struct GNUNET_MESH_Handle *h;
1150 struct GNUNET_MESH_TunnelMessage msg;
1152 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "mesh: Destroying tunnel\n");
1155 if (0 != tun->owner)
1156 GNUNET_PEER_change_rc (tun->owner, -1);
1158 msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY);
1159 msg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage));
1160 msg.tunnel_id = htonl (tun->tid);
1161 destroy_tunnel (h, tun->tid);
1162 send_packet (h, &msg.header);
1167 * Request that a peer should be added to the tunnel. The existing
1168 * connect handler will be called ONCE with either success or failure.
1169 * This function should NOT be called again with the same peer before the
1170 * connect handler is called.
1172 * @param tunnel handle to existing tunnel
1173 * @param timeout how long to try to establish a connection
1174 * @param peer peer to add
1177 GNUNET_MESH_peer_request_connect_add (struct GNUNET_MESH_Tunnel *tunnel,
1178 const struct GNUNET_PeerIdentity *peer)
1180 struct GNUNET_MESH_PeerControl msg;
1181 GNUNET_PEER_Id peer_id;
1184 peer_id = GNUNET_PEER_intern (peer);
1185 for (i = 0; i < tunnel->npeers; i++)
1187 if (tunnel->peers[i]->id == peer_id)
1189 GNUNET_PEER_change_rc (peer_id, -1);
1194 add_peer_to_tunnel (tunnel, peer);
1196 msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1197 msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
1198 msg.tunnel_id = htonl (tunnel->tid);
1200 send_packet (tunnel->mesh, &msg.header);
1207 * Request that a peer should be removed from the tunnel. The existing
1208 * disconnect handler will be called ONCE if we were connected.
1210 * @param tunnel handle to existing tunnel
1211 * @param peer peer to remove
1214 GNUNET_MESH_peer_request_connect_del (struct GNUNET_MESH_Tunnel *tunnel,
1215 const struct GNUNET_PeerIdentity *peer)
1217 struct GNUNET_MESH_PeerControl msg;
1218 GNUNET_PEER_Id peer_id;
1221 peer_id = GNUNET_PEER_search (peer);
1227 for (i = 0; i < tunnel->npeers; i++)
1228 if (tunnel->peers[i]->id == peer_id)
1230 if (i == tunnel->npeers)
1235 if (NULL != tunnel->disconnect_handler && tunnel->peers[i]->connected == 1)
1236 tunnel->disconnect_handler (tunnel->cls, peer);
1237 GNUNET_PEER_change_rc (peer_id, -1);
1238 GNUNET_free (tunnel->peers[i]);
1239 tunnel->peers[i] = tunnel->peers[tunnel->npeers - 1];
1240 GNUNET_array_grow (tunnel->peers, tunnel->npeers, tunnel->npeers - 1);
1242 msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
1243 msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL);
1244 msg.tunnel_id = htonl (tunnel->tid);
1245 memcpy (&msg.peer, peer, sizeof (struct GNUNET_PeerIdentity));
1246 send_packet (tunnel->mesh, &msg.header);
1251 * Request that the mesh should try to connect to a peer supporting the given
1254 * @param tunnel handle to existing tunnel
1255 * @param app_type application type that must be supported by the peer (MESH
1256 * should discover peer in proximity handling this type)
1259 GNUNET_MESH_peer_request_connect_by_type (struct GNUNET_MESH_Tunnel *tunnel,
1260 GNUNET_MESH_ApplicationType app_type)
1262 struct GNUNET_MESH_ConnectPeerByType msg;
1264 GNUNET_array_append (tunnel->apps, tunnel->napps, app_type);
1266 msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
1267 msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE);
1268 msg.tunnel_id = htonl (tunnel->tid);
1269 msg.type = htonl (app_type);
1270 send_packet (tunnel->mesh, &msg.header);
1275 * Ask the mesh to call "notify" once it is ready to transmit the
1276 * given number of bytes to the specified "target". If we are not yet
1277 * connected to the specified peer, a call to this function will cause
1278 * us to try to establish a connection.
1280 * @param tunnel tunnel to use for transmission
1281 * @param cork is corking allowed for this transmission?
1282 * @param priority how important is the message?
1283 * @param maxdelay how long can the message wait?
1284 * @param target destination for the message,
1285 * NULL for multicast to all tunnel targets
1286 * @param notify_size how many bytes of buffer space does notify want?
1287 * @param notify function to call when buffer space is available;
1288 * will be called with NULL on timeout or if the overall queue
1289 * for this peer is larger than queue_size and this is currently
1290 * the message with the lowest priority
1291 * @param notify_cls closure for notify
1292 * @return non-NULL if the notify callback was queued,
1293 * NULL if we can not even queue the request (insufficient
1294 * memory); if NULL is returned, "notify" will NOT be called.
1296 struct GNUNET_MESH_TransmitHandle *
1297 GNUNET_MESH_notify_transmit_ready (struct GNUNET_MESH_Tunnel *tunnel, int cork,
1299 struct GNUNET_TIME_Relative maxdelay,
1300 const struct GNUNET_PeerIdentity *target,
1302 GNUNET_CONNECTION_TransmitReadyNotify notify,
1305 struct GNUNET_MESH_TransmitHandle *th;
1308 if (tunnel->mesh->npackets >= tunnel->mesh->max_queue_size &&
1309 tunnel->npackets > 0)
1310 return NULL; /* queue full */
1312 tunnel->mesh->npackets++;
1313 th = GNUNET_malloc (sizeof (struct GNUNET_MESH_TransmitHandle));
1314 th->tunnel = tunnel;
1315 th->priority = priority;
1316 th->timeout = GNUNET_TIME_relative_to_absolute (maxdelay);
1317 th->target = GNUNET_PEER_intern (target);
1320 target) ? sizeof (struct GNUNET_MESH_Multicast) : sizeof (struct
1321 GNUNET_MESH_Unicast);
1322 th->size = notify_size + overhead;
1323 th->notify = notify;
1324 th->notify_cls = notify_cls;
1325 add_to_queue (tunnel->mesh, th);
1331 * Cancel the specified transmission-ready notification.
1333 * @param th handle that was returned by "notify_transmit_ready".
1336 GNUNET_MESH_notify_transmit_ready_cancel (struct GNUNET_MESH_TransmitHandle *th)
1338 struct GNUNET_MESH_Handle *mesh;
1340 mesh = th->tunnel->mesh;
1341 if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1342 GNUNET_SCHEDULER_cancel (th->timeout_task);
1343 GNUNET_CONTAINER_DLL_remove (mesh->th_head, mesh->th_tail, th);
1345 if ((NULL == mesh->th_head) && (NULL != mesh->th))
1347 /* queue empty, no point in asking for transmission */
1348 GNUNET_CLIENT_notify_transmit_ready_cancel (mesh->th);
1354 #if 0 /* keep Emacsens' auto-indent happy */