luc14n0/glib
1
0
Fork 0
mirror of https://gitlab.gnome.org/GNOME/glib.git synced 2024-12-26 07:26:15 +01:00
Code Issues Packages Projects Releases Wiki Activity

gio/tests/socket: add datagram version of test_ip_sync

Browse Source
This commit is contained in:
Tim-Philipp Müller 2014-06-13 19:45:25 +01:00
parent 4864850427
commit 3c3fc0e463
1 changed files with 163 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

162
gio/tests/socket.c
View File

@ -37,8 +37,44 @@ typedef struct {
GSocketFamily family;
GThread *thread;
GMainLoop *loop;
GCancellable *cancellable; /* to shut down dgram echo server thread */
} IPTestData;
static gpointer
echo_server_dgram_thread (gpointer user_data)
{
IPTestData *data = user_data;
GSocketAddress *sa;
GCancellable *cancellable = data->cancellable;
GSocket *sock;
GError *error = NULL;
gssize nread, nwrote;
gchar buf[128];
sock = data->server;
while (TRUE)
{
nread = g_socket_receive_from (sock, &sa, buf, sizeof (buf), cancellable, &error);
if (error && g_error_matches (error, G_IO_ERROR, G_IO_ERROR_CANCELLED))
break;
g_assert_no_error (error);
g_assert_cmpint (nread, >=, 0);
nwrote = g_socket_send_to (sock, sa, buf, nread, cancellable, &error);
if (error && g_error_matches (error, G_IO_ERROR, G_IO_ERROR_CANCELLED))
break;
g_assert_no_error (error);
g_assert_cmpint (nwrote, ==, nread);
g_object_unref (sa);
}
g_clear_error (&error);
return NULL;
}
static gpointer
echo_server_thread (gpointer user_data)
{
@ -72,7 +108,8 @@ echo_server_thread (gpointer user_data)
}
static IPTestData *
create_server (GSocketFamily family,
create_server_full (GSocketFamily family,
GSocketType socket_type,
GThreadFunc server_thread,
gboolean v4mapped)
{
@ -86,13 +123,13 @@ create_server (GSocketFamily family,
data->family = family;
data->server = server = g_socket_new (family,
G_SOCKET_TYPE_STREAM,
socket_type,
G_SOCKET_PROTOCOL_DEFAULT,
&error);
g_assert_no_error (error);
g_assert_cmpint (g_socket_get_family (server), ==, family);
g_assert_cmpint (g_socket_get_socket_type (server), ==, G_SOCKET_TYPE_STREAM);
g_assert_cmpint (g_socket_get_socket_type (server), ==, socket_type);
g_assert_cmpint (g_socket_get_protocol (server), ==, G_SOCKET_PROTOCOL_DEFAULT);
g_socket_set_blocking (server, TRUE);
@ -127,14 +164,29 @@ create_server (GSocketFamily family,
g_assert_cmpint (g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (addr)), !=, 0);
g_object_unref (addr);
if (socket_type == G_SOCKET_TYPE_STREAM)
{
g_socket_listen (server, &error);
g_assert_no_error (error);
}
else
{
data->cancellable = g_cancellable_new ();
}
data->thread = g_thread_new ("server", server_thread, data);
return data;
}
static IPTestData *
create_server (GSocketFamily family,
GThreadFunc server_thread,
gboolean v4mapped)
{
return create_server_full (family, G_SOCKET_TYPE_STREAM, server_thread, v4mapped);
}
static const gchar *testbuf = "0123456789abcdef";
static gboolean
@ -495,6 +547,108 @@ test_ipv6_sync (void)
test_ip_sync (G_SOCKET_FAMILY_IPV6);
}
static void
test_ip_sync_dgram (GSocketFamily family)
{
IPTestData *data;
GError *error = NULL;
GSocket *client;
GSocketAddress *dest_addr;
gssize len;
gchar buf[128];
data = create_server_full (family, G_SOCKET_TYPE_DATAGRAM,
echo_server_dgram_thread, FALSE);
dest_addr = g_socket_get_local_address (data->server, &error);
client = g_socket_new (family,
G_SOCKET_TYPE_DATAGRAM,
G_SOCKET_PROTOCOL_DEFAULT,
&error);
g_assert_no_error (error);
g_assert_cmpint (g_socket_get_family (client), ==, family);
g_assert_cmpint (g_socket_get_socket_type (client), ==, G_SOCKET_TYPE_DATAGRAM);
g_assert_cmpint (g_socket_get_protocol (client), ==, G_SOCKET_PROTOCOL_DEFAULT);
g_socket_set_blocking (client, TRUE);
g_socket_set_timeout (client, 1);
len = g_socket_send_to (client, dest_addr, testbuf, strlen (testbuf) + 1, NULL, &error);
g_assert_no_error (error);
g_assert_cmpint (len, ==, strlen (testbuf) + 1);
len = g_socket_receive_from (client, NULL, buf, sizeof (buf), NULL, &error);
g_assert_no_error (error);
g_assert_cmpint (len, ==, strlen (testbuf) + 1);
g_assert_cmpstr (testbuf, ==, buf);
{
GOutputVector v[7] = { { NULL, }, };
v[0].buffer = testbuf2 + 0;
v[0].size = 3;
v[1].buffer = testbuf2 + 3;
v[1].size = 5;
v[2].buffer = testbuf2 + 3 + 5;
v[2].size = 0;
v[3].buffer = testbuf2 + 3 + 5;
v[3].size = 6;
v[4].buffer = testbuf2 + 3 + 5 + 6;
v[4].size = 2;
v[5].buffer = testbuf2 + 3 + 5 + 6 + 2;
v[5].size = 1;
v[6].buffer = testbuf2 + 3 + 5 + 6 + 2 + 1;
v[6].size = strlen (testbuf2) - (3 + 5 + 6 + 2 + 1);
len = g_socket_send_message (client, dest_addr, v, G_N_ELEMENTS (v), NULL, 0, 0, NULL, &error);
g_assert_no_error (error);
g_assert_cmpint (len, ==, strlen (testbuf2));
memset (buf, 0, sizeof (buf));
len = g_socket_receive_from (client, NULL, buf, sizeof (buf), NULL, &error);
g_assert_no_error (error);
g_assert_cmpint (len, ==, strlen (testbuf2));
g_assert_cmpstr (testbuf2, ==, buf);
}
g_cancellable_cancel (data->cancellable);
g_thread_join (data->thread);
g_socket_close (client, &error);
g_assert_no_error (error);
g_socket_close (data->server, &error);
g_assert_no_error (error);
g_object_unref (data->server);
g_object_unref (data->cancellable);
g_object_unref (client);
g_object_unref (dest_addr);
g_slice_free (IPTestData, data);
}
static void
test_ipv4_sync_dgram (void)
{
test_ip_sync_dgram (G_SOCKET_FAMILY_IPV4);
}
static void
test_ipv6_sync_dgram (void)
{
if (!ipv6_supported)
{
g_test_skip ("No support for IPv6");
return;
}
test_ip_sync_dgram (G_SOCKET_FAMILY_IPV6);
}
static gpointer
graceful_server_thread (gpointer user_data)
{
@ -1138,6 +1292,8 @@ main (int argc,
g_test_add_func ("/socket/ipv4_async", test_ipv4_async);
g_test_add_func ("/socket/ipv6_sync", test_ipv6_sync);
g_test_add_func ("/socket/ipv6_async", test_ipv6_async);
g_test_add_func ("/socket/ipv4_sync/datagram", test_ipv4_sync_dgram);
g_test_add_func ("/socket/ipv6_sync/datagram", test_ipv6_sync_dgram);
#if defined (IPPROTO_IPV6) && defined (IPV6_V6ONLY)
g_test_add_func ("/socket/ipv6_v4mapped", test_ipv6_v4mapped);
#endif