| // Copyright (c) 2010 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/utility/utility_thread.h" |
| |
| #include <vector> |
| |
| #include "base/file_util.h" |
| #include "base/path_service.h" |
| #include "base/values.h" |
| #include "build/build_config.h" |
| #include "chrome/common/child_process.h" |
| #include "chrome/common/chrome_paths.h" |
| #include "chrome/common/extensions/extension_unpacker.h" |
| #include "chrome/common/extensions/update_manifest.h" |
| #include "chrome/common/indexed_db_key.h" |
| #include "chrome/common/serialized_script_value.h" |
| #include "chrome/common/utility_messages.h" |
| #include "chrome/common/web_resource/web_resource_unpacker.h" |
| #include "gfx/rect.h" |
| #include "printing/native_metafile.h" |
| #include "printing/page_range.h" |
| #include "printing/units.h" |
| #include "third_party/skia/include/core/SkBitmap.h" |
| #include "third_party/WebKit/WebKit/chromium/public/WebSerializedScriptValue.h" |
| #include "webkit/glue/idb_bindings.h" |
| #include "webkit/glue/image_decoder.h" |
| |
| #if defined(OS_WIN) |
| #include "app/win/iat_patch_function.h" |
| #endif |
| |
| namespace { |
| |
| template<typename SRC, typename DEST> |
| void ConvertVector(const SRC& src, DEST* dest) { |
| dest->reserve(src.size()); |
| for (typename SRC::const_iterator i = src.begin(); i != src.end(); ++i) |
| dest->push_back(typename DEST::value_type(*i)); |
| } |
| |
| } // namespace |
| |
| |
| UtilityThread::UtilityThread() |
| : batch_mode_(false) { |
| ChildProcess::current()->AddRefProcess(); |
| } |
| |
| UtilityThread::~UtilityThread() { |
| } |
| |
| void UtilityThread::OnControlMessageReceived(const IPC::Message& msg) { |
| IPC_BEGIN_MESSAGE_MAP(UtilityThread, msg) |
| IPC_MESSAGE_HANDLER(UtilityMsg_UnpackExtension, OnUnpackExtension) |
| IPC_MESSAGE_HANDLER(UtilityMsg_UnpackWebResource, OnUnpackWebResource) |
| IPC_MESSAGE_HANDLER(UtilityMsg_ParseUpdateManifest, OnParseUpdateManifest) |
| IPC_MESSAGE_HANDLER(UtilityMsg_DecodeImage, OnDecodeImage) |
| IPC_MESSAGE_HANDLER(UtilityMsg_RenderPDFPagesToMetafile, |
| OnRenderPDFPagesToMetafile) |
| IPC_MESSAGE_HANDLER(UtilityMsg_IDBKeysFromValuesAndKeyPath, |
| OnIDBKeysFromValuesAndKeyPath) |
| IPC_MESSAGE_HANDLER(UtilityMsg_BatchMode_Started, OnBatchModeStarted) |
| IPC_MESSAGE_HANDLER(UtilityMsg_BatchMode_Finished, OnBatchModeFinished) |
| IPC_END_MESSAGE_MAP() |
| } |
| |
| void UtilityThread::OnUnpackExtension(const FilePath& extension_path) { |
| ExtensionUnpacker unpacker(extension_path); |
| if (unpacker.Run() && unpacker.DumpImagesToFile() && |
| unpacker.DumpMessageCatalogsToFile()) { |
| Send(new UtilityHostMsg_UnpackExtension_Succeeded( |
| *unpacker.parsed_manifest())); |
| } else { |
| Send(new UtilityHostMsg_UnpackExtension_Failed(unpacker.error_message())); |
| } |
| |
| ReleaseProcessIfNeeded(); |
| } |
| |
| void UtilityThread::OnUnpackWebResource(const std::string& resource_data) { |
| // Parse json data. |
| // TODO(mrc): Add the possibility of a template that controls parsing, and |
| // the ability to download and verify images. |
| WebResourceUnpacker unpacker(resource_data); |
| if (unpacker.Run()) { |
| Send(new UtilityHostMsg_UnpackWebResource_Succeeded( |
| *unpacker.parsed_json())); |
| } else { |
| Send(new UtilityHostMsg_UnpackWebResource_Failed( |
| unpacker.error_message())); |
| } |
| |
| ReleaseProcessIfNeeded(); |
| } |
| |
| void UtilityThread::OnParseUpdateManifest(const std::string& xml) { |
| UpdateManifest manifest; |
| if (!manifest.Parse(xml)) { |
| Send(new UtilityHostMsg_ParseUpdateManifest_Failed(manifest.errors())); |
| } else { |
| Send(new UtilityHostMsg_ParseUpdateManifest_Succeeded(manifest.results())); |
| } |
| ReleaseProcessIfNeeded(); |
| } |
| |
| void UtilityThread::OnDecodeImage( |
| const std::vector<unsigned char>& encoded_data) { |
| webkit_glue::ImageDecoder decoder; |
| const SkBitmap& decoded_image = decoder.Decode(&encoded_data[0], |
| encoded_data.size()); |
| if (decoded_image.empty()) { |
| Send(new UtilityHostMsg_DecodeImage_Failed()); |
| } else { |
| Send(new UtilityHostMsg_DecodeImage_Succeeded(decoded_image)); |
| } |
| ReleaseProcessIfNeeded(); |
| } |
| |
| |
| void UtilityThread::OnRenderPDFPagesToMetafile( |
| base::PlatformFile pdf_file, |
| const FilePath& metafile_path, |
| const gfx::Rect& render_area, |
| int render_dpi, |
| const std::vector<printing::PageRange>& page_ranges) { |
| bool succeeded = false; |
| #if defined(OS_WIN) |
| printing::NativeMetafile metafile; |
| int highest_rendered_page_number = 0; |
| succeeded = RenderPDFToWinMetafile(pdf_file, metafile_path, render_area, |
| render_dpi, page_ranges, &metafile, |
| &highest_rendered_page_number); |
| if (succeeded) { |
| Send(new UtilityHostMsg_RenderPDFPagesToMetafile_Succeeded( |
| highest_rendered_page_number)); |
| } |
| #endif // defined(OS_WIN) |
| if (!succeeded) { |
| Send(new UtilityHostMsg_RenderPDFPagesToMetafile_Failed()); |
| } |
| ReleaseProcessIfNeeded(); |
| } |
| |
| #if defined(OS_WIN) |
| // Exported by pdf.dll |
| typedef bool (*RenderPDFPageToDCProc)( |
| const unsigned char* pdf_buffer, int buffer_size, int page_number, HDC dc, |
| int dpi_x, int dpi_y, int bounds_origin_x, int bounds_origin_y, |
| int bounds_width, int bounds_height, bool fit_to_bounds, |
| bool stretch_to_bounds, bool keep_aspect_ratio, bool center_in_bounds); |
| |
| typedef bool (*GetPDFDocInfoProc)(const unsigned char* pdf_buffer, |
| int buffer_size, int* page_count, |
| double* max_page_width); |
| |
| // The 2 below IAT patch functions are almost identical to the code in |
| // render_process_impl.cc. This is needed to work around specific Windows APIs |
| // used by the Chrome PDF plugin that will fail in the sandbox. |
| static app::win::IATPatchFunction g_iat_patch_createdca; |
| HDC WINAPI UtilityProcess_CreateDCAPatch(LPCSTR driver_name, |
| LPCSTR device_name, |
| LPCSTR output, |
| const DEVMODEA* init_data) { |
| if (driver_name && |
| (std::string("DISPLAY") == std::string(driver_name))) |
| // CreateDC fails behind the sandbox, but not CreateCompatibleDC. |
| return CreateCompatibleDC(NULL); |
| |
| NOTREACHED(); |
| return CreateDCA(driver_name, device_name, output, init_data); |
| } |
| |
| static app::win::IATPatchFunction g_iat_patch_get_font_data; |
| DWORD WINAPI UtilityProcess_GetFontDataPatch( |
| HDC hdc, DWORD table, DWORD offset, LPVOID buffer, DWORD length) { |
| int rv = GetFontData(hdc, table, offset, buffer, length); |
| if (rv == GDI_ERROR && hdc) { |
| HFONT font = static_cast<HFONT>(GetCurrentObject(hdc, OBJ_FONT)); |
| |
| LOGFONT logfont; |
| if (GetObject(font, sizeof(LOGFONT), &logfont)) { |
| std::vector<char> font_data; |
| if (UtilityThread::current()->Send( |
| new UtilityHostMsg_PreCacheFont(logfont))) |
| rv = GetFontData(hdc, table, offset, buffer, length); |
| } |
| } |
| return rv; |
| } |
| |
| bool UtilityThread::RenderPDFToWinMetafile( |
| base::PlatformFile pdf_file, |
| const FilePath& metafile_path, |
| const gfx::Rect& render_area, |
| int render_dpi, |
| const std::vector<printing::PageRange>& page_ranges, |
| printing::NativeMetafile* metafile, |
| int* highest_rendered_page_number) { |
| *highest_rendered_page_number = -1; |
| ScopedHandle file(pdf_file); |
| FilePath pdf_module_path; |
| PathService::Get(chrome::FILE_PDF_PLUGIN, &pdf_module_path); |
| HMODULE pdf_module = GetModuleHandle(pdf_module_path.value().c_str()); |
| if (!pdf_module) |
| return false; |
| |
| RenderPDFPageToDCProc render_proc = |
| reinterpret_cast<RenderPDFPageToDCProc>( |
| GetProcAddress(pdf_module, "RenderPDFPageToDC")); |
| if (!render_proc) |
| return false; |
| |
| GetPDFDocInfoProc get_info_proc = reinterpret_cast<GetPDFDocInfoProc>( |
| GetProcAddress(pdf_module, "GetPDFDocInfo")); |
| if (!get_info_proc) |
| return false; |
| |
| // Patch the IAT for handling specific APIs known to fail in the sandbox. |
| if (!g_iat_patch_createdca.is_patched()) |
| g_iat_patch_createdca.Patch(pdf_module_path.value().c_str(), |
| "gdi32.dll", "CreateDCA", |
| UtilityProcess_CreateDCAPatch); |
| |
| if (!g_iat_patch_get_font_data.is_patched()) |
| g_iat_patch_get_font_data.Patch(pdf_module_path.value().c_str(), |
| "gdi32.dll", "GetFontData", |
| UtilityProcess_GetFontDataPatch); |
| |
| // TODO(sanjeevr): Add a method to the PDF DLL that takes in a file handle |
| // and a page range array. That way we don't need to read the entire PDF into |
| // memory. |
| DWORD length = ::GetFileSize(file, NULL); |
| if (length == INVALID_FILE_SIZE) |
| return false; |
| |
| std::vector<uint8> buffer; |
| buffer.resize(length); |
| DWORD bytes_read = 0; |
| if (!ReadFile(pdf_file, &buffer.front(), length, &bytes_read, NULL) || |
| (bytes_read != length)) |
| return false; |
| |
| int total_page_count = 0; |
| if (!get_info_proc(&buffer.front(), buffer.size(), &total_page_count, NULL)) |
| return false; |
| |
| metafile->CreateFileBackedDc(NULL, NULL, metafile_path); |
| // Since we created the metafile using the screen DPI (but we actually want |
| // the PDF DLL to print using the passed in render_dpi, we apply the following |
| // transformation. |
| SetGraphicsMode(metafile->hdc(), GM_ADVANCED); |
| XFORM xform = {0}; |
| int screen_dpi = GetDeviceCaps(GetDC(NULL), LOGPIXELSX); |
| xform.eM11 = xform.eM22 = |
| static_cast<float>(screen_dpi) / static_cast<float>(render_dpi); |
| ModifyWorldTransform(metafile->hdc(), &xform, MWT_LEFTMULTIPLY); |
| |
| bool ret = false; |
| std::vector<printing::PageRange>::const_iterator iter; |
| for (iter = page_ranges.begin(); iter != page_ranges.end(); ++iter) { |
| for (int page_number = iter->from; page_number <= iter->to; ++page_number) { |
| if (page_number >= total_page_count) |
| break; |
| metafile->StartPage(); |
| if (render_proc(&buffer.front(), buffer.size(), page_number, |
| metafile->hdc(), render_dpi, render_dpi, |
| render_area.x(), render_area.y(), render_area.width(), |
| render_area.height(), true, false, true, true)) |
| if (*highest_rendered_page_number < page_number) |
| *highest_rendered_page_number = page_number; |
| ret = true; |
| metafile->EndPage(); |
| } |
| } |
| metafile->CloseDc(); |
| return ret; |
| } |
| #endif // defined(OS_WIN) |
| |
| void UtilityThread::OnIDBKeysFromValuesAndKeyPath( |
| int id, |
| const std::vector<SerializedScriptValue>& serialized_script_values, |
| const string16& idb_key_path) { |
| std::vector<WebKit::WebSerializedScriptValue> web_values; |
| ConvertVector(serialized_script_values, &web_values); |
| std::vector<WebKit::WebIDBKey> web_keys; |
| bool error = webkit_glue::IDBKeysFromValuesAndKeyPath( |
| web_values, idb_key_path, &web_keys); |
| if (error) { |
| Send(new UtilityHostMsg_IDBKeysFromValuesAndKeyPath_Failed(id)); |
| return; |
| } |
| std::vector<IndexedDBKey> keys; |
| ConvertVector(web_keys, &keys); |
| Send(new UtilityHostMsg_IDBKeysFromValuesAndKeyPath_Succeeded(id, keys)); |
| ReleaseProcessIfNeeded(); |
| } |
| |
| void UtilityThread::OnBatchModeStarted() { |
| batch_mode_ = true; |
| } |
| |
| void UtilityThread::OnBatchModeFinished() { |
| ChildProcess::current()->ReleaseProcess(); |
| } |
| |
| void UtilityThread::ReleaseProcessIfNeeded() { |
| if (!batch_mode_) |
| ChildProcess::current()->ReleaseProcess(); |
| } |