| /* |
| * 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); |
| } |