| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "printing/backend/print_backend.h" | 5 #include "printing/backend/print_backend.h" |
| 6 | 6 |
| 7 #include <algorithm> |
| 7 #include <objidl.h> | 8 #include <objidl.h> |
| 8 #include <winspool.h> | 9 #include <winspool.h> |
| 9 | 10 |
| 10 #include "base/file_path.h" | 11 #include "base/file_path.h" |
| 11 #include "base/file_version_info.h" | 12 #include "base/file_version_info.h" |
| 12 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
| 13 #include "base/string_piece.h" | 14 #include "base/string_piece.h" |
| 14 #include "base/utf_string_conversions.h" | 15 #include "base/utf_string_conversions.h" |
| 15 #include "base/win/scoped_bstr.h" | 16 #include "base/win/scoped_bstr.h" |
| 16 #include "base/win/scoped_comptr.h" | 17 #include "base/win/scoped_comptr.h" |
| (...skipping 26 matching lines...) Expand all Loading... |
| 43 virtual ~PrintBackendWin() {} | 44 virtual ~PrintBackendWin() {} |
| 44 | 45 |
| 45 virtual bool EnumeratePrinters(PrinterList* printer_list); | 46 virtual bool EnumeratePrinters(PrinterList* printer_list); |
| 46 | 47 |
| 47 virtual std::string GetDefaultPrinterName(); | 48 virtual std::string GetDefaultPrinterName(); |
| 48 | 49 |
| 49 virtual bool GetPrinterCapsAndDefaults(const std::string& printer_name, | 50 virtual bool GetPrinterCapsAndDefaults(const std::string& printer_name, |
| 50 PrinterCapsAndDefaults* printer_info); | 51 PrinterCapsAndDefaults* printer_info); |
| 51 | 52 |
| 52 virtual bool GetPrinterDriverInfo(const std::string& printer_name, | 53 virtual bool GetPrinterDriverInfo(const std::string& printer_name, |
| 53 PrinterDriverInfo* driver_info); | 54 std::string* driver_info); |
| 54 | 55 |
| 55 virtual bool IsValidPrinter(const std::string& printer_name); | 56 virtual bool IsValidPrinter(const std::string& printer_name); |
| 56 }; | 57 }; |
| 57 | 58 |
| 58 bool PrintBackendWin::EnumeratePrinters(PrinterList* printer_list) { | 59 bool PrintBackendWin::EnumeratePrinters(PrinterList* printer_list) { |
| 59 DCHECK(printer_list); | 60 DCHECK(printer_list); |
| 60 DWORD bytes_needed = 0; | 61 DWORD bytes_needed = 0; |
| 61 DWORD count_returned = 0; | 62 DWORD count_returned = 0; |
| 62 BOOL ret = EnumPrinters(PRINTER_ENUM_LOCAL|PRINTER_ENUM_CONNECTIONS, NULL, 2, | 63 BOOL ret = EnumPrinters(PRINTER_ENUM_LOCAL|PRINTER_ENUM_CONNECTIONS, NULL, 2, |
| 63 NULL, 0, &bytes_needed, &count_returned); | 64 NULL, 0, &bytes_needed, &count_returned); |
| (...skipping 107 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 171 } | 172 } |
| 172 } | 173 } |
| 173 } | 174 } |
| 174 XPSModule::CloseProvider(provider); | 175 XPSModule::CloseProvider(provider); |
| 175 } | 176 } |
| 176 return true; | 177 return true; |
| 177 } | 178 } |
| 178 | 179 |
| 179 // Gets the information about driver for a specific printer. | 180 // Gets the information about driver for a specific printer. |
| 180 bool PrintBackendWin::GetPrinterDriverInfo(const std::string& printer_name, | 181 bool PrintBackendWin::GetPrinterDriverInfo(const std::string& printer_name, |
| 181 PrinterDriverInfo* driver_info) { | 182 std::string* driver_info) { |
| 182 DCHECK(driver_info); | 183 DCHECK(driver_info); |
| 183 ScopedPrinterHandle printer_handle; | 184 ScopedPrinterHandle printer_handle; |
| 184 if (!::OpenPrinter(const_cast<LPTSTR>(UTF8ToWide(printer_name).c_str()), | 185 if (!::OpenPrinter(const_cast<LPTSTR>(UTF8ToWide(printer_name).c_str()), |
| 185 printer_handle.Receive(), NULL)) { | 186 printer_handle.Receive(), NULL)) { |
| 186 return false; | 187 return false; |
| 187 } | 188 } |
| 188 DCHECK(printer_handle.IsValid()); | 189 DCHECK(printer_handle.IsValid()); |
| 189 DWORD bytes_needed = 0; | 190 DWORD bytes_needed = 0; |
| 190 ::GetPrinterDriver(printer_handle, NULL, 6, NULL, 0, &bytes_needed); | 191 ::GetPrinterDriver(printer_handle, NULL, 6, NULL, 0, &bytes_needed); |
| 191 scoped_array<BYTE> driver_info_buffer(new BYTE[bytes_needed]); | 192 scoped_array<BYTE> driver_info_buffer(new BYTE[bytes_needed]); |
| 192 if (!bytes_needed || !driver_info_buffer.get()) | 193 if (!bytes_needed || !driver_info_buffer.get()) |
| 193 return false; | 194 return false; |
| 194 if (!::GetPrinterDriver(printer_handle, NULL, 6, driver_info_buffer.get(), | 195 if (!::GetPrinterDriver(printer_handle, NULL, 6, driver_info_buffer.get(), |
| 195 bytes_needed, &bytes_needed)) { | 196 bytes_needed, &bytes_needed)) { |
| 196 return false; | 197 return false; |
| 197 } | 198 } |
| 198 if (!bytes_needed) | 199 if (!bytes_needed) |
| 199 return false; | 200 return false; |
| 200 const DRIVER_INFO_6* driver_info_6 = | 201 const DRIVER_INFO_6* driver_info_6 = |
| 201 reinterpret_cast<DRIVER_INFO_6*>(driver_info_buffer.get()); | 202 reinterpret_cast<DRIVER_INFO_6*>(driver_info_buffer.get()); |
| 202 | 203 |
| 204 std::string info[4]; |
| 205 |
| 203 if (driver_info_6->pName) | 206 if (driver_info_6->pName) |
| 204 driver_info->driver_name = WideToUTF8(driver_info_6->pName); | 207 info[0] = WideToUTF8(driver_info_6->pName); |
| 205 | 208 |
| 206 if (driver_info_6->pDriverPath) { | 209 if (driver_info_6->pDriverPath) { |
| 207 scoped_ptr<FileVersionInfo> version_info( | 210 scoped_ptr<FileVersionInfo> version_info( |
| 208 FileVersionInfo::CreateFileVersionInfo( | 211 FileVersionInfo::CreateFileVersionInfo( |
| 209 FilePath(driver_info_6->pDriverPath))); | 212 FilePath(driver_info_6->pDriverPath))); |
| 210 driver_info->driver_version = WideToUTF8(version_info->file_version()); | 213 info[1] = WideToUTF8(version_info->file_version()); |
| 211 driver_info->product_name = WideToUTF8(version_info->product_name()); | 214 info[2] = WideToUTF8(version_info->product_name()); |
| 212 driver_info->product_version = WideToUTF8(version_info->product_version()); | 215 info[3] = WideToUTF8(version_info->product_version()); |
| 216 } |
| 217 |
| 218 for (size_t i = 0; i < arraysize(info); ++i) { |
| 219 std::replace(info[i].begin(), info[i].end(), ';', ','); |
| 220 driver_info->append(info[i]); |
| 221 driver_info->append(";"); |
| 213 } | 222 } |
| 214 | 223 |
| 215 return true; | 224 return true; |
| 216 } | 225 } |
| 217 | 226 |
| 218 bool PrintBackendWin::IsValidPrinter(const std::string& printer_name) { | 227 bool PrintBackendWin::IsValidPrinter(const std::string& printer_name) { |
| 219 ScopedPrinterHandle printer_handle; | 228 ScopedPrinterHandle printer_handle; |
| 220 OpenPrinter(const_cast<LPTSTR>(UTF8ToWide(printer_name).c_str()), | 229 OpenPrinter(const_cast<LPTSTR>(UTF8ToWide(printer_name).c_str()), |
| 221 printer_handle.Receive(), NULL); | 230 printer_handle.Receive(), NULL); |
| 222 return printer_handle.IsValid(); | 231 return printer_handle.IsValid(); |
| 223 } | 232 } |
| 224 | 233 |
| 225 scoped_refptr<PrintBackend> PrintBackend::CreateInstance( | 234 scoped_refptr<PrintBackend> PrintBackend::CreateInstance( |
| 226 const base::DictionaryValue* print_backend_settings) { | 235 const base::DictionaryValue* print_backend_settings) { |
| 227 return new PrintBackendWin; | 236 return new PrintBackendWin; |
| 228 } | 237 } |
| 229 | 238 |
| 230 } // namespace printing | 239 } // namespace printing |
| OLD | NEW |