diff options
author | Lei Zhang <thestig@chromium.org> | 2018-05-25 21:47:19 +0000 |
---|---|---|
committer | Chromium commit bot <commit-bot@chromium.org> | 2018-05-25 21:47:19 +0000 |
commit | fa4d93a08d5cd4f349c480b194a3e795273b27ed (patch) | |
tree | 522fdac3c130bf525814c0c745a9cffb2f6d6a4b /testing/fuzzers/pdf_codec_jbig2_fuzzer.cc | |
parent | 0d86f765b8361b7f9f3a5fcc659de2f52c806bd0 (diff) | |
download | pdfium-fa4d93a08d5cd4f349c480b194a3e795273b27ed.tar.xz |
Move fuzzers to testing/fuzzers.
Move them out of testing/libfuzzer, to make it possible to pull
libfuzzer into that directory. Leave testing/libfuzzer/BUILD.gn there
for now as a transitional build file.
BUG=pdfium:1088
Change-Id: I4126d89dd3e075ac63477a4860e029c135866dbe
Reviewed-on: https://pdfium-review.googlesource.com/32896
Commit-Queue: Lei Zhang <thestig@chromium.org>
Reviewed-by: dsinclair <dsinclair@chromium.org>
Diffstat (limited to 'testing/fuzzers/pdf_codec_jbig2_fuzzer.cc')
-rw-r--r-- | testing/fuzzers/pdf_codec_jbig2_fuzzer.cc | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/testing/fuzzers/pdf_codec_jbig2_fuzzer.cc b/testing/fuzzers/pdf_codec_jbig2_fuzzer.cc new file mode 100644 index 0000000000..7b8e2aac33 --- /dev/null +++ b/testing/fuzzers/pdf_codec_jbig2_fuzzer.cc @@ -0,0 +1,59 @@ +// Copyright 2016 The PDFium Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. + +#include <cstdint> + +#include "core/fpdfapi/parser/cpdf_stream.h" +#include "core/fpdfapi/parser/cpdf_stream_acc.h" +#include "core/fxcodec/JBig2_DocumentContext.h" +#include "core/fxcodec/codec/ccodec_jbig2module.h" +#include "core/fxcodec/jbig2/JBig2_Context.h" +#include "core/fxcrt/fx_safe_types.h" +#include "core/fxge/dib/cfx_dibitmap.h" +#include "core/fxge/fx_dib.h" +#include "third_party/base/ptr_util.h" + +static uint32_t GetInteger(const uint8_t* data) { + return data[0] | data[1] << 8 | data[2] << 16 | data[3] << 24; +} + +extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { + const size_t kParameterSize = 8; + if (size < kParameterSize) + return 0; + + uint32_t width = GetInteger(data); + uint32_t height = GetInteger(data + 4); + size -= kParameterSize; + data += kParameterSize; + + static constexpr uint32_t kMemLimit = 512000000; // 512 MB + static constexpr uint32_t k1bppRgbComponents = 4; // From CFX_DIBitmap impl. + FX_SAFE_UINT32 mem = width; + mem *= height; + mem *= k1bppRgbComponents; + if (!mem.IsValid() || mem.ValueOrDie() > kMemLimit) + return 0; + + auto bitmap = pdfium::MakeRetain<CFX_DIBitmap>(); + if (!bitmap->Create(width, height, FXDIB_1bppRgb)) + return 0; + + auto stream = pdfium::MakeUnique<CPDF_Stream>(); + stream->AsStream()->SetData(data, size); + + auto src_stream = pdfium::MakeRetain<CPDF_StreamAcc>(stream->AsStream()); + src_stream->LoadAllDataRaw(); + + CCodec_Jbig2Module module; + CCodec_Jbig2Context jbig2_context; + std::unique_ptr<JBig2_DocumentContext> document_context; + FXCODEC_STATUS status = module.StartDecode( + &jbig2_context, &document_context, width, height, src_stream, nullptr, + bitmap->GetBuffer(), bitmap->GetPitch(), nullptr); + + while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE) + status = module.ContinueDecode(&jbig2_context, nullptr); + return 0; +} |