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/cpdf_image.h" | 7 #include "core/fpdfapi/page/cpdf_image.h" |
8 | 8 |
9 #include <algorithm> | 9 #include <algorithm> |
10 #include <memory> | 10 #include <memory> |
11 #include <utility> | 11 #include <utility> |
12 #include <vector> | 12 #include <vector> |
13 | 13 |
14 #include "core/fpdfapi/cpdf_modulemgr.h" | 14 #include "core/fpdfapi/cpdf_modulemgr.h" |
15 #include "core/fpdfapi/page/cpdf_page.h" | 15 #include "core/fpdfapi/page/cpdf_page.h" |
16 #include "core/fpdfapi/parser/cpdf_array.h" | 16 #include "core/fpdfapi/parser/cpdf_array.h" |
17 #include "core/fpdfapi/parser/cpdf_boolean.h" | 17 #include "core/fpdfapi/parser/cpdf_boolean.h" |
18 #include "core/fpdfapi/parser/cpdf_dictionary.h" | 18 #include "core/fpdfapi/parser/cpdf_dictionary.h" |
19 #include "core/fpdfapi/parser/cpdf_document.h" | 19 #include "core/fpdfapi/parser/cpdf_document.h" |
20 #include "core/fpdfapi/parser/cpdf_name.h" | 20 #include "core/fpdfapi/parser/cpdf_name.h" |
21 #include "core/fpdfapi/parser/cpdf_number.h" | 21 #include "core/fpdfapi/parser/cpdf_number.h" |
22 #include "core/fpdfapi/parser/cpdf_reference.h" | 22 #include "core/fpdfapi/parser/cpdf_reference.h" |
23 #include "core/fpdfapi/parser/cpdf_stream.h" | 23 #include "core/fpdfapi/parser/cpdf_stream.h" |
24 #include "core/fpdfapi/parser/cpdf_string.h" | 24 #include "core/fpdfapi/parser/cpdf_string.h" |
25 #include "core/fpdfapi/render/cpdf_pagerendercache.h" | 25 #include "core/fpdfapi/render/cpdf_pagerendercache.h" |
26 #include "core/fpdfapi/render/render_int.h" | 26 #include "core/fpdfapi/render/render_int.h" |
27 #include "core/fxcodec/fx_codec.h" | 27 #include "core/fxcodec/fx_codec.h" |
28 #include "core/fxge/fx_dib.h" | 28 #include "core/fxge/fx_dib.h" |
| 29 #include "third_party/base/numerics/safe_conversions.h" |
29 #include "third_party/base/ptr_util.h" | 30 #include "third_party/base/ptr_util.h" |
30 | 31 |
31 CPDF_Image::CPDF_Image(CPDF_Document* pDoc) : m_pDocument(pDoc) {} | 32 CPDF_Image::CPDF_Image(CPDF_Document* pDoc) : m_pDocument(pDoc) {} |
32 | 33 |
33 CPDF_Image::CPDF_Image(CPDF_Document* pDoc, | 34 CPDF_Image::CPDF_Image(CPDF_Document* pDoc, |
34 std::unique_ptr<CPDF_Stream> pStream) | 35 std::unique_ptr<CPDF_Stream> pStream) |
35 : m_bIsInline(true), | 36 : m_bIsInline(true), |
36 m_pDocument(pDoc), | 37 m_pDocument(pDoc), |
37 m_pStream(pStream.get()), | 38 m_pStream(pStream.get()), |
38 m_pOwnedStream(std::move(pStream)) { | 39 m_pOwnedStream(std::move(pStream)) { |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
79 } | 80 } |
80 | 81 |
81 void CPDF_Image::ConvertStreamToIndirectObject() { | 82 void CPDF_Image::ConvertStreamToIndirectObject() { |
82 if (!m_pStream->IsInline()) | 83 if (!m_pStream->IsInline()) |
83 return; | 84 return; |
84 | 85 |
85 ASSERT(m_pOwnedStream); | 86 ASSERT(m_pOwnedStream); |
86 m_pDocument->AddIndirectObject(std::move(m_pOwnedStream)); | 87 m_pDocument->AddIndirectObject(std::move(m_pOwnedStream)); |
87 } | 88 } |
88 | 89 |
89 CPDF_Dictionary* CPDF_Image::InitJPEG(uint8_t* pData, uint32_t size) { | 90 std::unique_ptr<CPDF_Dictionary> CPDF_Image::InitJPEG(uint8_t* pData, |
| 91 uint32_t size) { |
90 int32_t width; | 92 int32_t width; |
91 int32_t height; | 93 int32_t height; |
92 int32_t num_comps; | 94 int32_t num_comps; |
93 int32_t bits; | 95 int32_t bits; |
94 bool color_trans; | 96 bool color_trans; |
95 if (!CPDF_ModuleMgr::Get()->GetJpegModule()->LoadInfo( | 97 if (!CPDF_ModuleMgr::Get()->GetJpegModule()->LoadInfo( |
96 pData, size, &width, &height, &num_comps, &bits, &color_trans)) { | 98 pData, size, &width, &height, &num_comps, &bits, &color_trans)) { |
97 return nullptr; | 99 return nullptr; |
98 } | 100 } |
99 | 101 |
100 CPDF_Dictionary* pDict = | 102 auto pDict = |
101 new CPDF_Dictionary(m_pDocument->GetByteStringPool()); | 103 pdfium::MakeUnique<CPDF_Dictionary>(m_pDocument->GetByteStringPool()); |
102 pDict->SetNewFor<CPDF_Name>("Type", "XObject"); | 104 pDict->SetNewFor<CPDF_Name>("Type", "XObject"); |
103 pDict->SetNewFor<CPDF_Name>("Subtype", "Image"); | 105 pDict->SetNewFor<CPDF_Name>("Subtype", "Image"); |
104 pDict->SetNewFor<CPDF_Number>("Width", width); | 106 pDict->SetNewFor<CPDF_Number>("Width", width); |
105 pDict->SetNewFor<CPDF_Number>("Height", height); | 107 pDict->SetNewFor<CPDF_Number>("Height", height); |
106 const FX_CHAR* csname = nullptr; | 108 const FX_CHAR* csname = nullptr; |
107 if (num_comps == 1) { | 109 if (num_comps == 1) { |
108 csname = "DeviceGray"; | 110 csname = "DeviceGray"; |
109 } else if (num_comps == 3) { | 111 } else if (num_comps == 3) { |
110 csname = "DeviceRGB"; | 112 csname = "DeviceRGB"; |
111 } else if (num_comps == 4) { | 113 } else if (num_comps == 4) { |
(...skipping 15 matching lines...) Expand all Loading... |
127 m_Width = width; | 129 m_Width = width; |
128 m_Height = height; | 130 m_Height = height; |
129 if (!m_pStream) { | 131 if (!m_pStream) { |
130 m_pOwnedStream = pdfium::MakeUnique<CPDF_Stream>(); | 132 m_pOwnedStream = pdfium::MakeUnique<CPDF_Stream>(); |
131 m_pStream = m_pOwnedStream.get(); | 133 m_pStream = m_pOwnedStream.get(); |
132 } | 134 } |
133 return pDict; | 135 return pDict; |
134 } | 136 } |
135 | 137 |
136 void CPDF_Image::SetJpegImage(IFX_SeekableReadStream* pFile) { | 138 void CPDF_Image::SetJpegImage(IFX_SeekableReadStream* pFile) { |
137 uint32_t size = (uint32_t)pFile->GetSize(); | 139 uint32_t size = pdfium::base::checked_cast<uint32_t>(pFile->GetSize()); |
138 if (!size) | 140 if (!size) |
139 return; | 141 return; |
140 | 142 |
141 uint32_t dwEstimateSize = std::min(size, 8192U); | 143 uint32_t dwEstimateSize = std::min(size, 8192U); |
142 std::vector<uint8_t> data(dwEstimateSize); | 144 std::vector<uint8_t> data(dwEstimateSize); |
143 pFile->ReadBlock(data.data(), 0, dwEstimateSize); | 145 if (!pFile->ReadBlock(data.data(), 0, dwEstimateSize)) |
144 CPDF_Dictionary* pDict = InitJPEG(data.data(), dwEstimateSize); | 146 return; |
| 147 |
| 148 std::unique_ptr<CPDF_Dictionary> pDict = |
| 149 InitJPEG(data.data(), dwEstimateSize); |
145 if (!pDict && size > dwEstimateSize) { | 150 if (!pDict && size > dwEstimateSize) { |
146 data.resize(size); | 151 data.resize(size); |
147 pFile->ReadBlock(data.data(), 0, size); | 152 pFile->ReadBlock(data.data(), 0, size); |
148 pDict = InitJPEG(data.data(), size); | 153 pDict = InitJPEG(data.data(), size); |
149 } | 154 } |
150 if (!pDict) | 155 if (!pDict) |
151 return; | 156 return; |
152 | 157 |
153 m_pStream->InitStreamFromFile(pFile, pdfium::WrapUnique(pDict)); | 158 m_pStream->InitStreamFromFile(pFile, std::move(pDict)); |
154 } | 159 } |
155 | 160 |
156 void CPDF_Image::SetImage(const CFX_DIBitmap* pBitmap) { | 161 void CPDF_Image::SetImage(const CFX_DIBitmap* pBitmap) { |
157 int32_t BitmapWidth = pBitmap->GetWidth(); | 162 int32_t BitmapWidth = pBitmap->GetWidth(); |
158 int32_t BitmapHeight = pBitmap->GetHeight(); | 163 int32_t BitmapHeight = pBitmap->GetHeight(); |
159 if (BitmapWidth < 1 || BitmapHeight < 1) | 164 if (BitmapWidth < 1 || BitmapHeight < 1) |
160 return; | 165 return; |
161 | 166 |
162 auto pDict = | 167 auto pDict = |
163 pdfium::MakeUnique<CPDF_Dictionary>(m_pDocument->GetByteStringPool()); | 168 pdfium::MakeUnique<CPDF_Dictionary>(m_pDocument->GetByteStringPool()); |
(...skipping 212 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
376 } | 381 } |
377 if (!ret) { | 382 if (!ret) { |
378 delete m_pDIBSource; | 383 delete m_pDIBSource; |
379 m_pDIBSource = nullptr; | 384 m_pDIBSource = nullptr; |
380 return false; | 385 return false; |
381 } | 386 } |
382 m_pMask = pSource->DetachMask(); | 387 m_pMask = pSource->DetachMask(); |
383 m_MatteColor = pSource->GetMatteColor(); | 388 m_MatteColor = pSource->GetMatteColor(); |
384 return false; | 389 return false; |
385 } | 390 } |
OLD | NEW |