Blame gio/tests/httpd.c

Packit ae235b
#include <gio/gio.h>
Packit ae235b
#include <string.h>
Packit ae235b
Packit ae235b
static int port = 8080;
Packit ae235b
static char *root = NULL;
Packit ae235b
static GOptionEntry cmd_entries[] = {
Packit ae235b
  {"port", 'p', 0, G_OPTION_ARG_INT, &port,
Packit ae235b
   "Local port to bind to", NULL},
Packit ae235b
  {NULL}
Packit ae235b
};
Packit ae235b
Packit ae235b
static void
Packit ae235b
send_error (GOutputStream *out,
Packit ae235b
	    int error_code,
Packit ae235b
	    const char *reason)
Packit ae235b
{
Packit ae235b
  char *res;
Packit ae235b
Packit ae235b
  res = g_strdup_printf ("HTTP/1.0 %d %s\r\n\r\n"
Packit ae235b
			 "<html><head><title>%d %s</title></head>"
Packit ae235b
			 "<body>%s</body></html>",
Packit ae235b
			 error_code, reason,
Packit ae235b
			 error_code, reason,
Packit ae235b
			 reason);
Packit ae235b
  g_output_stream_write_all (out, res, strlen (res), NULL, NULL, NULL);
Packit ae235b
  g_free (res);
Packit ae235b
}
Packit ae235b
Packit ae235b
static gboolean
Packit ae235b
handler (GThreadedSocketService *service,
Packit ae235b
	 GSocketConnection      *connection,
Packit ae235b
	 GSocketListener        *listener,
Packit ae235b
	 gpointer                user_data)
Packit ae235b
{
Packit ae235b
  GOutputStream *out;
Packit ae235b
  GInputStream *in;
Packit ae235b
  GFileInputStream *file_in;
Packit ae235b
  GDataInputStream *data;
Packit ae235b
  char *line, *escaped, *tmp, *query, *unescaped, *path, *version;
Packit ae235b
  GFile *f;
Packit ae235b
  GError *error;
Packit ae235b
  GFileInfo *info;
Packit ae235b
  GString *s;
Packit ae235b
Packit ae235b
  in = g_io_stream_get_input_stream (G_IO_STREAM (connection));
Packit ae235b
  out = g_io_stream_get_output_stream (G_IO_STREAM (connection));
Packit ae235b
Packit ae235b
  data = g_data_input_stream_new (in);
Packit ae235b
  /* Be tolerant of input */
Packit ae235b
  g_data_input_stream_set_newline_type (data, G_DATA_STREAM_NEWLINE_TYPE_ANY);
Packit ae235b
Packit ae235b
  line = g_data_input_stream_read_line (data, NULL, NULL, NULL);
Packit ae235b
Packit ae235b
  if (line == NULL)
Packit ae235b
    {
Packit ae235b
      send_error (out, 400, "Invalid request");
Packit ae235b
      goto out;
Packit ae235b
    }
Packit ae235b
Packit ae235b
  if (!g_str_has_prefix (line, "GET "))
Packit ae235b
    {
Packit ae235b
      send_error (out, 501, "Only GET implemented");
Packit ae235b
      goto out;
Packit ae235b
    }
Packit ae235b
Packit ae235b
  escaped = line + 4; /* Skip "GET " */
Packit ae235b
Packit ae235b
  version = NULL;
Packit ae235b
  tmp = strchr (escaped, ' ');
Packit ae235b
  if (tmp == NULL)
Packit ae235b
    {
Packit ae235b
      send_error (out, 400, "Bad Request");
Packit ae235b
      goto out;
Packit ae235b
    }
Packit ae235b
  *tmp = 0;
Packit ae235b
Packit ae235b
  version = tmp + 1;
Packit ae235b
  if (!g_str_has_prefix (version, "HTTP/1."))
Packit ae235b
    {
Packit ae235b
      send_error(out, 505, "HTTP Version Not Supported");
Packit ae235b
      goto out;
Packit ae235b
    }
Packit ae235b
Packit ae235b
  query = strchr (escaped, '?');
Packit ae235b
  if (query != NULL)
Packit ae235b
    *query++ = 0;
Packit ae235b
Packit ae235b
  unescaped = g_uri_unescape_string (escaped, NULL);
Packit ae235b
  path = g_build_filename (root, unescaped, NULL);
Packit ae235b
  g_free (unescaped);
Packit ae235b
  f = g_file_new_for_path (path);
Packit ae235b
  g_free (path);
Packit ae235b
Packit ae235b
  error = NULL;
Packit ae235b
  file_in = g_file_read (f, NULL, &error);
Packit ae235b
  if (file_in == NULL)
Packit ae235b
    {
Packit ae235b
      send_error (out, 404, error->message);
Packit ae235b
      g_error_free (error);
Packit ae235b
      goto out;
Packit ae235b
    }
Packit ae235b
Packit ae235b
  s = g_string_new ("HTTP/1.0 200 OK\r\n");
Packit ae235b
  info = g_file_input_stream_query_info (file_in,
Packit ae235b
					 G_FILE_ATTRIBUTE_STANDARD_SIZE ","
Packit ae235b
					 G_FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE,
Packit ae235b
					 NULL, NULL);
Packit ae235b
  if (info)
Packit ae235b
    {
Packit ae235b
      const char *content_type;
Packit ae235b
      char *mime_type;
Packit ae235b
Packit ae235b
      if (g_file_info_has_attribute (info, G_FILE_ATTRIBUTE_STANDARD_SIZE))
Packit ae235b
	g_string_append_printf (s, "Content-Length: %"G_GINT64_FORMAT"\r\n",
Packit ae235b
				g_file_info_get_size (info));
Packit ae235b
      content_type = g_file_info_get_content_type (info);
Packit ae235b
      if (content_type)
Packit ae235b
	{
Packit ae235b
	  mime_type = g_content_type_get_mime_type (content_type);
Packit ae235b
	  if (mime_type)
Packit ae235b
	    {
Packit ae235b
	      g_string_append_printf (s, "Content-Type: %s\r\n",
Packit ae235b
				      mime_type);
Packit ae235b
	      g_free (mime_type);
Packit ae235b
	    }
Packit ae235b
	}
Packit ae235b
    }
Packit ae235b
  g_string_append (s, "\r\n");
Packit ae235b
Packit ae235b
  if (g_output_stream_write_all (out,
Packit ae235b
				 s->str, s->len,
Packit ae235b
				 NULL, NULL, NULL))
Packit ae235b
    {
Packit ae235b
      g_output_stream_splice (out,
Packit ae235b
			      G_INPUT_STREAM (file_in),
Packit ae235b
			      0, NULL, NULL);
Packit ae235b
    }
Packit ae235b
  g_string_free (s, TRUE);
Packit ae235b
Packit ae235b
  g_input_stream_close (G_INPUT_STREAM (file_in), NULL, NULL);
Packit ae235b
  g_object_unref (file_in);
Packit ae235b
Packit ae235b
 out:
Packit ae235b
  g_object_unref (data);
Packit ae235b
Packit ae235b
  return TRUE;
Packit ae235b
}
Packit ae235b
Packit ae235b
int
Packit ae235b
main (int argc, char *argv[])
Packit ae235b
{
Packit ae235b
  GSocketService *service;
Packit ae235b
  GOptionContext *context;
Packit ae235b
  GError *error = NULL;
Packit ae235b
Packit ae235b
  context = g_option_context_new ("<http root dir> - Simple HTTP server");
Packit ae235b
  g_option_context_add_main_entries (context, cmd_entries, NULL);
Packit ae235b
  if (!g_option_context_parse (context, &argc, &argv, &error))
Packit ae235b
    {
Packit ae235b
      g_printerr ("%s: %s\n", argv[0], error->message);
Packit ae235b
      return 1;
Packit ae235b
    }
Packit ae235b
Packit ae235b
  if (argc != 2)
Packit ae235b
    {
Packit ae235b
      g_printerr ("Root directory not specified\n");
Packit ae235b
      return 1;
Packit ae235b
    }
Packit ae235b
Packit ae235b
  root = g_strdup (argv[1]);
Packit ae235b
Packit ae235b
  service = g_threaded_socket_service_new (10);
Packit ae235b
  if (!g_socket_listener_add_inet_port (G_SOCKET_LISTENER (service),
Packit ae235b
					port,
Packit ae235b
					NULL,
Packit ae235b
					&error))
Packit ae235b
    {
Packit ae235b
      g_printerr ("%s: %s\n", argv[0], error->message);
Packit ae235b
      return 1;
Packit ae235b
    }
Packit ae235b
Packit ae235b
  g_print ("Http server listening on port %d\n", port);
Packit ae235b
Packit ae235b
  g_signal_connect (service, "run", G_CALLBACK (handler), NULL);
Packit ae235b
Packit ae235b
  g_main_loop_run (g_main_loop_new (NULL, FALSE));
Packit ae235b
  g_assert_not_reached ();
Packit ae235b
}