| /* |
| * 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 <wtf/glib/GRefPtr.h> |
| |
| 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<WebKitAuthenticationRequest> 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[] = |
| "<html>" |
| "<head><title>WebKit2Gtk+ Authentication test</title></head>" |
| "<body></body></html>"; |
| static const char authFailureHTMLString[] = |
| "<html>" |
| "<head><title>401 Authorization Required</title></head>" |
| "<body></body></html>"; |
| |
| 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 USE(LIBSECRET) |
| 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 testWebViewAuthenticationEmptyRealm(AuthenticationTest* test, gconstpointer) |
| { |
| test->loadURI(kServer->getURIForPath("/empty-realm.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); |
| } |
| |
| 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") || !strcmp(path, "/empty-realm.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); |
| if (!strcmp(path, "/empty-realm.html")) |
| soup_message_headers_append(message->response_headers, "WWW-Authenticate", "Basic"); |
| else |
| 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-success", testWebViewAuthenticationSuccess); |
| AuthenticationTest::add("WebKitWebView", "authentication-failure", testWebViewAuthenticationFailure); |
| AuthenticationTest::add("WebKitWebView", "authentication-no-credential", testWebViewAuthenticationNoCredential); |
| AuthenticationTest::add("WebKitWebView", "authentication-storage", testWebViewAuthenticationStorage); |
| AuthenticationTest::add("WebKitWebView", "authentication-empty-realm", testWebViewAuthenticationEmptyRealm); |
| } |
| |
| void afterAll() |
| { |
| delete kServer; |
| } |