diff options
author | Bruno Cabral <bcabral@uw.edu> | 2014-09-05 12:49:58 +0000 |
---|---|---|
committer | Bruno Cabral <bcabral@uw.edu> | 2014-09-05 12:49:58 +0000 |
commit | a25c73343c817385ed151f911b70dbb82361d8d3 (patch) | |
tree | 6ddada178ddbc99e1a0e1687f18a2944dad1f9b3 /src/fs/test_fs_search_with_and.c | |
parent | ab98625d6e8f82dc7c897c48081b0f1fef947f8d (diff) | |
download | gnunet-a25c73343c817385ed151f911b70dbb82361d8d3.tar.gz gnunet-a25c73343c817385ed151f911b70dbb82361d8d3.zip |
Add fs search test with multiple files using an AND operator
Diffstat (limited to 'src/fs/test_fs_search_with_and.c')
-rw-r--r-- | src/fs/test_fs_search_with_and.c | 263 |
1 files changed, 263 insertions, 0 deletions
diff --git a/src/fs/test_fs_search_with_and.c b/src/fs/test_fs_search_with_and.c new file mode 100644 index 000000000..de016c2ae --- /dev/null +++ b/src/fs/test_fs_search_with_and.c | |||
@@ -0,0 +1,263 @@ | |||
1 | /* | ||
2 | This file is part of GNUnet. | ||
3 | (C) 2004-2013 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 | * @file fs/test_fs_search_with_and.c | ||
22 | * @brief testcase for publishing multiple files and search with a and operator | ||
23 | * @author Bruno Cabral - 99% based on Christian Grothoff code | ||
24 | */ | ||
25 | #include "platform.h" | ||
26 | #include "gnunet_util_lib.h" | ||
27 | #include "gnunet_testing_lib.h" | ||
28 | #include "gnunet_fs_service.h" | ||
29 | |||
30 | |||
31 | /** | ||
32 | * File-size we use for testing. | ||
33 | */ | ||
34 | #define FILESIZE 1024 | ||
35 | |||
36 | /** | ||
37 | * Number of files for testing. | ||
38 | */ | ||
39 | #define NUM_FILES 10 | ||
40 | |||
41 | |||
42 | /** | ||
43 | * How long until we give up on transmitting the message? | ||
44 | */ | ||
45 | #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 60) | ||
46 | |||
47 | /** | ||
48 | * How long should our test-content live? | ||
49 | */ | ||
50 | #define LIFETIME GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 15) | ||
51 | |||
52 | |||
53 | static struct GNUNET_TIME_Absolute start; | ||
54 | |||
55 | static struct GNUNET_FS_Handle *fs; | ||
56 | |||
57 | static struct GNUNET_FS_SearchContext *search; | ||
58 | |||
59 | static struct GNUNET_FS_PublishContext *publish; | ||
60 | |||
61 | static GNUNET_SCHEDULER_TaskIdentifier timeout_task; | ||
62 | |||
63 | static int err; | ||
64 | |||
65 | static int processed_files; | ||
66 | |||
67 | |||
68 | static void | ||
69 | abort_publish_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
70 | { | ||
71 | if (NULL != publish) | ||
72 | { | ||
73 | GNUNET_FS_publish_stop (publish); | ||
74 | publish = NULL; | ||
75 | } | ||
76 | if (GNUNET_SCHEDULER_NO_TASK != timeout_task) | ||
77 | { | ||
78 | GNUNET_SCHEDULER_cancel (timeout_task); | ||
79 | timeout_task = GNUNET_SCHEDULER_NO_TASK; | ||
80 | } | ||
81 | } | ||
82 | |||
83 | |||
84 | static void | ||
85 | abort_error (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
86 | { | ||
87 | fprintf (stderr, | ||
88 | "Timeout\n"); | ||
89 | timeout_task = GNUNET_SCHEDULER_NO_TASK; | ||
90 | if (NULL != search) | ||
91 | { | ||
92 | GNUNET_FS_search_stop (search); | ||
93 | search = NULL; | ||
94 | } | ||
95 | if (NULL != publish) | ||
96 | { | ||
97 | GNUNET_FS_publish_stop (publish); | ||
98 | publish = NULL; | ||
99 | } | ||
100 | err = 1; | ||
101 | } | ||
102 | |||
103 | |||
104 | static void | ||
105 | abort_search_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
106 | { | ||
107 | if (NULL != search) | ||
108 | { | ||
109 | GNUNET_FS_search_stop (search); | ||
110 | search = NULL; | ||
111 | } | ||
112 | } | ||
113 | |||
114 | |||
115 | static void * | ||
116 | progress_cb (void *cls, const struct GNUNET_FS_ProgressInfo *event) | ||
117 | { | ||
118 | |||
119 | struct GNUNET_FS_Uri *kuri; | ||
120 | |||
121 | switch (event->status) | ||
122 | { | ||
123 | case GNUNET_FS_STATUS_PUBLISH_PROGRESS: | ||
124 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, | ||
125 | "Publish is progressing (%llu/%llu at level %u off %llu)...\n", | ||
126 | (unsigned long long) event->value.publish.completed, | ||
127 | (unsigned long long) event->value.publish.size, | ||
128 | event->value.publish.specifics.progress.depth, | ||
129 | (unsigned long long) event->value.publish.specifics. | ||
130 | progress.offset); | ||
131 | break; | ||
132 | case GNUNET_FS_STATUS_PUBLISH_PROGRESS_DIRECTORY: | ||
133 | break; | ||
134 | case GNUNET_FS_STATUS_PUBLISH_COMPLETED: | ||
135 | if(processed_files == NUM_FILES) | ||
136 | { | ||
137 | kuri = GNUNET_FS_uri_ksk_create ("+down_foo +down_bar", NULL); | ||
138 | GNUNET_assert (kuri != NULL); | ||
139 | |||
140 | start = GNUNET_TIME_absolute_get (); | ||
141 | search = | ||
142 | GNUNET_FS_search_start (fs, kuri, 1, GNUNET_FS_SEARCH_OPTION_NONE, | ||
143 | "search"); | ||
144 | GNUNET_FS_uri_destroy (kuri); | ||
145 | GNUNET_assert (search != NULL); | ||
146 | } | ||
147 | break; | ||
148 | case GNUNET_FS_STATUS_SEARCH_RESULT: | ||
149 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, | ||
150 | "Search complete.\n"); | ||
151 | GNUNET_SCHEDULER_add_continuation (&abort_search_task, NULL, | ||
152 | GNUNET_SCHEDULER_REASON_PREREQ_DONE); | ||
153 | break; | ||
154 | case GNUNET_FS_STATUS_PUBLISH_ERROR: | ||
155 | FPRINTF (stderr, "Error publishing file: %s\n", | ||
156 | event->value.publish.specifics.error.message); | ||
157 | GNUNET_break (0); | ||
158 | GNUNET_SCHEDULER_add_continuation (&abort_publish_task, NULL, | ||
159 | GNUNET_SCHEDULER_REASON_PREREQ_DONE); | ||
160 | break; | ||
161 | case GNUNET_FS_STATUS_SEARCH_ERROR: | ||
162 | FPRINTF (stderr, "Error searching file: %s\n", | ||
163 | event->value.search.specifics.error.message); | ||
164 | GNUNET_SCHEDULER_add_continuation (&abort_search_task, NULL, | ||
165 | GNUNET_SCHEDULER_REASON_PREREQ_DONE); | ||
166 | break; | ||
167 | case GNUNET_FS_STATUS_PUBLISH_START: | ||
168 | GNUNET_assert (0 == strcmp ("publish-context", event->value.publish.cctx)); | ||
169 | GNUNET_assert (NULL == event->value.publish.pctx); | ||
170 | GNUNET_assert (FILESIZE == event->value.publish.size); | ||
171 | GNUNET_assert (0 == event->value.publish.completed); | ||
172 | GNUNET_assert (1 == event->value.publish.anonymity); | ||
173 | processed_files++; | ||
174 | break; | ||
175 | case GNUNET_FS_STATUS_PUBLISH_STOPPED: | ||
176 | GNUNET_assert (publish == event->value.publish.pc); | ||
177 | GNUNET_assert (FILESIZE == event->value.publish.size); | ||
178 | GNUNET_assert (1 == event->value.publish.anonymity); | ||
179 | GNUNET_FS_stop (fs); | ||
180 | fs = NULL; | ||
181 | break; | ||
182 | case GNUNET_FS_STATUS_SEARCH_START: | ||
183 | GNUNET_assert (search == NULL); | ||
184 | GNUNET_assert (0 == strcmp ("search", event->value.search.cctx)); | ||
185 | GNUNET_assert (1 == event->value.search.anonymity); | ||
186 | break; | ||
187 | case GNUNET_FS_STATUS_SEARCH_RESULT_STOPPED: | ||
188 | break; | ||
189 | case GNUNET_FS_STATUS_SEARCH_STOPPED: | ||
190 | GNUNET_assert (search == event->value.search.sc); | ||
191 | GNUNET_SCHEDULER_add_continuation (&abort_publish_task, NULL, | ||
192 | GNUNET_SCHEDULER_REASON_PREREQ_DONE); | ||
193 | break; | ||
194 | default: | ||
195 | FPRINTF (stderr, "Unexpected event: %d\n", event->status); | ||
196 | break; | ||
197 | } | ||
198 | return NULL; | ||
199 | } | ||
200 | |||
201 | |||
202 | static void | ||
203 | run (void *cls, | ||
204 | const struct GNUNET_CONFIGURATION_Handle *cfg, | ||
205 | struct GNUNET_TESTING_Peer *peer) | ||
206 | { | ||
207 | const char *keywords[] = { | ||
208 | "down_foo", | ||
209 | "down_bar" | ||
210 | }; | ||
211 | char *buf; | ||
212 | struct GNUNET_CONTAINER_MetaData *meta; | ||
213 | struct GNUNET_FS_Uri *kuri; | ||
214 | struct GNUNET_FS_BlockOptions bo; | ||
215 | struct GNUNET_FS_FileInformation *fi; | ||
216 | size_t i; | ||
217 | size_t j; | ||
218 | |||
219 | fs = GNUNET_FS_start (cfg, "test-fs-search", &progress_cb, NULL, | ||
220 | GNUNET_FS_FLAGS_NONE, GNUNET_FS_OPTIONS_END); | ||
221 | GNUNET_assert (NULL != fs); | ||
222 | |||
223 | processed_files = 0; | ||
224 | for(j = 0; j < NUM_FILES; j++){ | ||
225 | buf = GNUNET_malloc (FILESIZE); | ||
226 | for (i = 0; i < FILESIZE; i++) | ||
227 | buf[i] = GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, 256); | ||
228 | meta = GNUNET_CONTAINER_meta_data_create (); | ||
229 | kuri = GNUNET_FS_uri_ksk_create_from_args (2, keywords); | ||
230 | bo.content_priority = 42; | ||
231 | bo.anonymity_level = 1; | ||
232 | bo.replication_level = 0; | ||
233 | bo.expiration_time = GNUNET_TIME_relative_to_absolute (LIFETIME); | ||
234 | fi = GNUNET_FS_file_information_create_from_data (fs, "publish-context", | ||
235 | FILESIZE, buf, kuri, meta, | ||
236 | GNUNET_NO, &bo); | ||
237 | GNUNET_FS_uri_destroy (kuri); | ||
238 | GNUNET_CONTAINER_meta_data_destroy (meta); | ||
239 | GNUNET_assert (NULL != fi); | ||
240 | start = GNUNET_TIME_absolute_get (); | ||
241 | publish = | ||
242 | GNUNET_FS_publish_start (fs, fi, NULL, NULL, NULL, | ||
243 | GNUNET_FS_PUBLISH_OPTION_NONE); | ||
244 | GNUNET_assert (publish != NULL); | ||
245 | } | ||
246 | |||
247 | |||
248 | timeout_task = GNUNET_SCHEDULER_add_delayed (LIFETIME, | ||
249 | &abort_error, NULL); | ||
250 | } | ||
251 | |||
252 | |||
253 | int | ||
254 | main (int argc, char *argv[]) | ||
255 | { | ||
256 | if (0 != GNUNET_TESTING_peer_run ("test-fs-search-with-and", | ||
257 | "test_fs_search_data.conf", | ||
258 | &run, NULL)) | ||
259 | return 1; | ||
260 | return err; | ||
261 | } | ||
262 | |||
263 | /* end of test_fs_search.c */ | ||