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