Wrapers for PRINTER_INFO_* and DRIVER_INFO_*.

printing/backend/print_backend_win.cc

 // Copyright (c) 2012 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 "printing/backend/print_backend.h"
 
 #include <objidl.h>
 #include <winspool.h>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/string_piece.h"
 #include "base/utf_string_conversions.h"
 #include "base/win/scoped_bstr.h"
 #include "base/win/scoped_comptr.h"
 #include "base/win/scoped_hglobal.h"
 #include "printing/backend/print_backend_consts.h"
+#include "printing/backend/printing_info_win.h"
 #include "printing/backend/win_helper.h"
 
+
 namespace {
 
-// This class is designed to work with PRINTER_INFO_X structures
-// and calls GetPrinter internally with correctly allocated buffer.
-template <typename T>
-class PrinterInfo {
- public:
-  bool GetPrinterInfo(HANDLE printer, int level) {
-    DWORD buf_size = 0;
-    GetPrinter(printer, level, NULL, 0, &buf_size);
-    if (buf_size == 0) {
-      LOG(WARNING) << "Failed to GetPrinter, error = " << GetLastError();
-      return false;
-    }
-    buffer_.reset(new uint8[buf_size]);
-    memset(buffer_.get(), 0, buf_size);
-    return !!GetPrinter(printer, level, buffer_.get(), buf_size, &buf_size);
-  }
-
-  const T* get() const {
-    return reinterpret_cast<T*>(buffer_.get());
-  }
-
- private:
-  scoped_array<uint8> buffer_;
-};
-
 HRESULT StreamOnHGlobalToString(IStream* stream, std::string* out) {
   DCHECK(stream);
   DCHECK(out);
   HGLOBAL hdata = NULL;
   HRESULT hr = GetHGlobalFromStream(stream, &hdata);
   if (SUCCEEDED(hr)) {
     DCHECK(hdata);
     base::win::ScopedHGlobal<char> locked_data(hdata);
     out->assign(locked_data.release(), locked_data.Size());
   }
@@ skipping 70 matching lines @@
     PrinterSemanticCapsAndDefaults* printer_info) {
   ScopedPrinterHandle printer_handle;
   OpenPrinter(const_cast<LPTSTR>(UTF8ToWide(printer_name).c_str()),
               printer_handle.Receive(), NULL);
   DCHECK(printer_handle);
   if (!printer_handle.IsValid()) {
     LOG(WARNING) << "Failed to open printer, error = " << GetLastError();
     return false;
   }
 
-  PrinterInfo<PRINTER_INFO_5> info_5;
-  if (!info_5.GetPrinterInfo(printer_handle, 5)) {
-    LOG(WARNING) << "Failed to get PRINTER_INFO_5, error = " << GetLastError();
+  PrinterInfo5 info_5;
+  if (!info_5.Init(printer_handle)) {
     return false;
   }
+  DCHECK_EQ(info_5.get()->pPrinterName, UTF8ToUTF16(printer_name));
+
+  PrinterSemanticCapsAndDefaults caps;
 
   // Get printer capabilities. For more info see here:
   // http://msdn.microsoft.com/en-us/library/windows/desktop/dd183552(v=vs.85).aspx
-  bool color_supported = (DeviceCapabilities(info_5.get()->pPrinterName,
+  caps.color_capable = (::DeviceCapabilities(info_5.get()->pPrinterName,
                                              info_5.get()->pPortName,
                                              DC_COLORDEVICE,
                                              NULL,
                                              NULL) == 1);
 
-  bool duplex_supported = (DeviceCapabilities(info_5.get()->pPrinterName,
+  caps.duplex_capable = (::DeviceCapabilities(info_5.get()->pPrinterName,
                                               info_5.get()->pPortName,
                                               DC_DUPLEX,
                                               NULL,
                                               NULL) == 1);
 
-  DEVMODE* devmode = NULL;
-  // PRINTER_INFO_9 retrieves current user settings.
-  PrinterInfo<PRINTER_INFO_9> info_9;
-  if (info_9.GetPrinterInfo(printer_handle, 9)) {
-    devmode = info_9.get()->pDevMode;
-  } else {
-    LOG(WARNING) << "Failed to get PRINTER_INFO_9, error = " << GetLastError();
-  }
+  UserDefaultDevMode user_settings;
 
-  // Sometimes user settings are not available (have not been setted up yet).
-  // Use printer default settings (PRINTER_INFO_8) in this case.
-  PrinterInfo<PRINTER_INFO_8> info_8;
-  if (!devmode) {
-    if (info_8.GetPrinterInfo(printer_handle, 8)) {
-      devmode = info_8.get()->pDevMode;
-    } else {
-      LOG(WARNING) << "Failed to get PRINTER_INFO_8, error = " <<
-                      GetLastError();
-    }
-  }
-  if (!devmode)
-    return false;
+  if (user_settings.Init(printer_handle)) {
+    if ((user_settings.get()->dmFields & DM_COLOR) == DM_COLOR)
+      caps.color_default = (user_settings.get()->dmColor == DMCOLOR_COLOR);
 
-  PrinterSemanticCapsAndDefaults caps;
-  caps.color_capable = color_supported;
-  if ((devmode->dmFields & DM_COLOR) == DM_COLOR)
-    caps.color_default = (devmode->dmColor == DMCOLOR_COLOR);
-
-  caps.duplex_capable = duplex_supported;
-  if ((devmode->dmFields & DM_DUPLEX) == DM_DUPLEX) {
-    switch (devmode->dmDuplex) {
+    if ((user_settings.get()->dmFields & DM_DUPLEX) == DM_DUPLEX) {
+      switch (user_settings.get()->dmDuplex) {
       case DMDUP_SIMPLEX:
         caps.duplex_default = SIMPLEX;
         break;
       case DMDUP_VERTICAL:
         caps.duplex_default = LONG_EDGE;
         break;
       case DMDUP_HORIZONTAL:
         caps.duplex_default = SHORT_EDGE;
         break;
       default:
         NOTREACHED();
+      }
     }
+  } else {
+    LOG(WARNING) << "Fallback to color/simplex mode.";
+    caps.color_default = caps.color_capable;
+    caps.duplex_default = SIMPLEX;
   }
 
   *printer_info = caps;
   return true;
 }
 
 bool PrintBackendWin::GetPrinterCapsAndDefaults(
     const std::string& printer_name,
     PrinterCapsAndDefaults* printer_info) {
   ScopedXPSInitializer xps_initializer;
@@ skipping 85 matching lines @@
               printer_handle.Receive(), NULL);
   return printer_handle.IsValid();
 }
 
 scoped_refptr<PrintBackend> PrintBackend::CreateInstance(
     const base::DictionaryValue* print_backend_settings) {
   return new PrintBackendWin;
 }
 
 }  // namespace printing