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 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
6 | 6 |
7 #include "core/fpdfapi/page/pageint.h" | 7 #include "core/fpdfapi/page/pageint.h" |
8 | 8 |
9 #include <limits.h> | 9 #include <limits.h> |
10 | 10 |
| 11 #include <memory> |
11 #include <utility> | 12 #include <utility> |
12 | 13 |
13 #include "core/fpdfapi/cpdf_modulemgr.h" | 14 #include "core/fpdfapi/cpdf_modulemgr.h" |
14 #include "core/fpdfapi/page/cpdf_docpagedata.h" | 15 #include "core/fpdfapi/page/cpdf_docpagedata.h" |
15 #include "core/fpdfapi/parser/cpdf_array.h" | 16 #include "core/fpdfapi/parser/cpdf_array.h" |
16 #include "core/fpdfapi/parser/cpdf_boolean.h" | 17 #include "core/fpdfapi/parser/cpdf_boolean.h" |
17 #include "core/fpdfapi/parser/cpdf_dictionary.h" | 18 #include "core/fpdfapi/parser/cpdf_dictionary.h" |
18 #include "core/fpdfapi/parser/cpdf_document.h" | 19 #include "core/fpdfapi/parser/cpdf_document.h" |
19 #include "core/fpdfapi/parser/cpdf_name.h" | 20 #include "core/fpdfapi/parser/cpdf_name.h" |
20 #include "core/fpdfapi/parser/cpdf_null.h" | 21 #include "core/fpdfapi/parser/cpdf_null.h" |
(...skipping 84 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
105 dest_buf = 0; | 106 dest_buf = 0; |
106 return (uint32_t)-1; | 107 return (uint32_t)-1; |
107 } | 108 } |
108 | 109 |
109 } // namespace | 110 } // namespace |
110 | 111 |
111 CPDF_StreamParser::CPDF_StreamParser(const uint8_t* pData, uint32_t dwSize) | 112 CPDF_StreamParser::CPDF_StreamParser(const uint8_t* pData, uint32_t dwSize) |
112 : m_pBuf(pData), | 113 : m_pBuf(pData), |
113 m_Size(dwSize), | 114 m_Size(dwSize), |
114 m_Pos(0), | 115 m_Pos(0), |
115 m_pLastObj(nullptr), | |
116 m_pPool(nullptr) {} | 116 m_pPool(nullptr) {} |
117 | 117 |
118 CPDF_StreamParser::CPDF_StreamParser( | 118 CPDF_StreamParser::CPDF_StreamParser( |
119 const uint8_t* pData, | 119 const uint8_t* pData, |
120 uint32_t dwSize, | 120 uint32_t dwSize, |
121 const CFX_WeakPtr<CFX_ByteStringPool>& pPool) | 121 const CFX_WeakPtr<CFX_ByteStringPool>& pPool) |
122 : m_pBuf(pData), | 122 : m_pBuf(pData), |
123 m_Size(dwSize), | 123 m_Size(dwSize), |
124 m_Pos(0), | 124 m_Pos(0), |
125 m_pLastObj(nullptr), | |
126 m_pPool(pPool) {} | 125 m_pPool(pPool) {} |
127 | 126 |
128 CPDF_StreamParser::~CPDF_StreamParser() { | 127 CPDF_StreamParser::~CPDF_StreamParser() {} |
129 delete m_pLastObj; | |
130 } | |
131 | 128 |
132 CPDF_Stream* CPDF_StreamParser::ReadInlineStream(CPDF_Document* pDoc, | 129 std::unique_ptr<CPDF_Stream> CPDF_StreamParser::ReadInlineStream( |
133 CPDF_Dictionary* pDict, | 130 CPDF_Document* pDoc, |
134 CPDF_Object* pCSObj) { | 131 CPDF_Dictionary* pDict, |
| 132 CPDF_Object* pCSObj) { |
135 if (m_Pos == m_Size) | 133 if (m_Pos == m_Size) |
136 return nullptr; | 134 return nullptr; |
137 | 135 |
138 if (PDFCharIsWhitespace(m_pBuf[m_Pos])) | 136 if (PDFCharIsWhitespace(m_pBuf[m_Pos])) |
139 m_Pos++; | 137 m_Pos++; |
140 | 138 |
141 CFX_ByteString Decoder; | 139 CFX_ByteString Decoder; |
142 CPDF_Dictionary* pParam = nullptr; | 140 CPDF_Dictionary* pParam = nullptr; |
143 CPDF_Object* pFilter = pDict->GetDirectObjectFor("Filter"); | 141 CPDF_Object* pFilter = pDict->GetDirectObjectFor("Filter"); |
144 if (pFilter) { | 142 if (pFilter) { |
(...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
226 break; | 224 break; |
227 } | 225 } |
228 dwStreamSize += m_Pos - dwPrevPos; | 226 dwStreamSize += m_Pos - dwPrevPos; |
229 } | 227 } |
230 m_Pos = dwSavePos; | 228 m_Pos = dwSavePos; |
231 pData = FX_Alloc(uint8_t, dwStreamSize); | 229 pData = FX_Alloc(uint8_t, dwStreamSize); |
232 FXSYS_memcpy(pData, m_pBuf + m_Pos, dwStreamSize); | 230 FXSYS_memcpy(pData, m_pBuf + m_Pos, dwStreamSize); |
233 m_Pos += dwStreamSize; | 231 m_Pos += dwStreamSize; |
234 } | 232 } |
235 pDict->SetNewFor<CPDF_Number>("Length", (int)dwStreamSize); | 233 pDict->SetNewFor<CPDF_Number>("Length", (int)dwStreamSize); |
236 return new CPDF_Stream(pData, dwStreamSize, pdfium::WrapUnique(pDict)); | 234 return pdfium::MakeUnique<CPDF_Stream>(pData, dwStreamSize, |
| 235 pdfium::WrapUnique(pDict)); |
237 } | 236 } |
238 | 237 |
239 CPDF_StreamParser::SyntaxType CPDF_StreamParser::ParseNextElement() { | 238 CPDF_StreamParser::SyntaxType CPDF_StreamParser::ParseNextElement() { |
240 delete m_pLastObj; | 239 m_pLastObj.reset(); |
241 m_pLastObj = nullptr; | |
242 | |
243 m_WordSize = 0; | 240 m_WordSize = 0; |
244 bool bIsNumber = true; | |
245 if (!PositionIsInBounds()) | 241 if (!PositionIsInBounds()) |
246 return EndOfData; | 242 return EndOfData; |
247 | 243 |
248 int ch = m_pBuf[m_Pos++]; | 244 int ch = m_pBuf[m_Pos++]; |
249 while (1) { | 245 while (1) { |
250 while (PDFCharIsWhitespace(ch)) { | 246 while (PDFCharIsWhitespace(ch)) { |
251 if (!PositionIsInBounds()) | 247 if (!PositionIsInBounds()) |
252 return EndOfData; | 248 return EndOfData; |
253 | 249 |
254 ch = m_pBuf[m_Pos++]; | 250 ch = m_pBuf[m_Pos++]; |
(...skipping 11 matching lines...) Expand all Loading... |
266 break; | 262 break; |
267 } | 263 } |
268 } | 264 } |
269 | 265 |
270 if (PDFCharIsDelimiter(ch) && ch != '/') { | 266 if (PDFCharIsDelimiter(ch) && ch != '/') { |
271 m_Pos--; | 267 m_Pos--; |
272 m_pLastObj = ReadNextObject(false, 0); | 268 m_pLastObj = ReadNextObject(false, 0); |
273 return Others; | 269 return Others; |
274 } | 270 } |
275 | 271 |
| 272 bool bIsNumber = true; |
276 while (1) { | 273 while (1) { |
277 if (m_WordSize < kMaxWordBuffer) | 274 if (m_WordSize < kMaxWordBuffer) |
278 m_WordBuffer[m_WordSize++] = ch; | 275 m_WordBuffer[m_WordSize++] = ch; |
279 | 276 |
280 if (!PDFCharIsNumeric(ch)) | 277 if (!PDFCharIsNumeric(ch)) |
281 bIsNumber = false; | 278 bIsNumber = false; |
282 | 279 |
283 if (!PositionIsInBounds()) | 280 if (!PositionIsInBounds()) |
284 break; | 281 break; |
285 | 282 |
286 ch = m_pBuf[m_Pos++]; | 283 ch = m_pBuf[m_Pos++]; |
287 | 284 |
288 if (PDFCharIsDelimiter(ch) || PDFCharIsWhitespace(ch)) { | 285 if (PDFCharIsDelimiter(ch) || PDFCharIsWhitespace(ch)) { |
289 m_Pos--; | 286 m_Pos--; |
290 break; | 287 break; |
291 } | 288 } |
292 } | 289 } |
293 | 290 |
294 m_WordBuffer[m_WordSize] = 0; | 291 m_WordBuffer[m_WordSize] = 0; |
295 if (bIsNumber) | 292 if (bIsNumber) |
296 return Number; | 293 return Number; |
297 | 294 |
298 if (m_WordBuffer[0] == '/') | 295 if (m_WordBuffer[0] == '/') |
299 return Name; | 296 return Name; |
300 | 297 |
301 if (m_WordSize == 4) { | 298 if (m_WordSize == 4) { |
302 if (memcmp(m_WordBuffer, "true", 4) == 0) { | 299 if (memcmp(m_WordBuffer, "true", 4) == 0) { |
303 m_pLastObj = new CPDF_Boolean(true); | 300 m_pLastObj = pdfium::MakeUnique<CPDF_Boolean>(true); |
304 return Others; | 301 return Others; |
305 } | 302 } |
306 if (memcmp(m_WordBuffer, "null", 4) == 0) { | 303 if (memcmp(m_WordBuffer, "null", 4) == 0) { |
307 m_pLastObj = new CPDF_Null; | 304 m_pLastObj = pdfium::MakeUnique<CPDF_Null>(); |
308 return Others; | 305 return Others; |
309 } | 306 } |
310 } else if (m_WordSize == 5) { | 307 } else if (m_WordSize == 5) { |
311 if (memcmp(m_WordBuffer, "false", 5) == 0) { | 308 if (memcmp(m_WordBuffer, "false", 5) == 0) { |
312 m_pLastObj = new CPDF_Boolean(false); | 309 m_pLastObj = pdfium::MakeUnique<CPDF_Boolean>(false); |
313 return Others; | 310 return Others; |
314 } | 311 } |
315 } | 312 } |
316 return Keyword; | 313 return Keyword; |
317 } | 314 } |
318 | 315 |
319 CPDF_Object* CPDF_StreamParser::GetObject() { | 316 std::unique_ptr<CPDF_Object> CPDF_StreamParser::ReadNextObject( |
320 CPDF_Object* pObj = m_pLastObj; | 317 bool bAllowNestedArray, |
321 m_pLastObj = nullptr; | 318 uint32_t dwInArrayLevel) { |
322 return pObj; | |
323 } | |
324 | |
325 CPDF_Object* CPDF_StreamParser::ReadNextObject(bool bAllowNestedArray, | |
326 uint32_t dwInArrayLevel) { | |
327 bool bIsNumber; | 319 bool bIsNumber; |
328 GetNextWord(bIsNumber); | 320 GetNextWord(bIsNumber); |
329 if (!m_WordSize) | 321 if (!m_WordSize) |
330 return nullptr; | 322 return nullptr; |
331 | 323 |
332 if (bIsNumber) { | 324 if (bIsNumber) { |
333 m_WordBuffer[m_WordSize] = 0; | 325 m_WordBuffer[m_WordSize] = 0; |
334 return new CPDF_Number(CFX_ByteStringC(m_WordBuffer, m_WordSize)); | 326 return pdfium::MakeUnique<CPDF_Number>( |
| 327 CFX_ByteStringC(m_WordBuffer, m_WordSize)); |
335 } | 328 } |
336 | 329 |
337 int first_char = m_WordBuffer[0]; | 330 int first_char = m_WordBuffer[0]; |
338 if (first_char == '/') { | 331 if (first_char == '/') { |
339 CFX_ByteString name = | 332 CFX_ByteString name = |
340 PDF_NameDecode(CFX_ByteStringC(m_WordBuffer + 1, m_WordSize - 1)); | 333 PDF_NameDecode(CFX_ByteStringC(m_WordBuffer + 1, m_WordSize - 1)); |
341 return new CPDF_Name(m_pPool, name); | 334 return pdfium::MakeUnique<CPDF_Name>(m_pPool, name); |
342 } | 335 } |
343 | 336 |
344 if (first_char == '(') { | 337 if (first_char == '(') { |
345 CFX_ByteString str = ReadString(); | 338 CFX_ByteString str = ReadString(); |
346 return new CPDF_String(m_pPool, str, false); | 339 return pdfium::MakeUnique<CPDF_String>(m_pPool, str, false); |
347 } | 340 } |
348 | 341 |
349 if (first_char == '<') { | 342 if (first_char == '<') { |
350 if (m_WordSize == 1) | 343 if (m_WordSize == 1) |
351 return new CPDF_String(m_pPool, ReadHexString(), true); | 344 return pdfium::MakeUnique<CPDF_String>(m_pPool, ReadHexString(), true); |
352 | 345 |
353 CPDF_Dictionary* pDict = new CPDF_Dictionary(m_pPool); | 346 auto pDict = pdfium::MakeUnique<CPDF_Dictionary>(m_pPool); |
354 while (1) { | 347 while (1) { |
355 GetNextWord(bIsNumber); | 348 GetNextWord(bIsNumber); |
356 if (m_WordSize == 2 && m_WordBuffer[0] == '>') | 349 if (m_WordSize == 2 && m_WordBuffer[0] == '>') |
357 break; | 350 break; |
358 | 351 |
359 if (!m_WordSize || m_WordBuffer[0] != '/') { | 352 if (!m_WordSize || m_WordBuffer[0] != '/') |
360 delete pDict; | |
361 return nullptr; | 353 return nullptr; |
362 } | |
363 | 354 |
364 CFX_ByteString key = | 355 CFX_ByteString key = |
365 PDF_NameDecode(CFX_ByteStringC(m_WordBuffer + 1, m_WordSize - 1)); | 356 PDF_NameDecode(CFX_ByteStringC(m_WordBuffer + 1, m_WordSize - 1)); |
366 auto pObj = pdfium::WrapUnique(ReadNextObject(true, 0)); | 357 std::unique_ptr<CPDF_Object> pObj = ReadNextObject(true, 0); |
367 if (!pObj) { | 358 if (!pObj) |
368 delete pDict; | |
369 return nullptr; | 359 return nullptr; |
370 } | 360 |
371 if (!key.IsEmpty()) | 361 if (!key.IsEmpty()) |
372 pDict->SetFor(key, std::move(pObj)); | 362 pDict->SetFor(key, std::move(pObj)); |
373 } | 363 } |
374 return pDict; | 364 return std::move(pDict); |
375 } | 365 } |
376 | 366 |
377 if (first_char == '[') { | 367 if (first_char == '[') { |
378 if ((!bAllowNestedArray && dwInArrayLevel) || | 368 if ((!bAllowNestedArray && dwInArrayLevel) || |
379 dwInArrayLevel > kMaxNestedArrayLevel) { | 369 dwInArrayLevel > kMaxNestedArrayLevel) { |
380 return nullptr; | 370 return nullptr; |
381 } | 371 } |
382 | 372 |
383 CPDF_Array* pArray = new CPDF_Array; | 373 auto pArray = pdfium::MakeUnique<CPDF_Array>(); |
384 while (1) { | 374 while (1) { |
385 CPDF_Object* pObj = ReadNextObject(bAllowNestedArray, dwInArrayLevel + 1); | 375 std::unique_ptr<CPDF_Object> pObj = |
| 376 ReadNextObject(bAllowNestedArray, dwInArrayLevel + 1); |
386 if (pObj) { | 377 if (pObj) { |
387 pArray->Add(pdfium::WrapUnique(pObj)); | 378 pArray->Add(std::move(pObj)); |
388 continue; | 379 continue; |
389 } | 380 } |
390 if (!m_WordSize || m_WordBuffer[0] == ']') | 381 if (!m_WordSize || m_WordBuffer[0] == ']') |
391 break; | 382 break; |
392 } | 383 } |
393 return pArray; | 384 return std::move(pArray); |
394 } | 385 } |
395 | 386 |
396 if (m_WordSize == 5 && !memcmp(m_WordBuffer, "false", 5)) | 387 if (m_WordSize == 5 && !memcmp(m_WordBuffer, "false", 5)) |
397 return new CPDF_Boolean(false); | 388 return pdfium::MakeUnique<CPDF_Boolean>(false); |
398 | 389 |
399 if (m_WordSize == 4) { | 390 if (m_WordSize == 4) { |
400 if (memcmp(m_WordBuffer, "true", 4) == 0) | 391 if (memcmp(m_WordBuffer, "true", 4) == 0) |
401 return new CPDF_Boolean(true); | 392 return pdfium::MakeUnique<CPDF_Boolean>(true); |
402 | |
403 if (memcmp(m_WordBuffer, "null", 4) == 0) | 393 if (memcmp(m_WordBuffer, "null", 4) == 0) |
404 return new CPDF_Null; | 394 return pdfium::MakeUnique<CPDF_Null>(); |
405 } | 395 } |
406 | 396 |
407 return nullptr; | 397 return nullptr; |
408 } | 398 } |
409 | 399 |
410 // TODO(npm): the following methods are almost identical in cpdf_syntaxparser | 400 // TODO(npm): the following methods are almost identical in cpdf_syntaxparser |
411 void CPDF_StreamParser::GetNextWord(bool& bIsNumber) { | 401 void CPDF_StreamParser::GetNextWord(bool& bIsNumber) { |
412 m_WordSize = 0; | 402 m_WordSize = 0; |
413 bIsNumber = true; | 403 bIsNumber = true; |
414 if (!PositionIsInBounds()) | 404 if (!PositionIsInBounds()) |
(...skipping 201 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
616 | 606 |
617 if (buf.GetLength() > kMaxStringLength) | 607 if (buf.GetLength() > kMaxStringLength) |
618 return CFX_ByteString(buf.GetBuffer(), kMaxStringLength); | 608 return CFX_ByteString(buf.GetBuffer(), kMaxStringLength); |
619 | 609 |
620 return buf.MakeString(); | 610 return buf.MakeString(); |
621 } | 611 } |
622 | 612 |
623 bool CPDF_StreamParser::PositionIsInBounds() const { | 613 bool CPDF_StreamParser::PositionIsInBounds() const { |
624 return m_Pos < m_Size; | 614 return m_Pos < m_Size; |
625 } | 615 } |
OLD | NEW |