Unreviewed, fix iOS build with recent SDKs.
[WebKit-https.git] / Source / WebKit / UIProcess / API / Cocoa / WKURLSchemeTask.mm
1 /*
2  * Copyright (C) 2017 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 "WKURLSchemeTaskInternal.h"
28
29 #import "WebURLSchemeHandler.h"
30 #import "WebURLSchemeTask.h"
31 #import <WebCore/ResourceError.h>
32 #import <WebCore/ResourceResponse.h>
33 #import <WebCore/SharedBuffer.h>
34 #import <wtf/BlockPtr.h>
35
36 static void raiseExceptionIfNecessary(WebKit::WebURLSchemeTask::ExceptionType exceptionType)
37 {
38     switch (exceptionType) {
39     case WebKit::WebURLSchemeTask::ExceptionType::None:
40         return;
41     case WebKit::WebURLSchemeTask::ExceptionType::TaskAlreadyStopped:
42         [NSException raise:NSInternalInconsistencyException format:@"This task has already been stopped"];
43         break;
44     case WebKit::WebURLSchemeTask::ExceptionType::CompleteAlreadyCalled:
45         [NSException raise:NSInternalInconsistencyException format:@"[WKURLSchemeTask taskDidCompleteWithError:] has already been called for this task"];
46         break;
47     case WebKit::WebURLSchemeTask::ExceptionType::DataAlreadySent:
48         [NSException raise:NSInternalInconsistencyException format:@"[WKURLSchemeTask taskDidReceiveData:] has already been called for this task"];
49         break;
50     case WebKit::WebURLSchemeTask::ExceptionType::NoResponseSent:
51         [NSException raise:NSInternalInconsistencyException format:@"No response has been sent for this task"];
52         break;
53     case WebKit::WebURLSchemeTask::ExceptionType::RedirectAfterResponse:
54         [NSException raise:NSInternalInconsistencyException format:@"No redirects are allowed after the response"];
55         break;
56     }
57 }
58
59 @implementation WKURLSchemeTaskImpl
60
61 - (void)dealloc
62 {
63     _urlSchemeTask->API::URLSchemeTask::~URLSchemeTask();
64
65     [super dealloc];
66 }
67
68 - (NSURLRequest *)request
69 {
70     return _urlSchemeTask->task().request().nsURLRequest(WebCore::HTTPBodyUpdatePolicy::UpdateHTTPBody);
71 }
72
73 - (void)didReceiveResponse:(NSURLResponse *)response
74 {
75     auto result = _urlSchemeTask->task().didReceiveResponse(response);
76     raiseExceptionIfNecessary(result);
77 }
78
79 - (void)didReceiveData:(NSData *)data
80 {
81     auto result = _urlSchemeTask->task().didReceiveData(WebCore::SharedBuffer::create(data));
82     raiseExceptionIfNecessary(result);
83 }
84
85 - (void)didFinish
86 {
87     auto result = _urlSchemeTask->task().didComplete({ });
88     raiseExceptionIfNecessary(result);
89 }
90
91 - (void)didFailWithError:(NSError *)error
92 {
93     auto result = _urlSchemeTask->task().didComplete(error);
94     raiseExceptionIfNecessary(result);
95 }
96
97 - (void)_didPerformRedirection:(NSURLResponse *)response newRequest:(NSURLRequest *)request
98 {
99     auto result = _urlSchemeTask->task().didPerformRedirection(response, request);
100     raiseExceptionIfNecessary(result);
101 }
102
103 #pragma mark WKObject protocol implementation
104
105 - (API::Object&)_apiObject
106 {
107     return *_urlSchemeTask;
108 }
109
110 @end