2006-12-04 Don Gibson <dgibson77@gmail.com>
[WebKit-https.git] / WebKitTools / WebKitLauncher / WebKitNightlyEnabler.m
index 589826736ddfb8b1be3dd236a1e5bd7c5ba0ca9d..e4750125906d44f4ab085cb283a011995eb3e647 100644 (file)
 #include <string.h>
 
 static void cleanUpAfterOurselves(void) __attribute__ ((constructor));
+static void *symbol_lookup(char *symbol);
 
-void *symbol_lookup(char *symbol);
+static NSString *WKNERunState = @"WKNERunState";
+static NSString *WKNEShouldMonitorShutdowns = @"WKNEShouldMonitorShutdowns";
+
+typedef enum {
+    RunStateShutDown,
+    RunStateInitializing,
+    RunStateRunning
+} WKNERunStates;
+
+static bool extensionBundlesWereLoaded = NO;
+static NSSet *extensionPaths = nil;
+
+static void myBundleDidLoad(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo)
+{
+    // Break out early if we have already detected an extension
+    if (extensionBundlesWereLoaded)
+        return;
+
+    NSBundle *bundle = (NSBundle *)object;
+    NSString *bundlePath = [[bundle bundlePath] stringByAbbreviatingWithTildeInPath];
+    NSString *bundleFileName = [bundlePath lastPathComponent];
+
+    // Explicitly ignore SIMBL.bundle, as its only purpose is to load extensions
+    // on a per-application basis.  It's presence indicates a user has application
+    // extensions, but not that any will be loaded into Safari
+    if ([bundleFileName isEqualToString:@"SIMBL.bundle"])
+        return;
+
+    // If the bundle lives inside a known extension path, flag it as an extension
+    NSEnumerator *e = [extensionPaths objectEnumerator];
+    NSString *path = nil;
+    while (path = [e nextObject]) {
+        if ([bundlePath length] < [path length])
+            continue;
+
+        if ([[bundlePath substringToIndex:[path length]] isEqualToString:path]) {
+            extensionBundlesWereLoaded = YES;
+            break;
+        }
+    }
+}
+
+static void myApplicationWillFinishLaunching(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo)
+{
+    CFNotificationCenterRemoveObserver(CFNotificationCenterGetLocalCenter(), &myApplicationWillFinishLaunching, NULL, NULL);
+    CFNotificationCenterRemoveObserver(CFNotificationCenterGetLocalCenter(), &myBundleDidLoad, NULL, NULL);
+    [extensionPaths release];
+    extensionPaths = nil;
+
+    NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
+    [userDefaults setInteger:RunStateRunning forKey:WKNERunState];
+    [userDefaults synchronize];
+
+    if (extensionBundlesWereLoaded)
+        NSRunInformationalAlertPanel(@"Safari extensions detected",
+                                     @"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",
+                                     nil, nil);
+}
+
+static void myApplicationWillTerminate(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo)
+{
+    NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
+    [userDefaults setInteger:RunStateShutDown forKey:WKNERunState];
+    [userDefaults synchronize];
+}
 
 void cleanUpAfterOurselves(void)
 {
@@ -48,6 +113,45 @@ void cleanUpAfterOurselves(void)
     *procPath = procPathBackup;
     unsetenv("DYLD_INSERT_LIBRARIES");
     unsetenv("CFProcessPath");
+
+    extensionPaths = [[NSSet alloc] initWithObjects:@"~/Library/InputManagers/", @"/Library/InputManagers/",
+                                                    @"~/Library/Application Support/SIMBL/Plugins/", @"/Library/Application Support/SIMBL/Plugins/",
+                                                    @"~/Library/Application Enhancers/", @"/Library/Application Enhancers/",
+                                                    nil];
+
+    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
+    NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
+    NSDictionary *defaultPrefs = [NSDictionary dictionaryWithObjectsAndKeys:[NSNumber numberWithBool:RunStateShutDown], WKNERunState,
+                                                                            [NSNumber numberWithBool:YES], WKNEShouldMonitorShutdowns, nil];
+    [userDefaults registerDefaults:defaultPrefs];
+    if ([userDefaults boolForKey:WKNEShouldMonitorShutdowns]) {
+        WKNERunStates savedState = (WKNERunStates)[userDefaults integerForKey:WKNERunState];
+        if (savedState == RunStateInitializing) {
+            // Use CoreFoundation here as AppKit hasn't been initialized at this stage of Safari's lifetime
+            CFOptionFlags responseFlags;
+            CFUserNotificationDisplayAlert(0, kCFUserNotificationCautionAlertLevel,
+                                           NULL, NULL, NULL,
+                                           CFSTR("WebKit failed to open correctly"),
+                                           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"), 
+                                           CFSTR("Continue"), NULL, NULL, &responseFlags);
+        }
+        else if (savedState == RunStateRunning) {
+            NSLog(@"WebKit failed to shut down cleanly.  Checking for Safari extensions.");
+            CFNotificationCenterAddObserver(CFNotificationCenterGetLocalCenter(), &myBundleDidLoad,
+                                            myBundleDidLoad, (CFStringRef) NSBundleDidLoadNotification,
+                                            NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
+        }
+    }
+    [userDefaults setInteger:RunStateInitializing forKey:WKNERunState];
+    [userDefaults synchronize];
+
+    CFNotificationCenterAddObserver(CFNotificationCenterGetLocalCenter(), &myApplicationWillFinishLaunching,
+                                    myApplicationWillFinishLaunching, (CFStringRef) NSApplicationWillFinishLaunchingNotification,
+                                    NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
+    CFNotificationCenterAddObserver(CFNotificationCenterGetLocalCenter(), &myApplicationWillTerminate,
+                                    myApplicationWillTerminate, (CFStringRef) NSApplicationWillTerminateNotification,
+                                    NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
+    [pool release];
 }
 
 #if __LP64__
@@ -65,13 +169,12 @@ void cleanUpAfterOurselves(void)
 #define macho_nlist nlist
 #endif
 
-void *GDSymbolLookup(const struct macho_header *header, const char *symbol);
+static void *GDSymbolLookup(const struct macho_header *header, const char *symbol);
 
-void *symbol_lookup(char *symbol)
+static void *symbol_lookup(char *symbol)
 {
     int i;
-    for(i=0;i<_dyld_image_count();i++)
-    {    
+    for (i = 0; i < _dyld_image_count(); i++) {
         void *symbolResult = GDSymbolLookup((const struct macho_header*)_dyld_get_image_header(i), symbol);
         if (symbolResult)
             return symbolResult;
@@ -79,90 +182,84 @@ void *symbol_lookup(char *symbol)
     return NULL;
 }
 
-void *GDSymbolLookup(const struct macho_header *header, const char *symbol)
+static void *GDSymbolLookup(const struct macho_header *header, const char *symbol)
 {
     if (!header || !symbol)
         return NULL;
-    if ((header->magic != MH_MAGIC) && (header->magic != MH_MAGIC_64))
+
+    if (header->magic != MH_MAGIC && header->magic != MH_MAGIC_64)
         return NULL;
-    
+
     uint32_t currCommand;
     const struct load_command *loadCommand = (const struct load_command *)( ((void *)header) + sizeof(struct macho_header));
     const struct macho_segment_command *segCommand;
-    
+
     const struct symtab_command *symtabCommand = NULL;
     const struct dysymtab_command *dysymtabCommand = NULL;
     const struct macho_segment_command *textSegment = NULL;
     const struct macho_segment_command *linkEditSegment = NULL;
-    
-    for (currCommand = 0; currCommand < header->ncmds; currCommand++)
-    {
-        switch (loadCommand->cmd)
-        {
+
+    for (currCommand = 0; currCommand < header->ncmds; currCommand++) {
+        switch (loadCommand->cmd) {
             case LC_SEGMENT_COMMAND:
                 segCommand = (const struct macho_segment_command *)loadCommand;
-                if (strcmp(segCommand->segname, "__TEXT")==0)
+                if (!strcmp(segCommand->segname, "__TEXT"))
                     textSegment = segCommand;
-                else if (strcmp(segCommand->segname, "__LINKEDIT")==0)
+                else if (!strcmp(segCommand->segname, "__LINKEDIT"))
                     linkEditSegment = segCommand;
                     break;
-                
+
             case LC_SYMTAB:
                 symtabCommand = (const struct symtab_command *)loadCommand;
                 break;
-                
+
             case LC_DYSYMTAB:
                 dysymtabCommand = (const struct dysymtab_command *)loadCommand;
                 break;
         }
-        
+
         loadCommand = (const struct load_command *)(((void*)loadCommand) + loadCommand->cmdsize);
     }
-    if (textSegment==NULL || linkEditSegment==NULL || symtabCommand==NULL || dysymtabCommand==NULL) {
+    if (!textSegment || !linkEditSegment || !symtabCommand || !dysymtabCommand)
         return NULL;
-    }
-    
+
     typedef enum { Start = 0, LocalSymbols, ExternalSymbols, Done } SymbolSearchState;
     uint32_t currentSymbolIndex;
     uint32_t maximumSymbolIndex;
     SymbolSearchState state;
-    
-    for (state = Start + 1; state < Done; state++)
-    {
-        switch(state) {
+
+    for (state = Start + 1; state < Done; state++) {
+        switch (state) {
             case LocalSymbols:
                 currentSymbolIndex = dysymtabCommand->ilocalsym;
                 maximumSymbolIndex = dysymtabCommand->ilocalsym + dysymtabCommand->nlocalsym;
                 break;
-                
+
             case ExternalSymbols:
                 currentSymbolIndex = dysymtabCommand->iextdefsym;
                 maximumSymbolIndex = dysymtabCommand->nextdefsym;
                 break;
-                
+
             default:
                 return NULL;
         }
-        for (; currentSymbolIndex < maximumSymbolIndex; currentSymbolIndex++)
-        {
+        for (; currentSymbolIndex < maximumSymbolIndex; currentSymbolIndex++) {
             const struct macho_nlist *symbolTableEntry;
             symbolTableEntry = (const struct macho_nlist *)(currentSymbolIndex*sizeof(struct macho_nlist)
                                                             + (ptrdiff_t)header + symtabCommand->symoff
                                                             + linkEditSegment->vmaddr - linkEditSegment->fileoff
                                                             - textSegment->vmaddr);
             int32_t stringTableIndex = symbolTableEntry->n_un.n_strx;
-            if (stringTableIndex<0)
+            if (stringTableIndex < 0)
                 continue;
             const char *stringTableEntry = (const char*)(stringTableIndex + (ptrdiff_t)header + symtabCommand->stroff
                                                          + linkEditSegment->vmaddr - linkEditSegment->fileoff
                                                          - textSegment->vmaddr);
-            if (strcmp(symbol, stringTableEntry)==0) {
+            if (!strcmp(symbol, stringTableEntry))
                 return ((void*)header) - textSegment->vmaddr + symbolTableEntry->n_value;
-            }
         }
         state++;
     }
     return NULL;
 }
 
-