blob: 126a2998698939b81abd2618751d83b8755d1f64 [file] [log] [blame]
// Copyright 2019 The PDFium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "core/fxcodec/progressive_decoder.h"
#include <stddef.h>
#include <stdint.h>
#include <array>
#include <numeric>
#include <tuple>
#include <utility>
#include "core/fxcodec/fx_codec.h"
#include "core/fxcodec/fx_codec_def.h"
#include "core/fxcodec/jpeg/jpeg_progressive_decoder.h"
#include "core/fxcrt/cfx_read_only_span_stream.h"
#include "core/fxcrt/cfx_read_only_vector_stream.h"
#include "core/fxcrt/data_vector.h"
#include "core/fxcrt/retain_ptr.h"
#include "core/fxcrt/span.h"
#include "core/fxge/dib/cfx_dibitmap.h"
#include "core/fxge/dib/fx_dib.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#ifdef PDF_ENABLE_XFA_BMP
#include "core/fxcodec/bmp/bmp_decoder.h"
#include "core/fxcodec/bmp/bmp_progressive_decoder.h"
#endif // PDF_ENABLE_XFA_BMP
#ifdef PDF_ENABLE_XFA_GIF
#include "core/fxcodec/gif/gif_decoder.h"
#include "core/fxcodec/gif/gif_progressive_decoder.h"
#endif // PDF_ENABLE_XFA_GIF
namespace fxcodec {
namespace {
using ::testing::ElementsAre;
using ::testing::ElementsAreArray;
template <size_t Size>
constexpr std::array<uint8_t, Size> IotaArray(uint8_t start) {
std::array<uint8_t, Size> result;
std::iota(result.begin(), result.end(), start);
return result;
}
FXCODEC_STATUS DecodeToBitmap(ProgressiveDecoder& decoder,
RetainPtr<CFX_DIBitmap> bitmap) {
FXCODEC_STATUS status = decoder.StartDecode(std::move(bitmap));
while (status == FXCODEC_STATUS::kDecodeToBeContinued) {
status = decoder.ContinueDecode();
}
return status;
}
class ProgressiveDecoderTest : public testing::Test {
void SetUp() override {
#ifdef PDF_ENABLE_XFA_BMP
BmpProgressiveDecoder::InitializeGlobals();
#endif
#ifdef PDF_ENABLE_XFA_GIF
GifProgressiveDecoder::InitializeGlobals();
#endif
JpegProgressiveDecoder::InitializeGlobals();
}
void TearDown() override {
JpegProgressiveDecoder::DestroyGlobals();
#ifdef PDF_ENABLE_XFA_GIF
GifProgressiveDecoder::DestroyGlobals();
#endif
#ifdef PDF_ENABLE_XFA_BMP
BmpProgressiveDecoder::DestroyGlobals();
#endif
}
};
} // namespace
#ifdef PDF_ENABLE_XFA_BMP
TEST_F(ProgressiveDecoderTest, Indexed8Bmp) {
static constexpr uint8_t kInput[] = {
0x42, 0x4d, 0x3e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x3a,
0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x08, 0x00, 0x00, 0x00, 0x00,
0x00, 0x04, 0x00, 0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x13, 0x0b,
0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0xc0,
0x80, 0x40, 0x00, 0x00, 0x00, 0x00, 0x00};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_BMP, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(1, decoder.GetWidth());
ASSERT_EQ(1, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kRgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
EXPECT_THAT(bitmap->GetScanline(0), ElementsAre(0xc0, 0x80, 0x40, 0x00));
}
TEST_F(ProgressiveDecoderTest, Indexed8BmpWithInvalidIndex) {
static constexpr uint8_t kInput[] = {
0x42, 0x4d, 0x3e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x3a,
0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x08, 0x00, 0x00, 0x00, 0x00,
0x00, 0x04, 0x00, 0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x13, 0x0b,
0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0xc0,
0x80, 0x40, 0x00, 0x01, 0x00, 0x00, 0x00};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_BMP, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(1, decoder.GetWidth());
ASSERT_EQ(1, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kRgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kError, status);
}
TEST_F(ProgressiveDecoderTest, Direct24Bmp) {
static constexpr uint8_t kInput[] = {
0x42, 0x4d, 0x3a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x36, 0x00,
0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00,
0x00, 0x00, 0x01, 0x00, 0x18, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00,
0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xc0, 0x80, 0x40, 0x00};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_BMP, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(1, decoder.GetWidth());
ASSERT_EQ(1, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kRgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
EXPECT_THAT(bitmap->GetScanline(0), ElementsAre(0xc0, 0x80, 0x40, 0x00));
}
TEST_F(ProgressiveDecoderTest, Direct32Bmp) {
static constexpr uint8_t kInput[] = {
0x42, 0x4d, 0x3a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x36, 0x00,
0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00,
0x00, 0x00, 0x01, 0x00, 0x20, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00,
0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xc0, 0x80, 0x40, 0xff};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_BMP, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(1, decoder.GetWidth());
ASSERT_EQ(1, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kRgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
EXPECT_THAT(bitmap->GetScanline(0), ElementsAre(0xc0, 0x80, 0x40, 0x00));
}
TEST_F(ProgressiveDecoderTest, BmpWithDataOffsetBeforeEndOfHeader) {
static constexpr uint8_t kInput[] = {
0x42, 0x4d, 0x3a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x35, 0x00,
0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00,
0x00, 0x00, 0x01, 0x00, 0x18, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00,
0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xc0, 0x80, 0x40, 0x00};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_BMP, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(1, decoder.GetWidth());
ASSERT_EQ(1, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kRgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
EXPECT_THAT(bitmap->GetScanline(0), ElementsAre(0xc0, 0x80, 0x40, 0x00));
}
TEST_F(ProgressiveDecoderTest, BmpWithDataOffsetAfterEndOfHeader) {
static constexpr uint8_t kInput[] = {
0x42, 0x4d, 0x3b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x37, 0x00,
0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00,
0x00, 0x00, 0x01, 0x00, 0x18, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00,
0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xc0, 0x80, 0x40, 0x00};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_BMP, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(1, decoder.GetWidth());
ASSERT_EQ(1, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kRgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
EXPECT_THAT(bitmap->GetScanline(0), ElementsAre(0xc0, 0x80, 0x40, 0x00));
}
TEST_F(ProgressiveDecoderTest, LargeBmp) {
// Construct a 24-bit BMP larger than `kBlockSize` (4096 bytes).
static constexpr uint8_t kWidth = 37;
static constexpr uint8_t kHeight = 38;
static constexpr size_t kScanlineSize = kWidth * 3 + 1;
DataVector<uint8_t> input = {
0x42, 0x4d, 0xd6, 0x10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x36,
0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, kWidth, 0x00, 0x00, 0x00,
kHeight, 0x00, 0x00, 0x00, 0x01, 0x00, 0x18, 0x00, 0x00, 0x00, 0x00,
0x00, 0xa0, 0x10, 0x00, 0x00, 0x13, 0x0b, 0x00, 0x00, 0x13, 0x0b,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
input.resize(54 + kScanlineSize * kHeight);
std::iota(input.begin() + 54, input.end(), 0);
ASSERT_EQ(4310u, input.size());
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlyVectorStream>(std::move(input));
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_BMP, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(kWidth, decoder.GetWidth());
ASSERT_EQ(kHeight, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kRgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
for (size_t row = 0; row < kHeight; ++row) {
// BMP encodes rows from bottom to top by default.
pdfium::span<const uint8_t> scanline =
bitmap->GetScanline(kHeight - row - 1);
EXPECT_THAT(
scanline.first(kScanlineSize - 1),
ElementsAreArray(IotaArray<kScanlineSize - 1>(row * kScanlineSize)));
// Last byte is padding to a 32-bit boundary.
EXPECT_EQ(0, scanline[kScanlineSize - 1]);
}
}
#endif // PDF_ENABLE_XFA_BMP
#ifdef PDF_ENABLE_XFA_GIF
TEST_F(ProgressiveDecoderTest, Gif87a) {
static constexpr uint8_t kInput[] = {
0x47, 0x49, 0x46, 0x38, 0x37, 0x61, 0x01, 0x00, 0x01, 0x00, 0x80, 0x01,
0x00, 0x40, 0x80, 0xc0, 0x80, 0x80, 0x80, 0x2c, 0x00, 0x00, 0x00, 0x00,
0x01, 0x00, 0x01, 0x00, 0x00, 0x02, 0x02, 0x44, 0x01, 0x00, 0x3b};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_GIF, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(1, decoder.GetWidth());
ASSERT_EQ(1, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kArgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
EXPECT_THAT(bitmap->GetScanline(0), ElementsAre(0xc0, 0x80, 0x40, 0xff));
}
TEST_F(ProgressiveDecoderTest, Gif89a) {
static constexpr uint8_t kInput[] = {
0x47, 0x49, 0x46, 0x38, 0x39, 0x61, 0x01, 0x00, 0x01, 0x00, 0x80,
0x01, 0x00, 0x40, 0x80, 0xc0, 0x80, 0x80, 0x80, 0x21, 0xf9, 0x04,
0x00, 0x00, 0x00, 0x00, 0x00, 0x2c, 0x00, 0x00, 0x00, 0x00, 0x01,
0x00, 0x01, 0x00, 0x00, 0x02, 0x02, 0x44, 0x01, 0x00, 0x3b};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_GIF, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(1, decoder.GetWidth());
ASSERT_EQ(1, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kArgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
EXPECT_THAT(bitmap->GetScanline(0), ElementsAre(0xc0, 0x80, 0x40, 0xff));
}
TEST_F(ProgressiveDecoderTest, GifInsufficientCodeSize) {
// This GIF causes `LZWDecompressor::Create()` to fail because the minimum
// code size is too small for the palette.
static constexpr uint8_t kInput[] = {
0x47, 0x49, 0x46, 0x38, 0x37, 0x61, 0x01, 0x00, 0x01, 0x00, 0x82,
0x01, 0x00, 0x40, 0x80, 0xc0, 0x80, 0x80, 0x80, 0x81, 0x81, 0x81,
0x82, 0x82, 0x82, 0x83, 0x83, 0x83, 0x84, 0x84, 0x84, 0x85, 0x85,
0x85, 0x86, 0x86, 0x86, 0x2c, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00,
0x01, 0x00, 0x00, 0x02, 0x2, 0x44, 0x01, 0x00, 0x3b};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_GIF, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(1, decoder.GetWidth());
ASSERT_EQ(1, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kArgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kError, status);
}
TEST_F(ProgressiveDecoderTest, GifDecodeAcrossScanlines) {
// This GIF contains an LZW code unit split across 2 scanlines. The decoder
// must continue decoding the second scanline using the residual data.
static constexpr uint8_t kInput[] = {
0x47, 0x49, 0x46, 0x38, 0x37, 0x61, 0x04, 0x00, 0x02, 0x00, 0x80, 0x01,
0x00, 0x40, 0x80, 0xc0, 0x80, 0x80, 0x80, 0x2c, 0x00, 0x00, 0x00, 0x00,
0x04, 0x00, 0x02, 0x00, 0x00, 0x02, 0x03, 0x84, 0x6f, 0x05, 0x00, 0x3b};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_GIF, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(4, decoder.GetWidth());
ASSERT_EQ(2, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kArgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
EXPECT_THAT(bitmap->GetScanline(0),
ElementsAre(0xc0, 0x80, 0x40, 0xff, 0xc0, 0x80, 0x40, 0xff, 0xc0,
0x80, 0x40, 0xff, 0xc0, 0x80, 0x40, 0xff));
EXPECT_THAT(bitmap->GetScanline(1),
ElementsAre(0xc0, 0x80, 0x40, 0xff, 0xc0, 0x80, 0x40, 0xff, 0xc0,
0x80, 0x40, 0xff, 0xc0, 0x80, 0x40, 0xff));
}
TEST_F(ProgressiveDecoderTest, GifDecodeAcrossSubblocks) {
// This GIF contains a scanline split across 2 data sub-blocks. The decoder
// must continue decoding in the second sub-block.
static constexpr uint8_t kInput[] = {
0x47, 0x49, 0x46, 0x38, 0x37, 0x61, 0x04, 0x00, 0x02, 0x00,
0x80, 0x01, 0x00, 0x40, 0x80, 0xc0, 0x80, 0x80, 0x80, 0x2c,
0x00, 0x00, 0x00, 0x00, 0x04, 0x00, 0x02, 0x00, 0x00, 0x02,
0x02, 0x84, 0x6f, 0x01, 0x05, 0x00, 0x3b};
ProgressiveDecoder decoder;
auto source = pdfium::MakeRetain<CFX_ReadOnlySpanStream>(kInput);
CFX_DIBAttribute attr;
FXCODEC_STATUS status =
decoder.LoadImageInfo(std::move(source), FXCODEC_IMAGE_GIF, &attr, true);
ASSERT_EQ(FXCODEC_STATUS::kFrameReady, status);
ASSERT_EQ(4, decoder.GetWidth());
ASSERT_EQ(2, decoder.GetHeight());
auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>();
ASSERT_TRUE(bitmap->Create(decoder.GetWidth(), decoder.GetHeight(),
FXDIB_Format::kArgb));
size_t frames;
std::tie(status, frames) = decoder.GetFrames();
ASSERT_EQ(FXCODEC_STATUS::kDecodeReady, status);
ASSERT_EQ(1u, frames);
status = DecodeToBitmap(decoder, bitmap);
EXPECT_EQ(FXCODEC_STATUS::kDecodeFinished, status);
EXPECT_THAT(bitmap->GetScanline(0),
ElementsAre(0xc0, 0x80, 0x40, 0xff, 0xc0, 0x80, 0x40, 0xff, 0xc0,
0x80, 0x40, 0xff, 0xc0, 0x80, 0x40, 0xff));
EXPECT_THAT(bitmap->GetScanline(1),
ElementsAre(0xc0, 0x80, 0x40, 0xff, 0xc0, 0x80, 0x40, 0xff, 0xc0,
0x80, 0x40, 0xff, 0xc0, 0x80, 0x40, 0xff));
}
#endif // PDF_ENABLE_XFA_GIF
} // namespace fxcodec