Home | History | Annotate | Download | only in WebKitLauncher
      1 /*
      2  * Copyright (C) 2006, 2007, 2008, 2009 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 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 <Cocoa/Cocoa.h>
     30 #import <CoreFoundation/CoreFoundation.h>
     31 
     32 // We need to weak-import posix_spawn and friends as they're not available on Tiger.
     33 // The BSD-level system headers do not have availability macros, so we redeclare the
     34 // functions ourselves with the "weak" attribute.
     35 
     36 #define WEAK_IMPORT __attribute__((weak))
     37 
     38 #define POSIX_SPAWN_SETEXEC 0x0040
     39 typedef void *posix_spawnattr_t;
     40 typedef void *posix_spawn_file_actions_t;
     41 int posix_spawnattr_init(posix_spawnattr_t *) WEAK_IMPORT;
     42 int posix_spawn(pid_t * __restrict, const char * __restrict, const posix_spawn_file_actions_t *, const posix_spawnattr_t * __restrict, char *const __argv[ __restrict], char *const __envp[ __restrict]) WEAK_IMPORT;
     43 int posix_spawnattr_setbinpref_np(posix_spawnattr_t * __restrict, size_t, cpu_type_t *__restrict, size_t *__restrict) WEAK_IMPORT;
     44 int posix_spawnattr_setflags(posix_spawnattr_t *, short) WEAK_IMPORT;
     45 
     46 
     47 static void displayErrorAndQuit(NSString *title, NSString *message)
     48 {
     49     NSApplicationLoad();
     50     NSRunCriticalAlertPanel(title, message, @"Quit", nil, nil);
     51     exit(0);
     52 }
     53 
     54 static int getLastVersionShown()
     55 {
     56     [[NSUserDefaults standardUserDefaults] registerDefaults:[NSDictionary dictionaryWithObject:@"-1" forKey:@"StartPageShownInVersion"]];
     57     return [[NSUserDefaults standardUserDefaults] integerForKey:@"StartPageShownInVersion"];
     58 }
     59 
     60 static void saveLastVersionShown(int lastVersion)
     61 {
     62     [[NSUserDefaults standardUserDefaults] setInteger:lastVersion forKey:@"StartPageShownInVersion"];
     63     [[NSUserDefaults standardUserDefaults] synchronize];
     64 }
     65 
     66 static NSString *getPathForStartPage()
     67 {
     68     return [[[NSBundle mainBundle] resourcePath] stringByAppendingPathComponent:@"start.html"];
     69 }
     70 
     71 static int getCurrentVersion()
     72 {
     73     return [[[[NSBundle mainBundle] infoDictionary] valueForKey:(NSString *)kCFBundleVersionKey] intValue];
     74 }
     75 
     76 static int getShowStartPageVersion()
     77 {
     78     return getCurrentVersion() + 1;
     79 }
     80 
     81 static BOOL startPageDisabled()
     82 {
     83     return [[NSUserDefaults standardUserDefaults] boolForKey:@"StartPageDisabled"];
     84 }
     85 
     86 static void addStartPageToArgumentsIfNeeded(NSMutableArray *arguments)
     87 {
     88     if (startPageDisabled())
     89         return;
     90 
     91     if (getLastVersionShown() < getShowStartPageVersion()) {
     92         saveLastVersionShown(getCurrentVersion());
     93         NSString *startPagePath = getPathForStartPage();
     94         if (startPagePath)
     95             [arguments addObject:startPagePath];
     96     }
     97 }
     98 
     99 static cpu_type_t preferredArchitecture()
    100 {
    101 #if defined(__ppc__)
    102     return CPU_TYPE_POWERPC;
    103 #elif defined(__LP64__)
    104     return CPU_TYPE_X86_64;
    105 #else
    106     return CPU_TYPE_X86;
    107 #endif
    108 }
    109 
    110 static void myExecve(NSString *executable, NSArray *args, NSDictionary *environment)
    111 {
    112     char **argv = (char **)calloc(sizeof(char *), [args count] + 1);
    113     char **env = (char **)calloc(sizeof(char *), [environment count] + 1);
    114 
    115     NSEnumerator *e = [args objectEnumerator];
    116     NSString *s;
    117     int i = 0;
    118     while ((s = [e nextObject]))
    119         argv[i++] = (char *) [s UTF8String];
    120 
    121     e = [environment keyEnumerator];
    122     i = 0;
    123     while ((s = [e nextObject]))
    124         env[i++] = (char *) [[NSString stringWithFormat:@"%@=%@", s, [environment objectForKey:s]] UTF8String];
    125 
    126     if (posix_spawnattr_init && posix_spawn && posix_spawnattr_setbinpref_np && posix_spawnattr_setflags) {
    127         posix_spawnattr_t attr;
    128         posix_spawnattr_init(&attr);
    129         cpu_type_t architecturePreference[] = { preferredArchitecture(), CPU_TYPE_X86 };
    130         posix_spawnattr_setbinpref_np(&attr, 2, architecturePreference, 0);
    131         short flags = POSIX_SPAWN_SETEXEC;
    132         posix_spawnattr_setflags(&attr, flags);
    133         posix_spawn(NULL, [executable fileSystemRepresentation], NULL, &attr, argv, env);
    134     } else
    135         execve([executable fileSystemRepresentation], argv, env);
    136 }
    137 
    138 static NSBundle *locateSafariBundle()
    139 {
    140     NSArray *applicationDirectories = NSSearchPathForDirectoriesInDomains(NSApplicationDirectory, NSAllDomainsMask, YES);
    141     NSEnumerator *e = [applicationDirectories objectEnumerator];
    142     NSString *applicationDirectory;
    143     while ((applicationDirectory = [e nextObject])) {
    144         NSString *possibleSafariPath = [applicationDirectory stringByAppendingPathComponent:@"Safari.app"];
    145         NSBundle *possibleSafariBundle = [NSBundle bundleWithPath:possibleSafariPath];
    146         if ([[possibleSafariBundle bundleIdentifier] isEqualToString:@"com.apple.Safari"])
    147             return possibleSafariBundle;
    148     }
    149 
    150     CFURLRef safariURL = nil;
    151     OSStatus err = LSFindApplicationForInfo(kLSUnknownCreator, CFSTR("com.apple.Safari"), nil, nil, &safariURL);
    152     if (err != noErr)
    153         displayErrorAndQuit(@"Unable to locate Safari", @"Nightly builds of WebKit require Safari to run.  Please check that it is available and then try again.");
    154 
    155     NSBundle *safariBundle = [NSBundle bundleWithPath:[(NSURL *)safariURL path]];
    156     CFRelease(safariURL);
    157     return safariBundle;
    158 }
    159 
    160 static NSString *currentMacOSXVersion()
    161 {
    162     SInt32 version;
    163     if (Gestalt(gestaltSystemVersion, &version) != noErr)
    164         return @"10.4";
    165 
    166     return [NSString stringWithFormat:@"%x.%x", (version & 0xFF00) >> 8, (version & 0x00F0) >> 4];
    167 }
    168 
    169 static NSString *fallbackMacOSXVersion(NSString *systemVersion)
    170 {
    171     NSDictionary *fallbackVersionMap = [[NSUserDefaults standardUserDefaults] dictionaryForKey:@"FallbackSystemVersions"];
    172     if (!fallbackVersionMap)
    173         return nil;
    174     NSString *fallbackSystemVersion = [fallbackVersionMap objectForKey:systemVersion];
    175     if (!fallbackSystemVersion || ![fallbackSystemVersion isKindOfClass:[NSString class]])
    176         return nil;
    177     return fallbackSystemVersion;
    178 }
    179 
    180 static BOOL checkFrameworkPath(NSString *frameworkPath)
    181 {
    182     BOOL isDirectory = NO;
    183     return [[NSFileManager defaultManager] fileExistsAtPath:frameworkPath isDirectory:&isDirectory] && isDirectory;
    184 }
    185 
    186 static BOOL checkSafariVersion(NSBundle *safariBundle)
    187 {
    188     NSString *safariBundleVersion = [[safariBundle infoDictionary] objectForKey:(NSString *)kCFBundleVersionKey];
    189     NSString *majorComponent = [[safariBundleVersion componentsSeparatedByString:@"."] objectAtIndex:0];
    190     NSString *majorVersion = [majorComponent substringFromIndex:[majorComponent length] - 3];
    191     return [majorVersion intValue] >= 530;
    192 }
    193 
    194 int main(int argc, char *argv[])
    195 {
    196     NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    197 
    198     NSString *systemVersion = currentMacOSXVersion();
    199     NSString *frameworkPath = [[[NSBundle mainBundle] privateFrameworksPath] stringByAppendingPathComponent:systemVersion];
    200 
    201     BOOL frameworkPathIsUsable = checkFrameworkPath(frameworkPath);
    202 
    203     if (!frameworkPathIsUsable) {
    204         NSString *fallbackSystemVersion = fallbackMacOSXVersion(systemVersion);
    205         if (fallbackSystemVersion) {
    206             frameworkPath = [[[NSBundle mainBundle] privateFrameworksPath] stringByAppendingPathComponent:fallbackSystemVersion];
    207             frameworkPathIsUsable = checkFrameworkPath(frameworkPath);
    208         }
    209     }
    210 
    211     if (!frameworkPathIsUsable)
    212         displayErrorAndQuit([NSString stringWithFormat:@"Mac OS X %@ is not supported", systemVersion],
    213                             [NSString stringWithFormat:@"Nightly builds of WebKit are not supported on Mac OS X %@ at this time.", systemVersion]);
    214 
    215     NSString *pathToEnablerLib = [[NSBundle mainBundle] pathForResource:@"WebKitNightlyEnabler" ofType:@"dylib"];
    216 
    217     NSBundle *safariBundle = locateSafariBundle();
    218     NSString *executablePath = [safariBundle executablePath];
    219 
    220     if (!checkSafariVersion(safariBundle)) {
    221         NSString *safariVersion = [[safariBundle localizedInfoDictionary] objectForKey:@"CFBundleShortVersionString"];
    222         displayErrorAndQuit([NSString stringWithFormat:@"Safari %@ is not supported", safariVersion],
    223                             [NSString stringWithFormat:@"Nightly builds of WebKit are not supported with Safari %@ at this time. Please update to a newer version of Safari.", safariVersion]);
    224     }
    225 
    226     if ([frameworkPath rangeOfString:@":"].location != NSNotFound ||
    227         [pathToEnablerLib rangeOfString:@":"].location != NSNotFound)
    228         displayErrorAndQuit(@"Unable to launch Safari",
    229                             @"WebKit is located at a path containing an unsupported character.  Please move WebKit to a different location and try again.");
    230 
    231     NSMutableArray *arguments = [NSMutableArray arrayWithObject:executablePath];
    232     NSMutableDictionary *environment = [[[NSDictionary dictionaryWithObjectsAndKeys:frameworkPath, @"DYLD_FRAMEWORK_PATH", @"YES", @"WEBKIT_UNSET_DYLD_FRAMEWORK_PATH",
    233                                                                                     pathToEnablerLib, @"DYLD_INSERT_LIBRARIES", [[NSBundle mainBundle] executablePath], @"WebKitAppPath", nil] mutableCopy] autorelease];
    234     [environment addEntriesFromDictionary:[[NSProcessInfo processInfo] environment]];
    235     addStartPageToArgumentsIfNeeded(arguments);
    236 
    237     while (*++argv)
    238         [arguments addObject:[NSString stringWithUTF8String:*argv]];
    239 
    240     myExecve(executablePath, arguments, environment);
    241 
    242     char *error = strerror(errno);
    243     NSString *errorMessage = [NSString stringWithFormat:@"Launching Safari at %@ failed with the error '%s' (%d)", [safariBundle bundlePath], error, errno];
    244     displayErrorAndQuit(@"Unable to launch Safari", errorMessage);
    245 
    246     [pool release];
    247     return 0;
    248 }
    249