diff options
author | Fabian Oehlmann <oehlmann@in.tum.de> | 2013-11-22 18:08:19 +0000 |
---|---|---|
committer | Fabian Oehlmann <oehlmann@in.tum.de> | 2013-11-22 18:08:19 +0000 |
commit | 9aeaad0a88ea5ea5f15c50071a1c3a888e63e131 (patch) | |
tree | 1d8d48964dda3d4974a7003fc49bcc500bcc87fe /src/ats | |
parent | ccf016b4b6e47bf0f37e3d2b00b046784253f003 (diff) | |
download | gnunet-9aeaad0a88ea5ea5f15c50071a1c3a888e63e131.tar.gz gnunet-9aeaad0a88ea5ea5f15c50071a1c3a888e63e131.zip |
added convergence test
Diffstat (limited to 'src/ats')
-rw-r--r-- | src/ats/Makefile.am | 10 | ||||
-rwxr-xr-x | src/ats/test_ats_solver_convergence.c | 341 |
2 files changed, 351 insertions, 0 deletions
diff --git a/src/ats/Makefile.am b/src/ats/Makefile.am index 4851ff8a5..fe07944ff 100644 --- a/src/ats/Makefile.am +++ b/src/ats/Makefile.am | |||
@@ -493,6 +493,16 @@ test_ats_solver_preferences_ril_LDADD = \ | |||
493 | $(top_builddir)/src/hello/libgnunethello.la \ | 493 | $(top_builddir)/src/hello/libgnunethello.la \ |
494 | $(top_builddir)/src/ats/libgnunetats.la \ | 494 | $(top_builddir)/src/ats/libgnunetats.la \ |
495 | $(top_builddir)/src/statistics/libgnunetstatistics.la | 495 | $(top_builddir)/src/statistics/libgnunetstatistics.la |
496 | |||
497 | test_ats_solver_convergence_ril_SOURCES = \ | ||
498 | test_ats_solver_convergence.c \ | ||
499 | test_ats_api_common.c test_ats_api_common.h | ||
500 | test_ats_solver_convergence_ril_LDADD = \ | ||
501 | $(top_builddir)/src/util/libgnunetutil.la \ | ||
502 | $(top_builddir)/src/testing/libgnunettesting.la \ | ||
503 | $(top_builddir)/src/hello/libgnunethello.la \ | ||
504 | $(top_builddir)/src/ats/libgnunetats.la \ | ||
505 | $(top_builddir)/src/statistics/libgnunetstatistics.la | ||
496 | 506 | ||
497 | EXTRA_DIST = \ | 507 | EXTRA_DIST = \ |
498 | ats.h \ | 508 | ats.h \ |
diff --git a/src/ats/test_ats_solver_convergence.c b/src/ats/test_ats_solver_convergence.c new file mode 100755 index 000000000..7a03ecb53 --- /dev/null +++ b/src/ats/test_ats_solver_convergence.c | |||
@@ -0,0 +1,341 @@ | |||
1 | /* | ||
2 | This file is part of GNUnet. | ||
3 | (C) 2010-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 ats/test_ats_solver_add_address.c | ||
22 | * @brief solver test: add address, request address and wait for suggest | ||
23 | * @author Christian Grothoff | ||
24 | * @author Matthias Wachs | ||
25 | * @author Fabian Oehlmann | ||
26 | */ | ||
27 | #include "platform.h" | ||
28 | #include "gnunet_util_lib.h" | ||
29 | #include "gnunet_testbed_service.h" | ||
30 | #include "gnunet_ats_service.h" | ||
31 | #include "test_ats_api_common.h" | ||
32 | |||
33 | /** | ||
34 | * Timeout task | ||
35 | */ | ||
36 | static GNUNET_SCHEDULER_TaskIdentifier end_task; | ||
37 | |||
38 | /** | ||
39 | * Statistics handle | ||
40 | */ | ||
41 | struct GNUNET_STATISTICS_Handle *stats; | ||
42 | |||
43 | /** | ||
44 | * Scheduling handle | ||
45 | */ | ||
46 | static struct GNUNET_ATS_SchedulingHandle *sched_ats; | ||
47 | |||
48 | /** | ||
49 | * Return value | ||
50 | */ | ||
51 | static int ret; | ||
52 | |||
53 | /** | ||
54 | * Test address | ||
55 | */ | ||
56 | static struct Test_Address test_addr; | ||
57 | |||
58 | /** | ||
59 | * Test peer | ||
60 | */ | ||
61 | static struct PeerContext p; | ||
62 | |||
63 | /** | ||
64 | * HELLO address | ||
65 | */ | ||
66 | struct GNUNET_HELLO_Address test_hello_address; | ||
67 | |||
68 | /** | ||
69 | * Session | ||
70 | */ | ||
71 | static void *test_session; | ||
72 | |||
73 | /** | ||
74 | * Test ats info | ||
75 | */ | ||
76 | struct GNUNET_ATS_Information test_ats_info[2]; | ||
77 | |||
78 | /** | ||
79 | * Test ats count | ||
80 | */ | ||
81 | uint32_t test_ats_count; | ||
82 | |||
83 | /** | ||
84 | * Seconds to run the test | ||
85 | */ | ||
86 | unsigned int seconds; | ||
87 | |||
88 | /** | ||
89 | * When the test starts | ||
90 | */ | ||
91 | struct GNUNET_TIME_Absolute time_start; | ||
92 | |||
93 | /** | ||
94 | * Whether to write a data file | ||
95 | */ | ||
96 | int write_data_file; | ||
97 | |||
98 | /** | ||
99 | * File handle | ||
100 | */ | ||
101 | struct GNUNET_DISK_FileHandle *data_file_handle; | ||
102 | |||
103 | /** | ||
104 | * File name | ||
105 | */ | ||
106 | char *data_file_name; | ||
107 | |||
108 | static int | ||
109 | stat_cb(void *cls, const char *subsystem, const char *name, uint64_t value, | ||
110 | int is_persistent); | ||
111 | |||
112 | static void | ||
113 | end (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
114 | { | ||
115 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Done!\n"); | ||
116 | if (end_task == GNUNET_SCHEDULER_NO_TASK) | ||
117 | { | ||
118 | GNUNET_SCHEDULER_cancel (end_task); | ||
119 | end_task = GNUNET_SCHEDULER_NO_TASK; | ||
120 | } | ||
121 | |||
122 | if (NULL != sched_ats) | ||
123 | { | ||
124 | GNUNET_ATS_scheduling_done (sched_ats); | ||
125 | sched_ats = NULL; | ||
126 | } | ||
127 | |||
128 | GNUNET_STATISTICS_watch_cancel (stats, "ats", "# addresses", &stat_cb, NULL); | ||
129 | if (NULL != stats) | ||
130 | { | ||
131 | GNUNET_STATISTICS_destroy (stats, GNUNET_NO); | ||
132 | stats = NULL; | ||
133 | } | ||
134 | |||
135 | /* Close data file */ | ||
136 | if (write_data_file) | ||
137 | { | ||
138 | if (GNUNET_SYSERR == GNUNET_DISK_file_close (data_file_handle)) | ||
139 | GNUNET_log(GNUNET_ERROR_TYPE_ERROR, "Cannot close log file '%s'\n", | ||
140 | data_file_name); | ||
141 | GNUNET_free_non_null(data_file_name); | ||
142 | } | ||
143 | |||
144 | free_test_address (&test_addr); | ||
145 | |||
146 | ret = 0; | ||
147 | } | ||
148 | |||
149 | |||
150 | static void | ||
151 | end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
152 | { | ||
153 | end (NULL, NULL); | ||
154 | ret = GNUNET_SYSERR; | ||
155 | } | ||
156 | |||
157 | static void | ||
158 | address_suggest_cb (void *cls, const struct GNUNET_HELLO_Address *address, | ||
159 | struct Session *session, | ||
160 | struct GNUNET_BANDWIDTH_Value32NBO bandwidth_out, | ||
161 | struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in, | ||
162 | const struct GNUNET_ATS_Information *atsi, | ||
163 | uint32_t ats_count) | ||
164 | { | ||
165 | struct GNUNET_TIME_Relative time_delta; | ||
166 | char *data; | ||
167 | |||
168 | GNUNET_assert (NULL != address); | ||
169 | GNUNET_assert (NULL == session); | ||
170 | GNUNET_assert (ntohl(bandwidth_in.value__) > 0); | ||
171 | GNUNET_assert (ntohl(bandwidth_out.value__) > 0); | ||
172 | |||
173 | time_delta = GNUNET_TIME_absolute_get_difference(time_start, GNUNET_TIME_absolute_get()); | ||
174 | |||
175 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Received suggestion for peer '%s': IN %lu - OUT %lu\n", | ||
176 | GNUNET_i2s (&address->peer), ntohl(bandwidth_in.value__)/1024, ntohl(bandwidth_out.value__)/1024); | ||
177 | |||
178 | if (write_data_file) | ||
179 | { | ||
180 | GNUNET_asprintf(&data,"%f\tIN %u\tOUT %u\n", | ||
181 | (double) time_delta.rel_value_us / 1000000., | ||
182 | ntohl(bandwidth_in.value__)/1024, | ||
183 | ntohl(bandwidth_out.value__)/1024); | ||
184 | if (GNUNET_SYSERR == GNUNET_DISK_file_write(data_file_handle, data, strlen(data))) | ||
185 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Cannot write data to file `%s'\n", data_file_name); | ||
186 | GNUNET_free(data); | ||
187 | } | ||
188 | |||
189 | return; | ||
190 | } | ||
191 | |||
192 | |||
193 | static int | ||
194 | stat_cb(void *cls, const char *subsystem, | ||
195 | const char *name, uint64_t value, | ||
196 | int is_persistent) | ||
197 | { | ||
198 | |||
199 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, "ATS statistics: `%s' `%s' %llu\n", | ||
200 | subsystem,name, value); | ||
201 | GNUNET_ATS_suggest_address (sched_ats, &p.id); | ||
202 | return GNUNET_OK; | ||
203 | } | ||
204 | |||
205 | static void | ||
206 | run (void *cls, const struct GNUNET_CONFIGURATION_Handle *mycfg, | ||
207 | struct GNUNET_TESTING_Peer *peer) | ||
208 | { | ||
209 | stats = GNUNET_STATISTICS_create ("ats", mycfg); | ||
210 | GNUNET_STATISTICS_watch (stats, "ats", "# addresses", &stat_cb, NULL); | ||
211 | |||
212 | |||
213 | /* Connect to ATS scheduling */ | ||
214 | sched_ats = GNUNET_ATS_scheduling_init (mycfg, &address_suggest_cb, NULL); | ||
215 | if (sched_ats == NULL) | ||
216 | { | ||
217 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Could not connect to ATS scheduling!\n"); | ||
218 | GNUNET_SCHEDULER_add_now (&end_badly, NULL); | ||
219 | return; | ||
220 | } | ||
221 | |||
222 | /* Set up file handle */ | ||
223 | GNUNET_asprintf (&data_file_name, "test_convergence_s%d.data", seconds); | ||
224 | data_file_handle = GNUNET_DISK_file_open (data_file_name, | ||
225 | GNUNET_DISK_OPEN_WRITE | GNUNET_DISK_OPEN_CREATE | GNUNET_DISK_OPEN_TRUNCATE, | ||
226 | GNUNET_DISK_PERM_USER_EXEC | GNUNET_DISK_PERM_USER_READ | GNUNET_DISK_PERM_USER_WRITE); | ||
227 | if (NULL == data_file_handle) { | ||
228 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Could not open data file\n"); | ||
229 | GNUNET_free(data_file_handle); | ||
230 | GNUNET_free(data_file_name); | ||
231 | GNUNET_SCHEDULER_add_now (&end_badly, NULL); | ||
232 | return; | ||
233 | } | ||
234 | |||
235 | /* Set up peer */ | ||
236 | memset (&p.id, '1', sizeof (p.id)); | ||
237 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Created peer `%s'\n", | ||
238 | GNUNET_i2s_full(&p.id)); | ||
239 | |||
240 | /* Prepare ATS Information */ | ||
241 | test_ats_info[0].type = htonl (GNUNET_ATS_NETWORK_TYPE); | ||
242 | test_ats_info[0].value = htonl(GNUNET_ATS_NET_WAN); | ||
243 | test_ats_info[1].type = htonl (GNUNET_ATS_QUALITY_NET_DISTANCE); | ||
244 | test_ats_info[1].value = htonl(1); | ||
245 | test_ats_count = 2; | ||
246 | |||
247 | /* Adding address without session */ | ||
248 | test_session = NULL; | ||
249 | create_test_address (&test_addr, "test", test_session, "test", strlen ("test") + 1); | ||
250 | test_hello_address.peer = p.id; | ||
251 | test_hello_address.transport_name = test_addr.plugin; | ||
252 | test_hello_address.address = test_addr.addr; | ||
253 | test_hello_address.address_length = test_addr.addr_len; | ||
254 | |||
255 | /* Adding address */ | ||
256 | GNUNET_ATS_address_add (sched_ats, &test_hello_address, NULL, test_ats_info, test_ats_count); | ||
257 | time_start = GNUNET_TIME_absolute_get(); | ||
258 | |||
259 | end_task = GNUNET_SCHEDULER_add_delayed(GNUNET_TIME_relative_multiply(GNUNET_TIME_relative_get_second_(), seconds), &end, NULL); | ||
260 | } | ||
261 | |||
262 | |||
263 | void | ||
264 | test_run (void *cls, char *const *args, | ||
265 | const char *cfgfile, | ||
266 | const struct GNUNET_CONFIGURATION_Handle * | ||
267 | cfg) | ||
268 | { | ||
269 | char *sep; | ||
270 | char *src_filename = GNUNET_strdup (__FILE__); | ||
271 | char *test_filename = cls; | ||
272 | char *config_file = "none"; | ||
273 | char *solver; | ||
274 | |||
275 | ret = 0; | ||
276 | |||
277 | if (NULL == (sep = (strstr (src_filename,".c")))) | ||
278 | { | ||
279 | GNUNET_break (0); | ||
280 | ret = -1; | ||
281 | //return -1; | ||
282 | } | ||
283 | sep[0] = '\0'; | ||
284 | |||
285 | if (NULL != (sep = strstr (test_filename, ".exe"))) | ||
286 | sep[0] = '\0'; | ||
287 | |||
288 | if (NULL == (solver = strstr (test_filename, src_filename))) | ||
289 | { | ||
290 | GNUNET_break (0); | ||
291 | ret = -1; | ||
292 | } | ||
293 | solver += strlen (src_filename) +1; | ||
294 | |||
295 | if (0 == strcmp(solver, "proportional")) | ||
296 | { | ||
297 | config_file = "test_ats_solver_proportional.conf"; | ||
298 | } | ||
299 | else if (0 == strcmp(solver, "mlp")) | ||
300 | { | ||
301 | config_file = "test_ats_solver_mlp.conf"; | ||
302 | } | ||
303 | else if ((0 == strcmp(solver, "ril"))) | ||
304 | { | ||
305 | config_file = "test_ats_solver_ril.conf"; | ||
306 | } | ||
307 | else | ||
308 | { | ||
309 | GNUNET_break (0); | ||
310 | GNUNET_free (src_filename); | ||
311 | ret = 1; | ||
312 | } | ||
313 | |||
314 | GNUNET_free (src_filename); | ||
315 | |||
316 | if (0 != GNUNET_TESTING_peer_run ("test-ats-solver", | ||
317 | config_file, &run, NULL )) | ||
318 | ret = GNUNET_SYSERR; | ||
319 | } | ||
320 | |||
321 | int | ||
322 | main (int argc, char *argv[]) | ||
323 | { | ||
324 | seconds = 5; | ||
325 | |||
326 | static struct GNUNET_GETOPT_CommandLineOption options[] = { | ||
327 | { 's', "seconds", NULL, | ||
328 | gettext_noop ("seconds to run the test"), | ||
329 | 1, &GNUNET_GETOPT_set_uint, &seconds }, | ||
330 | { 'd', "data-file", NULL, | ||
331 | gettext_noop ("generate data file"), | ||
332 | 0, &GNUNET_GETOPT_set_one, &write_data_file}, | ||
333 | GNUNET_GETOPT_OPTION_END | ||
334 | }; | ||
335 | |||
336 | GNUNET_PROGRAM_run2 (argc, argv, argv[0], NULL, options, &test_run, argv[0], GNUNET_YES); | ||
337 | |||
338 | return ret; | ||
339 | } | ||
340 | |||
341 | /* end of file test_ats_solver_convergence.c */ | ||