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 #ifndef HTTP_BASE_H_
18 
19 #define HTTP_BASE_H_
20 
21 #include <media/DataSource.h>
22 #include <media/stagefright/foundation/ABase.h>
23 #include <media/stagefright/MediaErrors.h>
24 #include <utils/List.h>
25 #include <utils/threads.h>
26 
27 namespace android {
28 
29 struct HTTPBase : public DataSource {
30     enum Flags {
31         // Don't log any URLs.
32         kFlagIncognito = 1
33     };
34 
35     HTTPBase();
36 
37     virtual status_t connect(
38             const char *uri,
39             const KeyedVector<String8, String8> *headers = NULL,
40             off64_t offset = 0) = 0;
41 
42     virtual void disconnect() = 0;
43 
44     // Returns true if bandwidth could successfully be estimated,
45     // false otherwise.
46     virtual bool estimateBandwidth(int32_t *bandwidth_bps);
47 
48     virtual status_t getEstimatedBandwidthKbps(int32_t *kbps);
49 
50     virtual status_t setBandwidthStatCollectFreq(int32_t freqMs);
51 
52     virtual void setBandwidthHistorySize(size_t numHistoryItems);
53 
54     static void RegisterSocketUserTag(int sockfd, uid_t uid, uint32_t kTag);
55     static void UnRegisterSocketUserTag(int sockfd);
56 
57     static void RegisterSocketUserMark(int sockfd, uid_t uid);
58     static void UnRegisterSocketUserMark(int sockfd);
59 
toStringHTTPBase60     virtual String8 toString() {
61         return mName;
62     }
63 
64 protected:
65     virtual void addBandwidthMeasurement(size_t numBytes, int64_t delayUs);
66     String8 mName;
67 
68 private:
69     struct BandwidthEntry {
70         int64_t mDelayUs;
71         size_t mNumBytes;
72     };
73 
74     Mutex mLock;
75 
76     List<BandwidthEntry> mBandwidthHistory;
77     size_t mNumBandwidthHistoryItems;
78     int64_t mTotalTransferTimeUs;
79     size_t mTotalTransferBytes;
80     size_t mMaxBandwidthHistoryItems;
81 
82     enum {
83         kMinBandwidthCollectFreqMs = 1000,   // 1 second
84         kMaxBandwidthCollectFreqMs = 60000,  // one minute
85     };
86 
87     int64_t mPrevBandwidthMeasureTimeUs;
88     int32_t mPrevEstimatedBandWidthKbps;
89     int32_t mBandWidthCollectFreqMs;
90 
91     DISALLOW_EVIL_CONSTRUCTORS(HTTPBase);
92 };
93 
94 }  // namespace android
95 
96 #endif  // HTTP_BASE_H_
97