| /* |
| * Copyright (C) 2005, 2006 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. |
| * 3. Neither the name of Apple Inc. ("Apple") nor the names of |
| * its contributors may be used to endorse or promote products derived |
| * from this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY APPLE 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 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 "WebPluginController.h" |
| |
| #import "DOMNodeInternal.h" |
| #import "WebBasePluginPackage.h" |
| #import "WebDataSourceInternal.h" |
| #import "WebFrameInternal.h" |
| #import "WebFrameView.h" |
| #import "WebHTMLViewPrivate.h" |
| #import "WebKitErrorsPrivate.h" |
| #import "WebKitLogging.h" |
| #import "WebNSObjectExtras.h" |
| #import "WebNSURLExtras.h" |
| #import "WebNSViewExtras.h" |
| #import "WebPlugin.h" |
| #import "WebPluginContainer.h" |
| #import "WebPluginContainerCheck.h" |
| #import "WebPluginPackage.h" |
| #import "WebPluginViewFactory.h" |
| #import "WebUIDelegate.h" |
| #import "WebViewInternal.h" |
| #import <Foundation/NSURLRequest.h> |
| #import <JavaScriptCore/JSLock.h> |
| #import <WebCore/CommonVM.h> |
| #import <WebCore/DocumentLoader.h> |
| #import <WebCore/Frame.h> |
| #import <WebCore/FrameLoadRequest.h> |
| #import <WebCore/FrameLoader.h> |
| #import <WebCore/HTMLMediaElement.h> |
| #import <WebCore/HTMLNames.h> |
| #import <WebCore/ResourceRequest.h> |
| #import <WebCore/ScriptController.h> |
| #import <WebCore/UserGestureIndicator.h> |
| #import <WebCore/WebCoreURLResponse.h> |
| #import <objc/runtime.h> |
| #import <wtf/text/WTFString.h> |
| |
| #if PLATFORM(IOS_FAMILY) |
| #import "DOMElementInternal.h" |
| #import "WebUIKitDelegate.h" |
| #import <WebCore/AudioSession.h> |
| #import <WebCore/FrameView.h> |
| #import <WebCore/GraphicsLayer.h> |
| #import <WebCore/RuntimeApplicationChecks.h> |
| #import <WebCore/WebCoreThreadRun.h> |
| #import <wtf/SoftLinking.h> |
| #endif |
| |
| @interface NSView (PluginSecrets) |
| - (void)setContainingWindow:(NSWindow *)w; |
| @end |
| |
| #if !PLATFORM(IOS_FAMILY) |
| // For compatibility only. |
| @interface NSObject (OldPluginAPI) |
| + (NSView *)pluginViewWithArguments:(NSDictionary *)arguments; |
| @end |
| #endif |
| |
| @interface NSView (OldPluginAPI) |
| - (void)pluginInitialize; |
| - (void)pluginStart; |
| - (void)pluginStop; |
| - (void)pluginDestroy; |
| @end |
| |
| #if !PLATFORM(IOS_FAMILY) |
| static bool isKindOfClass(id, NSString* className); |
| static void installFlip4MacPlugInWorkaroundIfNecessary(); |
| #endif |
| |
| |
| static NSMutableSet *pluginViews = nil; |
| |
| #if PLATFORM(IOS_FAMILY) |
| static void initializeAudioSession() |
| { |
| static bool wasAudioSessionInitialized; |
| if (wasAudioSessionInitialized) |
| return; |
| |
| wasAudioSessionInitialized = true; |
| if (!WebCore::IOSApplication::isMobileSafari()) |
| return; |
| |
| WebCore::AudioSession::sharedSession().setCategory(WebCore::AudioSession::MediaPlayback, WebCore::RouteSharingPolicy::Default); |
| } |
| #endif |
| |
| @implementation WebPluginController |
| |
| - (NSView *)plugInViewWithArguments:(NSDictionary *)arguments fromPluginPackage:(WebPluginPackage *)pluginPackage |
| { |
| #if PLATFORM(IOS_FAMILY) |
| initializeAudioSession(); |
| #endif |
| |
| [pluginPackage load]; |
| |
| NSView *view = nil; |
| |
| #if PLATFORM(IOS_FAMILY) |
| { |
| WebView *webView = [_documentView _webView]; |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| view = [[webView _UIKitDelegateForwarder] webView:webView plugInViewWithArguments:arguments fromPlugInPackage:pluginPackage]; |
| } |
| #else |
| Class viewFactory = [pluginPackage viewFactory]; |
| if ([viewFactory respondsToSelector:@selector(plugInViewWithArguments:)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| view = [viewFactory plugInViewWithArguments:arguments]; |
| } else if ([viewFactory respondsToSelector:@selector(pluginViewWithArguments:)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| view = [viewFactory pluginViewWithArguments:arguments]; |
| } |
| #endif |
| |
| if (view == nil) { |
| return nil; |
| } |
| |
| if (pluginViews == nil) { |
| pluginViews = [[NSMutableSet alloc] init]; |
| } |
| [pluginViews addObject:view]; |
| |
| return view; |
| } |
| |
| #if PLATFORM(IOS_FAMILY) |
| + (void)addPlugInView:(NSView *)view |
| { |
| if (pluginViews == nil) |
| pluginViews = [[NSMutableSet alloc] init]; |
| |
| ASSERT(view); |
| if (view) |
| [pluginViews addObject:view]; |
| } |
| #endif |
| |
| + (BOOL)isPlugInView:(NSView *)view |
| { |
| return [pluginViews containsObject:view]; |
| } |
| |
| - (id)initWithDocumentView:(NSView *)view |
| { |
| self = [super init]; |
| if (!self) |
| return nil; |
| _documentView = view; |
| _views = [[NSMutableArray alloc] init]; |
| _checksInProgress = (NSMutableSet *)CFSetCreateMutable(NULL, 0, NULL); |
| return self; |
| } |
| |
| - (void)setDataSource:(WebDataSource *)dataSource |
| { |
| _dataSource = dataSource; |
| } |
| |
| - (void)dealloc |
| { |
| [_views release]; |
| [_checksInProgress release]; |
| [super dealloc]; |
| } |
| |
| #if PLATFORM(IOS_FAMILY) |
| - (BOOL)plugInsAreRunning |
| { |
| NSUInteger pluginViewCount = [_views count]; |
| return _started && pluginViewCount; |
| } |
| |
| - (CALayer *)superlayerForPluginView:(NSView *)view |
| { |
| auto* coreFrame = core([self webFrame]); |
| auto* coreView = coreFrame ? coreFrame->view() : nullptr; |
| if (!coreView) |
| return nil; |
| |
| // Get a GraphicsLayer; |
| WebCore::GraphicsLayer* layerForWidget = coreView->graphicsLayerForPlatformWidget(view); |
| if (!layerForWidget) |
| return nil; |
| |
| return layerForWidget->platformLayer(); |
| } |
| #endif |
| |
| - (void)stopOnePlugin:(NSView *)view |
| { |
| if ([view respondsToSelector:@selector(webPlugInStop)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view webPlugInStop]; |
| } else if ([view respondsToSelector:@selector(pluginStop)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view pluginStop]; |
| } |
| } |
| |
| #if PLATFORM(IOS_FAMILY) |
| - (void)stopOnePluginForPageCache:(NSView *)view |
| { |
| if ([view respondsToSelector:@selector(webPlugInStopForPageCache)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view webPlugInStopForPageCache]; |
| } else |
| [self stopOnePlugin:view]; |
| } |
| #endif |
| |
| - (void)destroyOnePlugin:(NSView *)view |
| { |
| if ([view respondsToSelector:@selector(webPlugInDestroy)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view webPlugInDestroy]; |
| } else if ([view respondsToSelector:@selector(pluginDestroy)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view pluginDestroy]; |
| } |
| } |
| |
| - (void)startAllPlugins |
| { |
| if (_started) |
| return; |
| |
| if ([_views count] > 0) |
| LOG(Plugins, "starting WebKit plugins : %@", [_views description]); |
| |
| int count = [_views count]; |
| for (int i = 0; i < count; i++) { |
| id aView = [_views objectAtIndex:i]; |
| if ([aView respondsToSelector:@selector(webPlugInStart)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [aView webPlugInStart]; |
| } else if ([aView respondsToSelector:@selector(pluginStart)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [aView pluginStart]; |
| } |
| } |
| _started = YES; |
| } |
| |
| - (void)stopAllPlugins |
| { |
| if (!_started) |
| return; |
| |
| if ([_views count] > 0) { |
| LOG(Plugins, "stopping WebKit plugins: %@", [_views description]); |
| } |
| |
| int viewsCount = [_views count]; |
| for (int i = 0; i < viewsCount; i++) |
| [self stopOnePlugin:[_views objectAtIndex:i]]; |
| |
| _started = NO; |
| } |
| |
| #if PLATFORM(IOS_FAMILY) |
| - (void)stopPluginsForPageCache |
| { |
| if (!_started) |
| return; |
| |
| NSUInteger viewsCount = [_views count]; |
| if (viewsCount > 0) |
| LOG(Plugins, "stopping WebKit plugins for PageCache: %@", [_views description]); |
| |
| for (NSUInteger i = 0; i < viewsCount; ++i) |
| [self stopOnePluginForPageCache:[_views objectAtIndex:i]]; |
| |
| _started = NO; |
| } |
| |
| - (void)restorePluginsFromCache |
| { |
| WebView *webView = [_documentView _webView]; |
| |
| NSUInteger viewsCount = [_views count]; |
| if (viewsCount > 0) |
| LOG(Plugins, "restoring WebKit plugins from PageCache: %@", [_views description]); |
| |
| for (NSUInteger i = 0; i < viewsCount; ++i) |
| [[webView _UIKitDelegateForwarder] webView:webView willAddPlugInView:[_views objectAtIndex:i]]; |
| } |
| #endif // PLATFORM(IOS_FAMILY) |
| |
| - (void)addPlugin:(NSView *)view |
| { |
| if (!_documentView) { |
| LOG_ERROR("can't add a plug-in to a defunct WebPluginController"); |
| return; |
| } |
| |
| if (![_views containsObject:view]) { |
| [_views addObject:view]; |
| #if !PLATFORM(IOS_FAMILY) |
| [[_documentView _webView] addPluginInstanceView:view]; |
| #endif |
| |
| #if !PLATFORM(IOS_FAMILY) |
| BOOL oldDefersCallbacks = [[self webView] defersCallbacks]; |
| if (!oldDefersCallbacks) |
| [[self webView] setDefersCallbacks:YES]; |
| |
| if (isKindOfClass(view, @"WmvPlugin")) |
| installFlip4MacPlugInWorkaroundIfNecessary(); |
| #endif |
| |
| LOG(Plugins, "initializing plug-in %@", view); |
| if ([view respondsToSelector:@selector(webPlugInInitialize)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view webPlugInInitialize]; |
| } else if ([view respondsToSelector:@selector(pluginInitialize)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view pluginInitialize]; |
| } |
| |
| #if !PLATFORM(IOS_FAMILY) |
| if (!oldDefersCallbacks) |
| [[self webView] setDefersCallbacks:NO]; |
| #endif |
| |
| if (_started) { |
| LOG(Plugins, "starting plug-in %@", view); |
| if ([view respondsToSelector:@selector(webPlugInStart)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view webPlugInStart]; |
| } else if ([view respondsToSelector:@selector(pluginStart)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view pluginStart]; |
| } |
| |
| if ([view respondsToSelector:@selector(setContainingWindow:)]) { |
| JSC::JSLock::DropAllLocks dropAllLocks(WebCore::commonVM()); |
| [view setContainingWindow:[_documentView window]]; |
| } |
| } |
| } |
| } |
| |
| - (void)destroyPlugin:(NSView *)view |
| { |
| if ([_views containsObject:view]) { |
| if (_started) |
| [self stopOnePlugin:view]; |
| [self destroyOnePlugin:view]; |
| |
| #if ENABLE(NETSCAPE_PLUGIN_API) |
| if (auto* frame = core([self webFrame])) |
| frame->script().cleanupScriptObjectsForPlugin(self); |
| #endif |
| |
| [pluginViews removeObject:view]; |
| #if !PLATFORM(IOS_FAMILY) |
| [[_documentView _webView] removePluginInstanceView:view]; |
| #endif |
| [_views removeObject:view]; |
| } |
| } |
| |
| - (void)_webPluginContainerCancelCheckIfAllowedToLoadRequest:(id)checkIdentifier |
| { |
| [checkIdentifier cancel]; |
| [_checksInProgress removeObject:checkIdentifier]; |
| } |
| |
| static void cancelOutstandingCheck(const void *item, void *context) |
| { |
| [(id)item cancel]; |
| } |
| |
| - (void)_cancelOutstandingChecks |
| { |
| if (_checksInProgress) { |
| CFSetApplyFunction((__bridge CFSetRef)_checksInProgress, cancelOutstandingCheck, NULL); |
| [_checksInProgress release]; |
| _checksInProgress = nil; |
| } |
| } |
| |
| - (void)destroyAllPlugins |
| { |
| [self stopAllPlugins]; |
| |
| if ([_views count] > 0) { |
| LOG(Plugins, "destroying WebKit plugins: %@", [_views description]); |
| } |
| |
| [self _cancelOutstandingChecks]; |
| |
| int viewsCount = [_views count]; |
| for (int i = 0; i < viewsCount; i++) { |
| id aView = [_views objectAtIndex:i]; |
| [self destroyOnePlugin:aView]; |
| |
| #if ENABLE(NETSCAPE_PLUGIN_API) |
| if (auto* frame = core([self webFrame])) |
| frame->script().cleanupScriptObjectsForPlugin(self); |
| #endif |
| |
| [pluginViews removeObject:aView]; |
| #if !PLATFORM(IOS_FAMILY) |
| [[_documentView _webView] removePluginInstanceView:aView]; |
| #endif |
| } |
| |
| #if !PLATFORM(IOS_FAMILY) |
| [_views makeObjectsPerformSelector:@selector(removeFromSuperviewWithoutNeedingDisplay)]; |
| #else |
| [_views makeObjectsPerformSelector:@selector(removeFromSuperview)]; |
| #endif |
| [_views release]; |
| _views = nil; |
| |
| _documentView = nil; |
| } |
| |
| #if PLATFORM(IOS_FAMILY) |
| - (BOOL)processingUserGesture |
| { |
| return WebCore::UserGestureIndicator::processingUserGesture(); |
| } |
| #endif |
| |
| - (id)_webPluginContainerCheckIfAllowedToLoadRequest:(NSURLRequest *)request inFrame:(NSString *)target resultObject:(id)obj selector:(SEL)selector |
| { |
| WebPluginContainerCheck *check = [WebPluginContainerCheck checkWithRequest:request target:target resultObject:obj selector:selector controller:self contextInfo:nil]; |
| [_checksInProgress addObject:check]; |
| [check start]; |
| |
| return check; |
| } |
| |
| - (void)webPlugInContainerLoadRequest:(NSURLRequest *)request inFrame:(NSString *)target |
| { |
| if (!request) { |
| LOG_ERROR("nil URL passed"); |
| return; |
| } |
| if (!_documentView) { |
| LOG_ERROR("could not load URL %@ because plug-in has already been destroyed", request); |
| return; |
| } |
| WebFrame *frame = [_dataSource webFrame]; |
| if (!frame) { |
| LOG_ERROR("could not load URL %@ because plug-in has already been stopped", request); |
| return; |
| } |
| if (!target) { |
| target = @"_top"; |
| } |
| NSString *JSString = [[request URL] _webkit_scriptIfJavaScriptURL]; |
| if (JSString) { |
| if ([frame findFrameNamed:target] != frame) { |
| LOG_ERROR("JavaScript requests can only be made on the frame that contains the plug-in"); |
| return; |
| } |
| [frame _stringByEvaluatingJavaScriptFromString:JSString]; |
| } else { |
| if (!request) { |
| LOG_ERROR("could not load URL %@", [request URL]); |
| return; |
| } |
| WebCore::FrameLoadRequest frameLoadRequest { *core(frame), request, WebCore::ShouldOpenExternalURLsPolicy::ShouldNotAllow }; |
| frameLoadRequest.setFrameName(target); |
| frameLoadRequest.setShouldCheckNewWindowPolicy(true); |
| core(frame)->loader().load(WTFMove(frameLoadRequest)); |
| } |
| } |
| |
| #if PLATFORM(IOS_FAMILY) |
| - (void)webPlugInContainerWillShowFullScreenForView:(id)plugInView |
| { |
| WebView *webView = [_dataSource _webView]; |
| [[webView _UIKitDelegateForwarder] webView:webView willShowFullScreenForPlugInView:plugInView]; |
| } |
| |
| - (void)webPlugInContainerDidHideFullScreenForView:(id)plugInView |
| { |
| WebView *webView = [_dataSource _webView]; |
| [[webView _UIKitDelegateForwarder] webView:webView didHideFullScreenForPlugInView:plugInView]; |
| } |
| #endif |
| |
| - (void)webPlugInContainerShowStatus:(NSString *)message |
| { |
| if (!message) |
| message = @""; |
| |
| WebView *v = [_dataSource _webView]; |
| [[v _UIDelegateForwarder] webView:v setStatusText:message]; |
| } |
| |
| // For compatibility only. |
| - (void)showStatus:(NSString *)message |
| { |
| [self webPlugInContainerShowStatus:message]; |
| } |
| |
| #if !PLATFORM(IOS_FAMILY) |
| - (NSColor *)webPlugInContainerSelectionColor |
| { |
| bool primary = true; |
| if (auto* frame = core([self webFrame])) |
| primary = frame->selection().isFocusedAndActive(); |
| return primary ? [NSColor selectedTextBackgroundColor] : [NSColor secondarySelectedControlColor]; |
| } |
| |
| // For compatibility only. |
| - (NSColor *)selectionColor |
| { |
| return [self webPlugInContainerSelectionColor]; |
| } |
| #endif |
| |
| - (WebFrame *)webFrame |
| { |
| return [_dataSource webFrame]; |
| } |
| |
| - (WebView *)webView |
| { |
| return [[self webFrame] webView]; |
| } |
| |
| - (NSString *)URLPolicyCheckReferrer |
| { |
| NSURL *responseURL = [[[[self webFrame] _dataSource] response] URL]; |
| ASSERT(responseURL); |
| return [responseURL _web_originalDataAsString]; |
| } |
| |
| - (void)pluginView:(NSView *)pluginView receivedResponse:(NSURLResponse *)response |
| { |
| if ([pluginView respondsToSelector:@selector(webPlugInMainResourceDidReceiveResponse:)]) |
| [pluginView webPlugInMainResourceDidReceiveResponse:response]; |
| else { |
| // Cancel the load since this plug-in does its own loading. |
| // FIXME: See <rdar://problem/4258008> for a problem with this. |
| NSError *error = [[NSError alloc] _initWithPluginErrorCode:WebKitErrorPlugInWillHandleLoad |
| contentURL:[response URL] |
| pluginPageURL:nil |
| pluginName:nil // FIXME: Get this from somewhere |
| MIMEType:[response MIMEType]]; |
| [_dataSource _documentLoader]->cancelMainResourceLoad(error); |
| [error release]; |
| } |
| } |
| |
| - (void)pluginView:(NSView *)pluginView receivedData:(NSData *)data |
| { |
| if ([pluginView respondsToSelector:@selector(webPlugInMainResourceDidReceiveData:)]) |
| [pluginView webPlugInMainResourceDidReceiveData:data]; |
| } |
| |
| - (void)pluginView:(NSView *)pluginView receivedError:(NSError *)error |
| { |
| if ([pluginView respondsToSelector:@selector(webPlugInMainResourceDidFailWithError:)]) |
| [pluginView webPlugInMainResourceDidFailWithError:error]; |
| } |
| |
| - (void)pluginViewFinishedLoading:(NSView *)pluginView |
| { |
| if ([pluginView respondsToSelector:@selector(webPlugInMainResourceDidFinishLoading)]) |
| [pluginView webPlugInMainResourceDidFinishLoading]; |
| } |
| |
| @end |
| |
| #if !PLATFORM(IOS_FAMILY) |
| static bool isKindOfClass(id object, NSString *className) |
| { |
| Class cls = NSClassFromString(className); |
| |
| if (!cls) |
| return false; |
| |
| return [object isKindOfClass:cls]; |
| } |
| |
| |
| // Existing versions of the Flip4Mac WebKit plug-in have an object lifetime bug related to an NSAlert that is |
| // used to notify the user about updates to the plug-in. This bug can result in Safari crashing if the page |
| // containing the plug-in navigates while the alert is displayed (<rdar://problem/7313430>). |
| // |
| // The gist of the bug is thus: Flip4Mac sets an instance of the TSUpdateCheck class as the modal delegate of the |
| // NSAlert instance. This TSUpdateCheck instance itself has a delegate. The delegate is set to the WmvPlugin |
| // instance which is the NSView subclass that is exposed to WebKit as the plug-in view. Since this relationship |
| // is that of delegates the TSUpdateCheck does not retain the WmvPlugin. This leads to a bug if the WmvPlugin |
| // instance is destroyed before the TSUpdateCheck instance as the TSUpdateCheck instance will be left with a |
| // pointer to a stale object. This will happen if a page containing the Flip4Mac plug-in triggers a navigation |
| // while the update sheet is visible as the WmvPlugin instance is removed from the view hierarchy and there are |
| // no other references to keep the object alive. |
| // |
| // We work around this bug by patching the following two messages: |
| // |
| // 1) -[NSAlert beginSheetModalForWindow:modalDelegate:didEndSelector:contextInfo:] |
| // 2) -[TSUpdateCheck alertDidEnd:returnCode:contextInfo:] |
| // |
| // Our override of 1) detects whether it is Flip4Mac's update sheet triggering the alert by checking whether the |
| // modal delegate is an instance of TSUpdateCheck. If it is, it retains the modal delegate's delegate. |
| // |
| // Our override of 2) then autoreleases the delegate, balancing the retain we added in 1). |
| // |
| // These two overrides have the effect of ensuring that the WmvPlugin instance will always outlive the TSUpdateCheck |
| // instance, preventing the TSUpdateCheck instance from accessing a stale delegate pointer and crashing the application. |
| |
| |
| typedef void (*beginSheetModalForWindowIMP)(id, SEL, NSWindow *, id, SEL, void*); |
| static beginSheetModalForWindowIMP original_NSAlert_beginSheetModalForWindow_modalDelegate_didEndSelector_contextInfo_; |
| |
| typedef void (*alertDidEndIMP)(id, SEL, NSAlert *, NSInteger, void*); |
| static alertDidEndIMP original_TSUpdateCheck_alertDidEnd_returnCode_contextInfo_; |
| |
| @class TSUpdateCheck; |
| |
| static void WebKit_TSUpdateCheck_alertDidEnd_returnCode_contextInfo_(id object, SEL selector, NSAlert *alert, NSInteger returnCode, void* contextInfo) |
| { |
| ALLOW_DEPRECATED_DECLARATIONS_BEGIN |
| [[(TSUpdateCheck *)object delegate] autorelease]; |
| ALLOW_DEPRECATED_DECLARATIONS_END |
| |
| original_TSUpdateCheck_alertDidEnd_returnCode_contextInfo_(object, selector, alert, returnCode, contextInfo); |
| } |
| |
| static void WebKit_NSAlert_beginSheetModalForWindow_modalDelegate_didEndSelector_contextInfo_(id object, SEL selector, NSWindow *window, id modalDelegate, SEL didEndSelector, void* contextInfo) |
| { |
| ALLOW_DEPRECATED_DECLARATIONS_BEGIN |
| if (isKindOfClass(modalDelegate, @"TSUpdateCheck")) |
| [[(TSUpdateCheck *)modalDelegate delegate] retain]; |
| ALLOW_DEPRECATED_DECLARATIONS_END |
| |
| original_NSAlert_beginSheetModalForWindow_modalDelegate_didEndSelector_contextInfo_(object, selector, window, modalDelegate, didEndSelector, contextInfo); |
| } |
| |
| static void installFlip4MacPlugInWorkaroundIfNecessary() |
| { |
| static bool hasInstalledFlip4MacPlugInWorkaround; |
| if (!hasInstalledFlip4MacPlugInWorkaround) { |
| Class TSUpdateCheck = objc_lookUpClass("TSUpdateCheck"); |
| if (!TSUpdateCheck) |
| return; |
| |
| IGNORE_WARNINGS_BEGIN("undeclared-selector") |
| Method methodToPatch = class_getInstanceMethod(TSUpdateCheck, @selector(alertDidEnd:returnCode:contextInfo:)); |
| IGNORE_WARNINGS_END |
| if (!methodToPatch) |
| return; |
| |
| IMP originalMethod = method_setImplementation(methodToPatch, reinterpret_cast<IMP>(WebKit_TSUpdateCheck_alertDidEnd_returnCode_contextInfo_)); |
| original_TSUpdateCheck_alertDidEnd_returnCode_contextInfo_ = reinterpret_cast<alertDidEndIMP>(originalMethod); |
| |
| methodToPatch = class_getInstanceMethod(objc_getRequiredClass("NSAlert"), @selector(beginSheetModalForWindow:modalDelegate:didEndSelector:contextInfo:)); |
| originalMethod = method_setImplementation(methodToPatch, reinterpret_cast<IMP>(WebKit_NSAlert_beginSheetModalForWindow_modalDelegate_didEndSelector_contextInfo_)); |
| original_NSAlert_beginSheetModalForWindow_modalDelegate_didEndSelector_contextInfo_ = reinterpret_cast<beginSheetModalForWindowIMP>(originalMethod); |
| |
| hasInstalledFlip4MacPlugInWorkaround = true; |
| } |
| } |
| #endif // !PLATFORM(IOS_FAMILY) |