1 /* 2 * Copyright (C) 2006, 2007, 2008, 2009 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 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 #import "WebKitNightlyEnablerSparkle.h" 32 33 static void enableWebKitNightlyBehaviour() __attribute__ ((constructor)); 34 35 static NSString *WKNERunState = @"WKNERunState"; 36 static NSString *WKNEShouldMonitorShutdowns = @"WKNEShouldMonitorShutdowns"; 37 38 typedef enum { 39 RunStateShutDown, 40 RunStateInitializing, 41 RunStateRunning 42 } WKNERunStates; 43 44 static char *webKitAppPath; 45 static bool extensionBundlesWereLoaded = NO; 46 static NSSet *extensionPaths = nil; 47 48 static int32_t systemVersion() 49 { 50 static SInt32 version = 0; 51 if (!version) 52 Gestalt(gestaltSystemVersion, &version); 53 54 return version; 55 } 56 57 58 static void myBundleDidLoad(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo) 59 { 60 NSBundle *bundle = (NSBundle *)object; 61 NSString *bundlePath = [[bundle bundlePath] stringByAbbreviatingWithTildeInPath]; 62 NSString *bundleFileName = [bundlePath lastPathComponent]; 63 64 // Explicitly ignore SIMBL.bundle, as its only purpose is to load extensions 65 // on a per-application basis. It's presence indicates a user has application 66 // extensions, but not that any will be loaded into Safari 67 if ([bundleFileName isEqualToString:@"SIMBL.bundle"]) 68 return; 69 70 // If the bundle lives inside a known extension path, flag it as an extension 71 NSEnumerator *e = [extensionPaths objectEnumerator]; 72 NSString *path = nil; 73 while ((path = [e nextObject])) { 74 if ([bundlePath length] < [path length]) 75 continue; 76 77 if ([[bundlePath substringToIndex:[path length]] isEqualToString:path]) { 78 NSLog(@"Extension detected: %@", bundlePath); 79 extensionBundlesWereLoaded = YES; 80 break; 81 } 82 } 83 } 84 85 static void myApplicationWillFinishLaunching(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo) 86 { 87 CFNotificationCenterRemoveObserver(CFNotificationCenterGetLocalCenter(), &myApplicationWillFinishLaunching, NULL, NULL); 88 CFNotificationCenterRemoveObserver(CFNotificationCenterGetLocalCenter(), &myBundleDidLoad, NULL, NULL); 89 [extensionPaths release]; 90 extensionPaths = nil; 91 92 NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults]; 93 [userDefaults setInteger:RunStateRunning forKey:WKNERunState]; 94 [userDefaults synchronize]; 95 96 if (extensionBundlesWereLoaded) 97 NSRunInformationalAlertPanel(@"Safari extensions detected", 98 @"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", 99 nil, nil); 100 101 initializeSparkle(); 102 } 103 104 static void myApplicationWillTerminate(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo) 105 { 106 NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults]; 107 [userDefaults setInteger:RunStateShutDown forKey:WKNERunState]; 108 [userDefaults synchronize]; 109 } 110 111 NSBundle *webKitLauncherBundle() 112 { 113 NSString *executablePath = [NSString stringWithUTF8String:webKitAppPath]; 114 NSRange appLocation = [executablePath rangeOfString:@".app/" options:NSBackwardsSearch]; 115 NSString *appPath = [executablePath substringToIndex:appLocation.location + appLocation.length]; 116 return [NSBundle bundleWithPath:appPath]; 117 } 118 119 extern char **_CFGetProcessPath() __attribute__((weak)); 120 extern OSStatus _RegisterApplication(CFDictionaryRef additionalAppInfoRef, ProcessSerialNumber* myPSN) __attribute__((weak)); 121 122 static void poseAsWebKitApp() 123 { 124 webKitAppPath = strdup(getenv("WebKitAppPath")); 125 if (!webKitAppPath) 126 return; 127 128 unsetenv("WebKitAppPath"); 129 130 // Set up the main bundle early so it points at Safari.app 131 CFBundleGetMainBundle(); 132 133 if (systemVersion() < 0x1060) { 134 if (!_CFGetProcessPath) 135 return; 136 137 // Fiddle with CoreFoundation to have it pick up the executable path as being within WebKit.app 138 char **processPath = _CFGetProcessPath(); 139 *processPath = NULL; 140 setenv("CFProcessPath", webKitAppPath, 1); 141 _CFGetProcessPath(); 142 unsetenv("CFProcessPath"); 143 } else { 144 if (!_RegisterApplication) 145 return; 146 147 // Register the application with LaunchServices, passing a customized registration dictionary that 148 // uses the WebKit launcher as the application bundle. 149 NSBundle *bundle = webKitLauncherBundle(); 150 NSMutableDictionary *checkInDictionary = [[bundle infoDictionary] mutableCopy]; 151 [checkInDictionary setObject:[bundle bundlePath] forKey:@"LSBundlePath"]; 152 [checkInDictionary setObject:[checkInDictionary objectForKey:(NSString *)kCFBundleNameKey] forKey:@"LSDisplayName"]; 153 _RegisterApplication((CFDictionaryRef)checkInDictionary, 0); 154 [checkInDictionary release]; 155 } 156 } 157 158 static BOOL insideSafari4OnTigerTrampoline() 159 { 160 // If we're not on Tiger then we can't be in the trampoline state. 161 if ((systemVersion() & 0xFFF0) != 0x1040) 162 return NO; 163 164 // If we're running Safari < 4.0 then we can't be in the trampoline state. 165 CFBundleRef safariBundle = CFBundleGetMainBundle(); 166 CFStringRef safariVersion = CFBundleGetValueForInfoDictionaryKey(safariBundle, CFSTR("CFBundleShortVersionString")); 167 if ([(NSString *)safariVersion intValue] < 4) 168 return NO; 169 170 const char* frameworkPath = getenv("DYLD_FRAMEWORK_PATH"); 171 if (!frameworkPath) 172 frameworkPath = ""; 173 174 // If the framework search path is empty or otherwise does not contain the Safari 175 // framework's Frameworks directory then we are in the trampoline state. 176 const char safariFrameworkSearchPath[] = "/System/Library/PrivateFrameworks/Safari.framework/Frameworks"; 177 return strstr(frameworkPath, safariFrameworkSearchPath) == 0; 178 } 179 180 static void enableWebKitNightlyBehaviour() 181 { 182 // If we're inside Safari in its trampoline state, it will very shortly relaunch itself. 183 // We bail out here so that we'll be called again in the freshly-launched Safari process. 184 if (insideSafari4OnTigerTrampoline()) 185 return; 186 187 NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init]; 188 189 unsetenv("DYLD_INSERT_LIBRARIES"); 190 poseAsWebKitApp(); 191 192 extensionPaths = [[NSSet alloc] initWithObjects:@"~/Library/InputManagers/", @"/Library/InputManagers/", 193 @"~/Library/Application Support/SIMBL/Plugins/", @"/Library/Application Support/SIMBL/Plugins/", 194 @"~/Library/Application Enhancers/", @"/Library/Application Enhancers/", 195 nil]; 196 197 // As of 2008-11 attempting to load Saft would cause a crash on launch, so prevent it from being loaded. 198 NSArray *disabledInputManagers = [NSArray arrayWithObjects:@"Saft", nil]; 199 200 NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults]; 201 NSDictionary *defaultPrefs = [NSDictionary dictionaryWithObjectsAndKeys:[NSNumber numberWithBool:RunStateShutDown], WKNERunState, 202 [NSNumber numberWithBool:YES], WKNEShouldMonitorShutdowns, 203 disabledInputManagers, @"NSDisabledInputManagers", nil]; 204 [userDefaults registerDefaults:defaultPrefs]; 205 if ([userDefaults boolForKey:WKNEShouldMonitorShutdowns]) { 206 WKNERunStates savedState = (WKNERunStates)[userDefaults integerForKey:WKNERunState]; 207 if (savedState == RunStateInitializing) { 208 // Use CoreFoundation here as AppKit hasn't been initialized at this stage of Safari's lifetime 209 CFOptionFlags responseFlags; 210 CFUserNotificationDisplayAlert(0, kCFUserNotificationCautionAlertLevel, 211 NULL, NULL, NULL, 212 CFSTR("WebKit failed to open correctly"), 213 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"), 214 CFSTR("Continue"), NULL, NULL, &responseFlags); 215 } 216 else if (savedState == RunStateRunning) { 217 NSLog(@"WebKit failed to shut down cleanly. Checking for Safari extensions."); 218 CFNotificationCenterAddObserver(CFNotificationCenterGetLocalCenter(), &myBundleDidLoad, 219 myBundleDidLoad, (CFStringRef) NSBundleDidLoadNotification, 220 NULL, CFNotificationSuspensionBehaviorDeliverImmediately); 221 } 222 } 223 [userDefaults setInteger:RunStateInitializing forKey:WKNERunState]; 224 [userDefaults synchronize]; 225 226 CFNotificationCenterAddObserver(CFNotificationCenterGetLocalCenter(), &myApplicationWillFinishLaunching, 227 myApplicationWillFinishLaunching, (CFStringRef) NSApplicationWillFinishLaunchingNotification, 228 NULL, CFNotificationSuspensionBehaviorDeliverImmediately); 229 CFNotificationCenterAddObserver(CFNotificationCenterGetLocalCenter(), &myApplicationWillTerminate, 230 myApplicationWillTerminate, (CFStringRef) NSApplicationWillTerminateNotification, 231 NULL, CFNotificationSuspensionBehaviorDeliverImmediately); 232 233 NSLog(@"WebKit %@ initialized.", [webKitLauncherBundle() objectForInfoDictionaryKey:@"CFBundleShortVersionString"]); 234 235 [pool release]; 236 } 237
