[glib] Add test apps for GSocket API

[Alexander Larsson <alexl src gnome org>]

commit 8f67f47e05a9244d86d9661caa73e5a2fc573e55
Author: Alexander Larsson <alexl redhat com>
Date:   Fri May 15 10:08:18 2009 +0200

    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 deletions(-)

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 <gio/gio.h>
+#include <glib.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+
+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 (" <hostname>[: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 <gio/gio.h>
+#include <glib.h>
+#include <stdlib.h>
+
+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;
+}