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