FileSource.cpp revision 83b0fd9997b558f6c2ebf5e6e4db20570cb233b8
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 174f82a25c07c3ffc8d44d61797043f9864058afdfMarco Nelissen//#define LOG_NDEBUG 0 184f82a25c07c3ffc8d44d61797043f9864058afdfMarco Nelissen#define LOG_TAG "FileSource" 194f82a25c07c3ffc8d44d61797043f9864058afdfMarco Nelissen#include <utils/Log.h> 204f82a25c07c3ffc8d44d61797043f9864058afdfMarco Nelissen 21f1d5aa162c02a16b7195a43a9bcea4d592600ac4James Dong#include <media/stagefright/foundation/ADebug.h> 2220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/FileSource.h> 2383b0fd9997b558f6c2ebf5e6e4db20570cb233b8Marco Nelissen#include <media/stagefright/Utils.h> 24c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong#include <sys/types.h> 25c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong#include <unistd.h> 26674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong#include <sys/types.h> 27674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong#include <sys/stat.h> 28674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong#include <fcntl.h> 2920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 3020111aa043c5f404472bc63b90bc5aad906b1101Andreas Hubernamespace android { 3120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 3220111aa043c5f404472bc63b90bc5aad906b1101Andreas HuberFileSource::FileSource(const char *filename) 33674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong : mFd(-1), 3403475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber mOffset(0), 35dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mLength(-1), 36dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDecryptHandle(NULL), 37dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmManagerClient(NULL), 38dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBufOffset(0), 39dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBufSize(0), 40dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBuf(NULL){ 41674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong 4283b0fd9997b558f6c2ebf5e6e4db20570cb233b8Marco Nelissen ALOGV("%s", filename); 43674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong mFd = open(filename, O_LARGEFILE | O_RDONLY); 444ee31e2603b4019e4ae76e027115f80b818527aaAndreas Huber 454ee31e2603b4019e4ae76e027115f80b818527aaAndreas Huber if (mFd >= 0) { 464ee31e2603b4019e4ae76e027115f80b818527aaAndreas Huber mLength = lseek64(mFd, 0, SEEK_END); 47f4b7d94062c896dca565d849c6188cd3489be257Andreas Huber } else { 48f4b7d94062c896dca565d849c6188cd3489be257Andreas Huber ALOGE("Failed to open file '%s'. (%s)", filename, strerror(errno)); 494ee31e2603b4019e4ae76e027115f80b818527aaAndreas Huber } 5003475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber} 5103475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber 5203475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas HuberFileSource::FileSource(int fd, int64_t offset, int64_t length) 53674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong : mFd(fd), 5403475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber mOffset(offset), 55dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mLength(length), 56dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDecryptHandle(NULL), 57dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmManagerClient(NULL), 58dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBufOffset(0), 59dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBufSize(0), 60dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBuf(NULL){ 6183b0fd9997b558f6c2ebf5e6e4db20570cb233b8Marco Nelissen ALOGV("fd=%d (%s), offset=%lld, length=%lld", 6283b0fd9997b558f6c2ebf5e6e4db20570cb233b8Marco Nelissen fd, nameForFd(fd).c_str(), (long long) offset, (long long) length); 6383b0fd9997b558f6c2ebf5e6e4db20570cb233b8Marco Nelissen 6403475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber CHECK(offset >= 0); 6503475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber CHECK(length >= 0); 6620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} 6720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 6820111aa043c5f404472bc63b90bc5aad906b1101Andreas HuberFileSource::~FileSource() { 69674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong if (mFd >= 0) { 70674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong close(mFd); 71674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong mFd = -1; 7220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 73dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang 74dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang if (mDrmBuf != NULL) { 75dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang delete[] mDrmBuf; 76dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBuf = NULL; 77dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } 78889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang 79889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang if (mDecryptHandle != NULL) { 80889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang // To release mDecryptHandle 818f64134f749e4f7861a08a3063450fc714c4651dGloria Wang CHECK(mDrmManagerClient); 82889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang mDrmManagerClient->closeDecryptSession(mDecryptHandle); 83889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang mDecryptHandle = NULL; 84889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang } 85889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang 86889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang if (mDrmManagerClient != NULL) { 87889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang delete mDrmManagerClient; 88889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang mDrmManagerClient = NULL; 89889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang } 9020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} 9120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 9234769bc913e9f6bb138e666d94a9d685bf3da217Andreas Huberstatus_t FileSource::initCheck() const { 93674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong return mFd >= 0 ? OK : NO_INIT; 9420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} 9520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 96c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dongssize_t FileSource::readAt(off64_t offset, void *data, size_t size) { 97674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong if (mFd < 0) { 983d8055ac96170de7a8c49b9ffbe83ed447ccddceAndreas Huber return NO_INIT; 993d8055ac96170de7a8c49b9ffbe83ed447ccddceAndreas Huber } 1003d8055ac96170de7a8c49b9ffbe83ed447ccddceAndreas Huber 10120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber Mutex::Autolock autoLock(mLock); 10220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 10303475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber if (mLength >= 0) { 10403475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber if (offset >= mLength) { 10503475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber return 0; // read beyond EOF. 10603475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber } 1070eaf820756bf584284f246b6ba79968efbd08fafWei Jia uint64_t numAvailable = mLength - offset; 1080eaf820756bf584284f246b6ba79968efbd08fafWei Jia if ((uint64_t)size > numAvailable) { 10903475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber size = numAvailable; 11003475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber } 11103475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber } 11203475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber 113dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang if (mDecryptHandle != NULL && DecryptApiType::CONTAINER_BASED 114dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang == mDecryptHandle->decryptApiType) { 115dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang return readAtDRM(offset, data, size); 116dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } else { 117c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong off64_t result = lseek64(mFd, offset + mOffset, SEEK_SET); 118c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong if (result == -1) { 119ee4e1b1a63758941460ae79a064249d3a5189443Lajos Molnar ALOGE("seek to %lld failed", (long long)(offset + mOffset)); 120dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang return UNKNOWN_ERROR; 121dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } 12220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 123c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong return ::read(mFd, data, size); 124dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } 12503475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber} 12603475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber 127c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dongstatus_t FileSource::getSize(off64_t *size) { 1283a13fad63af40a8364fce796b1a54a8f0a2fbf32Andreas Huber Mutex::Autolock autoLock(mLock); 1293a13fad63af40a8364fce796b1a54a8f0a2fbf32Andreas Huber 130674ebd0b4e1143e38392a4e3bb38b4679a4577bcJames Dong if (mFd < 0) { 1313d8055ac96170de7a8c49b9ffbe83ed447ccddceAndreas Huber return NO_INIT; 1323d8055ac96170de7a8c49b9ffbe83ed447ccddceAndreas Huber } 1333d8055ac96170de7a8c49b9ffbe83ed447ccddceAndreas Huber 1344ee31e2603b4019e4ae76e027115f80b818527aaAndreas Huber *size = mLength; 13520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 13603475f5af25e4aa32d46d41dd2ec756853d8dd9dAndreas Huber return OK; 13720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} 13820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 1399d2f386dd2885eaffa11fd494ae258bb09fe6397James Dongsp<DecryptHandle> FileSource::DrmInitialization(const char *mime) { 140889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang if (mDrmManagerClient == NULL) { 141889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang mDrmManagerClient = new DrmManagerClient(); 142889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang } 143889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang 144889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang if (mDrmManagerClient == NULL) { 145b371426ce4cf2fa6d8c3d1903b61322feb165d35Gloria Wang return NULL; 146b371426ce4cf2fa6d8c3d1903b61322feb165d35Gloria Wang } 147b371426ce4cf2fa6d8c3d1903b61322feb165d35Gloria Wang 148dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang if (mDecryptHandle == NULL) { 149dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDecryptHandle = mDrmManagerClient->openDecryptSession( 1509d2f386dd2885eaffa11fd494ae258bb09fe6397James Dong mFd, mOffset, mLength, mime); 151dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } 152dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang 153dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang if (mDecryptHandle == NULL) { 154889b340ec736a9d3e3e690256d305cc8740f0c4bGloria Wang delete mDrmManagerClient; 155dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmManagerClient = NULL; 156dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } 157dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang 158dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang return mDecryptHandle; 159dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang} 160dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang 161b5ce361d19e69fe156f7188c9ee0f4734b259874Gloria Wangvoid FileSource::getDrmInfo(sp<DecryptHandle> &handle, DrmManagerClient **client) { 162b5ce361d19e69fe156f7188c9ee0f4734b259874Gloria Wang handle = mDecryptHandle; 163dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang 164dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang *client = mDrmManagerClient; 165dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang} 166dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang 167c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dongssize_t FileSource::readAtDRM(off64_t offset, void *data, size_t size) { 168dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang size_t DRM_CACHE_SIZE = 1024; 169dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang if (mDrmBuf == NULL) { 170dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBuf = new unsigned char[DRM_CACHE_SIZE]; 171dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } 172dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang 173dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang if (mDrmBuf != NULL && mDrmBufSize > 0 && (offset + mOffset) >= mDrmBufOffset 174dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang && (offset + mOffset + size) <= (mDrmBufOffset + mDrmBufSize)) { 175dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang /* Use buffered data */ 176dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang memcpy(data, (void*)(mDrmBuf+(offset+mOffset-mDrmBufOffset)), size); 177dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang return size; 178dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } else if (size <= DRM_CACHE_SIZE) { 179dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang /* Buffer new data */ 180dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBufOffset = offset + mOffset; 181dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang mDrmBufSize = mDrmManagerClient->pread(mDecryptHandle, mDrmBuf, 182dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang DRM_CACHE_SIZE, offset + mOffset); 183dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang if (mDrmBufSize > 0) { 184dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang int64_t dataRead = 0; 185dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang dataRead = size > mDrmBufSize ? mDrmBufSize : size; 186dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang memcpy(data, (void*)mDrmBuf, dataRead); 187dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang return dataRead; 188dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } else { 189dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang return mDrmBufSize; 190dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } 191dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } else { 192dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang /* Too big chunk to cache. Call DRM directly */ 193dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang return mDrmManagerClient->pread(mDecryptHandle, data, size, offset + mOffset); 194dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang } 195dcd25efb46c41c8d24a0a9cf61fb57f84149709eGloria Wang} 19620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} // namespace android 197