Home | History | Annotate | Download | only in libstagefright 
      1 /*
      2  * Copyright (C) 2011 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 //#define LOG_NDEBUG 0
     18 #define LOG_TAG "HTTPBase"
     19 #include <utils/Log.h>
     20 
     21 #include "include/HTTPBase.h"
     22 
     23 #if CHROMIUM_AVAILABLE
     24 #include "include/chromium_http_stub.h"
     25 #endif
     26 
     27 #include <media/stagefright/foundation/ADebug.h>
     28 #include <media/stagefright/foundation/ALooper.h>
     29 
     30 #include <cutils/properties.h>
     31 #include <cutils/qtaguid.h>
     32 
     33 #include <ConnectivityManager.h>
     34 
     35 namespace android {
     36 
     37 HTTPBase::HTTPBase()
     38     : mNumBandwidthHistoryItems(0),
     39       mTotalTransferTimeUs(0),
     40       mTotalTransferBytes(0),
     41       mPrevBandwidthMeasureTimeUs(0),
     42       mPrevEstimatedBandWidthKbps(0),
     43       mBandWidthCollectFreqMs(5000),
     44       mUIDValid(false),
     45       mUID(0) {
     46 }
     47 
     48 // static
     49 sp<HTTPBase> HTTPBase::Create(uint32_t flags) {
     50 #if CHROMIUM_AVAILABLE
     51         HTTPBase *dataSource = createChromiumHTTPDataSource(flags);
     52         if (dataSource) {
     53            return dataSource;
     54         }
     55 #endif
     56     {
     57         TRESPASS();
     58 
     59         return NULL;
     60     }
     61 }
     62 
     63 // static
     64 status_t HTTPBase::UpdateProxyConfig(
     65         const char *host, int32_t port, const char *exclusionList) {
     66 #if CHROMIUM_AVAILABLE
     67     return UpdateChromiumHTTPDataSourceProxyConfig(host, port, exclusionList);
     68 #else
     69     return INVALID_OPERATION;
     70 #endif
     71 }
     72 
     73 void HTTPBase::addBandwidthMeasurement(
     74         size_t numBytes, int64_t delayUs) {
     75     Mutex::Autolock autoLock(mLock);
     76 
     77     BandwidthEntry entry;
     78     entry.mDelayUs = delayUs;
     79     entry.mNumBytes = numBytes;
     80     mTotalTransferTimeUs += delayUs;
     81     mTotalTransferBytes += numBytes;
     82 
     83     mBandwidthHistory.push_back(entry);
     84     if (++mNumBandwidthHistoryItems > 100) {
     85         BandwidthEntry *entry = &*mBandwidthHistory.begin();
     86         mTotalTransferTimeUs -= entry->mDelayUs;
     87         mTotalTransferBytes -= entry->mNumBytes;
     88         mBandwidthHistory.erase(mBandwidthHistory.begin());
     89         --mNumBandwidthHistoryItems;
     90 
     91         int64_t timeNowUs = ALooper::GetNowUs();
     92         if (timeNowUs - mPrevBandwidthMeasureTimeUs >=
     93                 mBandWidthCollectFreqMs * 1000LL) {
     94 
     95             if (mPrevBandwidthMeasureTimeUs != 0) {
     96                 mPrevEstimatedBandWidthKbps =
     97                     (mTotalTransferBytes * 8E3 / mTotalTransferTimeUs);
     98             }
     99             mPrevBandwidthMeasureTimeUs = timeNowUs;
    100         }
    101     }
    102 
    103 }
    104 
    105 bool HTTPBase::estimateBandwidth(int32_t *bandwidth_bps) {
    106     Mutex::Autolock autoLock(mLock);
    107 
    108     if (mNumBandwidthHistoryItems < 2) {
    109         return false;
    110     }
    111 
    112     *bandwidth_bps = ((double)mTotalTransferBytes * 8E6 / mTotalTransferTimeUs);
    113 
    114     return true;
    115 }
    116 
    117 status_t HTTPBase::getEstimatedBandwidthKbps(int32_t *kbps) {
    118     Mutex::Autolock autoLock(mLock);
    119     *kbps = mPrevEstimatedBandWidthKbps;
    120     return OK;
    121 }
    122 
    123 status_t HTTPBase::setBandwidthStatCollectFreq(int32_t freqMs) {
    124     Mutex::Autolock autoLock(mLock);
    125 
    126     if (freqMs < kMinBandwidthCollectFreqMs
    127             || freqMs > kMaxBandwidthCollectFreqMs) {
    128 
    129         ALOGE("frequency (%d ms) is out of range [1000, 60000]", freqMs);
    130         return BAD_VALUE;
    131     }
    132 
    133     ALOGI("frequency set to %d ms", freqMs);
    134     mBandWidthCollectFreqMs = freqMs;
    135     return OK;
    136 }
    137 
    138 void HTTPBase::setUID(uid_t uid) {
    139     mUIDValid = true;
    140     mUID = uid;
    141 }
    142 
    143 bool HTTPBase::getUID(uid_t *uid) const {
    144     if (!mUIDValid) {
    145         return false;
    146     }
    147 
    148     *uid = mUID;
    149 
    150     return true;
    151 }
    152 
    153 // static
    154 void HTTPBase::RegisterSocketUserTag(int sockfd, uid_t uid, uint32_t kTag) {
    155     int res = qtaguid_tagSocket(sockfd, kTag, uid);
    156     if (res != 0) {
    157         ALOGE("Failed tagging socket %d for uid %d (My UID=%d)", sockfd, uid, geteuid());
    158     }
    159 }
    160 
    161 // static
    162 void HTTPBase::UnRegisterSocketUserTag(int sockfd) {
    163     int res = qtaguid_untagSocket(sockfd);
    164     if (res != 0) {
    165         ALOGE("Failed untagging socket %d (My UID=%d)", sockfd, geteuid());
    166     }
    167 }
    168 
    169 // static
    170 void HTTPBase::RegisterSocketUserMark(int sockfd, uid_t uid) {
    171     ConnectivityManager::markSocketAsUser(sockfd, uid);
    172 }
    173 
    174 // static
    175 void HTTPBase::UnRegisterSocketUserMark(int sockfd) {
    176     RegisterSocketUserMark(sockfd, geteuid());
    177 }
    178 
    179 }  // namespace android
    180