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