From 833619b4441915c7c55085d44b3221eaef0d9800 Mon Sep 17 00:00:00 2001 From: tsepez Date: Wed, 7 Dec 2016 09:21:17 -0800 Subject: Refcount all the IFX_ stream classes all the time. We can remove a lot of "bOwnsStream" logic in the process. Always pass these by const reference, in case the called method wants to hang on to the stream (one exception is where we stick a raw pointer into a void* slot in a context from another layer). Review-Url: https://codereview.chromium.org/2451493002 --- testing/libfuzzer/pdf_cfx_saxreader_fuzzer.cc | 9 +++------ testing/libfuzzer/xfa_codec_fuzzer.h | 8 ++------ 2 files changed, 5 insertions(+), 12 deletions(-) (limited to 'testing') diff --git a/testing/libfuzzer/pdf_cfx_saxreader_fuzzer.cc b/testing/libfuzzer/pdf_cfx_saxreader_fuzzer.cc index 63d0e3c75d..94decbfc2c 100644 --- a/testing/libfuzzer/pdf_cfx_saxreader_fuzzer.cc +++ b/testing/libfuzzer/pdf_cfx_saxreader_fuzzer.cc @@ -16,17 +16,14 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { if (!stream) return 0; - std::unique_ptr> - fileRead(stream->MakeSeekableReadStream()); + CFX_RetainPtr fileRead = + stream->MakeSeekableReadStream(); if (!fileRead) return 0; CFX_SAXReader reader; - if (reader.StartParse(fileRead.get(), 0, -1, CFX_SaxParseMode_NotSkipSpace) < - 0) { + if (reader.StartParse(fileRead, 0, -1, CFX_SaxParseMode_NotSkipSpace) < 0) return 0; - } while (1) { int32_t ret = reader.ContinueParse(nullptr); diff --git a/testing/libfuzzer/xfa_codec_fuzzer.h b/testing/libfuzzer/xfa_codec_fuzzer.h index 4281db0ac9..8608993396 100644 --- a/testing/libfuzzer/xfa_codec_fuzzer.h +++ b/testing/libfuzzer/xfa_codec_fuzzer.h @@ -17,10 +17,8 @@ class XFACodecFuzzer { std::unique_ptr mgr(new CCodec_ModuleMgr()); std::unique_ptr decoder( mgr->CreateProgressiveDecoder()); - Reader source(data, size); - - FXCODEC_STATUS status = - decoder->LoadImageInfo(&source, type, nullptr, true); + CFX_RetainPtr source(new Reader(data, size)); + FXCODEC_STATUS status = decoder->LoadImageInfo(source, type, nullptr, true); if (status != FXCODEC_STATUS_FRAME_READY) return 0; @@ -46,8 +44,6 @@ class XFACodecFuzzer { Reader(const uint8_t* data, size_t size) : m_data(data), m_size(size) {} ~Reader() {} - void Release() override {} - bool ReadBlock(void* buffer, FX_FILESIZE offset, size_t size) override { if (offset < 0 || static_cast(offset) >= m_size) return false; -- cgit v1.2.3