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 |