/*
* Copyright (C) 2010 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.
*/
#define LOG_TAG "MtpDevice"
#include "MtpDebug.h"
#include "MtpDevice.h"
#include "MtpDeviceInfo.h"
#include "MtpEventPacket.h"
#include "MtpObjectInfo.h"
#include "MtpProperty.h"
#include "MtpStorageInfo.h"
#include "MtpStringBuffer.h"
#include "MtpUtils.h"
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <endian.h>
#include <usbhost/usbhost.h>
namespace {
static constexpr int USB_CONTROL_TRANSFER_TIMEOUT_MS = 200;
} // namespace
#if 0
static bool isMtpDevice(uint16_t vendor, uint16_t product) {
// Sandisk Sansa Fuze
if (vendor == 0x0781 && product == 0x74c2)
return true;
// Samsung YP-Z5
if (vendor == 0x04e8 && product == 0x503c)
return true;
return false;
}
#endif
namespace {
bool writeToFd(void* data, uint32_t /* unused_offset */, uint32_t length, void* clientData) {
const int fd = *static_cast<int*>(clientData);
const ssize_t result = write(fd, data, length);
if (result < 0) {
return false;
}
return static_cast<uint32_t>(result) == length;
}
} // namespace
MtpDevice* MtpDevice::open(const char* deviceName, int fd) {
struct usb_device *device = usb_device_new(deviceName, fd);
if (!device) {
MTPE("usb_device_new failed for %s", deviceName);
return NULL;
}
struct usb_descriptor_header* desc;
struct usb_descriptor_iter iter;
usb_descriptor_iter_init(device, &iter);
while ((desc = usb_descriptor_iter_next(&iter)) != NULL) {
if (desc->bDescriptorType == USB_DT_INTERFACE) {
struct usb_interface_descriptor *interface = (struct usb_interface_descriptor *)desc;
if (interface->bInterfaceClass == USB_CLASS_STILL_IMAGE &&
interface->bInterfaceSubClass == 1 && // Still Image Capture
interface->bInterfaceProtocol == 1) // Picture Transfer Protocol (PIMA 15470)
{
char* manufacturerName = usb_device_get_manufacturer_name(device,
USB_CONTROL_TRANSFER_TIMEOUT_MS);
char* productName = usb_device_get_product_name(device,
USB_CONTROL_TRANSFER_TIMEOUT_MS);
MTPD("Found camera: \"%s\" \"%s\"\n", manufacturerName, productName);
free(manufacturerName);
free(productName);
} else if (interface->bInterfaceClass == 0xFF &&
interface->bInterfaceSubClass == 0xFF &&
interface->bInterfaceProtocol == 0) {
char* interfaceName = usb_device_get_string(device, interface->iInterface,
USB_CONTROL_TRANSFER_TIMEOUT_MS);
if (!interfaceName) {
continue;
} else if (strcmp(interfaceName, "MTP")) {
free(interfaceName);
continue;
}
free(interfaceName);
// Looks like an android style MTP device
char* manufacturerName = usb_device_get_manufacturer_name(device,
USB_CONTROL_TRANSFER_TIMEOUT_MS);
char* productName = usb_device_get_product_name(device,
USB_CONTROL_TRANSFER_TIMEOUT_MS);
MTPD("Found MTP device: \"%s\" \"%s\"\n", manufacturerName, productName);
free(manufacturerName);
free(productName);
}
#if 0
else {
// look for special cased devices based on vendor/product ID
// we are doing this mainly for testing purposes
uint16_t vendor = usb_device_get_vendor_id(device);
uint16_t product = usb_device_get_product_id(device);
if (!isMtpDevice(vendor, product)) {
// not an MTP or PTP device
continue;
}
// request MTP OS string and descriptor
// some music players need to see this before entering MTP mode.
char buffer[256];
memset(buffer, 0, sizeof(buffer));
int ret = usb_device_control_transfer(device,
USB_DIR_IN|USB_RECIP_DEVICE|USB_TYPE_STANDARD,
USB_REQ_GET_DESCRIPTOR, (USB_DT_STRING << 8) | 0xEE,
0, buffer, sizeof(buffer), 0);
printf("usb_device_control_transfer returned %d errno: %d\n", ret, errno);
if (ret > 0) {
printf("got MTP string %s\n", buffer);
ret = usb_device_control_transfer(device,
USB_DIR_IN|USB_RECIP_DEVICE|USB_TYPE_VENDOR, 1,
0, 4, buffer, sizeof(buffer), 0);
printf("OS descriptor got %d\n", ret);
} else {
printf("no MTP string\n");
}
}
#else
else {
continue;
}
#endif
// if we got here, then we have a likely MTP or PTP device
// interface should be followed by three endpoints
struct usb_endpoint_descriptor *ep;
struct usb_endpoint_descriptor *ep_in_desc = NULL;
struct usb_endpoint_descriptor *ep_out_desc = NULL;
struct usb_endpoint_descriptor *ep_intr_desc = NULL;
//USB3 add USB_DT_SS_ENDPOINT_COMP as companion descriptor;
struct usb_ss_ep_comp_descriptor *ep_ss_ep_comp_desc = NULL;
for (int i = 0; i < 3; i++) {
ep = (struct usb_endpoint_descriptor *)usb_descriptor_iter_next(&iter);
if (ep && ep->bDescriptorType == USB_DT_SS_ENDPOINT_COMP) {
MTPD("Descriptor type is USB_DT_SS_ENDPOINT_COMP for USB3 \n");
ep_ss_ep_comp_desc = (usb_ss_ep_comp_descriptor*)ep;
ep = (struct usb_endpoint_descriptor *)usb_descriptor_iter_next(&iter);
}
if (!ep || ep->bDescriptorType != USB_DT_ENDPOINT) {
MTPE("endpoints not found\n");
usb_device_close(device);
return NULL;
}
if (ep->bmAttributes == USB_ENDPOINT_XFER_BULK) {
if (ep->bEndpointAddress & USB_ENDPOINT_DIR_MASK)
ep_in_desc = ep;
else
ep_out_desc = ep;
} else if (ep->bmAttributes == USB_ENDPOINT_XFER_INT &&
ep->bEndpointAddress & USB_ENDPOINT_DIR_MASK) {
ep_intr_desc = ep;
}
}
if (!ep_in_desc || !ep_out_desc || !ep_intr_desc) {
MTPE("endpoints not found\n");
usb_device_close(device);
return NULL;
}
int ret = usb_device_claim_interface(device, interface->bInterfaceNumber);
if (ret && errno == EBUSY) {
// disconnect kernel driver and try again
usb_device_connect_kernel_driver(device, interface->bInterfaceNumber, false);
ret = usb_device_claim_interface(device, interface->bInterfaceNumber);
}
if (ret) {
MTPE("usb_device_claim_interface failed errno: %d\n", errno);
usb_device_close(device);
return NULL;
}
MtpDevice* mtpDevice = new MtpDevice(device, interface->bInterfaceNumber,
ep_in_desc, ep_out_desc, ep_intr_desc);
mtpDevice->initialize();
return mtpDevice;
}
}
usb_device_close(device);
MTPE("device not found");
return NULL;
}
MtpDevice::MtpDevice(struct usb_device* device, int interface,
const struct usb_endpoint_descriptor *ep_in,
const struct usb_endpoint_descriptor *ep_out,
const struct usb_endpoint_descriptor *ep_intr)
: mDevice(device),
mInterface(interface),
mRequestIn1(NULL),
mRequestIn2(NULL),
mRequestOut(NULL),
mRequestIntr(NULL),
mDeviceInfo(NULL),
mSessionID(0),
mTransactionID(0),
mReceivedResponse(false),
mProcessingEvent(false),
mCurrentEventHandle(0),
mLastSendObjectInfoTransactionID(0),
mLastSendObjectInfoObjectHandle(0),
mPacketDivisionMode(FIRST_PACKET_HAS_PAYLOAD)
{
mRequestIn1 = usb_request_new(device, ep_in);
mRequestIn2 = usb_request_new(device, ep_in);
mRequestOut = usb_request_new(device, ep_out);
mRequestIntr = usb_request_new(device, ep_intr);
}
MtpDevice::~MtpDevice() {
close();
for (size_t i = 0; i < mDeviceProperties.size(); i++)
delete mDeviceProperties[i];
usb_request_free(mRequestIn1);
usb_request_free(mRequestIn2);
usb_request_free(mRequestOut);
usb_request_free(mRequestIntr);
}
void MtpDevice::initialize() {
openSession();
mDeviceInfo = getDeviceInfo();
if (mDeviceInfo) {
if (mDeviceInfo->mDeviceProperties) {
int count = mDeviceInfo->mDeviceProperties->size();
for (int i = 0; i < count; i++) {
MtpDeviceProperty propCode = (*mDeviceInfo->mDeviceProperties)[i];
MtpProperty* property = getDevicePropDesc(propCode);
if (property)
mDeviceProperties.push_back(property);
}
}
}
}
void MtpDevice::close() {
if (mDevice) {
usb_device_release_interface(mDevice, mInterface);
usb_device_close(mDevice);
mDevice = NULL;
}
}
void MtpDevice::print() {
if (!mDeviceInfo)
return;
mDeviceInfo->print();
if (mDeviceInfo->mDeviceProperties) {
MTPD("***** DEVICE PROPERTIES *****\n");
int count = mDeviceInfo->mDeviceProperties->size();
for (int i = 0; i < count; i++) {
MtpDeviceProperty propCode = (*mDeviceInfo->mDeviceProperties)[i];
MtpProperty* property = getDevicePropDesc(propCode);
if (property) {
property->print();
delete property;
}
}
}
if (mDeviceInfo->mPlaybackFormats) {
MTPD("***** OBJECT PROPERTIES *****\n");
int count = mDeviceInfo->mPlaybackFormats->size();
for (int i = 0; i < count; i++) {
MtpObjectFormat format = (*mDeviceInfo->mPlaybackFormats)[i];
MTPD("*** FORMAT: %s\n", MtpDebug::getFormatCodeName(format));
MtpObjectPropertyList* props = getObjectPropsSupported(format);
if (props) {
for (size_t j = 0; j < props->size(); j++) {
MtpObjectProperty prop = (*props)[j];
MtpProperty* property = getObjectPropDesc(prop, format);
if (property) {
property->print();
delete property;
} else {
MTPE("could not fetch property: %s",
MtpDebug::getObjectPropCodeName(prop));
}
}
}
}
}
}
const char* MtpDevice::getDeviceName() {
if (mDevice)
return usb_device_get_name(mDevice);
else
return "???";
}
bool MtpDevice::openSession() {
std::lock_guard<std::mutex> lg(mMutex);
mSessionID = 0;
mTransactionID = 0;
MtpSessionID newSession = 1;
mRequest.reset();
mRequest.setParameter(1, newSession);
if (!sendRequest(MTP_OPERATION_OPEN_SESSION))
return false;
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_SESSION_ALREADY_OPEN)
newSession = mResponse.getParameter(1);
else if (ret != MTP_RESPONSE_OK)
return false;
mSessionID = newSession;
mTransactionID = 1;
return true;
}
bool MtpDevice::closeSession() {
// FIXME
return true;
}
MtpDeviceInfo* MtpDevice::getDeviceInfo() {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
if (!sendRequest(MTP_OPERATION_GET_DEVICE_INFO))
return NULL;
if (!readData())
return NULL;
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
MtpDeviceInfo* info = new MtpDeviceInfo;
if (info->read(mData))
return info;
else
delete info;
}
return NULL;
}
MtpStorageIDList* MtpDevice::getStorageIDs() {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
if (!sendRequest(MTP_OPERATION_GET_STORAGE_IDS))
return NULL;
if (!readData())
return NULL;
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
return mData.getAUInt32();
}
return NULL;
}
MtpStorageInfo* MtpDevice::getStorageInfo(MtpStorageID storageID) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, storageID);
if (!sendRequest(MTP_OPERATION_GET_STORAGE_INFO))
return NULL;
if (!readData())
return NULL;
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
MtpStorageInfo* info = new MtpStorageInfo(storageID);
if (info->read(mData))
return info;
else
delete info;
}
return NULL;
}
MtpObjectHandleList* MtpDevice::getObjectHandles(MtpStorageID storageID,
MtpObjectFormat format, MtpObjectHandle parent) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, storageID);
mRequest.setParameter(2, format);
mRequest.setParameter(3, parent);
if (!sendRequest(MTP_OPERATION_GET_OBJECT_HANDLES))
return NULL;
if (!readData())
return NULL;
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
return mData.getAUInt32();
}
return NULL;
}
MtpObjectInfo* MtpDevice::getObjectInfo(MtpObjectHandle handle) {
std::lock_guard<std::mutex> lg(mMutex);
// FIXME - we might want to add some caching here
mRequest.reset();
mRequest.setParameter(1, handle);
if (!sendRequest(MTP_OPERATION_GET_OBJECT_INFO))
return NULL;
if (!readData())
return NULL;
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
MtpObjectInfo* info = new MtpObjectInfo(handle);
if (info->read(mData))
return info;
else
delete info;
}
return NULL;
}
void* MtpDevice::getThumbnail(MtpObjectHandle handle, int& outLength) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, handle);
if (sendRequest(MTP_OPERATION_GET_THUMB) && readData()) {
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
return mData.getData(&outLength);
}
}
outLength = 0;
return NULL;
}
MtpObjectHandle MtpDevice::sendObjectInfo(MtpObjectInfo* info) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
MtpObjectHandle parent = info->mParent;
if (parent == 0)
parent = MTP_PARENT_ROOT;
mRequest.setParameter(1, info->mStorageID);
mRequest.setParameter(2, parent);
mData.reset();
mData.putUInt32(info->mStorageID);
mData.putUInt16(info->mFormat);
mData.putUInt16(info->mProtectionStatus);
mData.putUInt32(info->mCompressedSize);
mData.putUInt16(info->mThumbFormat);
mData.putUInt32(info->mThumbCompressedSize);
mData.putUInt32(info->mThumbPixWidth);
mData.putUInt32(info->mThumbPixHeight);
mData.putUInt32(info->mImagePixWidth);
mData.putUInt32(info->mImagePixHeight);
mData.putUInt32(info->mImagePixDepth);
mData.putUInt32(info->mParent);
mData.putUInt16(info->mAssociationType);
mData.putUInt32(info->mAssociationDesc);
mData.putUInt32(info->mSequenceNumber);
mData.putString(info->mName);
char created[100], modified[100];
formatDateTime(info->mDateCreated, created, sizeof(created));
formatDateTime(info->mDateModified, modified, sizeof(modified));
mData.putString(created);
mData.putString(modified);
if (info->mKeywords)
mData.putString(info->mKeywords);
else
mData.putEmptyString();
if (sendRequest(MTP_OPERATION_SEND_OBJECT_INFO) && sendData()) {
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
mLastSendObjectInfoTransactionID = mRequest.getTransactionID();
mLastSendObjectInfoObjectHandle = mResponse.getParameter(3);
info->mStorageID = mResponse.getParameter(1);
info->mParent = mResponse.getParameter(2);
info->mHandle = mResponse.getParameter(3);
return info->mHandle;
}
}
return (MtpObjectHandle)-1;
}
bool MtpDevice::sendObject(MtpObjectHandle handle, int size, int srcFD) {
std::lock_guard<std::mutex> lg(mMutex);
if (mLastSendObjectInfoTransactionID + 1 != mTransactionID ||
mLastSendObjectInfoObjectHandle != handle) {
MTPE("A sendObject request must follow the sendObjectInfo request.");
return false;
}
mRequest.reset();
if (sendRequest(MTP_OPERATION_SEND_OBJECT)) {
mData.setOperationCode(mRequest.getOperationCode());
mData.setTransactionID(mRequest.getTransactionID());
const int writeResult = mData.write(mRequestOut, mPacketDivisionMode, srcFD, size);
const MtpResponseCode ret = readResponse();
return ret == MTP_RESPONSE_OK && writeResult > 0;
}
return false;
}
bool MtpDevice::deleteObject(MtpObjectHandle handle) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, handle);
if (sendRequest(MTP_OPERATION_DELETE_OBJECT)) {
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK)
return true;
}
return false;
}
MtpObjectHandle MtpDevice::getParent(MtpObjectHandle handle) {
MtpObjectInfo* info = getObjectInfo(handle);
if (info) {
MtpObjectHandle parent = info->mParent;
delete info;
return parent;
} else {
return -1;
}
}
MtpObjectHandle MtpDevice::getStorageID(MtpObjectHandle handle) {
MtpObjectInfo* info = getObjectInfo(handle);
if (info) {
MtpObjectHandle storageId = info->mStorageID;
delete info;
return storageId;
} else {
return -1;
}
}
MtpObjectPropertyList* MtpDevice::getObjectPropsSupported(MtpObjectFormat format) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, format);
if (!sendRequest(MTP_OPERATION_GET_OBJECT_PROPS_SUPPORTED))
return NULL;
if (!readData())
return NULL;
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
return mData.getAUInt16();
}
return NULL;
}
MtpProperty* MtpDevice::getDevicePropDesc(MtpDeviceProperty code) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, code);
if (!sendRequest(MTP_OPERATION_GET_DEVICE_PROP_DESC))
return NULL;
if (!readData())
return NULL;
MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
MtpProperty* property = new MtpProperty;
if (property->read(mData))
return property;
else
delete property;
}
return NULL;
}
MtpProperty* MtpDevice::getObjectPropDesc(MtpObjectProperty code, MtpObjectFormat format) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, code);
mRequest.setParameter(2, format);
if (!sendRequest(MTP_OPERATION_GET_OBJECT_PROP_DESC))
return NULL;
if (!readData())
return NULL;
const MtpResponseCode ret = readResponse();
if (ret == MTP_RESPONSE_OK) {
MtpProperty* property = new MtpProperty;
if (property->read(mData))
return property;
else
delete property;
}
return NULL;
}
bool MtpDevice::getObjectPropValue(MtpObjectHandle handle, MtpProperty* property) {
if (property == nullptr)
return false;
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, handle);
mRequest.setParameter(2, property->getPropertyCode());
if (!sendRequest(MTP_OPERATION_GET_OBJECT_PROP_VALUE))
return false;
if (!readData())
return false;
if (readResponse() != MTP_RESPONSE_OK)
return false;
property->setCurrentValue(mData);
return true;
}
bool MtpDevice::readObject(MtpObjectHandle handle,
ReadObjectCallback callback,
uint32_t expectedLength,
void* clientData) {
return readObjectInternal(handle, callback, &expectedLength, clientData);
}
// reads the object's data and writes it to the specified file path
bool MtpDevice::readObject(MtpObjectHandle handle, const char* destPath, int group, int perm) {
MTPD("readObject: %s", destPath);
int fd = ::open(destPath, O_RDWR | O_CREAT | O_TRUNC, S_IRUSR | S_IWUSR);
if (fd < 0) {
MTPE("open failed for %s", destPath);
return false;
}
fchown(fd, getuid(), group);
// set permissions
int mask = umask(0);
fchmod(fd, perm);
umask(mask);
bool result = readObject(handle, fd);
::close(fd);
return result;
}
bool MtpDevice::readObject(MtpObjectHandle handle, int fd) {
MTPD("readObject: %d", fd);
return readObjectInternal(handle, writeToFd, NULL /* expected size */, &fd);
}
bool MtpDevice::readObjectInternal(MtpObjectHandle handle,
ReadObjectCallback callback,
const uint32_t* expectedLength,
void* clientData) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, handle);
if (!sendRequest(MTP_OPERATION_GET_OBJECT)) {
MTPE("Failed to send a read request.");
return false;
}
return readData(callback, expectedLength, nullptr, clientData);
}
bool MtpDevice::readData(ReadObjectCallback callback,
const uint32_t* expectedLength,
uint32_t* writtenSize,
void* clientData) {
if (!mData.readDataHeader(mRequestIn1)) {
MTPE("Failed to read header.");
return false;
}
// If object size 0 byte, the remote device may reply a response packet without sending any data
// packets.
if (mData.getContainerType() == MTP_CONTAINER_TYPE_RESPONSE) {
mResponse.copyFrom(mData);
return mResponse.getResponseCode() == MTP_RESPONSE_OK;
}
const uint32_t fullLength = mData.getContainerLength();
if (fullLength < MTP_CONTAINER_HEADER_SIZE) {
MTPE("fullLength is too short: %d", fullLength);
return false;
}
const uint32_t length = fullLength - MTP_CONTAINER_HEADER_SIZE;
if (expectedLength && length != *expectedLength) {
MTPE("readObject error length: %d", fullLength);
return false;
}
uint32_t offset = 0;
bool writingError = false;
{
int initialDataLength = 0;
void* const initialData = mData.getData(&initialDataLength);
if (fullLength > MTP_CONTAINER_HEADER_SIZE && initialDataLength == 0) {
// According to the MTP spec, the responder (MTP device) can choose two ways of sending
// data. a) The first packet contains the head and as much of the payload as possible
// b) The first packet contains only the header. The initiator (MTP host) needs
// to remember which way the responder used, and send upcoming data in the same way.
MTPD("Found short packet that contains only a header.");
mPacketDivisionMode = FIRST_PACKET_ONLY_HEADER;
}
if (initialData) {
if (initialDataLength > 0) {
if (!callback(initialData, offset, initialDataLength, clientData)) {
MTPE("Failed to write initial data.");
writingError = true;
}
offset += initialDataLength;
}
free(initialData);
}
}
// USB reads greater than 16K don't work.
char buffer1[MTP_BUFFER_SIZE], buffer2[MTP_BUFFER_SIZE];
mRequestIn1->buffer = buffer1;
mRequestIn2->buffer = buffer2;
struct usb_request* req = NULL;
while (offset < length) {
// Wait for previous read to complete.
void* writeBuffer = NULL;
int writeLength = 0;
if (req) {
const int read = mData.readDataWait(mDevice);
if (read < 0) {
MTPE("readDataWait failed.");
return false;
}
writeBuffer = req->buffer;
writeLength = read;
}
// Request to read next chunk.
const uint32_t nextOffset = offset + writeLength;
if (nextOffset < length) {
// Queue up a read request.
const size_t remaining = length - nextOffset;
req = (req == mRequestIn1 ? mRequestIn2 : mRequestIn1);
req->buffer_length = remaining > MTP_BUFFER_SIZE ?
static_cast<size_t>(MTP_BUFFER_SIZE) : remaining;
if (mData.readDataAsync(req) != 0) {
MTPE("readDataAsync failed");
return false;
}
}
// Write previous buffer.
if (writeBuffer && !writingError) {
if (!callback(writeBuffer, offset, writeLength, clientData)) {
MTPE("write failed");
writingError = true;
}
}
offset = nextOffset;
}
if (writtenSize) {
*writtenSize = length;
}
return readResponse() == MTP_RESPONSE_OK;
}
bool MtpDevice::readPartialObject(MtpObjectHandle handle,
uint32_t offset,
uint32_t size,
uint32_t *writtenSize,
ReadObjectCallback callback,
void* clientData) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, handle);
mRequest.setParameter(2, offset);
mRequest.setParameter(3, size);
if (!sendRequest(MTP_OPERATION_GET_PARTIAL_OBJECT)) {
MTPE("Failed to send a read request.");
return false;
}
// The expected size is null because it requires the exact number of bytes to read though
// MTP_OPERATION_GET_PARTIAL_OBJECT allows devices to return shorter length of bytes than
// requested. Destination's buffer length should be checked in |callback|.
return readData(callback, nullptr /* expected size */, writtenSize, clientData);
}
bool MtpDevice::readPartialObject64(MtpObjectHandle handle,
uint64_t offset,
uint32_t size,
uint32_t *writtenSize,
ReadObjectCallback callback,
void* clientData) {
std::lock_guard<std::mutex> lg(mMutex);
mRequest.reset();
mRequest.setParameter(1, handle);
mRequest.setParameter(2, 0xffffffff & offset);
mRequest.setParameter(3, 0xffffffff & (offset >> 32));
mRequest.setParameter(4, size);
if (!sendRequest(MTP_OPERATION_GET_PARTIAL_OBJECT_64)) {
MTPE("Failed to send a read request.");
return false;
}
// The expected size is null because it requires the exact number of bytes to read though
// MTP_OPERATION_GET_PARTIAL_OBJECT_64 allows devices to return shorter length of bytes than
// requested. Destination's buffer length should be checked in |callback|.
return readData(callback, nullptr /* expected size */, writtenSize, clientData);
}
bool MtpDevice::sendRequest(MtpOperationCode operation) {
MTPD("sendRequest: %s\n", MtpDebug::getOperationCodeName(operation));
mReceivedResponse = false;
mRequest.setOperationCode(operation);
if (mTransactionID > 0)
mRequest.setTransactionID(mTransactionID++);
int ret = mRequest.write(mRequestOut);
mRequest.dump();
return (ret > 0);
}
bool MtpDevice::sendData() {
MTPD("sendData\n");
mData.setOperationCode(mRequest.getOperationCode());
mData.setTransactionID(mRequest.getTransactionID());
int ret = mData.write(mRequestOut, mPacketDivisionMode);
mData.dump();
return (ret >= 0);
}
bool MtpDevice::readData() {
mData.reset();
int ret = mData.read(mRequestIn1);
MTPD("readData returned %d\n", ret);
if (ret >= MTP_CONTAINER_HEADER_SIZE) {
if (mData.getContainerType() == MTP_CONTAINER_TYPE_RESPONSE) {
MTPD("got response packet instead of data packet");
// we got a response packet rather than data
// copy it to mResponse
mResponse.copyFrom(mData);
mReceivedResponse = true;
return false;
}
mData.dump();
return true;
}
else {
MTPD("readResponse failed\n");
return false;
}
}
MtpResponseCode MtpDevice::readResponse() {
MTPD("readResponse\n");
if (mReceivedResponse) {
mReceivedResponse = false;
return mResponse.getResponseCode();
}
int ret = mResponse.read(mRequestIn1);
// handle zero length packets, which might occur if the data transfer
// ends on a packet boundary
if (ret == 0)
ret = mResponse.read(mRequestIn1);
if (ret >= MTP_CONTAINER_HEADER_SIZE) {
mResponse.dump();
return mResponse.getResponseCode();
} else {
MTPD("readResponse failed\n");
return -1;
}
}
int MtpDevice::submitEventRequest() {
if (!mEventMutex.try_lock()) {
// An event is being reaped on another thread.
return -1;
}
if (mProcessingEvent) {
// An event request was submitted, but no reapEventRequest called so far.
return -1;
}
std::lock_guard<std::mutex> lg(mEventMutexForInterrupt);
mEventPacket.sendRequest(mRequestIntr);
const int currentHandle = ++mCurrentEventHandle;
mProcessingEvent = true;
mEventMutex.unlock();
return currentHandle;
}
int MtpDevice::reapEventRequest(int handle, uint32_t (*parameters)[3]) {
std::lock_guard<std::mutex> lg(mEventMutex);
if (!mProcessingEvent || mCurrentEventHandle != handle || !parameters) {
return -1;
}
mProcessingEvent = false;
const int readSize = mEventPacket.readResponse(mRequestIntr->dev);
const int result = mEventPacket.getEventCode();
// MTP event has three parameters.
(*parameters)[0] = mEventPacket.getParameter(1);
(*parameters)[1] = mEventPacket.getParameter(2);
(*parameters)[2] = mEventPacket.getParameter(3);
return readSize != 0 ? result : 0;
}
void MtpDevice::discardEventRequest(int handle) {
std::lock_guard<std::mutex> lg(mEventMutexForInterrupt);
if (mCurrentEventHandle != handle) {
return;
}
usb_request_cancel(mRequestIntr);
}
