1/*
2 * Copyright (C) 2017 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include "guest/commands/usbforward/transport_request.h"
18
19#include <cutils/log.h>
20
21namespace usb_forward {
22
23TransportRequest::TransportRequest(std::shared_ptr<libusb_device_handle> handle,
24                                   CallbackType callback,
25                                   const ControlTransfer& transfer)
26    : handle_{std::move(handle)},
27      callback_{std::move(callback)},
28      is_control_{true},
29      transfer_{libusb_alloc_transfer(0), libusb_free_transfer} {
30  // NOTE: libusb places setup structure as part of user data!
31  buffer_.reset(new uint8_t[transfer.length + LIBUSB_CONTROL_SETUP_SIZE]);
32
33  // NOTE: libusb places a structure of size LIBUSB_CONTROL_SETUP_SIZE directly
34  // in the data buffer.
35  libusb_fill_control_setup(buffer_.get(), transfer.type, transfer.cmd,
36                            transfer.value, transfer.index, transfer.length);
37
38  // NOTE: despite libusb requires user to allocate buffer large enough to
39  // accommodate SETUP structure and actual data, it requires user to provide
40  // only data length here, while setup length is added internally.
41  libusb_fill_control_transfer(transfer_.get(), handle_.get(), buffer_.get(),
42                               OnTransferComplete, this, transfer.timeout);
43}
44
45TransportRequest::TransportRequest(std::shared_ptr<libusb_device_handle> handle,
46                                   CallbackType callback,
47                                   const DataTransfer& transfer)
48    : handle_{std::move(handle)},
49      callback_{std::move(callback)},
50      is_control_{false},
51      transfer_{libusb_alloc_transfer(0), libusb_free_transfer} {
52  buffer_.reset(new uint8_t[transfer.length]);
53  libusb_fill_bulk_transfer(
54      transfer_.get(), handle_.get(),
55      transfer.endpoint_id | (transfer.is_host_to_device ? LIBUSB_ENDPOINT_OUT
56                                                         : LIBUSB_ENDPOINT_IN),
57      buffer_.get(), transfer.length, OnTransferComplete, this,
58      transfer.timeout);
59}
60
61uint8_t* TransportRequest::Buffer() {
62  if (is_control_) {
63    return &buffer_[LIBUSB_CONTROL_SETUP_SIZE];
64  } else {
65    return buffer_.get();
66  }
67}
68
69bool TransportRequest::Submit() {
70  if (handle_) {
71    auto err = libusb_submit_transfer(transfer_.get());
72    if (err != 0) {
73      ALOGE("libusb transfer failed: %d", err);
74    }
75    return err == 0;
76  } else {
77    ALOGE("Initiated transfer, but device not opened.");
78    return false;
79  }
80}
81
82void TransportRequest::OnTransferComplete(libusb_transfer* req) {
83  auto treq = static_cast<TransportRequest*>(req->user_data);
84  treq->callback_(req->status == 0, treq->Buffer(), req->actual_length);
85}
86
87}  // namespace usb_forward
88