* Function to call when notify_size bytes are available
* for transmission.
*/
- GNUNET_CONNECTION_TransmitReadyNotify notify;
+ GNUNET_TRANSPORT_TransmitReadyNotify notify;
/**
* Closure for notify.
*/
size_t notify_size;
- /**
- * How important is this message? Not used for control messages.
- */
- uint32_t priority;
-
};
GNUNET_TRANSPORT_TryConnectCallback cb;
+ int connect;
+
/**
* Closure for @e cb.
*/
*/
GNUNET_TRANSPORT_NotifyDisconnect nd_cb;
+ /**
+ * function to call on excess bandwidth events
+ */
+ GNUNET_TRANSPORT_NotifyExcessBandwidth neb_cb;
+
/**
* Head of DLL of control messages.
*/
}
+/**
+ * The outbound quota has changed in a way that may require
+ * us to reset the timeout. Update the timeout.
+ *
+ * @param cls the `struct Neighbour` for which the timeout changed
+ */
+static void
+outbound_bw_tracker_update (void *cls)
+{
+ struct Neighbour *n = cls;
+ struct GNUNET_TIME_Relative delay;
+
+ if (NULL == n->hn)
+ return;
+ delay = GNUNET_BANDWIDTH_tracker_get_delay (&n->out_tracker,
+ n->th->notify_size + n->traffic_overhead);
+ LOG (GNUNET_ERROR_TYPE_DEBUG,
+ "New outbound delay %llu us\n",
+ GNUNET_STRINGS_relative_time_to_string (delay,
+ GNUNET_NO));
+ GNUNET_CONTAINER_heap_update_cost (n->h->ready_heap,
+ n->hn, delay.rel_value_us);
+ schedule_transmission (n->h);
+}
+
+
+/**
+ * Function called by the bandwidth tracker if we have excess
+ * bandwidth.
+ *
+ * @param cls the `struct Neighbour` that has excess bandwidth
+ */
+static void
+notify_excess_cb (void *cls)
+{
+ struct Neighbour *n = cls;
+ struct GNUNET_TRANSPORT_Handle *h = n->h;
+
+ if (NULL != h->neb_cb)
+ h->neb_cb (h->cls,
+ &n->id);
+}
+
+
/**
* Add neighbour to our list
*
n->h = h;
n->is_ready = GNUNET_YES;
n->traffic_overhead = 0;
- GNUNET_BANDWIDTH_tracker_init (&n->out_tracker,
- GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT,
- MAX_BANDWIDTH_CARRY_S);
+ GNUNET_BANDWIDTH_tracker_init2 (&n->out_tracker,
+ &outbound_bw_tracker_update, n,
+ GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT,
+ MAX_BANDWIDTH_CARRY_S,
+ ¬ify_excess_cb,
+ n);
GNUNET_assert (GNUNET_OK ==
GNUNET_CONTAINER_multipeermap_put (h->neighbours,
&n->id, n,
GNUNET_assert (GNUNET_YES ==
GNUNET_CONTAINER_multipeermap_remove (handle->neighbours, key,
n));
+ GNUNET_BANDWIDTH_tracker_notification_stop (&n->out_tracker);
GNUNET_free (n);
return GNUNET_YES;
}
/**
* Function we use for handling incoming messages.
*
- * @param cls closure (struct GNUNET_TRANSPORT_Handle *)
+ * @param cls closure, a `struct GNUNET_TRANSPORT_Handle *`
* @param msg message received, NULL on timeout or fatal error
*/
static void
uint32_t bytes_physical;
GNUNET_assert (NULL != h->client);
+ if (GNUNET_YES == h->reconnecting)
+ {
+ return;
+ }
if (NULL == msg)
{
LOG (GNUNET_ERROR_TYPE_DEBUG,
"Error receiving from transport service, disconnecting temporarily.\n");
+ h->reconnecting = GNUNET_YES;
disconnect_and_schedule_reconnect (h);
return;
}
okm = (const struct SendOkMessage *) msg;
bytes_msg = ntohl (okm->bytes_msg);
bytes_physical = ntohl (okm->bytes_physical);
- LOG (GNUNET_ERROR_TYPE_DEBUG, "Receiving `%s' message, transmission %s.\n",
- "SEND_OK", ntohl (okm->success) == GNUNET_OK ? "succeeded" : "failed");
+ LOG (GNUNET_ERROR_TYPE_DEBUG,
+ "Receiving SEND_OK message, transmission %s.\n",
+ ntohl (okm->success) == GNUNET_OK ? "succeeded" : "failed");
n = neighbour_find (h, &okm->peer);
if (NULL == n)
if (bytes_physical >= bytes_msg)
{
- LOG (GNUNET_ERROR_TYPE_DEBUG, "Overhead for %u byte message: %u \n",
+ LOG (GNUNET_ERROR_TYPE_DEBUG,
+ "Overhead for %u byte message: %u\n",
bytes_msg, bytes_physical - bytes_msg);
n->traffic_overhead += bytes_physical - bytes_msg;
}
* Transmit message(s) to service.
*
* @param cls handle to transport
- * @param size number of bytes available in buf
+ * @param size number of bytes available in @a buf
* @param buf where to copy the message
- * @return number of bytes copied to buf
+ * @return number of bytes copied to @a buf
*/
static size_t
transport_notify_ready (void *cls, size_t size, void *buf)
{
GNUNET_CONTAINER_DLL_remove (h->control_head, h->control_tail, th);
nret = th->notify (th->notify_cls, size, &cbuf[ret]);
- LOG (GNUNET_ERROR_TYPE_DEBUG, "Added %u bytes of control message at %u\n",
+ LOG (GNUNET_ERROR_TYPE_DEBUG,
+ "Added %u bytes of control message at %u\n",
nret, ret);
GNUNET_free (th);
ret += nret;
GNUNET_SERVER_MAX_MESSAGE_SIZE);
obm.header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_SEND);
obm.header.size = htons (mret + sizeof (struct OutboundMessage));
- obm.priority = htonl (th->priority);
+ obm.reserved = htonl (0);
obm.timeout =
GNUNET_TIME_relative_hton (GNUNET_TIME_absolute_get_remaining
(th->timeout));
return; /* no pending messages */
size = n->th->notify_size + sizeof (struct OutboundMessage);
}
- LOG (GNUNET_ERROR_TYPE_DEBUG, "Calling notify_transmit_ready\n");
+ LOG (GNUNET_ERROR_TYPE_DEBUG,
+ "Calling notify_transmit_ready\n");
h->cth =
GNUNET_CLIENT_notify_transmit_ready (h->client, size,
GNUNET_TIME_UNIT_FOREVER_REL,
* @param h handle to the transport service
* @param size number of bytes to be transmitted
* @param notify function to call to get the content
- * @param notify_cls closure for notify
+ * @param notify_cls closure for @a notify
* @return a `struct GNUNET_TRANSPORT_TransmitHandle`
*/
static struct GNUNET_TRANSPORT_TransmitHandle *
schedule_control_transmit (struct GNUNET_TRANSPORT_Handle *h, size_t size,
- GNUNET_CONNECTION_TransmitReadyNotify notify,
+ GNUNET_TRANSPORT_TransmitReadyNotify notify,
void *notify_cls)
{
struct GNUNET_TRANSPORT_TransmitHandle *th;
* Transmit START message to service.
*
* @param cls unused
- * @param size number of bytes available in buf
+ * @param size number of bytes available in @a buf
* @param buf where to copy the message
- * @return number of bytes copied to buf
+ * @return number of bytes copied to @a buf
*/
static size_t
send_start (void *cls, size_t size, void *buf)
GNUNET_assert (NULL == h->client);
GNUNET_assert (NULL == h->control_head);
GNUNET_assert (NULL == h->control_tail);
+ h->reconnecting = GNUNET_NO;
h->client = GNUNET_CLIENT_connect ("transport", h->cfg);
+
GNUNET_assert (NULL != h->client);
- schedule_control_transmit (h, sizeof (struct StartMessage), &send_start, h);
+ schedule_control_transmit (h, sizeof (struct StartMessage),
+ &send_start, h);
}
{
GNUNET_CLIENT_disconnect (h->client);
h->client = NULL;
+/* LOG (GNUNET_ERROR_TYPE_ERROR,
+ "Client disconnect done \n");*/
}
/* Forget about all neighbours that we used to be connected to */
GNUNET_CONTAINER_multipeermap_iterate (h->neighbours,
struct GNUNET_TRANSPORT_TryConnectHandle *tch = cls;
struct TransportRequestConnectMessage msg;
- if (buf == NULL)
+ if (NULL == buf)
{
if (NULL != tch->cb)
tch->cb (tch->cb_cls, GNUNET_SYSERR);
GNUNET_assert (size >= sizeof (struct TransportRequestConnectMessage));
msg.header.size = htons (sizeof (struct TransportRequestConnectMessage));
msg.header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_REQUEST_CONNECT);
- msg.reserved = htonl (0);
+ msg.connect = htonl (tch->connect);
msg.peer = tch->pid;
memcpy (buf, &msg, sizeof (msg));
if (NULL != tch->cb)
tch->pid = *(target);
tch->cb = cb;
tch->cb_cls = cb_cls;
+ tch->connect = GNUNET_YES;
tch->tth = schedule_control_transmit (handle,
sizeof (struct TransportRequestConnectMessage),
&send_try_connect, tch);
GNUNET_TRANSPORT_try_connect_cancel (struct GNUNET_TRANSPORT_TryConnectHandle *tch)
{
struct GNUNET_TRANSPORT_Handle *th;
+ GNUNET_assert (GNUNET_YES == tch->connect);
+
+ th = tch->th;
+ cancel_control_transmit (th, tch->tth);
+ GNUNET_CONTAINER_DLL_remove (th->tc_head, th->tc_tail, tch);
+ GNUNET_free (tch);
+}
+
+/**
+ * Ask the transport service to shutdown a connection to
+ * the given peer.
+ *
+ * @param handle connection to transport service
+ * @param target who we should try to connect to
+ * @param cb callback to be called when request was transmitted to transport
+ * service
+ * @param cb_cls closure for the callback
+ * @return a `struct GNUNET_TRANSPORT_TryConnectHandle` handle or
+ * NULL on failure (cb will not be called)
+ */
+struct GNUNET_TRANSPORT_TryConnectHandle *
+GNUNET_TRANSPORT_try_disconnect (struct GNUNET_TRANSPORT_Handle *handle,
+ const struct GNUNET_PeerIdentity *target,
+ GNUNET_TRANSPORT_TryConnectCallback cb,
+ void *cb_cls)
+{
+ struct GNUNET_TRANSPORT_TryConnectHandle *tch = NULL;
+
+ if (NULL == handle->client)
+ return NULL;
+ tch = GNUNET_new (struct GNUNET_TRANSPORT_TryConnectHandle);
+ tch->th = handle;
+ tch->pid = *(target);
+ tch->cb = cb;
+ tch->cb_cls = cb_cls;
+ tch->connect = GNUNET_NO;
+ tch->tth = schedule_control_transmit (handle,
+ sizeof (struct TransportRequestConnectMessage),
+ &send_try_connect, tch);
+ GNUNET_CONTAINER_DLL_insert(handle->tc_head, handle->tc_tail, tch);
+ return tch;
+}
+
+
+/**
+ * Cancel the request to transport to try a disconnect
+ * Callback will not be called
+ *
+ * @param tch the handle to cancel
+ */
+void
+GNUNET_TRANSPORT_try_disconnect_cancel (struct GNUNET_TRANSPORT_TryConnectHandle *tch)
+{
+ struct GNUNET_TRANSPORT_Handle *th;
+ GNUNET_assert (GNUNET_NO == tch->connect);
th = tch->th;
cancel_control_transmit (th, tch->tth);
GNUNET_free (tch);
}
+
+
+
/**
* Send HELLO message to the service.
*
msg->ats_count = htons (ats_count);
msg->peer = (*peer);
memcpy (&msg[1], ats, ats_count * sizeof (struct GNUNET_ATS_Information));
- schedule_control_transmit (handle, len, &send_metric, msg);
+ schedule_control_transmit (handle, len,
+ &send_metric, msg);
}
-
/**
* Offer the transport service the HELLO of another peer. Note that
* the transport service may just ignore this message if the HELLO is
ohh->cont = cont;
ohh->cls = cls;
ohh->msg = msg;
- ohh->tth = schedule_control_transmit (handle, size, &send_hello, ohh);
+ ohh->tth = schedule_control_transmit (handle, size,
+ &send_hello, ohh);
GNUNET_CONTAINER_DLL_insert (handle->oh_head, handle->oh_tail, ohh);
return ohh;
}
*/
int
GNUNET_TRANSPORT_check_peer_connected (struct GNUNET_TRANSPORT_Handle *handle,
- const struct GNUNET_PeerIdentity *peer)
+ const struct GNUNET_PeerIdentity *peer)
{
if (GNUNET_YES ==
GNUNET_CONTAINER_multipeermap_contains (handle->neighbours,
* @param rec receive function to call
* @param nc function to call on connect events
* @param nd function to call on disconnect events
+ * @return NULL on error
*/
struct GNUNET_TRANSPORT_Handle *
GNUNET_TRANSPORT_connect (const struct GNUNET_CONFIGURATION_Handle *cfg,
GNUNET_TRANSPORT_ReceiveCallback rec,
GNUNET_TRANSPORT_NotifyConnect nc,
GNUNET_TRANSPORT_NotifyDisconnect nd)
+{
+ return GNUNET_TRANSPORT_connect2 (cfg, self, cls,
+ rec, nc, nd, NULL);
+}
+
+
+/**
+ * Connect to the transport service. Note that the connection may
+ * complete (or fail) asynchronously.
+ *
+ * @param cfg configuration to use
+ * @param self our own identity (API should check that it matches
+ * the identity found by transport), or NULL (no check)
+ * @param cls closure for the callbacks
+ * @param rec receive function to call
+ * @param nc function to call on connect events
+ * @param nd function to call on disconnect events
+ * @param neb function to call if we have excess bandwidth to a peer
+ * @return NULL on error
+ */
+struct GNUNET_TRANSPORT_Handle *
+GNUNET_TRANSPORT_connect2 (const struct GNUNET_CONFIGURATION_Handle *cfg,
+ const struct GNUNET_PeerIdentity *self, void *cls,
+ GNUNET_TRANSPORT_ReceiveCallback rec,
+ GNUNET_TRANSPORT_NotifyConnect nc,
+ GNUNET_TRANSPORT_NotifyDisconnect nd,
+ GNUNET_TRANSPORT_NotifyExcessBandwidth neb)
{
struct GNUNET_TRANSPORT_Handle *ret;
ret->rec = rec;
ret->nc_cb = nc;
ret->nd_cb = nd;
+ ret->neb_cb = neb;
ret->reconnect_delay = GNUNET_TIME_UNIT_ZERO;
LOG (GNUNET_ERROR_TYPE_DEBUG,
"Connecting to transport service.\n");
ret->ready_heap =
GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MIN);
schedule_control_transmit (ret, sizeof (struct StartMessage),
- &send_start, ret);
+ &send_start, ret);
return ret;
}
/**
* Disconnect from the transport service.
*
- * @param handle handle to the service as returned from GNUNET_TRANSPORT_connect
+ * @param handle handle to the service as returned from #GNUNET_TRANSPORT_connect()
*/
void
GNUNET_TRANSPORT_disconnect (struct GNUNET_TRANSPORT_Handle *handle)
}
GNUNET_free_non_null (handle->my_hello);
handle->my_hello = NULL;
- GNUNET_assert (handle->tc_head == NULL);
- GNUNET_assert (handle->tc_tail == NULL);
- GNUNET_assert (handle->hwl_head == NULL);
- GNUNET_assert (handle->hwl_tail == NULL);
+ GNUNET_assert (NULL == handle->tc_head);
+ GNUNET_assert (NULL == handle->tc_tail);
+ GNUNET_assert (NULL == handle->hwl_head);
+ GNUNET_assert (NULL == handle->hwl_tail);
GNUNET_CONTAINER_heap_destroy (handle->ready_heap);
handle->ready_heap = NULL;
GNUNET_free (handle);
* @param handle connection to transport service
* @param target who should receive the message
* @param size how big is the message we want to transmit?
- * @param priority how important is the message?
* @param timeout after how long should we give up (and call
* notify with buf NULL and size 0)?
* @param notify function to call when we are ready to
struct GNUNET_TRANSPORT_TransmitHandle *
GNUNET_TRANSPORT_notify_transmit_ready (struct GNUNET_TRANSPORT_Handle *handle,
const struct GNUNET_PeerIdentity *target,
- size_t size, uint32_t priority,
+ size_t size,
struct GNUNET_TIME_Relative timeout,
- GNUNET_CONNECTION_TransmitReadyNotify notify,
+ GNUNET_TRANSPORT_TransmitReadyNotify notify,
void *notify_cls)
{
struct Neighbour *n;
th->notify_cls = notify_cls;
th->timeout = GNUNET_TIME_relative_to_absolute (timeout);
th->notify_size = size;
- th->priority = priority;
n->th = th;
/* calculate when our transmission should be ready */
delay = GNUNET_BANDWIDTH_tracker_get_delay (&n->out_tracker, size + n->traffic_overhead);