123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255 |
- /*
- This file is part of GNUnet.
- Copyright (C) 2004, 2009 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 peerinfo/test_peerinfo_api_notify_friend_only.c
- * @brief testcase friend only HELLO restrictions in for peerinfo
- * @author Christian Grothoff
- * @author Matthias Wachs
- *
- * TODO:
- * - test merging of HELLOs (add same peer twice...)
- */
- #include "platform.h"
- #include "gnunet_hello_lib.h"
- #include "gnunet_util_lib.h"
- #include "gnunet_peerinfo_service.h"
- #include "gnunet_testing_lib.h"
- #include "peerinfo.h"
- #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
- static struct GNUNET_PEERINFO_Handle *h;
- static struct GNUNET_PEERINFO_NotifyContext *pnc_w_fo;
- static struct GNUNET_PEERINFO_NotifyContext *pnc_wo_fo;
- static const struct GNUNET_CONFIGURATION_Handle *mycfg;
- static int global_ret;
- /**
- * Did we get a HELLO callback for notification handle with friend HELLOS
- * (expected)
- */
- static int res_cb_w_fo;
- /**
- * Did we get a HELLO callback for notification handle without friend HELLOS
- * (not expected)
- */
- static int res_cb_wo_fo;
- struct GNUNET_PeerIdentity pid;
- struct GNUNET_SCHEDULER_Task *timeout_task;
- static void
- end_badly (void *cls)
- {
- timeout_task = NULL;
- GNUNET_break(0);
- if (NULL != pnc_wo_fo)
- {
- GNUNET_PEERINFO_notify_cancel (pnc_wo_fo);
- pnc_wo_fo = NULL;
- }
- if (NULL != pnc_w_fo)
- {
- GNUNET_PEERINFO_notify_cancel (pnc_w_fo);
- pnc_w_fo = NULL;
- }
- if (NULL != h)
- {
- GNUNET_PEERINFO_disconnect (h);
- h = NULL;
- }
- global_ret = 255;
- }
- static void
- done (void *cls)
- {
- if (NULL != pnc_w_fo)
- GNUNET_PEERINFO_notify_cancel (pnc_w_fo);
- pnc_w_fo = NULL;
- if (NULL != pnc_wo_fo)
- GNUNET_PEERINFO_notify_cancel (pnc_wo_fo);
- pnc_wo_fo = NULL;
- GNUNET_PEERINFO_disconnect (h);
- h = NULL;
- if (NULL != timeout_task)
- {
- GNUNET_SCHEDULER_cancel (timeout_task);
- timeout_task = NULL;
- }
- if ((GNUNET_YES == res_cb_w_fo) && (GNUNET_NO == res_cb_wo_fo))
- global_ret = 0;
- else
- GNUNET_break(0);
- }
- static ssize_t
- address_generator (void *cls, size_t max, void *buf)
- {
- size_t *agc = cls;
- ssize_t ret;
- struct GNUNET_HELLO_Address address;
- if (0 == *agc)
- return GNUNET_SYSERR; /* Done */
- memset (&address.peer, 0, sizeof(struct GNUNET_PeerIdentity));
- address.address = "Address";
- address.transport_name = "peerinfotest";
- address.address_length = *agc;
- ret = GNUNET_HELLO_add_address (&address,
- GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_HOURS), buf, max);
- (*agc)--;
- return ret;
- }
- static void
- process_w_fo (void *cls,
- const struct GNUNET_PeerIdentity *peer,
- const struct GNUNET_HELLO_Message *hello,
- const char *err_msg)
- {
- if (err_msg != NULL)
- {
- GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
- _("Error in communication with PEERINFO service\n"));
- GNUNET_SCHEDULER_add_now (&done, NULL);
- return;
- }
- if (NULL != peer)
- {
- GNUNET_log(GNUNET_ERROR_TYPE_INFO,
- "Received callback for peer `%s' %s HELLO\n", GNUNET_i2s (peer),
- (NULL != hello) ? "with" : "without");
- if (NULL == hello)
- return;
- if (GNUNET_NO == GNUNET_HELLO_is_friend_only (hello))
- {
- GNUNET_break(0);
- return;
- }
- GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Received %s HELLO for peer `%s'\n",
- (GNUNET_YES == GNUNET_HELLO_is_friend_only (hello)) ? "friend only" : "public",
- GNUNET_i2s (peer));
- if (0 == memcmp (&pid, peer, sizeof(pid)))
- {
- res_cb_w_fo = GNUNET_YES;
- GNUNET_SCHEDULER_add_now (&done, NULL);
- }
- return;
- }
- }
- static void
- process_wo_fo (void *cls, const struct GNUNET_PeerIdentity *peer,
- const struct GNUNET_HELLO_Message *hello, const char *err_msg)
- {
- if (err_msg != NULL)
- {
- GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
- _("Error in communication with PEERINFO service\n"));
- GNUNET_SCHEDULER_add_now (&done, NULL);
- return;
- }
- if (NULL != peer)
- {
- GNUNET_log(GNUNET_ERROR_TYPE_INFO,
- "Received callback for peer `%s' %s HELLO\n", GNUNET_i2s (peer),
- (NULL != hello) ? "with" : "without");
- if (NULL == hello)
- return;
- if (GNUNET_YES == GNUNET_HELLO_is_friend_only (hello))
- {
- GNUNET_break(0);
- return;
- }
- GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Received %s HELLO for peer `%s'\n",
- (GNUNET_YES == GNUNET_HELLO_is_friend_only (hello)) ? "friend only" : "public",
- GNUNET_i2s (peer));
- if (0 == memcmp (&pid, peer, sizeof(pid)))
- {
- GNUNET_break(0);
- res_cb_wo_fo = GNUNET_YES;
- }
- }
- }
- static void
- add_peer ()
- {
- struct GNUNET_HELLO_Message *h2;
- size_t agc;
- agc = 2;
- memset (&pid, 32, sizeof(pid));
- h2 = GNUNET_HELLO_create (&pid.public_key, &address_generator, &agc,
- GNUNET_YES);
- GNUNET_PEERINFO_add_peer (h, h2, NULL, NULL);
- GNUNET_free(h2);
- }
- static void
- run (void *cls,
- const struct GNUNET_CONFIGURATION_Handle *cfg,
- struct GNUNET_TESTING_Peer *peer)
- {
- timeout_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL);
- mycfg = cfg;
- pnc_w_fo = GNUNET_PEERINFO_notify (mycfg, GNUNET_YES, &process_w_fo, NULL);
- pnc_wo_fo = GNUNET_PEERINFO_notify (mycfg, GNUNET_NO, &process_wo_fo, NULL);
- h = GNUNET_PEERINFO_connect (cfg);
- GNUNET_assert(NULL != h);
- add_peer ();
- }
- int
- main (int argc, char *argv[])
- {
- res_cb_w_fo = GNUNET_NO;
- res_cb_wo_fo = GNUNET_NO;
- global_ret = 3;
- if (0 != GNUNET_TESTING_service_run ("test-peerinfo-api-friend-only",
- "peerinfo",
- "test_peerinfo_api_data.conf",
- &run, NULL))
- return 1;
- return global_ret;
- }
- /* end of test_peerinfo_api_notify_friend_only.c */
|