Home | History | Annotate | Download | only in mtp
      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 
     44     inline const uint8_t*     getData() const { return mBuffer + MTP_CONTAINER_HEADER_SIZE; }
     45     inline uint8_t      getUInt8() { return (uint8_t)mBuffer[mOffset++]; }
     46     inline int8_t       getInt8() { return (int8_t)mBuffer[mOffset++]; }
     47     uint16_t            getUInt16();
     48     inline int16_t      getInt16() { return (int16_t)getUInt16(); }
     49     uint32_t            getUInt32();
     50     inline int32_t      getInt32() { return (int32_t)getUInt32(); }
     51     uint64_t            getUInt64();
     52     inline int64_t      getInt64() { return (int64_t)getUInt64(); }
     53     void                getUInt128(uint128_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);
     92     inline void         putEmptyString() { putUInt8(0); }
     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 
    117     inline bool         hasData() const { return mPacketSize > MTP_CONTAINER_HEADER_SIZE; }
    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