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