1 /* 2 * Copyright (C) 2009 Apple, 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 #include "config.h" 30 #include "DumpRenderTree.h" 31 #include "LayoutTestController.h" 32 #include "PixelDumpSupport.h" 33 #include <wtf/Assertions.h> 34 #include <wtf/RefPtr.h> 35 #include <wtf/RetainPtr.h> 36 37 #if PLATFORM(CG) 38 #include "PixelDumpSupportCG.h" 39 #elif PLATFORM(CAIRO) 40 #include "PixelDumpSupportCairo.h" 41 #endif 42 43 void dumpWebViewAsPixelsAndCompareWithExpected(const std::string& expectedHash) 44 { 45 RefPtr<BitmapContext> context = createBitmapContextFromWebView(gLayoutTestController->testOnscreen(), gLayoutTestController->testRepaint(), gLayoutTestController->testRepaintSweepHorizontally(), gLayoutTestController->dumpSelectionRect()); 46 ASSERT(context); 47 48 // Compute the hash of the bitmap context pixels 49 char actualHash[33]; 50 computeMD5HashStringForBitmapContext(context.get(), actualHash); 51 printf("\nActualHash: %s\n", actualHash); 52 53 // Check the computed hash against the expected one and dump image on mismatch 54 bool dumpImage = true; 55 if (expectedHash.length() > 0) { 56 ASSERT(expectedHash.length() == 32); 57 58 printf("\nExpectedHash: %s\n", expectedHash.c_str()); 59 60 if (expectedHash == actualHash) // FIXME: do case insensitive compare 61 dumpImage = false; 62 } 63 64 if (dumpImage) 65 dumpBitmap(context.get()); 66 } 67 68 void printPNG(const unsigned char* data, const size_t dataLength) 69 { 70 printf("Content-Type: %s\n", "image/png"); 71 printf("Content-Length: %lu\n", static_cast<unsigned long>(dataLength)); 72 73 const size_t bytesToWriteInOneChunk = 1 << 15; 74 size_t dataRemainingToWrite = dataLength; 75 while (dataRemainingToWrite) { 76 size_t bytesToWriteInThisChunk = std::min(dataRemainingToWrite, bytesToWriteInOneChunk); 77 size_t bytesWritten = fwrite(data, 1, bytesToWriteInThisChunk, stdout); 78 if (bytesWritten != bytesToWriteInThisChunk) 79 break; 80 dataRemainingToWrite -= bytesWritten; 81 data += bytesWritten; 82 } 83 } 84