[Chromium] Introduce TestInterfaces to hold all the JavaScript interfaces needed...
[WebKit-https.git] / Tools / DumpRenderTree / chromium / TestShell.cpp
1 /*
2  * Copyright (C) 2012 Google 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 are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32 #include "TestShell.h"
33
34 #include "DRTDevToolsAgent.h"
35 #include "DRTDevToolsClient.h"
36 #include "LayoutTestController.h"
37 #include "MockWebPrerenderingSupport.h"
38 #include "platform/WebArrayBufferView.h"
39 #include "WebCompositor.h"
40 #include "WebDataSource.h"
41 #include "WebDocument.h"
42 #include "WebElement.h"
43 #include "WebFrame.h"
44 #include "WebHistoryItem.h"
45 #include "WebIDBFactory.h"
46 #include "WebTestingSupport.h"
47 #include "platform/WebThread.h"
48 #include "WebKit.h"
49 #include "platform/WebKitPlatformSupport.h"
50 #include "WebPermissions.h"
51 #include "platform/WebPoint.h"
52 #include "WebRuntimeFeatures.h"
53 #include "WebScriptController.h"
54 #include "WebSettings.h"
55 #include "platform/WebSize.h"
56 #include "platform/WebString.h"
57 #include "platform/WebURLRequest.h"
58 #include "platform/WebURLResponse.h"
59 #include "WebView.h"
60 #include "WebViewHost.h"
61 #include "skia/ext/platform_canvas.h"
62 #include "webkit/support/webkit_support.h"
63 #include "webkit/support/webkit_support_gfx.h"
64 #include <algorithm>
65 #include <cctype>
66 #include <vector>
67 #include <wtf/MD5.h>
68 #include <wtf/OwnArrayPtr.h>
69
70
71 using namespace WebKit;
72 using namespace std;
73
74 // Content area size for newly created windows.
75 static const int testWindowWidth = 800;
76 static const int testWindowHeight = 600;
77
78 // The W3C SVG layout tests use a different size than the other layout tests.
79 static const int SVGTestWindowWidth = 480;
80 static const int SVGTestWindowHeight = 360;
81
82 static const char layoutTestsPattern[] = "/LayoutTests/";
83 static const string::size_type layoutTestsPatternSize = sizeof(layoutTestsPattern) - 1;
84 static const char fileUrlPattern[] = "file:/";
85 static const char fileTestPrefix[] = "(file test):";
86 static const char dataUrlPattern[] = "data:";
87 static const string::size_type dataUrlPatternSize = sizeof(dataUrlPattern) - 1;
88
89 // FIXME: Move this to a common place so that it can be shared with
90 // WebCore::TransparencyWin::makeLayerOpaque().
91 static void makeCanvasOpaque(SkCanvas* canvas)
92 {
93     const SkBitmap& bitmap = canvas->getTopDevice()->accessBitmap(true);
94     ASSERT(bitmap.config() == SkBitmap::kARGB_8888_Config);
95
96     SkAutoLockPixels lock(bitmap);
97     for (int y = 0; y < bitmap.height(); y++) {
98         uint32_t* row = bitmap.getAddr32(0, y);
99         for (int x = 0; x < bitmap.width(); x++)
100             row[x] |= 0xFF000000; // Set alpha bits to 1.
101     }
102 }
103
104 TestShell::TestShell()
105     : m_testIsPending(false)
106     , m_testIsPreparing(false)
107     , m_focusedWidget(0)
108     , m_testShellMode(false)
109     , m_devTools(0)
110     , m_allowExternalPages(false)
111     , m_acceleratedCompositingForVideoEnabled(false)
112     , m_threadedCompositingEnabled(false)
113     , m_forceCompositingMode(false)
114     , m_accelerated2dCanvasEnabled(false)
115     , m_deferred2dCanvasEnabled(false)
116     , m_acceleratedPaintingEnabled(false)
117     , m_stressOpt(false)
118     , m_stressDeopt(false)
119     , m_dumpWhenFinished(true)
120     , m_isDisplayingModalDialog(false)
121 {
122     WebRuntimeFeatures::enableDataTransferItems(true);
123     WebRuntimeFeatures::enableGeolocation(true);
124     WebRuntimeFeatures::enablePointerLock(true);
125     WebRuntimeFeatures::enableIndexedDatabase(true);
126     WebRuntimeFeatures::enableFileSystem(true);
127     WebRuntimeFeatures::enableJavaScriptI18NAPI(true);
128     WebRuntimeFeatures::enableMediaSource(true);
129     WebRuntimeFeatures::enableEncryptedMedia(true);
130     WebRuntimeFeatures::enableMediaStream(true);
131     WebRuntimeFeatures::enablePeerConnection(true);
132     WebRuntimeFeatures::enableWebAudio(true);
133     WebRuntimeFeatures::enableVideoTrack(true);
134     WebRuntimeFeatures::enableGamepad(true);
135     WebRuntimeFeatures::enableShadowDOM(true);
136     WebRuntimeFeatures::enableStyleScoped(true);
137     WebRuntimeFeatures::enableScriptedSpeech(true);
138
139     // 30 second is the same as the value in Mac DRT.
140     // If we use a value smaller than the timeout value of
141     // (new-)run-webkit-tests, (new-)run-webkit-tests misunderstands that a
142     // timed-out DRT process was crashed.
143     m_timeout = 30 * 1000;
144 }
145
146 void TestShell::initialize()
147 {
148     m_webPermissions = adoptPtr(new WebPermissions(this));
149     m_accessibilityController = adoptPtr(new AccessibilityController(this));
150     m_testInterfaces = adoptPtr(new TestInterfaces());
151     m_layoutTestController = adoptPtr(new LayoutTestController(this));
152     m_eventSender = adoptPtr(new EventSender(this));
153     m_textInputController = adoptPtr(new TextInputController(this));
154 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
155     m_notificationPresenter = adoptPtr(new NotificationPresenter(this));
156 #endif
157     m_printer = m_testShellMode ? TestEventPrinter::createTestShellPrinter() : TestEventPrinter::createDRTPrinter();
158 #if ENABLE(LINK_PRERENDER)
159     m_prerenderingSupport = adoptPtr(new MockWebPrerenderingSupport());
160 #endif
161
162     WTF::initializeThreading();
163
164     if (m_threadedCompositingEnabled) {
165         m_webCompositorThread = adoptPtr(WebKit::webKitPlatformSupport()->createThread("Compositor"));
166         WebCompositor::initialize(m_webCompositorThread.get());
167     } else
168         WebCompositor::initialize(0);
169
170     createMainWindow();
171 }
172
173 void TestShell::createMainWindow()
174 {
175     m_drtDevToolsAgent = adoptPtr(new DRTDevToolsAgent);
176     m_webViewHost = adoptPtr(createNewWindow(WebURL(), m_drtDevToolsAgent.get()));
177     m_webView = m_webViewHost->webView();
178     m_drtDevToolsAgent->setWebView(m_webView);
179 }
180
181 TestShell::~TestShell()
182 {
183     // Note: DevTools are closed together with all the other windows in the
184     // windows list.
185
186     // Destroy the WebView before its WebViewHost.
187     m_drtDevToolsAgent->setWebView(0);
188 }
189
190 void TestShell::createDRTDevToolsClient(DRTDevToolsAgent* agent)
191 {
192     m_drtDevToolsClient = adoptPtr(new DRTDevToolsClient(agent, m_devTools->webView()));
193 }
194
195 void TestShell::showDevTools()
196 {
197     if (!m_devTools) {
198         WebURL url = webkit_support::GetDevToolsPathAsURL();
199         if (!url.isValid()) {
200             ASSERT(false);
201             return;
202         }
203         m_devTools = createNewWindow(url);
204         m_devTools->webView()->settings()->setMemoryInfoEnabled(true);
205         m_devTools->setLogConsoleOutput(false);
206         ASSERT(m_devTools);
207         createDRTDevToolsClient(m_drtDevToolsAgent.get());
208     }
209     m_devTools->show(WebKit::WebNavigationPolicyNewWindow);
210 }
211
212 void TestShell::closeDevTools()
213 {
214     if (m_devTools) {
215         m_devTools->webView()->settings()->setMemoryInfoEnabled(false);
216         m_drtDevToolsAgent->reset();
217         m_drtDevToolsClient.clear();
218         closeWindow(m_devTools);
219         m_devTools = 0;
220     }
221 }
222
223 void TestShell::resetWebSettings(WebView& webView)
224 {
225     m_prefs.reset();
226     m_prefs.acceleratedCompositingEnabled = true;
227     m_prefs.acceleratedCompositingForVideoEnabled = m_acceleratedCompositingForVideoEnabled;
228     m_prefs.forceCompositingMode = m_forceCompositingMode;
229     m_prefs.accelerated2dCanvasEnabled = m_accelerated2dCanvasEnabled;
230     m_prefs.deferred2dCanvasEnabled = m_deferred2dCanvasEnabled;
231     m_prefs.acceleratedPaintingEnabled = m_acceleratedPaintingEnabled;
232     m_prefs.applyTo(&webView);
233 }
234
235 void TestShell::runFileTest(const TestParams& params)
236 {
237     ASSERT(params.testUrl.isValid());
238     m_testIsPreparing = true;
239     m_params = params;
240     string testUrl = m_params.testUrl.spec();
241
242     if (testUrl.find("loading/") != string::npos
243         || testUrl.find("loading\\") != string::npos)
244         m_layoutTestController->setShouldDumpFrameLoadCallbacks(true);
245
246     if (testUrl.find("compositing/") != string::npos || testUrl.find("compositing\\") != string::npos) {
247         m_prefs.acceleratedCompositingForVideoEnabled = true;
248         m_prefs.accelerated2dCanvasEnabled = true;
249         m_prefs.deferred2dCanvasEnabled = true;
250         m_prefs.mockScrollbarsEnabled = true;
251         m_prefs.applyTo(m_webView);
252     }
253
254     if (testUrl.find("/dumpAsText/") != string::npos
255         || testUrl.find("\\dumpAsText\\") != string::npos) {
256         m_layoutTestController->setShouldDumpAsText(true);
257         m_layoutTestController->setShouldGeneratePixelResults(false);
258     }
259
260     if (testUrl.find("/inspector/") != string::npos
261         || testUrl.find("\\inspector\\") != string::npos)
262         showDevTools();
263
264     if (m_params.debugLayerTree)
265         m_layoutTestController->setShowDebugLayerTree(true);
266
267     if (m_dumpWhenFinished)
268         m_printer->handleTestHeader(testUrl.c_str());
269     loadURL(m_params.testUrl);
270
271     m_testIsPreparing = false;
272     waitTestFinished();
273 }
274
275 static inline bool isSVGTestURL(const WebURL& url)
276 {
277     return url.isValid() && string(url.spec()).find("W3C-SVG-1.1") != string::npos;
278 }
279
280 void TestShell::resizeWindowForTest(WebViewHost* window, const WebURL& url)
281 {
282     int width, height;
283     if (isSVGTestURL(url)) {
284         width = SVGTestWindowWidth;
285         height = SVGTestWindowHeight;
286     } else {
287         width = testWindowWidth;
288         height = testWindowHeight;
289     }
290     window->setWindowRect(WebRect(0, 0, width + virtualWindowBorder * 2, height + virtualWindowBorder * 2));
291 }
292
293 void TestShell::resetTestController()
294 {
295     resetWebSettings(*webView());
296     m_webPermissions->reset();
297     m_accessibilityController->reset();
298     m_testInterfaces->resetAll();
299     m_layoutTestController->reset();
300     m_eventSender->reset();
301     m_webViewHost->reset();
302 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
303     m_notificationPresenter->reset();
304 #endif
305 #if OS(ANDROID)
306     webkit_support::ReleaseMediaResources();
307 #endif
308     m_drtDevToolsAgent->reset();
309     if (m_drtDevToolsClient)
310         m_drtDevToolsClient->reset();
311     webView()->setPageScaleFactor(1, WebPoint(0, 0));
312     webView()->enableFixedLayoutMode(false);
313     webView()->setFixedLayoutSize(WebSize(0, 0));
314     webView()->mainFrame()->clearOpener();
315     WebTestingSupport::resetInternalsObject(webView()->mainFrame());
316 }
317
318 void TestShell::loadURL(const WebURL& url)
319 {
320     m_webViewHost->loadURLForFrame(url, WebString());
321 }
322
323 void TestShell::reload()
324 {
325     m_webViewHost->navigationController()->reload();
326 }
327
328 void TestShell::goToOffset(int offset)
329 {
330      m_webViewHost->navigationController()->goToOffset(offset);
331 }
332
333 int TestShell::navigationEntryCount() const
334 {
335     return m_webViewHost->navigationController()->entryCount();
336 }
337
338 void TestShell::callJSGC()
339 {
340     m_webView->mainFrame()->collectGarbage();
341 }
342
343 void TestShell::setFocus(WebWidget* widget, bool enable)
344 {
345     // Simulate the effects of InteractiveSetFocus(), which includes calling
346     // both setFocus() and setIsActive().
347     if (enable) {
348         if (m_focusedWidget != widget) {
349             if (m_focusedWidget)
350                 m_focusedWidget->setFocus(false);
351             webView()->setIsActive(enable);
352             widget->setFocus(enable);
353             m_focusedWidget = widget;
354         }
355     } else {
356         if (m_focusedWidget == widget) {
357             widget->setFocus(enable);
358             webView()->setIsActive(enable);
359             m_focusedWidget = 0;
360         }
361     }
362 }
363
364 void TestShell::testFinished()
365 {
366     if (!m_testIsPending)
367         return;
368     m_testIsPending = false;
369     if (m_dumpWhenFinished)
370         dump();
371     webkit_support::QuitMessageLoop();
372 }
373
374 void TestShell::testTimedOut()
375 {
376     m_printer->handleTimedOut();
377     testFinished();
378 }
379
380 void TestShell::setPerTilePaintingEnabled(bool enabled)
381 {
382     WebCompositor::setPerTilePaintingEnabled(enabled);
383 }
384
385 static string dumpDocumentText(WebFrame* frame)
386 {
387     // We use the document element's text instead of the body text here because
388     // not all documents have a body, such as XML documents.
389     WebElement documentElement = frame->document().documentElement();
390     if (documentElement.isNull())
391         return string();
392     return documentElement.innerText().utf8();
393 }
394
395 static string dumpFramesAsText(WebFrame* frame, bool recursive)
396 {
397     string result;
398
399     // Add header for all but the main frame. Skip empty frames.
400     if (frame->parent() && !frame->document().documentElement().isNull()) {
401         result.append("\n--------\nFrame: '");
402         result.append(frame->name().utf8().data());
403         result.append("'\n--------\n");
404     }
405
406     result.append(dumpDocumentText(frame));
407     result.append("\n");
408
409     if (recursive) {
410         for (WebFrame* child = frame->firstChild(); child; child = child->nextSibling())
411             result.append(dumpFramesAsText(child, recursive));
412     }
413
414     return result;
415 }
416
417 static string dumpFramesAsPrintedText(WebFrame* frame, bool recursive)
418 {
419     string result;
420
421     // Cannot do printed format for anything other than HTML
422     if (!frame->document().isHTMLDocument())
423         return string();
424
425     // Add header for all but the main frame. Skip empty frames.
426     if (frame->parent() && !frame->document().documentElement().isNull()) {
427         result.append("\n--------\nFrame: '");
428         result.append(frame->name().utf8().data());
429         result.append("'\n--------\n");
430     }
431
432     result.append(frame->renderTreeAsText(WebFrame::RenderAsTextPrinting).utf8());
433     result.append("\n");
434
435     if (recursive) {
436         for (WebFrame* child = frame->firstChild(); child; child = child->nextSibling())
437             result.append(dumpFramesAsPrintedText(child, recursive));
438     }
439
440     return result;
441 }
442
443 static void dumpFrameScrollPosition(WebFrame* frame, bool recursive)
444 {
445     WebSize offset = frame->scrollOffset();
446     if (offset.width > 0 || offset.height > 0) {
447         if (frame->parent())
448             printf("frame '%s' ", frame->name().utf8().data());
449         printf("scrolled to %d,%d\n", offset.width, offset.height);
450     }
451
452     if (!recursive)
453         return;
454     for (WebFrame* child = frame->firstChild(); child; child = child->nextSibling())
455         dumpFrameScrollPosition(child, recursive);
456 }
457
458 struct ToLower {
459     char16 operator()(char16 c) { return tolower(c); }
460 };
461
462 // FIXME: Eliminate std::transform(), std::vector, and std::sort().
463
464 // Returns True if item1 < item2.
465 static bool HistoryItemCompareLess(const WebHistoryItem& item1, const WebHistoryItem& item2)
466 {
467     string16 target1 = item1.target();
468     string16 target2 = item2.target();
469     std::transform(target1.begin(), target1.end(), target1.begin(), ToLower());
470     std::transform(target2.begin(), target2.end(), target2.begin(), ToLower());
471     return target1 < target2;
472 }
473
474 static string normalizeLayoutTestURLInternal(const string& url)
475 {
476     string result = url;
477     size_t pos;
478     if (!url.find(fileUrlPattern) && ((pos = url.find(layoutTestsPattern)) != string::npos)) {
479         // adjust file URLs to match upstream results.
480         result.replace(0, pos + layoutTestsPatternSize, fileTestPrefix);
481     } else if (!url.find(dataUrlPattern)) {
482         // URL-escape data URLs to match results upstream.
483         string path = url.substr(dataUrlPatternSize);
484         result.replace(dataUrlPatternSize, url.length(), path);
485     }
486     return result;
487 }
488
489 static string dumpHistoryItem(const WebHistoryItem& item, int indent, bool isCurrent)
490 {
491     string result;
492
493     if (isCurrent) {
494         result.append("curr->");
495         result.append(indent - 6, ' '); // 6 == "curr->".length()
496     } else
497         result.append(indent, ' ');
498
499     string url = normalizeLayoutTestURLInternal(item.urlString().utf8());
500     result.append(url);
501     if (!item.target().isEmpty()) {
502         result.append(" (in frame \"");
503         result.append(item.target().utf8());
504         result.append("\")");
505     }
506     if (item.isTargetItem())
507         result.append("  **nav target**");
508     result.append("\n");
509
510     const WebVector<WebHistoryItem>& children = item.children();
511     if (!children.isEmpty()) {
512         // Must sort to eliminate arbitrary result ordering which defeats
513         // reproducible testing.
514         // FIXME: WebVector should probably just be a std::vector!!
515         std::vector<WebHistoryItem> sortedChildren;
516         for (size_t i = 0; i < children.size(); ++i)
517             sortedChildren.push_back(children[i]);
518         std::sort(sortedChildren.begin(), sortedChildren.end(), HistoryItemCompareLess);
519         for (size_t i = 0; i < sortedChildren.size(); ++i)
520             result += dumpHistoryItem(sortedChildren[i], indent + 4, false);
521     }
522
523     return result;
524 }
525
526 static void dumpBackForwardList(const TestNavigationController& navigationController, string& result)
527 {
528     result.append("\n============== Back Forward List ==============\n");
529     for (int index = 0; index < navigationController.entryCount(); ++index) {
530         int currentIndex = navigationController.lastCommittedEntryIndex();
531         WebHistoryItem historyItem = navigationController.entryAtIndex(index)->contentState();
532         if (historyItem.isNull()) {
533             historyItem.initialize();
534             historyItem.setURLString(navigationController.entryAtIndex(index)->URL().spec().utf16());
535         }
536         result.append(dumpHistoryItem(historyItem, 8, index == currentIndex));
537     }
538     result.append("===============================================\n");
539 }
540
541 string TestShell::dumpAllBackForwardLists()
542 {
543     string result;
544     for (unsigned i = 0; i < m_windowList.size(); ++i)
545         dumpBackForwardList(*m_windowList[i]->navigationController(), result);
546     return result;
547 }
548
549 void TestShell::dump()
550 {
551     WebScriptController::flushConsoleMessages();
552
553     // Dump the requested representation.
554     WebFrame* frame = m_webView->mainFrame();
555     if (!frame)
556         return;
557     bool shouldDumpAsText = m_layoutTestController->shouldDumpAsText();
558     bool shouldDumpAsAudio = m_layoutTestController->shouldDumpAsAudio();
559     bool shouldGeneratePixelResults = m_layoutTestController->shouldGeneratePixelResults();
560     bool shouldDumpAsPrinted = m_layoutTestController->isPrinting();
561     bool dumpedAnything = false;
562
563     if (shouldDumpAsAudio) {
564         m_printer->handleAudioHeader();
565
566         const WebKit::WebArrayBufferView& webArrayBufferView = m_layoutTestController->audioData();
567         printf("Content-Length: %d\n", webArrayBufferView.byteLength());
568
569         if (fwrite(webArrayBufferView.baseAddress(), 1, webArrayBufferView.byteLength(), stdout) != webArrayBufferView.byteLength())
570             FATAL("Short write to stdout, disk full?\n");
571         m_printer->handleAudioFooter();
572         m_printer->handleTestFooter(true);
573
574         fflush(stdout);
575         fflush(stderr);
576         return;
577     }
578
579     if (m_params.dumpTree) {
580         dumpedAnything = true;
581         m_printer->handleTextHeader();
582         // Text output: the test page can request different types of output
583         // which we handle here.
584         if (!shouldDumpAsText) {
585             // Plain text pages should be dumped as text
586             string mimeType = frame->dataSource()->response().mimeType().utf8();
587             if (mimeType == "text/plain") {
588                 shouldDumpAsText = true;
589                 shouldGeneratePixelResults = false;
590             }
591         }
592         if (shouldDumpAsText) {
593             bool recursive = m_layoutTestController->shouldDumpChildFramesAsText();
594             string dataUtf8 = shouldDumpAsPrinted ? dumpFramesAsPrintedText(frame, recursive) : dumpFramesAsText(frame, recursive);
595             if (fwrite(dataUtf8.c_str(), 1, dataUtf8.size(), stdout) != dataUtf8.size())
596                 FATAL("Short write to stdout, disk full?\n");
597         } else {
598           WebFrame::RenderAsTextControls renderTextBehavior = WebFrame::RenderAsTextNormal;
599             if (shouldDumpAsPrinted)
600                 renderTextBehavior |= WebFrame::RenderAsTextPrinting;
601             if (m_params.debugRenderTree)
602                 renderTextBehavior |= WebFrame::RenderAsTextDebug;
603             printf("%s", frame->renderTreeAsText(renderTextBehavior).utf8().data());
604             bool recursive = m_layoutTestController->shouldDumpChildFrameScrollPositions();
605             dumpFrameScrollPosition(frame, recursive);
606         }
607         if (m_layoutTestController->shouldDumpBackForwardList())
608             printf("%s", dumpAllBackForwardLists().c_str());
609     }
610     if (dumpedAnything && m_params.printSeparators)
611         m_printer->handleTextFooter();
612
613     if (m_params.dumpPixels && shouldGeneratePixelResults) {
614         // Image output: we write the image data to the file given on the
615         // command line (for the dump pixels argument), and the MD5 sum to
616         // stdout.
617         dumpedAnything = true;
618         m_webView->layout();
619         if (m_layoutTestController->testRepaint()) {
620             WebSize viewSize = m_webView->size();
621             int width = viewSize.width;
622             int height = viewSize.height;
623             if (m_layoutTestController->sweepHorizontally()) {
624                 for (WebRect column(0, 0, 1, height); column.x < width; column.x++)
625                     m_webViewHost->paintRect(column);
626             } else {
627                 for (WebRect line(0, 0, width, 1); line.y < height; line.y++)
628                     m_webViewHost->paintRect(line);
629             }
630         } else if (m_layoutTestController->isPrinting())
631             m_webViewHost->paintPagesWithBoundaries();
632         else
633             m_webViewHost->paintInvalidatedRegion();
634
635         // See if we need to draw the selection bounds rect. Selection bounds
636         // rect is the rect enclosing the (possibly transformed) selection.
637         // The rect should be drawn after everything is laid out and painted.
638         if (m_layoutTestController->shouldDumpSelectionRect()) {
639             // If there is a selection rect - draw a red 1px border enclosing rect
640             WebRect wr = frame->selectionBoundsRect();
641             if (!wr.isEmpty()) {
642                 // Render a red rectangle bounding selection rect
643                 SkPaint paint;
644                 paint.setColor(0xFFFF0000); // Fully opaque red
645                 paint.setStyle(SkPaint::kStroke_Style);
646                 paint.setFlags(SkPaint::kAntiAlias_Flag);
647                 paint.setStrokeWidth(1.0f);
648                 SkIRect rect; // Bounding rect
649                 rect.set(wr.x, wr.y, wr.x + wr.width, wr.y + wr.height);
650                 m_webViewHost->canvas()->drawIRect(rect, paint);
651             }
652         }
653
654         dumpImage(m_webViewHost->canvas());
655     }
656     m_printer->handleTestFooter(dumpedAnything);
657     fflush(stdout);
658     fflush(stderr);
659 }
660
661 void TestShell::dumpImage(SkCanvas* canvas) const
662 {
663     // Fix the alpha. The expected PNGs on Mac have an alpha channel, so we want
664     // to keep it. On Windows, the alpha channel is wrong since text/form control
665     // drawing may have erased it in a few places. So on Windows we force it to
666     // opaque and also don't write the alpha channel for the reference. Linux
667     // doesn't have the wrong alpha like Windows, but we match Windows.
668 #if OS(MAC_OS_X)
669     bool discardTransparency = false;
670 #else
671     bool discardTransparency = true;
672     makeCanvasOpaque(canvas);
673 #endif
674
675     const SkBitmap& sourceBitmap = canvas->getTopDevice()->accessBitmap(false);
676     SkAutoLockPixels sourceBitmapLock(sourceBitmap);
677
678     // Compute MD5 sum.
679     MD5 digester;
680     Vector<uint8_t, 16> digestValue;
681 #if OS(ANDROID)
682     // On Android, pixel layout is RGBA (see third_party/skia/include/core/SkColorPriv.h);
683     // however, other Chrome platforms use BGRA (see skia/config/SkUserConfig.h).
684     // To match the checksum of other Chrome platforms, we need to reorder the layout of pixels.
685     // NOTE: The following code assumes we use SkBitmap::kARGB_8888_Config,
686     // which has been checked in device.makeOpaque() (see above).
687     const uint8_t* rawPixels = reinterpret_cast<const uint8_t*>(sourceBitmap.getPixels());
688     size_t bitmapSize = sourceBitmap.getSize();
689     OwnArrayPtr<uint8_t> reorderedPixels = adoptArrayPtr(new uint8_t[bitmapSize]);
690     for (size_t i = 0; i < bitmapSize; i += 4) {
691         reorderedPixels[i] = rawPixels[i + 2]; // R
692         reorderedPixels[i + 1] = rawPixels[i + 1]; // G
693         reorderedPixels[i + 2] = rawPixels[i]; // B
694         reorderedPixels[i + 3] = rawPixels[i + 3]; // A
695     }
696     digester.addBytes(reorderedPixels.get(), bitmapSize);
697     reorderedPixels.clear();
698 #else
699     digester.addBytes(reinterpret_cast<const uint8_t*>(sourceBitmap.getPixels()), sourceBitmap.getSize());
700 #endif
701     digester.checksum(digestValue);
702     string md5hash;
703     md5hash.reserve(16 * 2);
704     for (unsigned i = 0; i < 16; ++i) {
705         char hex[3];
706         // Use "x", not "X". The string must be lowercased.
707         sprintf(hex, "%02x", digestValue[i]);
708         md5hash.append(hex);
709     }
710
711     // Only encode and dump the png if the hashes don't match. Encoding the
712     // image is really expensive.
713     if (md5hash.compare(m_params.pixelHash)) {
714         std::vector<unsigned char> png;
715 #if OS(ANDROID)
716         webkit_support::EncodeRGBAPNGWithChecksum(reinterpret_cast<const unsigned char*>(sourceBitmap.getPixels()), sourceBitmap.width(),
717             sourceBitmap.height(), static_cast<int>(sourceBitmap.rowBytes()), discardTransparency, md5hash, &png);
718 #else
719         webkit_support::EncodeBGRAPNGWithChecksum(reinterpret_cast<const unsigned char*>(sourceBitmap.getPixels()), sourceBitmap.width(),
720             sourceBitmap.height(), static_cast<int>(sourceBitmap.rowBytes()), discardTransparency, md5hash, &png);
721 #endif
722
723         m_printer->handleImage(md5hash.c_str(), m_params.pixelHash.c_str(), &png[0], png.size(), m_params.pixelFileName.c_str());
724     } else
725         m_printer->handleImage(md5hash.c_str(), m_params.pixelHash.c_str(), 0, 0, m_params.pixelFileName.c_str());
726 }
727
728 void TestShell::bindJSObjectsToWindow(WebFrame* frame)
729 {
730     WebTestingSupport::injectInternalsObject(frame);
731     m_accessibilityController->bindToJavascript(frame, WebString::fromUTF8("accessibilityController"));
732     m_testInterfaces->bindTo(frame);
733     m_layoutTestController->bindToJavascript(frame, WebString::fromUTF8("layoutTestController"));
734     m_layoutTestController->bindToJavascript(frame, WebString::fromUTF8("testRunner"));
735     m_eventSender->bindToJavascript(frame, WebString::fromUTF8("eventSender"));
736     m_textInputController->bindToJavascript(frame, WebString::fromUTF8("textInputController"));
737 }
738
739 WebViewHost* TestShell::createNewWindow(const WebKit::WebURL& url)
740 {
741     return createNewWindow(url, 0);
742 }
743
744 WebViewHost* TestShell::createNewWindow(const WebKit::WebURL& url, DRTDevToolsAgent* devToolsAgent)
745 {
746     WebViewHost* host = new WebViewHost(this);
747     WebView* view = WebView::create(host);
748     view->setPermissionClient(webPermissions());
749     view->setDevToolsAgentClient(devToolsAgent);
750     host->setWebWidget(view);
751     m_prefs.applyTo(view);
752     view->initializeMainFrame(host);
753     m_windowList.append(host);
754     host->loadURLForFrame(url, WebString());
755     return host;
756 }
757
758 void TestShell::closeWindow(WebViewHost* window)
759 {
760     size_t i = m_windowList.find(window);
761     if (i == notFound) {
762         ASSERT_NOT_REACHED();
763         return;
764     }
765     m_windowList.remove(i);
766     WebWidget* focusedWidget = m_focusedWidget;
767     if (window->webWidget() == m_focusedWidget)
768         focusedWidget = 0;
769
770     delete window;
771     // We set the focused widget after deleting the web view host because it
772     // can change the focus.
773     m_focusedWidget = focusedWidget;
774     if (m_focusedWidget) {
775         webView()->setIsActive(true);
776         m_focusedWidget->setFocus(true);
777     }
778 }
779
780 void TestShell::closeRemainingWindows()
781 {
782     // Just close devTools window manually because we have custom deinitialization code for it.
783     closeDevTools();
784
785     // Iterate through the window list and close everything except the main
786     // window. We don't want to delete elements as we're iterating, so we copy
787     // to a temp vector first.
788     Vector<WebViewHost*> windowsToDelete;
789     for (unsigned i = 0; i < m_windowList.size(); ++i) {
790         if (m_windowList[i] != webViewHost())
791             windowsToDelete.append(m_windowList[i]);
792     }
793     ASSERT(windowsToDelete.size() + 1 == m_windowList.size());
794     for (unsigned i = 0; i < windowsToDelete.size(); ++i)
795         closeWindow(windowsToDelete[i]);
796     ASSERT(m_windowList.size() == 1);
797 }
798
799 int TestShell::windowCount()
800 {
801     return m_windowList.size();
802 }
803
804 string TestShell::normalizeLayoutTestURL(const string& url)
805 {
806     return normalizeLayoutTestURLInternal(url);
807 }