OLD | NEW |
(Empty) | |
| 1 // Copyright 2016 The 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 #include <cstdint> |
| 6 |
| 7 #include "core/fpdfapi/fpdf_parser/include/cpdf_stream.h" |
| 8 #include "core/fpdfapi/fpdf_parser/include/cpdf_stream_acc.h" |
| 9 #include "core/fxcodec/codec/ccodec_jbig2module.h" |
| 10 #include "core/fxcodec/include/JBig2_DocumentContext.h" |
| 11 #include "core/fxcodec/jbig2/JBig2_Context.h" |
| 12 #include "core/fxge/include/fx_dib.h" |
| 13 |
| 14 static uint32_t GetInteger(const uint8_t* data) { |
| 15 return data[0] | data[1] << 8 | data[2] << 16 | data[3] << 24; |
| 16 } |
| 17 |
| 18 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { |
| 19 const size_t kParameterSize = 8; |
| 20 if (size < kParameterSize) |
| 21 return 0; |
| 22 |
| 23 uint32_t width = GetInteger(data); |
| 24 uint32_t height = GetInteger(data + 4); |
| 25 size -= kParameterSize; |
| 26 data += kParameterSize; |
| 27 |
| 28 std::unique_ptr<CFX_DIBitmap> bitmap(new CFX_DIBitmap); |
| 29 if (!bitmap->Create(width, height, FXDIB_1bppRgb)) |
| 30 return 0; |
| 31 |
| 32 std::unique_ptr<CPDF_Object, ReleaseDeleter<CPDF_Object>> stream( |
| 33 new CPDF_Stream); |
| 34 stream->AsStream()->SetData(data, size); |
| 35 CPDF_StreamAcc src_stream; |
| 36 src_stream.LoadAllData(stream->AsStream(), TRUE); |
| 37 |
| 38 CCodec_Jbig2Module module; |
| 39 CCodec_Jbig2Context jbig2_context; |
| 40 std::unique_ptr<JBig2_DocumentContext> document_context; |
| 41 FXCODEC_STATUS status = module.StartDecode( |
| 42 &jbig2_context, &document_context, width, height, &src_stream, nullptr, |
| 43 bitmap->GetBuffer(), bitmap->GetPitch(), nullptr); |
| 44 |
| 45 while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE) |
| 46 status = module.ContinueDecode(&jbig2_context, nullptr); |
| 47 return 0; |
| 48 } |
OLD | NEW |