diff options
Diffstat (limited to 'Tools/TestWebKitAPI/Tests/WebKit2Gtk/TestWebKitWebContext.cpp')
-rw-r--r-- | Tools/TestWebKitAPI/Tests/WebKit2Gtk/TestWebKitWebContext.cpp | 412 |
1 files changed, 386 insertions, 26 deletions
diff --git a/Tools/TestWebKitAPI/Tests/WebKit2Gtk/TestWebKitWebContext.cpp b/Tools/TestWebKitAPI/Tests/WebKit2Gtk/TestWebKitWebContext.cpp index 74e8d6817..e7b03fe0d 100644 --- a/Tools/TestWebKitAPI/Tests/WebKit2Gtk/TestWebKitWebContext.cpp +++ b/Tools/TestWebKitAPI/Tests/WebKit2Gtk/TestWebKitWebContext.cpp @@ -22,10 +22,13 @@ #include "LoadTrackingTest.h" #include "WebKitTestServer.h" #include <gtk/gtk.h> +#include <limits.h> +#include <stdlib.h> #include <webkit2/webkit2.h> #include <wtf/HashMap.h> -#include <wtf/gobject/GRefPtr.h> -#include <wtf/gobject/GUniquePtr.h> +#include <wtf/glib/GRefPtr.h> +#include <wtf/glib/GUniquePtr.h> +#include <wtf/text/StringBuilder.h> #include <wtf/text/StringHash.h> static WebKitTestServer* kServer; @@ -34,6 +37,45 @@ static void testWebContextDefault(Test* test, gconstpointer) { // Check there's a single instance of the default web context. g_assert(webkit_web_context_get_default() == webkit_web_context_get_default()); + g_assert(webkit_web_context_get_default() != test->m_webContext.get()); +} + +static void testWebContextEphemeral(Test* test, gconstpointer) +{ + // By default web contexts are not ephemeral. + g_assert(!webkit_web_context_is_ephemeral(webkit_web_context_get_default())); + g_assert(!webkit_web_context_is_ephemeral(test->m_webContext.get())); + + WebKitWebsiteDataManager* manager = webkit_web_context_get_website_data_manager(webkit_web_context_get_default()); + g_assert(WEBKIT_IS_WEBSITE_DATA_MANAGER(manager)); + g_assert(!webkit_website_data_manager_is_ephemeral(manager)); + manager = webkit_web_context_get_website_data_manager(test->m_webContext.get()); + g_assert(WEBKIT_IS_WEBSITE_DATA_MANAGER(manager)); + g_assert(!webkit_website_data_manager_is_ephemeral(manager)); + + GRefPtr<WebKitWebView> webView = WEBKIT_WEB_VIEW(webkit_web_view_new()); + g_assert(!webkit_web_view_is_ephemeral(webView.get())); + g_assert(webkit_web_view_get_website_data_manager(webView.get()) == webkit_web_context_get_website_data_manager(webkit_web_context_get_default())); + + webView = WEBKIT_WEB_VIEW(webkit_web_view_new_with_context(test->m_webContext.get())); + g_assert(!webkit_web_view_is_ephemeral(webView.get())); + g_assert(webkit_web_view_get_website_data_manager(webView.get()) == manager); + + GRefPtr<WebKitWebContext> context = adoptGRef(webkit_web_context_new_ephemeral()); + g_assert(webkit_web_context_is_ephemeral(context.get())); + manager = webkit_web_context_get_website_data_manager(context.get()); + g_assert(WEBKIT_IS_WEBSITE_DATA_MANAGER(manager)); + g_assert(webkit_website_data_manager_is_ephemeral(manager)); + g_assert(webkit_web_view_get_website_data_manager(webView.get()) != manager); + + webView = WEBKIT_WEB_VIEW(webkit_web_view_new_with_context(context.get())); + g_assert(webkit_web_view_is_ephemeral(webView.get())); + g_assert(webkit_web_view_get_website_data_manager(webView.get()) == manager); + + GRefPtr<WebKitWebsiteDataManager> ephemeralManager = adoptGRef(webkit_website_data_manager_new_ephemeral()); + g_assert(webkit_website_data_manager_is_ephemeral(ephemeralManager.get())); + context = adoptGRef(webkit_web_context_new_with_website_data_manager(ephemeralManager.get())); + g_assert(webkit_web_context_is_ephemeral(context.get())); } class PluginsTest: public Test { @@ -41,11 +83,10 @@ public: MAKE_GLIB_TEST_FIXTURE(PluginsTest); PluginsTest() - : m_context(webkit_web_context_get_default()) - , m_mainLoop(g_main_loop_new(0, TRUE)) - , m_plugins(0) + : m_mainLoop(g_main_loop_new(nullptr, TRUE)) + , m_plugins(nullptr) { - webkit_web_context_set_additional_plugins_directory(m_context, WEBKIT_TEST_PLUGIN_DIR); + webkit_web_context_set_additional_plugins_directory(m_webContext.get(), WEBKIT_TEST_PLUGIN_DIR); } ~PluginsTest() @@ -56,19 +97,18 @@ public: static void getPluginsAsyncReadyCallback(GObject*, GAsyncResult* result, PluginsTest* test) { - test->m_plugins = webkit_web_context_get_plugins_finish(test->m_context, result, 0); + test->m_plugins = webkit_web_context_get_plugins_finish(test->m_webContext.get(), result, nullptr); g_main_loop_quit(test->m_mainLoop); } GList* getPlugins() { g_list_free_full(m_plugins, g_object_unref); - webkit_web_context_get_plugins(m_context, 0, reinterpret_cast<GAsyncReadyCallback>(getPluginsAsyncReadyCallback), this); + webkit_web_context_get_plugins(m_webContext.get(), nullptr, reinterpret_cast<GAsyncReadyCallback>(getPluginsAsyncReadyCallback), this); g_main_loop_run(m_mainLoop); return m_plugins; } - WebKitWebContext* m_context; GMainLoop* m_mainLoop; GList* m_plugins; }; @@ -89,7 +129,9 @@ static void testWebContextGetPlugins(PluginsTest* test, gconstpointer) } g_assert(WEBKIT_IS_PLUGIN(testPlugin.get())); - GUniquePtr<char> pluginPath(g_build_filename(WEBKIT_TEST_PLUGIN_DIR, "libTestNetscapePlugin.so", nullptr)); + char normalizedPath[PATH_MAX]; + g_assert(realpath(WEBKIT_TEST_PLUGIN_DIR, normalizedPath)); + GUniquePtr<char> pluginPath(g_build_filename(normalizedPath, "libTestNetscapePlugIn.so", nullptr)); g_assert_cmpstr(webkit_plugin_get_path(testPlugin.get()), ==, pluginPath.get()); g_assert_cmpstr(webkit_plugin_get_description(testPlugin.get()), ==, "Simple Netscape® plug-in that handles test content for WebKit"); GList* mimeInfoList = webkit_plugin_get_mime_info_list(testPlugin.get()); @@ -116,7 +158,10 @@ static const char* kBarHTML = "<html><body>Bar</body></html>"; static const char* kEchoHTMLFormat = "<html><body>%s</body></html>"; static const char* errorDomain = "test"; static const int errorCode = 10; -static const char* errorMessage = "Error message."; + +static const char* genericErrorMessage = "Error message."; +static const char* beforeReceiveResponseErrorMessage = "Error before didReceiveResponse."; +static const char* afterInitialChunkErrorMessage = "Error after reading the initial chunk."; class URISchemeTest: public LoadTrackingTest { public: @@ -148,23 +193,38 @@ public: g_assert(webkit_uri_scheme_request_get_web_view(request) == test->m_webView); - GRefPtr<GInputStream> inputStream = adoptGRef(g_memory_input_stream_new()); - test->assertObjectIsDeletedWhenTestFinishes(G_OBJECT(inputStream.get())); - const char* scheme = webkit_uri_scheme_request_get_scheme(request); g_assert(scheme); g_assert(test->m_handlersMap.contains(String::fromUTF8(scheme))); + const URISchemeHandler& handler = test->m_handlersMap.get(String::fromUTF8(scheme)); + + GRefPtr<GInputStream> inputStream = adoptGRef(g_memory_input_stream_new()); + test->assertObjectIsDeletedWhenTestFinishes(G_OBJECT(inputStream.get())); + + const gchar* requestPath = webkit_uri_scheme_request_get_path(request); + if (!g_strcmp0(scheme, "error")) { - GUniquePtr<GError> error(g_error_new_literal(g_quark_from_string(errorDomain), errorCode, errorMessage)); - webkit_uri_scheme_request_finish_error(request, error.get()); + if (!g_strcmp0(requestPath, "before-response")) { + GUniquePtr<GError> error(g_error_new_literal(g_quark_from_string(errorDomain), errorCode, beforeReceiveResponseErrorMessage)); + // We call finish() and then finish_error() to make sure that not even + // the didReceiveResponse message is processed at the time of failing. + webkit_uri_scheme_request_finish(request, G_INPUT_STREAM(inputStream.get()), handler.replyLength, handler.mimeType.data()); + webkit_uri_scheme_request_finish_error(request, error.get()); + } else if (!g_strcmp0(requestPath, "after-first-chunk")) { + g_memory_input_stream_add_data(G_MEMORY_INPUT_STREAM(inputStream.get()), handler.reply.data(), handler.reply.length(), 0); + webkit_uri_scheme_request_finish(request, inputStream.get(), handler.replyLength, handler.mimeType.data()); + // We need to wait until we reach the load-committed state before calling webkit_uri_scheme_request_finish_error(), + // so we rely on the test using finishOnCommittedAndWaitUntilLoadFinished() to actually call it from loadCommitted(). + } else { + GUniquePtr<GError> error(g_error_new_literal(g_quark_from_string(errorDomain), errorCode, genericErrorMessage)); + webkit_uri_scheme_request_finish_error(request, error.get()); + } return; } - const URISchemeHandler& handler = test->m_handlersMap.get(String::fromUTF8(scheme)); - if (!g_strcmp0(scheme, "echo")) { - char* replyHTML = g_strdup_printf(handler.reply.data(), webkit_uri_scheme_request_get_path(request)); + char* replyHTML = g_strdup_printf(handler.reply.data(), requestPath); g_memory_input_stream_add_data(G_MEMORY_INPUT_STREAM(inputStream.get()), replyHTML, strlen(replyHTML), g_free); } else if (!g_strcmp0(scheme, "closed")) g_input_stream_close(inputStream.get(), 0, 0); @@ -177,13 +237,58 @@ public: void registerURISchemeHandler(const char* scheme, const char* reply, int replyLength, const char* mimeType) { m_handlersMap.set(String::fromUTF8(scheme), URISchemeHandler(reply, replyLength, mimeType)); - webkit_web_context_register_uri_scheme(webkit_web_context_get_default(), scheme, uriSchemeRequestCallback, this, 0); + webkit_web_context_register_uri_scheme(m_webContext.get(), scheme, uriSchemeRequestCallback, this, 0); + } + + void loadCommitted() override + { + if (m_finishOnCommitted) { + GUniquePtr<GError> error(g_error_new_literal(g_quark_from_string(errorDomain), errorCode, afterInitialChunkErrorMessage)); + webkit_uri_scheme_request_finish_error(m_uriSchemeRequest.get(), error.get()); + } + + LoadTrackingTest::loadCommitted(); + } + + void finishOnCommittedAndWaitUntilLoadFinished() + { + m_finishOnCommitted = true; + waitUntilLoadFinished(); + m_finishOnCommitted = false; } GRefPtr<WebKitURISchemeRequest> m_uriSchemeRequest; HashMap<String, URISchemeHandler> m_handlersMap; + bool m_finishOnCommitted { false }; }; +String generateHTMLContent(unsigned contentLength) +{ + String baseString("abcdefghijklmnopqrstuvwxyz0123457890"); + unsigned baseLength = baseString.length(); + + StringBuilder builder; + builder.append("<html><body>"); + + if (contentLength <= baseLength) + builder.append(baseString, 0, contentLength); + else { + unsigned currentLength = 0; + while (currentLength < contentLength) { + if ((currentLength + baseLength) <= contentLength) + builder.append(baseString); + else + builder.append(baseString, 0, contentLength - currentLength); + + // Account for the 12 characters of the '<html><body>' prefix. + currentLength = builder.length() - 12; + } + } + builder.append("</body></html>"); + + return builder.toString(); +} + static void testWebContextURIScheme(URISchemeTest* test, gconstpointer) { test->registerURISchemeHandler("foo", kBarHTML, strlen(kBarHTML), "text/html"); @@ -203,6 +308,16 @@ static void testWebContextURIScheme(URISchemeTest* test, gconstpointer) g_assert_cmpint(mainResourceDataSize, ==, strlen(echoHTML.get())); g_assert(!strncmp(mainResourceData, echoHTML.get(), mainResourceDataSize)); + test->loadURI("echo:with#fragment"); + test->waitUntilLoadFinished(); + g_assert_cmpstr(webkit_uri_scheme_request_get_path(test->m_uriSchemeRequest.get()), ==, "with"); + g_assert_cmpstr(webkit_uri_scheme_request_get_uri(test->m_uriSchemeRequest.get()), ==, "echo:with#fragment"); + echoHTML.reset(g_strdup_printf(kEchoHTMLFormat, webkit_uri_scheme_request_get_path(test->m_uriSchemeRequest.get()))); + mainResourceDataSize = 0; + mainResourceData = test->mainResourceData(mainResourceDataSize); + g_assert_cmpint(mainResourceDataSize, ==, strlen(echoHTML.get())); + g_assert(!strncmp(mainResourceData, echoHTML.get(), mainResourceDataSize)); + test->registerURISchemeHandler("nomime", kBarHTML, -1, 0); test->m_loadEvents.clear(); test->loadURI("nomime:foo-bar"); @@ -216,14 +331,35 @@ static void testWebContextURIScheme(URISchemeTest* test, gconstpointer) g_assert(!test->m_loadEvents.contains(LoadTrackingTest::ProvisionalLoadFailed)); g_assert(!test->m_loadEvents.contains(LoadTrackingTest::LoadFailed)); - test->registerURISchemeHandler("error", 0, 0, 0); + // Anything over 8192 bytes will get multiple calls to g_input_stream_read_async in + // WebKitURISchemeRequest when reading data, but we still need way more than that to + // ensure that we reach the load-committed state before failing, so we use an 8MB HTML. + String longHTMLContent = generateHTMLContent(8 * 1024 * 1024); + test->registerURISchemeHandler("error", longHTMLContent.utf8().data(), -1, "text/html"); test->m_loadEvents.clear(); test->loadURI("error:error"); test->waitUntilLoadFinished(); g_assert(test->m_loadEvents.contains(LoadTrackingTest::ProvisionalLoadFailed)); g_assert(test->m_loadFailed); g_assert_error(test->m_error.get(), g_quark_from_string(errorDomain), errorCode); - g_assert_cmpstr(test->m_error->message, ==, errorMessage); + g_assert_cmpstr(test->m_error->message, ==, genericErrorMessage); + + test->m_loadEvents.clear(); + test->loadURI("error:before-response"); + test->waitUntilLoadFinished(); + g_assert(test->m_loadEvents.contains(LoadTrackingTest::ProvisionalLoadFailed)); + g_assert(test->m_loadFailed); + g_assert_error(test->m_error.get(), g_quark_from_string(errorDomain), errorCode); + g_assert_cmpstr(test->m_error->message, ==, beforeReceiveResponseErrorMessage); + + test->m_loadEvents.clear(); + test->loadURI("error:after-first-chunk"); + test->finishOnCommittedAndWaitUntilLoadFinished(); + g_assert(!test->m_loadEvents.contains(LoadTrackingTest::ProvisionalLoadFailed)); + g_assert(test->m_loadEvents.contains(LoadTrackingTest::LoadFailed)); + g_assert(test->m_loadFailed); + g_assert_error(test->m_error.get(), g_quark_from_string(errorDomain), errorCode); + g_assert_cmpstr(test->m_error->message, ==, afterInitialChunkErrorMessage); test->registerURISchemeHandler("closed", 0, 0, 0); test->m_loadEvents.clear(); @@ -236,7 +372,7 @@ static void testWebContextURIScheme(URISchemeTest* test, gconstpointer) static void testWebContextSpellChecker(Test* test, gconstpointer) { - WebKitWebContext* webContext = webkit_web_context_get_default(); + WebKitWebContext* webContext = test->m_webContext.get(); // Check what happens if no spell checking language has been set. const gchar* const* currentLanguage = webkit_web_context_get_spell_checking_languages(webContext); @@ -289,7 +425,7 @@ static void testWebContextSpellChecker(Test* test, gconstpointer) static void testWebContextLanguages(WebViewTest* test, gconstpointer) { - static const char* expectedDefaultLanguage = "en"; + static const char* expectedDefaultLanguage = "en-US"; test->loadURI(kServer->getURIForPath("/").data()); test->waitUntilLoadFinished(); size_t mainResourceDataSize = 0; @@ -302,7 +438,7 @@ static void testWebContextLanguages(WebViewTest* test, gconstpointer) g_ptr_array_add(languages.get(), const_cast<gpointer>(static_cast<const void*>("ES_es"))); g_ptr_array_add(languages.get(), const_cast<gpointer>(static_cast<const void*>("dE"))); g_ptr_array_add(languages.get(), 0); - webkit_web_context_set_preferred_languages(webkit_web_context_get_default(), reinterpret_cast<const char* const*>(languages->pdata)); + webkit_web_context_set_preferred_languages(test->m_webContext.get(), reinterpret_cast<const char* const*>(languages->pdata)); static const char* expectedLanguages = "en, es-es;q=0.90, de;q=0.80"; test->loadURI(kServer->getURIForPath("/").data()); @@ -311,6 +447,32 @@ static void testWebContextLanguages(WebViewTest* test, gconstpointer) mainResourceData = test->mainResourceData(mainResourceDataSize); g_assert_cmpuint(mainResourceDataSize, ==, strlen(expectedLanguages)); g_assert(!strncmp(mainResourceData, expectedLanguages, mainResourceDataSize)); + + // When using the C locale, en-US should be used as default. + const char* cLanguage[] = { "C", nullptr }; + webkit_web_context_set_preferred_languages(test->m_webContext.get(), cLanguage); + GUniqueOutPtr<GError> error; + WebKitJavascriptResult* javascriptResult = test->runJavaScriptAndWaitUntilFinished("Intl.DateTimeFormat().resolvedOptions().locale", &error.outPtr()); + g_assert(javascriptResult); + g_assert(!error); + GUniquePtr<char> locale(WebViewTest::javascriptResultToCString(javascriptResult)); + g_assert_cmpstr(locale.get(), ==, expectedDefaultLanguage); + + // When using the POSIX locale, en-US should be used as default. + const char* posixLanguage[] = { "POSIX", nullptr }; + webkit_web_context_set_preferred_languages(test->m_webContext.get(), posixLanguage); + javascriptResult = test->runJavaScriptAndWaitUntilFinished("Intl.DateTimeFormat().resolvedOptions().locale", &error.outPtr()); + g_assert(javascriptResult); + g_assert(!error); + locale.reset(WebViewTest::javascriptResultToCString(javascriptResult)); + g_assert_cmpstr(locale.get(), ==, expectedDefaultLanguage); + + // An invalid locale should throw an exception. + const char* invalidLanguage[] = { "A", nullptr }; + webkit_web_context_set_preferred_languages(test->m_webContext.get(), invalidLanguage); + javascriptResult = test->runJavaScriptAndWaitUntilFinished("Intl.DateTimeFormat().resolvedOptions().locale", &error.outPtr()); + g_assert(!javascriptResult); + g_assert_error(error.get(), WEBKIT_JAVASCRIPT_ERROR, WEBKIT_JAVASCRIPT_ERROR_SCRIPT_FAILED); } static void serverCallback(SoupServer* server, SoupMessage* message, const char* path, GHashTable*, SoupClientContext*, gpointer) @@ -325,6 +487,16 @@ static void serverCallback(SoupServer* server, SoupMessage* message, const char* soup_message_set_status(message, SOUP_STATUS_OK); soup_message_body_append(message->response_body, SOUP_MEMORY_COPY, acceptLanguage, strlen(acceptLanguage)); soup_message_body_complete(message->response_body); + } else if (g_str_equal(path, "/empty")) { + const char* emptyHTML = "<html><body></body></html>"; + soup_message_body_append(message->response_body, SOUP_MEMORY_STATIC, emptyHTML, strlen(emptyHTML)); + soup_message_body_complete(message->response_body); + soup_message_set_status(message, SOUP_STATUS_OK); + } else if (g_str_equal(path, "/echoPort")) { + char* port = g_strdup_printf("%u", soup_server_get_port(server)); + soup_message_body_append(message->response_body, SOUP_MEMORY_TAKE, port, strlen(port)); + soup_message_body_complete(message->response_body); + soup_message_set_status(message, SOUP_STATUS_OK); } else soup_message_set_status(message, SOUP_STATUS_NOT_FOUND); } @@ -343,7 +515,7 @@ public: }; SecurityPolicyTest() - : m_manager(webkit_web_context_get_security_manager(webkit_web_context_get_default())) + : m_manager(webkit_web_context_get_security_manager(m_webContext.get())) { } @@ -406,17 +578,205 @@ static void testWebContextSecurityPolicy(SecurityPolicyTest* test, gconstpointer | SecurityPolicyTest::CORSEnabled | SecurityPolicyTest::EmptyDocument); } +static void consoleMessageReceivedCallback(WebKitUserContentManager*, WebKitJavascriptResult* message, WebKitJavascriptResult** result) +{ + g_assert(result); + g_assert(!*result); + *result = webkit_javascript_result_ref(message); +} + +static void testWebContextSecurityFileXHR(WebViewTest* test, gconstpointer) +{ + GUniquePtr<char> fileURL(g_strdup_printf("file://%s/simple.html", Test::getResourcesDir(Test::WebKit2Resources).data())); + test->loadURI(fileURL.get()); + test->waitUntilLoadFinished(); + + GUniquePtr<char> jsonURL(g_strdup_printf("file://%s/simple.json", Test::getResourcesDir().data())); + GUniquePtr<char> xhr(g_strdup_printf("var xhr = new XMLHttpRequest; xhr.open(\"GET\", \"%s\"); xhr.send();", jsonURL.get())); + + WebKitJavascriptResult* consoleMessage = nullptr; + webkit_user_content_manager_register_script_message_handler(test->m_userContentManager.get(), "console"); + g_signal_connect(test->m_userContentManager.get(), "script-message-received::console", G_CALLBACK(consoleMessageReceivedCallback), &consoleMessage); + + // By default file access is not allowed, this will show a console message with a cross-origin error. + GUniqueOutPtr<GError> error; + WebKitJavascriptResult* javascriptResult = test->runJavaScriptAndWaitUntilFinished(xhr.get(), &error.outPtr()); + g_assert(javascriptResult); + g_assert(!error); + g_assert(consoleMessage); + GUniquePtr<char> messageString(WebViewTest::javascriptResultToCString(consoleMessage)); + GRefPtr<GVariant> variant = g_variant_parse(G_VARIANT_TYPE("(uusus)"), messageString.get(), nullptr, nullptr, nullptr); + g_assert(variant.get()); + unsigned level; + const char* messageText; + g_variant_get(variant.get(), "(uu&su&s)", nullptr, &level, &messageText, nullptr, nullptr); + g_assert_cmpuint(level, ==, 3); // Console error message. + GUniquePtr<char> expectedErrorMessage(g_strdup_printf("XMLHttpRequest cannot load %s. Cross origin requests are only supported for HTTP.", jsonURL.get())); + g_assert_cmpstr(messageText, ==, expectedErrorMessage.get()); + webkit_javascript_result_unref(consoleMessage); + consoleMessage = nullptr; + level = 0; + messageText = nullptr; + variant = nullptr; + + // Allow file access from file URLs. + webkit_settings_set_allow_file_access_from_file_urls(webkit_web_view_get_settings(test->m_webView), TRUE); + test->loadURI(fileURL.get()); + test->waitUntilLoadFinished(); + javascriptResult = test->runJavaScriptAndWaitUntilFinished(xhr.get(), &error.outPtr()); + g_assert(javascriptResult); + g_assert(!error); + + // It isn't still possible to load file from an HTTP URL. + test->loadURI(kServer->getURIForPath("/").data()); + test->waitUntilLoadFinished(); + javascriptResult = test->runJavaScriptAndWaitUntilFinished(xhr.get(), &error.outPtr()); + g_assert(javascriptResult); + g_assert(!error); + g_assert(consoleMessage); + variant = g_variant_parse(G_VARIANT_TYPE("(uusus)"), messageString.get(), nullptr, nullptr, nullptr); + g_assert(variant.get()); + g_variant_get(variant.get(), "(uu&su&s)", nullptr, &level, &messageText, nullptr, nullptr); + g_assert_cmpuint(level, ==, 3); // Console error message. + g_assert_cmpstr(messageText, ==, expectedErrorMessage.get()); + webkit_javascript_result_unref(consoleMessage); + + g_signal_handlers_disconnect_matched(test->m_userContentManager.get(), G_SIGNAL_MATCH_DATA, 0, 0, nullptr, nullptr, &consoleMessage); + webkit_user_content_manager_unregister_script_message_handler(test->m_userContentManager.get(), "console"); + + webkit_settings_set_allow_file_access_from_file_urls(webkit_web_view_get_settings(test->m_webView), FALSE); +} + +class ProxyTest : public WebViewTest { +public: + MAKE_GLIB_TEST_FIXTURE(ProxyTest); + + ProxyTest() + { + // This "proxy server" is actually just a different instance of the main + // test server (kServer), listening on a different port. Requests + // will not actually be proxied to kServer because proxyServer is not + // actually a proxy server. We're testing whether the proxy settings + // work, not whether we can write a soup proxy server. + m_proxyServer.run(serverCallback); + g_assert(m_proxyServer.baseURI()); + } + + CString loadURIAndGetMainResourceData(const char* uri) + { + loadURI(uri); + waitUntilLoadFinished(); + size_t dataSize = 0; + const char* data = mainResourceData(dataSize); + return CString(data, dataSize); + } + + GUniquePtr<char> proxyServerPortAsString() + { + GUniquePtr<char> port(g_strdup_printf("%u", soup_uri_get_port(m_proxyServer.baseURI()))); + return port; + } + + WebKitTestServer m_proxyServer; +}; + +static void ephemeralViewloadChanged(WebKitWebView* webView, WebKitLoadEvent loadEvent, WebViewTest* test) +{ + if (loadEvent != WEBKIT_LOAD_FINISHED) + return; + g_signal_handlers_disconnect_by_func(webView, reinterpret_cast<void*>(ephemeralViewloadChanged), test); + test->quitMainLoop(); +} + +static void testWebContextProxySettings(ProxyTest* test, gconstpointer) +{ + // Proxy URI is unset by default. Requests to kServer should be received by kServer. + GUniquePtr<char> serverPortAsString(g_strdup_printf("%u", soup_uri_get_port(kServer->baseURI()))); + auto mainResourceData = test->loadURIAndGetMainResourceData(kServer->getURIForPath("/echoPort").data()); + ASSERT_CMP_CSTRING(mainResourceData, ==, serverPortAsString.get()); + + // Set default proxy URI to point to proxyServer. Requests to kServer should be received by proxyServer instead. + GUniquePtr<char> proxyURI(soup_uri_to_string(test->m_proxyServer.baseURI(), FALSE)); + WebKitNetworkProxySettings* settings = webkit_network_proxy_settings_new(proxyURI.get(), nullptr); + webkit_web_context_set_network_proxy_settings(test->m_webContext.get(), WEBKIT_NETWORK_PROXY_MODE_CUSTOM, settings); + GUniquePtr<char> proxyServerPortAsString = test->proxyServerPortAsString(); + mainResourceData = test->loadURIAndGetMainResourceData(kServer->getURIForPath("/echoPort").data()); + ASSERT_CMP_CSTRING(mainResourceData, ==, proxyServerPortAsString.get()); + webkit_network_proxy_settings_free(settings); + + // Proxy settings also affect ephemeral web views. + GRefPtr<WebKitWebView> webView = WEBKIT_WEB_VIEW(g_object_new(WEBKIT_TYPE_WEB_VIEW, + "web-context", test->m_webContext.get(), + "is-ephemeral", TRUE, + nullptr)); + g_assert(webkit_web_view_is_ephemeral(webView.get())); + g_assert(!webkit_web_context_is_ephemeral(webkit_web_view_get_context(webView.get()))); + + g_signal_connect(webView.get(), "load-changed", G_CALLBACK(ephemeralViewloadChanged), test); + webkit_web_view_load_uri(webView.get(), kServer->getURIForPath("/echoPort").data()); + g_main_loop_run(test->m_mainLoop); + WebKitWebResource* resource = webkit_web_view_get_main_resource(webView.get()); + g_assert(WEBKIT_IS_WEB_RESOURCE(resource)); + webkit_web_resource_get_data(resource, nullptr, [](GObject* object, GAsyncResult* result, gpointer userData) { + size_t dataSize; + GUniquePtr<char> data(reinterpret_cast<char*>(webkit_web_resource_get_data_finish(WEBKIT_WEB_RESOURCE(object), result, &dataSize, nullptr))); + g_assert(data); + auto* test = static_cast<ProxyTest*>(userData); + GUniquePtr<char> proxyServerPortAsString = test->proxyServerPortAsString(); + ASSERT_CMP_CSTRING(CString(data.get(), dataSize), ==, proxyServerPortAsString.get()); + test->quitMainLoop(); + }, test); + g_main_loop_run(test->m_mainLoop); + + // Remove the proxy. Requests to kServer should be received by kServer again. + webkit_web_context_set_network_proxy_settings(test->m_webContext.get(), WEBKIT_NETWORK_PROXY_MODE_NO_PROXY, nullptr); + mainResourceData = test->loadURIAndGetMainResourceData(kServer->getURIForPath("/echoPort").data()); + ASSERT_CMP_CSTRING(mainResourceData, ==, serverPortAsString.get()); + + // Use a default proxy uri, but ignoring requests to localhost. + static const char* ignoreHosts[] = { "localhost", nullptr }; + settings = webkit_network_proxy_settings_new(proxyURI.get(), ignoreHosts); + webkit_web_context_set_network_proxy_settings(test->m_webContext.get(), WEBKIT_NETWORK_PROXY_MODE_CUSTOM, settings); + mainResourceData = test->loadURIAndGetMainResourceData(kServer->getURIForPath("/echoPort").data()); + ASSERT_CMP_CSTRING(mainResourceData, ==, proxyServerPortAsString.get()); + GUniquePtr<char> localhostEchoPortURI(g_strdup_printf("http://localhost:%s/echoPort", serverPortAsString.get())); + mainResourceData = test->loadURIAndGetMainResourceData(localhostEchoPortURI.get()); + ASSERT_CMP_CSTRING(mainResourceData, ==, serverPortAsString.get()); + webkit_network_proxy_settings_free(settings); + + // Remove the proxy again to ensure next test is not using any previous values. + webkit_web_context_set_network_proxy_settings(test->m_webContext.get(), WEBKIT_NETWORK_PROXY_MODE_NO_PROXY, nullptr); + mainResourceData = test->loadURIAndGetMainResourceData(kServer->getURIForPath("/echoPort").data()); + ASSERT_CMP_CSTRING(mainResourceData, ==, serverPortAsString.get()); + + // Use scheme specific proxy instead of the default. + settings = webkit_network_proxy_settings_new(nullptr, nullptr); + webkit_network_proxy_settings_add_proxy_for_scheme(settings, "http", proxyURI.get()); + webkit_web_context_set_network_proxy_settings(test->m_webContext.get(), WEBKIT_NETWORK_PROXY_MODE_CUSTOM, settings); + mainResourceData = test->loadURIAndGetMainResourceData(kServer->getURIForPath("/echoPort").data()); + ASSERT_CMP_CSTRING(mainResourceData, ==, proxyServerPortAsString.get()); + webkit_network_proxy_settings_free(settings); + + // Reset to use the default resolver. + webkit_web_context_set_network_proxy_settings(test->m_webContext.get(), WEBKIT_NETWORK_PROXY_MODE_DEFAULT, nullptr); + mainResourceData = test->loadURIAndGetMainResourceData(kServer->getURIForPath("/echoPort").data()); + ASSERT_CMP_CSTRING(mainResourceData, ==, serverPortAsString.get()); +} + void beforeAll() { kServer = new WebKitTestServer(); kServer->run(serverCallback); Test::add("WebKitWebContext", "default-context", testWebContextDefault); + Test::add("WebKitWebContext", "ephemeral", testWebContextEphemeral); PluginsTest::add("WebKitWebContext", "get-plugins", testWebContextGetPlugins); URISchemeTest::add("WebKitWebContext", "uri-scheme", testWebContextURIScheme); Test::add("WebKitWebContext", "spell-checker", testWebContextSpellChecker); WebViewTest::add("WebKitWebContext", "languages", testWebContextLanguages); SecurityPolicyTest::add("WebKitSecurityManager", "security-policy", testWebContextSecurityPolicy); + WebViewTest::add("WebKitSecurityManager", "file-xhr", testWebContextSecurityFileXHR); + ProxyTest::add("WebKitWebContext", "proxy", testWebContextProxySettings); } void afterAll() |