Home | History | Annotate | Download | only in libmediaplayerservice
      1 /*
      2 **
      3 ** Copyright (C) 2008 The Android Open Source Project
      4 **
      5 ** Licensed under the Apache License, Version 2.0 (the "License");
      6 ** you may not use this file except in compliance with the License.
      7 ** You may obtain a copy of the License at
      8 **
      9 **     http://www.apache.org/licenses/LICENSE-2.0
     10 **
     11 ** Unless required by applicable law or agreed to in writing, software
     12 ** distributed under the License is distributed on an "AS IS" BASIS,
     13 ** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     14 ** See the License for the specific language governing permissions and
     15 ** limitations under the License.
     16 */
     17 
     18 //#define LOG_NDEBUG 0
     19 #define LOG_TAG "MetadataRetrieverClient"
     20 #include <utils/Log.h>
     21 
     22 #include <sys/types.h>
     23 #include <sys/stat.h>
     24 #include <dirent.h>
     25 #include <unistd.h>
     26 
     27 #include <string.h>
     28 #include <cutils/atomic.h>
     29 #include <cutils/properties.h>
     30 #include <binder/MemoryBase.h>
     31 #include <binder/MemoryHeapBase.h>
     32 #include <binder/IPCThreadState.h>
     33 #include <binder/IServiceManager.h>
     34 #include <media/IMediaHTTPService.h>
     35 #include <media/MediaMetadataRetrieverInterface.h>
     36 #include <media/MediaPlayerInterface.h>
     37 #include <media/stagefright/DataSource.h>
     38 #include <media/stagefright/Utils.h>
     39 #include <private/media/VideoFrame.h>
     40 #include "MetadataRetrieverClient.h"
     41 #include "StagefrightMetadataRetriever.h"
     42 #include "MediaPlayerFactory.h"
     43 
     44 namespace android {
     45 
     46 MetadataRetrieverClient::MetadataRetrieverClient(pid_t pid)
     47 {
     48     ALOGV("MetadataRetrieverClient constructor pid(%d)", pid);
     49     mPid = pid;
     50     mThumbnail = NULL;
     51     mAlbumArt = NULL;
     52     mRetriever = NULL;
     53 }
     54 
     55 MetadataRetrieverClient::~MetadataRetrieverClient()
     56 {
     57     ALOGV("MetadataRetrieverClient destructor");
     58     disconnect();
     59 }
     60 
     61 status_t MetadataRetrieverClient::dump(int fd, const Vector<String16>& /*args*/)
     62 {
     63     const size_t SIZE = 256;
     64     char buffer[SIZE];
     65     String8 result;
     66     result.append(" MetadataRetrieverClient\n");
     67     snprintf(buffer, 255, "  pid(%d)\n", mPid);
     68     result.append(buffer);
     69     write(fd, result.string(), result.size());
     70     write(fd, "\n", 1);
     71     return NO_ERROR;
     72 }
     73 
     74 void MetadataRetrieverClient::disconnect()
     75 {
     76     ALOGV("disconnect from pid %d", mPid);
     77     Mutex::Autolock lock(mLock);
     78     mRetriever.clear();
     79     mThumbnail.clear();
     80     mAlbumArt.clear();
     81     IPCThreadState::self()->flushCommands();
     82 }
     83 
     84 static sp<MediaMetadataRetrieverBase> createRetriever(player_type playerType)
     85 {
     86     sp<MediaMetadataRetrieverBase> p;
     87     switch (playerType) {
     88         case STAGEFRIGHT_PLAYER:
     89         case NU_PLAYER:
     90         {
     91             p = new StagefrightMetadataRetriever;
     92             break;
     93         }
     94         default:
     95             // TODO:
     96             // support for TEST_PLAYER
     97             ALOGE("player type %d is not supported",  playerType);
     98             break;
     99     }
    100     if (p == NULL) {
    101         ALOGE("failed to create a retriever object");
    102     }
    103     return p;
    104 }
    105 
    106 status_t MetadataRetrieverClient::setDataSource(
    107         const sp<IMediaHTTPService> &httpService,
    108         const char *url,
    109         const KeyedVector<String8, String8> *headers)
    110 {
    111     ALOGV("setDataSource(%s)", url);
    112     Mutex::Autolock lock(mLock);
    113     if (url == NULL) {
    114         return UNKNOWN_ERROR;
    115     }
    116 
    117     // When asking the MediaPlayerFactory subsystem to choose a media player for
    118     // a given URL, a pointer to an outer IMediaPlayer can be passed to the
    119     // factory system to be taken into consideration along with the URL.  In the
    120     // case of choosing an instance of a MediaPlayerBase for a
    121     // MetadataRetrieverClient, there is no outer IMediaPlayer which will
    122     // eventually encapsulate the result of this selection.  In this case, just
    123     // pass NULL to getPlayerType to indicate that there is no outer
    124     // IMediaPlayer to consider during selection.
    125     player_type playerType =
    126         MediaPlayerFactory::getPlayerType(NULL /* client */, url);
    127     ALOGV("player type = %d", playerType);
    128     sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
    129     if (p == NULL) return NO_INIT;
    130     status_t ret = p->setDataSource(httpService, url, headers);
    131     if (ret == NO_ERROR) mRetriever = p;
    132     return ret;
    133 }
    134 
    135 status_t MetadataRetrieverClient::setDataSource(int fd, int64_t offset, int64_t length)
    136 {
    137     ALOGV("setDataSource fd=%d (%s), offset=%lld, length=%lld",
    138             fd, nameForFd(fd).c_str(), (long long) offset, (long long) length);
    139     Mutex::Autolock lock(mLock);
    140     struct stat sb;
    141     int ret = fstat(fd, &sb);
    142     if (ret != 0) {
    143         ALOGE("fstat(%d) failed: %d, %s", fd, ret, strerror(errno));
    144         return BAD_VALUE;
    145     }
    146     ALOGV("st_dev  = %llu", static_cast<unsigned long long>(sb.st_dev));
    147     ALOGV("st_mode = %u", sb.st_mode);
    148     ALOGV("st_uid  = %lu", static_cast<unsigned long>(sb.st_uid));
    149     ALOGV("st_gid  = %lu", static_cast<unsigned long>(sb.st_gid));
    150     ALOGV("st_size = %llu", static_cast<unsigned long long>(sb.st_size));
    151 
    152     if (offset >= sb.st_size) {
    153         ALOGE("offset (%lld) bigger than file size (%llu)",
    154                 (long long)offset, (unsigned long long)sb.st_size);
    155         return BAD_VALUE;
    156     }
    157     if (offset + length > sb.st_size) {
    158         length = sb.st_size - offset;
    159         ALOGV("calculated length = %lld", (long long)length);
    160     }
    161 
    162     player_type playerType =
    163         MediaPlayerFactory::getPlayerType(NULL /* client */,
    164                                           fd,
    165                                           offset,
    166                                           length);
    167     ALOGV("player type = %d", playerType);
    168     sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
    169     if (p == NULL) {
    170         return NO_INIT;
    171     }
    172     status_t status = p->setDataSource(fd, offset, length);
    173     if (status == NO_ERROR) mRetriever = p;
    174     return status;
    175 }
    176 
    177 status_t MetadataRetrieverClient::setDataSource(
    178         const sp<IDataSource>& source)
    179 {
    180     ALOGV("setDataSource(IDataSource)");
    181     Mutex::Autolock lock(mLock);
    182 
    183     sp<DataSource> dataSource = DataSource::CreateFromIDataSource(source);
    184     player_type playerType =
    185         MediaPlayerFactory::getPlayerType(NULL /* client */, dataSource);
    186     ALOGV("player type = %d", playerType);
    187     sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
    188     if (p == NULL) return NO_INIT;
    189     status_t ret = p->setDataSource(dataSource);
    190     if (ret == NO_ERROR) mRetriever = p;
    191     return ret;
    192 }
    193 
    194 Mutex MetadataRetrieverClient::sLock;
    195 
    196 sp<IMemory> MetadataRetrieverClient::getFrameAtTime(int64_t timeUs, int option)
    197 {
    198     ALOGV("getFrameAtTime: time(%lld us) option(%d)", (long long)timeUs, option);
    199     Mutex::Autolock lock(mLock);
    200     Mutex::Autolock glock(sLock);
    201     mThumbnail.clear();
    202     if (mRetriever == NULL) {
    203         ALOGE("retriever is not initialized");
    204         return NULL;
    205     }
    206     VideoFrame *frame = mRetriever->getFrameAtTime(timeUs, option);
    207     if (frame == NULL) {
    208         ALOGE("failed to capture a video frame");
    209         return NULL;
    210     }
    211     size_t size = sizeof(VideoFrame) + frame->mSize;
    212     sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient");
    213     if (heap == NULL) {
    214         ALOGE("failed to create MemoryDealer");
    215         delete frame;
    216         return NULL;
    217     }
    218     mThumbnail = new MemoryBase(heap, 0, size);
    219     if (mThumbnail == NULL) {
    220         ALOGE("not enough memory for VideoFrame size=%zu", size);
    221         delete frame;
    222         return NULL;
    223     }
    224     VideoFrame *frameCopy = static_cast<VideoFrame *>(mThumbnail->pointer());
    225     frameCopy->mWidth = frame->mWidth;
    226     frameCopy->mHeight = frame->mHeight;
    227     frameCopy->mDisplayWidth = frame->mDisplayWidth;
    228     frameCopy->mDisplayHeight = frame->mDisplayHeight;
    229     frameCopy->mSize = frame->mSize;
    230     frameCopy->mRotationAngle = frame->mRotationAngle;
    231     ALOGV("rotation: %d", frameCopy->mRotationAngle);
    232     frameCopy->mData = (uint8_t *)frameCopy + sizeof(VideoFrame);
    233     memcpy(frameCopy->mData, frame->mData, frame->mSize);
    234     frameCopy->mData = 0;
    235     delete frame;  // Fix memory leakage
    236     return mThumbnail;
    237 }
    238 
    239 sp<IMemory> MetadataRetrieverClient::extractAlbumArt()
    240 {
    241     ALOGV("extractAlbumArt");
    242     Mutex::Autolock lock(mLock);
    243     mAlbumArt.clear();
    244     if (mRetriever == NULL) {
    245         ALOGE("retriever is not initialized");
    246         return NULL;
    247     }
    248     MediaAlbumArt *albumArt = mRetriever->extractAlbumArt();
    249     if (albumArt == NULL) {
    250         ALOGE("failed to extract an album art");
    251         return NULL;
    252     }
    253     size_t size = sizeof(MediaAlbumArt) + albumArt->size();
    254     sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient");
    255     if (heap == NULL) {
    256         ALOGE("failed to create MemoryDealer object");
    257         delete albumArt;
    258         return NULL;
    259     }
    260     mAlbumArt = new MemoryBase(heap, 0, size);
    261     if (mAlbumArt == NULL) {
    262         ALOGE("not enough memory for MediaAlbumArt size=%zu", size);
    263         delete albumArt;
    264         return NULL;
    265     }
    266     MediaAlbumArt::init((MediaAlbumArt *) mAlbumArt->pointer(),
    267                         albumArt->size(), albumArt->data());
    268     delete albumArt;  // We've taken our copy.
    269     return mAlbumArt;
    270 }
    271 
    272 const char* MetadataRetrieverClient::extractMetadata(int keyCode)
    273 {
    274     ALOGV("extractMetadata");
    275     Mutex::Autolock lock(mLock);
    276     if (mRetriever == NULL) {
    277         ALOGE("retriever is not initialized");
    278         return NULL;
    279     }
    280     return mRetriever->extractMetadata(keyCode);
    281 }
    282 
    283 }; // namespace android
    284