123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345 |
- /*
- This file is part of GNUnet
- Copyright (C) 2008--2013, 2016 GNUnet e.V.
- GNUnet is free software: you can redistribute it and/or modify it
- under the terms of the GNU Affero General Public License as published
- by the Free Software Foundation, either version 3 of the License,
- or (at your option) any later version.
- GNUnet is distributed in the hope that it will be useful, but
- WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- Affero General Public License for more details.
- You should have received a copy of the GNU Affero General Public License
- along with this program. If not, see <http://www.gnu.org/licenses/>.
- SPDX-License-Identifier: AGPL3.0-or-later
- */
- /**
- * @file testbed-logger/testbed_logger_api.c
- * @brief Client-side routines for communicating with the tesbted logger service
- * @author Sree Harsha Totakura <sreeharsha@totakura.in>
- * @author Christian Grothoff
- */
- #include "platform.h"
- #include "gnunet_util_lib.h"
- #include "gnunet_testbed_logger_service.h"
- /**
- * Generic logging shorthand
- */
- #define LOG(kind, ...) \
- GNUNET_log_from (kind, "testbed-logger-api", __VA_ARGS__)
- /**
- * The size of the buffer we fill before sending out the message
- */
- #define BUFFER_SIZE (GNUNET_MAX_MESSAGE_SIZE - sizeof(struct \
- GNUNET_MessageHeader))
- /**
- * Connection handle for the logger service
- */
- struct GNUNET_TESTBED_LOGGER_Handle
- {
- /**
- * Client connection
- */
- struct GNUNET_MQ_Handle *mq;
- /**
- * Flush completion callback
- */
- GNUNET_TESTBED_LOGGER_FlushCompletion cb;
- /**
- * Closure for @e cb
- */
- void *cb_cls;
- /**
- * Local buffer for data to be transmitted
- */
- char buf[BUFFER_SIZE];
- /**
- * How many bytes in @a buf are in use?
- */
- size_t buse;
- /**
- * Number of bytes wrote since last flush
- */
- size_t bwrote;
- /**
- * How long after should we retry sending a message to the service?
- */
- struct GNUNET_TIME_Relative retry_backoff;
- /**
- * Task to call the flush completion callback
- */
- struct GNUNET_SCHEDULER_Task *flush_completion_task;
- /**
- * Number of entries in the MQ.
- */
- unsigned int mq_len;
- };
- /**
- * Task to call the flush completion notification
- *
- * @param cls the logger handle
- */
- static void
- call_flush_completion (void *cls)
- {
- struct GNUNET_TESTBED_LOGGER_Handle *h = cls;
- GNUNET_TESTBED_LOGGER_FlushCompletion cb;
- void *cb_cls;
- size_t bw;
- h->flush_completion_task = NULL;
- bw = h->bwrote;
- h->bwrote = 0;
- cb = h->cb;
- h->cb = NULL;
- cb_cls = h->cb_cls;
- h->cb_cls = NULL;
- if (NULL != cb)
- cb (cb_cls, bw);
- }
- /**
- * Schedule the flush completion notification task
- *
- * @param h logger handle
- */
- static void
- trigger_flush_notification (struct GNUNET_TESTBED_LOGGER_Handle *h)
- {
- if (NULL != h->flush_completion_task)
- GNUNET_SCHEDULER_cancel (h->flush_completion_task);
- h->flush_completion_task
- = GNUNET_SCHEDULER_add_now (&call_flush_completion,
- h);
- }
- /**
- * Send the buffered data to the service
- *
- * @param h the logger handle
- */
- static void
- dispatch_buffer (struct GNUNET_TESTBED_LOGGER_Handle *h);
- /**
- * MQ successfully sent a message.
- *
- * @param cls our handle
- */
- static void
- notify_sent (void *cls)
- {
- struct GNUNET_TESTBED_LOGGER_Handle *h = cls;
- h->mq_len--;
- if ((0 == h->mq_len) &&
- (NULL != h->cb))
- {
- if (0 == h->buse)
- trigger_flush_notification (h);
- else
- dispatch_buffer (h);
- }
- }
- /**
- * Send the buffered data to the service
- *
- * @param h the logger handle
- */
- static void
- dispatch_buffer (struct GNUNET_TESTBED_LOGGER_Handle *h)
- {
- struct GNUNET_MessageHeader *msg;
- struct GNUNET_MQ_Envelope *env;
- env = GNUNET_MQ_msg_extra (msg,
- h->buse,
- GNUNET_MESSAGE_TYPE_TESTBED_LOGGER_MSG);
- GNUNET_memcpy (&msg[1],
- h->buf,
- h->buse);
- h->bwrote += h->buse;
- h->buse = 0;
- h->mq_len++;
- GNUNET_MQ_notify_sent (env,
- ¬ify_sent,
- h);
- GNUNET_MQ_send (h->mq,
- env);
- }
- /**
- * We got disconnected from the logger. Stop logging.
- *
- * @param cls the `struct GNUNET_TESTBED_LOGGER_Handle`
- * @param error error code
- */
- static void
- mq_error_handler (void *cls,
- enum GNUNET_MQ_Error error)
- {
- struct GNUNET_TESTBED_LOGGER_Handle *h = cls;
- GNUNET_break (0);
- GNUNET_MQ_destroy (h->mq);
- h->mq = NULL;
- }
- /**
- * Connect to the testbed logger service
- *
- * @param cfg configuration to use
- * @return the handle which can be used for sending data to the service; NULL
- * upon any error
- */
- struct GNUNET_TESTBED_LOGGER_Handle *
- GNUNET_TESTBED_LOGGER_connect (const struct GNUNET_CONFIGURATION_Handle *cfg)
- {
- struct GNUNET_TESTBED_LOGGER_Handle *h;
- h = GNUNET_new (struct GNUNET_TESTBED_LOGGER_Handle);
- h->mq = GNUNET_CLIENT_connect (cfg,
- "testbed-logger",
- NULL,
- &mq_error_handler,
- h);
- if (NULL == h->mq)
- {
- GNUNET_free (h);
- return NULL;
- }
- return h;
- }
- /**
- * Disconnect from the logger service.
- *
- * @param h the logger handle
- */
- void
- GNUNET_TESTBED_LOGGER_disconnect (struct GNUNET_TESTBED_LOGGER_Handle *h)
- {
- if (NULL != h->flush_completion_task)
- {
- GNUNET_SCHEDULER_cancel (h->flush_completion_task);
- h->flush_completion_task = NULL;
- }
- if (0 != h->mq_len)
- LOG (GNUNET_ERROR_TYPE_WARNING,
- "Disconnect lost %u logger message[s]\n",
- h->mq_len);
- if (NULL != h->mq)
- {
- GNUNET_MQ_destroy (h->mq);
- h->mq = NULL;
- }
- GNUNET_free (h);
- }
- /**
- * Send data to be logged to the logger service. The data will be buffered and
- * will be sent upon an explicit call to GNUNET_TESTBED_LOGGER_flush() or upon
- * exceeding a threshold size.
- *
- * @param h the logger handle
- * @param data the data to send;
- * @param size how many bytes of @a data to send
- */
- void
- GNUNET_TESTBED_LOGGER_write (struct GNUNET_TESTBED_LOGGER_Handle *h,
- const void *data,
- size_t size)
- {
- if (NULL == h->mq)
- return;
- while (0 != size)
- {
- size_t fit_size = GNUNET_MIN (size,
- BUFFER_SIZE - h->buse);
- GNUNET_memcpy (&h->buf[h->buse],
- data,
- fit_size);
- h->buse += fit_size;
- data += fit_size;
- size -= fit_size;
- if (0 != size)
- dispatch_buffer (h);
- }
- }
- /**
- * Flush the buffered data to the logger service
- *
- * @param h the logger handle
- * @param cb the callback to call after the data is flushed
- * @param cb_cls the closure for the above callback
- */
- void
- GNUNET_TESTBED_LOGGER_flush (struct GNUNET_TESTBED_LOGGER_Handle *h,
- GNUNET_TESTBED_LOGGER_FlushCompletion cb,
- void *cb_cls)
- {
- GNUNET_assert (NULL == h->cb);
- h->cb = cb;
- h->cb_cls = cb_cls;
- if ((NULL == h->mq) ||
- (0 == h->buse))
- {
- trigger_flush_notification (h);
- return;
- }
- dispatch_buffer (h);
- }
- /**
- * Cancel notification upon flush. Should only be used when the flush
- * completion callback given to GNUNET_TESTBED_LOGGER_flush() is not already
- * called.
- *
- * @param h the logger handle
- */
- void
- GNUNET_TESTBED_LOGGER_flush_cancel (struct GNUNET_TESTBED_LOGGER_Handle *h)
- {
- if (NULL != h->flush_completion_task)
- {
- GNUNET_SCHEDULER_cancel (h->flush_completion_task);
- h->flush_completion_task = NULL;
- }
- h->cb = NULL;
- h->cb_cls = NULL;
- }
- /* End of testbed_logger_api.c */
|