Tools/TestWebKitAPI/Tests/WebKitCocoa/CreateWebArchive.mm - external/github.com/WebKit/webkit - Git at Google


 /*
  * Copyright (C) 2020 Apple Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
  *
  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  * THE POSSIBILITY OF SUCH DAMAGE.
  */

 #import "config.h"

 #if PLATFORM(MAC) || PLATFORM(IOS_FAMILY)

 #import "HTTPServer.h"
 #import "PlatformUtilities.h"
 #import "Test.h"
 #import "TestNavigationDelegate.h"
 #import "TestURLSchemeHandler.h"
 #import "TestWKWebView.h"

 #if PLATFORM(IOS_FAMILY)
 #import <MobileCoreServices/MobileCoreServices.h>
 #endif

 namespace TestWebKitAPI {

 static const char* mainBytes = R"TESTRESOURCE(
 <head>
 <script src="webarchivetest://host/script.js"></script>
 </head>
 )TESTRESOURCE";

 static const char* scriptBytes = R"TESTRESOURCE(
 window.webkit.messageHandlers.testHandler.postMessage("done");
 )TESTRESOURCE";

 TEST(WebArchive, CreateCustomScheme)
 {
     auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
     auto schemeHandler = adoptNS([[TestURLSchemeHandler alloc] init]);
     [configuration setURLSchemeHandler:schemeHandler.get() forURLScheme:@"webarchivetest"];

     [schemeHandler setStartURLSchemeTaskHandler:^(WKWebView *, id<WKURLSchemeTask> task) {
         const char* bytes = nullptr;
         if ([task.request.URL.absoluteString isEqualToString:@"webarchivetest://host/main.html"])
             bytes = mainBytes;
         else if ([task.request.URL.absoluteString isEqualToString:@"webarchivetest://host/script.js"])
             bytes = scriptBytes;
         else
             FAIL();

         auto response = adoptNS([[NSURLResponse alloc] initWithURL:task.request.URL MIMEType:@"text/html" expectedContentLength:0 textEncodingName:nil]);
         [task didReceiveResponse:response.get()];
         [task didReceiveData:[NSData dataWithBytes:bytes length:strlen(bytes)]];
         [task didFinish];
     }];

     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);

     static bool done = false;
     [webView performAfterReceivingMessage:@"done" action:[&] { done = true; }];
     [webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"webarchivetest://host/main.html"]]];

     Util::run(&done);
     done = false;

     static RetainPtr<NSData> archiveData;
     [webView createWebArchiveDataWithCompletionHandler:^(NSData *result, NSError *error) {
         EXPECT_NULL(error);
         EXPECT_TRUE(!!result);
         archiveData = result;
         done = true;
     }];

     Util::run(&done);
     done = false;

     [webView performAfterReceivingMessage:@"done" action:[&] { done = true; }];
     [webView loadData:archiveData.get() MIMEType:(NSString *)kUTTypeArchive characterEncodingName:@"utf-8" baseURL:[NSURL URLWithString:@"about:blank"]];

     Util::run(&done);
     done = false;
 }

 static RetainPtr<NSData> webArchiveAccessingCookies()
 {
     HTTPServer server({ { "/"_s, { "<script>document.cookie</script>"_s } } }, HTTPServer::Protocol::HttpsProxy);
     auto webView = adoptNS([[WKWebView alloc] initWithFrame:CGRectZero configuration:server.httpsProxyConfiguration()]);
     auto navigationDelegate = adoptNS([TestNavigationDelegate new]);
     [navigationDelegate allowAnyTLSCertificate];
     webView.get().navigationDelegate = navigationDelegate.get();
     [webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"https://webkit.org/"]]];
     [navigationDelegate waitForDidFinishNavigation];

     __block RetainPtr<NSData> webArchive;
     [webView createWebArchiveDataWithCompletionHandler:^(NSData *result, NSError *error) {
         webArchive = result;
     }];
     while (!webArchive)
         Util::spinRunLoop();
     return webArchive;
 }

 TEST(WebArchive, CookieAccessAfterLoadRequest)
 {
     auto webArchive = webArchiveAccessingCookies();

     NSString *path = [NSTemporaryDirectory() stringByAppendingPathComponent:@"CookieAccessTest.webarchive"];
     [[NSFileManager defaultManager] removeItemAtPath:path error:nil];
     BOOL success = [webArchive writeToFile:path atomically:YES];
     EXPECT_TRUE(success);

     auto webView = adoptNS([WKWebView new]);
     [webView loadRequest:[NSURLRequest requestWithURL:[NSURL fileURLWithPath:path isDirectory:NO]]];
     [webView _test_waitForDidFinishNavigation];
     [[NSFileManager defaultManager] removeItemAtPath:path error:nil];
 }

 TEST(WebArchive, CookieAccessAfterLoadData)
 {
     auto webArchive = webArchiveAccessingCookies();
     auto webView = adoptNS([WKWebView new]);
     [webView loadData:webArchive.get() MIMEType:@"application/x-webarchive" characterEncodingName:@"utf-8" baseURL:[NSURL URLWithString:@"http://example.com/"]];
     [webView _test_waitForDidFinishNavigation];
 }

 TEST(WebArchive, ApplicationXWebarchiveMIMETypeDoesNotLoadHTML)
 {
     HTTPServer server({ { "/"_s, { { { "Content-Type"_s, "application/x-webarchive"_s } }, "Not web archive content, should not load"_s } } });

     auto webView = adoptNS([WKWebView new]);
     [webView loadRequest:server.request()];
     [webView _test_waitForDidFailProvisionalNavigation];
 }

 } // namespace TestWebKitAPI

 #endif // PLATFORM(MAC) || PLATFORM(IOS_FAMILY)

	/*
	* Copyright (C) 2020 Apple Inc. All rights reserved.
	*
	* Redistribution and use in source and binary forms, with or without
	* modification, are permitted provided that the following conditions
	* are met:
	* 1. Redistributions of source code must retain the above copyright
	* notice, this list of conditions and the following disclaimer.
	* 2. Redistributions in binary form must reproduce the above copyright
	* notice, this list of conditions and the following disclaimer in the
	* documentation and/or other materials provided with the distribution.
	*
	* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
	* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
	* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
	* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
	* THE POSSIBILITY OF SUCH DAMAGE.
	*/

	#import "config.h"

	#if PLATFORM(MAC) \|\| PLATFORM(IOS_FAMILY)

	#import "HTTPServer.h"
	#import "PlatformUtilities.h"
	#import "Test.h"
	#import "TestNavigationDelegate.h"
	#import "TestURLSchemeHandler.h"
	#import "TestWKWebView.h"

	#if PLATFORM(IOS_FAMILY)
	#import <MobileCoreServices/MobileCoreServices.h>
	#endif

	namespace TestWebKitAPI {

	static const char* mainBytes = R"TESTRESOURCE(
	<head>
	<script src="webarchivetest://host/script.js"></script>
	</head>
	)TESTRESOURCE";

	static const char* scriptBytes = R"TESTRESOURCE(
	window.webkit.messageHandlers.testHandler.postMessage("done");
	)TESTRESOURCE";

	TEST(WebArchive, CreateCustomScheme)
	{
	auto configuration = adoptNS([[WKWebViewConfiguration alloc] init]);
	auto schemeHandler = adoptNS([[TestURLSchemeHandler alloc] init]);
	[configuration setURLSchemeHandler:schemeHandler.get() forURLScheme:@"webarchivetest"];

	[schemeHandler setStartURLSchemeTaskHandler:^(WKWebView *, id<WKURLSchemeTask> task) {
	const char* bytes = nullptr;
	if ([task.request.URL.absoluteString isEqualToString:@"webarchivetest://host/main.html"])
	bytes = mainBytes;
	else if ([task.request.URL.absoluteString isEqualToString:@"webarchivetest://host/script.js"])
	bytes = scriptBytes;
	else
	FAIL();

	auto response = adoptNS([[NSURLResponse alloc] initWithURL:task.request.URL MIMEType:@"text/html" expectedContentLength:0 textEncodingName:nil]);
	[task didReceiveResponse:response.get()];
	[task didReceiveData:[NSData dataWithBytes:bytes length:strlen(bytes)]];
	[task didFinish];
	}];

	auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);

	static bool done = false;
	[webView performAfterReceivingMessage:@"done" action:[&] { done = true; }];
	[webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"webarchivetest://host/main.html"]]];

	Util::run(&done);
	done = false;

	static RetainPtr<NSData> archiveData;
	[webView createWebArchiveDataWithCompletionHandler:^(NSData result, NSError error) {
	EXPECT_NULL(error);
	EXPECT_TRUE(!!result);
	archiveData = result;
	done = true;
	}];

	Util::run(&done);
	done = false;

	[webView performAfterReceivingMessage:@"done" action:[&] { done = true; }];
	[webView loadData:archiveData.get() MIMEType:(NSString *)kUTTypeArchive characterEncodingName:@"utf-8" baseURL:[NSURL URLWithString:@"about:blank"]];

	Util::run(&done);
	done = false;
	}

	static RetainPtr<NSData> webArchiveAccessingCookies()
	{
	HTTPServer server({ { "/"_s, { "<script>document.cookie</script>"_s } } }, HTTPServer::Protocol::HttpsProxy);
	auto webView = adoptNS([[WKWebView alloc] initWithFrame:CGRectZero configuration:server.httpsProxyConfiguration()]);
	auto navigationDelegate = adoptNS([TestNavigationDelegate new]);
	[navigationDelegate allowAnyTLSCertificate];
	webView.get().navigationDelegate = navigationDelegate.get();
	[webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"https://webkit.org/"]]];
	[navigationDelegate waitForDidFinishNavigation];

	__block RetainPtr<NSData> webArchive;
	[webView createWebArchiveDataWithCompletionHandler:^(NSData result, NSError error) {
	webArchive = result;
	}];
	while (!webArchive)
	Util::spinRunLoop();
	return webArchive;
	}

	TEST(WebArchive, CookieAccessAfterLoadRequest)
	{
	auto webArchive = webArchiveAccessingCookies();

	NSString *path = [NSTemporaryDirectory() stringByAppendingPathComponent:@"CookieAccessTest.webarchive"];
	[[NSFileManager defaultManager] removeItemAtPath:path error:nil];
	BOOL success = [webArchive writeToFile:path atomically:YES];
	EXPECT_TRUE(success);

	auto webView = adoptNS([WKWebView new]);
	[webView loadRequest:[NSURLRequest requestWithURL:[NSURL fileURLWithPath:path isDirectory:NO]]];
	[webView _test_waitForDidFinishNavigation];
	[[NSFileManager defaultManager] removeItemAtPath:path error:nil];
	}

	TEST(WebArchive, CookieAccessAfterLoadData)
	{
	auto webArchive = webArchiveAccessingCookies();
	auto webView = adoptNS([WKWebView new]);
	[webView loadData:webArchive.get() MIMEType:@"application/x-webarchive" characterEncodingName:@"utf-8" baseURL:[NSURL URLWithString:@"http://example.com/"]];
	[webView _test_waitForDidFinishNavigation];
	}

	TEST(WebArchive, ApplicationXWebarchiveMIMETypeDoesNotLoadHTML)
	{
	HTTPServer server({ { "/"_s, { { { "Content-Type"_s, "application/x-webarchive"_s } }, "Not web archive content, should not load"_s } } });

	auto webView = adoptNS([WKWebView new]);
	[webView loadRequest:server.request()];
	[webView _test_waitForDidFailProvisionalNavigation];
	}

	} // namespace TestWebKitAPI

	#endif // PLATFORM(MAC) \|\| PLATFORM(IOS_FAMILY)