GNOME.org
 

[glib] Add test apps for highlevel socket classes

commit 34e74378c9fc99a3dc75eb9680bb00cb784029bb
Author: Alexander Larsson <alexl redhat com>
Date:   Fri May 15 21:34:14 2009 +0200

    Add test apps for highlevel socket classes
    
    echo-server - simple echo server
    httpd - simple http server
---
 gio/tests/Makefile.am   |   10 +++-
 gio/tests/echo-server.c |   73 +++++++++++++++++++
 gio/tests/httpd.c       |  183 +++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 265 insertions(+), 1 deletions(-)

diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am
index 977e1f2..7a20869 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 socket-server socket-client
+SAMPLE_PROGS = resolver socket-server socket-client echo-server httpd
 
 if OS_UNIX
 TEST_PROGS += live-g-file unix-streams desktop-app-info
@@ -94,4 +94,12 @@ socket_client_SOURCES	  = socket-client.c
 socket_client_LDADD	  = $(progs_ldadd) \
 	$(top_builddir)/gthread/libgthread-2.0.la
 
+echo_server_SOURCES	  = echo-server.c
+echo_server_LDADD	  = $(progs_ldadd) \
+	$(top_builddir)/gthread/libgthread-2.0.la
+
+httpd_SOURCES		  = httpd.c
+httpd_LDADD		  = $(progs_ldadd) \
+	$(top_builddir)/gthread/libgthread-2.0.la
+
 DISTCLEAN_FILES = applications/mimeinfo.cache
diff --git a/gio/tests/echo-server.c b/gio/tests/echo-server.c
new file mode 100644
index 0000000..1aa3166
--- /dev/null
+++ b/gio/tests/echo-server.c
@@ -0,0 +1,73 @@
+#include <gio/gio.h>
+#include <string.h>
+
+#define MESSAGE "Welcome to the echo service!\n"
+
+int port = 7777;
+static GOptionEntry cmd_entries[] = {
+  {"port", 'p', 0, G_OPTION_ARG_INT, &port,
+   "Local port to bind to", NULL},
+  {NULL}
+};
+
+
+static gboolean
+handler (GThreadedSocketService *service,
+         GSocketConnection      *connection,
+         GSocketListener        *listener,
+         gpointer                user_data)
+{
+  GOutputStream *out;
+  GInputStream *in;
+  char buffer[1024];
+  gssize size;
+
+  out = g_io_stream_get_output_stream (G_IO_STREAM (connection));
+  in = g_io_stream_get_input_stream (G_IO_STREAM (connection));
+
+  g_output_stream_write_all (out, MESSAGE, strlen (MESSAGE),
+                             NULL, NULL, NULL);
+
+  while (0 < (size = g_input_stream_read (in, buffer,
+                                          sizeof buffer, NULL, NULL)))
+    g_output_stream_write (out, buffer, size, NULL, NULL);
+
+  return TRUE;
+}
+
+int
+main (int argc, char *argv[])
+{
+  GSocketService *service;
+  GOptionContext *context;
+  GError *error = NULL;
+
+  g_type_init ();
+  g_thread_init (NULL);
+
+  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;
+    }
+
+  service = g_threaded_socket_service_new ();
+
+  if (!g_socket_listener_add_inet_port (G_SOCKET_LISTENER (service),
+					port,
+					NULL,
+					&error))
+    {
+      g_printerr ("%s: %s\n", argv[0], error->message);
+      return 1;
+    }
+
+  g_print ("Echo service listening on port %d\n", port);
+
+  g_signal_connect (service, "run", G_CALLBACK (handler), NULL);
+
+  g_main_loop_run (g_main_loop_new (NULL, FALSE));
+  g_assert_not_reached ();
+}
diff --git a/gio/tests/httpd.c b/gio/tests/httpd.c
new file mode 100644
index 0000000..6ad0b43
--- /dev/null
+++ b/gio/tests/httpd.c
@@ -0,0 +1,183 @@
+#include <gio/gio.h>
+#include <string.h>
+
+static int port = 8080;
+static char *root = NULL;
+static GOptionEntry cmd_entries[] = {
+  {"port", 'p', 0, G_OPTION_ARG_INT, &port,
+   "Local port to bind to", NULL},
+  {NULL}
+};
+
+static void
+send_error (GOutputStream *out,
+	    int error_code,
+	    char *reason)
+{
+  char *res;
+
+  res = g_strdup_printf ("HTTP/1.0 %d %s\r\n\r\n"
+			 "<html><head><title>%d %s</title></head>"
+			 "<body>%s</body></html>",
+			 error_code, reason,
+			 error_code, reason,
+			 reason);
+  g_output_stream_write_all (out, res, strlen (res), NULL, NULL, NULL);
+  g_free (res);
+}
+
+static gboolean
+handler (GThreadedSocketService *service,
+	 GSocketConnection      *connection,
+	 GSocketListener        *listener,
+	 gpointer                user_data)
+{
+  GOutputStream *out;
+  GInputStream *in;
+  GFileInputStream *file_in;
+  GDataInputStream *data;
+  char *line, *escaped, *tmp, *query, *unescaped, *path, *version;
+  GFile *f;
+  GError *error;
+  GFileInfo *info;
+  GString *s;
+
+  in = g_io_stream_get_input_stream (G_IO_STREAM (connection));
+  out = g_io_stream_get_output_stream (G_IO_STREAM (connection));
+
+  data = g_data_input_stream_new (in);
+  /* Be tolerant of input */
+  g_data_input_stream_set_newline_type (data, G_DATA_STREAM_NEWLINE_TYPE_ANY);
+
+  line = g_data_input_stream_read_line (data, NULL, NULL, NULL);
+
+  if (line == NULL)
+    {
+      send_error (out, 400, "Invalid request");
+      goto out;
+    }
+
+  if (!g_str_has_prefix (line, "GET "))
+    {
+      send_error (out, 501, "Only GET implemented");
+      goto out;
+    }
+
+  escaped = line + 4; /* Skip "GET " */
+
+  version = NULL;
+  tmp = strchr (escaped, ' ');
+  if (tmp != NULL)
+    {
+      *tmp = 0;
+      version = tmp + 1;
+    }
+
+  query = strchr (escaped, '?');
+  if (query != NULL)
+    *query++ = 0;
+
+  unescaped = g_uri_unescape_string (escaped, NULL);
+  path = g_build_filename (root, unescaped, NULL);
+  g_free (unescaped);
+  f = g_file_new_for_path (path);
+  g_free (path);
+
+  error = NULL;
+  file_in = g_file_read (f, NULL, &error);
+  if (file_in == NULL)
+    {
+      send_error (out, 404, error->message);
+      g_error_free (error);
+      goto out;
+    }
+
+  s = g_string_new ("HTTP/1.0 200 OK\r\n");
+  info = g_file_input_stream_query_info (file_in,
+					 G_FILE_ATTRIBUTE_STANDARD_SIZE ","
+					 G_FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE,
+					 NULL, NULL);
+  if (info)
+    {
+      const char *content_type;
+      char *mime_type;
+
+      if (g_file_info_has_attribute (info, G_FILE_ATTRIBUTE_STANDARD_SIZE))
+	g_string_append_printf (s, "Content-Length: %"G_GINT64_FORMAT"\r\n",
+				g_file_info_get_size (info));
+      content_type = g_file_info_get_content_type (info);
+      if (content_type)
+	{
+	  mime_type = g_content_type_get_mime_type (content_type);
+	  if (mime_type)
+	    {
+	      g_string_append_printf (s, "Content-Type: %s\r\n",
+				      mime_type);
+	      g_free (mime_type);
+	    }
+	}
+    }
+  g_string_append (s, "\r\n");
+
+  if (g_output_stream_write_all (out,
+				 s->str, s->len,
+				 NULL, NULL, NULL))
+    {
+      g_output_stream_splice (out,
+			      G_INPUT_STREAM (file_in),
+			      0, NULL, NULL);
+    }
+  g_string_free (s, TRUE);
+
+  g_input_stream_close (G_INPUT_STREAM (file_in), NULL, NULL);
+  g_object_unref (file_in);
+
+ out:
+  g_object_unref (data);
+
+  return TRUE;
+}
+
+int
+main (int argc, char *argv[])
+{
+  GSocketService *service;
+  GOptionContext *context;
+  GError *error = NULL;
+
+  g_type_init ();
+  g_thread_init (NULL);
+
+  context = g_option_context_new ("<http root dir> - Simple HTTP server");
+  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 ("Root directory not specified\n");
+      return 1;
+    }
+
+  root = g_strdup (argv[1]);
+
+  service = g_threaded_socket_service_new ();
+  if (!g_socket_listener_add_inet_port (G_SOCKET_LISTENER (service),
+					port,
+					NULL,
+					&error))
+    {
+      g_printerr ("%s: %s\n", argv[0], error->message);
+      return 1;
+    }
+
+  g_print ("Http server listening on port %d\n", port);
+
+  g_signal_connect (service, "run", G_CALLBACK (handler), NULL);
+
+  g_main_loop_run (g_main_loop_new (NULL, FALSE));
+  g_assert_not_reached ();
+}
[Date Prev][Date Next]   [Thread Prev][Thread Next]   [Thread Index] [Date Index] [Author Index]