blob: 32291c8b61147ce5bbeedf7359f6bbe8db65d493 [file] [log] [blame]
/*
* Copyright (C) 2011 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
//#define LOG_NDEBUG 0
#define LOG_TAG "HTTPBase"
#include <utils/Log.h>
#include "include/HTTPBase.h"
#include <media/stagefright/foundation/ADebug.h>
#include <media/stagefright/foundation/ALooper.h>
#include <cutils/properties.h>
#include <cutils/qtaguid.h>
#include <NetdClient.h>
namespace android {
HTTPBase::HTTPBase()
: mNumBandwidthHistoryItems(0),
mTotalTransferTimeUs(0),
mTotalTransferBytes(0),
mPrevBandwidthMeasureTimeUs(0),
mPrevEstimatedBandWidthKbps(0),
mBandWidthCollectFreqMs(5000) {
}
void HTTPBase::addBandwidthMeasurement(
size_t numBytes, int64_t delayUs) {
Mutex::Autolock autoLock(mLock);
BandwidthEntry entry;
entry.mDelayUs = delayUs;
entry.mNumBytes = numBytes;
mTotalTransferTimeUs += delayUs;
mTotalTransferBytes += numBytes;
mBandwidthHistory.push_back(entry);
if (++mNumBandwidthHistoryItems > 100) {
BandwidthEntry *entry = &*mBandwidthHistory.begin();
mTotalTransferTimeUs -= entry->mDelayUs;
mTotalTransferBytes -= entry->mNumBytes;
mBandwidthHistory.erase(mBandwidthHistory.begin());
--mNumBandwidthHistoryItems;
int64_t timeNowUs = ALooper::GetNowUs();
if (timeNowUs - mPrevBandwidthMeasureTimeUs >=
mBandWidthCollectFreqMs * 1000LL) {
if (mPrevBandwidthMeasureTimeUs != 0) {
mPrevEstimatedBandWidthKbps =
(mTotalTransferBytes * 8E3 / mTotalTransferTimeUs);
}
mPrevBandwidthMeasureTimeUs = timeNowUs;
}
}
}
bool HTTPBase::estimateBandwidth(int32_t *bandwidth_bps) {
Mutex::Autolock autoLock(mLock);
if (mNumBandwidthHistoryItems < 2) {
return false;
}
*bandwidth_bps = ((double)mTotalTransferBytes * 8E6 / mTotalTransferTimeUs);
return true;
}
status_t HTTPBase::getEstimatedBandwidthKbps(int32_t *kbps) {
Mutex::Autolock autoLock(mLock);
*kbps = mPrevEstimatedBandWidthKbps;
return OK;
}
status_t HTTPBase::setBandwidthStatCollectFreq(int32_t freqMs) {
Mutex::Autolock autoLock(mLock);
if (freqMs < kMinBandwidthCollectFreqMs
|| freqMs > kMaxBandwidthCollectFreqMs) {
ALOGE("frequency (%d ms) is out of range [1000, 60000]", freqMs);
return BAD_VALUE;
}
ALOGI("frequency set to %d ms", freqMs);
mBandWidthCollectFreqMs = freqMs;
return OK;
}
// static
void HTTPBase::RegisterSocketUserTag(int sockfd, uid_t uid, uint32_t kTag) {
int res = qtaguid_tagSocket(sockfd, kTag, uid);
if (res != 0) {
ALOGE("Failed tagging socket %d for uid %d (My UID=%d)", sockfd, uid, geteuid());
}
}
// static
void HTTPBase::UnRegisterSocketUserTag(int sockfd) {
int res = qtaguid_untagSocket(sockfd);
if (res != 0) {
ALOGE("Failed untagging socket %d (My UID=%d)", sockfd, geteuid());
}
}
// static
void HTTPBase::RegisterSocketUserMark(int sockfd, uid_t uid) {
setNetworkForUser(uid, sockfd);
}
// static
void HTTPBase::UnRegisterSocketUserMark(int sockfd) {
RegisterSocketUserMark(sockfd, geteuid());
}
} // namespace android