WebKitTools/WebKitLauncher/WebKitNightlyEnabler.m

   1 /*
   2  * Copyright (C) 2006, 2007 Apple Inc.  All rights reserved.
   3  * Copyright (C) 2006 Graham Dennis.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  * 1.  Redistributions of source code must retain the above copyright
  10  *     notice, this list of conditions and the following disclaimer.
  11  * 2.  Redistributions in binary form must reproduce the above copyright
  12  *     notice, this list of conditions and the following disclaimer in the
  13  *     documentation and/or other materials provided with the distribution.
  14  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
  15  *     its contributors may be used to endorse or promote products derived
  16  *     from this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  19  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  21  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  22  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  23  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  25  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29
  30 #import <Cocoa/Cocoa.h>
  31
  32 static void enableWebKitNightlyBehaviour() __attribute__ ((constructor));
  33
  34 static NSString *WKNERunState = @"WKNERunState";
  35 static NSString *WKNEShouldMonitorShutdowns = @"WKNEShouldMonitorShutdowns";
  36
  37 typedef enum {
  38     RunStateShutDown,
  39     RunStateInitializing,
  40     RunStateRunning
  41 } WKNERunStates;
  42
  43 static bool extensionBundlesWereLoaded = NO;
  44 static NSSet *extensionPaths = nil;
  45
  46 static void myBundleDidLoad(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo)
  47 {
  48     // Break out early if we have already detected an extension
  49     if (extensionBundlesWereLoaded)
  50         return;
  51
  52     NSBundle *bundle = (NSBundle *)object;
  53     NSString *bundlePath = [[bundle bundlePath] stringByAbbreviatingWithTildeInPath];
  54     NSString *bundleFileName = [bundlePath lastPathComponent];
  55
  56     // Explicitly ignore SIMBL.bundle, as its only purpose is to load extensions
  57     // on a per-application basis.  It's presence indicates a user has application
  58     // extensions, but not that any will be loaded into Safari
  59     if ([bundleFileName isEqualToString:@"SIMBL.bundle"])
  60         return;
  61
  62     // If the bundle lives inside a known extension path, flag it as an extension
  63     NSEnumerator *e = [extensionPaths objectEnumerator];
  64     NSString *path = nil;
  65     while (path = [e nextObject]) {
  66         if ([bundlePath length] < [path length])
  67             continue;
  68
  69         if ([[bundlePath substringToIndex:[path length]] isEqualToString:path]) {
  70             extensionBundlesWereLoaded = YES;
  71             break;
  72         }
  73     }
  74 }
  75
  76 static void myApplicationWillFinishLaunching(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo)
  77 {
  78     CFNotificationCenterRemoveObserver(CFNotificationCenterGetLocalCenter(), &myApplicationWillFinishLaunching, NULL, NULL);
  79     CFNotificationCenterRemoveObserver(CFNotificationCenterGetLocalCenter(), &myBundleDidLoad, NULL, NULL);
  80     [extensionPaths release];
  81     extensionPaths = nil;
  82
  83     NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
  84     [userDefaults setInteger:RunStateRunning forKey:WKNERunState];
  85     [userDefaults synchronize];
  86
  87     if (extensionBundlesWereLoaded)
  88         NSRunInformationalAlertPanel(@"Safari extensions detected",
  89                                      @"Safari extensions were detected on your system.  Extensions are incompatible with nightly builds of WebKit, and may cause crashes or incorrect behavior.  Please disable them if you experience such behavior.", @"Continue",
  90                                      nil, nil);
  91 }
  92
  93 static void myApplicationWillTerminate(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo)
  94 {
  95     NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
  96     [userDefaults setInteger:RunStateShutDown forKey:WKNERunState];
  97     [userDefaults synchronize];
  98 }
  99
 100 extern char **_CFGetProcessPath() __attribute__((weak));
 101
 102 static void poseAsWebKitApp()
 103 {
 104     char *webKitAppPath = getenv("WebKitAppPath");
 105     if (!webKitAppPath || !_CFGetProcessPath)
 106         return;
 107
 108     // Set up the main bundle early so it points at Safari.app
 109     CFBundleGetMainBundle();
 110
 111     // Fiddle with CoreFoundation to have it pick up the executable path as being within WebKit.app
 112     char **processPath = _CFGetProcessPath();
 113     *processPath = NULL;
 114     setenv("CFProcessPath", webKitAppPath, 1);
 115     _CFGetProcessPath();
 116
 117     // Clean up
 118     unsetenv("CFProcessPath");
 119     unsetenv("WebKitAppPath");
 120 }
 121
 122 static void enableWebKitNightlyBehaviour()
 123 {
 124     unsetenv("DYLD_INSERT_LIBRARIES");
 125     poseAsWebKitApp();
 126
 127     extensionPaths = [[NSSet alloc] initWithObjects:@"~/Library/InputManagers/", @"/Library/InputManagers/",
 128                                                     @"~/Library/Application Support/SIMBL/Plugins/", @"/Library/Application Support/SIMBL/Plugins/",
 129                                                     @"~/Library/Application Enhancers/", @"/Library/Application Enhancers/",
 130                                                     nil];
 131
 132     NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
 133     NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
 134     NSDictionary *defaultPrefs = [NSDictionary dictionaryWithObjectsAndKeys:[NSNumber numberWithBool:RunStateShutDown], WKNERunState,
 135                                                                             [NSNumber numberWithBool:YES], WKNEShouldMonitorShutdowns, nil];
 136     [userDefaults registerDefaults:defaultPrefs];
 137     if ([userDefaults boolForKey:WKNEShouldMonitorShutdowns]) {
 138         WKNERunStates savedState = (WKNERunStates)[userDefaults integerForKey:WKNERunState];
 139         if (savedState == RunStateInitializing) {
 140             // Use CoreFoundation here as AppKit hasn't been initialized at this stage of Safari's lifetime
 141             CFOptionFlags responseFlags;
 142             CFUserNotificationDisplayAlert(0, kCFUserNotificationCautionAlertLevel,
 143                                            NULL, NULL, NULL,
 144                                            CFSTR("WebKit failed to open correctly"),
 145                                            CFSTR("WebKit failed to open correctly on your previous attempt. Please disable any Safari extensions that you may have installed.  If the problem continues to occur, please file a bug report at http://webkit.org/quality/reporting.html"),
 146                                            CFSTR("Continue"), NULL, NULL, &responseFlags);
 147         }
 148         else if (savedState == RunStateRunning) {
 149             NSLog(@"WebKit failed to shut down cleanly.  Checking for Safari extensions.");
 150             CFNotificationCenterAddObserver(CFNotificationCenterGetLocalCenter(), &myBundleDidLoad,
 151                                             myBundleDidLoad, (CFStringRef) NSBundleDidLoadNotification,
 152                                             NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
 153         }
 154     }
 155     [userDefaults setInteger:RunStateInitializing forKey:WKNERunState];
 156     [userDefaults synchronize];
 157
 158     CFNotificationCenterAddObserver(CFNotificationCenterGetLocalCenter(), &myApplicationWillFinishLaunching,
 159                                     myApplicationWillFinishLaunching, (CFStringRef) NSApplicationWillFinishLaunchingNotification,
 160                                     NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
 161     CFNotificationCenterAddObserver(CFNotificationCenterGetLocalCenter(), &myApplicationWillTerminate,
 162                                     myApplicationWillTerminate, (CFStringRef) NSApplicationWillTerminateNotification,
 163                                     NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
 164     [pool release];
 165 }