MetadataRetrieverClient.cpp revision 6b3359d4669f539c173bfc40809deb8ffde3dd0c 
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 <media/PVMetadataRetriever.h>
39#include <private/media/VideoFrame.h>
40#include "VorbisMetadataRetriever.h"
41#include "MidiMetadataRetriever.h"
42#include "MetadataRetrieverClient.h"
43#include "StagefrightMetadataRetriever.h"
44
45/* desktop Linux needs a little help with gettid() */
46#if defined(HAVE_GETTID) && !defined(HAVE_ANDROID_OS)
47#define __KERNEL__
48# include <linux/unistd.h>
49#ifdef _syscall0
50_syscall0(pid_t,gettid)
51#else
52pid_t gettid() { return syscall(__NR_gettid);}
53#endif
54#undef __KERNEL__
55#endif
56
57namespace android {
58
59extern player_type getPlayerType(const char* url);
60extern player_type getPlayerType(int fd, int64_t offset, int64_t length);
61
62MetadataRetrieverClient::MetadataRetrieverClient(pid_t pid)
63{
64    LOGV("MetadataRetrieverClient constructor pid(%d)", pid);
65    mPid = pid;
66    mThumbnail = NULL;
67    mAlbumArt = NULL;
68    mRetriever = NULL;
69    mMode = METADATA_MODE_FRAME_CAPTURE_AND_METADATA_RETRIEVAL;
70}
71
72MetadataRetrieverClient::~MetadataRetrieverClient()
73{
74    LOGV("MetadataRetrieverClient destructor");
75    disconnect();
76}
77
78status_t MetadataRetrieverClient::dump(int fd, const Vector<String16>& args) const
79{
80    const size_t SIZE = 256;
81    char buffer[SIZE];
82    String8 result;
83    result.append(" MetadataRetrieverClient\n");
84    snprintf(buffer, 255, "  pid(%d) mode(%d)\n", mPid, mMode);
85    result.append(buffer);
86    write(fd, result.string(), result.size());
87    write(fd, "\n", 1);
88    return NO_ERROR;
89}
90
91void MetadataRetrieverClient::disconnect()
92{
93    LOGV("disconnect from pid %d", mPid);
94    Mutex::Autolock lock(mLock);
95    mRetriever.clear();
96    mThumbnail.clear();
97    mAlbumArt.clear();
98    mMode = METADATA_MODE_FRAME_CAPTURE_AND_METADATA_RETRIEVAL;
99    IPCThreadState::self()->flushCommands();
100}
101
102static sp<MediaMetadataRetrieverBase> createRetriever(player_type playerType)
103{
104    sp<MediaMetadataRetrieverBase> p;
105    switch (playerType) {
106#if BUILD_WITH_FULL_STAGEFRIGHT
107        case STAGEFRIGHT_PLAYER:
108        {
109            char value[PROPERTY_VALUE_MAX];
110            if (property_get("media.stagefright.enable-meta", value, NULL)
111                && (!strcmp(value, "1") || !strcasecmp(value, "true"))) {
112                LOGV("create StagefrightMetadataRetriever");
113                p = new StagefrightMetadataRetriever;
114                break;
115            }
116
117            // fall through
118        }
119#endif
120#ifndef NO_OPENCORE
121        case PV_PLAYER:
122            LOGV("create pv metadata retriever");
123            p = new PVMetadataRetriever();
124            break;
125#endif
126        case VORBIS_PLAYER:
127            LOGV("create vorbis metadata retriever");
128            p = new VorbisMetadataRetriever();
129            break;
130        case SONIVOX_PLAYER:
131            LOGV("create midi metadata retriever");
132            p = new MidiMetadataRetriever();
133            break;
134        default:
135            // TODO:
136            // support for TEST_PLAYER
137            LOGE("player type %d is not supported",  playerType);
138            break;
139    }
140    if (p == NULL) {
141        LOGE("failed to create a retriever object");
142    }
143    return p;
144}
145
146status_t MetadataRetrieverClient::setDataSource(const char *url)
147{
148    LOGV("setDataSource(%s)", url);
149    Mutex::Autolock lock(mLock);
150    if (url == NULL) {
151        return UNKNOWN_ERROR;
152    }
153    player_type playerType = getPlayerType(url);
154    LOGV("player type = %d", playerType);
155    sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
156    if (p == NULL) return NO_INIT;
157    status_t ret = p->setMode(mMode);
158    if (ret == NO_ERROR) {
159        ret = p->setDataSource(url);
160    }
161    if (ret == NO_ERROR) mRetriever = p;
162    return ret;
163}
164
165status_t MetadataRetrieverClient::setDataSource(int fd, int64_t offset, int64_t length)
166{
167    LOGV("setDataSource fd=%d, offset=%lld, length=%lld", fd, offset, length);
168    Mutex::Autolock lock(mLock);
169    struct stat sb;
170    int ret = fstat(fd, &sb);
171    if (ret != 0) {
172        LOGE("fstat(%d) failed: %d, %s", fd, ret, strerror(errno));
173        return BAD_VALUE;
174    }
175    LOGV("st_dev  = %llu", sb.st_dev);
176    LOGV("st_mode = %u", sb.st_mode);
177    LOGV("st_uid  = %lu", sb.st_uid);
178    LOGV("st_gid  = %lu", sb.st_gid);
179    LOGV("st_size = %llu", sb.st_size);
180
181    if (offset >= sb.st_size) {
182        LOGE("offset (%lld) bigger than file size (%llu)", offset, sb.st_size);
183        ::close(fd);
184        return BAD_VALUE;
185    }
186    if (offset + length > sb.st_size) {
187        length = sb.st_size - offset;
188        LOGV("calculated length = %lld", length);
189    }
190
191    player_type playerType = getPlayerType(fd, offset, length);
192    LOGV("player type = %d", playerType);
193    sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
194    if (p == NULL) {
195        ::close(fd);
196        return NO_INIT;
197    }
198    status_t status = p->setMode(mMode);
199    if (status == NO_ERROR) {
200        p->setDataSource(fd, offset, length);
201    }
202    if (status == NO_ERROR) mRetriever = p;
203    ::close(fd);
204    return status;
205}
206
207status_t MetadataRetrieverClient::setMode(int mode)
208{
209    LOGV("setMode");
210    Mutex::Autolock lock(mLock);
211    if (mode < METADATA_MODE_NOOP ||
212        mode > METADATA_MODE_FRAME_CAPTURE_AND_METADATA_RETRIEVAL) {
213        LOGE("invalid mode %d", mode);
214        return BAD_VALUE;
215    }
216    mMode = mode;
217    return NO_ERROR;
218}
219
220status_t MetadataRetrieverClient::getMode(int* mode) const
221{
222    LOGV("getMode");
223    Mutex::Autolock lock(mLock);
224
225    // TODO:
226    // This may not be necessary.
227    // If setDataSource() has not been called, return the cached value
228    // otherwise, return the value retrieved from the retriever
229    if (mRetriever == NULL) {
230        *mode = mMode;
231    } else {
232        mRetriever->getMode(mode);
233    }
234    return NO_ERROR;
235}
236
237sp<IMemory> MetadataRetrieverClient::captureFrame()
238{
239    LOGV("captureFrame");
240    Mutex::Autolock lock(mLock);
241    mThumbnail.clear();
242    if (mRetriever == NULL) {
243        LOGE("retriever is not initialized");
244        return NULL;
245    }
246    VideoFrame *frame = mRetriever->captureFrame();
247    if (frame == NULL) {
248        LOGE("failed to capture a video frame");
249        return NULL;
250    }
251    size_t size = sizeof(VideoFrame) + frame->mSize;
252    sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient");
253    if (heap == NULL) {
254        LOGE("failed to create MemoryDealer");
255        delete frame;
256        return NULL;
257    }
258    mThumbnail = new MemoryBase(heap, 0, size);
259    if (mThumbnail == NULL) {
260        LOGE("not enough memory for VideoFrame size=%u", size);
261        delete frame;
262        return NULL;
263    }
264    VideoFrame *frameCopy = static_cast<VideoFrame *>(mThumbnail->pointer());
265    frameCopy->mWidth = frame->mWidth;
266    frameCopy->mHeight = frame->mHeight;
267    frameCopy->mDisplayWidth = frame->mDisplayWidth;
268    frameCopy->mDisplayHeight = frame->mDisplayHeight;
269    frameCopy->mSize = frame->mSize;
270    frameCopy->mData = (uint8_t *)frameCopy + sizeof(VideoFrame);
271    memcpy(frameCopy->mData, frame->mData, frame->mSize);
272    delete frame;  // Fix memory leakage
273    return mThumbnail;
274}
275
276sp<IMemory> MetadataRetrieverClient::extractAlbumArt()
277{
278    LOGV("extractAlbumArt");
279    Mutex::Autolock lock(mLock);
280    mAlbumArt.clear();
281    if (mRetriever == NULL) {
282        LOGE("retriever is not initialized");
283        return NULL;
284    }
285    MediaAlbumArt *albumArt = mRetriever->extractAlbumArt();
286    if (albumArt == NULL) {
287        LOGE("failed to extract an album art");
288        return NULL;
289    }
290    size_t size = sizeof(MediaAlbumArt) + albumArt->mSize;
291    sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient");
292    if (heap == NULL) {
293        LOGE("failed to create MemoryDealer object");
294        delete albumArt;
295        return NULL;
296    }
297    mAlbumArt = new MemoryBase(heap, 0, size);
298    if (mAlbumArt == NULL) {
299        LOGE("not enough memory for MediaAlbumArt size=%u", size);
300        delete albumArt;
301        return NULL;
302    }
303    MediaAlbumArt *albumArtCopy = static_cast<MediaAlbumArt *>(mAlbumArt->pointer());
304    albumArtCopy->mSize = albumArt->mSize;
305    albumArtCopy->mData = (uint8_t *)albumArtCopy + sizeof(MediaAlbumArt);
306    memcpy(albumArtCopy->mData, albumArt->mData, albumArt->mSize);
307    delete albumArt;  // Fix memory leakage
308    return mAlbumArt;
309}
310
311const char* MetadataRetrieverClient::extractMetadata(int keyCode)
312{
313    LOGV("extractMetadata");
314    Mutex::Autolock lock(mLock);
315    if (mRetriever == NULL) {
316        LOGE("retriever is not initialized");
317        return NULL;
318    }
319    return mRetriever->extractMetadata(keyCode);
320}
321
322}; // namespace android
323