15994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber/*
25994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber * Copyright (C) 2010 The Android Open Source Project
35994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber *
45994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber * Licensed under the Apache License, Version 2.0 (the "License");
55994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber * you may not use this file except in compliance with the License.
65994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber * You may obtain a copy of the License at
75994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber *
85994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber *      http://www.apache.org/licenses/LICENSE-2.0
95994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber *
105994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber * Unless required by applicable law or agreed to in writing, software
115994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber * distributed under the License is distributed on an "AS IS" BASIS,
125994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
135994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber * See the License for the specific language governing permissions and
145994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber * limitations under the License.
155994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber */
165994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
175994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber#ifndef NU_CACHED_SOURCE_2_H_
185994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
195994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber#define NU_CACHED_SOURCE_2_H_
205994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
215994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber#include <media/stagefright/foundation/ABase.h>
225994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber#include <media/stagefright/foundation/AHandlerReflector.h>
235994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber#include <media/stagefright/DataSource.h>
245994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
255994b4798b01f3dd340577c9ea9657f09093a770Andreas Hubernamespace android {
265994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
275994b4798b01f3dd340577c9ea9657f09093a770Andreas Huberstruct ALooper;
285994b4798b01f3dd340577c9ea9657f09093a770Andreas Huberstruct PageCache;
295994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
305994b4798b01f3dd340577c9ea9657f09093a770Andreas Huberstruct NuCachedSource2 : public DataSource {
3149c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber    NuCachedSource2(
3249c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber            const sp<DataSource> &source,
3349c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber            const char *cacheConfig = NULL,
3449c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber            bool disconnectAtHighwatermark = false);
355994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
365994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    virtual status_t initCheck() const;
375994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
38c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong    virtual ssize_t readAt(off64_t offset, void *data, size_t size);
395994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
40c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong    virtual status_t getSize(off64_t *size);
415994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    virtual uint32_t flags();
425994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
439d2f386dd2885eaffa11fd494ae258bb09fe6397James Dong    virtual sp<DecryptHandle> DrmInitialization(const char* mime);
44b5ce361d19e69fe156f7188c9ee0f4734b259874Gloria Wang    virtual void getDrmInfo(sp<DecryptHandle> &handle, DrmManagerClient **client);
45771b85d9245a24273497792a2515d88d31c99e1eGloria Wang    virtual String8 getUri();
466511c9755c3a3360ba869772600c7aae048a7ffcAndreas Huber
476511c9755c3a3360ba869772600c7aae048a7ffcAndreas Huber    virtual String8 getMIMEType() const;
486511c9755c3a3360ba869772600c7aae048a7ffcAndreas Huber
495994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    ////////////////////////////////////////////////////////////////////////////
505994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
515994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    size_t cachedSize();
52a53d87c7b1428fe02f535c31dafd64cb1362fde9Andreas Huber    size_t approxDataRemaining(status_t *finalStatus) const;
535994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
5434ef0f32c8fc0186236a27e07405328cc1f7c56dAndreas Huber    void resumeFetchingIfNecessary();
55b33d2ac90cfce0fe6db8c3e979e7ae2bbfc28163James Dong
56b33d2ac90cfce0fe6db8c3e979e7ae2bbfc28163James Dong    // The following methods are supported only if the
57b33d2ac90cfce0fe6db8c3e979e7ae2bbfc28163James Dong    // data source is HTTP-based; otherwise, ERROR_UNSUPPORTED
58b33d2ac90cfce0fe6db8c3e979e7ae2bbfc28163James Dong    // is returned.
595b1b8a93a07326f1cbc627f09e02988375189e0aJames Dong    status_t getEstimatedBandwidthKbps(int32_t *kbps);
60b33d2ac90cfce0fe6db8c3e979e7ae2bbfc28163James Dong    status_t setCacheStatCollectFreq(int32_t freqMs);
6134ef0f32c8fc0186236a27e07405328cc1f7c56dAndreas Huber
6249c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber    static void RemoveCacheSpecificHeaders(
6349c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber            KeyedVector<String8, String8> *headers,
6449c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber            String8 *cacheConfig,
6549c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber            bool *disconnectAtHighwatermark);
6649c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber
675994b4798b01f3dd340577c9ea9657f09093a770Andreas Huberprotected:
685994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    virtual ~NuCachedSource2();
695994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
705994b4798b01f3dd340577c9ea9657f09093a770Andreas Huberprivate:
715994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    friend struct AHandlerReflector<NuCachedSource2>;
725994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
735994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    enum {
74a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber        kPageSize                       = 65536,
75a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber        kDefaultHighWaterThreshold      = 20 * 1024 * 1024,
76a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber        kDefaultLowWaterThreshold       = 4 * 1024 * 1024,
77a5273ebd1746368662a597643d6701a5046d5c7bAndreas Huber
78a5273ebd1746368662a597643d6701a5046d5c7bAndreas Huber        // Read data after a 15 sec timeout whether we're actively
79a5273ebd1746368662a597643d6701a5046d5c7bAndreas Huber        // fetching or not.
80a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber        kDefaultKeepAliveIntervalUs     = 15000000,
815994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    };
825994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
835994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    enum {
845994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber        kWhatFetchMore  = 'fetc',
855994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber        kWhatRead       = 'read',
865994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    };
875994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
880683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber    enum {
890683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber        kMaxNumRetries = 10,
900683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber    };
910683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber
925994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    sp<DataSource> mSource;
935994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    sp<AHandlerReflector<NuCachedSource2> > mReflector;
945994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    sp<ALooper> mLooper;
955994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
965994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    Mutex mSerializer;
97a53d87c7b1428fe02f535c31dafd64cb1362fde9Andreas Huber    mutable Mutex mLock;
985994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    Condition mCondition;
995994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
1005994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    PageCache *mCache;
101c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong    off64_t mCacheOffset;
1025994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    status_t mFinalStatus;
103c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong    off64_t mLastAccessPos;
1045994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    sp<AMessage> mAsyncResult;
1055994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    bool mFetching;
106a5273ebd1746368662a597643d6701a5046d5c7bAndreas Huber    int64_t mLastFetchTimeUs;
1075994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
1080683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber    int32_t mNumRetriesLeft;
1090683eba6b35c396c21f10e926709f2f8fc05f090Andreas Huber
110a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber    size_t mHighwaterThresholdBytes;
111a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber    size_t mLowwaterThresholdBytes;
112a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber
113a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber    // If the keep-alive interval is 0, keep-alives are disabled.
114a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber    int64_t mKeepAliveIntervalUs;
115a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber
11649c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber    bool mDisconnectAtHighwatermark;
11749c59815369616b0fd5451ccabd377e8fe1dc3faAndreas Huber
1185994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    void onMessageReceived(const sp<AMessage> &msg);
1195994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    void onFetch();
1205994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    void onRead(const sp<AMessage> &msg);
1215994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
1225994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    void fetchInternal();
123c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong    ssize_t readInternal(off64_t offset, void *data, size_t size);
124c7fc37a3dab9bd1f96713649f351b5990e6316ffJames Dong    status_t seekInternal_l(off64_t offset);
1255994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
126a53d87c7b1428fe02f535c31dafd64cb1362fde9Andreas Huber    size_t approxDataRemaining_l(status_t *finalStatus) const;
1277bf8413f91fc072452f315a91618aeef2574d420Andreas Huber
1287bf8413f91fc072452f315a91618aeef2574d420Andreas Huber    void restartPrefetcherIfNecessary_l(
1297bf8413f91fc072452f315a91618aeef2574d420Andreas Huber            bool ignoreLowWaterThreshold = false, bool force = false);
1305994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
131a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber    void updateCacheParamsFromSystemProperty();
132a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber    void updateCacheParamsFromString(const char *s);
133a045cb0e77097120e86e367e1cab5494ce2a5d5eAndreas Huber
1345994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber    DISALLOW_EVIL_CONSTRUCTORS(NuCachedSource2);
1355994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber};
1365994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
1375994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber}  // namespace android
1385994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber
1395994b4798b01f3dd340577c9ea9657f09093a770Andreas Huber#endif  // NU_CACHED_SOURCE_2_H_
140