/* * Copyright (C) 2005, 2006, 2007 Apple, Inc. All rights reserved. * (C) 2007 Graham Dennis (graham.dennis@gmail.com) * * 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 Computer, 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 "DumpRenderTree.h" #import "EditingDelegate.h" #import "EventSendingController.h" #import "FrameLoadDelegate.h" #import "NavigationController.h" #import "ObjCPlugin.h" #import "ObjCPluginFunction.h" #import "PolicyDelegate.h" #import "ResourceLoadDelegate.h" #import "UIDelegate.h" #import // for CMSetDefaultProfileBySpace #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import #import // for class_poseAs #import #define COMMON_DIGEST_FOR_OPENSSL #import // for MD5 functions @interface DumpRenderTreeWindow : NSWindow @end @interface DumpRenderTreePasteboard : NSPasteboard - (int)declareType:(NSString *)type owner:(id)newOwner; @end @interface DumpRenderTreeEvent : NSEvent @end @interface LocalPasteboard : NSPasteboard { NSMutableArray *typesArray; NSMutableSet *typesSet; NSMutableDictionary *dataByType; int changeCount; } @end BOOL windowIsKey = YES; WebFrame *mainFrame = 0; BOOL shouldDumpSubframesAsText; BOOL shouldDumpEditingCallbacks; BOOL shouldDumpResourceLoadCallbacks; BOOL shouldDumpFrameLoadCallbacks; NSMutableSet *disallowedURLs = 0; BOOL waitToDump; // TRUE if waitUntilDone() has been called, but notifyDone() has not yet been called BOOL canOpenWindows; BOOL closeWebViews; BOOL closeRemainingWindowsWhenComplete = YES; BOOL addFileToPasteboardOnDrag = NO; static void runTest(const char *pathOrURL); static NSString *md5HashStringForBitmap(CGImageRef bitmap); static void displayWebView(); volatile BOOL done; NavigationController *navigationController = nil; static NSTimer *waitToDumpWatchdog; static NSTimeInterval waitToDumpWatchdogInterval = 10; // seconds // Delegates static FrameLoadDelegate *frameLoadDelegate; static UIDelegate *uiDelegate; static EditingDelegate *editingDelegate; static ResourceLoadDelegate *resourceLoadDelegate; static PolicyDelegate *policyDelegate; // Deciding when it's OK to dump out the state is a bit tricky. All these must be true: // - There is no load in progress // - There is no work queued up (see workQueue var, below) // - waitToDump==NO. This means either waitUntilDone was never called, or it was called // and notifyDone was called subsequently. // Note that the call to notifyDone and the end of the load can happen in either order. // This is the topmost frame that is loading, during a given load, or nil when no load is // in progress. Usually this is the same as the main frame, but not always. In the case // where a frameset is loaded, and then new content is loaded into one of the child frames, // that child frame is the "topmost frame that is loading". WebFrame *topLoadingFrame = nil; // !nil iff a load is in progress static BOOL dumpAsText; static BOOL dumpDOMAsWebArchive; static BOOL dumpSourceAsWebArchive; static BOOL dumpSelectionRect; BOOL dumpTitleChanges = NO; static BOOL dumpBackForwardList; static BOOL dumpChildFrameScrollPositions; static BOOL dumpChildFramesAsText; static int dumpPixels; static int paint; static int dumpAllPixels; static int threaded; static BOOL readFromWindow; static int testRepaintDefault; static BOOL testRepaint; static int repaintSweepHorizontallyDefault; static BOOL repaintSweepHorizontally; static int dumpTree = YES; static BOOL printSeparators; static NSString *currentTest = nil; static NSMutableDictionary *localPasteboards; static WebHistoryItem *prevTestBFItem = nil; // current b/f item at the end of the previous test static unsigned char* screenCaptureBuffer; static CGColorSpaceRef sharedColorSpace; // a queue of NSInvocations, queued by callouts from the test, to be exec'ed when the load is done NSMutableArray *workQueue = nil; // to prevent infinite loops, only the first page of a test can add to a work queue // (since we may well come back to that same page) BOOL workQueueFrozen = NO; const unsigned maxViewHeight = 600; const unsigned maxViewWidth = 800; static CFMutableArrayRef allWindowsRef; static pthread_mutex_t javaScriptThreadsMutex = PTHREAD_MUTEX_INITIALIZER; static BOOL javaScriptThreadsShouldTerminate; static const int javaScriptThreadsCount = 4; static CFMutableDictionaryRef javaScriptThreads() { assert(pthread_mutex_trylock(&javaScriptThreadsMutex) == EBUSY); static CFMutableDictionaryRef staticJavaScriptThreads; if (!staticJavaScriptThreads) staticJavaScriptThreads = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, NULL, NULL); return staticJavaScriptThreads; } // Loops forever, running a script and randomly respawning, until // javaScriptThreadsShouldTerminate becomes true. void* runJavaScriptThread(void* arg) { const char* const script = "var array = [];" "for (var i = 0; i < 10; i++) {" " array.push(String(i));" "}"; while(1) { JSGlobalContextRef ctx = JSGlobalContextCreate(NULL); JSStringRef scriptRef = JSStringCreateWithUTF8CString(script); JSValueRef exception = NULL; JSEvaluateScript(ctx, scriptRef, NULL, NULL, 0, &exception); assert(!exception); JSGlobalContextRelease(ctx); JSStringRelease(scriptRef); JSGarbageCollect(ctx); pthread_mutex_lock(&javaScriptThreadsMutex); // Check for cancellation. if (javaScriptThreadsShouldTerminate) { pthread_mutex_unlock(&javaScriptThreadsMutex); return 0; } // Respawn probabilistically. if (random() % 5 == 0) { pthread_t pthread; pthread_create(&pthread, NULL, &runJavaScriptThread, NULL); pthread_detach(pthread); CFDictionaryRemoveValue(javaScriptThreads(), pthread_self()); CFDictionaryAddValue(javaScriptThreads(), pthread, NULL); pthread_mutex_unlock(&javaScriptThreadsMutex); return 0; } pthread_mutex_unlock(&javaScriptThreadsMutex); } } static void startJavaScriptThreads(void) { pthread_mutex_lock(&javaScriptThreadsMutex); for (int i = 0; i < javaScriptThreadsCount; i++) { pthread_t pthread; pthread_create(&pthread, NULL, &runJavaScriptThread, NULL); pthread_detach(pthread); CFDictionaryAddValue(javaScriptThreads(), pthread, NULL); } pthread_mutex_unlock(&javaScriptThreadsMutex); } static void stopJavaScriptThreads(void) { pthread_mutex_lock(&javaScriptThreadsMutex); javaScriptThreadsShouldTerminate = YES; const pthread_t pthreads[javaScriptThreadsCount]; assert(CFDictionaryGetCount(javaScriptThreads()) == javaScriptThreadsCount); CFDictionaryGetKeysAndValues(javaScriptThreads(), (const void**)pthreads, NULL); pthread_mutex_unlock(&javaScriptThreadsMutex); for (int i = 0; i < javaScriptThreadsCount; i++) { pthread_t pthread = pthreads[i]; pthread_join(pthread, NULL); } } static BOOL shouldIgnoreWebCoreNodeLeaks(CFStringRef URLString) { static CFStringRef const ignoreSet[] = { // Keeping this infrastructure around in case we ever need it again. }; static const int ignoreSetCount = sizeof(ignoreSet) / sizeof(CFStringRef); for (int i = 0; i < ignoreSetCount; i++) { CFStringRef ignoreString = ignoreSet[i]; CFRange range = CFRangeMake(0, CFStringGetLength(URLString)); CFOptionFlags flags = kCFCompareAnchored | kCFCompareBackwards | kCFCompareCaseInsensitive; if (CFStringFindWithOptions(URLString, ignoreString, range, flags, NULL)) return YES; } return NO; } static CMProfileRef currentColorProfile = 0; static void restoreColorSpace(int ignored) { if (currentColorProfile) { int error = CMSetDefaultProfileByUse(cmDisplayUse, currentColorProfile); if (error) fprintf(stderr, "Failed to retore previous color profile! You may need to open System Preferences : Displays : Color and manually restore your color settings. (Error: %i)", error); currentColorProfile = 0; } } static void crashHandler(int sig) { fprintf(stderr, "%s\n", strsignal(sig)); restoreColorSpace(0); exit(128 + sig); } static void activateAhemFont(void) { unsigned long fontDataLength; char* fontData = getsectdata("__DATA", "Ahem", &fontDataLength); if (!fontData) { fprintf(stderr, "Failed to locate the Ahem font.\n"); exit(1); } ATSFontContainerRef fontContainer; OSStatus status = ATSFontActivateFromMemory(fontData, fontDataLength, kATSFontContextLocal, kATSFontFormatUnspecified, NULL, kATSOptionFlagsDefault, &fontContainer); if (status != noErr) { fprintf(stderr, "Failed to activate the Ahem font.\n"); exit(1); } } static void setDefaultColorProfileToRGB(void) { CMProfileRef genericProfile = [[NSColorSpace genericRGBColorSpace] colorSyncProfile]; CMProfileRef previousProfile; int error = CMGetDefaultProfileByUse(cmDisplayUse, &previousProfile); if (error) { fprintf(stderr, "Failed to get current color profile. I will not be able to restore your current profile, thus I'm not changing it. Many pixel tests may fail as a result. (Error: %i)\n", error); return; } if (previousProfile == genericProfile) return; CFStringRef previousProfileName; CFStringRef genericProfileName; char previousProfileNameString[1024]; char genericProfileNameString[1024]; CMCopyProfileDescriptionString(previousProfile, &previousProfileName); CMCopyProfileDescriptionString(genericProfile, &genericProfileName); CFStringGetCString(previousProfileName, previousProfileNameString, sizeof(previousProfileNameString), kCFStringEncodingUTF8); CFStringGetCString(genericProfileName, genericProfileNameString, sizeof(previousProfileNameString), kCFStringEncodingUTF8); CFRelease(genericProfileName); CFRelease(previousProfileName); fprintf(stderr, "\n\nWARNING: Temporarily changing your system color profile from \"%s\" to \"%s\".\n", previousProfileNameString, genericProfileNameString); fprintf(stderr, "This allows the WebKit pixel-based regression tests to have consistent color values across all machines.\n"); fprintf(stderr, "The colors on your screen will change for the duration of the testing.\n\n"); if ((error = CMSetDefaultProfileByUse(cmDisplayUse, genericProfile))) fprintf(stderr, "Failed to set color profile to \"%s\"! Many pixel tests will fail as a result. (Error: %i)", genericProfileNameString, error); else { currentColorProfile = previousProfile; signal(SIGINT, restoreColorSpace); signal(SIGHUP, restoreColorSpace); signal(SIGTERM, restoreColorSpace); } } static void* (*savedMalloc)(malloc_zone_t*, size_t); static void* (*savedRealloc)(malloc_zone_t*, void*, size_t); static void* checkedMalloc(malloc_zone_t* zone, size_t size) { if (size >= 0x10000000) return 0; return savedMalloc(zone, size); } static void* checkedRealloc(malloc_zone_t* zone, void* ptr, size_t size) { if (size >= 0x10000000) return 0; return savedRealloc(zone, ptr, size); } static void makeLargeMallocFailSilently(void) { malloc_zone_t* zone = malloc_default_zone(); savedMalloc = zone->malloc; savedRealloc = zone->realloc; zone->malloc = checkedMalloc; zone->realloc = checkedRealloc; } WebView *createWebView() { NSRect rect = NSMakeRect(0, 0, maxViewWidth, maxViewHeight); WebView *webView = [[WebView alloc] initWithFrame:rect frameName:nil groupName:@"org.webkit.DumpRenderTree"]; [webView setUIDelegate:uiDelegate]; [webView setFrameLoadDelegate:frameLoadDelegate]; [webView setEditingDelegate:editingDelegate]; [webView setResourceLoadDelegate:resourceLoadDelegate]; // Register the same schemes that Safari does [WebView registerURLSchemeAsLocal:@"feed"]; [WebView registerURLSchemeAsLocal:@"feeds"]; [WebView registerURLSchemeAsLocal:@"feedsearch"]; // The back/forward cache is causing problems due to layouts during transition from one page to another. // So, turn it off for now, but we might want to turn it back on some day. [[webView backForwardList] setPageCacheSize:0]; [webView setContinuousSpellCheckingEnabled:YES]; // To make things like certain NSViews, dragging, and plug-ins work, put the WebView a window, but put it off-screen so you don't see it. // Put it at -10000, -10000 in "flipped coordinates", since WebCore and the DOM use flipped coordinates. NSRect windowRect = NSOffsetRect(rect, -10000, [[[NSScreen screens] objectAtIndex:0] frame].size.height - rect.size.height + 10000); DumpRenderTreeWindow *window = [[DumpRenderTreeWindow alloc] initWithContentRect:windowRect styleMask:NSBorderlessWindowMask backing:NSBackingStoreBuffered defer:YES]; [[window contentView] addSubview:webView]; [window orderBack:nil]; [window setAutodisplay:NO]; // For reasons that are not entirely clear, the following pair of calls makes WebView handle its // dynamic scrollbars properly. Without it, every frame will always have scrollbars. NSBitmapImageRep *imageRep = [webView bitmapImageRepForCachingDisplayInRect:[webView bounds]]; [webView cacheDisplayInRect:[webView bounds] toBitmapImageRep:imageRep]; return webView; } void testStringByEvaluatingJavaScriptFromString() { // maps expected result <= JavaScript expression NSDictionary *expressions = [NSDictionary dictionaryWithObjectsAndKeys: @"0", @"0", @"0", @"'0'", @"", @"", @"", @"''", @"", @"new String()", @"", @"new String('0')", @"", @"throw 1", @"", @"{ }", @"", @"[ ]", @"", @"//", @"", @"a.b.c", @"", @"(function() { throw 'error'; })()", @"", @"null", @"", @"undefined", @"true", @"true", @"false", @"false", nil ]; NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init]; WebView *webView = [[WebView alloc] initWithFrame:NSZeroRect frameName:@"" groupName:@""]; NSEnumerator *enumerator = [expressions keyEnumerator]; id expression; while ((expression = [enumerator nextObject])) { NSString *expectedResult = [expressions objectForKey:expression]; NSString *result = [webView stringByEvaluatingJavaScriptFromString:expression]; assert([result isEqualToString:expectedResult]); } [webView close]; [webView release]; [pool release]; } void dumpRenderTree(int argc, const char *argv[]) { [NSApplication sharedApplication]; class_poseAs(objc_getClass("DumpRenderTreePasteboard"), objc_getClass("NSPasteboard")); class_poseAs(objc_getClass("DumpRenderTreeEvent"), objc_getClass("NSEvent")); struct option options[] = { {"dump-all-pixels", no_argument, &dumpAllPixels, YES}, {"horizontal-sweep", no_argument, &repaintSweepHorizontallyDefault, YES}, {"notree", no_argument, &dumpTree, NO}, {"pixel-tests", no_argument, &dumpPixels, YES}, {"paint", no_argument, &paint, YES}, {"repaint", no_argument, &testRepaintDefault, YES}, {"tree", no_argument, &dumpTree, YES}, {"threaded", no_argument, &threaded, YES}, {NULL, 0, NULL, 0} }; NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults]; [defaults setObject:@"DoubleMax" forKey:@"AppleScrollBarVariant"]; [defaults setInteger:4 forKey:@"AppleAntiAliasingThreshold"]; // 2 is the "Medium" font smoothing mode [defaults setInteger:2 forKey:@"AppleFontSmoothing"]; [defaults setInteger:1 forKey:@"AppleAquaColorVariant"]; [defaults setObject:@"0.709800 0.835300 1.000000" forKey:@"AppleHighlightColor"]; [defaults setObject:@"0.500000 0.500000 0.500000" forKey:@"AppleOtherHighlightColor"]; [defaults setObject:[NSArray arrayWithObject:@"en"] forKey:@"AppleLanguages"]; WebPreferences *preferences = [WebPreferences standardPreferences]; [preferences setStandardFontFamily:@"Times"]; [preferences setFixedFontFamily:@"Courier"]; [preferences setSerifFontFamily:@"Times"]; [preferences setSansSerifFontFamily:@"Helvetica"]; [preferences setCursiveFontFamily:@"Apple Chancery"]; [preferences setFantasyFontFamily:@"Papyrus"]; [preferences setDefaultFontSize:16]; [preferences setDefaultFixedFontSize:13]; [preferences setMinimumFontSize:1]; [preferences setJavaEnabled:NO]; [preferences setJavaScriptCanOpenWindowsAutomatically:YES]; [preferences setEditableLinkBehavior:WebKitEditableLinkOnlyLiveWithShiftKey]; [preferences setTabsToLinks:NO]; [preferences setDOMPasteAllowed:YES]; int option; while ((option = getopt_long(argc, (char * const *)argv, "", options, NULL)) != -1) switch (option) { case '?': // unknown or ambiguous option case ':': // missing argument exit(1); break; } activateAhemFont(); if (dumpPixels) { setDefaultColorProfileToRGB(); screenCaptureBuffer = malloc(maxViewHeight * maxViewWidth * 4); sharedColorSpace = CGColorSpaceCreateDeviceRGB(); } localPasteboards = [[NSMutableDictionary alloc] init]; navigationController = [[NavigationController alloc] init]; frameLoadDelegate = [[FrameLoadDelegate alloc] init]; uiDelegate = [[UIDelegate alloc] init]; editingDelegate = [[EditingDelegate alloc] init]; resourceLoadDelegate = [[ResourceLoadDelegate alloc] init]; policyDelegate = [[PolicyDelegate alloc] init]; NSString *pwd = [[NSString stringWithUTF8String:argv[0]] stringByDeletingLastPathComponent]; [WebPluginDatabase setAdditionalWebPlugInPaths:[NSArray arrayWithObject:pwd]]; [[WebPluginDatabase sharedDatabase] refresh]; WebView *webView = createWebView(); mainFrame = [webView mainFrame]; NSWindow *window = [webView window]; workQueue = [[NSMutableArray alloc] init]; makeLargeMallocFailSilently(); signal(SIGILL, crashHandler); /* 4: illegal instruction (not reset when caught) */ signal(SIGTRAP, crashHandler); /* 5: trace trap (not reset when caught) */ signal(SIGEMT, crashHandler); /* 7: EMT instruction */ signal(SIGFPE, crashHandler); /* 8: floating point exception */ signal(SIGBUS, crashHandler); /* 10: bus error */ signal(SIGSEGV, crashHandler); /* 11: segmentation violation */ signal(SIGSYS, crashHandler); /* 12: bad argument to system call */ signal(SIGPIPE, crashHandler); /* 13: write on a pipe with no reader */ signal(SIGXCPU, crashHandler); /* 24: exceeded CPU time limit */ signal(SIGXFSZ, crashHandler); /* 25: exceeded file size limit */ [[NSURLCache sharedURLCache] removeAllCachedResponses]; // testStringByEvaluatingJavaScriptFromString(); if (threaded) startJavaScriptThreads(); if (argc == optind+1 && strcmp(argv[optind], "-") == 0) { char filenameBuffer[2048]; printSeparators = YES; while (fgets(filenameBuffer, sizeof(filenameBuffer), stdin)) { char *newLineCharacter = strchr(filenameBuffer, '\n'); if (newLineCharacter) *newLineCharacter = '\0'; if (strlen(filenameBuffer) == 0) continue; runTest(filenameBuffer); } } else { printSeparators = (optind < argc-1 || (dumpPixels && dumpTree)); for (int i = optind; i != argc; ++i) runTest(argv[i]); } if (threaded) stopJavaScriptThreads(); [workQueue release]; [WebCoreStatistics emptyCache]; // Otherwise SVGImages trigger false positives for Frame/Node counts [webView close]; mainFrame = nil; // Work around problem where registering drag types leaves an outstanding // "perform selector" on the window, which retains the window. It's a bit // inelegant and perhaps dangerous to just blow them all away, but in practice // it probably won't cause any trouble (and this is just a test tool, after all). [NSObject cancelPreviousPerformRequestsWithTarget:window]; [window close]; // releases when closed [webView release]; [frameLoadDelegate release]; [editingDelegate release]; [resourceLoadDelegate release]; [uiDelegate release]; [policyDelegate release]; [localPasteboards release]; localPasteboards = nil; [navigationController release]; navigationController = nil; [disallowedURLs release]; disallowedURLs = nil; if (dumpPixels) restoreColorSpace(0); } int main(int argc, const char *argv[]) { NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init]; dumpRenderTree(argc, argv); [WebCoreStatistics garbageCollectJavaScriptObjects]; [pool release]; return 0; } static int compareHistoryItems(id item1, id item2, void *context) { return [[item1 target] caseInsensitiveCompare:[item2 target]]; } static void dumpHistoryItem(WebHistoryItem *item, int indent, BOOL current) { int start = 0; if (current) { printf("curr->"); start = 6; } for (int i = start; i < indent; i++) putchar(' '); printf("%s", [[item URLString] UTF8String]); NSString *target = [item target]; if (target && [target length] > 0) printf(" (in frame \"%s\")", [target UTF8String]); if ([item isTargetItem]) printf(" **nav target**"); putchar('\n'); NSArray *kids = [item children]; if (kids) { // must sort to eliminate arbitrary result ordering which defeats reproducible testing kids = [kids sortedArrayUsingFunction:&compareHistoryItems context:nil]; for (unsigned i = 0; i < [kids count]; i++) dumpHistoryItem([kids objectAtIndex:i], indent+4, NO); } } static void dumpFrameScrollPosition(WebFrame *f) { NSPoint scrollPosition = [[[[f frameView] documentView] superview] bounds].origin; if (ABS(scrollPosition.x) > 0.00000001 || ABS(scrollPosition.y) > 0.00000001) { if ([f parentFrame] != nil) printf("frame '%s' ", [[f name] UTF8String]); printf("scrolled to %.f,%.f\n", scrollPosition.x, scrollPosition.y); } if (dumpChildFrameScrollPositions) { NSArray *kids = [f childFrames]; if (kids) for (unsigned i = 0; i < [kids count]; i++) dumpFrameScrollPosition([kids objectAtIndex:i]); } } static NSString *dumpFramesAsText(WebFrame *frame) { if (!frame) return @""; DOMDocument *document = [frame DOMDocument]; if (!document) return @""; DOMElement *documentElement = [document documentElement]; if (!documentElement) return @""; NSMutableString *result = [[[NSMutableString alloc] init] autorelease]; // Add header for all but the main frame. if ([frame parentFrame]) result = [NSMutableString stringWithFormat:@"\n--------\nFrame: '%@'\n--------\n", [frame name]]; [result appendFormat:@"%@\n", [documentElement innerText]]; if (dumpChildFramesAsText) { NSArray *kids = [frame childFrames]; if (kids) { for (unsigned i = 0; i < [kids count]; i++) [result appendString:dumpFramesAsText([kids objectAtIndex:i])]; } } return result; } static void convertMIMEType(NSMutableString *mimeType) { if ([mimeType isEqualToString:@"application/x-javascript"]) [mimeType setString:@"text/javascript"]; } static void convertWebResourceDataToString(NSMutableDictionary *resource) { NSMutableString *mimeType = [resource objectForKey:@"WebResourceMIMEType"]; convertMIMEType(mimeType); if ([mimeType hasPrefix:@"text/"]) { NSData *data = [resource objectForKey:@"WebResourceData"]; NSString *dataAsString = [[[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding] autorelease]; [resource setObject:dataAsString forKey:@"WebResourceData"]; } } static void normalizeWebResourceURL(NSMutableString *webResourceURL, NSString *oldURLBase) { [webResourceURL replaceOccurrencesOfString:oldURLBase withString:@"file://" options:NSLiteralSearch range:NSMakeRange(0, [webResourceURL length])]; } static void convertWebResourceResponseToDictionary(NSMutableDictionary *propertyList, NSString *oldURLBase) { NSURLResponse *response = nil; NSData *responseData = [propertyList objectForKey:@"WebResourceResponse"]; // WebResourceResponseKey in WebResource.m if ([responseData isKindOfClass:[NSData class]]) { // Decode NSURLResponse NSKeyedUnarchiver *unarchiver = [[NSKeyedUnarchiver alloc] initForReadingWithData:responseData]; response = [unarchiver decodeObjectForKey:@"WebResourceResponse"]; // WebResourceResponseKey in WebResource.m [unarchiver finishDecoding]; [unarchiver release]; } NSMutableDictionary *responseDictionary = [[NSMutableDictionary alloc] init]; NSMutableString *urlString = [[[response URL] description] mutableCopy]; normalizeWebResourceURL(urlString, oldURLBase); [responseDictionary setObject:urlString forKey:@"URL"]; [urlString release]; NSMutableString *mimeTypeString = [[response MIMEType] mutableCopy]; convertMIMEType(mimeTypeString); [responseDictionary setObject:mimeTypeString forKey:@"MIMEType"]; [mimeTypeString release]; NSString *textEncodingName = [response textEncodingName]; if (textEncodingName) [responseDictionary setObject:textEncodingName forKey:@"textEncodingName"]; [responseDictionary setObject:[NSNumber numberWithLongLong:[response expectedContentLength]] forKey:@"expectedContentLength"]; if ([response isKindOfClass:[NSHTTPURLResponse class]]) { NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)response; [responseDictionary setObject:[httpResponse allHeaderFields] forKey:@"allHeaderFields"]; [responseDictionary setObject:[NSNumber numberWithInt:[httpResponse statusCode]] forKey:@"statusCode"]; } [propertyList setObject:responseDictionary forKey:@"WebResourceResponse"]; [responseDictionary release]; } static NSString *serializeWebArchiveToXML(WebArchive *webArchive) { NSString *errorString; NSMutableDictionary *propertyList = [NSPropertyListSerialization propertyListFromData:[webArchive data] mutabilityOption:NSPropertyListMutableContainersAndLeaves format:NULL errorDescription:&errorString]; if (!propertyList) return errorString; // Normalize WebResourceResponse and WebResourceURL values in plist for testing NSString *cwdURL = [@"file://" stringByAppendingString:[[[NSFileManager defaultManager] currentDirectoryPath] stringByExpandingTildeInPath]]; NSMutableArray *resources = [NSMutableArray arrayWithCapacity:1]; [resources addObject:propertyList]; while ([resources count]) { NSMutableDictionary *resourcePropertyList = [resources objectAtIndex:0]; [resources removeObjectAtIndex:0]; NSMutableDictionary *mainResource = [resourcePropertyList objectForKey:@"WebMainResource"]; normalizeWebResourceURL([mainResource objectForKey:@"WebResourceURL"], cwdURL); convertWebResourceDataToString(mainResource); // Add subframeArchives to list for processing NSMutableArray *subframeArchives = [resourcePropertyList objectForKey:@"WebSubframeArchives"]; // WebSubframeArchivesKey in WebArchive.m if (subframeArchives) [resources addObjectsFromArray:subframeArchives]; NSMutableArray *subresources = [resourcePropertyList objectForKey:@"WebSubresources"]; // WebSubresourcesKey in WebArchive.m NSEnumerator *enumerator = [subresources objectEnumerator]; NSMutableDictionary *subresourcePropertyList; while ((subresourcePropertyList = [enumerator nextObject])) { normalizeWebResourceURL([subresourcePropertyList objectForKey:@"WebResourceURL"], cwdURL); convertWebResourceResponseToDictionary(subresourcePropertyList, cwdURL); convertWebResourceDataToString(subresourcePropertyList); } } NSData *xmlData = [NSPropertyListSerialization dataFromPropertyList:propertyList format:NSPropertyListXMLFormat_v1_0 errorDescription:&errorString]; if (!xmlData) return errorString; NSMutableString *string = [[[NSMutableString alloc] initWithData:xmlData encoding:NSUTF8StringEncoding] autorelease]; // Replace "Apple Computer" with "Apple" in the DTD declaration. NSRange range = [string rangeOfString:@"-//Apple Computer//"]; if (range.location != NSNotFound) [string replaceCharactersInRange:range withString:@"-//Apple//"]; return string; } static void dumpBackForwardListForWebView(WebView *view) { printf("\n============== Back Forward List ==============\n"); WebBackForwardList *bfList = [view backForwardList]; // Print out all items in the list after prevTestBFItem, which was from the previous test // Gather items from the end of the list, the print them out from oldest to newest NSMutableArray *itemsToPrint = [[NSMutableArray alloc] init]; for (int i = [bfList forwardListCount]; i > 0; i--) { WebHistoryItem *item = [bfList itemAtIndex:i]; // something is wrong if the item from the last test is in the forward part of the b/f list assert(item != prevTestBFItem); [itemsToPrint addObject:item]; } assert([bfList currentItem] != prevTestBFItem); [itemsToPrint addObject:[bfList currentItem]]; int currentItemIndex = [itemsToPrint count] - 1; for (int i = -1; i >= -[bfList backListCount]; i--) { WebHistoryItem *item = [bfList itemAtIndex:i]; if (item == prevTestBFItem) break; [itemsToPrint addObject:item]; } for (int i = [itemsToPrint count]-1; i >= 0; i--) { dumpHistoryItem([itemsToPrint objectAtIndex:i], 8, i == currentItemIndex); } [itemsToPrint release]; printf("===============================================\n"); } void dump(void) { [waitToDumpWatchdog invalidate]; [waitToDumpWatchdog release]; waitToDumpWatchdog = nil; if (dumpTree) { NSString *result = nil; dumpAsText |= [[[[mainFrame dataSource] response] MIMEType] isEqualToString:@"text/plain"]; if (dumpAsText) { result = dumpFramesAsText(mainFrame); } else if (dumpDOMAsWebArchive) { WebArchive *webArchive = [[mainFrame DOMDocument] webArchive]; result = serializeWebArchiveToXML(webArchive); } else if (dumpSourceAsWebArchive) { WebArchive *webArchive = [[mainFrame dataSource] webArchive]; result = serializeWebArchiveToXML(webArchive); } else { bool isSVGW3CTest = ([currentTest rangeOfString:@"svg/W3C-SVG-1.1"].length); if (isSVGW3CTest) [[mainFrame webView] setFrameSize:NSMakeSize(480, 360)]; else [[mainFrame webView] setFrameSize:NSMakeSize(maxViewWidth, maxViewHeight)]; result = [mainFrame renderTreeAsExternalRepresentation]; } if (!result) { const char *errorMessage; if (dumpAsText) errorMessage = "[documentElement innerText]"; else if (dumpDOMAsWebArchive) errorMessage = "[[mainFrame DOMDocument] webArchive]"; else if (dumpSourceAsWebArchive) errorMessage = "[[mainFrame dataSource] webArchive]"; else errorMessage = "[mainFrame renderTreeAsExternalRepresentation]"; printf("ERROR: nil result from %s", errorMessage); } else { NSData *data = [result dataUsingEncoding:NSUTF8StringEncoding]; fwrite([data bytes], 1, [data length], stdout); if (!dumpAsText && !dumpDOMAsWebArchive && !dumpSourceAsWebArchive) dumpFrameScrollPosition(mainFrame); } if (dumpBackForwardList) { unsigned count = [(NSArray *)allWindowsRef count]; for (unsigned i = 0; i < count; i++) { NSWindow *window = [(NSArray *)allWindowsRef objectAtIndex:i]; WebView *webView = [[[window contentView] subviews] objectAtIndex:0]; dumpBackForwardListForWebView(webView); } } if (printSeparators) puts("#EOF"); } if (dumpPixels) { if (!dumpAsText && !dumpDOMAsWebArchive && !dumpSourceAsWebArchive) { // grab a bitmap from the view WebView* view = [mainFrame webView]; NSSize webViewSize = [view frame].size; CGContextRef cgContext = CGBitmapContextCreate(screenCaptureBuffer, webViewSize.width, webViewSize.height, 8, webViewSize.width * 4, sharedColorSpace, kCGBitmapByteOrder32Host | kCGImageAlphaPremultipliedLast); NSGraphicsContext* savedContext = [[[NSGraphicsContext currentContext] retain] autorelease]; NSGraphicsContext* nsContext = [NSGraphicsContext graphicsContextWithGraphicsPort:cgContext flipped:NO]; [NSGraphicsContext setCurrentContext:nsContext]; if (readFromWindow) { NSBitmapImageRep *imageRep; [view displayIfNeeded]; [view lockFocus]; imageRep = [[NSBitmapImageRep alloc] initWithFocusedViewRect:[view frame]]; [view unlockFocus]; [imageRep draw]; [imageRep release]; } else if (!testRepaint) [view displayRectIgnoringOpacity:NSMakeRect(0, 0, webViewSize.width, webViewSize.height) inContext:nsContext]; else if (!repaintSweepHorizontally) { NSRect line = NSMakeRect(0, 0, webViewSize.width, 1); while (line.origin.y < webViewSize.height) { [view displayRectIgnoringOpacity:line inContext:nsContext]; line.origin.y++; } } else { NSRect column = NSMakeRect(0, 0, 1, webViewSize.height); while (column.origin.x < webViewSize.width) { [view displayRectIgnoringOpacity:column inContext:nsContext]; column.origin.x++; } } if (dumpSelectionRect) { NSView *documentView = [[mainFrame frameView] documentView]; if ([documentView conformsToProtocol:@protocol(WebDocumentSelection)]) { [[NSColor redColor] set]; [NSBezierPath strokeRect:[documentView convertRect:[(id )documentView selectionRect] fromView:nil]]; } } [NSGraphicsContext setCurrentContext:savedContext]; CGImageRef bitmapImage = CGBitmapContextCreateImage(cgContext); CGContextRelease(cgContext); // compute the actual hash to compare to the expected image's hash NSString *actualHash = md5HashStringForBitmap(bitmapImage); printf("\nActualHash: %s\n", [actualHash UTF8String]); BOOL dumpImage; if (dumpAllPixels) dumpImage = YES; else { // FIXME: It's unfortunate that we hardcode the file naming scheme here. // At one time, the perl script had all the knowledge about file layout. // Some day we should restore that setup by passing in more parameters to this tool. NSString *baseTestPath = [currentTest stringByDeletingPathExtension]; NSString *baselineHashPath = [baseTestPath stringByAppendingString:@"-expected.checksum"]; NSString *baselineHash = [NSString stringWithContentsOfFile:baselineHashPath encoding:NSUTF8StringEncoding error:nil]; NSString *baselineImagePath = [baseTestPath stringByAppendingString:@"-expected.png"]; printf("BaselineHash: %s\n", [baselineHash UTF8String]); /// send the image to stdout if the hash mismatches or if there's no file in the file system dumpImage = ![baselineHash isEqualToString:actualHash] || access([baselineImagePath fileSystemRepresentation], F_OK) != 0; } if (dumpImage) { CFMutableDataRef imageData = CFDataCreateMutable(0, 0); CGImageDestinationRef imageDest = CGImageDestinationCreateWithData(imageData, CFSTR("public.png"), 1, 0); CGImageDestinationAddImage(imageDest, bitmapImage, 0); CGImageDestinationFinalize(imageDest); CFRelease(imageDest); printf("Content-length: %lu\n", CFDataGetLength(imageData)); fwrite(CFDataGetBytePtr(imageData), 1, CFDataGetLength(imageData), stdout); CFRelease(imageData); } CGImageRelease(bitmapImage); } printf("#EOF\n"); } fflush(stdout); if (paint) displayWebView(); done = YES; } @implementation LayoutTestController + (BOOL)isSelectorExcludedFromWebScript:(SEL)aSelector { if (0 || aSelector == @selector(accessStoredWebScriptObject) || aSelector == @selector(addDisallowedURL:) || aSelector == @selector(addFileToPasteboardOnDrag) || aSelector == @selector(clearBackForwardList) || aSelector == @selector(decodeHostName:) || aSelector == @selector(display) || aSelector == @selector(dumpAsText) || aSelector == @selector(dumpBackForwardList) || aSelector == @selector(dumpChildFrameScrollPositions) || aSelector == @selector(dumpChildFramesAsText) || aSelector == @selector(dumpDOMAsWebArchive) || aSelector == @selector(dumpEditingCallbacks) || aSelector == @selector(dumpFrameLoadCallbacks) || aSelector == @selector(dumpResourceLoadCallbacks) || aSelector == @selector(dumpSelectionRect) || aSelector == @selector(dumpSourceAsWebArchive) || aSelector == @selector(dumpTitleChanges) || aSelector == @selector(encodeHostName:) || aSelector == @selector(keepWebHistory) || aSelector == @selector(notifyDone) || aSelector == @selector(objCClassNameOf:) || aSelector == @selector(objCIdentityIsEqual::) || aSelector == @selector(objCObjectOfClass:) || aSelector == @selector(objCLongLongRoundTrip:) || aSelector == @selector(objCUnsignedLongLongRoundTrip:) || aSelector == @selector(queueBackNavigation:) || aSelector == @selector(queueForwardNavigation:) || aSelector == @selector(queueLoad:target:) || aSelector == @selector(queueReload) || aSelector == @selector(queueScript:) || aSelector == @selector(repaintSweepHorizontally) || aSelector == @selector(setAcceptsEditing:) || aSelector == @selector(setCallCloseOnWebViews:) || aSelector == @selector(setCanOpenWindows) || aSelector == @selector(setCloseRemainingWindowsWhenComplete:) || aSelector == @selector(setCustomPolicyDelegate:) || aSelector == @selector(setMainFrameIsFirstResponder:) || aSelector == @selector(setTabKeyCyclesThroughElements:) || aSelector == @selector(setUseDashboardCompatibilityMode:) || aSelector == @selector(setUserStyleSheetEnabled:) || aSelector == @selector(setUserStyleSheetLocation:) || aSelector == @selector(setWindowIsKey:) || aSelector == @selector(storeWebScriptObject:) || aSelector == @selector(testRepaint) || aSelector == @selector(testWrapperRoundTripping:) || aSelector == @selector(waitUntilDone) || aSelector == @selector(windowCount) ) return NO; return YES; } + (NSString *)webScriptNameForSelector:(SEL)aSelector { if (aSelector == @selector(setWindowIsKey:)) return @"setWindowIsKey"; if (aSelector == @selector(setMainFrameIsFirstResponder:)) return @"setMainFrameIsFirstResponder"; if (aSelector == @selector(queueBackNavigation:)) return @"queueBackNavigation"; if (aSelector == @selector(queueForwardNavigation:)) return @"queueForwardNavigation"; if (aSelector == @selector(queueScript:)) return @"queueScript"; if (aSelector == @selector(queueLoad:target:)) return @"queueLoad"; if (aSelector == @selector(setAcceptsEditing:)) return @"setAcceptsEditing"; if (aSelector == @selector(setTabKeyCyclesThroughElements:)) return @"setTabKeyCyclesThroughElements"; if (aSelector == @selector(storeWebScriptObject:)) return @"storeWebScriptObject"; if (aSelector == @selector(testWrapperRoundTripping:)) return @"testWrapperRoundTripping"; if (aSelector == @selector(setUserStyleSheetLocation:)) return @"setUserStyleSheetLocation"; if (aSelector == @selector(setUserStyleSheetEnabled:)) return @"setUserStyleSheetEnabled"; if (aSelector == @selector(objCClassNameOf:)) return @"objCClassName"; if (aSelector == @selector(objCObjectOfClass:)) return @"objCObjectOfClass"; if (aSelector == @selector(objCIdentityIsEqual::)) return @"objCIdentityIsEqual"; if (aSelector == @selector(addDisallowedURL:)) return @"addDisallowedURL"; if (aSelector == @selector(setCallCloseOnWebViews:)) return @"setCallCloseOnWebViews"; if (aSelector == @selector(setCloseRemainingWindowsWhenComplete:)) return @"setCloseRemainingWindowsWhenComplete"; if (aSelector == @selector(setCustomPolicyDelegate:)) return @"setCustomPolicyDelegate"; if (aSelector == @selector(setUseDashboardCompatibilityMode:)) return @"setUseDashboardCompatiblityMode"; if (aSelector == @selector(encodeHostName:)) return @"encodeHostName"; if (aSelector == @selector(decodeHostName:)) return @"decodeHostName"; if (aSelector == @selector(objCLongLongRoundTrip:)) return @"objCLongLongRoundTrip"; if (aSelector == @selector(objCUnsignedLongLongRoundTrip:)) return @"objCUnsignedLongLongRoundTrip"; return nil; } - (void)clearBackForwardList { WebBackForwardList *backForwardList = [[mainFrame webView] backForwardList]; WebHistoryItem *item = [[backForwardList currentItem] retain]; // We clear the history by setting the back/forward list's capacity to 0 // then restoring it back and adding back the current item. int capacity = [backForwardList capacity]; [backForwardList setCapacity:0]; [backForwardList setCapacity:capacity]; [backForwardList addItem:item]; [backForwardList goToItem:item]; [item release]; } - (void)setUseDashboardCompatibilityMode:(BOOL)flag { [[mainFrame webView] _setDashboardBehavior:WebDashboardBehaviorUseBackwardCompatibilityMode to:flag]; } - (void)setCloseRemainingWindowsWhenComplete:(BOOL)closeWindows { closeRemainingWindowsWhenComplete = closeWindows; } - (void)setCustomPolicyDelegate:(BOOL)setDelegate { if (setDelegate) [[mainFrame webView] setPolicyDelegate:policyDelegate]; else [[mainFrame webView] setPolicyDelegate:nil]; } - (void)keepWebHistory { if (![WebHistory optionalSharedHistory]) { WebHistory *history = [[WebHistory alloc] init]; [WebHistory setOptionalSharedHistory:history]; [history release]; } } - (void)setCallCloseOnWebViews:(BOOL)callClose { closeWebViews = callClose; } - (void)setCanOpenWindows { canOpenWindows = YES; } - (void)waitUntilDone { waitToDump = YES; if (!waitToDumpWatchdog) waitToDumpWatchdog = [[NSTimer scheduledTimerWithTimeInterval:waitToDumpWatchdogInterval target:self selector:@selector(waitUntilDoneWatchdogFired) userInfo:nil repeats:NO] retain]; } - (void)waitUntilDoneWatchdogFired { const char* message = "FAIL: Timed out waiting for notifyDone to be called\n"; fprintf(stderr, message); fprintf(stdout, message); dump(); } - (void)notifyDone { if (waitToDump && !topLoadingFrame && [workQueue count] == 0) dump(); waitToDump = NO; } - (void)dumpAsText { dumpAsText = YES; } - (void)addFileToPasteboardOnDrag { addFileToPasteboardOnDrag = YES; } - (void)addDisallowedURL:(NSString *)urlString { if (!disallowedURLs) disallowedURLs = [[NSMutableSet alloc] init]; // Canonicalize the URL NSURLRequest* request = [NSURLRequest requestWithURL:[NSURL URLWithString:urlString]]; request = [NSURLProtocol canonicalRequestForRequest:request]; [disallowedURLs addObject:[request URL]]; } - (void)setUserStyleSheetLocation:(NSString *)path { NSURL *url = [NSURL URLWithString:path]; [[WebPreferences standardPreferences] setUserStyleSheetLocation:url]; } - (void)setUserStyleSheetEnabled:(BOOL)flag { [[WebPreferences standardPreferences] setUserStyleSheetEnabled:flag]; } - (void)dumpDOMAsWebArchive { dumpDOMAsWebArchive = YES; } - (void)dumpSourceAsWebArchive { dumpSourceAsWebArchive = YES; } - (void)dumpSelectionRect { dumpSelectionRect = YES; } - (void)dumpTitleChanges { dumpTitleChanges = YES; } - (void)dumpBackForwardList { dumpBackForwardList = YES; } - (int)windowCount { return CFArrayGetCount(allWindowsRef); } - (void)dumpChildFrameScrollPositions { dumpChildFrameScrollPositions = YES; } - (void)dumpChildFramesAsText { dumpChildFramesAsText = YES; } - (void)dumpEditingCallbacks { shouldDumpEditingCallbacks = YES; } - (void)dumpResourceLoadCallbacks { shouldDumpResourceLoadCallbacks = YES; } - (void)dumpFrameLoadCallbacks { shouldDumpFrameLoadCallbacks = YES; } - (void)setWindowIsKey:(BOOL)flag { windowIsKey = flag; NSView *documentView = [[mainFrame frameView] documentView]; if ([documentView isKindOfClass:[WebHTMLView class]]) [(WebHTMLView *)documentView _updateActiveState]; } - (void)setMainFrameIsFirstResponder:(BOOL)flag { NSView *documentView = [[mainFrame frameView] documentView]; NSResponder *firstResponder = flag ? documentView : nil; [[[mainFrame webView] window] makeFirstResponder:firstResponder]; if ([documentView isKindOfClass:[WebHTMLView class]]) [(WebHTMLView *)documentView _updateActiveState]; } - (void)display { displayWebView(); } - (void)testRepaint { testRepaint = YES; } - (void)repaintSweepHorizontally { repaintSweepHorizontally = YES; } - (id)invokeUndefinedMethodFromWebScript:(NSString *)name withArguments:(NSArray *)args { return nil; } - (void)_addWorkForTarget:(id)target selector:(SEL)selector arg1:(id)arg1 arg2:(id)arg2 { if (workQueueFrozen) return; NSMethodSignature *sig = [target methodSignatureForSelector:selector]; NSInvocation *work = [NSInvocation invocationWithMethodSignature:sig]; [work retainArguments]; [work setTarget:target]; [work setSelector:selector]; if (arg1) { [work setArgument:&arg1 atIndex:2]; if (arg2) [work setArgument:&arg2 atIndex:3]; } [workQueue addObject:work]; } - (void)_doLoad:(NSURL *)url target:(NSString *)target { WebFrame *targetFrame; if (target && ![target isKindOfClass:[WebUndefined class]]) targetFrame = [mainFrame findFrameNamed:target]; else targetFrame = mainFrame; [targetFrame loadRequest:[NSURLRequest requestWithURL:url]]; } - (void)_doBackOrForwardNavigation:(NSNumber *)index { int bfIndex = [index intValue]; if (bfIndex == 1) [[mainFrame webView] goForward]; if (bfIndex == -1) [[mainFrame webView] goBack]; else { WebBackForwardList *bfList = [[mainFrame webView] backForwardList]; [[mainFrame webView] goToBackForwardItem:[bfList itemAtIndex:bfIndex]]; } } - (void)queueBackNavigation:(int)howFarBack { [self _addWorkForTarget:self selector:@selector(_doBackOrForwardNavigation:) arg1:[NSNumber numberWithInt:-howFarBack] arg2:nil]; } - (void)queueForwardNavigation:(int)howFarForward { [self _addWorkForTarget:self selector:@selector(_doBackOrForwardNavigation:) arg1:[NSNumber numberWithInt:howFarForward] arg2:nil]; } - (void)queueReload { [self _addWorkForTarget:[mainFrame webView] selector:@selector(reload:) arg1:self arg2:nil]; } - (void)queueScript:(NSString *)script { [self _addWorkForTarget:[mainFrame webView] selector:@selector(stringByEvaluatingJavaScriptFromString:) arg1:script arg2:nil]; } - (void)queueLoad:(NSString *)URLString target:(NSString *)target { NSURL *URL = [NSURL URLWithString:URLString relativeToURL:[[[mainFrame dataSource] response] URL]]; [self _addWorkForTarget:self selector:@selector(_doLoad:target:) arg1:URL arg2:target]; } - (void)setAcceptsEditing:(BOOL)newAcceptsEditing { [(EditingDelegate *)[[mainFrame webView] editingDelegate] setAcceptsEditing:newAcceptsEditing]; } - (void)setTabKeyCyclesThroughElements:(BOOL)newTabKeyCyclesThroughElements { [[mainFrame webView] setTabKeyCyclesThroughElements:newTabKeyCyclesThroughElements]; } - (void)storeWebScriptObject:(WebScriptObject *)webScriptObject { if (webScriptObject == storedWebScriptObject) return; [storedWebScriptObject release]; storedWebScriptObject = [webScriptObject retain]; } - (void)accessStoredWebScriptObject { JSObjectRef jsObject = [storedWebScriptObject JSObject]; ASSERT(!jsObject); [storedWebScriptObject callWebScriptMethod:@"" withArguments:nil]; [storedWebScriptObject evaluateWebScript:@""]; [storedWebScriptObject setValue:[WebUndefined undefined] forKey:@"key"]; [storedWebScriptObject valueForKey:@"key"]; [storedWebScriptObject removeWebScriptKey:@"key"]; [storedWebScriptObject stringRepresentation]; [storedWebScriptObject webScriptValueAtIndex:0]; [storedWebScriptObject setWebScriptValueAtIndex:0 value:[WebUndefined undefined]]; [storedWebScriptObject setException:@"exception"]; } - (BOOL)testWrapperRoundTripping:(WebScriptObject *)webScriptObject { JSObjectRef jsObject = [webScriptObject JSObject]; if (!jsObject) return false; if (!webScriptObject) return false; if ([[webScriptObject evaluateWebScript:@"({ })"] class] != [webScriptObject class]) return false; [webScriptObject setValue:[NSNumber numberWithInt:666] forKey:@"key"]; if (![[webScriptObject valueForKey:@"key"] isKindOfClass:[NSNumber class]] || ![[webScriptObject valueForKey:@"key"] isEqualToNumber:[NSNumber numberWithInt:666]]) return false; [webScriptObject removeWebScriptKey:@"key"]; @try { if ([webScriptObject valueForKey:@"key"]) return false; } @catch(NSException *exception) { // NSObject throws an exception if the key doesn't exist. } [webScriptObject setWebScriptValueAtIndex:0 value:webScriptObject]; if ([webScriptObject webScriptValueAtIndex:0] != webScriptObject) return false; if ([[webScriptObject stringRepresentation] isEqualToString:@"[Object object]"]) return false; if ([webScriptObject callWebScriptMethod:@"returnThis" withArguments:nil] != webScriptObject) return false; return true; } - (void)dealloc { [storedWebScriptObject release]; [super dealloc]; } - (NSString *)objCClassNameOf:(id)object { if (!object) return @"nil"; return NSStringFromClass([object class]); } - (id)objCObjectOfClass:(NSString *)aClass { if ([aClass isEqualToString:@"NSNull"]) return [NSNull null]; if ([aClass isEqualToString:@"WebUndefined"]) return [WebUndefined undefined]; if ([aClass isEqualToString:@"NSCFBoolean"]) return [NSNumber numberWithBool:true]; if ([aClass isEqualToString:@"NSCFNumber"]) return [NSNumber numberWithInt:1]; if ([aClass isEqualToString:@"NSCFString"]) return @""; if ([aClass isEqualToString:@"WebScriptObject"]) return self; if ([aClass isEqualToString:@"NSArray"]) return [NSArray array]; return nil; } - (BOOL)objCIdentityIsEqual:(WebScriptObject *)a :(WebScriptObject *)b { return a == b; } - (NSString*)decodeHostName:(NSString*)name { return [name _web_decodeHostName]; } - (NSString*)encodeHostName:(NSString*)name { return [name _web_encodeHostName]; } - (long long)objCLongLongRoundTrip:(long long)num { return num; } - (unsigned long long)objCUnsignedLongLongRoundTrip:(unsigned long long)num { return num; } @end static bool shouldLogFrameLoadDelegates(const char *pathOrURL) { return strstr(pathOrURL, "loading/"); } static void runTest(const char *pathOrURL) { CFStringRef pathOrURLString = CFStringCreateWithCString(NULL, pathOrURL, kCFStringEncodingUTF8); if (!pathOrURLString) { fprintf(stderr, "can't parse filename as UTF-8\n"); return; } CFURLRef URL; if (CFStringHasPrefix(pathOrURLString, CFSTR("http://")) || CFStringHasPrefix(pathOrURLString, CFSTR("https://"))) URL = CFURLCreateWithString(NULL, pathOrURLString, NULL); else URL = CFURLCreateWithFileSystemPath(NULL, pathOrURLString, kCFURLPOSIXPathStyle, FALSE); if (!URL) { CFRelease(pathOrURLString); fprintf(stderr, "can't turn %s into a CFURL\n", pathOrURL); return; } [(EditingDelegate *)[[mainFrame webView] editingDelegate] setAcceptsEditing:YES]; [[mainFrame webView] makeTextStandardSize:nil]; [[mainFrame webView] setTabKeyCyclesThroughElements: YES]; [[mainFrame webView] setPolicyDelegate:nil]; [WebView _setUsesTestModeFocusRingColor:YES]; done = NO; topLoadingFrame = nil; waitToDump = NO; dumpAsText = NO; dumpDOMAsWebArchive = NO; dumpSourceAsWebArchive = NO; dumpChildFrameScrollPositions = NO; dumpChildFramesAsText = NO; shouldDumpEditingCallbacks = NO; shouldDumpResourceLoadCallbacks = NO; shouldDumpFrameLoadCallbacks = NO; dumpSelectionRect = NO; dumpTitleChanges = NO; dumpBackForwardList = NO; readFromWindow = NO; canOpenWindows = NO; closeWebViews = YES; addFileToPasteboardOnDrag = NO; [[mainFrame webView] _setDashboardBehavior:WebDashboardBehaviorUseBackwardCompatibilityMode to:NO]; testRepaint = testRepaintDefault; repaintSweepHorizontally = repaintSweepHorizontallyDefault; if ([WebHistory optionalSharedHistory]) [WebHistory setOptionalSharedHistory:nil]; lastMousePosition = NSMakePoint(0, 0); [disallowedURLs removeAllObjects]; if (currentTest != nil) CFRelease(currentTest); currentTest = (NSString *)pathOrURLString; [prevTestBFItem release]; prevTestBFItem = [[[[mainFrame webView] backForwardList] currentItem] retain]; [workQueue removeAllObjects]; workQueueFrozen = NO; BOOL _shouldIgnoreWebCoreNodeLeaks = shouldIgnoreWebCoreNodeLeaks(CFURLGetString(URL)); if (_shouldIgnoreWebCoreNodeLeaks) [WebCoreStatistics startIgnoringWebCoreNodeLeaks]; NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init]; if (shouldLogFrameLoadDelegates(pathOrURL)) shouldDumpFrameLoadCallbacks = YES; [mainFrame loadRequest:[NSURLRequest requestWithURL:(NSURL *)URL]]; CFRelease(URL); [pool release]; while (!done) { pool = [[NSAutoreleasePool alloc] init]; [[NSRunLoop currentRunLoop] runMode:NSDefaultRunLoopMode beforeDate:[NSDate distantPast]]; [pool release]; } pool = [[NSAutoreleasePool alloc] init]; [EventSendingController clearSavedEvents]; [[mainFrame webView] setSelectedDOMRange:nil affinity:NSSelectionAffinityDownstream]; if (closeRemainingWindowsWhenComplete) { NSArray* array = [(NSArray *)allWindowsRef copy]; unsigned count = [array count]; for (unsigned i = 0; i < count; i++) { NSWindow *window = [array objectAtIndex:i]; // Don't try to close the main window if (window == [[mainFrame webView] window]) continue; WebView *webView = [[[window contentView] subviews] objectAtIndex:0]; [webView close]; [window close]; } [array release]; } [pool release]; // We should only have our main window left when we're done ASSERT(CFArrayGetCount(allWindowsRef) == 1); ASSERT(CFArrayGetValueAtIndex(allWindowsRef, 0) == [[mainFrame webView] window]); if (_shouldIgnoreWebCoreNodeLeaks) [WebCoreStatistics stopIgnoringWebCoreNodeLeaks]; } /* Hashes a bitmap and returns a text string for comparison and saving to a file */ static NSString *md5HashStringForBitmap(CGImageRef bitmap) { MD5_CTX md5Context; unsigned char hash[16]; unsigned bitsPerPixel = CGImageGetBitsPerPixel(bitmap); assert(bitsPerPixel == 32); // ImageDiff assumes 32 bit RGBA, we must as well. unsigned bytesPerPixel = bitsPerPixel / 8; unsigned pixelsHigh = CGImageGetHeight(bitmap); unsigned pixelsWide = CGImageGetWidth(bitmap); unsigned bytesPerRow = CGImageGetBytesPerRow(bitmap); assert(bytesPerRow >= (pixelsWide * bytesPerPixel)); MD5_Init(&md5Context); unsigned char *bitmapData = screenCaptureBuffer; for (unsigned row = 0; row < pixelsHigh; row++) { MD5_Update(&md5Context, bitmapData, pixelsWide * bytesPerPixel); bitmapData += bytesPerRow; } MD5_Final(hash, &md5Context); char hex[33] = ""; for (int i = 0; i < 16; i++) { snprintf(hex, 33, "%s%02x", hex, hash[i]); } return [NSString stringWithUTF8String:hex]; } static void displayWebView() { NSView *webView = [mainFrame webView]; [webView display]; [webView lockFocus]; [[[NSColor blackColor] colorWithAlphaComponent:0.66] set]; NSRectFillUsingOperation([webView frame], NSCompositeSourceOver); [webView unlockFocus]; readFromWindow = YES; } @implementation DumpRenderTreePasteboard // Return a local pasteboard so we don't disturb the real pasteboards when running tests. + (NSPasteboard *)_pasteboardWithName:(NSString *)name { static int number = 0; if (!name) name = [NSString stringWithFormat:@"LocalPasteboard%d", ++number]; LocalPasteboard *pasteboard = [localPasteboards objectForKey:name]; if (pasteboard) return pasteboard; pasteboard = [[LocalPasteboard alloc] init]; [localPasteboards setObject:pasteboard forKey:name]; [pasteboard release]; return pasteboard; } // Convenience method for JS so that it doesn't have to try and create a NSArray on the objc side instead // of the usual WebScriptObject that is passed around - (int)declareType:(NSString *)type owner:(id)newOwner { return [self declareTypes:[NSArray arrayWithObject:type] owner:newOwner]; } @end @implementation LocalPasteboard + (id)alloc { return NSAllocateObject(self, 0, 0); } - (id)init { typesArray = [[NSMutableArray alloc] init]; typesSet = [[NSMutableSet alloc] init]; dataByType = [[NSMutableDictionary alloc] init]; return self; } - (void)dealloc { [typesArray release]; [typesSet release]; [dataByType release]; [super dealloc]; } - (NSString *)name { return nil; } - (void)releaseGlobally { } - (int)declareTypes:(NSArray *)newTypes owner:(id)newOwner { [typesArray removeAllObjects]; [typesSet removeAllObjects]; [dataByType removeAllObjects]; return [self addTypes:newTypes owner:newOwner]; } - (int)addTypes:(NSArray *)newTypes owner:(id)newOwner { unsigned count = [newTypes count]; unsigned i; for (i = 0; i < count; ++i) { NSString *type = [newTypes objectAtIndex:i]; NSString *setType = [typesSet member:type]; if (!setType) { setType = [type copy]; [typesArray addObject:setType]; [typesSet addObject:setType]; [setType release]; } if (newOwner && [newOwner respondsToSelector:@selector(pasteboard:provideDataForType:)]) [newOwner pasteboard:self provideDataForType:setType]; } return ++changeCount; } - (int)changeCount { return changeCount; } - (NSArray *)types { return typesArray; } - (NSString *)availableTypeFromArray:(NSArray *)types { unsigned count = [types count]; unsigned i; for (i = 0; i < count; ++i) { NSString *type = [types objectAtIndex:i]; NSString *setType = [typesSet member:type]; if (setType) return setType; } return nil; } - (BOOL)setData:(NSData *)data forType:(NSString *)dataType { if (data == nil) data = [NSData data]; if (![typesSet containsObject:dataType]) return NO; [dataByType setObject:data forKey:dataType]; ++changeCount; return YES; } - (NSData *)dataForType:(NSString *)dataType { return [dataByType objectForKey:dataType]; } - (BOOL)setPropertyList:(id)propertyList forType:(NSString *)dataType; { CFDataRef data = NULL; if (propertyList) data = CFPropertyListCreateXMLData(NULL, propertyList); BOOL result = [self setData:(NSData *)data forType:dataType]; if (data) CFRelease(data); return result; } - (BOOL)setString:(NSString *)string forType:(NSString *)dataType { CFDataRef data = NULL; if (string) { if ([string length] == 0) data = CFDataCreate(NULL, NULL, 0); else data = CFStringCreateExternalRepresentation(NULL, (CFStringRef)string, kCFStringEncodingUTF8, 0); } BOOL result = [self setData:(NSData *)data forType:dataType]; if (data) CFRelease(data); return result; } @end static CFArrayCallBacks NonRetainingArrayCallbacks = { 0, NULL, NULL, CFCopyDescription, CFEqual }; @implementation DumpRenderTreeWindow - (id)initWithContentRect:(NSRect)contentRect styleMask:(unsigned int)styleMask backing:(NSBackingStoreType)bufferingType defer:(BOOL)deferCreation { if (!allWindowsRef) allWindowsRef = CFArrayCreateMutable(NULL, 0, &NonRetainingArrayCallbacks); CFArrayAppendValue(allWindowsRef, self); return [super initWithContentRect:contentRect styleMask:styleMask backing:bufferingType defer:deferCreation]; } - (void)dealloc { CFRange arrayRange = CFRangeMake(0, CFArrayGetCount(allWindowsRef)); CFIndex i = CFArrayGetFirstIndexOfValue(allWindowsRef, arrayRange, self); assert(i != -1); CFArrayRemoveValueAtIndex(allWindowsRef, i); [super dealloc]; } - (BOOL)isKeyWindow { return windowIsKey; } - (void)keyDown:(id)sender { // Do nothing, avoiding the beep we'd otherwise get from NSResponder, // once we get to the end of the responder chain. } @end @implementation DumpRenderTreeEvent + (NSPoint)mouseLocation { return [[[mainFrame webView] window] convertBaseToScreen:lastMousePosition]; } @end