Source/WebKit2/UIProcess/API/Cocoa/WKProcessPool.mm

   1 /*
   2  * Copyright (C) 2014 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  23  * THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #import "config.h"
  27 #import "WKProcessPoolInternal.h"
  28
  29 #if WK_API_ENABLED
  30
  31 #import "CacheModel.h"
  32 #import "HistoryClient.h"
  33 #import "ProcessModel.h"
  34 #import "WKObject.h"
  35 #import "WKProcessPoolConfigurationPrivate.h"
  36 #import "WebCertificateInfo.h"
  37 #import "WebContext.h"
  38 #import "WebCookieManagerProxy.h"
  39 #import "WebProcessMessages.h"
  40 #import <WebCore/CertificateInfo.h>
  41 #import <wtf/RetainPtr.h>
  42
  43 #if PLATFORM(IOS)
  44 #import <WebCore/WebCoreThreadSystemInterface.h>
  45 #endif
  46
  47 #if __has_include(<CFNetwork/CFNSURLConnection.h>)
  48 #import <CFNetwork/CFNSURLConnection.h>
  49 #else
  50 enum : NSUInteger {
  51     NSHTTPCookieAcceptPolicyExclusivelyFromMainDocumentDomain = 3,
  52 };
  53 #endif
  54
  55 #if PLATFORM(MAC) && __MAC_OS_X_VERSION_MIN_REQUIRED == 1080
  56 @interface NSKeyedArchiver (WKDetails)
  57 - (void)setRequiresSecureCoding:(BOOL)b;
  58 @end
  59 #endif
  60
  61 @implementation WKProcessPool
  62
  63 - (instancetype)init
  64 {
  65     return [self _initWithConfiguration:adoptNS([[_WKProcessPoolConfiguration alloc] init]).get()];
  66 }
  67
  68 - (void)dealloc
  69 {
  70     _context->~WebContext();
  71
  72     [super dealloc];
  73 }
  74
  75 - (NSString *)description
  76 {
  77     return [NSString stringWithFormat:@"<%@: %p; configuration = %@>", NSStringFromClass(self.class), self, _configuration.get()];
  78 }
  79
  80 - (_WKProcessPoolConfiguration *)_configuration
  81 {
  82     return [[_configuration copy] autorelease];
  83 }
  84
  85 - (API::Object&)_apiObject
  86 {
  87     return *_context;
  88 }
  89
  90 @end
  91
  92 @implementation WKProcessPool (WKPrivate)
  93
  94 - (instancetype)_initWithConfiguration:(_WKProcessPoolConfiguration *)configuration
  95 {
  96     if (!(self = [super init]))
  97         return nil;
  98
  99     _configuration = adoptNS([configuration copy]);
 100
 101 #if PLATFORM(IOS)
 102     // FIXME: Remove once <rdar://problem/15256572> is fixed.
 103     InitWebCoreThreadSystemInterface();
 104 #endif
 105
 106     String bundlePath;
 107     if (NSURL *bundleURL = [_configuration injectedBundleURL]) {
 108         if (!bundleURL.isFileURL)
 109             [NSException raise:NSInvalidArgumentException format:@"Injected Bundle URL must be a file URL"];
 110
 111         bundlePath = bundleURL.path;
 112     }
 113
 114     API::Object::constructInWrapper<WebKit::WebContext>(self, bundlePath);
 115     _context->setHistoryClient(std::make_unique<WebKit::HistoryClient>());
 116     _context->setUsesNetworkProcess(true);
 117     _context->setProcessModel(WebKit::ProcessModelMultipleSecondaryProcesses);
 118
 119     // FIXME: Add a way to configure the cache model, see <rdar://problem/16206857>.
 120     _context->setCacheModel(WebKit::CacheModelPrimaryWebBrowser);
 121
 122     return self;
 123 }
 124
 125 - (void)_setAllowsSpecificHTTPSCertificate:(NSArray *)certificateChain forHost:(NSString *)host
 126 {
 127     _context->allowSpecificHTTPSCertificateForHost(WebKit::WebCertificateInfo::create(WebCore::CertificateInfo((CFArrayRef)certificateChain)).get(), host);
 128 }
 129
 130 static WebKit::HTTPCookieAcceptPolicy toHTTPCookieAcceptPolicy(NSHTTPCookieAcceptPolicy policy)
 131 {
 132     switch (static_cast<NSUInteger>(policy)) {
 133     case NSHTTPCookieAcceptPolicyAlways:
 134         return WebKit::HTTPCookieAcceptPolicyAlways;
 135     case NSHTTPCookieAcceptPolicyNever:
 136         return WebKit::HTTPCookieAcceptPolicyNever;
 137     case NSHTTPCookieAcceptPolicyOnlyFromMainDocumentDomain:
 138         return WebKit::HTTPCookieAcceptPolicyOnlyFromMainDocumentDomain;
 139     case NSHTTPCookieAcceptPolicyExclusivelyFromMainDocumentDomain:
 140         return WebKit::HTTPCookieAcceptPolicyExclusivelyFromMainDocumentDomain;
 141     }
 142
 143     ASSERT_NOT_REACHED();
 144     return WebKit::HTTPCookieAcceptPolicyAlways;
 145 }
 146
 147 - (void)_setCookieAcceptPolicy:(NSHTTPCookieAcceptPolicy)policy
 148 {
 149     _context->supplement<WebKit::WebCookieManagerProxy>()->setHTTPCookieAcceptPolicy(toHTTPCookieAcceptPolicy(policy));
 150 }
 151
 152 - (id)_objectForBundleParameter:(NSString *)parameter
 153 {
 154     return [_context->bundleParameters() objectForKey:parameter];
 155 }
 156
 157 - (void)_setObject:(id <NSCopying, NSSecureCoding>)object forBundleParameter:(NSString *)parameter
 158 {
 159     auto copy = adoptNS([(NSObject *)object copy]);
 160
 161     auto data = adoptNS([[NSMutableData alloc] init]);
 162     auto keyedArchiver = adoptNS([[NSKeyedArchiver alloc] initForWritingWithMutableData:data.get()]);
 163     [keyedArchiver setRequiresSecureCoding:YES];
 164
 165     @try {
 166         [keyedArchiver encodeObject:copy.get() forKey:@"parameter"];
 167         [keyedArchiver finishEncoding];
 168     } @catch (NSException *exception) {
 169         LOG_ERROR("Failed to encode bundle parameter: %@", exception);
 170     }
 171
 172     [_context->ensureBundleParameters() setObject:copy.get() forKey:parameter];
 173     _context->sendToAllProcesses(Messages::WebProcess::SetInjectedBundleParameter(parameter, IPC::DataReference(static_cast<const uint8_t*>([data bytes]), [data length])));
 174 }
 175
 176 @end
 177
 178 #endif // WK_API_ENABLED