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_SeekableReadStream { | 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), current_pos_(0) {} |
20 | 20 |
21 // IFX_Stream | 21 // IFX_Stream |
22 void Release() override { delete this; } | 22 void Release() override { delete this; } |
23 | 23 |
24 // IFX_SeekableReadStream | 24 // IFX_SeekableReadStream |
25 bool ReadBlock(void* buffer, FX_FILESIZE offset, size_t size) override { | 25 bool IsEOF() override { return current_pos_ >= total_size_; } |
26 if (offset < 0 || offset + size > total_size_) | |
27 return false; | |
28 | 26 |
| 27 size_t ReadBlock(void* buffer, FX_FILESIZE offset, size_t size) override { |
| 28 if (offset < 0 || size > total_size_ || |
| 29 offset > total_size_ - pdfium::base::checked_cast<FX_FILESIZE>(size)) { |
| 30 return 0; |
| 31 } |
29 memcpy(buffer, buffer_ + offset, size); | 32 memcpy(buffer, buffer_ + offset, size); |
30 return true; | 33 current_pos_ = offset + pdfium::base::checked_cast<FX_FILESIZE>(size); |
| 34 return size; |
31 } | 35 } |
32 | 36 |
33 FX_FILESIZE GetSize() override { return (FX_FILESIZE)total_size_; }; | 37 FX_FILESIZE GetSize() override { return total_size_; }; |
34 | 38 |
35 protected: | 39 protected: |
36 const unsigned char* buffer_; | 40 const unsigned char* buffer_; |
37 size_t total_size_; | 41 FX_FILESIZE total_size_; |
| 42 FX_FILESIZE current_pos_; |
38 }; | 43 }; |
39 | 44 |
40 // A wrapper class to help test member functions of CPDF_Parser. | 45 // A wrapper class to help test member functions of CPDF_Parser. |
41 class CPDF_TestParser : public CPDF_Parser { | 46 class CPDF_TestParser : public CPDF_Parser { |
42 public: | 47 public: |
43 CPDF_TestParser() {} | 48 CPDF_TestParser() {} |
44 ~CPDF_TestParser() {} | 49 ~CPDF_TestParser() {} |
45 | 50 |
46 // Setup reading from a file and initial states. | 51 // Setup reading from a file and initial states. |
47 bool InitTestFromFile(const FX_CHAR* path) { | 52 bool InitTestFromFile(const FX_CHAR* path) { |
(...skipping 145 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
193 | 198 |
194 ASSERT_TRUE(parser.LoadCrossRefV4(0, 0, false)); | 199 ASSERT_TRUE(parser.LoadCrossRefV4(0, 0, false)); |
195 const FX_FILESIZE offsets[] = {0, 23, 0, 0, 0, 45, 179}; | 200 const FX_FILESIZE offsets[] = {0, 23, 0, 0, 0, 45, 179}; |
196 const uint8_t types[] = {0, 1, 0, 0, 0, 1, 1}; | 201 const uint8_t types[] = {0, 1, 0, 0, 0, 1, 1}; |
197 for (size_t i = 0; i < FX_ArraySize(offsets); ++i) { | 202 for (size_t i = 0; i < FX_ArraySize(offsets); ++i) { |
198 EXPECT_EQ(offsets[i], parser.m_ObjectInfo[i].pos); | 203 EXPECT_EQ(offsets[i], parser.m_ObjectInfo[i].pos); |
199 EXPECT_EQ(types[i], parser.m_ObjectInfo[i].type); | 204 EXPECT_EQ(types[i], parser.m_ObjectInfo[i].type); |
200 } | 205 } |
201 } | 206 } |
202 } | 207 } |
OLD | NEW |