From 8f67f47e05a9244d86d9661caa73e5a2fc573e55 Mon Sep 17 00:00:00 2001 From: Alexander Larsson Date: Fri, 15 May 2009 10:08:18 +0200 Subject: [PATCH] Add test apps for GSocket API --- gio/tests/Makefile.am | 10 +- gio/tests/socket-client.c | 294 ++++++++++++++++++++++++++++++++++++++++++++ gio/tests/socket-server.c | 304 ++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 607 insertions(+), 1 deletion(-) create mode 100644 gio/tests/socket-client.c create mode 100644 gio/tests/socket-server.c diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am index f327443..977e1f2 100644 --- a/gio/tests/Makefile.am +++ b/gio/tests/Makefile.am @@ -30,7 +30,7 @@ TEST_PROGS += \ filter-streams \ simple-async-result -SAMPLE_PROGS = resolver +SAMPLE_PROGS = resolver socket-server socket-client if OS_UNIX TEST_PROGS += live-g-file unix-streams desktop-app-info @@ -86,4 +86,12 @@ resolver_SOURCES = resolver.c resolver_LDADD = $(progs_ldadd) \ $(top_builddir)/gthread/libgthread-2.0.la +socket_server_SOURCES = socket-server.c +socket_server_LDADD = $(progs_ldadd) \ + $(top_builddir)/gthread/libgthread-2.0.la + +socket_client_SOURCES = socket-client.c +socket_client_LDADD = $(progs_ldadd) \ + $(top_builddir)/gthread/libgthread-2.0.la + DISTCLEAN_FILES = applications/mimeinfo.cache diff --git a/gio/tests/socket-client.c b/gio/tests/socket-client.c new file mode 100644 index 0000000..abc28af --- /dev/null +++ b/gio/tests/socket-client.c @@ -0,0 +1,294 @@ +#include +#include +#include +#include +#include + +GMainLoop *loop; + +gboolean verbose = FALSE; +gboolean non_blocking = FALSE; +gboolean use_udp = FALSE; +gboolean use_source = FALSE; +int cancel_timeout = 0; + +static GOptionEntry cmd_entries[] = { + {"cancel", 'c', 0, G_OPTION_ARG_INT, &cancel_timeout, + "Cancel any op after the specified amount of seconds", NULL}, + {"udp", 'u', 0, G_OPTION_ARG_NONE, &use_udp, + "Use udp instead of tcp", NULL}, + {"verbose", 'v', 0, G_OPTION_ARG_NONE, &verbose, + "Be verbose", NULL}, + {"non-blocking", 'n', 0, G_OPTION_ARG_NONE, &non_blocking, + "Enable non-blocking i/o", NULL}, + {"use-source", 's', 0, G_OPTION_ARG_NONE, &use_source, + "Use GSource to wait for non-blocking i/o", NULL}, + {NULL} +}; + +static char * +socket_address_to_string (GSocketAddress *address) +{ + GInetAddress *inet_address; + char *str, *res; + int port; + + inet_address = g_inet_socket_address_get_address (G_INET_SOCKET_ADDRESS (address)); + str = g_inet_address_to_string (inet_address); + port = g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (address)); + res = g_strdup_printf ("%s:%d", str, port); + g_free (str); + return res; +} + +static gboolean +source_ready (gpointer data, + GIOCondition condition) +{ + g_main_loop_quit (loop); + return FALSE; +} + +static void +ensure_condition (GSocket *socket, + const char *where, + GCancellable *cancellable, + GIOCondition condition) +{ + GError *error = NULL; + GSource *source; + + if (!non_blocking) + return; + + if (use_source) + { + source = g_socket_create_source (socket, + condition, + cancellable); + g_source_set_callback (source, + (GSourceFunc) source_ready, + NULL, NULL); + g_source_attach (source, NULL); + g_source_unref (source); + g_main_loop_run (loop); + } + else + { + if (!g_socket_condition_wait (socket, condition, cancellable, &error)) + { + g_printerr ("condition wait error for %s: %s\n", + where, + error->message); + exit (1); + } + } +} + +static gpointer +cancel_thread (gpointer data) +{ + GCancellable *cancellable = data; + + g_usleep (1000*1000*cancel_timeout); + g_print ("Cancelling\n"); + g_cancellable_cancel (cancellable); + return NULL; +} + +int +main (int argc, + char *argv[]) +{ + GSocket *socket; + GSocketAddress *src_address; + GSocketAddress *address; + GSocketType socket_type; + GError *error = NULL; + GOptionContext *context; + GCancellable *cancellable; + GSocketAddressEnumerator *enumerator; + GSocketConnectable *connectable; + + g_thread_init (NULL); + + g_type_init (); + + context = g_option_context_new (" [:port] - Test GSocket client stuff"); + g_option_context_add_main_entries (context, cmd_entries, NULL); + if (!g_option_context_parse (context, &argc, &argv, &error)) + { + g_printerr ("%s: %s\n", argv[0], error->message); + return 1; + } + + if (argc != 2) + { + g_printerr ("%s: %s\n", argv[0], "Need to specify hostname"); + return 1; + } + + if (cancel_timeout) + { + cancellable = g_cancellable_new (); + g_thread_create (cancel_thread, cancellable, FALSE, NULL); + } + else + { + cancellable = NULL; + } + + loop = g_main_loop_new (NULL, FALSE); + + if (use_udp) + socket_type = G_SOCKET_TYPE_DATAGRAM; + else + socket_type = G_SOCKET_TYPE_STREAM; + + socket = g_socket_new (G_SOCKET_FAMILY_IPV4, socket_type, 0, &error); + if (socket == NULL) + { + g_printerr ("%s: %s\n", argv[0], error->message); + return 1; + } + + connectable = g_network_address_parse (argv[1], 7777, &error); + if (connectable == NULL) + { + g_printerr ("%s: %s\n", argv[0], error->message); + return 1; + } + + enumerator = g_socket_connectable_enumerate (connectable); + while (TRUE) + { + address = g_socket_address_enumerator_next (enumerator, cancellable, &error); + if (address == NULL) + { + if (error == NULL) + g_printerr ("%s: No more addresses to try\n", argv[0]); + else + g_printerr ("%s: %s\n", argv[0], error->message); + return 1; + } + + if (g_socket_connect (socket, address, &error)) + break; + g_printerr ("%s: Connection to %s failed: %s, trying next\n", argv[0], socket_address_to_string (address), error->message); + g_error_free (error); + error = NULL; + + g_object_unref (address); + } + g_object_unref (enumerator); + + g_print ("Connected to %s\n", + socket_address_to_string (address)); + + /* TODO: Test non-blocking connect */ + if (non_blocking) + g_socket_set_blocking (socket, FALSE); + + src_address = g_socket_get_local_address (socket, &error); + if (!src_address) + { + g_printerr ("Error getting local address: %s\n", + error->message); + return 1; + } + g_print ("local address: %s\n", + socket_address_to_string (src_address)); + g_object_unref (src_address); + + while (TRUE) + { + gchar buffer[4096] = { }; + gssize size; + gsize to_send; + + if (fgets (buffer, sizeof buffer, stdin) == NULL) + break; + + to_send = strlen (buffer); + while (to_send > 0) + { + ensure_condition (socket, "send", cancellable, G_IO_OUT); + if (use_udp) + size = g_socket_send_to (socket, address, + buffer, to_send, &error); + else + size = g_socket_send (socket, buffer, to_send, &error); + + if (size < 0) + { + if (g_error_matches (error, + G_IO_ERROR, + G_IO_ERROR_WOULD_BLOCK)) + { + g_print ("socket send would block, handling\n"); + g_error_free (error); + error = NULL; + continue; + } + else + { + g_printerr ("Error sending to socket: %s\n", + error->message); + return 1; + } + } + + g_print ("sent %" G_GSSIZE_FORMAT " bytes of data\n", size); + + if (size == 0) + { + g_printerr ("Unexpected short write\n"); + return 1; + } + + to_send -= size; + } + + ensure_condition (socket, "receive", cancellable, G_IO_IN); + if (use_udp) + size = g_socket_receive_from (socket, &src_address, + buffer, sizeof buffer, &error); + else + size = g_socket_receive (socket, buffer, sizeof buffer, &error); + + if (size < 0) + { + g_printerr ("Error receiving from socket: %s\n", + error->message); + return 1; + } + + if (size == 0) + break; + + g_print ("received %" G_GSSIZE_FORMAT " bytes of data", size); + if (use_udp) + g_print (" from %s", socket_address_to_string (src_address)); + g_print ("\n"); + + if (verbose) + g_print ("-------------------------\n" + "%.*s" + "-------------------------\n", + (int)size, buffer); + + } + + g_print ("closing socket\n"); + + if (!g_socket_close (socket, &error)) + { + g_printerr ("Error closing master socket: %s\n", + error->message); + return 1; + } + + g_object_unref (G_OBJECT (socket)); + + return 0; +} diff --git a/gio/tests/socket-server.c b/gio/tests/socket-server.c new file mode 100644 index 0000000..0e7d1ad --- /dev/null +++ b/gio/tests/socket-server.c @@ -0,0 +1,304 @@ +#include +#include +#include + +GMainLoop *loop; + +int port = 7777; +gboolean verbose = FALSE; +gboolean dont_reuse_address = FALSE; +gboolean non_blocking = FALSE; +gboolean use_udp = FALSE; +gboolean use_source = FALSE; +int cancel_timeout = 0; + +static GOptionEntry cmd_entries[] = { + {"port", 'p', 0, G_OPTION_ARG_INT, &port, + "Local port to bind to", NULL}, + {"cancel", 'c', 0, G_OPTION_ARG_INT, &cancel_timeout, + "Cancel any op after the specified amount of seconds", NULL}, + {"udp", 'u', 0, G_OPTION_ARG_NONE, &use_udp, + "Use udp instead of tcp", NULL}, + {"verbose", 'v', 0, G_OPTION_ARG_NONE, &verbose, + "Be verbose", NULL}, + {"no-reuse", 0, 0, G_OPTION_ARG_NONE, &dont_reuse_address, + "Don't SOADDRREUSE", NULL}, + {"non-blocking", 'n', 0, G_OPTION_ARG_NONE, &non_blocking, + "Enable non-blocking i/o", NULL}, + {"use-source", 's', 0, G_OPTION_ARG_NONE, &use_source, + "Use GSource to wait for non-blocking i/o", NULL}, + {NULL} +}; + +static char * +socket_address_to_string (GSocketAddress *address) +{ + GInetAddress *inet_address; + char *str, *res; + int port; + + inet_address = g_inet_socket_address_get_address (G_INET_SOCKET_ADDRESS (address)); + str = g_inet_address_to_string (inet_address); + port = g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (address)); + res = g_strdup_printf ("%s:%d", str, port); + g_free (str); + return res; +} + +static gboolean +source_ready (gpointer data, + GIOCondition condition) +{ + g_main_loop_quit (loop); + return FALSE; +} + +static void +ensure_condition (GSocket *socket, + const char *where, + GCancellable *cancellable, + GIOCondition condition) +{ + GError *error = NULL; + GSource *source; + + if (!non_blocking) + return; + + if (use_source) + { + source = g_socket_create_source (socket, + condition, + cancellable); + g_source_set_callback (source, + (GSourceFunc) source_ready, + NULL, NULL); + g_source_attach (source, NULL); + g_source_unref (source); + g_main_loop_run (loop); + } + else + { + if (!g_socket_condition_wait (socket, condition, cancellable, &error)) + { + g_printerr ("condition wait error for %s: %s\n", + where, + error->message); + exit (1); + } + } +} + +static gpointer +cancel_thread (gpointer data) +{ + GCancellable *cancellable = data; + + g_usleep (1000*1000*cancel_timeout); + g_print ("Cancelling\n"); + g_cancellable_cancel (cancellable); + return NULL; +} + +int +main (int argc, + char *argv[]) +{ + GSocket *socket, *new_socket, *recv_socket; + GSocketAddress *src_address; + GSocketAddress *address; + GSocketType socket_type; + GError *error = NULL; + GOptionContext *context; + GCancellable *cancellable; + + g_thread_init (NULL); + + g_type_init (); + + context = g_option_context_new (" - Test GSocket server stuff"); + g_option_context_add_main_entries (context, cmd_entries, NULL); + if (!g_option_context_parse (context, &argc, &argv, &error)) + { + g_printerr ("%s: %s\n", argv[0], error->message); + return 1; + } + + if (cancel_timeout) + { + cancellable = g_cancellable_new (); + g_thread_create (cancel_thread, cancellable, FALSE, NULL); + } + else + { + cancellable = NULL; + } + + loop = g_main_loop_new (NULL, FALSE); + + if (use_udp) + socket_type = G_SOCKET_TYPE_DATAGRAM; + else + socket_type = G_SOCKET_TYPE_STREAM; + + socket = g_socket_new (G_SOCKET_FAMILY_IPV4, socket_type, 0, &error); + + if (socket == NULL) + { + g_printerr ("%s: %s\n", argv[0], error->message); + return 1; + } + + if (non_blocking) + g_socket_set_blocking (socket, FALSE); + + src_address = g_inet_socket_address_new (g_inet_address_new_any (G_SOCKET_FAMILY_IPV4), port); + if (!g_socket_bind (socket, src_address, !dont_reuse_address, &error)) + { + g_printerr ("Can't bind socket: %s\n", error->message); + return 1; + } + + if (!use_udp) + { + if (!g_socket_listen (socket, &error)) + { + g_printerr ("Can't listen on socket: %s\n", error->message); + return 1; + } + + g_print ("listening on port %d...\n", port); + + ensure_condition (socket, "accept", cancellable, G_IO_IN); + new_socket = g_socket_accept (socket, &error); + if (!new_socket) + { + g_printerr ("Error accepting socket: %s\n", + error->message); + return 1; + } + + if (non_blocking) + g_socket_set_blocking (new_socket, FALSE); + + address = g_socket_get_remote_address (new_socket, &error); + if (!address) + { + g_printerr ("Error getting remote address: %s\n", + error->message); + return 1; + } + + g_print ("got a new connection from %s\n", + socket_address_to_string (address)); + + recv_socket = new_socket; + } + else + { + recv_socket = socket; + new_socket = NULL; + } + + + while (TRUE) + { + gchar buffer[4096] = { }; + gssize size; + gsize to_send; + + ensure_condition (recv_socket, "receive", cancellable, G_IO_IN); + if (use_udp) + size = g_socket_receive_from (recv_socket, &address, + buffer, sizeof buffer, &error); + else + size = g_socket_receive (recv_socket, buffer, sizeof buffer, &error); + + if (size < 0) + { + g_printerr ("Error receiving from socket: %s\n", + error->message); + return 1; + } + + if (size == 0) + break; + + g_print ("received %" G_GSSIZE_FORMAT " bytes of data", size); + if (use_udp) + g_print (" from %s", socket_address_to_string (address)); + g_print ("\n"); + + if (verbose) + g_print ("-------------------------\n" + "%.*s\n" + "-------------------------\n", + (int)size, buffer); + + to_send = size; + + while (to_send > 0) + { + ensure_condition (recv_socket, "send", cancellable, G_IO_OUT); + if (use_udp) + size = g_socket_send_to (recv_socket, address, + buffer, to_send, &error); + else + size = g_socket_send (recv_socket, buffer, to_send, &error); + + if (size < 0) + { + if (g_error_matches (error, + G_IO_ERROR, + G_IO_ERROR_WOULD_BLOCK)) + { + g_print ("socket send would block, handling\n"); + g_error_free (error); + error = NULL; + continue; + } + else + { + g_printerr ("Error sending to socket: %s\n", + error->message); + return 1; + } + } + + g_print ("sent %" G_GSSIZE_FORMAT " bytes of data\n", size); + + if (size == 0) + { + g_printerr ("Unexpected short write\n"); + return 1; + } + + to_send -= size; + } + } + + g_print ("connection closed\n"); + + if (new_socket) + { + if (!g_socket_close (new_socket, &error)) + { + g_printerr ("Error closing connection socket: %s\n", + error->message); + return 1; + } + + g_object_unref (G_OBJECT (new_socket)); + } + + if (!g_socket_close (socket, &error)) + { + g_printerr ("Error closing master socket: %s\n", + error->message); + return 1; + } + + g_object_unref (G_OBJECT (socket)); + + return 0; +} -- 2.7.4