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
  76   query = strchr (escaped, '?');
  77   if (query != NULL)
  78     *query++ = 0;
  79
  80   unescaped = g_uri_unescape_string (escaped, NULL);
  81   path = g_build_filename (root, unescaped, NULL);
  82   g_free (unescaped);
  83   f = g_file_new_for_path (path);
  84   g_free (path);
  85
  86   error = NULL;
  87   file_in = g_file_read (f, NULL, &error);
  88   if (file_in == NULL)
  89     {
  90       send_error (out, 404, error->message);
  91       g_error_free (error);
  92       goto out;
  93     }
  94
  95   s = g_string_new ("HTTP/1.0 200 OK\r\n");
  96   info = g_file_input_stream_query_info (file_in,
  97                                          G_FILE_ATTRIBUTE_STANDARD_SIZE ","
  98                                          G_FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE,
  99                                          NULL, NULL);
 100   if (info)
 101     {
 102       const char *content_type;
 103       char *mime_type;
 104
 105       if (g_file_info_has_attribute (info, G_FILE_ATTRIBUTE_STANDARD_SIZE))
 106         g_string_append_printf (s, "Content-Length: %"G_GINT64_FORMAT"\r\n",
 107                                 g_file_info_get_size (info));
 108       content_type = g_file_info_get_content_type (info);
 109       if (content_type)
 110         {
 111           mime_type = g_content_type_get_mime_type (content_type);
 112           if (mime_type)
 113             {
 114               g_string_append_printf (s, "Content-Type: %s\r\n",
 115                                       mime_type);
 116               g_free (mime_type);
 117             }
 118         }
 119     }
 120   g_string_append (s, "\r\n");
 121
 122   if (g_output_stream_write_all (out,
 123                                  s->str, s->len,
 124                                  NULL, NULL, NULL))
 125     {
 126       g_output_stream_splice (out,
 127                               G_INPUT_STREAM (file_in),
 128                               0, NULL, NULL);
 129     }
 130   g_string_free (s, TRUE);
 131
 132   g_input_stream_close (G_INPUT_STREAM (file_in), NULL, NULL);
 133   g_object_unref (file_in);
 134
 135  out:
 136   g_object_unref (data);
 137
 138   return TRUE;
 139 }
 140
 141 int
 142 main (int argc, char *argv[])
 143 {
 144   GSocketService *service;
 145   GOptionContext *context;
 146   GError *error = NULL;
 147
 148   g_type_init ();
 149   g_thread_init (NULL);
 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 }