* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
- * version 2 of the License, or (at your option) any later version.
+ * version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
#include <gio/gio.h>
+#include <gio/gcredentialsprivate.h>
#ifdef G_OS_UNIX
#include <errno.h>
#include <sys/wait.h>
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)
{
IPTestData *data = user_data;
}
static IPTestData *
-create_server (GSocketFamily family,
- GThreadFunc server_thread,
- gboolean v4mapped)
+create_server_full (GSocketFamily family,
+ GSocketType socket_type,
+ GThreadFunc server_thread,
+ gboolean v4mapped,
+ GError **error)
{
IPTestData *data;
GSocket *server;
- GError *error = NULL;
GSocketAddress *addr;
GInetAddress *iaddr;
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);
+ error);
+ if (server == NULL)
+ goto 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);
if (v4mapped)
{
g_socket_set_option (data->server, IPPROTO_IPV6, IPV6_V6ONLY, FALSE, NULL);
- if (! g_socket_speaks_ipv4 (data->server))
+ if (!g_socket_speaks_ipv4 (data->server))
{
- g_object_unref (data->server);
- g_slice_free (IPTestData, data);
- return NULL;
+ g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED,
+ "IPv6-only server cannot speak IPv4");
+ goto error;
}
}
#endif
g_object_unref (iaddr);
g_assert_cmpint (g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (addr)), ==, 0);
- g_socket_bind (server, addr, TRUE, &error);
- g_assert_no_error (error);
+ if (!g_socket_bind (server, addr, TRUE, error))
+ {
+ g_object_unref (addr);
+ goto error;
+ }
g_object_unref (addr);
- addr = g_socket_get_local_address (server, &error);
- g_assert_no_error (error);
+ addr = g_socket_get_local_address (server, error);
+ if (addr == NULL)
+ goto error;
g_assert_cmpint (g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (addr)), !=, 0);
g_object_unref (addr);
- g_socket_listen (server, &error);
- g_assert_no_error (error);
+ if (socket_type == G_SOCKET_TYPE_STREAM)
+ {
+ if (!g_socket_listen (server, error))
+ goto error;
+ }
+ else
+ {
+ data->cancellable = g_cancellable_new ();
+ }
data->thread = g_thread_new ("server", server_thread, data);
return data;
+
+error:
+ g_clear_object (&data->server);
+ g_slice_free (IPTestData, data);
+
+ return NULL;
+}
+
+static IPTestData *
+create_server (GSocketFamily family,
+ GThreadFunc server_thread,
+ gboolean v4mapped,
+ GError **error)
+{
+ return create_server_full (family, G_SOCKET_TYPE_STREAM, server_thread, v4mapped, error);
}
static const gchar *testbuf = "0123456789abcdef";
gssize len;
gchar buf[128];
- data = create_server (family, echo_server_thread, FALSE);
+ data = create_server (family, echo_server_thread, FALSE, &error);
+ if (error != NULL)
+ {
+ gchar *message = g_strdup_printf ("Failed to create server: %s", error->message);
+ g_test_skip (message);
+ g_free (message);
+ g_clear_error (&error);
+ return;
+ }
+ g_assert_nonnull (data);
+
addr = g_socket_get_local_address (data->server, &error);
g_assert_no_error (error);
gssize len;
gchar buf[128];
- data = create_server (family, echo_server_thread, FALSE);
+ data = create_server (family, echo_server_thread, FALSE, &error);
+ if (error != NULL)
+ {
+ gchar *message = g_strdup_printf ("Failed to create server: %s", error->message);
+ g_test_skip (message);
+ g_free (message);
+ g_clear_error (&error);
+ return;
+ }
+
addr = g_socket_get_local_address (data->server, &error);
g_assert_no_error (error);
g_assert_no_error (error);
g_assert_cmpint (len, ==, strlen (testbuf2));
+ memset (buf, 0, sizeof (buf));
len = g_socket_receive (client, buf, sizeof (buf), NULL, &error);
g_assert_no_error (error);
g_assert_cmpint (len, ==, strlen (testbuf2));
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, &error);
+ if (error != NULL)
+ {
+ gchar *message = g_strdup_printf ("Failed to create server: %s", error->message);
+ g_test_skip (message);
+ g_free (message);
+ g_clear_error (&error);
+ return;
+ }
+
+ 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);
+
+ {
+ GOutputMessage m[3] = { { NULL, }, };
+ GInputMessage im[3] = { { NULL, }, };
+ GOutputVector v[7] = { { NULL, }, };
+ GInputVector iv[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);
+
+ iv[0].buffer = buf + 0;
+ iv[0].size = 3;
+ iv[1].buffer = buf + 3;
+ iv[1].size = 5;
+ iv[2].buffer = buf + 3 + 5;
+ iv[2].size = 0;
+ iv[3].buffer = buf + 3 + 5;
+ iv[3].size = 6;
+ iv[4].buffer = buf + 3 + 5 + 6;
+ iv[4].size = 2;
+ iv[5].buffer = buf + 3 + 5 + 6 + 2;
+ iv[5].size = 1;
+ iv[6].buffer = buf + 3 + 5 + 6 + 2 + 1;
+ iv[6].size = sizeof (buf) - (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);
+
+ m[0].vectors = &v[0];
+ m[0].num_vectors = 1;
+ m[0].address = dest_addr;
+ m[1].vectors = &v[0];
+ m[1].num_vectors = 6;
+ m[1].address = dest_addr;
+ m[2].vectors = &v[6];
+ m[2].num_vectors = 1;
+ m[2].address = dest_addr;
+
+ len = g_socket_send_messages (client, m, G_N_ELEMENTS (m), 0, NULL, &error);
+ g_assert_no_error (error);
+ g_assert_cmpint (len, ==, G_N_ELEMENTS (m));
+ g_assert_cmpint (m[0].bytes_sent, ==, 3);
+ g_assert_cmpint (m[1].bytes_sent, ==, 17);
+ g_assert_cmpint (m[2].bytes_sent, ==, v[6].size);
+
+ 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, ==, 3);
+
+ memset (buf, 0, sizeof (buf));
+ len = g_socket_receive_from (client, NULL, buf, sizeof (buf), NULL, &error);
+ g_assert_no_error (error);
+ /* v[0].size + v[1].size + v[2].size + v[3].size + v[4].size + v[5].size */
+ g_assert_cmpint (len, ==, 17);
+ g_assert (memcmp (testbuf2, buf, 17) == 0);
+
+ 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, ==, v[6].size);
+ g_assert_cmpstr (buf, ==, v[6].buffer);
+
+ /* reset since we're re-using the message structs */
+ m[0].bytes_sent = 0;
+ m[1].bytes_sent = 0;
+ m[2].bytes_sent = 0;
+
+ /* now try receiving multiple messages */
+ len = g_socket_send_messages (client, m, G_N_ELEMENTS (m), 0, NULL, &error);
+ g_assert_no_error (error);
+ g_assert_cmpint (len, ==, G_N_ELEMENTS (m));
+ g_assert_cmpint (m[0].bytes_sent, ==, 3);
+ g_assert_cmpint (m[1].bytes_sent, ==, 17);
+ g_assert_cmpint (m[2].bytes_sent, ==, v[6].size);
+
+ im[0].vectors = &iv[0];
+ im[0].num_vectors = 1;
+ im[1].vectors = &iv[0];
+ im[1].num_vectors = 6;
+ im[2].vectors = &iv[6];
+ im[2].num_vectors = 1;
+
+ memset (buf, 0, sizeof (buf));
+ len = g_socket_receive_messages (client, im, G_N_ELEMENTS (im), 0,
+ NULL, &error);
+ g_assert_no_error (error);
+ g_assert_cmpint (len, ==, G_N_ELEMENTS (im));
+
+ g_assert_cmpuint (im[0].bytes_received, ==, 3);
+ /* v[0].size + v[1].size + v[2].size + v[3].size + v[4].size + v[5].size */
+ g_assert_cmpuint (im[1].bytes_received, ==, 17);
+ g_assert_cmpuint (im[2].bytes_received, ==, v[6].size);
+
+ /* reset since we're re-using the message structs */
+ m[0].bytes_sent = 0;
+ m[1].bytes_sent = 0;
+ m[2].bytes_sent = 0;
+
+ /* now try to generate an early return by omitting the destination address on [1] */
+ m[1].address = NULL;
+ len = g_socket_send_messages (client, m, G_N_ELEMENTS (m), 0, NULL, &error);
+ g_assert_no_error (error);
+ g_assert_cmpint (len, ==, 1);
+
+ g_assert_cmpint (m[0].bytes_sent, ==, 3);
+ g_assert_cmpint (m[1].bytes_sent, ==, 0);
+ g_assert_cmpint (m[2].bytes_sent, ==, 0);
+
+ /* reset since we're re-using the message structs */
+ m[0].bytes_sent = 0;
+ m[1].bytes_sent = 0;
+ m[2].bytes_sent = 0;
+
+ /* now try to generate an error by omitting all destination addresses */
+ m[0].address = NULL;
+ m[1].address = NULL;
+ m[2].address = NULL;
+ len = g_socket_send_messages (client, m, G_N_ELEMENTS (m), 0, NULL, &error);
+ /* This error code may vary between platforms and over time; it is not guaranteed API: */
+#ifndef G_OS_WIN32
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_FAILED);
+#else
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_NOT_CONNECTED);
+#endif
+ g_clear_error (&error);
+ g_assert_cmpint (len, ==, -1);
+
+ g_assert_cmpint (m[0].bytes_sent, ==, 0);
+ g_assert_cmpint (m[1].bytes_sent, ==, 0);
+ g_assert_cmpint (m[2].bytes_sent, ==, 0);
+
+ len = g_socket_receive_from (client, NULL, buf, sizeof (buf), NULL, &error);
+ g_assert_cmpint (len, ==, 3);
+ }
+
+ 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
+cancellable_thread_cb (gpointer data)
+{
+ GCancellable *cancellable = data;
+
+ g_usleep (0.1 * G_USEC_PER_SEC);
+ g_cancellable_cancel (cancellable);
+ g_object_unref (cancellable);
+
+ return NULL;
+}
+
+static void
+test_ip_sync_dgram_timeouts (GSocketFamily family)
+{
+ GError *error = NULL;
+ GSocket *client = NULL;
+ GCancellable *cancellable = NULL;
+ GThread *cancellable_thread = NULL;
+ gssize len;
+#ifdef G_OS_WIN32
+ GInetAddress *iaddr;
+ GSocketAddress *addr;
+#endif
+
+ 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);
+
+#ifdef G_OS_WIN32
+ /* Winsock can't recv() on unbound udp socket */
+ iaddr = g_inet_address_new_loopback (family);
+ addr = g_inet_socket_address_new (iaddr, 0);
+ g_object_unref (iaddr);
+ g_socket_bind (client, addr, TRUE, &error);
+ g_object_unref (addr);
+ g_assert_no_error (error);
+#endif
+
+ /* No overall timeout: test the per-operation timeouts instead. */
+ g_socket_set_timeout (client, 0);
+
+ cancellable = g_cancellable_new ();
+
+ /* Check for timeouts when no server is running. */
+ {
+ gint64 start_time;
+ GInputMessage im = { NULL, };
+ GInputVector iv = { NULL, };
+ guint8 buf[128];
+
+ iv.buffer = buf;
+ iv.size = sizeof (buf);
+
+ im.vectors = &iv;
+ im.num_vectors = 1;
+
+ memset (buf, 0, sizeof (buf));
+
+ /* Try a non-blocking read. */
+ g_socket_set_blocking (client, FALSE);
+ len = g_socket_receive_messages (client, &im, 1, 0 /* flags */,
+ NULL, &error);
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK);
+ g_assert_cmpint (len, ==, -1);
+ g_clear_error (&error);
+
+ /* Try a timeout read. Can’t really validate the time taken more than
+ * checking it’s positive. */
+ g_socket_set_timeout (client, 1);
+ g_socket_set_blocking (client, TRUE);
+ start_time = g_get_monotonic_time ();
+ len = g_socket_receive_messages (client, &im, 1, 0 /* flags */,
+ NULL, &error);
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_TIMED_OUT);
+ g_assert_cmpint (len, ==, -1);
+ g_assert_cmpint (g_get_monotonic_time () - start_time, >, 0);
+ g_clear_error (&error);
+
+ /* Try a blocking read, cancelled from another thread. */
+ g_socket_set_timeout (client, 0);
+ cancellable_thread = g_thread_new ("cancellable",
+ cancellable_thread_cb,
+ g_object_ref (cancellable));
+
+ start_time = g_get_monotonic_time ();
+ len = g_socket_receive_messages (client, &im, 1, 0 /* flags */,
+ cancellable, &error);
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
+ g_assert_cmpint (len, ==, -1);
+ g_assert_cmpint (g_get_monotonic_time () - start_time, >, 0);
+ g_clear_error (&error);
+
+ g_thread_join (cancellable_thread);
+ }
+
+ g_socket_close (client, &error);
+ g_assert_no_error (error);
+
+ g_object_unref (client);
+ g_object_unref (cancellable);
+}
+
+static void
+test_ipv4_sync_dgram_timeouts (void)
+{
+ test_ip_sync_dgram_timeouts (G_SOCKET_FAMILY_IPV4);
+}
+
+static void
+test_ipv6_sync_dgram_timeouts (void)
+{
+ if (!ipv6_supported)
+ {
+ g_test_skip ("No support for IPv6");
+ return;
+ }
+
+ test_ip_sync_dgram_timeouts (G_SOCKET_FAMILY_IPV6);
+}
+
static gpointer
graceful_server_thread (gpointer user_data)
{
gssize len;
gchar buf[128];
- data = create_server (family, graceful_server_thread, FALSE);
+ data = create_server (family, graceful_server_thread, FALSE, &error);
+ if (error != NULL)
+ {
+ gchar *message = g_strdup_printf ("Failed to create server: %s", error->message);
+ g_test_skip (message);
+ g_free (message);
+ g_clear_error (&error);
+ return;
+ }
+
addr = g_socket_get_local_address (data->server, &error);
g_assert_no_error (error);
return;
}
- data = create_server (G_SOCKET_FAMILY_IPV6, v4mapped_server_thread, TRUE);
-
- if (data == NULL)
+ data = create_server (G_SOCKET_FAMILY_IPV6, v4mapped_server_thread, TRUE, &error);
+ if (error != NULL)
{
- g_test_message ("Test not run: not supported by the OS");
+ gchar *message = g_strdup_printf ("Failed to create server: %s", error->message);
+ g_test_skip (message);
+ g_free (message);
+ g_clear_error (&error);
return;
}
gint64 start_time;
gint poll_duration;
- data = create_server (G_SOCKET_FAMILY_IPV4, echo_server_thread, FALSE);
+ if (!g_test_thorough ())
+ {
+ g_test_skip ("Not running timing heavy test");
+ return;
+ }
+
+ data = create_server (G_SOCKET_FAMILY_IPV4, echo_server_thread, FALSE, &error);
+ if (error != NULL)
+ {
+ gchar *message = g_strdup_printf ("Failed to create server: %s", error->message);
+ g_test_skip (message);
+ g_free (message);
+ g_clear_error (&error);
+ return;
+ }
+
addr = g_socket_get_local_address (data->server, &error);
g_assert_no_error (error);
g_slice_free (IPTestData, data);
}
+static int
+duplicate_fd (int fd)
+{
+#ifdef G_OS_WIN32
+ HANDLE newfd;
+
+ if (!DuplicateHandle (GetCurrentProcess (),
+ (HANDLE)fd,
+ GetCurrentProcess (),
+ &newfd,
+ 0,
+ FALSE,
+ DUPLICATE_SAME_ACCESS))
+ {
+ return -1;
+ }
+
+ return (int)newfd;
+#else
+ return dup (fd);
+#endif
+}
+
static void
-test_sockaddr (void)
+test_fd_reuse (void)
{
- struct sockaddr_in6 sin6, gsin6;
- GSocketAddress *saddr;
- GInetSocketAddress *isaddr;
- GInetAddress *iaddr;
+ IPTestData *data;
GError *error = NULL;
+ GSocket *client;
+ GSocket *client2;
+ GSocketAddress *addr;
+ int fd;
+ gssize len;
+ gchar buf[128];
- memset (&sin6, 0, sizeof (sin6));
- sin6.sin6_family = AF_INET6;
- sin6.sin6_addr = in6addr_loopback;
- sin6.sin6_port = g_htons (42);
- sin6.sin6_scope_id = 17;
- sin6.sin6_flowinfo = 1729;
+ g_test_bug ("741707");
- saddr = g_socket_address_new_from_native (&sin6, sizeof (sin6));
+ data = create_server (G_SOCKET_FAMILY_IPV4, echo_server_thread, FALSE, &error);
+ if (error != NULL)
+ {
+ gchar *message = g_strdup_printf ("Failed to create server: %s", error->message);
+ g_test_skip (message);
+ g_free (message);
+ g_clear_error (&error);
+ return;
+ }
+
+ addr = g_socket_get_local_address (data->server, &error);
+ g_assert_no_error (error);
+
+ client = g_socket_new (G_SOCKET_FAMILY_IPV4,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &error);
+ g_assert_no_error (error);
+
+ g_socket_set_blocking (client, TRUE);
+ g_socket_set_timeout (client, 1);
+
+ g_socket_connect (client, addr, NULL, &error);
+ g_assert_no_error (error);
+ g_assert (g_socket_is_connected (client));
+ g_object_unref (addr);
+
+ /* we have to dup otherwise the fd gets closed twice on unref */
+ fd = duplicate_fd (g_socket_get_fd (client));
+ client2 = g_socket_new_from_fd (fd, &error);
+ g_assert_no_error (error);
+
+ g_assert_cmpint (g_socket_get_family (client2), ==, g_socket_get_family (client));
+ g_assert_cmpint (g_socket_get_socket_type (client2), ==, g_socket_get_socket_type (client));
+ g_assert_cmpint (g_socket_get_protocol (client2), ==, G_SOCKET_PROTOCOL_TCP);
+
+ len = g_socket_send (client2, testbuf, strlen (testbuf) + 1, NULL, &error);
+ g_assert_no_error (error);
+ g_assert_cmpint (len, ==, strlen (testbuf) + 1);
+
+ len = g_socket_receive (client2, buf, sizeof (buf), NULL, &error);
+ g_assert_no_error (error);
+ g_assert_cmpint (len, ==, strlen (testbuf) + 1);
+
+ g_assert_cmpstr (testbuf, ==, buf);
+
+ g_socket_shutdown (client, FALSE, TRUE, &error);
+ g_assert_no_error (error);
+ /* The semantics of dup()+shutdown() are ambiguous; this call will succeed
+ * on Linux, but return ENOTCONN on OS X.
+ */
+ g_socket_shutdown (client2, FALSE, TRUE, NULL);
+
+ g_thread_join (data->thread);
+
+ g_socket_close (client, &error);
+ g_assert_no_error (error);
+ g_socket_close (client2, &error);
+ g_assert_no_error (error);
+ g_socket_close (data->server, &error);
+ g_assert_no_error (error);
+
+ g_assert_cmpint (g_socket_get_fd (client), ==, -1);
+ g_assert_cmpint (g_socket_get_fd (client2), ==, -1);
+ g_assert_cmpint (g_socket_get_fd (data->server), ==, -1);
+
+ g_object_unref (data->server);
+ g_object_unref (client);
+ g_object_unref (client2);
+
+ g_slice_free (IPTestData, data);
+}
+
+static void
+test_sockaddr (void)
+{
+ struct sockaddr_in6 sin6, gsin6;
+ GSocketAddress *saddr;
+ GInetSocketAddress *isaddr;
+ GInetAddress *iaddr;
+ GError *error = NULL;
+
+ memset (&sin6, 0, sizeof (sin6));
+ sin6.sin6_family = AF_INET6;
+ sin6.sin6_addr = in6addr_loopback;
+ sin6.sin6_port = g_htons (42);
+ sin6.sin6_scope_id = 17;
+ sin6.sin6_flowinfo = 1729;
+
+ saddr = g_socket_address_new_from_native (&sin6, sizeof (sin6));
g_assert (G_IS_INET_SOCKET_ADDRESS (saddr));
isaddr = G_INET_SOCKET_ADDRESS (saddr);
* g_unix_connection_receive_credentials().
*/
}
+
+static gboolean
+postmortem_source_cb (GSocket *socket,
+ GIOCondition condition,
+ gpointer user_data)
+{
+ gboolean *been_here = user_data;
+
+ g_assert_cmpint (condition, ==, G_IO_NVAL);
+
+ *been_here = TRUE;
+ return FALSE;
+}
+
+static void
+test_source_postmortem (void)
+{
+ GMainContext *context;
+ GSocket *socket;
+ GSource *source;
+ GError *error = NULL;
+ gboolean callback_visited = FALSE;
+
+ socket = g_socket_new (G_SOCKET_FAMILY_UNIX, G_SOCKET_TYPE_STREAM, G_SOCKET_PROTOCOL_DEFAULT, &error);
+ g_assert_no_error (error);
+
+ context = g_main_context_new ();
+
+ source = g_socket_create_source (socket, G_IO_IN, NULL);
+ g_source_set_callback (source, (GSourceFunc) postmortem_source_cb,
+ &callback_visited, NULL);
+ g_source_attach (source, context);
+ g_source_unref (source);
+
+ g_socket_close (socket, &error);
+ g_assert_no_error (error);
+ g_object_unref (socket);
+
+ /* Test that, after a socket is closed, its source callback should be called
+ * exactly once. */
+ g_main_context_iteration (context, FALSE);
+ g_assert (callback_visited);
+ g_assert (!g_main_context_pending (context));
+
+ g_main_context_unref (context);
+}
+
#endif /* G_OS_UNIX */
static void
GError *err = NULL;
GSocket *listener, *server, *client;
GInetAddress *addr;
- GSocketAddress *saddr;
+ GSocketAddress *saddr, *boundaddr;
gchar data[] = "0123456789abcdef";
gchar buf[34];
gssize nread;
g_object_unref (saddr);
g_object_unref (addr);
- saddr = g_socket_get_local_address (listener, &err);
+ boundaddr = g_socket_get_local_address (listener, &err);
g_assert_no_error (err);
+ addr = g_inet_address_new_loopback (G_SOCKET_FAMILY_IPV4);
+ saddr = g_inet_socket_address_new (addr, g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (boundaddr)));
+ g_object_unref (addr);
+ g_object_unref (boundaddr);
+
if (socket_type == G_SOCKET_TYPE_STREAM)
{
g_socket_listen (listener, &err);
g_object_unref (client);
}
+typedef struct {
+ GInputStream *is;
+ GOutputStream *os;
+ const guint8 *write_data;
+ guint8 *read_data;
+} TestReadWriteData;
+
+static gpointer
+test_read_write_write_thread (gpointer user_data)
+{
+ TestReadWriteData *data = user_data;
+ gsize bytes_written;
+ GError *error = NULL;
+ gboolean res;
+
+ res = g_output_stream_write_all (data->os, data->write_data, 1024, &bytes_written, NULL, &error);
+ g_assert_true (res);
+ g_assert_no_error (error);
+ g_assert_cmpint (bytes_written, ==, 1024);
+
+ return NULL;
+}
+
+static gpointer
+test_read_write_read_thread (gpointer user_data)
+{
+ TestReadWriteData *data = user_data;
+ gsize bytes_read;
+ GError *error = NULL;
+ gboolean res;
+
+ res = g_input_stream_read_all (data->is, data->read_data, 1024, &bytes_read, NULL, &error);
+ g_assert_true (res);
+ g_assert_no_error (error);
+ g_assert_cmpint (bytes_read, ==, 1024);
+
+ return NULL;
+}
+
+static gpointer
+test_read_write_writev_thread (gpointer user_data)
+{
+ TestReadWriteData *data = user_data;
+ gsize bytes_written;
+ GError *error = NULL;
+ gboolean res;
+ GOutputVector vectors[3];
+
+ vectors[0].buffer = data->write_data;
+ vectors[0].size = 256;
+ vectors[1].buffer = data->write_data + 256;
+ vectors[1].size = 256;
+ vectors[2].buffer = data->write_data + 512;
+ vectors[2].size = 512;
+
+ res = g_output_stream_writev_all (data->os, vectors, G_N_ELEMENTS (vectors), &bytes_written, NULL, &error);
+ g_assert_true (res);
+ g_assert_no_error (error);
+ g_assert_cmpint (bytes_written, ==, 1024);
+
+ return NULL;
+}
+
+/* test if normal read/write/writev via the GSocket*Streams works on TCP sockets */
+static void
+test_read_write (gconstpointer user_data)
+{
+ gboolean writev = GPOINTER_TO_INT (user_data);
+ GError *err = NULL;
+ GSocket *listener, *server, *client;
+ GInetAddress *addr;
+ GSocketAddress *saddr, *boundaddr;
+ TestReadWriteData data;
+ guint8 data_write[1024], data_read[1024];
+ GSocketConnection *server_stream, *client_stream;
+ GThread *write_thread, *read_thread;
+ guint i;
+
+ listener = g_socket_new (G_SOCKET_FAMILY_IPV4,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &err);
+ g_assert_no_error (err);
+ g_assert (G_IS_SOCKET (listener));
+
+ client = g_socket_new (G_SOCKET_FAMILY_IPV4,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &err);
+ g_assert_no_error (err);
+ g_assert (G_IS_SOCKET (client));
+
+ addr = g_inet_address_new_any (G_SOCKET_FAMILY_IPV4);
+ saddr = g_inet_socket_address_new (addr, 0);
+
+ g_socket_bind (listener, saddr, TRUE, &err);
+ g_assert_no_error (err);
+ g_object_unref (saddr);
+ g_object_unref (addr);
+
+ boundaddr = g_socket_get_local_address (listener, &err);
+ g_assert_no_error (err);
+
+ g_socket_listen (listener, &err);
+ g_assert_no_error (err);
+
+ addr = g_inet_address_new_loopback (G_SOCKET_FAMILY_IPV4);
+ saddr = g_inet_socket_address_new (addr, g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (boundaddr)));
+ g_object_unref (addr);
+ g_object_unref (boundaddr);
+
+ g_socket_connect (client, saddr, NULL, &err);
+ g_assert_no_error (err);
+
+ server = g_socket_accept (listener, NULL, &err);
+ g_assert_no_error (err);
+ g_socket_set_blocking (server, FALSE);
+ g_object_unref (listener);
+
+ server_stream = g_socket_connection_factory_create_connection (server);
+ g_assert_nonnull (server_stream);
+ client_stream = g_socket_connection_factory_create_connection (client);
+ g_assert_nonnull (client_stream);
+
+ for (i = 0; i < sizeof (data_write); i++)
+ data_write[i] = i;
+
+ data.is = g_io_stream_get_input_stream (G_IO_STREAM (server_stream));
+ data.os = g_io_stream_get_output_stream (G_IO_STREAM (client_stream));
+ data.read_data = data_read;
+ data.write_data = data_write;
+
+ if (writev)
+ write_thread = g_thread_new ("writer", test_read_write_writev_thread, &data);
+ else
+ write_thread = g_thread_new ("writer", test_read_write_write_thread, &data);
+ read_thread = g_thread_new ("reader", test_read_write_read_thread, &data);
+
+ g_thread_join (write_thread);
+ g_thread_join (read_thread);
+
+ g_assert_cmpmem (data_write, sizeof data_write, data_read, sizeof data_read);
+
+ g_socket_close (server, &err);
+ g_assert_no_error (err);
+
+ g_object_unref (server_stream);
+ g_object_unref (client_stream);
+
+ g_object_unref (saddr);
+ g_object_unref (server);
+ g_object_unref (client);
+}
+
+#if G_CREDENTIALS_SUPPORTED
+static gpointer client_setup_thread (gpointer user_data);
+
+static void
+test_credentials_tcp_client (void)
+{
+ const GSocketFamily family = G_SOCKET_FAMILY_IPV4;
+ IPTestData *data;
+ GError *error = NULL;
+ GSocket *client;
+ GSocketAddress *addr;
+ GCredentials *creds;
+
+ data = create_server (family, echo_server_thread, FALSE, &error);
+ if (error != NULL)
+ {
+ gchar *message = g_strdup_printf ("Failed to create server: %s", error->message);
+ g_test_skip (message);
+ g_free (message);
+ g_clear_error (&error);
+ return;
+ }
+
+ addr = g_socket_get_local_address (data->server, &error);
+ g_assert_no_error (error);
+
+ client = g_socket_new (family,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &error);
+ g_assert_no_error (error);
+
+ g_socket_set_blocking (client, TRUE);
+ g_socket_set_timeout (client, 1);
+
+ g_socket_connect (client, addr, NULL, &error);
+ g_assert_no_error (error);
+ g_object_unref (addr);
+
+ creds = g_socket_get_credentials (client, &error);
+ if (creds != NULL)
+ {
+ gchar *str = g_credentials_to_string (creds);
+ g_print ("Supported on this OS: %s\n", str);
+ g_free (str);
+ g_clear_object (&creds);
+ }
+ else
+ {
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED);
+ g_print ("Unsupported on this OS: %s\n", error->message);
+ g_clear_error (&error);
+ }
+
+ g_socket_close (client, &error);
+ g_assert_no_error (error);
+
+ g_thread_join (data->thread);
+
+ g_socket_close (data->server, &error);
+ g_assert_no_error (error);
+
+ g_object_unref (data->server);
+ g_object_unref (client);
+
+ g_slice_free (IPTestData, data);
+}
+
+static void
+test_credentials_tcp_server (void)
+{
+ const GSocketFamily family = G_SOCKET_FAMILY_IPV4;
+ IPTestData *data;
+ GSocket *server;
+ GError *error = NULL;
+ GSocketAddress *addr = NULL;
+ GInetAddress *iaddr = NULL;
+ GSocket *sock = NULL;
+ GCredentials *creds;
+
+ data = g_slice_new0 (IPTestData);
+ data->family = family;
+ data->server = server = g_socket_new (family,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &error);
+ if (error != NULL)
+ goto skip;
+
+ g_socket_set_blocking (server, TRUE);
+
+ iaddr = g_inet_address_new_loopback (family);
+ addr = g_inet_socket_address_new (iaddr, 0);
+
+ if (!g_socket_bind (server, addr, TRUE, &error))
+ goto skip;
+
+ if (!g_socket_listen (server, &error))
+ goto skip;
+
+ data->thread = g_thread_new ("client", client_setup_thread, data);
+
+ sock = g_socket_accept (server, NULL, &error);
+ g_assert_no_error (error);
+
+ creds = g_socket_get_credentials (sock, &error);
+ if (creds != NULL)
+ {
+ gchar *str = g_credentials_to_string (creds);
+ g_print ("Supported on this OS: %s\n", str);
+ g_free (str);
+ g_clear_object (&creds);
+ }
+ else
+ {
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED);
+ g_print ("Unsupported on this OS: %s\n", error->message);
+ g_clear_error (&error);
+ }
+
+ goto beach;
+
+skip:
+ {
+ gchar *message = g_strdup_printf ("Failed to create server: %s", error->message);
+ g_test_skip (message);
+ g_free (message);
+
+ goto beach;
+ }
+beach:
+ {
+ g_clear_error (&error);
+
+ g_clear_object (&sock);
+ g_clear_object (&addr);
+ g_clear_object (&iaddr);
+
+ g_clear_pointer (&data->thread, g_thread_join);
+ g_clear_object (&data->server);
+ g_clear_object (&data->client);
+
+ g_slice_free (IPTestData, data);
+ }
+}
+
+static gpointer
+client_setup_thread (gpointer user_data)
+{
+ IPTestData *data = user_data;
+ GSocketAddress *addr;
+ GSocket *client;
+ GError *error = NULL;
+
+ addr = g_socket_get_local_address (data->server, &error);
+ g_assert_no_error (error);
+
+ data->client = client = g_socket_new (data->family,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &error);
+ g_assert_no_error (error);
+
+ g_socket_set_blocking (client, TRUE);
+ g_socket_set_timeout (client, 1);
+
+ g_socket_connect (client, addr, NULL, &error);
+ g_assert_no_error (error);
+
+ g_object_unref (addr);
+
+ return NULL;
+}
+
+#ifdef G_OS_UNIX
+static void
+test_credentials_unix_socketpair (void)
+{
+ gint fds[2];
+ gint status;
+ GSocket *sock;
+ GError *error = NULL;
+ GCredentials *creds;
+
+ status = socketpair (PF_UNIX, SOCK_STREAM, 0, fds);
+ g_assert_cmpint (status, ==, 0);
+
+ sock = g_socket_new_from_fd (fds[0], &error);
+
+ creds = g_socket_get_credentials (sock, &error);
+ if (creds != NULL)
+ {
+ gchar *str = g_credentials_to_string (creds);
+ g_print ("Supported on this OS: %s\n", str);
+ g_free (str);
+ g_clear_object (&creds);
+ }
+ else
+ {
+ g_assert_error (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED);
+ g_print ("Unsupported on this OS: %s\n", error->message);
+ g_clear_error (&error);
+ }
+
+ g_object_unref (sock);
+ close (fds[1]);
+}
+#endif
+#endif
+
int
main (int argc,
char *argv[])
GError *error = NULL;
g_test_init (&argc, &argv, NULL);
+ g_test_bug_base ("https://bugzilla.gnome.org/");
sock = g_socket_new (G_SOCKET_FAMILY_IPV6,
G_SOCKET_TYPE_STREAM,
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/ipv4_sync/datagram/timeouts", test_ipv4_sync_dgram_timeouts);
+ g_test_add_func ("/socket/ipv6_sync/datagram", test_ipv6_sync_dgram);
+ g_test_add_func ("/socket/ipv6_sync/datagram/timeouts", test_ipv6_sync_dgram_timeouts);
#if defined (IPPROTO_IPV6) && defined (IPV6_V6ONLY)
g_test_add_func ("/socket/ipv6_v4mapped", test_ipv6_v4mapped);
#endif
g_test_add_func ("/socket/close_graceful", test_close_graceful);
g_test_add_func ("/socket/timed_wait", test_timed_wait);
+ g_test_add_func ("/socket/fd_reuse", test_fd_reuse);
g_test_add_func ("/socket/address", test_sockaddr);
#ifdef G_OS_UNIX
g_test_add_func ("/socket/unix-from-fd", test_unix_from_fd);
g_test_add_func ("/socket/unix-connection", test_unix_connection);
g_test_add_func ("/socket/unix-connection-ancillary-data", test_unix_connection_ancillary_data);
+ g_test_add_func ("/socket/source-postmortem", test_source_postmortem);
#endif
g_test_add_func ("/socket/reuse/tcp", test_reuse_tcp);
g_test_add_func ("/socket/reuse/udp", test_reuse_udp);
test_get_available);
g_test_add_data_func ("/socket/get_available/stream", GUINT_TO_POINTER (G_SOCKET_TYPE_STREAM),
test_get_available);
+ g_test_add_data_func ("/socket/read_write", GUINT_TO_POINTER (FALSE),
+ test_read_write);
+ g_test_add_data_func ("/socket/read_writev", GUINT_TO_POINTER (TRUE),
+ test_read_write);
+#if G_CREDENTIALS_SUPPORTED
+ g_test_add_func ("/socket/credentials/tcp_client", test_credentials_tcp_client);
+ g_test_add_func ("/socket/credentials/tcp_server", test_credentials_tcp_server);
+#ifdef G_OS_UNIX
+ g_test_add_func ("/socket/credentials/unix_socketpair", test_credentials_unix_socketpair);
+#endif
+#endif
return g_test_run();
}
-
projects / platform / upstream / glib.git / blobdiff