struct GNUNET_MESH_Handle* handle;
+ /* The message-type requested for this tunnel. Is only needed for pending
+ * by_tupe-tunnels
+ */
+ uint16_t message_type;
+
/* The context of the receive-function. */
void *ctx;
};
struct peer_list_element
{
struct GNUNET_PeerIdentity peer;
+
+ /* how many Message-Types can this peer receive */
+ unsigned int num_types;
+
+ /* array of message-types */
+ GNUNET_MESH_ApplicationType *types;
+
struct GNUNET_TRANSPORT_ATS_Information atsi;
struct peer_list_element *next, *prev;
};
struct peer_list connected_peers;
struct tunnel_list established_tunnels;
struct tunnel_list pending_tunnels;
+ struct tunnel_list pending_by_type_tunnels;
void *cls;
GNUNET_MESH_TunnelEndHandler *cleaner;
+ size_t hello_message_size;
+ uint16_t *hello_message;
};
static void
GNUNET_SCHEDULER_add_now(send_end_connect, tunnel);
}
+static void
+call_connect_handler (void *cls,
+ const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+ struct GNUNET_MESH_Tunnel *tunnel = cls;
+
+ tunnel->connect_handler (tunnel->handler_cls, &tunnel->peer,
+ NULL);
+ GNUNET_SCHEDULER_add_now (send_end_connect, tunnel);
+}
+
static void
core_startup (void *cls,
struct GNUNET_CORE_Handle *core,
handle->connected_to_core = GNUNET_YES;
}
+static size_t
+send_hello_message (void *cls, size_t size, void *buf)
+{
+ struct GNUNET_MESH_Handle *handle = cls;
+ struct GNUNET_MessageHeader *hdr = buf;
+
+ size_t sent = sizeof(struct GNUNET_MessageHeader) + handle->hello_message_size;
+
+ hdr->type = htons(GNUNET_MESSAGE_TYPE_MESH_HELLO);
+ hdr->size = htons(size);
+
+ memcpy(hdr+1, handle->hello_message, handle->hello_message_size);
+ return sent;
+}
+
+
/**
* Core calls this if we are connected to a new peer.
*
- * If core tells us that we are connected to ourself, we ignore it. Otherwise, the
- * peer is added to the connected_peers-list.
+ * The peer is added to the connected_peers-list.
*
*/
static void
const struct GNUNET_TRANSPORT_ATS_Information *atsi)
{
struct GNUNET_MESH_Handle *handle = cls;
- /* Check for connect-to-self-message, which we ignore */
- if (0 ==
- memcmp (peer, &handle->myself, sizeof (struct GNUNET_PeerIdentity)))
- return;
+ /* Send a hello to this peer */
+ GNUNET_CORE_notify_transmit_ready(handle->core,
+ GNUNET_NO,
+ 42,
+ GNUNET_TIME_UNIT_SECONDS,
+ peer,
+ sizeof(struct GNUNET_MessageHeader) + handle->hello_message_size,
+ &send_hello_message,
+ cls);
/* put the new peer into the list of connected peers */
struct peer_list_element *element =
GNUNET_malloc (sizeof (struct peer_list_element));
memcpy (&element->peer, peer, sizeof (struct GNUNET_PeerIdentity));
- memcpy (&element->atsi, atsi,
- sizeof (struct GNUNET_TRANSPORT_ATS_Information));
+
+ if (NULL != atsi)
+ memcpy (&element->atsi, atsi,
+ sizeof (struct GNUNET_TRANSPORT_ATS_Information));
GNUNET_CONTAINER_DLL_insert_after (handle->connected_peers.head,
handle->connected_peers.tail,
{
GNUNET_CONTAINER_DLL_remove (handle->connected_peers.head,
handle->connected_peers.tail, element);
+ GNUNET_free_non_null(element->types);
GNUNET_free (element);
}
{
/* disconnect tunnels */
/* outbound tunnels */
- if (telement->tunnel.connect_handler != NULL)
+ if (telement->tunnel.connect_handler != NULL && NULL != telement->tunnel.disconnect_handler)
telement->tunnel.disconnect_handler (telement->tunnel.handler_cls,
peer);
/* inbound tunnels */
- else
+ else if (NULL != handle->cleaner)
handle->cleaner (handle->cls, &telement->tunnel,
&telement->tunnel.ctx);
}
}
+/**
+ * Receive a message from core.
+ * This is a hello-message, containing the message-types the other peer can receive
+ */
+static int
+receive_hello (void *cls,
+ const struct GNUNET_PeerIdentity *other,
+ const struct GNUNET_MessageHeader *message,
+ const struct GNUNET_TRANSPORT_ATS_Information *atsi)
+{
+ struct GNUNET_MESH_Handle *handle = cls;
+ uint16_t *num = (uint16_t *) (message + 1);
+ uint16_t *ports = num + 1;
+ unsigned int i;
+
+ struct peer_list_element *element = handle->connected_peers.head;
+ while (element != NULL)
+ {
+ if (0 ==
+ memcmp (&element->peer, other, sizeof (struct GNUNET_PeerIdentity)))
+ break;
+ element = element->next;
+ }
+
+ /* TODO: add, not replace! */
+ /* TODO: if this changes anything: send new hello */
+ element->num_types = *num;
+ element->types = GNUNET_malloc (*num * sizeof (GNUNET_MESH_ApplicationType));
+
+ for (i = 0; i < *num; i++)
+ element->types[i] = (GNUNET_MESH_ApplicationType)ntohs (ports[i]);
+
+ struct tunnel_list_element *tunnel = handle->pending_by_type_tunnels.head;
+ while (tunnel != NULL)
+ {
+ struct tunnel_list_element *next = tunnel->next;
+ for (i = 0; i < *num; i++)
+ {
+ if (ntohs (ports[i]) == tunnel->tunnel.message_type)
+ {
+ GNUNET_CONTAINER_DLL_remove (handle->pending_tunnels.head,
+ handle->pending_tunnels.tail,
+ tunnel);
+ GNUNET_CONTAINER_DLL_insert_after (handle->established_tunnels.
+ head,
+ handle->established_tunnels.
+ tail,
+ handle->established_tunnels.
+ tail, tunnel);
+ tunnel->tunnel.connect_handler (tunnel->tunnel.handler_cls,
+ &tunnel->tunnel.peer, atsi);
+ GNUNET_SCHEDULER_add_now (send_end_connect, tunnel);
+ break;
+ }
+ }
+ if (ntohs (ports[i]) == tunnel->tunnel.message_type)
+ tunnel = next;
+ else
+ tunnel = tunnel->next;
+ }
+ return GNUNET_OK;
+}
+
/**
* Receive a message from core.
*/
}
return handler->callback (handle->cls, &tunnel->tunnel,
- &tunnel->tunnel.ctx, rmessage, atsi);
+ &tunnel->tunnel.ctx, other, rmessage, atsi);
}
+struct GNUNET_MESH_Tunnel *
+GNUNET_MESH_peer_request_connect_by_type (struct GNUNET_MESH_Handle *handle,
+ struct GNUNET_TIME_Relative timeout,
+ GNUNET_MESH_ApplicationType message_type,
+ GNUNET_MESH_TunnelConnectHandler
+ connect_handler,
+ GNUNET_MESH_TunnelDisconnectHandler
+ disconnect_handler,
+ void *handler_cls)
+{
+ /* Look in the list of connected peers */
+ struct peer_list_element *element = handle->connected_peers.head;
+ while (element != NULL)
+ {
+ unsigned int i;
+ for (i = 0; i < element->num_types; i++)
+ if (message_type == element->types[i])
+ return GNUNET_MESH_peer_request_connect_all (handle, timeout, 1,
+ &handle->myself,
+ connect_handler,
+ disconnect_handler,
+ handler_cls);
+ element = element->next;
+ }
+
+ /* Put into pending list */
+ struct tunnel_list_element *tunnel =
+ GNUNET_malloc (sizeof (struct tunnel_list_element));
+
+ tunnel->tunnel.connect_handler = connect_handler;
+ tunnel->tunnel.disconnect_handler = disconnect_handler;
+ tunnel->tunnel.handler_cls = handler_cls;
+ tunnel->tunnel.ctx = NULL;
+ tunnel->tunnel.handle = handle;
+ memcpy (&tunnel->tunnel.id.initiator, &handle->myself,
+ sizeof (struct GNUNET_PeerIdentity));
+ tunnel->tunnel.id.id = current_id++;
+ tunnel->tunnel.message_type = message_type;
+
+ GNUNET_CONTAINER_DLL_insert_after (handle->pending_by_type_tunnels.head,
+ handle->pending_by_type_tunnels.tail,
+ handle->pending_by_type_tunnels.tail,
+ tunnel);
+ return &tunnel->tunnel;
+}
+
+
struct GNUNET_MESH_Tunnel *
GNUNET_MESH_peer_request_connect_all (struct GNUNET_MESH_Handle *handle,
handle->established_tunnels.tail,
handle->established_tunnels.tail,
tunnel);
- connect_handler (handler_cls, &element->peer, &element->atsi);
+ GNUNET_SCHEDULER_add_now(call_connect_handler, tunnel);
}
else if (0 ==
memcmp (peers, &handle->myself,
(void) GNUNET_CORE_peer_request_connect (handle->core,
timeout,
peers,
- NULL, NULL);
+ NULL, NULL);
}
return &tunnel->tunnel;
return sent;
}
+
+/**
+ * Ask the mesh to call "notify" once it is ready to transmit the
+ * given number of bytes to the specified "target". If we are not yet
+ * connected to the specified peer, a call to this function will cause
+ * us to try to establish a connection.
+ *
+ * @param tunnel tunnel to use for transmission
+ * @param cork is corking allowed for this transmission?
+ * @param priority how important is the message?
+ * @param maxdelay how long can the message wait?
+ * @param target destination for the message, NULL for multicast to all tunnel targets
+ * @param notify_size how many bytes of buffer space does notify want?
+ * @param notify function to call when buffer space is available;
+ * will be called with NULL on timeout or if the overall queue
+ * for this peer is larger than queue_size and this is currently
+ * the message with the lowest priority
+ * @param notify_cls closure for notify
+ * @return non-NULL if the notify callback was queued,
+ * NULL if we can not even queue the request (insufficient
+ * memory); if NULL is returned, "notify" will NOT be called.
+ */
struct GNUNET_MESH_TransmitHandle *
GNUNET_MESH_notify_transmit_ready (struct
GNUNET_MESH_Tunnel
struct
GNUNET_TIME_Relative
maxdelay,
+ const struct GNUNET_PeerIdentity *target,
size_t
notify_size,
GNUNET_CONNECTION_TransmitReadyNotify
cls->notify = notify;
cls->tunnel = tunnel;
GNUNET_CORE_notify_transmit_ready(tunnel->handle->core,
+ cork,
priority,
maxdelay,
&tunnel->peer,
return (struct GNUNET_MESH_TransmitHandle*) 1;
}
+void build_hello_message(struct GNUNET_MESH_Handle* handle, int num)
+{
+ handle->hello_message_size = sizeof(uint16_t) + /* For the number of types */
+ num * sizeof(uint16_t); /* For the types */
+
+ uint16_t *nums = GNUNET_malloc(handle->hello_message_size);
+ uint16_t *types = nums + 1;
+
+ *nums = num;
+
+ unsigned int i;
+ for(i = 0; i < num; i++)
+ types[i] = handle->handlers[i].type;
+
+ handle->hello_message = nums;
+}
+
struct GNUNET_MESH_Handle *
GNUNET_MESH_connect (const struct
*cfg, void *cls,
GNUNET_MESH_TunnelEndHandler
cleaner,
- const struct GNUNET_MESH_MessageHandler *handlers)
+ const struct GNUNET_MESH_MessageHandler *handlers,
+ const GNUNET_MESH_ApplicationType *stypes)
{
struct GNUNET_MESH_Handle *ret =
GNUNET_malloc (sizeof (struct GNUNET_MESH_Handle));
ret->connected_peers.tail = NULL;
ret->cleaner = cleaner;
ret->cls = cls;
-
+
const struct GNUNET_MESH_MessageHandler *it;
unsigned int len = 1;
for (it = handlers; it->callback != NULL; it++)
memcpy (ret->handlers, handlers,
len * sizeof (struct GNUNET_MESH_MessageHandler));
+ build_hello_message(ret, len);
+
const static struct GNUNET_CORE_MessageHandler core_handlers[] = {
{&core_receive, GNUNET_MESSAGE_TYPE_MESH, 0},
+ {&receive_hello, GNUNET_MESSAGE_TYPE_MESH_HELLO, 0},
{NULL, 0, 0}
};
GNUNET_MESH_disconnect (struct GNUNET_MESH_Handle *handle)
{
GNUNET_free (handle->handlers);
+ GNUNET_free (handle->hello_message);
GNUNET_CORE_disconnect (handle->core);
struct peer_list_element *element = handle->connected_peers.head;
while (element != NULL)
{
struct peer_list_element *next = element->next;
+ GNUNET_free_non_null(element->types);
GNUNET_free (element);
element = next;
}