Fix potential integer overflow in fpdf_render_loadimage.cpp
[pdfium.git] / core / src / fpdfapi / fpdf_render / fpdf_render_loadimage.cpp
1 // Copyright 2014 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4  
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "../../../include/fxge/fx_ge.h"
8 #include "../../../include/fxcodec/fx_codec.h"
9 #include "../../../include/fpdfapi/fpdf_module.h"
10 #include "../../../include/fpdfapi/fpdf_render.h"
11 #include "../../../include/fpdfapi/fpdf_pageobj.h"
12 #include "../fpdf_page/pageint.h"
13 #include "render_int.h"
14 #include "../../../../third_party/numerics/safe_math.h"
15
16 static unsigned int _GetBits8(FX_LPCBYTE pData, int bitpos, int nbits)
17 {
18     unsigned int byte = pData[bitpos / 8];
19     if (nbits == 8) {
20         return byte;
21     } else if (nbits == 4) {
22         return (bitpos % 8) ? (byte & 0x0f) : (byte >> 4);
23     } else if (nbits == 2) {
24         return (byte >> (6 - bitpos % 8)) & 0x03;
25     } else if (nbits == 1) {
26         return (byte >> (7 - bitpos % 8)) & 0x01;
27     } else if (nbits == 16) {
28         return byte * 256 + pData[bitpos / 8 + 1];
29     }
30     return 0;
31 }
32 CFX_DIBSource* CPDF_Image::LoadDIBSource(CFX_DIBSource** ppMask, FX_DWORD* pMatteColor, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask) const
33 {
34     CPDF_DIBSource* pSource = FX_NEW CPDF_DIBSource;
35     if (pSource->Load(m_pDocument, m_pStream, (CPDF_DIBSource**)ppMask, pMatteColor, NULL, NULL, bStdCS, GroupFamily, bLoadMask)) {
36         return pSource;
37     }
38     delete pSource;
39     return NULL;
40 }
41 CFX_DIBSource* CPDF_Image::DetachBitmap()
42 {
43     CFX_DIBSource* pBitmap = m_pDIBSource;
44     m_pDIBSource = NULL;
45     return pBitmap;
46 }
47 CFX_DIBSource* CPDF_Image::DetachMask()
48 {
49     CFX_DIBSource* pBitmap = m_pMask;
50     m_pMask = NULL;
51     return pBitmap;
52 }
53 FX_BOOL CPDF_Image::StartLoadDIBSource(CPDF_Dictionary* pFormResource, CPDF_Dictionary* pPageResource, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask)
54 {
55     m_pDIBSource = FX_NEW CPDF_DIBSource;
56     int ret = ((CPDF_DIBSource*)m_pDIBSource)->StartLoadDIBSource(m_pDocument, m_pStream, TRUE, pFormResource, pPageResource, bStdCS, GroupFamily, bLoadMask);
57     if (ret == 2) {
58         return TRUE;
59     }
60     if (!ret) {
61         delete m_pDIBSource;
62         m_pDIBSource = NULL;
63         return FALSE;
64     }
65     m_pMask = ((CPDF_DIBSource*)m_pDIBSource)->DetachMask();
66     m_MatteColor = ((CPDF_DIBSource*)m_pDIBSource)->m_MatteColor;
67     return FALSE;
68 }
69 FX_BOOL CPDF_Image::Continue(IFX_Pause* pPause)
70 {
71     int ret = ((CPDF_DIBSource*)m_pDIBSource)->ContinueLoadDIBSource(pPause);
72     if (ret == 2) {
73         return TRUE;
74     }
75     if (!ret) {
76         delete m_pDIBSource;
77         m_pDIBSource = NULL;
78         return FALSE;
79     }
80     m_pMask = ((CPDF_DIBSource*)m_pDIBSource)->DetachMask();
81     m_MatteColor = ((CPDF_DIBSource*)m_pDIBSource)->m_MatteColor;
82     return FALSE;
83 }
84 CPDF_DIBSource::CPDF_DIBSource()
85 {
86     m_pDocument = NULL;
87     m_pStreamAcc = NULL;
88     m_pDict = NULL;
89     m_bpp = 0;
90     m_Width = m_Height = 0;
91     m_pColorSpace = NULL;
92     m_bDefaultDecode = TRUE;
93     m_bImageMask = FALSE;
94     m_pPalette = NULL;
95     m_pCompData = NULL;
96     m_bColorKey = FALSE;
97     m_pMaskedLine = m_pLineBuf = NULL;
98     m_pCachedBitmap = NULL;
99     m_pDecoder = NULL;
100     m_nComponents = 0;
101     m_bpc = 0;
102     m_bLoadMask = FALSE;
103     m_Family = 0;
104     m_pMask = NULL;
105     m_MatteColor = 0;
106     m_pJbig2Context = NULL;
107     m_pGlobalStream = NULL;
108     m_bStdCS = FALSE;
109     m_pMaskStream = NULL;
110     m_Status = 0;
111     m_bHasMask = FALSE;
112 }
113 CPDF_DIBSource::~CPDF_DIBSource()
114 {
115     if (m_pStreamAcc) {
116         delete m_pStreamAcc;
117     }
118     if (m_pMaskedLine) {
119         FX_Free(m_pMaskedLine);
120     }
121     if (m_pLineBuf) {
122         FX_Free(m_pLineBuf);
123     }
124     if (m_pCachedBitmap) {
125         delete m_pCachedBitmap;
126     }
127     if (m_pDecoder) {
128         delete m_pDecoder;
129     }
130     if (m_pCompData) {
131         FX_Free(m_pCompData);
132     }
133     CPDF_ColorSpace* pCS = m_pColorSpace;
134     if (pCS && m_pDocument) {
135         m_pDocument->GetPageData()->ReleaseColorSpace(pCS->GetArray());
136     }
137     if (m_pJbig2Context) {
138         ICodec_Jbig2Module* pJbig2Moudle = CPDF_ModuleMgr::Get()->GetJbig2Module();
139         pJbig2Moudle->DestroyJbig2Context(m_pJbig2Context);
140         m_pJbig2Context = NULL;
141     }
142     if (m_pGlobalStream) {
143         delete m_pGlobalStream;
144     }
145     m_pGlobalStream = NULL;
146 }
147 CFX_DIBitmap* CPDF_DIBSource::GetBitmap() const
148 {
149     if (m_pCachedBitmap) {
150         return m_pCachedBitmap;
151     }
152     return Clone();
153 }
154 void CPDF_DIBSource::ReleaseBitmap(CFX_DIBitmap* pBitmap) const
155 {
156     if (pBitmap && pBitmap != m_pCachedBitmap) {
157         delete pBitmap;
158     }
159 }
160 FX_BOOL CPDF_DIBSource::Load(CPDF_Document* pDoc, const CPDF_Stream* pStream, CPDF_DIBSource** ppMask,
161                              FX_DWORD* pMatteColor, CPDF_Dictionary* pFormResources, CPDF_Dictionary* pPageResources, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask)
162 {
163     if (pStream == NULL) {
164         return FALSE;
165     }
166     m_pDocument = pDoc;
167     m_pDict = pStream->GetDict();
168     m_pStream = pStream;
169     m_Width = m_pDict->GetInteger(FX_BSTRC("Width"));
170     m_Height = m_pDict->GetInteger(FX_BSTRC("Height"));
171     if (m_Width <= 0 || m_Height <= 0 || m_Width > 0x01ffff || m_Height > 0x01ffff) {
172         return FALSE;
173     }
174     m_GroupFamily = GroupFamily;
175     m_bLoadMask = bLoadMask;
176     if (!LoadColorInfo(m_pStream->GetObjNum() != 0 ? NULL : pFormResources, pPageResources)) {
177         return FALSE;
178     }
179
180     if (m_bpc == 0 || m_nComponents == 0) {
181         return FALSE;
182     }
183
184     FX_SAFE_DWORD src_pitch = m_bpc;
185     src_pitch *= m_nComponents;
186     src_pitch *= m_Width;
187     src_pitch += 7;
188     src_pitch /= 8;
189     src_pitch *= m_Height;
190     if (!src_pitch.IsValid()) {
191         return FALSE;
192     }
193
194     m_pStreamAcc = FX_NEW CPDF_StreamAcc;
195     m_pStreamAcc->LoadAllData(pStream, FALSE, src_pitch.ValueOrDie(), TRUE);
196     if (m_pStreamAcc->GetSize() == 0 || m_pStreamAcc->GetData() == NULL) {
197         return FALSE;
198     }
199     const CFX_ByteString& decoder = m_pStreamAcc->GetImageDecoder();
200     if (!decoder.IsEmpty() && decoder == FX_BSTRC("CCITTFaxDecode")) {
201         m_bpc = 1;
202     }
203     if (!CreateDecoder()) {
204         return FALSE;
205     }
206     if (m_bImageMask) {
207         m_bpp = 1;
208         m_bpc = 1;
209         m_nComponents = 1;
210         m_AlphaFlag = 1;
211     } else if (m_bpc * m_nComponents == 1) {
212         m_bpp = 1;
213     } else if (m_bpc * m_nComponents <= 8) {
214         m_bpp = 8;
215     } else {
216         m_bpp = 24;
217     }
218
219     FX_SAFE_DWORD pitch = m_Width;
220     pitch *= m_bpp;
221     pitch += 31;
222     pitch /= 8;
223     if (!pitch.IsValid()) {
224         return FALSE;
225     }
226   
227     m_pLineBuf = FX_Alloc(FX_BYTE, pitch.ValueOrDie());
228     if (m_pColorSpace && bStdCS) {
229         m_pColorSpace->EnableStdConversion(TRUE);
230     }
231     LoadPalette();
232     if (m_bColorKey) {
233         m_bpp = 32;
234         m_AlphaFlag = 2;
235         pitch = m_Width;
236         pitch *= m_bpp;
237         pitch += 31;
238         pitch /= 8;
239         if (!pitch.IsValid()) {
240             return FALSE;
241         }
242
243         m_pMaskedLine = FX_Alloc(FX_BYTE, pitch.ValueOrDie());
244     }
245     m_Pitch = pitch.ValueOrDie();
246     if (ppMask) {
247         *ppMask = LoadMask(*pMatteColor);
248     }
249     if (m_pColorSpace && bStdCS) {
250         m_pColorSpace->EnableStdConversion(FALSE);
251     }
252     return TRUE;
253 }
254 int     CPDF_DIBSource::ContinueToLoadMask()
255 {
256     if (m_bImageMask) {
257         m_bpp = 1;
258         m_bpc = 1;
259         m_nComponents = 1;
260         m_AlphaFlag = 1;
261     } else if (m_bpc * m_nComponents == 1) {
262         m_bpp = 1;
263     } else if (m_bpc * m_nComponents <= 8) {
264         m_bpp = 8;
265     } else {
266         m_bpp = 24;
267     }
268     if (!m_bpc || !m_nComponents) {
269         return 0;
270     }
271     FX_SAFE_DWORD pitch = m_Width;
272     pitch *= m_bpp;
273     pitch += 31;
274     pitch /= 8;
275     if (!pitch.IsValid()) {
276         return 0;
277     }
278     m_pLineBuf = FX_Alloc(FX_BYTE, pitch.ValueOrDie());
279     if (m_pColorSpace && m_bStdCS) {
280         m_pColorSpace->EnableStdConversion(TRUE);
281     }
282     LoadPalette();
283     if (m_bColorKey) {
284         m_bpp = 32;
285         m_AlphaFlag = 2;
286         pitch = m_Width;
287         pitch *= m_bpp;
288         pitch += 31;
289         pitch /= 8;
290         if (!pitch.IsValid()) {
291             return 0;
292         }
293         m_pMaskedLine = FX_Alloc(FX_BYTE, pitch.ValueOrDie());
294     }
295     m_Pitch = pitch.ValueOrDie();
296     return 1;
297 }
298 int     CPDF_DIBSource::StartLoadDIBSource(CPDF_Document* pDoc, const CPDF_Stream* pStream, FX_BOOL bHasMask,
299                                        CPDF_Dictionary* pFormResources, CPDF_Dictionary* pPageResources,
300                                        FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask)
301 {
302     if (pStream == NULL) {
303         return 0;
304     }
305     m_pDocument = pDoc;
306     m_pDict = pStream->GetDict();
307     m_pStream = pStream;
308     m_bStdCS = bStdCS;
309     m_bHasMask = bHasMask;
310     m_Width = m_pDict->GetInteger(FX_BSTRC("Width"));
311     m_Height = m_pDict->GetInteger(FX_BSTRC("Height"));
312     if (m_Width <= 0 || m_Height <= 0 || m_Width > 0x01ffff || m_Height > 0x01ffff) {
313         return 0;
314     }
315     m_GroupFamily = GroupFamily;
316     m_bLoadMask = bLoadMask;
317     if (!LoadColorInfo(m_pStream->GetObjNum() != 0 ? NULL : pFormResources, pPageResources)) {
318         return 0;
319     }
320
321     if (m_bpc == 0 || m_nComponents == 0) {
322         return 0;
323     }
324
325     FX_SAFE_DWORD src_pitch = m_bpc;
326     src_pitch *= m_nComponents;
327     src_pitch *= m_Width;
328     src_pitch += 7;
329     src_pitch /= 8;
330     src_pitch *= m_Height;
331     if (!src_pitch.IsValid()) {
332         return 0;
333     }
334
335     m_pStreamAcc = FX_NEW CPDF_StreamAcc;
336     m_pStreamAcc->LoadAllData(pStream, FALSE, src_pitch.ValueOrDie(), TRUE);
337     if (m_pStreamAcc->GetSize() == 0 || m_pStreamAcc->GetData() == NULL) {
338         return 0;
339     }
340     const CFX_ByteString& decoder = m_pStreamAcc->GetImageDecoder();
341     if (!decoder.IsEmpty() && decoder == FX_BSTRC("CCITTFaxDecode")) {
342         m_bpc = 1;
343     }
344     int ret = CreateDecoder();
345     if (ret != 1) {
346         if (!ret) {
347             return ret;
348         }
349         if (!ContinueToLoadMask()) {
350             return 0;
351         }
352         if (m_bHasMask) {
353             StratLoadMask();
354         }
355         return ret;
356     }
357     if (!ContinueToLoadMask()) {
358         return 0;
359     }
360     if (m_bHasMask) {
361         ret = StratLoadMask();
362     }
363     if (ret == 2) {
364         return ret;
365     }
366     if (m_pColorSpace && m_bStdCS) {
367         m_pColorSpace->EnableStdConversion(FALSE);
368     }
369     return ret;
370 }
371 int     CPDF_DIBSource::ContinueLoadDIBSource(IFX_Pause* pPause)
372 {
373     FXCODEC_STATUS ret;
374     if (m_Status == 1) {
375         const CFX_ByteString& decoder = m_pStreamAcc->GetImageDecoder();
376         if (decoder == FX_BSTRC("JPXDecode")) {
377             return 0;
378         }
379         ICodec_Jbig2Module* pJbig2Moudle = CPDF_ModuleMgr::Get()->GetJbig2Module();
380         if (m_pJbig2Context == NULL) {
381             m_pJbig2Context = pJbig2Moudle->CreateJbig2Context();
382             if (m_pStreamAcc->GetImageParam()) {
383                 CPDF_Stream* pGlobals = m_pStreamAcc->GetImageParam()->GetStream(FX_BSTRC("JBIG2Globals"));
384                 if (pGlobals) {
385                     m_pGlobalStream = FX_NEW CPDF_StreamAcc;
386                     m_pGlobalStream->LoadAllData(pGlobals, FALSE);
387                 }
388             }
389             ret = pJbig2Moudle->StartDecode(m_pJbig2Context, m_Width, m_Height, m_pStreamAcc->GetData(), m_pStreamAcc->GetSize(),
390                                             m_pGlobalStream ? m_pGlobalStream->GetData() : NULL, m_pGlobalStream ? m_pGlobalStream->GetSize() : 0, m_pCachedBitmap->GetBuffer(),
391                                             m_pCachedBitmap->GetPitch(), pPause);
392             if (ret < 0) {
393                 delete m_pCachedBitmap;
394                 m_pCachedBitmap = NULL;
395                 if (m_pGlobalStream) {
396                     delete m_pGlobalStream;
397                 }
398                 m_pGlobalStream = NULL;
399                 pJbig2Moudle->DestroyJbig2Context(m_pJbig2Context);
400                 m_pJbig2Context = NULL;
401                 return 0;
402             }
403             if (ret == FXCODEC_STATUS_DECODE_TOBECONTINUE) {
404                 return 2;
405             }
406             int ret1 = 1;
407             if (m_bHasMask) {
408                 ret1 = ContinueLoadMaskDIB(pPause);
409                 m_Status = 2;
410             }
411             if (ret1 == 2) {
412                 return ret1;
413             }
414             if (m_pColorSpace && m_bStdCS) {
415                 m_pColorSpace->EnableStdConversion(FALSE);
416             }
417             return ret1;
418         }
419         FXCODEC_STATUS ret = pJbig2Moudle->ContinueDecode(m_pJbig2Context, pPause);
420         if (ret < 0) {
421             delete m_pCachedBitmap;
422             m_pCachedBitmap = NULL;
423             if (m_pGlobalStream) {
424                 delete m_pGlobalStream;
425             }
426             m_pGlobalStream = NULL;
427             pJbig2Moudle->DestroyJbig2Context(m_pJbig2Context);
428             m_pJbig2Context = NULL;
429             return 0;
430         }
431         if (ret == FXCODEC_STATUS_DECODE_TOBECONTINUE) {
432             return 2;
433         }
434         int ret1 = 1;
435         if (m_bHasMask) {
436             ret1 = ContinueLoadMaskDIB(pPause);
437             m_Status = 2;
438         }
439         if (ret1 == 2) {
440             return ret1;
441         }
442         if (m_pColorSpace && m_bStdCS) {
443             m_pColorSpace->EnableStdConversion(FALSE);
444         }
445         return ret1;
446     } else if (m_Status == 2) {
447         return ContinueLoadMaskDIB(pPause);
448     }
449     return 0;
450 }
451 FX_BOOL CPDF_DIBSource::LoadColorInfo(CPDF_Dictionary* pFormResources, CPDF_Dictionary* pPageResources)
452 {
453     if (m_pDict->GetInteger("ImageMask")) {
454         m_bImageMask = TRUE;
455     }
456     if (m_bImageMask || !m_pDict->KeyExist(FX_BSTRC("ColorSpace"))) {
457         if (!m_bImageMask) {
458             CPDF_Object* pFilter = m_pDict->GetElementValue(FX_BSTRC("Filter"));
459             if (pFilter) {
460                 CFX_ByteString filter;
461                 if (pFilter->GetType() == PDFOBJ_NAME) {
462                     filter = pFilter->GetString();
463                     if (filter == FX_BSTRC("JPXDecode")) {
464                         return TRUE;
465                     }
466                 } else if (pFilter->GetType() == PDFOBJ_ARRAY) {
467                     CPDF_Array* pArray = (CPDF_Array*)pFilter;
468                     if (pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("JPXDecode")) {
469                         return TRUE;
470                     }
471                 }
472             }
473         }
474         m_bImageMask = TRUE;
475         m_bpc = m_nComponents = 1;
476         CPDF_Array* pDecode = m_pDict->GetArray(FX_BSTRC("Decode"));
477         m_bDefaultDecode = pDecode == NULL || pDecode->GetInteger(0) == 0;
478         return TRUE;
479     }
480     CPDF_Object* pCSObj = m_pDict->GetElementValue(FX_BSTRC("ColorSpace"));
481     if (pCSObj == NULL) {
482         return FALSE;
483     }
484     CPDF_DocPageData* pDocPageData = m_pDocument->GetPageData();
485     if (pFormResources) {
486         m_pColorSpace = pDocPageData->GetColorSpace(pCSObj, pFormResources);
487     }
488     if (m_pColorSpace == NULL) {
489         m_pColorSpace = pDocPageData->GetColorSpace(pCSObj, pPageResources);
490     }
491     if (m_pColorSpace == NULL) {
492         return FALSE;
493     }
494     m_bpc = m_pDict->GetInteger(FX_BSTRC("BitsPerComponent"));
495     m_Family = m_pColorSpace->GetFamily();
496     m_nComponents = m_pColorSpace->CountComponents();
497     if (m_Family == PDFCS_ICCBASED && pCSObj->GetType() == PDFOBJ_NAME) {
498         CFX_ByteString cs = pCSObj->GetString();
499         if (cs == FX_BSTRC("DeviceGray")) {
500             m_nComponents = 1;
501         } else if (cs == FX_BSTRC("DeviceRGB")) {
502             m_nComponents = 3;
503         } else if (cs == FX_BSTRC("DeviceCMYK")) {
504             m_nComponents = 4;
505         }
506     }
507     m_pCompData = FX_Alloc(DIB_COMP_DATA, m_nComponents);
508     if (m_bpc == 0) {
509         return TRUE;
510     }
511     int max_data = (1 << m_bpc) - 1;
512     CPDF_Array* pDecode = m_pDict->GetArray(FX_BSTRC("Decode"));
513     if (pDecode) {
514         for (FX_DWORD i = 0; i < m_nComponents; i ++) {
515             m_pCompData[i].m_DecodeMin = pDecode->GetNumber(i * 2);
516             FX_FLOAT max = pDecode->GetNumber(i * 2 + 1);
517             m_pCompData[i].m_DecodeStep = (max - m_pCompData[i].m_DecodeMin) / max_data;
518             FX_FLOAT def_value, def_min, def_max;
519             m_pColorSpace->GetDefaultValue(i, def_value, def_min, def_max);
520             if (m_Family == PDFCS_INDEXED) {
521                 def_max = (FX_FLOAT)max_data;
522             }
523             if (def_min != m_pCompData[i].m_DecodeMin || def_max != max) {
524                 m_bDefaultDecode = FALSE;
525             }
526         }
527     } else {
528         for (FX_DWORD i = 0; i < m_nComponents; i ++) {
529             FX_FLOAT def_value;
530             m_pColorSpace->GetDefaultValue(i, def_value, m_pCompData[i].m_DecodeMin, m_pCompData[i].m_DecodeStep);
531             if (m_Family == PDFCS_INDEXED) {
532                 m_pCompData[i].m_DecodeStep = (FX_FLOAT)max_data;
533             }
534             m_pCompData[i].m_DecodeStep = (m_pCompData[i].m_DecodeStep - m_pCompData[i].m_DecodeMin) / max_data;
535         }
536     }
537     if (!m_pDict->KeyExist(FX_BSTRC("SMask"))) {
538         CPDF_Object* pMask = m_pDict->GetElementValue(FX_BSTRC("Mask"));
539         if (pMask == NULL) {
540             return TRUE;
541         }
542         if (pMask->GetType() == PDFOBJ_ARRAY) {
543             CPDF_Array* pArray = (CPDF_Array*)pMask;
544             if (pArray->GetCount() >= m_nComponents * 2)
545                 for (FX_DWORD i = 0; i < m_nComponents * 2; i ++) {
546                     if (i % 2) {
547                         m_pCompData[i / 2].m_ColorKeyMax = pArray->GetInteger(i);
548                     } else {
549                         m_pCompData[i / 2].m_ColorKeyMin = pArray->GetInteger(i);
550                     }
551                 }
552             m_bColorKey = TRUE;
553         }
554     }
555     return TRUE;
556 }
557 ICodec_ScanlineDecoder* FPDFAPI_CreateFaxDecoder(FX_LPCBYTE src_buf, FX_DWORD src_size, int width, int height,
558         const CPDF_Dictionary* pParams);
559 ICodec_ScanlineDecoder* FPDFAPI_CreateFlateDecoder(FX_LPCBYTE src_buf, FX_DWORD src_size, int width, int height,
560         int nComps, int bpc, const CPDF_Dictionary* pParams);
561 int CPDF_DIBSource::CreateDecoder()
562 {
563     const CFX_ByteString& decoder = m_pStreamAcc->GetImageDecoder();
564     if (decoder.IsEmpty()) {
565         return 1;
566     }
567     FX_LPCBYTE src_data = m_pStreamAcc->GetData();
568     FX_DWORD src_size = m_pStreamAcc->GetSize();
569     const CPDF_Dictionary* pParams = m_pStreamAcc->GetImageParam();
570     if (decoder == FX_BSTRC("CCITTFaxDecode")) {
571         m_pDecoder = FPDFAPI_CreateFaxDecoder(src_data, src_size, m_Width, m_Height, pParams);
572     } else if (decoder == FX_BSTRC("DCTDecode")) {
573         m_pDecoder = CPDF_ModuleMgr::Get()->GetJpegModule()->CreateDecoder(src_data, src_size, m_Width, m_Height,
574                      m_nComponents, pParams ? pParams->GetInteger(FX_BSTR("ColorTransform"), 1) : 1);
575         if (NULL == m_pDecoder) {
576             FX_BOOL bTransform = FALSE;
577             int comps, bpc;
578             ICodec_JpegModule* pJpegModule = CPDF_ModuleMgr::Get()->GetJpegModule();
579             if (pJpegModule->LoadInfo(src_data, src_size, m_Width, m_Height, comps, bpc, bTransform)) {
580                 m_nComponents = comps;
581                 m_bpc = bpc;
582                 m_pDecoder = CPDF_ModuleMgr::Get()->GetJpegModule()->CreateDecoder(src_data, src_size, m_Width, m_Height,
583                              m_nComponents, bTransform);
584             }
585         }
586     } else if (decoder == FX_BSTRC("FlateDecode")) {
587         m_pDecoder = FPDFAPI_CreateFlateDecoder(src_data, src_size, m_Width, m_Height, m_nComponents, m_bpc, pParams);
588     } else if (decoder == FX_BSTRC("JPXDecode")) {
589         LoadJpxBitmap();
590         return m_pCachedBitmap != NULL ? 1 : 0;
591     } else if (decoder == FX_BSTRC("JBIG2Decode")) {
592         m_pCachedBitmap = FX_NEW CFX_DIBitmap;
593         if (!m_pCachedBitmap->Create(m_Width, m_Height, m_bImageMask ? FXDIB_1bppMask : FXDIB_1bppRgb)) {
594             delete m_pCachedBitmap;
595             m_pCachedBitmap = NULL;
596             return 0;
597         }
598         m_Status = 1;
599         return 2;
600     } else if (decoder == FX_BSTRC("RunLengthDecode")) {
601         m_pDecoder = CPDF_ModuleMgr::Get()->GetCodecModule()->GetBasicModule()->CreateRunLengthDecoder(src_data, src_size, m_Width, m_Height, m_nComponents, m_bpc);
602     }
603     if (m_pDecoder) {
604         int requested_pitch = (m_Width * m_nComponents * m_bpc + 7) / 8;
605         int provided_pitch = (m_pDecoder->GetWidth() * m_pDecoder->CountComps() * m_pDecoder->GetBPC() + 7) / 8;
606         if (provided_pitch < requested_pitch) {
607             return 0;
608         }
609         return 1;
610     }
611     return 0;
612 }
613 void CPDF_DIBSource::LoadJpxBitmap()
614 {
615     ICodec_JpxModule* pJpxModule = CPDF_ModuleMgr::Get()->GetJpxModule();
616     if (pJpxModule == NULL) {
617         return;
618     }
619     FX_LPVOID ctx = pJpxModule->CreateDecoder(m_pStreamAcc->GetData(), m_pStreamAcc->GetSize(), m_pColorSpace != NULL);
620     if (ctx == NULL) {
621         return;
622     }
623     FX_DWORD width = 0, height = 0, codestream_nComps = 0, image_nComps = 0;
624     pJpxModule->GetImageInfo(ctx, width, height, codestream_nComps, image_nComps);
625     if ((int)width < m_Width || (int)height < m_Height) {
626         pJpxModule->DestroyDecoder(ctx);
627         return;
628     }
629     int output_nComps;
630     FX_BOOL bTranslateColor, bSwapRGB = FALSE;
631     if (m_pColorSpace) {
632         if (codestream_nComps != (FX_DWORD)m_pColorSpace->CountComponents()) {
633             return;
634         }
635         output_nComps = codestream_nComps;
636         bTranslateColor = FALSE;
637         if (m_pColorSpace == CPDF_ColorSpace::GetStockCS(PDFCS_DEVICERGB)) {
638             bSwapRGB = TRUE;
639             m_pColorSpace = NULL;
640         }
641     } else {
642         bTranslateColor = TRUE;
643         if (image_nComps) {
644             output_nComps = image_nComps;
645         } else {
646             output_nComps = codestream_nComps;
647         }
648         if (output_nComps == 3) {
649             bSwapRGB = TRUE;
650         } else if (output_nComps == 4) {
651             m_pColorSpace = CPDF_ColorSpace::GetStockCS(PDFCS_DEVICECMYK);
652             bTranslateColor = FALSE;
653         }
654         m_nComponents = output_nComps;
655     }
656     FXDIB_Format format;
657     if (output_nComps == 1) {
658         format = FXDIB_8bppRgb;
659     } else if (output_nComps <= 3) {
660         format = FXDIB_Rgb;
661     } else if (output_nComps == 4) {
662         format = FXDIB_Rgb32;
663     } else {
664         width = (width * output_nComps + 2) / 3;
665         format = FXDIB_Rgb;
666     }
667     m_pCachedBitmap = FX_NEW CFX_DIBitmap;
668     if (!m_pCachedBitmap->Create(width, height, format)) {
669         delete m_pCachedBitmap;
670         m_pCachedBitmap = NULL;
671         return;
672     }
673     m_pCachedBitmap->Clear(0xFFFFFFFF);
674     FX_LPBYTE output_offsets = FX_Alloc(FX_BYTE, output_nComps);
675     for (int i = 0; i < output_nComps; i ++) {
676         output_offsets[i] = i;
677     }
678     if (bSwapRGB) {
679         output_offsets[0] = 2;
680         output_offsets[2] = 0;
681     }
682     if (!pJpxModule->Decode(ctx, m_pCachedBitmap->GetBuffer(), m_pCachedBitmap->GetPitch(), bTranslateColor, output_offsets)) {
683         delete m_pCachedBitmap;
684         m_pCachedBitmap = NULL;
685         return;
686     }
687     FX_Free(output_offsets);
688     pJpxModule->DestroyDecoder(ctx);
689     if (m_pColorSpace && m_pColorSpace->GetFamily() == PDFCS_INDEXED && m_bpc < 8) {
690         int scale = 8 - m_bpc;
691         for (FX_DWORD row = 0; row < height; row ++) {
692             FX_LPBYTE scanline = (FX_LPBYTE)m_pCachedBitmap->GetScanline(row);
693             for (FX_DWORD col = 0; col < width; col ++) {
694                 *scanline = (*scanline) >> scale;
695                 scanline++;
696             }
697         }
698     }
699     m_bpc = 8;
700 }
701 void CPDF_DIBSource::LoadJbig2Bitmap()
702 {
703     ICodec_Jbig2Module* pJbig2Module = CPDF_ModuleMgr::Get()->GetJbig2Module();
704     if (pJbig2Module == NULL) {
705         return;
706     }
707     CPDF_StreamAcc* pGlobalStream = NULL;
708     if (m_pStreamAcc->GetImageParam()) {
709         CPDF_Stream* pGlobals = m_pStreamAcc->GetImageParam()->GetStream(FX_BSTRC("JBIG2Globals"));
710         if (pGlobals) {
711             pGlobalStream = FX_NEW CPDF_StreamAcc;
712             pGlobalStream->LoadAllData(pGlobals, FALSE);
713         }
714     }
715     m_pCachedBitmap = FX_NEW CFX_DIBitmap;
716     if (!m_pCachedBitmap->Create(m_Width, m_Height, m_bImageMask ? FXDIB_1bppMask : FXDIB_1bppRgb)) {
717         return;
718     }
719     int ret = pJbig2Module->Decode(m_Width, m_Height, m_pStreamAcc->GetData(), m_pStreamAcc->GetSize(),
720                                    pGlobalStream ? pGlobalStream->GetData() : NULL, pGlobalStream ? pGlobalStream->GetSize() : 0,
721                                    m_pCachedBitmap->GetBuffer(), m_pCachedBitmap->GetPitch());
722     if (ret < 0) {
723         delete m_pCachedBitmap;
724         m_pCachedBitmap = NULL;
725     }
726     if (pGlobalStream) {
727         delete pGlobalStream;
728     }
729     m_bpc = 1;
730     m_nComponents = 1;
731 }
732 CPDF_DIBSource* CPDF_DIBSource::LoadMask(FX_DWORD& MatteColor)
733 {
734     MatteColor = 0xffffffff;
735     CPDF_Stream* pSoftMask = m_pDict->GetStream(FX_BSTRC("SMask"));
736     if (pSoftMask) {
737         CPDF_Array* pMatte = pSoftMask->GetDict()->GetArray(FX_BSTRC("Matte"));
738         if (pMatte != NULL && m_pColorSpace && (FX_DWORD)m_pColorSpace->CountComponents() <= m_nComponents) {
739             FX_FLOAT* pColor = FX_Alloc(FX_FLOAT, m_nComponents);
740             for (FX_DWORD i = 0; i < m_nComponents; i ++) {
741                 pColor[i] = pMatte->GetFloat(i);
742             }
743             FX_FLOAT R, G, B;
744             m_pColorSpace->GetRGB(pColor, R, G, B);
745             FX_Free(pColor);
746             MatteColor = FXARGB_MAKE(0, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255));
747         }
748         return LoadMaskDIB(pSoftMask);
749     }
750     CPDF_Object* pMask = m_pDict->GetElementValue(FX_BSTRC("Mask"));
751     if (pMask == NULL) {
752         return NULL;
753     }
754     if (pMask->GetType() == PDFOBJ_STREAM) {
755         return LoadMaskDIB((CPDF_Stream*)pMask);
756     }
757     return NULL;
758 }
759 int     CPDF_DIBSource::StratLoadMask()
760 {
761     m_MatteColor = 0xffffffff;
762     m_pMaskStream = m_pDict->GetStream(FX_BSTRC("SMask"));
763     if (m_pMaskStream) {
764         CPDF_Array* pMatte = m_pMaskStream->GetDict()->GetArray(FX_BSTRC("Matte"));
765         if (pMatte != NULL && m_pColorSpace && (FX_DWORD)m_pColorSpace->CountComponents() <= m_nComponents) {
766             FX_FLOAT R, G, B;
767             FX_FLOAT* pColor = FX_Alloc(FX_FLOAT, m_nComponents);
768             for (FX_DWORD i = 0; i < m_nComponents; i ++) {
769                 pColor[i] = pMatte->GetFloat(i);
770             }
771             m_pColorSpace->GetRGB(pColor, R, G, B);
772             FX_Free(pColor);
773             m_MatteColor = FXARGB_MAKE(0, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255));
774         }
775         return StartLoadMaskDIB();
776     }
777     m_pMaskStream = m_pDict->GetElementValue(FX_BSTRC("Mask"));
778     if (m_pMaskStream == NULL) {
779         return 1;
780     }
781     if (m_pMaskStream->GetType() == PDFOBJ_STREAM) {
782         return StartLoadMaskDIB();
783     }
784     return 1;
785 }
786 int     CPDF_DIBSource::ContinueLoadMaskDIB(IFX_Pause* pPause)
787 {
788     if (m_pMask == NULL) {
789         return 1;
790     }
791     int ret = m_pMask->ContinueLoadDIBSource(pPause);
792     if (ret == 2) {
793         return ret;
794     }
795     if (m_pColorSpace && m_bStdCS) {
796         m_pColorSpace->EnableStdConversion(FALSE);
797     }
798     if (!ret) {
799         delete m_pMask;
800         m_pMask = NULL;
801         return ret;
802     }
803     return 1;
804 }
805 CPDF_DIBSource* CPDF_DIBSource::DetachMask()
806 {
807     CPDF_DIBSource* pDIBSource = m_pMask;
808     m_pMask = NULL;
809     return pDIBSource;
810 }
811 CPDF_DIBSource* CPDF_DIBSource::LoadMaskDIB(CPDF_Stream* pMask)
812 {
813     CPDF_DIBSource* pMaskSource = FX_NEW CPDF_DIBSource;
814     if (!pMaskSource->Load(m_pDocument, pMask, NULL, NULL, NULL, NULL, TRUE)) {
815         delete pMaskSource;
816         return NULL;
817     }
818     return pMaskSource;
819 }
820 int CPDF_DIBSource::StartLoadMaskDIB()
821 {
822     m_pMask = FX_NEW CPDF_DIBSource;
823     int ret = m_pMask->StartLoadDIBSource(m_pDocument, (CPDF_Stream*)m_pMaskStream, FALSE, NULL, NULL, TRUE);
824     if (ret == 2) {
825         if (m_Status == 0) {
826             m_Status = 2;
827         }
828         return 2;
829     }
830     if (!ret) {
831         delete m_pMask;
832         m_pMask = NULL;
833         return 1;
834     }
835     return 1;
836 }
837 void CPDF_DIBSource::LoadPalette()
838 {
839     if (m_bpc * m_nComponents > 8) {
840         return;
841     }
842     if (m_pColorSpace == NULL) {
843         return;
844     }
845     if (m_bpc * m_nComponents == 1) {
846         if (m_bDefaultDecode && (m_Family == PDFCS_DEVICEGRAY || m_Family == PDFCS_DEVICERGB)) {
847             return;
848         }
849         if (m_pColorSpace->CountComponents() > 3) {
850             return;
851         }
852         FX_FLOAT color_values[3];
853         color_values[0] = m_pCompData[0].m_DecodeMin;
854         color_values[1] = color_values[2] = color_values[0];
855         FX_FLOAT R=0.0f, G=0.0f, B=0.0f;
856         m_pColorSpace->GetRGB(color_values, R, G, B);
857         FX_ARGB argb0 = ArgbEncode(255, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255));
858         color_values[0] += m_pCompData[0].m_DecodeStep;
859         color_values[1] += m_pCompData[0].m_DecodeStep;
860         color_values[2] += m_pCompData[0].m_DecodeStep;
861         m_pColorSpace->GetRGB(color_values, R, G, B);
862         FX_ARGB argb1 = ArgbEncode(255, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255));
863         if (argb0 != 0xFF000000 || argb1 != 0xFFFFFFFF) {
864             SetPaletteArgb(0, argb0);
865             SetPaletteArgb(1, argb1);
866         }
867         return;
868     }
869     if (m_pColorSpace == CPDF_ColorSpace::GetStockCS(PDFCS_DEVICEGRAY) && m_bpc == 8 && m_bDefaultDecode) {
870     } else {
871         int palette_count = 1 << (m_bpc * m_nComponents);
872         CFX_FixedBufGrow<FX_FLOAT, 16> color_values(m_nComponents);
873         FX_FLOAT* color_value = color_values;
874         for (int i = 0; i < palette_count; i ++) {
875             int color_data = i;
876             for (FX_DWORD j = 0; j < m_nComponents; j ++) {
877                 int encoded_component = color_data % (1 << m_bpc);
878                 color_data /= 1 << m_bpc;
879                 color_value[j] = m_pCompData[j].m_DecodeMin + m_pCompData[j].m_DecodeStep * encoded_component;
880             }
881             FX_FLOAT R = 0, G = 0, B = 0;
882             if (m_nComponents == 1 && m_Family == PDFCS_ICCBASED && m_pColorSpace->CountComponents() > 1) {
883                 int nComponents = m_pColorSpace->CountComponents();
884                 FX_FLOAT* temp_buf = FX_Alloc(FX_FLOAT, nComponents);
885                 for (int i = 0; i < nComponents; i++) {
886                     temp_buf[i] = *color_value;
887                 }
888                 m_pColorSpace->GetRGB(temp_buf, R, G, B);
889                 FX_Free(temp_buf);
890             } else {
891                 m_pColorSpace->GetRGB(color_value, R, G, B);
892             }
893             SetPaletteArgb(i, ArgbEncode(255, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255)));
894         }
895     }
896 }
897 FX_DWORD CPDF_DIBSource::GetValidBpc() const
898 {
899     FX_DWORD bpc = m_bpc;
900         CPDF_Object * pFilter = m_pDict ? m_pDict->GetElementValue(FX_BSTRC("Filter")) : NULL;
901     if (pFilter) {
902         if (pFilter->GetType() == PDFOBJ_NAME) {
903             CFX_ByteString filter = pFilter->GetString();
904             if (filter == FX_BSTRC("CCITTFaxDecode") || filter == FX_BSTRC("JBIG2Decode")) {
905                 bpc = 1;
906             }
907             if (filter == FX_BSTRC("RunLengthDecode") || filter == FX_BSTRC("DCTDecode")) {
908                 bpc = 8;
909             }
910         } else if (pFilter->GetType() == PDFOBJ_ARRAY) {
911             CPDF_Array *pArray = (CPDF_Array *)pFilter;
912             if (pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("CCITTFacDecode") ||
913                     pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("JBIG2Decode")) {
914                 bpc = 1;
915             }
916             if (pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("RunLengthDecode") ||
917                     pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("DCTDecode")) {
918                 bpc = 8;
919             }
920         }
921     }
922     return bpc;
923 }
924 #define NORMALCOLOR_MAX(color, max) (color) > (max) ? (max) : (color) < 0 ? 0 : (color);
925 void CPDF_DIBSource::TranslateScanline24bpp(FX_LPBYTE dest_scan, FX_LPCBYTE src_scan) const
926 {
927     int max_data = (1 << m_bpc) - 1;
928     if (m_bDefaultDecode) {
929         if (m_Family == PDFCS_DEVICERGB || m_Family == PDFCS_CALRGB) {
930             if (m_bpc == 16) {
931                 FX_LPCBYTE src_pos = src_scan;
932                 for (int col = 0; col < m_Width; col ++) {
933                     *dest_scan++ = src_pos[4];
934                     *dest_scan++ = src_pos[2];
935                     *dest_scan++ = *src_pos;
936                     src_pos += 6;
937                 }
938             } else if (m_bpc == 8) {
939                 FX_LPCBYTE src_pos = src_scan;
940                 for (int column = 0; column < m_Width; column ++) {
941                     *dest_scan++ = src_pos[2];
942                     *dest_scan++ = src_pos[1];
943                     *dest_scan++ = *src_pos;
944                     src_pos += 3;
945                 }
946             } else {
947                 int src_bit_pos = 0;
948                 int dest_byte_pos = 0;
949                 FX_DWORD bpc = GetValidBpc();
950                 for (int column = 0; column < m_Width; column ++) {
951                     int R = _GetBits8(src_scan, src_bit_pos, bpc);
952                     src_bit_pos += bpc;
953                     int G = _GetBits8(src_scan, src_bit_pos, bpc);
954                     src_bit_pos += bpc;
955                     int B = _GetBits8(src_scan, src_bit_pos, bpc);
956                     src_bit_pos += bpc;
957                     R = NORMALCOLOR_MAX(R, max_data);
958                     G = NORMALCOLOR_MAX(G, max_data);
959                     B = NORMALCOLOR_MAX(B, max_data);
960                     dest_scan[dest_byte_pos] = B * 255 / max_data;
961                     dest_scan[dest_byte_pos + 1] = G * 255 / max_data;
962                     dest_scan[dest_byte_pos + 2] = R * 255 / max_data;
963                     dest_byte_pos += 3;
964                 }
965             }
966             return;
967         } else if (m_bpc == 8) {
968             if (m_nComponents == m_pColorSpace->CountComponents())
969                 m_pColorSpace->TranslateImageLine(dest_scan, src_scan, m_Width, m_Width, m_Height,
970                                                   m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK);
971             return;
972         }
973     }
974     CFX_FixedBufGrow<FX_FLOAT, 16> color_values1(m_nComponents);
975     FX_FLOAT* color_values = color_values1;
976     FX_FLOAT R = 0.0f, G = 0.0f, B = 0.0f;
977     if (m_bpc == 8) {
978         int src_byte_pos = 0;
979         int dest_byte_pos = 0;
980         for (int column = 0; column < m_Width; column ++) {
981             for (FX_DWORD color = 0; color < m_nComponents; color ++) {
982                 int data = src_scan[src_byte_pos ++];
983                 color_values[color] = m_pCompData[color].m_DecodeMin +
984                                       m_pCompData[color].m_DecodeStep * data;
985             }
986             if (m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK) {
987                 FX_FLOAT k = 1.0f - color_values[3];
988                 R = (1.0f - color_values[0]) * k;
989                 G = (1.0f - color_values[1]) * k;
990                 B = (1.0f - color_values[2]) * k;
991             } else {
992                 m_pColorSpace->GetRGB(color_values, R, G, B);
993             }
994             R = NORMALCOLOR_MAX(R, 1);
995             G = NORMALCOLOR_MAX(G, 1);
996             B = NORMALCOLOR_MAX(B, 1);
997             dest_scan[dest_byte_pos] = (FX_INT32)(B * 255);
998             dest_scan[dest_byte_pos + 1] = (FX_INT32)(G * 255);
999             dest_scan[dest_byte_pos + 2] = (FX_INT32)(R * 255);
1000             dest_byte_pos += 3;
1001         }
1002     } else {
1003         int src_bit_pos = 0;
1004         int dest_byte_pos = 0;
1005         FX_DWORD bpc = GetValidBpc();
1006         for (int column = 0; column < m_Width; column ++) {
1007             for (FX_DWORD color = 0; color < m_nComponents; color ++) {
1008                 int data = _GetBits8(src_scan, src_bit_pos, bpc);
1009                 color_values[color] = m_pCompData[color].m_DecodeMin +
1010                                       m_pCompData[color].m_DecodeStep * data;
1011                 src_bit_pos += bpc;
1012             }
1013             if (m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK) {
1014                 FX_FLOAT k = 1.0f - color_values[3];
1015                 R = (1.0f - color_values[0]) * k;
1016                 G = (1.0f - color_values[1]) * k;
1017                 B = (1.0f - color_values[2]) * k;
1018             } else {
1019                 m_pColorSpace->GetRGB(color_values, R, G, B);
1020             }
1021             R = NORMALCOLOR_MAX(R, 1);
1022             G = NORMALCOLOR_MAX(G, 1);
1023             B = NORMALCOLOR_MAX(B, 1);
1024             dest_scan[dest_byte_pos] = (FX_INT32)(B * 255);
1025             dest_scan[dest_byte_pos + 1] = (FX_INT32)(G * 255);
1026             dest_scan[dest_byte_pos + 2] = (FX_INT32)(R * 255);
1027             dest_byte_pos += 3;
1028         }
1029     }
1030 }
1031 FX_LPBYTE CPDF_DIBSource::GetBuffer() const
1032 {
1033     if (m_pCachedBitmap) {
1034         return m_pCachedBitmap->GetBuffer();
1035     }
1036     return NULL;
1037 }
1038 FX_LPCBYTE CPDF_DIBSource::GetScanline(int line) const
1039 {
1040     FX_DWORD src_pitch = (m_Width * m_bpc * m_nComponents + 7) / 8;
1041     FX_LPCBYTE pSrcLine = NULL;
1042     if (m_pCachedBitmap) {
1043         if (line >= m_pCachedBitmap->GetHeight()) {
1044             line = m_pCachedBitmap->GetHeight() - 1;
1045         }
1046         pSrcLine = m_pCachedBitmap->GetScanline(line);
1047     } else if (m_pDecoder) {
1048         pSrcLine = m_pDecoder->GetScanline(line);
1049     } else {
1050         if (m_pStreamAcc->GetSize() >= (line + 1) * src_pitch) {
1051             pSrcLine = m_pStreamAcc->GetData() + line * src_pitch;
1052         }
1053     }
1054     if (pSrcLine == NULL) {
1055         FX_LPBYTE pLineBuf = m_pMaskedLine ? m_pMaskedLine : m_pLineBuf;
1056         FXSYS_memset8(pLineBuf, 0xff, m_Pitch);
1057         return pLineBuf;
1058     }
1059     if (m_bpc * m_nComponents == 1) {
1060         if (m_bImageMask && m_bDefaultDecode) {
1061             for (FX_DWORD i = 0; i < src_pitch; i ++) {
1062                 m_pLineBuf[i] = ~pSrcLine[i];
1063             }
1064         } else if (m_bColorKey) {
1065             FX_DWORD reset_argb, set_argb;
1066             reset_argb = m_pPalette ? m_pPalette[0] : 0xff000000;
1067             set_argb = m_pPalette ? m_pPalette[1] : 0xffffffff;
1068             if (m_pCompData[0].m_ColorKeyMin == 0) {
1069                 reset_argb = 0;
1070             }
1071             if (m_pCompData[0].m_ColorKeyMax == 1) {
1072                 set_argb = 0;
1073             }
1074             set_argb = FXARGB_TODIB(set_argb);
1075             reset_argb = FXARGB_TODIB(reset_argb);
1076             FX_DWORD* dest_scan = (FX_DWORD*)m_pMaskedLine;
1077             for (int col = 0; col < m_Width; col ++) {
1078                 if (pSrcLine[col / 8] & (1 << (7 - col % 8))) {
1079                     *dest_scan = set_argb;
1080                 } else {
1081                     *dest_scan = reset_argb;
1082                 }
1083                 dest_scan ++;
1084             }
1085             return m_pMaskedLine;
1086         } else {
1087             FXSYS_memcpy32(m_pLineBuf, pSrcLine, src_pitch);
1088         }
1089         return m_pLineBuf;
1090     }
1091     if (m_bpc * m_nComponents <= 8) {
1092         if (m_bpc == 8) {
1093             FXSYS_memcpy32(m_pLineBuf, pSrcLine, src_pitch);
1094         } else {
1095             int src_bit_pos = 0;
1096             for (int col = 0; col < m_Width; col ++) {
1097                 int color_index = 0;
1098                 for (FX_DWORD color = 0; color < m_nComponents; color ++) {
1099                     int data = _GetBits8(pSrcLine, src_bit_pos, m_bpc);
1100                     color_index |= data << (color * m_bpc);
1101                     src_bit_pos += m_bpc;
1102                 }
1103                 m_pLineBuf[col] = color_index;
1104             }
1105         }
1106         if (m_bColorKey) {
1107             FX_LPBYTE pDestPixel = m_pMaskedLine;
1108             FX_LPCBYTE pSrcPixel = m_pLineBuf;
1109             for (int col = 0; col < m_Width; col ++) {
1110                 FX_BYTE index = *pSrcPixel++;
1111                 if (m_pPalette) {
1112                     *pDestPixel++ = FXARGB_B(m_pPalette[index]);
1113                     *pDestPixel++ = FXARGB_G(m_pPalette[index]);
1114                     *pDestPixel++ = FXARGB_R(m_pPalette[index]);
1115                 } else {
1116                     *pDestPixel++ = index;
1117                     *pDestPixel++ = index;
1118                     *pDestPixel++ = index;
1119                 }
1120                 *pDestPixel = (index < m_pCompData[0].m_ColorKeyMin || index > m_pCompData[0].m_ColorKeyMax) ? 0xff : 0;
1121                 pDestPixel ++ ;
1122             }
1123             return m_pMaskedLine;
1124         }
1125         return m_pLineBuf;
1126     }
1127     if (m_bColorKey) {
1128         if (m_nComponents == 3 && m_bpc == 8) {
1129             FX_LPBYTE alpha_channel = m_pMaskedLine + 3;
1130             for (int col = 0; col < m_Width; col ++) {
1131                 FX_LPCBYTE pPixel = pSrcLine + col * 3;
1132                 alpha_channel[col * 4] = (pPixel[0] < m_pCompData[0].m_ColorKeyMin ||
1133                                           pPixel[0] > m_pCompData[0].m_ColorKeyMax ||
1134                                           pPixel[1] < m_pCompData[1].m_ColorKeyMin || pPixel[1] > m_pCompData[1].m_ColorKeyMax ||
1135                                           pPixel[2] < m_pCompData[2].m_ColorKeyMin || pPixel[2] > m_pCompData[2].m_ColorKeyMax) ? 0xff : 0;
1136             }
1137         } else {
1138             FXSYS_memset8(m_pMaskedLine, 0xff, m_Pitch);
1139         }
1140     }
1141     if (m_pColorSpace) {
1142         TranslateScanline24bpp(m_pLineBuf, pSrcLine);
1143         pSrcLine = m_pLineBuf;
1144     }
1145     if (m_bColorKey) {
1146         FX_LPCBYTE pSrcPixel = pSrcLine;
1147         FX_LPBYTE pDestPixel = m_pMaskedLine;
1148         for (int col = 0; col < m_Width; col ++) {
1149             *pDestPixel++ = *pSrcPixel++;
1150             *pDestPixel++ = *pSrcPixel++;
1151             *pDestPixel++ = *pSrcPixel++;
1152             pDestPixel ++;
1153         }
1154         return m_pMaskedLine;
1155     }
1156     return pSrcLine;
1157 }
1158 FX_BOOL CPDF_DIBSource::SkipToScanline(int line, IFX_Pause* pPause) const
1159 {
1160     if (m_pDecoder) {
1161         return m_pDecoder->SkipToScanline(line, pPause);
1162     }
1163     return FALSE;
1164 }
1165 void CPDF_DIBSource::DownSampleScanline(int line, FX_LPBYTE dest_scan, int dest_bpp,
1166                                         int dest_width, FX_BOOL bFlipX, int clip_left, int clip_width) const
1167 {
1168     if (line < 0 || dest_scan == NULL || dest_bpp <= 0 ||
1169         dest_width <= 0 || clip_left < 0 || clip_width <= 0) {
1170         return;
1171     }
1172
1173     FX_DWORD bpc = GetValidBpc();
1174     FX_DWORD src_width = m_Width;
1175     FX_SAFE_DWORD pitch = src_width;
1176     pitch *= bpc;
1177     pitch *= m_nComponents;
1178     pitch += 7;
1179     pitch /= 8;
1180     if (!pitch.IsValid()) { 
1181         return;
1182     }
1183
1184     FX_LPCBYTE pSrcLine = NULL;
1185     if (m_pCachedBitmap) {
1186         pSrcLine = m_pCachedBitmap->GetScanline(line);
1187     } else if (m_pDecoder) {
1188         pSrcLine = m_pDecoder->GetScanline(line);
1189     } else {
1190         FX_DWORD src_pitch = pitch.ValueOrDie();
1191         pitch *= (line+1);
1192         if (!pitch.IsValid()) { 
1193             return;
1194         }
1195  
1196         if (m_pStreamAcc->GetSize() >= pitch.ValueOrDie()) {
1197             pSrcLine = m_pStreamAcc->GetData() + line * src_pitch;
1198         }
1199     }
1200     int orig_Bpp = bpc * m_nComponents / 8;
1201     int dest_Bpp = dest_bpp / 8;
1202     if (pSrcLine == NULL) {
1203         FXSYS_memset32(dest_scan, 0xff, dest_Bpp * clip_width);
1204         return;
1205     }
1206
1207     FX_SAFE_INT max_src_x = clip_left;
1208     max_src_x += clip_width - 1;
1209     max_src_x *= src_width;
1210     max_src_x /= dest_width;
1211     if (!max_src_x.IsValid()) {
1212         return;
1213     }
1214
1215     CFX_FixedBufGrow<FX_BYTE, 128> temp(orig_Bpp);
1216     if (bpc * m_nComponents == 1) {
1217         FX_DWORD set_argb = (FX_DWORD) - 1, reset_argb = 0;
1218         if (m_bImageMask) {
1219             if (m_bDefaultDecode) {
1220                 set_argb = 0;
1221                 reset_argb = (FX_DWORD) - 1;
1222             }
1223         } else if (m_bColorKey) {
1224             reset_argb = m_pPalette ? m_pPalette[0] : 0xff000000;
1225             set_argb = m_pPalette ? m_pPalette[1] : 0xffffffff;
1226             if (m_pCompData[0].m_ColorKeyMin == 0) {
1227                 reset_argb = 0;
1228             }
1229             if (m_pCompData[0].m_ColorKeyMax == 1) {
1230                 set_argb = 0;
1231             }
1232             set_argb = FXARGB_TODIB(set_argb);
1233             reset_argb = FXARGB_TODIB(reset_argb);
1234             for (int i = 0; i < clip_width; i ++) {
1235                 FX_DWORD src_x = (clip_left + i) * src_width / dest_width;
1236                 if (bFlipX) {
1237                     src_x = src_width - src_x - 1;
1238                 }
1239                 src_x %= src_width;
1240                 if (pSrcLine[src_x / 8] & (1 << (7 - src_x % 8))) {
1241                     ((FX_DWORD*)dest_scan)[i] = set_argb;
1242                 } else {
1243                     ((FX_DWORD*)dest_scan)[i] = reset_argb;
1244                 }
1245             }
1246             return;
1247         } else {
1248             if (dest_Bpp == 1) {
1249             } else if (m_pPalette) {
1250                 reset_argb = m_pPalette[0];
1251                 set_argb = m_pPalette[1];
1252             }
1253         }
1254         for (int i = 0; i < clip_width; i ++) {
1255             FX_DWORD src_x = (clip_left + i) * src_width / dest_width;
1256             if (bFlipX) {
1257                 src_x = src_width - src_x - 1;
1258             }
1259             src_x %= src_width;
1260             int dest_pos = i * dest_Bpp;
1261             if (pSrcLine[src_x / 8] & (1 << (7 - src_x % 8))) {
1262                 if (dest_Bpp == 1) {
1263                     dest_scan[dest_pos] = (FX_BYTE)set_argb;
1264                 } else if (dest_Bpp == 3) {
1265                     dest_scan[dest_pos] = FXARGB_B(set_argb);
1266                     dest_scan[dest_pos + 1] = FXARGB_G(set_argb);
1267                     dest_scan[dest_pos + 2] = FXARGB_R(set_argb);
1268                 } else {
1269                     *(FX_DWORD*)(dest_scan + dest_pos) = set_argb;
1270                 }
1271             } else {
1272                 if (dest_Bpp == 1) {
1273                     dest_scan[dest_pos] = (FX_BYTE)reset_argb;
1274                 } else if (dest_Bpp == 3) {
1275                     dest_scan[dest_pos] = FXARGB_B(reset_argb);
1276                     dest_scan[dest_pos + 1] = FXARGB_G(reset_argb);
1277                     dest_scan[dest_pos + 2] = FXARGB_R(reset_argb);
1278                 } else {
1279                     *(FX_DWORD*)(dest_scan + dest_pos) = reset_argb;
1280                 }
1281             }
1282         }
1283         return;
1284     } else if (bpc * m_nComponents <= 8) {
1285         if (bpc < 8) {
1286             int src_bit_pos = 0;
1287             for (FX_DWORD col = 0; col < src_width; col ++) {
1288                 int color_index = 0;
1289                 for (FX_DWORD color = 0; color < m_nComponents; color ++) {
1290                     int data = _GetBits8(pSrcLine, src_bit_pos, bpc);
1291                     color_index |= data << (color * bpc);
1292                     src_bit_pos += bpc;
1293                 }
1294                 m_pLineBuf[col] = color_index;
1295             }
1296             pSrcLine = m_pLineBuf;
1297         }
1298         if (m_bColorKey) {
1299             for (int i = 0; i < clip_width; i ++) {
1300                 FX_DWORD src_x = (clip_left + i) * src_width / dest_width;
1301                 if (bFlipX) {
1302                     src_x = src_width - src_x - 1;
1303                 }
1304                 src_x %= src_width;
1305                 FX_LPBYTE pDestPixel = dest_scan + i * 4;
1306                 FX_BYTE index = pSrcLine[src_x];
1307                 if (m_pPalette) {
1308                     *pDestPixel++ = FXARGB_B(m_pPalette[index]);
1309                     *pDestPixel++ = FXARGB_G(m_pPalette[index]);
1310                     *pDestPixel++ = FXARGB_R(m_pPalette[index]);
1311                 } else {
1312                     *pDestPixel++ = index;
1313                     *pDestPixel++ = index;
1314                     *pDestPixel++ = index;
1315                 }
1316                 *pDestPixel = (index < m_pCompData[0].m_ColorKeyMin || index > m_pCompData[0].m_ColorKeyMax) ? 0xff : 0;
1317             }
1318             return;
1319         }
1320         for (int i = 0; i < clip_width; i ++) {
1321             FX_DWORD src_x = (clip_left + i) * src_width / dest_width;
1322             if (bFlipX) {
1323                 src_x = src_width - src_x - 1;
1324             }
1325             src_x %= src_width;
1326             FX_BYTE index = pSrcLine[src_x];
1327             if (dest_Bpp == 1) {
1328                 dest_scan[i] = index;
1329             } else {
1330                 int dest_pos = i * dest_Bpp;
1331                 FX_ARGB argb = m_pPalette[index];
1332                 dest_scan[dest_pos] = FXARGB_B(argb);
1333                 dest_scan[dest_pos + 1] = FXARGB_G(argb);
1334                 dest_scan[dest_pos + 2] = FXARGB_R(argb);
1335             }
1336         }
1337         return;
1338     } else {
1339         int last_src_x = -1;
1340         FX_ARGB last_argb;
1341         FX_FLOAT orig_Not8Bpp = (FX_FLOAT)bpc * (FX_FLOAT)m_nComponents / 8.0f;
1342         FX_FLOAT unit_To8Bpc = 255.0f / ((1 << bpc) - 1);
1343         for (int i = 0; i < clip_width; i ++) {
1344             int dest_x = clip_left + i;
1345             FX_DWORD src_x = (bFlipX ? (dest_width - dest_x - 1) : dest_x) * (FX_INT64)src_width / dest_width;
1346             src_x %= src_width;
1347             FX_LPCBYTE pSrcPixel = NULL;
1348             if (bpc % 8 == 0) {
1349                 pSrcPixel = pSrcLine + src_x * orig_Bpp;
1350             } else {
1351                 pSrcPixel = pSrcLine + (int)(src_x * orig_Not8Bpp);
1352             }
1353             FX_LPBYTE pDestPixel = dest_scan + i * dest_Bpp;
1354             FX_ARGB argb;
1355             if (src_x == last_src_x) {
1356                 argb = last_argb;
1357             } else {
1358                 if (m_pColorSpace) {
1359                     FX_BYTE color[4];
1360                     if (!m_bDefaultDecode) {
1361                         for (int i = 0; i < orig_Bpp; i ++) {
1362                             int color_value = (int)((m_pCompData[i].m_DecodeMin + m_pCompData[i].m_DecodeStep * (FX_FLOAT)pSrcPixel[i]) * 255.0f + 0.5f);
1363                             temp[i] = color_value > 255 ? 255 : (color_value < 0 ? 0 : color_value);
1364                         }
1365                         m_pColorSpace->TranslateImageLine(color, temp, 1, 0, 0, m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK);
1366                     } else {
1367                         if (bpc < 8) {
1368                             int src_bit_pos = 0;
1369                             if (src_x % 2) {
1370                                 src_bit_pos = 4;
1371                             }
1372                             for (FX_DWORD i = 0; i < m_nComponents; i ++) {
1373                                 temp[i] = (FX_BYTE)(_GetBits8(pSrcPixel, src_bit_pos, bpc) * unit_To8Bpc);
1374                                 src_bit_pos += bpc;
1375                             }
1376                             m_pColorSpace->TranslateImageLine(color, temp, 1, 0, 0, m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK);
1377                         } else {
1378                             m_pColorSpace->TranslateImageLine(color, pSrcPixel, 1, 0, 0, m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK);
1379                         }
1380                     }
1381                     argb = FXARGB_MAKE(0xff, color[2], color[1], color[0]);
1382                 } else {
1383                     argb = FXARGB_MAKE(0xff, pSrcPixel[2], pSrcPixel[1], pSrcPixel[0]);
1384                 }
1385                 if (m_bColorKey) {
1386                     int alpha = 0xff;
1387                     if (m_nComponents == 3 && bpc == 8) {
1388                         alpha = (pSrcPixel[0] < m_pCompData[0].m_ColorKeyMin ||
1389                                  pSrcPixel[0] > m_pCompData[0].m_ColorKeyMax ||
1390                                  pSrcPixel[1] < m_pCompData[1].m_ColorKeyMin ||
1391                                  pSrcPixel[1] > m_pCompData[1].m_ColorKeyMax ||
1392                                  pSrcPixel[2] < m_pCompData[2].m_ColorKeyMin ||
1393                                  pSrcPixel[2] > m_pCompData[2].m_ColorKeyMax) ? 0xff : 0;
1394                     }
1395                     argb &= 0xffffff;
1396                     argb |= alpha << 24;
1397                 }
1398                 last_src_x = src_x;
1399                 last_argb = argb;
1400             }
1401             if (dest_Bpp == 4) {
1402                 *(FX_DWORD*)pDestPixel = FXARGB_TODIB(argb);
1403             } else {
1404                 *pDestPixel++ = FXARGB_B(argb);
1405                 *pDestPixel++ = FXARGB_G(argb);
1406                 *pDestPixel = FXARGB_R(argb);
1407             }
1408         }
1409     }
1410 }
1411 void CPDF_DIBSource::SetDownSampleSize(int dest_width, int dest_height) const
1412 {
1413     if (m_pDecoder) {
1414         m_pDecoder->DownScale(dest_width, dest_height);
1415         ((CPDF_DIBSource*)this)->m_Width = m_pDecoder->GetWidth();
1416         ((CPDF_DIBSource*)this)->m_Height = m_pDecoder->GetHeight();
1417     }
1418 }
1419 void CPDF_DIBSource::ClearImageData()
1420 {
1421     if (m_pDecoder) {
1422         m_pDecoder->ClearImageData();
1423     }
1424 }
1425 CPDF_ProgressiveImageLoaderHandle::CPDF_ProgressiveImageLoaderHandle()
1426 {
1427     m_pImageLoader = NULL;
1428     m_pCache = NULL;
1429     m_pImage = NULL;
1430 }
1431 CPDF_ProgressiveImageLoaderHandle::~CPDF_ProgressiveImageLoaderHandle()
1432 {
1433     m_pImageLoader = NULL;
1434     m_pCache = NULL;
1435     m_pImage = NULL;
1436 }
1437 FX_BOOL CPDF_ProgressiveImageLoaderHandle::Start(CPDF_ImageLoader* pImageLoader, const CPDF_ImageObject* pImage, CPDF_PageRenderCache* pCache, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask, CPDF_RenderStatus* pRenderStatus, FX_INT32 nDownsampleWidth, FX_INT32 nDownsampleHeight)
1438 {
1439     m_pImageLoader = pImageLoader;
1440     m_pCache = pCache;
1441     m_pImage = (CPDF_ImageObject*)pImage;
1442     m_nDownsampleWidth = nDownsampleWidth;
1443     m_nDownsampleHeight = nDownsampleHeight;
1444     FX_BOOL ret;
1445     if (pCache) {
1446         ret = pCache->StartGetCachedBitmap(pImage->m_pImage->GetStream(), bStdCS, GroupFamily, bLoadMask, pRenderStatus, m_nDownsampleWidth, m_nDownsampleHeight);
1447         if (ret == FALSE) {
1448             m_pImageLoader->m_bCached = TRUE;
1449             m_pImageLoader->m_pBitmap = pCache->m_pCurImageCache->DetachBitmap();
1450             m_pImageLoader->m_pMask = pCache->m_pCurImageCache->DetachMask();
1451             m_pImageLoader->m_MatteColor = pCache->m_pCurImageCache->m_MatteColor;
1452         }
1453     } else {
1454         ret = pImage->m_pImage->StartLoadDIBSource(pRenderStatus->m_pFormResource, pRenderStatus->m_pPageResource, bStdCS, GroupFamily, bLoadMask);
1455         if (ret == FALSE) {
1456             m_pImageLoader->m_bCached = FALSE;
1457             m_pImageLoader->m_pBitmap = m_pImage->m_pImage->DetachBitmap();
1458             m_pImageLoader->m_pMask = m_pImage->m_pImage->DetachMask();
1459             m_pImageLoader->m_MatteColor = m_pImage->m_pImage->m_MatteColor;
1460         }
1461     }
1462     return ret;
1463 }
1464 FX_BOOL CPDF_ProgressiveImageLoaderHandle::Continue(IFX_Pause* pPause)
1465 {
1466     FX_BOOL ret;
1467     if (m_pCache) {
1468         ret = m_pCache->Continue(pPause);
1469         if (ret == FALSE) {
1470             m_pImageLoader->m_bCached = TRUE;
1471             m_pImageLoader->m_pBitmap = m_pCache->m_pCurImageCache->DetachBitmap();
1472             m_pImageLoader->m_pMask = m_pCache->m_pCurImageCache->DetachMask();
1473             m_pImageLoader->m_MatteColor = m_pCache->m_pCurImageCache->m_MatteColor;
1474         }
1475     } else {
1476         ret = m_pImage->m_pImage->Continue(pPause);
1477         if (ret == FALSE) {
1478             m_pImageLoader->m_bCached = FALSE;
1479             m_pImageLoader->m_pBitmap = m_pImage->m_pImage->DetachBitmap();
1480             m_pImageLoader->m_pMask = m_pImage->m_pImage->DetachMask();
1481             m_pImageLoader->m_MatteColor = m_pImage->m_pImage->m_MatteColor;
1482         }
1483     }
1484     return ret;
1485 }
1486 FX_BOOL CPDF_ImageLoader::Load(const CPDF_ImageObject* pImage, CPDF_PageRenderCache* pCache, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask, CPDF_RenderStatus* pRenderStatus)
1487 {
1488     if (pImage == NULL) {
1489         return FALSE;
1490     }
1491     if (pCache) {
1492         pCache->GetCachedBitmap(pImage->m_pImage->GetStream(), m_pBitmap, m_pMask, m_MatteColor, bStdCS, GroupFamily, bLoadMask, pRenderStatus, m_nDownsampleWidth, m_nDownsampleHeight);
1493         m_bCached = TRUE;
1494     } else {
1495         m_pBitmap = pImage->m_pImage->LoadDIBSource(&m_pMask, &m_MatteColor, bStdCS, GroupFamily, bLoadMask);
1496         m_bCached = FALSE;
1497     }
1498     return FALSE;
1499 }
1500 FX_BOOL CPDF_ImageLoader::StartLoadImage(const CPDF_ImageObject* pImage, CPDF_PageRenderCache* pCache, FX_LPVOID& LoadHandle, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask, CPDF_RenderStatus* pRenderStatus, FX_INT32 nDownsampleWidth, FX_INT32 nDownsampleHeight)
1501 {
1502     m_nDownsampleWidth = nDownsampleWidth;
1503     m_nDownsampleHeight = nDownsampleHeight;
1504     CPDF_ProgressiveImageLoaderHandle* pLoaderHandle = NULL;
1505     pLoaderHandle =     FX_NEW CPDF_ProgressiveImageLoaderHandle;
1506     FX_BOOL ret = pLoaderHandle->Start(this, pImage, pCache, bStdCS, GroupFamily, bLoadMask, pRenderStatus, m_nDownsampleWidth, m_nDownsampleHeight);
1507     LoadHandle = pLoaderHandle;
1508     return ret;
1509 }
1510 FX_BOOL CPDF_ImageLoader::Continue(FX_LPVOID LoadHandle, IFX_Pause* pPause)
1511 {
1512     return ((CPDF_ProgressiveImageLoaderHandle*)LoadHandle)->Continue(pPause);
1513 }
1514 CPDF_ImageLoader::~CPDF_ImageLoader()
1515 {
1516     if (!m_bCached) {
1517         if (m_pBitmap) {
1518             delete m_pBitmap;
1519             m_pBitmap = NULL;
1520         }
1521         if (m_pMask) {
1522             delete m_pMask;
1523         }
1524     }
1525 }