1 /* |
|
2 * Copyright (C) 2007, 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 * |
|
8 * 1. Redistributions of source code must retain the above copyright |
|
9 * notice, this list of conditions and the following disclaimer. |
|
10 * 2. Redistributions in binary form must reproduce the above copyright |
|
11 * notice, this list of conditions and the following disclaimer in the |
|
12 * documentation and/or other materials provided with the distribution. |
|
13 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of |
|
14 * its contributors may be used to endorse or promote products derived |
|
15 * from this software without specific prior written permission. |
|
16 * |
|
17 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY |
|
18 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
|
19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
|
20 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY |
|
21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
|
22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
|
23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
|
24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
|
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
27 */ |
|
28 |
|
29 #import "config.h" |
|
30 #import "ResourceLoadDelegate.h" |
|
31 |
|
32 #import "DumpRenderTree.h" |
|
33 #import "LayoutTestController.h" |
|
34 #import <WebKit/WebKit.h> |
|
35 #import <WebKit/WebTypesInternal.h> |
|
36 #import <WebKit/WebDataSourcePrivate.h> |
|
37 #import <wtf/Assertions.h> |
|
38 |
|
39 using namespace std; |
|
40 |
|
41 @interface NSURL (DRTExtras) |
|
42 - (NSString *)_drt_descriptionSuitableForTestResult; |
|
43 @end |
|
44 |
|
45 @interface NSError (DRTExtras) |
|
46 - (NSString *)_drt_descriptionSuitableForTestResult; |
|
47 @end |
|
48 |
|
49 @interface NSURLResponse (DRTExtras) |
|
50 - (NSString *)_drt_descriptionSuitableForTestResult; |
|
51 @end |
|
52 |
|
53 @interface NSURLRequest (DRTExtras) |
|
54 - (NSString *)_drt_descriptionSuitableForTestResult; |
|
55 @end |
|
56 |
|
57 @implementation NSError (DRTExtras) |
|
58 - (NSString *)_drt_descriptionSuitableForTestResult |
|
59 { |
|
60 NSString *str = [NSString stringWithFormat:@"<NSError domain %@, code %d", [self domain], [self code]]; |
|
61 NSURL *failingURL; |
|
62 |
|
63 if ((failingURL = [[self userInfo] objectForKey:@"NSErrorFailingURLKey"])) |
|
64 str = [str stringByAppendingFormat:@", failing URL \"%@\"", [failingURL _drt_descriptionSuitableForTestResult]]; |
|
65 |
|
66 str = [str stringByAppendingFormat:@">"]; |
|
67 |
|
68 return str; |
|
69 } |
|
70 |
|
71 @end |
|
72 |
|
73 @implementation NSURL (DRTExtras) |
|
74 |
|
75 - (NSString *)_drt_descriptionSuitableForTestResult |
|
76 { |
|
77 if (![self isFileURL]) |
|
78 return [self absoluteString]; |
|
79 |
|
80 WebDataSource *dataSource = [mainFrame dataSource]; |
|
81 if (!dataSource) |
|
82 dataSource = [mainFrame provisionalDataSource]; |
|
83 |
|
84 NSString *basePath = [[[[dataSource request] URL] path] stringByDeletingLastPathComponent]; |
|
85 |
|
86 return [[self path] substringFromIndex:[basePath length] + 1]; |
|
87 } |
|
88 |
|
89 @end |
|
90 |
|
91 @implementation NSURLResponse (DRTExtras) |
|
92 |
|
93 - (NSString *)_drt_descriptionSuitableForTestResult |
|
94 { |
|
95 int statusCode = 0; |
|
96 if ([self isKindOfClass:[NSHTTPURLResponse class]]) |
|
97 statusCode = [(NSHTTPURLResponse *)self statusCode]; |
|
98 return [NSString stringWithFormat:@"<NSURLResponse %@, http status code %i>", [[self URL] _drt_descriptionSuitableForTestResult], statusCode]; |
|
99 } |
|
100 |
|
101 @end |
|
102 |
|
103 @implementation NSURLRequest (DRTExtras) |
|
104 |
|
105 - (NSString *)_drt_descriptionSuitableForTestResult |
|
106 { |
|
107 NSString *httpMethod = [self HTTPMethod]; |
|
108 if (!httpMethod) |
|
109 httpMethod = @"(none)"; |
|
110 return [NSString stringWithFormat:@"<NSURLRequest URL %@, main document URL %@, http method %@>", [[self URL] _drt_descriptionSuitableForTestResult], [[self mainDocumentURL] _drt_descriptionSuitableForTestResult], httpMethod]; |
|
111 } |
|
112 |
|
113 @end |
|
114 |
|
115 @implementation ResourceLoadDelegate |
|
116 |
|
117 - webView: (WebView *)wv identifierForInitialRequest: (NSURLRequest *)request fromDataSource: (WebDataSource *)dataSource |
|
118 { |
|
119 ASSERT([[dataSource webFrame] dataSource] || [[dataSource webFrame] provisionalDataSource]); |
|
120 |
|
121 if (!done && gLayoutTestController->dumpResourceLoadCallbacks()) |
|
122 return [[request URL] _drt_descriptionSuitableForTestResult]; |
|
123 |
|
124 return @"<unknown>"; |
|
125 } |
|
126 |
|
127 -(NSURLRequest *)webView: (WebView *)wv resource:identifier willSendRequest: (NSURLRequest *)request redirectResponse:(NSURLResponse *)redirectResponse fromDataSource:(WebDataSource *)dataSource |
|
128 { |
|
129 if (!done && gLayoutTestController->dumpResourceLoadCallbacks()) { |
|
130 NSString *string = [NSString stringWithFormat:@"%@ - willSendRequest %@ redirectResponse %@", identifier, [request _drt_descriptionSuitableForTestResult], |
|
131 [redirectResponse _drt_descriptionSuitableForTestResult]]; |
|
132 printf("%s\n", [string UTF8String]); |
|
133 } |
|
134 |
|
135 if (!done && !gLayoutTestController->deferMainResourceDataLoad()) { |
|
136 [dataSource _setDeferMainResourceDataLoad:false]; |
|
137 } |
|
138 |
|
139 if (!done && gLayoutTestController->willSendRequestReturnsNull()) |
|
140 return nil; |
|
141 |
|
142 if (!done && gLayoutTestController->willSendRequestReturnsNullOnRedirect() && redirectResponse) { |
|
143 printf("Returning null for this redirect\n"); |
|
144 return nil; |
|
145 } |
|
146 |
|
147 NSURL *url = [request URL]; |
|
148 NSString *host = [url host]; |
|
149 if (host |
|
150 && (NSOrderedSame == [[url scheme] caseInsensitiveCompare:@"http"] || NSOrderedSame == [[url scheme] caseInsensitiveCompare:@"https"]) |
|
151 && NSOrderedSame != [host compare:@"127.0.0.1"] |
|
152 && NSOrderedSame != [host compare:@"255.255.255.255"] // used in some tests that expect to get back an error |
|
153 && NSOrderedSame != [host caseInsensitiveCompare:@"localhost"]) { |
|
154 printf("Blocked access to external URL %s\n", [[url absoluteString] cStringUsingEncoding:NSUTF8StringEncoding]); |
|
155 return nil; |
|
156 } |
|
157 |
|
158 if (disallowedURLs && CFSetContainsValue(disallowedURLs, url)) |
|
159 return nil; |
|
160 |
|
161 NSMutableURLRequest *newRequest = [request mutableCopy]; |
|
162 const set<string>& clearHeaders = gLayoutTestController->willSendRequestClearHeaders(); |
|
163 for (set<string>::const_iterator header = clearHeaders.begin(); header != clearHeaders.end(); ++header) { |
|
164 NSString *nsHeader = [[NSString alloc] initWithUTF8String:header->c_str()]; |
|
165 [newRequest setValue:nil forHTTPHeaderField:nsHeader]; |
|
166 [nsHeader release]; |
|
167 } |
|
168 |
|
169 return [newRequest autorelease]; |
|
170 } |
|
171 |
|
172 - (void)webView:(WebView *)wv resource:(id)identifier didReceiveAuthenticationChallenge:(NSURLAuthenticationChallenge *)challenge fromDataSource:(WebDataSource *)dataSource |
|
173 { |
|
174 if (!gLayoutTestController->handlesAuthenticationChallenges()) { |
|
175 NSString *string = [NSString stringWithFormat:@"%@ - didReceiveAuthenticationChallenge - Simulating cancelled authentication sheet", identifier]; |
|
176 printf("%s\n", [string UTF8String]); |
|
177 |
|
178 [[challenge sender] continueWithoutCredentialForAuthenticationChallenge:challenge]; |
|
179 return; |
|
180 } |
|
181 |
|
182 const char* user = gLayoutTestController->authenticationUsername().c_str(); |
|
183 NSString *nsUser = [NSString stringWithFormat:@"%s", user ? user : ""]; |
|
184 |
|
185 const char* password = gLayoutTestController->authenticationPassword().c_str(); |
|
186 NSString *nsPassword = [NSString stringWithFormat:@"%s", password ? password : ""]; |
|
187 |
|
188 NSString *string = [NSString stringWithFormat:@"%@ - didReceiveAuthenticationChallenge - Responding with %@:%@", identifier, nsUser, nsPassword]; |
|
189 printf("%s\n", [string UTF8String]); |
|
190 |
|
191 [[challenge sender] useCredential:[NSURLCredential credentialWithUser:nsUser password:nsPassword persistence:NSURLCredentialPersistenceForSession] |
|
192 forAuthenticationChallenge:challenge]; |
|
193 } |
|
194 |
|
195 - (void)webView:(WebView *)wv resource:(id)identifier didCancelAuthenticationChallenge:(NSURLAuthenticationChallenge *)challenge fromDataSource:(WebDataSource *)dataSource |
|
196 { |
|
197 } |
|
198 |
|
199 -(void)webView: (WebView *)wv resource:identifier didReceiveResponse: (NSURLResponse *)response fromDataSource:(WebDataSource *)dataSource |
|
200 { |
|
201 if (!done && gLayoutTestController->dumpResourceLoadCallbacks()) { |
|
202 NSString *string = [NSString stringWithFormat:@"%@ - didReceiveResponse %@", identifier, [response _drt_descriptionSuitableForTestResult]]; |
|
203 printf("%s\n", [string UTF8String]); |
|
204 } |
|
205 if (!done && gLayoutTestController->dumpResourceResponseMIMETypes()) |
|
206 printf("%s has MIME type %s\n", [[[[response URL] relativePath] lastPathComponent] UTF8String], [[response MIMEType] UTF8String]); |
|
207 } |
|
208 |
|
209 -(void)webView: (WebView *)wv resource:identifier didReceiveContentLength: (NSInteger)length fromDataSource:(WebDataSource *)dataSource |
|
210 { |
|
211 } |
|
212 |
|
213 -(void)webView: (WebView *)wv resource:identifier didFinishLoadingFromDataSource:(WebDataSource *)dataSource |
|
214 { |
|
215 if (!done && gLayoutTestController->dumpResourceLoadCallbacks()) { |
|
216 NSString *string = [NSString stringWithFormat:@"%@ - didFinishLoading", identifier]; |
|
217 printf("%s\n", [string UTF8String]); |
|
218 } |
|
219 } |
|
220 |
|
221 -(void)webView: (WebView *)wv resource:identifier didFailLoadingWithError:(NSError *)error fromDataSource:(WebDataSource *)dataSource |
|
222 { |
|
223 if (!done && gLayoutTestController->dumpResourceLoadCallbacks()) { |
|
224 NSString *string = [NSString stringWithFormat:@"%@ - didFailLoadingWithError: %@", identifier, [error _drt_descriptionSuitableForTestResult]]; |
|
225 printf("%s\n", [string UTF8String]); |
|
226 } |
|
227 } |
|
228 |
|
229 - (void)webView: (WebView *)wv plugInFailedWithError:(NSError *)error dataSource:(WebDataSource *)dataSource |
|
230 { |
|
231 // The call to -display here simulates the "Plug-in not found" sheet that Safari shows. |
|
232 // It is used for platform/mac/plugins/update-widget-from-style-recalc.html |
|
233 [wv display]; |
|
234 } |
|
235 |
|
236 -(NSCachedURLResponse *) webView: (WebView *)wv resource:(id)identifier willCacheResponse:(NSCachedURLResponse *)response fromDataSource:(WebDataSource *)dataSource |
|
237 { |
|
238 if (!done && gLayoutTestController->dumpWillCacheResponse()) { |
|
239 NSString *string = [NSString stringWithFormat:@"%@ - willCacheResponse: called", identifier]; |
|
240 printf("%s\n", [string UTF8String]); |
|
241 } |
|
242 return response; |
|
243 } |
|
244 |
|
245 @end |
|