1 /* 2 * Copyright (C) 2010 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 #ifndef _MTP_DATA_PACKET_H 18 #define _MTP_DATA_PACKET_H 19 20 #include "MtpPacket.h" 21 #include "mtp.h" 22 23 struct usb_device; 24 struct usb_request; 25 26 namespace android { 27 28 class MtpStringBuffer; 29 30 class MtpDataPacket : public MtpPacket { 31 private: 32 // current offset for get/put methods 33 int mOffset; 34 35 public: 36 MtpDataPacket(); 37 virtual ~MtpDataPacket(); 38 39 virtual void reset(); 40 41 void setOperationCode(MtpOperationCode code); 42 void setTransactionID(MtpTransactionID id); 43 getData()44 inline const uint8_t* getData() const { return mBuffer + MTP_CONTAINER_HEADER_SIZE; } getUInt8()45 inline uint8_t getUInt8() { return (uint8_t)mBuffer[mOffset++]; } getInt8()46 inline int8_t getInt8() { return (int8_t)mBuffer[mOffset++]; } 47 uint16_t getUInt16(); getInt16()48 inline int16_t getInt16() { return (int16_t)getUInt16(); } 49 uint32_t getUInt32(); getInt32()50 inline int32_t getInt32() { return (int32_t)getUInt32(); } 51 uint64_t getUInt64(); getInt64()52 inline int64_t getInt64() { return (int64_t)getUInt64(); } 53 void getUInt128(uint128_t& value); getInt128(int128_t & value)54 inline void getInt128(int128_t& value) { getUInt128((uint128_t&)value); } 55 void getString(MtpStringBuffer& string); 56 57 Int8List* getAInt8(); 58 UInt8List* getAUInt8(); 59 Int16List* getAInt16(); 60 UInt16List* getAUInt16(); 61 Int32List* getAInt32(); 62 UInt32List* getAUInt32(); 63 Int64List* getAInt64(); 64 UInt64List* getAUInt64(); 65 66 void putInt8(int8_t value); 67 void putUInt8(uint8_t value); 68 void putInt16(int16_t value); 69 void putUInt16(uint16_t value); 70 void putInt32(int32_t value); 71 void putUInt32(uint32_t value); 72 void putInt64(int64_t value); 73 void putUInt64(uint64_t value); 74 void putInt128(const int128_t& value); 75 void putUInt128(const uint128_t& value); 76 void putInt128(int64_t value); 77 void putUInt128(uint64_t value); 78 79 void putAInt8(const int8_t* values, int count); 80 void putAUInt8(const uint8_t* values, int count); 81 void putAInt16(const int16_t* values, int count); 82 void putAUInt16(const uint16_t* values, int count); 83 void putAUInt16(const UInt16List* values); 84 void putAInt32(const int32_t* values, int count); 85 void putAUInt32(const uint32_t* values, int count); 86 void putAUInt32(const UInt32List* list); 87 void putAInt64(const int64_t* values, int count); 88 void putAUInt64(const uint64_t* values, int count); 89 void putString(const MtpStringBuffer& string); 90 void putString(const char* string); 91 void putString(const uint16_t* string); putEmptyString()92 inline void putEmptyString() { putUInt8(0); } putEmptyArray()93 inline void putEmptyArray() { putUInt32(0); } 94 95 96 #ifdef MTP_DEVICE 97 // fill our buffer with data from the given file descriptor 98 int read(int fd); 99 100 // write our data to the given file descriptor 101 int write(int fd); 102 int writeData(int fd, void* data, uint32_t length); 103 #endif 104 105 #ifdef MTP_HOST 106 int read(struct usb_request *request); 107 int readData(struct usb_request *request, void* buffer, int length); 108 int readDataAsync(struct usb_request *req); 109 int readDataWait(struct usb_device *device); 110 int readDataHeader(struct usb_request *ep); 111 112 int writeDataHeader(struct usb_request *ep, uint32_t length); 113 int write(struct usb_request *ep); 114 int write(struct usb_request *ep, void* buffer, uint32_t length); 115 #endif 116 hasData()117 inline bool hasData() const { return mPacketSize > MTP_CONTAINER_HEADER_SIZE; } getContainerLength()118 inline uint32_t getContainerLength() const { return MtpPacket::getUInt32(MTP_CONTAINER_LENGTH_OFFSET); } 119 void* getData(int& outLength) const; 120 }; 121 122 }; // namespace android 123 124 #endif // _MTP_DATA_PACKET_H 125