| OLD | NEW |
| 1 // Copyright 2016 PDFium Authors. All rights reserved. | 1 // Copyright 2016 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 TEST(cpdf_syntax_parser, ReadHexString) { | 15 TEST(cpdf_syntax_parser, ReadHexString) { |
| 16 { | 16 { |
| 17 // Empty string. | 17 // Empty string. |
| 18 uint8_t data[] = ""; | 18 uint8_t data[] = ""; |
| 19 ScopedFileStream stream(IFX_MemoryStream::Create(data, 0, false)); | |
| 20 | |
| 21 CPDF_SyntaxParser parser; | 19 CPDF_SyntaxParser parser; |
| 22 parser.InitParser(stream.get(), 0); | 20 parser.InitParser(IFX_MemoryStream::Create(data, 0, false), 0); |
| 23 EXPECT_EQ("", parser.ReadHexString()); | 21 EXPECT_EQ("", parser.ReadHexString()); |
| 24 EXPECT_EQ(0, parser.SavePos()); | 22 EXPECT_EQ(0, parser.SavePos()); |
| 25 } | 23 } |
| 26 | 24 |
| 27 { | 25 { |
| 28 // Blank string. | 26 // Blank string. |
| 29 uint8_t data[] = " "; | 27 uint8_t data[] = " "; |
| 30 ScopedFileStream stream(IFX_MemoryStream::Create(data, 2, false)); | |
| 31 | |
| 32 CPDF_SyntaxParser parser; | 28 CPDF_SyntaxParser parser; |
| 33 parser.InitParser(stream.get(), 0); | 29 parser.InitParser(IFX_MemoryStream::Create(data, 2, false), 0); |
| 34 EXPECT_EQ("", parser.ReadHexString()); | 30 EXPECT_EQ("", parser.ReadHexString()); |
| 35 EXPECT_EQ(2, parser.SavePos()); | 31 EXPECT_EQ(2, parser.SavePos()); |
| 36 } | 32 } |
| 37 | 33 |
| 38 { | 34 { |
| 39 // Skips unknown characters. | 35 // Skips unknown characters. |
| 40 uint8_t data[] = "z12b"; | 36 uint8_t data[] = "z12b"; |
| 41 ScopedFileStream stream(IFX_MemoryStream::Create(data, 4, false)); | |
| 42 | |
| 43 CPDF_SyntaxParser parser; | 37 CPDF_SyntaxParser parser; |
| 44 parser.InitParser(stream.get(), 0); | 38 parser.InitParser(IFX_MemoryStream::Create(data, 4, false), 0); |
| 45 EXPECT_EQ("\x12\xb0", parser.ReadHexString()); | 39 EXPECT_EQ("\x12\xb0", parser.ReadHexString()); |
| 46 EXPECT_EQ(4, parser.SavePos()); | 40 EXPECT_EQ(4, parser.SavePos()); |
| 47 } | 41 } |
| 48 | 42 |
| 49 { | 43 { |
| 50 // Skips unknown characters. | 44 // Skips unknown characters. |
| 51 uint8_t data[] = "*<&*#$^&@1"; | 45 uint8_t data[] = "*<&*#$^&@1"; |
| 52 ScopedFileStream stream(IFX_MemoryStream::Create(data, 10, false)); | |
| 53 | |
| 54 CPDF_SyntaxParser parser; | 46 CPDF_SyntaxParser parser; |
| 55 parser.InitParser(stream.get(), 0); | 47 parser.InitParser(IFX_MemoryStream::Create(data, 10, false), 0); |
| 56 EXPECT_EQ("\x10", parser.ReadHexString()); | 48 EXPECT_EQ("\x10", parser.ReadHexString()); |
| 57 EXPECT_EQ(10, parser.SavePos()); | 49 EXPECT_EQ(10, parser.SavePos()); |
| 58 } | 50 } |
| 59 | 51 |
| 60 { | 52 { |
| 61 // Skips unknown characters. | 53 // Skips unknown characters. |
| 62 uint8_t data[] = "\x80zab"; | 54 uint8_t data[] = "\x80zab"; |
| 63 ScopedFileStream stream(IFX_MemoryStream::Create(data, 4, false)); | |
| 64 | |
| 65 CPDF_SyntaxParser parser; | 55 CPDF_SyntaxParser parser; |
| 66 parser.InitParser(stream.get(), 0); | 56 parser.InitParser(IFX_MemoryStream::Create(data, 4, false), 0); |
| 67 EXPECT_EQ("\xab", parser.ReadHexString()); | 57 EXPECT_EQ("\xab", parser.ReadHexString()); |
| 68 EXPECT_EQ(4, parser.SavePos()); | 58 EXPECT_EQ(4, parser.SavePos()); |
| 69 } | 59 } |
| 70 | 60 |
| 71 { | 61 { |
| 72 // Skips unknown characters. | 62 // Skips unknown characters. |
| 73 uint8_t data[] = "\xffzab"; | 63 uint8_t data[] = "\xffzab"; |
| 74 ScopedFileStream stream(IFX_MemoryStream::Create(data, 4, false)); | |
| 75 | |
| 76 CPDF_SyntaxParser parser; | 64 CPDF_SyntaxParser parser; |
| 77 parser.InitParser(stream.get(), 0); | 65 parser.InitParser(IFX_MemoryStream::Create(data, 4, false), 0); |
| 78 EXPECT_EQ("\xab", parser.ReadHexString()); | 66 EXPECT_EQ("\xab", parser.ReadHexString()); |
| 79 EXPECT_EQ(4, parser.SavePos()); | 67 EXPECT_EQ(4, parser.SavePos()); |
| 80 } | 68 } |
| 81 | 69 |
| 82 { | 70 { |
| 83 // Regular conversion. | 71 // Regular conversion. |
| 84 uint8_t data[] = "1A2b>abcd"; | 72 uint8_t data[] = "1A2b>abcd"; |
| 85 ScopedFileStream stream(IFX_MemoryStream::Create(data, 9, false)); | |
| 86 | |
| 87 CPDF_SyntaxParser parser; | 73 CPDF_SyntaxParser parser; |
| 88 parser.InitParser(stream.get(), 0); | 74 parser.InitParser(IFX_MemoryStream::Create(data, 9, false), 0); |
| 89 EXPECT_EQ("\x1a\x2b", parser.ReadHexString()); | 75 EXPECT_EQ("\x1a\x2b", parser.ReadHexString()); |
| 90 EXPECT_EQ(5, parser.SavePos()); | 76 EXPECT_EQ(5, parser.SavePos()); |
| 91 } | 77 } |
| 92 | 78 |
| 93 { | 79 { |
| 94 // Position out of bounds. | 80 // Position out of bounds. |
| 95 uint8_t data[] = "12ab>"; | 81 uint8_t data[] = "12ab>"; |
| 96 ScopedFileStream stream(IFX_MemoryStream::Create(data, 5, false)); | |
| 97 | |
| 98 CPDF_SyntaxParser parser; | 82 CPDF_SyntaxParser parser; |
| 99 parser.InitParser(stream.get(), 0); | 83 parser.InitParser(IFX_MemoryStream::Create(data, 5, false), 0); |
| 100 parser.RestorePos(5); | 84 parser.RestorePos(5); |
| 101 EXPECT_EQ("", parser.ReadHexString()); | 85 EXPECT_EQ("", parser.ReadHexString()); |
| 102 | 86 |
| 103 parser.RestorePos(6); | 87 parser.RestorePos(6); |
| 104 EXPECT_EQ("", parser.ReadHexString()); | 88 EXPECT_EQ("", parser.ReadHexString()); |
| 105 | 89 |
| 106 parser.RestorePos(-1); | 90 parser.RestorePos(-1); |
| 107 EXPECT_EQ("", parser.ReadHexString()); | 91 EXPECT_EQ("", parser.ReadHexString()); |
| 108 | 92 |
| 109 parser.RestorePos(std::numeric_limits<FX_FILESIZE>::max()); | 93 parser.RestorePos(std::numeric_limits<FX_FILESIZE>::max()); |
| 110 EXPECT_EQ("", parser.ReadHexString()); | 94 EXPECT_EQ("", parser.ReadHexString()); |
| 111 | 95 |
| 112 // Check string still parses when set to 0. | 96 // Check string still parses when set to 0. |
| 113 parser.RestorePos(0); | 97 parser.RestorePos(0); |
| 114 EXPECT_EQ("\x12\xab", parser.ReadHexString()); | 98 EXPECT_EQ("\x12\xab", parser.ReadHexString()); |
| 115 } | 99 } |
| 116 | 100 |
| 117 { | 101 { |
| 118 // Missing ending >. | 102 // Missing ending >. |
| 119 uint8_t data[] = "1A2b"; | 103 uint8_t data[] = "1A2b"; |
| 120 ScopedFileStream stream(IFX_MemoryStream::Create(data, 4, false)); | |
| 121 | |
| 122 CPDF_SyntaxParser parser; | 104 CPDF_SyntaxParser parser; |
| 123 parser.InitParser(stream.get(), 0); | 105 parser.InitParser(IFX_MemoryStream::Create(data, 4, false), 0); |
| 124 EXPECT_EQ("\x1a\x2b", parser.ReadHexString()); | 106 EXPECT_EQ("\x1a\x2b", parser.ReadHexString()); |
| 125 EXPECT_EQ(4, parser.SavePos()); | 107 EXPECT_EQ(4, parser.SavePos()); |
| 126 } | 108 } |
| 127 | 109 |
| 128 { | 110 { |
| 129 // Missing ending >. | 111 // Missing ending >. |
| 130 uint8_t data[] = "12abz"; | 112 uint8_t data[] = "12abz"; |
| 131 ScopedFileStream stream(IFX_MemoryStream::Create(data, 5, false)); | |
| 132 | |
| 133 CPDF_SyntaxParser parser; | 113 CPDF_SyntaxParser parser; |
| 134 parser.InitParser(stream.get(), 0); | 114 parser.InitParser(IFX_MemoryStream::Create(data, 5, false), 0); |
| 135 EXPECT_EQ("\x12\xab", parser.ReadHexString()); | 115 EXPECT_EQ("\x12\xab", parser.ReadHexString()); |
| 136 EXPECT_EQ(5, parser.SavePos()); | 116 EXPECT_EQ(5, parser.SavePos()); |
| 137 } | 117 } |
| 138 | 118 |
| 139 { | 119 { |
| 140 // Uneven number of bytes. | 120 // Uneven number of bytes. |
| 141 uint8_t data[] = "1A2>asdf"; | 121 uint8_t data[] = "1A2>asdf"; |
| 142 ScopedFileStream stream(IFX_MemoryStream::Create(data, 8, false)); | |
| 143 | |
| 144 CPDF_SyntaxParser parser; | 122 CPDF_SyntaxParser parser; |
| 145 parser.InitParser(stream.get(), 0); | 123 parser.InitParser(IFX_MemoryStream::Create(data, 8, false), 0); |
| 146 EXPECT_EQ("\x1a\x20", parser.ReadHexString()); | 124 EXPECT_EQ("\x1a\x20", parser.ReadHexString()); |
| 147 EXPECT_EQ(4, parser.SavePos()); | 125 EXPECT_EQ(4, parser.SavePos()); |
| 148 } | 126 } |
| 149 | 127 |
| 150 { | 128 { |
| 151 // Uneven number of bytes. | 129 // Uneven number of bytes. |
| 152 uint8_t data[] = "1A2zasdf"; | 130 uint8_t data[] = "1A2zasdf"; |
| 153 ScopedFileStream stream(IFX_MemoryStream::Create(data, 8, false)); | |
| 154 | |
| 155 CPDF_SyntaxParser parser; | 131 CPDF_SyntaxParser parser; |
| 156 parser.InitParser(stream.get(), 0); | 132 parser.InitParser(IFX_MemoryStream::Create(data, 8, false), 0); |
| 157 EXPECT_EQ("\x1a\x2a\xdf", parser.ReadHexString()); | 133 EXPECT_EQ("\x1a\x2a\xdf", parser.ReadHexString()); |
| 158 EXPECT_EQ(8, parser.SavePos()); | 134 EXPECT_EQ(8, parser.SavePos()); |
| 159 } | 135 } |
| 160 | 136 |
| 161 { | 137 { |
| 162 // Just ending character. | 138 // Just ending character. |
| 163 uint8_t data[] = ">"; | 139 uint8_t data[] = ">"; |
| 164 ScopedFileStream stream(IFX_MemoryStream::Create(data, 1, false)); | |
| 165 | |
| 166 CPDF_SyntaxParser parser; | 140 CPDF_SyntaxParser parser; |
| 167 parser.InitParser(stream.get(), 0); | 141 parser.InitParser(IFX_MemoryStream::Create(data, 1, false), 0); |
| 168 EXPECT_EQ("", parser.ReadHexString()); | 142 EXPECT_EQ("", parser.ReadHexString()); |
| 169 EXPECT_EQ(1, parser.SavePos()); | 143 EXPECT_EQ(1, parser.SavePos()); |
| 170 } | 144 } |
| 171 } | 145 } |
| OLD | NEW |