LiveSession.cpp revision a44153c1a57202fb538659eb50706e60454d6273
1a44153c1a57202fb538659eb50706e60454d6273Andreas Huber/* 2a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * Copyright (C) 2010 The Android Open Source Project 3a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * 4a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * Licensed under the Apache License, Version 2.0 (the "License"); 5a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * you may not use this file except in compliance with the License. 6a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * You may obtain a copy of the License at 7a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * 8a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * http://www.apache.org/licenses/LICENSE-2.0 9a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * 10a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * Unless required by applicable law or agreed to in writing, software 11a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * distributed under the License is distributed on an "AS IS" BASIS, 12a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * See the License for the specific language governing permissions and 14a44153c1a57202fb538659eb50706e60454d6273Andreas Huber * limitations under the License. 15a44153c1a57202fb538659eb50706e60454d6273Andreas Huber */ 16a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 17a44153c1a57202fb538659eb50706e60454d6273Andreas Huber//#define LOG_NDEBUG 0 18a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#define LOG_TAG "LiveSession" 19a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <utils/Log.h> 20a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 21a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include "include/LiveSession.h" 22a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 23a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include "LiveDataSource.h" 24a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 25a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include "include/M3UParser.h" 26a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include "include/NuHTTPDataSource.h" 27a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 28a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <cutils/properties.h> 29a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <media/stagefright/foundation/hexdump.h> 30a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <media/stagefright/foundation/ABuffer.h> 31a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <media/stagefright/foundation/ADebug.h> 32a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <media/stagefright/foundation/AMessage.h> 33a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <media/stagefright/DataSource.h> 34a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <media/stagefright/FileSource.h> 35a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <media/stagefright/MediaErrors.h> 36a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 37a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <ctype.h> 38a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#include <openssl/aes.h> 39a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 40a44153c1a57202fb538659eb50706e60454d6273Andreas Hubernamespace android { 41a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 42a44153c1a57202fb538659eb50706e60454d6273Andreas Huberconst int64_t LiveSession::kMaxPlaylistAgeUs = 15000000ll; 43a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 44a44153c1a57202fb538659eb50706e60454d6273Andreas HuberLiveSession::LiveSession() 45a44153c1a57202fb538659eb50706e60454d6273Andreas Huber : mDataSource(new LiveDataSource), 46a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mHTTPDataSource(new NuHTTPDataSource), 47a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mPrevBandwidthIndex(-1), 48a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mLastPlaylistFetchTimeUs(-1), 49a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeqNumber(-1), 50a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeekTimeUs(-1), 51a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mNumRetries(0), 52a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mDurationUs(-1), 53a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeekDone(false), 54a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mMonitorQueueGeneration(0) { 55a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 56a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 57a44153c1a57202fb538659eb50706e60454d6273Andreas HuberLiveSession::~LiveSession() { 58a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 59a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 60a44153c1a57202fb538659eb50706e60454d6273Andreas Hubersp<DataSource> LiveSession::getDataSource() { 61a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return mDataSource; 62a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 63a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 64a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::connect(const char *url) { 65a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<AMessage> msg = new AMessage(kWhatConnect, id()); 66a44153c1a57202fb538659eb50706e60454d6273Andreas Huber msg->setString("url", url); 67a44153c1a57202fb538659eb50706e60454d6273Andreas Huber msg->post(); 68a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 69a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 70a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::disconnect() { 71a44153c1a57202fb538659eb50706e60454d6273Andreas Huber (new AMessage(kWhatDisconnect, id()))->post(); 72a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 73a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 74a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::seekTo(int64_t timeUs) { 75a44153c1a57202fb538659eb50706e60454d6273Andreas Huber Mutex::Autolock autoLock(mLock); 76a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeekDone = false; 77a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 78a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<AMessage> msg = new AMessage(kWhatSeek, id()); 79a44153c1a57202fb538659eb50706e60454d6273Andreas Huber msg->setInt64("timeUs", timeUs); 80a44153c1a57202fb538659eb50706e60454d6273Andreas Huber msg->post(); 81a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 82a44153c1a57202fb538659eb50706e60454d6273Andreas Huber while (!mSeekDone) { 83a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mCondition.wait(mLock); 84a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 85a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 86a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 87a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::onMessageReceived(const sp<AMessage> &msg) { 88a44153c1a57202fb538659eb50706e60454d6273Andreas Huber switch (msg->what()) { 89a44153c1a57202fb538659eb50706e60454d6273Andreas Huber case kWhatConnect: 90a44153c1a57202fb538659eb50706e60454d6273Andreas Huber onConnect(msg); 91a44153c1a57202fb538659eb50706e60454d6273Andreas Huber break; 92a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 93a44153c1a57202fb538659eb50706e60454d6273Andreas Huber case kWhatDisconnect: 94a44153c1a57202fb538659eb50706e60454d6273Andreas Huber onDisconnect(); 95a44153c1a57202fb538659eb50706e60454d6273Andreas Huber break; 96a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 97a44153c1a57202fb538659eb50706e60454d6273Andreas Huber case kWhatMonitorQueue: 98a44153c1a57202fb538659eb50706e60454d6273Andreas Huber { 99a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int32_t generation; 100a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK(msg->findInt32("generation", &generation)); 101a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 102a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (generation != mMonitorQueueGeneration) { 103a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // Stale event 104a44153c1a57202fb538659eb50706e60454d6273Andreas Huber break; 105a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 106a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 107a44153c1a57202fb538659eb50706e60454d6273Andreas Huber onMonitorQueue(); 108a44153c1a57202fb538659eb50706e60454d6273Andreas Huber break; 109a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 110a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 111a44153c1a57202fb538659eb50706e60454d6273Andreas Huber case kWhatSeek: 112a44153c1a57202fb538659eb50706e60454d6273Andreas Huber onSeek(msg); 113a44153c1a57202fb538659eb50706e60454d6273Andreas Huber break; 114a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 115a44153c1a57202fb538659eb50706e60454d6273Andreas Huber default: 116a44153c1a57202fb538659eb50706e60454d6273Andreas Huber TRESPASS(); 117a44153c1a57202fb538659eb50706e60454d6273Andreas Huber break; 118a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 119a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 120a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 121a44153c1a57202fb538659eb50706e60454d6273Andreas Huber// static 122a44153c1a57202fb538659eb50706e60454d6273Andreas Huberint LiveSession::SortByBandwidth(const BandwidthItem *a, const BandwidthItem *b) { 123a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (a->mBandwidth < b->mBandwidth) { 124a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return -1; 125a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else if (a->mBandwidth == b->mBandwidth) { 126a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return 0; 127a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 128a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 129a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return 1; 130a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 131a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 132a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::onConnect(const sp<AMessage> &msg) { 133a44153c1a57202fb538659eb50706e60454d6273Andreas Huber AString url; 134a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK(msg->findString("url", &url)); 135a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 136a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGI("onConnect '%s'", url.c_str()); 137a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 138a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mMasterURL = url; 139a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 140a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<M3UParser> playlist = fetchPlaylist(url.c_str()); 141a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK(playlist != NULL); 142a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 143a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (playlist->isVariantPlaylist()) { 144a44153c1a57202fb538659eb50706e60454d6273Andreas Huber for (size_t i = 0; i < playlist->size(); ++i) { 145a44153c1a57202fb538659eb50706e60454d6273Andreas Huber BandwidthItem item; 146a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 147a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<AMessage> meta; 148a44153c1a57202fb538659eb50706e60454d6273Andreas Huber playlist->itemAt(i, &item.mURI, &meta); 149a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 150a44153c1a57202fb538659eb50706e60454d6273Andreas Huber unsigned long bandwidth; 151a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK(meta->findInt32("bandwidth", (int32_t *)&item.mBandwidth)); 152a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 153a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mBandwidthItems.push(item); 154a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 155a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 156a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK_GT(mBandwidthItems.size(), 0u); 157a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 158a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mBandwidthItems.sort(SortByBandwidth); 159a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 160a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mBandwidthItems.size() > 1) { 161a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // XXX Remove the lowest bitrate stream for now... 162a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mBandwidthItems.removeAt(0); 163a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 164a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 165a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 166a44153c1a57202fb538659eb50706e60454d6273Andreas Huber postMonitorQueue(); 167a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 168a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 169a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::onDisconnect() { 170a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGI("onDisconnect"); 171a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 172a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mDataSource->queueEOS(ERROR_END_OF_STREAM); 173a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 174a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 175a44153c1a57202fb538659eb50706e60454d6273Andreas Huberstatus_t LiveSession::fetchFile(const char *url, sp<ABuffer> *out) { 176a44153c1a57202fb538659eb50706e60454d6273Andreas Huber *out = NULL; 177a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 178a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<DataSource> source; 179a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 180a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (!strncasecmp(url, "file://", 7)) { 181a44153c1a57202fb538659eb50706e60454d6273Andreas Huber source = new FileSource(url + 7); 182a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else { 183a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK(!strncasecmp(url, "http://", 7)); 184a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 185a44153c1a57202fb538659eb50706e60454d6273Andreas Huber status_t err = mHTTPDataSource->connect(url); 186a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 187a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (err != OK) { 188a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return err; 189a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 190a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 191a44153c1a57202fb538659eb50706e60454d6273Andreas Huber source = mHTTPDataSource; 192a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 193a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 194a44153c1a57202fb538659eb50706e60454d6273Andreas Huber off64_t size; 195a44153c1a57202fb538659eb50706e60454d6273Andreas Huber status_t err = source->getSize(&size); 196a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 197a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (err != OK) { 198a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size = 65536; 199a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 200a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 201a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<ABuffer> buffer = new ABuffer(size); 202a44153c1a57202fb538659eb50706e60454d6273Andreas Huber buffer->setRange(0, 0); 203a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 204a44153c1a57202fb538659eb50706e60454d6273Andreas Huber for (;;) { 205a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t bufferRemaining = buffer->capacity() - buffer->size(); 206a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 207a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (bufferRemaining == 0) { 208a44153c1a57202fb538659eb50706e60454d6273Andreas Huber bufferRemaining = 32768; 209a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 210a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGV("increasing download buffer to %d bytes", 211a44153c1a57202fb538659eb50706e60454d6273Andreas Huber buffer->size() + bufferRemaining); 212a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 213a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<ABuffer> copy = new ABuffer(buffer->size() + bufferRemaining); 214a44153c1a57202fb538659eb50706e60454d6273Andreas Huber memcpy(copy->data(), buffer->data(), buffer->size()); 215a44153c1a57202fb538659eb50706e60454d6273Andreas Huber copy->setRange(0, buffer->size()); 216a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 217a44153c1a57202fb538659eb50706e60454d6273Andreas Huber buffer = copy; 218a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 219a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 220a44153c1a57202fb538659eb50706e60454d6273Andreas Huber ssize_t n = source->readAt( 221a44153c1a57202fb538659eb50706e60454d6273Andreas Huber buffer->size(), buffer->data() + buffer->size(), 222a44153c1a57202fb538659eb50706e60454d6273Andreas Huber bufferRemaining); 223a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 224a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (n < 0) { 225a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return err; 226a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 227a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 228a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (n == 0) { 229a44153c1a57202fb538659eb50706e60454d6273Andreas Huber break; 230a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 231a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 232a44153c1a57202fb538659eb50706e60454d6273Andreas Huber buffer->setRange(0, buffer->size() + (size_t)n); 233a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 234a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 235a44153c1a57202fb538659eb50706e60454d6273Andreas Huber *out = buffer; 236a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 237a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return OK; 238a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 239a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 240a44153c1a57202fb538659eb50706e60454d6273Andreas Hubersp<M3UParser> LiveSession::fetchPlaylist(const char *url) { 241a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<ABuffer> buffer; 242a44153c1a57202fb538659eb50706e60454d6273Andreas Huber status_t err = fetchFile(url, &buffer); 243a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 244a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (err != OK) { 245a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return NULL; 246a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 247a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 248a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<M3UParser> playlist = 249a44153c1a57202fb538659eb50706e60454d6273Andreas Huber new M3UParser(url, buffer->data(), buffer->size()); 250a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 251a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (playlist->initCheck() != OK) { 252a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return NULL; 253a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 254a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 255a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return playlist; 256a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 257a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 258a44153c1a57202fb538659eb50706e60454d6273Andreas Huberstatic double uniformRand() { 259a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return (double)rand() / RAND_MAX; 260a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 261a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 262a44153c1a57202fb538659eb50706e60454d6273Andreas Hubersize_t LiveSession::getBandwidthIndex() { 263a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mBandwidthItems.size() == 0) { 264a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return 0; 265a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 266a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 267a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#if 1 268a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int32_t bandwidthBps; 269a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mHTTPDataSource != NULL 270a44153c1a57202fb538659eb50706e60454d6273Andreas Huber && mHTTPDataSource->estimateBandwidth(&bandwidthBps)) { 271a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGV("bandwidth estimated at %.2f kbps", bandwidthBps / 1024.0f); 272a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else { 273a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGV("no bandwidth estimate."); 274a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return 0; // Pick the lowest bandwidth stream by default. 275a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 276a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 277a44153c1a57202fb538659eb50706e60454d6273Andreas Huber char value[PROPERTY_VALUE_MAX]; 278a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (property_get("media.httplive.max-bw", value, NULL)) { 279a44153c1a57202fb538659eb50706e60454d6273Andreas Huber char *end; 280a44153c1a57202fb538659eb50706e60454d6273Andreas Huber long maxBw = strtoul(value, &end, 10); 281a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (end > value && *end == '\0') { 282a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (maxBw > 0 && bandwidthBps > maxBw) { 283a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGV("bandwidth capped to %ld bps", maxBw); 284a44153c1a57202fb538659eb50706e60454d6273Andreas Huber bandwidthBps = maxBw; 285a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 286a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 287a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 288a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 289a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // Consider only 80% of the available bandwidth usable. 290a44153c1a57202fb538659eb50706e60454d6273Andreas Huber bandwidthBps = (bandwidthBps * 8) / 10; 291a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 292a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // Pick the highest bandwidth stream below or equal to estimated bandwidth. 293a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 294a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t index = mBandwidthItems.size() - 1; 295a44153c1a57202fb538659eb50706e60454d6273Andreas Huber while (index > 0 && mBandwidthItems.itemAt(index).mBandwidth 296a44153c1a57202fb538659eb50706e60454d6273Andreas Huber > (size_t)bandwidthBps) { 297a44153c1a57202fb538659eb50706e60454d6273Andreas Huber --index; 298a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 299a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#elif 0 300a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // Change bandwidth at random() 301a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t index = uniformRand() * mBandwidthItems.size(); 302a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#elif 0 303a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // There's a 50% chance to stay on the current bandwidth and 304a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // a 50% chance to switch to the next higher bandwidth (wrapping around 305a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // to lowest) 306a44153c1a57202fb538659eb50706e60454d6273Andreas Huber const size_t kMinIndex = 0; 307a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 308a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t index; 309a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mPrevBandwidthIndex < 0) { 310a44153c1a57202fb538659eb50706e60454d6273Andreas Huber index = kMinIndex; 311a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else if (uniformRand() < 0.5) { 312a44153c1a57202fb538659eb50706e60454d6273Andreas Huber index = (size_t)mPrevBandwidthIndex; 313a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else { 314a44153c1a57202fb538659eb50706e60454d6273Andreas Huber index = mPrevBandwidthIndex + 1; 315a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (index == mBandwidthItems.size()) { 316a44153c1a57202fb538659eb50706e60454d6273Andreas Huber index = kMinIndex; 317a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 318a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 319a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#elif 0 320a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // Pick the highest bandwidth stream below or equal to 1.2 Mbit/sec 321a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 322a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t index = mBandwidthItems.size() - 1; 323a44153c1a57202fb538659eb50706e60454d6273Andreas Huber while (index > 0 && mBandwidthItems.itemAt(index).mBandwidth > 1200000) { 324a44153c1a57202fb538659eb50706e60454d6273Andreas Huber --index; 325a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 326a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#else 327a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t index = mBandwidthItems.size() - 1; // Highest bandwidth stream 328a44153c1a57202fb538659eb50706e60454d6273Andreas Huber#endif 329a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 330a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return index; 331a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 332a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 333a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::onDownloadNext() { 334a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t bandwidthIndex = getBandwidthIndex(); 335a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 336a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int64_t nowUs = ALooper::GetNowUs(); 337a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 338a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mLastPlaylistFetchTimeUs < 0 339a44153c1a57202fb538659eb50706e60454d6273Andreas Huber || (ssize_t)bandwidthIndex != mPrevBandwidthIndex 340a44153c1a57202fb538659eb50706e60454d6273Andreas Huber || (!mPlaylist->isComplete() 341a44153c1a57202fb538659eb50706e60454d6273Andreas Huber && mLastPlaylistFetchTimeUs + kMaxPlaylistAgeUs <= nowUs)) { 342a44153c1a57202fb538659eb50706e60454d6273Andreas Huber AString url; 343a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mBandwidthItems.size() > 0) { 344a44153c1a57202fb538659eb50706e60454d6273Andreas Huber url = mBandwidthItems.editItemAt(bandwidthIndex).mURI; 345a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else { 346a44153c1a57202fb538659eb50706e60454d6273Andreas Huber url = mMasterURL; 347a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 348a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 349a44153c1a57202fb538659eb50706e60454d6273Andreas Huber bool firstTime = (mPlaylist == NULL); 350a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 351a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mPlaylist = fetchPlaylist(url.c_str()); 352a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK(mPlaylist != NULL); 353a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 354a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (firstTime) { 355a44153c1a57202fb538659eb50706e60454d6273Andreas Huber Mutex::Autolock autoLock(mLock); 356a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 357a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int32_t targetDuration; 358a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (!mPlaylist->isComplete() 359a44153c1a57202fb538659eb50706e60454d6273Andreas Huber || !mPlaylist->meta()->findInt32( 360a44153c1a57202fb538659eb50706e60454d6273Andreas Huber "target-duration", &targetDuration)) { 361a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mDurationUs = -1; 362a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else { 363a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mDurationUs = 1000000ll * targetDuration * mPlaylist->size(); 364a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 365a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 366a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 367a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mLastPlaylistFetchTimeUs = ALooper::GetNowUs(); 368a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 369a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 370a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int32_t firstSeqNumberInPlaylist; 371a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mPlaylist->meta() == NULL || !mPlaylist->meta()->findInt32( 372a44153c1a57202fb538659eb50706e60454d6273Andreas Huber "media-sequence", &firstSeqNumberInPlaylist)) { 373a44153c1a57202fb538659eb50706e60454d6273Andreas Huber firstSeqNumberInPlaylist = 0; 374a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 375a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 376a44153c1a57202fb538659eb50706e60454d6273Andreas Huber bool explicitDiscontinuity = false; 377a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 378a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mSeekTimeUs >= 0) { 379a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int32_t targetDuration; 380a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mPlaylist->isComplete() && 381a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mPlaylist->meta()->findInt32( 382a44153c1a57202fb538659eb50706e60454d6273Andreas Huber "target-duration", &targetDuration)) { 383a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int64_t seekTimeSecs = (mSeekTimeUs + 500000ll) / 1000000ll; 384a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int64_t index = seekTimeSecs / targetDuration; 385a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 386a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (index >= 0 && index < mPlaylist->size()) { 387a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeqNumber = firstSeqNumberInPlaylist + index; 388a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mDataSource->reset(); 389a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 390a44153c1a57202fb538659eb50706e60454d6273Andreas Huber explicitDiscontinuity = true; 391a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 392a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 393a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 394a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeekTimeUs = -1; 395a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 396a44153c1a57202fb538659eb50706e60454d6273Andreas Huber Mutex::Autolock autoLock(mLock); 397a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeekDone = true; 398a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mCondition.broadcast(); 399a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 400a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 401a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mSeqNumber < 0) { 402a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mPlaylist->isComplete()) { 403a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeqNumber = firstSeqNumberInPlaylist; 404a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else { 405a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeqNumber = firstSeqNumberInPlaylist + mPlaylist->size() / 2; 406a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 407a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 408a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 409a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int32_t lastSeqNumberInPlaylist = 410a44153c1a57202fb538659eb50706e60454d6273Andreas Huber firstSeqNumberInPlaylist + (int32_t)mPlaylist->size() - 1; 411a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 412a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mSeqNumber < firstSeqNumberInPlaylist 413a44153c1a57202fb538659eb50706e60454d6273Andreas Huber || mSeqNumber > lastSeqNumberInPlaylist) { 414a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (!mPlaylist->isComplete() 415a44153c1a57202fb538659eb50706e60454d6273Andreas Huber && mSeqNumber > lastSeqNumberInPlaylist 416a44153c1a57202fb538659eb50706e60454d6273Andreas Huber && mNumRetries < kMaxNumRetries) { 417a44153c1a57202fb538659eb50706e60454d6273Andreas Huber ++mNumRetries; 418a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 419a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mLastPlaylistFetchTimeUs = -1; 420a44153c1a57202fb538659eb50706e60454d6273Andreas Huber postMonitorQueue(1000000ll); 421a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return; 422a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 423a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 424a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGE("Cannot find sequence number %d in playlist " 425a44153c1a57202fb538659eb50706e60454d6273Andreas Huber "(contains %d - %d)", 426a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeqNumber, firstSeqNumberInPlaylist, 427a44153c1a57202fb538659eb50706e60454d6273Andreas Huber firstSeqNumberInPlaylist + mPlaylist->size() - 1); 428a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 429a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mDataSource->queueEOS(ERROR_END_OF_STREAM); 430a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return; 431a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 432a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 433a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mNumRetries = 0; 434a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 435a44153c1a57202fb538659eb50706e60454d6273Andreas Huber AString uri; 436a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<AMessage> itemMeta; 437a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK(mPlaylist->itemAt( 438a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeqNumber - firstSeqNumberInPlaylist, 439a44153c1a57202fb538659eb50706e60454d6273Andreas Huber &uri, 440a44153c1a57202fb538659eb50706e60454d6273Andreas Huber &itemMeta)); 441a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 442a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int32_t val; 443a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (itemMeta->findInt32("discontinuity", &val) && val != 0) { 444a44153c1a57202fb538659eb50706e60454d6273Andreas Huber explicitDiscontinuity = true; 445a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 446a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 447a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<ABuffer> buffer; 448a44153c1a57202fb538659eb50706e60454d6273Andreas Huber status_t err = fetchFile(uri.c_str(), &buffer); 449a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK_EQ(err, (status_t)OK); 450a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 451a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK_EQ((status_t)OK, 452a44153c1a57202fb538659eb50706e60454d6273Andreas Huber decryptBuffer(mSeqNumber - firstSeqNumberInPlaylist, buffer)); 453a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 454a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (buffer->size() == 0 || buffer->data()[0] != 0x47) { 455a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // Not a transport stream??? 456a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 457a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGE("This doesn't look like a transport stream..."); 458a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 459a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mDataSource->queueEOS(ERROR_UNSUPPORTED); 460a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return; 461a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 462a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 463a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (explicitDiscontinuity 464a44153c1a57202fb538659eb50706e60454d6273Andreas Huber || (mPrevBandwidthIndex >= 0 465a44153c1a57202fb538659eb50706e60454d6273Andreas Huber && (size_t)mPrevBandwidthIndex != bandwidthIndex)) { 466a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // Signal discontinuity. 467a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 468a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<ABuffer> tmp = new ABuffer(188); 469a44153c1a57202fb538659eb50706e60454d6273Andreas Huber memset(tmp->data(), 0, tmp->size()); 470a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 471a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mDataSource->queueBuffer(tmp); 472a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 473a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 474a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mDataSource->queueBuffer(buffer); 475a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 476a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mPrevBandwidthIndex = bandwidthIndex; 477a44153c1a57202fb538659eb50706e60454d6273Andreas Huber ++mSeqNumber; 478a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 479a44153c1a57202fb538659eb50706e60454d6273Andreas Huber postMonitorQueue(); 480a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 481a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 482a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::onMonitorQueue() { 483a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (mSeekTimeUs >= 0 484a44153c1a57202fb538659eb50706e60454d6273Andreas Huber || mDataSource->countQueuedBuffers() < kMaxNumQueuedFragments) { 485a44153c1a57202fb538659eb50706e60454d6273Andreas Huber onDownloadNext(); 486a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else { 487a44153c1a57202fb538659eb50706e60454d6273Andreas Huber postMonitorQueue(1000000ll); 488a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 489a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 490a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 491a44153c1a57202fb538659eb50706e60454d6273Andreas Huberstatus_t LiveSession::decryptBuffer( 492a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t playlistIndex, const sp<ABuffer> &buffer) { 493a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<AMessage> itemMeta; 494a44153c1a57202fb538659eb50706e60454d6273Andreas Huber bool found = false; 495a44153c1a57202fb538659eb50706e60454d6273Andreas Huber AString method; 496a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 497a44153c1a57202fb538659eb50706e60454d6273Andreas Huber for (ssize_t i = playlistIndex; i >= 0; --i) { 498a44153c1a57202fb538659eb50706e60454d6273Andreas Huber AString uri; 499a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK(mPlaylist->itemAt(i, &uri, &itemMeta)); 500a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 501a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (itemMeta->findString("cipher-method", &method)) { 502a44153c1a57202fb538659eb50706e60454d6273Andreas Huber found = true; 503a44153c1a57202fb538659eb50706e60454d6273Andreas Huber break; 504a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 505a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 506a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 507a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (!found) { 508a44153c1a57202fb538659eb50706e60454d6273Andreas Huber method = "NONE"; 509a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 510a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 511a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (method == "NONE") { 512a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return OK; 513a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else if (!(method == "AES-128")) { 514a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGE("Unsupported cipher method '%s'", method.c_str()); 515a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return ERROR_UNSUPPORTED; 516a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 517a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 518a44153c1a57202fb538659eb50706e60454d6273Andreas Huber AString keyURI; 519a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (!itemMeta->findString("cipher-uri", &keyURI)) { 520a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGE("Missing key uri"); 521a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return ERROR_MALFORMED; 522a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 523a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 524a44153c1a57202fb538659eb50706e60454d6273Andreas Huber ssize_t index = mAESKeyForURI.indexOfKey(keyURI); 525a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 526a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<ABuffer> key; 527a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (index >= 0) { 528a44153c1a57202fb538659eb50706e60454d6273Andreas Huber key = mAESKeyForURI.valueAt(index); 529a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else { 530a44153c1a57202fb538659eb50706e60454d6273Andreas Huber key = new ABuffer(16); 531a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 532a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<NuHTTPDataSource> keySource = new NuHTTPDataSource; 533a44153c1a57202fb538659eb50706e60454d6273Andreas Huber status_t err = keySource->connect(keyURI.c_str()); 534a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 535a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (err == OK) { 536a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t offset = 0; 537a44153c1a57202fb538659eb50706e60454d6273Andreas Huber while (offset < 16) { 538a44153c1a57202fb538659eb50706e60454d6273Andreas Huber ssize_t n = keySource->readAt( 539a44153c1a57202fb538659eb50706e60454d6273Andreas Huber offset, key->data() + offset, 16 - offset); 540a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (n <= 0) { 541a44153c1a57202fb538659eb50706e60454d6273Andreas Huber err = ERROR_IO; 542a44153c1a57202fb538659eb50706e60454d6273Andreas Huber break; 543a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 544a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 545a44153c1a57202fb538659eb50706e60454d6273Andreas Huber offset += n; 546a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 547a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 548a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 549a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (err != OK) { 550a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGE("failed to fetch cipher key from '%s'.", keyURI.c_str()); 551a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return ERROR_IO; 552a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 553a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 554a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mAESKeyForURI.add(keyURI, key); 555a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 556a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 557a44153c1a57202fb538659eb50706e60454d6273Andreas Huber AES_KEY aes_key; 558a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (AES_set_decrypt_key(key->data(), 128, &aes_key) != 0) { 559a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGE("failed to set AES decryption key."); 560a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return UNKNOWN_ERROR; 561a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 562a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 563a44153c1a57202fb538659eb50706e60454d6273Andreas Huber unsigned char aes_ivec[16]; 564a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 565a44153c1a57202fb538659eb50706e60454d6273Andreas Huber AString iv; 566a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (itemMeta->findString("cipher-iv", &iv)) { 567a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if ((!iv.startsWith("0x") && !iv.startsWith("0X")) 568a44153c1a57202fb538659eb50706e60454d6273Andreas Huber || iv.size() != 16 * 2 + 2) { 569a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGE("malformed cipher IV '%s'.", iv.c_str()); 570a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return ERROR_MALFORMED; 571a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 572a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 573a44153c1a57202fb538659eb50706e60454d6273Andreas Huber memset(aes_ivec, 0, sizeof(aes_ivec)); 574a44153c1a57202fb538659eb50706e60454d6273Andreas Huber for (size_t i = 0; i < 16; ++i) { 575a44153c1a57202fb538659eb50706e60454d6273Andreas Huber char c1 = tolower(iv.c_str()[2 + 2 * i]); 576a44153c1a57202fb538659eb50706e60454d6273Andreas Huber char c2 = tolower(iv.c_str()[3 + 2 * i]); 577a44153c1a57202fb538659eb50706e60454d6273Andreas Huber if (!isxdigit(c1) || !isxdigit(c2)) { 578a44153c1a57202fb538659eb50706e60454d6273Andreas Huber LOGE("malformed cipher IV '%s'.", iv.c_str()); 579a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return ERROR_MALFORMED; 580a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 581a44153c1a57202fb538659eb50706e60454d6273Andreas Huber uint8_t nibble1 = isdigit(c1) ? c1 - '0' : c1 - 'a' + 10; 582a44153c1a57202fb538659eb50706e60454d6273Andreas Huber uint8_t nibble2 = isdigit(c2) ? c2 - '0' : c2 - 'a' + 10; 583a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 584a44153c1a57202fb538659eb50706e60454d6273Andreas Huber aes_ivec[i] = nibble1 << 4 | nibble2; 585a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 586a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } else { 587a44153c1a57202fb538659eb50706e60454d6273Andreas Huber memset(aes_ivec, 0, sizeof(aes_ivec)); 588a44153c1a57202fb538659eb50706e60454d6273Andreas Huber aes_ivec[15] = mSeqNumber & 0xff; 589a44153c1a57202fb538659eb50706e60454d6273Andreas Huber aes_ivec[14] = (mSeqNumber >> 8) & 0xff; 590a44153c1a57202fb538659eb50706e60454d6273Andreas Huber aes_ivec[13] = (mSeqNumber >> 16) & 0xff; 591a44153c1a57202fb538659eb50706e60454d6273Andreas Huber aes_ivec[12] = (mSeqNumber >> 24) & 0xff; 592a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 593a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 594a44153c1a57202fb538659eb50706e60454d6273Andreas Huber AES_cbc_encrypt( 595a44153c1a57202fb538659eb50706e60454d6273Andreas Huber buffer->data(), buffer->data(), buffer->size(), 596a44153c1a57202fb538659eb50706e60454d6273Andreas Huber &aes_key, aes_ivec, AES_DECRYPT); 597a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 598a44153c1a57202fb538659eb50706e60454d6273Andreas Huber // hexdump(buffer->data(), buffer->size()); 599a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 600a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t n = buffer->size(); 601a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK_GT(n, 0u); 602a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 603a44153c1a57202fb538659eb50706e60454d6273Andreas Huber size_t pad = buffer->data()[n - 1]; 604a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 605a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK_GT(pad, 0u); 606a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK_LE(pad, 16u); 607a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK_GE((size_t)n, pad); 608a44153c1a57202fb538659eb50706e60454d6273Andreas Huber for (size_t i = 0; i < pad; ++i) { 609a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK_EQ((unsigned)buffer->data()[n - 1 - i], pad); 610a44153c1a57202fb538659eb50706e60454d6273Andreas Huber } 611a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 612a44153c1a57202fb538659eb50706e60454d6273Andreas Huber n -= pad; 613a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 614a44153c1a57202fb538659eb50706e60454d6273Andreas Huber buffer->setRange(buffer->offset(), n); 615a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 616a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return OK; 617a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 618a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 619a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::postMonitorQueue(int64_t delayUs) { 620a44153c1a57202fb538659eb50706e60454d6273Andreas Huber sp<AMessage> msg = new AMessage(kWhatMonitorQueue, id()); 621a44153c1a57202fb538659eb50706e60454d6273Andreas Huber msg->setInt32("generation", ++mMonitorQueueGeneration); 622a44153c1a57202fb538659eb50706e60454d6273Andreas Huber msg->post(delayUs); 623a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 624a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 625a44153c1a57202fb538659eb50706e60454d6273Andreas Hubervoid LiveSession::onSeek(const sp<AMessage> &msg) { 626a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int64_t timeUs; 627a44153c1a57202fb538659eb50706e60454d6273Andreas Huber CHECK(msg->findInt64("timeUs", &timeUs)); 628a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 629a44153c1a57202fb538659eb50706e60454d6273Andreas Huber mSeekTimeUs = timeUs; 630a44153c1a57202fb538659eb50706e60454d6273Andreas Huber postMonitorQueue(); 631a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 632a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 633a44153c1a57202fb538659eb50706e60454d6273Andreas Huberstatus_t LiveSession::getDuration(int64_t *durationUs) { 634a44153c1a57202fb538659eb50706e60454d6273Andreas Huber Mutex::Autolock autoLock(mLock); 635a44153c1a57202fb538659eb50706e60454d6273Andreas Huber *durationUs = mDurationUs; 636a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 637a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return OK; 638a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 639a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 640a44153c1a57202fb538659eb50706e60454d6273Andreas Huberbool LiveSession::isSeekable() { 641a44153c1a57202fb538659eb50706e60454d6273Andreas Huber int64_t durationUs; 642a44153c1a57202fb538659eb50706e60454d6273Andreas Huber return getDuration(&durationUs) == OK && durationUs >= 0; 643a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} 644a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 645a44153c1a57202fb538659eb50706e60454d6273Andreas Huber} // namespace android 646a44153c1a57202fb538659eb50706e60454d6273Andreas Huber 647