| /* |
| * Copyright (C) 2017 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| #include <glog/logging.h> |
| |
| #include "host/vadb/usb_cmd_control_transfer.h" |
| |
| namespace vadb { |
| USBCmdControlTransfer::USBCmdControlTransfer( |
| uint8_t bus_id, uint8_t dev_id, uint8_t type, uint8_t request, |
| uint16_t value, uint16_t index, uint32_t timeout, std::vector<uint8_t> data, |
| usbip::Device::AsyncTransferReadyCB callback) |
| : data_(std::move(data)), callback_(std::move(callback)) { |
| req_.bus_id = bus_id; |
| req_.dev_id = dev_id; |
| req_.type = type; |
| req_.cmd = request; |
| req_.value = value; |
| req_.index = index; |
| req_.length = data_.size(); |
| req_.timeout = timeout; |
| } |
| |
| bool USBCmdControlTransfer::OnRequest(const avd::SharedFD& fd) { |
| if (fd->Write(&req_, sizeof(req_)) != sizeof(req_)) { |
| LOG(ERROR) << "Short write: " << fd->StrError(); |
| return false; |
| } |
| |
| if ((req_.type & 0x80) == 0) { |
| if (data_.size() > 0) { |
| if (fd->Write(data_.data(), data_.size()) != data_.size()) { |
| LOG(ERROR) << "Short write: " << fd->StrError(); |
| return false; |
| } |
| } |
| } |
| |
| return true; |
| } |
| |
| bool USBCmdControlTransfer::OnResponse(bool is_success, |
| const avd::SharedFD& fd) { |
| if (!is_success) { |
| callback_(false, std::move(data_)); |
| return true; |
| } |
| |
| if (req_.type & 0x80) { |
| int32_t len; |
| if (fd->Read(&len, sizeof(len)) != sizeof(len)) { |
| LOG(ERROR) << "Short read: " << fd->StrError(); |
| callback_(false, std::move(data_)); |
| return false; |
| } |
| |
| if (len > 0) { |
| data_.resize(len); |
| if (fd->Read(data_.data(), len) != len) { |
| LOG(ERROR) << "Short read: " << fd->StrError(); |
| callback_(false, std::move(data_)); |
| return false; |
| } |
| } |
| } |
| |
| callback_(true, std::move(data_)); |
| return true; |
| } |
| } // namespace vadb |