gio/tests: add some more async stream tests

Add read_async() and skip_async() tests to buffered-input-stream.

Fix and re-enable filter-streams's existing close_async() tests, and
add read_async(), skip_async(), and write_async() tests as well. Also,
redo the tests to use dummy GFilterInputStream and GFilterOutputStream
subclasses rather than GBufferedInput/OutputStream, so that we're
testing the base filter stream implementations of everything (since
the buffered stream overrides are already getting tested in the
buffered-input-stream and buffered-output-stream tests anyway).

Add a skip_async() test to unix-streams. (This one would crash without
the bugfix in the previous commit.)
This commit is contained in:
Dan Winship 2013-01-16 10:09:10 -05:00
parent acfa6e2337
commit 578b657f95
3 changed files with 394 additions and 48 deletions

View File

@ -208,6 +208,98 @@ test_read (void)
g_object_unref (base); g_object_unref (base);
} }
static void
return_result_cb (GObject *object,
GAsyncResult *result,
gpointer user_data)
{
GAsyncResult **ret = user_data;
*ret = g_object_ref (result);
}
static void
test_read_async (void)
{
GInputStream *base;
GInputStream *in;
gchar buffer[20];
GError *error;
GAsyncResult *result;
base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ", -1, NULL);
in = g_buffered_input_stream_new_sized (base, 8);
g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 0);
error = NULL;
result = NULL;
g_buffered_input_stream_fill_async (G_BUFFERED_INPUT_STREAM (in), 8,
G_PRIORITY_DEFAULT, NULL,
return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_buffered_input_stream_fill_finish (G_BUFFERED_INPUT_STREAM (in), result, &error), ==, 8);
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 8);
memset (buffer, 0, 20);
g_input_stream_read_async (in, &buffer, 16, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_read_finish (in, result, &error), ==, 16);
g_assert_cmpstr (buffer, ==, "abcdefghijklmnop");
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 0);
memset (buffer, 0, 20);
g_input_stream_read_async (in, &buffer, 16, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_read_finish (in, result, &error), ==, 16);
g_assert_cmpstr (buffer, ==, "qrstuvwxyzABCDEF");
g_assert_no_error (error);
g_clear_object (&result);
memset (buffer, 0, 20);
g_input_stream_read_async (in, &buffer, 16, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_read_finish (in, result, &error), ==, 16);
g_assert_cmpstr (buffer, ==, "GHIJKLMNOPQRSTUV");
g_assert_no_error (error);
g_clear_object (&result);
memset (buffer, 0, 20);
g_input_stream_read_async (in, &buffer, 16, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_read_finish (in, result, &error), ==, 4);
g_assert_cmpstr (buffer, ==, "WXYZ");
g_assert_no_error (error);
g_clear_object (&result);
memset (buffer, 0, 20);
g_input_stream_read_async (in, &buffer, 16, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_read_finish (in, result, &error), ==, 0);
g_assert_no_error (error);
g_clear_object (&result);
g_object_unref (in);
g_object_unref (base);
}
static void static void
test_skip (void) test_skip (void)
{ {
@ -250,6 +342,76 @@ test_skip (void)
g_object_unref (base); g_object_unref (base);
} }
static void
test_skip_async (void)
{
GInputStream *base;
GInputStream *in;
GError *error;
GAsyncResult *result;
base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ", -1, NULL);
in = g_buffered_input_stream_new_sized (base, 5);
error = NULL;
g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'a');
g_assert_no_error (error);
g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'b');
g_assert_no_error (error);
g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'c');
g_assert_no_error (error);
result = NULL;
g_input_stream_skip_async (in, 7, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_skip_finish (in, result, &error), ==, 7);
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'k');
g_assert_no_error (error);
g_input_stream_skip_async (in, 10, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_skip_finish (in, result, &error), ==, 10);
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'v');
g_assert_no_error (error);
g_input_stream_skip_async (in, 20, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_skip_finish (in, result, &error), ==, 20);
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'Q');
g_assert_no_error (error);
g_input_stream_skip_async (in, 10, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_skip_finish (in, result, &error), ==, 8);
g_clear_object (&result);
g_assert_no_error (error);
g_input_stream_skip_async (in, 10, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
while (!result)
g_main_context_iteration (NULL, TRUE);
g_assert_cmpint (g_input_stream_skip_finish (in, result, &error), ==, 0);
g_clear_object (&result);
g_assert_no_error (error);
g_object_unref (in);
g_object_unref (base);
}
static void static void
test_close (void) test_close (void)
{ {
@ -381,7 +543,9 @@ main (int argc,
g_test_add_func ("/buffered-input-stream/set-buffer-size", test_set_buffer_size); g_test_add_func ("/buffered-input-stream/set-buffer-size", test_set_buffer_size);
g_test_add_func ("/buffered-input-stream/read-byte", test_read_byte); g_test_add_func ("/buffered-input-stream/read-byte", test_read_byte);
g_test_add_func ("/buffered-input-stream/read", test_read); g_test_add_func ("/buffered-input-stream/read", test_read);
g_test_add_func ("/buffered-input-stream/read-async", test_read_async);
g_test_add_func ("/buffered-input-stream/skip", test_skip); g_test_add_func ("/buffered-input-stream/skip", test_skip);
g_test_add_func ("/buffered-input-stream/skip-async", test_skip_async);
g_test_add_func ("/buffered-input-stream/seek", test_seek); g_test_add_func ("/buffered-input-stream/seek", test_seek);
g_test_add_func ("/filter-input-stream/close", test_close); g_test_add_func ("/filter-input-stream/close", test_close);

View File

@ -15,6 +15,51 @@
#include <glib/glib.h> #include <glib/glib.h>
#include <gio/gio.h> #include <gio/gio.h>
/* GFilterInputStream and GFilterOutputStream are abstract, so define
* minimal subclasses for testing. (This used to use
* GBufferedInputStream and GBufferedOutputStream, but those have
* their own test program, and they override some methods, meaning the
* core filter stream functionality wasn't getting fully tested.)
*/
GType test_filter_input_stream_get_type (void);
GType test_filter_output_stream_get_type (void);
#define TEST_TYPE_FILTER_INPUT_STREAM (test_filter_input_stream_get_type ())
#define TEST_FILTER_INPUT_STREAM(o) (G_TYPE_CHECK_INSTANCE_CAST ((o), TEST_TYPE_FILTER_INPUT_STREAM, TestFilterInputStream))
#define TEST_TYPE_FILTER_OUTPUT_STREAM (test_filter_output_stream_get_type ())
#define TEST_FILTER_OUTPUT_STREAM(o) (G_TYPE_CHECK_INSTANCE_CAST ((o), TEST_TYPE_FILTER_OUTPUT_STREAM, TestFilterOutputStream))
typedef GFilterInputStream TestFilterInputStream;
typedef GFilterInputStreamClass TestFilterInputStreamClass;
typedef GFilterOutputStream TestFilterOutputStream;
typedef GFilterOutputStreamClass TestFilterOutputStreamClass;
G_DEFINE_TYPE (TestFilterInputStream, test_filter_input_stream, G_TYPE_FILTER_INPUT_STREAM);
G_DEFINE_TYPE (TestFilterOutputStream, test_filter_output_stream, G_TYPE_FILTER_OUTPUT_STREAM);
static void
test_filter_input_stream_init (TestFilterInputStream *stream)
{
}
static void
test_filter_input_stream_class_init (TestFilterInputStreamClass *klass)
{
}
static void
test_filter_output_stream_init (TestFilterOutputStream *stream)
{
}
static void
test_filter_output_stream_class_init (TestFilterOutputStreamClass *klass)
{
}
/* Now the tests */
static void static void
test_input_filter (void) test_input_filter (void)
{ {
@ -25,10 +70,13 @@ test_input_filter (void)
g_test_bug ("568394"); g_test_bug ("568394");
base = g_memory_input_stream_new_from_data ("abcdefghijk", -1, NULL); base = g_memory_input_stream_new_from_data ("abcdefghijk", -1, NULL);
f1 = g_buffered_input_stream_new (base); f1 = g_object_new (TEST_TYPE_FILTER_INPUT_STREAM,
f2 = g_buffered_input_stream_new (base); "base-stream", base,
"close-base-stream", FALSE,
g_filter_input_stream_set_close_base_stream (G_FILTER_INPUT_STREAM (f1), FALSE); NULL);
f2 = g_object_new (TEST_TYPE_FILTER_INPUT_STREAM,
"base-stream", base,
NULL);
g_assert (g_filter_input_stream_get_base_stream (G_FILTER_INPUT_STREAM (f1)) == base); g_assert (g_filter_input_stream_get_base_stream (G_FILTER_INPUT_STREAM (f1)) == base);
g_assert (g_filter_input_stream_get_base_stream (G_FILTER_INPUT_STREAM (f2)) == base); g_assert (g_filter_input_stream_get_base_stream (G_FILTER_INPUT_STREAM (f2)) == base);
@ -71,10 +119,13 @@ test_output_filter (void)
GOutputStream *base, *f1, *f2; GOutputStream *base, *f1, *f2;
base = g_memory_output_stream_new (NULL, 0, g_realloc, g_free); base = g_memory_output_stream_new (NULL, 0, g_realloc, g_free);
f1 = g_buffered_output_stream_new (base); f1 = g_object_new (TEST_TYPE_FILTER_OUTPUT_STREAM,
f2 = g_buffered_output_stream_new (base); "base-stream", base,
"close-base-stream", FALSE,
g_filter_output_stream_set_close_base_stream (G_FILTER_OUTPUT_STREAM (f1), FALSE); NULL);
f2 = g_object_new (TEST_TYPE_FILTER_OUTPUT_STREAM,
"base-stream", base,
NULL);
g_assert (g_filter_output_stream_get_base_stream (G_FILTER_OUTPUT_STREAM (f1)) == base); g_assert (g_filter_output_stream_get_base_stream (G_FILTER_OUTPUT_STREAM (f1)) == base);
g_assert (g_filter_output_stream_get_base_stream (G_FILTER_OUTPUT_STREAM (f2)) == base); g_assert (g_filter_output_stream_get_base_stream (G_FILTER_OUTPUT_STREAM (f2)) == base);
@ -98,8 +149,19 @@ test_output_filter (void)
gpointer expected_obj; gpointer expected_obj;
gpointer expected_data; gpointer expected_data;
gboolean callback_happened; gboolean callback_happened;
GMainLoop *loop;
static void
return_result_cb (GObject *object,
GAsyncResult *result,
gpointer user_data)
{
GAsyncResult **ret = user_data;
*ret = g_object_ref (result);
g_main_loop_quit (loop);
}
#if 0
static void static void
in_cb (GObject *object, in_cb (GObject *object,
GAsyncResult *result, GAsyncResult *result,
@ -115,22 +177,64 @@ in_cb (GObject *object,
g_assert (error == NULL); g_assert (error == NULL);
callback_happened = TRUE; callback_happened = TRUE;
g_main_loop_quit (loop);
} }
static void static void
test_input_async (void) test_input_async (void)
{ {
GInputStream *base, *f1, *f2; GInputStream *base, *f1, *f2;
char buf[20];
GAsyncResult *result = NULL;
GError *error = NULL;
base = g_memory_input_stream_new_from_data ("abcdefghijk", -1, NULL); loop = g_main_loop_new (NULL, FALSE);
f1 = g_buffered_input_stream_new (base);
f2 = g_buffered_input_stream_new (base);
g_filter_input_stream_set_close_base_stream (G_FILTER_INPUT_STREAM (f1), FALSE); base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyz", -1, NULL);
f1 = g_object_new (TEST_TYPE_FILTER_INPUT_STREAM,
"base-stream", base,
"close-base-stream", FALSE,
NULL);
f2 = g_object_new (TEST_TYPE_FILTER_INPUT_STREAM,
"base-stream", base,
NULL);
g_assert (g_filter_input_stream_get_base_stream (G_FILTER_INPUT_STREAM (f1)) == base); g_assert (g_filter_input_stream_get_base_stream (G_FILTER_INPUT_STREAM (f1)) == base);
g_assert (g_filter_input_stream_get_base_stream (G_FILTER_INPUT_STREAM (f2)) == base); g_assert (g_filter_input_stream_get_base_stream (G_FILTER_INPUT_STREAM (f2)) == base);
memset (buf, 0, sizeof (buf));
g_input_stream_read_async (f1, buf, 10, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
g_main_loop_run (loop);
g_assert_cmpint (g_input_stream_read_finish (f1, result, &error), ==, 10);
g_assert_cmpstr (buf, ==, "abcdefghij");
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (base)), ==, 10);
g_input_stream_skip_async (f2, 10, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
g_main_loop_run (loop);
g_assert_cmpint (g_input_stream_skip_finish (f2, result, &error), ==, 10);
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (base)), ==, 20);
memset (buf, 0, sizeof (buf));
g_input_stream_read_async (f1, buf, 10, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
g_main_loop_run (loop);
g_assert_cmpint (g_input_stream_read_finish (f1, result, &error), ==, 6);
g_assert_cmpstr (buf, ==, "uvwxyz");
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (base)), ==, 26);
g_assert (!g_input_stream_is_closed (base)); g_assert (!g_input_stream_is_closed (base));
g_assert (!g_input_stream_is_closed (f1)); g_assert (!g_input_stream_is_closed (f1));
g_assert (!g_input_stream_is_closed (f2)); g_assert (!g_input_stream_is_closed (f2));
@ -141,8 +245,7 @@ test_input_async (void)
g_input_stream_close_async (f1, 0, NULL, in_cb, expected_data); g_input_stream_close_async (f1, 0, NULL, in_cb, expected_data);
g_assert (callback_happened == FALSE); g_assert (callback_happened == FALSE);
while (g_main_context_pending (NULL)) g_main_loop_run (loop);
g_main_context_iteration (NULL, FALSE);
g_assert (callback_happened == TRUE); g_assert (callback_happened == TRUE);
g_assert (!g_input_stream_is_closed (base)); g_assert (!g_input_stream_is_closed (base));
@ -158,8 +261,7 @@ test_input_async (void)
g_input_stream_close_async (f2, 0, NULL, in_cb, expected_data); g_input_stream_close_async (f2, 0, NULL, in_cb, expected_data);
g_assert (callback_happened == FALSE); g_assert (callback_happened == FALSE);
while (g_main_context_pending (NULL)) g_main_loop_run (loop);
g_main_context_iteration (NULL, FALSE);
g_assert (callback_happened == TRUE); g_assert (callback_happened == TRUE);
g_assert (g_input_stream_is_closed (base)); g_assert (g_input_stream_is_closed (base));
@ -169,6 +271,7 @@ test_input_async (void)
g_assert (g_input_stream_is_closed (base)); g_assert (g_input_stream_is_closed (base));
g_object_unref (base); g_object_unref (base);
g_main_loop_unref (loop);
} }
static void static void
@ -186,22 +289,55 @@ out_cb (GObject *object,
g_assert (error == NULL); g_assert (error == NULL);
callback_happened = TRUE; callback_happened = TRUE;
g_main_loop_quit (loop);
} }
static void static void
test_output_async (void) test_output_async (void)
{ {
GOutputStream *base, *f1, *f2; GOutputStream *base, *f1, *f2;
GAsyncResult *result = NULL;
GError *error = NULL;
loop = g_main_loop_new (NULL, FALSE);
base = g_memory_output_stream_new (NULL, 0, g_realloc, g_free); base = g_memory_output_stream_new (NULL, 0, g_realloc, g_free);
f1 = g_buffered_output_stream_new (base); f1 = g_object_new (TEST_TYPE_FILTER_OUTPUT_STREAM,
f2 = g_buffered_output_stream_new (base); "base-stream", base,
"close-base-stream", FALSE,
g_filter_output_stream_set_close_base_stream (G_FILTER_OUTPUT_STREAM (f1), FALSE); NULL);
f2 = g_object_new (TEST_TYPE_FILTER_OUTPUT_STREAM,
"base-stream", base,
NULL);
g_assert (g_filter_output_stream_get_base_stream (G_FILTER_OUTPUT_STREAM (f1)) == base); g_assert (g_filter_output_stream_get_base_stream (G_FILTER_OUTPUT_STREAM (f1)) == base);
g_assert (g_filter_output_stream_get_base_stream (G_FILTER_OUTPUT_STREAM (f2)) == base); g_assert (g_filter_output_stream_get_base_stream (G_FILTER_OUTPUT_STREAM (f2)) == base);
g_output_stream_write_async (f1, "abcdefghijklm", 13, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
g_main_loop_run (loop);
g_assert_cmpint (g_output_stream_write_finish (f1, result, &error), ==, 13);
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (base)), ==, 13);
g_output_stream_write_async (f2, "nopqrstuvwxyz", 13, G_PRIORITY_DEFAULT,
NULL, return_result_cb, &result);
g_main_loop_run (loop);
g_assert_cmpint (g_output_stream_write_finish (f2, result, &error), ==, 13);
g_assert_no_error (error);
g_clear_object (&result);
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (base)), ==, 26);
g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 26);
g_output_stream_write (base, "\0", 1, NULL, &error);
g_assert_no_error (error);
g_assert_cmpstr (g_memory_output_stream_get_data (G_MEMORY_OUTPUT_STREAM (base)), ==, "abcdefghijklmnopqrstuvwxyz");
g_assert (!g_output_stream_is_closed (base)); g_assert (!g_output_stream_is_closed (base));
g_assert (!g_output_stream_is_closed (f1)); g_assert (!g_output_stream_is_closed (f1));
g_assert (!g_output_stream_is_closed (f2)); g_assert (!g_output_stream_is_closed (f2));
@ -212,8 +348,7 @@ test_output_async (void)
g_output_stream_close_async (f1, 0, NULL, out_cb, expected_data); g_output_stream_close_async (f1, 0, NULL, out_cb, expected_data);
g_assert (callback_happened == FALSE); g_assert (callback_happened == FALSE);
while (g_main_context_pending (NULL)) g_main_loop_run (loop);
g_main_context_iteration (NULL, FALSE);
g_assert (callback_happened == TRUE); g_assert (callback_happened == TRUE);
g_assert (!g_output_stream_is_closed (base)); g_assert (!g_output_stream_is_closed (base));
@ -229,8 +364,7 @@ test_output_async (void)
g_output_stream_close_async (f2, 0, NULL, out_cb, expected_data); g_output_stream_close_async (f2, 0, NULL, out_cb, expected_data);
g_assert (callback_happened == FALSE); g_assert (callback_happened == FALSE);
while (g_main_context_pending (NULL)) g_main_loop_run (loop);
g_main_context_iteration (NULL, FALSE);
g_assert (callback_happened == TRUE); g_assert (callback_happened == TRUE);
g_assert (g_output_stream_is_closed (base)); g_assert (g_output_stream_is_closed (base));
@ -240,8 +374,8 @@ test_output_async (void)
g_assert (g_output_stream_is_closed (base)); g_assert (g_output_stream_is_closed (base));
g_object_unref (base); g_object_unref (base);
g_main_loop_unref (loop);
} }
#endif
int int
main (int argc, char **argv) main (int argc, char **argv)
@ -251,10 +385,8 @@ main (int argc, char **argv)
g_test_add_func ("/filter-stream/input", test_input_filter); g_test_add_func ("/filter-stream/input", test_input_filter);
g_test_add_func ("/filter-stream/output", test_output_filter); g_test_add_func ("/filter-stream/output", test_output_filter);
#if 0
g_test_add_func ("/filter-stream/async-input", test_input_async); g_test_add_func ("/filter-stream/async-input", test_input_async);
g_test_add_func ("/filter-stream/async-output", test_output_async); g_test_add_func ("/filter-stream/async-output", test_output_async);
#endif
return g_test_run(); return g_test_run();
} }

View File

@ -120,8 +120,9 @@ reader_thread (gpointer user_data)
char main_buf[sizeof (DATA)]; char main_buf[sizeof (DATA)];
gssize main_len, main_offset; gssize main_len, main_offset;
static void readable (GObject *source, GAsyncResult *res, gpointer user_data); static void main_thread_read (GObject *source, GAsyncResult *res, gpointer user_data);
static void writable (GObject *source, GAsyncResult *res, gpointer user_data); static void main_thread_skipped (GObject *source, GAsyncResult *res, gpointer user_data);
static void main_thread_wrote (GObject *source, GAsyncResult *res, gpointer user_data);
static void static void
do_main_cancel (GOutputStream *out) do_main_cancel (GOutputStream *out)
@ -131,13 +132,14 @@ do_main_cancel (GOutputStream *out)
} }
static void static void
readable (GObject *source, GAsyncResult *res, gpointer user_data) main_thread_skipped (GObject *source, GAsyncResult *res, gpointer user_data)
{ {
GInputStream *in = G_INPUT_STREAM (source); GInputStream *in = G_INPUT_STREAM (source);
GOutputStream *out = user_data; GOutputStream *out = user_data;
GError *err = NULL; GError *err = NULL;
gssize nskipped;
main_len = g_input_stream_read_finish (in, res, &err); nskipped = g_input_stream_skip_finish (in, res, &err);
if (g_cancellable_is_cancelled (main_cancel)) if (g_cancellable_is_cancelled (main_cancel))
{ {
@ -145,16 +147,62 @@ readable (GObject *source, GAsyncResult *res, gpointer user_data)
return; return;
} }
g_assert (err == NULL); g_assert_no_error (err);
main_offset = 0; main_offset += nskipped;
g_output_stream_write_async (out, main_buf, main_len, if (main_offset == main_len)
G_PRIORITY_DEFAULT, main_cancel, {
writable, in); main_offset = 0;
g_output_stream_write_async (out, main_buf, main_len,
G_PRIORITY_DEFAULT, main_cancel,
main_thread_wrote, in);
}
else
{
g_input_stream_skip_async (in, main_len - main_offset,
G_PRIORITY_DEFAULT, main_cancel,
main_thread_skipped, out);
}
} }
static void static void
writable (GObject *source, GAsyncResult *res, gpointer user_data) main_thread_read (GObject *source, GAsyncResult *res, gpointer user_data)
{
GInputStream *in = G_INPUT_STREAM (source);
GOutputStream *out = user_data;
GError *err = NULL;
gssize nread;
nread = g_input_stream_read_finish (in, res, &err);
if (g_cancellable_is_cancelled (main_cancel))
{
do_main_cancel (out);
return;
}
g_assert_no_error (err);
main_offset += nread;
if (main_offset == sizeof (DATA))
{
main_len = main_offset;
main_offset = 0;
/* Now skip the same amount */
g_input_stream_skip_async (in, main_len,
G_PRIORITY_DEFAULT, main_cancel,
main_thread_skipped, out);
}
else
{
g_input_stream_read_async (in, main_buf, sizeof (main_buf),
G_PRIORITY_DEFAULT, main_cancel,
main_thread_read, out);
}
}
static void
main_thread_wrote (GObject *source, GAsyncResult *res, gpointer user_data)
{ {
GOutputStream *out = G_OUTPUT_STREAM (source); GOutputStream *out = G_OUTPUT_STREAM (source);
GInputStream *in = user_data; GInputStream *in = user_data;
@ -169,22 +217,23 @@ writable (GObject *source, GAsyncResult *res, gpointer user_data)
return; return;
} }
g_assert (err == NULL); g_assert_no_error (err);
g_assert_cmpint (nwrote, <=, main_len - main_offset); g_assert_cmpint (nwrote, <=, main_len - main_offset);
main_offset += nwrote; main_offset += nwrote;
if (main_offset == main_len) if (main_offset == main_len)
{ {
main_offset = 0;
g_input_stream_read_async (in, main_buf, sizeof (main_buf), g_input_stream_read_async (in, main_buf, sizeof (main_buf),
G_PRIORITY_DEFAULT, main_cancel, G_PRIORITY_DEFAULT, main_cancel,
readable, out); main_thread_read, out);
} }
else else
{ {
g_output_stream_write_async (out, main_buf + main_offset, g_output_stream_write_async (out, main_buf + main_offset,
main_len - main_offset, main_len - main_offset,
G_PRIORITY_DEFAULT, main_cancel, G_PRIORITY_DEFAULT, main_cancel,
writable, in); main_thread_wrote, in);
} }
} }
@ -204,12 +253,13 @@ test_pipe_io (gconstpointer nonblocking)
/* Split off two (additional) threads, a reader and a writer. From /* Split off two (additional) threads, a reader and a writer. From
* the writer thread, write data synchronously in small chunks, * the writer thread, write data synchronously in small chunks,
* which gets read asynchronously by the main thread and then * which gets alternately read and skipped asynchronously by the
* written asynchronously to the reader thread, which reads it * main thread and then (if not skipped) written asynchronously to
* synchronously. Eventually a timeout in the main thread will cause * the reader thread, which reads it synchronously. Eventually a
* it to cancel the writer thread, which will in turn cancel the * timeout in the main thread will cause it to cancel the writer
* read op in the main thread, which will then close the pipe to * thread, which will in turn cancel the read op in the main thread,
* the reader thread, causing the read op to fail. * which will then close the pipe to the reader thread, causing the
* read op to fail.
*/ */
g_assert (pipe (writer_pipe) == 0 && pipe (reader_pipe) == 0); g_assert (pipe (writer_pipe) == 0 && pipe (reader_pipe) == 0);
@ -240,7 +290,7 @@ test_pipe_io (gconstpointer nonblocking)
g_input_stream_read_async (in, main_buf, sizeof (main_buf), g_input_stream_read_async (in, main_buf, sizeof (main_buf),
G_PRIORITY_DEFAULT, main_cancel, G_PRIORITY_DEFAULT, main_cancel,
readable, out); main_thread_read, out);
g_timeout_add (500, timeout, writer_cancel); g_timeout_add (500, timeout, writer_cancel);