]> git.saurik.com Git - cydia.git/blob - CyteKit/WebView.mm
Fix underlying Apple UIWebViewWebViewDelegate on 2.x retain/release bug.
[cydia.git] / CyteKit / WebView.mm
1 /* Cydia - iPhone UIKit Front-End for Debian APT
2 * Copyright (C) 2008-2011 Jay Freeman (saurik)
3 */
4
5 /* Modified BSD License {{{ */
6 /*
7 * Redistribution and use in source and binary
8 * forms, with or without modification, are permitted
9 * provided that the following conditions are met:
10 *
11 * 1. Redistributions of source code must retain the
12 * above copyright notice, this list of conditions
13 * and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the
15 * above copyright notice, this list of conditions
16 * and the following disclaimer in the documentation
17 * and/or other materials provided with the
18 * distribution.
19 * 3. The name of the author may not be used to endorse
20 * or promote products derived from this software
21 * without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
24 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
25 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
26 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
28 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
29 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
30 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
31 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
32 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
34 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
35 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
36 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37 */
38 /* }}} */
39
40 #include "CyteKit/dispatchEvent.h"
41 #include "CyteKit/WebView.h"
42
43 #include <CydiaSubstrate/CydiaSubstrate.h>
44
45 #include "iPhonePrivate.h"
46
47 // CYWebPolicyDecision* {{{
48 @interface CYWebPolicyDecisionMediator : NSObject <
49 WebPolicyDecisionListener
50 > {
51 id<WebPolicyDecisionListener> listener_;
52 CYWebPolicyDecision decision_;
53 }
54
55 - (id) initWithListener:(id<WebPolicyDecisionListener>)listener;
56
57 - (CYWebPolicyDecision) decision;
58 - (bool) decided;
59 - (bool) decide;
60
61 @end
62
63 @implementation CYWebPolicyDecisionMediator
64
65 - (id) initWithListener:(id<WebPolicyDecisionListener>)listener {
66 if ((self = [super init]) != nil) {
67 listener_ = listener;
68 } return self;
69 }
70
71 - (CYWebPolicyDecision) decision {
72 return decision_;
73 }
74
75 - (bool) decided {
76 return decision_ != CYWebPolicyDecisionUnknown;
77 }
78
79 - (bool) decide {
80 switch (decision_) {
81 case CYWebPolicyDecisionUnknown:
82 default:
83 NSLog(@"CYWebPolicyDecisionUnknown");
84 return false;
85
86 case CYWebPolicyDecisionDownload: [listener_ download]; break;
87 case CYWebPolicyDecisionIgnore: [listener_ ignore]; break;
88 case CYWebPolicyDecisionUse: [listener_ use]; break;
89 }
90
91 return true;
92 }
93
94 - (void) download {
95 decision_ = CYWebPolicyDecisionDownload;
96 }
97
98 - (void) ignore {
99 decision_ = CYWebPolicyDecisionIgnore;
100 }
101
102 - (void) use {
103 decision_ = CYWebPolicyDecisionUse;
104 }
105
106 @end
107 // }}}
108
109 @implementation CyteWebView : UIWebView
110
111 #if ShowInternals
112 #include "CyteKit/UCInternal.h"
113 #endif
114
115 - (id) initWithFrame:(CGRect)frame {
116 if ((self = [super initWithFrame:frame]) != nil) {
117 } return self;
118 }
119
120 - (void) dealloc {
121 if (kCFCoreFoundationVersionNumber < kCFCoreFoundationVersionNumber_iPhoneOS_3_0) {
122 UIWebViewInternal *&_internal(MSHookIvar<UIWebViewInternal *>(self, "_internal"));
123 if (&_internal != NULL) {
124 UIWebViewWebViewDelegate *&webViewDelegate(MSHookIvar<UIWebViewWebViewDelegate *>(_internal, "webViewDelegate"));
125 if (&webViewDelegate != NULL)
126 [webViewDelegate _clearUIWebView];
127 }
128 }
129
130 [super dealloc];
131 }
132
133 - (NSString *) description {
134 return [NSString stringWithFormat:@"<%s: %p, %@>", class_getName([self class]), self, [[[self request] URL] absoluteString]];
135 }
136
137 - (id<CyteWebViewDelegate>) delegate {
138 return (id<CyteWebViewDelegate>) [super delegate];
139 }
140
141 /*- (WebView *) webView:(WebView *)view createWebViewWithRequest:(NSURLRequest *)request {
142 id<CyteWebViewDelegate> delegate([self delegate]);
143 WebView *created(nil);
144 if (created == nil && [delegate respondsToSelector:@selector(webView:createWebViewWithRequest:)])
145 created = [delegate webView:view createWebViewWithRequest:request];
146 if (created == nil && [UIWebView instancesRespondToSelector:@selector(webView:createWebViewWithRequest:)])
147 created = [super webView:view createWebViewWithRequest:request];
148 return created;
149 }*/
150
151 // webView:addMessageToConsole: (X.Xx) {{{
152 static void $UIWebViewWebViewDelegate$webView$addMessageToConsole$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, NSDictionary *message) {
153 UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
154 if ([uiWebView respondsToSelector:@selector(webView:addMessageToConsole:)])
155 [uiWebView webView:view addMessageToConsole:message];
156 }
157
158 - (void) webView:(WebView *)view addMessageToConsole:(NSDictionary *)message {
159 id<CyteWebViewDelegate> delegate([self delegate]);
160 if ([delegate respondsToSelector:@selector(webView:addMessageToConsole:)])
161 [delegate webView:view addMessageToConsole:message];
162 if ([UIWebView instancesRespondToSelector:@selector(webView:addMessageToConsole:)])
163 [super webView:view addMessageToConsole:message];
164 }
165 // }}}
166 // webView:decidePolicyForNavigationAction:request:frame:decisionListener: (2.0+) {{{
167 - (void) webView:(WebView *)view decidePolicyForNavigationAction:(NSDictionary *)action request:(NSURLRequest *)request frame:(WebFrame *)frame decisionListener:(id<WebPolicyDecisionListener>)listener {
168 id<CyteWebViewDelegate> delegate([self delegate]);
169 CYWebPolicyDecisionMediator *mediator([[[CYWebPolicyDecisionMediator alloc] initWithListener:listener] autorelease]);
170 if (![mediator decided] && [delegate respondsToSelector:@selector(webView:decidePolicyForNavigationAction:request:frame:decisionListener:)])
171 [delegate webView:view decidePolicyForNavigationAction:action request:request frame:frame decisionListener:mediator];
172 if (![mediator decided] && [UIWebView instancesRespondToSelector:@selector(webView:decidePolicyForNavigationAction:request:frame:decisionListener:)])
173 [super webView:view decidePolicyForNavigationAction:action request:request frame:frame decisionListener:mediator];
174 [delegate webView:view didDecidePolicy:[mediator decision] forNavigationAction:action request:request frame:frame];
175 [mediator decide];
176 }
177 // }}}
178 // webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener: (3.0+) {{{
179 static void $UIWebViewWebViewDelegate$webView$decidePolicyForNewWindowAction$request$newFrameName$decisionListener$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, NSDictionary *action, NSURLRequest *request, NSString *frame, id<WebPolicyDecisionListener> listener) {
180 UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
181 if ([uiWebView respondsToSelector:@selector(webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener:)])
182 [uiWebView webView:view decidePolicyForNewWindowAction:action request:request newFrameName:frame decisionListener:listener];
183 }
184
185 - (void) webView:(WebView *)view decidePolicyForNewWindowAction:(NSDictionary *)action request:(NSURLRequest *)request newFrameName:(NSString *)frame decisionListener:(id<WebPolicyDecisionListener>)listener {
186 id<CyteWebViewDelegate> delegate([self delegate]);
187 CYWebPolicyDecisionMediator *mediator([[[CYWebPolicyDecisionMediator alloc] initWithListener:listener] autorelease]);
188 if (![mediator decided] && [delegate respondsToSelector:@selector(webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener:)])
189 [delegate webView:view decidePolicyForNewWindowAction:action request:request newFrameName:frame decisionListener:mediator];
190 if (![mediator decided] && [UIWebView instancesRespondToSelector:@selector(webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener:)])
191 [super webView:view decidePolicyForNewWindowAction:action request:request newFrameName:frame decisionListener:mediator];
192 [mediator decide];
193 }
194 // }}}
195 // webView:didClearWindowObject:forFrame: (3.2+) {{{
196 static void $UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, WebScriptObject *window, WebFrame *frame) {
197 UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
198 if ([uiWebView respondsToSelector:@selector(webView:didClearWindowObject:forFrame:)])
199 [uiWebView webView:view didClearWindowObject:window forFrame:frame];
200 }
201
202 - (void) webView:(WebView *)view didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame {
203 id<CyteWebViewDelegate> delegate([self delegate]);
204 if ([delegate respondsToSelector:@selector(webView:didClearWindowObject:forFrame:)])
205 [delegate webView:view didClearWindowObject:window forFrame:frame];
206 if ([UIWebView instancesRespondToSelector:@selector(webView:didClearWindowObject:forFrame:)])
207 [super webView:view didClearWindowObject:window forFrame:frame];
208 }
209 // }}}
210 // webView:didCommitLoadForFrame: (3.0+) {{{
211 static void $UIWebViewWebViewDelegate$webView$didCommitLoadForFrame$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, WebFrame *frame) {
212 UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
213 if ([uiWebView respondsToSelector:@selector(webView:didCommitLoadForFrame:)])
214 [uiWebView webView:view didCommitLoadForFrame:frame];
215 }
216
217 - (void) webView:(WebView *)view didCommitLoadForFrame:(WebFrame *)frame {
218 id<CyteWebViewDelegate> delegate([self delegate]);
219 if ([delegate respondsToSelector:@selector(webView:didCommitLoadForFrame:)])
220 [delegate webView:view didCommitLoadForFrame:frame];
221 if ([UIWebView instancesRespondToSelector:@selector(webView:didCommitLoadForFrame:)])
222 [super webView:view didCommitLoadForFrame:frame];
223 }
224 // }}}
225 // webView:didFailLoadWithError:forFrame: (2.0+) {{{
226 - (void) webView:(WebView *)view didFailLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
227 id<CyteWebViewDelegate> delegate([self delegate]);
228 if ([delegate respondsToSelector:@selector(webView:didFailLoadWithError:forFrame:)])
229 [delegate webView:view didFailLoadWithError:error forFrame:frame];
230 if ([UIWebView instancesRespondToSelector:@selector(webView:didFailLoadWithError:forFrame:)])
231 [super webView:view didFailLoadWithError:error forFrame:frame];
232 }
233 // }}}
234 // webView:didFailProvisionalLoadWithError:forFrame: (2.0+) {{{
235 - (void) webView:(WebView *)view didFailProvisionalLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
236 id<CyteWebViewDelegate> delegate([self delegate]);
237 if ([delegate respondsToSelector:@selector(webView:didFailProvisionalLoadWithError:forFrame:)])
238 [delegate webView:view didFailProvisionalLoadWithError:error forFrame:frame];
239 if ([UIWebView instancesRespondToSelector:@selector(webView:didFailProvisionalLoadWithError:forFrame:)])
240 [super webView:view didFailProvisionalLoadWithError:error forFrame:frame];
241 }
242 // }}}
243 // webView:didFinishLoadForFrame: (2.0+) {{{
244 - (void) webView:(WebView *)view didFinishLoadForFrame:(WebFrame *)frame {
245 id<CyteWebViewDelegate> delegate([self delegate]);
246 if ([delegate respondsToSelector:@selector(webView:didFinishLoadForFrame:)])
247 [delegate webView:view didFinishLoadForFrame:frame];
248 if ([UIWebView instancesRespondToSelector:@selector(webView:didFinishLoadForFrame:)])
249 [super webView:view didFinishLoadForFrame:frame];
250 }
251 // }}}
252 // webView:didReceiveTitle:forFrame: (3.2+) {{{
253 static void $UIWebViewWebViewDelegate$webView$didReceiveTitle$forFrame$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, NSString *title, WebFrame *frame) {
254 UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
255 if ([uiWebView respondsToSelector:@selector(webView:didReceiveTitle:forFrame:)])
256 [uiWebView webView:view didReceiveTitle:title forFrame:frame];
257 }
258
259 - (void) webView:(WebView *)view didReceiveTitle:(NSString *)title forFrame:(WebFrame *)frame {
260 id<CyteWebViewDelegate> delegate([self delegate]);
261 if ([delegate respondsToSelector:@selector(webView:didReceiveTitle:forFrame:)])
262 [delegate webView:view didReceiveTitle:title forFrame:frame];
263 if ([UIWebView instancesRespondToSelector:@selector(webView:didReceiveTitle:forFrame:)])
264 [super webView:view didReceiveTitle:title forFrame:frame];
265 }
266 // }}}
267 // webView:didStartProvisionalLoadForFrame: (2.0+) {{{
268 - (void) webView:(WebView *)view didStartProvisionalLoadForFrame:(WebFrame *)frame {
269 id<CyteWebViewDelegate> delegate([self delegate]);
270 if ([delegate respondsToSelector:@selector(webView:didStartProvisionalLoadForFrame:)])
271 [delegate webView:view didStartProvisionalLoadForFrame:frame];
272 if ([UIWebView instancesRespondToSelector:@selector(webView:didStartProvisionalLoadForFrame:)])
273 [super webView:view didStartProvisionalLoadForFrame:frame];
274 }
275 // }}}
276 // webView:resource:willSendRequest:redirectResponse:fromDataSource: (3.2+) {{{
277 static NSURLRequest *$UIWebViewWebViewDelegate$webView$resource$willSendRequest$redirectResponse$fromDataSource$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, id identifier, NSURLRequest *request, NSURLResponse *response, WebDataSource *source) {
278 UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
279 if ([uiWebView respondsToSelector:@selector(webView:resource:willSendRequest:redirectResponse:fromDataSource:)])
280 request = [uiWebView webView:view resource:identifier willSendRequest:request redirectResponse:response fromDataSource:source];
281 return request;
282 }
283
284 - (NSURLRequest *) webView:(WebView *)view resource:(id)identifier willSendRequest:(NSURLRequest *)request redirectResponse:(NSURLResponse *)response fromDataSource:(WebDataSource *)source {
285 id<CyteWebViewDelegate> delegate([self delegate]);
286 if ([UIWebView instancesRespondToSelector:@selector(webView:resource:willSendRequest:redirectResponse:fromDataSource:)])
287 request = [super webView:view resource:identifier willSendRequest:request redirectResponse:response fromDataSource:source];
288 if ([delegate respondsToSelector:@selector(webView:resource:willSendRequest:redirectResponse:fromDataSource:)])
289 request = [delegate webView:view resource:identifier willSendRequest:request redirectResponse:response fromDataSource:source];
290 return request;
291 }
292 // }}}
293 // webView:runJavaScriptAlertPanelWithMessage:initiatedByFrame: (2.1+) {{{
294 - (void) webView:(WebView *)view runJavaScriptAlertPanelWithMessage:(NSString *)message initiatedByFrame:(WebFrame *)frame {
295 [[self retain] autorelease];
296 id<CyteWebViewDelegate> delegate([self delegate]);
297 if ([UIWebView instancesRespondToSelector:@selector(webView:runJavaScriptAlertPanelWithMessage:initiatedByFrame:)])
298 if (
299 ![delegate respondsToSelector:@selector(webView:shouldRunJavaScriptAlertPanelWithMessage:initiatedByFrame:)] ||
300 [delegate webView:view shouldRunJavaScriptAlertPanelWithMessage:message initiatedByFrame:frame]
301 )
302 [super webView:view runJavaScriptAlertPanelWithMessage:message initiatedByFrame:frame];
303 }
304 // }}}
305 // webView:runJavaScriptConfirmPanelWithMessage:initiatedByFrame: (2.1+) {{{
306 - (BOOL) webView:(WebView *)view runJavaScriptConfirmPanelWithMessage:(NSString *)message initiatedByFrame:(WebFrame *)frame {
307 [[self retain] autorelease];
308 id<CyteWebViewDelegate> delegate([self delegate]);
309 if ([UIWebView instancesRespondToSelector:@selector(webView:runJavaScriptConfirmPanelWithMessage:initiatedByFrame:)])
310 if (
311 ![delegate respondsToSelector:@selector(webView:shouldRunJavaScriptConfirmPanelWithMessage:initiatedByFrame:)] ||
312 [delegate webView:view shouldRunJavaScriptConfirmPanelWithMessage:message initiatedByFrame:frame]
313 )
314 return [super webView:view runJavaScriptConfirmPanelWithMessage:message initiatedByFrame:frame];
315 return NO;
316 }
317 // }}}
318 // webView:runJavaScriptTextInputPanelWithPrompt:defaultText:initiatedByFrame: (2.1+) {{{
319 - (NSString *) webView:(WebView *)view runJavaScriptTextInputPanelWithPrompt:(NSString *)prompt defaultText:(NSString *)text initiatedByFrame:(WebFrame *)frame {
320 [[self retain] autorelease];
321 id<CyteWebViewDelegate> delegate([self delegate]);
322 if ([UIWebView instancesRespondToSelector:@selector(webView:runJavaScriptTextInputPanelWithPrompt:defaultText:initiatedByFrame:)])
323 if (
324 ![delegate respondsToSelector:@selector(webView:shouldRunJavaScriptTextInputPanelWithPrompt:defaultText:initiatedByFrame:)] ||
325 [delegate webView:view shouldRunJavaScriptTextInputPanelWithPrompt:prompt defaultText:text initiatedByFrame:frame]
326 )
327 return [super webView:view runJavaScriptTextInputPanelWithPrompt:prompt defaultText:text initiatedByFrame:frame];
328 return nil;
329 }
330 // }}}
331 // webViewClose: (3.2+) {{{
332 static void $UIWebViewWebViewDelegate$webViewClose$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view) {
333 UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
334 if ([uiWebView respondsToSelector:@selector(webViewClose:)])
335 [uiWebView webViewClose:view];
336 }
337
338 - (void) webViewClose:(WebView *)view {
339 id<CyteWebViewDelegate> delegate([self delegate]);
340 if ([delegate respondsToSelector:@selector(webViewClose:)])
341 [delegate webViewClose:view];
342 if ([UIWebView instancesRespondToSelector:@selector(webViewClose:)])
343 [super webViewClose:view];
344 }
345 // }}}
346
347 - (void) _updateViewSettings {
348 [super _updateViewSettings];
349
350 id<CyteWebViewDelegate> delegate([self delegate]);
351 if ([delegate respondsToSelector:@selector(webViewUpdateViewSettings:)])
352 [delegate webViewUpdateViewSettings:self];
353 }
354
355 - (void) dispatchEvent:(NSString *)event {
356 [[self _documentView] dispatchEvent:event];
357 }
358
359 - (void) reloadFromOrigin {
360 [[[self _documentView] webView] reloadFromOrigin:nil];
361 }
362
363 @end
364
365 static void $UIWebViewWebViewDelegate$_clearUIWebView(UIWebViewWebViewDelegate *self, SEL sel) {
366 MSHookIvar<UIWebView *>(self, "uiWebView") = nil;
367 }
368
369 __attribute__((__constructor__)) static void $() {
370 if (Class $UIWebViewWebViewDelegate = objc_getClass("UIWebViewWebViewDelegate")) {
371 class_addMethod($UIWebViewWebViewDelegate, @selector(webView:addMessageToConsole:), (IMP) &$UIWebViewWebViewDelegate$webView$addMessageToConsole$, "v16@0:4@8@12");
372 class_addMethod($UIWebViewWebViewDelegate, @selector(webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener:), (IMP) &$UIWebViewWebViewDelegate$webView$decidePolicyForNewWindowAction$request$newFrameName$decisionListener$, "v28@0:4@8@12@16@20@24");
373 class_addMethod($UIWebViewWebViewDelegate, @selector(webView:didClearWindowObject:forFrame:), (IMP) &$UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$, "v20@0:4@8@12@16");
374 class_addMethod($UIWebViewWebViewDelegate, @selector(webView:didCommitLoadForFrame:), (IMP) &$UIWebViewWebViewDelegate$webView$didCommitLoadForFrame$, "v16@0:4@8@12");
375 class_addMethod($UIWebViewWebViewDelegate, @selector(webView:didReceiveTitle:forFrame:), (IMP) &$UIWebViewWebViewDelegate$webView$didReceiveTitle$forFrame$, "v20@0:4@8@12@16");
376 class_addMethod($UIWebViewWebViewDelegate, @selector(webView:resource:willSendRequest:redirectResponse:fromDataSource:), (IMP) &$UIWebViewWebViewDelegate$webView$resource$willSendRequest$redirectResponse$fromDataSource$, "@28@0:4@8@12@16@20@24");
377 class_addMethod($UIWebViewWebViewDelegate, @selector(webViewClose:), (IMP) &$UIWebViewWebViewDelegate$webViewClose$, "v12@0:4@8");
378 class_addMethod($UIWebViewWebViewDelegate, @selector(_clearUIWebView), (IMP) &$UIWebViewWebViewDelegate$_clearUIWebView, "v8@0:4");
379 }
380 }