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