2f20a2d3f3
Replace a pointer/size pair with a span. Change-Id: I1773fc3a65464eb891db7324eef1be8e5791928c Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/4214272 Reviewed-by: Tom Sepez <tsepez@chromium.org> Commit-Queue: Lei Zhang <thestig@chromium.org> Cr-Commit-Position: refs/heads/main@{#1100137}
103 lines
2.8 KiB
C++
103 lines
2.8 KiB
C++
// Copyright 2018 The Chromium Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#ifndef PDF_LOADER_DOCUMENT_LOADER_IMPL_H_
|
|
#define PDF_LOADER_DOCUMENT_LOADER_IMPL_H_
|
|
|
|
#include <stdint.h>
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "base/memory/raw_ptr.h"
|
|
#include "base/memory/weak_ptr.h"
|
|
#include "pdf/loader/chunk_stream.h"
|
|
#include "pdf/loader/document_loader.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(const DocumentLoaderImpl&) = delete;
|
|
DocumentLoaderImpl& operator=(const DocumentLoaderImpl&) = delete;
|
|
~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(uint32_t input_size);
|
|
void SaveChunkData();
|
|
|
|
uint32_t EndOfCurrentChunk() const;
|
|
|
|
const raw_ptr<Client> client_;
|
|
std::string url_;
|
|
std::unique_ptr<URLLoaderWrapper> loader_;
|
|
|
|
DataStream chunk_stream_;
|
|
bool partial_loading_enabled_; // Default determined by `kPdfPartialLoading`.
|
|
bool is_partial_loader_active_ = false;
|
|
|
|
std::vector<char> buffer_;
|
|
|
|
// The current chunk DocumentLoader is working with.
|
|
Chunk chunk_;
|
|
|
|
// In units of Chunks.
|
|
RangeSet pending_requests_;
|
|
|
|
uint32_t bytes_received_ = 0;
|
|
|
|
base::WeakPtrFactory<DocumentLoaderImpl> weak_factory_{this};
|
|
};
|
|
|
|
} // namespace chrome_pdf
|
|
|
|
#endif // PDF_LOADER_DOCUMENT_LOADER_IMPL_H_
|