blob: 60194e29ebbe5e6fb28c06568e0deea17e1fb401 [file] [log] [blame]
// Copyright (c) 2012 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.
#include "chrome/browser/printing/print_preview_data_service.h"
#include <utility>
#include "base/macros.h"
#include "base/memory/ref_counted_memory.h"
#include "base/memory/singleton.h"
#include "base/stl_util.h"
#include "printing/print_job_constants.h"
namespace {
void ValidatePreviewData(const scoped_refptr<base::RefCountedMemory>& data) {
// PDFs are generally much bigger. This is just a sanity check on size.
DCHECK_GE(data->size(), 50U);
static const char kPdfHeader[] = "%PDF-";
const char* content = data->front_as<const char>();
DCHECK_EQ(0, memcmp(content, kPdfHeader, strlen(kPdfHeader)));
} // namespace
// PrintPreviewDataStore stores data for preview workflow and preview printing
// workflow.
// NOTE:
// This class stores a list of PDFs. The list |index| is zero-based and can
// be |printing::COMPLETE_PREVIEW_DOCUMENT_INDEX| to represent complete preview
// document. The PDF stored at |printing::COMPLETE_PREVIEW_DOCUMENT_INDEX| is
// optimized with font subsetting, compression, etc. PDF's stored at all other
// indices are unoptimized.
// PrintPreviewDataStore owns the data and is responsible for freeing it when
// either:
// a) There is a new data.
// b) When PrintPreviewDataStore is destroyed.
class PrintPreviewDataStore {
PrintPreviewDataStore() {}
~PrintPreviewDataStore() {}
// Get the preview page for the specified |index|.
void GetPreviewDataForIndex(
int index,
scoped_refptr<base::RefCountedMemory>* data) const {
if (IsInvalidIndex(index))
auto it = page_data_map_.find(index);
if (it != page_data_map_.end())
*data = it->second.get();
// Set/Update the preview data entry for the specified |index|.
void SetPreviewDataForIndex(int index,
scoped_refptr<base::RefCountedMemory> data) {
if (IsInvalidIndex(index))
page_data_map_[index] = std::move(data);
// 1:1 relationship between page index and its associated preview data.
// Key: Page index is zero-based and can be
// |printing::COMPLETE_PREVIEW_DOCUMENT_INDEX| to represent complete preview
// document.
// Value: Preview data.
using PreviewPageDataMap =
std::map<int, scoped_refptr<base::RefCountedMemory>>;
static bool IsInvalidIndex(int index) {
return (index != printing::COMPLETE_PREVIEW_DOCUMENT_INDEX &&
index < printing::FIRST_PAGE_INDEX);
PreviewPageDataMap page_data_map_;
// static
PrintPreviewDataService* PrintPreviewDataService::GetInstance() {
return base::Singleton<PrintPreviewDataService>::get();
PrintPreviewDataService::PrintPreviewDataService() {
PrintPreviewDataService::~PrintPreviewDataService() {
void PrintPreviewDataService::GetDataEntry(
int32_t preview_ui_id,
int index,
scoped_refptr<base::RefCountedMemory>* data_bytes) const {
*data_bytes = nullptr;
auto it = data_store_map_.find(preview_ui_id);
if (it != data_store_map_.end())
it->second->GetPreviewDataForIndex(index, data_bytes);
void PrintPreviewDataService::SetDataEntry(
int32_t preview_ui_id,
int index,
scoped_refptr<base::RefCountedMemory> data_bytes) {
if (!base::ContainsKey(data_store_map_, preview_ui_id))
data_store_map_[preview_ui_id] = std::make_unique<PrintPreviewDataStore>();
void PrintPreviewDataService::RemoveEntry(int32_t preview_ui_id) {