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 44 namespace android { 45 46 MetadataRetrieverClient::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 55 MetadataRetrieverClient::~MetadataRetrieverClient() 56 { 57 ALOGV("MetadataRetrieverClient destructor"); 58 disconnect(); 59 } 60 61 status_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 74 void 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 84 static 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 106 status_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 135 status_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<unsigned long long>(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", static_cast<unsigned long long>(sb.st_size)); 151 152 if (offset >= sb.st_size) { 153 ALOGE("offset (%lld) bigger than file size (%llu)", 154 (long long)offset, (unsigned long long)sb.st_size); 155 return BAD_VALUE; 156 } 157 if (offset + length > sb.st_size) { 158 length = sb.st_size - offset; 159 ALOGV("calculated length = %lld", (long long)length); 160 } 161 162 player_type playerType = 163 MediaPlayerFactory::getPlayerType(NULL /* client */, 164 fd, 165 offset, 166 length); 167 ALOGV("player type = %d", playerType); 168 sp<MediaMetadataRetrieverBase> p = createRetriever(playerType); 169 if (p == NULL) { 170 return NO_INIT; 171 } 172 status_t status = p->setDataSource(fd, offset, length); 173 if (status == NO_ERROR) mRetriever = p; 174 return status; 175 } 176 177 status_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 194 Mutex MetadataRetrieverClient::sLock; 195 196 sp<IMemory> MetadataRetrieverClient::getFrameAtTime(int64_t timeUs, int option) 197 { 198 ALOGV("getFrameAtTime: time(%lld us) option(%d)", (long long)timeUs, option); 199 Mutex::Autolock lock(mLock); 200 Mutex::Autolock glock(sLock); 201 mThumbnail.clear(); 202 if (mRetriever == NULL) { 203 ALOGE("retriever is not initialized"); 204 return NULL; 205 } 206 VideoFrame *frame = mRetriever->getFrameAtTime(timeUs, option); 207 if (frame == NULL) { 208 ALOGE("failed to capture a video frame"); 209 return NULL; 210 } 211 size_t size = sizeof(VideoFrame) + frame->mSize; 212 sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient"); 213 if (heap == NULL) { 214 ALOGE("failed to create MemoryDealer"); 215 delete frame; 216 return NULL; 217 } 218 mThumbnail = new MemoryBase(heap, 0, size); 219 if (mThumbnail == NULL) { 220 ALOGE("not enough memory for VideoFrame size=%zu", size); 221 delete frame; 222 return NULL; 223 } 224 VideoFrame *frameCopy = static_cast<VideoFrame *>(mThumbnail->pointer()); 225 frameCopy->mWidth = frame->mWidth; 226 frameCopy->mHeight = frame->mHeight; 227 frameCopy->mDisplayWidth = frame->mDisplayWidth; 228 frameCopy->mDisplayHeight = frame->mDisplayHeight; 229 frameCopy->mSize = frame->mSize; 230 frameCopy->mRotationAngle = frame->mRotationAngle; 231 ALOGV("rotation: %d", frameCopy->mRotationAngle); 232 frameCopy->mData = (uint8_t *)frameCopy + sizeof(VideoFrame); 233 memcpy(frameCopy->mData, frame->mData, frame->mSize); 234 frameCopy->mData = 0; 235 delete frame; // Fix memory leakage 236 return mThumbnail; 237 } 238 239 sp<IMemory> MetadataRetrieverClient::extractAlbumArt() 240 { 241 ALOGV("extractAlbumArt"); 242 Mutex::Autolock lock(mLock); 243 mAlbumArt.clear(); 244 if (mRetriever == NULL) { 245 ALOGE("retriever is not initialized"); 246 return NULL; 247 } 248 MediaAlbumArt *albumArt = mRetriever->extractAlbumArt(); 249 if (albumArt == NULL) { 250 ALOGE("failed to extract an album art"); 251 return NULL; 252 } 253 size_t size = sizeof(MediaAlbumArt) + albumArt->size(); 254 sp<MemoryHeapBase> heap = new MemoryHeapBase(size, 0, "MetadataRetrieverClient"); 255 if (heap == NULL) { 256 ALOGE("failed to create MemoryDealer object"); 257 delete albumArt; 258 return NULL; 259 } 260 mAlbumArt = new MemoryBase(heap, 0, size); 261 if (mAlbumArt == NULL) { 262 ALOGE("not enough memory for MediaAlbumArt size=%zu", size); 263 delete albumArt; 264 return NULL; 265 } 266 MediaAlbumArt::init((MediaAlbumArt *) mAlbumArt->pointer(), 267 albumArt->size(), albumArt->data()); 268 delete albumArt; // We've taken our copy. 269 return mAlbumArt; 270 } 271 272 const char* MetadataRetrieverClient::extractMetadata(int keyCode) 273 { 274 ALOGV("extractMetadata"); 275 Mutex::Autolock lock(mLock); 276 if (mRetriever == NULL) { 277 ALOGE("retriever is not initialized"); 278 return NULL; 279 } 280 return mRetriever->extractMetadata(keyCode); 281 } 282 283 }; // namespace android 284