|
- /*
- * This file is part of GNUnet
- * Copyright (C) 2013 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 psyc/test_psyc.c
- * @brief Tests for the PSYC API.
- * @author Gabor X Toth
- * @author Christian Grothoff
- */
-
- #include <inttypes.h>
-
- #include <gnunet/platform.h>
- #include <gnunet/gnunet_crypto_lib.h>
- #include <gnunet/gnunet_common.h>
- #include <gnunet/gnunet_util_lib.h>
- #include <gnunet/gnunet_testing_lib.h>
- #include "gnunet_psyc_util_lib.h"
- #include "gnunet_psyc_service.h"
-
- #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
-
- /**
- * Return value from 'main'.
- */
- static int res;
-
- static const struct GNUNET_CONFIGURATION_Handle *cfg;
-
- static struct GNUNET_PeerIdentity this_peer;
-
- /**
- * Handle for task for timeout termination.
- */
- static struct GNUNET_SCHEDULER_Task * end_badly_task;
-
- static struct GNUNET_PSYC_Master *mst;
- static struct GNUNET_PSYC_Slave *slv;
-
- static struct GNUNET_PSYC_Channel *mst_chn, *slv_chn;
-
- static struct GNUNET_CRYPTO_EddsaPrivateKey *channel_key;
- static struct GNUNET_CRYPTO_EcdsaPrivateKey *slave_key;
-
- static struct GNUNET_CRYPTO_EddsaPublicKey channel_pub_key;
- static struct GNUNET_CRYPTO_EcdsaPublicKey slave_pub_key;
-
- struct TransmitClosure
- {
- struct GNUNET_PSYC_MasterTransmitHandle *mst_tmit;
- struct GNUNET_PSYC_SlaveTransmitHandle *slv_tmit;
- struct GNUNET_PSYC_Environment *env;
- struct GNUNET_PSYC_Modifier *mod;
- char *data[16];
- const char *mod_value;
- size_t mod_value_size;
- uint8_t data_delay[16];
- uint8_t data_count;
- uint8_t paused;
- uint8_t n;
- };
-
- static struct TransmitClosure *tmit;
-
- static uint8_t join_req_count, end_count;
-
- enum
- {
- TEST_NONE = 0,
- TEST_MASTER_START = 1,
- TEST_SLAVE_JOIN_REJECT = 2,
- TEST_SLAVE_JOIN_ACCEPT = 3,
- TEST_SLAVE_ADD = 4,
- TEST_SLAVE_REMOVE = 5,
- TEST_SLAVE_TRANSMIT = 6,
- TEST_MASTER_TRANSMIT = 7,
- TEST_MASTER_HISTORY_REPLAY_LATEST = 8,
- TEST_SLAVE_HISTORY_REPLAY_LATEST = 9,
- TEST_MASTER_HISTORY_REPLAY = 10,
- TEST_SLAVE_HISTORY_REPLAY = 11,
- TEST_MASTER_STATE_GET = 12,
- TEST_SLAVE_STATE_GET = 13,
- TEST_MASTER_STATE_GET_PREFIX = 14,
- TEST_SLAVE_STATE_GET_PREFIX = 15,
- } test;
-
-
- static void
- master_transmit ();
-
- static void
- master_history_replay_latest ();
-
-
- static void
- master_stopped (void *cls)
- {
- if (NULL != tmit)
- {
- GNUNET_PSYC_env_destroy (tmit->env);
- GNUNET_free (tmit);
- tmit = NULL;
- }
- GNUNET_SCHEDULER_shutdown ();
- }
-
-
- static void
- slave_parted (void *cls)
- {
- if (NULL != mst)
- {
- GNUNET_PSYC_master_stop (mst, GNUNET_NO, &master_stopped, NULL);
- mst = NULL;
- }
- else
- master_stopped (NULL);
- }
-
-
- /**
- * Clean up all resources used.
- */
- static void
- cleanup ()
- {
- if (NULL != slv)
- {
- GNUNET_PSYC_slave_part (slv, GNUNET_NO, &slave_parted, NULL);
- slv = NULL;
- }
- else
- slave_parted (NULL);
- }
-
-
- /**
- * Terminate the test case (failure).
- *
- * @param cls NULL
- */
- static void
- end_badly (void *cls)
- {
- res = 1;
- cleanup ();
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Test FAILED.\n");
- }
-
-
- /**
- * Terminate the test case (success).
- *
- * @param cls NULL
- */
- static void
- end_normally (void *cls)
- {
- res = 0;
- cleanup ();
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Test PASSED.\n");
- }
-
-
- /**
- * Finish the test case (successfully).
- */
- static void
- end ()
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Ending tests.\n");
-
- if (end_badly_task != NULL)
- {
- GNUNET_SCHEDULER_cancel (end_badly_task);
- end_badly_task = NULL;
- }
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MILLISECONDS,
- &end_normally, NULL);
- }
-
-
- static void
- master_message_cb (void *cls, const struct GNUNET_PSYC_MessageHeader *msg)
- {
- GNUNET_assert (NULL != msg);
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Master got PSYC message fragment of size %u "
- "belonging to message ID %" PRIu64 " with flags %x\n",
- test, ntohs (msg->header.size),
- GNUNET_ntohll (msg->message_id), ntohl (msg->flags));
- // FIXME
- }
-
-
- static void
- master_message_part_cb (void *cls, const struct GNUNET_PSYC_MessageHeader *msg,
- const struct GNUNET_MessageHeader *pmsg)
- {
- GNUNET_assert (NULL != msg && NULL != pmsg);
-
- uint64_t message_id = GNUNET_ntohll (msg->message_id);
- uint32_t flags = ntohl (msg->flags);
-
- uint16_t type = ntohs (pmsg->type);
- uint16_t size = ntohs (pmsg->size);
-
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Master got message part of type %u and size %u "
- "belonging to message ID %" PRIu64 " with flags %x\n",
- test, type, size, message_id, flags);
-
- switch (test)
- {
- case TEST_SLAVE_TRANSMIT:
- if (GNUNET_PSYC_MESSAGE_REQUEST != flags)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Test #%d: Unexpected request flags: %x" PRIu32 "\n",
- test, flags);
- GNUNET_assert (0);
- return;
- }
- // FIXME: check rest of message
-
- if (GNUNET_MESSAGE_TYPE_PSYC_MESSAGE_END == type)
- master_transmit ();
- break;
-
- case TEST_MASTER_TRANSMIT:
- if (GNUNET_MESSAGE_TYPE_PSYC_MESSAGE_END == type && 2 == ++end_count)
- master_history_replay_latest ();
- break;
-
- case TEST_MASTER_HISTORY_REPLAY:
- case TEST_MASTER_HISTORY_REPLAY_LATEST:
- if (GNUNET_PSYC_MESSAGE_HISTORIC != flags)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Test #%d: Unexpected flags for historic message: %x" PRIu32 "\n",
- test, flags);
- GNUNET_assert (0);
- return;
- }
- break;
-
- default:
- GNUNET_assert (0);
- }
- }
-
-
- static void
- slave_message_cb (void *cls, const struct GNUNET_PSYC_MessageHeader *msg)
- {
- GNUNET_assert (NULL != msg);
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Slave got PSYC message fragment of size %u "
- "belonging to message ID %" PRIu64 " with flags %x\n",
- test, ntohs (msg->header.size),
- GNUNET_ntohll (msg->message_id), ntohl (msg->flags));
- // FIXME
- }
-
-
- static void
- slave_message_part_cb (void *cls,
- const struct GNUNET_PSYC_MessageHeader *msg,
- const struct GNUNET_MessageHeader *pmsg)
- {
- GNUNET_assert (NULL != msg && NULL != pmsg);
-
- uint64_t message_id = GNUNET_ntohll (msg->message_id);
- uint32_t flags = ntohl (msg->flags);
-
- uint16_t type = ntohs (pmsg->type);
- uint16_t size = ntohs (pmsg->size);
-
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Slave got message part of type %u and size %u "
- "belonging to message ID %" PRIu64 " with flags %x\n",
- test, type, size, message_id, flags);
-
- switch (test)
- {
- case TEST_MASTER_TRANSMIT:
- if (GNUNET_MESSAGE_TYPE_PSYC_MESSAGE_END == type && 2 == ++end_count)
- master_history_replay_latest ();
- break;
-
- case TEST_SLAVE_HISTORY_REPLAY:
- case TEST_SLAVE_HISTORY_REPLAY_LATEST:
- if (GNUNET_PSYC_MESSAGE_HISTORIC != flags)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Test #%d: Unexpected flags for historic message: %x" PRIu32 "\n",
- test, flags);
- GNUNET_assert (0);
- return;
- }
- break;
-
- default:
- GNUNET_assert (0);
- }
- }
-
-
- static void
- state_get_var (void *cls, const struct GNUNET_MessageHeader *mod,
- const char *name, const void *value,
- uint32_t value_size, uint32_t full_value_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Got state var: %s\n%.*s\n",
- name,
- (int) value_size,
- (const char *) value);
- }
-
-
- /*** Slave state_get_prefix() ***/
-
- static void
- slave_state_get_prefix_result (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: slave_state_get_prefix:\t%" PRId64 " (%.*s)\n",
- test, result,
- (int) err_msg_size,
- (const char *) err_msg);
- // FIXME: GNUNET_assert (2 == result);
- end ();
- }
-
-
- static void
- slave_state_get_prefix ()
- {
- test = TEST_SLAVE_STATE_GET_PREFIX;
- GNUNET_PSYC_channel_state_get_prefix (slv_chn, "_foo", state_get_var,
- slave_state_get_prefix_result, NULL);
- }
-
-
- /*** Master state_get_prefix() ***/
-
-
- static void
- master_state_get_prefix_result (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: master_state_get_prefix:\t%" PRId64 " (%s)\n",
- test, result, (char *) err_msg);
- // FIXME: GNUNET_assert (2 == result);
- slave_state_get_prefix ();
- }
-
-
- static void
- master_state_get_prefix ()
- {
- test = TEST_MASTER_STATE_GET_PREFIX;
- GNUNET_PSYC_channel_state_get_prefix (mst_chn, "_foo", state_get_var,
- master_state_get_prefix_result, NULL);
- }
-
-
- /*** Slave state_get() ***/
-
-
- static void
- slave_state_get_result (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: slave_state_get:\t%" PRId64 " (%.*s)\n",
- test, result, err_msg_size, (char *) err_msg);
- // FIXME: GNUNET_assert (2 == result);
- master_state_get_prefix ();
- }
-
-
- static void
- slave_state_get ()
- {
- test = TEST_SLAVE_STATE_GET;
- GNUNET_PSYC_channel_state_get (slv_chn, "_foo_bar_baz", state_get_var,
- slave_state_get_result, NULL);
- }
-
-
- /*** Master state_get() ***/
-
-
- static void
- master_state_get_result (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: master_state_get:\t%" PRId64 " (%.*s)\n",
- test, result, err_msg_size, (char *) err_msg);
- // FIXME: GNUNET_assert (1 == result);
- slave_state_get ();
- }
-
-
- static void
- master_state_get ()
- {
- test = TEST_MASTER_STATE_GET;
- GNUNET_PSYC_channel_state_get (mst_chn, "_foo_bar_baz", state_get_var,
- master_state_get_result, NULL);
- }
-
-
- /*** Slave history_replay() ***/
-
- static void
- slave_history_replay_result (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: slave_history_replay:\t%" PRId64 " (%.*s)\n",
- test, result,
- (int) err_msg_size,
- (const char *) err_msg);
- GNUNET_assert (9 == result);
-
- master_state_get ();
- }
-
-
- static void
- slave_history_replay ()
- {
- test = TEST_SLAVE_HISTORY_REPLAY;
- GNUNET_PSYC_channel_history_replay (slv_chn, 1, 1, "",
- GNUNET_PSYC_HISTORY_REPLAY_LOCAL,
- slave_message_cb,
- slave_message_part_cb,
- slave_history_replay_result, NULL);
- }
-
-
- /*** Master history_replay() ***/
-
-
- static void
- master_history_replay_result (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: master_history_replay:\t%" PRId64 " (%.*s)\n",
- test, result,
- (int) err_msg_size,
- (const char *) err_msg);
- GNUNET_assert (9 == result);
-
- slave_history_replay ();
- }
-
-
- static void
- master_history_replay ()
- {
- test = TEST_MASTER_HISTORY_REPLAY;
- GNUNET_PSYC_channel_history_replay (mst_chn, 1, 1, "",
- GNUNET_PSYC_HISTORY_REPLAY_LOCAL,
- master_message_cb,
- master_message_part_cb,
- master_history_replay_result, NULL);
- }
-
-
- /*** Slave history_replay_latest() ***/
-
-
- static void
- slave_history_replay_latest_result (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: slave_history_replay_latest:\t%" PRId64 " (%.*s)\n",
- test, result,
- (int) err_msg_size,
- (const char *) err_msg);
- GNUNET_assert (9 == result);
-
- master_history_replay ();
- }
-
-
- static void
- slave_history_replay_latest ()
- {
- test = TEST_SLAVE_HISTORY_REPLAY_LATEST;
- GNUNET_PSYC_channel_history_replay_latest (slv_chn, 1, "",
- GNUNET_PSYC_HISTORY_REPLAY_LOCAL,
- &slave_message_cb,
- &slave_message_part_cb,
- &slave_history_replay_latest_result,
- NULL);
- }
-
-
- /*** Master history_replay_latest() ***/
-
-
- static void
- master_history_replay_latest_result (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: master_history_replay_latest:\t%" PRId64 " (%.*s)\n",
- test, result, err_msg_size, (char *) err_msg);
- GNUNET_assert (9 == result);
-
- slave_history_replay_latest ();
- }
-
-
- static void
- master_history_replay_latest ()
- {
- test = TEST_MASTER_HISTORY_REPLAY_LATEST;
- GNUNET_PSYC_channel_history_replay_latest (mst_chn, 1, "",
- GNUNET_PSYC_HISTORY_REPLAY_LOCAL,
- &master_message_cb,
- &master_message_part_cb,
- &master_history_replay_latest_result,
- NULL);
- }
-
-
- static void
- transmit_resume (void *cls)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Test #%d: Transmission resumed.\n", test);
- struct TransmitClosure *tmit = cls;
- if (NULL != tmit->mst_tmit)
- GNUNET_PSYC_master_transmit_resume (tmit->mst_tmit);
- else
- GNUNET_PSYC_slave_transmit_resume (tmit->slv_tmit);
- }
-
-
- static int
- tmit_notify_data (void *cls, uint16_t *data_size, void *data)
- {
- struct TransmitClosure *tmit = cls;
- if (0 == tmit->data_count)
- {
- *data_size = 0;
- return GNUNET_YES;
- }
-
- uint16_t size = strlen (tmit->data[tmit->n]);
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Test #%d: Transmit notify data: %u bytes available, "
- "processing fragment %u/%u (size %u).\n",
- test, *data_size, tmit->n + 1, tmit->data_count, size);
- if (*data_size < size)
- {
- *data_size = 0;
- GNUNET_assert (0);
- return GNUNET_SYSERR;
- }
-
- if (GNUNET_YES != tmit->paused && 0 < tmit->data_delay[tmit->n])
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Test #%d: Transmission paused.\n", test);
- tmit->paused = GNUNET_YES;
- GNUNET_SCHEDULER_add_delayed (
- GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,
- tmit->data_delay[tmit->n]),
- &transmit_resume, tmit);
- *data_size = 0;
- return GNUNET_NO;
- }
- tmit->paused = GNUNET_NO;
-
- *data_size = size;
- GNUNET_memcpy (data, tmit->data[tmit->n], size);
-
- return ++tmit->n < tmit->data_count ? GNUNET_NO : GNUNET_YES;
- }
-
-
- static int
- tmit_notify_mod (void *cls, uint16_t *data_size, void *data, uint8_t *oper,
- uint32_t *full_value_size)
- {
- struct TransmitClosure *tmit = cls;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Test #%d: Transmit notify modifier: %u bytes available, "
- "%u modifiers left to process.\n",
- test, *data_size, (unsigned int) GNUNET_PSYC_env_get_count (tmit->env));
-
- uint16_t name_size = 0;
- size_t value_size = 0;
- const char *value = NULL;
-
- if (NULL != oper && NULL != tmit->mod)
- { /* New modifier */
- tmit->mod = tmit->mod->next;
- if (NULL == tmit->mod)
- { /* No more modifiers, continue with data */
- *data_size = 0;
- return GNUNET_YES;
- }
-
- GNUNET_assert (tmit->mod->value_size < UINT32_MAX);
- *full_value_size = tmit->mod->value_size;
- *oper = tmit->mod->oper;
- name_size = strlen (tmit->mod->name);
-
- if (name_size + 1 + tmit->mod->value_size <= *data_size)
- {
- *data_size = name_size + 1 + tmit->mod->value_size;
- }
- else
- {
- tmit->mod_value_size = tmit->mod->value_size;
- value_size = *data_size - name_size - 1;
- tmit->mod_value_size -= value_size;
- tmit->mod_value = tmit->mod->value + value_size;
- }
-
- GNUNET_memcpy (data, tmit->mod->name, name_size);
- ((char *)data)[name_size] = '\0';
- GNUNET_memcpy ((char *)data + name_size + 1, tmit->mod->value, value_size);
- }
- else if (NULL != tmit->mod_value && 0 < tmit->mod_value_size)
- { /* Modifier continuation */
- value = tmit->mod_value;
- if (tmit->mod_value_size <= *data_size)
- {
- value_size = tmit->mod_value_size;
- tmit->mod_value = NULL;
- }
- else
- {
- value_size = *data_size;
- tmit->mod_value += value_size;
- }
- tmit->mod_value_size -= value_size;
-
- if (*data_size < value_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "value larger than buffer: %u < %zu\n",
- *data_size, value_size);
- *data_size = 0;
- return GNUNET_NO;
- }
-
- *data_size = value_size;
- GNUNET_memcpy (data, value, value_size);
- }
-
- return GNUNET_NO;
- }
-
-
- static void
- slave_join ();
-
-
- static void
- slave_transmit ()
- {
- test = TEST_SLAVE_TRANSMIT;
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Slave sending request to master.\n", test);
-
- tmit = GNUNET_new (struct TransmitClosure);
- tmit->env = GNUNET_PSYC_env_create ();
- GNUNET_PSYC_env_add (tmit->env, GNUNET_PSYC_OP_ASSIGN,
- "_abc", "abc def", 7);
- GNUNET_PSYC_env_add (tmit->env, GNUNET_PSYC_OP_ASSIGN,
- "_abc_def", "abc def ghi", 11);
- tmit->mod = GNUNET_PSYC_env_head (tmit->env);
- tmit->n = 0;
- tmit->data[0] = "slave test";
- tmit->data_count = 1;
- tmit->slv_tmit
- = GNUNET_PSYC_slave_transmit (slv, "_request_test", &tmit_notify_mod,
- &tmit_notify_data, tmit,
- GNUNET_PSYC_SLAVE_TRANSMIT_NONE);
- }
-
-
- static void
- slave_remove_cb (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: slave_remove:\t%" PRId64 " (%.*s)\n",
- test, result, err_msg_size, (char *) err_msg);
-
- slave_transmit ();
- }
-
-
- static void
- slave_remove ()
- {
- test = TEST_SLAVE_REMOVE;
- struct GNUNET_PSYC_Channel *chn = GNUNET_PSYC_master_get_channel (mst);
- GNUNET_PSYC_channel_slave_remove (chn, &slave_pub_key, 2,
- &slave_remove_cb, chn);
- }
-
-
- static void
- slave_add_cb (void *cls, int64_t result,
- const void *err_msg, uint16_t err_msg_size)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: slave_add:\t%" PRId64 " (%.*s)\n",
- test, result, err_msg_size, (char *) err_msg);
- slave_remove ();
- }
-
-
- static void
- slave_add ()
- {
- test = TEST_SLAVE_ADD;
- struct GNUNET_PSYC_Channel *chn = GNUNET_PSYC_master_get_channel (mst);
- GNUNET_PSYC_channel_slave_add (chn, &slave_pub_key, 2, 2, &slave_add_cb, chn);
- }
-
-
- static void
- schedule_second_slave_join (void *cls)
- {
- slave_join (TEST_SLAVE_JOIN_ACCEPT);
- }
-
-
- static void
- first_slave_parted (void *cls)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "First slave parted.\n");
- GNUNET_SCHEDULER_add_now (&schedule_second_slave_join, NULL);
- }
-
-
- static void
- schedule_first_slave_part (void *cls)
- {
- GNUNET_PSYC_slave_part (slv, GNUNET_NO, &first_slave_parted, NULL);
- }
-
-
- static void
- join_decision_cb (void *cls,
- const struct GNUNET_PSYC_JoinDecisionMessage *dcsn,
- int is_admitted,
- const struct GNUNET_PSYC_Message *join_msg)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Slave got join decision: %d\n", test, is_admitted);
-
- switch (test)
- {
- case TEST_SLAVE_JOIN_REJECT:
- GNUNET_assert (0 == is_admitted);
- GNUNET_assert (1 == join_req_count);
- GNUNET_SCHEDULER_add_now (&schedule_first_slave_part, NULL);
- break;
-
- case TEST_SLAVE_JOIN_ACCEPT:
- GNUNET_assert (1 == is_admitted);
- GNUNET_assert (2 == join_req_count);
- slave_add ();
- break;
-
- default:
- GNUNET_break (0);
- }
- }
-
-
- static void
- join_request_cb (void *cls,
- const struct GNUNET_PSYC_JoinRequestMessage *req,
- const struct GNUNET_CRYPTO_EcdsaPublicKey *slave_key,
- const struct GNUNET_PSYC_Message *join_msg,
- struct GNUNET_PSYC_JoinHandle *jh)
- {
- struct GNUNET_HashCode slave_key_hash;
- GNUNET_CRYPTO_hash (slave_key, sizeof (*slave_key), &slave_key_hash);
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Got join request #%u from %s.\n",
- test, join_req_count, GNUNET_h2s (&slave_key_hash));
-
- /* Reject first request */
- int is_admitted = (0 < join_req_count++) ? GNUNET_YES : GNUNET_NO;
- GNUNET_PSYC_join_decision (jh, is_admitted, 0, NULL, NULL);
- }
-
-
- static void
- slave_connect_cb (void *cls, int result, uint64_t max_message_id)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Slave connected: %d, max_message_id: %" PRIu64 "\n",
- test, result, max_message_id);
- GNUNET_assert (TEST_SLAVE_JOIN_REJECT == test || TEST_SLAVE_JOIN_ACCEPT == test);
- GNUNET_assert (GNUNET_OK == result || GNUNET_NO == result);
- }
-
-
- static void
- slave_join (int t)
- {
- test = t;
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Joining slave.\n", t);
-
- struct GNUNET_PeerIdentity origin = this_peer;
- struct GNUNET_PSYC_Environment *env = GNUNET_PSYC_env_create ();
- GNUNET_PSYC_env_add (env, GNUNET_PSYC_OP_ASSIGN,
- "_foo", "bar baz", 7);
- GNUNET_PSYC_env_add (env, GNUNET_PSYC_OP_ASSIGN,
- "_foo_bar", "foo bar baz", 11);
- struct GNUNET_PSYC_Message *
- join_msg = GNUNET_PSYC_message_create ("_request_join", env, "some data", 9);
-
- slv = GNUNET_PSYC_slave_join (cfg,
- &channel_pub_key,
- slave_key,
- GNUNET_PSYC_SLAVE_JOIN_NONE,
- &origin,
- 0,
- NULL,
- &slave_message_cb,
- &slave_message_part_cb,
- &slave_connect_cb,
- &join_decision_cb,
- NULL,
- join_msg);
- GNUNET_free (join_msg);
- slv_chn = GNUNET_PSYC_slave_get_channel (slv);
- GNUNET_PSYC_env_destroy (env);
- }
-
-
- static void
- master_transmit ()
- {
- test = TEST_MASTER_TRANSMIT;
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Master sending message to all.\n", test);
- end_count = 0;
-
- uint32_t i, j;
-
- char *name_max = "_test_max";
- uint8_t name_max_size = sizeof ("_test_max");
- char *val_max = GNUNET_malloc (GNUNET_PSYC_MODIFIER_MAX_PAYLOAD);
- for (i = 0; i < GNUNET_PSYC_MODIFIER_MAX_PAYLOAD; i++)
- val_max[i] = (0 == i % 10000) ? '0' + i / 10000 : '.';
-
- char *name_cont = "_test_cont";
- uint8_t name_cont_size = sizeof ("_test_cont");
- char *val_cont = GNUNET_malloc (GNUNET_PSYC_MODIFIER_MAX_PAYLOAD
- + GNUNET_PSYC_MOD_CONT_MAX_PAYLOAD);
- for (i = 0; i < GNUNET_PSYC_MODIFIER_MAX_PAYLOAD - name_cont_size; i++)
- val_cont[i] = (0 == i % 10000) ? '0' + i / 10000 : ':';
- for (j = 0; j < GNUNET_PSYC_MOD_CONT_MAX_PAYLOAD; j++, i++)
- val_cont[i] = (0 == j % 10000) ? '0' + j / 10000 : '!';
-
- tmit = GNUNET_new (struct TransmitClosure);
- tmit->env = GNUNET_PSYC_env_create ();
- GNUNET_PSYC_env_add (tmit->env, GNUNET_PSYC_OP_ASSIGN,
- "_foo", "bar baz", 7);
- GNUNET_PSYC_env_add (tmit->env, GNUNET_PSYC_OP_ASSIGN,
- name_max, val_max,
- GNUNET_PSYC_MODIFIER_MAX_PAYLOAD
- - name_max_size);
- GNUNET_PSYC_env_add (tmit->env, GNUNET_PSYC_OP_ASSIGN,
- "_foo_bar", "foo bar baz", 11);
- GNUNET_PSYC_env_add (tmit->env, GNUNET_PSYC_OP_ASSIGN,
- name_cont, val_cont,
- GNUNET_PSYC_MODIFIER_MAX_PAYLOAD - name_cont_size
- + GNUNET_PSYC_MOD_CONT_MAX_PAYLOAD);
- tmit->mod = GNUNET_PSYC_env_head (tmit->env);
- tmit->data[0] = "foo";
- tmit->data[1] = GNUNET_malloc (GNUNET_PSYC_DATA_MAX_PAYLOAD + 1);
- for (i = 0; i < GNUNET_PSYC_DATA_MAX_PAYLOAD; i++)
- tmit->data[1][i] = (0 == i % 10000) ? '0' + i / 10000 : '_';
- tmit->data[2] = "foo bar";
- tmit->data[3] = "foo bar baz";
- tmit->data_delay[1] = 3;
- tmit->data_count = 4;
- tmit->mst_tmit
- = GNUNET_PSYC_master_transmit (mst, "_notice_test", &tmit_notify_mod,
- &tmit_notify_data, tmit,
- GNUNET_PSYC_MASTER_TRANSMIT_INC_GROUP_GEN);
- }
-
-
- static void
- master_start_cb (void *cls, int result, uint64_t max_message_id)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Test #%d: Master started: %d, max_message_id: %" PRIu64 "\n",
- test, result, max_message_id);
- GNUNET_assert (TEST_MASTER_START == test);
- GNUNET_assert (GNUNET_OK == result || GNUNET_NO == result);
- slave_join (TEST_SLAVE_JOIN_REJECT);
- }
-
-
- static void
- master_start ()
- {
- test = TEST_MASTER_START;
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Test #%d: Starting master.\n", test);
- mst = GNUNET_PSYC_master_start (cfg, channel_key, GNUNET_PSYC_CHANNEL_PRIVATE,
- &master_start_cb, &join_request_cb,
- &master_message_cb, &master_message_part_cb,
- NULL);
- mst_chn = GNUNET_PSYC_master_get_channel (mst);
- }
-
-
- static void
- schedule_master_start (void *cls)
- {
- master_start ();
- }
-
-
- /**
- * Main function of the test, run from scheduler.
- *
- * @param cls NULL
- * @param cfg configuration we use (also to connect to PSYC service)
- * @param peer handle to access more of the peer (not used)
- */
- static void
- #if DEBUG_TEST_PSYC
- run (void *cls, char *const *args, const char *cfgfile,
- const struct GNUNET_CONFIGURATION_Handle *c)
- #else
- run (void *cls,
- const struct GNUNET_CONFIGURATION_Handle *c,
- struct GNUNET_TESTING_Peer *peer)
- #endif
- {
- cfg = c;
- end_badly_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL);
-
- GNUNET_CRYPTO_get_peer_identity (cfg, &this_peer);
-
- channel_key = GNUNET_CRYPTO_eddsa_key_create ();
- slave_key = GNUNET_CRYPTO_ecdsa_key_create ();
-
- GNUNET_CRYPTO_eddsa_key_get_public (channel_key, &channel_pub_key);
- GNUNET_CRYPTO_ecdsa_key_get_public (slave_key, &slave_pub_key);
-
- #if DEBUG_TEST_PSYC
- master_start ();
- #else
- /* Allow some time for the services to initialize. */
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
- &schedule_master_start, NULL);
- #endif
- }
-
-
- int
- main (int argc, char *argv[])
- {
- res = 1;
- #if DEBUG_TEST_PSYC
- const struct GNUNET_GETOPT_CommandLineOption opts[] = {
- GNUNET_GETOPT_OPTION_END
- };
- if (GNUNET_OK != GNUNET_PROGRAM_run (argc, argv, "test-psyc",
- "test-psyc [options]",
- opts, &run, NULL))
- return 1;
- #else
- if (0 != GNUNET_TESTING_peer_run ("test-psyc", "test_psyc.conf", &run, NULL))
- return 1;
- #endif
- return res;
- }
-
- /* end of test_psyc.c */
|