/*
This file is part of GNUnet.
- (C) 2009 Christian Grothoff (and other contributing authors)
+ (C) 2009, 2010 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 2, or (at your
+ 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
* @author Christian Grothoff
*/
#include "platform.h"
+#include "gnunet_constants.h"
#include "gnunet_core_service.h"
#include "core.h"
*/
GNUNET_CORE_StartupCallback init;
- /**
- * Function to call whenever we're notified about a peer connecting
- * (pre-connects, no session key exchange yet).
- */
- GNUNET_CORE_ConnectEventHandler pre_connects;
-
/**
* Function to call whenever we're notified about a peer connecting.
*/
*/
GNUNET_CORE_DisconnectEventHandler disconnects;
+ /**
+ * Function to call whenever we're notified about a peer changing status.
+ */
+ GNUNET_CORE_PeerStatusEventHandler status_events;
+
/**
* Function to call whenever we receive an inbound message.
*/
const struct GNUNET_CORE_MessageHandler *handlers;
/**
- * Our connection to the service.
+ * Our connection to the service for notifications.
*/
- struct GNUNET_CLIENT_Connection *client;
+ struct GNUNET_CLIENT_Connection *client_notifications;
/**
* Handle for our current transmission request.
*/
- struct GNUNET_CLIENT_TransmitHandle *th;
+ struct GNUNET_CLIENT_TransmitHandle *cth;
/**
* Head of doubly-linked list of pending requests.
*/
void *get_message_cls;
- /**
- * If this entry is for a configuration request, pointer
- * to the information callback; otherwise NULL.
- */
- GNUNET_CORE_PeerConfigurationInfoCallback info;
-
- /**
- * Closure for info.
- */
- void *info_cls;
-
/**
* If this entry is for a transmission request, pointer
* to the notify callback; otherwise NULL.
};
+static void
+reconnect_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
+
+
/**
* Function called when we are ready to transmit our
* "START" message (or when this operation timed out).
static void
reconnect (struct GNUNET_CORE_Handle *h)
{
- GNUNET_CLIENT_disconnect (h->client);
+#if DEBUG_CORE
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Reconnecting to CORE service\n");
+#endif
+ if (h->client_notifications != NULL)
+ GNUNET_CLIENT_disconnect (h->client_notifications, GNUNET_NO);
h->currently_down = GNUNET_YES;
- h->client = GNUNET_CLIENT_connect (h->sched, "core", h->cfg);
- h->th = GNUNET_CLIENT_notify_transmit_ready (h->client,
- sizeof (struct InitMessage) +
- sizeof (uint16_t) * h->hcnt,
- GNUNET_TIME_UNIT_SECONDS,
- GNUNET_NO,
- &transmit_start, h);
+ h->client_notifications = GNUNET_CLIENT_connect (h->sched, "core", h->cfg);
+ if (h->client_notifications == NULL)
+ h->reconnect_task = GNUNET_SCHEDULER_add_delayed (h->sched,
+ GNUNET_TIME_UNIT_SECONDS,
+ &reconnect_task,
+ h);
+ else
+ h->cth = GNUNET_CLIENT_notify_transmit_ready (h->client_notifications,
+ sizeof (struct InitMessage) +
+ sizeof (uint16_t) * h->hcnt,
+ GNUNET_TIME_UNIT_SECONDS,
+ GNUNET_NO,
+ &transmit_start, h);
}
* @param tc context, can be NULL (!)
*/
static void
-timeout_request (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+timeout_request (void *cls,
+ const struct GNUNET_SCHEDULER_TaskContext *tc)
{
struct GNUNET_CORE_TransmitHandle *th = cls;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Transmission request timed out.\n");
th->timeout_task = GNUNET_SCHEDULER_NO_TASK;
+#if DEBUG_CORE
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Signalling timeout of request for transmission to CORE service\n");
+#endif
GNUNET_assert (0 == th->get_message (th->get_message_cls, 0, NULL));
}
struct GNUNET_CORE_TransmitHandle *th;
size_t ret;
- h->th = NULL;
+ h->cth = NULL;
th = h->pending_head;
+ if (th == NULL)
+ return 0;
if (buf == NULL)
{
timeout_request (th, NULL);
return 0;
}
- /* create new timeout task (in case core takes too long to respond!) */
- th->timeout_task = GNUNET_SCHEDULER_add_delayed (h->sched,
- GNUNET_TIME_absolute_get_remaining
- (th->timeout),
- &timeout_request, th);
- /* remove th from doubly-linked pending list, move to submitted */
- GNUNET_assert (th->prev == NULL);
- h->pending_head = th->next;
- if (th->next == NULL)
- h->pending_tail = NULL;
- else
- th->next->prev = NULL;
+ GNUNET_CONTAINER_DLL_remove (h->pending_head,
+ h->pending_tail,
+ th);
GNUNET_assert (h->submitted == NULL);
h->submitted = th;
GNUNET_assert (size >= th->msize);
ret = th->get_message (th->get_message_cls, size, buf);
GNUNET_assert (ret <= size);
+#if DEBUG_CORE
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Transmitting %u bytes to core\n",
+ ret);
+#endif
return ret;
}
trigger_next_request (struct GNUNET_CORE_Handle *h)
{
struct GNUNET_CORE_TransmitHandle *th;
+
if (h->currently_down)
- return; /* connection temporarily down */
- if (NULL == (th = h->pending_head))
- return; /* no requests pending */
- GNUNET_assert (NULL == h->th);
- if (GNUNET_SCHEDULER_NO_TASK != th->timeout_task)
{
- GNUNET_SCHEDULER_cancel (h->sched, th->timeout_task);
- th->timeout_task = GNUNET_SCHEDULER_NO_TASK;
+#if DEBUG_CORE
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "In trigger_next_request, connection currently down...\n");
+#endif
+ return; /* connection temporarily down */
}
- h->th = GNUNET_CLIENT_notify_transmit_ready (h->client,
- th->msize,
- GNUNET_TIME_absolute_get_remaining
- (th->timeout),
- GNUNET_NO,
- &request_start,
- h);
+ if (NULL == (th = h->pending_head))
+ return; /* no requests pending */
+ GNUNET_assert (NULL == h->cth);
+ h->cth = GNUNET_CLIENT_notify_transmit_ready (h->client_notifications,
+ th->msize,
+ GNUNET_TIME_absolute_get_remaining
+ (th->timeout),
+ GNUNET_NO,
+ &request_start,
+ h);
}
/**
- * Handler for most messages received from the core.
+ * Handler for notification messages received from the core.
*
* @param cls our "struct GNUNET_CORE_Handle"
* @param msg the message received from the core service
*/
static void
-main_handler (void *cls, const struct GNUNET_MessageHeader *msg)
+main_notify_handler (void *cls, const struct GNUNET_MessageHeader *msg)
{
struct GNUNET_CORE_Handle *h = cls;
unsigned int hpos;
const struct ConnectNotifyMessage *cnm;
+ const struct DisconnectNotifyMessage *dnm;
const struct NotifyTrafficMessage *ntm;
- const struct ConfigurationInfoMessage *cim;
const struct GNUNET_MessageHeader *em;
+ const struct PeerStatusNotifyMessage *psnm;
uint16_t msize;
uint16_t et;
const struct GNUNET_CORE_MessageHandler *mh;
#endif
switch (ntohs (msg->type))
{
- case GNUNET_MESSAGE_TYPE_CORE_NOTIFY_PRE_CONNECT:
- if (NULL == h->pre_connects)
- {
- GNUNET_break (0);
- break;
- }
- if (msize != sizeof (struct ConnectNotifyMessage))
- {
- GNUNET_break (0);
- break;
- }
- cnm = (const struct ConnectNotifyMessage *) msg;
- h->pre_connects (h->cls,
- &cnm->peer,
- GNUNET_TIME_relative_ntoh (cnm->latency),
- ntohl (cnm->distance));
- break;
case GNUNET_MESSAGE_TYPE_CORE_NOTIFY_CONNECT:
if (NULL == h->connects)
{
GNUNET_break (0);
break;
}
- if (msize != sizeof (struct ConnectNotifyMessage))
+ if (msize != sizeof (struct DisconnectNotifyMessage))
{
GNUNET_break (0);
break;
}
- cnm = (const struct ConnectNotifyMessage *) msg;
+ dnm = (const struct DisconnectNotifyMessage *) msg;
h->disconnects (h->cls,
- &cnm->peer);
+ &dnm->peer);
+ break;
+ case GNUNET_MESSAGE_TYPE_CORE_NOTIFY_STATUS_CHANGE:
+ if (NULL == h->status_events)
+ {
+ GNUNET_break (0);
+ break;
+ }
+ if (msize != sizeof (struct PeerStatusNotifyMessage))
+ {
+ GNUNET_break (0);
+ break;
+ }
+ psnm = (const struct PeerStatusNotifyMessage *) msg;
+ h->status_events (h->cls,
+ &psnm->peer,
+ GNUNET_TIME_relative_ntoh (psnm->latency),
+ ntohl (psnm->distance),
+ psnm->bandwidth_in,
+ psnm->bandwidth_out,
+ GNUNET_TIME_absolute_ntoh (psnm->timeout));
break;
case GNUNET_MESSAGE_TYPE_CORE_NOTIFY_INBOUND:
if (msize <
em = (const struct GNUNET_MessageHeader *) &ntm[1];
#if DEBUG_CORE
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Received message of type %u from peer `%4s'\n",
- ntohs (em->type), GNUNET_i2s (&ntm->peer));
+ "Received message of type %u and size %u from peer `%4s'\n",
+ ntohs (em->type),
+ ntohs (em->size),
+ GNUNET_i2s (&ntm->peer));
#endif
if ((GNUNET_NO == h->inbound_hdr_only) &&
(msize != ntohs (em->size) + sizeof (struct NotifyTrafficMessage)))
GNUNET_TIME_relative_ntoh (ntm->latency),
ntohl (ntm->distance));
break;
- case GNUNET_MESSAGE_TYPE_CORE_CONFIGURATION_INFO:
- if (msize != sizeof (struct ConfigurationInfoMessage))
- {
- GNUNET_break (0);
- break;
- }
- if (NULL == h->submitted)
- break;
- cim = (const struct ConfigurationInfoMessage *) msg;
-
- /* process configuration data */
- if (h->submitted->info != NULL)
- h->submitted->info (h->submitted->info_cls,
- &h->submitted->peer,
- ntohl (cim->bpm_in),
- ntohl (cim->bpm_out),
- (int) ntohl (cim->reserved_amount),
- cim->preference);
- /* done, clean up! */
- GNUNET_CORE_notify_transmit_ready_cancel (h->submitted); // HUH?
- trigger_next_request (h);
- break;
default:
GNUNET_break (0);
break;
}
- GNUNET_CLIENT_receive (h->client,
- &main_handler, h, GNUNET_TIME_UNIT_FOREVER_REL);
+ GNUNET_CLIENT_receive (h->client_notifications,
+ &main_notify_handler, h, GNUNET_TIME_UNIT_FOREVER_REL);
}
-
/**
* Function called when we are ready to transmit our
* "START" message (or when this operation timed out).
(ntohs (msg->size) != sizeof (struct InitReplyMessage)) ||
(ntohs (msg->type) != GNUNET_MESSAGE_TYPE_CORE_INIT_REPLY))
{
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _
- ("Error connecting to core service (failed to receive `%s' message).\n"),
- "INIT_REPLY");
- GNUNET_break (msg == NULL);
+ if (msg != NULL)
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+ _
+ ("Error connecting to core service (failed to receive `%s' message, got message of type %u and size %u).\n"),
+ "INIT_REPLY",
+ ntohs (msg->type),
+ ntohs (msg->size));
+ GNUNET_break (0);
+ }
+ else
+ {
+#if DEBUG_CORE
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ _("Failed to connect to core service, will retry.\n"));
+#endif
+ }
transmit_start (h, 0, NULL);
return;
}
/* start our message processing loop */
#if DEBUG_CORE
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- _
- ("Successfully connected to core service, starting processing loop.\n"));
+ "Successfully connected to core service, starting processing loop.\n");
#endif
h->currently_down = GNUNET_NO;
trigger_next_request (h);
- GNUNET_CLIENT_receive (h->client,
- &main_handler, h, GNUNET_TIME_UNIT_FOREVER_REL);
+ GNUNET_CLIENT_receive (h->client_notifications,
+ &main_notify_handler, h, GNUNET_TIME_UNIT_FOREVER_REL);
if (NULL != (init = h->init))
{
/* mark so we don't call init on reconnect */
h->init = NULL;
-#if DEBUG_CORE
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _("Successfully connected to core service.\n"));
-#endif
GNUNET_CRYPTO_hash (&m->publicKey,
sizeof (struct
GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded),
unsigned int hpos;
struct GNUNET_TIME_Relative delay;
- h->th = NULL;
+ h->cth = NULL;
if (size == 0)
{
if ((h->init == NULL) ||
init->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_INIT);
init->header.size = htons (msize);
opt = GNUNET_CORE_OPTION_NOTHING;
- if (h->pre_connects != NULL)
- opt |= GNUNET_CORE_OPTION_SEND_PRE_CONNECT;
if (h->connects != NULL)
opt |= GNUNET_CORE_OPTION_SEND_CONNECT;
if (h->disconnects != NULL)
opt |= GNUNET_CORE_OPTION_SEND_DISCONNECT;
+ if (h->status_events != NULL)
+ opt |= GNUNET_CORE_OPTION_SEND_STATUS_CHANGE;
if (h->inbound_notify != NULL)
{
if (h->inbound_hdr_only)
ts = (uint16_t *) & init[1];
for (hpos = 0; hpos < h->hcnt; hpos++)
ts[hpos] = htons (h->handlers[hpos].type);
- GNUNET_CLIENT_receive (h->client,
+ GNUNET_CLIENT_receive (h->client_notifications,
&init_reply_handler,
h,
GNUNET_TIME_absolute_get_remaining
* @param cls closure for the various callbacks that follow (including handlers in the handlers array)
* @param init callback to call on timeout or once we have successfully
* connected to the core service; note that timeout is only meaningful if init is not NULL
- * @param pre_connects function to call on peer pre-connect (no session key yet), can be NULL
* @param connects function to call on peer connect, can be NULL
* @param disconnects function to call on peer disconnect / timeout, can be NULL
+ * @param status_events function to call on changes to peer connection status, can be NULL
* @param inbound_notify function to call for all inbound messages, can be NULL
* @param inbound_hdr_only set to GNUNET_YES if inbound_notify will only read the
* GNUNET_MessageHeader and hence we do not need to give it the full message;
struct GNUNET_TIME_Relative timeout,
void *cls,
GNUNET_CORE_StartupCallback init,
- GNUNET_CORE_ConnectEventHandler pre_connects,
GNUNET_CORE_ConnectEventHandler connects,
GNUNET_CORE_DisconnectEventHandler disconnects,
+ GNUNET_CORE_PeerStatusEventHandler status_events,
GNUNET_CORE_MessageCallback inbound_notify,
int inbound_hdr_only,
GNUNET_CORE_MessageCallback outbound_notify,
h->cfg = cfg;
h->cls = cls;
h->init = init;
- h->pre_connects = pre_connects;
h->connects = connects;
h->disconnects = disconnects;
+ h->status_events = status_events;
h->inbound_notify = inbound_notify;
h->outbound_notify = outbound_notify;
h->inbound_hdr_only = inbound_hdr_only;
h->outbound_hdr_only = outbound_hdr_only;
h->handlers = handlers;
- h->client = GNUNET_CLIENT_connect (sched, "core", cfg);
- if (h->client == NULL)
+ h->client_notifications = GNUNET_CLIENT_connect (sched, "core", cfg);
+ if (h->client_notifications == NULL)
{
GNUNET_free (h);
return NULL;
"Trying to connect to core service in next %llu ms.\n",
timeout.value);
#endif
- h->th =
- GNUNET_CLIENT_notify_transmit_ready (h->client,
+ h->cth =
+ GNUNET_CLIENT_notify_transmit_ready (h->client_notifications,
sizeof (struct InitMessage) +
sizeof (uint16_t) * h->hcnt, timeout,
GNUNET_YES,
void
GNUNET_CORE_disconnect (struct GNUNET_CORE_Handle *handle)
{
- if (handle->th != NULL)
- GNUNET_CLIENT_notify_transmit_ready_cancel (handle->th);
+ if (handle->cth != NULL)
+ GNUNET_CLIENT_notify_transmit_ready_cancel (handle->cth);
if (handle->solicit_transmit_req != NULL)
GNUNET_CORE_notify_transmit_ready_cancel (handle->solicit_transmit_req);
if (handle->reconnect_task != GNUNET_SCHEDULER_NO_TASK)
GNUNET_SCHEDULER_cancel (handle->sched, handle->reconnect_task);
- GNUNET_CLIENT_disconnect (handle->client);
+ if (handle->client_notifications != NULL)
+ GNUNET_CLIENT_disconnect (handle->client_notifications, GNUNET_NO);
+ GNUNET_break (handle->pending_head == NULL);
GNUNET_free_non_null (handle->solicit_buffer);
GNUNET_free (handle);
}
trigger_next_request (h);
return 0;
}
-#if DEBUG_CORE
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Preparing for P2P transmission to `%4s'.\n",
- GNUNET_i2s(&th->peer));
-#endif
- GNUNET_assert (th->timeout_task != GNUNET_SCHEDULER_NO_TASK);
sm = (struct SendMessage *) buf;
sm->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_SEND);
sm->priority = htonl (th->priority);
notify_cls = th->notify_cls;
GNUNET_CORE_notify_transmit_ready_cancel (th);
trigger_next_request (h);
+ size = GNUNET_MIN (size,
+ GNUNET_CONSTANTS_MAX_ENCRYPTED_MESSAGE_SIZE);
GNUNET_assert (size >= sizeof (struct SendMessage));
dt = notify (notify_cls, size - sizeof (struct SendMessage), &sm[1]);
if (0 == dt)
{
+#if DEBUG_CORE
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Size of clients message to peer %s is 0!\n",
+ GNUNET_i2s(&th->peer));
+#endif
/* client decided to send nothing! */
return 0;
}
+#if DEBUG_CORE
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Produced SEND message to core with %u bytes payload\n",
+ dt);
+#endif
GNUNET_assert (dt >= sizeof (struct GNUNET_MessageHeader));
+ if (dt + sizeof (struct SendMessage) >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
+ {
+ GNUNET_break (0);
+ return 0;
+ }
+#if DEBUG_CORE
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Preparing for P2P transmission of %u bytes to `%4s'.\n",
+ dt,
+ GNUNET_i2s(&th->peer));
+#endif
sm->header.size = htons (dt + sizeof (struct SendMessage));
- GNUNET_assert (dt + sizeof (struct SendMessage) < size);
+ GNUNET_assert (dt + sizeof (struct SendMessage) <= size);
return dt + sizeof (struct SendMessage);
}
GNUNET_SERVER_MAX_MESSAGE_SIZE);
th = GNUNET_malloc (sizeof (struct GNUNET_CORE_TransmitHandle));
th->ch = handle;
- /* append to list */
- th->prev = handle->pending_tail;
- if (handle->pending_tail == NULL)
- handle->pending_head = th;
- else
- handle->pending_tail->next = th;
+ GNUNET_CONTAINER_DLL_insert_after (handle->pending_head,
+ handle->pending_tail,
+ handle->pending_tail,
+ th);
th->get_message = &produce_send;
th->get_message_cls = th;
th->notify = notify;
th->msize = sizeof (struct SendMessage) + notify_size;
/* was the request queue previously empty? */
if ( (handle->pending_head == th) &&
- (handle->th == NULL) )
+ (handle->cth == NULL) )
trigger_next_request (handle);
return th;
}
/**
* Cancel the specified transmission-ready notification.
*
- * @param h handle that was returned by "notify_transmit_ready".
+ * @param th handle that was returned by "notify_transmit_ready".
*/
void
GNUNET_CORE_notify_transmit_ready_cancel (struct GNUNET_CORE_TransmitHandle
- *h)
+ *th)
{
- struct GNUNET_CORE_Handle *handle = h->ch;
-
- if (handle->submitted == h)
- {
- handle->submitted = NULL;
- }
- else
- {
- if (h->prev == NULL)
- handle->pending_head = h->next;
- else
- h->prev->next = h->next;
- if (h->next == NULL)
- handle->pending_tail = h->prev;
- else
- h->next->prev = h->prev;
- }
- if (h->timeout_task != GNUNET_SCHEDULER_NO_TASK)
- GNUNET_SCHEDULER_cancel (handle->sched, h->timeout_task);
- GNUNET_free (h);
+ struct GNUNET_CORE_Handle *h = th->ch;
+
+ if (h->submitted == th)
+ h->submitted = NULL;
+ else
+ GNUNET_CONTAINER_DLL_remove (h->pending_head,
+ h->pending_tail,
+ th);
+ if (th->timeout_task != GNUNET_SCHEDULER_NO_TASK)
+ GNUNET_SCHEDULER_cancel (h->sched, th->timeout_task);
+ GNUNET_free (th);
}