X-Git-Url: http://review.tizen.org/git/?a=blobdiff_plain;f=gio%2Ftests%2Fwin32-streams.c;h=b9d95f9aea6e553698faff250424837d90f13212;hb=b5ba22f163f884f14724b54d001bd044308f9f63;hp=fd4b27b920a69c16a53fd8ea9965337cfb5da4b3;hpb=52389bd01067284565b007bc156643d534282842;p=platform%2Fupstream%2Fglib.git

diff --git a/gio/tests/win32-streams.c b/gio/tests/win32-streams.c
index fd4b27b..b9d95f9 100644
--- a/gio/tests/win32-streams.c
+++ b/gio/tests/win32-streams.c
@@ -278,14 +278,258 @@ test_pipe_io (void)
   g_object_unref (out);
 }
 
+typedef struct _PipeIOOverlapReader
+{
+  char buf[sizeof (DATA)];
+  GInputStream *in;
+  GThread *thread;
+  GCancellable *cancellable;
+  gboolean success;
+} PipeIOOverlapReader;
+
+#define TEST_PIPE_IO_OVERLAP (1024 * 4)
+
+static gpointer
+pipe_io_overlap_reader_thread (gpointer user_data)
+{
+  PipeIOOverlapReader *p = user_data;
+  GError *err = NULL;
+  gsize read;
+  guint i;
+
+  for (i = 0; i < TEST_PIPE_IO_OVERLAP; ++i) {
+    memset (p->buf, 0, sizeof (p->buf));
+    g_input_stream_read_all (p->in, p->buf, sizeof (p->buf),
+                             &read, NULL, &err);
+
+    g_assert_cmpuint (read, ==, sizeof (p->buf));
+    g_assert_no_error (err);
+    g_assert_cmpstr (p->buf, ==, DATA);
+  }
+
+  return NULL;
+}
+
+static gpointer
+pipe_io_overlap_writer_thread (gpointer user_data)
+{
+  GOutputStream *out = user_data;
+  GError *err = NULL;
+  gsize bytes_written;
+  guint i;
+
+  for (i = 0; i < TEST_PIPE_IO_OVERLAP; ++i) {
+    g_output_stream_write_all (out, DATA, sizeof (DATA),
+                               &bytes_written, NULL, &err);
+
+    g_assert_cmpuint (bytes_written, ==, sizeof (DATA));
+    g_assert_no_error (err);
+  }
+
+  return NULL;
+}
+
+static void
+test_pipe_io_overlap (void)
+{
+  GOutputStream *out_server, *out_client;
+  GThread *writer_server, *writer_client;
+  PipeIOOverlapReader rs, rc;
+  HANDLE server, client;
+  gchar name[256];
+
+  g_snprintf (name, sizeof (name),
+              "\\\\.\\pipe\\gtest-io-overlap-%u", (guint) getpid ());
+
+  server = CreateNamedPipe (name,
+                            PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED,
+                            PIPE_READMODE_BYTE | PIPE_WAIT,
+                            1, 0, 0, 0, NULL);
+  g_assert (server != INVALID_HANDLE_VALUE);
+
+  client = CreateFile (name, GENERIC_WRITE | GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
+  g_assert (client != INVALID_HANDLE_VALUE);
+
+  out_server = g_win32_output_stream_new (server, TRUE);
+  writer_server = g_thread_new ("writer_server", pipe_io_overlap_writer_thread, out_server);
+  rs.in = g_win32_input_stream_new (server, TRUE);
+  rs.thread = g_thread_new ("reader_server", pipe_io_overlap_reader_thread, &rs);
+
+  out_client = g_win32_output_stream_new (client, TRUE);
+  writer_client = g_thread_new ("writer_client", pipe_io_overlap_writer_thread, out_client);
+  rc.in = g_win32_input_stream_new (client, TRUE);
+  rc.thread = g_thread_new ("reader_client", pipe_io_overlap_reader_thread, &rc);
+
+  g_thread_join (writer_client);
+  g_thread_join (writer_server);
+  g_thread_join (rc.thread);
+  g_thread_join (rs.thread);
+
+  g_object_unref (rs.in);
+  g_object_unref (rc.in);
+  g_object_unref (out_server);
+  g_object_unref (out_client);
+}
+
+static gpointer
+pipe_io_concurrent_writer_thread (gpointer user_data)
+{
+  GOutputStream *out = user_data;
+  GError *err = NULL;
+  gsize bytes_written;
+
+  g_output_stream_write_all (out, DATA, 1, &bytes_written, NULL, &err);
+
+  g_assert_cmpuint (bytes_written, ==, 1);
+  g_assert_no_error (err);
+
+  return NULL;
+}
+
+static gpointer
+pipe_io_concurrent_reader_thread (gpointer user_data)
+{
+  PipeIOOverlapReader *p = user_data;
+  GError *err = NULL;
+  gsize read;
+
+  memset (p->buf, 0, sizeof (p->buf));
+  p->success = g_input_stream_read_all (p->in, p->buf, 1, &read, p->cancellable, &err);
+
+  /* only one thread will succeed, the other will be cancelled */
+  if (p->success)
+    {
+      /* continue the main thread */
+      write (writer_pipe[1], "", 1);
+      g_assert_cmpuint (read, ==, 1);
+      g_assert_no_error (err);
+    }
+
+  return NULL;
+}
+
+static void
+test_pipe_io_concurrent (void)
+{
+  GOutputStream *out_server;
+  GThread *writer_server;
+  PipeIOOverlapReader rc1, rc2;
+  HANDLE server, client;
+  gchar name[256], c;
+
+  g_snprintf (name, sizeof (name),
+              "\\\\.\\pipe\\gtest-io-concurrent-%u", (guint) getpid ());
+
+  server = CreateNamedPipe (name,
+                            PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED,
+                            PIPE_READMODE_BYTE | PIPE_WAIT,
+                            1, 0, 0, 0, NULL);
+  g_assert (server != INVALID_HANDLE_VALUE);
+  g_assert (_pipe (writer_pipe, 10, _O_BINARY) == 0);
+
+  client = CreateFile (name, GENERIC_WRITE | GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL);
+  g_assert (client != INVALID_HANDLE_VALUE);
+
+  rc1.in = g_win32_input_stream_new (client, TRUE);
+  rc1.success = FALSE;
+  rc1.cancellable = g_cancellable_new ();
+  rc1.thread = g_thread_new ("reader_client", pipe_io_concurrent_reader_thread, &rc1);
+
+  rc2.in = g_win32_input_stream_new (client, TRUE);
+  rc2.success = FALSE;
+  rc2.cancellable = g_cancellable_new ();
+  rc2.thread = g_thread_new ("reader_client", pipe_io_concurrent_reader_thread, &rc2);
+
+  /* FIXME: how to synchronize on both reader thread waiting in read,
+     before starting the writer thread? */
+  g_usleep (G_USEC_PER_SEC / 10);
+
+  out_server = g_win32_output_stream_new (server, TRUE);
+  writer_server = g_thread_new ("writer_server", pipe_io_concurrent_writer_thread, out_server);
+
+  read (writer_pipe[0], &c, 1);
+
+  g_assert (rc1.success ^ rc2.success);
+
+  g_cancellable_cancel (rc1.cancellable);
+  g_cancellable_cancel (rc2.cancellable);
+
+  g_thread_join (writer_server);
+  g_thread_join (rc1.thread);
+  g_thread_join (rc2.thread);
+
+  g_object_unref (rc1.in);
+  g_object_unref (rc2.in);
+  g_object_unref (out_server);
+
+  close (writer_pipe[0]);
+  close (writer_pipe[1]);
+}
+
+static void
+readable_cancel (GObject *source, GAsyncResult *res, gpointer user_data)
+{
+  GInputStream *in = G_INPUT_STREAM (source);
+  GError *err = NULL;
+  gssize len;
+
+  len = g_input_stream_read_finish (in, res, &err);
+  g_assert_cmpint (len, ==, -1);
+  g_assert_error (err, G_IO_ERROR, G_IO_ERROR_CANCELLED);
+  g_error_free (err);
+
+  g_main_loop_quit (loop);
+}
+
+static void
+test_pipe_io_cancel (void)
+{
+  GInputStream *in;
+  GOutputStream *out;
+  HANDLE in_handle, out_handle;
+  gchar name[256];
+
+  g_snprintf (name, sizeof (name),
+              "\\\\.\\pipe\\gtest-io-cancel-%u", (guint) getpid ());
+
+  in_handle = CreateNamedPipe (name,
+                               PIPE_ACCESS_INBOUND | FILE_FLAG_OVERLAPPED,
+                               PIPE_READMODE_BYTE | PIPE_WAIT,
+                               1, 0, 0, 0, NULL);
+  g_assert (in_handle != INVALID_HANDLE_VALUE);
+
+  out_handle = CreateFile (name, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
+  g_assert (out_handle != INVALID_HANDLE_VALUE);
+
+  in = g_win32_input_stream_new (in_handle, TRUE);
+  out = g_win32_output_stream_new (out_handle, TRUE);
+
+  reader_cancel = g_cancellable_new ();
+  g_input_stream_read_async (in, main_buf, sizeof (main_buf),
+                             G_PRIORITY_DEFAULT, reader_cancel,
+                             readable_cancel, out);
+
+  g_timeout_add (500, timeout, reader_cancel);
+
+  loop = g_main_loop_new (NULL, TRUE);
+  g_main_loop_run (loop);
+  g_main_loop_unref (loop);
+
+  g_object_unref (reader_cancel);
+  g_object_unref (in);
+  g_object_unref (out);
+}
+
 int
 main (int   argc,
       char *argv[])
 {
-  g_type_init ();
   g_test_init (&argc, &argv, NULL);
 
   g_test_add_func ("/win32-streams/pipe-io-test", test_pipe_io);
+  g_test_add_func ("/win32-streams/pipe-io-cancel-test", test_pipe_io_cancel);
+  g_test_add_func ("/win32-streams/pipe-io-overlap-test", test_pipe_io_overlap);
+  g_test_add_func ("/win32-streams/pipe-io-concurrent-test", test_pipe_io_concurrent);
 
   return g_test_run();
 }