Tools/WebKitLauncher/main.m - platform/external/webkit - Git at Google

 /*
  * Copyright (C) 2006, 2007, 2008, 2009 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 <Cocoa/Cocoa.h>
 #import <CoreFoundation/CoreFoundation.h>

 // We need to weak-import posix_spawn and friends as they're not available on Tiger.
 // The BSD-level system headers do not have availability macros, so we redeclare the
 // functions ourselves with the "weak" attribute.

 #define WEAK_IMPORT __attribute__((weak))

 #define POSIX_SPAWN_SETEXEC 0x0040
 typedef void *posix_spawnattr_t;
 typedef void *posix_spawn_file_actions_t;
 int posix_spawnattr_init(posix_spawnattr_t *) WEAK_IMPORT;
 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;
 int posix_spawnattr_setbinpref_np(posix_spawnattr_t * __restrict, size_t, cpu_type_t *__restrict, size_t *__restrict) WEAK_IMPORT;
 int posix_spawnattr_setflags(posix_spawnattr_t *, short) WEAK_IMPORT;


 static void displayErrorAndQuit(NSString *title, NSString *message)
 {
     NSApplicationLoad();
     NSRunCriticalAlertPanel(title, message, @"Quit", nil, nil);
     exit(0);
 }

 static int getLastVersionShown()
 {
     [[NSUserDefaults standardUserDefaults] registerDefaults:[NSDictionary dictionaryWithObject:@"-1" forKey:@"StartPageShownInVersion"]];
     return [[NSUserDefaults standardUserDefaults] integerForKey:@"StartPageShownInVersion"];
 }

 static void saveLastVersionShown(int lastVersion)
 {
     [[NSUserDefaults standardUserDefaults] setInteger:lastVersion forKey:@"StartPageShownInVersion"];
     [[NSUserDefaults standardUserDefaults] synchronize];
 }

 static NSString *getPathForStartPage()
 {
     return [[[NSBundle mainBundle] resourcePath] stringByAppendingPathComponent:@"start.html"];
 }

 static int getCurrentVersion()
 {
     return [[[[NSBundle mainBundle] infoDictionary] valueForKey:(NSString *)kCFBundleVersionKey] intValue];
 }

 static int getShowStartPageVersion()
 {
     return getCurrentVersion() + 1;
 }

 static BOOL startPageDisabled()
 {
     return [[NSUserDefaults standardUserDefaults] boolForKey:@"StartPageDisabled"];
 }

 static void addStartPageToArgumentsIfNeeded(NSMutableArray *arguments)
 {
     if (startPageDisabled())
         return;

     if (getLastVersionShown() < getShowStartPageVersion()) {
         saveLastVersionShown(getCurrentVersion());
         NSString *startPagePath = getPathForStartPage();
         if (startPagePath)
             [arguments addObject:startPagePath];
     }
 }

 static cpu_type_t preferredArchitecture()
 {
 #if defined(__ppc__)
     return CPU_TYPE_POWERPC;
 #elif defined(__LP64__)
     return CPU_TYPE_X86_64;
 #else
     return CPU_TYPE_X86;
 #endif
 }

 static void myExecve(NSString *executable, NSArray *args, NSDictionary *environment)
 {
     char **argv = (char **)calloc(sizeof(char *), [args count] + 1);
     char **env = (char **)calloc(sizeof(char *), [environment count] + 1);

     NSEnumerator *e = [args objectEnumerator];
     NSString *s;
     int i = 0;
     while ((s = [e nextObject]))
         argv[i++] = (char *) [s UTF8String];

     e = [environment keyEnumerator];
     i = 0;
     while ((s = [e nextObject]))
         env[i++] = (char *) [[NSString stringWithFormat:@"%@=%@", s, [environment objectForKey:s]] UTF8String];

     if (posix_spawnattr_init && posix_spawn && posix_spawnattr_setbinpref_np && posix_spawnattr_setflags) {
         posix_spawnattr_t attr;
         posix_spawnattr_init(&attr);
         cpu_type_t architecturePreference[] = { preferredArchitecture(), CPU_TYPE_X86 };
         posix_spawnattr_setbinpref_np(&attr, 2, architecturePreference, 0);
         short flags = POSIX_SPAWN_SETEXEC;
         posix_spawnattr_setflags(&attr, flags);
         posix_spawn(NULL, [executable fileSystemRepresentation], NULL, &attr, argv, env);
     } else
         execve([executable fileSystemRepresentation], argv, env);
 }

 static NSBundle *locateSafariBundle()
 {
     NSArray *applicationDirectories = NSSearchPathForDirectoriesInDomains(NSApplicationDirectory, NSAllDomainsMask, YES);
     NSEnumerator *e = [applicationDirectories objectEnumerator];
     NSString *applicationDirectory;
     while ((applicationDirectory = [e nextObject])) {
         NSString *possibleSafariPath = [applicationDirectory stringByAppendingPathComponent:@"Safari.app"];
         NSBundle *possibleSafariBundle = [NSBundle bundleWithPath:possibleSafariPath];
         if ([[possibleSafariBundle bundleIdentifier] isEqualToString:@"com.apple.Safari"])
             return possibleSafariBundle;
     }

     CFURLRef safariURL = nil;
     OSStatus err = LSFindApplicationForInfo(kLSUnknownCreator, CFSTR("com.apple.Safari"), nil, nil, &safariURL);
     if (err != noErr)
         displayErrorAndQuit(@"Unable to locate Safari", @"Nightly builds of WebKit require Safari to run.  Please check that it is available and then try again.");

     NSBundle *safariBundle = [NSBundle bundleWithPath:[(NSURL *)safariURL path]];
     CFRelease(safariURL);
     return safariBundle;
 }

 static NSString *currentMacOSXVersion()
 {
     SInt32 version;
     if (Gestalt(gestaltSystemVersion, &version) != noErr)
         return @"10.4";

     return [NSString stringWithFormat:@"%x.%x", (version & 0xFF00) >> 8, (version & 0x00F0) >> 4];
 }

 static NSString *fallbackMacOSXVersion(NSString *systemVersion)
 {
     NSDictionary *fallbackVersionMap = [[NSUserDefaults standardUserDefaults] dictionaryForKey:@"FallbackSystemVersions"];
     if (!fallbackVersionMap)
         return nil;
     NSString *fallbackSystemVersion = [fallbackVersionMap objectForKey:systemVersion];
     if (!fallbackSystemVersion || ![fallbackSystemVersion isKindOfClass:[NSString class]])
         return nil;
     return fallbackSystemVersion;
 }

 static BOOL checkFrameworkPath(NSString *frameworkPath)
 {
     BOOL isDirectory = NO;
     return [[NSFileManager defaultManager] fileExistsAtPath:frameworkPath isDirectory:&isDirectory] && isDirectory;
 }

 static BOOL checkSafariVersion(NSBundle *safariBundle)
 {
     NSString *safariBundleVersion = [[safariBundle infoDictionary] objectForKey:(NSString *)kCFBundleVersionKey];
     NSString *majorComponent = [[safariBundleVersion componentsSeparatedByString:@"."] objectAtIndex:0];
     NSString *majorVersion = [majorComponent substringFromIndex:[majorComponent length] - 3];
     return [majorVersion intValue] >= 530;
 }

 int main(int argc, char *argv[])
 {
     NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];

     NSString *systemVersion = currentMacOSXVersion();
     NSString *frameworkPath = [[[NSBundle mainBundle] privateFrameworksPath] stringByAppendingPathComponent:systemVersion];

     BOOL frameworkPathIsUsable = checkFrameworkPath(frameworkPath);

     if (!frameworkPathIsUsable) {
         NSString *fallbackSystemVersion = fallbackMacOSXVersion(systemVersion);
         if (fallbackSystemVersion) {
             frameworkPath = [[[NSBundle mainBundle] privateFrameworksPath] stringByAppendingPathComponent:fallbackSystemVersion];
             frameworkPathIsUsable = checkFrameworkPath(frameworkPath);
         }
     }

     if (!frameworkPathIsUsable)
         displayErrorAndQuit([NSString stringWithFormat:@"Mac OS X %@ is not supported", systemVersion],
                             [NSString stringWithFormat:@"Nightly builds of WebKit are not supported on Mac OS X %@ at this time.", systemVersion]);

     NSString *pathToEnablerLib = [[NSBundle mainBundle] pathForResource:@"WebKitNightlyEnabler" ofType:@"dylib"];

     NSBundle *safariBundle = locateSafariBundle();
     NSString *executablePath = [safariBundle executablePath];

     if (!checkSafariVersion(safariBundle)) {
         NSString *safariVersion = [[safariBundle localizedInfoDictionary] objectForKey:@"CFBundleShortVersionString"];
         displayErrorAndQuit([NSString stringWithFormat:@"Safari %@ is not supported", safariVersion],
                             [NSString stringWithFormat:@"Nightly builds of WebKit are not supported with Safari %@ at this time. Please update to a newer version of Safari.", safariVersion]);
     }

     if ([frameworkPath rangeOfString:@":"].location != NSNotFound ||
         [pathToEnablerLib rangeOfString:@":"].location != NSNotFound)
         displayErrorAndQuit(@"Unable to launch Safari",
                             @"WebKit is located at a path containing an unsupported character.  Please move WebKit to a different location and try again.");

     NSMutableArray *arguments = [NSMutableArray arrayWithObject:executablePath];
     NSMutableDictionary *environment = [[[NSDictionary dictionaryWithObjectsAndKeys:frameworkPath, @"DYLD_FRAMEWORK_PATH", @"YES", @"WEBKIT_UNSET_DYLD_FRAMEWORK_PATH",
                                                                                     pathToEnablerLib, @"DYLD_INSERT_LIBRARIES", [[NSBundle mainBundle] executablePath], @"WebKitAppPath", nil] mutableCopy] autorelease];
     [environment addEntriesFromDictionary:[[NSProcessInfo processInfo] environment]];
     addStartPageToArgumentsIfNeeded(arguments);

     while (*++argv)
         [arguments addObject:[NSString stringWithUTF8String:*argv]];

     myExecve(executablePath, arguments, environment);

     char *error = strerror(errno);
     NSString *errorMessage = [NSString stringWithFormat:@"Launching Safari at %@ failed with the error '%s' (%d)", [safariBundle bundlePath], error, errno];
     displayErrorAndQuit(@"Unable to launch Safari", errorMessage);

     [pool release];
     return 0;
 }
	/*
	* Copyright (C) 2006, 2007, 2008, 2009 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 <Cocoa/Cocoa.h>
	#import <CoreFoundation/CoreFoundation.h>

	// We need to weak-import posix_spawn and friends as they're not available on Tiger.
	// The BSD-level system headers do not have availability macros, so we redeclare the
	// functions ourselves with the "weak" attribute.

	#define WEAK_IMPORT __attribute__((weak))

	#define POSIX_SPAWN_SETEXEC 0x0040
	typedef void *posix_spawnattr_t;
	typedef void *posix_spawn_file_actions_t;
	int posix_spawnattr_init(posix_spawnattr_t *) WEAK_IMPORT;
	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;
	int posix_spawnattr_setbinpref_np(posix_spawnattr_t * __restrict, size_t, cpu_type_t __restrict, size_t __restrict) WEAK_IMPORT;
	int posix_spawnattr_setflags(posix_spawnattr_t *, short) WEAK_IMPORT;


	static void displayErrorAndQuit(NSString title, NSString message)
	{
	NSApplicationLoad();
	NSRunCriticalAlertPanel(title, message, @"Quit", nil, nil);
	exit(0);
	}

	static int getLastVersionShown()
	{
	[[NSUserDefaults standardUserDefaults] registerDefaults:[NSDictionary dictionaryWithObject:@"-1" forKey:@"StartPageShownInVersion"]];
	return [[NSUserDefaults standardUserDefaults] integerForKey:@"StartPageShownInVersion"];
	}

	static void saveLastVersionShown(int lastVersion)
	{
	[[NSUserDefaults standardUserDefaults] setInteger:lastVersion forKey:@"StartPageShownInVersion"];
	[[NSUserDefaults standardUserDefaults] synchronize];
	}

	static NSString *getPathForStartPage()
	{
	return [[[NSBundle mainBundle] resourcePath] stringByAppendingPathComponent:@"start.html"];
	}

	static int getCurrentVersion()
	{
	return [[[[NSBundle mainBundle] infoDictionary] valueForKey:(NSString *)kCFBundleVersionKey] intValue];
	}

	static int getShowStartPageVersion()
	{
	return getCurrentVersion() + 1;
	}

	static BOOL startPageDisabled()
	{
	return [[NSUserDefaults standardUserDefaults] boolForKey:@"StartPageDisabled"];
	}

	static void addStartPageToArgumentsIfNeeded(NSMutableArray *arguments)
	{
	if (startPageDisabled())
	return;

	if (getLastVersionShown() < getShowStartPageVersion()) {
	saveLastVersionShown(getCurrentVersion());
	NSString *startPagePath = getPathForStartPage();
	if (startPagePath)
	[arguments addObject:startPagePath];
	}
	}

	static cpu_type_t preferredArchitecture()
	{
	#if defined(__ppc__)
	return CPU_TYPE_POWERPC;
	#elif defined(__LP64__)
	return CPU_TYPE_X86_64;
	#else
	return CPU_TYPE_X86;
	#endif
	}

	static void myExecve(NSString executable, NSArray args, NSDictionary *environment)
	{
	char argv = (char )calloc(sizeof(char *), [args count] + 1);
	char env = (char )calloc(sizeof(char *), [environment count] + 1);

	NSEnumerator *e = [args objectEnumerator];
	NSString *s;
	int i = 0;
	while ((s = [e nextObject]))
	argv[i++] = (char *) [s UTF8String];

	e = [environment keyEnumerator];
	i = 0;
	while ((s = [e nextObject]))
	env[i++] = (char *) [[NSString stringWithFormat:@"%@=%@", s, [environment objectForKey:s]] UTF8String];

	if (posix_spawnattr_init && posix_spawn && posix_spawnattr_setbinpref_np && posix_spawnattr_setflags) {
	posix_spawnattr_t attr;
	posix_spawnattr_init(&attr);
	cpu_type_t architecturePreference[] = { preferredArchitecture(), CPU_TYPE_X86 };
	posix_spawnattr_setbinpref_np(&attr, 2, architecturePreference, 0);
	short flags = POSIX_SPAWN_SETEXEC;
	posix_spawnattr_setflags(&attr, flags);
	posix_spawn(NULL, [executable fileSystemRepresentation], NULL, &attr, argv, env);
	} else
	execve([executable fileSystemRepresentation], argv, env);
	}

	static NSBundle *locateSafariBundle()
	{
	NSArray *applicationDirectories = NSSearchPathForDirectoriesInDomains(NSApplicationDirectory, NSAllDomainsMask, YES);
	NSEnumerator *e = [applicationDirectories objectEnumerator];
	NSString *applicationDirectory;
	while ((applicationDirectory = [e nextObject])) {
	NSString *possibleSafariPath = [applicationDirectory stringByAppendingPathComponent:@"Safari.app"];
	NSBundle *possibleSafariBundle = [NSBundle bundleWithPath:possibleSafariPath];
	if ([[possibleSafariBundle bundleIdentifier] isEqualToString:@"com.apple.Safari"])
	return possibleSafariBundle;
	}

	CFURLRef safariURL = nil;
	OSStatus err = LSFindApplicationForInfo(kLSUnknownCreator, CFSTR("com.apple.Safari"), nil, nil, &safariURL);
	if (err != noErr)
	displayErrorAndQuit(@"Unable to locate Safari", @"Nightly builds of WebKit require Safari to run. Please check that it is available and then try again.");

	NSBundle safariBundle = [NSBundle bundleWithPath:[(NSURL )safariURL path]];
	CFRelease(safariURL);
	return safariBundle;
	}

	static NSString *currentMacOSXVersion()
	{
	SInt32 version;
	if (Gestalt(gestaltSystemVersion, &version) != noErr)
	return @"10.4";

	return [NSString stringWithFormat:@"%x.%x", (version & 0xFF00) >> 8, (version & 0x00F0) >> 4];
	}

	static NSString fallbackMacOSXVersion(NSString systemVersion)
	{
	NSDictionary *fallbackVersionMap = [[NSUserDefaults standardUserDefaults] dictionaryForKey:@"FallbackSystemVersions"];
	if (!fallbackVersionMap)
	return nil;
	NSString *fallbackSystemVersion = [fallbackVersionMap objectForKey:systemVersion];
	if (!fallbackSystemVersion \|\| ![fallbackSystemVersion isKindOfClass:[NSString class]])
	return nil;
	return fallbackSystemVersion;
	}

	static BOOL checkFrameworkPath(NSString *frameworkPath)
	{
	BOOL isDirectory = NO;
	return [[NSFileManager defaultManager] fileExistsAtPath:frameworkPath isDirectory:&isDirectory] && isDirectory;
	}

	static BOOL checkSafariVersion(NSBundle *safariBundle)
	{
	NSString safariBundleVersion = [[safariBundle infoDictionary] objectForKey:(NSString )kCFBundleVersionKey];
	NSString *majorComponent = [[safariBundleVersion componentsSeparatedByString:@"."] objectAtIndex:0];
	NSString *majorVersion = [majorComponent substringFromIndex:[majorComponent length] - 3];
	return [majorVersion intValue] >= 530;
	}

	int main(int argc, char *argv[])
	{
	NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];

	NSString *systemVersion = currentMacOSXVersion();
	NSString *frameworkPath = [[[NSBundle mainBundle] privateFrameworksPath] stringByAppendingPathComponent:systemVersion];

	BOOL frameworkPathIsUsable = checkFrameworkPath(frameworkPath);

	if (!frameworkPathIsUsable) {
	NSString *fallbackSystemVersion = fallbackMacOSXVersion(systemVersion);
	if (fallbackSystemVersion) {
	frameworkPath = [[[NSBundle mainBundle] privateFrameworksPath] stringByAppendingPathComponent:fallbackSystemVersion];
	frameworkPathIsUsable = checkFrameworkPath(frameworkPath);
	}
	}

	if (!frameworkPathIsUsable)
	displayErrorAndQuit([NSString stringWithFormat:@"Mac OS X %@ is not supported", systemVersion],
	[NSString stringWithFormat:@"Nightly builds of WebKit are not supported on Mac OS X %@ at this time.", systemVersion]);

	NSString *pathToEnablerLib = [[NSBundle mainBundle] pathForResource:@"WebKitNightlyEnabler" ofType:@"dylib"];

	NSBundle *safariBundle = locateSafariBundle();
	NSString *executablePath = [safariBundle executablePath];

	if (!checkSafariVersion(safariBundle)) {
	NSString *safariVersion = [[safariBundle localizedInfoDictionary] objectForKey:@"CFBundleShortVersionString"];
	displayErrorAndQuit([NSString stringWithFormat:@"Safari %@ is not supported", safariVersion],
	[NSString stringWithFormat:@"Nightly builds of WebKit are not supported with Safari %@ at this time. Please update to a newer version of Safari.", safariVersion]);
	}

	if ([frameworkPath rangeOfString:@":"].location != NSNotFound \|\|
	[pathToEnablerLib rangeOfString:@":"].location != NSNotFound)
	displayErrorAndQuit(@"Unable to launch Safari",
	@"WebKit is located at a path containing an unsupported character. Please move WebKit to a different location and try again.");

	NSMutableArray *arguments = [NSMutableArray arrayWithObject:executablePath];
	NSMutableDictionary *environment = [[[NSDictionary dictionaryWithObjectsAndKeys:frameworkPath, @"DYLD_FRAMEWORK_PATH", @"YES", @"WEBKIT_UNSET_DYLD_FRAMEWORK_PATH",
	pathToEnablerLib, @"DYLD_INSERT_LIBRARIES", [[NSBundle mainBundle] executablePath], @"WebKitAppPath", nil] mutableCopy] autorelease];
	[environment addEntriesFromDictionary:[[NSProcessInfo processInfo] environment]];
	addStartPageToArgumentsIfNeeded(arguments);

	while (*++argv)
	[arguments addObject:[NSString stringWithUTF8String:*argv]];

	myExecve(executablePath, arguments, environment);

	char *error = strerror(errno);
	NSString *errorMessage = [NSString stringWithFormat:@"Launching Safari at %@ failed with the error '%s' (%d)", [safariBundle bundlePath], error, errno];
	displayErrorAndQuit(@"Unable to launch Safari", errorMessage);

	[pool release];
	return 0;
	}