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   G_OPTION_ENTRY_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       send_error (out, 400, "Bad Request");
  73       goto out;
  74     }
  75   *tmp = 0;
  76
  77   version = tmp + 1;
  78   if (!g_str_has_prefix (version, "HTTP/1."))
  79     {
  80       send_error(out, 505, "HTTP Version Not Supported");
  81       goto out;
  82     }
  83
  84   query = strchr (escaped, '?');
  85   if (query != NULL)
  86     *query++ = 0;
  87
  88   unescaped = g_uri_unescape_string (escaped, NULL);
  89   path = g_build_filename (root, unescaped, NULL);
  90   g_free (unescaped);
  91   f = g_file_new_for_path (path);
  92   g_free (path);
  93
  94   error = NULL;
  95   file_in = g_file_read (f, NULL, &error);
  96   if (file_in == NULL)
  97     {
  98       send_error (out, 404, error->message);
  99       g_error_free (error);
 100       goto out;
 101     }
 102
 103   s = g_string_new ("HTTP/1.0 200 OK\r\n");
 104   info = g_file_input_stream_query_info (file_in,
 105                                          G_FILE_ATTRIBUTE_STANDARD_SIZE ","
 106                                          G_FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE,
 107                                          NULL, NULL);
 108   if (info)
 109     {
 110       const char *content_type;
 111       char *mime_type;
 112
 113       if (g_file_info_has_attribute (info, G_FILE_ATTRIBUTE_STANDARD_SIZE))
 114         g_string_append_printf (s, "Content-Length: %"G_GINT64_FORMAT"\r\n",
 115                                 g_file_info_get_size (info));
 116       content_type = g_file_info_get_content_type (info);
 117       if (content_type)
 118         {
 119           mime_type = g_content_type_get_mime_type (content_type);
 120           if (mime_type)
 121             {
 122               g_string_append_printf (s, "Content-Type: %s\r\n",
 123                                       mime_type);
 124               g_free (mime_type);
 125             }
 126         }
 127     }
 128   g_string_append (s, "\r\n");
 129
 130   if (g_output_stream_write_all (out,
 131                                  s->str, s->len,
 132                                  NULL, NULL, NULL))
 133     {
 134       g_output_stream_splice (out,
 135                               G_INPUT_STREAM (file_in),
 136                               0, NULL, NULL);
 137     }
 138   g_string_free (s, TRUE);
 139
 140   g_input_stream_close (G_INPUT_STREAM (file_in), NULL, NULL);
 141   g_object_unref (file_in);
 142
 143  out:
 144   g_object_unref (data);
 145
 146   return TRUE;
 147 }
 148
 149 int
 150 main (int argc, char *argv[])
 151 {
 152   GSocketService *service;
 153   GOptionContext *context;
 154   GError *error = NULL;
 155
 156   context = g_option_context_new ("<http root dir> - Simple HTTP server");
 157   g_option_context_add_main_entries (context, cmd_entries, NULL);
 158   if (!g_option_context_parse (context, &argc, &argv, &error))
 159     {
 160       g_printerr ("%s: %s\n", argv[0], error->message);
 161       return 1;
 162     }
 163
 164   if (argc != 2)
 165     {
 166       g_printerr ("Root directory not specified\n");
 167       return 1;
 168     }
 169
 170   root = g_strdup (argv[1]);
 171
 172   service = g_threaded_socket_service_new (10);
 173   if (!g_socket_listener_add_inet_port (G_SOCKET_LISTENER (service),
 174                                         port,
 175                                         NULL,
 176                                         &error))
 177     {
 178       g_printerr ("%s: %s\n", argv[0], error->message);
 179       return 1;
 180     }
 181
 182   g_print ("Http server listening on port %d\n", port);
 183
 184   g_signal_connect (service, "run", G_CALLBACK (handler), NULL);
 185
 186   g_main_loop_run (g_main_loop_new (NULL, FALSE));
 187   g_assert_not_reached ();
 188 }