1 // Copyright (c) 2012 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 #ifndef PRINTING_BACKEND_PRINTING_INFO_WIN_H_ 6 #define PRINTING_BACKEND_PRINTING_INFO_WIN_H_ 7 8 #include <objidl.h> 9 #include <winspool.h> 10 11 #include "base/memory/scoped_ptr.h" 12 #include "printing/printing_export.h" 13 14 namespace printing { 15 16 namespace internal { 17 18 PRINTING_EXPORT uint8* GetDriverInfo(HANDLE printer, int level); 19 PRINTING_EXPORT uint8* GetPrinterInfo(HANDLE printer, int level); 20 21 // This class is designed to work with PRINTER_INFO_X structures 22 // and calls GetPrinter internally with correctly allocated buffer. 23 template <typename PrinterInfoType, int level> 24 class PrinterInfo { 25 public: Init(HANDLE printer)26 bool Init(HANDLE printer) { 27 buffer_.reset(GetPrinterInfo(printer, level)); 28 return buffer_; 29 } 30 get()31 const PrinterInfoType* get() const { 32 return reinterpret_cast<const PrinterInfoType*>(buffer_.get()); 33 } 34 35 private: 36 scoped_ptr<uint8[]> buffer_; 37 }; 38 39 // This class is designed to work with DRIVER_INFO_X structures 40 // and calls GetDriverInfo internally with correctly allocated buffer. 41 template <typename DriverInfoType, int level> 42 class DriverInfo { 43 public: Init(HANDLE printer)44 bool Init(HANDLE printer) { 45 buffer_.reset(GetDriverInfo(printer, level)); 46 return buffer_; 47 } 48 get()49 const DriverInfoType* get() const { 50 return reinterpret_cast<const DriverInfoType*>(buffer_.get()); 51 } 52 53 private: 54 scoped_ptr<uint8[]> buffer_; 55 }; 56 57 } // namespace internal 58 59 typedef internal::PrinterInfo<PRINTER_INFO_2, 2> PrinterInfo2; 60 typedef internal::PrinterInfo<PRINTER_INFO_5, 5> PrinterInfo5; 61 typedef internal::PrinterInfo<PRINTER_INFO_8, 8> PrinterInfo8; 62 typedef internal::PrinterInfo<PRINTER_INFO_9, 9> PrinterInfo9; 63 64 typedef internal::DriverInfo<DRIVER_INFO_6, 6> DriverInfo6; 65 66 // Retrieves DEVMODE from PRINTER_INFO_* structures. 67 // Requests in following order: 68 // 9 (user-default), 69 // 8 (admin-default), 70 // 2 (printer-default). 71 class PRINTING_EXPORT UserDefaultDevMode { 72 public: 73 UserDefaultDevMode(); 74 ~UserDefaultDevMode(); 75 76 bool Init(HANDLE printer); 77 get()78 const DEVMODE* get() const { 79 return dev_mode_; 80 } 81 82 private: 83 PrinterInfo2 info_2_; 84 PrinterInfo8 info_8_; 85 PrinterInfo9 info_9_; 86 const DEVMODE* dev_mode_; 87 }; 88 89 } // namespace printing 90 91 #endif // PRINTING_BACKEND_PRINTING_INFO_WIN_H_ 92