1 /* GLib testing framework examples and tests
2 * Copyright (C) 2008 Red Hat, Inc
4 * This work is provided "as is"; redistribution and modification
5 * in whole or in part, in any medium, physical or electronic is
6 * permitted without restriction.
8 * This work is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
12 * In no event shall the authors or contributors be liable for any
13 * direct, indirect, incidental, special, exemplary, or consequential
14 * damages (including, but not limited to, procurement of substitute
15 * goods or services; loss of use, data, or profits; or business
16 * interruption) however caused and on any theory of liability, whether
17 * in contract, strict liability, or tort (including negligence or
18 * otherwise) arising in any way out of the use of this software, even
19 * if advised of the possibility of such damage.
23 #include <gio/gunixinputstream.h>
24 #include <gio/gunixoutputstream.h>
25 #include <glib/glib-unix.h>
31 #define DATA "abcdefghijklmnopqrstuvwxyz"
33 int writer_pipe[2], reader_pipe[2];
34 GCancellable *writer_cancel, *reader_cancel, *main_cancel;
39 writer_thread (gpointer user_data)
42 gssize nwrote, offset;
45 out = g_unix_output_stream_new (writer_pipe[1], TRUE);
52 while (offset < (gssize) sizeof (DATA))
54 nwrote = g_output_stream_write (out, DATA + offset,
55 sizeof (DATA) - offset,
57 if (nwrote <= 0 || err != NULL)
62 g_assert (nwrote > 0 || err != NULL);
66 if (g_cancellable_is_cancelled (writer_cancel))
68 g_cancellable_cancel (main_cancel);
73 g_warning ("writer: %s", err->message);
74 g_assert_not_reached ();
78 reader_thread (gpointer user_data)
81 gssize nread = 0, total;
83 char buf[sizeof (DATA)];
85 in = g_unix_input_stream_new (reader_pipe[0], TRUE);
90 while (total < (gssize) sizeof (DATA))
92 nread = g_input_stream_read (in, buf + total, sizeof (buf) - total,
94 if (nread <= 0 || err != NULL)
104 g_assert (err == NULL);
110 g_assert_cmpstr (buf, ==, DATA);
111 g_assert (!g_cancellable_is_cancelled (reader_cancel));
115 g_warning ("reader: %s", err->message);
116 g_assert_not_reached ();
119 char main_buf[sizeof (DATA)];
120 gssize main_len, main_offset;
122 static void readable (GObject *source, GAsyncResult *res, gpointer user_data);
123 static void writable (GObject *source, GAsyncResult *res, gpointer user_data);
126 do_main_cancel (GOutputStream *out)
128 g_output_stream_close (out, NULL, NULL);
129 g_main_loop_quit (loop);
133 readable (GObject *source, GAsyncResult *res, gpointer user_data)
135 GInputStream *in = G_INPUT_STREAM (source);
136 GOutputStream *out = user_data;
139 main_len = g_input_stream_read_finish (in, res, &err);
141 if (g_cancellable_is_cancelled (main_cancel))
143 do_main_cancel (out);
147 g_assert (err == NULL);
150 g_output_stream_write_async (out, main_buf, main_len,
151 G_PRIORITY_DEFAULT, main_cancel,
156 writable (GObject *source, GAsyncResult *res, gpointer user_data)
158 GOutputStream *out = G_OUTPUT_STREAM (source);
159 GInputStream *in = user_data;
163 nwrote = g_output_stream_write_finish (out, res, &err);
165 if (g_cancellable_is_cancelled (main_cancel))
167 do_main_cancel (out);
171 g_assert (err == NULL);
172 g_assert_cmpint (nwrote, <=, main_len - main_offset);
174 main_offset += nwrote;
175 if (main_offset == main_len)
177 g_input_stream_read_async (in, main_buf, sizeof (main_buf),
178 G_PRIORITY_DEFAULT, main_cancel,
183 g_output_stream_write_async (out, main_buf + main_offset,
184 main_len - main_offset,
185 G_PRIORITY_DEFAULT, main_cancel,
191 timeout (gpointer cancellable)
193 g_cancellable_cancel (cancellable);
198 test_pipe_io (gconstpointer nonblocking)
200 GThread *writer, *reader;
204 /* Split off two (additional) threads, a reader and a writer. From
205 * the writer thread, write data synchronously in small chunks,
206 * which gets read asynchronously by the main thread and then
207 * written asynchronously to the reader thread, which reads it
208 * synchronously. Eventually a timeout in the main thread will cause
209 * it to cancel the writer thread, which will in turn cancel the
210 * read op in the main thread, which will then close the pipe to
211 * the reader thread, causing the read op to fail.
214 g_assert (pipe (writer_pipe) == 0 && pipe (reader_pipe) == 0);
218 GError *error = NULL;
220 g_unix_set_fd_nonblocking (writer_pipe[0], TRUE, &error);
221 g_assert_no_error (error);
222 g_unix_set_fd_nonblocking (writer_pipe[1], TRUE, &error);
223 g_assert_no_error (error);
224 g_unix_set_fd_nonblocking (reader_pipe[0], TRUE, &error);
225 g_assert_no_error (error);
226 g_unix_set_fd_nonblocking (reader_pipe[1], TRUE, &error);
227 g_assert_no_error (error);
230 writer_cancel = g_cancellable_new ();
231 reader_cancel = g_cancellable_new ();
232 main_cancel = g_cancellable_new ();
234 writer = g_thread_new ("writer", writer_thread, NULL);
235 reader = g_thread_new ("reader", reader_thread, NULL);
237 in = g_unix_input_stream_new (writer_pipe[0], TRUE);
238 out = g_unix_output_stream_new (reader_pipe[1], TRUE);
240 g_input_stream_read_async (in, main_buf, sizeof (main_buf),
241 G_PRIORITY_DEFAULT, main_cancel,
244 g_timeout_add (500, timeout, writer_cancel);
246 loop = g_main_loop_new (NULL, TRUE);
247 g_main_loop_run (loop);
248 g_main_loop_unref (loop);
250 g_thread_join (reader);
251 g_thread_join (writer);
253 g_object_unref (main_cancel);
254 g_object_unref (reader_cancel);
255 g_object_unref (writer_cancel);
257 g_object_unref (out);
263 GUnixInputStream *is;
264 GUnixOutputStream *os;
268 is = G_UNIX_INPUT_STREAM (g_unix_input_stream_new (0, TRUE));
271 "close-fd", &close_fd,
273 g_assert_cmpint (fd, ==, 0);
276 g_unix_input_stream_set_close_fd (is, FALSE);
277 g_assert (!g_unix_input_stream_get_close_fd (is));
278 g_assert_cmpint (g_unix_input_stream_get_fd (is), ==, 0);
282 os = G_UNIX_OUTPUT_STREAM (g_unix_output_stream_new (1, TRUE));
285 "close-fd", &close_fd,
287 g_assert_cmpint (fd, ==, 1);
290 g_unix_output_stream_set_close_fd (os, FALSE);
291 g_assert (!g_unix_output_stream_get_close_fd (os));
292 g_assert_cmpint (g_unix_output_stream_get_fd (os), ==, 1);
302 g_test_init (&argc, &argv, NULL);
304 g_test_add_func ("/unix-streams/basic", test_basic);
305 g_test_add_data_func ("/unix-streams/pipe-io-test",
306 GINT_TO_POINTER (FALSE),
308 g_test_add_data_func ("/unix-streams/nonblocking-io-test",
309 GINT_TO_POINTER (TRUE),