DataSource.h revision a3630a418b4f65277a42cd4018cd3b0b7e134d0c
120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber/* 220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * Copyright (C) 2009 The Android Open Source Project 320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * 420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * Licensed under the Apache License, Version 2.0 (the "License"); 520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * you may not use this file except in compliance with the License. 620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * You may obtain a copy of the License at 720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * 820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * http://www.apache.org/licenses/LICENSE-2.0 920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * 1020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * Unless required by applicable law or agreed to in writing, software 1120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * distributed under the License is distributed on an "AS IS" BASIS, 1220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * See the License for the specific language governing permissions and 1420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * limitations under the License. 1520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber */ 1620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 1720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#ifndef DATA_SOURCE_H_ 1820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 1920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#define DATA_SOURCE_H_ 2020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 2120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <sys/types.h> 22a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa#include <media/stagefright/foundation/ADebug.h> 230683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber#include <media/stagefright/MediaErrors.h> 2420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <utils/Errors.h> 255561ccf4a8db88a2e44eac1b3ed13b4ff53a7f20Andreas Huber#include <utils/KeyedVector.h> 2620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <utils/List.h> 27693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber#include <utils/RefBase.h> 2820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <utils/threads.h> 29dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang#include <drm/DrmManagerClient.h> 3020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 3120111aa043c5f404472bc63b90bc5aad906b1101Andreas Hubernamespace android { 3220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 335a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huberstruct AMessage; 3420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huberclass String8; 3520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 36693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huberclass DataSource : public RefBase { 3720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huberpublic: 3866b0a35c9a4136a84c3f73b1b25fcb5220138b95Andreas Huber enum Flags { 39acdd9d0f59c3d8bdc3de80664a609527ec3d21e2Andreas Huber kWantsPrefetching = 1, 40acdd9d0f59c3d8bdc3de80664a609527ec3d21e2Andreas Huber kStreamedFromLocalHost = 2, 41bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber kIsCachingDataSource = 4, 42b33d2ac90cfce0fe6db8c3e979e7ae2bbfc28163James Dong kIsHTTPBasedSource = 8, 4366b0a35c9a4136a84c3f73b1b25fcb5220138b95Andreas Huber }; 4466b0a35c9a4136a84c3f73b1b25fcb5220138b95Andreas Huber 455561ccf4a8db88a2e44eac1b3ed13b4ff53a7f20Andreas Huber static sp<DataSource> CreateFromURI( 465561ccf4a8db88a2e44eac1b3ed13b4ff53a7f20Andreas Huber const char *uri, 475561ccf4a8db88a2e44eac1b3ed13b4ff53a7f20Andreas Huber const KeyedVector<String8, String8> *headers = NULL); 48fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber 4920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber DataSource() {} 5020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 5134769bc913e9f6bb138e666d94a9d685bf3da217Andreas Huber virtual status_t initCheck() const = 0; 5234769bc913e9f6bb138e666d94a9d685bf3da217Andreas Huber 53c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong virtual ssize_t readAt(off64_t offset, void *data, size_t size) = 0; 5420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 55693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber // Convenience methods: 56c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong bool getUInt16(off64_t offset, uint16_t *x); 57ec77122351b4e78c1fe5b60a208f76baf8c67591Marco Nelissen bool getUInt24(off64_t offset, uint32_t *x); // 3 byte int, returned as a 32-bit int 5805f625c46b992ab66b8d1527a366fe2746b4e3c7Marco Nelissen bool getUInt32(off64_t offset, uint32_t *x); 5905f625c46b992ab66b8d1527a366fe2746b4e3c7Marco Nelissen bool getUInt64(off64_t offset, uint64_t *x); 60693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 61a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // Reads in "count" entries of type T into vector *x. 62a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // Returns true if "count" entries can be read. 63a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // If fewer than "count" entries can be read, return false. In this case, 64a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // the output vector *x will still have those entries that were read. Call 65a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // x->size() to obtain the number of entries read. 66a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // The optional parameter chunkSize specifies how many entries should be 67a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // read from the data source at one time into a temporary buffer. Increasing 68a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // chunkSize can improve the performance at the cost of extra memory usage. 69a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // The default value for chunkSize is set to read at least 4k bytes at a 70a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // time, depending on sizeof(T). 71a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa template <typename T> 72a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa bool getVector(off64_t offset, Vector<T>* x, size_t count, 73a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa size_t chunkSize = (4095 / sizeof(T)) + 1); 74a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa 7520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber // May return ERROR_UNSUPPORTED. 76c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong virtual status_t getSize(off64_t *size); 7720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 7866b0a35c9a4136a84c3f73b1b25fcb5220138b95Andreas Huber virtual uint32_t flags() { 7966b0a35c9a4136a84c3f73b1b25fcb5220138b95Andreas Huber return 0; 8066b0a35c9a4136a84c3f73b1b25fcb5220138b95Andreas Huber } 8166b0a35c9a4136a84c3f73b1b25fcb5220138b95Andreas Huber 820683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber virtual status_t reconnectAtOffset(off64_t offset) { 830683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber return ERROR_UNSUPPORTED; 840683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber } 850683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber 8620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber //////////////////////////////////////////////////////////////////////////// 8720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 885a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber bool sniff(String8 *mimeType, float *confidence, sp<AMessage> *meta); 8920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 905a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber // The sniffer can optionally fill in "meta" with an AMessage containing 915a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber // a dictionary of values that helps the corresponding extractor initialize 925a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber // its state without duplicating effort already exerted by the sniffer. 9320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber typedef bool (*SnifferFunc)( 945a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber const sp<DataSource> &source, String8 *mimeType, 955a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber float *confidence, sp<AMessage> *meta); 9620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 9720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber static void RegisterDefaultSniffers(); 9820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 99dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang // for DRM 1009d2f386dd2885eaffa11fd494ae258bb09fe6397James Dong virtual sp<DecryptHandle> DrmInitialization(const char *mime = NULL) { 101dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang return NULL; 102dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } 103b5ce361d19e69fe156f7188c9ee0f4734b259874Gloria Wang virtual void getDrmInfo(sp<DecryptHandle> &handle, DrmManagerClient **client) {}; 104dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang 105771b85d9245a24273497792a2515d88d31c99e1eGloria Wang virtual String8 getUri() { 106771b85d9245a24273497792a2515d88d31c99e1eGloria Wang return String8(); 107771b85d9245a24273497792a2515d88d31c99e1eGloria Wang } 108dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang 1096511c9755c3a3360ba869772600c7aae048a7ffcAndreas Huber virtual String8 getMIMEType() const; 1106511c9755c3a3360ba869772600c7aae048a7ffcAndreas Huber 111693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huberprotected: 112693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber virtual ~DataSource() {} 113693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 11420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huberprivate: 11520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber static Mutex gSnifferMutex; 11620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber static List<SnifferFunc> gSniffers; 117e3067743d6ef2c6c1aac4361b6ed65e6d7bbe3e3Marco Nelissen static bool gSniffersRegistered; 118e3067743d6ef2c6c1aac4361b6ed65e6d7bbe3e3Marco Nelissen 119e3067743d6ef2c6c1aac4361b6ed65e6d7bbe3e3Marco Nelissen static void RegisterSniffer_l(SnifferFunc func); 12020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 12120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber DataSource(const DataSource &); 12220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber DataSource &operator=(const DataSource &); 12320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber}; 12420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 125a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasatemplate <typename T> 126a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasabool DataSource::getVector(off64_t offset, Vector<T>* x, size_t count, 127a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa size_t chunkSize) 128a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa{ 129a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa x->clear(); 130a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa if (chunkSize == 0) { 131a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa return false; 132a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa } 133a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa if (count == 0) { 134a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa return true; 135a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa } 136a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa 137a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa T tmp[chunkSize]; 138a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa ssize_t numBytesRead; 139a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa size_t numBytesPerChunk = chunkSize * sizeof(T); 140a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa size_t i; 141a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa 142a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa for (i = 0; i + chunkSize < count; i += chunkSize) { 143a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // This loops is executed when more than chunkSize records need to be 144a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // read. 145a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa numBytesRead = this->readAt(offset, (void*)&tmp, numBytesPerChunk); 146a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa if (numBytesRead == -1) { // If readAt() returns -1, there is an error. 147a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa return false; 148a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa } 149a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa if (numBytesRead < numBytesPerChunk) { 150a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // This case is triggered when the stream ends before the whole 151a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // chunk is read. 152a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa x->appendArray(tmp, (size_t)numBytesRead / sizeof(T)); 153a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa return false; 154a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa } 155a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa x->appendArray(tmp, chunkSize); 156a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa offset += numBytesPerChunk; 157a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa } 158a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa 159a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // There are (count - i) more records to read. 160a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // Right now, (count - i) <= chunkSize. 161a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa // We do the same thing as above, but with chunkSize replaced by count - i. 162a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa numBytesRead = this->readAt(offset, (void*)&tmp, (count - i) * sizeof(T)); 163a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa if (numBytesRead == -1) { 164a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa return false; 165a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa } 166a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa x->appendArray(tmp, (size_t)numBytesRead / sizeof(T)); 167a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa return x->size() == count; 168a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa} 169a3630a418b4f65277a42cd4018cd3b0b7e134d0cPawin Vongmasa 17020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} // namespace android 17120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 17220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#endif // DATA_SOURCE_H_ 173