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