| OLD | NEW |
| 1 // Copyright 2015 PDFium Authors. All rights reserved. | 1 // Copyright 2015 PDFium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // 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/fxcodec/jbig2/JBig2_SddProc.h" | 7 #include "core/fxcodec/jbig2/JBig2_SddProc.h" |
| 8 | 8 |
| 9 #include <memory> | 9 #include <memory> |
| 10 #include <vector> | 10 #include <vector> |
| 11 | 11 |
| 12 #include "core/fxcodec/jbig2/JBig2_ArithIntDecoder.h" | 12 #include "core/fxcodec/jbig2/JBig2_ArithIntDecoder.h" |
| 13 #include "core/fxcodec/jbig2/JBig2_GrdProc.h" | 13 #include "core/fxcodec/jbig2/JBig2_GrdProc.h" |
| 14 #include "core/fxcodec/jbig2/JBig2_GrrdProc.h" | 14 #include "core/fxcodec/jbig2/JBig2_GrrdProc.h" |
| 15 #include "core/fxcodec/jbig2/JBig2_HuffmanDecoder.h" | 15 #include "core/fxcodec/jbig2/JBig2_HuffmanDecoder.h" |
| 16 #include "core/fxcodec/jbig2/JBig2_HuffmanTable.h" | 16 #include "core/fxcodec/jbig2/JBig2_HuffmanTable.h" |
| 17 #include "core/fxcodec/jbig2/JBig2_HuffmanTable_Standard.h" | 17 #include "core/fxcodec/jbig2/JBig2_HuffmanTable_Standard.h" |
| 18 #include "core/fxcodec/jbig2/JBig2_SymbolDict.h" | 18 #include "core/fxcodec/jbig2/JBig2_SymbolDict.h" |
| 19 #include "core/fxcodec/jbig2/JBig2_TrdProc.h" | 19 #include "core/fxcodec/jbig2/JBig2_TrdProc.h" |
| 20 #include "core/fxcrt/fx_basic.h" | 20 #include "core/fxcrt/fx_basic.h" |
| 21 #include "third_party/base/ptr_util.h" |
| 21 | 22 |
| 22 CJBig2_SymbolDict* CJBig2_SDDProc::decode_Arith( | 23 CJBig2_SymbolDict* CJBig2_SDDProc::decode_Arith( |
| 23 CJBig2_ArithDecoder* pArithDecoder, | 24 CJBig2_ArithDecoder* pArithDecoder, |
| 24 std::vector<JBig2ArithCtx>* gbContext, | 25 std::vector<JBig2ArithCtx>* gbContext, |
| 25 std::vector<JBig2ArithCtx>* grContext) { | 26 std::vector<JBig2ArithCtx>* grContext) { |
| 26 CJBig2_Image** SDNEWSYMS; | 27 CJBig2_Image** SDNEWSYMS; |
| 27 uint32_t HCHEIGHT, NSYMSDECODED; | 28 uint32_t HCHEIGHT, NSYMSDECODED; |
| 28 int32_t HCDH; | 29 int32_t HCDH; |
| 29 uint32_t SYMWIDTH, TOTWIDTH; | 30 uint32_t SYMWIDTH, TOTWIDTH; |
| 30 int32_t DW; | 31 int32_t DW; |
| (...skipping 21 matching lines...) Expand all Loading... |
| 52 std::unique_ptr<CJBig2_ArithIntDecoder> IAFS(new CJBig2_ArithIntDecoder); | 53 std::unique_ptr<CJBig2_ArithIntDecoder> IAFS(new CJBig2_ArithIntDecoder); |
| 53 std::unique_ptr<CJBig2_ArithIntDecoder> IADS(new CJBig2_ArithIntDecoder); | 54 std::unique_ptr<CJBig2_ArithIntDecoder> IADS(new CJBig2_ArithIntDecoder); |
| 54 std::unique_ptr<CJBig2_ArithIntDecoder> IAIT(new CJBig2_ArithIntDecoder); | 55 std::unique_ptr<CJBig2_ArithIntDecoder> IAIT(new CJBig2_ArithIntDecoder); |
| 55 std::unique_ptr<CJBig2_ArithIntDecoder> IARI(new CJBig2_ArithIntDecoder); | 56 std::unique_ptr<CJBig2_ArithIntDecoder> IARI(new CJBig2_ArithIntDecoder); |
| 56 std::unique_ptr<CJBig2_ArithIntDecoder> IARDW(new CJBig2_ArithIntDecoder); | 57 std::unique_ptr<CJBig2_ArithIntDecoder> IARDW(new CJBig2_ArithIntDecoder); |
| 57 std::unique_ptr<CJBig2_ArithIntDecoder> IARDH(new CJBig2_ArithIntDecoder); | 58 std::unique_ptr<CJBig2_ArithIntDecoder> IARDH(new CJBig2_ArithIntDecoder); |
| 58 nTmp = 0; | 59 nTmp = 0; |
| 59 while ((uint32_t)(1 << nTmp) < (SDNUMINSYMS + SDNUMNEWSYMS)) { | 60 while ((uint32_t)(1 << nTmp) < (SDNUMINSYMS + SDNUMNEWSYMS)) { |
| 60 nTmp++; | 61 nTmp++; |
| 61 } | 62 } |
| 62 IAID.reset(new CJBig2_ArithIaidDecoder((uint8_t)nTmp)); | 63 IAID = pdfium::MakeUnique<CJBig2_ArithIaidDecoder>((uint8_t)nTmp); |
| 63 SDNEWSYMS = FX_Alloc(CJBig2_Image*, SDNUMNEWSYMS); | 64 SDNEWSYMS = FX_Alloc(CJBig2_Image*, SDNUMNEWSYMS); |
| 64 FXSYS_memset(SDNEWSYMS, 0, SDNUMNEWSYMS * sizeof(CJBig2_Image*)); | 65 FXSYS_memset(SDNEWSYMS, 0, SDNUMNEWSYMS * sizeof(CJBig2_Image*)); |
| 65 | 66 |
| 66 HCHEIGHT = 0; | 67 HCHEIGHT = 0; |
| 67 NSYMSDECODED = 0; | 68 NSYMSDECODED = 0; |
| 68 while (NSYMSDECODED < SDNUMNEWSYMS) { | 69 while (NSYMSDECODED < SDNUMNEWSYMS) { |
| 69 BS = nullptr; | 70 BS = nullptr; |
| 70 IADH->decode(pArithDecoder, &HCDH); | 71 IADH->decode(pArithDecoder, &HCDH); |
| 71 HCHEIGHT = HCHEIGHT + HCDH; | 72 HCHEIGHT = HCHEIGHT + HCDH; |
| 72 if ((int)HCHEIGHT < 0 || (int)HCHEIGHT > JBIG2_MAX_IMAGE_SIZE) { | 73 if ((int)HCHEIGHT < 0 || (int)HCHEIGHT > JBIG2_MAX_IMAGE_SIZE) { |
| (...skipping 178 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 251 } | 252 } |
| 252 } | 253 } |
| 253 EXINDEX = EXINDEX + EXRUNLENGTH; | 254 EXINDEX = EXINDEX + EXRUNLENGTH; |
| 254 CUREXFLAG = !CUREXFLAG; | 255 CUREXFLAG = !CUREXFLAG; |
| 255 } | 256 } |
| 256 if (num_ex_syms > SDNUMEXSYMS) { | 257 if (num_ex_syms > SDNUMEXSYMS) { |
| 257 FX_Free(EXFLAGS); | 258 FX_Free(EXFLAGS); |
| 258 goto failed; | 259 goto failed; |
| 259 } | 260 } |
| 260 | 261 |
| 261 pDict.reset(new CJBig2_SymbolDict); | 262 pDict = pdfium::MakeUnique<CJBig2_SymbolDict>(); |
| 262 I = J = 0; | 263 I = J = 0; |
| 263 for (I = 0; I < SDNUMINSYMS + SDNUMNEWSYMS; I++) { | 264 for (I = 0; I < SDNUMINSYMS + SDNUMNEWSYMS; I++) { |
| 264 if (EXFLAGS[I] && J < SDNUMEXSYMS) { | 265 if (EXFLAGS[I] && J < SDNUMEXSYMS) { |
| 265 if (I < SDNUMINSYMS) { | 266 if (I < SDNUMINSYMS) { |
| 266 pDict->AddImage(SDINSYMS[I] ? new CJBig2_Image(*SDINSYMS[I]) : nullptr); | 267 pDict->AddImage(SDINSYMS[I] ? new CJBig2_Image(*SDINSYMS[I]) : nullptr); |
| 267 } else { | 268 } else { |
| 268 pDict->AddImage(SDNEWSYMS[I - SDNUMINSYMS]); | 269 pDict->AddImage(SDNEWSYMS[I - SDNUMINSYMS]); |
| 269 } | 270 } |
| 270 ++J; | 271 ++J; |
| 271 } else if (!EXFLAGS[I] && I >= SDNUMINSYMS) { | 272 } else if (!EXFLAGS[I] && I >= SDNUMINSYMS) { |
| (...skipping 283 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 555 for (I = HCFIRSTSYM; I < NSYMSDECODED; I++) { | 556 for (I = HCFIRSTSYM; I < NSYMSDECODED; I++) { |
| 556 SDNEWSYMS[I] = BHC->subImage(nTmp, 0, SDNEWSYMWIDTHS[I], HCHEIGHT); | 557 SDNEWSYMS[I] = BHC->subImage(nTmp, 0, SDNEWSYMWIDTHS[I], HCHEIGHT); |
| 557 nTmp += SDNEWSYMWIDTHS[I]; | 558 nTmp += SDNEWSYMWIDTHS[I]; |
| 558 } | 559 } |
| 559 delete BHC; | 560 delete BHC; |
| 560 BHC = nullptr; | 561 BHC = nullptr; |
| 561 } | 562 } |
| 562 } | 563 } |
| 563 EXINDEX = 0; | 564 EXINDEX = 0; |
| 564 CUREXFLAG = 0; | 565 CUREXFLAG = 0; |
| 565 pTable.reset(new CJBig2_HuffmanTable(HuffmanTable_B1, HuffmanTable_B1_Size, | 566 pTable = pdfium::MakeUnique<CJBig2_HuffmanTable>( |
| 566 HuffmanTable_HTOOB_B1)); | 567 HuffmanTable_B1, HuffmanTable_B1_Size, HuffmanTable_HTOOB_B1); |
| 567 EXFLAGS = FX_Alloc(bool, SDNUMINSYMS + SDNUMNEWSYMS); | 568 EXFLAGS = FX_Alloc(bool, SDNUMINSYMS + SDNUMNEWSYMS); |
| 568 num_ex_syms = 0; | 569 num_ex_syms = 0; |
| 569 while (EXINDEX < SDNUMINSYMS + SDNUMNEWSYMS) { | 570 while (EXINDEX < SDNUMINSYMS + SDNUMNEWSYMS) { |
| 570 if (pHuffmanDecoder->decodeAValue(pTable.get(), (int*)&EXRUNLENGTH) != 0) { | 571 if (pHuffmanDecoder->decodeAValue(pTable.get(), (int*)&EXRUNLENGTH) != 0) { |
| 571 FX_Free(EXFLAGS); | 572 FX_Free(EXFLAGS); |
| 572 goto failed; | 573 goto failed; |
| 573 } | 574 } |
| 574 if (EXINDEX + EXRUNLENGTH > SDNUMINSYMS + SDNUMNEWSYMS) { | 575 if (EXINDEX + EXRUNLENGTH > SDNUMINSYMS + SDNUMNEWSYMS) { |
| 575 FX_Free(EXFLAGS); | 576 FX_Free(EXFLAGS); |
| 576 goto failed; | 577 goto failed; |
| (...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 613 failed: | 614 failed: |
| 614 for (I = 0; I < NSYMSDECODED; I++) { | 615 for (I = 0; I < NSYMSDECODED; I++) { |
| 615 delete SDNEWSYMS[I]; | 616 delete SDNEWSYMS[I]; |
| 616 } | 617 } |
| 617 FX_Free(SDNEWSYMS); | 618 FX_Free(SDNEWSYMS); |
| 618 if (SDREFAGG == 0) { | 619 if (SDREFAGG == 0) { |
| 619 FX_Free(SDNEWSYMWIDTHS); | 620 FX_Free(SDNEWSYMWIDTHS); |
| 620 } | 621 } |
| 621 return nullptr; | 622 return nullptr; |
| 622 } | 623 } |
| OLD | NEW |