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 #define LOG_TAG "MtpDataPacket"
18
19 #include <stdio.h>
20 #include <sys/types.h>
21 #include <fcntl.h>
22
23 #include <usbhost/usbhost.h>
24
25 #include "MtpDataPacket.h"
26 #include "MtpStringBuffer.h"
27
28 #define MTP_BUFFER_SIZE 16384
29
30 namespace android {
31
MtpDataPacket()32 MtpDataPacket::MtpDataPacket()
33 : MtpPacket(MTP_BUFFER_SIZE), // MAX_USBFS_BUFFER_SIZE
34 mOffset(MTP_CONTAINER_HEADER_SIZE)
35 {
36 }
37
~MtpDataPacket()38 MtpDataPacket::~MtpDataPacket() {
39 }
40
reset()41 void MtpDataPacket::reset() {
42 MtpPacket::reset();
43 mOffset = MTP_CONTAINER_HEADER_SIZE;
44 }
45
setOperationCode(MtpOperationCode code)46 void MtpDataPacket::setOperationCode(MtpOperationCode code) {
47 MtpPacket::putUInt16(MTP_CONTAINER_CODE_OFFSET, code);
48 }
49
setTransactionID(MtpTransactionID id)50 void MtpDataPacket::setTransactionID(MtpTransactionID id) {
51 MtpPacket::putUInt32(MTP_CONTAINER_TRANSACTION_ID_OFFSET, id);
52 }
53
getUInt16()54 uint16_t MtpDataPacket::getUInt16() {
55 int offset = mOffset;
56 uint16_t result = (uint16_t)mBuffer[offset] | ((uint16_t)mBuffer[offset + 1] << 8);
57 mOffset += 2;
58 return result;
59 }
60
getUInt32()61 uint32_t MtpDataPacket::getUInt32() {
62 int offset = mOffset;
63 uint32_t result = (uint32_t)mBuffer[offset] | ((uint32_t)mBuffer[offset + 1] << 8) |
64 ((uint32_t)mBuffer[offset + 2] << 16) | ((uint32_t)mBuffer[offset + 3] << 24);
65 mOffset += 4;
66 return result;
67 }
68
getUInt64()69 uint64_t MtpDataPacket::getUInt64() {
70 int offset = mOffset;
71 uint64_t result = (uint64_t)mBuffer[offset] | ((uint64_t)mBuffer[offset + 1] << 8) |
72 ((uint64_t)mBuffer[offset + 2] << 16) | ((uint64_t)mBuffer[offset + 3] << 24) |
73 ((uint64_t)mBuffer[offset + 4] << 32) | ((uint64_t)mBuffer[offset + 5] << 40) |
74 ((uint64_t)mBuffer[offset + 6] << 48) | ((uint64_t)mBuffer[offset + 7] << 56);
75 mOffset += 8;
76 return result;
77 }
78
getUInt128(uint128_t & value)79 void MtpDataPacket::getUInt128(uint128_t& value) {
80 value[0] = getUInt32();
81 value[1] = getUInt32();
82 value[2] = getUInt32();
83 value[3] = getUInt32();
84 }
85
getString(MtpStringBuffer & string)86 void MtpDataPacket::getString(MtpStringBuffer& string)
87 {
88 string.readFromPacket(this);
89 }
90
getAInt8()91 Int8List* MtpDataPacket::getAInt8() {
92 Int8List* result = new Int8List;
93 int count = getUInt32();
94 for (int i = 0; i < count; i++)
95 result->push(getInt8());
96 return result;
97 }
98
getAUInt8()99 UInt8List* MtpDataPacket::getAUInt8() {
100 UInt8List* result = new UInt8List;
101 int count = getUInt32();
102 for (int i = 0; i < count; i++)
103 result->push(getUInt8());
104 return result;
105 }
106
getAInt16()107 Int16List* MtpDataPacket::getAInt16() {
108 Int16List* result = new Int16List;
109 int count = getUInt32();
110 for (int i = 0; i < count; i++)
111 result->push(getInt16());
112 return result;
113 }
114
getAUInt16()115 UInt16List* MtpDataPacket::getAUInt16() {
116 UInt16List* result = new UInt16List;
117 int count = getUInt32();
118 for (int i = 0; i < count; i++)
119 result->push(getUInt16());
120 return result;
121 }
122
getAInt32()123 Int32List* MtpDataPacket::getAInt32() {
124 Int32List* result = new Int32List;
125 int count = getUInt32();
126 for (int i = 0; i < count; i++)
127 result->push(getInt32());
128 return result;
129 }
130
getAUInt32()131 UInt32List* MtpDataPacket::getAUInt32() {
132 UInt32List* result = new UInt32List;
133 int count = getUInt32();
134 for (int i = 0; i < count; i++)
135 result->push(getUInt32());
136 return result;
137 }
138
getAInt64()139 Int64List* MtpDataPacket::getAInt64() {
140 Int64List* result = new Int64List;
141 int count = getUInt32();
142 for (int i = 0; i < count; i++)
143 result->push(getInt64());
144 return result;
145 }
146
getAUInt64()147 UInt64List* MtpDataPacket::getAUInt64() {
148 UInt64List* result = new UInt64List;
149 int count = getUInt32();
150 for (int i = 0; i < count; i++)
151 result->push(getUInt64());
152 return result;
153 }
154
putInt8(int8_t value)155 void MtpDataPacket::putInt8(int8_t value) {
156 allocate(mOffset + 1);
157 mBuffer[mOffset++] = (uint8_t)value;
158 if (mPacketSize < mOffset)
159 mPacketSize = mOffset;
160 }
161
putUInt8(uint8_t value)162 void MtpDataPacket::putUInt8(uint8_t value) {
163 allocate(mOffset + 1);
164 mBuffer[mOffset++] = (uint8_t)value;
165 if (mPacketSize < mOffset)
166 mPacketSize = mOffset;
167 }
168
putInt16(int16_t value)169 void MtpDataPacket::putInt16(int16_t value) {
170 allocate(mOffset + 2);
171 mBuffer[mOffset++] = (uint8_t)(value & 0xFF);
172 mBuffer[mOffset++] = (uint8_t)((value >> 8) & 0xFF);
173 if (mPacketSize < mOffset)
174 mPacketSize = mOffset;
175 }
176
putUInt16(uint16_t value)177 void MtpDataPacket::putUInt16(uint16_t value) {
178 allocate(mOffset + 2);
179 mBuffer[mOffset++] = (uint8_t)(value & 0xFF);
180 mBuffer[mOffset++] = (uint8_t)((value >> 8) & 0xFF);
181 if (mPacketSize < mOffset)
182 mPacketSize = mOffset;
183 }
184
putInt32(int32_t value)185 void MtpDataPacket::putInt32(int32_t value) {
186 allocate(mOffset + 4);
187 mBuffer[mOffset++] = (uint8_t)(value & 0xFF);
188 mBuffer[mOffset++] = (uint8_t)((value >> 8) & 0xFF);
189 mBuffer[mOffset++] = (uint8_t)((value >> 16) & 0xFF);
190 mBuffer[mOffset++] = (uint8_t)((value >> 24) & 0xFF);
191 if (mPacketSize < mOffset)
192 mPacketSize = mOffset;
193 }
194
putUInt32(uint32_t value)195 void MtpDataPacket::putUInt32(uint32_t value) {
196 allocate(mOffset + 4);
197 mBuffer[mOffset++] = (uint8_t)(value & 0xFF);
198 mBuffer[mOffset++] = (uint8_t)((value >> 8) & 0xFF);
199 mBuffer[mOffset++] = (uint8_t)((value >> 16) & 0xFF);
200 mBuffer[mOffset++] = (uint8_t)((value >> 24) & 0xFF);
201 if (mPacketSize < mOffset)
202 mPacketSize = mOffset;
203 }
204
putInt64(int64_t value)205 void MtpDataPacket::putInt64(int64_t value) {
206 allocate(mOffset + 8);
207 mBuffer[mOffset++] = (uint8_t)(value & 0xFF);
208 mBuffer[mOffset++] = (uint8_t)((value >> 8) & 0xFF);
209 mBuffer[mOffset++] = (uint8_t)((value >> 16) & 0xFF);
210 mBuffer[mOffset++] = (uint8_t)((value >> 24) & 0xFF);
211 mBuffer[mOffset++] = (uint8_t)((value >> 32) & 0xFF);
212 mBuffer[mOffset++] = (uint8_t)((value >> 40) & 0xFF);
213 mBuffer[mOffset++] = (uint8_t)((value >> 48) & 0xFF);
214 mBuffer[mOffset++] = (uint8_t)((value >> 56) & 0xFF);
215 if (mPacketSize < mOffset)
216 mPacketSize = mOffset;
217 }
218
putUInt64(uint64_t value)219 void MtpDataPacket::putUInt64(uint64_t value) {
220 allocate(mOffset + 8);
221 mBuffer[mOffset++] = (uint8_t)(value & 0xFF);
222 mBuffer[mOffset++] = (uint8_t)((value >> 8) & 0xFF);
223 mBuffer[mOffset++] = (uint8_t)((value >> 16) & 0xFF);
224 mBuffer[mOffset++] = (uint8_t)((value >> 24) & 0xFF);
225 mBuffer[mOffset++] = (uint8_t)((value >> 32) & 0xFF);
226 mBuffer[mOffset++] = (uint8_t)((value >> 40) & 0xFF);
227 mBuffer[mOffset++] = (uint8_t)((value >> 48) & 0xFF);
228 mBuffer[mOffset++] = (uint8_t)((value >> 56) & 0xFF);
229 if (mPacketSize < mOffset)
230 mPacketSize = mOffset;
231 }
232
putInt128(const int128_t & value)233 void MtpDataPacket::putInt128(const int128_t& value) {
234 putInt32(value[0]);
235 putInt32(value[1]);
236 putInt32(value[2]);
237 putInt32(value[3]);
238 }
239
putUInt128(const uint128_t & value)240 void MtpDataPacket::putUInt128(const uint128_t& value) {
241 putUInt32(value[0]);
242 putUInt32(value[1]);
243 putUInt32(value[2]);
244 putUInt32(value[3]);
245 }
246
putInt128(int64_t value)247 void MtpDataPacket::putInt128(int64_t value) {
248 putInt64(value);
249 putInt64(value < 0 ? -1 : 0);
250 }
251
putUInt128(uint64_t value)252 void MtpDataPacket::putUInt128(uint64_t value) {
253 putUInt64(value);
254 putUInt64(0);
255 }
256
putAInt8(const int8_t * values,int count)257 void MtpDataPacket::putAInt8(const int8_t* values, int count) {
258 putUInt32(count);
259 for (int i = 0; i < count; i++)
260 putInt8(*values++);
261 }
262
putAUInt8(const uint8_t * values,int count)263 void MtpDataPacket::putAUInt8(const uint8_t* values, int count) {
264 putUInt32(count);
265 for (int i = 0; i < count; i++)
266 putUInt8(*values++);
267 }
268
putAInt16(const int16_t * values,int count)269 void MtpDataPacket::putAInt16(const int16_t* values, int count) {
270 putUInt32(count);
271 for (int i = 0; i < count; i++)
272 putInt16(*values++);
273 }
274
putAUInt16(const uint16_t * values,int count)275 void MtpDataPacket::putAUInt16(const uint16_t* values, int count) {
276 putUInt32(count);
277 for (int i = 0; i < count; i++)
278 putUInt16(*values++);
279 }
280
putAUInt16(const UInt16List * values)281 void MtpDataPacket::putAUInt16(const UInt16List* values) {
282 size_t count = (values ? values->size() : 0);
283 putUInt32(count);
284 for (size_t i = 0; i < count; i++)
285 putUInt16((*values)[i]);
286 }
287
putAInt32(const int32_t * values,int count)288 void MtpDataPacket::putAInt32(const int32_t* values, int count) {
289 putUInt32(count);
290 for (int i = 0; i < count; i++)
291 putInt32(*values++);
292 }
293
putAUInt32(const uint32_t * values,int count)294 void MtpDataPacket::putAUInt32(const uint32_t* values, int count) {
295 putUInt32(count);
296 for (int i = 0; i < count; i++)
297 putUInt32(*values++);
298 }
299
putAUInt32(const UInt32List * list)300 void MtpDataPacket::putAUInt32(const UInt32List* list) {
301 if (!list) {
302 putEmptyArray();
303 } else {
304 size_t size = list->size();
305 putUInt32(size);
306 for (size_t i = 0; i < size; i++)
307 putUInt32((*list)[i]);
308 }
309 }
310
putAInt64(const int64_t * values,int count)311 void MtpDataPacket::putAInt64(const int64_t* values, int count) {
312 putUInt32(count);
313 for (int i = 0; i < count; i++)
314 putInt64(*values++);
315 }
316
putAUInt64(const uint64_t * values,int count)317 void MtpDataPacket::putAUInt64(const uint64_t* values, int count) {
318 putUInt32(count);
319 for (int i = 0; i < count; i++)
320 putUInt64(*values++);
321 }
322
putString(const MtpStringBuffer & string)323 void MtpDataPacket::putString(const MtpStringBuffer& string) {
324 string.writeToPacket(this);
325 }
326
putString(const char * s)327 void MtpDataPacket::putString(const char* s) {
328 MtpStringBuffer string(s);
329 string.writeToPacket(this);
330 }
331
putString(const uint16_t * string)332 void MtpDataPacket::putString(const uint16_t* string) {
333 int count = 0;
334 for (int i = 0; i < 256; i++) {
335 if (string[i])
336 count++;
337 else
338 break;
339 }
340 putUInt8(count > 0 ? count + 1 : 0);
341 for (int i = 0; i < count; i++)
342 putUInt16(string[i]);
343 // only terminate with zero if string is not empty
344 if (count > 0)
345 putUInt16(0);
346 }
347
348 #ifdef MTP_DEVICE
read(int fd)349 int MtpDataPacket::read(int fd) {
350 int ret = ::read(fd, mBuffer, MTP_BUFFER_SIZE);
351 if (ret < MTP_CONTAINER_HEADER_SIZE)
352 return -1;
353 mPacketSize = ret;
354 mOffset = MTP_CONTAINER_HEADER_SIZE;
355 return ret;
356 }
357
write(int fd)358 int MtpDataPacket::write(int fd) {
359 MtpPacket::putUInt32(MTP_CONTAINER_LENGTH_OFFSET, mPacketSize);
360 MtpPacket::putUInt16(MTP_CONTAINER_TYPE_OFFSET, MTP_CONTAINER_TYPE_DATA);
361 int ret = ::write(fd, mBuffer, mPacketSize);
362 return (ret < 0 ? ret : 0);
363 }
364
writeData(int fd,void * data,uint32_t length)365 int MtpDataPacket::writeData(int fd, void* data, uint32_t length) {
366 allocate(length);
367 memcpy(mBuffer + MTP_CONTAINER_HEADER_SIZE, data, length);
368 length += MTP_CONTAINER_HEADER_SIZE;
369 MtpPacket::putUInt32(MTP_CONTAINER_LENGTH_OFFSET, length);
370 MtpPacket::putUInt16(MTP_CONTAINER_TYPE_OFFSET, MTP_CONTAINER_TYPE_DATA);
371 int ret = ::write(fd, mBuffer, length);
372 return (ret < 0 ? ret : 0);
373 }
374
375 #endif // MTP_DEVICE
376
377 #ifdef MTP_HOST
read(struct usb_request * request)378 int MtpDataPacket::read(struct usb_request *request) {
379 // first read the header
380 request->buffer = mBuffer;
381 request->buffer_length = mBufferSize;
382 int length = transfer(request);
383 if (length >= MTP_CONTAINER_HEADER_SIZE) {
384 // look at the length field to see if the data spans multiple packets
385 uint32_t totalLength = MtpPacket::getUInt32(MTP_CONTAINER_LENGTH_OFFSET);
386 allocate(totalLength);
387 while (totalLength > length) {
388 request->buffer = mBuffer + length;
389 request->buffer_length = totalLength - length;
390 int ret = transfer(request);
391 if (ret >= 0)
392 length += ret;
393 else {
394 length = ret;
395 break;
396 }
397 }
398 }
399 if (length >= 0)
400 mPacketSize = length;
401 return length;
402 }
403
readData(struct usb_request * request,void * buffer,int length)404 int MtpDataPacket::readData(struct usb_request *request, void* buffer, int length) {
405 int read = 0;
406 while (read < length) {
407 request->buffer = (char *)buffer + read;
408 request->buffer_length = length - read;
409 int ret = transfer(request);
410 if (ret < 0) {
411 return ret;
412 }
413 read += ret;
414 }
415 return read;
416 }
417
418 // Queue a read request. Call readDataWait to wait for result
readDataAsync(struct usb_request * req)419 int MtpDataPacket::readDataAsync(struct usb_request *req) {
420 if (usb_request_queue(req)) {
421 ALOGE("usb_endpoint_queue failed, errno: %d", errno);
422 return -1;
423 }
424 return 0;
425 }
426
427 // Wait for result of readDataAsync
readDataWait(struct usb_device * device)428 int MtpDataPacket::readDataWait(struct usb_device *device) {
429 struct usb_request *req = usb_request_wait(device);
430 return (req ? req->actual_length : -1);
431 }
432
readDataHeader(struct usb_request * request)433 int MtpDataPacket::readDataHeader(struct usb_request *request) {
434 request->buffer = mBuffer;
435 request->buffer_length = request->max_packet_size;
436 int length = transfer(request);
437 if (length >= 0)
438 mPacketSize = length;
439 return length;
440 }
441
writeDataHeader(struct usb_request * request,uint32_t length)442 int MtpDataPacket::writeDataHeader(struct usb_request *request, uint32_t length) {
443 MtpPacket::putUInt32(MTP_CONTAINER_LENGTH_OFFSET, length);
444 MtpPacket::putUInt16(MTP_CONTAINER_TYPE_OFFSET, MTP_CONTAINER_TYPE_DATA);
445 request->buffer = mBuffer;
446 request->buffer_length = MTP_CONTAINER_HEADER_SIZE;
447 int ret = transfer(request);
448 return (ret < 0 ? ret : 0);
449 }
450
write(struct usb_request * request)451 int MtpDataPacket::write(struct usb_request *request) {
452 MtpPacket::putUInt32(MTP_CONTAINER_LENGTH_OFFSET, mPacketSize);
453 MtpPacket::putUInt16(MTP_CONTAINER_TYPE_OFFSET, MTP_CONTAINER_TYPE_DATA);
454
455 // send header separately from data
456 request->buffer = mBuffer;
457 request->buffer_length = MTP_CONTAINER_HEADER_SIZE;
458 int ret = transfer(request);
459 if (ret == MTP_CONTAINER_HEADER_SIZE) {
460 request->buffer = mBuffer + MTP_CONTAINER_HEADER_SIZE;
461 request->buffer_length = mPacketSize - MTP_CONTAINER_HEADER_SIZE;
462 ret = transfer(request);
463 }
464 return (ret < 0 ? ret : 0);
465 }
466
write(struct usb_request * request,void * buffer,uint32_t length)467 int MtpDataPacket::write(struct usb_request *request, void* buffer, uint32_t length) {
468 request->buffer = buffer;
469 request->buffer_length = length;
470 int ret = transfer(request);
471 return (ret < 0 ? ret : 0);
472 }
473
474 #endif // MTP_HOST
475
getData(int & outLength) const476 void* MtpDataPacket::getData(int& outLength) const {
477 int length = mPacketSize - MTP_CONTAINER_HEADER_SIZE;
478 if (length > 0) {
479 void* result = malloc(length);
480 if (result) {
481 memcpy(result, mBuffer + MTP_CONTAINER_HEADER_SIZE, length);
482 outLength = length;
483 return result;
484 }
485 }
486 outLength = 0;
487 return NULL;
488 }
489
490 } // namespace android
491