host/commands/virtual_usb_manager/vadb/usb_cmd_control_transfer.cpp

/*
 * 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/commands/virtual_usb_manager/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 cvd::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 (static_cast<size_t>(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 cvd::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