// Copyright 2017 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. #ifndef TESTING_FAKE_FILE_ACCESS_H_ #define TESTING_FAKE_FILE_ACCESS_H_ #include <memory> #include "core/fxcrt/unowned_ptr.h" #include "public/fpdf_dataavail.h" #include "public/fpdfview.h" #include "testing/range_set.h" class FakeFileAccess { public: explicit FakeFileAccess(FPDF_FILEACCESS* file_access); ~FakeFileAccess(); FPDF_FILEACCESS* GetFileAccess() const { return file_access_wrapper_.get(); } FX_FILEAVAIL* GetFileAvail() const { return file_avail_.get(); } FX_DOWNLOADHINTS* GetDownloadHints() const { return download_hints_.get(); } FPDF_BOOL IsDataAvail(size_t offset, size_t size) const; void AddSegment(size_t offset, size_t size); unsigned long GetFileSize(); int GetBlock(unsigned long position, unsigned char* pBuf, unsigned long size); void SetRequestedDataAvailable(); void SetWholeFileAvailable(); private: fxcrt::UnownedPtr<FPDF_FILEACCESS> file_access_; std::unique_ptr<FPDF_FILEACCESS> file_access_wrapper_; std::unique_ptr<FX_FILEAVAIL> file_avail_; std::unique_ptr<FX_DOWNLOADHINTS> download_hints_; RangeSet available_data_; RangeSet requested_data_; }; #endif // TESTING_FAKE_FILE_ACCESS_H_