| 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 |