mirror of
https://gitlab.gnome.org/GNOME/glib.git
synced 2025-04-01 13:23:07 +02:00
Add test for async read/write(v)/close on GSocketConnections
This commit is contained in:
Sebastian Dröge
parent
708aa8f4ee
commit
61520ae601
@ -232,6 +232,316 @@ test_threaded_712570 (void)
|
|||||||
g_mutex_unlock (&mutex_712570);
|
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
|
int
|
||||||
main (int argc,
|
main (int argc,
|
||||||
char *argv[])
|
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/start-stop", test_start_stop);
|
||||||
g_test_add_func ("/socket-service/threaded/712570", test_threaded_712570);
|
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();
|
return g_test_run();
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user