Eliminate copies of encoded image data

third_party/WebKit/Source/platform/image-decoders/webp/WEBPImageDecoderTest.cpp

 /*
  * Copyright (C) 2013 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 15 matching lines @@
  * 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.
  */
 
 #include "platform/image-decoders/webp/WEBPImageDecoder.h"
 
 #include "platform/RuntimeEnabledFeatures.h"
 #include "platform/SharedBuffer.h"
 #include "platform/image-decoders/ImageDecoderTestHelpers.h"
+#include "platform/image-decoders/SharedBufferSegmentReader.h"
 #include "public/platform/WebData.h"
 #include "public/platform/WebSize.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/OwnPtr.h"
 #include "wtf/PassOwnPtr.h"
 #include "wtf/Vector.h"
 #include "wtf/dtoa/utils.h"
 
 namespace blink {
 
@@ skipping 14 matching lines @@
     SCOPED_TRACE(webpFile);
 
     RefPtr<SharedBuffer> fullData = readFile(webpFile);
     ASSERT_TRUE(fullData.get());
     Vector<unsigned> baselineHashes;
     createDecodingBaseline(&createDecoder, fullData.get(), &baselineHashes);
     size_t frameCount = baselineHashes.size();
 
     // Random decoding should get the same results as sequential decoding.
     OwnPtr<ImageDecoder> decoder = createDecoder();
-    decoder->setData(fullData.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(fullData));
+    decoder->setData(segmentReader.get(), true);
     const size_t skippingStep = 5;
     for (size_t i = 0; i < skippingStep; ++i) {
         for (size_t j = i; j < frameCount; j += skippingStep) {
             SCOPED_TRACE(testing::Message() << "Random i:" << i << " j:" << j);
             ImageFrame* frame = decoder->frameBufferAtIndex(j);
             EXPECT_EQ(baselineHashes[j], hashBitmap(frame->getSkBitmap()));
         }
     }
 
     // Decoding in reverse order.
     decoder = createDecoder();
-    decoder->setData(fullData.get(), true);
+    decoder->setData(segmentReader.get(), true);
     for (size_t i = frameCount; i; --i) {
         SCOPED_TRACE(testing::Message() << "Reverse i:" << i);
         ImageFrame* frame = decoder->frameBufferAtIndex(i - 1);
         EXPECT_EQ(baselineHashes[i - 1], hashBitmap(frame->getSkBitmap()));
     }
 }
 
 void testRandomDecodeAfterClearFrameBufferCache(const char* webpFile)
 {
     SCOPED_TRACE(webpFile);
 
     RefPtr<SharedBuffer> data = readFile(webpFile);
     ASSERT_TRUE(data.get());
     Vector<unsigned> baselineHashes;
     createDecodingBaseline(&createDecoder, data.get(), &baselineHashes);
     size_t frameCount = baselineHashes.size();
 
     OwnPtr<ImageDecoder> decoder = createDecoder();
-    decoder->setData(data.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
     for (size_t clearExceptFrame = 0; clearExceptFrame < frameCount; ++clearExceptFrame) {
         decoder->clearCacheExceptFrame(clearExceptFrame);
         const size_t skippingStep = 5;
         for (size_t i = 0; i < skippingStep; ++i) {
             for (size_t j = 0; j < frameCount; j += skippingStep) {
                 SCOPED_TRACE(testing::Message() << "Random i:" << i << " j:" << j);
                 ImageFrame* frame = decoder->frameBufferAtIndex(j);
                 EXPECT_EQ(baselineHashes[j], hashBitmap(frame->getSkBitmap()));
             }
         }
     }
 }
 
 void testDecodeAfterReallocatingData(const char* webpFile)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
     RefPtr<SharedBuffer> data = readFile(webpFile);
     ASSERT_TRUE(data.get());
 
     // Parse from 'data'.
-    decoder->setData(data.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
     size_t frameCount = decoder->frameCount();
 
     // ... and then decode frames from 'reallocatedData'.
     RefPtr<SharedBuffer> reallocatedData = data.get()->copy();
     ASSERT_TRUE(reallocatedData.get());
     data.clear();
-    decoder->setData(reallocatedData.get(), true);
+    segmentReader = adoptRef(new SharedBufferSegmentReader(reallocatedData));
+    decoder->setData(segmentReader.get(), true);
 
     for (size_t i = 0; i < frameCount; ++i) {
         const ImageFrame* const frame = decoder->frameBufferAtIndex(i);
         EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
     }
 }
 
 void testByteByByteSizeAvailable(const char* webpFile, size_t frameOffset, bool hasColorProfile, int expectedRepetitionCount)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
     RefPtr<SharedBuffer> data = readFile(webpFile);
     ASSERT_TRUE(data.get());
     EXPECT_LT(frameOffset, data->size());
 
     // Send data to the decoder byte-by-byte and use the provided frame offset in the data to check
     // isSizeAvailable() changes state only when that offset is reached, and the associated decoder
     // state also: size, colorProfile, frameCount, repetitionCount ...
 
     for (size_t length = 1; length <= frameOffset; ++length) {
         RefPtr<SharedBuffer> tempData = SharedBuffer::create(data->data(), length);
-        decoder->setData(tempData.get(), false);
+        RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(tempData));
+        decoder->setData(segmentReader.get(), false);
 
         if (length < frameOffset) {
             EXPECT_FALSE(decoder->isSizeAvailable());
             EXPECT_TRUE(decoder->size().isEmpty());
             EXPECT_FALSE(decoder->hasColorProfile());
             EXPECT_EQ(0u, decoder->frameCount());
             EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());
             EXPECT_FALSE(decoder->frameBufferAtIndex(0));
         } else {
             EXPECT_TRUE(decoder->isSizeAvailable());
@@ skipping 17 matching lines @@
 }
 
 // If 'parseErrorExpected' is true, error is expected during parse (frameCount()
 // call); else error is expected during decode (frameBufferAtIndex() call).
 void testInvalidImage(const char* webpFile, bool parseErrorExpected)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
 
     RefPtr<SharedBuffer> data = readFile(webpFile);
     ASSERT_TRUE(data.get());
-    decoder->setData(data.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
 
     if (parseErrorExpected) {
         EXPECT_EQ(0u, decoder->frameCount());
         EXPECT_FALSE(decoder->frameBufferAtIndex(0));
     } else {
         EXPECT_GT(decoder->frameCount(), 0u);
         ImageFrame* frame = decoder->frameBufferAtIndex(0);
         ASSERT_TRUE(frame);
         EXPECT_EQ(ImageFrame::FramePartial, frame->status());
     }
@@ skipping 35 matching lines @@
     // Pre-multiplication could round the RGBA channel values. So, we declare
     // that the frames match if the RGBA channel values differ by at most 2.
     EXPECT_GE(2, maxDifference) << webpFile;
 }
 
 // Verify that result of alpha blending is similar for AlphaPremultiplied and AlphaNotPremultiplied cases.
 void testAlphaBlending(const char* webpFile)
 {
     RefPtr<SharedBuffer> data = readFile(webpFile);
     ASSERT_TRUE(data.get());
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
 
     OwnPtr<ImageDecoder> decoderA = createDecoder(ImageDecoder::AlphaPremultiplied);
-    decoderA->setData(data.get(), true);
+    decoderA->setData(segmentReader.get(), true);
 
     OwnPtr<ImageDecoder> decoderB = createDecoder(ImageDecoder::AlphaNotPremultiplied);
-    decoderB->setData(data.get(), true);
+    decoderB->setData(segmentReader.get(), true);
 
     size_t frameCount = decoderA->frameCount();
     ASSERT_EQ(frameCount, decoderB->frameCount());
 
     for (size_t i = 0; i < frameCount; ++i)
         verifyFramesMatch(webpFile, decoderA->frameBufferAtIndex(i), decoderB->frameBufferAtIndex(i));
 }
 
 } // anonymous namespace
 
 TEST(AnimatedWebPTests, uniqueGenerationIDs)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
 
     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
     ASSERT_TRUE(data.get());
-    decoder->setData(data.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
 
     ImageFrame* frame = decoder->frameBufferAtIndex(0);
     uint32_t generationID0 = frame->getSkBitmap().getGenerationID();
     frame = decoder->frameBufferAtIndex(1);
     uint32_t generationID1 = frame->getSkBitmap().getGenerationID();
 
     EXPECT_TRUE(generationID0 != generationID1);
 }
 
 TEST(AnimatedWebPTests, verifyAnimationParametersTransparentImage)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
     EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());
 
     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
     ASSERT_TRUE(data.get());
-    decoder->setData(data.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
 
     const int canvasWidth = 11;
     const int canvasHeight = 29;
     const struct AnimParam {
         int xOffset, yOffset, width, height;
         ImageFrame::DisposalMethod disposalMethod;
         ImageFrame::AlphaBlendSource alphaBlendSource;
         unsigned duration;
         bool hasAlpha;
     } frameParameters[] = {
@@ skipping 21 matching lines @@
     EXPECT_EQ(cAnimationLoopInfinite, decoder->repetitionCount());
 }
 
 TEST(AnimatedWebPTests, verifyAnimationParametersOpaqueFramesTransparentBackground)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
     EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());
 
     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated-opaque.webp");
     ASSERT_TRUE(data.get());
-    decoder->setData(data.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
 
     const int canvasWidth = 94;
     const int canvasHeight = 87;
     const struct AnimParam {
         int xOffset, yOffset, width, height;
         ImageFrame::DisposalMethod disposalMethod;
         ImageFrame::AlphaBlendSource alphaBlendSource;
         unsigned duration;
         bool hasAlpha;
     } frameParameters[] = {
@@ skipping 22 matching lines @@
     EXPECT_EQ(cAnimationLoopInfinite, decoder->repetitionCount());
 }
 
 TEST(AnimatedWebPTests, verifyAnimationParametersBlendOverwrite)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
     EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());
 
     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated-no-blend.webp");
     ASSERT_TRUE(data.get());
-    decoder->setData(data.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
 
     const int canvasWidth = 94;
     const int canvasHeight = 87;
     const struct AnimParam {
         int xOffset, yOffset, width, height;
         ImageFrame::DisposalMethod disposalMethod;
         ImageFrame::AlphaBlendSource alphaBlendSource;
         unsigned duration;
         bool hasAlpha;
     } frameParameters[] = {
@@ skipping 35 matching lines @@
     // One of the frame rectangles extends outside the image boundary.
     testInvalidImage("/LayoutTests/fast/images/resources/invalid-animated-webp3.webp", true);
 }
 
 TEST(AnimatedWebPTests, truncatedLastFrame)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
 
     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/invalid-animated-webp2.webp");
     ASSERT_TRUE(data.get());
-    decoder->setData(data.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
 
     size_t frameCount = 8;
     EXPECT_EQ(frameCount, decoder->frameCount());
     ImageFrame* frame = decoder->frameBufferAtIndex(0);
     ASSERT_TRUE(frame);
     EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
     EXPECT_FALSE(decoder->failed());
     frame = decoder->frameBufferAtIndex(frameCount - 1);
     ASSERT_TRUE(frame);
     EXPECT_EQ(ImageFrame::FramePartial, frame->status());
     EXPECT_TRUE(decoder->failed());
     frame = decoder->frameBufferAtIndex(0);
     ASSERT_TRUE(frame);
     EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
 }
 
 TEST(AnimatedWebPTests, truncatedInBetweenFrame)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
 
     RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/invalid-animated-webp4.webp");
     ASSERT_TRUE(fullData.get());
     RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), fullData->size() - 1);
-    decoder->setData(data.get(), false);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), false);
 
     ImageFrame* frame = decoder->frameBufferAtIndex(1);
     ASSERT_TRUE(frame);
     EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
     frame = decoder->frameBufferAtIndex(2);
     ASSERT_TRUE(frame);
     EXPECT_EQ(ImageFrame::FramePartial, frame->status());
     EXPECT_TRUE(decoder->failed());
 }
 
 // Reproduce a crash that used to happen for a specific file with specific sequence of method calls.
 TEST(AnimatedWebPTests, reproCrash)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
 
     RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/invalid_vp8_vp8x.webp");
     ASSERT_TRUE(fullData.get());
 
     // Parse partial data up to which error in bitstream is not detected.
     const size_t partialSize = 32768;
     ASSERT_GT(fullData->size(), partialSize);
     RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), partialSize);
-    decoder->setData(data.get(), false);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), false);
     EXPECT_EQ(1u, decoder->frameCount());
     ImageFrame* frame = decoder->frameBufferAtIndex(0);
     ASSERT_TRUE(frame);
     EXPECT_EQ(ImageFrame::FramePartial, frame->status());
     EXPECT_FALSE(decoder->failed());
 
     // Parse full data now. The error in bitstream should now be detected.
-    decoder->setData(fullData.get(), true);
+    segmentReader = adoptRef(new SharedBufferSegmentReader(fullData));
+    decoder->setData(segmentReader.get(), true);
     EXPECT_EQ(1u, decoder->frameCount());
     frame = decoder->frameBufferAtIndex(0);
     ASSERT_TRUE(frame);
     EXPECT_EQ(ImageFrame::FramePartial, frame->status());
     EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());
     EXPECT_TRUE(decoder->failed());
 }
 
 TEST(AnimatedWebPTests, progressiveDecode)
 {
     RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
     ASSERT_TRUE(fullData.get());
     const size_t fullLength = fullData->size();
 
     OwnPtr<ImageDecoder>  decoder;
     ImageFrame* frame;
 
     Vector<unsigned> truncatedHashes;
     Vector<unsigned> progressiveHashes;
 
     // Compute hashes when the file is truncated.
     const size_t increment = 1;
     for (size_t i = 1; i <= fullLength; i += increment) {
         decoder = createDecoder();
         RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), i);
-        decoder->setData(data.get(), i == fullLength);
+        RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+        decoder->setData(segmentReader.get(), i == fullLength);
         frame = decoder->frameBufferAtIndex(0);
         if (!frame) {
             truncatedHashes.append(0);
             continue;
         }
         truncatedHashes.append(hashBitmap(frame->getSkBitmap()));
     }
 
     // Compute hashes when the file is progressively decoded.
     decoder = createDecoder();
     for (size_t i = 1; i <= fullLength; i += increment) {
         RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), i);
-        decoder->setData(data.get(), i == fullLength);
+        RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+        decoder->setData(segmentReader.get(), i == fullLength);
         frame = decoder->frameBufferAtIndex(0);
         if (!frame) {
             progressiveHashes.append(0);
             continue;
         }
         progressiveHashes.append(hashBitmap(frame->getSkBitmap()));
     }
 
     bool match = true;
     for (size_t i = 0; i < truncatedHashes.size(); ++i) {
         if (truncatedHashes[i] != progressiveHashes[i]) {
             match = false;
             break;
         }
     }
     EXPECT_TRUE(match);
 }
 
 TEST(AnimatedWebPTests, frameIsCompleteAndDuration)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
 
     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
     ASSERT_TRUE(data.get());
 
     ASSERT_GE(data->size(), 10u);
     RefPtr<SharedBuffer> tempData = SharedBuffer::create(data->data(), data->size() - 10);
-    decoder->setData(tempData.get(), false);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(tempData));
+    decoder->setData(segmentReader.get(), false);
 
     EXPECT_EQ(2u, decoder->frameCount());
     EXPECT_FALSE(decoder->failed());
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(0));
     EXPECT_EQ(1000, decoder->frameDurationAtIndex(0));
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(1));
     EXPECT_EQ(500, decoder->frameDurationAtIndex(1));
 
-    decoder->setData(data.get(), true);
+    segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
     EXPECT_EQ(3u, decoder->frameCount());
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(0));
     EXPECT_EQ(1000, decoder->frameDurationAtIndex(0));
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(1));
     EXPECT_EQ(500, decoder->frameDurationAtIndex(1));
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(2));
     EXPECT_EQ(1000.0, decoder->frameDurationAtIndex(2));
 }
 
 TEST(AnimatedWebPTests, updateRequiredPreviousFrameAfterFirstDecode)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
 
     RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
     ASSERT_TRUE(fullData.get());
 
     // Give it data that is enough to parse but not decode in order to check the status
     // of requiredPreviousFrameIndex before decoding.
     size_t partialSize = 1;
     do {
         RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), partialSize);
-        decoder->setData(data.get(), false);
+        RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+        decoder->setData(segmentReader.get(), false);
         ++partialSize;
     } while (!decoder->frameCount() || decoder->frameBufferAtIndex(0)->status() == ImageFrame::FrameEmpty);
 
     EXPECT_EQ(kNotFound, decoder->frameBufferAtIndex(0)->requiredPreviousFrameIndex());
     size_t frameCount = decoder->frameCount();
     for (size_t i = 1; i < frameCount; ++i)
         EXPECT_EQ(i - 1, decoder->frameBufferAtIndex(i)->requiredPreviousFrameIndex());
 
-    decoder->setData(fullData.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(fullData));
+    decoder->setData(segmentReader.get(), true);
     for (size_t i = 0; i < frameCount; ++i)
         EXPECT_EQ(kNotFound, decoder->frameBufferAtIndex(i)->requiredPreviousFrameIndex());
 }
 
 TEST(AnimatedWebPTests, randomFrameDecode)
 {
     testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated.webp");
     testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated-opaque.webp");
     testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated-large.webp");
     testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated-icc-xmp.webp");
@@ skipping 14 matching lines @@
     Vector<unsigned> baselineHashes;
     createDecodingBaseline(&createDecoder, fullData.get(), &baselineHashes);
     size_t frameCount = baselineHashes.size();
 
     OwnPtr<ImageDecoder> decoder = createDecoder();
 
     // Let frame 0 be partially decoded.
     size_t partialSize = 1;
     do {
         RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), partialSize);
-        decoder->setData(data.get(), false);
+        RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+        decoder->setData(segmentReader.get(), false);
         ++partialSize;
     } while (!decoder->frameCount() || decoder->frameBufferAtIndex(0)->status() == ImageFrame::FrameEmpty);
 
     // Skip to the last frame and clear.
-    decoder->setData(fullData.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(fullData));
+    decoder->setData(segmentReader.get(), true);
     EXPECT_EQ(frameCount, decoder->frameCount());
     ImageFrame* lastFrame = decoder->frameBufferAtIndex(frameCount - 1);
     EXPECT_EQ(baselineHashes[frameCount - 1], hashBitmap(lastFrame->getSkBitmap()));
     decoder->clearCacheExceptFrame(kNotFound);
 
     // Resume decoding of the first frame.
     ImageFrame* firstFrame = decoder->frameBufferAtIndex(0);
     EXPECT_EQ(ImageFrame::FrameComplete, firstFrame->status());
     EXPECT_EQ(baselineHashes[0], hashBitmap(firstFrame->getSkBitmap()));
 }
@@ skipping 38 matching lines @@
 {
     testByteByByteSizeAvailable("/LayoutTests/fast/images/resources/webp-color-profile-lossy.webp", 520u, true, cAnimationNone);
     testByteByByteSizeAvailable("/LayoutTests/fast/images/resources/test.webp", 30u, false, cAnimationNone);
 }
 
 TEST(StaticWebPTests, notAnimated)
 {
     OwnPtr<ImageDecoder> decoder = createDecoder();
     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-color-profile-lossy.webp");
     ASSERT_TRUE(data.get());
-    decoder->setData(data.get(), true);
+    RefPtr<SharedBufferSegmentReader> segmentReader = adoptRef(new SharedBufferSegmentReader(data));
+    decoder->setData(segmentReader.get(), true);
     EXPECT_EQ(1u, decoder->frameCount());
     EXPECT_EQ(cAnimationNone, decoder->repetitionCount());
 }
 
 } // namespace blink