MetadataRetrieverClient.cpp revision 0512ab559d4670c2204078470d7ef5d376811c57 
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 <sys/resource.h>
25#include <dirent.h>
26#include <unistd.h>
27
28#include <string.h>
29#include <cutils/atomic.h>
30#include <cutils/properties.h>
31#include <binder/MemoryBase.h>
32#include <binder/MemoryHeapBase.h>
33#include <android_runtime/ActivityManager.h>
34#include <binder/IPCThreadState.h>
35#include <binder/IServiceManager.h>
36#include <media/MediaMetadataRetrieverInterface.h>
37#include <media/MediaPlayerInterface.h>
38#include <private/media/VideoFrame.h>
39#include "MidiMetadataRetriever.h"
40#include "MetadataRetrieverClient.h"
41#include "StagefrightMetadataRetriever.h"
42
43namespace android {
44
45extern player_type getPlayerType(const char* url);
46extern player_type getPlayerType(int fd, int64_t offset, int64_t length);
47
48MetadataRetrieverClient::MetadataRetrieverClient(pid_t pid)
49{
50    LOGV("MetadataRetrieverClient constructor pid(%d)", pid);
51    mPid = pid;
52    mThumbnail = NULL;
53    mAlbumArt = NULL;
54    mRetriever = NULL;
55}
56
57MetadataRetrieverClient::~MetadataRetrieverClient()
58{
59    LOGV("MetadataRetrieverClient destructor");
60    disconnect();
61}
62
63status_t MetadataRetrieverClient::dump(int fd, const Vector<String16>& args) const
64{
65    const size_t SIZE = 256;
66    char buffer[SIZE];
67    String8 result;
68    result.append(" MetadataRetrieverClient\n");
69    snprintf(buffer, 255, "  pid(%d)\n", mPid);
70    result.append(buffer);
71    write(fd, result.string(), result.size());
72    write(fd, "\n", 1);
73    return NO_ERROR;
74}
75
76void MetadataRetrieverClient::disconnect()
77{
78    LOGV("disconnect from pid %d", mPid);
79    Mutex::Autolock lock(mLock);
80    mRetriever.clear();
81    mThumbnail.clear();
82    mAlbumArt.clear();
83    IPCThreadState::self()->flushCommands();
84}
85
86static sp<MediaMetadataRetrieverBase> createRetriever(player_type playerType)
87{
88    sp<MediaMetadataRetrieverBase> p;
89    switch (playerType) {
90        case STAGEFRIGHT_PLAYER:
91        {
92            p = new StagefrightMetadataRetriever;
93            break;
94        }
95        case SONIVOX_PLAYER:
96            LOGV("create midi metadata retriever");
97            p = new MidiMetadataRetriever();
98            break;
99        default:
100            // TODO:
101            // support for TEST_PLAYER
102            LOGE("player type %d is not supported",  playerType);
103            break;
104    }
105    if (p == NULL) {
106        LOGE("failed to create a retriever object");
107    }
108    return p;
109}
110
111status_t MetadataRetrieverClient::setDataSource(
112        const char *url, const KeyedVector<String8, String8> *headers)
113{
114    LOGV("setDataSource(%s)", url);
115    Mutex::Autolock lock(mLock);
116    if (url == NULL) {
117        return UNKNOWN_ERROR;
118    }
119    player_type playerType = getPlayerType(url);
120    LOGV("player type = %d", playerType);
121    sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
122    if (p == NULL) return NO_INIT;
123    status_t ret = p->setDataSource(url, headers);
124    if (ret == NO_ERROR) mRetriever = p;
125    return ret;
126}
127
128status_t MetadataRetrieverClient::setDataSource(int fd, int64_t offset, int64_t length)
129{
130    LOGV("setDataSource fd=%d, offset=%lld, length=%lld", fd, offset, length);
131    Mutex::Autolock lock(mLock);
132    struct stat sb;
133    int ret = fstat(fd, &sb);
134    if (ret != 0) {
135        LOGE("fstat(%d) failed: %d, %s", fd, ret, strerror(errno));
136        return BAD_VALUE;
137    }
138    LOGV("st_dev  = %llu", sb.st_dev);
139    LOGV("st_mode = %u", sb.st_mode);
140    LOGV("st_uid  = %lu", sb.st_uid);
141    LOGV("st_gid  = %lu", sb.st_gid);
142    LOGV("st_size = %llu", sb.st_size);
143
144    if (offset >= sb.st_size) {
145        LOGE("offset (%lld) bigger than file size (%llu)", offset, sb.st_size);
146        ::close(fd);
147        return BAD_VALUE;
148    }
149    if (offset + length > sb.st_size) {
150        length = sb.st_size - offset;
151        LOGV("calculated length = %lld", length);
152    }
153
154    player_type playerType = getPlayerType(fd, offset, length);
155    LOGV("player type = %d", playerType);
156    sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
157    if (p == NULL) {
158        ::close(fd);
159        return NO_INIT;
160    }
161    status_t status = p->setDataSource(fd, offset, length);
162    if (status == NO_ERROR) mRetriever = p;
163    ::close(fd);
164    return status;
165}
166
167sp<IMemory> MetadataRetrieverClient::getFrameAtTime(int64_t timeUs, int option)
168{
169    LOGV("getFrameAtTime: time(%lld us) option(%d)", timeUs, option);
170    Mutex::Autolock lock(mLock);
171    mThumbnail.clear();
172    if (mRetriever == NULL) {
173        LOGE("retriever is not initialized");
174        return NULL;
175    }
176    VideoFrame *frame = mRetriever->getFrameAtTime(timeUs, option);
177    if (frame == NULL) {
178        LOGE("failed to capture a video frame");
179        return NULL;
180    }
181    size_t size = sizeof(VideoFrame) + frame->mSize;
182    sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient");
183    if (heap == NULL) {
184        LOGE("failed to create MemoryDealer");
185        delete frame;
186        return NULL;
187    }
188    mThumbnail = new MemoryBase(heap, 0, size);
189    if (mThumbnail == NULL) {
190        LOGE("not enough memory for VideoFrame size=%u", size);
191        delete frame;
192        return NULL;
193    }
194    VideoFrame *frameCopy = static_cast<VideoFrame *>(mThumbnail->pointer());
195    frameCopy->mWidth = frame->mWidth;
196    frameCopy->mHeight = frame->mHeight;
197    frameCopy->mDisplayWidth = frame->mDisplayWidth;
198    frameCopy->mDisplayHeight = frame->mDisplayHeight;
199    frameCopy->mSize = frame->mSize;
200    frameCopy->mRotationAngle = frame->mRotationAngle;
201    LOGV("rotation: %d", frameCopy->mRotationAngle);
202    frameCopy->mData = (uint8_t *)frameCopy + sizeof(VideoFrame);
203    memcpy(frameCopy->mData, frame->mData, frame->mSize);
204    delete frame;  // Fix memory leakage
205    return mThumbnail;
206}
207
208sp<IMemory> MetadataRetrieverClient::extractAlbumArt()
209{
210    LOGV("extractAlbumArt");
211    Mutex::Autolock lock(mLock);
212    mAlbumArt.clear();
213    if (mRetriever == NULL) {
214        LOGE("retriever is not initialized");
215        return NULL;
216    }
217    MediaAlbumArt *albumArt = mRetriever->extractAlbumArt();
218    if (albumArt == NULL) {
219        LOGE("failed to extract an album art");
220        return NULL;
221    }
222    size_t size = sizeof(MediaAlbumArt) + albumArt->mSize;
223    sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient");
224    if (heap == NULL) {
225        LOGE("failed to create MemoryDealer object");
226        delete albumArt;
227        return NULL;
228    }
229    mAlbumArt = new MemoryBase(heap, 0, size);
230    if (mAlbumArt == NULL) {
231        LOGE("not enough memory for MediaAlbumArt size=%u", size);
232        delete albumArt;
233        return NULL;
234    }
235    MediaAlbumArt *albumArtCopy = static_cast<MediaAlbumArt *>(mAlbumArt->pointer());
236    albumArtCopy->mSize = albumArt->mSize;
237    albumArtCopy->mData = (uint8_t *)albumArtCopy + sizeof(MediaAlbumArt);
238    memcpy(albumArtCopy->mData, albumArt->mData, albumArt->mSize);
239    delete albumArt;  // Fix memory leakage
240    return mAlbumArt;
241}
242
243const char* MetadataRetrieverClient::extractMetadata(int keyCode)
244{
245    LOGV("extractMetadata");
246    Mutex::Autolock lock(mLock);
247    if (mRetriever == NULL) {
248        LOGE("retriever is not initialized");
249        return NULL;
250    }
251    return mRetriever->extractMetadata(keyCode);
252}
253
254}; // namespace android
255