diff options
Diffstat (limited to 'src/mesh/mesh_api.c')
| -rw-r--r-- | src/mesh/mesh_api.c | 1713 |
1 files changed, 1713 insertions, 0 deletions
diff --git a/src/mesh/mesh_api.c b/src/mesh/mesh_api.c new file mode 100644 index 0000000..7b9aa06 --- /dev/null +++ b/src/mesh/mesh_api.c @@ -0,0 +1,1713 @@ +/* + This file is part of GNUnet. + (C) 2011 Christian Grothoff (and other contributing authors) + GNUnet is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published + by the Free Software Foundation; either version 3, or (at your + option) any later version. + GNUnet is distributed in the hope that it will be useful, but + WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + General Public License for more details. + You should have received a copy of the GNU General Public License + along with GNUnet; see the file COPYING. If not, write to the + Free Software Foundation, Inc., 59 Temple Place - Suite 330, + Boston, MA 02111-1307, USA. +*/ + +/** + * @file mesh/mesh_api.c + * @brief mesh api: client implementation of mesh service + * @author Bartlomiej Polot + * + * STRUCTURE: + * - CONSTANTS + * - DATA STRUCTURES + * - AUXILIARY FUNCTIONS + * - RECEIVE HANDLERS + * - SEND FUNCTIONS + * - API CALL DEFINITIONS + */ +#ifdef __cplusplus +extern "C" +{ +#if 0 /* keep Emacsens' auto-indent happy */ +} +#endif +#endif + +#include "platform.h" +#include "gnunet_common.h" +#include "gnunet_client_lib.h" +#include "gnunet_util_lib.h" +#include "gnunet_peer_lib.h" +#include "gnunet_mesh_service.h" +#include "mesh.h" +#include "mesh_protocol.h" + +#define MESH_API_DEBUG GNUNET_YES + +#if MESH_API_DEBUG +#define LOG(kind,...) GNUNET_log_from (kind, "mesh-api",__VA_ARGS__) +#else +#define LOG(kind,...) +#endif + +/******************************************************************************/ +/************************ DATA STRUCTURES ****************************/ +/******************************************************************************/ + +/** + * Transmission queue to the service + */ +struct GNUNET_MESH_TransmitHandle +{ + + /** + * Double Linked list + */ + struct GNUNET_MESH_TransmitHandle *next; + + /** + * Double Linked list + */ + struct GNUNET_MESH_TransmitHandle *prev; + + /** + * Tunnel this message is sent on / for (may be NULL for control messages). + */ + struct GNUNET_MESH_Tunnel *tunnel; + + /** + * Callback to obtain the message to transmit, or NULL if we + * got the message in 'data'. Notice that messages built + * by 'notify' need to be encapsulated with information about + * the 'target'. + */ + GNUNET_CONNECTION_TransmitReadyNotify notify; + + /** + * Closure for 'notify' + */ + void *notify_cls; + + /** + * How long is this message valid. Once the timeout has been + * reached, the message must no longer be sent. If this + * is a message with a 'notify' callback set, the 'notify' + * function should be called with 'buf' NULL and size 0. + */ + struct GNUNET_TIME_Absolute timeout; + + /** + * Task triggering a timeout, can be NO_TASK if the timeout is FOREVER. + */ + GNUNET_SCHEDULER_TaskIdentifier timeout_task; + + /** + * Priority of the message. The queue is sorted by priority, + * control messages have the maximum priority (UINT32_MAX). + */ + uint32_t priority; + + /** + * Target of the message, 0 for multicast. This field + * is only valid if 'notify' is non-NULL. + */ + GNUNET_PEER_Id target; + + /** + * Size of 'data' -- or the desired size of 'notify' if 'data' is NULL. + */ + size_t size; +}; + + +/** + * Opaque handle to the service. + */ +struct GNUNET_MESH_Handle +{ + + /** + * Handle to the server connection, to send messages later + */ + struct GNUNET_CLIENT_Connection *client; + + /** + * Set of handlers used for processing incoming messages in the tunnels + */ + const struct GNUNET_MESH_MessageHandler *message_handlers; + + /** + * Set of applications that should be claimed to be offered at this node. + * Note that this is just informative, the appropiate handlers must be + * registered independently and the mapping is up to the developer of the + * client application. + */ + const GNUNET_MESH_ApplicationType *applications; + + /** + * Double linked list of the tunnels this client is connected to. + */ + struct GNUNET_MESH_Tunnel *tunnels_head; + struct GNUNET_MESH_Tunnel *tunnels_tail; + + /** + * Callback for inbound tunnel creation + */ + GNUNET_MESH_InboundTunnelNotificationHandler *new_tunnel; + + /** + * Callback for inbound tunnel disconnection + */ + GNUNET_MESH_TunnelEndHandler *cleaner; + + /** + * Handle to cancel pending transmissions in case of disconnection + */ + struct GNUNET_CLIENT_TransmitHandle *th; + + /** + * Closure for all the handlers given by the client + */ + void *cls; + + /** + * Messages to send to the service + */ + struct GNUNET_MESH_TransmitHandle *th_head; + struct GNUNET_MESH_TransmitHandle *th_tail; + + /** + * tid of the next tunnel to create (to avoid reusing IDs often) + */ + MESH_TunnelNumber next_tid; + unsigned int n_handlers; + unsigned int n_applications; + unsigned int max_queue_size; + + /** + * Have we started the task to receive messages from the service + * yet? We do this after we send the 'MESH_LOCAL_CONNECT' message. + */ + int in_receive; + + /** + * Number of packets queued + */ + unsigned int npackets; + + /** + * Configuration given by the client, in case of reconnection + */ + const struct GNUNET_CONFIGURATION_Handle *cfg; + + /** + * Time to the next reconnect in case one reconnect fails + */ + struct GNUNET_TIME_Relative reconnect_time; + + /** + * Task for trying to reconnect. + */ + GNUNET_SCHEDULER_TaskIdentifier reconnect_task; +}; + + +/** + * Description of a peer + */ +struct GNUNET_MESH_Peer +{ + /** + * ID of the peer in short form + */ + GNUNET_PEER_Id id; + + /** + * Tunnel this peer belongs to + */ + struct GNUNET_MESH_Tunnel *t; + + /** + * Flag indicating whether service has informed about its connection + */ + int connected; + +}; + + +/** + * Opaque handle to a tunnel. + */ +struct GNUNET_MESH_Tunnel +{ + + /** + * DLL + */ + struct GNUNET_MESH_Tunnel *next; + struct GNUNET_MESH_Tunnel *prev; + + /** + * Callback to execute when peers connect to the tunnel + */ + GNUNET_MESH_PeerConnectHandler connect_handler; + + /** + * Callback to execute when peers disconnect from the tunnel + */ + GNUNET_MESH_PeerDisconnectHandler disconnect_handler; + + /** + * Closure for the connect/disconnect handlers + */ + void *cls; + + /** + * Handle to the mesh this tunnel belongs to + */ + struct GNUNET_MESH_Handle *mesh; + + /** + * Local ID of the tunnel + */ + MESH_TunnelNumber tid; + + /** + * Owner of the tunnel. 0 if the tunnel is the local client. + */ + GNUNET_PEER_Id owner; + + /** + * All peers added to the tunnel + */ + struct GNUNET_MESH_Peer **peers; + + /** + * List of application types that have been requested for this tunnel + */ + GNUNET_MESH_ApplicationType *apps; + + /** + * Any data the caller wants to put in here + */ + void *ctx; + + /** + * Number of peers added to the tunnel + */ + unsigned int npeers; + + /** + * Number of packets queued in this tunnel + */ + unsigned int npackets; + + /** + * Number of applications requested this tunnel + */ + unsigned int napps; + +}; + + +/******************************************************************************/ +/*********************** AUXILIARY FUNCTIONS *************************/ +/******************************************************************************/ + +/** + * Get the tunnel handler for the tunnel specified by id from the given handle + * @param h Mesh handle + * @param tid ID of the wanted tunnel + * @return handle to the required tunnel or NULL if not found + */ +static struct GNUNET_MESH_Tunnel * +retrieve_tunnel (struct GNUNET_MESH_Handle *h, MESH_TunnelNumber tid) +{ + struct GNUNET_MESH_Tunnel *t; + + t = h->tunnels_head; + while (t != NULL) + { + if (t->tid == tid) + return t; + t = t->next; + } + return NULL; +} + + +/** + * Create a new tunnel and insert it in the tunnel list of the mesh handle + * @param h Mesh handle + * @param tid desired tid of the tunnel, 0 to assign one automatically + * @return handle to the created tunnel + */ +static struct GNUNET_MESH_Tunnel * +create_tunnel (struct GNUNET_MESH_Handle *h, MESH_TunnelNumber tid) +{ + struct GNUNET_MESH_Tunnel *t; + + t = GNUNET_malloc (sizeof (struct GNUNET_MESH_Tunnel)); + GNUNET_CONTAINER_DLL_insert (h->tunnels_head, h->tunnels_tail, t); + t->mesh = h; + if (0 == tid) + { + t->tid = h->next_tid; + while (NULL != retrieve_tunnel (h, h->next_tid)) + { + h->next_tid++; + h->next_tid &= ~GNUNET_MESH_LOCAL_TUNNEL_ID_SERV; + h->next_tid |= GNUNET_MESH_LOCAL_TUNNEL_ID_CLI; + } + } + else + { + t->tid = tid; + } + return t; +} + + +/** + * Destroy the specified tunnel. + * - Destroys all peers, calling the disconnect callback on each if needed + * - Cancels all outgoing traffic for that tunnel, calling respective notifys + * - Calls cleaner if tunnel was inbound + * - Frees all memory used + * + * @param t Pointer to the tunnel. + * @param call_cleaner Whether to call the cleaner handler. + * + * @return Handle to the required tunnel or NULL if not found. + */ +static void +destroy_tunnel (struct GNUNET_MESH_Tunnel *t, int call_cleaner) +{ + struct GNUNET_MESH_Handle *h; + struct GNUNET_PeerIdentity pi; + struct GNUNET_MESH_TransmitHandle *th; + struct GNUNET_MESH_TransmitHandle *next; + unsigned int i; + + if (NULL == t) + { + GNUNET_break (0); + return; + } + h = t->mesh; + + /* disconnect all peers */ + GNUNET_CONTAINER_DLL_remove (h->tunnels_head, h->tunnels_tail, t); + for (i = 0; i < t->npeers; i++) + { + if ( (NULL != t->disconnect_handler) && t->peers[i]->connected) + { + GNUNET_PEER_resolve (t->peers[i]->id, &pi); + t->disconnect_handler (t->cls, &pi); + } + GNUNET_PEER_change_rc (t->peers[i]->id, -1); + GNUNET_free (t->peers[i]); + } + + /* signal tunnel destruction */ + if ( (NULL != h->cleaner) && (0 != t->owner) && (GNUNET_YES == call_cleaner) ) + h->cleaner (h->cls, t, t->ctx); + + /* check that clients did not leave messages behind in the queue */ + for (th = h->th_head; NULL != th; th = next) + { + next = th->next; + if (th->tunnel != t) + continue; + /* Clients should have aborted their requests already. + * Management traffic should be ok, as clients can't cancel that */ + GNUNET_break (NULL == th->notify); + GNUNET_CONTAINER_DLL_remove (h->th_head, h->th_tail, th); + + /* clean up request */ + if (GNUNET_SCHEDULER_NO_TASK != th->timeout_task) + GNUNET_SCHEDULER_cancel (th->timeout_task); + GNUNET_free (th); + } + + /* if there are no more pending requests with mesh service, cancel active request */ + /* Note: this should be unnecessary... */ + if ( (NULL == h->th_head) && (NULL != h->th)) + { + GNUNET_CLIENT_notify_transmit_ready_cancel (h->th); + h->th = NULL; + } + + + if (t->npeers > 0) + GNUNET_free (t->peers); + if (0 != t->owner) + GNUNET_PEER_change_rc (t->owner, -1); + if (0 != t->napps && t->apps) + GNUNET_free (t->apps); + GNUNET_free (t); + return; +} + + +/** + * Get the peer descriptor for the peer with id from the given tunnel + * @param t Tunnel handle + * @param id Short form ID of the wanted peer + * @return handle to the requested peer or NULL if not found + */ +static struct GNUNET_MESH_Peer * +retrieve_peer (struct GNUNET_MESH_Tunnel *t, GNUNET_PEER_Id id) +{ + unsigned int i; + + for (i = 0; i < t->npeers; i++) + if (t->peers[i]->id == id) + return t->peers[i]; + return NULL; +} + + +/** + * Add a peer into a tunnel + * @param t Tunnel handle + * @param pi Full ID of the new peer + * @return handle to the newly created peer + */ +static struct GNUNET_MESH_Peer * +add_peer_to_tunnel (struct GNUNET_MESH_Tunnel *t, + const struct GNUNET_PeerIdentity *pi) +{ + struct GNUNET_MESH_Peer *p; + GNUNET_PEER_Id id; + + if (0 != t->owner) + { + GNUNET_break (0); + return NULL; + } + id = GNUNET_PEER_intern (pi); + + p = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer)); + p->id = id; + p->t = t; + GNUNET_array_append (t->peers, t->npeers, p); + return p; +} + + +/** + * Remove a peer from a tunnel + * @param p Peer handle + */ +static void +remove_peer_from_tunnel (struct GNUNET_MESH_Peer *p) +{ + unsigned int i; + + for (i = 0; i < p->t->npeers; i++) + { + if (p->t->peers[i] == p) + break; + } + if (i == p->t->npeers) + { + GNUNET_break (0); + return; + } + p->t->peers[i] = p->t->peers[p->t->npeers - 1]; + GNUNET_array_grow (p->t->peers, p->t->npeers, p->t->npeers - 1); +} + + +/** + * Notify client that the transmission has timed out + * @param cls closure + * @param tc task context + */ +static void +timeout_transmission (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + struct GNUNET_MESH_TransmitHandle *th = cls; + struct GNUNET_MESH_Handle *mesh; + + mesh = th->tunnel->mesh; + GNUNET_CONTAINER_DLL_remove (mesh->th_head, mesh->th_tail, th); + if (th->notify != NULL) + th->notify (th->notify_cls, 0, NULL); + GNUNET_free (th); + if ((NULL == mesh->th_head) && (NULL != mesh->th)) + { + /* queue empty, no point in asking for transmission */ + GNUNET_CLIENT_notify_transmit_ready_cancel (mesh->th); + mesh->th = NULL; + } +} + + +/** + * Add a transmit handle to the transmission queue by priority and set the + * timeout if needed. + * + * @param h mesh handle with the queue head and tail + * @param th handle to the packet to be transmitted + */ +static void +add_to_queue (struct GNUNET_MESH_Handle *h, + struct GNUNET_MESH_TransmitHandle *th) +{ + struct GNUNET_MESH_TransmitHandle *p; + + p = h->th_head; + while ((NULL != p) && (th->priority <= p->priority)) + p = p->next; + if (NULL == p) + p = h->th_tail; + else + p = p->prev; + GNUNET_CONTAINER_DLL_insert_after (h->th_head, h->th_tail, p, th); + if (GNUNET_TIME_UNIT_FOREVER_ABS.abs_value == th->timeout.abs_value) + return; + th->timeout_task = + GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_absolute_get_remaining + (th->timeout), &timeout_transmission, th); +} + + +/** + * Auxiliary function to send an already constructed packet to the service. + * Takes care of creating a new queue element, copying the message and + * calling the tmt_rdy function if necessary. + * + * @param h mesh handle + * @param msg message to transmit + * @param tunnel tunnel this send is related to (NULL if N/A) + */ +static void +send_packet (struct GNUNET_MESH_Handle *h, + const struct GNUNET_MessageHeader *msg, + struct GNUNET_MESH_Tunnel *tunnel); + + +/** + * Reconnect callback: tries to reconnect again after a failer previous + * reconnecttion + * @param cls closure (mesh handle) + * @param tc task context + */ +static void +reconnect_cbk (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc); + + +/** + * Send a connect packet to the service with the applications and types + * requested by the user. + * + * @param h The mesh handle. + * + */ +static void +send_connect (struct GNUNET_MESH_Handle *h) +{ + size_t size; + + size = sizeof (struct GNUNET_MESH_ClientConnect); + size += h->n_applications * sizeof (GNUNET_MESH_ApplicationType); + size += h->n_handlers * sizeof (uint16_t); + { + char buf[size]; + struct GNUNET_MESH_ClientConnect *msg; + GNUNET_MESH_ApplicationType *apps; + uint16_t napps; + uint16_t *types; + uint16_t ntypes; + + /* build connection packet */ + msg = (struct GNUNET_MESH_ClientConnect *) buf; + msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT); + msg->header.size = htons (size); + apps = (GNUNET_MESH_ApplicationType *) &msg[1]; + for (napps = 0; napps < h->n_applications; napps++) + { + apps[napps] = htonl (h->applications[napps]); + LOG (GNUNET_ERROR_TYPE_DEBUG, " app %u\n", h->applications[napps]); + } + types = (uint16_t *) & apps[napps]; + for (ntypes = 0; ntypes < h->n_handlers; ntypes++) + types[ntypes] = htons (h->message_handlers[ntypes].type); + msg->applications = htons (napps); + msg->types = htons (ntypes); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Sending %lu bytes long message %d types and %d apps\n", + ntohs (msg->header.size), ntypes, napps); + send_packet (h, &msg->header, NULL); + } +} + + +/** + * Reconnect to the service, retransmit all infomation to try to restore the + * original state. + * + * @param h handle to the mesh + * + * @return GNUNET_YES in case of sucess, GNUNET_NO otherwise (service down...) + */ +static int +do_reconnect (struct GNUNET_MESH_Handle *h) +{ + struct GNUNET_MESH_Tunnel *t; + unsigned int i; + + LOG (GNUNET_ERROR_TYPE_DEBUG, "*****************************\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, "******* RECONNECT *******\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, "*****************************\n"); + + h->in_receive = GNUNET_NO; + /* disconnect */ + if (NULL != h->th) + { + GNUNET_CLIENT_notify_transmit_ready_cancel (h->th); + h->th = NULL; + } + if (NULL != h->client) + { + GNUNET_CLIENT_disconnect (h->client, GNUNET_NO); + } + + /* connect again */ + h->client = GNUNET_CLIENT_connect ("mesh", h->cfg); + if (h->client == NULL) + { + h->reconnect_task = GNUNET_SCHEDULER_add_delayed (h->reconnect_time, + &reconnect_cbk, h); + h->reconnect_time = + GNUNET_TIME_relative_min (GNUNET_TIME_UNIT_SECONDS, + GNUNET_TIME_relative_multiply + (h->reconnect_time, 2)); + LOG (GNUNET_ERROR_TYPE_DEBUG, " Next retry in %sms\n", + GNUNET_TIME_relative_to_string (h->reconnect_time)); + GNUNET_break (0); + return GNUNET_NO; + } + else + { + h->reconnect_time = GNUNET_TIME_UNIT_MILLISECONDS; + } + send_connect (h); + /* Rebuild all tunnels */ + for (t = h->tunnels_head; NULL != t; t = t->next) + { + struct GNUNET_MESH_TunnelMessage tmsg; + struct GNUNET_MESH_PeerControl pmsg; + + if (t->tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV) + { + /* Tunnel was created by service (incoming tunnel) */ + /* TODO: Notify service of missing tunnel, to request + * creator to recreate path (find a path to him via DHT?) + */ + continue; + } + tmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE); + tmsg.header.size = htons (sizeof (struct GNUNET_MESH_TunnelMessage)); + tmsg.tunnel_id = htonl (t->tid); + send_packet (h, &tmsg.header, t); + + pmsg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl)); + pmsg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD); + pmsg.tunnel_id = htonl (t->tid); + + /* Reconnect all peers */ + for (i = 0; i < t->npeers; i++) + { + GNUNET_PEER_resolve (t->peers[i]->id, &pmsg.peer); + if (NULL != t->disconnect_handler && t->peers[i]->connected) + t->disconnect_handler (t->cls, &pmsg.peer); + /* If the tunnel was "by type", dont connect individual peers */ + if (0 == t->napps) + send_packet (t->mesh, &pmsg.header, t); + } + /* Reconnect all types, if any */ + for (i = 0; i < t->napps; i++) + { + struct GNUNET_MESH_ConnectPeerByType msg; + + msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType)); + msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE); + msg.tunnel_id = htonl (t->tid); + msg.type = htonl (t->apps[i]); + send_packet (t->mesh, &msg.header, t); + } + } + return GNUNET_YES; +} + +/** + * Reconnect callback: tries to reconnect again after a failer previous + * reconnecttion + * @param cls closure (mesh handle) + * @param tc task context + */ +static void +reconnect_cbk (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + struct GNUNET_MESH_Handle *h = cls; + + h->reconnect_task = GNUNET_SCHEDULER_NO_TASK; + if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN)) + return; + do_reconnect (h); +} + + +/** + * Reconnect to the service, retransmit all infomation to try to restore the + * original state. + * + * @param h handle to the mesh + * + * @return GNUNET_YES in case of sucess, GNUNET_NO otherwise (service down...) + */ +static void +reconnect (struct GNUNET_MESH_Handle *h) +{ + LOG (GNUNET_ERROR_TYPE_DEBUG, "Requested RECONNECT\n"); + if (GNUNET_SCHEDULER_NO_TASK == h->reconnect_task) + h->reconnect_task = GNUNET_SCHEDULER_add_delayed (h->reconnect_time, + &reconnect_cbk, h); +} + + +/******************************************************************************/ +/*********************** RECEIVE HANDLERS ****************************/ +/******************************************************************************/ + +/** + * Process the new tunnel notification and add it to the tunnels in the handle + * + * @param h The mesh handle + * @param msg A message with the details of the new incoming tunnel + */ +static void +process_tunnel_created (struct GNUNET_MESH_Handle *h, + const struct GNUNET_MESH_TunnelNotification *msg) +{ + struct GNUNET_MESH_Tunnel *t; + MESH_TunnelNumber tid; + + tid = ntohl (msg->tunnel_id); + if (tid < GNUNET_MESH_LOCAL_TUNNEL_ID_SERV) + { + GNUNET_break (0); + return; + } + t = create_tunnel (h, tid); + t->owner = GNUNET_PEER_intern (&msg->peer); + t->npeers = 1; + t->peers = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer *)); + t->peers[0] = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer)); + t->peers[0]->t = t; + t->peers[0]->connected = 1; + t->peers[0]->id = t->owner; + GNUNET_PEER_change_rc (t->owner, 1); + t->mesh = h; + t->tid = tid; + if (NULL != h->new_tunnel) + { + struct GNUNET_ATS_Information atsi; + + atsi.type = 0; + atsi.value = 0; + t->ctx = h->new_tunnel (h->cls, t, &msg->peer, &atsi); + } + LOG (GNUNET_ERROR_TYPE_DEBUG, "new incoming tunnel %X\n", t->tid); + return; +} + + +/** + * Process the tunnel destroy notification and free associated resources + * + * @param h The mesh handle + * @param msg A message with the details of the tunnel being destroyed + */ +static void +process_tunnel_destroy (struct GNUNET_MESH_Handle *h, + const struct GNUNET_MESH_TunnelMessage *msg) +{ + struct GNUNET_MESH_Tunnel *t; + MESH_TunnelNumber tid; + + tid = ntohl (msg->tunnel_id); + t = retrieve_tunnel (h, tid); + + if (NULL == t) + { + return; + } + if (0 == t->owner) + { + GNUNET_break (0); + } + LOG (GNUNET_ERROR_TYPE_DEBUG, "tunnel %u destroyed\n", t->tid); + destroy_tunnel (t, GNUNET_YES); + return; +} + + +/** + * Process the new peer event and notify the upper level of it + * + * @param h The mesh handle + * @param msg A message with the details of the peer event + */ +static void +process_peer_event (struct GNUNET_MESH_Handle *h, + const struct GNUNET_MESH_PeerControl *msg) +{ + struct GNUNET_MESH_Tunnel *t; + struct GNUNET_MESH_Peer *p; + struct GNUNET_ATS_Information atsi; + GNUNET_PEER_Id id; + uint16_t size; + + LOG (GNUNET_ERROR_TYPE_DEBUG, "processig peer event\n"); + size = ntohs (msg->header.size); + if (size != sizeof (struct GNUNET_MESH_PeerControl)) + { + GNUNET_break (0); + return; + } + t = retrieve_tunnel (h, ntohl (msg->tunnel_id)); + if (NULL == t) + { + GNUNET_break (0); + return; + } + id = GNUNET_PEER_search (&msg->peer); + if ((p = retrieve_peer (t, id)) == NULL) + p = add_peer_to_tunnel (t, &msg->peer); + if (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD == ntohs (msg->header.type)) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, "adding peer\n"); + if (NULL != t->connect_handler) + { + atsi.type = 0; + atsi.value = 0; + t->connect_handler (t->cls, &msg->peer, &atsi); + } + p->connected = 1; + } + else + { + LOG (GNUNET_ERROR_TYPE_DEBUG, "removing peer\n"); + if (NULL != t->disconnect_handler && p->connected) + { + t->disconnect_handler (t->cls, &msg->peer); + } + remove_peer_from_tunnel (p); + GNUNET_free (p); + } + LOG (GNUNET_ERROR_TYPE_DEBUG, "processing peer event END\n"); +} + + +/** + * Process the incoming data packets + * + * @param h The mesh handle + * @param message A message encapsulating the data + * + * @return GNUNET_YES if everything went fine + * GNUNET_NO if client closed connection (h no longer valid) + */ +static int +process_incoming_data (struct GNUNET_MESH_Handle *h, + const struct GNUNET_MessageHeader *message) +{ + const struct GNUNET_MessageHeader *payload; + const struct GNUNET_MESH_MessageHandler *handler; + const struct GNUNET_PeerIdentity *peer; + struct GNUNET_MESH_Unicast *ucast; + struct GNUNET_MESH_Multicast *mcast; + struct GNUNET_MESH_ToOrigin *to_orig; + struct GNUNET_MESH_Tunnel *t; + unsigned int i; + uint16_t type; + + LOG (GNUNET_ERROR_TYPE_DEBUG, "Got a data message!\n"); + type = ntohs (message->type); + switch (type) + { + case GNUNET_MESSAGE_TYPE_MESH_UNICAST: + ucast = (struct GNUNET_MESH_Unicast *) message; + + t = retrieve_tunnel (h, ntohl (ucast->tid)); + payload = (struct GNUNET_MessageHeader *) &ucast[1]; + peer = &ucast->oid; + LOG (GNUNET_ERROR_TYPE_DEBUG, " ucast on tunnel %s [%x]\n", + GNUNET_i2s (peer), ntohl (ucast->tid)); + break; + case GNUNET_MESSAGE_TYPE_MESH_MULTICAST: + mcast = (struct GNUNET_MESH_Multicast *) message; + t = retrieve_tunnel (h, ntohl (mcast->tid)); + payload = (struct GNUNET_MessageHeader *) &mcast[1]; + peer = &mcast->oid; + LOG (GNUNET_ERROR_TYPE_DEBUG, " mcast on tunnel %s [%x]\n", + GNUNET_i2s (peer), ntohl (mcast->tid)); + break; + case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN: + to_orig = (struct GNUNET_MESH_ToOrigin *) message; + t = retrieve_tunnel (h, ntohl (to_orig->tid)); + payload = (struct GNUNET_MessageHeader *) &to_orig[1]; + peer = &to_orig->sender; + LOG (GNUNET_ERROR_TYPE_DEBUG, " torig on tunnel %s [%x]\n", + GNUNET_i2s (peer), ntohl (to_orig->tid)); + break; + default: + GNUNET_break (0); + return GNUNET_YES; + } + if (NULL == t) + { + GNUNET_break (0); + return GNUNET_YES; + } + type = ntohs (payload->type); + for (i = 0; i < h->n_handlers; i++) + { + handler = &h->message_handlers[i]; + if (handler->type == type) + { + struct GNUNET_ATS_Information atsi; + + atsi.type = 0; + atsi.value = 0; + if (GNUNET_OK != + handler->callback (h->cls, t, &t->ctx, peer, payload, &atsi)) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, "MESH: callback caused disconnection\n"); + GNUNET_MESH_disconnect (h); + return GNUNET_NO; + } + else + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "MESH: callback completed successfully\n"); + + } + } + } + return GNUNET_YES; +} + + +/** + * Function to process all messages received from the service + * + * @param cls closure + * @param msg message received, NULL on timeout or fatal error + */ +static void +msg_received (void *cls, const struct GNUNET_MessageHeader *msg) +{ + struct GNUNET_MESH_Handle *h = cls; + + if (msg == NULL) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, "Received NULL msg\n"); + reconnect (h); + return; + } + LOG (GNUNET_ERROR_TYPE_DEBUG, "received a message type %hu from MESH\n", + ntohs (msg->type)); + switch (ntohs (msg->type)) + { + /* Notify of a new incoming tunnel */ + case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE: + process_tunnel_created (h, (struct GNUNET_MESH_TunnelNotification *) msg); + break; + /* Notify of a tunnel disconnection */ + case GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY: + process_tunnel_destroy (h, (struct GNUNET_MESH_TunnelMessage *) msg); + break; + /* Notify of a new peer or a peer disconnect in the tunnel */ + case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD: + case GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL: + process_peer_event (h, (struct GNUNET_MESH_PeerControl *) msg); + break; + /* Notify of a new data packet in the tunnel */ + case GNUNET_MESSAGE_TYPE_MESH_UNICAST: + case GNUNET_MESSAGE_TYPE_MESH_MULTICAST: + case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN: + if (GNUNET_NO == process_incoming_data (h, msg)) + return; + break; + /* We shouldn't get any other packages, log and ignore */ + default: + LOG (GNUNET_ERROR_TYPE_WARNING, + "MESH: unsolicited message form service (type %d)\n", + ntohs (msg->type)); + } + LOG (GNUNET_ERROR_TYPE_DEBUG, "message processed\n"); + GNUNET_CLIENT_receive (h->client, &msg_received, h, + GNUNET_TIME_UNIT_FOREVER_REL); +} + + +/******************************************************************************/ +/************************ SEND FUNCTIONS ****************************/ +/******************************************************************************/ + +/** + * Function called to send a message to the service. + * "buf" will be NULL and "size" zero if the socket was closed for writing in + * the meantime. + * + * @param cls closure, the mesh handle + * @param size number of bytes available in buf + * @param buf where the callee should write the connect message + * @return number of bytes written to buf + */ +static size_t +send_callback (void *cls, size_t size, void *buf) +{ + struct GNUNET_MESH_Handle *h = cls; + struct GNUNET_MESH_TransmitHandle *th; + char *cbuf = buf; + size_t tsize; + size_t psize; + + LOG (GNUNET_ERROR_TYPE_DEBUG, "Send packet() Buffer %u\n", size); + h->th = NULL; + if ((0 == size) || (NULL == buf)) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, "Received NULL callback\n"); + reconnect (h); + return 0; + } + tsize = 0; + while ((NULL != (th = h->th_head)) && (size >= th->size)) + { + if (NULL != th->notify) + { + if (th->tunnel->tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV) + { + /* traffic to origin */ + struct GNUNET_MESH_ToOrigin to; + struct GNUNET_MessageHeader *mh; + + GNUNET_assert (size >= th->size); + mh = (struct GNUNET_MessageHeader *) &cbuf[sizeof (to)]; + psize = th->notify (th->notify_cls, size - sizeof (to), mh); + LOG (GNUNET_ERROR_TYPE_DEBUG, " to origin, type %u\n", + ntohs (mh->type)); + if (psize > 0) + { + psize += sizeof (to); + GNUNET_assert (size >= psize); + to.header.size = htons (psize); + to.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN); + to.tid = htonl (th->tunnel->tid); + memset (&to.oid, 0, sizeof (struct GNUNET_PeerIdentity)); + memset (&to.sender, 0, sizeof (struct GNUNET_PeerIdentity)); + memcpy (cbuf, &to, sizeof (to)); + } + } + else if (th->target == 0) + { + /* multicast */ + struct GNUNET_MESH_Multicast mc; + struct GNUNET_MessageHeader *mh; + + GNUNET_assert (size >= th->size); + mh = (struct GNUNET_MessageHeader *) &cbuf[sizeof (mc)]; + psize = th->notify (th->notify_cls, size - sizeof (mc), mh); + LOG (GNUNET_ERROR_TYPE_DEBUG, " multicast, type %u\n", + ntohs (mh->type)); + if (psize > 0) + { + psize += sizeof (mc); + GNUNET_assert (size >= psize); + mc.header.size = htons (psize); + mc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_MULTICAST); + mc.tid = htonl (th->tunnel->tid); + mc.mid = 0; + mc.ttl = 0; + memset (&mc.oid, 0, sizeof (struct GNUNET_PeerIdentity)); + memcpy (cbuf, &mc, sizeof (mc)); + } + } + else + { + /* unicast */ + struct GNUNET_MESH_Unicast uc; + struct GNUNET_MessageHeader *mh; + + GNUNET_assert (size >= th->size); + mh = (struct GNUNET_MessageHeader *) &cbuf[sizeof (uc)]; + psize = th->notify (th->notify_cls, size - sizeof (uc), mh); + LOG (GNUNET_ERROR_TYPE_DEBUG, " unicast, type %u\n", + ntohs (mh->type)); + if (psize > 0) + { + psize += sizeof (uc); + GNUNET_assert (size >= psize); + uc.header.size = htons (psize); + uc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_UNICAST); + uc.tid = htonl (th->tunnel->tid); + memset (&uc.oid, 0, sizeof (struct GNUNET_PeerIdentity)); + GNUNET_PEER_resolve (th->target, &uc.destination); + memcpy (cbuf, &uc, sizeof (uc)); + } + } + } + else + { + memcpy (cbuf, &th[1], th->size); + psize |