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