Use XPS to set color for HP printers.

Wrap DocumentProperties into printing::CreateDevMode().

BUG=166587

Review URL: https://codereview.chromium.org/165993002

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@251408 0039d316-1c4b-4281-b951-d872f2087c98

3 files changed, 156 insertions, 15 deletions

diff --git a/printing/backend/print_backend_win.cc b/printing/backend/print_backend_win.cc
index aaf8da1..bb7989f 100644
--- a/printing/backend/print_backend_win.cc
+++ b/printing/backend/print_backend_win.cc
@@ -314,27 +314,18 @@ bool PrintBackendWin::GetPrinterCapsAndDefaults(
     }
     ScopedPrinterHandle printer_handle;
     if (printer_handle.OpenPrinter(printer_name_wide.c_str())) {
-      LONG devmode_size = DocumentProperties(
-          NULL, printer_handle, const_cast<LPTSTR>(printer_name_wide.c_str()),
-          NULL, NULL, 0);
-      if (devmode_size <= 0)
+      scoped_ptr<DEVMODE[]> devmode_out(CreateDevMode(printer_handle, NULL));
+      if (!devmode_out)
         return false;
-      scoped_ptr<BYTE[]> devmode_out_buffer(new BYTE[devmode_size]);
-      DEVMODE* devmode_out =
-          reinterpret_cast<DEVMODE*>(devmode_out_buffer.get());
-      DocumentProperties(
-          NULL, printer_handle, const_cast<LPTSTR>(printer_name_wide.c_str()),
-          devmode_out, NULL, DM_OUT_BUFFER);
       base::win::ScopedComPtr<IStream> printer_defaults_stream;
       hr = CreateStreamOnHGlobal(NULL, TRUE,
                                  printer_defaults_stream.Receive());
       DCHECK(SUCCEEDED(hr));
       if (printer_defaults_stream) {
-        hr = XPSModule::ConvertDevModeToPrintTicket(provider,
-                                                    devmode_size,
-                                                    devmode_out,
-                                                    kPTJobScope,
-                                                    printer_defaults_stream);
+        DWORD dm_size = devmode_out.get()->dmSize +
+                        devmode_out.get()->dmDriverExtra;
+        hr = XPSModule::ConvertDevModeToPrintTicket(provider, dm_size,
+            devmode_out.get(), kPTJobScope, printer_defaults_stream);
         DCHECK(SUCCEEDED(hr));
         if (SUCCEEDED(hr)) {
           hr = StreamOnHGlobalToString(printer_defaults_stream.get(),
diff --git a/printing/backend/win_helper.cc b/printing/backend/win_helper.cc
index f7b3310..8a66720 100644
--- a/printing/backend/win_helper.cc
+++ b/printing/backend/win_helper.cc
@@ -10,7 +10,11 @@
 #include "base/files/file_path.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/numerics/safe_conversions.h"
+#include "base/strings/string_util.h"
+#include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
+#include "base/win/scoped_comptr.h"
 #include "printing/backend/print_backend.h"
 #include "printing/backend/print_backend_consts.h"
 #include "printing/backend/printing_info_win.h"
@@ -75,6 +79,42 @@ PTReleaseMemoryProc g_release_memory_proc = NULL;
 PTCloseProviderProc g_close_provider_proc = NULL;
 StartXpsPrintJobProc g_start_xps_print_job_proc = NULL;
 
+HRESULT StreamFromPrintTicket(const std::string& print_ticket,
+                              IStream** stream) {
+  DCHECK(stream);
+  HRESULT hr = CreateStreamOnHGlobal(NULL, TRUE, stream);
+  if (FAILED(hr)) {
+    return hr;
+  }
+  ULONG bytes_written = 0;
+  (*stream)->Write(print_ticket.c_str(),
+                   base::checked_cast<ULONG>(print_ticket.length()),
+                   &bytes_written);
+  DCHECK(bytes_written == print_ticket.length());
+  LARGE_INTEGER pos = {0};
+  ULARGE_INTEGER new_pos = {0};
+  (*stream)->Seek(pos, STREAM_SEEK_SET, &new_pos);
+  return S_OK;
+}
+
+const char kXpsTicketTemplate[] =
+  "<?xml version='1.0' encoding='UTF-8'?>"
+  "<psf:PrintTicket "
+  "xmlns:psf='"
+  "http://schemas.microsoft.com/windows/2003/08/printing/printschemaframework' "
+  "xmlns:psk="
+  "'http://schemas.microsoft.com/windows/2003/08/printing/printschemakeywords' "
+  "version='1'>"
+  "<psf:Feature name='psk:PageOutputColor'>"
+  "<psf:Option name='psk:%s'>"
+  "</psf:Option>"
+  "</psf:Feature>"
+  "</psf:PrintTicket>";
+
+const char kXpsTicketColor[] = "Color";
+const char kXpsTicketMonochrome[] = "Monochrome";
+
+
 }  // namespace
 
 
@@ -345,4 +385,98 @@ std::string GetDriverInfo(HANDLE printer) {
   return driver_info;
 }
 
+scoped_ptr<DEVMODE[]> XpsTicketToDevMode(const base::string16& printer_name,
+                                         const std::string& print_ticket) {
+  scoped_ptr<DEVMODE[]> scoped_dev_mode;
+  printing::ScopedXPSInitializer xps_initializer;
+  if (!xps_initializer.initialized()) {
+    // TODO(sanjeevr): Handle legacy proxy case (with no prntvpt.dll)
+    return scoped_dev_mode.Pass();
+  }
+
+  printing::ScopedPrinterHandle printer;
+  if (!printer.OpenPrinter(printer_name.c_str()))
+    return scoped_dev_mode.Pass();
+
+  base::win::ScopedComPtr<IStream> pt_stream;
+  HRESULT hr = StreamFromPrintTicket(print_ticket, pt_stream.Receive());
+  if (FAILED(hr))
+    return scoped_dev_mode.Pass();
+
+  HPTPROVIDER provider = NULL;
+  hr = printing::XPSModule::OpenProvider(printer_name, 1, &provider);
+  if (SUCCEEDED(hr)) {
+    ULONG size = 0;
+    DEVMODE* dm = NULL;
+    // Use kPTJobScope, because kPTDocumentScope breaks duplex.
+    hr = printing::XPSModule::ConvertPrintTicketToDevMode(provider,
+                                                          pt_stream,
+                                                          kUserDefaultDevmode,
+                                                          kPTJobScope,
+                                                          &size,
+                                                          &dm,
+                                                          NULL);
+    if (SUCCEEDED(hr)) {
+      // Correct DEVMODE using DocumentProperties. See documentation for
+      // PTConvertPrintTicketToDevMode.
+      scoped_dev_mode = CreateDevMode(printer, dm);
+      printing::XPSModule::ReleaseMemory(dm);
+    }
+    printing::XPSModule::CloseProvider(provider);
+  }
+  return scoped_dev_mode.Pass();
+}
+
+scoped_ptr<DEVMODE[]> CreateDevModeWithColor(HANDLE printer,
+                                             const base::string16& printer_name,
+                                             bool color) {
+  scoped_ptr<DEVMODE[]> default = CreateDevMode(printer, NULL);
+  if (default && (default.get()->dmFields & DM_COLOR) &&
+      (default.get()->dmColor == DMCOLOR_COLOR) == color) {
+    return default.Pass();
+  }
+
+  default.get()->dmFields |= DM_COLOR;
+  default.get()->dmColor = color ? DMCOLOR_COLOR : DMCOLOR_MONOCHROME;
+
+  DriverInfo6 info_6;
+  if (!info_6.Init(printer))
+    return default.Pass();
+
+  const DRIVER_INFO_6* p = info_6.get();
+
+  // Only HP known to have issues.
+  if (wcscmp(p->pszMfgName, L"HP") != 0)
+    return default.Pass();
+
+  // Need XPS for this workaround.
+  printing::ScopedXPSInitializer xps_initializer;
+  if (!xps_initializer.initialized())
+    return default.Pass();
+
+  const char* xps_color = color ? kXpsTicketColor : kXpsTicketMonochrome;
+  std::string xps_ticket = base::StringPrintf(kXpsTicketTemplate, xps_color);
+  scoped_ptr<DEVMODE[]> ticket = printing::XpsTicketToDevMode(printer_name,
+                                                              xps_ticket);
+  if (!ticket)
+    return default.Pass();
+
+  return ticket.Pass();
+}
+
+PRINTING_EXPORT scoped_ptr<DEVMODE[]> CreateDevMode(HANDLE printer,
+                                                    DEVMODE* in) {
+  DWORD flags = in ? (DM_IN_BUFFER) : 0;
+  LONG buffer_size = DocumentProperties(NULL, printer, L"", NULL, in, flags);
+  if (buffer_size <= 0)
+    return scoped_ptr<DEVMODE[]>();
+  scoped_ptr<DEVMODE[]> out(
+      reinterpret_cast<DEVMODE*>(new uint8[buffer_size]));
+  flags |= DM_OUT_BUFFER;
+  if (DocumentProperties(NULL, printer, L"", out.get(), in, flags) != IDOK)
+    return scoped_ptr<DEVMODE[]>();
+  DCHECK_EQ(buffer_size, out.get()->dmSize + out.get()->dmDriverExtra);
+  return out.Pass();
+}
+
 }  // namespace printing
diff --git a/printing/backend/win_helper.h b/printing/backend/win_helper.h
index db66bf8..9b998d7 100644
--- a/printing/backend/win_helper.h
+++ b/printing/backend/win_helper.h
@@ -12,6 +12,7 @@
 
 #include <string>
 
+#include "base/memory/scoped_ptr.h"
 #include "base/strings/string16.h"
 #include "base/win/scoped_handle.h"
 #include "printing/printing_export.h"
@@ -169,6 +170,21 @@ PRINTING_EXPORT bool InitBasicPrinterInfo(HANDLE printer,
 
 PRINTING_EXPORT std::string GetDriverInfo(HANDLE printer);
 
+PRINTING_EXPORT scoped_ptr<DEVMODE[]> XpsTicketToDevMode(
+    const base::string16& printer_name,
+    const std::string& print_ticket);
+
+// Creates default DEVMODE and sets color option. Some devices need special
+// workaround for color.
+PRINTING_EXPORT scoped_ptr<DEVMODE[]> CreateDevModeWithColor(
+    HANDLE printer,
+    const base::string16& printer_name,
+    bool color);
+
+// Creates new DEVMODE. If |in| is not NULL copy settings from there.
+PRINTING_EXPORT scoped_ptr<DEVMODE[]> CreateDevMode(HANDLE printer,
+                                                    DEVMODE* in);
+
 }  // namespace printing
 
 #endif  // PRINTING_BACKEND_WIN_HELPER_H_