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/parser/cpdf_stream.h"
8 #include "core/fpdfapi/parser/cpdf_stream_acc.h"
9 #include "core/fxcodec/JBig2_DocumentContext.h"
10 #include "core/fxcodec/codec/ccodec_jbig2module.h"
11 #include "core/fxcodec/jbig2/JBig2_Context.h"
12 #include "core/fxge/fx_dib.h"
13
GetInteger(const uint8_t * data)14 static uint32_t GetInteger(const uint8_t* data) {
15 return data[0] | data[1] << 8 | data[2] << 16 | data[3] << 24;
16 }
17
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)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> stream(new CPDF_Stream);
33 stream->AsStream()->SetData(data, size);
34 CPDF_StreamAcc src_stream;
35 src_stream.LoadAllData(stream->AsStream(), true);
36
37 CCodec_Jbig2Module module;
38 CCodec_Jbig2Context jbig2_context;
39 std::unique_ptr<JBig2_DocumentContext> document_context;
40 FXCODEC_STATUS status = module.StartDecode(
41 &jbig2_context, &document_context, width, height, &src_stream, nullptr,
42 bitmap->GetBuffer(), bitmap->GetPitch(), nullptr);
43
44 while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE)
45 status = module.ContinueDecode(&jbig2_context, nullptr);
46 return 0;
47 }
48