| 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 "cloud_print/virtual_driver/win/port_monitor/port_monitor.h" | 5 #include "cloud_print/virtual_driver/win/port_monitor/port_monitor.h" |
| 6 | 6 |
| 7 #include <winspool.h> | 7 #include <winspool.h> |
| 8 | 8 |
| 9 #include "base/files/file_util.h" | 9 #include "base/files/file_util.h" |
| 10 #include "base/path_service.h" | 10 #include "base/path_service.h" |
| (...skipping 125 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 136 TEST_F(PortMonitorTest, EnumPortsTest) { | 136 TEST_F(PortMonitorTest, EnumPortsTest) { |
| 137 DWORD needed_bytes = 0; | 137 DWORD needed_bytes = 0; |
| 138 DWORD returned = 0; | 138 DWORD returned = 0; |
| 139 EXPECT_FALSE(Monitor2EnumPorts(NULL, | 139 EXPECT_FALSE(Monitor2EnumPorts(NULL, |
| 140 NULL, | 140 NULL, |
| 141 1, | 141 1, |
| 142 NULL, | 142 NULL, |
| 143 0, | 143 0, |
| 144 &needed_bytes, | 144 &needed_bytes, |
| 145 &returned)); | 145 &returned)); |
| 146 EXPECT_EQ(ERROR_INSUFFICIENT_BUFFER, GetLastError()); | 146 EXPECT_EQ(static_cast<DWORD>(ERROR_INSUFFICIENT_BUFFER), GetLastError()); |
| 147 EXPECT_NE(0u, needed_bytes); | 147 EXPECT_NE(0u, needed_bytes); |
| 148 EXPECT_EQ(0u, returned); | 148 EXPECT_EQ(0u, returned); |
| 149 | 149 |
| 150 BYTE* buffer = new BYTE[needed_bytes]; | 150 BYTE* buffer = new BYTE[needed_bytes]; |
| 151 ASSERT_TRUE(buffer != NULL); | 151 ASSERT_TRUE(buffer != NULL); |
| 152 EXPECT_TRUE(Monitor2EnumPorts(NULL, | 152 EXPECT_TRUE(Monitor2EnumPorts(NULL, |
| 153 NULL, | 153 NULL, |
| 154 1, | 154 1, |
| 155 buffer, | 155 buffer, |
| 156 needed_bytes, | 156 needed_bytes, |
| 157 &needed_bytes, | 157 &needed_bytes, |
| 158 &returned)); | 158 &returned)); |
| 159 EXPECT_NE(0u, needed_bytes); | 159 EXPECT_NE(0u, needed_bytes); |
| 160 EXPECT_EQ(1u, returned); | 160 EXPECT_EQ(1u, returned); |
| 161 PORT_INFO_1* port_info_1 = reinterpret_cast<PORT_INFO_1*>(buffer); | 161 PORT_INFO_1* port_info_1 = reinterpret_cast<PORT_INFO_1*>(buffer); |
| 162 EXPECT_TRUE(port_info_1->pName != NULL); | 162 EXPECT_TRUE(port_info_1->pName != NULL); |
| 163 delete[] buffer; | 163 delete[] buffer; |
| 164 | 164 |
| 165 returned = 0; | 165 returned = 0; |
| 166 needed_bytes = 0; | 166 needed_bytes = 0; |
| 167 EXPECT_FALSE(Monitor2EnumPorts(NULL, | 167 EXPECT_FALSE(Monitor2EnumPorts(NULL, |
| 168 NULL, | 168 NULL, |
| 169 2, | 169 2, |
| 170 NULL, | 170 NULL, |
| 171 0, | 171 0, |
| 172 &needed_bytes, | 172 &needed_bytes, |
| 173 &returned)); | 173 &returned)); |
| 174 EXPECT_EQ(ERROR_INSUFFICIENT_BUFFER, GetLastError()); | 174 EXPECT_EQ(static_cast<DWORD>(ERROR_INSUFFICIENT_BUFFER), GetLastError()); |
| 175 EXPECT_NE(0u, needed_bytes); | 175 EXPECT_NE(0u, needed_bytes); |
| 176 EXPECT_EQ(0u, returned); | 176 EXPECT_EQ(0u, returned); |
| 177 | 177 |
| 178 buffer = new BYTE[needed_bytes]; | 178 buffer = new BYTE[needed_bytes]; |
| 179 ASSERT_TRUE(buffer != NULL); | 179 ASSERT_TRUE(buffer != NULL); |
| 180 EXPECT_TRUE(Monitor2EnumPorts(NULL, | 180 EXPECT_TRUE(Monitor2EnumPorts(NULL, |
| 181 NULL, | 181 NULL, |
| 182 2, | 182 2, |
| 183 buffer, | 183 buffer, |
| 184 needed_bytes, | 184 needed_bytes, |
| (...skipping 21 matching lines...) Expand all Loading... |
| 206 MONITOR2* monitor2 = InitializePrintMonitor2(&monitor_init, &monitor_handle); | 206 MONITOR2* monitor2 = InitializePrintMonitor2(&monitor_init, &monitor_handle); |
| 207 EXPECT_TRUE(monitor2 != NULL); | 207 EXPECT_TRUE(monitor2 != NULL); |
| 208 EXPECT_TRUE(monitor_handle != NULL); | 208 EXPECT_TRUE(monitor_handle != NULL); |
| 209 | 209 |
| 210 // Test the XCV functions. Used for reporting the location of the | 210 // Test the XCV functions. Used for reporting the location of the |
| 211 // UI portion of the port monitor. | 211 // UI portion of the port monitor. |
| 212 EXPECT_TRUE(monitor2->pfnXcvOpenPort != NULL); | 212 EXPECT_TRUE(monitor2->pfnXcvOpenPort != NULL); |
| 213 EXPECT_TRUE(monitor2->pfnXcvOpenPort(monitor_handle, NULL, 0, &xcv_handle)); | 213 EXPECT_TRUE(monitor2->pfnXcvOpenPort(monitor_handle, NULL, 0, &xcv_handle)); |
| 214 EXPECT_TRUE(xcv_handle != NULL); | 214 EXPECT_TRUE(xcv_handle != NULL); |
| 215 EXPECT_TRUE(monitor2->pfnXcvDataPort != NULL); | 215 EXPECT_TRUE(monitor2->pfnXcvDataPort != NULL); |
| 216 EXPECT_EQ(ERROR_ACCESS_DENIED, | 216 EXPECT_EQ(static_cast<DWORD>(ERROR_ACCESS_DENIED), |
| 217 monitor2->pfnXcvDataPort(xcv_handle, | 217 monitor2->pfnXcvDataPort(xcv_handle, kXcvDataItem, NULL, 0, buffer, |
| 218 kXcvDataItem, | 218 kBufferSize, &bytes_needed)); |
| 219 NULL, | |
| 220 0, | |
| 221 buffer, | |
| 222 kBufferSize, | |
| 223 &bytes_needed)); | |
| 224 EXPECT_TRUE(monitor2->pfnXcvClosePort != NULL); | 219 EXPECT_TRUE(monitor2->pfnXcvClosePort != NULL); |
| 225 EXPECT_TRUE(monitor2->pfnXcvClosePort(xcv_handle)); | 220 EXPECT_TRUE(monitor2->pfnXcvClosePort(xcv_handle)); |
| 226 EXPECT_TRUE(monitor2->pfnXcvOpenPort(monitor_handle, | 221 EXPECT_TRUE(monitor2->pfnXcvOpenPort(monitor_handle, |
| 227 NULL, | 222 NULL, |
| 228 SERVER_ACCESS_ADMINISTER, | 223 SERVER_ACCESS_ADMINISTER, |
| 229 &xcv_handle)); | 224 &xcv_handle)); |
| 230 EXPECT_TRUE(xcv_handle != NULL); | 225 EXPECT_TRUE(xcv_handle != NULL); |
| 231 EXPECT_TRUE(monitor2->pfnXcvDataPort != NULL); | 226 EXPECT_TRUE(monitor2->pfnXcvDataPort != NULL); |
| 232 EXPECT_EQ(ERROR_SUCCESS, | 227 EXPECT_EQ(static_cast<DWORD>(ERROR_SUCCESS), |
| 233 monitor2->pfnXcvDataPort(xcv_handle, | 228 monitor2->pfnXcvDataPort(xcv_handle, kXcvDataItem, NULL, 0, buffer, |
| 234 kXcvDataItem, | 229 kBufferSize, &bytes_needed)); |
| 235 NULL, | |
| 236 0, | |
| 237 buffer, | |
| 238 kBufferSize, | |
| 239 &bytes_needed)); | |
| 240 EXPECT_TRUE(monitor2->pfnXcvClosePort != NULL); | 230 EXPECT_TRUE(monitor2->pfnXcvClosePort != NULL); |
| 241 EXPECT_TRUE(monitor2->pfnXcvClosePort(xcv_handle)); | 231 EXPECT_TRUE(monitor2->pfnXcvClosePort(xcv_handle)); |
| 242 | 232 |
| 243 // Test opening the port and running a print job. | 233 // Test opening the port and running a print job. |
| 244 EXPECT_TRUE(monitor2->pfnOpenPort != NULL); | 234 EXPECT_TRUE(monitor2->pfnOpenPort != NULL); |
| 245 EXPECT_TRUE(monitor2->pfnOpenPort(monitor_handle, NULL, &port_handle)); | 235 EXPECT_TRUE(monitor2->pfnOpenPort(monitor_handle, NULL, &port_handle)); |
| 246 EXPECT_TRUE(port_handle != NULL); | 236 EXPECT_TRUE(port_handle != NULL); |
| 247 EXPECT_TRUE(monitor2->pfnStartDocPort != NULL); | 237 EXPECT_TRUE(monitor2->pfnStartDocPort != NULL); |
| 248 EXPECT_TRUE(monitor2->pfnWritePort != NULL); | 238 EXPECT_TRUE(monitor2->pfnWritePort != NULL); |
| 249 EXPECT_TRUE(monitor2->pfnReadPort != NULL); | 239 EXPECT_TRUE(monitor2->pfnReadPort != NULL); |
| 250 EXPECT_TRUE(monitor2->pfnEndDocPort != NULL); | 240 EXPECT_TRUE(monitor2->pfnEndDocPort != NULL); |
| 251 | 241 |
| 252 // These functions should fail if we have not impersonated the user. | 242 // These functions should fail if we have not impersonated the user. |
| 253 EXPECT_FALSE(monitor2->pfnStartDocPort( | 243 EXPECT_FALSE(monitor2->pfnStartDocPort( |
| 254 port_handle, const_cast<wchar_t*>(L""), 0, 0, NULL)); | 244 port_handle, const_cast<wchar_t*>(L""), 0, 0, NULL)); |
| 255 EXPECT_FALSE(monitor2->pfnWritePort(port_handle, | 245 EXPECT_FALSE(monitor2->pfnWritePort(port_handle, |
| 256 buffer, | 246 buffer, |
| 257 kBufferSize, | 247 kBufferSize, |
| 258 &bytes_processed)); | 248 &bytes_processed)); |
| 259 EXPECT_EQ(0, bytes_processed); | 249 EXPECT_EQ(0u, bytes_processed); |
| 260 EXPECT_FALSE(monitor2->pfnReadPort(port_handle, | 250 EXPECT_FALSE(monitor2->pfnReadPort(port_handle, |
| 261 buffer, | 251 buffer, |
| 262 sizeof(buffer), | 252 sizeof(buffer), |
| 263 &bytes_processed)); | 253 &bytes_processed)); |
| 264 EXPECT_EQ(0u, bytes_processed); | 254 EXPECT_EQ(0u, bytes_processed); |
| 265 EXPECT_FALSE(monitor2->pfnEndDocPort(port_handle)); | 255 EXPECT_FALSE(monitor2->pfnEndDocPort(port_handle)); |
| 266 | 256 |
| 267 // Now impersonate so we can test the success case. | 257 // Now impersonate so we can test the success case. |
| 268 ASSERT_TRUE(ImpersonateSelf(SecurityImpersonation)); | 258 ASSERT_TRUE(ImpersonateSelf(SecurityImpersonation)); |
| 269 EXPECT_TRUE(monitor2->pfnStartDocPort( | 259 EXPECT_TRUE(monitor2->pfnStartDocPort( |
| (...skipping 11 matching lines...) Expand all Loading... |
| 281 EXPECT_TRUE(monitor2->pfnEndDocPort(port_handle)); | 271 EXPECT_TRUE(monitor2->pfnEndDocPort(port_handle)); |
| 282 RevertToSelf(); | 272 RevertToSelf(); |
| 283 EXPECT_TRUE(monitor2->pfnClosePort != NULL); | 273 EXPECT_TRUE(monitor2->pfnClosePort != NULL); |
| 284 EXPECT_TRUE(monitor2->pfnClosePort(port_handle)); | 274 EXPECT_TRUE(monitor2->pfnClosePort(port_handle)); |
| 285 // Shutdown the port monitor. | 275 // Shutdown the port monitor. |
| 286 Monitor2Shutdown(monitor_handle); | 276 Monitor2Shutdown(monitor_handle); |
| 287 } | 277 } |
| 288 | 278 |
| 289 } // namespace cloud_print | 279 } // namespace cloud_print |
| 290 | 280 |
| OLD | NEW |