chrome/renderer/chrome_mock_render_thread.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/renderer/chrome_mock_render_thread.h"
   6
   7 #include <vector>
   8
   9 #include "base/values.h"
  10 #include "chrome/renderer/mock_printer.h"
  11 #include "extensions/common/extension_messages.h"
  12 #include "ipc/ipc_sync_message.h"
  13 #include "printing/page_range.h"
  14 #include "printing/print_job_constants.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16
  17 #if defined(OS_CHROMEOS)
  18 #include <fcntl.h>
  19
  20 #include "base/file_util.h"
  21 #endif
  22
  23 #if defined(ENABLE_PRINTING)
  24 #include "chrome/common/print_messages.h"
  25 #endif
  26
  27 ChromeMockRenderThread::ChromeMockRenderThread()
  28 #if defined(ENABLE_PRINTING)
  29     : printer_(new MockPrinter),
  30       print_dialog_user_response_(true),
  31       print_preview_cancel_page_number_(-1),
  32       print_preview_pages_remaining_(0)
  33 #endif
  34 {
  35 }
  36
  37 ChromeMockRenderThread::~ChromeMockRenderThread() {
  38 }
  39
  40 scoped_refptr<base::MessageLoopProxy>
  41 ChromeMockRenderThread::GetIOMessageLoopProxy() {
  42   return io_message_loop_proxy_;
  43 }
  44
  45 void ChromeMockRenderThread::set_io_message_loop_proxy(
  46     const scoped_refptr<base::MessageLoopProxy>& proxy) {
  47   io_message_loop_proxy_ = proxy;
  48 }
  49
  50 bool ChromeMockRenderThread::OnMessageReceived(const IPC::Message& msg) {
  51   if (content::MockRenderThread::OnMessageReceived(msg))
  52     return true;
  53
  54   // Some messages we do special handling.
  55   bool handled = true;
  56   bool msg_is_ok = true;
  57   IPC_BEGIN_MESSAGE_MAP_EX(ChromeMockRenderThread, msg, msg_is_ok)
  58     IPC_MESSAGE_HANDLER(ExtensionHostMsg_OpenChannelToExtension,
  59                         OnOpenChannelToExtension)
  60 #if defined(ENABLE_PRINTING)
  61     IPC_MESSAGE_HANDLER(PrintHostMsg_GetDefaultPrintSettings,
  62                         OnGetDefaultPrintSettings)
  63     IPC_MESSAGE_HANDLER(PrintHostMsg_ScriptedPrint, OnScriptedPrint)
  64     IPC_MESSAGE_HANDLER(PrintHostMsg_UpdatePrintSettings, OnUpdatePrintSettings)
  65     IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPrintedPagesCount,
  66                         OnDidGetPrintedPagesCount)
  67     IPC_MESSAGE_HANDLER(PrintHostMsg_DidPrintPage, OnDidPrintPage)
  68     IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPreviewPageCount,
  69                         OnDidGetPreviewPageCount)
  70     IPC_MESSAGE_HANDLER(PrintHostMsg_DidPreviewPage, OnDidPreviewPage)
  71     IPC_MESSAGE_HANDLER(PrintHostMsg_CheckForCancel, OnCheckForCancel)
  72 #if defined(OS_WIN)
  73     IPC_MESSAGE_HANDLER(PrintHostMsg_DuplicateSection, OnDuplicateSection)
  74 #endif
  75 #if defined(OS_CHROMEOS)
  76     IPC_MESSAGE_HANDLER(PrintHostMsg_AllocateTempFileForPrinting,
  77                         OnAllocateTempFileForPrinting)
  78     IPC_MESSAGE_HANDLER(PrintHostMsg_TempFileForPrintingWritten,
  79                         OnTempFileForPrintingWritten)
  80 #endif  // defined(OS_CHROMEOS)
  81 #endif  // defined(ENABLE_PRINTING)
  82     IPC_MESSAGE_UNHANDLED(handled = false)
  83   IPC_END_MESSAGE_MAP_EX()
  84   return handled;
  85 }
  86
  87 void ChromeMockRenderThread::OnOpenChannelToExtension(
  88     int routing_id,
  89     const ExtensionMsg_ExternalConnectionInfo& info,
  90     const std::string& channel_name,
  91     bool include_tls_channel_id,
  92     int* port_id) {
  93   *port_id = 0;
  94 }
  95
  96 #if defined(ENABLE_PRINTING)
  97 #if defined(OS_CHROMEOS)
  98 void ChromeMockRenderThread::OnAllocateTempFileForPrinting(
  99     int render_view_id,
 100     base::FileDescriptor* renderer_fd,
 101     int* browser_fd) {
 102   renderer_fd->fd = *browser_fd = -1;
 103   renderer_fd->auto_close = false;
 104
 105   base::FilePath path;
 106   if (base::CreateTemporaryFile(&path)) {
 107     int fd = open(path.value().c_str(), O_WRONLY);
 108     DCHECK_GE(fd, 0);
 109     renderer_fd->fd = *browser_fd = fd;
 110   }
 111 }
 112
 113 void ChromeMockRenderThread::OnTempFileForPrintingWritten(int render_view_id,
 114                                                           int browser_fd) {
 115   close(browser_fd);
 116 }
 117 #endif  // defined(OS_CHROMEOS)
 118
 119 void ChromeMockRenderThread::OnGetDefaultPrintSettings(
 120     PrintMsg_Print_Params* params) {
 121   printer_->GetDefaultPrintSettings(params);
 122 }
 123
 124 void ChromeMockRenderThread::OnScriptedPrint(
 125     const PrintHostMsg_ScriptedPrint_Params& params,
 126     PrintMsg_PrintPages_Params* settings) {
 127   if (print_dialog_user_response_) {
 128     printer_->ScriptedPrint(params.cookie,
 129                             params.expected_pages_count,
 130                             params.has_selection,
 131                             settings);
 132   }
 133 }
 134
 135 void ChromeMockRenderThread::OnDidGetPrintedPagesCount(
 136     int cookie, int number_pages) {
 137   printer_->SetPrintedPagesCount(cookie, number_pages);
 138 }
 139
 140 void ChromeMockRenderThread::OnDidPrintPage(
 141     const PrintHostMsg_DidPrintPage_Params& params) {
 142   printer_->PrintPage(params);
 143 }
 144
 145 void ChromeMockRenderThread::OnDidGetPreviewPageCount(
 146     const PrintHostMsg_DidGetPreviewPageCount_Params& params) {
 147   print_preview_pages_remaining_ = params.page_count;
 148 }
 149
 150 void ChromeMockRenderThread::OnDidPreviewPage(
 151     const PrintHostMsg_DidPreviewPage_Params& params) {
 152   DCHECK_GE(params.page_number, printing::FIRST_PAGE_INDEX);
 153   print_preview_pages_remaining_--;
 154 }
 155
 156 void ChromeMockRenderThread::OnCheckForCancel(int32 preview_ui_id,
 157                                               int preview_request_id,
 158                                               bool* cancel) {
 159   *cancel =
 160       (print_preview_pages_remaining_ == print_preview_cancel_page_number_);
 161 }
 162
 163 void ChromeMockRenderThread::OnUpdatePrintSettings(
 164     int document_cookie,
 165     const base::DictionaryValue& job_settings,
 166     PrintMsg_PrintPages_Params* params) {
 167   // Check and make sure the required settings are all there.
 168   // We don't actually care about the values.
 169   std::string dummy_string;
 170   int margins_type = 0;
 171   if (!job_settings.GetBoolean(printing::kSettingLandscape, NULL) ||
 172       !job_settings.GetBoolean(printing::kSettingCollate, NULL) ||
 173       !job_settings.GetInteger(printing::kSettingColor, NULL) ||
 174       !job_settings.GetBoolean(printing::kSettingPrintToPDF, NULL) ||
 175       !job_settings.GetBoolean(printing::kIsFirstRequest, NULL) ||
 176       !job_settings.GetString(printing::kSettingDeviceName, &dummy_string) ||
 177       !job_settings.GetInteger(printing::kSettingDuplexMode, NULL) ||
 178       !job_settings.GetInteger(printing::kSettingCopies, NULL) ||
 179       !job_settings.GetInteger(printing::kPreviewUIID, NULL) ||
 180       !job_settings.GetInteger(printing::kPreviewRequestID, NULL) ||
 181       !job_settings.GetInteger(printing::kSettingMarginsType, &margins_type)) {
 182     return;
 183   }
 184
 185   // Just return the default settings.
 186   const base::ListValue* page_range_array;
 187   printing::PageRanges new_ranges;
 188   if (job_settings.GetList(printing::kSettingPageRange, &page_range_array)) {
 189     for (size_t index = 0; index < page_range_array->GetSize(); ++index) {
 190       const base::DictionaryValue* dict;
 191       if (!page_range_array->GetDictionary(index, &dict))
 192         continue;
 193       printing::PageRange range;
 194       if (!dict->GetInteger(printing::kSettingPageRangeFrom, &range.from) ||
 195           !dict->GetInteger(printing::kSettingPageRangeTo, &range.to)) {
 196         continue;
 197       }
 198       // Page numbers are 1-based in the dictionary.
 199       // Page numbers are 0-based for the printing context.
 200       range.from--;
 201       range.to--;
 202       new_ranges.push_back(range);
 203     }
 204   }
 205   std::vector<int> pages(printing::PageRange::GetPages(new_ranges));
 206   printer_->UpdateSettings(document_cookie, params, pages, margins_type);
 207
 208   job_settings.GetBoolean(printing::kSettingShouldPrintSelectionOnly,
 209                           &params->params.selection_only);
 210   job_settings.GetBoolean(printing::kSettingShouldPrintBackgrounds,
 211                           &params->params.should_print_backgrounds);
 212 }
 213
 214 MockPrinter* ChromeMockRenderThread::printer() {
 215   return printer_.get();
 216 }
 217
 218 void ChromeMockRenderThread::set_print_dialog_user_response(bool response) {
 219   print_dialog_user_response_ = response;
 220 }
 221
 222 void ChromeMockRenderThread::set_print_preview_cancel_page_number(int page) {
 223   print_preview_cancel_page_number_ = page;
 224 }
 225
 226 int ChromeMockRenderThread::print_preview_pages_remaining() const {
 227   return print_preview_pages_remaining_;
 228 }
 229 #endif  // defined(ENABLE_PRINTING)