OLD | NEW |
1 // Copyright (c) 2014 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2014 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 "device/hid/hid_connection_win.h" | 5 #include "device/hid/hid_connection_win.h" |
6 | 6 |
7 #include <cstring> | 7 #include <cstring> |
8 | 8 |
9 #include "base/files/file.h" | 9 #include "base/files/file.h" |
10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
| 11 #include "base/stl_util.h" |
| 12 #include "base/threading/thread_restrictions.h" |
11 #include "base/win/object_watcher.h" | 13 #include "base/win/object_watcher.h" |
| 14 #include "base/win/scoped_handle.h" |
| 15 #include "device/hid/hid_service.h" |
| 16 #include "device/hid/hid_service_win.h" |
12 | 17 |
13 #define INITGUID | 18 #define INITGUID |
14 | 19 |
15 #include <windows.h> | 20 #include <windows.h> |
16 #include <hidclass.h> | 21 #include <hidclass.h> |
17 | 22 |
18 extern "C" { | 23 extern "C" { |
19 #include <hidsdi.h> | 24 #include <hidsdi.h> |
20 } | 25 } |
21 | 26 |
(...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
91 Release(); | 96 Release(); |
92 } | 97 } |
93 | 98 |
94 void PendingHidTransfer::WillDestroyCurrentMessageLoop() { | 99 void PendingHidTransfer::WillDestroyCurrentMessageLoop() { |
95 watcher_.StopWatching(); | 100 watcher_.StopWatching(); |
96 connection_->OnTransferCanceled(this); | 101 connection_->OnTransferCanceled(this); |
97 } | 102 } |
98 | 103 |
99 HidConnectionWin::HidConnectionWin(const HidDeviceInfo& device_info) | 104 HidConnectionWin::HidConnectionWin(const HidDeviceInfo& device_info) |
100 : HidConnection(device_info) { | 105 : HidConnection(device_info) { |
| 106 DCHECK(thread_checker_.CalledOnValidThread()); |
101 file_.Set(CreateFileA(device_info.device_id.c_str(), | 107 file_.Set(CreateFileA(device_info.device_id.c_str(), |
102 GENERIC_WRITE | GENERIC_READ, | 108 GENERIC_WRITE | GENERIC_READ, |
103 FILE_SHARE_READ | FILE_SHARE_WRITE, | 109 FILE_SHARE_READ | FILE_SHARE_WRITE, |
104 NULL, | 110 NULL, |
105 OPEN_EXISTING, | 111 OPEN_EXISTING, |
106 FILE_FLAG_OVERLAPPED, | 112 FILE_FLAG_OVERLAPPED, |
107 NULL)); | 113 NULL)); |
108 | 114 |
109 if (!file_.IsValid() && | 115 if (!file_.IsValid() && |
110 GetLastError() == base::File::FILE_ERROR_ACCESS_DENIED) { | 116 GetLastError() == base::File::FILE_ERROR_ACCESS_DENIED) { |
111 file_.Set(CreateFileA(device_info.device_id.c_str(), | 117 file_.Set(CreateFileA(device_info.device_id.c_str(), |
112 GENERIC_READ, | 118 GENERIC_READ, |
113 FILE_SHARE_READ, | 119 FILE_SHARE_READ, |
114 NULL, | 120 NULL, |
115 OPEN_EXISTING, | 121 OPEN_EXISTING, |
116 FILE_FLAG_OVERLAPPED, | 122 FILE_FLAG_OVERLAPPED, |
117 NULL)); | 123 NULL)); |
118 } | 124 } |
119 } | 125 } |
120 | 126 |
| 127 bool HidConnectionWin::available() const { |
| 128 return file_.IsValid(); |
| 129 } |
| 130 |
121 HidConnectionWin::~HidConnectionWin() { | 131 HidConnectionWin::~HidConnectionWin() { |
| 132 DCHECK(thread_checker_.CalledOnValidThread()); |
122 CancelIo(file_.Get()); | 133 CancelIo(file_.Get()); |
123 } | 134 } |
124 | 135 |
125 void HidConnectionWin::PlatformRead(scoped_refptr<net::IOBufferWithSize> buffer, | 136 void HidConnectionWin::Read(scoped_refptr<net::IOBufferWithSize> buffer, |
126 const HidConnection::IOCallback& callback) { | 137 const HidConnection::IOCallback& callback) { |
127 scoped_refptr<net::IOBufferWithSize> receive_buffer = | 138 DCHECK(thread_checker_.CalledOnValidThread()); |
128 new net::IOBufferWithSize(device_info().max_input_report_size); | 139 if (device_info().input_report_size == 0) { |
| 140 // The device does not support input reports. |
| 141 callback.Run(false, 0); |
| 142 return; |
| 143 } |
129 | 144 |
| 145 // This fairly awkward logic is correct: If Windows does not expect a device |
| 146 // to supply a report ID in its input reports, it requires the buffer to be |
| 147 // 1 byte larger than what the device actually sends. |
| 148 int receive_buffer_size = device_info().input_report_size; |
| 149 int expected_buffer_size = receive_buffer_size; |
| 150 if (!device_info().has_report_id) |
| 151 expected_buffer_size -= 1; |
| 152 |
| 153 if (buffer->size() < expected_buffer_size) { |
| 154 callback.Run(false, 0); |
| 155 return; |
| 156 } |
| 157 |
| 158 scoped_refptr<net::IOBufferWithSize> receive_buffer(buffer); |
| 159 if (receive_buffer_size != expected_buffer_size) |
| 160 receive_buffer = new net::IOBufferWithSize(receive_buffer_size); |
130 scoped_refptr<PendingHidTransfer> transfer( | 161 scoped_refptr<PendingHidTransfer> transfer( |
131 new PendingHidTransfer(this, buffer, receive_buffer, callback)); | 162 new PendingHidTransfer(this, buffer, receive_buffer, callback)); |
132 transfers_.insert(transfer); | 163 transfers_.insert(transfer); |
133 transfer->TakeResultFromWindowsAPI( | 164 transfer->TakeResultFromWindowsAPI( |
134 ReadFile(file_.Get(), | 165 ReadFile(file_.Get(), |
135 receive_buffer->data(), | 166 receive_buffer->data(), |
136 static_cast<DWORD>(receive_buffer->size()), | 167 static_cast<DWORD>(receive_buffer->size()), |
137 NULL, | 168 NULL, |
138 transfer->GetOverlapped())); | 169 transfer->GetOverlapped())); |
139 } | 170 } |
140 | 171 |
141 void HidConnectionWin::PlatformWrite( | 172 void HidConnectionWin::Write(uint8_t report_id, |
142 uint8_t report_id, | 173 scoped_refptr<net::IOBufferWithSize> buffer, |
143 scoped_refptr<net::IOBufferWithSize> buffer, | 174 const HidConnection::IOCallback& callback) { |
144 const HidConnection::IOCallback& callback) { | 175 DCHECK(thread_checker_.CalledOnValidThread()); |
| 176 if (device_info().output_report_size == 0) { |
| 177 // The device does not support output reports. |
| 178 callback.Run(false, 0); |
| 179 return; |
| 180 } |
| 181 |
145 // The Windows API always wants either a report ID (if supported) or | 182 // The Windows API always wants either a report ID (if supported) or |
146 // zero at the front of every output report. | 183 // zero at the front of every output report. |
147 scoped_refptr<net::IOBufferWithSize> output_buffer(buffer); | 184 scoped_refptr<net::IOBufferWithSize> output_buffer(buffer); |
148 output_buffer = new net::IOBufferWithSize(buffer->size() + 1); | 185 output_buffer = new net::IOBufferWithSize(buffer->size() + 1); |
149 output_buffer->data()[0] = report_id; | 186 output_buffer->data()[0] = report_id; |
150 memcpy(output_buffer->data() + 1, buffer->data(), buffer->size()); | 187 memcpy(output_buffer->data() + 1, buffer->data(), buffer->size()); |
151 | 188 |
152 scoped_refptr<PendingHidTransfer> transfer( | 189 scoped_refptr<PendingHidTransfer> transfer( |
153 new PendingHidTransfer(this, output_buffer, NULL, callback)); | 190 new PendingHidTransfer(this, output_buffer, NULL, callback)); |
154 transfers_.insert(transfer); | 191 transfers_.insert(transfer); |
155 transfer->TakeResultFromWindowsAPI( | 192 transfer->TakeResultFromWindowsAPI( |
156 WriteFile(file_.Get(), | 193 WriteFile(file_.Get(), |
157 output_buffer->data(), | 194 output_buffer->data(), |
158 static_cast<DWORD>(output_buffer->size()), | 195 static_cast<DWORD>(output_buffer->size()), |
159 NULL, | 196 NULL, |
160 transfer->GetOverlapped())); | 197 transfer->GetOverlapped())); |
161 } | 198 } |
162 | 199 |
163 void HidConnectionWin::PlatformGetFeatureReport( | 200 void HidConnectionWin::GetFeatureReport( |
164 uint8_t report_id, | 201 uint8_t report_id, |
165 scoped_refptr<net::IOBufferWithSize> buffer, | 202 scoped_refptr<net::IOBufferWithSize> buffer, |
166 const IOCallback& callback) { | 203 const IOCallback& callback) { |
167 scoped_refptr<net::IOBufferWithSize> receive_buffer = | 204 DCHECK(thread_checker_.CalledOnValidThread()); |
168 new net::IOBufferWithSize(device_info().max_feature_report_size); | 205 if (device_info().feature_report_size == 0) { |
| 206 // The device does not support feature reports. |
| 207 callback.Run(false, 0); |
| 208 return; |
| 209 } |
| 210 |
| 211 int receive_buffer_size = device_info().feature_report_size; |
| 212 int expected_buffer_size = receive_buffer_size; |
| 213 if (!device_info().has_report_id) |
| 214 expected_buffer_size -= 1; |
| 215 if (buffer->size() < expected_buffer_size) { |
| 216 callback.Run(false, 0); |
| 217 return; |
| 218 } |
| 219 |
| 220 scoped_refptr<net::IOBufferWithSize> receive_buffer(buffer); |
| 221 if (receive_buffer_size != expected_buffer_size) |
| 222 receive_buffer = new net::IOBufferWithSize(receive_buffer_size); |
| 223 |
169 // The first byte of the destination buffer is the report ID being requested. | 224 // The first byte of the destination buffer is the report ID being requested. |
170 receive_buffer->data()[0] = report_id; | 225 receive_buffer->data()[0] = report_id; |
171 | |
172 scoped_refptr<PendingHidTransfer> transfer( | 226 scoped_refptr<PendingHidTransfer> transfer( |
173 new PendingHidTransfer(this, buffer, receive_buffer, callback)); | 227 new PendingHidTransfer(this, buffer, receive_buffer, callback)); |
174 transfers_.insert(transfer); | 228 transfers_.insert(transfer); |
175 transfer->TakeResultFromWindowsAPI( | 229 transfer->TakeResultFromWindowsAPI( |
176 DeviceIoControl(file_.Get(), | 230 DeviceIoControl(file_.Get(), |
177 IOCTL_HID_GET_FEATURE, | 231 IOCTL_HID_GET_FEATURE, |
178 NULL, | 232 NULL, |
179 0, | 233 0, |
180 receive_buffer->data(), | 234 receive_buffer->data(), |
181 static_cast<DWORD>(receive_buffer->size()), | 235 static_cast<DWORD>(receive_buffer->size()), |
182 NULL, | 236 NULL, |
183 transfer->GetOverlapped())); | 237 transfer->GetOverlapped())); |
184 } | 238 } |
185 | 239 |
186 void HidConnectionWin::PlatformSendFeatureReport( | 240 void HidConnectionWin::SendFeatureReport( |
187 uint8_t report_id, | 241 uint8_t report_id, |
188 scoped_refptr<net::IOBufferWithSize> buffer, | 242 scoped_refptr<net::IOBufferWithSize> buffer, |
189 const IOCallback& callback) { | 243 const IOCallback& callback) { |
| 244 DCHECK(thread_checker_.CalledOnValidThread()); |
| 245 if (device_info().feature_report_size == 0) { |
| 246 // The device does not support feature reports. |
| 247 callback.Run(false, 0); |
| 248 return; |
| 249 } |
| 250 |
190 // The Windows API always wants either a report ID (if supported) or | 251 // The Windows API always wants either a report ID (if supported) or |
191 // zero at the front of every output report. | 252 // zero at the front of every output report. |
192 scoped_refptr<net::IOBufferWithSize> output_buffer(buffer); | 253 scoped_refptr<net::IOBufferWithSize> output_buffer(buffer); |
193 output_buffer = new net::IOBufferWithSize(buffer->size() + 1); | 254 output_buffer = new net::IOBufferWithSize(buffer->size() + 1); |
194 output_buffer->data()[0] = report_id; | 255 output_buffer->data()[0] = report_id; |
195 memcpy(output_buffer->data() + 1, buffer->data(), buffer->size()); | 256 memcpy(output_buffer->data() + 1, buffer->data(), buffer->size()); |
196 | 257 |
197 scoped_refptr<PendingHidTransfer> transfer( | 258 scoped_refptr<PendingHidTransfer> transfer( |
198 new PendingHidTransfer(this, output_buffer, NULL, callback)); | 259 new PendingHidTransfer(this, output_buffer, NULL, callback)); |
199 transfer->TakeResultFromWindowsAPI( | 260 transfer->TakeResultFromWindowsAPI( |
200 DeviceIoControl(file_.Get(), | 261 DeviceIoControl(file_.Get(), |
201 IOCTL_HID_SET_FEATURE, | 262 IOCTL_HID_SET_FEATURE, |
202 output_buffer->data(), | 263 output_buffer->data(), |
203 static_cast<DWORD>(output_buffer->size()), | 264 static_cast<DWORD>(output_buffer->size()), |
204 NULL, | 265 NULL, |
205 0, | 266 0, |
206 NULL, | 267 NULL, |
207 transfer->GetOverlapped())); | 268 transfer->GetOverlapped())); |
208 } | 269 } |
209 | 270 |
210 void HidConnectionWin::OnTransferFinished( | 271 void HidConnectionWin::OnTransferFinished( |
211 scoped_refptr<PendingHidTransfer> transfer) { | 272 scoped_refptr<PendingHidTransfer> transfer) { |
| 273 DWORD bytes_transferred; |
212 transfers_.erase(transfer); | 274 transfers_.erase(transfer); |
213 | |
214 DWORD bytes_transferred; | |
215 if (GetOverlappedResult( | 275 if (GetOverlappedResult( |
216 file_, transfer->GetOverlapped(), &bytes_transferred, FALSE)) { | 276 file_, transfer->GetOverlapped(), &bytes_transferred, FALSE)) { |
217 if (bytes_transferred == 0) { | 277 if (bytes_transferred == 0) |
218 transfer->callback_.Run(true, 0); | 278 transfer->callback_.Run(true, 0); |
219 return; | 279 // If this is an input transfer and the receive buffer is not the same as |
| 280 // the target buffer, we need to copy the receive buffer into the target |
| 281 // buffer, discarding the first byte. This is because the target buffer's |
| 282 // owner is not expecting a report ID but Windows will always provide one. |
| 283 if (transfer->receive_buffer_ && |
| 284 transfer->receive_buffer_ != transfer->target_buffer_) { |
| 285 // Move one byte forward. |
| 286 --bytes_transferred; |
| 287 memcpy(transfer->target_buffer_->data(), |
| 288 transfer->receive_buffer_->data() + 1, |
| 289 bytes_transferred); |
220 } | 290 } |
221 | 291 transfer->callback_.Run(true, bytes_transferred); |
222 if (transfer->receive_buffer_) { | |
223 // If owner HID top-level collection does not have report ID, we need to | |
224 // copy the receive buffer into the target buffer, discarding the first | |
225 // byte. This is because the target buffer's owner is not expecting a | |
226 // report ID but Windows will always provide one. | |
227 if (!has_report_id()) { | |
228 uint8_t report_id = transfer->receive_buffer_->data()[0]; | |
229 // Assert first byte is 0x00 | |
230 if (report_id != HidConnection::kNullReportId) { | |
231 VLOG(1) << "Unexpected report ID in HID report:" << report_id; | |
232 transfer->callback_.Run(false, 0); | |
233 } else { | |
234 // Move one byte forward. | |
235 --bytes_transferred; | |
236 memcpy(transfer->target_buffer_->data(), | |
237 transfer->receive_buffer_->data() + 1, | |
238 bytes_transferred); | |
239 } | |
240 } else { | |
241 memcpy(transfer->target_buffer_->data(), | |
242 transfer->receive_buffer_->data(), | |
243 bytes_transferred); | |
244 } | |
245 } | |
246 | |
247 CompleteRead(transfer->target_buffer_, transfer->callback_); | |
248 } else { | 292 } else { |
249 transfer->callback_.Run(false, 0); | 293 transfer->callback_.Run(false, 0); |
250 } | 294 } |
251 } | 295 } |
252 | 296 |
253 void HidConnectionWin::OnTransferCanceled( | 297 void HidConnectionWin::OnTransferCanceled( |
254 scoped_refptr<PendingHidTransfer> transfer) { | 298 scoped_refptr<PendingHidTransfer> transfer) { |
255 transfers_.erase(transfer); | 299 transfers_.erase(transfer); |
256 transfer->callback_.Run(false, 0); | 300 transfer->callback_.Run(false, 0); |
257 } | 301 } |
258 | 302 |
259 } // namespace device | 303 } // namespace device |
OLD | NEW |