aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/peerinfo/Makefile.am9
-rw-r--r--src/peerinfo/test_peerinfo_api_notify_friend_only.c258
2 files changed, 267 insertions, 0 deletions
diff --git a/src/peerinfo/Makefile.am b/src/peerinfo/Makefile.am
index 5a2558048..89ce66a26 100644
--- a/src/peerinfo/Makefile.am
+++ b/src/peerinfo/Makefile.am
@@ -49,6 +49,7 @@ endif
49check_PROGRAMS = \ 49check_PROGRAMS = \
50 test_peerinfo_api \ 50 test_peerinfo_api \
51 test_peerinfo_api_friend_only \ 51 test_peerinfo_api_friend_only \
52 test_peerinfo_api_notify_friend_only \
52 $(PEERINFO_BENCHMARKS) 53 $(PEERINFO_BENCHMARKS)
53 54
54if ENABLE_TEST_RUN 55if ENABLE_TEST_RUN
@@ -71,6 +72,14 @@ test_peerinfo_api_friend_only_LDADD = \
71 $(top_builddir)/src/testing/libgnunettesting.la \ 72 $(top_builddir)/src/testing/libgnunettesting.la \
72 $(top_builddir)/src/util/libgnunetutil.la 73 $(top_builddir)/src/util/libgnunetutil.la
73 74
75test_peerinfo_api_notify_friend_only_SOURCES = \
76 test_peerinfo_api_notify_friend_only.c
77test_peerinfo_api_notify_friend_only_LDADD = \
78 $(top_builddir)/src/hello/libgnunethello.la \
79 $(top_builddir)/src/peerinfo/libgnunetpeerinfo.la \
80 $(top_builddir)/src/testing/libgnunettesting.la \
81 $(top_builddir)/src/util/libgnunetutil.la
82
74perf_peerinfo_api_SOURCES = \ 83perf_peerinfo_api_SOURCES = \
75 perf_peerinfo_api.c 84 perf_peerinfo_api.c
76perf_peerinfo_api_LDADD = \ 85perf_peerinfo_api_LDADD = \
diff --git a/src/peerinfo/test_peerinfo_api_notify_friend_only.c b/src/peerinfo/test_peerinfo_api_notify_friend_only.c
new file mode 100644
index 000000000..3afd50e68
--- /dev/null
+++ b/src/peerinfo/test_peerinfo_api_notify_friend_only.c
@@ -0,0 +1,258 @@
1/*
2 This file is part of GNUnet.
3 (C) 2004, 2009 Christian Grothoff (and other contributing authors)
4
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.
9
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.
14
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., 59 Temple Place - Suite 330,
18 Boston, MA 02111-1307, USA.
19*/
20
21/**
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
26 *
27 * TODO:
28 * - test merging of HELLOs (add same peer twice...)
29 */
30#include "platform.h"
31#include "gnunet_hello_lib.h"
32#include "gnunet_util_lib.h"
33#include "gnunet_peerinfo_service.h"
34#include "gnunet_testing_lib.h"
35#include "peerinfo.h"
36
37#define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
38
39static struct GNUNET_PEERINFO_Handle *h;
40static struct GNUNET_PEERINFO_NotifyContext *pnc_w_fo;
41static struct GNUNET_PEERINFO_NotifyContext *pnc_wo_fo;
42
43static const struct GNUNET_CONFIGURATION_Handle *mycfg;
44
45
46static int global_ret;
47
48/**
49 * Did we get a HELLO callback for notification handle with friend HELLOS
50 * (expected)
51 */
52static int res_cb_w_fo;
53
54/**
55 * Did we get a HELLO callback for notification handle without friend HELLOS
56 * (not expected)
57 */
58static int res_cb_wo_fo;
59
60GNUNET_SCHEDULER_TaskIdentifier timeout_task;
61
62
63static void
64end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
65{
66 timeout_task = GNUNET_SCHEDULER_NO_TASK;
67 GNUNET_break (0);
68 if (NULL != pnc_wo_fo)
69 {
70 GNUNET_PEERINFO_notify_cancel (pnc_wo_fo);
71 pnc_wo_fo = NULL;
72 }
73 if (NULL != pnc_w_fo)
74 {
75 GNUNET_PEERINFO_notify_cancel (pnc_w_fo);
76 pnc_w_fo = NULL;
77 }
78 if (NULL != h)
79 {
80 GNUNET_PEERINFO_disconnect (h);
81 h = NULL;
82 }
83 global_ret = 255;
84}
85
86static void
87done (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
88{
89 if (NULL != pnc_w_fo)
90 GNUNET_PEERINFO_notify_cancel (pnc_w_fo);
91 pnc_w_fo = NULL;
92 if (NULL != pnc_wo_fo)
93 GNUNET_PEERINFO_notify_cancel (pnc_wo_fo);
94 pnc_wo_fo = NULL;
95 GNUNET_PEERINFO_disconnect (h);
96 h = NULL;
97
98 if (GNUNET_SCHEDULER_NO_TASK != timeout_task)
99 {
100 GNUNET_SCHEDULER_cancel (timeout_task);
101 timeout_task = GNUNET_SCHEDULER_NO_TASK;
102 }
103
104 if ((GNUNET_YES == res_cb_w_fo) && (GNUNET_NO == res_cb_wo_fo))
105 global_ret = 0;
106 else
107 GNUNET_break (0);
108}
109
110static size_t
111address_generator (void *cls, size_t max, void *buf)
112{
113 size_t *agc = cls;
114 size_t ret;
115 struct GNUNET_HELLO_Address address;
116
117 if (0 == *agc)
118 return 0;
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 =
124 GNUNET_HELLO_add_address (&address,
125 GNUNET_TIME_relative_to_absolute
126 (GNUNET_TIME_UNIT_HOURS), buf, max);
127 (*agc)--;
128 return ret;
129}
130
131static void
132process_w_fo (void *cls, const struct GNUNET_PeerIdentity *peer,
133 const struct GNUNET_HELLO_Message *hello, const char *err_msg)
134{
135 if (err_msg != NULL)
136 {
137 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
138 _("Error in communication with PEERINFO service\n"));
139 GNUNET_SCHEDULER_add_now(&done, NULL);
140 return;
141 }
142
143 if (NULL != peer)
144 {
145 if (NULL == hello)
146 return;
147
148 if (GNUNET_NO == GNUNET_HELLO_is_friend_only(hello))
149 {
150 GNUNET_break (0);
151 return;
152 }
153
154 GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Received %s HELLO\n",
155 (GNUNET_YES == GNUNET_HELLO_is_friend_only (hello)) ? "friend only" : "public");
156 res_cb_w_fo = GNUNET_YES;
157 GNUNET_SCHEDULER_add_now(&done, NULL);
158 return;
159 }
160}
161
162static void
163process_wo_fo (void *cls, const struct GNUNET_PeerIdentity *peer,
164 const struct GNUNET_HELLO_Message *hello, const char *err_msg)
165{
166 if (err_msg != NULL)
167 {
168 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
169 _("Error in communication with PEERINFO service\n"));
170 GNUNET_SCHEDULER_add_now(&done, NULL);
171 return;
172 }
173
174 if (NULL != peer)
175 {
176 if (NULL == hello)
177 return;
178
179 GNUNET_break (0);
180
181 if (GNUNET_YES == GNUNET_HELLO_is_friend_only(hello))
182 {
183 GNUNET_break (0);
184 return;
185 }
186
187 GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Received %s HELLO\n",
188 (GNUNET_YES == GNUNET_HELLO_is_friend_only (hello)) ? "friend only" : "public");
189 res_cb_wo_fo = GNUNET_YES;
190 }
191}
192
193static void
194add_peer_done (void *cls, const char *emsg)
195{
196 if (NULL == emsg)
197 {
198 pnc_w_fo = GNUNET_PEERINFO_notify (mycfg, GNUNET_YES, &process_w_fo, NULL);
199 pnc_wo_fo = GNUNET_PEERINFO_notify (mycfg, GNUNET_NO, &process_wo_fo, NULL);
200 return;
201 }
202 else
203 {
204 GNUNET_break (0);
205 GNUNET_SCHEDULER_cancel (timeout_task);
206 timeout_task = GNUNET_SCHEDULER_add_now (&end_badly, NULL);
207 }
208}
209
210
211
212static void
213add_peer ()
214{
215 struct GNUNET_CRYPTO_EccPublicKeyBinaryEncoded pkey;
216 struct GNUNET_PeerIdentity pid;
217 struct GNUNET_HELLO_Message *h2;
218 size_t agc;
219
220 agc = 2;
221 memset (&pkey, 32, sizeof (pkey));
222 GNUNET_CRYPTO_hash (&pkey, sizeof (pkey), &pid.hashPubKey);
223 h2 = GNUNET_HELLO_create (&pkey, &address_generator, &agc, GNUNET_YES);
224 GNUNET_PEERINFO_add_peer (h, h2, &add_peer_done, NULL);
225 GNUNET_free (h2);
226
227}
228
229
230
231static void
232run (void *cls,
233 const struct GNUNET_CONFIGURATION_Handle *cfg,
234 struct GNUNET_TESTING_Peer *peer)
235{
236 timeout_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL);
237 mycfg = cfg;
238 h = GNUNET_PEERINFO_connect (cfg);
239 GNUNET_assert (NULL != h);
240 add_peer ();
241}
242
243
244int
245main (int argc, char *argv[])
246{
247 res_cb_w_fo = GNUNET_NO;
248 res_cb_wo_fo = GNUNET_NO;
249 global_ret = 3;
250 if (0 != GNUNET_TESTING_service_run ("test-peerinfo-api-friend-only",
251 "peerinfo",
252 "test_peerinfo_api_data.conf",
253 &run, NULL))
254 return 1;
255 return global_ret;
256}
257
258/* end of test_peerinfo_api_notify_friend_only.c */