| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/renderer/chrome_mock_render_thread.h" | 5 #include "chrome/renderer/chrome_mock_render_thread.h" |
| 6 | 6 |
| 7 #include <fcntl.h> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/file_util.h" | 9 #include "base/values.h" |
| 10 #include "base/process_util.h" | |
| 11 #include "chrome/common/extensions/extension_messages.h" | 10 #include "chrome/common/extensions/extension_messages.h" |
| 12 #include "chrome/common/print_messages.h" | 11 #include "chrome/common/print_messages.h" |
| 13 #include "chrome/common/render_messages.h" | 12 #include "chrome/renderer/mock_printer.h" |
| 14 #include "chrome/common/url_constants.h" | |
| 15 #include "ipc/ipc_message_utils.h" | |
| 16 #include "ipc/ipc_sync_message.h" | 13 #include "ipc/ipc_sync_message.h" |
| 17 #include "printing/print_job_constants.h" | 14 #include "printing/print_job_constants.h" |
| 18 #include "printing/page_range.h" | 15 #include "printing/page_range.h" |
| 19 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 20 | 17 |
| 18 #if defined(OS_CHROMEOS) |
| 19 #include <fcntl.h> |
| 20 |
| 21 #include "base/file_util.h" |
| 22 #endif |
| 23 |
| 21 ChromeMockRenderThread::ChromeMockRenderThread() | 24 ChromeMockRenderThread::ChromeMockRenderThread() |
| 22 : printer_(new MockPrinter), | 25 : printer_(new MockPrinter), |
| 23 print_dialog_user_response_(true), | 26 print_dialog_user_response_(true), |
| 24 print_preview_cancel_page_number_(-1), | 27 print_preview_cancel_page_number_(-1), |
| 25 print_preview_pages_remaining_(0) { | 28 print_preview_pages_remaining_(0) { |
| 26 } | 29 } |
| 27 | 30 |
| 28 ChromeMockRenderThread::~ChromeMockRenderThread() { | 31 ChromeMockRenderThread::~ChromeMockRenderThread() { |
| 29 } | 32 } |
| 30 | 33 |
| (...skipping 26 matching lines...) Expand all Loading... |
| 57 OnAllocateTempFileForPrinting) | 60 OnAllocateTempFileForPrinting) |
| 58 IPC_MESSAGE_HANDLER(PrintHostMsg_TempFileForPrintingWritten, | 61 IPC_MESSAGE_HANDLER(PrintHostMsg_TempFileForPrintingWritten, |
| 59 OnTempFileForPrintingWritten) | 62 OnTempFileForPrintingWritten) |
| 60 #endif | 63 #endif |
| 61 IPC_MESSAGE_UNHANDLED(handled = false) | 64 IPC_MESSAGE_UNHANDLED(handled = false) |
| 62 IPC_END_MESSAGE_MAP_EX() | 65 IPC_END_MESSAGE_MAP_EX() |
| 63 return handled; | 66 return handled; |
| 64 } | 67 } |
| 65 | 68 |
| 66 void ChromeMockRenderThread::OnMsgOpenChannelToExtension( | 69 void ChromeMockRenderThread::OnMsgOpenChannelToExtension( |
| 67 int routing_id, const std::string& source_extension_id, | 70 int routing_id, |
| 71 const std::string& source_extension_id, |
| 68 const std::string& target_extension_id, | 72 const std::string& target_extension_id, |
| 69 const std::string& channel_name, int* port_id) { | 73 const std::string& channel_name, |
| 74 int* port_id) { |
| 70 *port_id = 0; | 75 *port_id = 0; |
| 71 } | 76 } |
| 72 | 77 |
| 73 #if defined(OS_CHROMEOS) | 78 #if defined(OS_CHROMEOS) |
| 74 void ChromeMockRenderThread::OnAllocateTempFileForPrinting( | 79 void ChromeMockRenderThread::OnAllocateTempFileForPrinting( |
| 75 base::FileDescriptor* renderer_fd, | 80 base::FileDescriptor* renderer_fd, |
| 76 int* browser_fd) { | 81 int* browser_fd) { |
| 77 renderer_fd->fd = *browser_fd = -1; | 82 renderer_fd->fd = *browser_fd = -1; |
| 78 renderer_fd->auto_close = false; | 83 renderer_fd->auto_close = false; |
| 79 | 84 |
| 80 FilePath path; | 85 FilePath path; |
| 81 if (file_util::CreateTemporaryFile(&path)) { | 86 if (file_util::CreateTemporaryFile(&path)) { |
| 82 int fd = open(path.value().c_str(), O_WRONLY); | 87 int fd = open(path.value().c_str(), O_WRONLY); |
| 83 DCHECK_GE(fd, 0); | 88 DCHECK_GE(fd, 0); |
| 84 renderer_fd->fd = *browser_fd = fd; | 89 renderer_fd->fd = *browser_fd = fd; |
| 85 } | 90 } |
| 86 } | 91 } |
| 87 | 92 |
| 88 void ChromeMockRenderThread::OnTempFileForPrintingWritten(int render_view_id, | 93 void ChromeMockRenderThread::OnTempFileForPrintingWritten(int render_view_id, |
| 89 int browser_fd) { | 94 int browser_fd) { |
| 90 close(browser_fd); | 95 close(browser_fd); |
| 91 } | 96 } |
| 92 #endif // defined(OS_CHROMEOS) | 97 #endif // defined(OS_CHROMEOS) |
| 93 | 98 |
| 94 void ChromeMockRenderThread::OnGetDefaultPrintSettings( | 99 void ChromeMockRenderThread::OnGetDefaultPrintSettings( |
| 95 PrintMsg_Print_Params* params) { | 100 PrintMsg_Print_Params* params) { |
| 96 if (printer_.get()) | 101 printer_->GetDefaultPrintSettings(params); |
| 97 printer_->GetDefaultPrintSettings(params); | |
| 98 } | 102 } |
| 99 | 103 |
| 100 void ChromeMockRenderThread::OnScriptedPrint( | 104 void ChromeMockRenderThread::OnScriptedPrint( |
| 101 const PrintHostMsg_ScriptedPrint_Params& params, | 105 const PrintHostMsg_ScriptedPrint_Params& params, |
| 102 PrintMsg_PrintPages_Params* settings) { | 106 PrintMsg_PrintPages_Params* settings) { |
| 103 if (print_dialog_user_response_ && printer_.get()) { | 107 if (print_dialog_user_response_) { |
| 104 printer_->ScriptedPrint(params.cookie, | 108 printer_->ScriptedPrint(params.cookie, |
| 105 params.expected_pages_count, | 109 params.expected_pages_count, |
| 106 params.has_selection, | 110 params.has_selection, |
| 107 settings); | 111 settings); |
| 108 } | 112 } |
| 109 } | 113 } |
| 110 | 114 |
| 111 void ChromeMockRenderThread::OnDidGetPrintedPagesCount( | 115 void ChromeMockRenderThread::OnDidGetPrintedPagesCount( |
| 112 int cookie, int number_pages) { | 116 int cookie, int number_pages) { |
| 113 if (printer_.get()) | 117 printer_->SetPrintedPagesCount(cookie, number_pages); |
| 114 printer_->SetPrintedPagesCount(cookie, number_pages); | |
| 115 } | 118 } |
| 116 | 119 |
| 117 void ChromeMockRenderThread::OnDidPrintPage( | 120 void ChromeMockRenderThread::OnDidPrintPage( |
| 118 const PrintHostMsg_DidPrintPage_Params& params) { | 121 const PrintHostMsg_DidPrintPage_Params& params) { |
| 119 if (printer_.get()) | 122 printer_->PrintPage(params); |
| 120 printer_->PrintPage(params); | |
| 121 } | 123 } |
| 122 | 124 |
| 123 void ChromeMockRenderThread::OnDidGetPreviewPageCount( | 125 void ChromeMockRenderThread::OnDidGetPreviewPageCount( |
| 124 const PrintHostMsg_DidGetPreviewPageCount_Params& params) { | 126 const PrintHostMsg_DidGetPreviewPageCount_Params& params) { |
| 125 print_preview_pages_remaining_ = params.page_count; | 127 print_preview_pages_remaining_ = params.page_count; |
| 126 } | 128 } |
| 127 | 129 |
| 128 void ChromeMockRenderThread::OnDidPreviewPage( | 130 void ChromeMockRenderThread::OnDidPreviewPage( |
| 129 const PrintHostMsg_DidPreviewPage_Params& params) { | 131 const PrintHostMsg_DidPreviewPage_Params& params) { |
| 130 DCHECK(params.page_number >= printing::FIRST_PAGE_INDEX); | 132 DCHECK_GE(params.page_number, printing::FIRST_PAGE_INDEX); |
| 131 print_preview_pages_remaining_--; | 133 print_preview_pages_remaining_--; |
| 132 } | 134 } |
| 133 | 135 |
| 134 void ChromeMockRenderThread::OnCheckForCancel( | 136 void ChromeMockRenderThread::OnCheckForCancel(int32 preview_ui_id, |
| 135 const std::string& preview_ui_addr, | 137 int preview_request_id, |
| 136 int preview_request_id, | 138 bool* cancel) { |
| 137 bool* cancel) { | |
| 138 *cancel = | 139 *cancel = |
| 139 (print_preview_pages_remaining_ == print_preview_cancel_page_number_); | 140 (print_preview_pages_remaining_ == print_preview_cancel_page_number_); |
| 140 } | 141 } |
| 141 | 142 |
| 142 void ChromeMockRenderThread::OnUpdatePrintSettings( | 143 void ChromeMockRenderThread::OnUpdatePrintSettings( |
| 143 int document_cookie, | 144 int document_cookie, |
| 144 const base::DictionaryValue& job_settings, | 145 const base::DictionaryValue& job_settings, |
| 145 PrintMsg_PrintPages_Params* params) { | 146 PrintMsg_PrintPages_Params* params) { |
| 146 // Check and make sure the required settings are all there. | 147 // Check and make sure the required settings are all there. |
| 147 // We don't actually care about the values. | 148 // We don't actually care about the values. |
| 148 std::string dummy_string; | 149 std::string dummy_string; |
| 149 int margins_type = 0; | 150 int margins_type = 0; |
| 150 if (!job_settings.GetBoolean(printing::kSettingLandscape, NULL) || | 151 if (!job_settings.GetBoolean(printing::kSettingLandscape, NULL) || |
| 151 !job_settings.GetBoolean(printing::kSettingCollate, NULL) || | 152 !job_settings.GetBoolean(printing::kSettingCollate, NULL) || |
| 152 !job_settings.GetInteger(printing::kSettingColor, NULL) || | 153 !job_settings.GetInteger(printing::kSettingColor, NULL) || |
| 153 !job_settings.GetBoolean(printing::kSettingPrintToPDF, NULL) || | 154 !job_settings.GetBoolean(printing::kSettingPrintToPDF, NULL) || |
| 154 !job_settings.GetBoolean(printing::kIsFirstRequest, NULL) || | 155 !job_settings.GetBoolean(printing::kIsFirstRequest, NULL) || |
| 155 !job_settings.GetString(printing::kSettingDeviceName, &dummy_string) || | 156 !job_settings.GetString(printing::kSettingDeviceName, &dummy_string) || |
| 156 !job_settings.GetInteger(printing::kSettingDuplexMode, NULL) || | 157 !job_settings.GetInteger(printing::kSettingDuplexMode, NULL) || |
| 157 !job_settings.GetInteger(printing::kSettingCopies, NULL) || | 158 !job_settings.GetInteger(printing::kSettingCopies, NULL) || |
| 158 !job_settings.GetString(printing::kPreviewUIAddr, &dummy_string) || | 159 !job_settings.GetInteger(printing::kPreviewUIID, NULL) || |
| 159 !job_settings.GetInteger(printing::kPreviewRequestID, NULL) || | 160 !job_settings.GetInteger(printing::kPreviewRequestID, NULL) || |
| 160 !job_settings.GetInteger(printing::kSettingMarginsType, &margins_type)) { | 161 !job_settings.GetInteger(printing::kSettingMarginsType, &margins_type)) { |
| 161 return; | 162 return; |
| 162 } | 163 } |
| 163 | 164 |
| 164 // Just return the default settings. | 165 // Just return the default settings. |
| 165 if (printer_.get()) { | 166 const ListValue* page_range_array; |
| 166 const ListValue* page_range_array; | 167 printing::PageRanges new_ranges; |
| 167 printing::PageRanges new_ranges; | 168 if (job_settings.GetList(printing::kSettingPageRange, &page_range_array)) { |
| 168 if (job_settings.GetList(printing::kSettingPageRange, &page_range_array)) { | 169 for (size_t index = 0; index < page_range_array->GetSize(); ++index) { |
| 169 for (size_t index = 0; index < page_range_array->GetSize(); ++index) { | 170 const base::DictionaryValue* dict; |
| 170 const base::DictionaryValue* dict; | 171 if (!page_range_array->GetDictionary(index, &dict)) |
| 171 if (!page_range_array->GetDictionary(index, &dict)) | 172 continue; |
| 172 continue; | 173 printing::PageRange range; |
| 173 printing::PageRange range; | 174 if (!dict->GetInteger(printing::kSettingPageRangeFrom, &range.from) || |
| 174 if (!dict->GetInteger(printing::kSettingPageRangeFrom, &range.from) || | 175 !dict->GetInteger(printing::kSettingPageRangeTo, &range.to)) { |
| 175 !dict->GetInteger(printing::kSettingPageRangeTo, &range.to)) { | 176 continue; |
| 176 continue; | |
| 177 } | |
| 178 // Page numbers are 1-based in the dictionary. | |
| 179 // Page numbers are 0-based for the printing context. | |
| 180 range.from--; | |
| 181 range.to--; | |
| 182 new_ranges.push_back(range); | |
| 183 } | 177 } |
| 178 // Page numbers are 1-based in the dictionary. |
| 179 // Page numbers are 0-based for the printing context. |
| 180 range.from--; |
| 181 range.to--; |
| 182 new_ranges.push_back(range); |
| 184 } | 183 } |
| 185 std::vector<int> pages(printing::PageRange::GetPages(new_ranges)); | |
| 186 printer_->UpdateSettings(document_cookie, params, pages, margins_type); | |
| 187 } | 184 } |
| 185 std::vector<int> pages(printing::PageRange::GetPages(new_ranges)); |
| 186 printer_->UpdateSettings(document_cookie, params, pages, margins_type); |
| 187 } |
| 188 |
| 189 MockPrinter* ChromeMockRenderThread::printer() { |
| 190 return printer_.get(); |
| 188 } | 191 } |
| 189 | 192 |
| 190 void ChromeMockRenderThread::set_print_dialog_user_response(bool response) { | 193 void ChromeMockRenderThread::set_print_dialog_user_response(bool response) { |
| 191 print_dialog_user_response_ = response; | 194 print_dialog_user_response_ = response; |
| 192 } | 195 } |
| 193 | 196 |
| 194 void ChromeMockRenderThread::set_print_preview_cancel_page_number(int page) { | 197 void ChromeMockRenderThread::set_print_preview_cancel_page_number(int page) { |
| 195 print_preview_cancel_page_number_ = page; | 198 print_preview_cancel_page_number_ = page; |
| 196 } | 199 } |
| 197 | 200 |
| 198 int ChromeMockRenderThread::print_preview_pages_remaining() { | 201 int ChromeMockRenderThread::print_preview_pages_remaining() const { |
| 199 return print_preview_pages_remaining_; | 202 return print_preview_pages_remaining_; |
| 200 } | 203 } |
| OLD | NEW |