2 This file is part of GNUnet.
3 Copyright (C) 2004, 2009 GNUnet e.V.
5 GNUnet is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published
7 by the Free Software Foundation; either version 3, or (at your
8 option) any later version.
10 GNUnet is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with GNUnet; see the file COPYING. If not, write to the
17 Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
18 Boston, MA 02110-1301, USA.
22 * @file peerinfo/test_peerinfo_api_notify_friend_only.c
23 * @brief testcase friend only HELLO restrictions in for peerinfo
24 * @author Christian Grothoff
25 * @author Matthias Wachs
28 * - test merging of HELLOs (add same peer twice...)
31 #include "gnunet_hello_lib.h"
32 #include "gnunet_util_lib.h"
33 #include "gnunet_peerinfo_service.h"
34 #include "gnunet_testing_lib.h"
37 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
39 static struct GNUNET_PEERINFO_Handle *h;
40 static struct GNUNET_PEERINFO_NotifyContext *pnc_w_fo;
41 static struct GNUNET_PEERINFO_NotifyContext *pnc_wo_fo;
43 static const struct GNUNET_CONFIGURATION_Handle *mycfg;
45 static int global_ret;
48 * Did we get a HELLO callback for notification handle with friend HELLOS
51 static int res_cb_w_fo;
54 * Did we get a HELLO callback for notification handle without friend HELLOS
57 static int res_cb_wo_fo;
59 struct GNUNET_PeerIdentity pid;
61 struct GNUNET_SCHEDULER_Task * timeout_task;
64 end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
68 if (NULL != pnc_wo_fo)
70 GNUNET_PEERINFO_notify_cancel (pnc_wo_fo);
75 GNUNET_PEERINFO_notify_cancel (pnc_w_fo);
80 GNUNET_PEERINFO_disconnect (h);
87 done (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
90 GNUNET_PEERINFO_notify_cancel (pnc_w_fo);
92 if (NULL != pnc_wo_fo)
93 GNUNET_PEERINFO_notify_cancel (pnc_wo_fo);
95 GNUNET_PEERINFO_disconnect (h);
98 if (NULL != timeout_task)
100 GNUNET_SCHEDULER_cancel (timeout_task);
104 if ((GNUNET_YES == res_cb_w_fo) && (GNUNET_NO == res_cb_wo_fo))
111 address_generator (void *cls, size_t max, void *buf)
115 struct GNUNET_HELLO_Address address;
118 return GNUNET_SYSERR; /* Done */
119 memset (&address.peer, 0, sizeof(struct GNUNET_PeerIdentity));
120 address.address = "Address";
121 address.transport_name = "peerinfotest";
122 address.address_length = *agc;
123 ret = GNUNET_HELLO_add_address (&address,
124 GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_HOURS), buf, max);
130 process_w_fo (void *cls, const struct GNUNET_PeerIdentity *peer,
131 const struct GNUNET_HELLO_Message *hello, const char *err_msg)
133 if (err_msg != NULL )
135 GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
136 _("Error in communication with PEERINFO service\n"));
137 GNUNET_SCHEDULER_add_now (&done, NULL );
143 GNUNET_log(GNUNET_ERROR_TYPE_INFO,
144 "Received callback for peer `%s' %s HELLO\n", GNUNET_i2s (peer),
145 (NULL != hello) ? "with" : "without");
150 if (GNUNET_NO == GNUNET_HELLO_is_friend_only (hello))
156 GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Received %s HELLO for peer `%s'\n",
157 (GNUNET_YES == GNUNET_HELLO_is_friend_only (hello)) ? "friend only" : "public",
159 if (0 == memcmp (&pid, peer, sizeof(pid)))
161 res_cb_w_fo = GNUNET_YES;
162 GNUNET_SCHEDULER_add_now (&done, NULL );
169 process_wo_fo (void *cls, const struct GNUNET_PeerIdentity *peer,
170 const struct GNUNET_HELLO_Message *hello, const char *err_msg)
172 if (err_msg != NULL )
174 GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
175 _("Error in communication with PEERINFO service\n"));
176 GNUNET_SCHEDULER_add_now (&done, NULL );
182 GNUNET_log(GNUNET_ERROR_TYPE_INFO,
183 "Received callback for peer `%s' %s HELLO\n", GNUNET_i2s (peer),
184 (NULL != hello) ? "with" : "without");
189 if (GNUNET_YES == GNUNET_HELLO_is_friend_only (hello))
195 GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Received %s HELLO for peer `%s'\n",
196 (GNUNET_YES == GNUNET_HELLO_is_friend_only (hello)) ? "friend only" : "public",
198 if (0 == memcmp (&pid, peer, sizeof(pid)))
201 res_cb_wo_fo = GNUNET_YES;
207 add_peer_done (void *cls, const char *emsg)
216 GNUNET_SCHEDULER_cancel (timeout_task);
217 timeout_task = GNUNET_SCHEDULER_add_now (&end_badly, NULL );
224 struct GNUNET_HELLO_Message *h2;
228 memset (&pid, 32, sizeof(pid));
229 h2 = GNUNET_HELLO_create (&pid.public_key, &address_generator, &agc,
231 GNUNET_PEERINFO_add_peer (h, h2, &add_peer_done, NULL );
237 run (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg,
238 struct GNUNET_TESTING_Peer *peer)
240 timeout_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL );
242 pnc_w_fo = GNUNET_PEERINFO_notify (mycfg, GNUNET_YES, &process_w_fo, NULL );
243 pnc_wo_fo = GNUNET_PEERINFO_notify (mycfg, GNUNET_NO, &process_wo_fo, NULL );
244 h = GNUNET_PEERINFO_connect (cfg);
245 GNUNET_assert(NULL != h);
250 main (int argc, char *argv[])
252 res_cb_w_fo = GNUNET_NO;
253 res_cb_wo_fo = GNUNET_NO;
256 != GNUNET_TESTING_service_run ("test-peerinfo-api-friend-only",
257 "peerinfo", "test_peerinfo_api_data.conf", &run, NULL ))
262 /* end of test_peerinfo_api_notify_friend_only.c */