MetadataRetrieverClient.cpp revision 83b0fd9997b558f6c2ebf5e6e4db20570cb233b8 
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 <cutils/properties.h>
30#include <binder/MemoryBase.h>
31#include <binder/MemoryHeapBase.h>
32#include <binder/IPCThreadState.h>
33#include <binder/IServiceManager.h>
34#include <media/IMediaHTTPService.h>
35#include <media/MediaMetadataRetrieverInterface.h>
36#include <media/MediaPlayerInterface.h>
37#include <media/stagefright/DataSource.h>
38#include <media/stagefright/Utils.h>
39#include <private/media/VideoFrame.h>
40#include "MetadataRetrieverClient.h"
41#include "StagefrightMetadataRetriever.h"
42#include "MediaPlayerFactory.h"
43
44namespace android {
45
46MetadataRetrieverClient::MetadataRetrieverClient(pid_t pid)
47{
48    ALOGV("MetadataRetrieverClient constructor pid(%d)", pid);
49    mPid = pid;
50    mThumbnail = NULL;
51    mAlbumArt = NULL;
52    mRetriever = NULL;
53}
54
55MetadataRetrieverClient::~MetadataRetrieverClient()
56{
57    ALOGV("MetadataRetrieverClient destructor");
58    disconnect();
59}
60
61status_t MetadataRetrieverClient::dump(int fd, const Vector<String16>& /*args*/)
62{
63    const size_t SIZE = 256;
64    char buffer[SIZE];
65    String8 result;
66    result.append(" MetadataRetrieverClient\n");
67    snprintf(buffer, 255, "  pid(%d)\n", mPid);
68    result.append(buffer);
69    write(fd, result.string(), result.size());
70    write(fd, "\n", 1);
71    return NO_ERROR;
72}
73
74void MetadataRetrieverClient::disconnect()
75{
76    ALOGV("disconnect from pid %d", mPid);
77    Mutex::Autolock lock(mLock);
78    mRetriever.clear();
79    mThumbnail.clear();
80    mAlbumArt.clear();
81    IPCThreadState::self()->flushCommands();
82}
83
84static sp<MediaMetadataRetrieverBase> createRetriever(player_type playerType)
85{
86    sp<MediaMetadataRetrieverBase> p;
87    switch (playerType) {
88        case STAGEFRIGHT_PLAYER:
89        case NU_PLAYER:
90        {
91            p = new StagefrightMetadataRetriever;
92            break;
93        }
94        default:
95            // TODO:
96            // support for TEST_PLAYER
97            ALOGE("player type %d is not supported",  playerType);
98            break;
99    }
100    if (p == NULL) {
101        ALOGE("failed to create a retriever object");
102    }
103    return p;
104}
105
106status_t MetadataRetrieverClient::setDataSource(
107        const sp<IMediaHTTPService> &httpService,
108        const char *url,
109        const KeyedVector<String8, String8> *headers)
110{
111    ALOGV("setDataSource(%s)", url);
112    Mutex::Autolock lock(mLock);
113    if (url == NULL) {
114        return UNKNOWN_ERROR;
115    }
116
117    // When asking the MediaPlayerFactory subsystem to choose a media player for
118    // a given URL, a pointer to an outer IMediaPlayer can be passed to the
119    // factory system to be taken into consideration along with the URL.  In the
120    // case of choosing an instance of a MediaPlayerBase for a
121    // MetadataRetrieverClient, there is no outer IMediaPlayer which will
122    // eventually encapsulate the result of this selection.  In this case, just
123    // pass NULL to getPlayerType to indicate that there is no outer
124    // IMediaPlayer to consider during selection.
125    player_type playerType =
126        MediaPlayerFactory::getPlayerType(NULL /* client */, url);
127    ALOGV("player type = %d", playerType);
128    sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
129    if (p == NULL) return NO_INIT;
130    status_t ret = p->setDataSource(httpService, url, headers);
131    if (ret == NO_ERROR) mRetriever = p;
132    return ret;
133}
134
135status_t MetadataRetrieverClient::setDataSource(int fd, int64_t offset, int64_t length)
136{
137    ALOGV("setDataSource fd=%d (%s), offset=%lld, length=%lld",
138            fd, nameForFd(fd).c_str(), (long long) offset, (long long) length);
139    Mutex::Autolock lock(mLock);
140    struct stat sb;
141    int ret = fstat(fd, &sb);
142    if (ret != 0) {
143        ALOGE("fstat(%d) failed: %d, %s", fd, ret, strerror(errno));
144        return BAD_VALUE;
145    }
146    ALOGV("st_dev  = %llu", static_cast<uint64_t>(sb.st_dev));
147    ALOGV("st_mode = %u", sb.st_mode);
148    ALOGV("st_uid  = %lu", static_cast<unsigned long>(sb.st_uid));
149    ALOGV("st_gid  = %lu", static_cast<unsigned long>(sb.st_gid));
150    ALOGV("st_size = %llu", sb.st_size);
151
152    if (offset >= sb.st_size) {
153        ALOGE("offset (%lld) bigger than file size (%llu)", offset, sb.st_size);
154        return BAD_VALUE;
155    }
156    if (offset + length > sb.st_size) {
157        length = sb.st_size - offset;
158        ALOGV("calculated length = %lld", length);
159    }
160
161    player_type playerType =
162        MediaPlayerFactory::getPlayerType(NULL /* client */,
163                                          fd,
164                                          offset,
165                                          length);
166    ALOGV("player type = %d", playerType);
167    sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
168    if (p == NULL) {
169        return NO_INIT;
170    }
171    status_t status = p->setDataSource(fd, offset, length);
172    if (status == NO_ERROR) mRetriever = p;
173    return status;
174}
175
176status_t MetadataRetrieverClient::setDataSource(
177        const sp<IDataSource>& source)
178{
179    ALOGV("setDataSource(IDataSource)");
180    Mutex::Autolock lock(mLock);
181
182    sp<DataSource> dataSource = DataSource::CreateFromIDataSource(source);
183    player_type playerType =
184        MediaPlayerFactory::getPlayerType(NULL /* client */, dataSource);
185    ALOGV("player type = %d", playerType);
186    sp<MediaMetadataRetrieverBase> p = createRetriever(playerType);
187    if (p == NULL) return NO_INIT;
188    status_t ret = p->setDataSource(dataSource);
189    if (ret == NO_ERROR) mRetriever = p;
190    return ret;
191}
192
193Mutex MetadataRetrieverClient::sLock;
194
195sp<IMemory> MetadataRetrieverClient::getFrameAtTime(int64_t timeUs, int option)
196{
197    ALOGV("getFrameAtTime: time(%lld us) option(%d)", timeUs, option);
198    Mutex::Autolock lock(mLock);
199    Mutex::Autolock glock(sLock);
200    mThumbnail.clear();
201    if (mRetriever == NULL) {
202        ALOGE("retriever is not initialized");
203        return NULL;
204    }
205    VideoFrame *frame = mRetriever->getFrameAtTime(timeUs, option);
206    if (frame == NULL) {
207        ALOGE("failed to capture a video frame");
208        return NULL;
209    }
210    size_t size = sizeof(VideoFrame) + frame->mSize;
211    sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient");
212    if (heap == NULL) {
213        ALOGE("failed to create MemoryDealer");
214        delete frame;
215        return NULL;
216    }
217    mThumbnail = new MemoryBase(heap, 0, size);
218    if (mThumbnail == NULL) {
219        ALOGE("not enough memory for VideoFrame size=%u", size);
220        delete frame;
221        return NULL;
222    }
223    VideoFrame *frameCopy = static_cast<VideoFrame *>(mThumbnail->pointer());
224    frameCopy->mWidth = frame->mWidth;
225    frameCopy->mHeight = frame->mHeight;
226    frameCopy->mDisplayWidth = frame->mDisplayWidth;
227    frameCopy->mDisplayHeight = frame->mDisplayHeight;
228    frameCopy->mSize = frame->mSize;
229    frameCopy->mRotationAngle = frame->mRotationAngle;
230    ALOGV("rotation: %d", frameCopy->mRotationAngle);
231    frameCopy->mData = (uint8_t *)frameCopy + sizeof(VideoFrame);
232    memcpy(frameCopy->mData, frame->mData, frame->mSize);
233    delete frame;  // Fix memory leakage
234    return mThumbnail;
235}
236
237sp<IMemory> MetadataRetrieverClient::extractAlbumArt()
238{
239    ALOGV("extractAlbumArt");
240    Mutex::Autolock lock(mLock);
241    mAlbumArt.clear();
242    if (mRetriever == NULL) {
243        ALOGE("retriever is not initialized");
244        return NULL;
245    }
246    MediaAlbumArt *albumArt = mRetriever->extractAlbumArt();
247    if (albumArt == NULL) {
248        ALOGE("failed to extract an album art");
249        return NULL;
250    }
251    size_t size = sizeof(MediaAlbumArt) + albumArt->size();
252    sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient");
253    if (heap == NULL) {
254        ALOGE("failed to create MemoryDealer object");
255        delete albumArt;
256        return NULL;
257    }
258    mAlbumArt = new MemoryBase(heap, 0, size);
259    if (mAlbumArt == NULL) {
260        ALOGE("not enough memory for MediaAlbumArt size=%u", size);
261        delete albumArt;
262        return NULL;
263    }
264    MediaAlbumArt::init((MediaAlbumArt *) mAlbumArt->pointer(),
265                        albumArt->size(), albumArt->data());
266    delete albumArt;  // We've taken our copy.
267    return mAlbumArt;
268}
269
270const char* MetadataRetrieverClient::extractMetadata(int keyCode)
271{
272    ALOGV("extractMetadata");
273    Mutex::Autolock lock(mLock);
274    if (mRetriever == NULL) {
275        ALOGE("retriever is not initialized");
276        return NULL;
277    }
278    return mRetriever->extractMetadata(keyCode);
279}
280
281}; // namespace android
282