diff options
author | Christian Grothoff <christian@grothoff.org> | 2015-03-25 14:31:03 +0000 |
---|---|---|
committer | Christian Grothoff <christian@grothoff.org> | 2015-03-25 14:31:03 +0000 |
commit | 00f87691363acaf94f533793654e610e46bf4e1d (patch) | |
tree | 7598432b2335d378d240dcba84aabea7e0f31a14 /src/fragmentation/test_fragmentation_parallel.c | |
parent | f1ba43deea5e7a1fbc22b429196d1f9fb4e1a9f3 (diff) | |
download | gnunet-00f87691363acaf94f533793654e610e46bf4e1d.tar.gz gnunet-00f87691363acaf94f533793654e610e46bf4e1d.zip |
-trying to fix AE's problem on Guix with more fancy retransmission logic
Diffstat (limited to 'src/fragmentation/test_fragmentation_parallel.c')
-rw-r--r-- | src/fragmentation/test_fragmentation_parallel.c | 253 |
1 files changed, 253 insertions, 0 deletions
diff --git a/src/fragmentation/test_fragmentation_parallel.c b/src/fragmentation/test_fragmentation_parallel.c new file mode 100644 index 000000000..b55ec58bf --- /dev/null +++ b/src/fragmentation/test_fragmentation_parallel.c | |||
@@ -0,0 +1,253 @@ | |||
1 | /* | ||
2 | This file is part of GNUnet | ||
3 | Copyright (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 | * @file fragmentation/test_fragmentation.c | ||
22 | * @brief test for fragmentation.c | ||
23 | * @author Christian Grothoff | ||
24 | */ | ||
25 | #include "platform.h" | ||
26 | #include "gnunet_fragmentation_lib.h" | ||
27 | |||
28 | #define DETAILS GNUNET_NO | ||
29 | |||
30 | /** | ||
31 | * Number of messages to transmit (note: each uses ~32k memory!) | ||
32 | */ | ||
33 | #define NUM_MSGS 500 | ||
34 | |||
35 | /** | ||
36 | * MTU to force on fragmentation (must be > 1k + 12) | ||
37 | */ | ||
38 | #define MTU 1111 | ||
39 | |||
40 | /** | ||
41 | * Simulate dropping of 1 out of how many messages? (must be > 1) | ||
42 | */ | ||
43 | #define DROPRATE 5 | ||
44 | |||
45 | static int ret = 1; | ||
46 | |||
47 | static unsigned int dups; | ||
48 | |||
49 | static unsigned int fragc; | ||
50 | |||
51 | static unsigned int frag_drops; | ||
52 | |||
53 | static unsigned int acks; | ||
54 | |||
55 | static unsigned int ack_drops; | ||
56 | |||
57 | static struct GNUNET_DEFRAGMENT_Context *defrag; | ||
58 | |||
59 | static struct GNUNET_BANDWIDTH_Tracker trackers[NUM_MSGS]; | ||
60 | |||
61 | static struct GNUNET_FRAGMENT_Context *frags[NUM_MSGS]; | ||
62 | |||
63 | static struct GNUNET_SCHEDULER_Task * shutdown_task; | ||
64 | |||
65 | static void | ||
66 | do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
67 | { | ||
68 | unsigned int i; | ||
69 | |||
70 | ret = 0; | ||
71 | shutdown_task = NULL; | ||
72 | GNUNET_DEFRAGMENT_context_destroy (defrag); | ||
73 | defrag = NULL; | ||
74 | for (i = 0; i < NUM_MSGS; i++) | ||
75 | { | ||
76 | if (frags[i] == NULL) | ||
77 | continue; | ||
78 | GNUNET_FRAGMENT_context_destroy (frags[i], NULL, NULL); | ||
79 | frags[i] = NULL; | ||
80 | } | ||
81 | } | ||
82 | |||
83 | |||
84 | static void | ||
85 | proc_msgs (void *cls, const struct GNUNET_MessageHeader *hdr) | ||
86 | { | ||
87 | static unsigned int total; | ||
88 | unsigned int i; | ||
89 | const char *buf; | ||
90 | |||
91 | #if DETAILS | ||
92 | FPRINTF (stderr, "%s", "!"); /* message complete, good! */ | ||
93 | #endif | ||
94 | buf = (const char *) hdr; | ||
95 | for (i = sizeof (struct GNUNET_MessageHeader); i < ntohs (hdr->size); i++) | ||
96 | GNUNET_assert (buf[i] == (char) i); | ||
97 | total++; | ||
98 | #if ! DETAILS | ||
99 | if (0 == (total % (NUM_MSGS / 100))) | ||
100 | FPRINTF (stderr, "%s", "."); | ||
101 | #endif | ||
102 | /* tolerate 10% loss, i.e. due to duplicate fragment IDs */ | ||
103 | if ((total >= NUM_MSGS - (NUM_MSGS / 10)) && (ret != 0)) | ||
104 | { | ||
105 | if (NULL == shutdown_task) | ||
106 | shutdown_task = GNUNET_SCHEDULER_add_now (&do_shutdown, NULL); | ||
107 | } | ||
108 | } | ||
109 | |||
110 | |||
111 | /** | ||
112 | * Process ACK (by passing to fragmenter) | ||
113 | */ | ||
114 | static void | ||
115 | proc_acks (void *cls, uint32_t msg_id, const struct GNUNET_MessageHeader *hdr) | ||
116 | { | ||
117 | unsigned int i; | ||
118 | int ret; | ||
119 | |||
120 | if (0 == GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, DROPRATE)) | ||
121 | { | ||
122 | ack_drops++; | ||
123 | return; /* random drop */ | ||
124 | } | ||
125 | for (i = 0; i < NUM_MSGS; i++) | ||
126 | { | ||
127 | if (frags[i] == NULL) | ||
128 | continue; | ||
129 | ret = GNUNET_FRAGMENT_process_ack (frags[i], hdr); | ||
130 | if (ret == GNUNET_OK) | ||
131 | { | ||
132 | #if DETAILS | ||
133 | FPRINTF (stderr, "%s", "@"); /* good ACK */ | ||
134 | #endif | ||
135 | GNUNET_FRAGMENT_context_destroy (frags[i], NULL, NULL); | ||
136 | frags[i] = NULL; | ||
137 | acks++; | ||
138 | return; | ||
139 | } | ||
140 | if (ret == GNUNET_NO) | ||
141 | { | ||
142 | #if DETAILS | ||
143 | FPRINTF (stderr, "%s", "@"); /* good ACK */ | ||
144 | #endif | ||
145 | acks++; | ||
146 | return; | ||
147 | } | ||
148 | } | ||
149 | #if DETAILS | ||
150 | FPRINTF (stderr, "%s", "_"); /* BAD: ack that nobody feels responsible for... */ | ||
151 | #endif | ||
152 | } | ||
153 | |||
154 | |||
155 | /** | ||
156 | * Process fragment (by passing to defrag). | ||
157 | */ | ||
158 | static void | ||
159 | proc_frac (void *cls, const struct GNUNET_MessageHeader *hdr) | ||
160 | { | ||
161 | struct GNUNET_FRAGMENT_Context **fc = cls; | ||
162 | int ret; | ||
163 | |||
164 | GNUNET_FRAGMENT_context_transmission_done (*fc); | ||
165 | if (0 == GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, DROPRATE)) | ||
166 | { | ||
167 | frag_drops++; | ||
168 | return; /* random drop */ | ||
169 | } | ||
170 | if (NULL == defrag) | ||
171 | { | ||
172 | FPRINTF (stderr, "%s", "E"); /* Error: frag after shutdown!? */ | ||
173 | return; | ||
174 | } | ||
175 | ret = GNUNET_DEFRAGMENT_process_fragment (defrag, hdr); | ||
176 | if (ret == GNUNET_NO) | ||
177 | { | ||
178 | #if DETAILS | ||
179 | FPRINTF (stderr, "%s", "?"); /* duplicate fragment */ | ||
180 | #endif | ||
181 | dups++; | ||
182 | } | ||
183 | else if (ret == GNUNET_OK) | ||
184 | { | ||
185 | #if DETAILS | ||
186 | FPRINTF (stderr, "%s", "."); /* good fragment */ | ||
187 | #endif | ||
188 | fragc++; | ||
189 | } | ||
190 | } | ||
191 | |||
192 | |||
193 | /** | ||
194 | * Main function run with scheduler. | ||
195 | */ | ||
196 | static void | ||
197 | run (void *cls, char *const *args, const char *cfgfile, | ||
198 | const struct GNUNET_CONFIGURATION_Handle *cfg) | ||
199 | { | ||
200 | unsigned int i; | ||
201 | struct GNUNET_MessageHeader *msg; | ||
202 | char buf[MTU + 32 * 1024]; | ||
203 | |||
204 | defrag = GNUNET_DEFRAGMENT_context_create (NULL, MTU, NUM_MSGS /* enough space for all */ | ||
205 | , NULL, &proc_msgs, &proc_acks); | ||
206 | for (i = 0; i < sizeof (buf); i++) | ||
207 | buf[i] = (char) i; | ||
208 | msg = (struct GNUNET_MessageHeader *) buf; | ||
209 | for (i = 0; i < NUM_MSGS; i++) | ||
210 | { | ||
211 | msg->type = htons ((uint16_t) i); | ||
212 | msg->size = | ||
213 | htons (sizeof (struct GNUNET_MessageHeader) + (17 * i) % (32 * 1024)); | ||
214 | frags[i] = GNUNET_FRAGMENT_context_create (NULL /* no stats */ , | ||
215 | MTU, &trackers[i], | ||
216 | GNUNET_TIME_UNIT_MILLISECONDS, | ||
217 | GNUNET_TIME_UNIT_SECONDS, | ||
218 | msg, | ||
219 | &proc_frac, &frags[i]); | ||
220 | } | ||
221 | } | ||
222 | |||
223 | |||
224 | int | ||
225 | main (int argc, char *argv[]) | ||
226 | { | ||
227 | struct GNUNET_GETOPT_CommandLineOption options[] = { | ||
228 | GNUNET_GETOPT_OPTION_END | ||
229 | }; | ||
230 | char *const argv_prog[] = { | ||
231 | "test-fragmentation", | ||
232 | "-c", | ||
233 | "test_fragmentation_data.conf", | ||
234 | "-L", | ||
235 | "WARNING", | ||
236 | NULL | ||
237 | }; | ||
238 | unsigned int i; | ||
239 | |||
240 | GNUNET_log_setup ("test-fragmentation", | ||
241 | "WARNING", | ||
242 | NULL); | ||
243 | for (i = 0; i < NUM_MSGS; i++) | ||
244 | GNUNET_BANDWIDTH_tracker_init (&trackers[i], NULL, NULL, | ||
245 | GNUNET_BANDWIDTH_value_init ((i + 1) * 1024), | ||
246 | 100); | ||
247 | GNUNET_PROGRAM_run (5, argv_prog, "test-fragmentation", "nohelp", options, | ||
248 | &run, NULL); | ||
249 | FPRINTF (stderr, | ||
250 | "\nHad %u good fragments, %u duplicate fragments, %u acks and %u simulated drops of acks\n", | ||
251 | fragc, dups, acks, ack_drops); | ||
252 | return ret; | ||
253 | } | ||