OLD | NEW |
1 // Copyright 2015 PDFium Authors. All rights reserved. | 1 // Copyright 2015 PDFium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <limits> | 5 #include <limits> |
6 #include <string> | 6 #include <string> |
7 | 7 |
8 #include "core/fpdfapi/parser/cpdf_parser.h" | 8 #include "core/fpdfapi/parser/cpdf_parser.h" |
9 #include "core/fpdfapi/parser/cpdf_syntax_parser.h" | 9 #include "core/fpdfapi/parser/cpdf_syntax_parser.h" |
10 #include "core/fxcrt/fx_ext.h" | 10 #include "core/fxcrt/fx_ext.h" |
11 #include "core/fxcrt/fx_stream.h" | 11 #include "core/fxcrt/fx_stream.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
13 #include "testing/utils/path_service.h" | 13 #include "testing/utils/path_service.h" |
14 | 14 |
15 // Provide a way to read test data from a buffer instead of a file. | 15 // Provide a way to read test data from a buffer instead of a file. |
16 class CFX_TestBufferRead : public IFX_FileRead { | 16 class CFX_TestBufferRead : public IFX_SeekableReadStream { |
17 public: | 17 public: |
18 CFX_TestBufferRead(const unsigned char* buffer_in, size_t buf_size) | 18 CFX_TestBufferRead(const unsigned char* buffer_in, size_t buf_size) |
19 : buffer_(buffer_in), total_size_(buf_size) {} | 19 : buffer_(buffer_in), total_size_(buf_size) {} |
20 | 20 |
21 // IFX_Stream | 21 // IFX_Stream |
22 void Release() override { delete this; } | 22 void Release() override { delete this; } |
23 | 23 |
24 // IFX_FileRead | 24 // IFX_SeekableReadStream |
25 FX_BOOL ReadBlock(void* buffer, FX_FILESIZE offset, size_t size) override { | 25 FX_BOOL ReadBlock(void* buffer, FX_FILESIZE offset, size_t size) override { |
26 if (offset < 0 || offset + size > total_size_) { | 26 if (offset < 0 || offset + size > total_size_) { |
27 return FALSE; | 27 return FALSE; |
28 } | 28 } |
29 | 29 |
30 memcpy(buffer, buffer_ + offset, size); | 30 memcpy(buffer, buffer_ + offset, size); |
31 return TRUE; | 31 return TRUE; |
32 } | 32 } |
33 FX_FILESIZE GetSize() override { return (FX_FILESIZE)total_size_; }; | 33 FX_FILESIZE GetSize() override { return (FX_FILESIZE)total_size_; }; |
34 | 34 |
35 protected: | 35 protected: |
36 const unsigned char* buffer_; | 36 const unsigned char* buffer_; |
37 size_t total_size_; | 37 size_t total_size_; |
38 }; | 38 }; |
39 | 39 |
40 // A wrapper class to help test member functions of CPDF_Parser. | 40 // A wrapper class to help test member functions of CPDF_Parser. |
41 class CPDF_TestParser : public CPDF_Parser { | 41 class CPDF_TestParser : public CPDF_Parser { |
42 public: | 42 public: |
43 CPDF_TestParser() {} | 43 CPDF_TestParser() {} |
44 ~CPDF_TestParser() {} | 44 ~CPDF_TestParser() {} |
45 | 45 |
46 // Setup reading from a file and initial states. | 46 // Setup reading from a file and initial states. |
47 bool InitTestFromFile(const FX_CHAR* path) { | 47 bool InitTestFromFile(const FX_CHAR* path) { |
48 IFX_FileRead* pFileAccess = FX_CreateFileRead(path); | 48 IFX_SeekableReadStream* pFileAccess = FX_CreateFileRead(path); |
49 if (!pFileAccess) | 49 if (!pFileAccess) |
50 return false; | 50 return false; |
51 | 51 |
52 // For the test file, the header is set at the beginning. | 52 // For the test file, the header is set at the beginning. |
53 m_pSyntax->InitParser(pFileAccess, 0); | 53 m_pSyntax->InitParser(pFileAccess, 0); |
54 return true; | 54 return true; |
55 } | 55 } |
56 | 56 |
57 // Setup reading from a buffer and initial states. | 57 // Setup reading from a buffer and initial states. |
58 bool InitTestFromBuffer(const unsigned char* buffer, size_t len) { | 58 bool InitTestFromBuffer(const unsigned char* buffer, size_t len) { |
(...skipping 134 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
193 | 193 |
194 ASSERT_TRUE(parser.LoadCrossRefV4(0, 0, FALSE)); | 194 ASSERT_TRUE(parser.LoadCrossRefV4(0, 0, FALSE)); |
195 const FX_FILESIZE offsets[] = {0, 23, 0, 0, 0, 45, 179}; | 195 const FX_FILESIZE offsets[] = {0, 23, 0, 0, 0, 45, 179}; |
196 const uint8_t types[] = {0, 1, 0, 0, 0, 1, 1}; | 196 const uint8_t types[] = {0, 1, 0, 0, 0, 1, 1}; |
197 for (size_t i = 0; i < FX_ArraySize(offsets); ++i) { | 197 for (size_t i = 0; i < FX_ArraySize(offsets); ++i) { |
198 EXPECT_EQ(offsets[i], parser.m_ObjectInfo[i].pos); | 198 EXPECT_EQ(offsets[i], parser.m_ObjectInfo[i].pos); |
199 EXPECT_EQ(types[i], parser.m_ObjectInfo[i].type); | 199 EXPECT_EQ(types[i], parser.m_ObjectInfo[i].type); |
200 } | 200 } |
201 } | 201 } |
202 } | 202 } |
OLD | NEW |