6f6c6e44bf
TestDocumentLoader currently loads the entire test PDF file as soon as PDFiumEngine calls DocumentLoader::RequestData(0, 1). This change adds a new TestDocumentLoader::SimulateLoadData(uint32_t) method that allows a test to control how quickly the document loads. A future test will take advantage of this to put the PDFiumEngine into a partially-loaded state. A test can request the TestDocumentLoader by calling PDFiumTestBase's InitializeEngineWithoutLoading() method instead of InitializeEngine(). This change also removes the DocumentLoader::SetDocumentSize() method, which only was used internally by DocumentLoaderImpl anyway. Bug: 1051548 Change-Id: Ib5256aa70c468284e0f01964b409f2a85a9a066a Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2067692 Commit-Queue: K Moon <kmoon@chromium.org> Reviewed-by: Lei Zhang <thestig@chromium.org> Reviewed-by: Henrique Nakashima <hnakashima@chromium.org> Reviewed-by: Daniel Hosseinian <dhoss@chromium.org> Cr-Commit-Position: refs/heads/master@{#751597}
101 lines
2.7 KiB
C++
101 lines
2.7 KiB
C++
// Copyright 2018 The Chromium 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 PDF_DOCUMENT_LOADER_IMPL_H_
|
|
#define PDF_DOCUMENT_LOADER_IMPL_H_
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "base/macros.h"
|
|
#include "pdf/chunk_stream.h"
|
|
#include "pdf/document_loader.h"
|
|
#include "ppapi/utility/completion_callback_factory.h"
|
|
|
|
namespace chrome_pdf {
|
|
|
|
class DocumentLoaderImpl : public DocumentLoader {
|
|
public:
|
|
// Number was chosen in https://crbug.com/78264#c8
|
|
static constexpr uint32_t kDefaultRequestSize = 65536;
|
|
|
|
explicit DocumentLoaderImpl(Client* client);
|
|
~DocumentLoaderImpl() override;
|
|
|
|
// DocumentLoader:
|
|
bool Init(std::unique_ptr<URLLoaderWrapper> loader,
|
|
const std::string& url) override;
|
|
bool GetBlock(uint32_t position, uint32_t size, void* buf) const override;
|
|
bool IsDataAvailable(uint32_t position, uint32_t size) const override;
|
|
void RequestData(uint32_t position, uint32_t size) override;
|
|
bool IsDocumentComplete() const override;
|
|
uint32_t GetDocumentSize() const override;
|
|
uint32_t BytesReceived() const override;
|
|
void ClearPendingRequests() override;
|
|
|
|
// Exposed for unit tests.
|
|
void SetPartialLoadingEnabled(bool enabled);
|
|
bool is_partial_loader_active() const { return is_partial_loader_active_; }
|
|
|
|
private:
|
|
using DataStream = ChunkStream<kDefaultRequestSize>;
|
|
struct Chunk {
|
|
Chunk();
|
|
~Chunk();
|
|
|
|
void Clear();
|
|
|
|
uint32_t chunk_index = 0;
|
|
uint32_t data_size = 0;
|
|
std::unique_ptr<DataStream::ChunkData> chunk_data;
|
|
};
|
|
|
|
// Called by the completion callback of the document's URLLoader.
|
|
void DidOpenPartial(int32_t result);
|
|
|
|
// Call to read data from the document's URLLoader.
|
|
void ReadMore();
|
|
|
|
// Called by the completion callback of the document's URLLoader.
|
|
void DidRead(int32_t result);
|
|
|
|
bool ShouldCancelLoading() const;
|
|
void ContinueDownload();
|
|
|
|
// Called when we complete server request.
|
|
void ReadComplete();
|
|
|
|
bool SaveBuffer(char* input, uint32_t input_size);
|
|
void SaveChunkData();
|
|
|
|
uint32_t EndOfCurrentChunk() const;
|
|
|
|
Client* const client_;
|
|
std::string url_;
|
|
std::unique_ptr<URLLoaderWrapper> loader_;
|
|
|
|
pp::CompletionCallbackFactory<DocumentLoaderImpl> loader_factory_;
|
|
|
|
DataStream chunk_stream_;
|
|
bool partial_loading_enabled_ = true;
|
|
bool is_partial_loader_active_ = false;
|
|
|
|
static constexpr uint32_t kReadBufferSize = 256 * 1024;
|
|
char buffer_[kReadBufferSize];
|
|
|
|
// The current chunk DocumentLoader is working with.
|
|
Chunk chunk_;
|
|
|
|
// In units of Chunks.
|
|
RangeSet pending_requests_;
|
|
|
|
uint32_t bytes_received_ = 0;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(DocumentLoaderImpl);
|
|
};
|
|
|
|
} // namespace chrome_pdf
|
|
|
|
#endif // PDF_DOCUMENT_LOADER_IMPL_H_
|