DataSource.cpp revision 5a1c3529e4fa2f8a11054181294e0ce79fff8dd3
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 1766326a5ee0869f1ee4d136a477e6effba428b3cbAndreas Huber#include "include/AMRExtractor.h" 1866326a5ee0869f1ee4d136a477e6effba428b3cbAndreas Huber#include "include/MP3Extractor.h" 1966326a5ee0869f1ee4d136a477e6effba428b3cbAndreas Huber#include "include/MPEG4Extractor.h" 200dba73763a04d39faf999dcc5ef12af3c99535a7Andreas Huber#include "include/WAVExtractor.h" 21ee7ff20e69498ebd53dd9717a0f984188341a75eAndreas Huber#include "include/OggExtractor.h" 22cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include "include/MPEG2TSExtractor.h" 230a5baa9b411fe086013d2a5e9126ed63fbad046cAndreas Huber#include "include/NuCachedSource2.h" 240a5baa9b411fe086013d2a5e9126ed63fbad046cAndreas Huber#include "include/NuHTTPDataSource.h" 2566326a5ee0869f1ee4d136a477e6effba428b3cbAndreas Huber 26093437c388e5dff6903a3d43f2ca9f8a1ba4744aAndreas Huber#include "matroska/MatroskaExtractor.h" 27093437c388e5dff6903a3d43f2ca9f8a1ba4744aAndreas Huber 285a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber#include <media/stagefright/foundation/AMessage.h> 2920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/DataSource.h> 30fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber#include <media/stagefright/FileSource.h> 3120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/MediaErrors.h> 3220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <utils/String8.h> 3320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 3420111aa043c5f404472bc63b90bc5aad906b1101Andreas Hubernamespace android { 3520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 36693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huberbool DataSource::getUInt16(off_t offset, uint16_t *x) { 37693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber *x = 0; 38693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 39693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber uint8_t byte[2]; 4034769bc913e9f6bb138e666d94a9d685bf3da217Andreas Huber if (readAt(offset, byte, 2) != 2) { 41693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber return false; 42693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber } 43693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 44693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber *x = (byte[0] << 8) | byte[1]; 45693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 46693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber return true; 47693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber} 48693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 4920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huberstatus_t DataSource::getSize(off_t *size) { 5020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber *size = 0; 5120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 5220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber return ERROR_UNSUPPORTED; 5320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} 5420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 5520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber//////////////////////////////////////////////////////////////////////////////// 5620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 5720111aa043c5f404472bc63b90bc5aad906b1101Andreas HuberMutex DataSource::gSnifferMutex; 5820111aa043c5f404472bc63b90bc5aad906b1101Andreas HuberList<DataSource::SnifferFunc> DataSource::gSniffers; 5920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 605a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huberbool DataSource::sniff( 615a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber String8 *mimeType, float *confidence, sp<AMessage> *meta) { 6220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber *mimeType = ""; 6320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber *confidence = 0.0f; 645a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber meta->clear(); 6520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 6620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber Mutex::Autolock autoLock(gSnifferMutex); 6720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber for (List<SnifferFunc>::iterator it = gSniffers.begin(); 6820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber it != gSniffers.end(); ++it) { 6920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber String8 newMimeType; 7020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber float newConfidence; 715a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber sp<AMessage> newMeta; 725a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber if ((*it)(this, &newMimeType, &newConfidence, &newMeta)) { 7320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber if (newConfidence > *confidence) { 7420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber *mimeType = newMimeType; 7520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber *confidence = newConfidence; 765a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber *meta = newMeta; 7720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 7820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 7920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 8020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 8120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber return *confidence > 0.0; 8220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} 8320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 8420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber// static 8520111aa043c5f404472bc63b90bc5aad906b1101Andreas Hubervoid DataSource::RegisterSniffer(SnifferFunc func) { 8620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber Mutex::Autolock autoLock(gSnifferMutex); 8720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 8820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber for (List<SnifferFunc>::iterator it = gSniffers.begin(); 8920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber it != gSniffers.end(); ++it) { 9020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber if (*it == func) { 9120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber return; 9220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 9320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 9420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 9520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber gSniffers.push_back(func); 9620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} 9720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 9820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber// static 9920111aa043c5f404472bc63b90bc5aad906b1101Andreas Hubervoid DataSource::RegisterDefaultSniffers() { 10020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber RegisterSniffer(SniffMPEG4); 101093437c388e5dff6903a3d43f2ca9f8a1ba4744aAndreas Huber RegisterSniffer(SniffMatroska); 1025a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber RegisterSniffer(SniffOgg); 1035a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber RegisterSniffer(SniffWAV); 1045a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber RegisterSniffer(SniffAMR); 105cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber RegisterSniffer(SniffMPEG2TS); 1065a1c3529e4fa2f8a11054181294e0ce79fff8dd3Andreas Huber RegisterSniffer(SniffMP3); 10720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} 10820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 109fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber// static 1105561ccf4a8db88a2e44eac1b3ed13b4ff53a7f20Andreas Hubersp<DataSource> DataSource::CreateFromURI( 1115561ccf4a8db88a2e44eac1b3ed13b4ff53a7f20Andreas Huber const char *uri, const KeyedVector<String8, String8> *headers) { 112fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber sp<DataSource> source; 113fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber if (!strncasecmp("file://", uri, 7)) { 114fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber source = new FileSource(uri + 7); 115fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber } else if (!strncasecmp("http://", uri, 7)) { 1160a5baa9b411fe086013d2a5e9126ed63fbad046cAndreas Huber sp<NuHTTPDataSource> httpSource = new NuHTTPDataSource; 11779f77ef3b0a37660ba8c5bcb2dfbfda3860f2135Andreas Huber if (httpSource->connect(uri, headers) != OK) { 118e94bd14078d327ef2f800e69907efce641a13272Andreas Huber return NULL; 119e94bd14078d327ef2f800e69907efce641a13272Andreas Huber } 1200a5baa9b411fe086013d2a5e9126ed63fbad046cAndreas Huber source = new NuCachedSource2(httpSource); 121fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber } else { 122fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber // Assume it's a filename. 123fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber source = new FileSource(uri); 124fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber } 125fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber 126fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber if (source == NULL || source->initCheck() != OK) { 127fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber return NULL; 128fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber } 129fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber 130fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber return source; 131fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber} 132fc9ba09e3bb368f823d473f5e2bb9aa32dba6289Andreas Huber 13320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} // namespace android 134