Add Postscript Printing

chrome/browser/printing/pdf_to_emf_converter.cc

 // Copyright 2014 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/pdf_to_emf_converter.h"
 
 #include <stdint.h>
 #include <windows.h>
 
 #include <memory>
@@ skipping 59 matching lines @@
  public:
   LazyEmf(const scoped_refptr<RefCountedTempDir>& temp_dir, ScopedTempFile file)
       : temp_dir_(temp_dir), file_(std::move(file)) {
     CHECK(file_);
   }
   ~LazyEmf() override { Close(); }
 
  protected:
   // MetafilePlayer:
   bool SafePlayback(HDC hdc) const override;
-
   void Close() const;
   bool LoadEmf(Emf* emf) const;
 
  private:
   mutable scoped_refptr<RefCountedTempDir> temp_dir_;
   mutable ScopedTempFile file_;  // Mutable because of consts in base class.
 
   bool GetDataAsVector(std::vector<char>* buffer) const override;
   bool SaveTo(base::File* file) const override;
 
@@ skipping 14 matching lines @@
 // All these steps work sequentially, so no data should be accessed
 // simultaneously by several threads.
 class PdfConverterUtilityProcessHostClient
     : public content::UtilityProcessHostClient {
  public:
   PdfConverterUtilityProcessHostClient(
       base::WeakPtr<PdfConverterImpl> converter,
       const PdfRenderSettings& settings);
 
   void Start(const scoped_refptr<base::RefCountedMemory>& data,
-             bool print_text_with_gdi,
              const PdfConverter::StartCallback& start_callback);
 
   void GetPage(int page_number,
                const PdfConverter::GetPageCallback& get_page_callback);
 
   void Stop();
 
   // UtilityProcessHostClient implementation.
   void OnProcessCrashed(int exit_code) override;
   void OnProcessLaunchFailed(int exit_code) override;
@@ skipping 19 matching lines @@
       return *this;
     }
 
     int page_number() const { return page_number_; }
     const PdfConverter::GetPageCallback& callback() const { return callback_; }
     ScopedTempFile TakeFile() { return std::move(file_); }
     void set_file(ScopedTempFile file) { file_ = std::move(file); }
 
    private:
     int page_number_;
-
     PdfConverter::GetPageCallback callback_;
     ScopedTempFile file_;
 
     DISALLOW_COPY_AND_ASSIGN(GetPageCallbackData);
   };
 
   ~PdfConverterUtilityProcessHostClient() override;
 
   // Helper functions: must be overridden by subclasses
   // Set the process name
   virtual base::string16 GetName() const = 0;
   // Create a metafileplayer subclass file from a temporary file.
   virtual std::unique_ptr<MetafilePlayer> GetFileFromTemp(
       std::unique_ptr<base::File, content::BrowserThread::DeleteOnFileThread>
           temp_file) = 0;
   // Send the messages to Start, GetPage, and Stop.
-  virtual void SendStartMessage(IPC::PlatformFileForTransit transit,
-                                bool print_text_with_gdi) = 0;
+  virtual void SendStartMessage(IPC::PlatformFileForTransit transit) = 0;
   virtual void SendGetPageMessage(int page_number,
                                   IPC::PlatformFileForTransit transit) = 0;
   virtual void SendStopMessage() = 0;
 
   // Message handlers:
   void OnPageCount(int page_count);
   void OnPageDone(bool success, float scale_factor);
 
   void OnFailed();
-  void OnTempPdfReady(bool print_text_with_gdi, ScopedTempFile pdf);
+  void OnTempPdfReady(ScopedTempFile pdf);
   void OnTempFileReady(GetPageCallbackData* callback_data,
                        ScopedTempFile temp_file);
-
   scoped_refptr<RefCountedTempDir> temp_dir_;
 
   // Used to suppress callbacks after PdfConverter is deleted.
   base::WeakPtr<PdfConverterImpl> converter_;
   PdfRenderSettings settings_;
 
   // Document loaded callback.
   PdfConverter::StartCallback start_callback_;
 
   // Process host for IPC.
@@ skipping 16 matching lines @@
 
   bool OnMessageReceived(const IPC::Message& message) override;
 
  private:
   ~PdfToEmfUtilityProcessHostClient() override;
   // Helpers to send messages and set process name
   base::string16 GetName() const override;
   std::unique_ptr<MetafilePlayer> GetFileFromTemp(
       std::unique_ptr<base::File, content::BrowserThread::DeleteOnFileThread>
           temp_file) override;
-  void SendStartMessage(IPC::PlatformFileForTransit transit,
-                        bool print_text_with_gdi) override;
+  void SendStartMessage(IPC::PlatformFileForTransit transit) override;
   void SendGetPageMessage(int page_number,
                           IPC::PlatformFileForTransit transit) override;
   void SendStopMessage() override;
 
   // Additional message handler needed for Pdf to Emf
   void OnPreCacheFontCharacters(const LOGFONT& log_font,
                                 const base::string16& characters);
 
   DISALLOW_COPY_AND_ASSIGN(PdfToEmfUtilityProcessHostClient);
 };
 
+// Converts PDF into PostScript.
+class PdfToPostScriptUtilityProcessHostClient
+    : public PdfConverterUtilityProcessHostClient {
+ public:
+  PdfToPostScriptUtilityProcessHostClient(
+      base::WeakPtr<PdfConverterImpl> converter,
+      const PdfRenderSettings& settings)
+      : PdfConverterUtilityProcessHostClient(converter, settings) {}
+
+  // UtilityProcessHostClient implementation.
+  bool OnMessageReceived(const IPC::Message& message) override;
+
+ private:
+  // Helpers to send messages and set process name
+  base::string16 GetName() const override;
+  std::unique_ptr<MetafilePlayer> GetFileFromTemp(
+      std::unique_ptr<base::File, content::BrowserThread::DeleteOnFileThread>
+          temp_file) override;
+  void SendStartMessage(IPC::PlatformFileForTransit transit) override;
+  void SendGetPageMessage(int page_number,
+                          IPC::PlatformFileForTransit transit) override;
+  void SendStopMessage() override;
+
+  // Additional message handler needed for Pdf to PostScript
+  void OnPageDone(bool success);
+
+  DISALLOW_COPY_AND_ASSIGN(PdfToPostScriptUtilityProcessHostClient);
+};
+
 class PdfConverterImpl : public PdfConverter {
  public:
   PdfConverterImpl();
 
   ~PdfConverterImpl() override;
 
   void Start(const scoped_refptr<base::RefCountedMemory>& data,
              const PdfRenderSettings& conversion_settings,
-             bool print_text_with_gdi,
              const StartCallback& start_callback) override;
 
   void GetPage(int page_number,
                const GetPageCallback& get_page_callback) override;
 
   // Helps to cancel callbacks if this object is destroyed.
   void RunCallback(const base::Closure& callback);
 
  protected:
   scoped_refptr<PdfConverterUtilityProcessHostClient> utility_client_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(PdfConverterImpl);
 };
 
 class PdfToEmfConverterImpl : public PdfConverterImpl {
  public:
   PdfToEmfConverterImpl();
 
   ~PdfToEmfConverterImpl() override;
 
   void Start(const scoped_refptr<base::RefCountedMemory>& data,
              const PdfRenderSettings& conversion_settings,
-             bool print_text_with_gdi,
              const StartCallback& start_callback) override;
 
  private:
   base::WeakPtrFactory<PdfToEmfConverterImpl> weak_ptr_factory_;
+  DISALLOW_COPY_AND_ASSIGN(PdfToEmfConverterImpl);
+};
 
-  DISALLOW_COPY_AND_ASSIGN(PdfToEmfConverterImpl);
+class PdfToPostScriptConverterImpl : public PdfConverterImpl {
+ public:
+  PdfToPostScriptConverterImpl();
+  ~PdfToPostScriptConverterImpl() {}
+
+  void Start(const scoped_refptr<base::RefCountedMemory>& data,
+             const PdfRenderSettings& conversion_settings,
+             const StartCallback& start_callback) override;
+
+ private:
+  base::WeakPtrFactory<PdfToPostScriptConverterImpl> weak_ptr_factory_;
+  DISALLOW_COPY_AND_ASSIGN(PdfToPostScriptConverterImpl);
 };
 
 ScopedTempFile CreateTempFile(scoped_refptr<RefCountedTempDir>* temp_dir) {
   if (!temp_dir->get())
     *temp_dir = new RefCountedTempDir();
   ScopedTempFile file;
   if (!(*temp_dir)->IsValid())
     return file;
   base::FilePath path;
   if (!base::CreateTemporaryFileInDir((*temp_dir)->GetPath(), &path)) {
@@ skipping 60 matching lines @@
   file_->Seek(base::File::FROM_BEGIN, 0);
   int64_t size = file_->GetLength();
   if (size <= 0)
     return false;
   std::vector<char> data(size);
   if (file_->ReadAtCurrentPos(data.data(), data.size()) != size)
     return false;
   return emf->InitFromData(data.data(), data.size());
 }
 
+// Postscript metafile subclass to override SafePlayback.
+class PostScriptMetaFile : public LazyEmf {
+ public:
+  PostScriptMetaFile(const scoped_refptr<RefCountedTempDir>& temp_dir,
+                     ScopedTempFile file)
+      : LazyEmf(temp_dir, std::move(file)) {}
+  ~PostScriptMetaFile() {}
+
+ protected:
+  // MetafilePlayer:
+  bool SafePlayback(HDC hdc) const override;
+
+  DISALLOW_COPY_AND_ASSIGN(PostScriptMetaFile);
+};
+
+bool PostScriptMetaFile::SafePlayback(HDC hdc) const {
+  // TODO(thestig): Fix destruction of metafiles. For some reasons
+  // instances of Emf are not deleted. https://crbug.com/260806
+  // It's known that the Emf going to be played just once to a printer. So just
+  // release |file_| before returning.
+  Emf emf;
+  if (!LoadEmf(&emf)) {
+    Close();
+    return false;
+  }
+
+  {
+    // Ensure enumerator destruction before calling Close() below.
+    Emf::Enumerator emf_enum(emf, nullptr, nullptr);

[Vitaly Buka (NO REVIEWS), 2017/01/27 00:38:49]

Why postscript is encoded as GDI comment into EMF?
Why not to send as regular postscript file?

[rbpotter, 2017/01/27 16:37:21]

This was originally done by thestig@, so I am not sure what the
reason was. I will look into writing new files similar to emf_win.cc/h
for postscript.

+    for (const Emf::Record& record : emf_enum) {
+      auto* emf_record = record.record();
+      if (emf_record->iType != EMR_GDICOMMENT)
+        continue;
+
+      const EMRGDICOMMENT* comment =
+          reinterpret_cast<const EMRGDICOMMENT*>(emf_record);
+      const char* data = reinterpret_cast<const char*>(comment->Data);
+      const uint16_t* ptr = reinterpret_cast<const uint16_t*>(data);
+      int ret = ExtEscape(hdc, PASSTHROUGH, 2 + *ptr, data, 0, nullptr);
+      DCHECK_EQ(*ptr, ret);
+    }
+  }
+  Close();
+  return true;
+}
+
 PdfConverterUtilityProcessHostClient::PdfConverterUtilityProcessHostClient(
     base::WeakPtr<PdfConverterImpl> converter,
     const PdfRenderSettings& settings)
     : converter_(converter), settings_(settings) {}
 
 PdfConverterUtilityProcessHostClient::~PdfConverterUtilityProcessHostClient() {}
 
 void PdfConverterUtilityProcessHostClient::Start(
     const scoped_refptr<base::RefCountedMemory>& data,
-    bool print_text_with_gdi,
     const PdfConverter::StartCallback& start_callback) {
   if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) {
     BrowserThread::PostTask(
         BrowserThread::IO, FROM_HERE,
         base::Bind(&PdfConverterUtilityProcessHostClient::Start, this, data,
-                   print_text_with_gdi, start_callback));
+                   start_callback));
     return;
   }
 
   // Store callback before any OnFailed() call to make it called on failure.
   start_callback_ = start_callback;
 
   // NOTE: This process _must_ be sandboxed, otherwise the pdf dll will load
   // gdiplus.dll, change how rendering happens, and not be able to correctly
   // generate when sent to a metafile DC.
   utility_process_host_ = content::UtilityProcessHost::Create(
                               this, base::ThreadTaskRunnerHandle::Get())
                               ->AsWeakPtr();
   utility_process_host_->SetName(GetName());
 
   BrowserThread::PostTaskAndReplyWithResult(
       BrowserThread::FILE, FROM_HERE,
       base::Bind(&CreateTempPdfFile, data, &temp_dir_),
-      base::Bind(&PdfConverterUtilityProcessHostClient::OnTempPdfReady, this,
-                 print_text_with_gdi));
+      base::Bind(&PdfConverterUtilityProcessHostClient::OnTempPdfReady, this));
 }
 
-void PdfConverterUtilityProcessHostClient::OnTempPdfReady(
-    bool print_text_with_gdi,
-    ScopedTempFile pdf) {
+void PdfConverterUtilityProcessHostClient::OnTempPdfReady(ScopedTempFile pdf) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   if (!utility_process_host_ || !pdf)
     return OnFailed();
   // Should reply with OnPageCount().
   SendStartMessage(
-      IPC::GetPlatformFileForTransit(pdf->GetPlatformFile(), false),
-      print_text_with_gdi);
+      IPC::GetPlatformFileForTransit(pdf->GetPlatformFile(), false));
 }
 
 void PdfConverterUtilityProcessHostClient::OnPageCount(int page_count) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   if (start_callback_.is_null())
     return OnFailed();
   BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
                           base::Bind(&PdfConverterImpl::RunCallback, converter_,
                                      base::Bind(start_callback_, page_count)));
   start_callback_.Reset();
@@ skipping 155 matching lines @@
 }
 
 void PdfToEmfUtilityProcessHostClient::SendGetPageMessage(
     int page_number,
     IPC::PlatformFileForTransit transit) {
   Send(new ChromeUtilityMsg_RenderPDFPagesToMetafiles_GetPage(page_number,
                                                               transit));
 }
 
 void PdfToEmfUtilityProcessHostClient::SendStartMessage(
-    IPC::PlatformFileForTransit transit,
-    bool print_text_with_gdi) {
-  Send(new ChromeUtilityMsg_RenderPDFPagesToMetafiles(transit, settings_,
-                                                      print_text_with_gdi));
+    IPC::PlatformFileForTransit transit) {
+  Send(new ChromeUtilityMsg_RenderPDFPagesToMetafiles(transit, settings_));
 }
 
 void PdfToEmfUtilityProcessHostClient::SendStopMessage() {
   Send(new ChromeUtilityMsg_RenderPDFPagesToMetafiles_Stop());
 }
 
+// Pdf to PostScript
+bool PdfToPostScriptUtilityProcessHostClient::OnMessageReceived(
+    const IPC::Message& message) {
+  bool handled = true;
+  IPC_BEGIN_MESSAGE_MAP(PdfToPostScriptUtilityProcessHostClient, message)
+    IPC_MESSAGE_HANDLER(
+        ChromeUtilityHostMsg_RenderPDFPagesToPostScript_PageCount, OnPageCount)
+    IPC_MESSAGE_HANDLER(
+        ChromeUtilityHostMsg_RenderPDFPagesToPostScript_PageDone, OnPageDone)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP()
+  return handled;
+}
+
+base::string16 PdfToPostScriptUtilityProcessHostClient::GetName() const {
+  return l10n_util::GetStringUTF16(IDS_UTILITY_PROCESS_PS_CONVERTOR_NAME);
+}
+
+std::unique_ptr<MetafilePlayer>
+PdfToPostScriptUtilityProcessHostClient::GetFileFromTemp(
+    std::unique_ptr<base::File, content::BrowserThread::DeleteOnFileThread>
+        temp_file) {
+  return base::MakeUnique<PostScriptMetaFile>(temp_dir_, std::move(temp_file));
+}
+
+void PdfToPostScriptUtilityProcessHostClient::SendGetPageMessage(
+    int page_number,
+    IPC::PlatformFileForTransit transit) {
+  Send(new ChromeUtilityMsg_RenderPDFPagesToPostScript_GetPage(page_number,
+                                                               transit));
+}
+
+void PdfToPostScriptUtilityProcessHostClient::SendStartMessage(
+    IPC::PlatformFileForTransit transit) {
+  Send(new ChromeUtilityMsg_RenderPDFPagesToPostScript_Start(transit,
+                                                             settings_));
+}
+
+void PdfToPostScriptUtilityProcessHostClient::SendStopMessage() {
+  Send(new ChromeUtilityMsg_RenderPDFPagesToPostScript_Stop());
+}
+
+void PdfToPostScriptUtilityProcessHostClient::OnPageDone(bool success) {

[Vitaly Buka (NO REVIEWS), 2017/01/27 00:38:49]

Maybe we can have single PdfConverterUtilityProcessHostClient for both
converters?

[rbpotter, 2017/01/27 16:37:21]

Sorry, I am not sure what you mean exactly. Do you think the classes
should be combined entirely? They send different messages and have to
set the metafile differently, which is why I had them separate.
I could split out the message handlers again and have one large
OnMessageReceived function; is that closer to what you are thinking?

+  PdfConverterUtilityProcessHostClient::OnPageDone(success, 0.0f);
+}
+
 // Pdf Converter Impl and subclasses
 PdfConverterImpl::PdfConverterImpl() {}
 
 PdfConverterImpl::~PdfConverterImpl() {}
 
 void PdfConverterImpl::Start(const scoped_refptr<base::RefCountedMemory>& data,
                              const PdfRenderSettings& conversion_settings,
-                             bool print_text_with_gdi,
                              const StartCallback& start_callback) {
   DCHECK(!utility_client_.get());
 }
 
 void PdfConverterImpl::GetPage(int page_number,
                                const GetPageCallback& get_page_callback) {
   utility_client_->GetPage(page_number, get_page_callback);
 }
 
 void PdfConverterImpl::RunCallback(const base::Closure& callback) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   callback.Run();
 }
 
 PdfToEmfConverterImpl::PdfToEmfConverterImpl() : weak_ptr_factory_(this) {}
 
 PdfToEmfConverterImpl::~PdfToEmfConverterImpl() {
   if (utility_client_.get())
     utility_client_->Stop();
 }
 
 void PdfToEmfConverterImpl::Start(
     const scoped_refptr<base::RefCountedMemory>& data,
     const PdfRenderSettings& conversion_settings,
-    bool print_text_with_gdi,
     const StartCallback& start_callback) {
   DCHECK(!utility_client_.get());
   utility_client_ = new PdfToEmfUtilityProcessHostClient(
       weak_ptr_factory_.GetWeakPtr(), conversion_settings);
-  utility_client_->Start(data, print_text_with_gdi, start_callback);
+  utility_client_->Start(data, start_callback);
+}
+
+PdfToPostScriptConverterImpl::PdfToPostScriptConverterImpl()
+    : weak_ptr_factory_(this) {}
+
+void PdfToPostScriptConverterImpl::Start(
+    const scoped_refptr<base::RefCountedMemory>& data,
+    const PdfRenderSettings& conversion_settings,
+    const StartCallback& start_callback) {
+  DCHECK(!utility_client_.get());

[Vitaly Buka (NO REVIEWS), 2017/01/27 00:38:49]

There is also
https://cs.chromium.org/chromium/src/chrome/service/service_utility_process_h...

which may also benefit from postscript printing

[rbpotter, 2017/01/27 16:37:21]

Should we add postscript printing there now or in a later patch?

+  utility_client_ = new PdfToPostScriptUtilityProcessHostClient(
+      weak_ptr_factory_.GetWeakPtr(), conversion_settings);
+  utility_client_->Start(data, start_callback);
 }
 
 }  // namespace
 
 PdfConverter::~PdfConverter() {}
 
 // static
 std::unique_ptr<PdfConverter> PdfConverter::CreatePdfToEmfConverter() {
   return base::MakeUnique<PdfToEmfConverterImpl>();
 }
 
+// static
+std::unique_ptr<PdfConverter> PdfConverter::CreatePdfToPostScriptConverter() {
+  return base::MakeUnique<PdfToPostScriptConverterImpl>();
+}
+
 }  // namespace printing