mirror of
https://gitlab.gnome.org/GNOME/glib.git
synced 2024-12-26 07:26:15 +01:00
185 lines
4.3 KiB
C
185 lines
4.3 KiB
C
#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,
|
|
const 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;
|
|
}
|
|
version = version; /* To avoid -Wunused-but-set-variable */
|
|
|
|
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 (10);
|
|
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 ();
|
|
}
|