Fix for http://bugzilla.opendarwin.org/show_bug.cgi?id=7223
[WebKit-https.git] / WebKitTools / DumpRenderTree / DumpRenderTree.m
1 /*
2  * Copyright (C) 2005, 2006 Apple Computer, 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 Computer, 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 "DumpRenderTree.h"
30
31 #import <WebKit/DOMExtensions.h>
32 #import <WebKit/DOMRange.h>
33 #import <WebKit/WebBackForwardList.h>
34 #import <WebKit/WebCoreStatistics.h>
35 #import <WebKit/WebDataSource.h>
36 #import <WebKit/WebEditingDelegate.h>
37 #import <WebKit/WebFrameView.h>
38 #import <WebKit/WebPreferences.h>
39 #import <WebKit/WebView.h>
40 #import <WebKit/WebHTMLViewPrivate.h>
41 #import <WebKit/WebPluginDatabase.h>
42
43 #import <ApplicationServices/ApplicationServices.h> // for CMSetDefaultProfileBySpace
44 #import <objc/objc-runtime.h>                       // for class_poseAs
45
46 #define COMMON_DIGEST_FOR_OPENSSL
47 #import <CommonCrypto/CommonDigest.h>               // for MD5 functions
48
49 #import <getopt.h>
50 #import <malloc/malloc.h>
51
52 #import "TextInputController.h"
53 #import "NavigationController.h"
54 #import "AppleScriptController.h"
55 #import "EventSendingController.h"
56 #import "EditingDelegate.h"
57
58 @interface DumpRenderTreeWindow : NSWindow
59 @end
60
61 @interface DumpRenderTreePasteboard : NSPasteboard
62 @end
63
64 @interface WaitUntilDoneDelegate : NSObject
65 @end
66
67 @interface LayoutTestController : NSObject
68 @end
69
70 static void dumpRenderTree(const char *pathOrURL);
71 static NSString *md5HashStringForBitmap(CGImageRef bitmap);
72
73 WebFrame *frame = 0;
74 DumpRenderTreeDraggingInfo *draggingInfo = 0;
75
76 static volatile BOOL done;
77 static NavigationController *navigationController;
78 static BOOL readyToDump;
79 static BOOL waitToDump;
80 static BOOL dumpAsText;
81 static BOOL dumpTitleChanges;
82 static int dumpPixels = NO;
83 static int dumpTree = YES;
84 static BOOL printSeparators;
85 static NSString *currentTest = nil;
86 static NSPasteboard *localPasteboard;
87 static BOOL windowIsKey = YES;
88 static unsigned char* screenCaptureBuffer;
89 static CGColorSpaceRef sharedColorSpace;
90
91 const unsigned maxViewHeight = 600;
92 const unsigned maxViewWidth = 800;
93
94 BOOL doneLoading()
95 {
96     return done;
97 }
98
99 static CMProfileRef currentColorProfile = 0;
100 static void restoreColorSpace(int ignored)
101 {
102     if (currentColorProfile) {
103         int error = CMSetDefaultProfileByUse(cmDisplayUse, currentColorProfile);
104         if (error)
105             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);
106         currentColorProfile = 0;
107     }
108 }
109
110 static void setDefaultColorProfileToRGB(void)
111 {
112     CMProfileRef genericProfile = [[NSColorSpace genericRGBColorSpace] colorSyncProfile];
113     CMProfileRef previousProfile;
114     int error = CMGetDefaultProfileByUse(cmDisplayUse, &previousProfile);
115     if (error) {
116         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);
117         return;
118     }
119     if (previousProfile == genericProfile)
120         return;
121     CFStringRef previousProfileName;
122     CFStringRef genericProfileName;
123     CMCopyProfileDescriptionString(previousProfile, &previousProfileName);
124     CMCopyProfileDescriptionString(genericProfile, &genericProfileName);
125     
126     fprintf(stderr, "\n\nWARNING: Temporarily changing your system color profile from \"%s\" to \"%s\".\n",
127         CFStringGetCStringPtr(previousProfileName, kCFStringEncodingMacRoman),
128         CFStringGetCStringPtr(genericProfileName, kCFStringEncodingMacRoman));
129     fprintf(stderr, "This allows the WebKit pixel-based regression tests to have consistent color values across all machines.\n");
130     fprintf(stderr, "The colors on your screen will change for the duration of the testing.\n\n");
131     
132     if ((error = CMSetDefaultProfileByUse(cmDisplayUse, genericProfile)))
133         fprintf(stderr, "Failed to set color profile to \"%s\"! Many pixel tests will fail as a result.  (Error: %i)",
134             CFStringGetCStringPtr(genericProfileName, kCFStringEncodingMacRoman), error);
135     else {
136         currentColorProfile = previousProfile;
137         signal(SIGINT, restoreColorSpace);
138         signal(SIGHUP, restoreColorSpace);
139         signal(SIGTERM, restoreColorSpace);
140     }
141     CFRelease(genericProfileName);
142     CFRelease(previousProfileName);
143 }
144
145 static void* (*savedMalloc)(malloc_zone_t*, size_t);
146 static void* (*savedRealloc)(malloc_zone_t*, void*, size_t);
147
148 static void* checkedMalloc(malloc_zone_t* zone, size_t size)
149 {
150     if (size >= 0x10000000)
151         return 0;
152     return savedMalloc(zone, size);
153 }
154
155 static void* checkedRealloc(malloc_zone_t* zone, void* ptr, size_t size)
156 {
157     if (size >= 0x10000000)
158         return 0;
159     return savedRealloc(zone, ptr, size);
160 }
161
162 static void makeLargeMallocFailSilently(void)
163 {
164     malloc_zone_t* zone = malloc_default_zone();
165     savedMalloc = zone->malloc;
166     savedRealloc = zone->realloc;
167     zone->malloc = checkedMalloc;
168     zone->realloc = checkedRealloc;
169 }
170
171 int main(int argc, const char *argv[])
172 {
173     NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
174     
175     [NSApplication sharedApplication];
176
177     class_poseAs(objc_getClass("DumpRenderTreePasteboard"), objc_getClass("NSPasteboard"));
178     class_poseAs(objc_getClass("DumpRenderTreeWindow"), objc_getClass("NSWindow"));
179     
180     struct option options[] = {
181         {"pixel-tests", no_argument, &dumpPixels, YES},
182         {"tree", no_argument, &dumpTree, YES},
183         {"notree", no_argument, &dumpTree, NO},
184         {NULL, 0, NULL, 0}
185     };
186
187     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
188     [defaults setObject:@"DoubleMax" forKey:@"AppleScrollBarVariant"];
189     [defaults setInteger:4 forKey:@"AppleAntiAliasingThreshold"];
190     // 2 is the "Medium" font smoothing mode
191     [defaults setInteger:2 forKey:@"AppleFontSmoothing"];
192
193     [defaults setInteger:1 forKey:@"AppleAquaColorVariant"];
194     [defaults setObject:@"0.709800 0.835300 1.000000" forKey:@"AppleHighlightColor"];
195     [defaults setObject:@"0.500000 0.500000 0.500000" forKey:@"AppleOtherHighlightColor"];
196     
197     [defaults setObject:[NSArray arrayWithObject:@"en"] forKey:@"AppleLanguages"];
198     
199     WebPreferences *preferences = [WebPreferences standardPreferences];
200     
201     [preferences setStandardFontFamily:@"Times"];
202     [preferences setFixedFontFamily:@"Courier"];
203     [preferences setSerifFontFamily:@"Times"];
204     [preferences setSansSerifFontFamily:@"Helvetica"];
205     [preferences setCursiveFontFamily:@"Apple Chancery"];
206     [preferences setFantasyFontFamily:@"Papyrus"];
207     [preferences setDefaultFontSize:16];
208     [preferences setDefaultFixedFontSize:13];
209     [preferences setMinimumFontSize:9];
210     [preferences setJavaEnabled:NO];
211     [preferences setJavaScriptCanOpenWindowsAutomatically:NO];
212
213     int option;
214     while ((option = getopt_long(argc, (char * const *)argv, "", options, NULL)) != -1)
215         switch (option) {
216             case '?':   // unknown or ambiguous option
217             case ':':   // missing argument
218                 exit(1);
219                 break;
220         }
221     
222     if ([[[NSFontManager sharedFontManager] availableMembersOfFontFamily:@"Ahem"] count] == 0) {
223         fprintf(stderr, "\nAhem font is not available. This special simple font is used to construct certain types of predictable tests.\n\nTo run regression tests, please get it from <http://webkit.opendarwin.org/quality/Ahem.ttf>.\n");
224         exit(1);
225     }
226     
227     if (dumpPixels) {
228         setDefaultColorProfileToRGB();
229         screenCaptureBuffer = malloc(maxViewHeight * maxViewWidth * 4);
230         sharedColorSpace = CGColorSpaceCreateDeviceRGB();
231     }
232     
233     localPasteboard = [NSPasteboard pasteboardWithUniqueName];
234     navigationController = [[NavigationController alloc] init];
235
236     NSRect rect = NSMakeRect(0, 0, maxViewWidth, maxViewHeight);
237     
238     WebView *webView = [[WebView alloc] initWithFrame:rect];
239     WaitUntilDoneDelegate *delegate = [[WaitUntilDoneDelegate alloc] init];
240     EditingDelegate *editingDelegate = [[EditingDelegate alloc] init];
241     [webView setFrameLoadDelegate:delegate];
242     [webView setEditingDelegate:editingDelegate];
243     [webView setUIDelegate:delegate];
244     frame = [webView mainFrame];
245     
246     [[webView preferences] setTabsToLinks:YES];
247     
248     NSString *pwd = [[NSString stringWithUTF8String:argv[0]] stringByDeletingLastPathComponent];
249     [WebPluginDatabase setAdditionalWebPlugInPaths:[NSArray arrayWithObject:pwd]];
250     [[WebPluginDatabase installedPlugins] refresh];
251
252     // The back/forward cache is causing problems due to layouts during transition from one page to another.
253     // So, turn it off for now, but we might want to turn it back on some day.
254     [[webView backForwardList] setPageCacheSize:0];
255
256     // 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.
257     NSRect windowRect = NSOffsetRect(rect, -10000, -10000);
258     NSWindow *window = [[NSWindow alloc] initWithContentRect:windowRect styleMask:NSBorderlessWindowMask backing:NSBackingStoreBuffered defer:YES];
259     [[window contentView] addSubview:webView];
260     [window orderBack:nil];
261     [window setAutodisplay:NO];
262
263     [webView setContinuousSpellCheckingEnabled:YES];
264
265     makeLargeMallocFailSilently();
266     
267     // For reasons that are not entirely clear, the following pair of calls makes WebView handle its
268     // dynamic scrollbars properly. Without it, every frame will always have scrollbars.
269     NSBitmapImageRep *imageRep = [webView bitmapImageRepForCachingDisplayInRect:[webView bounds]];
270     [webView cacheDisplayInRect:[webView bounds] toBitmapImageRep:imageRep];
271     
272     if (argc == optind+1 && strcmp(argv[optind], "-") == 0) {
273         char filenameBuffer[2048];
274         printSeparators = YES;
275         while (fgets(filenameBuffer, sizeof(filenameBuffer), stdin)) {
276             char *newLineCharacter = strchr(filenameBuffer, '\n');
277             if (newLineCharacter)
278                 *newLineCharacter = '\0';
279             
280             if (strlen(filenameBuffer) == 0)
281                 continue;
282                 
283             dumpRenderTree(filenameBuffer);
284             fflush(stdout);
285         }
286     } else {
287         printSeparators = (optind < argc-1 || (dumpPixels && dumpTree));
288         for (int i = optind; i != argc; ++i) {
289             dumpRenderTree(argv[i]);
290         }
291     }
292     
293     [webView setFrameLoadDelegate:nil];
294     [webView setEditingDelegate:nil];
295     [webView setUIDelegate:nil];
296     frame = nil;
297
298     // Work around problem where registering drag types leaves an outstanding
299     // "perform selector" on the window, which retains the window. It's a bit
300     // inelegant and perhaps dangerous to just blow them all away, but in practice
301     // it probably won't cause any trouble (and this is just a test tool, after all).
302     [NSObject cancelPreviousPerformRequestsWithTarget:window];
303     
304     [window release];
305     [webView release];
306     [delegate release];
307     [editingDelegate release];
308
309     [localPasteboard releaseGlobally];
310     localPasteboard = nil;
311     
312     [navigationController release];
313     navigationController = nil;
314     
315     if (dumpPixels)
316         restoreColorSpace(0);
317     
318     [pool release];
319
320     return 0;
321 }
322
323 static void dump(void)
324 {
325     NSString *result = nil;
326     if (dumpTree) {
327         if (dumpAsText) {
328             DOMElement *documentElement = [[frame DOMDocument] documentElement];
329             if ([documentElement isKindOfClass:[DOMHTMLElement class]])
330                 result = [[(DOMHTMLElement *)documentElement innerText] stringByAppendingString:@"\n"];
331         } else {
332             bool isSVGW3CTest = ([currentTest rangeOfString:@"svg/W3C-SVG-1.1"].length);
333             if (isSVGW3CTest)
334                 [[frame webView] setFrameSize:NSMakeSize(480, 360)];
335             else 
336                 [[frame webView] setFrameSize:NSMakeSize(maxViewWidth, maxViewHeight)];
337             result = [frame renderTreeAsExternalRepresentation];
338         }
339         
340         if (!result)
341             printf("ERROR: nil result from %s", dumpAsText ? "[documentElement innerText]" : "[frame renderTreeAsExternalRepresentation]");
342         else
343             fputs([result UTF8String], stdout);
344         
345         if (printSeparators)
346             puts("#EOF");
347     }
348     
349     if (dumpPixels) {
350         if (!dumpAsText) {
351             // FIXME: It's unfortunate that we hardcode the file naming scheme here.
352             // At one time, the perl script had all the knowledge about file layout.
353             // Some day we should restore that setup by passing in more parameters to this tool.
354
355             NSString *baseTestPath = [currentTest stringByDeletingPathExtension];
356             NSString *baselineHashPath = [baseTestPath stringByAppendingString:@"-expected.checksum"];
357             NSString *baselineHash = [NSString stringWithContentsOfFile:baselineHashPath encoding:NSUTF8StringEncoding error:nil];
358             NSString *baselineImagePath = [baseTestPath stringByAppendingString:@"-expected.png"];
359             
360             // grab a bitmap from the view
361             WebView *view = [frame webView];
362             NSSize webViewSize = [[frame webView] frame].size;
363             CGContextRef cgContext = CGBitmapContextCreate(screenCaptureBuffer, webViewSize.width, webViewSize.height, 8, webViewSize.width * 4, sharedColorSpace, kCGBitmapByteOrder32Host | kCGImageAlphaPremultipliedLast);
364             NSGraphicsContext *nsContext = [NSGraphicsContext graphicsContextWithGraphicsPort:cgContext flipped:NO];
365             [NSGraphicsContext saveGraphicsState];
366             [NSGraphicsContext setCurrentContext:nsContext];
367             [view displayRectIgnoringOpacity:NSMakeRect(0, 0, webViewSize.width, webViewSize.height) inContext:nsContext];
368             [NSGraphicsContext restoreGraphicsState];
369             
370             // has the actual hash to compare to the expected image's hash
371             CGImageRef bitmapImage = CGBitmapContextCreateImage(cgContext);
372             NSString *actualHash = md5HashStringForBitmap(bitmapImage);
373             printf("\nActualHash: %s\n", [actualHash UTF8String]);
374             printf("BaselineHash: %s\n", [baselineHash UTF8String]);
375             
376             // if the hashes don't match, send image back to stdout for diff comparision
377             if (![baselineHash isEqualToString:actualHash] || access([baselineImagePath fileSystemRepresentation], F_OK) != 0) {
378                 CFMutableDataRef imageData = CFDataCreateMutable(0, 0);
379                 CGImageDestinationRef imageDest = CGImageDestinationCreateWithData(imageData, CFSTR("public.png"), 1, 0);
380                 CGImageDestinationAddImage(imageDest, bitmapImage, 0);
381                 CGImageDestinationFinalize(imageDest);
382                 CFRelease(imageDest);
383                 printf("Content-length: %lu\n", CFDataGetLength(imageData));
384                 fwrite(CFDataGetBytePtr(imageData), 1, CFDataGetLength(imageData), stdout);
385                 CFRelease(imageData);
386             }
387             CGImageRelease(bitmapImage);
388             CGContextRelease(cgContext);
389         }
390
391         printf("#EOF\n");
392     }
393
394     done = YES;
395 }
396
397 @implementation WaitUntilDoneDelegate
398
399 - (void)webView:(WebView *)c locationChangeDone:(NSError *)error forDataSource:(WebDataSource *)dataSource
400 {
401     if ([dataSource webFrame] == frame) {
402         if (waitToDump)
403             readyToDump = YES;
404         else
405             dump();
406     }
407 }
408
409 - (void)webView:(WebView *)sender didCommitLoadForFrame:(WebFrame *)f
410 {
411     if (frame == f)
412         readyToDump = NO;
413
414     windowIsKey = YES;
415     NSView *documentView = [[frame frameView] documentView];
416     [[[frame webView] window] makeFirstResponder:documentView];
417     if ([documentView isKindOfClass:[WebHTMLView class]])
418         [(WebHTMLView *)documentView _updateFocusState];
419 }
420
421 - (void)webView:(WebView *)sender didFailProvisionalLoadWithError:(NSError *)error forFrame:(WebFrame *)frame
422 {
423     [self webView:sender locationChangeDone:error forDataSource:[frame provisionalDataSource]];
424 }
425
426 - (void)webView:(WebView *)sender didFinishLoadForFrame:(WebFrame *)frame
427 {
428     [self webView:sender locationChangeDone:nil forDataSource:[frame dataSource]];
429     [navigationController webView:sender didFinishLoadForFrame:frame];
430 }
431
432 - (void)webView:(WebView *)sender didFailLoadWithError:(NSError *)error forFrame:(WebFrame *)frame
433 {
434     [self webView:sender locationChangeDone:error forDataSource:[frame dataSource]];
435 }
436
437 - (void)webView:(WebView *)sender windowScriptObjectAvailable:(WebScriptObject *)obj 
438
439     LayoutTestController *ltc = [[LayoutTestController alloc] init];
440     [obj setValue:ltc forKey:@"layoutTestController"];
441     [ltc release];
442     EventSendingController *esc = [[EventSendingController alloc] init];
443     [obj setValue:esc forKey:@"eventSender"];
444     [esc release];
445     TextInputController *tic = [[TextInputController alloc] initWithWebView:sender];
446     [obj setValue:tic forKey:@"textInputController"];
447     [tic release];
448     AppleScriptController *asc = [[AppleScriptController alloc] initWithWebView:sender];
449     [obj setValue:asc forKey:@"appleScriptController"];
450     [asc release];
451     [obj setValue:navigationController forKey:@"navigationController"];
452 }
453
454 - (void)webView:(WebView *)sender runJavaScriptAlertPanelWithMessage:(NSString *)message
455 {
456     printf("ALERT: %s\n", [message UTF8String]);
457 }
458
459 - (void)webView:(WebView *)sender didReceiveTitle:(NSString *)title forFrame:(WebFrame *)frame
460 {
461     if (dumpTitleChanges)
462         printf("TITLE CHANGED: %s\n", [title UTF8String]);
463 }
464
465 - (void)webView:(WebView *)sender dragImage:(NSImage *)anImage at:(NSPoint)viewLocation offset:(NSSize)initialOffset event:(NSEvent *)event pasteboard:(NSPasteboard *)pboard source:(id)sourceObj slideBack:(BOOL)slideFlag forView:(NSView *)view
466 {
467     // A new drag was started before the old one ended.  Probably shouldn't happen.
468     if (draggingInfo) {
469         [[draggingInfo draggingSource] draggedImage:[draggingInfo draggedImage] endedAt:lastMousePosition operation:NSDragOperationNone];
470         [draggingInfo release];
471     }
472     draggingInfo = [[DumpRenderTreeDraggingInfo alloc] initWithImage:anImage offset:initialOffset pasteboard:pboard source:sourceObj];
473 }
474
475 - (void)webViewFocus:(WebView *)webView
476 {
477     windowIsKey = YES;
478     NSView *documentView = [[frame frameView] documentView];
479     if ([documentView isKindOfClass:[WebHTMLView class]])
480         [(WebHTMLView *)documentView _updateFocusState];
481 }
482
483 @end
484
485 @implementation LayoutTestController
486
487 + (BOOL)isSelectorExcludedFromWebScript:(SEL)aSelector
488 {
489     if (aSelector == @selector(waitUntilDone)
490             || aSelector == @selector(notifyDone)
491             || aSelector == @selector(dumpAsText)
492             || aSelector == @selector(dumpTitleChanges)
493             || aSelector == @selector(setWindowIsKey:)
494             || aSelector == @selector(setMainFrameIsFirstResponder:))
495         return NO;
496     return YES;
497 }
498
499 + (NSString *)webScriptNameForSelector:(SEL)aSelector
500 {
501     if (aSelector == @selector(setWindowIsKey:))
502         return @"setWindowIsKey";
503     if (aSelector == @selector(setMainFrameIsFirstResponder:))
504         return @"setMainFrameIsFirstResponder";
505     return nil;
506 }
507
508 - (void)waitUntilDone 
509 {
510     waitToDump = YES;
511 }
512
513 - (void)notifyDone
514 {
515     if (waitToDump && readyToDump)
516         dump();
517     waitToDump = NO;
518 }
519
520 - (void)dumpAsText
521 {
522     dumpAsText = YES;
523 }
524
525 - (void)dumpTitleChanges
526 {
527     dumpTitleChanges = YES;
528 }
529
530 - (void)setWindowIsKey:(BOOL)flag
531 {
532     windowIsKey = flag;
533     NSView *documentView = [[frame frameView] documentView];
534     if ([documentView isKindOfClass:[WebHTMLView class]])
535         [(WebHTMLView *)documentView _updateFocusState];
536 }
537
538 - (void)setMainFrameIsFirstResponder:(BOOL)flag
539 {
540     NSView *documentView = [[frame frameView] documentView];
541     
542     NSResponder *firstResponder = flag ? documentView : nil;
543     [[[frame webView] window] makeFirstResponder:firstResponder];
544         
545     if ([documentView isKindOfClass:[WebHTMLView class]])
546         [(WebHTMLView *)documentView _updateFocusState];
547 }
548
549 - (id)invokeUndefinedMethodFromWebScript:(NSString *)name withArguments:(NSArray *)args
550 {
551     return nil;
552 }
553
554 @end
555
556 static void dumpRenderTree(const char *pathOrURL)
557 {
558     CFStringRef pathOrURLString = CFStringCreateWithCString(NULL, pathOrURL, kCFStringEncodingUTF8);
559     if (!pathOrURLString) {
560         fprintf(stderr, "can't parse filename as UTF-8\n");
561         return;
562     }
563
564     CFURLRef URL;
565     if (CFStringHasPrefix(pathOrURLString, CFSTR("http://")))
566         URL = CFURLCreateWithString(NULL, pathOrURLString, NULL);
567     else
568         URL = CFURLCreateWithFileSystemPath(NULL, pathOrURLString, kCFURLPOSIXPathStyle, FALSE);
569     
570     if (!URL) {
571         CFRelease(pathOrURLString);
572         fprintf(stderr, "can't turn %s into a CFURL\n", pathOrURL);
573         return;
574     }
575
576     done = NO;
577     readyToDump = NO;
578     waitToDump = NO;
579     dumpAsText = NO;
580     dumpTitleChanges = NO;
581     if (currentTest != nil)
582         CFRelease(currentTest);
583     currentTest = (NSString *)pathOrURLString;
584
585     NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
586     [frame loadRequest:[NSURLRequest requestWithURL:(NSURL *)URL]];
587     CFRelease(URL);
588     [pool release];
589     while (!done) {
590         pool = [[NSAutoreleasePool alloc] init];
591         [[NSRunLoop currentRunLoop] runMode:NSDefaultRunLoopMode beforeDate:[NSDate distantPast]];
592         [pool release];
593     }
594     pool = [[NSAutoreleasePool alloc] init];
595     [[frame webView] setSelectedDOMRange:nil affinity:NSSelectionAffinityDownstream];
596     if (draggingInfo)
597         [draggingInfo release];
598     draggingInfo = nil;
599     [pool release];
600 }
601
602 /* Hashes a bitmap and returns a text string for comparison and saving to a file */
603 static NSString *md5HashStringForBitmap(CGImageRef bitmap)
604 {
605     MD5_CTX md5Context;
606     unsigned char hash[16];
607     
608     unsigned bitsPerPixel = CGImageGetBitsPerPixel(bitmap);
609     assert(bitsPerPixel == 32); // ImageDiff assumes 32 bit RGBA, we must as well.
610     unsigned bytesPerPixel = bitsPerPixel / 8;
611     unsigned pixelsHigh = CGImageGetHeight(bitmap);
612     unsigned pixelsWide = CGImageGetWidth(bitmap);
613     unsigned bytesPerRow = CGImageGetBytesPerRow(bitmap);
614     assert(bytesPerRow >= (pixelsWide * bytesPerPixel));
615     
616     MD5_Init(&md5Context);
617     unsigned char *bitmapData = screenCaptureBuffer;
618     for (unsigned row = 0; row < pixelsHigh; row++) {
619         MD5_Update(&md5Context, bitmapData, pixelsWide * bytesPerPixel);
620         bitmapData += bytesPerRow;
621     }
622     MD5_Final(hash, &md5Context);
623     
624     char hex[33] = "";
625     for (int i = 0; i < 16; i++) {
626        snprintf(hex, 33, "%s%02x", hex, hash[i]);
627     }
628
629     return [NSString stringWithUTF8String:hex];
630 }
631
632 @implementation DumpRenderTreePasteboard
633
634 // Return a local pasteboard so we don't disturb the real pasteboard when running tests.
635 + (NSPasteboard *)generalPasteboard
636 {
637     return localPasteboard;
638 }
639
640 @end
641
642 @implementation DumpRenderTreeWindow
643
644 - (BOOL)isKeyWindow
645 {
646     return windowIsKey;
647 }
648
649 @end