[glib: 12/15] Add test for async read/write(v)/close on GSocketConnections

[Philip Withnall <pwithnall src gnome org>]

commit 61520ae601c1dd7b8c6924e0c1501fea593b749a
Author: Sebastian Dröge <sebastian centricular com>
Date:   Fri Jan 4 17:55:43 2019 +0200

    Add test for async read/write(v)/close on GSocketConnections

 gio/tests/socket-service.c | 312 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 312 insertions(+)
---
diff --git a/gio/tests/socket-service.c b/gio/tests/socket-service.c
index 9ae76d082..14cf24e55 100644
--- a/gio/tests/socket-service.c
+++ b/gio/tests/socket-service.c
@@ -232,6 +232,316 @@ test_threaded_712570 (void)
   g_mutex_unlock (&mutex_712570);
 }
 
+static void
+closed_read_write_async_cb (GSocketConnection *conn,
+                            GAsyncResult      *result,
+                            gpointer           user_data)
+{
+  GError *error = NULL;
+  gboolean res;
+
+  res = g_io_stream_close_finish (G_IO_STREAM (conn), result, &error);
+  g_assert_no_error (error);
+  g_assert_true (res);
+}
+
+typedef struct {
+  GSocketConnection *conn;
+  guint8 *data;
+} WriteAsyncData;
+
+static void
+written_read_write_async_cb (GOutputStream *ostream,
+                             GAsyncResult  *result,
+                             gpointer       user_data)
+{
+  WriteAsyncData *data = user_data;
+  GError *error = NULL;
+  gboolean res;
+  gsize bytes_written;
+  GSocketConnection *conn;
+
+  conn = data->conn;
+
+  g_free (data->data);
+  g_free (data);
+
+  res = g_output_stream_write_all_finish (ostream, result, &bytes_written, &error);
+  g_assert_no_error (error);
+  g_assert_true (res);
+  g_assert_cmpint (bytes_written, ==, 20);
+
+  g_io_stream_close_async (G_IO_STREAM (conn),
+                           G_PRIORITY_DEFAULT,
+                           NULL,
+                           (GAsyncReadyCallback) closed_read_write_async_cb,
+                           NULL);
+  g_object_unref (conn);
+}
+
+static void
+connected_read_write_async_cb (GObject      *client,
+                               GAsyncResult *result,
+                               gpointer      user_data)
+{
+  GSocketConnection *conn;
+  GOutputStream *ostream;
+  GError *error = NULL;
+  WriteAsyncData *data;
+  gsize i;
+  GSocketConnection **sconn = user_data;
+
+  conn = g_socket_client_connect_finish (G_SOCKET_CLIENT (client), result, &error);
+  g_assert_no_error (error);
+  g_assert_nonnull (conn);
+
+  ostream = g_io_stream_get_output_stream (G_IO_STREAM (conn));
+
+  data = g_new0 (WriteAsyncData, 1);
+  data->conn = conn;
+  data->data = g_new0 (guint8, 20);
+  for (i = 0; i < 20; i++)
+    data->data[i] = i;
+
+  g_output_stream_write_all_async (ostream,
+                                   data->data,
+                                   20,
+                                   G_PRIORITY_DEFAULT,
+                                   NULL,
+                                   (GAsyncReadyCallback) written_read_write_async_cb,
+                                   data /* stolen */);
+
+  *sconn = g_object_ref (conn);
+}
+
+typedef struct {
+  GSocketConnection *conn;
+  GOutputVector *vectors;
+  guint n_vectors;
+  guint8 *data;
+} WritevAsyncData;
+
+static void
+writtenv_read_write_async_cb (GOutputStream *ostream,
+                              GAsyncResult  *result,
+                              gpointer       user_data)
+{
+  WritevAsyncData *data = user_data;
+  GError *error = NULL;
+  gboolean res;
+  gsize bytes_written;
+  GSocketConnection *conn;
+
+  conn = data->conn;
+  g_free (data->data);
+  g_free (data);
+
+  res = g_output_stream_writev_all_finish (ostream, result, &bytes_written, &error);
+  g_assert_no_error (error);
+  g_assert_true (res);
+  g_assert_cmpint (bytes_written, ==, 20);
+
+  g_io_stream_close_async (G_IO_STREAM (conn),
+                           G_PRIORITY_DEFAULT,
+                           NULL,
+                           (GAsyncReadyCallback) closed_read_write_async_cb,
+                           NULL);
+  g_object_unref (conn);
+}
+
+static void
+connected_read_writev_async_cb (GObject      *client,
+                               GAsyncResult *result,
+                               gpointer      user_data)
+{
+  GSocketConnection *conn;
+  GOutputStream *ostream;
+  GError *error = NULL;
+  WritevAsyncData *data;
+  gsize i;
+  GSocketConnection **sconn = user_data;
+
+  conn = g_socket_client_connect_finish (G_SOCKET_CLIENT (client), result, &error);
+  g_assert_no_error (error);
+  g_assert_nonnull (conn);
+
+  ostream = g_io_stream_get_output_stream (G_IO_STREAM (conn));
+
+  data = g_new0 (WritevAsyncData, 1);
+  data->conn = conn;
+  data->vectors = g_new0 (GOutputVector, 3);
+  data->n_vectors = 3;
+  data->data = g_new0 (guint8, 20);
+  for (i = 0; i < 20; i++)
+    data->data[i] = i;
+
+  data->vectors[0].buffer = data->data;
+  data->vectors[0].size = 5;
+  data->vectors[1].buffer = data->data + 5;
+  data->vectors[1].size = 10;
+  data->vectors[2].buffer = data->data + 15;
+  data->vectors[2].size = 5;
+
+  g_output_stream_writev_all_async (ostream,
+                                    data->vectors,
+                                    data->n_vectors,
+                                    G_PRIORITY_DEFAULT,
+                                    NULL,
+                                    (GAsyncReadyCallback) writtenv_read_write_async_cb,
+                                    data /* stolen */);
+
+  *sconn = g_object_ref (conn);
+}
+
+typedef struct {
+  GSocketConnection *conn;
+  guint8 *data;
+} ReadAsyncData;
+
+static void
+read_read_write_async_cb (GInputStream *istream,
+                          GAsyncResult *result,
+                          gpointer      user_data)
+{
+  ReadAsyncData *data = user_data;
+  GError *error = NULL;
+  gboolean res;
+  gsize bytes_read;
+  GSocketConnection *conn;
+  const guint8 expected_data[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 };
+
+  res = g_input_stream_read_all_finish (istream, result, &bytes_read, &error);
+  g_assert_no_error (error);
+  g_assert_true (res);
+
+  g_assert_cmpmem (expected_data, sizeof expected_data, data->data, bytes_read);
+
+  conn = data->conn;
+  g_object_set_data (G_OBJECT (conn), "test-data-read", GINT_TO_POINTER (TRUE));
+
+  g_free (data->data);
+  g_free (data);
+
+  g_io_stream_close_async (G_IO_STREAM (conn),
+                           G_PRIORITY_DEFAULT,
+                           NULL,
+                           (GAsyncReadyCallback) closed_read_write_async_cb,
+                           NULL);
+  g_object_unref (conn);
+}
+
+static void
+incoming_read_write_async_cb (GSocketService    *service,
+                              GSocketConnection *conn,
+                              GObject           *source_object,
+                              gpointer           user_data)
+{
+  ReadAsyncData *data;
+  GSocketConnection **cconn = user_data;
+  GInputStream *istream;
+
+  istream = g_io_stream_get_input_stream (G_IO_STREAM (conn));
+
+  data = g_new0 (ReadAsyncData, 1);
+  data->conn = g_object_ref (conn);
+  data->data = g_new0 (guint8, 20);
+
+  g_input_stream_read_all_async (istream,
+                                 data->data,
+                                 20,
+                                 G_PRIORITY_DEFAULT,
+                                 NULL,
+                                 (GAsyncReadyCallback) read_read_write_async_cb,
+                                 data /* stolen */);
+
+  *cconn = g_object_ref (conn);
+}
+
+static void
+test_read_write_async_internal (gboolean writev)
+{
+  GInetAddress *iaddr;
+  GSocketAddress *saddr, *listening_addr;
+  GSocketService *service;
+  GError *error = NULL;
+  GSocketClient *client;
+  GSocketConnection *sconn = NULL, *cconn = NULL;
+
+  iaddr = g_inet_address_new_loopback (G_SOCKET_FAMILY_IPV4);
+  saddr = g_inet_socket_address_new (iaddr, 0);
+  g_object_unref (iaddr);
+
+  service = g_socket_service_new ();
+
+  g_socket_listener_add_address (G_SOCKET_LISTENER (service),
+                                 saddr,
+                                 G_SOCKET_TYPE_STREAM,
+                                 G_SOCKET_PROTOCOL_TCP,
+                                 NULL,
+                                 &listening_addr,
+                                 &error);
+  g_assert_no_error (error);
+  g_object_unref (saddr);
+
+  g_signal_connect (service, "incoming", G_CALLBACK (incoming_read_write_async_cb), &sconn);
+
+  client = g_socket_client_new ();
+
+  if (writev)
+    g_socket_client_connect_async (client,
+                                   G_SOCKET_CONNECTABLE (listening_addr),
+                                   NULL,
+                                   connected_read_writev_async_cb,
+                                   &cconn);
+  else
+    g_socket_client_connect_async (client,
+                                   G_SOCKET_CONNECTABLE (listening_addr),
+                                   NULL,
+                                   connected_read_write_async_cb,
+                                   &cconn);
+
+  g_object_unref (client);
+  g_object_unref (listening_addr);
+
+  g_socket_service_start (service);
+  g_assert_true (g_socket_service_is_active (service));
+
+  do
+    {
+      g_main_context_iteration (NULL, TRUE);
+    }
+  while (!sconn || !cconn ||
+         !g_io_stream_is_closed (G_IO_STREAM (sconn)) ||
+         !g_io_stream_is_closed (G_IO_STREAM (cconn)));
+
+  g_assert_true (GPOINTER_TO_INT (g_object_get_data (G_OBJECT (sconn), "test-data-read")));
+
+  g_object_unref (sconn);
+  g_object_unref (cconn);
+  g_object_unref (service);
+}
+
+/* Test if connecting to a socket service and asynchronously writing data on
+ * one side followed by reading the same data on the other side of the
+ * connection works correctly
+ */
+static void
+test_read_write_async (void)
+{
+  test_read_write_async_internal (FALSE);
+}
+
+/* Test if connecting to a socket service and asynchronously writing data on
+ * one side followed by reading the same data on the other side of the
+ * connection works correctly. This uses writev() instead of normal write().
+ */
+static void
+test_read_writev_async (void)
+{
+  test_read_write_async_internal (TRUE);
+}
+
+
 int
 main (int   argc,
       char *argv[])
@@ -242,6 +552,8 @@ main (int   argc,
 
   g_test_add_func ("/socket-service/start-stop", test_start_stop);
   g_test_add_func ("/socket-service/threaded/712570", test_threaded_712570);
+  g_test_add_func ("/socket-service/read_write_async", test_read_write_async);
+  g_test_add_func ("/socket-service/read_writev_async", test_read_writev_async);
 
   return g_test_run();
 }