OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "cloud_print/virtual_driver/win/virtual_driver_helpers.h" | 5 #include "cloud_print/virtual_driver/win/virtual_driver_helpers.h" |
6 #include <windows.h> | 6 #include <windows.h> |
7 #include <winspool.h> | 7 #include <winspool.h> |
8 #include "base/file_util.h" | 8 #include "base/file_util.h" |
9 #include "cloud_print/virtual_driver/win/virtual_driver_consts.h" | 9 #include "base/logging.h" |
| 10 #include "base/string16.h" |
| 11 #include "base/win/windows_version.h" |
10 | 12 |
11 namespace cloud_print { | 13 namespace cloud_print { |
12 | 14 |
13 const size_t kMaxMessageLen = 100; | 15 const size_t kMaxMessageLen = 100; |
14 | 16 |
15 void DisplayWindowsMessage(HWND hwnd, HRESULT message_id) { | 17 void DisplayWindowsMessage(HWND hwnd, |
| 18 HRESULT message_id, |
| 19 const string16 &caption) { |
16 wchar_t message_text[kMaxMessageLen + 1] = L""; | 20 wchar_t message_text[kMaxMessageLen + 1] = L""; |
17 | 21 |
18 ::FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS, | 22 ::FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS, |
19 NULL, | 23 NULL, |
20 message_id, | 24 message_id, |
21 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), | 25 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), |
22 message_text, | 26 message_text, |
23 kMaxMessageLen, | 27 kMaxMessageLen, |
24 NULL); | 28 NULL); |
25 ::MessageBox(hwnd, message_text, kVirtualDriverName, MB_OK); | 29 ::MessageBox(hwnd, message_text, caption.c_str(), MB_OK); |
26 } | 30 } |
27 | 31 |
28 HRESULT GetLastHResult() { | 32 HRESULT GetLastHResult() { |
29 DWORD error_code = GetLastError(); | 33 DWORD error_code = GetLastError(); |
30 return HRESULT_FROM_WIN32(error_code); | 34 return HRESULT_FROM_WIN32(error_code); |
31 } | 35 } |
32 | 36 |
| 37 string16 GetPortMonitorDllName() { |
| 38 if (IsSystem64Bit()) { |
| 39 return string16(L"gcp_portmon64.dll"); |
| 40 } else { |
| 41 return string16(L"gcp_portmon32.dll"); |
| 42 } |
| 43 } |
| 44 |
33 HRESULT GetPrinterDriverDir(FilePath* path) { | 45 HRESULT GetPrinterDriverDir(FilePath* path) { |
34 BYTE driver_dir_buffer[MAX_PATH * sizeof(wchar_t)]; | 46 BYTE driver_dir_buffer[MAX_PATH * sizeof(wchar_t)]; |
35 DWORD needed = 0; | 47 DWORD needed = 0; |
36 if (!GetPrinterDriverDirectory(NULL, | 48 if (!GetPrinterDriverDirectory(NULL, |
37 NULL, | 49 NULL, |
38 1, | 50 1, |
39 driver_dir_buffer, | 51 driver_dir_buffer, |
40 MAX_PATH * sizeof(wchar_t), | 52 MAX_PATH * sizeof(wchar_t), |
41 &needed)) { | 53 &needed)) { |
42 // We could try to allocate a larger buffer if needed > MAX_PATH | 54 // We could try to allocate a larger buffer if needed > MAX_PATH |
43 // but that really shouldn't happen. | 55 // but that really shouldn't happen. |
44 return cloud_print::GetLastHResult(); | 56 return cloud_print::GetLastHResult(); |
45 } | 57 } |
46 *path = FilePath(reinterpret_cast<wchar_t*>(driver_dir_buffer)); | 58 *path = FilePath(reinterpret_cast<wchar_t*>(driver_dir_buffer)); |
47 | 59 |
48 // The XPS driver is a "Level 3" driver | 60 // The XPS driver is a "Level 3" driver |
49 *path = path->Append(L"3"); | 61 *path = path->Append(L"3"); |
50 return S_OK; | 62 return S_OK; |
51 } | 63 } |
| 64 |
| 65 bool IsSystem64Bit() { |
| 66 base::win::OSInfo::WindowsArchitecture arch = |
| 67 base::win::OSInfo::GetInstance()->architecture(); |
| 68 return (arch == base::win::OSInfo::X64_ARCHITECTURE) || |
| 69 (arch == base::win::OSInfo::IA64_ARCHITECTURE); |
52 } | 70 } |
53 | 71 |
| 72 string16 LoadLocalString(DWORD string_id) { |
| 73 static wchar_t dummy = L'\0'; |
| 74 // We never expect strings longer than MAX_PATH characters. |
| 75 static wchar_t buffer[MAX_PATH]; |
| 76 HMODULE module = NULL; |
| 77 GetModuleHandleEx(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS | |
| 78 GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT, |
| 79 &dummy, |
| 80 &module); |
| 81 int count = LoadString(module, |
| 82 string_id, |
| 83 buffer, |
| 84 MAX_PATH); |
| 85 CHECK_NE(0, count); |
| 86 return string16(buffer); |
| 87 } |
| 88 } |
| 89 |
OLD | NEW |