#include "gnunet_time_lib.h"
+#define LOG(kind,...) GNUNET_log_from (kind, "util", __VA_ARGS__)
+
/**
* How much buffer space do we want to have at least
{
struct GNUNET_SERVER_TransmitContext *tc = cls;
size_t msize;
+
if (buf == NULL)
- {
- GNUNET_SERVER_receive_done (tc->client, GNUNET_SYSERR);
- GNUNET_free_non_null (tc->buf);
- GNUNET_free (tc);
- return 0;
- }
+ {
+ GNUNET_SERVER_transmit_context_destroy (tc, GNUNET_SYSERR);
+ return 0;
+ }
if (tc->total - tc->off > size)
msize = size;
else
memcpy (buf, &tc->buf[tc->off], msize);
tc->off += msize;
if (tc->total == tc->off)
- {
- GNUNET_SERVER_receive_done (tc->client, GNUNET_OK);
- GNUNET_free_non_null (tc->buf);
- GNUNET_free (tc);
- }
+ {
+
+ GNUNET_SERVER_receive_done (tc->client, GNUNET_OK);
+ GNUNET_SERVER_client_drop (tc->client);
+ GNUNET_free_non_null (tc->buf);
+ GNUNET_free (tc);
+ }
else
+ {
+ if (NULL ==
+ GNUNET_SERVER_notify_transmit_ready (tc->client,
+ GNUNET_MIN (MIN_BLOCK_SIZE,
+ tc->total - tc->off),
+ GNUNET_TIME_absolute_get_remaining
+ (tc->timeout), &transmit_response,
+ tc))
{
- if (NULL == GNUNET_SERVER_notify_transmit_ready (tc->client,
- GNUNET_MIN
- (MIN_BLOCK_SIZE,
- tc->total - tc->off),
- GNUNET_TIME_absolute_get_remaining
- (tc->timeout),
- &transmit_response,
- tc))
- {
- GNUNET_break (0);
- GNUNET_SERVER_receive_done (tc->client, GNUNET_SYSERR);
- GNUNET_free_non_null (tc->buf);
- GNUNET_free (tc);
- }
+ GNUNET_break (0);
+ GNUNET_SERVER_transmit_context_destroy (tc, GNUNET_SYSERR);
}
+ }
return msize;
}
GNUNET_assert (client != NULL);
tc = GNUNET_malloc (sizeof (struct GNUNET_SERVER_TransmitContext));
+ GNUNET_SERVER_client_keep (client);
tc->client = client;
return tc;
}
*/
void
GNUNET_SERVER_transmit_context_append_data (struct GNUNET_SERVER_TransmitContext
- *tc, const void *data, size_t length,
- uint16_t type)
+ *tc, const void *data,
+ size_t length, uint16_t type)
{
struct GNUNET_MessageHeader *msg;
size_t size;
* @param msg message to append
*/
void
-GNUNET_SERVER_transmit_context_append_message(struct GNUNET_SERVER_TransmitContext
- *tc, const struct GNUNET_MessageHeader *msg)
+GNUNET_SERVER_transmit_context_append_message (struct
+ GNUNET_SERVER_TransmitContext
+ *tc,
+ const struct GNUNET_MessageHeader
+ *msg)
{
struct GNUNET_MessageHeader *m;
uint16_t size;
GNUNET_MIN (MIN_BLOCK_SIZE,
tc->total), timeout,
&transmit_response, tc))
- {
- GNUNET_break (0);
- GNUNET_SERVER_receive_done (tc->client, GNUNET_SYSERR);
- GNUNET_free_non_null (tc->buf);
- GNUNET_free (tc);
- }
+ {
+ GNUNET_break (0);
+ GNUNET_SERVER_transmit_context_destroy (tc, GNUNET_SYSERR);
+ }
+}
+
+
+/**
+ * Destroy a transmission context. This function must not be called
+ * after 'GNUNET_SERVER_transmit_context_run'.
+ *
+ * @param tc transmission context to destroy
+ * @param success code to give to 'GNUNET_SERVER_receive_done' for
+ * the client: GNUNET_OK to keep the connection open and
+ * continue to receive
+ * GNUNET_NO to close the connection (normal behavior)
+ * GNUNET_SYSERR to close the connection (signal
+ * serious error)
+ */
+void
+GNUNET_SERVER_transmit_context_destroy (struct GNUNET_SERVER_TransmitContext
+ *tc, int success)
+{
+ GNUNET_SERVER_receive_done (tc->client, success);
+ GNUNET_SERVER_client_drop (tc->client);
+ GNUNET_free_non_null (tc->buf);
+ GNUNET_free (tc);
}
+
/* end of server_tc.c */