/* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 8 -*- */ /* * Copyright (C) 2001-2003, Ximian, Inc. */ #include #include #include #include #include #include #include static int compare_strings (gconstpointer a, gconstpointer b) { const char **sa = (const char **)a; const char **sb = (const char **)b; return strcmp (*sa, *sb); } static GString * get_directory_listing (const char *path) { GPtrArray *entries; GString *listing; char *escaped; GDir *dir; const gchar *d_name; int i; entries = g_ptr_array_new (); dir = g_dir_open (path, 0, NULL); if (dir) { while ((d_name = g_dir_read_name (dir))) { if (!strcmp (d_name, ".") || (!strcmp (d_name, "..") && !strcmp (path, "./"))) continue; escaped = g_markup_escape_text (d_name, -1); g_ptr_array_add (entries, escaped); } g_dir_close (dir); } g_ptr_array_sort (entries, (GCompareFunc)compare_strings); listing = g_string_new ("\r\n"); escaped = g_markup_escape_text (strchr (path, '/'), -1); g_string_append_printf (listing, "Index of %s\r\n", escaped); g_string_append_printf (listing, "

Index of %s

\r\n

\r\n", escaped); g_free (escaped); for (i = 0; i < entries->len; i++) { g_string_append_printf (listing, "%s
\r\n", (char *)entries->pdata[i], (char *)entries->pdata[i]); g_free (entries->pdata[i]); } g_string_append (listing, "\r\n\r\n"); g_ptr_array_free (entries, TRUE); return listing; } static void do_get (SoupServer *server, SoupServerMessage *msg, const char *path) { char *slash; GStatBuf st; if (g_stat (path, &st) == -1) { if (errno == EPERM) soup_server_message_set_status (msg, SOUP_STATUS_FORBIDDEN, NULL); else if (errno == ENOENT) soup_server_message_set_status (msg, SOUP_STATUS_NOT_FOUND, NULL); else soup_server_message_set_status (msg, SOUP_STATUS_INTERNAL_SERVER_ERROR, NULL); /* In a real program you should also provide some text explaining the * error to the user (via soup_server_message_set_response), and include in it * the request path, as otherwise some browsers (Firefox, WebKit) only * show a blank page, confusing the user. */ return; } if (g_file_test (path, G_FILE_TEST_IS_DIR)) { GString *listing; char *index_path; char *listing_str; gsize listing_len; slash = strrchr (path, '/'); if (!slash || slash[1]) { char *redir_uri; redir_uri = g_strdup_printf ("%s/", g_uri_get_path (soup_server_message_get_uri (msg))); soup_server_message_set_redirect (msg, SOUP_STATUS_MOVED_PERMANENTLY, redir_uri); g_free (redir_uri); return; } index_path = g_strdup_printf ("%s/index.html", path); if (g_stat (index_path, &st) != -1) { do_get (server, msg, index_path); g_free (index_path); return; } g_free (index_path); listing = get_directory_listing (path); listing_len = listing->len; listing_str = g_string_free (g_steal_pointer (&listing), FALSE); soup_server_message_set_response (msg, "text/html", SOUP_MEMORY_TAKE, g_steal_pointer (&listing_str), listing_len); soup_server_message_set_status (msg, SOUP_STATUS_OK, NULL); return; } if (soup_server_message_get_method (msg) == SOUP_METHOD_GET) { GMappedFile *mapping; GBytes *buffer; mapping = g_mapped_file_new (path, FALSE, NULL); if (!mapping) { soup_server_message_set_status (msg, SOUP_STATUS_INTERNAL_SERVER_ERROR, NULL); return; } buffer = g_bytes_new_with_free_func (g_mapped_file_get_contents (mapping), g_mapped_file_get_length (mapping), (GDestroyNotify)g_mapped_file_unref, mapping); soup_message_body_append_bytes (soup_server_message_get_response_body (msg), buffer); g_bytes_unref (buffer); } else /* soup_server_message_get_method (msg) == SOUP_METHOD_HEAD */ { char *length; /* We could just use the same code for both GET and * HEAD (soup-message-server-io.c will fix things up). * But we'll optimize and avoid the extra I/O. */ length = g_strdup_printf ("%lu", (gulong)st.st_size); soup_message_headers_append (soup_server_message_get_response_headers (msg), "Content-Length", length); g_free (length); } soup_server_message_set_status (msg, SOUP_STATUS_OK, NULL); } static void do_put (SoupServer *server, SoupServerMessage *msg, const char *path) { GStatBuf st; FILE *f; gboolean created = TRUE; SoupMessageBody *request_body; if (g_stat (path, &st) != -1) { const char *match = soup_message_headers_get_one (soup_server_message_get_request_headers (msg), "If-None-Match"); if (match && !strcmp (match, "*")) { soup_server_message_set_status (msg, SOUP_STATUS_CONFLICT, NULL); return; } if (!g_file_test (path, G_FILE_TEST_IS_REGULAR)) { soup_server_message_set_status (msg, SOUP_STATUS_FORBIDDEN, NULL); return; } created = FALSE; } f = fopen (path, "w"); if (!f) { soup_server_message_set_status (msg, SOUP_STATUS_INTERNAL_SERVER_ERROR, NULL); return; } request_body = soup_server_message_get_request_body (msg); fwrite (request_body->data, 1, request_body->length, f); fclose (f); soup_server_message_set_status (msg, created ? SOUP_STATUS_CREATED : SOUP_STATUS_OK, NULL); } static void server_callback (SoupServer *server, SoupServerMessage *msg, const char *path, GHashTable *query, gpointer user_data) { char *file_path; SoupMessageHeadersIter iter; SoupMessageBody *request_body; const char *name, *value; g_print ("%s %s HTTP/1.%d\n", soup_server_message_get_method (msg), path, soup_server_message_get_http_version (msg)); soup_message_headers_iter_init (&iter, soup_server_message_get_request_headers (msg)); while (soup_message_headers_iter_next (&iter, &name, &value)) g_print ("%s: %s\n", name, value); request_body = soup_server_message_get_request_body (msg); if (request_body->length) g_print ("%s\n", request_body->data); file_path = g_strdup_printf (".%s", path); if (soup_server_message_get_method (msg) == SOUP_METHOD_GET || soup_server_message_get_method (msg) == SOUP_METHOD_HEAD) do_get (server, msg, file_path); else if (soup_server_message_get_method (msg) == SOUP_METHOD_PUT) do_put (server, msg, file_path); else soup_server_message_set_status (msg, SOUP_STATUS_NOT_IMPLEMENTED, NULL); g_free (file_path); g_print (" -> %d %s\n\n", soup_server_message_get_status (msg), soup_server_message_get_reason_phrase (msg)); } static void quit (int sig) { /* Exit cleanly on ^C in case we're valgrinding. */ exit (0); } static int port; static const char *tls_cert_file, *tls_key_file; static GOptionEntry entries[] = { { "cert-file", 'c', 0, G_OPTION_ARG_STRING, &tls_cert_file, "Use FILE as the TLS certificate file", "FILE" }, { "key-file", 'k', 0, G_OPTION_ARG_STRING, &tls_key_file, "Use FILE as the TLS private key file", "FILE" }, { "port", 'p', 0, G_OPTION_ARG_INT, &port, "Port to listen on", NULL }, { NULL } }; int main (int argc, char **argv) { GOptionContext *opts; GMainLoop *loop; SoupServer *server; GSList *uris, *u; char *str; GTlsCertificate *cert; GError *error = NULL; opts = g_option_context_new (NULL); g_option_context_add_main_entries (opts, entries, NULL); if (!g_option_context_parse (opts, &argc, &argv, &error)) { g_printerr ("Could not parse arguments: %s\n", error->message); g_printerr ("%s", g_option_context_get_help (opts, TRUE, NULL)); exit (1); } if (argc != 1) { g_printerr ("%s", g_option_context_get_help (opts, TRUE, NULL)); exit (1); } g_option_context_free (opts); signal (SIGINT, quit); if (tls_cert_file && tls_key_file) { cert = g_tls_certificate_new_from_files (tls_cert_file, tls_key_file, &error); if (error) { g_printerr ("Unable to create server: %s\n", error->message); exit (1); } server = soup_server_new ("server-header", "simple-httpd ", "tls-certificate", cert, NULL); g_object_unref (cert); soup_server_listen_all (server, port, SOUP_SERVER_LISTEN_HTTPS, &error); } else { server = soup_server_new ("server-header", "simple-httpd ", NULL); soup_server_listen_all (server, port, 0, &error); } soup_server_add_handler (server, NULL, server_callback, NULL, NULL); uris = soup_server_get_uris (server); for (u = uris; u; u = u->next) { str = g_uri_to_string (u->data); g_print ("Listening on %s\n", str); g_free (str); g_uri_unref (u->data); } g_slist_free (uris); g_print ("\nWaiting for requests...\n"); loop = g_main_loop_new (NULL, TRUE); g_main_loop_run (loop); return 0; }