Implement acceptance policies in SoupCookieJar

Through the "accept-policy" property in the jar we can now set one
among three predetermined policies: accept all cookies
(SOUP_COOKIE_JAR_ACCEPT_ALWAYS), accept none
(SOUP_COOKIE_JAR_ACCEPT_NEVER) and only accept cookies set by the main
document we are loading (SOUP_COOKIE_JAR_ACCEPT_NO_THIRD_PARTY).

A new property, "first-party", is added to SoupMessage so that the
application using libsoup can inform us of what is the main document
URI for each message that is requested.

1 files changed, 119 insertions, 0 deletions

diff --git a/tests/cookies-test.c b/tests/cookies-test.c
new file mode 100644
index 00000000..4e0c4d1e
--- /dev/null
+++ b/tests/cookies-test.c
@@ -0,0 +1,119 @@
+/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
+/*
+ * Copyright (C) 2010 Igalia S.L.
+ */
+
+#include <glib.h>
+#include <libsoup/soup.h>
+
+#include "test-utils.h"
+
+SoupServer *server;
+SoupURI *first_party_uri, *third_party_uri;
+const char *first_party = "http://127.0.0.1/";
+const char *third_party = "http://localhost/";
+
+static void
+server_callback (SoupServer *server, SoupMessage *msg,
+		 const char *path, GHashTable *query,
+		 SoupClientContext *context, gpointer data)
+{
+    if (g_str_equal(path, "/index.html"))
+	soup_message_headers_replace (msg->response_headers,
+				      "Set-Cookie",
+				      "foo=bar");
+    else if (g_str_equal (path, "/foo.jpg"))
+	soup_message_headers_replace (msg->response_headers,
+				      "Set-Cookie",
+				      "baz=qux");
+    else
+	g_return_if_reached ();
+
+    soup_message_set_status (msg, SOUP_STATUS_OK);
+}
+
+typedef struct {
+    SoupCookieJarAcceptPolicy policy;
+    int n_cookies;
+} CookiesForPolicy;
+
+static const CookiesForPolicy validResults[] = {
+    { SOUP_COOKIE_JAR_ACCEPT_ALWAYS, 2 },
+    { SOUP_COOKIE_JAR_ACCEPT_NEVER, 0 },
+    { SOUP_COOKIE_JAR_ACCEPT_NO_THIRD_PARTY, 1 }
+};
+
+static void
+do_cookies_accept_policy_test (void)
+{
+	SoupSession *session;
+	SoupMessage *msg;
+	SoupURI *uri;
+	SoupCookieJar *jar;
+	GSList *l, *p;
+	int i;
+
+	session = soup_test_session_new (SOUP_TYPE_SESSION_ASYNC, NULL);
+	soup_session_add_feature_by_type (session, SOUP_TYPE_COOKIE_JAR);
+	jar = SOUP_COOKIE_JAR (soup_session_get_feature (session, SOUP_TYPE_COOKIE_JAR));
+
+	for (i = 0; i < G_N_ELEMENTS (validResults); i++) {
+	    soup_cookie_jar_set_accept_policy (jar, validResults[i].policy);
+
+	    uri = soup_uri_new_with_base (first_party_uri, "/index.html");
+	    msg = soup_message_new_from_uri ("GET", uri);
+	    soup_message_set_first_party (msg, first_party_uri);
+	    soup_session_send_message (session, msg);
+	    soup_uri_free (uri);
+	    g_object_unref (msg);
+
+	    /* We can't use to servers due to limitations in
+	     * test_server, so let's swap first and third party here
+	     * to simulate a cookie coming from a third party.
+	     */
+	    uri = soup_uri_new_with_base (first_party_uri, "/foo.jpg");
+	    msg = soup_message_new_from_uri ("GET", uri);
+	    soup_message_set_first_party (msg, third_party_uri);
+	    soup_session_send_message (session, msg);
+	    soup_uri_free (uri);
+	    g_object_unref (msg);
+
+	    l = soup_cookie_jar_all_cookies (jar);
+	    if (g_slist_length (l) < validResults[i].n_cookies) {
+		    debug_printf (1, " accepted less cookies than it should have\n");
+		    errors++;
+	    } else if (g_slist_length (l) > validResults[i].n_cookies) {
+		    debug_printf (1, " accepted more cookies than it should have\n");
+		    errors++;
+	    }
+
+	    for (p = l; p; p = p->next)
+		soup_cookie_jar_delete_cookie (jar, p->data);
+
+	    g_slist_free (l);
+	}
+
+	soup_test_session_abort_unref (session);
+}
+
+int
+main (int argc, char **argv)
+{
+	test_init (argc, argv, NULL);
+
+	server = soup_test_server_new (TRUE);
+	soup_server_add_handler (server, NULL, server_callback, NULL, NULL);
+	first_party_uri = soup_uri_new (first_party);
+	third_party_uri = soup_uri_new (third_party);
+	soup_uri_set_port (first_party_uri, soup_server_get_port (server));
+	soup_uri_set_port (third_party_uri, soup_server_get_port (server));
+
+	do_cookies_accept_policy_test ();
+
+	soup_uri_free (first_party_uri);
+	soup_uri_free (third_party_uri);
+
+	test_cleanup ();
+
+	return errors != 0;
+}