11 #define CRAWLER_TIMEOUT_RANGE 40
12 #define RECURSER_TIMEOUT 50
14 G_LOCK_DEFINE_STATIC (context_array_lock);
15 GPtrArray *context_array;
18 G_LOCK_DEFINE_STATIC (crawler_array_lock);
19 GPtrArray *crawler_array;
21 typedef struct _AddrData AddrData;
22 typedef struct _TestData TestData;
38 static void cleanup_crawlers (GMainContext *context);
41 read_all (GIOChannel *channel, char *buf, int len)
47 while (bytes_read < len)
49 err = g_io_channel_read (channel, buf + bytes_read, len - bytes_read, &count);
52 if (err != G_IO_ERROR_AGAIN)
65 write_all (GIOChannel *channel, char *buf, int len)
67 int bytes_written = 0;
71 while (bytes_written < len)
73 err = g_io_channel_write (channel, buf + bytes_written, len - bytes_written, &count);
74 if (err && err != G_IO_ERROR_AGAIN)
77 bytes_written += count;
84 adder_callback (GIOChannel *source,
85 GIOCondition condition,
93 AddrData *addr_data = data;
95 if (!read_all (source, buf1, 32) ||
96 !read_all (source, buf2, 32))
98 g_main_loop_quit (addr_data->loop);
102 sprintf (result, "%d", atoi(buf1) + atoi(buf2));
103 write_all (addr_data->dest, result, 32);
109 timeout_callback (gpointer data)
111 AddrData *addr_data = data;
119 adder_thread (gpointer data)
121 GMainContext *context;
122 GSource *adder_source;
123 GSource *timeout_source;
125 GIOChannel **channels = data;
128 context = g_main_context_get (g_thread_self());
130 G_LOCK (context_array_lock);
131 g_ptr_array_add (context_array, context);
132 G_UNLOCK (context_array_lock);
134 addr_data.dest = channels[1];
135 addr_data.loop = g_main_loop_new (context, FALSE);
138 adder_source = g_io_create_watch (channels[0], G_IO_IN | G_IO_HUP);
139 g_source_set_callback (adder_source, (GSourceFunc)adder_callback, &addr_data, NULL);
140 g_source_attach (adder_source, context);
141 g_source_unref (adder_source);
143 timeout_source = g_timeout_source_new (10);
144 g_source_set_callback (timeout_source, (GSourceFunc)timeout_callback, &addr_data, NULL);
145 g_source_set_priority (timeout_source, G_PRIORITY_HIGH);
146 g_source_attach (timeout_source, context);
147 g_source_unref (timeout_source);
149 g_main_run (addr_data.loop);
151 g_io_channel_close (channels[0]);
152 g_io_channel_close (channels[1]);
153 g_io_channel_unref (channels[0]);
154 g_io_channel_unref (channels[1]);
158 g_main_loop_destroy (addr_data.loop);
160 g_print ("Timeout run %d times\n", addr_data.count);
162 G_LOCK (context_array_lock);
163 g_ptr_array_remove (context_array, context);
164 if (context_array->len == 0)
165 g_main_loop_quit (main_loop);
166 G_UNLOCK (context_array_lock);
168 cleanup_crawlers (context);
172 io_pipe (GIOChannel **channels)
178 g_warning ("Cannot create pipe %s\n", g_strerror (errno));
182 channels[0] = g_io_channel_unix_new (fds[0]);
183 channels[1] = g_io_channel_unix_new (fds[1]);
187 do_add (GIOChannel *in, gint a, gint b)
192 sprintf (buf1, "%d", a);
193 sprintf (buf2, "%d", b);
195 write_all (in, buf1, 32);
196 write_all (in, buf2, 32);
200 adder_response (GIOChannel *source,
201 GIOCondition condition,
205 TestData *test_data = data;
207 if (!read_all (source, result, 32))
210 test_data->current_val = atoi (result);
213 if (test_data->iters == 0)
215 if (test_data->current_val != ITERS * INCREMENT)
217 g_print ("Addition failed: %d != %d\n",
218 test_data->current_val, ITERS * INCREMENT);
222 g_io_channel_close (source);
223 g_io_channel_close (test_data->in);
225 g_io_channel_unref (source);
226 g_io_channel_unref (test_data->in);
231 do_add (test_data->in, test_data->current_val, INCREMENT);
237 create_adder_thread (void)
242 GIOChannel *in_channels[2];
243 GIOChannel *out_channels[2];
245 GIOChannel **sub_channels;
247 sub_channels = g_new (GIOChannel *, 2);
249 io_pipe (in_channels);
250 io_pipe (out_channels);
252 sub_channels[0] = in_channels[0];
253 sub_channels[1] = out_channels[1];
255 g_thread_create (adder_thread, sub_channels, 0,
256 FALSE, TRUE, G_THREAD_PRIORITY_NORMAL, &err);
260 g_warning ("Cannot create thread: %s", err->message);
264 test_data = g_new (TestData, 1);
265 test_data->in = in_channels[1];
266 test_data->current_val = 0;
267 test_data->iters = ITERS;
269 g_io_add_watch (out_channels[0], G_IO_IN | G_IO_HUP,
270 adder_response, test_data);
272 do_add (test_data->in, test_data->current_val, INCREMENT);
275 static void create_crawler (void);
278 remove_crawler (void)
280 GSource *other_source;
282 if (crawler_array->len > 0)
284 other_source = crawler_array->pdata[g_random_int_range (0, crawler_array->len)];
285 g_source_destroy (other_source);
286 g_assert (g_ptr_array_remove_fast (crawler_array, other_source));
291 crawler_callback (gpointer data)
293 GSource *source = data;
295 G_LOCK (crawler_array_lock);
297 if (!g_ptr_array_remove_fast (crawler_array, source))
301 G_UNLOCK (crawler_array_lock);
310 create_crawler (void)
312 GSource *source = g_timeout_source_new (g_random_int_range (0, CRAWLER_TIMEOUT_RANGE));
313 g_source_set_callback (source, (GSourceFunc)crawler_callback, source, NULL);
315 G_LOCK (context_array_lock);
316 g_source_attach (source, context_array->pdata[g_random_int_range (0, context_array->len)]);
317 g_source_unref (source);
318 G_UNLOCK (context_array_lock);
320 G_LOCK (crawler_array_lock);
321 g_ptr_array_add (crawler_array, source);
322 G_UNLOCK (crawler_array_lock);
326 cleanup_crawlers (GMainContext *context)
330 G_LOCK (crawler_array_lock);
331 for (i=0; i < crawler_array->len; i++)
333 if (g_source_get_context (crawler_array->pdata[i]) == context)
335 g_source_destroy (g_ptr_array_remove_index (crawler_array, i));
339 G_UNLOCK (crawler_array_lock);
343 recurser_idle (gpointer data)
345 GMainContext *context = data;
348 for (i = 0; i < 10; i++)
349 g_main_context_iteration (context, TRUE);
355 recurser_start (gpointer data)
357 GMainContext *context;
360 G_LOCK (context_array_lock);
361 context = context_array->pdata[g_random_int_range (0, context_array->len)];
362 source = g_idle_source_new ();
363 g_source_set_callback (source, recurser_idle, context, NULL);
364 g_source_attach (source, context);
365 g_source_unref (source);
366 G_UNLOCK (context_array_lock);
375 /* Only run the test, if threads are enabled and a default thread
376 implementation is available */
377 #if defined(G_THREADS_ENABLED) && ! defined(G_THREADS_IMPL_NONE)
380 g_thread_init (NULL);
382 context_array = g_ptr_array_new ();
383 crawler_array = g_ptr_array_new ();
385 main_loop = g_main_loop_new (NULL, FALSE);
387 for (i = 0; i < NTHREADS; i++)
388 create_adder_thread ();
390 for (i = 0; i < NCRAWLERS; i++)
393 g_timeout_add (RECURSER_TIMEOUT, recurser_start, NULL);
395 g_main_loop_run (main_loop);
396 g_main_loop_destroy (main_loop);