/*
 * 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.
 *
 * Copyright (C) 2014 TeamWin - bigbiff and Dees_Troy mtp database conversion to c++
 *
 */

#ifndef _MTP_DATA_PACKET_H
#define _MTP_DATA_PACKET_H

#include "MtpPacket.h"
#include "mtp.h"

struct usb_device;
struct usb_request;


class MtpStringBuffer;

class MtpDataPacket : public MtpPacket {
private:
	// current offset for get/put methods
	uint64_t			mOffset;

public:
						MtpDataPacket();
	virtual				~MtpDataPacket();

	virtual void		reset();

	void				setOperationCode(MtpOperationCode code);
	void				setTransactionID(MtpTransactionID id);

	inline const uint8_t*	  getData() const { return mBuffer + MTP_CONTAINER_HEADER_SIZE; }
	inline uint8_t		getUInt8() { return (uint8_t)mBuffer[mOffset++]; }
	inline int8_t		getInt8() { return (int8_t)mBuffer[mOffset++]; }
	uint16_t			getUInt16();
	inline int16_t		getInt16() { return (int16_t)getUInt16(); }
	uint32_t			getUInt32();
	inline int32_t		getInt32() { return (int32_t)getUInt32(); }
	uint64_t			getUInt64();
	inline int64_t		getInt64() { return (int64_t)getUInt64(); }
	void				getUInt128(uint128_t& value);
	inline void			getInt128(int128_t& value) { getUInt128((uint128_t&)value); }
	void				getString(MtpStringBuffer& string);

	Int8List*			getAInt8();
	UInt8List*			getAUInt8();
	Int16List*			getAInt16();
	UInt16List*			getAUInt16();
	Int32List*			getAInt32();
	UInt32List*			getAUInt32();
	Int64List*			getAInt64();
	UInt64List*			getAUInt64();

	void				putInt8(int8_t value);
	void				putUInt8(uint8_t value);
	void				putInt16(int16_t value);
	void				putUInt16(uint16_t value);
	void				putInt32(int32_t value);
	void				putUInt32(uint32_t value);
	void				putInt64(int64_t value);
	void				putUInt64(uint64_t value);
	void				putInt128(const int128_t& value);
	void				putUInt128(const uint128_t& value);
	void				putInt128(int64_t value);
	void				putUInt128(uint64_t value);

	void				putAInt8(const int8_t* values, int count);
	void				putAUInt8(const uint8_t* values, int count);
	void				putAInt16(const int16_t* values, int count);
	void				putAUInt16(const uint16_t* values, int count);
	void				putAUInt16(const UInt16List* values);
	void				putAInt32(const int32_t* values, int count);
	void				putAUInt32(const uint32_t* values, int count);
	void				putAUInt32(const UInt32List* list);
	void				putAInt64(const int64_t* values, int count);
	void				putAUInt64(const uint64_t* values, int count);
	void				putString(const MtpStringBuffer& string);
	void				putString(const char* string);
	void				putString(const uint16_t* string);
	inline void			putEmptyString() { putUInt8(0); }
	inline void			putEmptyArray() { putUInt32(0); }


#ifdef MTP_DEVICE
	// fill our buffer with data from the given file descriptor
	int					read(int fd);

	// write our data to the given file descriptor
	int					write(int fd);
	int					writeData(int fd, void* data, uint32_t length);
#endif
#ifdef MTP_HOST
	int					read(struct usb_request *request);
	int					readData(struct usb_request *request, void* buffer, int length);
	int					readDataAsync(struct usb_request *req);
	int					readDataWait(struct usb_device *device);
	int					readDataHeader(struct usb_request *ep);

	int					writeDataHeader(struct usb_request *ep, uint32_t length);
	int					write(struct usb_request *ep);
	int					write(struct usb_request *ep, void* buffer, uint32_t length);
#endif
	inline bool			hasData() const { return mPacketSize > MTP_CONTAINER_HEADER_SIZE; }
	inline uint32_t		getContainerLength() const { return MtpPacket::getUInt32(MTP_CONTAINER_LENGTH_OFFSET); }
	void*				getData(int& outLength) const;
};


#endif // _MTP_DATA_PACKET_H