blob: 7f64cc6d3e3537de68d28d138e9f8acd0ed2bd9b [file] [log] [blame]
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "device/devices_app/usb/device_impl.h"
#include "base/bind.h"
#include "base/callback.h"
#include "base/stl_util.h"
#include "device/devices_app/usb/type_converters.h"
#include "device/usb/usb_descriptors.h"
#include "device/usb/usb_device.h"
#include "net/base/io_buffer.h"
namespace device {
namespace usb {
namespace {
template <typename... Args>
void CallMojoCallback(const mojo::Callback<void(Args...)>& callback,
Args... args) {
callback.Run(args...);
}
// Generic wrapper to convert a Mojo callback to something we can rebind and
// pass around. This is only usable for callbacks with no move-only arguments.
template <typename... Args>
base::Callback<void(Args...)> WrapMojoCallback(
const mojo::Callback<void(Args...)>& callback) {
return base::Bind(&CallMojoCallback<Args...>, callback);
}
void OnPermissionCheckComplete(
const base::Callback<void(bool)>& callback,
const base::Callback<void(const base::Callback<void(bool)>&)>& action,
bool allowed) {
if (allowed)
action.Run(callback);
else
callback.Run(false);
}
scoped_refptr<net::IOBuffer> CreateTransferBuffer(size_t size) {
scoped_refptr<net::IOBuffer> buffer = new net::IOBuffer(
std::max(static_cast<size_t>(1u), static_cast<size_t>(size)));
return buffer;
}
void OnTransferIn(const DeviceImpl::MojoTransferInCallback& callback,
UsbTransferStatus status,
scoped_refptr<net::IOBuffer> buffer,
size_t buffer_size) {
mojo::Array<uint8_t> data;
if (buffer) {
// TODO(rockot/reillyg): We should change UsbDeviceHandle to use a
// std::vector<uint8_t> instead of net::IOBuffer. Then we could move
// instead of copy.
std::vector<uint8_t> bytes(buffer_size);
std::copy(buffer->data(), buffer->data() + buffer_size, bytes.begin());
data.Swap(&bytes);
}
callback.Run(mojo::ConvertTo<TransferStatus>(status), data.Pass());
}
void OnControlTransferInPermissionCheckComplete(
scoped_refptr<UsbDeviceHandle> device_handle,
ControlTransferParamsPtr params,
int length,
int timeout,
const Device::ControlTransferInCallback& callback,
bool allowed) {
if (allowed) {
scoped_refptr<net::IOBuffer> buffer = CreateTransferBuffer(length);
device_handle->ControlTransfer(
USB_DIRECTION_INBOUND,
mojo::ConvertTo<UsbDeviceHandle::TransferRequestType>(params->type),
mojo::ConvertTo<UsbDeviceHandle::TransferRecipient>(params->recipient),
params->request, params->value, params->index, buffer, length, timeout,
base::Bind(&OnTransferIn, callback));
} else {
mojo::Array<uint8_t> data;
callback.Run(TRANSFER_STATUS_PERMISSION_DENIED, data.Pass());
}
}
void OnTransferOut(const DeviceImpl::MojoTransferOutCallback& callback,
UsbTransferStatus status,
scoped_refptr<net::IOBuffer> buffer,
size_t buffer_size) {
callback.Run(mojo::ConvertTo<TransferStatus>(status));
}
void OnControlTransferOutPermissionCheckComplete(
scoped_refptr<UsbDeviceHandle> device_handle,
ControlTransferParamsPtr params,
mojo::Array<uint8_t> data,
int timeout,
const Device::ControlTransferOutCallback& callback,
bool allowed) {
if (allowed) {
scoped_refptr<net::IOBuffer> buffer = CreateTransferBuffer(data.size());
const std::vector<uint8_t>& storage = data.storage();
std::copy(storage.begin(), storage.end(), buffer->data());
device_handle->ControlTransfer(
USB_DIRECTION_OUTBOUND,
mojo::ConvertTo<UsbDeviceHandle::TransferRequestType>(params->type),
mojo::ConvertTo<UsbDeviceHandle::TransferRecipient>(params->recipient),
params->request, params->value, params->index, buffer, data.size(),
timeout, base::Bind(&OnTransferOut, callback));
} else {
callback.Run(TRANSFER_STATUS_PERMISSION_DENIED);
}
}
void OnIsochronousTransferIn(
const Device::IsochronousTransferInCallback& callback,
uint32_t packet_size,
UsbTransferStatus status,
scoped_refptr<net::IOBuffer> buffer,
size_t buffer_size) {
size_t num_packets = buffer_size / packet_size;
mojo::Array<mojo::Array<uint8_t>> packets(num_packets);
if (buffer) {
for (size_t i = 0; i < num_packets; ++i) {
size_t packet_index = i * packet_size;
std::vector<uint8_t> bytes(packet_size);
std::copy(buffer->data() + packet_index,
buffer->data() + packet_index + packet_size, bytes.begin());
packets[i].Swap(&bytes);
}
}
callback.Run(mojo::ConvertTo<TransferStatus>(status), packets.Pass());
}
void OnIsochronousTransferOut(
const Device::IsochronousTransferOutCallback& callback,
UsbTransferStatus status,
scoped_refptr<net::IOBuffer> buffer,
size_t buffer_size) {
callback.Run(mojo::ConvertTo<TransferStatus>(status));
}
} // namespace
DeviceImpl::DeviceImpl(scoped_refptr<UsbDevice> device,
PermissionProviderPtr permission_provider,
mojo::InterfaceRequest<Device> request)
: binding_(this, request.Pass()),
device_(device),
permission_provider_(permission_provider.Pass()),
weak_factory_(this) {
// This object owns itself and will be destroyed if either the message pipe
// it is bound to is closed or the PermissionProvider it depends on is
// unavailable.
binding_.set_connection_error_handler([this]() { delete this; });
permission_provider_.set_connection_error_handler([this]() { delete this; });
}
DeviceImpl::~DeviceImpl() {
CloseHandle();
}
void DeviceImpl::CloseHandle() {
if (device_handle_)
device_handle_->Close();
device_handle_ = nullptr;
}
void DeviceImpl::HasControlTransferPermission(
ControlTransferRecipient recipient,
uint16_t index,
const base::Callback<void(bool)>& callback) {
DCHECK(device_handle_);
const UsbConfigDescriptor* config = device_->GetActiveConfiguration();
if (recipient == CONTROL_TRANSFER_RECIPIENT_INTERFACE ||
recipient == CONTROL_TRANSFER_RECIPIENT_ENDPOINT) {
if (!config) {
callback.Run(false);
return;
}
uint8_t interface_number = index & 0xff;
if (recipient == CONTROL_TRANSFER_RECIPIENT_ENDPOINT) {
if (!device_handle_->FindInterfaceByEndpoint(index & 0xff,
&interface_number)) {
callback.Run(false);
return;
}
}
permission_provider_->HasInterfacePermission(
interface_number, config->configuration_value,
DeviceInfo::From(*device_), callback);
} else if (config) {
permission_provider_->HasConfigurationPermission(
config->configuration_value, DeviceInfo::From(*device_), callback);
} else {
// Client must already have device permission to have gotten this far.
callback.Run(true);
}
}
void DeviceImpl::OnOpen(const OpenCallback& callback,
scoped_refptr<UsbDeviceHandle> handle) {
device_handle_ = handle;
callback.Run(handle ? OPEN_DEVICE_ERROR_OK : OPEN_DEVICE_ERROR_ACCESS_DENIED);
}
void DeviceImpl::GetDeviceInfo(const GetDeviceInfoCallback& callback) {
callback.Run(DeviceInfo::From(*device_));
}
void DeviceImpl::GetConfiguration(const GetConfigurationCallback& callback) {
const UsbConfigDescriptor* config = device_->GetActiveConfiguration();
callback.Run(config ? config->configuration_value : 0);
}
void DeviceImpl::Open(const OpenCallback& callback) {
device_->Open(
base::Bind(&DeviceImpl::OnOpen, weak_factory_.GetWeakPtr(), callback));
}
void DeviceImpl::Close(const CloseCallback& callback) {
CloseHandle();
callback.Run();
}
void DeviceImpl::SetConfiguration(uint8_t value,
const SetConfigurationCallback& callback) {
if (!device_handle_) {
callback.Run(false);
return;
}
auto set_configuration =
base::Bind(&UsbDeviceHandle::SetConfiguration, device_handle_, value);
permission_provider_->HasConfigurationPermission(
value, DeviceInfo::From(*device_),
base::Bind(&OnPermissionCheckComplete, WrapMojoCallback(callback),
set_configuration));
}
void DeviceImpl::ClaimInterface(uint8_t interface_number,
const ClaimInterfaceCallback& callback) {
if (!device_handle_) {
callback.Run(false);
return;
}
const UsbConfigDescriptor* config = device_->GetActiveConfiguration();
if (!config) {
callback.Run(false);
return;
}
auto claim_interface = base::Bind(&UsbDeviceHandle::ClaimInterface,
device_handle_, interface_number);
permission_provider_->HasInterfacePermission(
interface_number, config->configuration_value, DeviceInfo::From(*device_),
base::Bind(&OnPermissionCheckComplete, WrapMojoCallback(callback),
claim_interface));
}
void DeviceImpl::ReleaseInterface(uint8_t interface_number,
const ReleaseInterfaceCallback& callback) {
if (!device_handle_) {
callback.Run(false);
return;
}
callback.Run(device_handle_->ReleaseInterface(interface_number));
}
void DeviceImpl::SetInterfaceAlternateSetting(
uint8_t interface_number,
uint8_t alternate_setting,
const SetInterfaceAlternateSettingCallback& callback) {
if (!device_handle_) {
callback.Run(false);
return;
}
device_handle_->SetInterfaceAlternateSetting(
interface_number, alternate_setting, WrapMojoCallback(callback));
}
void DeviceImpl::Reset(const ResetCallback& callback) {
if (!device_handle_) {
callback.Run(false);
return;
}
device_handle_->ResetDevice(WrapMojoCallback(callback));
}
void DeviceImpl::ClearHalt(uint8_t endpoint,
const ClearHaltCallback& callback) {
if (!device_handle_) {
callback.Run(false);
return;
}
device_handle_->ClearHalt(endpoint, WrapMojoCallback(callback));
}
void DeviceImpl::ControlTransferIn(ControlTransferParamsPtr params,
uint32_t length,
uint32_t timeout,
const ControlTransferInCallback& callback) {
if (!device_handle_) {
callback.Run(TRANSFER_STATUS_ERROR, mojo::Array<uint8_t>());
return;
}
ControlTransferRecipient recipient = params->recipient;
uint16_t index = params->index;
HasControlTransferPermission(
recipient, index,
base::Bind(&OnControlTransferInPermissionCheckComplete, device_handle_,
base::Passed(&params), length, timeout, callback));
}
void DeviceImpl::ControlTransferOut(
ControlTransferParamsPtr params,
mojo::Array<uint8_t> data,
uint32_t timeout,
const ControlTransferOutCallback& callback) {
if (!device_handle_) {
callback.Run(TRANSFER_STATUS_ERROR);
return;
}
ControlTransferRecipient recipient = params->recipient;
uint16_t index = params->index;
HasControlTransferPermission(
recipient, index, base::Bind(&OnControlTransferOutPermissionCheckComplete,
device_handle_, base::Passed(&params),
base::Passed(&data), timeout, callback));
}
void DeviceImpl::GenericTransferIn(uint8_t endpoint_number,
uint32_t length,
uint32_t timeout,
const GenericTransferInCallback& callback) {
if (!device_handle_) {
callback.Run(TRANSFER_STATUS_ERROR, mojo::Array<uint8_t>());
return;
}
uint8_t endpoint_address = endpoint_number | 0x80;
scoped_refptr<net::IOBuffer> buffer = CreateTransferBuffer(length);
device_handle_->GenericTransfer(USB_DIRECTION_INBOUND, endpoint_address,
buffer, length, timeout,
base::Bind(&OnTransferIn, callback));
}
void DeviceImpl::GenericTransferOut(
uint8_t endpoint_number,
mojo::Array<uint8_t> data,
uint32_t timeout,
const GenericTransferOutCallback& callback) {
if (!device_handle_) {
callback.Run(TRANSFER_STATUS_ERROR);
return;
}
uint8_t endpoint_address = endpoint_number;
scoped_refptr<net::IOBuffer> buffer = CreateTransferBuffer(data.size());
const std::vector<uint8_t>& storage = data.storage();
std::copy(storage.begin(), storage.end(), buffer->data());
device_handle_->GenericTransfer(USB_DIRECTION_OUTBOUND, endpoint_address,
buffer, data.size(), timeout,
base::Bind(&OnTransferOut, callback));
}
void DeviceImpl::IsochronousTransferIn(
uint8_t endpoint_number,
uint32_t num_packets,
uint32_t packet_length,
uint32_t timeout,
const IsochronousTransferInCallback& callback) {
if (!device_handle_) {
callback.Run(TRANSFER_STATUS_ERROR, mojo::Array<mojo::Array<uint8_t>>());
return;
}
uint8_t endpoint_address = endpoint_number | 0x80;
size_t transfer_size = static_cast<size_t>(num_packets) * packet_length;
scoped_refptr<net::IOBuffer> buffer = CreateTransferBuffer(transfer_size);
device_handle_->IsochronousTransfer(
USB_DIRECTION_INBOUND, endpoint_address, buffer, transfer_size,
num_packets, packet_length, timeout,
base::Bind(&OnIsochronousTransferIn, callback, packet_length));
}
void DeviceImpl::IsochronousTransferOut(
uint8_t endpoint_number,
mojo::Array<mojo::Array<uint8_t>> packets,
uint32_t timeout,
const IsochronousTransferOutCallback& callback) {
if (!device_handle_) {
callback.Run(TRANSFER_STATUS_ERROR);
return;
}
uint8_t endpoint_address = endpoint_number;
uint32_t packet_size = 0;
for (size_t i = 0; i < packets.size(); ++i) {
packet_size =
std::max(packet_size, static_cast<uint32_t>(packets[i].size()));
}
size_t transfer_size = packet_size * packets.size();
scoped_refptr<net::IOBuffer> buffer = CreateTransferBuffer(transfer_size);
memset(buffer->data(), 0, transfer_size);
for (size_t i = 0; i < packets.size(); ++i) {
uint8_t* packet =
reinterpret_cast<uint8_t*>(&buffer->data()[i * packet_size]);
DCHECK_LE(packets[i].size(), static_cast<size_t>(packet_size));
memcpy(packet, packets[i].storage().data(), packets[i].size());
}
device_handle_->IsochronousTransfer(
USB_DIRECTION_OUTBOUND, endpoint_address, buffer, transfer_size,
static_cast<uint32_t>(packets.size()), packet_size, timeout,
base::Bind(&OnIsochronousTransferOut, callback));
}
} // namespace usb
} // namespace device