X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Fmesh%2Fgnunet-service-mesh_peer.c;h=515c4b4f0fdae87e218465702fca5530893d6019;hb=0f6d24a229e9149db26a4e667ed25032d19f533a;hp=43c6259e3cda71bba631a12a5e6e6aca10b62bf3;hpb=0b2fd8bc93e74fb4094e0e94ad7344da99b81874;p=oweals%2Fgnunet.git diff --git a/src/mesh/gnunet-service-mesh_peer.c b/src/mesh/gnunet-service-mesh_peer.c index 43c6259e3..515c4b4f0 100644 --- a/src/mesh/gnunet-service-mesh_peer.c +++ b/src/mesh/gnunet-service-mesh_peer.c @@ -22,15 +22,90 @@ #include "platform.h" #include "gnunet_util_lib.h" +#include "gnunet_core_service.h" +#include "gnunet_statistics_service.h" + +#include "mesh_protocol_enc.h" + #include "gnunet-service-mesh_peer.h" #include "gnunet-service-mesh_dht.h" #include "gnunet-service-mesh_connection.h" +#include "gnunet-service-mesh_local.h" +#include "gnunet-service-mesh_tunnel.h" #include "mesh_path.h" +#define LOG(level, ...) GNUNET_log_from (level,"mesh-p2p",__VA_ARGS__) + /******************************************************************************/ /******************************** STRUCTS **********************************/ /******************************************************************************/ +/** + * Struct containing info about a queued transmission to this peer + */ +struct MeshPeerQueue +{ + /** + * DLL next + */ + struct MeshPeerQueue *next; + + /** + * DLL previous + */ + struct MeshPeerQueue *prev; + + /** + * Peer this transmission is directed to. + */ + struct MeshPeer *peer; + + /** + * Connection this message belongs to. + */ + struct MeshConnection *c; + + /** + * Is FWD in c? + */ + int fwd; + + /** + * Channel this message belongs to, if known. + */ + struct MeshChannel *ch; + + /** + * Pointer to info stucture used as cls. + */ + void *cls; + + /** + * Type of message + */ + uint16_t type; + + /** + * Size of the message + */ + size_t size; + + /** + * Set when this message starts waiting for CORE. + */ + struct GNUNET_TIME_Absolute start_waiting; + + /** + * Function to call on sending. + */ + GMP_sent callback; + + /** + * Closure for callback. + */ + void *callback_cls; +}; + /** * Struct containing all information regarding a given peer */ @@ -64,7 +139,7 @@ struct MeshPeer /** * Tunnel to this peer, if any. */ - struct MeshTunnel2 *tunnel; + struct MeshTunnel3 *tunnel; /** * Connections that go through this peer, indexed by tid; @@ -97,6 +172,21 @@ struct MeshPeer /******************************* GLOBALS ***********************************/ /******************************************************************************/ +/** + * Global handle to the statistics service. + */ +extern struct GNUNET_STATISTICS_Handle *stats; + +/** + * Local peer own ID (full value). + */ +extern struct GNUNET_PeerIdentity my_full_id; + +/** + * Local peer own ID (short) + */ +extern GNUNET_PEER_Id myid; + /** * Peers known, indexed by PeerIdentity (MeshPeer). */ @@ -112,11 +202,178 @@ static unsigned long long max_peers; */ static unsigned long long drop_percent; +/** + * Handle to communicate with core. + */ +static struct GNUNET_CORE_Handle *core_handle; + + /******************************************************************************/ /***************************** CORE CALLBACKS *********************************/ /******************************************************************************/ +/** + * Iterator to notify all connections of a broken link. Mark connections + * to destroy after all traffic has been sent. + * + * @param cls Closure (peer disconnected). + * @param key Current key code (peer id). + * @param value Value in the hash map (connection). + * + * @return GNUNET_YES if we should continue to iterate, + * GNUNET_NO if not. + */ +static int +notify_broken (void *cls, + const struct GNUNET_HashCode *key, + void *value) +{ + struct MeshPeer *peer = cls; + struct MeshConnection *c = value; + + GMC_notify_broken (c, peer); + + return GNUNET_YES; +} + + +/** + * Method called whenever a given peer connects. + * + * @param cls closure + * @param peer peer identity this notification is about + */ +static void +core_connect (void *cls, const struct GNUNET_PeerIdentity *peer) +{ + struct MeshPeer *mp; + struct MeshPeerPath *path; + + LOG (GNUNET_ERROR_TYPE_DEBUG, "Peer connected\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, " %s\n", GNUNET_i2s (&my_full_id)); + mp = GMP_get (peer); + if (myid == mp->id) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, " (self)\n"); + path = path_new (1); + } + else + { + LOG (GNUNET_ERROR_TYPE_DEBUG, " %s\n", GNUNET_i2s (peer)); + path = path_new (2); + path->peers[1] = mp->id; + GNUNET_PEER_change_rc (mp->id, 1); + GNUNET_STATISTICS_update (stats, "# peers", 1, GNUNET_NO); + } + path->peers[0] = myid; + GNUNET_PEER_change_rc (myid, 1); + GMP_add_path (mp, path, GNUNET_YES); + + mp->connections = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_YES); + return; +} + + +/** + * Method called whenever a peer disconnects. + * + * @param cls closure + * @param peer peer identity this notification is about + */ +static void +core_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer) +{ + struct MeshPeer *pi; + + LOG (GNUNET_ERROR_TYPE_DEBUG, "Peer disconnected\n"); + pi = GNUNET_CONTAINER_multipeermap_get (peers, peer); + if (NULL == pi) + { + GNUNET_break (0); + return; + } + + GNUNET_CONTAINER_multihashmap_iterate (pi->connections, ¬ify_broken, pi); + GNUNET_CONTAINER_multihashmap_destroy (pi->connections); + pi->connections = NULL; + if (NULL != pi->core_transmit) + { + GNUNET_CORE_notify_transmit_ready_cancel (pi->core_transmit); + pi->core_transmit = NULL; + } + if (myid == pi->id) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, " (self)\n"); + } + GNUNET_STATISTICS_update (stats, "# peers", -1, GNUNET_NO); + + return; +} + + +/** + * Functions to handle messages from core + */ +static struct GNUNET_CORE_MessageHandler core_handlers[] = { + {&GMC_handle_create, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE, + 0}, + {&GMC_handle_confirm, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK, + sizeof (struct GNUNET_MESH_ConnectionACK)}, + {&GMC_handle_broken, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN, + sizeof (struct GNUNET_MESH_ConnectionBroken)}, + {&GMC_handle_destroy, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY, + sizeof (struct GNUNET_MESH_ConnectionDestroy)}, + {&GMC_handle_keepalive, GNUNET_MESSAGE_TYPE_MESH_FWD_KEEPALIVE, + sizeof (struct GNUNET_MESH_ConnectionKeepAlive)}, + {&GMC_handle_keepalive, GNUNET_MESSAGE_TYPE_MESH_BCK_KEEPALIVE, + sizeof (struct GNUNET_MESH_ConnectionKeepAlive)}, + {&GMC_handle_ack, GNUNET_MESSAGE_TYPE_MESH_ACK, + sizeof (struct GNUNET_MESH_ACK)}, + {&GMC_handle_poll, GNUNET_MESSAGE_TYPE_MESH_POLL, + sizeof (struct GNUNET_MESH_Poll)}, + {&GMC_handle_encrypted, GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED, 0}, + {NULL, 0, 0} +}; + + +/** + * To be called on core init/fail. + * + * @param cls Closure (config) + * @param identity the public identity of this peer + */ +static void +core_init (void *cls, + const struct GNUNET_PeerIdentity *identity) +{ + const struct GNUNET_CONFIGURATION_Handle *c = cls; + static int i = 0; + + LOG (GNUNET_ERROR_TYPE_DEBUG, "Core init\n"); + if (0 != memcmp (identity, &my_full_id, sizeof (my_full_id))) + { + LOG (GNUNET_ERROR_TYPE_ERROR, _("Wrong CORE service\n")); + LOG (GNUNET_ERROR_TYPE_ERROR, " core id %s\n", GNUNET_i2s (identity)); + LOG (GNUNET_ERROR_TYPE_ERROR, " my id %s\n", GNUNET_i2s (&my_full_id)); + GNUNET_CORE_disconnect (core_handle); + core_handle = GNUNET_CORE_connect (c, /* Main configuration */ + NULL, /* Closure passed to MESH functions */ + &core_init, /* Call core_init once connected */ + &core_connect, /* Handle connects */ + &core_disconnect, /* remove peers on disconnects */ + NULL, /* Don't notify about all incoming messages */ + GNUNET_NO, /* For header only in notification */ + NULL, /* Don't notify about all outbound messages */ + GNUNET_NO, /* For header-only out notification */ + core_handlers); /* Register these handlers */ + if (10 < i++) + GNUNET_abort(); + } + GML_start (); + return; +} + /** * Core callback to write a pre-constructed data packet to core buffer * @@ -159,11 +416,11 @@ send_core_connection_create (struct MeshConnection *c, size_t size, void *buf) { struct GNUNET_MESH_ConnectionCreate *msg; struct GNUNET_PeerIdentity *peer_ptr; - struct MeshPeerPath *p = c->path; + const struct MeshPeerPath *p = GMC_get_path (c); size_t size_needed; int i; - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION CREATE...\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION CREATE...\n"); size_needed = sizeof (struct GNUNET_MESH_ConnectionCreate) + p->length * sizeof (struct GNUNET_PeerIdentity); @@ -176,7 +433,7 @@ send_core_connection_create (struct MeshConnection *c, size_t size, void *buf) msg = (struct GNUNET_MESH_ConnectionCreate *) buf; msg->header.size = htons (size_needed); msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE); - msg->cid = c->id; + msg->cid = *GMC_get_id (c); peer_ptr = (struct GNUNET_PeerIdentity *) &msg[1]; for (i = 0; i < p->length; i++) @@ -184,8 +441,9 @@ send_core_connection_create (struct MeshConnection *c, size_t size, void *buf) GNUNET_PEER_resolve (p->peers[i], peer_ptr++); } - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, - "CONNECTION CREATE (%u bytes long) sent!\n", size_needed); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "CONNECTION CREATE (%u bytes long) sent!\n", + size_needed); return size_needed; } @@ -203,10 +461,8 @@ static size_t send_core_connection_ack (struct MeshConnection *c, size_t size, void *buf) { struct GNUNET_MESH_ConnectionACK *msg = buf; - struct MeshTunnel2 *t = c->t; - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION ACK...\n"); - GNUNET_assert (NULL != t); + LOG (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION ACK...\n"); if (sizeof (struct GNUNET_MESH_ConnectionACK) > size) { GNUNET_break (0); @@ -214,12 +470,12 @@ send_core_connection_ack (struct MeshConnection *c, size_t size, void *buf) } msg->header.size = htons (sizeof (struct GNUNET_MESH_ConnectionACK)); msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK); - msg->cid = c->id; + msg->cid = *GMC_get_id (c); msg->reserved = 0; /* TODO add signature */ - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CONNECTION ACK sent!\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, "CONNECTION ACK sent!\n"); return sizeof (struct GNUNET_MESH_ConnectionACK); } @@ -243,7 +499,7 @@ shutdown_tunnel (void *cls, void *value) { struct MeshPeer *p = value; - struct MeshTunnel2 *t = p->tunnel; + struct MeshTunnel3 *t = p->tunnel; if (NULL != t) GMT_destroy (t); @@ -273,24 +529,24 @@ peer_destroy (struct MeshPeer *peer) GNUNET_CONTAINER_multipeermap_remove (peers, &id, peer)) { GNUNET_break (0); - GNUNET_log (GNUNET_ERROR_TYPE_WARNING, + LOG (GNUNET_ERROR_TYPE_WARNING, "removing peer %s, not in peermap\n", GNUNET_i2s (&id)); } - if (NULL != peer->search_h) - { - GMD_search_stop (peer->search_h); - } - p = peer->path_head; - while (NULL != p) - { - nextp = p->next; - GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, p); - path_destroy (p); - p = nextp; - } - tunnel_destroy_empty (peer->tunnel); - GNUNET_free (peer); - return GNUNET_OK; + if (NULL != peer->search_h) + { + GMD_search_stop (peer->search_h); + } + p = peer->path_head; + while (NULL != p) + { + nextp = p->next; + GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, p); + path_destroy (p); + p = nextp; + } + GMT_destroy_empty (peer->tunnel); + GNUNET_free (peer); + return GNUNET_OK; } @@ -389,293 +645,169 @@ peer_delete_oldest (void) /** - * Retrieve the MeshPeer stucture associated with the peer, create one - * and insert it in the appropriate structures if the peer is not known yet. + * Choose the best path towards a peer considering the tunnel properties. * - * @param peer Full identity of the peer. + * @param peer The destination peer. * - * @return Existing or newly created peer info. + * @return Best current known path towards the peer, if any. */ -static struct MeshPeer * -peer_get (const struct GNUNET_PeerIdentity *peer_id) +static struct MeshPeerPath * +peer_get_best_path (const struct MeshPeer *peer) { - struct MeshPeer *peer; + struct MeshPeerPath *best_p; + struct MeshPeerPath *p; + unsigned int best_cost; + unsigned int cost; - peer = GNUNET_CONTAINER_multipeermap_get (peers, peer_id); - if (NULL == peer) + best_cost = UINT_MAX; + best_p = NULL; + for (p = peer->path_head; NULL != p; p = p->next) { - peer = GNUNET_new (struct MeshPeer); - if (GNUNET_CONTAINER_multipeermap_size (peers) > max_peers) + if (GNUNET_YES == GMT_is_path_used (peer->tunnel, p)) + continue; /* If path is already in use, skip it. */ + + if ((cost = GMT_get_path_cost (peer->tunnel, p)) < best_cost) { - peer_delete_oldest (); + best_cost = cost; + best_p = p; } - GNUNET_CONTAINER_multipeermap_put (peers, peer_id, peer, - GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST); - peer->id = GNUNET_PEER_intern (peer_id); } - peer->last_contact = GNUNET_TIME_absolute_get(); - - return peer; + return best_p; } -/** - * Retrieve the MeshPeer stucture associated with the peer, create one - * and insert it in the appropriate structures if the peer is not known yet. - * - * @param peer Short identity of the peer. - * - * @return Existing or newly created peer info. - */ -static struct MeshPeer * -peer_get_short (const GNUNET_PEER_Id peer) +static int +queue_is_sendable (struct MeshPeerQueue *q) { - return peer_get (GNUNET_PEER_resolve2 (peer)); + /* Is PID-independent? */ + switch (q->type) + { + case GNUNET_MESSAGE_TYPE_MESH_ACK: + case GNUNET_MESSAGE_TYPE_MESH_POLL: + return GNUNET_YES; + } + + if (GMC_is_sendable (q->c, q->fwd)) + return GNUNET_YES; + + return GNUNET_NO; } /** - * Get a cost of a path for a peer considering existing tunnel connections. + * Get first sendable message. * - * @param peer Peer towards which the path is considered. - * @param path Candidate path. + * @param peer The destination peer. * - * @return Cost of the path (path length + number of overlapping nodes) + * @return Best current known path towards the peer, if any. */ -static unsigned int -peer_get_path_cost (const struct MeshPeer *peer, - const struct MeshPeerPath *path) +static struct MeshPeerQueue * +peer_get_first_message (const struct MeshPeer *peer) { - struct MeshConnection *c; - unsigned int overlap; - unsigned int i; - unsigned int j; - - if (NULL == path) - return 0; - - overlap = 0; - GNUNET_assert (NULL != peer->tunnel); + struct MeshPeerQueue *q; - for (i = 0; i < path->length; i++) + for (q = peer->queue_head; NULL != q; q = q->next) { - for (c = peer->tunnel->connection_head; NULL != c; c = c->next) - { - for (j = 0; j < c->path->length; j++) - { - if (path->peers[i] == c->path->peers[j]) - { - overlap++; - break; - } - } - } + if (queue_is_sendable (q)) + return q; } - return (path->length + overlap) * (path->score * -1); + + return NULL; } /** - * Choose the best path towards a peer considering the tunnel properties. - * - * @param peer The destination peer. + * Function to process paths received for a new peer addition. The recorded + * paths form the initial tunnel, which can be optimized later. + * Called on each result obtained for the DHT search. * - * @return Best current known path towards the peer, if any. + * @param cls closure + * @param path */ -static struct MeshPeerPath * -peer_get_best_path (const struct MeshPeer *peer) +static void +search_handler (void *cls, const struct MeshPeerPath *path) { - struct MeshPeerPath *best_p; - struct MeshPeerPath *p; - struct MeshConnection *c; - unsigned int best_cost; - unsigned int cost; + struct MeshPeer *peer = cls; + unsigned int connection_count; - best_cost = UINT_MAX; - best_p = NULL; - for (p = peer->path_head; NULL != p; p = p->next) + GMP_add_path_to_all (path, GNUNET_NO); + + /* Count connections */ + connection_count = GMT_count_connections (peer->tunnel); + + /* If we already have 3 (or more (?!)) connections, it's enough */ + if (3 <= connection_count) + return; + + if (MESH_TUNNEL3_SEARCHING == GMT_get_state (peer->tunnel)) { - for (c = peer->tunnel->connection_head; NULL != c; c = c->next) - if (c->path == p) - break; - if (NULL != c) - continue; /* If path is in use in a connection, skip it. */ - - if ((cost = peer_get_path_cost (peer, p)) < best_cost) - { - best_cost = cost; - best_p = p; - } + LOG (GNUNET_ERROR_TYPE_DEBUG, " ... connect!\n"); + GMP_connect (peer); } - return best_p; + return; } /** - * Add the path to the peer and update the path used to reach it in case this - * is the shortest. + * Free a transmission that was already queued with all resources + * associated to the request. * - * @param peer_info Destination peer to add the path to. - * @param path New path to add. Last peer must be the peer in arg 1. - * Path will be either used of freed if already known. - * @param trusted Do we trust that this path is real? + * @param queue Queue handler to cancel. + * @param clear_cls Is it necessary to free associated cls? */ -void -peer_add_path (struct MeshPeer *peer_info, struct MeshPeerPath *path, - int trusted) +static void +queue_destroy (struct MeshPeerQueue *queue, int clear_cls) { - struct MeshPeerPath *aux; - unsigned int l; - unsigned int l2; + struct MeshPeer *peer; - if ((NULL == peer_info) || (NULL == path)) - { - GNUNET_break (0); - path_destroy (path); - return; - } - if (path->peers[path->length - 1] != peer_info->id) - { - GNUNET_break (0); - path_destroy (path); - return; - } - if (2 >= path->length && GNUNET_NO == trusted) - { - /* Only allow CORE to tell us about direct paths */ - path_destroy (path); - return; - } - for (l = 1; l < path->length; l++) - { - if (path->peers[l] == myid) - { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shortening path by %u\n", l); - for (l2 = 0; l2 < path->length - l; l2++) - { - path->peers[l2] = path->peers[l + l2]; - } - path->length -= l; - l = 1; - path->peers = - GNUNET_realloc (path->peers, path->length * sizeof (GNUNET_PEER_Id)); - } - } - - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "adding path [%u] to peer %s\n", - path->length, peer2s (peer_info)); - - l = path_get_length (path); - if (0 == l) - { - path_destroy (path); - return; - } - - GNUNET_assert (peer_info->id == path->peers[path->length - 1]); - for (aux = peer_info->path_head; aux != NULL; aux = aux->next) - { - l2 = path_get_length (aux); - if (l2 > l) - { - GNUNET_CONTAINER_DLL_insert_before (peer_info->path_head, - peer_info->path_tail, aux, path); - return; - } - else - { - if (l2 == l && memcmp (path->peers, aux->peers, l) == 0) - { - path_destroy (path); - return; - } - } - } - GNUNET_CONTAINER_DLL_insert_tail (peer_info->path_head, peer_info->path_tail, - path); - return; -} - - -/** - * Add the path to the origin peer and update the path used to reach it in case - * this is the shortest. - * The path is given in peer_info -> destination, therefore we turn the path - * upside down first. - * - * @param peer_info Peer to add the path to, being the origin of the path. - * @param path New path to add after being inversed. - * Path will be either used or freed. - * @param trusted Do we trust that this path is real? - */ -static void -peer_add_path_to_origin (struct MeshPeer *peer_info, - struct MeshPeerPath *path, int trusted) -{ - if (NULL == path) - return; - path_invert (path); - peer_add_path (peer_info, path, trusted); -} + peer = queue->peer; + GNUNET_assert (NULL != queue->c); + if (GNUNET_YES == clear_cls) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, "# queue destroy type %s\n", + GNUNET_MESH_DEBUG_M2S (queue->type)); + switch (queue->type) + { + case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY: + case GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY: + LOG (GNUNET_ERROR_TYPE_INFO, "destroying a DESTROY message\n"); + /* fall through */ + case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED: + case GNUNET_MESSAGE_TYPE_MESH_ACK: + case GNUNET_MESSAGE_TYPE_MESH_POLL: + case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK: + case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE: + case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN: + LOG (GNUNET_ERROR_TYPE_DEBUG, "# prebuilt message\n");; + GNUNET_free_non_null (queue->cls); + break; -/** - * Adds a path to the peer_infos of all the peers in the path - * - * @param p Path to process. - * @param confirmed Whether we know if the path works or not. - */ -static void -path_add_to_peers (struct MeshPeerPath *p, int confirmed) -{ - unsigned int i; + default: + GNUNET_break (0); + LOG (GNUNET_ERROR_TYPE_ERROR, "# type %s unknown!\n", + GNUNET_MESH_DEBUG_M2S (queue->type)); + } + } + GNUNET_CONTAINER_DLL_remove (peer->queue_head, peer->queue_tail, queue); - /* TODO: invert and add */ - for (i = 0; i < p->length && p->peers[i] != myid; i++) /* skip'em */ ; - for (i++; i < p->length; i++) + if (queue->type != GNUNET_MESSAGE_TYPE_MESH_ACK && + queue->type != GNUNET_MESSAGE_TYPE_MESH_POLL) { - struct MeshPeer *aux; - struct MeshPeerPath *copy; - - aux = peer_get_short (p->peers[i]); - copy = path_duplicate (p); - copy->length = i + 1; - peer_add_path (aux, copy, p->length < 3 ? GNUNET_NO : confirmed); + peer->queue_n--; } -} - - -/** - * Function to process paths received for a new peer addition. The recorded - * paths form the initial tunnel, which can be optimized later. - * Called on each result obtained for the DHT search. - * - * @param cls closure - * @param path - */ -static void -search_handler (void *cls, struct MeshPeerPath *path) -{ - struct MeshPeer *peer = cls; - unsigned int connection_count; - - path_add_to_peers (path, GNUNET_NO); - - /* Count connections */ - connection_count = GMC_count (peer->tunnel->connection_head); - - /* If we already have 3 (or more (?!)) connections, it's enough */ - if (3 <= connection_count) - return; - if (peer->tunnel->state == MESH_TUNNEL_SEARCHING) + if (NULL != queue->callback) { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ... connect!\n"); - GMP_connect (peer); + LOG (GNUNET_ERROR_TYPE_DEBUG, "# Calling callback\n"); + queue->callback (queue->callback_cls, + queue->c, queue->type, + queue->fwd, queue->size, + GNUNET_TIME_absolute_get_duration (queue->start_waiting)); } - return; -} + GNUNET_free (queue); +} /** * Core callback to write a queued packet to core buffer @@ -690,24 +822,17 @@ static size_t queue_send (void *cls, size_t size, void *buf) { struct MeshPeer *peer = cls; - struct MeshFlowControl *fc; struct MeshConnection *c; - struct GNUNET_MessageHeader *msg; struct MeshPeerQueue *queue; - struct MeshTunnel2 *t; - struct MeshChannel *ch; const struct GNUNET_PeerIdentity *dst_id; size_t data_size; - uint32_t pid; - uint16_t type; - int fwd; peer->core_transmit = NULL; - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* Queue send (max %u)\n", size); + LOG (GNUNET_ERROR_TYPE_DEBUG, "* Queue send (max %u)\n", size); if (NULL == buf || 0 == size) { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* Buffer size 0.\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, "* Buffer size 0.\n"); return 0; } @@ -719,15 +844,13 @@ queue_send (void *cls, size_t size, void *buf) return 0; } c = queue->c; - fwd = queue->fwd; - fc = fwd ? &c->fwd_fc : &c->bck_fc; dst_id = GNUNET_PEER_resolve2 (peer->id); - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* towards %s\n", GNUNET_i2s (dst_id)); + LOG (GNUNET_ERROR_TYPE_DEBUG, "* towards %s\n", GNUNET_i2s (dst_id)); /* Check if buffer size is enough for the message */ if (queue->size > size) { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* not enough room, reissue\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, "* not enough room, reissue\n"); peer->core_transmit = GNUNET_CORE_notify_transmit_ready (core_handle, GNUNET_NO, @@ -739,10 +862,7 @@ queue_send (void *cls, size_t size, void *buf) peer); return 0; } - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* size %u ok\n", queue->size); - - t = (NULL != c) ? c->t : NULL; - type = 0; + LOG (GNUNET_ERROR_TYPE_DEBUG, "* size %u ok\n", queue->size); /* Fill buf */ switch (queue->type) @@ -750,26 +870,23 @@ queue_send (void *cls, size_t size, void *buf) case GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY: case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY: case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN: - case GNUNET_MESSAGE_TYPE_MESH_FWD: - case GNUNET_MESSAGE_TYPE_MESH_BCK: + case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED: case GNUNET_MESSAGE_TYPE_MESH_ACK: case GNUNET_MESSAGE_TYPE_MESH_POLL: - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + LOG (GNUNET_ERROR_TYPE_DEBUG, "* raw: %s\n", GNUNET_MESH_DEBUG_M2S (queue->type)); data_size = send_core_data_raw (queue->cls, size, buf); - msg = (struct GNUNET_MessageHeader *) buf; - type = ntohs (msg->type); break; case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE: - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* path create\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, "* path create\n"); if (GMC_is_origin (c, GNUNET_YES)) data_size = send_core_connection_create (queue->c, size, buf); else data_size = send_core_data_raw (queue->cls, size, buf); break; case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK: - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* path ack\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, "* path ack\n"); if (GMC_is_origin (c, GNUNET_NO) || GMC_is_origin (c, GNUNET_YES)) data_size = send_core_connection_ack (queue->c, size, buf); @@ -785,44 +902,30 @@ queue_send (void *cls, size_t size, void *buf) break; default: GNUNET_break (0); - GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "* type unknown: %u\n", + LOG (GNUNET_ERROR_TYPE_WARNING, "* type unknown: %u\n", queue->type); data_size = 0; } if (0 < drop_percent && - GNUNET_CRYPTO_random_u32(GNUNET_CRYPTO_QUALITY_WEAK, 101) < drop_percent) + GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, 101) < drop_percent) { - GNUNET_log (GNUNET_ERROR_TYPE_WARNING, + LOG (GNUNET_ERROR_TYPE_WARNING, "Dropping message of type %s\n", GNUNET_MESH_DEBUG_M2S (queue->type)); data_size = 0; } /* Free queue, but cls was freed by send_core_* */ - ch = queue->ch; queue_destroy (queue, GNUNET_NO); - /* Send ACK if needed, after accounting for sent ID in fc->queue_n */ - switch (type) - { - case GNUNET_MESSAGE_TYPE_MESH_FWD: - case GNUNET_MESSAGE_TYPE_MESH_BCK: - pid = ntohl ( ((struct GNUNET_MESH_Encrypted *) buf)->pid ); - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* accounting pid %u\n", pid); - fc->last_pid_sent = pid; - send_ack (c, ch, fwd); - break; - default: - break; - } - /* If more data in queue, send next */ queue = peer_get_first_message (peer); if (NULL != queue) { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* more data!\n"); - if (NULL == peer->core_transmit) { + LOG (GNUNET_ERROR_TYPE_DEBUG, "* more data!\n"); + if (NULL == peer->core_transmit) + { peer->core_transmit = GNUNET_CORE_notify_transmit_ready(core_handle, 0, @@ -832,174 +935,33 @@ queue_send (void *cls, size_t size, void *buf) queue->size, &queue_send, peer); + queue->start_waiting = GNUNET_TIME_absolute_get (); } else { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + LOG (GNUNET_ERROR_TYPE_DEBUG, "* tmt rdy called somewhere else\n"); } - if (GNUNET_SCHEDULER_NO_TASK == fc->poll_task) - { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* starting poll timeout\n"); - fc->poll_task = - GNUNET_SCHEDULER_add_delayed (fc->poll_time, &connection_poll, fc); - } +// GMC_start_poll (); FIXME needed? } else { - if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task) - { - GNUNET_SCHEDULER_cancel (fc->poll_task); - fc->poll_task = GNUNET_SCHEDULER_NO_TASK; - } - } - if (NULL != c) - { - c->pending_messages--; - if (GNUNET_YES == c->destroy && 0 == c->pending_messages) - { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* destroying connection!\n"); - GMC_destroy (c); - } +// GMC_stop_poll(); FIXME needed? } - if (NULL != t) - { - t->pending_messages--; - if (GNUNET_YES == t->destroy && 0 == t->pending_messages) - { -// GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* destroying tunnel!\n"); - tunnel_destroy (t); - } - } - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* Return %d\n", data_size); + LOG (GNUNET_ERROR_TYPE_DEBUG, "* Return %d\n", data_size); return data_size; } - -/** - * Get first sendable message. - * - * @param peer The destination peer. - * - * @return Best current known path towards the peer, if any. - */ -static struct MeshPeerQueue * -peer_get_first_message (const struct MeshPeer *peer) -{ - struct MeshPeerQueue *q; - - for (q = peer->queue_head; NULL != q; q = q->next) - { - if (queue_is_sendable (q)) - return q; - } - - return NULL; -} - - -static int -queue_is_sendable (struct MeshPeerQueue *q) -{ - struct MeshFlowControl *fc; - - /* Is PID-independent? */ - switch (q->type) - { - case GNUNET_MESSAGE_TYPE_MESH_ACK: - case GNUNET_MESSAGE_TYPE_MESH_POLL: - return GNUNET_YES; - } - - /* Is PID allowed? */ - fc = q->fwd ? &q->c->fwd_fc : &q->c->bck_fc; - if (GMC_is_pid_bigger (fc->last_ack_recv, fc->last_pid_sent)) - return GNUNET_YES; - - return GNUNET_NO; -} - - /******************************************************************************/ /******************************** API ***********************************/ /******************************************************************************/ - -/** - * Free a transmission that was already queued with all resources - * associated to the request. - * - * @param queue Queue handler to cancel. - * @param clear_cls Is it necessary to free associated cls? - */ -void -GMP_queue_destroy (struct MeshPeerQueue *queue, int clear_cls) -{ - struct MeshPeer *peer; - struct MeshFlowControl *fc; - int fwd; - - fwd = queue->fwd; - peer = queue->peer; - GNUNET_assert (NULL != queue->c); - fc = fwd ? &queue->c->fwd_fc : &queue->c->bck_fc; - - if (GNUNET_YES == clear_cls) - { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " queue destroy type %s\n", - GNUNET_MESH_DEBUG_M2S (queue->type)); - switch (queue->type) - { - case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY: - case GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY: - GNUNET_log (GNUNET_ERROR_TYPE_INFO, "destroying a DESTROY message\n"); - GNUNET_break (GNUNET_YES == queue->c->destroy); - /* fall through */ - case GNUNET_MESSAGE_TYPE_MESH_FWD: - case GNUNET_MESSAGE_TYPE_MESH_BCK: - case GNUNET_MESSAGE_TYPE_MESH_ACK: - case GNUNET_MESSAGE_TYPE_MESH_POLL: - case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK: - case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE: - case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN: - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " prebuilt message\n");; - GNUNET_free_non_null (queue->cls); - break; - - default: - GNUNET_break (0); - GNUNET_log (GNUNET_ERROR_TYPE_ERROR, " type %s unknown!\n", - GNUNET_MESH_DEBUG_M2S (queue->type)); - } - - } - GNUNET_CONTAINER_DLL_remove (peer->queue_head, peer->queue_tail, queue); - - if (queue->type != GNUNET_MESSAGE_TYPE_MESH_ACK && - queue->type != GNUNET_MESSAGE_TYPE_MESH_POLL) - { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " Q_N- %p %u\n", fc, fc->queue_n); - fc->queue_n--; - peer->queue_n--; - } - if (NULL != queue->c) - { - queue->c->pending_messages--; - if (NULL != queue->c->t) - { - queue->c->t->pending_messages--; - } - } - - GNUNET_free (queue); -} - - /** * @brief Queue and pass message to core when possible. * + * @param peer Peer towards which to queue the message. * @param cls Closure (@c type dependant). It will be used by queue_send to * build the message to be sent if not already prebuilt. * @param type Type of the message, 0 for a raw message. @@ -1007,33 +969,23 @@ GMP_queue_destroy (struct MeshPeerQueue *queue, int clear_cls) * @param c Connection this message belongs to (cannot be NULL). * @param ch Channel this message belongs to, if applicable (otherwise NULL). * @param fwd Is this a message going root->dest? (FWD ACK are NOT FWD!) + * @param cont Continuation to be called once CORE has taken the message. + * @param cont_cls Closure for @c cont. */ void -GMP_queue_add (void *cls, uint16_t type, size_t size, - struct MeshConnection *c, - struct MeshChannel *ch, - int fwd) +GMP_queue_add (struct MeshPeer *peer, void *cls, uint16_t type, size_t size, + struct MeshConnection *c, struct MeshChannel *ch, int fwd, + GMP_sent cont, void *cont_cls) { struct MeshPeerQueue *queue; - struct MeshFlowControl *fc; - struct MeshPeer *peer; int priority; int call_core; - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + LOG (GNUNET_ERROR_TYPE_DEBUG, "queue add %s %s (%u) on c %p, ch %p\n", fwd ? "FWD" : "BCK", GNUNET_MESH_DEBUG_M2S (type), size, c, ch); GNUNET_assert (NULL != c); - fc = fwd ? &c->fwd_fc : &c->bck_fc; - peer = fwd ? connection_get_next_hop (c) : connection_get_prev_hop (c); - - if (NULL == fc) - { - GNUNET_break (0); - return; - } - if (NULL == peer->connections) { /* We are not connected to this peer, ignore request. */ @@ -1049,37 +1001,9 @@ GMP_queue_add (void *cls, uint16_t type, size_t size, priority = 100; } - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "priority %d\n", priority); - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "fc %p\n", fc); - if (fc->queue_n >= fc->queue_max && 0 == priority) - { - GNUNET_STATISTICS_update (stats, "# messages dropped (buffer full)", - 1, GNUNET_NO); - GNUNET_break (0); - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, - "queue full: %u/%u\n", - fc->queue_n, fc->queue_max); - return; /* Drop this message */ - } + LOG (GNUNET_ERROR_TYPE_DEBUG, "priority %d\n", priority); - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "last pid %u\n", fc->last_pid_sent); - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ack %u\n", fc->last_ack_recv); - if (GMC_is_pid_bigger (fc->last_pid_sent + 1, fc->last_ack_recv)) - { - call_core = GNUNET_NO; - if (GNUNET_SCHEDULER_NO_TASK == fc->poll_task && - GNUNET_MESSAGE_TYPE_MESH_POLL != type) - { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, - "no buffer space (%u > %u): starting poll\n", - fc->last_pid_sent + 1, fc->last_ack_recv); - fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time, - &connection_poll, - fc); - } - } - else - call_core = GNUNET_YES; + call_core = GMC_is_sendable (c, fwd); queue = GNUNET_malloc (sizeof (struct MeshPeerQueue)); queue->cls = cls; queue->type = type; @@ -1088,6 +1012,8 @@ GMP_queue_add (void *cls, uint16_t type, size_t size, queue->c = c; queue->ch = ch; queue->fwd = fwd; + queue->callback = cont; + queue->callback_cls = cont_cls; if (100 <= priority) { struct MeshPeerQueue *copy; @@ -1107,16 +1033,14 @@ GMP_queue_add (void *cls, uint16_t type, size_t size, else { GNUNET_CONTAINER_DLL_insert_tail (peer->queue_head, peer->queue_tail, queue); - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " Q_N+ %p %u\n", fc, fc->queue_n); - fc->queue_n++; peer->queue_n++; } if (NULL == peer->core_transmit && GNUNET_YES == call_core) { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + LOG (GNUNET_ERROR_TYPE_DEBUG, "calling core tmt rdy towards %s for %u bytes\n", - peer2s (peer), size); + GMP_2s (peer), size); peer->core_transmit = GNUNET_CORE_notify_transmit_ready (core_handle, 0, @@ -1126,36 +1050,125 @@ GMP_queue_add (void *cls, uint16_t type, size_t size, size, &queue_send, peer); + queue->start_waiting = GNUNET_TIME_absolute_get (); } else { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + LOG (GNUNET_ERROR_TYPE_DEBUG, "core tmt rdy towards %s already called\n", - peer2s (peer)); + GMP_2s (peer)); } - c->pending_messages++; - if (NULL != c->t) - c->t->pending_messages++; } - /** - * Initialize the peer subsystem. + * Cancel all queued messages to a peer that belong to a certain connection. * - * @param c Configuration. + * @param peer Peer towards whom to cancel. + * @param c Connection whose queued messages to cancel. */ void -GMP_init (const struct GNUNET_CONFIGURATION_Handle *c) +GMP_queue_cancel (struct MeshPeer *peer, struct MeshConnection *c) { - peers = GNUNET_CONTAINER_multipeermap_create (128, GNUNET_NO); - if (GNUNET_OK != - GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_PEERS", - &max_peers)) + struct MeshPeerQueue *q; + struct MeshPeerQueue *next; + + for (q = peer->queue_head; NULL != q; q = next) { - GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_WARNING, - "MESH", "MAX_PEERS", "USING DEFAULT"); + next = q->next; + if (q->c == c) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "connection_cancel_queue %s\n", + GNUNET_MESH_DEBUG_M2S (q->type)); + queue_destroy (q, GNUNET_YES); + } + } + if (NULL == peer->queue_head) + { + if (NULL != peer->core_transmit) + { + GNUNET_CORE_notify_transmit_ready_cancel (peer->core_transmit); + peer->core_transmit = NULL; + } + } +} + + +/** + * Get the first transmittable message for a connection. + * + * @param peer Neighboring peer. + * @param c Connection. + * + * @return First transmittable message. + */ +static struct MeshPeerQueue * +connection_get_first_message (struct MeshPeer *peer, struct MeshConnection *c) +{ + struct MeshPeerQueue *q; + + for (q = peer->queue_head; NULL != q; q = q->next) + { + if (q->c != c) + continue; + if (queue_is_sendable (q)) + return q; + } + + return NULL; +} + +void +GMP_queue_unlock (struct MeshPeer *peer, struct MeshConnection *c) +{ + struct MeshPeerQueue *q; + size_t size; + + if (NULL != peer->core_transmit) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, " already unlocked!\n"); + return; /* Already unlocked */ + } + + q = connection_get_first_message (peer, c); + if (NULL == q) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, " queue empty!\n"); + return; /* Nothing to transmit */ + } + + size = q->size; + peer->core_transmit = + GNUNET_CORE_notify_transmit_ready (core_handle, + GNUNET_NO, + 0, + GNUNET_TIME_UNIT_FOREVER_REL, + GNUNET_PEER_resolve2 (peer->id), + size, + &queue_send, + peer); +} + + +/** + * Initialize the peer subsystem. + * + * @param c Configuration. + * @param id Peer identity + */ +void +GMP_init (const struct GNUNET_CONFIGURATION_Handle *c) +{ + LOG (GNUNET_ERROR_TYPE_DEBUG, "init\n"); + peers = GNUNET_CONTAINER_multipeermap_create (128, GNUNET_NO); + if (GNUNET_OK != + GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_PEERS", + &max_peers)) + { + GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_WARNING, + "MESH", "MAX_PEERS", "USING DEFAULT"); max_peers = 1000; } @@ -1167,13 +1180,30 @@ GMP_init (const struct GNUNET_CONFIGURATION_Handle *c) } else { - GNUNET_log (GNUNET_ERROR_TYPE_WARNING, + LOG (GNUNET_ERROR_TYPE_WARNING, "\n***************************************\n" "Mesh is running with drop mode enabled.\n" "This is NOT a good idea!\n" "Remove the DROP_PERCENT option from your configuration.\n" "***************************************\n"); } + + core_handle = GNUNET_CORE_connect (c, /* Main configuration */ + NULL, /* Closure passed to MESH functions */ + &core_init, /* Call core_init once connected */ + &core_connect, /* Handle connects */ + &core_disconnect, /* remove peers on disconnects */ + NULL, /* Don't notify about all incoming messages */ + GNUNET_NO, /* For header only in notification */ + NULL, /* Don't notify about all outbound messages */ + GNUNET_NO, /* For header-only out notification */ + core_handlers); /* Register these handlers */ + if (NULL == core_handle) + { + GNUNET_break (0); + GNUNET_SCHEDULER_shutdown (); + return; + } } /** @@ -1183,39 +1213,91 @@ void GMP_shutdown (void) { GNUNET_CONTAINER_multipeermap_iterate (peers, &shutdown_tunnel, NULL); + + if (core_handle != NULL) + { + GNUNET_CORE_disconnect (core_handle); + core_handle = NULL; + } + GNUNET_PEER_change_rc (myid, -1); +} + +/** + * Retrieve the MeshPeer stucture associated with the peer, create one + * and insert it in the appropriate structures if the peer is not known yet. + * + * @param peer Full identity of the peer. + * + * @return Existing or newly created peer structure. + */ +struct MeshPeer * +GMP_get (const struct GNUNET_PeerIdentity *peer_id) +{ + struct MeshPeer *peer; + + peer = GNUNET_CONTAINER_multipeermap_get (peers, peer_id); + if (NULL == peer) + { + peer = GNUNET_new (struct MeshPeer); + if (GNUNET_CONTAINER_multipeermap_size (peers) > max_peers) + { + peer_delete_oldest (); + } + GNUNET_CONTAINER_multipeermap_put (peers, peer_id, peer, + GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST); + peer->id = GNUNET_PEER_intern (peer_id); + } + peer->last_contact = GNUNET_TIME_absolute_get(); + + return peer; } /** - * Try to establish a new connection to this peer in the given tunnel. + * Retrieve the MeshPeer stucture associated with the peer, create one + * and insert it in the appropriate structures if the peer is not known yet. + * + * @param peer Short identity of the peer. + * + * @return Existing or newly created peer structure. + */ +struct MeshPeer * +GMP_get_short (const GNUNET_PEER_Id peer) +{ + return GMP_get (GNUNET_PEER_resolve2 (peer)); +} + + +/** + * Try to establish a new connection to this peer (in its tunnel). * If the peer doesn't have any path to it yet, try to get one. * If the peer already has some path, send a CREATE CONNECTION towards it. * - * @param peer PeerInfo of the peer. + * @param peer Peer to connect to. */ void GMP_connect (struct MeshPeer *peer) { - struct MeshTunnel2 *t; + struct MeshTunnel3 *t; struct MeshPeerPath *p; struct MeshConnection *c; int rerun_search; - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + LOG (GNUNET_ERROR_TYPE_DEBUG, "peer_connect towards %s\n", - peer2s (peer)); + GMP_2s (peer)); t = peer->tunnel; c = NULL; rerun_search = GNUNET_NO; if (NULL != peer->path_head) { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "path exists\n"); + LOG (GNUNET_ERROR_TYPE_DEBUG, "path exists\n"); p = peer_get_best_path (peer); if (NULL != p) { - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " %u hops\n", p->length); - c = tunnel_use_path (t, p); + LOG (GNUNET_ERROR_TYPE_DEBUG, " %u hops\n", p->length); + c = GMT_use_path (t, p); if (NULL == c) { /* This case can happen when the path includes a first hop that is @@ -1235,7 +1317,7 @@ GMP_connect (struct MeshPeer *peer) } else { - send_connection_create (c); + GMC_send_create (c); return; } } @@ -1245,8 +1327,9 @@ GMP_connect (struct MeshPeer *peer) { GMD_search_stop (peer->search_h); peer->search_h = NULL; - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, - " Stopping DHT GET for peer %s\n", peer2s (peer)); + LOG (GNUNET_ERROR_TYPE_DEBUG, + " Stopping DHT GET for peer %s\n", + GMP_2s (peer)); } if (NULL == peer->search_h) @@ -1254,14 +1337,337 @@ GMP_connect (struct MeshPeer *peer) const struct GNUNET_PeerIdentity *id; id = GNUNET_PEER_resolve2 (peer->id); - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, - " Starting DHT GET for peer %s\n", peer2s (peer)); + LOG (GNUNET_ERROR_TYPE_DEBUG, + " Starting DHT GET for peer %s\n", GMP_2s (peer)); peer->search_h = GMD_search (id, &search_handler, peer); - if (MESH_TUNNEL_NEW == t->state) - tunnel_change_state (t, MESH_TUNNEL_SEARCHING); + if (MESH_TUNNEL3_NEW == GMT_get_state (t)) + GMT_change_state (t, MESH_TUNNEL3_SEARCHING); + } +} + + +/** + * Set tunnel. + * + * @param peer Peer. + * @param t Tunnel. + */ +void +GMP_set_tunnel (struct MeshPeer *peer, struct MeshTunnel3 *t) +{ + peer->tunnel = t; +} + + +/** + * Chech whether there is a direct (core level) connection to peer. + * + * @param peer Peer to check. + * + * @return GNUNET_YES if there is a direct connection. + */ +int +GMP_is_neighbor (const struct MeshPeer *peer) +{ + struct MeshPeerPath *path; + + if (NULL == peer->connections) + return GNUNET_NO; + + for (path = peer->path_head; NULL != path; path = path->next) + { + if (3 > path->length) + return GNUNET_YES; + } + + GNUNET_break (0); /* Is not a neighbor but connections is not NULL */ + return GNUNET_NO; +} + + +/** + * Create and initialize a new tunnel towards a peer, in case it has none. + * In case the peer already has a tunnel, nothing is done. + * + * Does not generate any traffic, just creates the local data structures. + * + * @param peer Peer towards which to create the tunnel. + */ +void +GMP_add_tunnel (struct MeshPeer *peer) +{ + if (NULL != peer->tunnel) + return; + peer->tunnel = GMT_new (peer); +} + + +/** + * Add a connection to a neighboring peer. + * + * Store that the peer is the first hop of the connection in one + * direction and that on peer disconnect the connection must be + * notified and destroyed, for it will no longer be valid. + * + * @param peer Peer to add connection to. + * @param c Connection to add. + * + * @return GNUNET_OK on success. + */ +int +GMP_add_connection (struct MeshPeer *peer, + struct MeshConnection *c) +{ + if (NULL == peer->connections) + { + GNUNET_break (0); + return GNUNET_SYSERR; + } + return GNUNET_CONTAINER_multihashmap_put (peer->connections, + GMC_get_id (c), + c, + GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST); +} + + +/** + * Add the path to the peer and update the path used to reach it in case this + * is the shortest. + * + * @param peer_info Destination peer to add the path to. + * @param path New path to add. Last peer must be the peer in arg 1. + * Path will be either used of freed if already known. + * @param trusted Do we trust that this path is real? + */ +void +GMP_add_path (struct MeshPeer *peer_info, struct MeshPeerPath *path, + int trusted) +{ + struct MeshPeerPath *aux; + unsigned int l; + unsigned int l2; + + if ((NULL == peer_info) || (NULL == path)) + { + GNUNET_break (0); + path_destroy (path); + return; + } + if (path->peers[path->length - 1] != peer_info->id) + { + GNUNET_break (0); + path_destroy (path); + return; + } + if (2 >= path->length && GNUNET_NO == trusted) + { + /* Only allow CORE to tell us about direct paths */ + path_destroy (path); + return; + } + for (l = 1; l < path->length; l++) + { + if (path->peers[l] == myid) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, "shortening path by %u\n", l); + for (l2 = 0; l2 < path->length - l; l2++) + { + path->peers[l2] = path->peers[l + l2]; + } + path->length -= l; + l = 1; + path->peers = + GNUNET_realloc (path->peers, path->length * sizeof (GNUNET_PEER_Id)); + } + } + + LOG (GNUNET_ERROR_TYPE_DEBUG, "adding path [%u] to peer %s\n", + path->length, GMP_2s (peer_info)); + + l = path_get_length (path); + if (0 == l) + { + path_destroy (path); + return; + } + + GNUNET_assert (peer_info->id == path->peers[path->length - 1]); + for (aux = peer_info->path_head; aux != NULL; aux = aux->next) + { + l2 = path_get_length (aux); + if (l2 > l) + { + GNUNET_CONTAINER_DLL_insert_before (peer_info->path_head, + peer_info->path_tail, aux, path); + return; + } + else + { + if (l2 == l && memcmp (path->peers, aux->peers, l) == 0) + { + path_destroy (path); + return; + } + } + } + GNUNET_CONTAINER_DLL_insert_tail (peer_info->path_head, peer_info->path_tail, + path); + return; +} + + +/** + * Add the path to the origin peer and update the path used to reach it in case + * this is the shortest. + * The path is given in peer_info -> destination, therefore we turn the path + * upside down first. + * + * @param peer_info Peer to add the path to, being the origin of the path. + * @param path New path to add after being inversed. + * Path will be either used or freed. + * @param trusted Do we trust that this path is real? + */ +void +GMP_add_path_to_origin (struct MeshPeer *peer, + struct MeshPeerPath *path, + int trusted) +{ + if (NULL == path) + return; + path_invert (path); + GMP_add_path (peer, path, trusted); +} + + +/** + * Adds a path to the info of all the peers in the path + * + * @param p Path to process. + * @param confirmed Whether we know if the path works or not. + */ +void +GMP_add_path_to_all (const struct MeshPeerPath *p, int confirmed) +{ + unsigned int i; + + /* TODO: invert and add */ + for (i = 0; i < p->length && p->peers[i] != myid; i++) /* skip'em */ ; + for (i++; i < p->length; i++) + { + struct MeshPeer *aux; + struct MeshPeerPath *copy; + + aux = GMP_get_short (p->peers[i]); + copy = path_duplicate (p); + copy->length = i + 1; + GMP_add_path (aux, copy, p->length < 3 ? GNUNET_NO : confirmed); + } +} + + +/** + * Remove a connection from a neighboring peer. + * + * @param peer Peer to remove connection from. + * @param c Connection to remove. + * + * @return GNUNET_OK on success. + */ +int +GMP_remove_connection (struct MeshPeer *peer, + const struct MeshConnection *c) +{ + if (NULL == peer || NULL == peer->connections) + { + GNUNET_break (0); + return GNUNET_SYSERR; + } + return GNUNET_CONTAINER_multihashmap_remove (peer->connections, + GMC_get_id (c), + c); +} + +/** + * Start the DHT search for new paths towards the peer: we don't have + * enough good connections. + * + * @param peer Destination peer. + */ +void +GMP_start_search (struct MeshPeer *peer) +{ + if (NULL != peer->search_h) + { + GNUNET_break (0); + return; + } + + peer->search_h = GMD_search (GMP_get_id (peer), &search_handler, peer); +} + + +/** + * Stop the DHT search for new paths towards the peer: we already have + * enough good connections. + * + * @param peer Destination peer. + */ +void +GMP_stop_search (struct MeshPeer *peer) +{ + if (NULL == peer->search_h) + { + GNUNET_break (0); + return; } + + GMD_search_stop (peer->search_h); + peer->search_h = NULL; } + +/** + * Get the Full ID of a peer. + * + * @param peer Peer to get from. + * + * @return Full ID of peer. + */ +const struct GNUNET_PeerIdentity * +GMP_get_id (const struct MeshPeer *peer) +{ + return GNUNET_PEER_resolve2 (peer->id); +} + + +/** + * Get the Short ID of a peer. + * + * @param peer Peer to get from. + * + * @return Short ID of peer. + */ +GNUNET_PEER_Id +GMP_get_short_id (const struct MeshPeer *peer) +{ + return peer->id; +} + + +/** + * Get the tunnel towards a peer. + * + * @param peer Peer to get from. + * + * @return Tunnel towards peer. + */ +struct MeshTunnel3 * +GMP_get_tunnel (const struct MeshPeer *peer) +{ + return peer->tunnel; +} + + /** * Get the static string for a peer ID. *