1 files changed, 200 insertions, 0 deletions
diff --git a/WebCore/platform/network/soup/cache/soup-directory-input-stream.c b/WebCore/platform/network/soup/cache/soup-directory-input-stream.c
new file mode 100644
index 0000000..c14863b
--- /dev/null
+++ b/WebCore/platform/network/soup/cache/soup-directory-input-stream.c
@@ -0,0 +1,200 @@
+/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
+/*
+ * Copyright (C) 2008 Red Hat, Inc.
+ * Copyright (C) 2010 Igalia, S.L.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Library General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Library General Public License for more details.
+ *
+ * You should have received a copy of the GNU Library General Public License
+ * along with this library; see the file COPYING.LIB.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include "soup-directory-input-stream.h"
+
+#include <libsoup/soup.h>
+#include <stdio.h>
+#include <string.h>
+
+#define INIT_STRING "<html><head><title>OMG!</title></head><body><table>"
+#define EXIT_STRING "</table></html>"
+
+G_DEFINE_TYPE (WebKitSoupDirectoryInputStream, webkit_soup_directory_input_stream, G_TYPE_INPUT_STREAM)
+
+static SoupBuffer *
+webkit_soup_directory_input_stream_parse_info (WebKitSoupDirectoryInputStream * stream,
+					       GFileInfo * info)
+{
+	SoupBuffer *buffer;
+	GString *string;
+	const char *s;
+	char *escaped, *path, *xml_string;
+
+	if (!g_file_info_get_name (info))
+		return NULL;
+
+	s = g_file_info_get_display_name (info);
+	if (!s) {
+		s = g_file_info_get_name (info);
+		/* FIXME: convert somehow? */
+		if (!g_utf8_validate (s, -1, NULL))
+			return NULL;
+	}
+	string = g_string_new ("<tr>");
+
+	xml_string = g_markup_escape_text (s, -1);
+	escaped = g_uri_escape_string (g_file_info_get_name (info), NULL, FALSE);
+	path = g_strconcat (stream->uri, "/", escaped, NULL);
+	g_free (escaped);
+	g_string_append_printf (string, "<td><a href=\"%s\">%s</a></td>", path, xml_string);
+	g_free (path);
+	g_free (xml_string);
+	g_string_append (string, "</tr>");
+
+	buffer = soup_buffer_new (SOUP_MEMORY_TAKE, string->str, string->len);
+	g_string_free (string, FALSE);
+
+	return buffer;
+}
+
+static SoupBuffer *
+webkit_soup_directory_input_stream_read_next_file (WebKitSoupDirectoryInputStream  *stream,
+						   GCancellable              *cancellable,
+						   GError                   **error)
+{
+	GFileInfo *info;
+	SoupBuffer *buffer;
+	GError *err = NULL;
+
+	do {
+		info = g_file_enumerator_next_file (stream->enumerator, cancellable, &err);
+		if (info == NULL) {
+			if (err) {
+				g_propagate_error (error, err);
+				return NULL;
+			} else if (!stream->done) {
+				stream->done = TRUE;
+				return soup_buffer_new (SOUP_MEMORY_STATIC,
+							EXIT_STRING,
+							sizeof (EXIT_STRING));
+			} else {
+				return NULL;
+			}
+		}
+
+		buffer = webkit_soup_directory_input_stream_parse_info (stream, info);
+	} while (buffer == NULL);
+
+	return buffer;
+}
+
+static gssize
+webkit_soup_directory_input_stream_read (GInputStream  *input,
+					 void          *buffer,
+					 gsize count,
+					 GCancellable  *cancellable,
+					 GError       **error)
+{
+	WebKitSoupDirectoryInputStream *stream = WEBKIT_SOUP_DIRECTORY_INPUT_STREAM (input);
+	gsize total, size;
+
+	for (total = 0; total < count; total += size) {
+		if (stream->buffer == NULL) {
+			stream->buffer = webkit_soup_directory_input_stream_read_next_file (stream, cancellable, error);
+			if (stream->buffer == NULL) {
+				/* FIXME: Is this correct or should we forward the error? */
+				if (total)
+					g_clear_error (error);
+				return total;
+			}
+		}
+
+		size = MIN (stream->buffer->length, count - total);
+		memcpy ((char *)buffer + total, stream->buffer->data, size);
+		if (size == stream->buffer->length) {
+			soup_buffer_free (stream->buffer);
+			stream->buffer = NULL;
+		} else {
+			SoupBuffer *sub = soup_buffer_new_subbuffer (stream->buffer,
+								     size,
+								     stream->buffer->length - size);
+			soup_buffer_free (stream->buffer);
+			stream->buffer = sub;
+		}
+	}
+
+	return total;
+}
+
+static gboolean
+webkit_soup_directory_input_stream_close (GInputStream  *input,
+					  GCancellable  *cancellable,
+					  GError       **error)
+{
+	WebKitSoupDirectoryInputStream *stream = WEBKIT_SOUP_DIRECTORY_INPUT_STREAM (input);
+	gboolean result;
+
+	if (stream->buffer) {
+		soup_buffer_free (stream->buffer);
+		stream->buffer = NULL;
+	}
+
+	result = g_file_enumerator_close (stream->enumerator,
+					  cancellable,
+					  error);
+	g_object_unref (stream->enumerator);
+	stream->enumerator = NULL;
+
+	g_free (stream->uri);
+	stream->uri = NULL;
+
+	return result;
+}
+
+static void
+webkit_soup_directory_input_stream_class_init (WebKitSoupDirectoryInputStreamClass *stream_class)
+{
+	GInputStreamClass *inputstream_class = G_INPUT_STREAM_CLASS (stream_class);
+
+	inputstream_class->read_fn = webkit_soup_directory_input_stream_read;
+	inputstream_class->close_fn = webkit_soup_directory_input_stream_close;
+}
+
+static void
+webkit_soup_directory_input_stream_init (WebKitSoupDirectoryInputStream *stream)
+{
+	stream->buffer = soup_buffer_new (SOUP_MEMORY_STATIC,
+					  INIT_STRING,
+					  sizeof (INIT_STRING));
+}
+
+GInputStream *
+webkit_soup_directory_input_stream_new (GFileEnumerator *enumerator,
+					SoupURI         *uri)
+{
+	GInputStream *stream;
+
+	g_return_val_if_fail (G_IS_FILE_ENUMERATOR (enumerator), NULL);
+	g_return_val_if_fail (uri != NULL, NULL);
+
+	stream = g_object_new (WEBKIT_TYPE_SOUP_DIRECTORY_INPUT_STREAM, NULL);
+
+	WEBKIT_SOUP_DIRECTORY_INPUT_STREAM (stream)->enumerator = g_object_ref (enumerator);
+	WEBKIT_SOUP_DIRECTORY_INPUT_STREAM (stream)->uri = soup_uri_to_string (uri, FALSE);
+
+	return stream;
+}
+