1// Copyright 2015 PDFium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "core/fpdfapi/parser/fpdf_parser_decode.h"
6
7#include <cstring>
8#include <string>
9
10#include "testing/embedder_test.h"
11#include "testing/gtest/include/gtest/gtest.h"
12#include "testing/test_support.h"
13
14using FPDFParserDecodeEmbeddertest = EmbedderTest;
15
16// NOTE: python's zlib.compress() and zlib.decompress() may be useful for
17// external validation of the FlateEncode/FlateDecode test cases.
18
19TEST_F(FPDFParserDecodeEmbeddertest, FlateEncode) {
20  static const pdfium::StrFuncTestData flate_encode_cases[] = {
21      STR_IN_OUT_CASE("", "\x78\x9c\x03\x00\x00\x00\x00\x01"),
22      STR_IN_OUT_CASE(" ", "\x78\x9c\x53\x00\x00\x00\x21\x00\x21"),
23      STR_IN_OUT_CASE("123", "\x78\x9c\x33\x34\x32\x06\x00\01\x2d\x00\x97"),
24      STR_IN_OUT_CASE("\x00\xff", "\x78\x9c\x63\xf8\x0f\x00\x01\x01\x01\x00"),
25      STR_IN_OUT_CASE(
26          "1 0 0 -1 29 763 cm\n0 0 555 735 re\nW n\nq\n0 0 555 734.394 re\n"
27          "W n\nq\n0.8009 0 0 0.8009 0 0 cm\n1 1 1 RG 1 1 1 rg\n/G0 gs\n"
28          "0 0 693 917 re\nf\nQ\nQ\n",
29          "\x78\x9c\x33\x54\x30\x00\x42\x5d\x43\x05\x23\x4b\x05\x73\x33\x63"
30          "\x85\xe4\x5c\x2e\x90\x80\xa9\xa9\xa9\x82\xb9\xb1\xa9\x42\x51\x2a"
31          "\x57\xb8\x42\x1e\x57\x21\x92\xa0\x89\x9e\xb1\xa5\x09\x92\x84\x9e"
32          "\x85\x81\x81\x25\xd8\x14\x24\x26\xd0\x18\x43\x05\x10\x0c\x72\x57"
33          "\x80\x30\x8a\xd2\xb9\xf4\xdd\x0d\x14\xd2\x8b\xc1\x46\x99\x59\x1a"
34          "\x2b\x58\x1a\x9a\x83\x8c\x49\xe3\x0a\x04\x42\x00\x37\x4c\x1b\x42"),
35  };
36
37  for (size_t i = 0; i < FX_ArraySize(flate_encode_cases); ++i) {
38    const pdfium::StrFuncTestData& data = flate_encode_cases[i];
39    unsigned char* buf = nullptr;
40    uint32_t buf_size;
41    EXPECT_TRUE(FlateEncode(data.input, data.input_size, &buf, &buf_size));
42    ASSERT_TRUE(buf);
43    EXPECT_EQ(data.expected_size, buf_size) << " for case " << i;
44    if (data.expected_size != buf_size)
45      continue;
46    EXPECT_EQ(0, memcmp(data.expected, buf, data.expected_size))
47        << " for case " << i;
48    FX_Free(buf);
49  }
50}
51
52TEST_F(FPDFParserDecodeEmbeddertest, FlateDecode) {
53  static const pdfium::DecodeTestData flate_decode_cases[] = {
54      STR_IN_OUT_CASE("", "", 0),
55      STR_IN_OUT_CASE("preposterous nonsense", "", 2),
56      STR_IN_OUT_CASE("\x78\x9c\x03\x00\x00\x00\x00\x01", "", 8),
57      STR_IN_OUT_CASE("\x78\x9c\x53\x00\x00\x00\x21\x00\x21", " ", 9),
58      STR_IN_OUT_CASE("\x78\x9c\x33\x34\x32\x06\x00\01\x2d\x00\x97", "123", 11),
59      STR_IN_OUT_CASE("\x78\x9c\x63\xf8\x0f\x00\x01\x01\x01\x00", "\x00\xff",
60                      10),
61      STR_IN_OUT_CASE(
62          "\x78\x9c\x33\x54\x30\x00\x42\x5d\x43\x05\x23\x4b\x05\x73\x33\x63"
63          "\x85\xe4\x5c\x2e\x90\x80\xa9\xa9\xa9\x82\xb9\xb1\xa9\x42\x51\x2a"
64          "\x57\xb8\x42\x1e\x57\x21\x92\xa0\x89\x9e\xb1\xa5\x09\x92\x84\x9e"
65          "\x85\x81\x81\x25\xd8\x14\x24\x26\xd0\x18\x43\x05\x10\x0c\x72\x57"
66          "\x80\x30\x8a\xd2\xb9\xf4\xdd\x0d\x14\xd2\x8b\xc1\x46\x99\x59\x1a"
67          "\x2b\x58\x1a\x9a\x83\x8c\x49\xe3\x0a\x04\x42\x00\x37\x4c\x1b\x42",
68          "1 0 0 -1 29 763 cm\n0 0 555 735 re\nW n\nq\n0 0 555 734.394 re\n"
69          "W n\nq\n0.8009 0 0 0.8009 0 0 cm\n1 1 1 RG 1 1 1 rg\n/G0 gs\n"
70          "0 0 693 917 re\nf\nQ\nQ\n",
71          96),
72  };
73
74  for (size_t i = 0; i < FX_ArraySize(flate_decode_cases); ++i) {
75    const pdfium::DecodeTestData& data = flate_decode_cases[i];
76    unsigned char* buf = nullptr;
77    uint32_t buf_size;
78    EXPECT_EQ(data.processed_size,
79              FlateDecode(data.input, data.input_size, &buf, &buf_size))
80        << " for case " << i;
81    ASSERT_TRUE(buf);
82    EXPECT_EQ(data.expected_size, buf_size) << " for case " << i;
83    if (data.expected_size != buf_size)
84      continue;
85    EXPECT_EQ(0, memcmp(data.expected, buf, data.expected_size))
86        << " for case " << i;
87    FX_Free(buf);
88  }
89}
90
91TEST_F(FPDFParserDecodeEmbeddertest, Bug_552046) {
92  // Tests specifying multiple image filters for a stream. Should not cause a
93  // crash when rendered.
94  EXPECT_TRUE(OpenDocument("bug_552046.pdf"));
95  FPDF_PAGE page = LoadPage(0);
96  FPDF_BITMAP bitmap = RenderPage(page);
97  CompareBitmap(bitmap, 612, 792, "1940568c9ba33bac5d0b1ee9558c76b3");
98  FPDFBitmap_Destroy(bitmap);
99  UnloadPage(page);
100}
101
102TEST_F(FPDFParserDecodeEmbeddertest, Bug_555784) {
103  // Tests bad input to the run length decoder that caused a heap overflow.
104  // Should not cause a crash when rendered.
105  EXPECT_TRUE(OpenDocument("bug_555784.pdf"));
106  FPDF_PAGE page = LoadPage(0);
107  FPDF_BITMAP bitmap = RenderPage(page);
108  CompareBitmap(bitmap, 612, 792, "1940568c9ba33bac5d0b1ee9558c76b3");
109  FPDFBitmap_Destroy(bitmap);
110  UnloadPage(page);
111}
112
113TEST_F(FPDFParserDecodeEmbeddertest, Bug_455199) {
114  // Tests object numbers with a value > 01000000.
115  // Should open successfully.
116  EXPECT_TRUE(OpenDocument("bug_455199.pdf"));
117  FPDF_PAGE page = LoadPage(0);
118  FPDF_BITMAP bitmap = RenderPage(page);
119#if _FX_PLATFORM_ == _FX_PLATFORM_APPLE_
120  const char kExpectedMd5sum[] = "b90475ca64d1348c3bf5e2b77ad9187a";
121#elif _FX_PLATFORM_ == _FX_PLATFORM_WINDOWS_
122  const char kExpectedMd5sum[] = "e5a6fa28298db07484cd922f3e210c88";
123#else
124  const char kExpectedMd5sum[] = "2baa4c0e1758deba1b9c908e1fbd04ed";
125#endif
126  CompareBitmap(bitmap, 200, 200, kExpectedMd5sum);
127  FPDFBitmap_Destroy(bitmap);
128  UnloadPage(page);
129}
130