/* * Copyright (C) 2011 Igalia S.L. * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2,1 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. */ #include "config.h" #include "LoadTrackingTest.h" #include "WebKitTestServer.h" #include static WebKitTestServer* kServer; class AuthenticationTest: public LoadTrackingTest { public: MAKE_GLIB_TEST_FIXTURE(AuthenticationTest); AuthenticationTest() { g_signal_connect(m_webView, "authenticate", G_CALLBACK(runAuthenticationCallback), this); } ~AuthenticationTest() { g_signal_handlers_disconnect_matched(m_webView, G_SIGNAL_MATCH_DATA, 0, 0, 0, 0, this); } static int authenticationRetries; static bool authenticationCancelledReceived; void loadURI(const char* uri) { // Reset the retry count of the fake server when a page is loaded. authenticationRetries = 0; authenticationCancelledReceived = false; LoadTrackingTest::loadURI(uri); } static gboolean runAuthenticationCallback(WebKitWebView*, WebKitAuthenticationRequest* request, AuthenticationTest* test) { g_signal_connect(request, "cancelled", G_CALLBACK(authenticationCancelledCallback), test); test->runAuthentication(request); return TRUE; } static void authenticationCancelledCallback(WebKitAuthenticationRequest*, AuthenticationTest*) { authenticationCancelledReceived = true; } void runAuthentication(WebKitAuthenticationRequest* request) { assertObjectIsDeletedWhenTestFinishes(G_OBJECT(request)); m_authenticationRequest = request; g_main_loop_quit(m_mainLoop); } WebKitAuthenticationRequest* waitForAuthenticationRequest() { g_main_loop_run(m_mainLoop); return m_authenticationRequest.get(); } private: GRefPtr m_authenticationRequest; }; int AuthenticationTest::authenticationRetries = 0; bool AuthenticationTest::authenticationCancelledReceived = false; static const char authTestUsername[] = "username"; static const char authTestPassword[] = "password"; static const char authExpectedSuccessTitle[] = "WebKit2Gtk+ Authentication test"; static const char authExpectedFailureTitle[] = "401 Authorization Required"; static const char authExpectedAuthorization[] = "Basic dXNlcm5hbWU6cGFzc3dvcmQ="; // Base64 encoding of "username:password". static const char authSuccessHTMLString[] = "" "WebKit2Gtk+ Authentication test" ""; static const char authFailureHTMLString[] = "" "401 Authorization Required" ""; static void testWebViewAuthenticationRequest(AuthenticationTest* test, gconstpointer) { // Test authentication request getters match soup authentication header. test->loadURI(kServer->getURIForPath("/auth-test.html").data()); WebKitAuthenticationRequest* request = test->waitForAuthenticationRequest(); g_assert_cmpstr(webkit_authentication_request_get_host(request), ==, soup_uri_get_host(kServer->baseURI())); g_assert_cmpuint(webkit_authentication_request_get_port(request), ==, soup_uri_get_port(kServer->baseURI())); g_assert_cmpstr(webkit_authentication_request_get_realm(request), ==, "my realm"); g_assert(webkit_authentication_request_get_scheme(request) == WEBKIT_AUTHENTICATION_SCHEME_HTTP_BASIC); g_assert(!webkit_authentication_request_is_for_proxy(request)); g_assert(!webkit_authentication_request_is_retry(request)); } static void testWebViewAuthenticationCancel(AuthenticationTest* test, gconstpointer) { // Test cancel. test->loadURI(kServer->getURIForPath("/auth-test.html").data()); WebKitAuthenticationRequest* request = test->waitForAuthenticationRequest(); webkit_authentication_request_cancel(request); // Server doesn't ask for new credentials. test->waitUntilLoadFinished(); g_assert_cmpint(test->m_loadEvents.size(), ==, 3); g_assert_cmpint(test->m_loadEvents[0], ==, LoadTrackingTest::ProvisionalLoadStarted); g_assert_cmpint(test->m_loadEvents[1], ==, LoadTrackingTest::ProvisionalLoadFailed); g_assert_cmpint(test->m_loadEvents[2], ==, LoadTrackingTest::LoadFinished); g_assert_error(test->m_error.get(), WEBKIT_NETWORK_ERROR, WEBKIT_NETWORK_ERROR_CANCELLED); } static void testWebViewAuthenticationLoadCancelled(AuthenticationTest* test, gconstpointer) { test->loadURI(kServer->getURIForPath("/auth-test.html").data()); test->waitForAuthenticationRequest(); webkit_web_view_stop_loading(test->m_webView); // Expect empty page. test->waitUntilLoadFinished(); g_assert(test->authenticationCancelledReceived); g_assert_cmpint(test->m_loadEvents.size(), ==, 3); g_assert_cmpint(test->m_loadEvents[0], ==, LoadTrackingTest::ProvisionalLoadStarted); g_assert_cmpint(test->m_loadEvents[1], ==, LoadTrackingTest::ProvisionalLoadFailed); g_assert_cmpint(test->m_loadEvents[2], ==, LoadTrackingTest::LoadFinished); g_assert_error(test->m_error.get(), WEBKIT_NETWORK_ERROR, WEBKIT_NETWORK_ERROR_CANCELLED); } static void testWebViewAuthenticationFailure(AuthenticationTest* test, gconstpointer) { // Test authentication failures. test->loadURI(kServer->getURIForPath("/auth-test.html").data()); WebKitAuthenticationRequest* request = test->waitForAuthenticationRequest(); g_assert(!webkit_authentication_request_is_retry(request)); WebKitCredential* credential = webkit_credential_new(authTestUsername, "wrongpassword", WEBKIT_CREDENTIAL_PERSISTENCE_NONE); webkit_authentication_request_authenticate(request, credential); webkit_credential_free(credential); // Expect a second authentication request. request = test->waitForAuthenticationRequest(); g_assert(webkit_authentication_request_is_retry(request)); // Test second failure. credential = webkit_credential_new(authTestUsername, "wrongpassword2", WEBKIT_CREDENTIAL_PERSISTENCE_NONE); webkit_authentication_request_authenticate(request, credential); webkit_credential_free(credential); // Expect authentication failed page. test->waitUntilLoadFinished(); g_assert_cmpint(test->m_loadEvents.size(), ==, 3); g_assert_cmpint(test->m_loadEvents[0], ==, LoadTrackingTest::ProvisionalLoadStarted); g_assert_cmpint(test->m_loadEvents[1], ==, LoadTrackingTest::LoadCommitted); g_assert_cmpint(test->m_loadEvents[2], ==, LoadTrackingTest::LoadFinished); g_assert_cmpstr(webkit_web_view_get_title(test->m_webView), ==, authExpectedFailureTitle); } static void testWebViewAuthenticationNoCredential(AuthenticationTest* test, gconstpointer) { // Test continue without credentials. test->loadURI(kServer->getURIForPath("/auth-test.html").data()); WebKitAuthenticationRequest* request = test->waitForAuthenticationRequest(); webkit_authentication_request_authenticate(request, 0); // Server doesn't ask for new credentials. test->waitUntilLoadFinished(); g_assert_cmpint(test->m_loadEvents.size(), ==, 3); g_assert_cmpint(test->m_loadEvents[0], ==, LoadTrackingTest::ProvisionalLoadStarted); g_assert_cmpint(test->m_loadEvents[1], ==, LoadTrackingTest::LoadCommitted); g_assert_cmpint(test->m_loadEvents[2], ==, LoadTrackingTest::LoadFinished); g_assert_cmpstr(webkit_web_view_get_title(test->m_webView), ==, authExpectedFailureTitle); } static void testWebViewAuthenticationStorage(AuthenticationTest* test, gconstpointer) { // Enable private browsing before authentication request to test that credentials can't be saved. webkit_settings_set_enable_private_browsing(webkit_web_view_get_settings(test->m_webView), TRUE); test->loadURI(kServer->getURIForPath("/auth-test.html").data()); WebKitAuthenticationRequest* request = test->waitForAuthenticationRequest(); g_assert(!webkit_authentication_request_get_proposed_credential(request)); g_assert(!webkit_authentication_request_can_save_credentials(request)); // If WebKit has been compiled with libsecret, and private browsing is disabled // then check that credentials can be saved. #if ENABLE(CREDENTIAL_STORAGE) webkit_settings_set_enable_private_browsing(webkit_web_view_get_settings(test->m_webView), FALSE); test->loadURI(kServer->getURIForPath("/auth-test.html").data()); request = test->waitForAuthenticationRequest(); g_assert(!webkit_authentication_request_get_proposed_credential(request)); g_assert(webkit_authentication_request_can_save_credentials(request)); #endif } static void testWebViewAuthenticationSuccess(AuthenticationTest* test, gconstpointer) { // Test correct authentication. test->loadURI(kServer->getURIForPath("/auth-test.html").data()); WebKitAuthenticationRequest* request = test->waitForAuthenticationRequest(); WebKitCredential* credential = webkit_credential_new(authTestUsername, authTestPassword, WEBKIT_CREDENTIAL_PERSISTENCE_FOR_SESSION); webkit_authentication_request_authenticate(request, credential); webkit_credential_free(credential); test->waitUntilLoadFinished(); g_assert_cmpint(test->m_loadEvents.size(), ==, 3); g_assert_cmpint(test->m_loadEvents[0], ==, LoadTrackingTest::ProvisionalLoadStarted); g_assert_cmpint(test->m_loadEvents[1], ==, LoadTrackingTest::LoadCommitted); g_assert_cmpint(test->m_loadEvents[2], ==, LoadTrackingTest::LoadFinished); g_assert_cmpstr(webkit_web_view_get_title(test->m_webView), ==, authExpectedSuccessTitle); // Test loading the same (authorized) page again. test->loadURI(kServer->getURIForPath("/auth-test.html").data()); // There is no authentication challenge. test->waitUntilLoadFinished(); g_assert_cmpint(test->m_loadEvents.size(), ==, 3); g_assert_cmpint(test->m_loadEvents[0], ==, LoadTrackingTest::ProvisionalLoadStarted); g_assert_cmpint(test->m_loadEvents[1], ==, LoadTrackingTest::LoadCommitted); g_assert_cmpint(test->m_loadEvents[2], ==, LoadTrackingTest::LoadFinished); g_assert_cmpstr(webkit_web_view_get_title(test->m_webView), ==, authExpectedSuccessTitle); } static void serverCallback(SoupServer*, SoupMessage* message, const char* path, GHashTable*, SoupClientContext*, void*) { if (message->method != SOUP_METHOD_GET) { soup_message_set_status(message, SOUP_STATUS_NOT_IMPLEMENTED); return; } if (!strcmp(path, "/auth-test.html")) { const char* authorization = soup_message_headers_get_one(message->request_headers, "Authorization"); // Require authentication. if (!g_strcmp0(authorization, authExpectedAuthorization)) { // Successful authentication. soup_message_set_status(message, SOUP_STATUS_OK); soup_message_body_append(message->response_body, SOUP_MEMORY_STATIC, authSuccessHTMLString, strlen(authSuccessHTMLString)); AuthenticationTest::authenticationRetries = 0; } else if (++AuthenticationTest::authenticationRetries < 3) { // No or invalid authorization header provided by the client, request authentication twice then fail. soup_message_set_status(message, SOUP_STATUS_UNAUTHORIZED); soup_message_headers_append(message->response_headers, "WWW-Authenticate", "Basic realm=\"my realm\""); // Include a failure message in case the user attempts to proceed without authentication. soup_message_body_append(message->response_body, SOUP_MEMORY_STATIC, authFailureHTMLString, strlen(authFailureHTMLString)); } else { // Authentication not successful, display a "401 Authorization Required" page. soup_message_set_status(message, SOUP_STATUS_OK); soup_message_body_append(message->response_body, SOUP_MEMORY_STATIC, authFailureHTMLString, strlen(authFailureHTMLString)); } } else soup_message_set_status(message, SOUP_STATUS_NOT_FOUND); soup_message_body_complete(message->response_body); } void beforeAll() { kServer = new WebKitTestServer(); kServer->run(serverCallback); AuthenticationTest::add("WebKitWebView", "authentication-request", testWebViewAuthenticationRequest); AuthenticationTest::add("WebKitWebView", "authentication-cancel", testWebViewAuthenticationCancel); AuthenticationTest::add("WebKitWebView", "authentication-load-cancelled", testWebViewAuthenticationLoadCancelled); AuthenticationTest::add("WebKitWebView", "authentication-failure", testWebViewAuthenticationFailure); AuthenticationTest::add("WebKitWebView", "authentication-no-credential", testWebViewAuthenticationNoCredential); AuthenticationTest::add("WebKitWebView", "authentication-storage", testWebViewAuthenticationStorage); // Testing authentication success must be done last because libsoup will never fire // the authenticate signal again once authentication is successful. AuthenticationTest::add("WebKitWebView", "authentication-success", testWebViewAuthenticationSuccess); } void afterAll() { delete kServer; }