gio/tests/httpd.c

   1 #include <gio/gio.h>
   2 #include <string.h>
   3
   4 static int port = 8080;
   5 static char *root = NULL;
   6 static GOptionEntry cmd_entries[] = {
   7   {"port", 'p', 0, G_OPTION_ARG_INT, &port,
   8    "Local port to bind to", NULL},
   9   {NULL}
  10 };
  11
  12 static void
  13 send_error (GOutputStream *out,
  14             int error_code,
  15             const char *reason)
  16 {
  17   char *res;
  18
  19   res = g_strdup_printf ("HTTP/1.0 %d %s\r\n\r\n"
  20                          "<html><head><title>%d %s</title></head>"
  21                          "<body>%s</body></html>",
  22                          error_code, reason,
  23                          error_code, reason,
  24                          reason);
  25   g_output_stream_write_all (out, res, strlen (res), NULL, NULL, NULL);
  26   g_free (res);
  27 }
  28
  29 static gboolean
  30 handler (GThreadedSocketService *service,
  31          GSocketConnection      *connection,
  32          GSocketListener        *listener,
  33          gpointer                user_data)
  34 {
  35   GOutputStream *out;
  36   GInputStream *in;
  37   GFileInputStream *file_in;
  38   GDataInputStream *data;
  39   char *line, *escaped, *tmp, *query, *unescaped, *path, *version;
  40   GFile *f;
  41   GError *error;
  42   GFileInfo *info;
  43   GString *s;
  44
  45   in = g_io_stream_get_input_stream (G_IO_STREAM (connection));
  46   out = g_io_stream_get_output_stream (G_IO_STREAM (connection));
  47
  48   data = g_data_input_stream_new (in);
  49   /* Be tolerant of input */
  50   g_data_input_stream_set_newline_type (data, G_DATA_STREAM_NEWLINE_TYPE_ANY);
  51
  52   line = g_data_input_stream_read_line (data, NULL, NULL, NULL);
  53
  54   if (line == NULL)
  55     {
  56       send_error (out, 400, "Invalid request");
  57       goto out;
  58     }
  59
  60   if (!g_str_has_prefix (line, "GET "))
  61     {
  62       send_error (out, 501, "Only GET implemented");
  63       goto out;
  64     }
  65
  66   escaped = line + 4; /* Skip "GET " */
  67
  68   version = NULL;
  69   tmp = strchr (escaped, ' ');
  70   if (tmp != NULL)
  71     {
  72       *tmp = 0;
  73       version = tmp + 1;
  74     }
  75   version = version; /* To avoid -Wunused-but-set-variable */
  76
  77   query = strchr (escaped, '?');
  78   if (query != NULL)
  79     *query++ = 0;
  80
  81   unescaped = g_uri_unescape_string (escaped, NULL);
  82   path = g_build_filename (root, unescaped, NULL);
  83   g_free (unescaped);
  84   f = g_file_new_for_path (path);
  85   g_free (path);
  86
  87   error = NULL;
  88   file_in = g_file_read (f, NULL, &error);
  89   if (file_in == NULL)
  90     {
  91       send_error (out, 404, error->message);
  92       g_error_free (error);
  93       goto out;
  94     }
  95
  96   s = g_string_new ("HTTP/1.0 200 OK\r\n");
  97   info = g_file_input_stream_query_info (file_in,
  98                                          G_FILE_ATTRIBUTE_STANDARD_SIZE ","
  99                                          G_FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE,
 100                                          NULL, NULL);
 101   if (info)
 102     {
 103       const char *content_type;
 104       char *mime_type;
 105
 106       if (g_file_info_has_attribute (info, G_FILE_ATTRIBUTE_STANDARD_SIZE))
 107         g_string_append_printf (s, "Content-Length: %"G_GINT64_FORMAT"\r\n",
 108                                 g_file_info_get_size (info));
 109       content_type = g_file_info_get_content_type (info);
 110       if (content_type)
 111         {
 112           mime_type = g_content_type_get_mime_type (content_type);
 113           if (mime_type)
 114             {
 115               g_string_append_printf (s, "Content-Type: %s\r\n",
 116                                       mime_type);
 117               g_free (mime_type);
 118             }
 119         }
 120     }
 121   g_string_append (s, "\r\n");
 122
 123   if (g_output_stream_write_all (out,
 124                                  s->str, s->len,
 125                                  NULL, NULL, NULL))
 126     {
 127       g_output_stream_splice (out,
 128                               G_INPUT_STREAM (file_in),
 129                               0, NULL, NULL);
 130     }
 131   g_string_free (s, TRUE);
 132
 133   g_input_stream_close (G_INPUT_STREAM (file_in), NULL, NULL);
 134   g_object_unref (file_in);
 135
 136  out:
 137   g_object_unref (data);
 138
 139   return TRUE;
 140 }
 141
 142 int
 143 main (int argc, char *argv[])
 144 {
 145   GSocketService *service;
 146   GOptionContext *context;
 147   GError *error = NULL;
 148
 149   g_type_init ();
 150
 151   context = g_option_context_new ("<http root dir> - Simple HTTP server");
 152   g_option_context_add_main_entries (context, cmd_entries, NULL);
 153   if (!g_option_context_parse (context, &argc, &argv, &error))
 154     {
 155       g_printerr ("%s: %s\n", argv[0], error->message);
 156       return 1;
 157     }
 158
 159   if (argc != 2)
 160     {
 161       g_printerr ("Root directory not specified\n");
 162       return 1;
 163     }
 164
 165   root = g_strdup (argv[1]);
 166
 167   service = g_threaded_socket_service_new (10);
 168   if (!g_socket_listener_add_inet_port (G_SOCKET_LISTENER (service),
 169                                         port,
 170                                         NULL,
 171                                         &error))
 172     {
 173       g_printerr ("%s: %s\n", argv[0], error->message);
 174       return 1;
 175     }
 176
 177   g_print ("Http server listening on port %d\n", port);
 178
 179   g_signal_connect (service, "run", G_CALLBACK (handler), NULL);
 180
 181   g_main_loop_run (g_main_loop_new (NULL, FALSE));
 182   g_assert_not_reached ();
 183 }