chrome/utility/printing_handler.cc

   1 // Copyright 2014 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/utility/printing_handler.h"
   6
   7 #include "base/files/file_util.h"
   8 #include "base/lazy_instance.h"
   9 #include "base/path_service.h"
  10 #include "base/scoped_native_library.h"
  11 #include "chrome/common/chrome_paths.h"
  12 #include "chrome/common/chrome_utility_printing_messages.h"
  13 #include "chrome/utility/cloud_print/bitmap_image.h"
  14 #include "chrome/utility/cloud_print/pwg_encoder.h"
  15 #include "content/public/utility/utility_thread.h"
  16 #include "printing/page_range.h"
  17 #include "printing/pdf_render_settings.h"
  18
  19 #if defined(OS_WIN)
  20 #include "base/win/iat_patch_function.h"
  21 #include "printing/emf_win.h"
  22 #include "ui/gfx/gdi_util.h"
  23 #endif
  24
  25 #if defined(ENABLE_FULL_PRINTING)
  26 #include "chrome/common/crash_keys.h"
  27 #include "printing/backend/print_backend.h"
  28 #endif
  29
  30 namespace {
  31
  32 bool Send(IPC::Message* message) {
  33   return content::UtilityThread::Get()->Send(message);
  34 }
  35
  36 void ReleaseProcessIfNeeded() {
  37   content::UtilityThread::Get()->ReleaseProcessIfNeeded();
  38 }
  39
  40 class PdfFunctionsBase {
  41  public:
  42   PdfFunctionsBase() : render_pdf_to_bitmap_func_(NULL),
  43                        get_pdf_doc_info_func_(NULL) {}
  44
  45   bool Init() {
  46     base::FilePath pdf_module_path;
  47     if (!PathService::Get(chrome::FILE_PDF_PLUGIN, &pdf_module_path) ||
  48         !base::PathExists(pdf_module_path)) {
  49       return false;
  50     }
  51
  52     pdf_lib_.Reset(base::LoadNativeLibrary(pdf_module_path, NULL));
  53     if (!pdf_lib_.is_valid()) {
  54       LOG(WARNING) << "Couldn't load PDF plugin";
  55       return false;
  56     }
  57
  58     render_pdf_to_bitmap_func_ =
  59         reinterpret_cast<RenderPDFPageToBitmapProc>(
  60             pdf_lib_.GetFunctionPointer("RenderPDFPageToBitmap"));
  61     LOG_IF(WARNING, !render_pdf_to_bitmap_func_) <<
  62         "Missing RenderPDFPageToBitmap";
  63
  64     get_pdf_doc_info_func_ =
  65         reinterpret_cast<GetPDFDocInfoProc>(
  66             pdf_lib_.GetFunctionPointer("GetPDFDocInfo"));
  67     LOG_IF(WARNING, !get_pdf_doc_info_func_) << "Missing GetPDFDocInfo";
  68
  69     if (!render_pdf_to_bitmap_func_ || !get_pdf_doc_info_func_ ||
  70         !PlatformInit(pdf_module_path, pdf_lib_)) {
  71       Reset();
  72     }
  73
  74     return IsValid();
  75   }
  76
  77   bool IsValid() const {
  78     return pdf_lib_.is_valid();
  79   }
  80
  81   void Reset() {
  82     pdf_lib_.Reset(NULL);
  83   }
  84
  85   bool RenderPDFPageToBitmap(const void* pdf_buffer,
  86                              int pdf_buffer_size,
  87                              int page_number,
  88                              void* bitmap_buffer,
  89                              int bitmap_width,
  90                              int bitmap_height,
  91                              int dpi_x,
  92                              int dpi_y,
  93                              bool autorotate) {
  94     if (!render_pdf_to_bitmap_func_)
  95       return false;
  96     return render_pdf_to_bitmap_func_(pdf_buffer, pdf_buffer_size, page_number,
  97                                       bitmap_buffer, bitmap_width,
  98                                       bitmap_height, dpi_x, dpi_y, autorotate);
  99   }
 100
 101   bool GetPDFDocInfo(const void* pdf_buffer,
 102                      int buffer_size,
 103                      int* page_count,
 104                      double* max_page_width) {
 105     if (!get_pdf_doc_info_func_)
 106       return false;
 107     return get_pdf_doc_info_func_(pdf_buffer, buffer_size, page_count,
 108                                   max_page_width);
 109   }
 110
 111  protected:
 112   virtual bool PlatformInit(
 113       const base::FilePath& pdf_module_path,
 114       const base::ScopedNativeLibrary& pdf_lib) {
 115     return true;
 116   }
 117
 118  private:
 119   // Exported by PDF plugin.
 120   typedef bool (*RenderPDFPageToBitmapProc)(const void* pdf_buffer,
 121                                             int pdf_buffer_size,
 122                                             int page_number,
 123                                             void* bitmap_buffer,
 124                                             int bitmap_width,
 125                                             int bitmap_height,
 126                                             int dpi_x,
 127                                             int dpi_y,
 128                                             bool autorotate);
 129   typedef bool (*GetPDFDocInfoProc)(const void* pdf_buffer,
 130                                     int buffer_size, int* page_count,
 131                                     double* max_page_width);
 132
 133   RenderPDFPageToBitmapProc render_pdf_to_bitmap_func_;
 134   GetPDFDocInfoProc get_pdf_doc_info_func_;
 135
 136   base::ScopedNativeLibrary pdf_lib_;
 137   DISALLOW_COPY_AND_ASSIGN(PdfFunctionsBase);
 138 };
 139
 140 #if defined(OS_WIN)
 141 // The 2 below IAT patch functions are almost identical to the code in
 142 // render_process_impl.cc. This is needed to work around specific Windows APIs
 143 // used by the Chrome PDF plugin that will fail in the sandbox.
 144 static base::win::IATPatchFunction g_iat_patch_createdca;
 145 HDC WINAPI UtilityProcess_CreateDCAPatch(LPCSTR driver_name,
 146                                          LPCSTR device_name,
 147                                          LPCSTR output,
 148                                          const DEVMODEA* init_data) {
 149   if (driver_name && (std::string("DISPLAY") == driver_name)) {
 150     // CreateDC fails behind the sandbox, but not CreateCompatibleDC.
 151     return CreateCompatibleDC(NULL);
 152   }
 153
 154   NOTREACHED();
 155   return CreateDCA(driver_name, device_name, output, init_data);
 156 }
 157
 158 static base::win::IATPatchFunction g_iat_patch_get_font_data;
 159 DWORD WINAPI UtilityProcess_GetFontDataPatch(
 160     HDC hdc, DWORD table, DWORD offset, LPVOID buffer, DWORD length) {
 161   int rv = GetFontData(hdc, table, offset, buffer, length);
 162   if (rv == GDI_ERROR && hdc) {
 163     HFONT font = static_cast<HFONT>(GetCurrentObject(hdc, OBJ_FONT));
 164
 165     LOGFONT logfont;
 166     if (GetObject(font, sizeof(LOGFONT), &logfont)) {
 167       content::UtilityThread::Get()->PreCacheFont(logfont);
 168       rv = GetFontData(hdc, table, offset, buffer, length);
 169       content::UtilityThread::Get()->ReleaseCachedFonts();
 170     }
 171   }
 172   return rv;
 173 }
 174
 175 class PdfFunctionsWin : public PdfFunctionsBase {
 176  public:
 177   PdfFunctionsWin() : render_pdf_to_dc_func_(NULL) {
 178   }
 179
 180   bool PlatformInit(
 181       const base::FilePath& pdf_module_path,
 182       const base::ScopedNativeLibrary& pdf_lib) OVERRIDE {
 183     // Patch the IAT for handling specific APIs known to fail in the sandbox.
 184     if (!g_iat_patch_createdca.is_patched()) {
 185       g_iat_patch_createdca.Patch(pdf_module_path.value().c_str(),
 186                                   "gdi32.dll", "CreateDCA",
 187                                   UtilityProcess_CreateDCAPatch);
 188     }
 189
 190     if (!g_iat_patch_get_font_data.is_patched()) {
 191       g_iat_patch_get_font_data.Patch(pdf_module_path.value().c_str(),
 192                                       "gdi32.dll", "GetFontData",
 193                                       UtilityProcess_GetFontDataPatch);
 194     }
 195     render_pdf_to_dc_func_ =
 196       reinterpret_cast<RenderPDFPageToDCProc>(
 197           pdf_lib.GetFunctionPointer("RenderPDFPageToDC"));
 198     LOG_IF(WARNING, !render_pdf_to_dc_func_) << "Missing RenderPDFPageToDC";
 199
 200     return render_pdf_to_dc_func_ != NULL;
 201   }
 202
 203   bool RenderPDFPageToDC(const void* pdf_buffer,
 204                          int buffer_size,
 205                          int page_number,
 206                          HDC dc,
 207                          int dpi_x,
 208                          int dpi_y,
 209                          int bounds_origin_x,
 210                          int bounds_origin_y,
 211                          int bounds_width,
 212                          int bounds_height,
 213                          bool fit_to_bounds,
 214                          bool stretch_to_bounds,
 215                          bool keep_aspect_ratio,
 216                          bool center_in_bounds,
 217                          bool autorotate) {
 218     if (!render_pdf_to_dc_func_)
 219       return false;
 220     return render_pdf_to_dc_func_(pdf_buffer, buffer_size, page_number,
 221                                   dc, dpi_x, dpi_y, bounds_origin_x,
 222                                   bounds_origin_y, bounds_width, bounds_height,
 223                                   fit_to_bounds, stretch_to_bounds,
 224                                   keep_aspect_ratio, center_in_bounds,
 225                                   autorotate);
 226   }
 227
 228  private:
 229   // Exported by PDF plugin.
 230   typedef bool (*RenderPDFPageToDCProc)(
 231       const void* pdf_buffer, int buffer_size, int page_number, HDC dc,
 232       int dpi_x, int dpi_y, int bounds_origin_x, int bounds_origin_y,
 233       int bounds_width, int bounds_height, bool fit_to_bounds,
 234       bool stretch_to_bounds, bool keep_aspect_ratio, bool center_in_bounds,
 235       bool autorotate);
 236   RenderPDFPageToDCProc render_pdf_to_dc_func_;
 237
 238   DISALLOW_COPY_AND_ASSIGN(PdfFunctionsWin);
 239 };
 240
 241 typedef PdfFunctionsWin PdfFunctions;
 242 #else  // OS_WIN
 243 typedef PdfFunctionsBase PdfFunctions;
 244 #endif  // OS_WIN
 245
 246 base::LazyInstance<PdfFunctions> g_pdf_lib = LAZY_INSTANCE_INITIALIZER;
 247
 248 }  // namespace
 249
 250 PrintingHandler::PrintingHandler() {}
 251
 252 PrintingHandler::~PrintingHandler() {}
 253
 254 // static
 255 void PrintingHandler::PreSandboxStartup() {
 256   g_pdf_lib.Get().Init();
 257 }
 258
 259 bool PrintingHandler::OnMessageReceived(const IPC::Message& message) {
 260   bool handled = true;
 261   IPC_BEGIN_MESSAGE_MAP(PrintingHandler, message)
 262 #if defined(WIN_PDF_METAFILE_FOR_PRINTING)
 263     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_RenderPDFPagesToMetafiles,
 264                         OnRenderPDFPagesToMetafile)
 265 #endif
 266     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_RenderPDFPagesToPWGRaster,
 267                         OnRenderPDFPagesToPWGRaster)
 268     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_GetPrinterCapsAndDefaults,
 269                         OnGetPrinterCapsAndDefaults)
 270     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_GetPrinterSemanticCapsAndDefaults,
 271                         OnGetPrinterSemanticCapsAndDefaults)
 272     IPC_MESSAGE_UNHANDLED(handled = false)
 273   IPC_END_MESSAGE_MAP()
 274   return handled;
 275 }
 276
 277 #if defined(WIN_PDF_METAFILE_FOR_PRINTING)
 278 void PrintingHandler::OnRenderPDFPagesToMetafile(
 279     IPC::PlatformFileForTransit pdf_transit,
 280     const base::FilePath& metafile_path,
 281     const printing::PdfRenderSettings& settings,
 282     const std::vector<printing::PageRange>& page_ranges_const) {
 283   bool succeeded = false;
 284   base::File pdf_file = IPC::PlatformFileForTransitToFile(pdf_transit);
 285   int highest_rendered_page_number = 0;
 286   double scale_factor = 1.0;
 287   std::vector<printing::PageRange> page_ranges = page_ranges_const;
 288   succeeded = RenderPDFToWinMetafile(pdf_file.Pass(),
 289                                      metafile_path,
 290                                      settings,
 291                                      &page_ranges,
 292                                      &highest_rendered_page_number,
 293                                      &scale_factor);
 294   if (succeeded) {
 295     // TODO(vitalybuka|scottmg): http://crbug.com/170859. These could
 296     // potentially be sent as each page is converted so that the spool could
 297     // start sooner.
 298     Send(new ChromeUtilityHostMsg_RenderPDFPagesToMetafiles_Succeeded(
 299         page_ranges, scale_factor));
 300   } else {
 301     Send(new ChromeUtilityHostMsg_RenderPDFPagesToMetafile_Failed());
 302   }
 303   ReleaseProcessIfNeeded();
 304 }
 305 #endif
 306
 307 void PrintingHandler::OnRenderPDFPagesToPWGRaster(
 308     IPC::PlatformFileForTransit pdf_transit,
 309     const printing::PdfRenderSettings& settings,
 310     const printing::PwgRasterSettings& bitmap_settings,
 311     IPC::PlatformFileForTransit bitmap_transit) {
 312   base::File pdf = IPC::PlatformFileForTransitToFile(pdf_transit);
 313   base::File bitmap = IPC::PlatformFileForTransitToFile(bitmap_transit);
 314   if (RenderPDFPagesToPWGRaster(pdf.Pass(), settings, bitmap_settings,
 315                                 bitmap.Pass())) {
 316     Send(new ChromeUtilityHostMsg_RenderPDFPagesToPWGRaster_Succeeded());
 317   } else {
 318     Send(new ChromeUtilityHostMsg_RenderPDFPagesToPWGRaster_Failed());
 319   }
 320   ReleaseProcessIfNeeded();
 321 }
 322
 323 #if defined(WIN_PDF_METAFILE_FOR_PRINTING)
 324 bool PrintingHandler::RenderPDFToWinMetafile(
 325     base::File pdf_file,
 326     const base::FilePath& metafile_path,
 327     const printing::PdfRenderSettings& settings,
 328     std::vector<printing::PageRange>* page_ranges,
 329     int* highest_rendered_page_number,
 330     double* scale_factor) {
 331   DCHECK(page_ranges);
 332   *highest_rendered_page_number = -1;
 333   *scale_factor = 1.0;
 334
 335   if (!g_pdf_lib.Get().IsValid())
 336     return false;
 337
 338   // TODO(sanjeevr): Add a method to the PDF DLL that takes in a file handle
 339   // and a page range array. That way we don't need to read the entire PDF into
 340   // memory.
 341   int64 length = pdf_file.GetLength();
 342   if (length < 0)
 343     return false;
 344
 345   std::vector<char> buffer;
 346   buffer.resize(length);
 347   if (length != pdf_file.Read(0, &buffer.front(), length))
 348     return false;
 349
 350   int total_page_count = 0;
 351   if (!g_pdf_lib.Get().GetPDFDocInfo(&buffer.front(), buffer.size(),
 352                                      &total_page_count, NULL)) {
 353     return false;
 354   }
 355
 356   // If no range supplied, do all pages.
 357   if (page_ranges->empty()) {
 358     printing::PageRange page_range_all;
 359     page_range_all.from = 0;
 360     page_range_all.to = total_page_count - 1;
 361     page_ranges->push_back(page_range_all);
 362   }
 363
 364   bool ret = false;
 365   std::vector<printing::PageRange>::const_iterator iter;
 366   for (iter = page_ranges->begin(); iter != page_ranges->end(); ++iter) {
 367     for (int page_number = iter->from; page_number <= iter->to; ++page_number) {
 368       if (page_number >= total_page_count)
 369         break;
 370
 371       printing::Emf metafile;
 372       metafile.InitToFile(metafile_path.InsertBeforeExtensionASCII(
 373           base::StringPrintf(".%d", page_number)));
 374
 375       // We need to scale down DC to fit an entire page into DC available area.
 376       // Current metafile is based on screen DC and have current screen size.
 377       // Writing outside of those boundaries will result in the cut-off output.
 378       // On metafiles (this is the case here), scaling down will still record
 379       // original coordinates and we'll be able to print in full resolution.
 380       // Before playback we'll need to counter the scaling up that will happen
 381       // in the service (print_system_win.cc).
 382       *scale_factor = gfx::CalculatePageScale(metafile.context(),
 383                                               settings.area().right(),
 384                                               settings.area().bottom());
 385       gfx::ScaleDC(metafile.context(), *scale_factor);
 386
 387       // The underlying metafile is of type Emf and ignores the arguments passed
 388       // to StartPage.
 389       metafile.StartPage(gfx::Size(), gfx::Rect(), 1);
 390       if (g_pdf_lib.Get().RenderPDFPageToDC(
 391               &buffer.front(), buffer.size(), page_number, metafile.context(),
 392               settings.dpi(), settings.dpi(), settings.area().x(),
 393               settings.area().y(), settings.area().width(),
 394               settings.area().height(), true, false, true, true,
 395               settings.autorotate())) {
 396         if (*highest_rendered_page_number < page_number)
 397           *highest_rendered_page_number = page_number;
 398         ret = true;
 399       }
 400       metafile.FinishPage();
 401       metafile.FinishDocument();
 402     }
 403   }
 404   return ret;
 405 }
 406 #endif  // defined(WIN_PDF_METAFILE_FOR_PRINTING)
 407
 408 bool PrintingHandler::RenderPDFPagesToPWGRaster(
 409     base::File pdf_file,
 410     const printing::PdfRenderSettings& settings,
 411     const printing::PwgRasterSettings& bitmap_settings,
 412     base::File bitmap_file) {
 413   bool autoupdate = true;
 414   if (!g_pdf_lib.Get().IsValid())
 415     return false;
 416
 417   base::File::Info info;
 418   if (!pdf_file.GetInfo(&info) || info.size <= 0)
 419     return false;
 420
 421   std::string data(info.size, 0);
 422   int data_size = pdf_file.Read(0, &data[0], data.size());
 423   if (data_size != static_cast<int>(data.size()))
 424     return false;
 425
 426   int total_page_count = 0;
 427   if (!g_pdf_lib.Get().GetPDFDocInfo(data.data(), data.size(),
 428                                      &total_page_count, NULL)) {
 429     return false;
 430   }
 431
 432   cloud_print::PwgEncoder encoder;
 433   std::string pwg_header;
 434   encoder.EncodeDocumentHeader(&pwg_header);
 435   int bytes_written = bitmap_file.WriteAtCurrentPos(pwg_header.data(),
 436                                                     pwg_header.size());
 437   if (bytes_written != static_cast<int>(pwg_header.size()))
 438     return false;
 439
 440   cloud_print::BitmapImage image(settings.area().size(),
 441                                  cloud_print::BitmapImage::BGRA);
 442   for (int i = 0; i < total_page_count; ++i) {
 443     int page_number = i;
 444
 445     if (bitmap_settings.reverse_page_order) {
 446       page_number = total_page_count - 1 - page_number;
 447     }
 448
 449     if (!g_pdf_lib.Get().RenderPDFPageToBitmap(data.data(),
 450                                                data.size(),
 451                                                page_number,
 452                                                image.pixel_data(),
 453                                                image.size().width(),
 454                                                image.size().height(),
 455                                                settings.dpi(),
 456                                                settings.dpi(),
 457                                                autoupdate)) {
 458       return false;
 459     }
 460
 461     cloud_print::PwgHeaderInfo header_info;
 462     header_info.dpi = settings.dpi();
 463     header_info.total_pages = total_page_count;
 464
 465     // Transform odd pages.
 466     if (page_number % 2) {
 467       switch (bitmap_settings.odd_page_transform) {
 468         case printing::TRANSFORM_NORMAL:
 469           break;
 470         case printing::TRANSFORM_ROTATE_180:
 471           header_info.flipx = true;
 472           header_info.flipy = true;
 473           break;
 474         case printing::TRANSFORM_FLIP_HORIZONTAL:
 475           header_info.flipx = true;
 476           break;
 477         case printing::TRANSFORM_FLIP_VERTICAL:
 478           header_info.flipy = true;
 479           break;
 480       }
 481     }
 482
 483     if (bitmap_settings.rotate_all_pages) {
 484       header_info.flipx = !header_info.flipx;
 485       header_info.flipy = !header_info.flipy;
 486     }
 487
 488     std::string pwg_page;
 489     if (!encoder.EncodePage(image, header_info, &pwg_page))
 490       return false;
 491     bytes_written = bitmap_file.WriteAtCurrentPos(pwg_page.data(),
 492                                                   pwg_page.size());
 493     if (bytes_written != static_cast<int>(pwg_page.size()))
 494       return false;
 495   }
 496   return true;
 497 }
 498
 499 void PrintingHandler::OnGetPrinterCapsAndDefaults(
 500     const std::string& printer_name) {
 501   scoped_refptr<printing::PrintBackend> print_backend =
 502       printing::PrintBackend::CreateInstance(NULL);
 503   printing::PrinterCapsAndDefaults printer_info;
 504
 505   crash_keys::ScopedPrinterInfo crash_key(
 506       print_backend->GetPrinterDriverInfo(printer_name));
 507
 508   if (print_backend->GetPrinterCapsAndDefaults(printer_name, &printer_info)) {
 509     Send(new ChromeUtilityHostMsg_GetPrinterCapsAndDefaults_Succeeded(
 510         printer_name, printer_info));
 511   } else  {
 512     Send(new ChromeUtilityHostMsg_GetPrinterCapsAndDefaults_Failed(
 513         printer_name));
 514   }
 515   ReleaseProcessIfNeeded();
 516 }
 517
 518 void PrintingHandler::OnGetPrinterSemanticCapsAndDefaults(
 519     const std::string& printer_name) {
 520   scoped_refptr<printing::PrintBackend> print_backend =
 521       printing::PrintBackend::CreateInstance(NULL);
 522   printing::PrinterSemanticCapsAndDefaults printer_info;
 523
 524   crash_keys::ScopedPrinterInfo crash_key(
 525       print_backend->GetPrinterDriverInfo(printer_name));
 526
 527   if (print_backend->GetPrinterSemanticCapsAndDefaults(printer_name,
 528                                                        &printer_info)) {
 529     Send(new ChromeUtilityHostMsg_GetPrinterSemanticCapsAndDefaults_Succeeded(
 530         printer_name, printer_info));
 531   } else {
 532     Send(new ChromeUtilityHostMsg_GetPrinterSemanticCapsAndDefaults_Failed(
 533         printer_name));
 534   }
 535   ReleaseProcessIfNeeded();
 536 }